Dmitry Timoshkov : wbemprox: Flesh out a bit SystemRestore::Enable() implementation.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: d2baf148a14b41d0347dfdc2bc6cf9efc4f63242
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=d2baf148a14b41d0347dfdc2b...

Author: Dmitry Timoshkov <dmitry(a)baikal.ru>
Date:   Wed Jan 20 11:15:58 2021 +0100

wbemprox: Flesh out a bit SystemRestore::Enable() implementation.

Signed-off-by: Dmitry Timoshkov <dmitry(a)baikal.ru>
Signed-off-by: Hans Leidekker <hans(a)codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>

---

 dlls/wbemprox/sysrestore.c  | 48 +++++++++++++++++++++++++++++++++++++++++++--
 dlls/wbemprox/tests/query.c |  2 +-
 2 files changed, 47 insertions(+), 3 deletions(-)

diff --git a/dlls/wbemprox/sysrestore.c b/dlls/wbemprox/sysrestore.c
index b840d220b5c..d59077381fb 100644
--- a/dlls/wbemprox/sysrestore.c
+++ b/dlls/wbemprox/sysrestore.c
@@ -45,8 +45,52 @@ HRESULT disable_restore( IWbemClassObject *obj, IWbemClassObject *in, IWbemClass
 
 HRESULT enable_restore( IWbemClassObject *obj, IWbemClassObject *in, IWbemClassObject **out )
 {
-    FIXME("stub\n");
-    return S_OK;
+    VARIANT drive, retval;
+    IWbemClassObject *sig, *out_params = NULL;
+    HRESULT hr;
+
+    TRACE("%p, %p\n", in, out);
+
+    hr = IWbemClassObject_Get( in, L"Drive", 0, &drive, NULL, NULL );
+    if (hr != S_OK) return hr;
+
+    hr = create_signature( L"SystemRestore", L"Enable", PARAM_OUT, &sig );
+    if (hr != S_OK)
+    {
+        VariantClear( &drive );
+        return hr;
+    }
+
+    if (out)
+    {
+        hr = IWbemClassObject_SpawnInstance( sig, 0, &out_params );
+        if (hr != S_OK)
+        {
+            VariantClear( &drive );
+            IWbemClassObject_Release( sig );
+            return hr;
+        }
+    }
+
+    FIXME("%s: stub\n", wine_dbgstr_variant(&drive));
+
+    VariantClear( &drive );
+    IWbemClassObject_Release( sig );
+
+    if (out_params)
+    {
+        set_variant( VT_UI4, ERROR_SUCCESS, NULL, &retval );
+        hr = IWbemClassObject_Put( out_params, L"ReturnValue", 0, &retval, CIM_UINT32 );
+    }
+
+    if (hr == S_OK && out)
+    {
+        *out = out_params;
+        IWbemClassObject_AddRef( out_params );
+    }
+    if (out_params) IWbemClassObject_Release( out_params );
+
+    return hr;
 }
 
 HRESULT get_last_restore_status( IWbemClassObject *obj, IWbemClassObject *in, IWbemClassObject **out )
diff --git a/dlls/wbemprox/tests/query.c b/dlls/wbemprox/tests/query.c
index 1c7a615dd4e..67c0b218ce1 100644
--- a/dlls/wbemprox/tests/query.c
+++ b/dlls/wbemprox/tests/query.c
@@ -1805,7 +1805,7 @@ static void test_SystemRestore( IWbemServices *services )
     out = NULL;
     hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, in, &out, NULL );
     ok( hr == S_OK || hr == WBEM_E_ACCESS_DENIED, "failed to execute method %08x\n", hr );
-    if (hr == S_OK && out)
+    if (hr == S_OK)
     {
         VariantInit( &var );
         hr = IWbemClassObject_Get( out, L"ReturnValue", 0, &var, NULL, NULL );