/* * Task Scheduler Service * * Copyright 2014 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 #include "windef.h" #include "winbase.h" #include "winsvc.h" #include "rpc.h" #include "atsvc.h" #include "schrpc.h" #include "wine/debug.h" #include "schedsvc_private.h" WINE_DEFAULT_DEBUG_CHANNEL(schedsvc); static const WCHAR scheduleW[] = {'S','c','h','e','d','u','l','e',0}; static SERVICE_STATUS_HANDLE schedsvc_handle; static HANDLE done_event, hjob_queue; void add_process_to_queue(HANDLE process) { if (!AssignProcessToJobObject(hjob_queue, process)) ERR("AssignProcessToJobObject failed\n"); } static DWORD WINAPI tasks_monitor_thread(void *arg) { static const WCHAR tasksW[] = { '\\','T','a','s','k','s','\\',0 }; WCHAR path[MAX_PATH]; HANDLE htasks, hport; JOBOBJECT_ASSOCIATE_COMPLETION_PORT info; OVERLAPPED ov; TRACE("Starting...\n"); GetWindowsDirectoryW(path, MAX_PATH); lstrcatW(path, tasksW); htasks = CreateFileW(path, FILE_LIST_DIRECTORY, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_OVERLAPPED, NULL); if (htasks == INVALID_HANDLE_VALUE) { ERR("Couldn't start monitoring %s for tasks, error %u\n", debugstr_w(path), GetLastError()); /* Probably this is an old prefix with disabled updates */ if (GetLastError() == ERROR_PATH_NOT_FOUND || GetLastError() == ERROR_FILE_NOT_FOUND) ERR("Please create the directory manually\n"); return -1; } hjob_queue = CreateJobObjectW(NULL, NULL); if (!hjob_queue) { ERR("CreateJobObject failed\n"); return -1; } hport = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 1); if (!hport) { ERR("CreateIoCompletionPort failed\n"); return -1; } info.CompletionKey = hjob_queue; info.CompletionPort = hport; if (!SetInformationJobObject(hjob_queue, JobObjectAssociateCompletionPortInformation, &info, sizeof(info))) { ERR("SetInformationJobObject failed\n"); return -1; } memset(&ov, 0, sizeof(ov)); ov.hEvent = CreateEventW(NULL, FALSE, FALSE, NULL); for (;;) { struct { FILE_NOTIFY_INFORMATION data; WCHAR name_buffer[MAX_PATH]; } info; HANDLE events[3]; DWORD ret; /* the buffer must be DWORD aligned */ C_ASSERT(!(sizeof(info) & 3)); memset(&info, 0, sizeof(info)); ret = ReadDirectoryChangesW(htasks, &info, sizeof(info), FALSE, FILE_NOTIFY_CHANGE_FILE_NAME | FILE_NOTIFY_CHANGE_SIZE | FILE_NOTIFY_CHANGE_LAST_WRITE, NULL, &ov, NULL); if (!ret) break; if (info.data.NextEntryOffset) FIXME("got multiple entries\n"); events[0] = done_event; events[1] = ov.hEvent; events[2] = hport; ret = WaitForMultipleObjects(3, events, FALSE, INFINITE); if (ret == WAIT_OBJECT_0) break; if (ret == WAIT_OBJECT_0 + 2) { DWORD msg; ULONG_PTR dummy, pid; if (GetQueuedCompletionStatus(hport, &msg, &dummy, (OVERLAPPED **)&pid, 0)) { if (msg == JOB_OBJECT_MSG_EXIT_PROCESS) { TRACE("got message: process %#lx has terminated\n", pid); update_process_status(pid); } else FIXME("got message %#x from the job\n", msg); } continue; } info.data.FileName[info.data.FileNameLength/sizeof(WCHAR)] = 0; switch (info.data.Action) { case FILE_ACTION_ADDED: TRACE("FILE_ACTION_ADDED %s\n", debugstr_w(info.data.FileName)); GetWindowsDirectoryW(path, MAX_PATH); lstrcatW(path, tasksW); lstrcatW(path, info.data.FileName); add_job(path); break; case FILE_ACTION_REMOVED: TRACE("FILE_ACTION_REMOVED %s\n", debugstr_w(info.data.FileName)); GetWindowsDirectoryW(path, MAX_PATH); lstrcatW(path, tasksW); lstrcatW(path, info.data.FileName); remove_job(path); break; case FILE_ACTION_MODIFIED: TRACE("FILE_ACTION_MODIFIED %s\n", debugstr_w(info.data.FileName)); GetWindowsDirectoryW(path, MAX_PATH); lstrcatW(path, tasksW); lstrcatW(path, info.data.FileName); remove_job(path); add_job(path); break; default: FIXME("%s: action %#x not handled\n", debugstr_w(info.data.FileName), info.data.Action); break; } check_task_state(); } CloseHandle(ov.hEvent); CloseHandle(hport); CloseHandle(hjob_queue); CloseHandle(htasks); TRACE("Finished.\n"); return 0; } void schedsvc_auto_start(void) { static DWORD start_type; SC_HANDLE scm, service; QUERY_SERVICE_CONFIGW *cfg; DWORD cfg_size; if (start_type == SERVICE_AUTO_START) return; TRACE("changing service start type to SERVICE_AUTO_START\n"); scm = OpenSCManagerW(NULL, NULL, 0); if (!scm) { WARN("failed to open SCM (%u)\n", GetLastError()); return; } service = OpenServiceW(scm, scheduleW, SERVICE_QUERY_CONFIG | SERVICE_CHANGE_CONFIG); if (service) { if (!QueryServiceConfigW(service, NULL, 0, &cfg_size) && GetLastError() == ERROR_INSUFFICIENT_BUFFER) { cfg = HeapAlloc(GetProcessHeap(), 0, cfg_size); if (cfg) { if (QueryServiceConfigW(service, cfg, cfg_size, &cfg_size)) { start_type = cfg->dwStartType; if (start_type != SERVICE_AUTO_START) { if (ChangeServiceConfigW(service, SERVICE_NO_CHANGE, SERVICE_AUTO_START, SERVICE_NO_CHANGE, NULL, NULL, NULL, NULL, NULL, NULL, NULL)) start_type = SERVICE_AUTO_START; } } HeapFree(GetProcessHeap(), 0, cfg); } } else WARN("failed to query service config (%u)\n", GetLastError()); CloseServiceHandle(service); } else WARN("failed to open service (%u)\n", GetLastError()); CloseServiceHandle(scm); } static void schedsvc_update_status(DWORD state) { SERVICE_STATUS status; status.dwServiceType = SERVICE_WIN32; status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN; status.dwWin32ExitCode = 0; status.dwServiceSpecificExitCode = 0; status.dwCheckPoint = 0; status.dwWaitHint = 0; status.dwControlsAccepted = 0; status.dwCurrentState = state; SetServiceStatus(schedsvc_handle, &status); } static void WINAPI schedsvc_handler(DWORD control) { TRACE("%#x\n", control); switch (control) { case SERVICE_CONTROL_STOP: case SERVICE_CONTROL_SHUTDOWN: schedsvc_update_status(SERVICE_STOP_PENDING); SetEvent(done_event); break; default: schedsvc_update_status(SERVICE_RUNNING); break; } } static RPC_BINDING_VECTOR *sched_bindings; static RPC_STATUS RPC_init(void) { static WCHAR ncacn_npW[] = { 'n','c','a','c','n','_','n','p',0 }; static WCHAR endpoint_npW[] = { '\\','p','i','p','e','\\','a','t','s','v','c',0 }; static WCHAR ncalrpcW[] = { 'n','c','a','l','r','p','c',0 }; static WCHAR endpoint_lrpcW[] = { 'a','t','s','v','c',0 }; RPC_STATUS status; status = RpcServerRegisterIf(ITaskSchedulerService_v1_0_s_ifspec, NULL, NULL); if (status != RPC_S_OK) { ERR("RpcServerRegisterIf error %#x\n", status); return status; } status = RpcServerRegisterIf(atsvc_v1_0_s_ifspec, NULL, NULL); if (status != RPC_S_OK) { ERR("RpcServerRegisterIf error %#x\n", status); RpcServerUnregisterIf(ITaskSchedulerService_v1_0_s_ifspec, NULL, FALSE); return status; } status = RpcServerUseProtseqEpW(ncacn_npW, RPC_C_PROTSEQ_MAX_REQS_DEFAULT, endpoint_npW, NULL); if (status != RPC_S_OK) { ERR("RpcServerUseProtseqEp error %#x\n", status); return status; } status = RpcServerUseProtseqEpW(ncalrpcW, RPC_C_PROTSEQ_MAX_REQS_DEFAULT, endpoint_lrpcW, NULL); if (status != RPC_S_OK) { ERR("RpcServerUseProtseqEp error %#x\n", status); return status; } status = RpcServerInqBindings(&sched_bindings); if (status != RPC_S_OK) { ERR("RpcServerInqBindings error %#x\n", status); return status; } status = RpcEpRegisterW(ITaskSchedulerService_v1_0_s_ifspec, sched_bindings, NULL, NULL); if (status != RPC_S_OK) { ERR("RpcEpRegister error %#x\n", status); return status; } status = RpcEpRegisterW(atsvc_v1_0_s_ifspec, sched_bindings, NULL, NULL); if (status != RPC_S_OK) { ERR("RpcEpRegister error %#x\n", status); return status; } status = RpcServerListen(1, RPC_C_LISTEN_MAX_CALLS_DEFAULT, TRUE); if (status != RPC_S_OK) { ERR("RpcServerListen error %#x\n", status); return status; } return RPC_S_OK; } static void RPC_finish(void) { RpcMgmtStopServerListening(NULL); RpcEpUnregister(ITaskSchedulerService_v1_0_s_ifspec, sched_bindings, NULL); RpcEpUnregister(atsvc_v1_0_s_ifspec, sched_bindings, NULL); RpcBindingVectorFree(&sched_bindings); RpcServerUnregisterIf(ITaskSchedulerService_v1_0_s_ifspec, NULL, FALSE); RpcServerUnregisterIf(atsvc_v1_0_s_ifspec, NULL, FALSE); } void WINAPI ServiceMain(DWORD argc, LPWSTR *argv) { HANDLE thread; DWORD tid; TRACE("starting Task Scheduler Service\n"); schedsvc_handle = RegisterServiceCtrlHandlerW(scheduleW, schedsvc_handler); if (!schedsvc_handle) { ERR("RegisterServiceCtrlHandler error %d\n", GetLastError()); return; } schedsvc_update_status(SERVICE_START_PENDING); done_event = CreateEventW(NULL, TRUE, FALSE, NULL); thread = CreateThread(NULL, 0, tasks_monitor_thread, NULL, 0, &tid); if (thread && RPC_init() == RPC_S_OK) { schedsvc_update_status(SERVICE_RUNNING); WaitForSingleObject(thread, INFINITE); CloseHandle(thread); RPC_finish(); } schedsvc_update_status(SERVICE_STOPPED); TRACE("exiting Task Scheduler Service\n"); } void __RPC_FAR * __RPC_USER MIDL_user_allocate(SIZE_T len) { return heap_alloc(len); } void __RPC_USER MIDL_user_free(void __RPC_FAR * ptr) { heap_free(ptr); }