mstask: Implement IPersistFile::Save().

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-20 16:37:31 +08:00 committed by Alexandre Julliard
parent be8a21f136
commit e4b7334574
1 changed files with 279 additions and 8 deletions

View File

@ -31,6 +31,25 @@
WINE_DEFAULT_DEBUG_CHANNEL(mstask);
typedef struct
{
USHORT product_version;
USHORT file_version;
UUID uuid;
USHORT name_size_offset;
USHORT trigger_offset;
USHORT error_retry_count;
USHORT error_retry_interval;
USHORT idle_deadline;
USHORT idle_wait;
UINT priority;
UINT maximum_runtime;
UINT exit_code;
HRESULT status;
UINT flags;
SYSTEMTIME last_runtime;
} FIXDLEN_DATA;
typedef struct
{
ITask ITask_iface;
@ -731,17 +750,270 @@ static HRESULT WINAPI MSTASK_IPersistFile_Load(
return E_NOTIMPL;
}
static HRESULT WINAPI MSTASK_IPersistFile_Save(
IPersistFile* iface,
LPCOLESTR pszFileName,
BOOL fRemember)
static BOOL write_signature(HANDLE hfile)
{
FIXME("(%p, %p, %d): stub\n", iface, pszFileName, fRemember);
WARN("Returning S_OK but not writing to disk: %s %d\n",
debugstr_w(pszFileName), fRemember);
struct
{
USHORT SignatureVersion;
USHORT ClientVersion;
BYTE md5[64];
} signature;
DWORD size;
signature.SignatureVersion = 0x0001;
signature.ClientVersion = 0x0001;
memset(&signature.md5, 0, sizeof(signature.md5));
return WriteFile(hfile, &signature, sizeof(signature), &size, NULL);
}
static BOOL write_reserved_data(HANDLE hfile)
{
static const struct
{
USHORT size;
BYTE data[8];
} user = { 8, { 0xff,0x0f,0x1d,0,0,0,0,0 } };
DWORD size;
return WriteFile(hfile, &user, sizeof(user), &size, NULL);
}
static BOOL write_user_data(HANDLE hfile, BYTE *data, WORD data_size)
{
DWORD size;
if (!WriteFile(hfile, &data_size, sizeof(data_size), &size, NULL))
return FALSE;
if (!data_size) return TRUE;
return WriteFile(hfile, data, data_size, &size, NULL);
}
static HRESULT write_triggers(ITask *task, HANDLE hfile)
{
WORD count, i;
DWORD size;
HRESULT hr;
hr = ITask_GetTriggerCount(task, &count);
if (hr != S_OK) return hr;
if (!WriteFile(hfile, &count, sizeof(count), &size, NULL))
return HRESULT_FROM_WIN32(GetLastError());
for (i = 0; i < count; i++)
{
ITaskTrigger *trigger;
TASK_TRIGGER task_trigger;
hr = ITask_GetTrigger(task, i, &trigger);
if (hr != S_OK) return hr;
hr = ITaskTrigger_GetTrigger(trigger, &task_trigger);
ITaskTrigger_Release(trigger);
if (hr != S_OK) return hr;
if (!WriteFile(hfile, &task_trigger, sizeof(task_trigger), &size, NULL))
return HRESULT_FROM_WIN32(GetLastError());
}
return S_OK;
}
static BOOL write_unicode_string(HANDLE hfile, const WCHAR *str)
{
USHORT count;
DWORD size;
count = str ? (lstrlenW(str) + 1) : 0;
if (!WriteFile(hfile, &count, sizeof(count), &size, NULL))
return FALSE;
if (!str) return TRUE;
count *= sizeof(WCHAR);
return WriteFile(hfile, str, count, &size, NULL);
}
static HRESULT WINAPI MSTASK_IPersistFile_Save(IPersistFile *iface, LPCOLESTR name, BOOL remember)
{
static const WCHAR tasksW[] = { '\\','T','a','s','k','s','\\',0 };
static const WCHAR jobW[] = { '.','j','o','b',0 };
static WCHAR authorW[] = { 'W','i','n','e',0 };
static WCHAR commentW[] = { 'C','r','e','a','t','e','d',' ','b','y',' ','W','i','n','e',0 };
FIXDLEN_DATA fixed;
WORD word, user_data_size = 0;
HANDLE hfile;
DWORD size, ver, disposition;
TaskImpl *This = impl_from_IPersistFile(iface);
ITask *task = &This->ITask_iface;
LPWSTR appname = NULL, params = NULL, workdir = NULL, creator = NULL, comment = NULL;
BYTE *user_data = NULL;
WCHAR task_name[MAX_PATH];
HRESULT hr;
TRACE("(%p, %s, %d)\n", iface, debugstr_w(name), remember);
disposition = name ? CREATE_NEW : OPEN_ALWAYS;
if (!name)
{
name = This->task_name;
remember = FALSE;
}
else if (strchrW(name, '.'))
return E_INVALIDARG;
GetWindowsDirectoryW(task_name, MAX_PATH);
lstrcatW(task_name, tasksW);
lstrcatW(task_name, name);
lstrcatW(task_name, jobW);
ITask_GetComment(task, &comment);
if (!comment) comment = commentW;
ITask_GetCreator(task, &creator);
if (!creator) creator = authorW;
ITask_GetApplicationName(task, &appname);
ITask_GetParameters(task, &params);
ITask_GetWorkingDirectory(task, &workdir);
ITask_GetWorkItemData(task, &user_data_size, &user_data);
ver = GetVersion();
fixed.product_version = MAKEWORD(ver >> 8, ver);
fixed.file_version = 0x0001;
CoCreateGuid(&fixed.uuid);
fixed.name_size_offset = sizeof(fixed) + sizeof(USHORT); /* FIXDLEN_DATA + Instance Count */
fixed.trigger_offset = sizeof(fixed) + sizeof(USHORT); /* FIXDLEN_DATA + Instance Count */
fixed.trigger_offset += sizeof(USHORT); /* Application Name */
if (appname)
fixed.trigger_offset += (lstrlenW(appname) + 1) * sizeof(WCHAR);
fixed.trigger_offset += sizeof(USHORT); /* Parameters */
if (params)
fixed.trigger_offset += (lstrlenW(params) + 1) * sizeof(WCHAR);
fixed.trigger_offset += sizeof(USHORT); /* Working Directory */
if (workdir)
fixed.trigger_offset += (lstrlenW(workdir) + 1) * sizeof(WCHAR);
fixed.trigger_offset += sizeof(USHORT); /* Author */
if (creator)
fixed.trigger_offset += (lstrlenW(creator) + 1) * sizeof(WCHAR);
fixed.trigger_offset += sizeof(USHORT); /* Comment */
if (comment)
fixed.trigger_offset += (lstrlenW(comment) + 1) * sizeof(WCHAR);
fixed.trigger_offset += sizeof(USHORT) + user_data_size; /* User Data */
fixed.trigger_offset += 10; /* Reserved Data */
fixed.error_retry_count = 0;
fixed.error_retry_interval = 0;
fixed.idle_wait = This->idle_minutes;
fixed.idle_deadline = This->deadline_minutes;
fixed.priority = This->priority;
fixed.maximum_runtime = This->maxRunTime;
fixed.exit_code = 0;
fixed.status = SCHED_S_TASK_HAS_NOT_RUN;
fixed.flags = 0;
memset(&fixed.last_runtime, 0, sizeof(fixed.last_runtime));
hfile = CreateFileW(task_name, GENERIC_WRITE, 0, NULL, disposition, 0, 0);
if (hfile == INVALID_HANDLE_VALUE)
return HRESULT_FROM_WIN32(GetLastError());
if (!WriteFile(hfile, &fixed, sizeof(fixed), &size, NULL))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Instance Count */
word = 0;
if (!WriteFile(hfile, &word, sizeof(word), &size, NULL))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Application Name */
if (!write_unicode_string(hfile, appname))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Parameters */
if (!write_unicode_string(hfile, params))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Working Directory */
if (!write_unicode_string(hfile, workdir))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Author */
if (!write_unicode_string(hfile, creator))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Comment */
if (!write_unicode_string(hfile, comment))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* User Data */
if (!write_user_data(hfile, user_data, user_data_size))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Reserved Data */
if (!write_reserved_data(hfile))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Trigegrs */
if (!write_triggers(hfile, task))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
/* Signature */
if (!write_signature(hfile))
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto failed;
}
hr = S_OK;
failed:
CoTaskMemFree(appname);
CoTaskMemFree(params);
CoTaskMemFree(workdir);
if (creator != authorW)
CoTaskMemFree(creator);
if (comment != commentW)
CoTaskMemFree(comment);
CoTaskMemFree(user_data);
CloseHandle(hfile);
if (hr != S_OK)
DeleteFileW(name);
else if (remember)
{
HeapFree(GetProcessHeap(), 0, This->task_name);
This->task_name = heap_strdupW(name);
}
return hr;
}
static HRESULT WINAPI MSTASK_IPersistFile_SaveCompleted(
IPersistFile* iface,
LPCOLESTR pszFileName)
@ -758,7 +1030,6 @@ static HRESULT WINAPI MSTASK_IPersistFile_GetCurFile(
return E_NOTIMPL;
}
static const ITaskVtbl MSTASK_ITaskVtbl =
{
MSTASK_ITask_QueryInterface,