X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=service.cpp;h=81e46efec92e2ca227466a5e58f7210e05adfe44;hb=9884e231415258809dfb1ef117b6bc9a5b011514;hp=766031ee9e9a3e186b42b07a014d32bdec996323;hpb=f2ec1c0c55a6b3e8ca02b3d66b78c87fe0ac1f47;p=nssm.git diff --git a/service.cpp b/service.cpp index 766031e..81e46ef 100644 --- a/service.cpp +++ b/service.cpp @@ -14,6 +14,7 @@ bool stopping; unsigned long throttle_delay; HANDLE throttle_timer; LARGE_INTEGER throttle_duetime; +FILETIME creation_time; static enum { NSSM_EXIT_RESTART, NSSM_EXIT_IGNORE, NSSM_EXIT_REALLY, NSSM_EXIT_UNCLEAN } exit_actions; static const char *exit_action_strings[] = { "Restart", "Ignore", "Exit", "Suicide", 0 }; @@ -90,7 +91,7 @@ int install_service(char *name, char *exe, char *flags) { print_message(stderr, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED); return 2; } - + /* Get path of this program */ char path[MAX_PATH]; GetModuleFileName(0, path, MAX_PATH); @@ -149,7 +150,7 @@ int remove_service(char *name) { print_message(stderr, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED); return 2; } - + /* Try to open the service */ SC_HANDLE service = OpenService(services, name, SC_MANAGER_ALL_ACCESS); if (! service) { @@ -391,6 +392,8 @@ int start_service() { process_handle = pi.hProcess; pid = pi.dwProcessId; + if (get_process_creation_time(process_handle, &creation_time)) ZeroMemory(&creation_time, sizeof(creation_time)); + /* Signal successful start */ service_status.dwCurrentState = SERVICE_RUNNING; SetServiceStatus(service_handle, &service_status); @@ -415,12 +418,11 @@ int stop_service(unsigned long exitcode, bool graceful, bool default_action) { SetServiceStatus(service_handle, &service_status); } - /* Nothing to do if server isn't running */ + /* Nothing to do if service isn't running */ if (pid) { - /* Shut down server */ + /* Shut down service */ log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_TERMINATEPROCESS, service_name, exe, 0); kill_process(service_name, process_handle, pid, 0); - process_handle = 0; } else log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_PROCESS_ALREADY_STOPPED, service_name, exe, 0); @@ -453,10 +455,23 @@ void CALLBACK end_service(void *arg, unsigned char why) { /* Check exit code */ unsigned long exitcode = 0; + char code[16]; + FILETIME exit_time; GetExitCodeProcess(process_handle, &exitcode); + if (exitcode == STILL_ACTIVE || get_process_exit_time(process_handle, &exit_time)) GetSystemTimeAsFileTime(&exit_time); + CloseHandle(process_handle); + + /* + Log that the service ended BEFORE logging about killing the process + tree. See below for the possible values of the why argument. + */ + if (! why) { + _snprintf(code, sizeof(code), "%d", exitcode); + log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_ENDED_SERVICE, exe, service_name, code, 0); + } /* Clean up. */ - kill_process_tree(service_name, pid, exitcode, pid); + kill_process_tree(service_name, pid, exitcode, pid, &creation_time, &exit_time); /* The why argument is true if our wait timed out or false otherwise. @@ -466,10 +481,6 @@ void CALLBACK end_service(void *arg, unsigned char why) { */ if (why) return; - char code[16]; - _snprintf(code, sizeof(code), "%d", exitcode); - log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_ENDED_SERVICE, exe, service_name, code, 0); - /* What action should we take? */ int action = NSSM_EXIT_RESTART; unsigned char action_string[ACTION_LEN];