mstask: Store comment using IRegistrationInfo.

Signed-off-by: Dmitry Timoshkov <dmitry@baikal.ru>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Dmitry Timoshkov 2018-04-02 13:27:07 +08:00 committed by Alexandre Julliard
parent f6d6365430
commit 6ff9f7e870
1 changed files with 29 additions and 14 deletions

View File

@ -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;
}
else
hr = E_OUTOFMEMORY;
hr = E_OUTOFMEMORY;
SysFreeString(id);
SysFreeString(description);
}
IRegistrationInfo_Release(info);
return hr;
}