| /* | |
| * Services.exe - RPC functions | |
| * | |
| * Copyright 2007 Google (Mikolaj Zalewski) | |
| * | |
| * 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 | |
| */ | |
| #define WIN32_LEAN_AND_MEAN | |
| #include <stdarg.h> | |
| #include <windows.h> | |
| #include <winternl.h> | |
| #include <winsvc.h> | |
| #include <ntsecapi.h> | |
| #include <rpc.h> | |
| #include "wine/list.h" | |
| #include "wine/unicode.h" | |
| #include "wine/debug.h" | |
| #include "services.h" | |
| #include "svcctl.h" | |
| extern HANDLE CDECL __wine_make_process_system(void); | |
| WINE_DEFAULT_DEBUG_CHANNEL(service); | |
| static const GENERIC_MAPPING g_scm_generic = | |
| { | |
| (STANDARD_RIGHTS_READ | SC_MANAGER_ENUMERATE_SERVICE | SC_MANAGER_QUERY_LOCK_STATUS), | |
| (STANDARD_RIGHTS_WRITE | SC_MANAGER_CREATE_SERVICE | SC_MANAGER_MODIFY_BOOT_CONFIG), | |
| (STANDARD_RIGHTS_EXECUTE | SC_MANAGER_CONNECT | SC_MANAGER_LOCK), | |
| SC_MANAGER_ALL_ACCESS | |
| }; | |
| static const GENERIC_MAPPING g_svc_generic = | |
| { | |
| (STANDARD_RIGHTS_READ | SERVICE_QUERY_CONFIG | SERVICE_QUERY_STATUS | SERVICE_INTERROGATE | SERVICE_ENUMERATE_DEPENDENTS), | |
| (STANDARD_RIGHTS_WRITE | SERVICE_CHANGE_CONFIG), | |
| (STANDARD_RIGHTS_EXECUTE | SERVICE_START | SERVICE_STOP | SERVICE_PAUSE_CONTINUE | SERVICE_USER_DEFINED_CONTROL), | |
| SERVICE_ALL_ACCESS | |
| }; | |
| typedef enum | |
| { | |
| SC_HTYPE_DONT_CARE = 0, | |
| SC_HTYPE_MANAGER, | |
| SC_HTYPE_SERVICE | |
| } SC_HANDLE_TYPE; | |
| struct sc_handle | |
| { | |
| SC_HANDLE_TYPE type; | |
| DWORD access; | |
| }; | |
| struct sc_manager_handle /* service control manager handle */ | |
| { | |
| struct sc_handle hdr; | |
| struct scmdatabase *db; | |
| }; | |
| struct sc_service_handle /* service handle */ | |
| { | |
| struct sc_handle hdr; | |
| struct service_entry *service_entry; | |
| }; | |
| struct sc_lock | |
| { | |
| struct scmdatabase *db; | |
| }; | |
| static void free_config_strings(QUERY_SERVICE_CONFIGW *old_cfg, QUERY_SERVICE_CONFIGW *new_cfg) | |
| { | |
| if (old_cfg->lpBinaryPathName != new_cfg->lpBinaryPathName) | |
| HeapFree(GetProcessHeap(), 0, old_cfg->lpBinaryPathName); | |
| if (old_cfg->lpLoadOrderGroup != new_cfg->lpLoadOrderGroup) | |
| HeapFree(GetProcessHeap(), 0, old_cfg->lpLoadOrderGroup); | |
| if (old_cfg->lpServiceStartName != new_cfg->lpServiceStartName) | |
| HeapFree(GetProcessHeap(), 0, old_cfg->lpServiceStartName); | |
| if (old_cfg->lpDisplayName != new_cfg->lpDisplayName) | |
| HeapFree(GetProcessHeap(), 0, old_cfg->lpDisplayName); | |
| } | |
| /* Check if the given handle is of the required type and allows the requested access. */ | |
| static DWORD validate_context_handle(SC_RPC_HANDLE handle, DWORD type, DWORD needed_access, struct sc_handle **out_hdr) | |
| { | |
| struct sc_handle *hdr = handle; | |
| if (type != SC_HTYPE_DONT_CARE && hdr->type != type) | |
| { | |
| WINE_ERR("Handle is of an invalid type (%d, %d)\n", hdr->type, type); | |
| return ERROR_INVALID_HANDLE; | |
| } | |
| if ((needed_access & hdr->access) != needed_access) | |
| { | |
| WINE_ERR("Access denied - handle created with access %x, needed %x\n", hdr->access, needed_access); | |
| return ERROR_ACCESS_DENIED; | |
| } | |
| *out_hdr = hdr; | |
| return ERROR_SUCCESS; | |
| } | |
| static DWORD validate_scm_handle(SC_RPC_HANDLE handle, DWORD needed_access, struct sc_manager_handle **manager) | |
| { | |
| struct sc_handle *hdr; | |
| DWORD err = validate_context_handle(handle, SC_HTYPE_MANAGER, needed_access, &hdr); | |
| if (err == ERROR_SUCCESS) | |
| *manager = (struct sc_manager_handle *)hdr; | |
| return err; | |
| } | |
| static DWORD validate_service_handle(SC_RPC_HANDLE handle, DWORD needed_access, struct sc_service_handle **service) | |
| { | |
| struct sc_handle *hdr; | |
| DWORD err = validate_context_handle(handle, SC_HTYPE_SERVICE, needed_access, &hdr); | |
| if (err == ERROR_SUCCESS) | |
| *service = (struct sc_service_handle *)hdr; | |
| return err; | |
| } | |
| DWORD svcctl_OpenSCManagerW( | |
| MACHINE_HANDLEW MachineName, /* Note: this parameter is ignored */ | |
| LPCWSTR DatabaseName, | |
| DWORD dwAccessMask, | |
| SC_RPC_HANDLE *handle) | |
| { | |
| struct sc_manager_handle *manager; | |
| WINE_TRACE("(%s, %s, %x)\n", wine_dbgstr_w(MachineName), wine_dbgstr_w(DatabaseName), dwAccessMask); | |
| if (DatabaseName != NULL && DatabaseName[0]) | |
| { | |
| if (strcmpW(DatabaseName, SERVICES_FAILED_DATABASEW) == 0) | |
| return ERROR_DATABASE_DOES_NOT_EXIST; | |
| if (strcmpW(DatabaseName, SERVICES_ACTIVE_DATABASEW) != 0) | |
| return ERROR_INVALID_NAME; | |
| } | |
| if (!(manager = HeapAlloc(GetProcessHeap(), 0, sizeof(*manager)))) | |
| return ERROR_NOT_ENOUGH_SERVER_MEMORY; | |
| manager->hdr.type = SC_HTYPE_MANAGER; | |
| if (dwAccessMask & MAXIMUM_ALLOWED) | |
| dwAccessMask |= SC_MANAGER_ALL_ACCESS; | |
| manager->hdr.access = dwAccessMask; | |
| RtlMapGenericMask(&manager->hdr.access, &g_scm_generic); | |
| manager->db = active_database; | |
| *handle = &manager->hdr; | |
| return ERROR_SUCCESS; | |
| } | |
| static void SC_RPC_HANDLE_destroy(SC_RPC_HANDLE handle) | |
| { | |
| struct sc_handle *hdr = handle; | |
| switch (hdr->type) | |
| { | |
| case SC_HTYPE_MANAGER: | |
| { | |
| struct sc_manager_handle *manager = (struct sc_manager_handle *)hdr; | |
| HeapFree(GetProcessHeap(), 0, manager); | |
| break; | |
| } | |
| case SC_HTYPE_SERVICE: | |
| { | |
| struct sc_service_handle *service = (struct sc_service_handle *)hdr; | |
| release_service(service->service_entry); | |
| HeapFree(GetProcessHeap(), 0, service); | |
| break; | |
| } | |
| default: | |
| WINE_ERR("invalid handle type %d\n", hdr->type); | |
| RpcRaiseException(ERROR_INVALID_HANDLE); | |
| } | |
| } | |
| DWORD svcctl_GetServiceDisplayNameW( | |
| SC_RPC_HANDLE hSCManager, | |
| LPCWSTR lpServiceName, | |
| WCHAR *lpBuffer, | |
| DWORD *cchBufSize) | |
| { | |
| struct sc_manager_handle *manager; | |
| struct service_entry *entry; | |
| DWORD err; | |
| WINE_TRACE("(%s, %d)\n", wine_dbgstr_w(lpServiceName), *cchBufSize); | |
| if ((err = validate_scm_handle(hSCManager, 0, &manager)) != ERROR_SUCCESS) | |
| return err; | |
| scmdatabase_lock_shared(manager->db); | |
| entry = scmdatabase_find_service(manager->db, lpServiceName); | |
| if (entry != NULL) | |
| { | |
| LPCWSTR name; | |
| int len; | |
| service_lock_shared(entry); | |
| name = get_display_name(entry); | |
| len = strlenW(name); | |
| if (len <= *cchBufSize) | |
| { | |
| err = ERROR_SUCCESS; | |
| memcpy(lpBuffer, name, (len + 1)*sizeof(*name)); | |
| } | |
| else | |
| err = ERROR_INSUFFICIENT_BUFFER; | |
| *cchBufSize = len; | |
| service_unlock(entry); | |
| } | |
| else | |
| err = ERROR_SERVICE_DOES_NOT_EXIST; | |
| scmdatabase_unlock(manager->db); | |
| if (err != ERROR_SUCCESS) | |
| lpBuffer[0] = 0; | |
| return err; | |
| } | |
| DWORD svcctl_GetServiceKeyNameW( | |
| SC_RPC_HANDLE hSCManager, | |
| LPCWSTR lpServiceDisplayName, | |
| WCHAR *lpBuffer, | |
| DWORD *cchBufSize) | |
| { | |
| struct service_entry *entry; | |
| struct sc_manager_handle *manager; | |
| DWORD err; | |
| WINE_TRACE("(%s, %d)\n", wine_dbgstr_w(lpServiceDisplayName), *cchBufSize); | |
| if ((err = validate_scm_handle(hSCManager, 0, &manager)) != ERROR_SUCCESS) | |
| return err; | |
| scmdatabase_lock_shared(manager->db); | |
| entry = scmdatabase_find_service_by_displayname(manager->db, lpServiceDisplayName); | |
| if (entry != NULL) | |
| { | |
| int len; | |
| service_lock_shared(entry); | |
| len = strlenW(entry->name); | |
| if (len <= *cchBufSize) | |
| { | |
| err = ERROR_SUCCESS; | |
| memcpy(lpBuffer, entry->name, (len + 1)*sizeof(*entry->name)); | |
| } | |
| else | |
| err = ERROR_INSUFFICIENT_BUFFER; | |
| *cchBufSize = len; | |
| service_unlock(entry); | |
| } | |
| else | |
| err = ERROR_SERVICE_DOES_NOT_EXIST; | |
| scmdatabase_unlock(manager->db); | |
| if (err != ERROR_SUCCESS) | |
| lpBuffer[0] = 0; | |
| return err; | |
| } | |
| static DWORD create_handle_for_service(struct service_entry *entry, DWORD dwDesiredAccess, SC_RPC_HANDLE *phService) | |
| { | |
| struct sc_service_handle *service; | |
| if (!(service = HeapAlloc(GetProcessHeap(), 0, sizeof(*service)))) | |
| { | |
| release_service(entry); | |
| return ERROR_NOT_ENOUGH_SERVER_MEMORY; | |
| } | |
| service->hdr.type = SC_HTYPE_SERVICE; | |
| service->hdr.access = dwDesiredAccess; | |
| RtlMapGenericMask(&service->hdr.access, &g_svc_generic); | |
| service->service_entry = entry; | |
| if (dwDesiredAccess & MAXIMUM_ALLOWED) | |
| dwDesiredAccess |= SERVICE_ALL_ACCESS; | |
| *phService = &service->hdr; | |
| return ERROR_SUCCESS; | |
| } | |
| DWORD svcctl_OpenServiceW( | |
| SC_RPC_HANDLE hSCManager, | |
| LPCWSTR lpServiceName, | |
| DWORD dwDesiredAccess, | |
| SC_RPC_HANDLE *phService) | |
| { | |
| struct sc_manager_handle *manager; | |
| struct service_entry *entry; | |
| DWORD err; | |
| WINE_TRACE("(%s, 0x%x)\n", wine_dbgstr_w(lpServiceName), dwDesiredAccess); | |
| if ((err = validate_scm_handle(hSCManager, 0, &manager)) != ERROR_SUCCESS) | |
| return err; | |
| if (!validate_service_name(lpServiceName)) | |
| return ERROR_INVALID_NAME; | |
| scmdatabase_lock_shared(manager->db); | |
| entry = scmdatabase_find_service(manager->db, lpServiceName); | |
| if (entry != NULL) | |
| InterlockedIncrement(&entry->ref_count); | |
| scmdatabase_unlock(manager->db); | |
| if (entry == NULL) | |
| return ERROR_SERVICE_DOES_NOT_EXIST; | |
| return create_handle_for_service(entry, dwDesiredAccess, phService); | |
| } | |
| DWORD svcctl_CreateServiceW( | |
| SC_RPC_HANDLE hSCManager, | |
| LPCWSTR lpServiceName, | |
| LPCWSTR lpDisplayName, | |
| DWORD dwDesiredAccess, | |
| DWORD dwServiceType, | |
| DWORD dwStartType, | |
| DWORD dwErrorControl, | |
| LPCWSTR lpBinaryPathName, | |
| LPCWSTR lpLoadOrderGroup, | |
| DWORD *lpdwTagId, | |
| const BYTE *lpDependencies, | |
| DWORD dwDependenciesSize, | |
| LPCWSTR lpServiceStartName, | |
| const BYTE *lpPassword, | |
| DWORD dwPasswordSize, | |
| SC_RPC_HANDLE *phService) | |
| { | |
| struct sc_manager_handle *manager; | |
| struct service_entry *entry; | |
| DWORD err; | |
| WINE_TRACE("(%s, %s, 0x%x, %s)\n", wine_dbgstr_w(lpServiceName), wine_dbgstr_w(lpDisplayName), dwDesiredAccess, wine_dbgstr_w(lpBinaryPathName)); | |
| if ((err = validate_scm_handle(hSCManager, SC_MANAGER_CREATE_SERVICE, &manager)) != ERROR_SUCCESS) | |
| return err; | |
| if (!validate_service_name(lpServiceName)) | |
| return ERROR_INVALID_NAME; | |
| if (!check_multisz((LPCWSTR)lpDependencies, dwDependenciesSize) || !lpServiceName[0] || !lpBinaryPathName[0]) | |
| return ERROR_INVALID_PARAMETER; | |
| if (lpPassword) | |
| WINE_FIXME("Don't know how to add a password\n"); /* I always get ERROR_GEN_FAILURE */ | |
| if (lpDependencies) | |
| WINE_FIXME("Dependencies not supported yet\n"); | |
| err = service_create(lpServiceName, &entry); | |
| if (err != ERROR_SUCCESS) | |
| return err; | |
| entry->ref_count = 1; | |
| entry->config.dwServiceType = entry->status.dwServiceType = dwServiceType; | |
| entry->config.dwStartType = dwStartType; | |
| entry->config.dwErrorControl = dwErrorControl; | |
| entry->config.lpBinaryPathName = strdupW(lpBinaryPathName); | |
| entry->config.lpLoadOrderGroup = strdupW(lpLoadOrderGroup); | |
| entry->config.lpServiceStartName = strdupW(lpServiceStartName); | |
| entry->config.lpDisplayName = strdupW(lpDisplayName); | |
| if (lpdwTagId) /* TODO: In most situations a non-NULL TagId will generate an ERROR_INVALID_PARAMETER. */ | |
| entry->config.dwTagId = *lpdwTagId; | |
| else | |
| entry->config.dwTagId = 0; | |
| /* other fields NULL*/ | |
| if (!validate_service_config(entry)) | |
| { | |
| WINE_ERR("Invalid data while trying to create service\n"); | |
| free_service_entry(entry); | |
| return ERROR_INVALID_PARAMETER; | |
| } | |
| scmdatabase_lock_exclusive(manager->db); | |
| if (scmdatabase_find_service(manager->db, lpServiceName)) | |
| { | |
| scmdatabase_unlock(manager->db); | |
| free_service_entry(entry); | |
| return ERROR_SERVICE_EXISTS; | |
| } | |
| if (scmdatabase_find_service_by_displayname(manager->db, get_display_name(entry))) | |
| { | |
| scmdatabase_unlock(manager->db); | |
| free_service_entry(entry); | |
| return ERROR_DUPLICATE_SERVICE_NAME; | |
| } | |
| err = scmdatabase_add_service(manager->db, entry); | |
| if (err != ERROR_SUCCESS) | |
| { | |
| scmdatabase_unlock(manager->db); | |
| free_service_entry(entry); | |
| return err; | |
| } | |
| scmdatabase_unlock(manager->db); | |
| return create_handle_for_service(entry, dwDesiredAccess, phService); | |
| } | |
| DWORD svcctl_DeleteService( | |
| SC_RPC_HANDLE hService) | |
| { | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| if ((err = validate_service_handle(hService, DELETE, &service)) != ERROR_SUCCESS) | |
| return err; | |
| scmdatabase_lock_exclusive(service->service_entry->db); | |
| service_lock_exclusive(service->service_entry); | |
| if (!is_marked_for_delete(service->service_entry)) | |
| err = scmdatabase_remove_service(service->service_entry->db, service->service_entry); | |
| else | |
| err = ERROR_SERVICE_MARKED_FOR_DELETE; | |
| service_unlock(service->service_entry); | |
| scmdatabase_unlock(service->service_entry->db); | |
| return err; | |
| } | |
| DWORD svcctl_QueryServiceConfigW( | |
| SC_RPC_HANDLE hService, | |
| QUERY_SERVICE_CONFIGW *config) | |
| { | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| WINE_TRACE("(%p)\n", config); | |
| if ((err = validate_service_handle(hService, SERVICE_QUERY_CONFIG, &service)) != 0) | |
| return err; | |
| service_lock_shared(service->service_entry); | |
| config->dwServiceType = service->service_entry->config.dwServiceType; | |
| config->dwStartType = service->service_entry->config.dwStartType; | |
| config->dwErrorControl = service->service_entry->config.dwErrorControl; | |
| config->lpBinaryPathName = strdupW(service->service_entry->config.lpBinaryPathName); | |
| config->lpLoadOrderGroup = strdupW(service->service_entry->config.lpLoadOrderGroup); | |
| config->dwTagId = service->service_entry->config.dwTagId; | |
| config->lpDependencies = NULL; /* TODO */ | |
| config->lpServiceStartName = strdupW(service->service_entry->config.lpServiceStartName); | |
| config->lpDisplayName = strdupW(service->service_entry->config.lpDisplayName); | |
| service_unlock(service->service_entry); | |
| return ERROR_SUCCESS; | |
| } | |
| DWORD svcctl_ChangeServiceConfigW( | |
| SC_RPC_HANDLE hService, | |
| DWORD dwServiceType, | |
| DWORD dwStartType, | |
| DWORD dwErrorControl, | |
| LPCWSTR lpBinaryPathName, | |
| LPCWSTR lpLoadOrderGroup, | |
| DWORD *lpdwTagId, | |
| const BYTE *lpDependencies, | |
| DWORD dwDependenciesSize, | |
| LPCWSTR lpServiceStartName, | |
| const BYTE *lpPassword, | |
| DWORD dwPasswordSize, | |
| LPCWSTR lpDisplayName) | |
| { | |
| struct service_entry new_entry, *entry; | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| WINE_TRACE("\n"); | |
| if ((err = validate_service_handle(hService, SERVICE_CHANGE_CONFIG, &service)) != 0) | |
| return err; | |
| if (!check_multisz((LPCWSTR)lpDependencies, dwDependenciesSize)) | |
| return ERROR_INVALID_PARAMETER; | |
| /* first check if the new configuration is correct */ | |
| service_lock_exclusive(service->service_entry); | |
| if (is_marked_for_delete(service->service_entry)) | |
| { | |
| service_unlock(service->service_entry); | |
| return ERROR_SERVICE_MARKED_FOR_DELETE; | |
| } | |
| if (lpDisplayName != NULL && | |
| (entry = scmdatabase_find_service_by_displayname(service->service_entry->db, lpDisplayName)) && | |
| (entry != service->service_entry)) | |
| { | |
| service_unlock(service->service_entry); | |
| return ERROR_DUPLICATE_SERVICE_NAME; | |
| } | |
| new_entry = *service->service_entry; | |
| if (dwServiceType != SERVICE_NO_CHANGE) | |
| new_entry.config.dwServiceType = dwServiceType; | |
| if (dwStartType != SERVICE_NO_CHANGE) | |
| new_entry.config.dwStartType = dwStartType; | |
| if (dwErrorControl != SERVICE_NO_CHANGE) | |
| new_entry.config.dwErrorControl = dwErrorControl; | |
| if (lpBinaryPathName != NULL) | |
| new_entry.config.lpBinaryPathName = (LPWSTR)lpBinaryPathName; | |
| if (lpLoadOrderGroup != NULL) | |
| new_entry.config.lpLoadOrderGroup = (LPWSTR)lpLoadOrderGroup; | |
| if (lpdwTagId != NULL) | |
| WINE_FIXME("Changing tag id not supported\n"); | |
| if (lpDependencies != NULL) | |
| WINE_FIXME("Changing dependencies not supported\n"); | |
| if (lpServiceStartName != NULL) | |
| new_entry.config.lpServiceStartName = (LPWSTR)lpServiceStartName; | |
| if (lpPassword != NULL) | |
| WINE_FIXME("Setting password not supported\n"); | |
| if (lpDisplayName != NULL) | |
| new_entry.config.lpDisplayName = (LPWSTR)lpDisplayName; | |
| if (!validate_service_config(&new_entry)) | |
| { | |
| WINE_ERR("The configuration after the change wouldn't be valid\n"); | |
| service_unlock(service->service_entry); | |
| return ERROR_INVALID_PARAMETER; | |
| } | |
| /* configuration OK. The strings needs to be duplicated */ | |
| if (lpBinaryPathName != NULL) | |
| new_entry.config.lpBinaryPathName = strdupW(lpBinaryPathName); | |
| if (lpLoadOrderGroup != NULL) | |
| new_entry.config.lpLoadOrderGroup = strdupW(lpLoadOrderGroup); | |
| if (lpServiceStartName != NULL) | |
| new_entry.config.lpServiceStartName = strdupW(lpServiceStartName); | |
| if (lpDisplayName != NULL) | |
| new_entry.config.lpDisplayName = strdupW(lpDisplayName); | |
| /* try to save to Registry, commit or rollback depending on success */ | |
| err = save_service_config(&new_entry); | |
| if (ERROR_SUCCESS == err) | |
| { | |
| free_config_strings(&service->service_entry->config,&new_entry.config); | |
| *service->service_entry = new_entry; | |
| } | |
| else free_config_strings(&new_entry.config,&service->service_entry->config); | |
| service_unlock(service->service_entry); | |
| return err; | |
| } | |
| DWORD svcctl_SetServiceStatus( | |
| SC_RPC_HANDLE hServiceStatus, | |
| LPSERVICE_STATUS lpServiceStatus) | |
| { | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| WINE_TRACE("(%p, %p)\n", hServiceStatus, lpServiceStatus); | |
| if ((err = validate_service_handle(hServiceStatus, SERVICE_SET_STATUS, &service)) != 0) | |
| return err; | |
| service_lock_exclusive(service->service_entry); | |
| /* FIXME: be a bit more discriminant about what parts of the status we set | |
| * and check that fields are valid */ | |
| service->service_entry->status.dwServiceType = lpServiceStatus->dwServiceType; | |
| service->service_entry->status.dwCurrentState = lpServiceStatus->dwCurrentState; | |
| service->service_entry->status.dwControlsAccepted = lpServiceStatus->dwControlsAccepted; | |
| service->service_entry->status.dwWin32ExitCode = lpServiceStatus->dwWin32ExitCode; | |
| service->service_entry->status.dwServiceSpecificExitCode = lpServiceStatus->dwServiceSpecificExitCode; | |
| service->service_entry->status.dwCheckPoint = lpServiceStatus->dwCheckPoint; | |
| service->service_entry->status.dwWaitHint = lpServiceStatus->dwWaitHint; | |
| service_unlock(service->service_entry); | |
| if (service->service_entry->status_changed_event) | |
| SetEvent(service->service_entry->status_changed_event); | |
| return ERROR_SUCCESS; | |
| } | |
| DWORD svcctl_ChangeServiceConfig2W( SC_RPC_HANDLE hService, DWORD level, SERVICE_CONFIG2W *config ) | |
| { | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| if ((err = validate_service_handle(hService, SERVICE_CHANGE_CONFIG, &service)) != 0) | |
| return err; | |
| switch (level) | |
| { | |
| case SERVICE_CONFIG_DESCRIPTION: | |
| { | |
| WCHAR *descr = NULL; | |
| if (config->descr.lpDescription[0]) | |
| { | |
| if (!(descr = strdupW( config->descr.lpDescription ))) | |
| return ERROR_NOT_ENOUGH_MEMORY; | |
| } | |
| WINE_TRACE( "changing service %p descr to %s\n", service, wine_dbgstr_w(descr) ); | |
| service_lock_exclusive( service->service_entry ); | |
| HeapFree( GetProcessHeap(), 0, service->service_entry->description ); | |
| service->service_entry->description = descr; | |
| save_service_config( service->service_entry ); | |
| service_unlock( service->service_entry ); | |
| } | |
| break; | |
| case SERVICE_CONFIG_FAILURE_ACTIONS: | |
| WINE_FIXME( "SERVICE_CONFIG_FAILURE_ACTIONS not implemented: period %u msg %s cmd %s\n", | |
| config->actions.dwResetPeriod, | |
| wine_dbgstr_w(config->actions.lpRebootMsg), | |
| wine_dbgstr_w(config->actions.lpCommand) ); | |
| break; | |
| default: | |
| WINE_FIXME("level %u not implemented\n", level); | |
| err = ERROR_INVALID_LEVEL; | |
| break; | |
| } | |
| return err; | |
| } | |
| DWORD svcctl_QueryServiceConfig2W( SC_RPC_HANDLE hService, DWORD level, | |
| BYTE *buffer, DWORD size, LPDWORD needed ) | |
| { | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| memset(buffer, 0, size); | |
| if ((err = validate_service_handle(hService, SERVICE_QUERY_STATUS, &service)) != 0) | |
| return err; | |
| switch (level) | |
| { | |
| case SERVICE_CONFIG_DESCRIPTION: | |
| { | |
| SERVICE_DESCRIPTIONW *descr = (SERVICE_DESCRIPTIONW *)buffer; | |
| service_lock_shared(service->service_entry); | |
| *needed = sizeof(*descr); | |
| if (service->service_entry->description) | |
| *needed += (strlenW(service->service_entry->description) + 1) * sizeof(WCHAR); | |
| if (size >= *needed) | |
| { | |
| if (service->service_entry->description) | |
| { | |
| /* store a buffer offset instead of a pointer */ | |
| descr->lpDescription = (WCHAR *)((BYTE *)(descr + 1) - buffer); | |
| strcpyW( (WCHAR *)(descr + 1), service->service_entry->description ); | |
| } | |
| else descr->lpDescription = NULL; | |
| } | |
| else err = ERROR_INSUFFICIENT_BUFFER; | |
| service_unlock(service->service_entry); | |
| } | |
| break; | |
| default: | |
| WINE_FIXME("level %u not implemented\n", level); | |
| err = ERROR_INVALID_LEVEL; | |
| break; | |
| } | |
| return err; | |
| } | |
| DWORD svcctl_QueryServiceStatusEx( | |
| SC_RPC_HANDLE hService, | |
| SC_STATUS_TYPE InfoLevel, | |
| BYTE *lpBuffer, | |
| DWORD cbBufSize, | |
| LPDWORD pcbBytesNeeded) | |
| { | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| LPSERVICE_STATUS_PROCESS pSvcStatusData; | |
| memset(lpBuffer, 0, cbBufSize); | |
| if ((err = validate_service_handle(hService, SERVICE_QUERY_STATUS, &service)) != 0) | |
| return err; | |
| if (InfoLevel != SC_STATUS_PROCESS_INFO) | |
| return ERROR_INVALID_LEVEL; | |
| pSvcStatusData = (LPSERVICE_STATUS_PROCESS) lpBuffer; | |
| if (pSvcStatusData == NULL) | |
| return ERROR_INVALID_PARAMETER; | |
| if (cbBufSize < sizeof(SERVICE_STATUS_PROCESS)) | |
| { | |
| if( pcbBytesNeeded != NULL) | |
| *pcbBytesNeeded = sizeof(SERVICE_STATUS_PROCESS); | |
| return ERROR_INSUFFICIENT_BUFFER; | |
| } | |
| service_lock_shared(service->service_entry); | |
| pSvcStatusData->dwServiceType = service->service_entry->status.dwServiceType; | |
| pSvcStatusData->dwCurrentState = service->service_entry->status.dwCurrentState; | |
| pSvcStatusData->dwControlsAccepted = service->service_entry->status.dwControlsAccepted; | |
| pSvcStatusData->dwWin32ExitCode = service->service_entry->status.dwWin32ExitCode; | |
| pSvcStatusData->dwServiceSpecificExitCode = service->service_entry->status.dwServiceSpecificExitCode; | |
| pSvcStatusData->dwCheckPoint = service->service_entry->status.dwCheckPoint; | |
| pSvcStatusData->dwWaitHint = service->service_entry->status.dwWaitHint; | |
| pSvcStatusData->dwProcessId = service->service_entry->status.dwProcessId; | |
| pSvcStatusData->dwServiceFlags = service->service_entry->status.dwServiceFlags; | |
| service_unlock(service->service_entry); | |
| return ERROR_SUCCESS; | |
| } | |
| /****************************************************************************** | |
| * service_accepts_control | |
| */ | |
| static BOOL service_accepts_control(const struct service_entry *service, DWORD dwControl) | |
| { | |
| DWORD a = service->status.dwControlsAccepted; | |
| switch (dwControl) | |
| { | |
| case SERVICE_CONTROL_INTERROGATE: | |
| return TRUE; | |
| case SERVICE_CONTROL_STOP: | |
| if (a&SERVICE_ACCEPT_STOP) | |
| return TRUE; | |
| break; | |
| case SERVICE_CONTROL_SHUTDOWN: | |
| if (a&SERVICE_ACCEPT_SHUTDOWN) | |
| return TRUE; | |
| break; | |
| case SERVICE_CONTROL_PAUSE: | |
| case SERVICE_CONTROL_CONTINUE: | |
| if (a&SERVICE_ACCEPT_PAUSE_CONTINUE) | |
| return TRUE; | |
| break; | |
| case SERVICE_CONTROL_PARAMCHANGE: | |
| if (a&SERVICE_ACCEPT_PARAMCHANGE) | |
| return TRUE; | |
| break; | |
| case SERVICE_CONTROL_NETBINDADD: | |
| case SERVICE_CONTROL_NETBINDREMOVE: | |
| case SERVICE_CONTROL_NETBINDENABLE: | |
| case SERVICE_CONTROL_NETBINDDISABLE: | |
| if (a&SERVICE_ACCEPT_NETBINDCHANGE) | |
| return TRUE; | |
| case SERVICE_CONTROL_HARDWAREPROFILECHANGE: | |
| if (a&SERVICE_ACCEPT_HARDWAREPROFILECHANGE) | |
| return TRUE; | |
| break; | |
| case SERVICE_CONTROL_POWEREVENT: | |
| if (a&SERVICE_ACCEPT_POWEREVENT) | |
| return TRUE; | |
| break; | |
| case SERVICE_CONTROL_SESSIONCHANGE: | |
| if (a&SERVICE_ACCEPT_SESSIONCHANGE) | |
| return TRUE; | |
| break; | |
| } | |
| return FALSE; | |
| } | |
| /****************************************************************************** | |
| * service_send_control | |
| */ | |
| static BOOL service_send_control(struct service_entry *service, HANDLE pipe, DWORD dwControl, DWORD *result) | |
| { | |
| service_start_info *ssi; | |
| DWORD len, count = 0; | |
| BOOL r; | |
| /* calculate how much space we need to send the startup info */ | |
| len = strlenW(service->name) + 1; | |
| ssi = HeapAlloc(GetProcessHeap(),0,FIELD_OFFSET(service_start_info, data[len])); | |
| ssi->cmd = WINESERV_SENDCONTROL; | |
| ssi->control = dwControl; | |
| ssi->total_size = FIELD_OFFSET(service_start_info, data[len]); | |
| ssi->name_size = strlenW(service->name) + 1; | |
| strcpyW( ssi->data, service->name ); | |
| r = WriteFile(pipe, ssi, ssi->total_size, &count, NULL); | |
| if (!r || count != ssi->total_size) | |
| { | |
| WINE_ERR("service protocol error - failed to write pipe!\n"); | |
| return r; | |
| } | |
| r = ReadFile(pipe, result, sizeof *result, &count, NULL); | |
| if (!r || count != sizeof *result) | |
| WINE_ERR("service protocol error - failed to read pipe " | |
| "r = %d count = %d!\n", r, count); | |
| return r; | |
| } | |
| DWORD svcctl_StartServiceW( | |
| SC_RPC_HANDLE hService, | |
| DWORD dwNumServiceArgs, | |
| LPCWSTR *lpServiceArgVectors) | |
| { | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| WINE_TRACE("(%p, %d, %p)\n", hService, dwNumServiceArgs, lpServiceArgVectors); | |
| if ((err = validate_service_handle(hService, SERVICE_START, &service)) != 0) | |
| return err; | |
| err = service_start(service->service_entry, dwNumServiceArgs, lpServiceArgVectors); | |
| return err; | |
| } | |
| DWORD svcctl_ControlService( | |
| SC_RPC_HANDLE hService, | |
| DWORD dwControl, | |
| SERVICE_STATUS *lpServiceStatus) | |
| { | |
| DWORD access_required; | |
| struct sc_service_handle *service; | |
| DWORD err; | |
| BOOL ret; | |
| HANDLE control_mutex; | |
| HANDLE control_pipe; | |
| WINE_TRACE("(%p, %d, %p)\n", hService, dwControl, lpServiceStatus); | |
| switch (dwControl) | |
| { | |
| case SERVICE_CONTROL_CONTINUE: | |
| case SERVICE_CONTROL_NETBINDADD: | |
| case SERVICE_CONTROL_NETBINDDISABLE: | |
| case SERVICE_CONTROL_NETBINDENABLE: | |
| case SERVICE_CONTROL_NETBINDREMOVE: | |
| case SERVICE_CONTROL_PARAMCHANGE: | |
| case SERVICE_CONTROL_PAUSE: | |
| access_required = SERVICE_PAUSE_CONTINUE; | |
| break; | |
| case SERVICE_CONTROL_INTERROGATE: | |
| access_required = SERVICE_INTERROGATE; | |
| break; | |
| case SERVICE_CONTROL_STOP: | |
| access_required = SERVICE_STOP; | |
| break; | |
| default: | |
| if (dwControl >= 128 && dwControl <= 255) | |
| access_required = SERVICE_USER_DEFINED_CONTROL; | |
| else | |
| return ERROR_INVALID_PARAMETER; | |
| } | |
| if ((err = validate_service_handle(hService, access_required, &service)) != 0) | |
| return err; | |
| service_lock_exclusive(service->service_entry); | |
| if (lpServiceStatus) | |
| { | |
| lpServiceStatus->dwServiceType = service->service_entry->status.dwServiceType; | |
| lpServiceStatus->dwCurrentState = service->service_entry->status.dwCurrentState; | |
| lpServiceStatus->dwControlsAccepted = service->service_entry->status.dwControlsAccepted; | |
| lpServiceStatus->dwWin32ExitCode = service->service_entry->status.dwWin32ExitCode; | |
| lpServiceStatus->dwServiceSpecificExitCode = service->service_entry->status.dwServiceSpecificExitCode; | |
| lpServiceStatus->dwCheckPoint = service->service_entry->status.dwCheckPoint; | |
| lpServiceStatus->dwWaitHint = service->service_entry->status.dwWaitHint; | |
| } | |
| if (!service_accepts_control(service->service_entry, dwControl)) | |
| { | |
| service_unlock(service->service_entry); | |
| return ERROR_INVALID_SERVICE_CONTROL; | |
| } | |
| switch (service->service_entry->status.dwCurrentState) | |
| { | |
| case SERVICE_STOPPED: | |
| service_unlock(service->service_entry); | |
| return ERROR_SERVICE_NOT_ACTIVE; | |
| case SERVICE_START_PENDING: | |
| if (dwControl==SERVICE_CONTROL_STOP) | |
| break; | |
| /* fall thru */ | |
| case SERVICE_STOP_PENDING: | |
| service_unlock(service->service_entry); | |
| return ERROR_SERVICE_CANNOT_ACCEPT_CTRL; | |
| } | |
| /* prevent races by caching these variables and clearing them on | |
| * stop here instead of outside the services lock */ | |
| control_mutex = service->service_entry->control_mutex; | |
| control_pipe = service->service_entry->control_pipe; | |
| if (dwControl == SERVICE_CONTROL_STOP) | |
| { | |
| service->service_entry->control_mutex = NULL; | |
| service->service_entry->control_pipe = INVALID_HANDLE_VALUE; | |
| } | |
| service_unlock(service->service_entry); | |
| ret = WaitForSingleObject(control_mutex, 30000); | |
| if (ret == WAIT_OBJECT_0) | |
| { | |
| DWORD result = ERROR_SUCCESS; | |
| ret = service_send_control(service->service_entry, control_pipe, dwControl, &result); | |
| if (dwControl == SERVICE_CONTROL_STOP) | |
| { | |
| CloseHandle(control_mutex); | |
| CloseHandle(control_pipe); | |
| } | |
| else | |
| ReleaseMutex(control_mutex); | |
| return result; | |
| } | |
| else | |
| { | |
| if (dwControl == SERVICE_CONTROL_STOP) | |
| { | |
| CloseHandle(control_mutex); | |
| CloseHandle(control_pipe); | |
| } | |
| return ERROR_SERVICE_REQUEST_TIMEOUT; | |
| } | |
| } | |
| DWORD svcctl_CloseServiceHandle( | |
| SC_RPC_HANDLE *handle) | |
| { | |
| WINE_TRACE("(&%p)\n", *handle); | |
| SC_RPC_HANDLE_destroy(*handle); | |
| *handle = NULL; | |
| return ERROR_SUCCESS; | |
| } | |
| static void SC_RPC_LOCK_destroy(SC_RPC_LOCK hLock) | |
| { | |
| struct sc_lock *lock = hLock; | |
| scmdatabase_unlock_startup(lock->db); | |
| HeapFree(GetProcessHeap(), 0, lock); | |
| } | |
| void __RPC_USER SC_RPC_LOCK_rundown(SC_RPC_LOCK hLock) | |
| { | |
| SC_RPC_LOCK_destroy(hLock); | |
| } | |
| DWORD svcctl_LockServiceDatabase( | |
| SC_RPC_HANDLE hSCManager, | |
| SC_RPC_LOCK *phLock) | |
| { | |
| struct sc_manager_handle *manager; | |
| struct sc_lock *lock; | |
| DWORD err; | |
| WINE_TRACE("(%p, %p)\n", hSCManager, phLock); | |
| if ((err = validate_scm_handle(hSCManager, SC_MANAGER_LOCK, &manager)) != ERROR_SUCCESS) | |
| return err; | |
| err = scmdatabase_lock_startup(manager->db); | |
| if (err != ERROR_SUCCESS) | |
| return err; | |
| lock = HeapAlloc(GetProcessHeap(), 0, sizeof(struct sc_lock)); | |
| if (!lock) | |
| { | |
| scmdatabase_unlock_startup(manager->db); | |
| return ERROR_NOT_ENOUGH_SERVER_MEMORY; | |
| } | |
| lock->db = manager->db; | |
| *phLock = lock; | |
| return ERROR_SUCCESS; | |
| } | |
| DWORD svcctl_UnlockServiceDatabase( | |
| SC_RPC_LOCK *phLock) | |
| { | |
| WINE_TRACE("(&%p)\n", *phLock); | |
| SC_RPC_LOCK_destroy(*phLock); | |
| *phLock = NULL; | |
| return ERROR_SUCCESS; | |
| } | |
| DWORD svcctl_QueryServiceObjectSecurity( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_SetServiceObjectSecurity( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_QueryServiceStatus( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_NotifyBootConfigStatus( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_SCSetServiceBitsW( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_EnumDependentServicesW( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_EnumServicesStatusW( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_QueryServiceLockStatusW( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_SCSetServiceBitsA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_ChangeServiceConfigA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_CreateServiceA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_EnumDependentServicesA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_EnumServicesStatusA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_OpenSCManagerA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_OpenServiceA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_QueryServiceConfigA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_QueryServiceLockStatusA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_StartServiceA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_GetServiceDisplayNameA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_GetServiceKeyNameA( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_GetCurrentGroupStateW( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_EnumServiceGroupW( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_ChangeServiceConfig2A( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD svcctl_QueryServiceConfig2A( | |
| void) | |
| { | |
| WINE_FIXME("\n"); | |
| return ERROR_CALL_NOT_IMPLEMENTED; | |
| } | |
| DWORD RPC_Init(void) | |
| { | |
| WCHAR transport[] = SVCCTL_TRANSPORT; | |
| WCHAR endpoint[] = SVCCTL_ENDPOINT; | |
| DWORD err; | |
| if ((err = RpcServerUseProtseqEpW(transport, 0, endpoint, NULL)) != ERROR_SUCCESS) | |
| { | |
| WINE_ERR("RpcServerUseProtseq failed with error %u\n", err); | |
| return err; | |
| } | |
| if ((err = RpcServerRegisterIf(svcctl_v2_0_s_ifspec, 0, 0)) != ERROR_SUCCESS) | |
| { | |
| WINE_ERR("RpcServerRegisterIf failed with error %u\n", err); | |
| return err; | |
| } | |
| if ((err = RpcServerListen(1, RPC_C_LISTEN_MAX_CALLS_DEFAULT, TRUE)) != ERROR_SUCCESS) | |
| { | |
| WINE_ERR("RpcServerListen failed with error %u\n", err); | |
| return err; | |
| } | |
| return ERROR_SUCCESS; | |
| } | |
| DWORD RPC_MainLoop(void) | |
| { | |
| DWORD err; | |
| HANDLE hExitEvent = __wine_make_process_system(); | |
| SetEvent(g_hStartedEvent); | |
| WINE_TRACE("Entered main loop\n"); | |
| do | |
| { | |
| err = WaitForSingleObjectEx(hExitEvent, INFINITE, TRUE); | |
| WINE_TRACE("Wait returned %d\n", err); | |
| } while (err != WAIT_OBJECT_0); | |
| WINE_TRACE("Object signaled - wine shutdown\n"); | |
| CloseHandle(hExitEvent); | |
| return ERROR_SUCCESS; | |
| } | |
| void __RPC_USER SC_RPC_HANDLE_rundown(SC_RPC_HANDLE handle) | |
| { | |
| SC_RPC_HANDLE_destroy(handle); | |
| } | |
| void __RPC_FAR * __RPC_USER MIDL_user_allocate(SIZE_T len) | |
| { | |
| return HeapAlloc(GetProcessHeap(), 0, len); | |
| } | |
| void __RPC_USER MIDL_user_free(void __RPC_FAR * ptr) | |
| { | |
| HeapFree(GetProcessHeap(), 0, ptr); | |
| } |