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;
unsigned long envlen = (unsigned long) _tcslen(value->string) + 1;
TCHAR *unformatted = 0;
unsigned long newlen;
- if (unformat_environment(value->string, envlen, &unformatted, &newlen)) return -1;
+ if (unformat_double_null(value->string, envlen, &unformatted, &newlen)) return -1;
if (test_environment(unformatted)) {
HeapFree(GetProcessHeap(), 0, unformatted);
TCHAR *formatted;
unsigned long newlen;
- if (format_environment(env, envlen, &formatted, &newlen)) return -1;
+ if (format_double_null(env, envlen, &formatted, &newlen)) return -1;
int ret;
if (additional) {
}
/* 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;
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;
{ 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_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 },
git://git.iain.cx/iain / nssm.git / blobdiff