mirror of http://git.nssm.cc/nssm/nssm.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1111 lines
42 KiB
C++
1111 lines
42 KiB
C++
#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 == (unsigned long) default_value) {
|
|
error = RegDeleteValue(key, name);
|
|
if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
|
|
print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
|
|
return -1;
|
|
}
|
|
|
|
if (set_number(key, (TCHAR *) name, number)) return -1;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int setting_get_number(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
HKEY key = (HKEY) param;
|
|
return get_number(key, (TCHAR *) name, &value->numeric, false);
|
|
}
|
|
|
|
static int setting_set_string(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
HKEY key = (HKEY) param;
|
|
if (! key) return -1;
|
|
|
|
long error;
|
|
|
|
/* Resetting to default? */
|
|
if (! value || ! value->string) {
|
|
if (default_value) value->string = (TCHAR *) default_value;
|
|
else {
|
|
error = RegDeleteValue(key, name);
|
|
if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
|
|
print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
|
|
return -1;
|
|
}
|
|
}
|
|
if (default_value && _tcslen((TCHAR *) default_value) && str_equiv(value->string, (TCHAR *) default_value)) {
|
|
error = RegDeleteValue(key, name);
|
|
if (error == ERROR_SUCCESS || error == ERROR_FILE_NOT_FOUND) return 0;
|
|
print_message(stderr, NSSM_MESSAGE_REGDELETEVALUE_FAILED, name, service_name, error_string(error));
|
|
return -1;
|
|
}
|
|
|
|
if (set_expand_string(key, (TCHAR *) name, value->string)) return -1;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int setting_get_string(const TCHAR *service_name, void *param, const TCHAR *name, void *default_value, value_t *value, const TCHAR *additional) {
|
|
HKEY key = (HKEY) param;
|
|
TCHAR buffer[VALUE_LENGTH];
|
|
|
|
if (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_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 = (unsigned long) setting->default_value;
|
|
if (setting->get) ret = setting->get(service_name, (void *) key, setting->name, setting->default_value, value, additional);
|
|
else ret = -1;
|
|
break;
|
|
|
|
default:
|
|
ret = -1;
|
|
break;
|
|
}
|
|
|
|
if (ret < 0) print_message(stderr, NSSM_MESSAGE_GET_SETTING_FAILED, setting->name, service_name);
|
|
|
|
return ret;
|
|
}
|
|
|
|
int get_setting(const TCHAR *service_name, SC_HANDLE service_handle, settings_t *setting, value_t *value, const TCHAR *additional) {
|
|
if (! service_handle) return -1;
|
|
return setting->get(service_name, service_handle, setting->name, 0, value, additional);
|
|
}
|
|
|
|
settings_t settings[] = {
|
|
{ NSSM_REG_EXE, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string },
|
|
{ NSSM_REG_FLAGS, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string },
|
|
{ NSSM_REG_DIR, REG_EXPAND_SZ, (void *) _T(""), false, 0, setting_set_string, setting_get_string },
|
|
{ NSSM_REG_EXIT, REG_SZ, (void *) exit_action_strings[NSSM_EXIT_RESTART], false, ADDITIONAL_MANDATORY, setting_set_exit_action, setting_get_exit_action },
|
|
{ NSSM_REG_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_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 }
|
|
};
|