int last;\r
} list_t;\r
\r
+/*\r
+ Check the status in response to a control.\r
+ Returns: 1 if the status is expected, eg STOP following CONTROL_STOP.\r
+ 0 if the status is desired, eg STOPPED following CONTROL_STOP.\r
+ -1 if the status is undesired, eg STOPPED following CONTROL_START.\r
+*/\r
+static inline int service_control_response(unsigned long control, unsigned long status) {\r
+ switch (control) {\r
+ case NSSM_SERVICE_CONTROL_START:\r
+ switch (status) {\r
+ case SERVICE_START_PENDING:\r
+ return 1;\r
+\r
+ case SERVICE_RUNNING:\r
+ return 0;\r
+\r
+ default:\r
+ return -1;\r
+ }\r
+\r
+ case SERVICE_CONTROL_STOP:\r
+ case SERVICE_CONTROL_SHUTDOWN:\r
+ switch (status) {\r
+ case SERVICE_STOP_PENDING:\r
+ return 1;\r
+\r
+ case SERVICE_STOPPED:\r
+ return 0;\r
+\r
+ default:\r
+ return -1;\r
+ }\r
+\r
+ case SERVICE_CONTROL_PAUSE:\r
+ switch (status) {\r
+ case SERVICE_PAUSE_PENDING:\r
+ return 1;\r
+\r
+ case SERVICE_PAUSED:\r
+ return 0;\r
+\r
+ default:\r
+ return -1;\r
+ }\r
+\r
+ case SERVICE_CONTROL_CONTINUE:\r
+ switch (status) {\r
+ case SERVICE_CONTINUE_PENDING:\r
+ return 1;\r
+\r
+ case SERVICE_RUNNING:\r
+ return 0;\r
+\r
+ default:\r
+ return -1;\r
+ }\r
+\r
+ case SERVICE_CONTROL_INTERROGATE:\r
+ return 0;\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
+static inline int await_service_control_response(unsigned long control, SC_HANDLE service_handle, SERVICE_STATUS *service_status, unsigned long initial_status) {\r
+ int tries = 0;\r
+ while (QueryServiceStatus(service_handle, service_status)) {\r
+ int response = service_control_response(control, service_status->dwCurrentState);\r
+ /* Alas we can't WaitForSingleObject() on an SC_HANDLE. */\r
+ if (! response) return response;\r
+ if (response > 0 || service_status->dwCurrentState == initial_status) {\r
+ if (++tries > 10) return response;\r
+ Sleep(50 * tries);\r
+ }\r
+ else return response;\r
+ }\r
+ return -1;\r
+}\r
+\r
int affinity_mask_to_string(__int64 mask, TCHAR **string) {\r
if (! string) return 1;\r
if (! mask) {\r
if (service->wait_handle) UnregisterWait(service->process_handle);\r
if (service->throttle_section_initialised) DeleteCriticalSection(&service->throttle_section);\r
if (service->throttle_timer) CloseHandle(service->throttle_timer);\r
+ if (service->initial_env) FreeEnvironmentStrings(service->initial_env);\r
HeapFree(GetProcessHeap(), 0, service);\r
}\r
\r
for (i = 0; settings[i].name; i++) _ftprintf(stderr, _T("%s\n"), settings[i].name);\r
return 1;\r
}\r
- if (argc < mandatory) return usage(1);\r
\r
additional = 0;\r
if (additional_mandatory) {\r
+ if (argc < mandatory) {\r
+ print_message(stderr, NSSM_MESSAGE_MISSING_SUBPARAMETER, parameter);\r
+ return 1;\r
+ }\r
additional = argv[3];\r
remainder = 4;\r
}\r
- else additional = argv[remainder];\r
+ else {\r
+ additional = argv[remainder];\r
+ if (argc < mandatory) return usage(1);\r
+ }\r
}\r
\r
nssm_service_t *service = alloc_nssm_service();\r
int ret;\r
unsigned long error;\r
SERVICE_STATUS service_status;\r
- if (control == 0) {\r
+ if (control == NSSM_SERVICE_CONTROL_START) {\r
+ unsigned long initial_status = SERVICE_STOPPED;\r
ret = StartService(service_handle, (unsigned long) argc, (const TCHAR **) argv);\r
error = GetLastError();\r
- CloseHandle(service_handle);\r
CloseServiceHandle(services);\r
\r
if (error == ERROR_IO_PENDING) {\r
/*\r
Older versions of Windows return immediately with ERROR_IO_PENDING\r
indicate that the operation is still in progress. Newer versions\r
- will return it if there really is a delay. As far as we're\r
- concerned the operation is a success. We don't claim to offer a\r
- fully-feature service control method; it's just a quick 'n' dirty\r
- interface.\r
-\r
- In the future we may identify and handle this situation properly.\r
+ will return it if there really is a delay.\r
*/\r
ret = 1;\r
error = ERROR_SUCCESS;\r
}\r
\r
if (ret) {\r
- _tprintf(_T("%s: %s"), canonical_name, error_string(error));\r
+ int response = await_service_control_response(control, service_handle, &service_status, initial_status);\r
+ CloseHandle(service_handle);\r
+\r
+ if (response) {\r
+ print_message(stderr, NSSM_MESSAGE_BAD_CONTROL_RESPONSE, canonical_name, service_status_text(service_status.dwCurrentState), service_control_text(control));\r
+ return 1;\r
+ }\r
+ else _tprintf(_T("%s: %s: %s"), canonical_name, service_control_text(control), error_string(error));\r
return 0;\r
}\r
else {\r
- _ftprintf(stderr, _T("%s: %s"), canonical_name, error_string(error));\r
+ CloseHandle(service_handle);\r
+ _ftprintf(stderr, _T("%s: %s: %s"), canonical_name, service_control_text(control), error_string(error));\r
return 1;\r
}\r
}\r
error = GetLastError();\r
\r
if (ret) {\r
- switch (service_status.dwCurrentState) {\r
- case SERVICE_STOPPED: _tprintf(_T("SERVICE_STOPPED\n")); break;\r
- case SERVICE_START_PENDING: _tprintf(_T("SERVICE_START_PENDING\n")); break;\r
- case SERVICE_STOP_PENDING: _tprintf(_T("SERVICE_STOP_PENDING\n")); break;\r
- case SERVICE_RUNNING: _tprintf(_T("SERVICE_RUNNING\n")); break;\r
- case SERVICE_CONTINUE_PENDING: _tprintf(_T("SERVICE_CONTINUE_PENDING\n")); break;\r
- case SERVICE_PAUSE_PENDING: _tprintf(_T("SERVICE_PAUSE_PENDING\n")); break;\r
- case SERVICE_PAUSED: _tprintf(_T("SERVICE_PAUSED\n")); break;\r
- default: _tprintf(_T("?\n")); return 1;\r
- }\r
+ _tprintf(_T("%s\n"), service_status_text(service_status.dwCurrentState));\r
return 0;\r
}\r
else {\r
}\r
else {\r
ret = ControlService(service_handle, control, &service_status);\r
+ unsigned long initial_status = service_status.dwCurrentState;\r
error = GetLastError();\r
- CloseHandle(service_handle);\r
CloseServiceHandle(services);\r
\r
if (error == ERROR_IO_PENDING) {\r
}\r
\r
if (ret) {\r
- _tprintf(_T("%s: %s"), canonical_name, error_string(error));\r
+ int response = await_service_control_response(control, service_handle, &service_status, initial_status);\r
+ CloseHandle(service_handle);\r
+\r
+ if (response) {\r
+ print_message(stderr, NSSM_MESSAGE_BAD_CONTROL_RESPONSE, canonical_name, service_status_text(service_status.dwCurrentState), service_control_text(control));\r
+ return 1;\r
+ }\r
+ else _tprintf(_T("%s: %s: %s"), canonical_name, service_control_text(control), error_string(error));\r
return 0;\r
}\r
else {\r
- _ftprintf(stderr, _T("%s: %s"), canonical_name, error_string(error));\r
+ CloseHandle(service_handle);\r
+ _ftprintf(stderr, _T("%s: %s: %s"), canonical_name, service_control_text(control), error_string(error));\r
+ if (error == ERROR_SERVICE_NOT_ACTIVE) {\r
+ if (control == SERVICE_CONTROL_SHUTDOWN || control == SERVICE_CONTROL_STOP) return 0;\r
+ }\r
return 1;\r
}\r
}\r
}\r
}\r
\r
+ /* Remember our initial environment. */\r
+ service->initial_env = GetEnvironmentStrings();\r
+\r
monitor_service(service);\r
}\r
\r
TCHAR *service_control_text(unsigned long control) {\r
switch (control) {\r
/* HACK: there is no SERVICE_CONTROL_START constant */\r
- case 0: return _T("START");\r
+ case NSSM_SERVICE_CONTROL_START: return _T("START");\r
case SERVICE_CONTROL_STOP: return _T("STOP");\r
case SERVICE_CONTROL_SHUTDOWN: return _T("SHUTDOWN");\r
case SERVICE_CONTROL_PAUSE: return _T("PAUSE");\r
case SERVICE_CONTROL_CONTINUE: return _T("CONTINUE");\r
case SERVICE_CONTROL_INTERROGATE: return _T("INTERROGATE");\r
+ case NSSM_SERVICE_CONTROL_ROTATE: return _T("ROTATE");\r
+ default: return 0;\r
+ }\r
+}\r
+\r
+TCHAR *service_status_text(unsigned long status) {\r
+ switch (status) {\r
+ case SERVICE_STOPPED: return _T("SERVICE_STOPPED");\r
+ case SERVICE_START_PENDING: return _T("SERVICE_START_PENDING");\r
+ case SERVICE_STOP_PENDING: return _T("SERVICE_STOP_PENDING");\r
+ case SERVICE_RUNNING: return _T("SERVICE_RUNNING");\r
+ case SERVICE_CONTINUE_PENDING: return _T("SERVICE_CONTINUE_PENDING");\r
+ case SERVICE_PAUSE_PENDING: return _T("SERVICE_PAUSE_PENDING");\r
+ case SERVICE_PAUSED: return _T("SERVICE_PAUSED");\r
default: return 0;\r
}\r
}\r
ZeroMemory(&service->throttle_duetime, sizeof(service->throttle_duetime));\r
SetWaitableTimer(service->throttle_timer, &service->throttle_duetime, 0, 0, 0, 0);\r
}\r
- service->status.dwCurrentState = SERVICE_CONTINUE_PENDING;\r
+ /* We can't continue if the application is running! */\r
+ if (! service->process_handle) service->status.dwCurrentState = SERVICE_CONTINUE_PENDING;\r
service->status.dwWaitHint = throttle_milliseconds(service->throttle) + NSSM_WAITHINT_MARGIN;\r
log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_RESET_THROTTLE, service->name, 0);\r
SetServiceStatus(service->status_handle, &service->status);\r
*/\r
log_service_control(service->name, control, false);\r
return ERROR_CALL_NOT_IMPLEMENTED;\r
+\r
+ case NSSM_SERVICE_CONTROL_ROTATE:\r
+ log_service_control(service->name, control, true);\r
+ if (service->rotate_stdout_online == NSSM_ROTATE_ONLINE) service->rotate_stdout_online = NSSM_ROTATE_ONLINE_ASAP;\r
+ if (service->rotate_stderr_online == NSSM_ROTATE_ONLINE) service->rotate_stderr_online = NSSM_ROTATE_ONLINE_ASAP;\r
+ return NO_ERROR;\r
}\r
\r
/* Unknown control */\r
\r
throttle_restart(service);\r
\r
+ /* Set the environment. */\r
+ if (service->env) duplicate_environment(service->env);\r
+ if (service->env_extra) set_environment_block(service->env_extra);\r
+\r
bool inherit_handles = false;\r
if (si.dwFlags & STARTF_USESTDHANDLES) inherit_handles = true;\r
unsigned long flags = service->priority & priority_mask();\r
+ if (service->stdin_pipe) flags |= DETACHED_PROCESS;\r
if (service->affinity) flags |= CREATE_SUSPENDED;\r
-#ifdef UNICODE\r
- flags |= CREATE_UNICODE_ENVIRONMENT;\r
-#endif\r
- if (! CreateProcess(0, cmd, 0, 0, inherit_handles, flags, service->env, service->dir, &si, &pi)) {\r
+ if (! CreateProcess(0, cmd, 0, 0, inherit_handles, flags, 0, service->dir, &si, &pi)) {\r
unsigned long exitcode = 3;\r
unsigned long error = GetLastError();\r
- if (error == ERROR_INVALID_PARAMETER && service->env) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEPROCESS_FAILED_INVALID_ENVIRONMENT, service->name, service->exe, NSSM_REG_ENV, 0);\r
- if (test_environment(service->env)) exitcode = 4;\r
- }\r
- else log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEPROCESS_FAILED, service->name, service->exe, error_string(error), 0);\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEPROCESS_FAILED, service->name, service->exe, error_string(error), 0);\r
close_output_handles(&si);\r
+ duplicate_environment(service->initial_env);\r
return stop_service(service, exitcode, true, true);\r
}\r
service->process_handle = pi.hProcess;\r
\r
if (get_process_creation_time(service->process_handle, &service->creation_time)) ZeroMemory(&service->creation_time, sizeof(service->creation_time));\r
\r
- close_output_handles(&si, ! service->rotate_stdout_online, ! service->rotate_stderr_online);\r
+ close_output_handles(&si);\r
+\r
+ /* Restore our environment. */\r
+ duplicate_environment(service->initial_env);\r
\r
if (service->affinity) {\r
/*\r
UnregisterWait(service->wait_handle);\r
service->wait_handle = 0;\r
}\r
+ if (service->stdin_pipe) {\r
+ CloseHandle(service->stdin_pipe);\r
+ service->stdin_pipe = 0;\r
+ }\r
+\r
+ service->rotate_stdout_online = service->rotate_stderr_online = NSSM_ROTATE_OFFLINE;\r
\r
if (default_action && ! exitcode && ! graceful) {\r
log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_GRACEFUL_SUICIDE, service->name, service->exe, exit_action_strings[NSSM_EXIT_UNCLEAN], exit_action_strings[NSSM_EXIT_UNCLEAN], exit_action_strings[NSSM_EXIT_UNCLEAN], exit_action_strings[NSSM_EXIT_REALLY], 0);\r
\r
service->stopping = true;\r
\r
+ service->rotate_stdout_online = service->rotate_stderr_online = NSSM_ROTATE_OFFLINE;\r
+\r
/* Check exit code */\r
unsigned long exitcode = 0;\r
TCHAR code[16];\r