[PATCH v3 3/3] taskschd: Implement ITaskDefinition Data property.

[Owen Rudge]

From: Owen Rudge

---
 dlls/taskschd/task.c            | 56 ++++++++++++++++++++++++++++++---
 dlls/taskschd/tests/scheduler.c | 18 +++++------
 2 files changed, 61 insertions(+), 13 deletions(-)

diff --git a/dlls/taskschd/task.c b/dlls/taskschd/task.c
index a3378a99117..0b0165bbf8b 100644
--- a/dlls/taskschd/task.c
+++ b/dlls/taskschd/task.c
@@ -2373,6 +2373,7 @@ typedef struct
     ITriggerCollection *triggers;
     IPrincipal *principal;
     IActionCollection *actions;
+    BSTR data;
 } TaskDefinition;

static inline TaskDefinition *impl_from_ITaskDefinition(ITaskDefinition *iface)
@@ -2405,6 +2406,8 @@ static ULONG WINAPI TaskDefinition_Release(ITaskDefinition *iface)
             IPrincipal_Release(taskdef->principal);
         if (taskdef->actions)
             IActionCollection_Release(taskdef->actions);
+        if (taskdef->data)
+            SysFreeString(taskdef->data);

free(taskdef);
     }
@@ -2576,14 +2579,41 @@ static HRESULT WINAPI TaskDefinition_put_Settings(ITaskDefinition *iface, ITaskS

static HRESULT WINAPI TaskDefinition_get_Data(ITaskDefinition *iface, BSTR *data)
 {
-    FIXME("%p,%p: stub\n", iface, data);
-    return E_NOTIMPL;
+    TaskDefinition *taskdef = impl_from_ITaskDefinition(iface);
+
+    TRACE("%p,%p\n", iface, data);
+
+    if (!data) return E_POINTER;
+
+    if (!taskdef->data)
+        *data = NULL;
+    else
+    {
+        *data = SysAllocString(taskdef->data);
+        if (!*data) return E_OUTOFMEMORY;
+    }
+
+    return S_OK;
 }

static HRESULT WINAPI TaskDefinition_put_Data(ITaskDefinition *iface, BSTR data)
 {
-    FIXME("%p,%p: stub\n", iface, data);
-    return E_NOTIMPL;
+    TaskDefinition *taskdef = impl_from_ITaskDefinition(iface);
+    BSTR copy = NULL;
+
+    TRACE("%p,%s\n", iface, debugstr_w(data));
+
+    if (data)
+    {
+        copy = SysAllocString(data);
+        if (!copy) return E_OUTOFMEMORY;
+    }
+
+    if (taskdef->data)
+        SysFreeString(taskdef->data);
+
+    taskdef->data = copy;
+    return S_OK;
 }

static HRESULT WINAPI TaskDefinition_get_Principal(ITaskDefinition *iface, IPrincipal **principal)
@@ -3127,6 +3157,12 @@ static HRESULT WINAPI TaskDefinition_get_XmlText(ITaskDefinition *iface, BSTR *x
     hr = write_actions(stream, taskdef->actions);
     if (hr != S_OK) goto failed;

+    if (taskdef->data)
+    {
+        hr = write_text_value(stream, L"Data", taskdef->data);
+        if (hr != S_OK) goto failed;
+    }
+
     pop_indent();

write_element_end(stream, L"Task");
@@ -3766,6 +3802,13 @@ static HRESULT read_task(IXmlReader *reader, ITaskDefinition *taskdef)
                 hr = read_principals(reader, taskdef);
             else if (!lstrcmpW(name, L"Actions"))
                 hr = read_actions(reader, taskdef);
+            else if (!lstrcmpW(name, L"Data"))
+            {
+                WCHAR *value;
+                hr = read_text_value(reader, &value);
+                if (hr == S_OK)
+                    ITaskDefinition_put_Data(taskdef, value);
+            }
             else
                 FIXME("unhandled Task element %s\n", debugstr_w(name));

@@ -3894,6 +3937,11 @@ static HRESULT WINAPI TaskDefinition_put_XmlText(ITaskDefinition *iface, BSTR xm
             IActionCollection_Release(taskdef->actions);
             taskdef->actions = NULL;
         }
+        if (taskdef->data)
+        {
+            SysFreeString(taskdef->data);
+            taskdef->data = NULL;
+        }

hr = read_xml(reader, iface);
     }
diff --git a/dlls/taskschd/tests/scheduler.c b/dlls/taskschd/tests/scheduler.c
index cc7a6afdd76..615f8da8084 100644
--- a/dlls/taskschd/tests/scheduler.c
+++ b/dlls/taskschd/tests/scheduler.c
@@ -1845,23 +1845,23 @@ static void test_TaskDefinition(void)
     IRegistrationInfo_Release(reginfo);

hr = ITaskDefinition_get_Data(taskdef, &bstr);
-    todo_wine ok(hr == S_OK, "get_Data error %#lx\n", hr);
-    if (!winetest_platform_is_wine) ok(!lstrcmpW(bstr, L"MyTestData"), "expected "MyTestData", got %s\n", wine_dbgstr_w(bstr));
-    if (!winetest_platform_is_wine) SysFreeString(bstr);
+    ok(hr == S_OK, "get_Data error %#lx\n", hr);
+    ok(!lstrcmpW(bstr, L"MyTestData"), "expected "MyTestData", got %s\n", wine_dbgstr_w(bstr));
+    SysFreeString(bstr);

hr = ITaskDefinition_put_Data(taskdef, (BSTR) L"NewTest");
-    todo_wine ok(hr == S_OK, "put_Data error %#lx\n", hr);
+    ok(hr == S_OK, "put_Data error %#lx\n", hr);
     bstr = (BSTR)0xdeadbeef;
     hr = ITaskDefinition_get_Data(taskdef, &bstr);
-    if (!winetest_platform_is_wine) ok(!lstrcmpW(bstr, L"NewTest"), "expected "NewTest", got %s\n", wine_dbgstr_w(bstr));
-    if (!winetest_platform_is_wine) SysFreeString(bstr);
+    ok(!lstrcmpW(bstr, L"NewTest"), "expected "NewTest", got %s\n", wine_dbgstr_w(bstr));
+    SysFreeString(bstr);

hr = ITaskDefinition_put_Data(taskdef, NULL);
-    todo_wine ok(hr == S_OK, "put_Data error %#lx\n", hr);
+    ok(hr == S_OK, "put_Data error %#lx\n", hr);
     bstr = (BSTR)0xdeadbeef;
     hr = ITaskDefinition_get_Data(taskdef, &bstr);
-    todo_wine ok(hr == S_OK, "get_Data error %#lx\n", hr);
-    todo_wine ok(!bstr, "expected NULL, got %s\n", wine_dbgstr_w(bstr));
+    ok(hr == S_OK, "get_Data error %#lx\n", hr);
+    ok(!bstr, "expected NULL, got %s\n", wine_dbgstr_w(bstr));

hr = ITaskDefinition_put_XmlText(taskdef, xml4);
     ok(hr == S_OK, "put_XmlText error %#lx\n", hr);