|
|
@@ -1,1130 +1,1130 @@
|
|
|
-#include "nssm.h"
|
|
|
-/* XXX: (value && value->string) is probably bogus because value is probably never null */
|
|
|
-
|
|
|
-/* Affinity. */
|
|
|
-#define NSSM_AFFINITY_ALL _T("All")
|
|
|
-
|
|
|
-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 setting? */
|
|
|
-static inline int is_default(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 == PtrToUlong(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 (get_string(key, (TCHAR *) name, (TCHAR *) buffer, (unsigned long) sizeof(buffer), false, 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(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 *) exit_action_strings[i], (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(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 inline bool split_hook_name(const TCHAR *hook_name, TCHAR *hook_event, TCHAR *hook_action) {
|
|
|
- TCHAR *s;
|
|
|
-
|
|
|
- for (s = (TCHAR *) hook_name; *s; s++) {
|
|
|
- if (*s == _T('/')) {
|
|
|
- *s = _T('\0');
|
|
|
- _sntprintf_s(hook_event, HOOK_NAME_LENGTH, _TRUNCATE, _T("%s"), hook_name);
|
|
|
- _sntprintf_s(hook_action, HOOK_NAME_LENGTH, _TRUNCATE, _T("%s"), ++s);
|
|
|
- return valid_hook_name(hook_event, hook_action, false);
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- print_message(stderr, NSSM_MESSAGE_INVALID_HOOK_NAME, hook_name);
|
|
|
- return false;
|
|
|
-}
|
|
|
-
|
|
|
-static int setting_set_hook(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
- TCHAR hook_event[HOOK_NAME_LENGTH];
|
|
|
- TCHAR hook_action[HOOK_NAME_LENGTH];
|
|
|
- if (! split_hook_name(additional, hook_event, hook_action)) return -1;
|
|
|
-
|
|
|
- TCHAR *cmd;
|
|
|
- if (value && value->string) cmd = value->string;
|
|
|
- else cmd = _T("");
|
|
|
-
|
|
|
- if (set_hook(service_name, hook_event, hook_action, cmd)) return -1;
|
|
|
- if (! _tcslen(cmd)) return 0;
|
|
|
- return 1;
|
|
|
-}
|
|
|
-
|
|
|
-static int setting_get_hook(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
- TCHAR hook_event[HOOK_NAME_LENGTH];
|
|
|
- TCHAR hook_action[HOOK_NAME_LENGTH];
|
|
|
- if (! split_hook_name(additional, hook_event, hook_action)) return -1;
|
|
|
-
|
|
|
- TCHAR cmd[CMD_LENGTH];
|
|
|
- if (get_hook(service_name, hook_event, hook_action, cmd, sizeof(cmd))) return -1;
|
|
|
-
|
|
|
- value_from_string(name, value, cmd);
|
|
|
-
|
|
|
- if (! _tcslen(cmd)) return 0;
|
|
|
- return 1;
|
|
|
-}
|
|
|
-
|
|
|
-static int setting_set_affinity(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;
|
|
|
- __int64 mask;
|
|
|
- __int64 system_affinity = 0LL;
|
|
|
-
|
|
|
- if (value && value->string) {
|
|
|
- DWORD_PTR affinity;
|
|
|
- if (! GetProcessAffinityMask(GetCurrentProcess(), &affinity, (DWORD_PTR *) &system_affinity)) system_affinity = ~0;
|
|
|
-
|
|
|
- if (is_default(value->string) || str_equiv(value->string, NSSM_AFFINITY_ALL)) mask = 0LL;
|
|
|
- else if (affinity_string_to_mask(value->string, &mask)) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_BOGUS_AFFINITY_MASK, value->string, num_cpus() - 1);
|
|
|
- return -1;
|
|
|
- }
|
|
|
- }
|
|
|
- else mask = 0LL;
|
|
|
-
|
|
|
- if (! mask) {
|
|
|
- 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;
|
|
|
- }
|
|
|
-
|
|
|
- /* Canonicalise. */
|
|
|
- TCHAR *canon = 0;
|
|
|
- if (affinity_mask_to_string(mask, &canon)) canon = value->string;
|
|
|
-
|
|
|
- __int64 effective_affinity = mask & system_affinity;
|
|
|
- if (effective_affinity != mask) {
|
|
|
- /* Requested CPUs did not intersect with available CPUs? */
|
|
|
- if (! effective_affinity) mask = effective_affinity = system_affinity;
|
|
|
-
|
|
|
- TCHAR *system = 0;
|
|
|
- if (! affinity_mask_to_string(system_affinity, &system)) {
|
|
|
- TCHAR *effective = 0;
|
|
|
- if (! affinity_mask_to_string(effective_affinity, &effective)) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_EFFECTIVE_AFFINITY_MASK, value->string, system, effective);
|
|
|
- HeapFree(GetProcessHeap(), 0, effective);
|
|
|
- }
|
|
|
- HeapFree(GetProcessHeap(), 0, system);
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- if (RegSetValueEx(key, name, 0, REG_SZ, (const unsigned char *) canon, (unsigned long) (_tcslen(canon) + 1) * sizeof(TCHAR)) != ERROR_SUCCESS) {
|
|
|
- if (canon != value->string) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, name, error_string(GetLastError()), 0);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- if (canon != value->string) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
- return 1;
|
|
|
-}
|
|
|
-
|
|
|
-static int setting_get_affinity(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 type;
|
|
|
- TCHAR *buffer = 0;
|
|
|
- unsigned long buflen = 0;
|
|
|
-
|
|
|
- int ret = RegQueryValueEx(key, name, 0, &type, 0, &buflen);
|
|
|
- if (ret == ERROR_FILE_NOT_FOUND) {
|
|
|
- if (value_from_string(name, value, NSSM_AFFINITY_ALL) == 1) return 0;
|
|
|
- return -1;
|
|
|
- }
|
|
|
- if (ret != ERROR_SUCCESS) return -1;
|
|
|
-
|
|
|
- if (type != REG_SZ) return -1;
|
|
|
-
|
|
|
- buffer = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, buflen);
|
|
|
- if (! buffer) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("affinity"), _T("setting_get_affinity"));
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- if (get_string(key, (TCHAR *) name, buffer, buflen, false, false, true)) {
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- __int64 affinity;
|
|
|
- if (affinity_string_to_mask(buffer, &affinity)) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_BOGUS_AFFINITY_MASK, buffer, num_cpus() - 1);
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
-
|
|
|
- /* Canonicalise. */
|
|
|
- if (affinity_mask_to_string(affinity, &buffer)) {
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- ret = value_from_string(name, value, buffer);
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return ret;
|
|
|
-}
|
|
|
-
|
|
|
-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_double_null(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 (get_environment((TCHAR *) service_name, key, (TCHAR *) name, &env, &envlen)) return -1;
|
|
|
- if (! envlen) return 0;
|
|
|
-
|
|
|
- TCHAR *formatted;
|
|
|
- unsigned long newlen;
|
|
|
- if (format_double_null(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. */
|
|
|
-static int native_set_dependongroup(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;
|
|
|
-
|
|
|
- /*
|
|
|
- Get existing service dependencies because we must set both types together.
|
|
|
- */
|
|
|
- TCHAR *buffer;
|
|
|
- unsigned long buflen;
|
|
|
- if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_SERVICES)) return -1;
|
|
|
-
|
|
|
- if (! value || ! value->string || ! value->string[0]) {
|
|
|
- if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, buffer, 0, 0, 0)) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return 0;
|
|
|
- }
|
|
|
-
|
|
|
- unsigned long len = (unsigned long) _tcslen(value->string) + 1;
|
|
|
- TCHAR *unformatted = 0;
|
|
|
- unsigned long newlen;
|
|
|
- if (unformat_double_null(value->string, len, &unformatted, &newlen)) {
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- /* Prepend group identifier. */
|
|
|
- unsigned long missing = 0;
|
|
|
- TCHAR *canon = unformatted;
|
|
|
- size_t canonlen = 0;
|
|
|
- TCHAR *s;
|
|
|
- for (s = unformatted; *s; s++) {
|
|
|
- if (*s != SC_GROUP_IDENTIFIER) missing++;
|
|
|
- size_t len = _tcslen(s);
|
|
|
- canonlen += len + 1;
|
|
|
- s += len;
|
|
|
- }
|
|
|
-
|
|
|
- if (missing) {
|
|
|
- /* Missing identifiers plus double NULL terminator. */
|
|
|
- canonlen += missing + 1;
|
|
|
- newlen = (unsigned long) canonlen;
|
|
|
-
|
|
|
- canon = (TCHAR *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, canonlen * sizeof(TCHAR));
|
|
|
- if (! canon) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("canon"), _T("native_set_dependongroup"));
|
|
|
- if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- size_t i = 0;
|
|
|
- for (s = unformatted; *s; s++) {
|
|
|
- if (*s != SC_GROUP_IDENTIFIER) canon[i++] = SC_GROUP_IDENTIFIER;
|
|
|
- size_t len = _tcslen(s);
|
|
|
- memmove(canon + i, s, (len + 1) * sizeof(TCHAR));
|
|
|
- i += len + 1;
|
|
|
- s += len;
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- TCHAR *dependencies;
|
|
|
- if (buflen > 2) {
|
|
|
- dependencies = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (newlen + buflen) * sizeof(TCHAR));
|
|
|
- if (! dependencies) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dependencies"), _T("native_set_dependongroup"));
|
|
|
- if (canon != unformatted) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
- if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- memmove(dependencies, buffer, buflen * sizeof(TCHAR));
|
|
|
- memmove(dependencies + buflen - 1, canon, newlen * sizeof(TCHAR));
|
|
|
- }
|
|
|
- else dependencies = canon;
|
|
|
-
|
|
|
- int ret = 1;
|
|
|
- if (set_service_dependencies(service_name, service_handle, dependencies)) ret = -1;
|
|
|
- if (dependencies != unformatted) HeapFree(GetProcessHeap(), 0, dependencies);
|
|
|
- if (canon != unformatted) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
- if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
-
|
|
|
- return ret;
|
|
|
-}
|
|
|
-
|
|
|
-static int native_get_dependongroup(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;
|
|
|
- unsigned long buflen;
|
|
|
- if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_GROUPS)) return -1;
|
|
|
-
|
|
|
- int ret;
|
|
|
- if (buflen) {
|
|
|
- TCHAR *formatted;
|
|
|
- unsigned long newlen;
|
|
|
- if (format_double_null(buffer, buflen, &formatted, &newlen)) {
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- ret = value_from_string(name, value, formatted);
|
|
|
- HeapFree(GetProcessHeap(), 0, formatted);
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- }
|
|
|
- else {
|
|
|
- value->string = 0;
|
|
|
- ret = 0;
|
|
|
- }
|
|
|
-
|
|
|
- return ret;
|
|
|
-}
|
|
|
-
|
|
|
-static int native_set_dependonservice(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;
|
|
|
-
|
|
|
- /*
|
|
|
- Get existing group dependencies because we must set both types together.
|
|
|
- */
|
|
|
- TCHAR *buffer;
|
|
|
- unsigned long buflen;
|
|
|
- if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_GROUPS)) return -1;
|
|
|
-
|
|
|
- if (! value || ! value->string || ! value->string[0]) {
|
|
|
- if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, buffer, 0, 0, 0)) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return 0;
|
|
|
- }
|
|
|
-
|
|
|
- unsigned long len = (unsigned long) _tcslen(value->string) + 1;
|
|
|
- TCHAR *unformatted = 0;
|
|
|
- unsigned long newlen;
|
|
|
- if (unformat_double_null(value->string, len, &unformatted, &newlen)) {
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- TCHAR *dependencies;
|
|
|
- if (buflen > 2) {
|
|
|
- dependencies = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (newlen + buflen) * sizeof(TCHAR));
|
|
|
- if (! dependencies) {
|
|
|
- print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dependencies"), _T("native_set_dependonservice"));
|
|
|
- if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- memmove(dependencies, buffer, buflen * sizeof(TCHAR));
|
|
|
- memmove(dependencies + buflen - 1, unformatted, newlen * sizeof(TCHAR));
|
|
|
- }
|
|
|
- else dependencies = unformatted;
|
|
|
-
|
|
|
- int ret = 1;
|
|
|
- if (set_service_dependencies(service_name, service_handle, dependencies)) ret = -1;
|
|
|
- if (dependencies != unformatted) HeapFree(GetProcessHeap(), 0, dependencies);
|
|
|
- if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
- if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
-
|
|
|
- return ret;
|
|
|
-}
|
|
|
-
|
|
|
-static int native_get_dependonservice(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;
|
|
|
- unsigned long buflen;
|
|
|
- if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_SERVICES)) return -1;
|
|
|
-
|
|
|
- int ret;
|
|
|
- if (buflen) {
|
|
|
- TCHAR *formatted;
|
|
|
- unsigned long newlen;
|
|
|
- if (format_double_null(buffer, buflen, &formatted, &newlen)) {
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- ret = value_from_string(name, value, formatted);
|
|
|
- HeapFree(GetProcessHeap(), 0, formatted);
|
|
|
- HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
- }
|
|
|
- else {
|
|
|
- value->string = 0;
|
|
|
- ret = 0;
|
|
|
- }
|
|
|
-
|
|
|
- return ret;
|
|
|
-}
|
|
|
-
|
|
|
-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_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
- HKEY key = open_service_registry(service_name, KEY_SET_VALUE, false);
|
|
|
- if (! key) return -1;
|
|
|
-
|
|
|
- int ret = setting_set_environment(service_name, (void *) key, NSSM_NATIVE_ENVIRONMENT, default_value, value, additional);
|
|
|
- RegCloseKey(key);
|
|
|
- return ret;
|
|
|
-}
|
|
|
-
|
|
|
-int native_get_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
- HKEY key = open_service_registry(service_name, KEY_READ, false);
|
|
|
- if (! key) return -1;
|
|
|
-
|
|
|
- ZeroMemory(value, sizeof(value_t));
|
|
|
- int ret = setting_get_environment(service_name, (void *) key, NSSM_NATIVE_ENVIRONMENT, default_value, value, additional);
|
|
|
- RegCloseKey(key);
|
|
|
- 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 = false;
|
|
|
- TCHAR *username = NSSM_LOCALSYSTEM_ACCOUNT;
|
|
|
- TCHAR *password = 0;
|
|
|
- if (additional) {
|
|
|
- username = (TCHAR *) additional;
|
|
|
- if (value && value->string) password = value->string;
|
|
|
- }
|
|
|
- else if (value && value->string) username = value->string;
|
|
|
-
|
|
|
- const TCHAR *well_known = well_known_username(username);
|
|
|
- size_t passwordsize = 0;
|
|
|
- if (well_known) {
|
|
|
- if (str_equiv(well_known, NSSM_LOCALSYSTEM_ACCOUNT)) localsystem = true;
|
|
|
- username = (TCHAR *) well_known;
|
|
|
- password = _T("");
|
|
|
- }
|
|
|
- else if (! password) {
|
|
|
- /* We need a password if the account requires it. */
|
|
|
- print_message(stderr, NSSM_MESSAGE_MISSING_PASSWORD, name);
|
|
|
- return -1;
|
|
|
- }
|
|
|
- else passwordsize = _tcslen(password) * sizeof(TCHAR);
|
|
|
-
|
|
|
- /*
|
|
|
- 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 (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- type = qsc->dwServiceType & ~SERVICE_INTERACTIVE_PROCESS;
|
|
|
- HeapFree(GetProcessHeap(), 0, qsc);
|
|
|
- }
|
|
|
-
|
|
|
- if (! well_known) {
|
|
|
- if (grant_logon_as_service(username)) {
|
|
|
- if (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
- 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 (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
- print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
|
|
|
- return -1;
|
|
|
- }
|
|
|
-
|
|
|
- if (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
-
|
|
|
- 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 = PtrToUlong(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_HOOK, REG_SZ, (void *) _T(""), false, ADDITIONAL_MANDATORY, setting_set_hook, setting_get_hook },
|
|
|
- { NSSM_REG_AFFINITY, REG_SZ, 0, false, 0, setting_set_affinity, setting_get_affinity },
|
|
|
- { 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_NO_CONSOLE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
|
|
|
- { NSSM_REG_PRIORITY, REG_SZ, (void *) priority_strings[NSSM_NORMAL_PRIORITY], false, 0, setting_set_priority, setting_get_priority },
|
|
|
- { NSSM_REG_RESTART_DELAY, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
|
|
|
- { 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_STDOUT NSSM_REG_STDIO_COPY_AND_TRUNCATE, REG_DWORD, 0, 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_STDERR NSSM_REG_STDIO_COPY_AND_TRUNCATE, REG_DWORD, 0, 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_KILL_PROCESS_TREE, REG_DWORD, (void *) 1, 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_ONLINE, 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_REG_ROTATE_DELAY, REG_DWORD, (void *) NSSM_ROTATE_DELAY, false, 0, setting_set_number, setting_get_number },
|
|
|
- { NSSM_NATIVE_DEPENDONGROUP, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_dependongroup, native_get_dependongroup },
|
|
|
- { NSSM_NATIVE_DEPENDONSERVICE, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_dependonservice, native_get_dependonservice },
|
|
|
- { 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_ENVIRONMENT, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_environment, native_get_environment },
|
|
|
- { NSSM_NATIVE_IMAGEPATH, REG_EXPAND_SZ, NULL, true, 0, native_set_imagepath, native_get_imagepath },
|
|
|
- { NSSM_NATIVE_OBJECTNAME, REG_SZ, NSSM_LOCALSYSTEM_ACCOUNT, true, 0, 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"
|
|
|
+/* XXX: (value && value->string) is probably bogus because value is probably never null */
|
|
|
+
|
|
|
+/* Affinity. */
|
|
|
+#define NSSM_AFFINITY_ALL _T("All")
|
|
|
+
|
|
|
+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 setting? */
|
|
|
+static inline int is_default(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 == PtrToUlong(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 (get_string(key, (TCHAR *) name, (TCHAR *) buffer, (unsigned long) sizeof(buffer), false, 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(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 *) exit_action_strings[i], (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(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 inline bool split_hook_name(const TCHAR *hook_name, TCHAR *hook_event, TCHAR *hook_action) {
|
|
|
+ TCHAR *s;
|
|
|
+
|
|
|
+ for (s = (TCHAR *) hook_name; *s; s++) {
|
|
|
+ if (*s == _T('/')) {
|
|
|
+ *s = _T('\0');
|
|
|
+ _sntprintf_s(hook_event, HOOK_NAME_LENGTH, _TRUNCATE, _T("%s"), hook_name);
|
|
|
+ _sntprintf_s(hook_action, HOOK_NAME_LENGTH, _TRUNCATE, _T("%s"), ++s);
|
|
|
+ return valid_hook_name(hook_event, hook_action, false);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ print_message(stderr, NSSM_MESSAGE_INVALID_HOOK_NAME, hook_name);
|
|
|
+ return false;
|
|
|
+}
|
|
|
+
|
|
|
+static int setting_set_hook(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
+ TCHAR hook_event[HOOK_NAME_LENGTH];
|
|
|
+ TCHAR hook_action[HOOK_NAME_LENGTH];
|
|
|
+ if (! split_hook_name(additional, hook_event, hook_action)) return -1;
|
|
|
+
|
|
|
+ TCHAR *cmd;
|
|
|
+ if (value && value->string) cmd = value->string;
|
|
|
+ else cmd = _T("");
|
|
|
+
|
|
|
+ if (set_hook(service_name, hook_event, hook_action, cmd)) return -1;
|
|
|
+ if (! _tcslen(cmd)) return 0;
|
|
|
+ return 1;
|
|
|
+}
|
|
|
+
|
|
|
+static int setting_get_hook(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
+ TCHAR hook_event[HOOK_NAME_LENGTH];
|
|
|
+ TCHAR hook_action[HOOK_NAME_LENGTH];
|
|
|
+ if (! split_hook_name(additional, hook_event, hook_action)) return -1;
|
|
|
+
|
|
|
+ TCHAR cmd[CMD_LENGTH];
|
|
|
+ if (get_hook(service_name, hook_event, hook_action, cmd, sizeof(cmd))) return -1;
|
|
|
+
|
|
|
+ value_from_string(name, value, cmd);
|
|
|
+
|
|
|
+ if (! _tcslen(cmd)) return 0;
|
|
|
+ return 1;
|
|
|
+}
|
|
|
+
|
|
|
+static int setting_set_affinity(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;
|
|
|
+ __int64 mask;
|
|
|
+ __int64 system_affinity = 0LL;
|
|
|
+
|
|
|
+ if (value && value->string) {
|
|
|
+ DWORD_PTR affinity;
|
|
|
+ if (! GetProcessAffinityMask(GetCurrentProcess(), &affinity, (DWORD_PTR *) &system_affinity)) system_affinity = ~0;
|
|
|
+
|
|
|
+ if (is_default(value->string) || str_equiv(value->string, NSSM_AFFINITY_ALL)) mask = 0LL;
|
|
|
+ else if (affinity_string_to_mask(value->string, &mask)) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_BOGUS_AFFINITY_MASK, value->string, num_cpus() - 1);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ else mask = 0LL;
|
|
|
+
|
|
|
+ if (! mask) {
|
|
|
+ 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;
|
|
|
+ }
|
|
|
+
|
|
|
+ /* Canonicalise. */
|
|
|
+ TCHAR *canon = 0;
|
|
|
+ if (affinity_mask_to_string(mask, &canon)) canon = value->string;
|
|
|
+
|
|
|
+ __int64 effective_affinity = mask & system_affinity;
|
|
|
+ if (effective_affinity != mask) {
|
|
|
+ /* Requested CPUs did not intersect with available CPUs? */
|
|
|
+ if (! effective_affinity) mask = effective_affinity = system_affinity;
|
|
|
+
|
|
|
+ TCHAR *system = 0;
|
|
|
+ if (! affinity_mask_to_string(system_affinity, &system)) {
|
|
|
+ TCHAR *effective = 0;
|
|
|
+ if (! affinity_mask_to_string(effective_affinity, &effective)) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_EFFECTIVE_AFFINITY_MASK, value->string, system, effective);
|
|
|
+ HeapFree(GetProcessHeap(), 0, effective);
|
|
|
+ }
|
|
|
+ HeapFree(GetProcessHeap(), 0, system);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (RegSetValueEx(key, name, 0, REG_SZ, (const unsigned char *) canon, (unsigned long) (_tcslen(canon) + 1) * sizeof(TCHAR)) != ERROR_SUCCESS) {
|
|
|
+ if (canon != value->string) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, name, error_string(GetLastError()), 0);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (canon != value->string) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
+ return 1;
|
|
|
+}
|
|
|
+
|
|
|
+static int setting_get_affinity(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 type;
|
|
|
+ TCHAR *buffer = 0;
|
|
|
+ unsigned long buflen = 0;
|
|
|
+
|
|
|
+ int ret = RegQueryValueEx(key, name, 0, &type, 0, &buflen);
|
|
|
+ if (ret == ERROR_FILE_NOT_FOUND) {
|
|
|
+ if (value_from_string(name, value, NSSM_AFFINITY_ALL) == 1) return 0;
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ if (ret != ERROR_SUCCESS) return -1;
|
|
|
+
|
|
|
+ if (type != REG_SZ) return -1;
|
|
|
+
|
|
|
+ buffer = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, buflen);
|
|
|
+ if (! buffer) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("affinity"), _T("setting_get_affinity"));
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (get_string(key, (TCHAR *) name, buffer, buflen, false, false, true)) {
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ __int64 affinity;
|
|
|
+ if (affinity_string_to_mask(buffer, &affinity)) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_BOGUS_AFFINITY_MASK, buffer, num_cpus() - 1);
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+
|
|
|
+ /* Canonicalise. */
|
|
|
+ if (affinity_mask_to_string(affinity, &buffer)) {
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ ret = value_from_string(name, value, buffer);
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+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_double_null(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 (get_environment((TCHAR *) service_name, key, (TCHAR *) name, &env, &envlen)) return -1;
|
|
|
+ if (! envlen) return 0;
|
|
|
+
|
|
|
+ TCHAR *formatted;
|
|
|
+ unsigned long newlen;
|
|
|
+ if (format_double_null(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. */
|
|
|
+static int native_set_dependongroup(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;
|
|
|
+
|
|
|
+ /*
|
|
|
+ Get existing service dependencies because we must set both types together.
|
|
|
+ */
|
|
|
+ TCHAR *buffer;
|
|
|
+ unsigned long buflen;
|
|
|
+ if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_SERVICES)) return -1;
|
|
|
+
|
|
|
+ if (! value || ! value->string || ! value->string[0]) {
|
|
|
+ if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, buffer, 0, 0, 0)) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ unsigned long len = (unsigned long) _tcslen(value->string) + 1;
|
|
|
+ TCHAR *unformatted = 0;
|
|
|
+ unsigned long newlen;
|
|
|
+ if (unformat_double_null(value->string, len, &unformatted, &newlen)) {
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ /* Prepend group identifier. */
|
|
|
+ unsigned long missing = 0;
|
|
|
+ TCHAR *canon = unformatted;
|
|
|
+ size_t canonlen = 0;
|
|
|
+ TCHAR *s;
|
|
|
+ for (s = unformatted; *s; s++) {
|
|
|
+ if (*s != SC_GROUP_IDENTIFIER) missing++;
|
|
|
+ size_t len = _tcslen(s);
|
|
|
+ canonlen += len + 1;
|
|
|
+ s += len;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (missing) {
|
|
|
+ /* Missing identifiers plus double NULL terminator. */
|
|
|
+ canonlen += missing + 1;
|
|
|
+ newlen = (unsigned long) canonlen;
|
|
|
+
|
|
|
+ canon = (TCHAR *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, canonlen * sizeof(TCHAR));
|
|
|
+ if (! canon) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("canon"), _T("native_set_dependongroup"));
|
|
|
+ if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ size_t i = 0;
|
|
|
+ for (s = unformatted; *s; s++) {
|
|
|
+ if (*s != SC_GROUP_IDENTIFIER) canon[i++] = SC_GROUP_IDENTIFIER;
|
|
|
+ size_t len = _tcslen(s);
|
|
|
+ memmove(canon + i, s, (len + 1) * sizeof(TCHAR));
|
|
|
+ i += len + 1;
|
|
|
+ s += len;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ TCHAR *dependencies;
|
|
|
+ if (buflen > 2) {
|
|
|
+ dependencies = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (newlen + buflen) * sizeof(TCHAR));
|
|
|
+ if (! dependencies) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dependencies"), _T("native_set_dependongroup"));
|
|
|
+ if (canon != unformatted) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
+ if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ memmove(dependencies, buffer, buflen * sizeof(TCHAR));
|
|
|
+ memmove(dependencies + buflen - 1, canon, newlen * sizeof(TCHAR));
|
|
|
+ }
|
|
|
+ else dependencies = canon;
|
|
|
+
|
|
|
+ int ret = 1;
|
|
|
+ if (set_service_dependencies(service_name, service_handle, dependencies)) ret = -1;
|
|
|
+ if (dependencies != unformatted) HeapFree(GetProcessHeap(), 0, dependencies);
|
|
|
+ if (canon != unformatted) HeapFree(GetProcessHeap(), 0, canon);
|
|
|
+ if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+static int native_get_dependongroup(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;
|
|
|
+ unsigned long buflen;
|
|
|
+ if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_GROUPS)) return -1;
|
|
|
+
|
|
|
+ int ret;
|
|
|
+ if (buflen) {
|
|
|
+ TCHAR *formatted;
|
|
|
+ unsigned long newlen;
|
|
|
+ if (format_double_null(buffer, buflen, &formatted, &newlen)) {
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ ret = value_from_string(name, value, formatted);
|
|
|
+ HeapFree(GetProcessHeap(), 0, formatted);
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ }
|
|
|
+ else {
|
|
|
+ value->string = 0;
|
|
|
+ ret = 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+static int native_set_dependonservice(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;
|
|
|
+
|
|
|
+ /*
|
|
|
+ Get existing group dependencies because we must set both types together.
|
|
|
+ */
|
|
|
+ TCHAR *buffer;
|
|
|
+ unsigned long buflen;
|
|
|
+ if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_GROUPS)) return -1;
|
|
|
+
|
|
|
+ if (! value || ! value->string || ! value->string[0]) {
|
|
|
+ if (! ChangeServiceConfig(service_handle, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, 0, 0, 0, buffer, 0, 0, 0)) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ unsigned long len = (unsigned long) _tcslen(value->string) + 1;
|
|
|
+ TCHAR *unformatted = 0;
|
|
|
+ unsigned long newlen;
|
|
|
+ if (unformat_double_null(value->string, len, &unformatted, &newlen)) {
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ TCHAR *dependencies;
|
|
|
+ if (buflen > 2) {
|
|
|
+ dependencies = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, (newlen + buflen) * sizeof(TCHAR));
|
|
|
+ if (! dependencies) {
|
|
|
+ print_message(stderr, NSSM_MESSAGE_OUT_OF_MEMORY, _T("dependencies"), _T("native_set_dependonservice"));
|
|
|
+ if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ memmove(dependencies, buffer, buflen * sizeof(TCHAR));
|
|
|
+ memmove(dependencies + buflen - 1, unformatted, newlen * sizeof(TCHAR));
|
|
|
+ }
|
|
|
+ else dependencies = unformatted;
|
|
|
+
|
|
|
+ int ret = 1;
|
|
|
+ if (set_service_dependencies(service_name, service_handle, dependencies)) ret = -1;
|
|
|
+ if (dependencies != unformatted) HeapFree(GetProcessHeap(), 0, dependencies);
|
|
|
+ if (unformatted) HeapFree(GetProcessHeap(), 0, unformatted);
|
|
|
+ if (buffer) HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+static int native_get_dependonservice(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;
|
|
|
+ unsigned long buflen;
|
|
|
+ if (get_service_dependencies(service_name, service_handle, &buffer, &buflen, DEPENDENCY_SERVICES)) return -1;
|
|
|
+
|
|
|
+ int ret;
|
|
|
+ if (buflen) {
|
|
|
+ TCHAR *formatted;
|
|
|
+ unsigned long newlen;
|
|
|
+ if (format_double_null(buffer, buflen, &formatted, &newlen)) {
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ ret = value_from_string(name, value, formatted);
|
|
|
+ HeapFree(GetProcessHeap(), 0, formatted);
|
|
|
+ HeapFree(GetProcessHeap(), 0, buffer);
|
|
|
+ }
|
|
|
+ else {
|
|
|
+ value->string = 0;
|
|
|
+ ret = 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+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_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
+ HKEY key = open_service_registry(service_name, KEY_SET_VALUE, false);
|
|
|
+ if (! key) return -1;
|
|
|
+
|
|
|
+ int ret = setting_set_environment(service_name, (void *) key, NSSM_NATIVE_ENVIRONMENT, default_value, value, additional);
|
|
|
+ RegCloseKey(key);
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+int native_get_environment(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
|
+ HKEY key = open_service_registry(service_name, KEY_READ, false);
|
|
|
+ if (! key) return -1;
|
|
|
+
|
|
|
+ ZeroMemory(value, sizeof(value_t));
|
|
|
+ int ret = setting_get_environment(service_name, (void *) key, NSSM_NATIVE_ENVIRONMENT, default_value, value, additional);
|
|
|
+ RegCloseKey(key);
|
|
|
+ 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 = false;
|
|
|
+ TCHAR *username = NSSM_LOCALSYSTEM_ACCOUNT;
|
|
|
+ TCHAR *password = 0;
|
|
|
+ if (additional) {
|
|
|
+ username = (TCHAR *) additional;
|
|
|
+ if (value && value->string) password = value->string;
|
|
|
+ }
|
|
|
+ else if (value && value->string) username = value->string;
|
|
|
+
|
|
|
+ const TCHAR *well_known = well_known_username(username);
|
|
|
+ size_t passwordsize = 0;
|
|
|
+ if (well_known) {
|
|
|
+ if (str_equiv(well_known, NSSM_LOCALSYSTEM_ACCOUNT)) localsystem = true;
|
|
|
+ username = (TCHAR *) well_known;
|
|
|
+ password = _T("");
|
|
|
+ }
|
|
|
+ else if (! password) {
|
|
|
+ /* We need a password if the account requires it. */
|
|
|
+ print_message(stderr, NSSM_MESSAGE_MISSING_PASSWORD, name);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ else passwordsize = _tcslen(password) * sizeof(TCHAR);
|
|
|
+
|
|
|
+ /*
|
|
|
+ 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 (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ type = qsc->dwServiceType & ~SERVICE_INTERACTIVE_PROCESS;
|
|
|
+ HeapFree(GetProcessHeap(), 0, qsc);
|
|
|
+ }
|
|
|
+
|
|
|
+ if (! well_known) {
|
|
|
+ if (grant_logon_as_service(username)) {
|
|
|
+ if (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
+ 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 (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
+ print_message(stderr, NSSM_MESSAGE_CHANGESERVICECONFIG_FAILED, error_string(GetLastError()));
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (passwordsize) SecureZeroMemory(password, passwordsize);
|
|
|
+
|
|
|
+ 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 = PtrToUlong(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_HOOK, REG_SZ, (void *) _T(""), false, ADDITIONAL_MANDATORY, setting_set_hook, setting_get_hook },
|
|
|
+ { NSSM_REG_AFFINITY, REG_SZ, 0, false, 0, setting_set_affinity, setting_get_affinity },
|
|
|
+ { 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_NO_CONSOLE, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
|
|
|
+ { NSSM_REG_PRIORITY, REG_SZ, (void *) priority_strings[NSSM_NORMAL_PRIORITY], false, 0, setting_set_priority, setting_get_priority },
|
|
|
+ { NSSM_REG_RESTART_DELAY, REG_DWORD, 0, false, 0, setting_set_number, setting_get_number },
|
|
|
+ { 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_STDOUT NSSM_REG_STDIO_COPY_AND_TRUNCATE, REG_DWORD, 0, 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_STDERR NSSM_REG_STDIO_COPY_AND_TRUNCATE, REG_DWORD, 0, 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_KILL_PROCESS_TREE, REG_DWORD, (void *) 1, 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_ONLINE, 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_REG_ROTATE_DELAY, REG_DWORD, (void *) NSSM_ROTATE_DELAY, false, 0, setting_set_number, setting_get_number },
|
|
|
+ { NSSM_NATIVE_DEPENDONGROUP, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_dependongroup, native_get_dependongroup },
|
|
|
+ { NSSM_NATIVE_DEPENDONSERVICE, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_dependonservice, native_get_dependonservice },
|
|
|
+ { 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_ENVIRONMENT, REG_MULTI_SZ, NULL, true, ADDITIONAL_CRLF, native_set_environment, native_get_environment },
|
|
|
+ { NSSM_NATIVE_IMAGEPATH, REG_EXPAND_SZ, NULL, true, 0, native_set_imagepath, native_get_imagepath },
|
|
|
+ { NSSM_NATIVE_OBJECTNAME, REG_SZ, NSSM_LOCALSYSTEM_ACCOUNT, true, 0, 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 }
|
|
|
+};
|
Iain Patterson