Sweden-Number/dlls/schedsvc/atsvc.c

715 lines
20 KiB
C

/*
* ATSvc RPC API
*
* Copyright 2018 Dmitry Timoshkov
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdarg.h>
#include "windef.h"
#include "atsvc.h"
#include "mstask.h"
#include "wine/list.h"
#include "wine/debug.h"
#include "schedsvc_private.h"
WINE_DEFAULT_DEBUG_CHANNEL(schedsvc);
/* lmat.h defines those, but other types in that file conflict
* with generated atsvc.h typedefs.
*/
#define JOB_ADD_CURRENT_DATE 0x08
#define JOB_NONINTERACTIVE 0x10
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;
UINT status;
UINT flags;
SYSTEMTIME last_runtime;
} FIXDLEN_DATA;
struct job_t
{
struct list entry;
WCHAR *name;
AT_ENUM info;
};
static LONG current_jobid = 1;
static struct list at_job_list = LIST_INIT(at_job_list);
static CRITICAL_SECTION at_job_list_section;
static CRITICAL_SECTION_DEBUG cs_debug =
{
0, 0, &at_job_list_section,
{ &cs_debug.ProcessLocksList, &cs_debug.ProcessLocksList },
0, 0, { (DWORD_PTR)(__FILE__ ": at_job_list_section") }
};
static CRITICAL_SECTION at_job_list_section = { &cs_debug, -1, 0, 0, 0, 0 };
static DWORD load_unicode_strings(const char *data, DWORD limit, AT_ENUM *info)
{
DWORD i, data_size = 0;
USHORT len;
for (i = 0; i < 5; i++)
{
if (limit < sizeof(USHORT))
{
TRACE("invalid string %u offset\n", i);
break;
}
len = *(USHORT *)data;
data += sizeof(USHORT);
data_size += sizeof(USHORT);
limit -= sizeof(USHORT);
if (limit < len * sizeof(WCHAR))
{
TRACE("invalid string %u size\n", i);
break;
}
TRACE("string %u: %s\n", i, wine_dbgstr_wn((const WCHAR *)data, len));
if (i == 0)
info->Command = heap_strdupW((const WCHAR *)data);
data += len * sizeof(WCHAR);
data_size += len * sizeof(WCHAR);
}
return data_size;
}
/* FIXME: read more data, currently only Command is handled */
static BOOL load_job_data(const char *data, DWORD size, AT_ENUM *info)
{
const FIXDLEN_DATA *fixed;
const SYSTEMTIME *st;
DWORD unicode_strings_size, data_size, triggers_size;
USHORT instance_count, triggers_count, i;
const USHORT *signature;
const TASK_TRIGGER *trigger;
memset(info, 0, sizeof(*info));
if (size < sizeof(*fixed))
{
TRACE("no space for FIXDLEN_DATA\n");
return FALSE;
}
fixed = (const FIXDLEN_DATA *)data;
TRACE("product_version %04x\n", fixed->product_version);
TRACE("file_version %04x\n", fixed->file_version);
TRACE("uuid %s\n", wine_dbgstr_guid(&fixed->uuid));
TRACE("name_size_offset %04x\n", fixed->name_size_offset);
TRACE("trigger_offset %04x\n", fixed->trigger_offset);
TRACE("error_retry_count %u\n", fixed->error_retry_count);
TRACE("error_retry_interval %u\n", fixed->error_retry_interval);
TRACE("idle_deadline %u\n", fixed->idle_deadline);
TRACE("idle_wait %u\n", fixed->idle_wait);
TRACE("priority %08x\n", fixed->priority);
TRACE("maximum_runtime %u\n", fixed->maximum_runtime);
TRACE("exit_code %#x\n", fixed->exit_code);
TRACE("status %08x\n", fixed->status);
TRACE("flags %08x\n", fixed->flags);
st = &fixed->last_runtime;
TRACE("last_runtime %d/%d/%d wday %d %d:%d:%d.%03d\n",
st->wDay, st->wMonth, st->wYear, st->wDayOfWeek,
st->wHour, st->wMinute, st->wSecond, st->wMilliseconds);
/* Instance Count */
if (size < sizeof(*fixed) + sizeof(USHORT))
{
TRACE("no space for instance count\n");
return FALSE;
}
instance_count = *(const USHORT *)(data + sizeof(*fixed));
TRACE("instance count %u\n", instance_count);
if (fixed->name_size_offset + sizeof(USHORT) < size)
unicode_strings_size = load_unicode_strings(data + fixed->name_size_offset, size - fixed->name_size_offset, info);
else
{
TRACE("invalid name_size_offset\n");
return FALSE;
}
TRACE("unicode strings end at %#x\n", fixed->name_size_offset + unicode_strings_size);
if (size < fixed->trigger_offset + sizeof(USHORT))
{
TRACE("no space for triggers count\n");
return FALSE;
}
triggers_count = *(const USHORT *)(data + fixed->trigger_offset);
TRACE("triggers_count %u\n", triggers_count);
triggers_size = size - fixed->trigger_offset - sizeof(USHORT);
TRACE("triggers_size %u\n", triggers_size);
trigger = (const TASK_TRIGGER *)(data + fixed->trigger_offset + sizeof(USHORT));
data += fixed->name_size_offset + unicode_strings_size;
size -= fixed->name_size_offset + unicode_strings_size;
/* User Data */
if (size < sizeof(USHORT))
{
TRACE("no space for user data size\n");
return FALSE;
}
data_size = *(const USHORT *)data;
if (size < sizeof(USHORT) + data_size)
{
TRACE("no space for user data\n");
return FALSE;
}
TRACE("User Data size %#x\n", data_size);
size -= sizeof(USHORT) + data_size;
data += sizeof(USHORT) + data_size;
/* Reserved Data */
if (size < sizeof(USHORT))
{
TRACE("no space for reserved data size\n");
return FALSE;
}
data_size = *(const USHORT *)data;
if (size < sizeof(USHORT) + data_size)
{
TRACE("no space for reserved data\n");
return FALSE;
}
TRACE("Reserved Data size %#x\n", data_size);
size -= sizeof(USHORT) + data_size;
data += sizeof(USHORT) + data_size;
/* Trigger Data */
TRACE("trigger_offset %04x, triggers end at %04x\n", fixed->trigger_offset,
(DWORD)(fixed->trigger_offset + sizeof(USHORT) + triggers_count * sizeof(TASK_TRIGGER)));
triggers_count = *(const USHORT *)data;
TRACE("triggers_count %u\n", triggers_count);
trigger = (const TASK_TRIGGER *)(data + sizeof(USHORT));
if (triggers_count * sizeof(TASK_TRIGGER) > triggers_size)
{
TRACE("no space for triggers data\n");
return FALSE;
}
for (i = 0; i < triggers_count; i++)
{
TRACE("%u: cbTriggerSize = %#x\n", i, trigger[i].cbTriggerSize);
if (trigger[i].cbTriggerSize != sizeof(TASK_TRIGGER))
TRACE("invalid cbTriggerSize\n");
TRACE("Reserved1 = %#x\n", trigger[i].Reserved1);
TRACE("wBeginYear = %u\n", trigger->wBeginYear);
TRACE("wBeginMonth = %u\n", trigger->wBeginMonth);
TRACE("wBeginDay = %u\n", trigger->wBeginDay);
TRACE("wEndYear = %u\n", trigger->wEndYear);
TRACE("wEndMonth = %u\n", trigger->wEndMonth);
TRACE("wEndDay = %u\n", trigger->wEndDay);
TRACE("wStartHour = %u\n", trigger->wStartHour);
TRACE("wStartMinute = %u\n", trigger->wStartMinute);
TRACE("MinutesDuration = %u\n", trigger->MinutesDuration);
TRACE("MinutesInterval = %u\n", trigger->MinutesInterval);
TRACE("rgFlags = %u\n", trigger->rgFlags);
TRACE("TriggerType = %u\n", trigger->TriggerType);
TRACE("Reserved2 = %u\n", trigger->Reserved2);
TRACE("wRandomMinutesInterval = %u\n", trigger->wRandomMinutesInterval);
}
size -= sizeof(USHORT) + triggers_count * sizeof(TASK_TRIGGER);
data += sizeof(USHORT) + triggers_count * sizeof(TASK_TRIGGER);
if (size < 2 * sizeof(USHORT) + 64)
{
TRACE("no space for signature\n");
return TRUE; /* signature is optional */
}
signature = (const USHORT *)data;
TRACE("signature version %04x, client version %04x\n", signature[0], signature[1]);
return TRUE;
}
void add_job(const WCHAR *name)
{
HANDLE file, mapping;
DWORD size, try;
void *data;
struct job_t *job;
job = heap_alloc_zero(sizeof(*job));
if (!job) return;
try = 1;
for (;;)
{
file = CreateFileW(name, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, 0);
if (file == INVALID_HANDLE_VALUE)
{
TRACE("Failed to open %s, error %u\n", debugstr_w(name), GetLastError());
if (try++ >= 3) break;
Sleep(100);
continue;
}
size = GetFileSize(file, NULL);
mapping = CreateFileMappingW(file, NULL, PAGE_READONLY, 0, 0, 0);
if (!mapping)
{
TRACE("Failed to create file mapping %s, error %u\n", debugstr_w(name), GetLastError());
CloseHandle(file);
break;
}
data = MapViewOfFile(mapping, FILE_MAP_READ, 0, 0, 0);
if (data)
{
if (load_job_data(data, size, &job->info))
{
EnterCriticalSection(&at_job_list_section);
job->name = heap_strdupW(name);
job->info.JobId = current_jobid++;
list_add_tail(&at_job_list, &job->entry);
LeaveCriticalSection(&at_job_list_section);
}
UnmapViewOfFile(data);
}
CloseHandle(mapping);
CloseHandle(file);
break;
}
if (!job->info.JobId)
{
heap_free(job->info.Command);
heap_free(job);
}
}
static BOOL write_signature(HANDLE hfile)
{
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_trigger(HANDLE hfile, const AT_INFO *info)
{
USHORT count;
DWORD size;
SYSTEMTIME st;
TASK_TRIGGER trigger;
count = 1;
if (!WriteFile(hfile, &count, sizeof(count), &size, NULL))
return FALSE;
GetSystemTime(&st);
if (!(info->Flags & JOB_ADD_CURRENT_DATE))
{
/* FIXME: parse AT_INFO */
}
trigger.cbTriggerSize = sizeof(trigger);
trigger.Reserved1 = 0;
trigger.wBeginYear = st.wYear;
trigger.wBeginMonth = st.wMonth;
trigger.wBeginDay = st.wDay;
trigger.wEndYear = st.wYear;
trigger.wEndMonth = st.wMonth;
trigger.wEndDay = st.wDay;
trigger.wStartHour = st.wHour;
trigger.wStartMinute = st.wMinute;
trigger.MinutesDuration = 0;
trigger.MinutesInterval = 0;
/* FIXME */
trigger.rgFlags = TASK_TRIGGER_FLAG_HAS_END_DATE;
trigger.TriggerType = TASK_TIME_TRIGGER_MONTHLYDATE;
trigger.Type.MonthlyDate.rgfDays = 0;
trigger.Type.MonthlyDate.rgfMonths = 0xffff;
trigger.Reserved2 = 0;
trigger.wRandomMinutesInterval = 0;
return WriteFile(hfile, &trigger, sizeof(trigger), &size, NULL);
}
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 BOOL create_job(const WCHAR *job_name, const AT_INFO *info)
{
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;
USHORT word;
HANDLE hfile;
DWORD size, ver;
BOOL ret = FALSE;
TRACE("trying to create job %s\n", debugstr_w(job_name));
hfile = CreateFileW(job_name, GENERIC_WRITE, 0, NULL, CREATE_NEW, 0, 0);
if (hfile == INVALID_HANDLE_VALUE)
return FALSE;
ver = GetVersion();
fixed.product_version = MAKEWORD(ver >> 8, ver);
fixed.file_version = 0x0001;
UuidCreate(&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) + (lstrlenW(info->Command) + 1) * sizeof(WCHAR); /* Application Name */
fixed.trigger_offset += sizeof(USHORT); /* Parameters */
fixed.trigger_offset += sizeof(USHORT); /* Working Directory */
fixed.trigger_offset += sizeof(USHORT) + (lstrlenW(authorW) + 1) * sizeof(WCHAR); /* Author */
fixed.trigger_offset += sizeof(USHORT) + (lstrlenW(commentW) + 1) * sizeof(WCHAR); /* Comment */
fixed.trigger_offset += sizeof(USHORT); /* User Data */
fixed.trigger_offset += 10; /* Reserved Data */
fixed.error_retry_count = 0;
fixed.error_retry_interval = 0;
fixed.idle_deadline = 60;
fixed.idle_wait = 10;
fixed.priority = NORMAL_PRIORITY_CLASS;
fixed.maximum_runtime = 259200000;
fixed.exit_code = 0;
fixed.status = SCHED_S_TASK_HAS_NOT_RUN;
fixed.flags = TASK_FLAG_DELETE_WHEN_DONE;
if (!(info->Flags & JOB_NONINTERACTIVE))
fixed.flags |= TASK_FLAG_INTERACTIVE;
/* FIXME: add other flags */
memset(&fixed.last_runtime, 0, sizeof(fixed.last_runtime));
if (!WriteFile(hfile, &fixed, sizeof(fixed), &size, NULL))
goto failed;
/* Instance Count */
word = 0;
if (!WriteFile(hfile, &word, sizeof(word), &size, NULL))
goto failed;
/* Application Name */
if (!write_unicode_string(hfile, info->Command))
goto failed;
/* Parameters */
if (!write_unicode_string(hfile, NULL))
goto failed;
/* Working Directory */
if (!write_unicode_string(hfile, NULL))
goto failed;
/* Author */
if (!write_unicode_string(hfile, authorW))
goto failed;
/* Comment */
if (!write_unicode_string(hfile, commentW))
goto failed;
/* User Data */
word = 0;
if (!WriteFile(hfile, &word, sizeof(word), &size, NULL))
goto failed;
/* Reserved Data */
if (!write_reserved_data(hfile))
goto failed;
/* Trigegrs */
if (!write_trigger(hfile, info))
goto failed;
/* Signature */
if (!write_signature(hfile))
goto failed;
ret = TRUE;
failed:
CloseHandle(hfile);
if (!ret) DeleteFileW(job_name);
return ret;
}
static struct job_t *find_job(DWORD jobid, const WCHAR *name)
{
struct job_t *job;
LIST_FOR_EACH_ENTRY(job, &at_job_list, struct job_t, entry)
{
if ((name && !lstrcmpiW(job->name, name)) || job->info.JobId == jobid)
return job;
}
return NULL;
}
void remove_job(const WCHAR *name)
{
struct job_t *job;
EnterCriticalSection(&at_job_list_section);
job = find_job(0, name);
if (job)
{
list_remove(&job->entry);
heap_free(job->name);
heap_free(job);
}
LeaveCriticalSection(&at_job_list_section);
}
DWORD __cdecl NetrJobAdd(ATSVC_HANDLE server_name, AT_INFO *info, DWORD *jobid)
{
WCHAR windir[MAX_PATH];
TRACE("%s,%p,%p\n", debugstr_w(server_name), info, jobid);
GetWindowsDirectoryW(windir, MAX_PATH);
for (;;)
{
static const WCHAR fmtW[] = { '\\','T','a','s','k','s','\\','A','t','%','u','.','j','o','b',0 };
WCHAR task_name[MAX_PATH], name[32];
strcpyW(task_name, windir);
sprintfW(name, fmtW, current_jobid);
strcatW(task_name, name);
if (create_job(task_name, info))
{
struct job_t *job;
int i;
for (i = 0; i < 5; i++)
{
EnterCriticalSection(&at_job_list_section);
job = find_job(0, task_name);
LeaveCriticalSection(&at_job_list_section);
if (job)
{
*jobid = job->info.JobId;
break;
}
Sleep(50);
}
if (!job)
{
ERR("couldn't find just created job %s\n", debugstr_w(task_name));
return ERROR_FILE_NOT_FOUND;
}
break;
}
if (GetLastError() != ERROR_FILE_EXISTS)
{
TRACE("create_job error %u\n", GetLastError());
return GetLastError();
}
InterlockedIncrement(&current_jobid);
}
return ERROR_SUCCESS;
}
DWORD __cdecl NetrJobDel(ATSVC_HANDLE server_name, DWORD min_jobid, DWORD max_jobid)
{
DWORD jobid, ret = ERROR_FILE_NOT_FOUND;
TRACE("%s,%u,%u\n", debugstr_w(server_name), min_jobid, max_jobid);
EnterCriticalSection(&at_job_list_section);
for (jobid = min_jobid; jobid <= max_jobid; jobid++)
{
struct job_t *job = find_job(jobid, NULL);
if (!job)
{
TRACE("job %u not found\n", jobid);
ret = ERROR_FILE_NOT_FOUND;
break;
}
TRACE("deleting job %s\n", debugstr_w(job->name));
if (!DeleteFileW(job->name))
{
ret = GetLastError();
break;
}
ret = ERROR_SUCCESS;
}
LeaveCriticalSection(&at_job_list_section);
return ret;
}
static void free_container(AT_ENUM_CONTAINER *container)
{
DWORD i;
for (i = 0; i < container->EntriesRead; i++)
heap_free(container->Buffer[i].Command);
heap_free(container->Buffer);
}
DWORD __cdecl NetrJobEnum(ATSVC_HANDLE server_name, AT_ENUM_CONTAINER *container,
DWORD max_length, DWORD *total, DWORD *resume)
{
DWORD allocated;
struct job_t *job;
TRACE("%s,%p,%u,%p,%p\n", debugstr_w(server_name), container, max_length, total, resume);
*total = 0;
*resume = 0;
container->EntriesRead = 0;
allocated = 64;
container->Buffer = heap_alloc(allocated * sizeof(AT_ENUM));
if (!container->Buffer) return ERROR_NOT_ENOUGH_MEMORY;
EnterCriticalSection(&at_job_list_section);
LIST_FOR_EACH_ENTRY(job, &at_job_list, struct job_t, entry)
{
if (container->EntriesRead >= max_length)
{
*resume = container->EntriesRead;
break;
}
if (allocated <= container->EntriesRead)
{
AT_ENUM *new_buffer;
allocated *= 2;
new_buffer = heap_realloc(container->Buffer, allocated * sizeof(AT_ENUM));
if (!new_buffer)
{
free_container(container);
LeaveCriticalSection(&at_job_list_section);
return ERROR_NOT_ENOUGH_MEMORY;
}
container->Buffer = new_buffer;
}
container->Buffer[container->EntriesRead] = job->info;
container->Buffer[container->EntriesRead].Command = heap_strdupW(job->info.Command);
container->EntriesRead++;
}
LeaveCriticalSection(&at_job_list_section);
*total = container->EntriesRead;
return ERROR_SUCCESS;
}
DWORD __cdecl NetrJobGetInfo(ATSVC_HANDLE server_name, DWORD jobid, AT_INFO **info)
{
struct job_t *job;
DWORD ret = ERROR_SUCCESS;
TRACE("%s,%u,%p\n", debugstr_w(server_name), jobid, info);
EnterCriticalSection(&at_job_list_section);
job = find_job(jobid, NULL);
if (job)
{
AT_INFO *info_ret = heap_alloc(sizeof(*info_ret));
if (!info_ret)
ret = ERROR_NOT_ENOUGH_MEMORY;
else
{
info_ret->JobTime = job->info.JobTime;
info_ret->DaysOfMonth = job->info.DaysOfMonth;
info_ret->DaysOfWeek = job->info.DaysOfWeek;
info_ret->Flags = job->info.Flags;
info_ret->Command = heap_strdupW(job->info.Command);
*info = info_ret;
}
}
LeaveCriticalSection(&at_job_list_section);
return ret;
}