[PATCH 8/9] mstask: Store comment using IRegistrationInfo.

[Dmitry Timoshkov]

Signed-off-by: Dmitry Timoshkov 
---
 dlls/mstask/task.c | 41 ++++++++++++++++++++++++++++-------------
 1 file changed, 28 insertions(+), 13 deletions(-)

diff --git a/dlls/mstask/task.c b/dlls/mstask/task.c
index 5a4023694c..d04666d76d 100644
--- a/dlls/mstask/task.c
+++ b/dlls/mstask/task.c
@@ -248,41 +248,56 @@ static HRESULT WINAPI MSTASK_ITask_GetExitCode(
 static HRESULT WINAPI MSTASK_ITask_SetComment(ITask *iface, LPCWSTR comment)
 {
     TaskImpl *This = impl_from_ITask(iface);
+    IRegistrationInfo *info;
+    HRESULT hr;

TRACE("(%p, %s)\n", iface, debugstr_w(comment));

if (!comment || !comment[0])
         comment = NULL;

-    return IExecAction_put_Id(This->action, (BSTR)comment);
+    hr = ITaskDefinition_get_RegistrationInfo(This->task, &info);
+    if (hr == S_OK)
+    {
+        hr = IRegistrationInfo_put_Description(info, (BSTR)comment);
+        IRegistrationInfo_Release(info);
+    }
+    return hr;
 }

static HRESULT WINAPI MSTASK_ITask_GetComment(ITask *iface, LPWSTR *comment)
 {
     TaskImpl *This = impl_from_ITask(iface);
+    IRegistrationInfo *info;
     HRESULT hr;
-    BSTR id;
+    BSTR description;
     DWORD len;

TRACE("(%p, %p)\n", iface, comment);

-    hr = IExecAction_get_Id(This->action, &id);
+    hr = ITaskDefinition_get_RegistrationInfo(This->task, &info);
     if (hr != S_OK) return hr;

-    len = id ? lstrlenW(id) + 1 : 1;
-    *comment = CoTaskMemAlloc(len * sizeof(WCHAR));
-    if (*comment)
+    hr = IRegistrationInfo_get_Description(info, &description);
+    if (hr == S_OK)
     {
-        if (!id)
-            *comment[0] = 0;
+        len = description ? lstrlenW(description) + 1 : 1;
+        *comment = CoTaskMemAlloc(len * sizeof(WCHAR));
+        if (*comment)
+        {
+            if (!description)
+                *comment[0] = 0;
+            else
+                lstrcpyW(*comment, description);
+            hr = S_OK;
+        }
         else
-            lstrcpyW(*comment, id);
-        hr = S_OK;
+            hr =  E_OUTOFMEMORY;
+
+        SysFreeString(description);
     }
-    else
-        hr =  E_OUTOFMEMORY;

-    SysFreeString(id);
+    IRegistrationInfo_Release(info);
     return hr;
 }