return ret;\r
}\r
\r
+QUERY_SERVICE_CONFIG *query_service_config(const TCHAR *service_name, SC_HANDLE service_handle) {\r
+ QUERY_SERVICE_CONFIG *qsc;\r
+ unsigned long bufsize;\r
+ unsigned long error;\r
+\r
+ QueryServiceConfig(service_handle, 0, 0, &bufsize);\r
+ error = GetLastError();\r
+ if (error == ERROR_INSUFFICIENT_BUFFER) {\r
+ qsc = (QUERY_SERVICE_CONFIG *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, bufsize);\r
+ if (! qsc) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("QUERY_SERVICE_CONFIG"), _T("query_service_config()"), 0);\r
+ return 0;\r
+ }\r
+ }\r
+ else {\r
+ print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG_FAILED, service_name, error_string(error), 0);\r
+ return 0;\r
+ }\r
+\r
+ if (! QueryServiceConfig(service_handle, qsc, bufsize, &bufsize)) {\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+ print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG_FAILED, service_name, error_string(GetLastError()), 0);\r
+ return 0;\r
+ }\r
+\r
+ return qsc;\r
+}\r
+\r
+int set_service_description(const TCHAR *service_name, SC_HANDLE service_handle, TCHAR *buffer) {\r
+ SERVICE_DESCRIPTION description;\r
+ ZeroMemory(&description, sizeof(description));\r
+ /*\r
+ lpDescription must be NULL if we aren't changing, the new description\r
+ or "".\r
+ */\r
+ if (buffer && buffer[0]) description.lpDescription = buffer;\r
+ else description.lpDescription = _T("");\r
+\r
+ if (ChangeServiceConfig2(service_handle, SERVICE_CONFIG_DESCRIPTION, &description)) return 0;\r
+\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SERVICE_CONFIG_DESCRIPTION_FAILED, service_name, error_string(GetLastError()), 0);\r
+ return 1;\r
+}\r
+\r
+int get_service_description(const TCHAR *service_name, SC_HANDLE service_handle, unsigned long len, TCHAR *buffer) {\r
+ if (! buffer) return 1;\r
+\r
+ unsigned long bufsize;\r
+ QueryServiceConfig2(service_handle, SERVICE_CONFIG_DESCRIPTION, 0, 0, &bufsize);\r
+ unsigned long error = GetLastError();\r
+ if (error == ERROR_INSUFFICIENT_BUFFER) {\r
+ SERVICE_DESCRIPTION *description = (SERVICE_DESCRIPTION *) HeapAlloc(GetProcessHeap(), 0, bufsize);\r
+ if (! description) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("SERVICE_CONFIG_DESCRIPTION"), _T("get_service_description()"));\r
+ return 2;\r
+ }\r
+\r
+ if (QueryServiceConfig2(service_handle, SERVICE_CONFIG_DESCRIPTION, (unsigned char *) description, bufsize, &bufsize)) {\r
+ if (description->lpDescription) _sntprintf_s(buffer, len, _TRUNCATE, _T("%s"), description->lpDescription);\r
+ else ZeroMemory(buffer, len * sizeof(TCHAR));\r
+ HeapFree(GetProcessHeap(), 0, description);\r
+ return 0;\r
+ }\r
+ else {\r
+ HeapFree(GetProcessHeap(), 0, description);\r
+ print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service_name, _T("SERVICE_CONFIG_DESCRIPTION"), error_string(error));\r
+ return 3;\r
+ }\r
+ }\r
+ else {\r
+ print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service_name, _T("SERVICE_CONFIG_DESCRIPTION"), error_string(error));\r
+ return 4;\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+int get_service_startup(const TCHAR *service_name, SC_HANDLE service_handle, const QUERY_SERVICE_CONFIG *qsc, unsigned long *startup) {\r
+ if (! qsc) return 1;\r
+\r
+ switch (qsc->dwStartType) {\r
+ case SERVICE_DEMAND_START: *startup = NSSM_STARTUP_MANUAL; break;\r
+ case SERVICE_DISABLED: *startup = NSSM_STARTUP_DISABLED; break;\r
+ default: *startup = NSSM_STARTUP_AUTOMATIC;\r
+ }\r
+\r
+ if (*startup != NSSM_STARTUP_AUTOMATIC) return 0;\r
+\r
+ /* Check for delayed start. */\r
+ unsigned long bufsize;\r
+ unsigned long error;\r
+ QueryServiceConfig2(service_handle, SERVICE_CONFIG_DELAYED_AUTO_START_INFO, 0, 0, &bufsize);\r
+ error = GetLastError();\r
+ if (error == ERROR_INSUFFICIENT_BUFFER) {\r
+ SERVICE_DELAYED_AUTO_START_INFO *info = (SERVICE_DELAYED_AUTO_START_INFO *) HeapAlloc(GetProcessHeap(), 0, bufsize);\r
+ if (! info) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("SERVICE_DELAYED_AUTO_START_INFO"), _T("get_service_startup()"));\r
+ return 2;\r
+ }\r
+\r
+ if (QueryServiceConfig2(service_handle, SERVICE_CONFIG_DELAYED_AUTO_START_INFO, (unsigned char *) info, bufsize, &bufsize)) {\r
+ if (info->fDelayedAutostart) *startup = NSSM_STARTUP_DELAYED;\r
+ HeapFree(GetProcessHeap(), 0, info);\r
+ return 0;\r
+ }\r
+ else {\r
+ error = GetLastError();\r
+ if (error != ERROR_INVALID_LEVEL) {\r
+ print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service_name, _T("SERVICE_CONFIG_DELAYED_AUTO_START_INFO"), error_string(error));\r
+ return 3;\r
+ }\r
+ }\r
+ }\r
+ else if (error != ERROR_INVALID_LEVEL) {\r
+ print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service_name, _T("SERVICE_DELAYED_AUTO_START_INFO"), error_string(error));\r
+ return 3;\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+int get_service_username(const TCHAR *service_name, const QUERY_SERVICE_CONFIG *qsc, TCHAR **username, size_t *usernamelen) {\r
+ if (! username) return 1;\r
+ if (! usernamelen) return 1;\r
+\r
+ *username = 0;\r
+ *usernamelen = 0;\r
+\r
+ if (! qsc) return 1;\r
+\r
+ if (str_equiv(qsc->lpServiceStartName, NSSM_LOCALSYSTEM_ACCOUNT)) return 0;\r
+\r
+ size_t len = _tcslen(qsc->lpServiceStartName);\r
+ *username = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(TCHAR));\r
+ if (! *username) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("username"), _T("get_service_username()"));\r
+ return 2;\r
+ }\r
+\r
+ memmove(*username, qsc->lpServiceStartName, (len + 1) * sizeof(TCHAR));\r
+ *usernamelen = len;\r
+\r
+ return 0;\r
+}\r
+\r
static int grant_logon_as_service(const TCHAR *username) {\r
if (str_equiv(username, NSSM_LOCALSYSTEM_ACCOUNT)) return 0;\r
\r
\r
/* Get system details. */\r
unsigned long bufsize;\r
- unsigned long error;\r
- QUERY_SERVICE_CONFIG *qsc;\r
-\r
- QueryServiceConfig(service->handle, 0, 0, &bufsize);\r
- error = GetLastError();\r
- if (error == ERROR_INSUFFICIENT_BUFFER) {\r
- qsc = (QUERY_SERVICE_CONFIG *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, bufsize);\r
- if (! qsc) {\r
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("QUERY_SERVICE_CONFIG"), _T("pre_edit_service()"), 0);\r
- return 4;\r
- }\r
- }\r
- else {\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service->name, service->handle);\r
+ if (! qsc) {\r
CloseHandle(service->handle);\r
CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG_FAILED, service->name, error_string(error), 0);\r
return 4;\r
}\r
\r
- if (! QueryServiceConfig(service->handle, qsc, bufsize, &bufsize)) {\r
+ service->type = qsc->dwServiceType;\r
+ if (! (service->type & SERVICE_WIN32_OWN_PROCESS)) {\r
HeapFree(GetProcessHeap(), 0, qsc);\r
CloseHandle(service->handle);\r
CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG_FAILED, service->name, error_string(GetLastError()), 0);\r
- return 4;\r
+ print_message(stderr, NSSM_MESSAGE_CANNOT_EDIT, service->name, _T("SERVICE_WIN32_OWN_PROCESS"), 0);\r
+ return 3;\r
}\r
\r
- service->type = qsc->dwServiceType;\r
- if (! (service->type & SERVICE_WIN32_OWN_PROCESS)) {\r
+ if (get_service_startup(service->name, service->handle, qsc, &service->startup)) {\r
HeapFree(GetProcessHeap(), 0, qsc);\r
CloseHandle(service->handle);\r
CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_CANNOT_EDIT, service->name, _T("SERVICE_WIN32_OWN_PROCESS"), 0);\r
- return 3;\r
+ return 4;\r
}\r
\r
- switch (qsc->dwStartType) {\r
- case SERVICE_DEMAND_START: service->startup = NSSM_STARTUP_MANUAL; break;\r
- case SERVICE_DISABLED: service->startup = NSSM_STARTUP_DISABLED; break;\r
- default: service->startup = NSSM_STARTUP_AUTOMATIC;\r
- }\r
- if (! str_equiv(qsc->lpServiceStartName, NSSM_LOCALSYSTEM_ACCOUNT)) {\r
- size_t len = _tcslen(qsc->lpServiceStartName);\r
- service->username = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(TCHAR));\r
- if (service->username) {\r
- memmove(service->username, qsc->lpServiceStartName, (len + 1) * sizeof(TCHAR));\r
- service->usernamelen = (unsigned long) len;\r
- }\r
- else {\r
- HeapFree(GetProcessHeap(), 0, qsc);\r
- CloseHandle(service->handle);\r
- CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("username"), _T("pre_edit_service()"));\r
- return 4;\r
- }\r
+ if (get_service_username(service->name, qsc, &service->username, &service->usernamelen)) {\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+ CloseHandle(service->handle);\r
+ CloseServiceHandle(services);\r
+ return 5;\r
}\r
+\r
_sntprintf_s(service->displayname, _countof(service->displayname), _TRUNCATE, _T("%s"), qsc->lpDisplayName);\r
\r
/* Get the canonical service name. We open it case insensitively. */\r
HeapFree(GetProcessHeap(), 0, qsc);\r
\r
/* Get extended system details. */\r
- if (service->startup == NSSM_STARTUP_AUTOMATIC) {\r
- QueryServiceConfig2(service->handle, SERVICE_CONFIG_DELAYED_AUTO_START_INFO, 0, 0, &bufsize);\r
- error = GetLastError();\r
- if (error == ERROR_INSUFFICIENT_BUFFER) {\r
- SERVICE_DELAYED_AUTO_START_INFO *info = (SERVICE_DELAYED_AUTO_START_INFO *) HeapAlloc(GetProcessHeap(), 0, bufsize);\r
- if (! info) {\r
- CloseHandle(service->handle);\r
- CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("SERVICE_DELAYED_AUTO_START_INFO"), _T("pre_edit_service()"));\r
- return 5;\r
- }\r
-\r
- if (QueryServiceConfig2(service->handle, SERVICE_CONFIG_DELAYED_AUTO_START_INFO, (unsigned char *) info, bufsize, &bufsize)) {\r
- if (info->fDelayedAutostart) service->startup = NSSM_STARTUP_DELAYED;\r
- }\r
- else {\r
- error = GetLastError();\r
- if (error != ERROR_INVALID_LEVEL) {\r
- CloseHandle(service->handle);\r
- CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service->name, _T("SERVICE_CONFIG_DELAYED_AUTO_START_INFO"), error_string(error));\r
- return 5;\r
- }\r
- }\r
- }\r
- else if (error != ERROR_INVALID_LEVEL) {\r
- CloseHandle(service->handle);\r
- CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service->name, _T("SERVICE_DELAYED_AUTO_START_INFO"), error_string(error));\r
- return 5;\r
- }\r
- }\r
-\r
- QueryServiceConfig2(service->handle, SERVICE_CONFIG_DESCRIPTION, 0, 0, &bufsize);\r
- error = GetLastError();\r
- if (error == ERROR_INSUFFICIENT_BUFFER) {\r
- SERVICE_DESCRIPTION *description = (SERVICE_DESCRIPTION *) HeapAlloc(GetProcessHeap(), 0, bufsize);\r
- if (! description) {\r
- CloseHandle(service->handle);\r
- CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("SERVICE_CONFIG_DESCRIPTION"), _T("pre_edit_service()"));\r
- return 6;\r
- }\r
-\r
- if (QueryServiceConfig2(service->handle, SERVICE_CONFIG_DESCRIPTION, (unsigned char *) description, bufsize, &bufsize)) {\r
- if (description->lpDescription) _sntprintf_s(service->description, _countof(service->description), _TRUNCATE, _T("%s"), description->lpDescription);\r
- HeapFree(GetProcessHeap(), 0, description);\r
- }\r
- else {\r
- HeapFree(GetProcessHeap(), 0, description);\r
- CloseHandle(service->handle);\r
- CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service->name, _T("SERVICE_CONFIG_DELAYED_AUTO_START_INFO"), error_string(error));\r
- return 6;\r
- }\r
- }\r
- else {\r
+ if (get_service_description(service->name, service->handle, _countof(service->description), service->description)) {\r
CloseHandle(service->handle);\r
CloseServiceHandle(services);\r
- print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service->name, _T("SERVICE_CONFIG_DELAYED_AUTO_START_INFO"), error_string(error));\r
return 6;\r
}\r
\r
}\r
\r
if (service->description[0] || editing) {\r
- SERVICE_DESCRIPTION description;\r
- ZeroMemory(&description, sizeof(description));\r
- if (service->description[0]) description.lpDescription = service->description;\r
- else description.lpDescription = 0;\r
- if (! ChangeServiceConfig2(service->handle, SERVICE_CONFIG_DESCRIPTION, &description)) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SERVICE_CONFIG_DESCRIPTION_FAILED, service->name, error_string(GetLastError()), 0);\r
- }\r
+ set_service_description(service->name, service->handle, service->description);\r
}\r
\r
SERVICE_DELAYED_AUTO_START_INFO delayed;\r