/* Service initialisation */
void WINAPI service_main(unsigned long argc, char **argv) {
if (_snprintf(service_name, sizeof(service_name), "%s", argv[0]) < 0) {
log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, "service_name", "service_main()", 0);
return;
}
/* Initialise status */
ZeroMemory(&service_status, sizeof(service_status));
service_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS;
service_status.dwControlsAccepted = SERVICE_ACCEPT_SHUTDOWN | SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE;
service_status.dwWin32ExitCode = NO_ERROR;
service_status.dwServiceSpecificExitCode = 0;
service_status.dwCheckPoint = 0;
service_status.dwWaitHint = NSSM_WAITHINT_MARGIN;
/* Signal we AREN'T running the server */
process_handle = 0;
pid = 0;
/* Register control handler */
service_handle = RegisterServiceCtrlHandlerEx(NSSM, service_control_handler, 0);
if (! service_handle) {
log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_REGISTERSERVICECTRLHANDER_FAILED, error_string(GetLastError()), 0);
return;
}
log_service_control(service_name, 0, true);
service_status.dwCurrentState = SERVICE_START_PENDING;
service_status.dwWaitHint = throttle_delay + NSSM_WAITHINT_MARGIN;
SetServiceStatus(service_handle, &service_status);
if (is_admin) {
/* Try to create the exit action parameters; we don't care if it fails */
create_exit_action(argv[0], exit_action_strings[0]);
set_service_recovery(0, service_name);
}
/* Used for signalling a resume if the service pauses when throttled. */
throttle_timer = CreateWaitableTimer(0, 1, 0);
if (! throttle_timer) {
log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_CREATEWAITABLETIMER_FAILED, service_name, error_string(GetLastError()), 0);
}
monitor_service();
}
int create_parameters(nssm_service_t *service, bool editing) {
/* Try to open the registry */
HKEY key = open_registry(service->name, KEY_WRITE);
if (! key) return 1;
/* Remember parameters in case we need to delete them. */
TCHAR registry[KEY_LENGTH];
int ret = service_registry_path(service->name, true, 0, registry, _countof(registry));
/* Try to create the parameters */
if (set_expand_string(key, NSSM_REG_EXE, service->exe)) {
if (ret > 0) RegDeleteKey(HKEY_LOCAL_MACHINE, registry);
RegCloseKey(key);
return 2;
}
if (set_expand_string(key, NSSM_REG_FLAGS, service->flags)) {
if (ret > 0) RegDeleteKey(HKEY_LOCAL_MACHINE, registry);
RegCloseKey(key);
return 3;
}
if (set_expand_string(key, NSSM_REG_DIR, service->dir)) {
if (ret > 0) RegDeleteKey(HKEY_LOCAL_MACHINE, registry);
RegCloseKey(key);
return 4;
}
/* Other non-default parameters. May fail. */
if (service->priority != NORMAL_PRIORITY_CLASS) set_number(key, NSSM_REG_PRIORITY, service->priority);
else if (editing) RegDeleteValue(key, NSSM_REG_PRIORITY);
if (service->affinity) {
TCHAR *string;
if (! affinity_mask_to_string(service->affinity, &string)) {
if (RegSetValueEx(key, NSSM_REG_AFFINITY, 0, REG_SZ, (const unsigned char *) string, (unsigned long) (_tcslen(string) + 1) * sizeof(TCHAR)) != ERROR_SUCCESS) {
log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, NSSM_REG_AFFINITY, error_string(GetLastError()), 0);
HeapFree(GetProcessHeap(), 0, string);
return 5;
}
}
if (string) HeapFree(GetProcessHeap(), 0, string);
}
else if (editing) RegDeleteValue(key, NSSM_REG_AFFINITY);
unsigned long stop_method_skip = ~service->stop_method;
if (stop_method_skip) set_number(key, NSSM_REG_STOP_METHOD_SKIP, stop_method_skip);
else if (editing) RegDeleteValue(key, NSSM_REG_STOP_METHOD_SKIP);
if (service->default_exit_action < NSSM_NUM_EXIT_ACTIONS) create_exit_action(service->name, exit_action_strings[service->default_exit_action], editing);
if (service->restart_delay) set_number(key, NSSM_REG_RESTART_DELAY, service->restart_delay);
else if (editing) RegDeleteValue(key, NSSM_REG_RESTART_DELAY);
if (service->throttle_delay != NSSM_RESET_THROTTLE_RESTART) set_number(key, NSSM_REG_THROTTLE, service->throttle_delay);
else if (editing) RegDeleteValue(key, NSSM_REG_THROTTLE);
if (service->kill_console_delay != NSSM_KILL_CONSOLE_GRACE_PERIOD) set_number(key, NSSM_REG_KILL_CONSOLE_GRACE_PERIOD, service->kill_console_delay);
else if (editing) RegDeleteValue(key, NSSM_REG_KILL_CONSOLE_GRACE_PERIOD);
if (service->kill_window_delay != NSSM_KILL_WINDOW_GRACE_PERIOD) set_number(key, NSSM_REG_KILL_WINDOW_GRACE_PERIOD, service->kill_window_delay);
else if (editing) RegDeleteValue(key, NSSM_REG_KILL_WINDOW_GRACE_PERIOD);
if (service->kill_threads_delay != NSSM_KILL_THREADS_GRACE_PERIOD) set_number(key, NSSM_REG_KILL_THREADS_GRACE_PERIOD, service->kill_threads_delay);
else if (editing) RegDeleteValue(key, NSSM_REG_KILL_THREADS_GRACE_PERIOD);
if (! service->kill_process_tree) set_number(key, NSSM_REG_KILL_PROCESS_TREE, 0);
else if (editing) RegDeleteValue(key, NSSM_REG_KILL_PROCESS_TREE);
if (service->stdin_path[0] || editing) {
if (service->stdin_path[0]) set_expand_string(key, NSSM_REG_STDIN, service->stdin_path);
else if (editing) RegDeleteValue(key, NSSM_REG_STDIN);
if (service->stdin_sharing != NSSM_STDIN_SHARING) set_createfile_parameter(key, NSSM_REG_STDIN, NSSM_REG_STDIO_SHARING, service->stdin_sharing);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDIN, NSSM_REG_STDIO_SHARING);
if (service->stdin_disposition != NSSM_STDIN_DISPOSITION) set_createfile_parameter(key, NSSM_REG_STDIN, NSSM_REG_STDIO_DISPOSITION, service->stdin_disposition);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDIN, NSSM_REG_STDIO_DISPOSITION);
if (service->stdin_flags != NSSM_STDIN_FLAGS) set_createfile_parameter(key, NSSM_REG_STDIN, NSSM_REG_STDIO_FLAGS, service->stdin_flags);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDIN, NSSM_REG_STDIO_FLAGS);
}
if (service->stdout_path[0] || editing) {
if (service->stdout_path[0]) set_expand_string(key, NSSM_REG_STDOUT, service->stdout_path);
else if (editing) RegDeleteValue(key, NSSM_REG_STDOUT);
if (service->stdout_sharing != NSSM_STDOUT_SHARING) set_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_SHARING, service->stdout_sharing);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_SHARING);
if (service->stdout_disposition != NSSM_STDOUT_DISPOSITION) set_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_DISPOSITION, service->stdout_disposition);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_DISPOSITION);
if (service->stdout_flags != NSSM_STDOUT_FLAGS) set_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_FLAGS, service->stdout_flags);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_FLAGS);
if (service->stdout_copy_and_truncate) set_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_COPY_AND_TRUNCATE, 1);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDOUT, NSSM_REG_STDIO_COPY_AND_TRUNCATE);
}
if (service->stderr_path[0] || editing) {
if (service->stderr_path[0]) set_expand_string(key, NSSM_REG_STDERR, service->stderr_path);
else if (editing) RegDeleteValue(key, NSSM_REG_STDERR);
if (service->stderr_sharing != NSSM_STDERR_SHARING) set_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_SHARING, service->stderr_sharing);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_SHARING);
if (service->stderr_disposition != NSSM_STDERR_DISPOSITION) set_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_DISPOSITION, service->stderr_disposition);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_DISPOSITION);
if (service->stderr_flags != NSSM_STDERR_FLAGS) set_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_FLAGS, service->stderr_flags);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_FLAGS);
if (service->stderr_copy_and_truncate) set_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_COPY_AND_TRUNCATE, 1);
else if (editing) delete_createfile_parameter(key, NSSM_REG_STDERR, NSSM_REG_STDIO_COPY_AND_TRUNCATE);
}
if (service->timestamp_log) set_number(key, NSSM_REG_TIMESTAMP_LOG, 1);
else if (editing) RegDeleteValue(key, NSSM_REG_TIMESTAMP_LOG);
if (service->hook_share_output_handles) set_number(key, NSSM_REG_HOOK_SHARE_OUTPUT_HANDLES, 1);
else if (editing) RegDeleteValue(key, NSSM_REG_HOOK_SHARE_OUTPUT_HANDLES);
if (service->rotate_files) set_number(key, NSSM_REG_ROTATE, 1);
else if (editing) RegDeleteValue(key, NSSM_REG_ROTATE);
if (service->rotate_stdout_online) set_number(key, NSSM_REG_ROTATE_ONLINE, 1);
else if (editing) RegDeleteValue(key, NSSM_REG_ROTATE_ONLINE);
if (service->rotate_seconds) set_number(key, NSSM_REG_ROTATE_SECONDS, service->rotate_seconds);
else if (editing) RegDeleteValue(key, NSSM_REG_ROTATE_SECONDS);
if (service->rotate_bytes_low) set_number(key, NSSM_REG_ROTATE_BYTES_LOW, service->rotate_bytes_low);
else if (editing) RegDeleteValue(key, NSSM_REG_ROTATE_BYTES_LOW);
if (service->rotate_bytes_high) set_number(key, NSSM_REG_ROTATE_BYTES_HIGH, service->rotate_bytes_high);
else if (editing) RegDeleteValue(key, NSSM_REG_ROTATE_BYTES_HIGH);
if (service->rotate_delay != NSSM_ROTATE_DELAY) set_number(key, NSSM_REG_ROTATE_DELAY, service->rotate_delay);
else if (editing) RegDeleteValue(key, NSSM_REG_ROTATE_DELAY);
if (service->no_console) set_number(key, NSSM_REG_NO_CONSOLE, 1);
else if (editing) RegDeleteValue(key, NSSM_REG_NO_CONSOLE);
/* Environment */
if (service->env) {
if (RegSetValueEx(key, NSSM_REG_ENV, 0, REG_MULTI_SZ, (const unsigned char *) service->env, (unsigned long) service->envlen * sizeof(TCHAR)) != ERROR_SUCCESS) {
log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, NSSM_REG_ENV, error_string(GetLastError()), 0);
}
}
else if (editing) RegDeleteValue(key, NSSM_REG_ENV);
if (service->env_extra) {
if (RegSetValueEx(key, NSSM_REG_ENV_EXTRA, 0, REG_MULTI_SZ, (const unsigned char *) service->env_extra, (unsigned long) service->env_extralen * sizeof(TCHAR)) != ERROR_SUCCESS) {
log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETVALUE_FAILED, NSSM_REG_ENV_EXTRA, error_string(GetLastError()), 0);
}
}
else if (editing) RegDeleteValue(key, NSSM_REG_ENV_EXTRA);
/* Close registry. */
RegCloseKey(key);
return 0;
}
