-#include "nssm.h"
-/* XXX: (value && value->string) is probably bogus because value is probably never null */
-
-extern const TCHAR *exit_action_strings[];
-extern const TCHAR *startup_strings[];
-extern const TCHAR *priority_strings[];
-
-/* Does the parameter refer to the default value of the AppExit setting? */
-static inline int is_default_exit_action(const TCHAR *value) {
- return (str_equiv(value, _T("default")) || str_equiv(value, _T("*")) || ! value[0]);
-}
-
-static int value_from_string(const TCHAR *name, value_t *value, const TCHAR *string) {
- size_t len = _tcslen(string);
- if (! len++) {
- value->string = 0;
- return 0;
- }
-
- value->string = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, len * sizeof(TCHAR));
- if (! value->string) {
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, name, _T("value_from_string()"));
- return -1;
- }
-
- if (_sntprintf_s(value->string, len, _TRUNCATE, _T("%s"), string) < 0) {
- HeapFree(GetProcessHeap(), 0, value->string);
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, name, _T("value_from_string()"));
- return -1;
- }
-
- return 1;
-}
-
-/* Functions to manage NSSM-specific settings in the registry. */
-static int setting_set_number(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- if (! key) return -1;
-
- unsigned long number;
- long error;
-
- /* Resetting to default? */
- if (! value || ! value->string) {
- error = RegDeleteValue(key, name);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
- return -1;
- }
- if (str_number(value->string, &number)) return -1;
-
- if (default_value && number == (unsigned long) default_value) {
- error = RegDeleteValue(key, name);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
- return -1;
- }
-
- if (set_number(key, (TCHAR *) name, number)) return -1;
-
- return 1;
-}
-
-static int setting_get_number(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- return get_number(key, (TCHAR *) name, &value->numeric, false);
-}
-
-static int setting_set_string(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- if (! key) return -1;
-
- long error;
-
- /* Resetting to default? */
- if (! value || ! value->string) {
- if (default_value) value->string = (TCHAR *) default_value;
- else {
- error = RegDeleteValue(key, name);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
- return -1;
- }
- }
- if (default_value && _tcslen((TCHAR *) default_value) && str_equiv(value->string, (TCHAR *) default_value)) {
- error = RegDeleteValue(key, name);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
- return -1;
- }
-
- if (set_expand_string(key, (TCHAR *) name, value->string)) return -1;
-
- return 1;
-}
-
-static int setting_get_string(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- TCHAR buffer[VALUE_LENGTH];
-
- if (expand_parameter(key, (TCHAR *) name, (TCHAR *) buffer, (unsigned long) sizeof(buffer), false, false)) return -1;
-
- return value_from_string(name, value, buffer);
-}
-
-static int setting_set_exit_action(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- unsigned long exitcode;
- TCHAR *code;
- TCHAR action_string[ACTION_LEN];
-
- if (additional) {
- /* Default action? */
- if (is_default_exit_action(additional)) code = 0;
- else {
- if (str_number(additional, &exitcode)) return -1;
- code = (TCHAR *) additional;
- }
- }
-
- HKEY key = open_registry(service_name, name, KEY_WRITE);
- if (! key) return -1;
-
- long error;
- int ret = 1;
-
- /* Resetting to default? */
- if (value && value->string) _sntprintf_s(action_string, _countof(action_string), _TRUNCATE, _T("%s"), value->string);
- else {
- if (code) {
- /* Delete explicit action. */
- error = RegDeleteValue(key, code);
- RegCloseKey(key);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, code, service_name, error_string(error));
- return -1;
- }
- else {
- /* Explicitly keep the default action. */
- if (default_value) _sntprintf_s(action_string, _countof(action_string), _TRUNCATE, _T("%s"), (TCHAR *) default_value);
- ret = 0;
- }
- }
-
- /* Validate the string. */
- for (int i = 0; exit_action_strings[i]; i++) {
- if (! _tcsnicmp((const TCHAR *) action_string, exit_action_strings[i], ACTION_LEN)) {
- if (default_value && str_equiv(action_string, (TCHAR *) default_value)) ret = 0;
- if (RegSetValueEx(key, code, 0, REG_SZ, (const unsigned char *) action_string, (unsigned long) (_tcslen(action_string) + 1) * sizeof(TCHAR)) != ERROR_SUCCESS) {
- print_message(stderr, NSSM_MESSAGE_SETVALUE_FAILED, code, service_name, error_string(GetLastError()));
- RegCloseKey(key);
- return -1;
- }
-
- RegCloseKey(key);
- return ret;
- }
- }
-
- print_message(stderr, NSSM_MESSAGE_INVALID_EXIT_ACTION, action_string);
- for (int i = 0; exit_action_strings[i]; i++) _ftprintf(stderr, _T("%s\n"), exit_action_strings[i]);
-
- return -1;
-}
-
-static int setting_get_exit_action(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- unsigned long exitcode = 0;
- unsigned long *code = 0;
-
- if (additional) {
- if (! is_default_exit_action(additional)) {
- if (str_number(additional, &exitcode)) return -1;
- code = &exitcode;
- }
- }
-
- TCHAR action_string[ACTION_LEN];
- bool default_action;
- if (get_exit_action(service_name, code, action_string, &default_action)) return -1;
-
- value_from_string(name, value, action_string);
-
- if (default_action && ! _tcsnicmp((const TCHAR *) action_string, (TCHAR *) default_value, ACTION_LEN)) return 0;
- return 1;
-}
-
-static int setting_set_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- if (! param) return -1;
-
- if (! value || ! value->string || ! value->string[0]) {
- long error = RegDeleteValue(key, name);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
- return -1;
- }
-
- unsigned long envlen = (unsigned long) _tcslen(value->string) + 1;
- TCHAR *unformatted = 0;
- unsigned long newlen;
- if (unformat_environment(value->string, envlen, &unformatted, &newlen)) return -1;
-
- if (test_environment(unformatted)) {
- HeapFree(GetProcessHeap(), 0, unformatted);
- print_message(stderr, NSSM_GUI_INVALID_ENVIRONMENT);
- return -1;
- }
-
- if (RegSetValueEx(key, name, 0, REG_MULTI_SZ, (const unsigned char *) unformatted, (unsigned long) newlen * sizeof(TCHAR)) != ERROR_SUCCESS) {
- if (newlen) HeapFree(GetProcessHeap(), 0, unformatted);
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, NSSM_REG_ENV, error_string(GetLastError()), 0);
- return -1;
- }
-
- if (newlen) HeapFree(GetProcessHeap(), 0, unformatted);
- return 1;
-}
-
-static int setting_get_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- if (! param) return -1;
-
- TCHAR *env = 0;
- unsigned long envlen;
- if (set_environment((TCHAR *) service_name, key, (TCHAR *) name, &env, &envlen)) return -1;
- if (! envlen) return 0;
-
- TCHAR *formatted;
- unsigned long newlen;
- if (format_environment(env, envlen, &formatted, &newlen)) return -1;
-
- int ret;
- if (additional) {
- /* Find named environment variable. */
- TCHAR *s;
- size_t len = _tcslen(additional);
- for (s = env; *s; s++) {
- /* Look for <additional>=<string> NULL NULL */
- if (! _tcsnicmp(s, additional, len) && s[len] == _T('=')) {
- /* Strip <key>= */
- s += len + 1;
- ret = value_from_string(name, value, s);
- HeapFree(GetProcessHeap(), 0, env);
- return ret;
- }
-
- /* Skip this string. */
- for ( ; *s; s++);
- }
- HeapFree(GetProcessHeap(), 0, env);
- return 0;
- }
-
- HeapFree(GetProcessHeap(), 0, env);
-
- ret = value_from_string(name, value, formatted);
- if (newlen) HeapFree(GetProcessHeap(), 0, formatted);
- return ret;
-}
-
-static int setting_set_priority(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- if (! param) return -1;
-
- TCHAR *priority_string;
- int i;
- long error;
-
- if (value && value->string) priority_string = value->string;
- else if (default_value) priority_string = (TCHAR *) default_value;
- else {
- error = RegDeleteValue(key, name);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
- return -1;
- }
-
- for (i = 0; priority_strings[i]; i++) {
- if (! str_equiv(priority_strings[i], priority_string)) continue;
-
- if (default_value && str_equiv(priority_string, (TCHAR *) default_value)) {
- error = RegDeleteValue(key, name);
- if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
- print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
- return -1;
- }
-
- if (set_number(key, (TCHAR *) name, priority_index_to_constant(i))) return -1;
- return 1;
- }
-
- print_message(stderr, NSSM_MESSAGE_INVALID_PRIORITY, priority_string);
- for (i = 0; priority_strings[i]; i++) _ftprintf(stderr, _T("%s\n"), priority_strings[i]);
-
- return -1;
-}
-
-static int setting_get_priority(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- HKEY key = (HKEY) param;
- if (! param) return -1;
-
- unsigned long constant;
- switch (get_number(key, (TCHAR *) name, &constant, false)) {
- case 0: return value_from_string(name, value, (const TCHAR *) default_value);
- case -1: return -1;
- }
-
- return value_from_string(name, value, priority_strings[priority_constant_to_index(constant)]);
-}
-
-/* Functions to manage native service settings. */
-int native_set_description(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- TCHAR *description = 0;
- if (value) description = value->string;
- if (set_service_description(service_name, service_handle, description)) return -1;
-
- if (description && description[0]) return 1;
-
- return 0;
-}
-
-int native_get_description(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- TCHAR buffer[VALUE_LENGTH];
- if (get_service_description(service_name, service_handle, _countof(buffer), buffer)) return -1;
-
- if (buffer[0]) return value_from_string(name, value, buffer);
- value->string = 0;
-
- return 0;
-}
-
-int native_set_displayname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- TCHAR *displayname = 0;
- if (value && value->string) displayname = value->string;
- else displayname = (TCHAR *) service_name;
-
- if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, 0, 0, 0, displayname)) {
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
- return -1;
- }
-
- /*
- If the display name and service name differ only in case,
- ChangeServiceConfig() will return success but the display name will be
- set to the service name, NOT the value passed to the function.
- This appears to be a quirk of Windows rather than a bug here.
- */
- if (displayname != service_name && ! str_equiv(displayname, service_name)) return 1;
-
- return 0;
-}
-
-int native_get_displayname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);
- if (! qsc) return -1;
-
- int ret = value_from_string(name, value, qsc->lpDisplayName);
- HeapFree(GetProcessHeap(), 0, qsc);
-
- return ret;
-}
-
-int native_set_imagepath(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- /* It makes no sense to try to reset the image path. */
- if (! value || ! value->string) {
- print_message(stderr, NSSM_MESSAGE_NO_DEFAULT_VALUE, name);
- return -1;
- }
-
- if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, value->string, 0, 0, 0, 0, 0, 0)) {
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
- return -1;
- }
-
- return 1;
-}
-
-int native_get_imagepath(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);
- if (! qsc) return -1;
-
- int ret = value_from_string(name, value, qsc->lpBinaryPathName);
- HeapFree(GetProcessHeap(), 0, qsc);
-
- return ret;
-}
-
-int native_set_name(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- print_message(stderr, NSSM_MESSAGE_CANNOT_RENAME_SERVICE);
- return -1;
-}
-
-int native_get_name(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- return value_from_string(name, value, service_name);
-}
-
-int native_set_objectname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- /*
- Logical syntax is: nssm set <service> ObjectName <username> <password>
- That means the username is actually passed in the additional parameter.
- */
- bool localsystem = true;
- TCHAR *username = NSSM_LOCALSYSTEM_ACCOUNT;
- TCHAR *password = 0;
- if (additional) {
- if (! str_equiv(additional, NSSM_LOCALSYSTEM_ACCOUNT)) {
- localsystem = false;
- username = (TCHAR *) additional;
- if (value && value->string) password = value->string;
- else {
- /* We need a password if the account is not LOCALSYSTEM. */
- print_message(stderr, NSSM_MESSAGE_MISSING_PASSWORD, name);
- return -1;
- }
- }
- }
-
- /*
- ChangeServiceConfig() will fail to set the username if the service is set
- to interact with the desktop.
- */
- unsigned long type = SERVICE_NO_CHANGE;
- if (! localsystem) {
- QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);
- if (! qsc) {
- if (password) SecureZeroMemory(password, _tcslen(password) * sizeof(TCHAR));
- return -1;
- }
-
- type = qsc->dwServiceType & ~SERVICE_INTERACTIVE_PROCESS;
- HeapFree(GetProcessHeap(), 0, qsc);
- }
-
- if (grant_logon_as_service(username)) {
- if (password) SecureZeroMemory(password, _tcslen(password) * sizeof(TCHAR));
- print_message(stderr, NSSM_MESSAGE_GRANT_LOGON_AS_SERVICE_FAILED, username);
- return -1;
- }
-
- if (! ChangeServiceConfig(service_handle, type, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, 0, username, password, 0)) {
- if (password) SecureZeroMemory(password, _tcslen(password) * sizeof(TCHAR));
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
- return -1;
- }
- if (password) SecureZeroMemory(password, _tcslen(password) * sizeof(TCHAR));
-
- if (localsystem) return 0;
-
- return 1;
-}
-
-int native_get_objectname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);
- if (! qsc) return -1;
-
- int ret = value_from_string(name, value, qsc->lpServiceStartName);
- HeapFree(GetProcessHeap(), 0, qsc);
-
- return ret;
-}
-
-int native_set_startup(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- /* It makes no sense to try to reset the startup type. */
- if (! value || ! value->string) {
- print_message(stderr, NSSM_MESSAGE_NO_DEFAULT_VALUE, name);
- return -1;
- }
-
- /* Map NSSM_STARTUP_* constant to Windows SERVICE_*_START constant. */
- int service_startup = -1;
- int i;
- for (i = 0; startup_strings[i]; i++) {
- if (str_equiv(value->string, startup_strings[i])) {
- service_startup = i;
- break;
- }
- }
-
- if (service_startup < 0) {
- print_message(stderr, NSSM_MESSAGE_INVALID_SERVICE_STARTUP, value->string);
- for (i = 0; startup_strings[i]; i++) _ftprintf(stderr, _T("%s\n"), startup_strings[i]);
- return -1;
- }
-
- unsigned long startup;
- switch (service_startup) {
- case NSSM_STARTUP_MANUAL: startup = SERVICE_DEMAND_START; break;
- case NSSM_STARTUP_DISABLED: startup = SERVICE_DISABLED; break;
- default: startup = SERVICE_AUTO_START;
- }
-
- if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, startup, SERVICE_NO_CHANGE, 0, 0, 0, 0, 0, 0, 0)) {
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
- return -1;
- }
-
- SERVICE_DELAYED_AUTO_START_INFO delayed;
- ZeroMemory(&delayed, sizeof(delayed));
- if (service_startup == NSSM_STARTUP_DELAYED) delayed.fDelayedAutostart = 1;
- else delayed.fDelayedAutostart = 0;
- if (! ChangeServiceConfig2(service_handle, SERVICE_CONFIG_DELAYED_AUTO_START_INFO, &delayed)) {
- unsigned long error = GetLastError();
- /* Pre-Vista we expect to fail with ERROR_INVALID_LEVEL */
- if (error != ERROR_INVALID_LEVEL) {
- log_event(EVENTLOG_ERROR_TYPE, NSSM_MESSAGE_SERVICE_CONFIG_DELAYED_AUTO_START_INFO_FAILED, service_name, error_string(error), 0);
- }
- }
-
- return 1;
-}
-
-int native_get_startup(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);
- if (! qsc) return -1;
-
- unsigned long startup;
- int ret = get_service_startup(service_name, service_handle, qsc, &startup);
- HeapFree(GetProcessHeap(), 0, qsc);
-
- if (ret) return -1;
-
- unsigned long i;
- for (i = 0; startup_strings[i]; i++);
- if (startup >= i) return -1;
-
- return value_from_string(name, value, startup_strings[startup]);
-}
-
-int native_set_type(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- /* It makes no sense to try to reset the service type. */
- if (! value || ! value->string) {
- print_message(stderr, NSSM_MESSAGE_NO_DEFAULT_VALUE, name);
- return -1;
- }
-
- /*
- We can only manage services of type SERVICE_WIN32_OWN_PROCESS
- and SERVICE_INTERACTIVE_PROCESS.
- */
- unsigned long type = SERVICE_WIN32_OWN_PROCESS;
- if (str_equiv(value->string, NSSM_INTERACTIVE_PROCESS)) type |= SERVICE_INTERACTIVE_PROCESS;
- else if (! str_equiv(value->string, NSSM_WIN32_OWN_PROCESS)) {
- print_message(stderr, NSSM_MESSAGE_INVALID_SERVICE_TYPE, value->string);
- _ftprintf(stderr, _T("%s\n"), NSSM_WIN32_OWN_PROCESS);
- _ftprintf(stderr, _T("%s\n"), NSSM_INTERACTIVE_PROCESS);
- return -1;
- }
-
- /*
- ChangeServiceConfig() will fail if the service runs under an account
- other than LOCALSYSTEM and we try to make it interactive.
- */
- if (type & SERVICE_INTERACTIVE_PROCESS) {
- QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);
- if (! qsc) return -1;
-
- if (! str_equiv(qsc->lpServiceStartName, NSSM_LOCALSYSTEM_ACCOUNT)) {
- HeapFree(GetProcessHeap(), 0, qsc);
- print_message(stderr, NSSM_MESSAGE_INTERACTIVE_NOT_LOCALSYSTEM, value->string, service_name, NSSM_LOCALSYSTEM_ACCOUNT);
- return -1;
- }
-
- HeapFree(GetProcessHeap(), 0, qsc);
- }
-
- if (! ChangeServiceConfig(service_handle, type, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, 0, 0, 0, 0)) {
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
- return -1;
- }
-
- return 1;
-}
-
-int native_get_type(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
- SC_HANDLE service_handle = (SC_HANDLE) param;
- if (! service_handle) return -1;
-
- QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);
- if (! qsc) return -1;
-
- value->numeric = qsc->dwServiceType;
- HeapFree(GetProcessHeap(), 0, qsc);
-
- const TCHAR *string;
- switch (value->numeric) {
- case SERVICE_KERNEL_DRIVER: string = NSSM_KERNEL_DRIVER; break;
- case SERVICE_FILE_SYSTEM_DRIVER: string = NSSM_FILE_SYSTEM_DRIVER; break;
- case SERVICE_WIN32_OWN_PROCESS: string = NSSM_WIN32_OWN_PROCESS; break;
- case SERVICE_WIN32_SHARE_PROCESS: string = NSSM_WIN32_SHARE_PROCESS; break;
- case SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS: string = NSSM_INTERACTIVE_PROCESS; break;
- case SERVICE_WIN32_SHARE_PROCESS|SERVICE_INTERACTIVE_PROCESS: string = NSSM_SHARE_INTERACTIVE_PROCESS; break;
- default: string = NSSM_UNKNOWN;
- }
-
- return value_from_string(name, value, string);
-}
-
-int set_setting(const TCHAR *service_name, HKEY key, settings_t *setting, value_t *value, const TCHAR *additional) {
- if (! key) return -1;
- int ret;
-
- if (setting->set) ret = setting->set(service_name, (void *) key, setting->name, setting->default_value, value, additional);
- else ret = -1;
-
- if (! ret) print_message(stdout, NSSM_MESSAGE_RESET_SETTING, setting->name, service_name);
- else if (ret > 0) print_message(stdout, NSSM_MESSAGE_SET_SETTING, setting->name, service_name);
- else print_message(stderr, NSSM_MESSAGE_SET_SETTING_FAILED, setting->name, service_name);
-
- return ret;
-}
-
-int set_setting(const TCHAR *service_name, SC_HANDLE service_handle, settings_t *setting, value_t *value, const TCHAR *additional) {
- if (! service_handle) return -1;
-
- int ret;
- if (setting->set) ret = setting->set(service_name, service_handle, setting->name, setting->default_value, value, additional);
- else ret = -1;
-
- if (! ret) print_message(stdout, NSSM_MESSAGE_RESET_SETTING, setting->name, service_name);
- else if (ret > 0) print_message(stdout, NSSM_MESSAGE_SET_SETTING, setting->name, service_name);
- else print_message(stderr, NSSM_MESSAGE_SET_SETTING_FAILED, setting->name, service_name);
-
- return ret;
-}
-
-/*
- Returns: 1 if the value was retrieved.
- 0 if the default value was retrieved.
- -1 on error.
-*/
-int get_setting(const TCHAR *service_name, HKEY key, settings_t *setting, value_t *value, const TCHAR *additional) {
- if (! key) return -1;
- int ret;
-
- switch (setting->type) {
- case REG_EXPAND_SZ:
- case REG_MULTI_SZ:
- case REG_SZ:
- value->string = (TCHAR *) setting->default_value;
- if (setting->get) ret = setting->get(service_name, (void *) key, setting->name, setting->default_value, value, additional);
- else ret = -1;
- break;
-
- case REG_DWORD:
- value->numeric = (unsigned long) setting->default_value;
- if (setting->get) ret = setting->get(service_name, (void *) key, setting->name, setting->default_value, value, additional);
- else ret = -1;
- break;
-
- default:
- ret = -1;
- break;
- }
-
- if (ret < 0) print_message(stderr, NSSM_MESSAGE_GET_SETTING_FAILED, setting->name, service_name);
-
- return ret;
-}
-
-int get_setting(const TCHAR *service_name, SC_HANDLE service_handle, settings_t *setting, value_t *value, const TCHAR *additional) {
- if (! service_handle) return -1;
- return setting->get(service_name, service_handle, setting->name, 0, value, additional);
-}
-
-settings_t settings[] = {
- { NSSM_REG_EXE, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string },
- { NSSM_REG_FLAGS, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string },
- { NSSM_REG_DIR, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string },
- { NSSM_REG_EXIT, REG_SZ, (void *) exit_action_strings[NSSM_EXIT_RESTART], false, ADDITIONAL_MANDATORY, setting_set_exit_action, setting_get_exit_action },
- { NSSM_REG_ENV, REG_MULTI_SZ, NULL, false, ADDITIONAL_CRLF, setting_set_environment, setting_get_environment },
- { NSSM_REG_ENV_EXTRA, REG_MULTI_SZ, NULL, false, ADDITIONAL_CRLF, setting_set_environment, setting_get_environment },
- { NSSM_REG_PRIORITY, REG_SZ, (void *) priority_strings[NSSM_NORMAL_PRIORITY], false, 0, setting_set_priority, setting_get_priority },
- { NSSM_REG_STDIN, REG_EXPAND_SZ, NULL, false, 0, setting_set_string, setting_get_string },
- { NSSM_REG_STDIN NSSM_REG_STDIO_SHARING, REG_DWORD, (void *) NSSM_STDIN_SHARING, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDIN NSSM_REG_STDIO_DISPOSITION, REG_DWORD, (void *) NSSM_STDIN_DISPOSITION, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDIN NSSM_REG_STDIO_FLAGS, REG_DWORD, (void *) NSSM_STDIN_FLAGS, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDOUT, REG_EXPAND_SZ, NULL, false, 0, setting_set_string, setting_get_string },
- { NSSM_REG_STDOUT NSSM_REG_STDIO_SHARING, REG_DWORD, (void *) NSSM_STDOUT_SHARING, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDOUT NSSM_REG_STDIO_DISPOSITION, REG_DWORD, (void *) NSSM_STDOUT_DISPOSITION, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDOUT NSSM_REG_STDIO_FLAGS, REG_DWORD, (void *) NSSM_STDOUT_FLAGS, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDERR, REG_EXPAND_SZ, NULL, false, 0, setting_set_string, setting_get_string },
- { NSSM_REG_STDERR NSSM_REG_STDIO_SHARING, REG_DWORD, (void *) NSSM_STDERR_SHARING, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDERR NSSM_REG_STDIO_DISPOSITION, REG_DWORD, (void *) NSSM_STDERR_DISPOSITION, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STDERR NSSM_REG_STDIO_FLAGS, REG_DWORD, (void *) NSSM_STDERR_FLAGS, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_STOP_METHOD_SKIP, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_KILL_CONSOLE_GRACE_PERIOD, REG_DWORD, (void *) NSSM_KILL_CONSOLE_GRACE_PERIOD, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_KILL_WINDOW_GRACE_PERIOD, REG_DWORD, (void *) NSSM_KILL_WINDOW_GRACE_PERIOD, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_KILL_THREADS_GRACE_PERIOD, REG_DWORD, (void *) NSSM_KILL_THREADS_GRACE_PERIOD, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_THROTTLE, REG_DWORD, (void *) NSSM_RESET_THROTTLE_RESTART, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_ROTATE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_ROTATE_SECONDS, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_ROTATE_BYTES_LOW, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
- { NSSM_REG_ROTATE_BYTES_HIGH, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
- { NSSM_NATIVE_DESCRIPTION, REG_SZ, _T(""), true, 0, native_set_description, native_get_description },
- { NSSM_NATIVE_DISPLAYNAME, REG_SZ, NULL, true, 0, native_set_displayname, native_get_displayname },
- { NSSM_NATIVE_IMAGEPATH, REG_EXPAND_SZ, NULL, true, 0, native_set_imagepath, native_get_imagepath },
- { NSSM_NATIVE_OBJECTNAME, REG_SZ, NSSM_LOCALSYSTEM_ACCOUNT, true, ADDITIONAL_SETTING, native_set_objectname, native_get_objectname },
- { NSSM_NATIVE_NAME, REG_SZ, NULL, true, 0, native_set_name, native_get_name },
- { NSSM_NATIVE_STARTUP, REG_SZ, NULL, true, 0, native_set_startup, native_get_startup },
- { NSSM_NATIVE_TYPE, REG_SZ, NULL, true, 0, native_set_type, native_get_type },
- { NULL, NULL, NULL, NULL, NULL }
-};
+#include "nssm.h"\r
+/* XXX: (value && value->string) is probably bogus because value is probably never null */\r
+\r
+/* Affinity. */\r
+#define NSSM_AFFINITY_ALL _T("All")\r
+/* Default value. */\r
+#define NSSM_DEFAULT_STRING _T("Default")\r
+\r
+extern const TCHAR *exit_action_strings[];\r
+extern const TCHAR *startup_strings[];\r
+extern const TCHAR *priority_strings[];\r
+extern const TCHAR *hook_event_strings[];\r
+extern const TCHAR *hook_action_strings[];\r
+\r
+/* Does the parameter refer to the default value of the setting? */\r
+static inline int is_default(const TCHAR *value) {\r
+ return (str_equiv(value, NSSM_DEFAULT_STRING) || str_equiv(value, _T("*")) || ! value[0]);\r
+}\r
+\r
+/* What type of parameter is it parameter? */\r
+static inline bool is_string_type(const unsigned long type) {\r
+ return (type == REG_MULTI_SZ || type == REG_EXPAND_SZ || type == REG_SZ);\r
+}\r
+static inline bool is_numeric_type(const unsigned long type) {\r
+ return (type == REG_DWORD);\r
+}\r
+\r
+static int value_from_string(const TCHAR *name, value_t *value, const TCHAR *string) {\r
+ size_t len = _tcslen(string);\r
+ if (! len++) {\r
+ value->string = 0;\r
+ return 0;\r
+ }\r
+\r
+ value->string = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, len * sizeof(TCHAR));\r
+ if (! value->string) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, name, _T("value_from_string()"));\r
+ return -1;\r
+ }\r
+\r
+ if (_sntprintf_s(value->string, len, _TRUNCATE, _T("%s"), string) < 0) {\r
+ HeapFree(GetProcessHeap(), 0, value->string);\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, name, _T("value_from_string()"));\r
+ return -1;\r
+ }\r
+\r
+ return 1;\r
+}\r
+\r
+/* Functions to manage NSSM-specific settings in the registry. */\r
+static int setting_set_number(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! key) return -1;\r
+\r
+ unsigned long number;\r
+ long error;\r
+\r
+ /* Resetting to default? */\r
+ if (! value || ! value->string) {\r
+ error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+ if (str_number(value->string, &number)) return -1;\r
+\r
+ if (default_value && number == PtrToUlong(default_value)) {\r
+ error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+\r
+ if (set_number(key, (TCHAR *) name, number)) return -1;\r
+\r
+ return 1;\r
+}\r
+\r
+static int setting_get_number(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ return get_number(key, (TCHAR *) name, &value->numeric, false);\r
+}\r
+\r
+static int setting_set_string(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! key) return -1;\r
+\r
+ long error;\r
+\r
+ /* Resetting to default? */\r
+ if (! value || ! value->string) {\r
+ if (default_value) value->string = (TCHAR *) default_value;\r
+ else {\r
+ error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+ }\r
+ if (default_value && _tcslen((TCHAR *) default_value) && str_equiv(value->string, (TCHAR *) default_value)) {\r
+ error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+\r
+ if (set_expand_string(key, (TCHAR *) name, value->string)) return -1;\r
+\r
+ return 1;\r
+}\r
+\r
+static int setting_get_string(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ TCHAR buffer[VALUE_LENGTH];\r
+\r
+ if (get_string(key, (TCHAR *) name, (TCHAR *) buffer, (unsigned long) sizeof(buffer), false, false, false)) return -1;\r
+\r
+ return value_from_string(name, value, buffer);\r
+}\r
+\r
+static int setting_not_dumpable(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ return 0;\r
+}\r
+\r
+static int setting_dump_string(const TCHAR *service_name, void *param, const TCHAR *name, const value_t *value, const TCHAR *additional) {\r
+ TCHAR quoted_service_name[SERVICE_NAME_LENGTH * 2];\r
+ TCHAR quoted_value[VALUE_LENGTH * 2];\r
+ TCHAR quoted_additional[VALUE_LENGTH * 2];\r
+ TCHAR quoted_nssm[EXE_LENGTH * 2];\r
+\r
+ if (quote(service_name, quoted_service_name, _countof(quoted_service_name))) return 1;\r
+\r
+ if (additional) {\r
+ if (_tcslen(additional)) {\r
+ if (quote(additional, quoted_additional, _countof(quoted_additional))) return 3;\r
+ }\r
+ else _sntprintf_s(quoted_additional, _countof(quoted_additional), _TRUNCATE, _T("\"\""));\r
+ }\r
+ else quoted_additional[0] = _T('\0');\r
+\r
+ unsigned long type = (unsigned long) param;\r
+ if (is_string_type(type)) {\r
+ if (_tcslen(value->string)) {\r
+ if (quote(value->string, quoted_value, _countof(quoted_value))) return 2;\r
+ }\r
+ else _sntprintf_s(quoted_value, _countof(quoted_value), _TRUNCATE, _T("\"\""));\r
+ }\r
+ else if (is_numeric_type(type)) _sntprintf_s(quoted_value, _countof(quoted_value), _TRUNCATE, _T("%lu"), value->numeric);\r
+ else return 2;\r
+\r
+ if (quote(nssm_exe(), quoted_nssm, _countof(quoted_nssm))) return 3;\r
+ if (_tcslen(quoted_additional)) _tprintf(_T("%s set %s %s %s %s\n"), quoted_nssm, quoted_service_name, name, quoted_additional, quoted_value);\r
+ else _tprintf(_T("%s set %s %s %s\n"), quoted_nssm, quoted_service_name, name, quoted_value);\r
+ return 0;\r
+}\r
+\r
+static int setting_set_exit_action(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ unsigned long exitcode;\r
+ TCHAR *code;\r
+ TCHAR action_string[ACTION_LEN];\r
+\r
+ if (additional) {\r
+ /* Default action? */\r
+ if (is_default(additional)) code = 0;\r
+ else {\r
+ if (str_number(additional, &exitcode)) return -1;\r
+ code = (TCHAR *) additional;\r
+ }\r
+ }\r
+\r
+ HKEY key = open_registry(service_name, name, KEY_WRITE);\r
+ if (! key) return -1;\r
+\r
+ long error;\r
+ int ret = 1;\r
+\r
+ /* Resetting to default? */\r
+ if (value && value->string) _sntprintf_s(action_string, _countof(action_string), _TRUNCATE, _T("%s"), value->string);\r
+ else {\r
+ if (code) {\r
+ /* Delete explicit action. */\r
+ error = RegDeleteValue(key, code);\r
+ RegCloseKey(key);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, code, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+ else {\r
+ /* Explicitly keep the default action. */\r
+ if (default_value) _sntprintf_s(action_string, _countof(action_string), _TRUNCATE, _T("%s"), (TCHAR *) default_value);\r
+ ret = 0;\r
+ }\r
+ }\r
+\r
+ /* Validate the string. */\r
+ for (int i = 0; exit_action_strings[i]; i++) {\r
+ if (! _tcsnicmp((const TCHAR *) action_string, exit_action_strings[i], ACTION_LEN)) {\r
+ if (default_value && str_equiv(action_string, (TCHAR *) default_value)) ret = 0;\r
+ if (RegSetValueEx(key, code, 0, REG_SZ, (const unsigned char *) exit_action_strings[i], (unsigned long) (_tcslen(action_string) + 1) * sizeof(TCHAR)) != ERROR_SUCCESS) {\r
+ print_message(stderr, NSSM_MESSAGE_SETVALUE_FAILED, code, service_name, error_string(GetLastError()));\r
+ RegCloseKey(key);\r
+ return -1;\r
+ }\r
+\r
+ RegCloseKey(key);\r
+ return ret;\r
+ }\r
+ }\r
+\r
+ print_message(stderr, NSSM_MESSAGE_INVALID_EXIT_ACTION, action_string);\r
+ for (int i = 0; exit_action_strings[i]; i++) _ftprintf(stderr, _T("%s\n"), exit_action_strings[i]);\r
+\r
+ return -1;\r
+}\r
+\r
+static int setting_get_exit_action(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ unsigned long exitcode = 0;\r
+ unsigned long *code = 0;\r
+\r
+ if (additional) {\r
+ if (! is_default(additional)) {\r
+ if (str_number(additional, &exitcode)) return -1;\r
+ code = &exitcode;\r
+ }\r
+ }\r
+\r
+ TCHAR action_string[ACTION_LEN];\r
+ bool default_action;\r
+ if (get_exit_action(service_name, code, action_string, &default_action)) return -1;\r
+\r
+ value_from_string(name, value, action_string);\r
+\r
+ if (default_action && ! _tcsnicmp((const TCHAR *) action_string, (TCHAR *) default_value, ACTION_LEN)) return 0;\r
+ return 1;\r
+}\r
+\r
+static int setting_dump_exit_action(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ int errors = 0;\r
+ HKEY key = open_registry(service_name, NSSM_REG_EXIT, KEY_READ);\r
+ if (! key) return -1;\r
+\r
+ TCHAR code[16];\r
+ unsigned long index = 0;\r
+ while (true) {\r
+ int ret = enumerate_registry_values(key, &index, code, _countof(code));\r
+ if (ret == ERROR_NO_MORE_ITEMS) break;\r
+ if (ret != ERROR_SUCCESS) continue;\r
+ bool valid = true;\r
+ int i;\r
+ for (i = 0; i < _countof(code); i++) {\r
+ if (! code[i]) break;\r
+ if (code[i] >= _T('0') || code[i] <= _T('9')) continue;\r
+ valid = false;\r
+ break;\r
+ }\r
+ if (! valid) continue;\r
+\r
+ TCHAR *additional = (code[i]) ? code : NSSM_DEFAULT_STRING;\r
+\r
+ ret = setting_get_exit_action(service_name, 0, name, default_value, value, additional);\r
+ if (ret == 1) {\r
+ if (setting_dump_string(service_name, (void *) REG_SZ, name, value, additional)) errors++;\r
+ }\r
+ else if (ret < 0) errors++;\r
+ }\r
+\r
+ if (errors) return -1;\r
+ return 0;\r
+}\r
+\r
+static inline bool split_hook_name(const TCHAR *hook_name, TCHAR *hook_event, TCHAR *hook_action) {\r
+ TCHAR *s;\r
+\r
+ for (s = (TCHAR *) hook_name; *s; s++) {\r
+ if (*s == _T('/')) {\r
+ *s = _T('\0');\r
+ _sntprintf_s(hook_event, HOOK_NAME_LENGTH, _TRUNCATE, _T("%s"), hook_name);\r
+ *s++ = _T('/');\r
+ _sntprintf_s(hook_action, HOOK_NAME_LENGTH, _TRUNCATE, _T("%s"), s);\r
+ return valid_hook_name(hook_event, hook_action, false);\r
+ }\r
+ }\r
+\r
+ print_message(stderr, NSSM_MESSAGE_INVALID_HOOK_NAME, hook_name);\r
+ return false;\r
+}\r
+\r
+static int setting_set_hook(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ TCHAR hook_event[HOOK_NAME_LENGTH];\r
+ TCHAR hook_action[HOOK_NAME_LENGTH];\r
+ if (! split_hook_name(additional, hook_event, hook_action)) return -1;\r
+\r
+ TCHAR *cmd;\r
+ if (value && value->string) cmd = value->string;\r
+ else cmd = _T("");\r
+\r
+ if (set_hook(service_name, hook_event, hook_action, cmd)) return -1;\r
+ if (! _tcslen(cmd)) return 0;\r
+ return 1;\r
+}\r
+\r
+static int setting_get_hook(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ TCHAR hook_event[HOOK_NAME_LENGTH];\r
+ TCHAR hook_action[HOOK_NAME_LENGTH];\r
+ if (! split_hook_name(additional, hook_event, hook_action)) return -1;\r
+\r
+ TCHAR cmd[CMD_LENGTH];\r
+ if (get_hook(service_name, hook_event, hook_action, cmd, sizeof(cmd))) return -1;\r
+\r
+ value_from_string(name, value, cmd);\r
+\r
+ if (! _tcslen(cmd)) return 0;\r
+ return 1;\r
+}\r
+\r
+static int setting_dump_hooks(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ int i, j;\r
+\r
+ int errors = 0;\r
+ for (i = 0; hook_event_strings[i]; i++) {\r
+ const TCHAR *hook_event = hook_event_strings[i];\r
+ for (j = 0; hook_action_strings[j]; j++) {\r
+ const TCHAR *hook_action = hook_action_strings[j];\r
+ if (! valid_hook_name(hook_event, hook_action, true)) continue;\r
+\r
+ TCHAR hook_name[HOOK_NAME_LENGTH];\r
+ _sntprintf_s(hook_name, _countof(hook_name), _TRUNCATE, _T("%s/%s"), hook_event, hook_action);\r
+\r
+ int ret = setting_get_hook(service_name, param, name, default_value, value, hook_name);\r
+ if (ret != 1) {\r
+ if (ret < 0) errors++;\r
+ continue;\r
+ }\r
+\r
+ if (setting_dump_string(service_name, (void *) REG_SZ, name, value, hook_name)) errors++;\r
+ }\r
+ }\r
+\r
+ if (errors) return -1;\r
+ return 0;\r
+}\r
+\r
+static int setting_set_affinity(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! key) return -1;\r
+\r
+ long error;\r
+ __int64 mask;\r
+ __int64 system_affinity = 0LL;\r
+\r
+ if (value && value->string) {\r
+ DWORD_PTR affinity;\r
+ if (! GetProcessAffinityMask(GetCurrentProcess(), &affinity, (DWORD_PTR *) &system_affinity)) system_affinity = ~0;\r
+\r
+ if (is_default(value->string) || str_equiv(value->string, NSSM_AFFINITY_ALL)) mask = 0LL;\r
+ else if (affinity_string_to_mask(value->string, &mask)) {\r
+ print_message(stderr, NSSM_MESSAGE_BOGUS_AFFINITY_MASK, value->string, num_cpus() - 1);\r
+ return -1;\r
+ }\r
+ }\r
+ else mask = 0LL;\r
+\r
+ if (! mask) {\r
+ error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+\r
+ /* Canonicalise. */\r
+ TCHAR *canon = 0;\r
+ if (affinity_mask_to_string(mask, &canon)) canon = value->string;\r
+\r
+ __int64 effective_affinity = mask & system_affinity;\r
+ if (effective_affinity != mask) {\r
+ /* Requested CPUs did not intersect with available CPUs? */\r
+ if (! effective_affinity) mask = effective_affinity = system_affinity;\r
+\r
+ TCHAR *system = 0;\r
+ if (! affinity_mask_to_string(system_affinity, &system)) {\r
+ TCHAR *effective = 0;\r
+ if (! affinity_mask_to_string(effective_affinity, &effective)) {\r
+ print_message(stderr, NSSM_MESSAGE_EFFECTIVE_AFFINITY_MASK, value->string, system, effective);\r
+ HeapFree(GetProcessHeap(), 0, effective);\r
+ }\r
+ HeapFree(GetProcessHeap(), 0, system);\r
+ }\r
+ }\r
+\r
+ if (RegSetValueEx(key, name, 0, REG_SZ, (const unsigned char *) canon, (unsigned long) (_tcslen(canon) + 1) * sizeof(TCHAR)) != ERROR_SUCCESS) {\r
+ if (canon != value->string) HeapFree(GetProcessHeap(), 0, canon);\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, name, error_string(GetLastError()), 0);\r
+ return -1;\r
+ }\r
+\r
+ if (canon != value->string) HeapFree(GetProcessHeap(), 0, canon);\r
+ return 1;\r
+}\r
+\r
+static int setting_get_affinity(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! key) return -1;\r
+\r
+ unsigned long type;\r
+ TCHAR *buffer = 0;\r
+ unsigned long buflen = 0;\r
+\r
+ int ret = RegQueryValueEx(key, name, 0, &type, 0, &buflen);\r
+ if (ret == ERROR_FILE_NOT_FOUND) {\r
+ if (value_from_string(name, value, NSSM_AFFINITY_ALL) == 1) return 0;\r
+ return -1;\r
+ }\r
+ if (ret != ERROR_SUCCESS) return -1;\r
+\r
+ if (type != REG_SZ) return -1;\r
+\r
+ buffer = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, buflen);\r
+ if (! buffer) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("affinity"), _T("setting_get_affinity"));\r
+ return -1;\r
+ }\r
+\r
+ if (get_string(key, (TCHAR *) name, buffer, buflen, false, false, true)) {\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+ return -1;\r
+ }\r
+\r
+ __int64 affinity;\r
+ if (affinity_string_to_mask(buffer, &affinity)) {\r
+ print_message(stderr, NSSM_MESSAGE_BOGUS_AFFINITY_MASK, buffer, num_cpus() - 1);\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+ return -1;\r
+ }\r
+\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+\r
+ /* Canonicalise. */\r
+ if (affinity_mask_to_string(affinity, &buffer)) {\r
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);\r
+ return -1;\r
+ }\r
+\r
+ ret = value_from_string(name, value, buffer);\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+ return ret;\r
+}\r
+\r
+static int setting_set_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! param) return -1;\r
+\r
+ TCHAR *string = 0;\r
+ TCHAR *unformatted = 0;\r
+ unsigned long envlen;\r
+ unsigned long newlen = 0;\r
+ int op = 0;\r
+ if (value && value->string && value->string[0]) {\r
+ string = value->string;\r
+ switch (string[0]) {\r
+ case _T('+'): op = 1; break;\r
+ case _T('-'): op = -1; break;\r
+ case _T(':'): string++; break;\r
+ }\r
+ }\r
+\r
+ if (op) {\r
+ string++;\r
+ TCHAR *env = 0;\r
+ if (get_environment((TCHAR *) service_name, key, (TCHAR *) name, &env, &envlen)) return -1;\r
+ if (env) {\r
+ int ret;\r
+ if (op > 0) ret = append_to_environment_block(env, envlen, string, &unformatted, &newlen);\r
+ else ret = remove_from_environment_block(env, envlen, string, &unformatted, &newlen);\r
+ if (envlen) HeapFree(GetProcessHeap(), 0, env);\r
+ if (ret) return -1;\r
+\r
+ string = unformatted;\r
+ }\r
+ else {\r
+ /*\r
+ No existing environment.\r
+ We can't remove from an empty environment so just treat an add\r
+ operation as setting a new string.\r
+ */\r
+ if (op < 0) return 0;\r
+ op = 0;\r
+ }\r
+ }\r
+\r
+ if (! string || ! string[0]) {\r
+ long error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+\r
+ if (! op) {\r
+ if (unformat_double_null(string, (unsigned long) _tcslen(string), &unformatted, &newlen)) return -1;\r
+ }\r
+\r
+ if (test_environment(unformatted)) {\r
+ HeapFree(GetProcessHeap(), 0, unformatted);\r
+ print_message(stderr, NSSM_GUI_INVALID_ENVIRONMENT);\r
+ return -1;\r
+ }\r
+\r
+ if (RegSetValueEx(key, name, 0, REG_MULTI_SZ, (const unsigned char *) unformatted, (unsigned long) newlen * sizeof(TCHAR)) != ERROR_SUCCESS) {\r
+ if (newlen) HeapFree(GetProcessHeap(), 0, unformatted);\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, NSSM_REG_ENV, error_string(GetLastError()), 0);\r
+ return -1;\r
+ }\r
+\r
+ if (newlen) HeapFree(GetProcessHeap(), 0, unformatted);\r
+ return 1;\r
+}\r
+\r
+static int setting_get_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! param) return -1;\r
+\r
+ TCHAR *env = 0;\r
+ unsigned long envlen;\r
+ if (get_environment((TCHAR *) service_name, key, (TCHAR *) name, &env, &envlen)) return -1;\r
+ if (! envlen) return 0;\r
+\r
+ TCHAR *formatted;\r
+ unsigned long newlen;\r
+ if (format_double_null(env, envlen, &formatted, &newlen)) return -1;\r
+\r
+ int ret;\r
+ if (additional) {\r
+ /* Find named environment variable. */\r
+ TCHAR *s;\r
+ size_t len = _tcslen(additional);\r
+ for (s = env; *s; s++) {\r
+ /* Look for <additional>=<string> NULL NULL */\r
+ if (! _tcsnicmp(s, additional, len) && s[len] == _T('=')) {\r
+ /* Strip <key>= */\r
+ s += len + 1;\r
+ ret = value_from_string(name, value, s);\r
+ HeapFree(GetProcessHeap(), 0, env);\r
+ return ret;\r
+ }\r
+\r
+ /* Skip this string. */\r
+ for ( ; *s; s++);\r
+ }\r
+ HeapFree(GetProcessHeap(), 0, env);\r
+ return 0;\r
+ }\r
+\r
+ HeapFree(GetProcessHeap(), 0, env);\r
+\r
+ ret = value_from_string(name, value, formatted);\r
+ if (newlen) HeapFree(GetProcessHeap(), 0, formatted);\r
+ return ret;\r
+}\r
+\r
+static int setting_dump_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ int errors = 0;\r
+ HKEY key = (HKEY) param;\r
+ if (! param) return -1;\r
+\r
+ TCHAR *env = 0;\r
+ unsigned long envlen;\r
+ if (get_environment((TCHAR *) service_name, key, (TCHAR *) name, &env, &envlen)) return -1;\r
+ if (! envlen) return 0;\r
+\r
+ TCHAR *s;\r
+ for (s = env; *s; s++) {\r
+ size_t len = _tcslen(s) + 2;\r
+ value->string = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, len * sizeof(TCHAR));\r
+ if (! value->string) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dump"), _T("setting_dump_environment"));\r
+ break;\r
+ }\r
+\r
+ _sntprintf_s(value->string, len, _TRUNCATE, _T("%c%s"), (s > env) ? _T('+') : _T(':'), s);\r
+ if (setting_dump_string(service_name, (void *) REG_SZ, name, value, 0)) errors++;\r
+ HeapFree(GetProcessHeap(), 0, value->string);\r
+ value->string = 0;\r
+\r
+ for ( ; *s; s++);\r
+ }\r
+\r
+ HeapFree(GetProcessHeap(), 0, env);\r
+\r
+ if (errors) return 1;\r
+ return 0;\r
+}\r
+\r
+static int setting_set_priority(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! param) return -1;\r
+\r
+ TCHAR *priority_string;\r
+ int i;\r
+ long error;\r
+\r
+ if (value && value->string) priority_string = value->string;\r
+ else if (default_value) priority_string = (TCHAR *) default_value;\r
+ else {\r
+ error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+\r
+ for (i = 0; priority_strings[i]; i++) {\r
+ if (! str_equiv(priority_strings[i], priority_string)) continue;\r
+\r
+ if (default_value && str_equiv(priority_string, (TCHAR *) default_value)) {\r
+ error = RegDeleteValue(key, name);\r
+ if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;\r
+ print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));\r
+ return -1;\r
+ }\r
+\r
+ if (set_number(key, (TCHAR *) name, priority_index_to_constant(i))) return -1;\r
+ return 1;\r
+ }\r
+\r
+ print_message(stderr, NSSM_MESSAGE_INVALID_PRIORITY, priority_string);\r
+ for (i = 0; priority_strings[i]; i++) _ftprintf(stderr, _T("%s\n"), priority_strings[i]);\r
+\r
+ return -1;\r
+}\r
+\r
+static int setting_get_priority(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = (HKEY) param;\r
+ if (! param) return -1;\r
+\r
+ unsigned long constant;\r
+ switch (get_number(key, (TCHAR *) name, &constant, false)) {\r
+ case 0:\r
+ if (value_from_string(name, value, (const TCHAR *) default_value) == -1) return -1;\r
+ return 0;\r
+ case -1: return -1;\r
+ }\r
+\r
+ return value_from_string(name, value, priority_strings[priority_constant_to_index(constant)]);\r
+}\r
+\r
+static int setting_dump_priority(const TCHAR *service_name, void *key_ptr, const TCHAR *name, void *setting_ptr, value_t *value, const TCHAR *additional) {\r
+ settings_t *setting = (settings_t *) setting_ptr;\r
+ int ret = setting_get_priority(service_name, key_ptr, name, (void *) setting->default_value, value, 0);\r
+ if (ret != 1) return ret;\r
+ return setting_dump_string(service_name, (void *) REG_SZ, name, value, 0);\r
+}\r
+\r
+/* Functions to manage native service settings. */\r
+static int native_set_dependon(const TCHAR *service_name, SC_HANDLE service_handle, TCHAR **dependencies, unsigned long *dependencieslen, value_t *value, int type) {\r
+ *dependencieslen = 0;\r
+ if (! value || ! value->string || ! value->string[0]) return 0;\r
+\r
+ TCHAR *string = value->string;\r
+ unsigned long buflen;\r
+ int op = 0;\r
+ switch (string[0]) {\r
+ case _T('+'): op = 1; break;\r
+ case _T('-'): op = -1; break;\r
+ case _T(':'): string++; break;\r
+ }\r
+\r
+ if (op) {\r
+ string++;\r
+ TCHAR *buffer = 0;\r
+ if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, type)) return -1;\r
+ if (buffer) {\r
+ int ret;\r
+ if (op > 0) ret = append_to_dependencies(buffer, buflen, string, dependencies, dependencieslen, type);\r
+ else ret = remove_from_dependencies(buffer, buflen, string, dependencies, dependencieslen, type);\r
+ if (buflen) HeapFree(GetProcessHeap(), 0, buffer);\r
+ return ret;\r
+ }\r
+ else {\r
+ /*\r
+ No existing list.\r
+ We can't remove from an empty list so just treat an add\r
+ operation as setting a new string.\r
+ */\r
+ if (op < 0) return 0;\r
+ op = 0;\r
+ }\r
+ }\r
+\r
+ if (! op) {\r
+ TCHAR *unformatted = 0;\r
+ unsigned long newlen;\r
+ if (unformat_double_null(string, (unsigned long) _tcslen(string), &unformatted, &newlen)) return -1;\r
+\r
+ if (type == DEPENDENCY_GROUPS) {\r
+ /* Prepend group identifier. */\r
+ unsigned long missing = 0;\r
+ TCHAR *canon = unformatted;\r
+ size_t canonlen = 0;\r
+ TCHAR *s;\r
+ for (s = unformatted; *s; s++) {\r
+ if (*s != SC_GROUP_IDENTIFIER) missing++;\r
+ size_t len = _tcslen(s);\r
+ canonlen += len + 1;\r
+ s += len;\r
+ }\r
+\r
+ if (missing) {\r
+ /* Missing identifiers plus double NULL terminator. */\r
+ canonlen += missing + 1;\r
+\r
+ canon = (TCHAR *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, canonlen * sizeof(TCHAR));\r
+ if (! canon) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("canon"), _T("native_set_dependon"));\r
+ if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);\r
+ return -1;\r
+ }\r
+\r
+ size_t i = 0;\r
+ for (s = unformatted; *s; s++) {\r
+ if (*s != SC_GROUP_IDENTIFIER) canon[i++] = SC_GROUP_IDENTIFIER;\r
+ size_t len = _tcslen(s);\r
+ memmove(canon + i, s, (len + 1) * sizeof(TCHAR));\r
+ i += len + 1;\r
+ s += len;\r
+ }\r
+\r
+ HeapFree(GetProcessHeap(), 0, unformatted);\r
+ unformatted = canon;\r
+ newlen = (unsigned long) canonlen;\r
+ }\r
+ }\r
+\r
+ *dependencies = unformatted;\r
+ *dependencieslen = newlen;\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+static int native_set_dependongroup(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ /*\r
+ Get existing service dependencies because we must set both types together.\r
+ */\r
+ TCHAR *services_buffer;\r
+ unsigned long services_buflen;\r
+ if (get_service_dependencies(service_name, service_handle, &services_buffer, &services_buflen, DEPENDENCY_SERVICES)) return -1;\r
+\r
+ if (! value || ! value->string || ! value->string[0]) {\r
+ if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, services_buffer, 0, 0, 0)) {\r
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));\r
+ if (services_buffer) HeapFree(GetProcessHeap(), 0, services_buffer);\r
+ return -1;\r
+ }\r
+\r
+ if (services_buffer) HeapFree(GetProcessHeap(), 0, services_buffer);\r
+ return 0;\r
+ }\r
+\r
+ /* Update the group list. */\r
+ TCHAR *groups_buffer;\r
+ unsigned long groups_buflen;\r
+ if (native_set_dependon(service_name, service_handle, &groups_buffer, &groups_buflen, value, DEPENDENCY_GROUPS)) return -1;\r
+\r
+ TCHAR *dependencies;\r
+ if (services_buflen > 2) {\r
+ dependencies = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (groups_buflen + services_buflen) * sizeof(TCHAR));\r
+ if (! dependencies) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dependencies"), _T("native_set_dependongroup"));\r
+ if (groups_buffer) HeapFree(GetProcessHeap(), 0, groups_buffer);\r
+ if (services_buffer) HeapFree(GetProcessHeap(), 0, services_buffer);\r
+ return -1;\r
+ }\r
+\r
+ memmove(dependencies, services_buffer, services_buflen * sizeof(TCHAR));\r
+ memmove(dependencies + services_buflen - 1, groups_buffer, groups_buflen * sizeof(TCHAR));\r
+ }\r
+ else dependencies = groups_buffer;\r
+\r
+ int ret = 1;\r
+ if (set_service_dependencies(service_name, service_handle, dependencies)) ret = -1;\r
+ if (dependencies != groups_buffer) HeapFree(GetProcessHeap(), 0, dependencies);\r
+ if (groups_buffer) HeapFree(GetProcessHeap(), 0, groups_buffer);\r
+ if (services_buffer) HeapFree(GetProcessHeap(), 0, services_buffer);\r
+\r
+ return ret;\r
+}\r
+\r
+static int native_get_dependongroup(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ TCHAR *buffer;\r
+ unsigned long buflen;\r
+ if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_GROUPS)) return -1;\r
+\r
+ int ret;\r
+ if (buflen) {\r
+ TCHAR *formatted;\r
+ unsigned long newlen;\r
+ if (format_double_null(buffer, buflen, &formatted, &newlen)) {\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+ return -1;\r
+ }\r
+\r
+ ret = value_from_string(name, value, formatted);\r
+ HeapFree(GetProcessHeap(), 0, formatted);\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+ }\r
+ else {\r
+ value->string = 0;\r
+ ret = 0;\r
+ }\r
+\r
+ return ret;\r
+}\r
+\r
+static int setting_dump_dependon(const TCHAR *service_name, SC_HANDLE service_handle, const TCHAR *name, int type, value_t *value) {\r
+ int errors = 0;\r
+\r
+ TCHAR *dependencies = 0;\r
+ unsigned long dependencieslen;\r
+ if (get_service_dependencies(service_name, service_handle, &dependencies, &dependencieslen, type)) return -1;\r
+ if (! dependencieslen) return 0;\r
+\r
+ TCHAR *s;\r
+ for (s = dependencies; *s; s++) {\r
+ size_t len = _tcslen(s) + 2;\r
+ value->string = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, len * sizeof(TCHAR));\r
+ if (! value->string) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dump"), _T("setting_dump_dependon"));\r
+ break;\r
+ }\r
+\r
+ _sntprintf_s(value->string, len, _TRUNCATE, _T("%c%s"), (s > dependencies) ? _T('+') : _T(':'), s);\r
+ if (setting_dump_string(service_name, (void *) REG_SZ, name, value, 0)) errors++;\r
+ HeapFree(GetProcessHeap(), 0, value->string);\r
+ value->string = 0;\r
+\r
+ for ( ; *s; s++);\r
+ }\r
+\r
+ HeapFree(GetProcessHeap(), 0, dependencies);\r
+\r
+ if (errors) return 1;\r
+ return 0;\r
+}\r
+\r
+static int native_dump_dependongroup(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ return setting_dump_dependon(service_name, (SC_HANDLE) param, name, DEPENDENCY_GROUPS, value);\r
+}\r
+\r
+static int native_set_dependonservice(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ /*\r
+ Get existing group dependencies because we must set both types together.\r
+ */\r
+ TCHAR *groups_buffer;\r
+ unsigned long groups_buflen;\r
+ if (get_service_dependencies(service_name, service_handle, &groups_buffer, &groups_buflen, DEPENDENCY_GROUPS)) return -1;\r
+\r
+ if (! value || ! value->string || ! value->string[0]) {\r
+ if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, groups_buffer, 0, 0, 0)) {\r
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));\r
+ if (groups_buffer) HeapFree(GetProcessHeap(), 0, groups_buffer);\r
+ return -1;\r
+ }\r
+\r
+ if (groups_buffer) HeapFree(GetProcessHeap(), 0, groups_buffer);\r
+ return 0;\r
+ }\r
+\r
+ /* Update the service list. */\r
+ TCHAR *services_buffer;\r
+ unsigned long services_buflen;\r
+ if (native_set_dependon(service_name, service_handle, &services_buffer, &services_buflen, value, DEPENDENCY_SERVICES)) return -1;\r
+\r
+ TCHAR *dependencies;\r
+ if (groups_buflen > 2) {\r
+ dependencies = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (services_buflen + groups_buflen) * sizeof(TCHAR));\r
+ if (! dependencies) {\r
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dependencies"), _T("native_set_dependonservice"));\r
+ if (groups_buffer) HeapFree(GetProcessHeap(), 0, groups_buffer);\r
+ if (services_buffer) HeapFree(GetProcessHeap(), 0, services_buffer);\r
+ return -1;\r
+ }\r
+\r
+ memmove(dependencies, services_buffer, services_buflen * sizeof(TCHAR));\r
+ memmove(dependencies + services_buflen - 1, groups_buffer, groups_buflen * sizeof(TCHAR));\r
+ }\r
+ else dependencies = services_buffer;\r
+\r
+ int ret = 1;\r
+ if (set_service_dependencies(service_name, service_handle, dependencies)) ret = -1;\r
+ if (dependencies != services_buffer) HeapFree(GetProcessHeap(), 0, dependencies);\r
+ if (groups_buffer) HeapFree(GetProcessHeap(), 0, groups_buffer);\r
+ if (services_buffer) HeapFree(GetProcessHeap(), 0, services_buffer);\r
+\r
+ return ret;\r
+}\r
+\r
+static int native_get_dependonservice(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ TCHAR *buffer;\r
+ unsigned long buflen;\r
+ if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_SERVICES)) return -1;\r
+\r
+ int ret;\r
+ if (buflen) {\r
+ TCHAR *formatted;\r
+ unsigned long newlen;\r
+ if (format_double_null(buffer, buflen, &formatted, &newlen)) {\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+ return -1;\r
+ }\r
+\r
+ ret = value_from_string(name, value, formatted);\r
+ HeapFree(GetProcessHeap(), 0, formatted);\r
+ HeapFree(GetProcessHeap(), 0, buffer);\r
+ }\r
+ else {\r
+ value->string = 0;\r
+ ret = 0;\r
+ }\r
+\r
+ return ret;\r
+}\r
+\r
+static int native_dump_dependonservice(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ return setting_dump_dependon(service_name, (SC_HANDLE) param, name, DEPENDENCY_SERVICES, value);\r
+}\r
+\r
+int native_set_description(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ TCHAR *description = 0;\r
+ if (value) description = value->string;\r
+ if (set_service_description(service_name, service_handle, description)) return -1;\r
+\r
+ if (description && description[0]) return 1;\r
+\r
+ return 0;\r
+}\r
+\r
+int native_get_description(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ TCHAR buffer[VALUE_LENGTH];\r
+ if (get_service_description(service_name, service_handle, _countof(buffer), buffer)) return -1;\r
+\r
+ if (buffer[0]) return value_from_string(name, value, buffer);\r
+ value->string = 0;\r
+\r
+ return 0;\r
+}\r
+\r
+int native_set_displayname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ TCHAR *displayname = 0;\r
+ if (value && value->string) displayname = value->string;\r
+ else displayname = (TCHAR *) service_name;\r
+\r
+ if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, 0, 0, 0, displayname)) {\r
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));\r
+ return -1;\r
+ }\r
+\r
+ /*\r
+ If the display name and service name differ only in case,\r
+ ChangeServiceConfig() will return success but the display name will be\r
+ set to the service name, NOT the value passed to the function.\r
+ This appears to be a quirk of Windows rather than a bug here.\r
+ */\r
+ if (displayname != service_name && ! str_equiv(displayname, service_name)) return 1;\r
+\r
+ return 0;\r
+}\r
+\r
+int native_get_displayname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);\r
+ if (! qsc) return -1;\r
+\r
+ int ret = value_from_string(name, value, qsc->lpDisplayName);\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+\r
+ return ret;\r
+}\r
+\r
+int native_set_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = open_service_registry(service_name, KEY_SET_VALUE, true);\r
+ if (! key) return -1;\r
+\r
+ int ret = setting_set_environment(service_name, (void *) key, name, default_value, value, additional);\r
+ RegCloseKey(key);\r
+ return ret;\r
+}\r
+\r
+int native_get_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = open_service_registry(service_name, KEY_READ, true);\r
+ if (! key) return -1;\r
+\r
+ ZeroMemory(value, sizeof(value_t));\r
+ int ret = setting_get_environment(service_name, (void *) key, name, default_value, value, additional);\r
+ RegCloseKey(key);\r
+ return ret;\r
+}\r
+\r
+static int native_dump_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ HKEY key = open_service_registry(service_name, KEY_READ, true);\r
+ if (! key) return -1;\r
+\r
+ int ret = setting_dump_environment(service_name, (void *) key, name, default_value, value, additional);\r
+ RegCloseKey(key);\r
+ return ret;\r
+}\r
+\r
+int native_set_imagepath(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ /* It makes no sense to try to reset the image path. */\r
+ if (! value || ! value->string) {\r
+ print_message(stderr, NSSM_MESSAGE_NO_DEFAULT_VALUE, name);\r
+ return -1;\r
+ }\r
+\r
+ if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, value->string, 0, 0, 0, 0, 0, 0)) {\r
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));\r
+ return -1;\r
+ }\r
+\r
+ return 1;\r
+}\r
+\r
+int native_get_imagepath(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);\r
+ if (! qsc) return -1;\r
+\r
+ int ret = value_from_string(name, value, qsc->lpBinaryPathName);\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+\r
+ return ret;\r
+}\r
+\r
+int native_set_name(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ print_message(stderr, NSSM_MESSAGE_CANNOT_RENAME_SERVICE);\r
+ return -1;\r
+}\r
+\r
+int native_get_name(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ return value_from_string(name, value, service_name);\r
+}\r
+\r
+int native_set_objectname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ /*\r
+ Logical syntax is: nssm set <service> ObjectName <username> <password>\r
+ That means the username is actually passed in the additional parameter.\r
+ */\r
+ bool localsystem = false;\r
+ TCHAR *username = NSSM_LOCALSYSTEM_ACCOUNT;\r
+ TCHAR *password = 0;\r
+ if (additional) {\r
+ username = (TCHAR *) additional;\r
+ if (value && value->string) password = value->string;\r
+ }\r
+ else if (value && value->string) username = value->string;\r
+\r
+ const TCHAR *well_known = well_known_username(username);\r
+ size_t passwordsize = 0;\r
+ if (well_known) {\r
+ if (str_equiv(well_known, NSSM_LOCALSYSTEM_ACCOUNT)) localsystem = true;\r
+ username = (TCHAR *) well_known;\r
+ password = _T("");\r
+ }\r
+ else if (! password) {\r
+ /* We need a password if the account requires it. */\r
+ print_message(stderr, NSSM_MESSAGE_MISSING_PASSWORD, name);\r
+ return -1;\r
+ }\r
+ else passwordsize = _tcslen(password) * sizeof(TCHAR);\r
+\r
+ /*\r
+ ChangeServiceConfig() will fail to set the username if the service is set\r
+ to interact with the desktop.\r
+ */\r
+ unsigned long type = SERVICE_NO_CHANGE;\r
+ if (! localsystem) {\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);\r
+ if (! qsc) {\r
+ if (passwordsize) SecureZeroMemory(password, passwordsize);\r
+ return -1;\r
+ }\r
+\r
+ type = qsc->dwServiceType & ~SERVICE_INTERACTIVE_PROCESS;\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+ }\r
+\r
+ if (! well_known) {\r
+ if (grant_logon_as_service(username)) {\r
+ if (passwordsize) SecureZeroMemory(password, passwordsize);\r
+ print_message(stderr, NSSM_MESSAGE_GRANT_LOGON_AS_SERVICE_FAILED, username);\r
+ return -1;\r
+ }\r
+ }\r
+\r
+ if (! ChangeServiceConfig(service_handle, type, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, 0, username, password, 0)) {\r
+ if (passwordsize) SecureZeroMemory(password, passwordsize);\r
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));\r
+ return -1;\r
+ }\r
+\r
+ if (passwordsize) SecureZeroMemory(password, passwordsize);\r
+\r
+ if (localsystem) return 0;\r
+\r
+ return 1;\r
+}\r
+\r
+int native_get_objectname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);\r
+ if (! qsc) return -1;\r
+\r
+ int ret = value_from_string(name, value, qsc->lpServiceStartName);\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+\r
+ return ret;\r
+}\r
+\r
+int native_dump_objectname(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ int ret = native_get_objectname(service_name, param, name, default_value, value, additional);\r
+ if (ret != 1) return ret;\r
+\r
+ /* Do we need to dump a dummy password? */\r
+ if (! well_known_username(value->string)) {\r
+ /* Parameters are the other way round. */\r
+ value_t inverted;\r
+ inverted.string = _T("****");\r
+ return setting_dump_string(service_name, (void *) REG_SZ, name, &inverted, value->string);\r
+ }\r
+ return setting_dump_string(service_name, (void *) REG_SZ, name, value, 0);\r
+}\r
+\r
+int native_set_startup(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ /* It makes no sense to try to reset the startup type. */\r
+ if (! value || ! value->string) {\r
+ print_message(stderr, NSSM_MESSAGE_NO_DEFAULT_VALUE, name);\r
+ return -1;\r
+ }\r
+\r
+ /* Map NSSM_STARTUP_* constant to Windows SERVICE_*_START constant. */\r
+ int service_startup = -1;\r
+ int i;\r
+ for (i = 0; startup_strings[i]; i++) {\r
+ if (str_equiv(value->string, startup_strings[i])) {\r
+ service_startup = i;\r
+ break;\r
+ }\r
+ }\r
+\r
+ if (service_startup < 0) {\r
+ print_message(stderr, NSSM_MESSAGE_INVALID_SERVICE_STARTUP, value->string);\r
+ for (i = 0; startup_strings[i]; i++) _ftprintf(stderr, _T("%s\n"), startup_strings[i]);\r
+ return -1;\r
+ }\r
+\r
+ unsigned long startup;\r
+ switch (service_startup) {\r
+ case NSSM_STARTUP_MANUAL: startup = SERVICE_DEMAND_START; break;\r
+ case NSSM_STARTUP_DISABLED: startup = SERVICE_DISABLED; break;\r
+ default: startup = SERVICE_AUTO_START;\r
+ }\r
+\r
+ if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, startup, SERVICE_NO_CHANGE, 0, 0, 0, 0, 0, 0, 0)) {\r
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));\r
+ return -1;\r
+ }\r
+\r
+ SERVICE_DELAYED_AUTO_START_INFO delayed;\r
+ ZeroMemory(&delayed, sizeof(delayed));\r
+ if (service_startup == NSSM_STARTUP_DELAYED) delayed.fDelayedAutostart = 1;\r
+ else delayed.fDelayedAutostart = 0;\r
+ if (! ChangeServiceConfig2(service_handle, SERVICE_CONFIG_DELAYED_AUTO_START_INFO, &delayed)) {\r
+ unsigned long error = GetLastError();\r
+ /* Pre-Vista we expect to fail with ERROR_INVALID_LEVEL */\r
+ if (error != ERROR_INVALID_LEVEL) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_MESSAGE_SERVICE_CONFIG_DELAYED_AUTO_START_INFO_FAILED, service_name, error_string(error), 0);\r
+ }\r
+ }\r
+\r
+ return 1;\r
+}\r
+\r
+int native_get_startup(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);\r
+ if (! qsc) return -1;\r
+\r
+ unsigned long startup;\r
+ int ret = get_service_startup(service_name, service_handle, qsc, &startup);\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+\r
+ if (ret) return -1;\r
+\r
+ unsigned long i;\r
+ for (i = 0; startup_strings[i]; i++);\r
+ if (startup >= i) return -1;\r
+\r
+ return value_from_string(name, value, startup_strings[startup]);\r
+}\r
+\r
+int native_set_type(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ /* It makes no sense to try to reset the service type. */\r
+ if (! value || ! value->string) {\r
+ print_message(stderr, NSSM_MESSAGE_NO_DEFAULT_VALUE, name);\r
+ return -1;\r
+ }\r
+\r
+ /*\r
+ We can only manage services of type SERVICE_WIN32_OWN_PROCESS\r
+ and SERVICE_INTERACTIVE_PROCESS.\r
+ */\r
+ unsigned long type = SERVICE_WIN32_OWN_PROCESS;\r
+ if (str_equiv(value->string, NSSM_INTERACTIVE_PROCESS)) type |= SERVICE_INTERACTIVE_PROCESS;\r
+ else if (! str_equiv(value->string, NSSM_WIN32_OWN_PROCESS)) {\r
+ print_message(stderr, NSSM_MESSAGE_INVALID_SERVICE_TYPE, value->string);\r
+ _ftprintf(stderr, _T("%s\n"), NSSM_WIN32_OWN_PROCESS);\r
+ _ftprintf(stderr, _T("%s\n"), NSSM_INTERACTIVE_PROCESS);\r
+ return -1;\r
+ }\r
+\r
+ /*\r
+ ChangeServiceConfig() will fail if the service runs under an account\r
+ other than LOCALSYSTEM and we try to make it interactive.\r
+ */\r
+ if (type & SERVICE_INTERACTIVE_PROCESS) {\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);\r
+ if (! qsc) return -1;\r
+\r
+ if (! str_equiv(qsc->lpServiceStartName, NSSM_LOCALSYSTEM_ACCOUNT)) {\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+ print_message(stderr, NSSM_MESSAGE_INTERACTIVE_NOT_LOCALSYSTEM, value->string, service_name, NSSM_LOCALSYSTEM_ACCOUNT);\r
+ return -1;\r
+ }\r
+\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+ }\r
+\r
+ if (! ChangeServiceConfig(service_handle, type, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, 0, 0, 0, 0)) {\r
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));\r
+ return -1;\r
+ }\r
+\r
+ return 1;\r
+}\r
+\r
+int native_get_type(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {\r
+ SC_HANDLE service_handle = (SC_HANDLE) param;\r
+ if (! service_handle) return -1;\r
+\r
+ QUERY_SERVICE_CONFIG *qsc = query_service_config(service_name, service_handle);\r
+ if (! qsc) return -1;\r
+\r
+ value->numeric = qsc->dwServiceType;\r
+ HeapFree(GetProcessHeap(), 0, qsc);\r
+\r
+ const TCHAR *string;\r
+ switch (value->numeric) {\r
+ case SERVICE_KERNEL_DRIVER: string = NSSM_KERNEL_DRIVER; break;\r
+ case SERVICE_FILE_SYSTEM_DRIVER: string = NSSM_FILE_SYSTEM_DRIVER; break;\r
+ case SERVICE_WIN32_OWN_PROCESS: string = NSSM_WIN32_OWN_PROCESS; break;\r
+ case SERVICE_WIN32_SHARE_PROCESS: string = NSSM_WIN32_SHARE_PROCESS; break;\r
+ case SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS: string = NSSM_INTERACTIVE_PROCESS; break;\r
+ case SERVICE_WIN32_SHARE_PROCESS|SERVICE_INTERACTIVE_PROCESS: string = NSSM_SHARE_INTERACTIVE_PROCESS; break;\r
+ default: string = NSSM_UNKNOWN;\r
+ }\r
+\r
+ return value_from_string(name, value, string);\r
+}\r
+\r
+int set_setting(const TCHAR *service_name, HKEY key, settings_t *setting, value_t *value, const TCHAR *additional) {\r
+ if (! key) return -1;\r
+ int ret;\r
+\r
+ if (setting->set) ret = setting->set(service_name, (void *) key, setting->name, setting->default_value, value, additional);\r
+ else ret = -1;\r
+\r
+ if (! ret) print_message(stdout, NSSM_MESSAGE_RESET_SETTING, setting->name, service_name);\r
+ else if (ret > 0) print_message(stdout, NSSM_MESSAGE_SET_SETTING, setting->name, service_name);\r
+ else print_message(stderr, NSSM_MESSAGE_SET_SETTING_FAILED, setting->name, service_name);\r
+\r
+ return ret;\r
+}\r
+\r
+int set_setting(const TCHAR *service_name, SC_HANDLE service_handle, settings_t *setting, value_t *value, const TCHAR *additional) {\r
+ if (! service_handle) return -1;\r
+\r
+ int ret;\r
+ if (setting->set) ret = setting->set(service_name, service_handle, setting->name, setting->default_value, value, additional);\r
+ else ret = -1;\r
+\r
+ if (! ret) print_message(stdout, NSSM_MESSAGE_RESET_SETTING, setting->name, service_name);\r
+ else if (ret > 0) print_message(stdout, NSSM_MESSAGE_SET_SETTING, setting->name, service_name);\r
+ else print_message(stderr, NSSM_MESSAGE_SET_SETTING_FAILED, setting->name, service_name);\r
+\r
+ return ret;\r
+}\r
+\r
+/*\r
+ Returns: 1 if the value was retrieved.\r
+ 0 if the default value was retrieved.\r
+ -1 on error.\r
+*/\r
+int get_setting(const TCHAR *service_name, HKEY key, settings_t *setting, value_t *value, const TCHAR *additional) {\r
+ if (! key) return -1;\r
+ int ret;\r
+\r
+ if (is_string_type(setting->type)) {\r
+ value->string = (TCHAR *) setting->default_value;\r
+ if (setting->get) ret = setting->get(service_name, (void *) key, setting->name, setting->default_value, value, additional);\r
+ else ret = -1;\r
+ }\r
+ else if (is_numeric_type(setting->type)) {\r
+ value->numeric = PtrToUlong(setting->default_value);\r
+ if (setting->get) ret = setting->get(service_name, (void *) key, setting->name, setting->default_value, value, additional);\r
+ else ret = -1;\r
+ }\r
+ else ret = -1;\r
+\r
+ if (ret < 0) print_message(stderr, NSSM_MESSAGE_GET_SETTING_FAILED, setting->name, service_name);\r
+\r
+ return ret;\r
+}\r
+\r
+int get_setting(const TCHAR *service_name, SC_HANDLE service_handle, settings_t *setting, value_t *value, const TCHAR *additional) {\r
+ if (! service_handle) return -1;\r
+ return setting->get(service_name, service_handle, setting->name, 0, value, additional);\r
+}\r
+\r
+int dump_setting(const TCHAR *service_name, HKEY key, SC_HANDLE service_handle, settings_t *setting) {\r
+ void *param;\r
+ if (setting->native) {\r
+ if (! service_handle) return -1;\r
+ param = (void *) service_handle;\r
+ }\r
+ else {\r
+ /* Will be null for native services. */\r
+ param = (void *) key;\r
+ }\r
+\r
+ value_t value = { 0 };\r
+ int ret;\r
+\r
+ if (setting->dump) return setting->dump(service_name, param, setting->name, (void *) setting, &value, 0);\r
+ if (setting->native) ret = get_setting(service_name, service_handle, setting, &value, 0);\r
+ else ret = get_setting(service_name, key, setting, &value, 0);\r
+ if (ret != 1) return ret;\r
+ return setting_dump_string(service_name, (void *) setting->type, setting->name, &value, 0);\r
+}\r
+\r
+settings_t settings[] = {\r
+ { NSSM_REG_EXE, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string, setting_not_dumpable },\r
+ { NSSM_REG_FLAGS, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string, 0 },\r
+ { NSSM_REG_DIR, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string, 0 },\r
+ { NSSM_REG_EXIT, REG_SZ, (void *) exit_action_strings[NSSM_EXIT_RESTART], false, ADDITIONAL_MANDATORY, setting_set_exit_action, setting_get_exit_action, setting_dump_exit_action },\r
+ { NSSM_REG_HOOK, REG_SZ, (void *) _T(""), false, ADDITIONAL_MANDATORY, setting_set_hook, setting_get_hook, setting_dump_hooks },\r
+ { NSSM_REG_AFFINITY, REG_SZ, 0, false, 0, setting_set_affinity, setting_get_affinity, 0 },\r
+ { NSSM_REG_ENV, REG_MULTI_SZ, NULL, false, ADDITIONAL_CRLF, setting_set_environment, setting_get_environment, setting_dump_environment },\r
+ { NSSM_REG_ENV_EXTRA, REG_MULTI_SZ, NULL, false, ADDITIONAL_CRLF, setting_set_environment, setting_get_environment, setting_dump_environment },\r
+ { NSSM_REG_NO_CONSOLE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_PRIORITY, REG_SZ, (void *) priority_strings[NSSM_NORMAL_PRIORITY], false, 0, setting_set_priority, setting_get_priority, setting_dump_priority },\r
+ { NSSM_REG_RESTART_DELAY, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDIN, REG_EXPAND_SZ, NULL, false, 0, setting_set_string, setting_get_string, 0 },\r
+ { NSSM_REG_STDIN NSSM_REG_STDIO_SHARING, REG_DWORD, (void *) NSSM_STDIN_SHARING, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDIN NSSM_REG_STDIO_DISPOSITION, REG_DWORD, (void *) NSSM_STDIN_DISPOSITION, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDIN NSSM_REG_STDIO_FLAGS, REG_DWORD, (void *) NSSM_STDIN_FLAGS, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDOUT, REG_EXPAND_SZ, NULL, false, 0, setting_set_string, setting_get_string, 0 },\r
+ { NSSM_REG_STDOUT NSSM_REG_STDIO_SHARING, REG_DWORD, (void *) NSSM_STDOUT_SHARING, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDOUT NSSM_REG_STDIO_DISPOSITION, REG_DWORD, (void *) NSSM_STDOUT_DISPOSITION, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDOUT NSSM_REG_STDIO_FLAGS, REG_DWORD, (void *) NSSM_STDOUT_FLAGS, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDOUT NSSM_REG_STDIO_COPY_AND_TRUNCATE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDERR, REG_EXPAND_SZ, NULL, false, 0, setting_set_string, setting_get_string, 0 },\r
+ { NSSM_REG_STDERR NSSM_REG_STDIO_SHARING, REG_DWORD, (void *) NSSM_STDERR_SHARING, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDERR NSSM_REG_STDIO_DISPOSITION, REG_DWORD, (void *) NSSM_STDERR_DISPOSITION, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDERR NSSM_REG_STDIO_FLAGS, REG_DWORD, (void *) NSSM_STDERR_FLAGS, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STDERR NSSM_REG_STDIO_COPY_AND_TRUNCATE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_STOP_METHOD_SKIP, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_KILL_CONSOLE_GRACE_PERIOD, REG_DWORD, (void *) NSSM_KILL_CONSOLE_GRACE_PERIOD, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_KILL_WINDOW_GRACE_PERIOD, REG_DWORD, (void *) NSSM_KILL_WINDOW_GRACE_PERIOD, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_KILL_THREADS_GRACE_PERIOD, REG_DWORD, (void *) NSSM_KILL_THREADS_GRACE_PERIOD, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_KILL_PROCESS_TREE, REG_DWORD, (void *) 1, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_THROTTLE, REG_DWORD, (void *) NSSM_RESET_THROTTLE_RESTART, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_HOOK_SHARE_OUTPUT_HANDLES, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_ROTATE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_ROTATE_ONLINE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_ROTATE_SECONDS, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_ROTATE_BYTES_LOW, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_ROTATE_BYTES_HIGH, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_REG_ROTATE_DELAY, REG_DWORD, (void *) NSSM_ROTATE_DELAY, false, 0, setting_set_number, setting_get_number, 0 },\r
+ { NSSM_NATIVE_DEPENDONGROUP, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_dependongroup, native_get_dependongroup, native_dump_dependongroup },\r
+ { NSSM_NATIVE_DEPENDONSERVICE, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_dependonservice, native_get_dependonservice, native_dump_dependonservice },\r
+ { NSSM_NATIVE_DESCRIPTION, REG_SZ, _T(""), true, 0, native_set_description, native_get_description, 0 },\r
+ { NSSM_NATIVE_DISPLAYNAME, REG_SZ, NULL, true, 0, native_set_displayname, native_get_displayname, 0 },\r
+ { NSSM_NATIVE_ENVIRONMENT, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_environment, native_get_environment, native_dump_environment },\r
+ { NSSM_NATIVE_IMAGEPATH, REG_EXPAND_SZ, NULL, true, 0, native_set_imagepath, native_get_imagepath, setting_not_dumpable },\r
+ { NSSM_NATIVE_OBJECTNAME, REG_SZ, NSSM_LOCALSYSTEM_ACCOUNT, true, 0, native_set_objectname, native_get_objectname, native_dump_objectname },\r
+ { NSSM_NATIVE_NAME, REG_SZ, NULL, true, 0, native_set_name, native_get_name, setting_not_dumpable },\r
+ { NSSM_NATIVE_STARTUP, REG_SZ, NULL, true, 0, native_set_startup, native_get_startup, 0 },\r
+ { NSSM_NATIVE_TYPE, REG_SZ, NULL, true, 0, native_set_type, native_get_type, 0 },\r
+ { NULL, NULL, NULL, NULL, NULL }\r
+};\r
git://git.iain.cx/iain / nssm.git / blobdiff