X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=process.cpp;h=624111cfbe6ad12d5a0cc0475427d88b6163d726;hb=f13f3f81d7ad3ccda728d06f615ce5f724023f4e;hp=59707c464243d694dda7c33f3a094a7f6a816733;hpb=e3b93d0c37ae200fa0375e4be6f435008050b385;p=nssm.git diff --git a/process.cpp b/process.cpp index 59707c4..624111c 100644 --- a/process.cpp +++ b/process.cpp @@ -23,6 +23,7 @@ int get_process_exit_time(HANDLE process_handle, FILETIME *ft) { return 1; } + if (! (exit_time.dwLowDateTime || exit_time.dwHighDateTime)) return 2; memmove(ft, &exit_time, sizeof(exit_time)); return 0; @@ -149,16 +150,9 @@ int kill_process(nssm_service_t *service, HANDLE process_handle, unsigned long p kill_t k = { pid, exitcode, 0 }; - /* Close the stdin pipe. */ - if (service->stdin_pipe) { - CloseHandle(service->stdin_pipe); - service->stdin_pipe = 0; - if (! await_shutdown(service, _T(__FUNCTION__), service->kill_console_delay)) return 1; - } - /* Try to send a Control-C event to the console. */ if (service->stop_method & NSSM_STOP_METHOD_CONSOLE) { - if (! kill_console(service)) return 1; + if (! kill_console(service, &k)) return 1; } /* @@ -186,14 +180,14 @@ int kill_process(nssm_service_t *service, HANDLE process_handle, unsigned long p /* We tried being nice. Time for extreme prejudice. */ if (service->stop_method & NSSM_STOP_METHOD_TERMINATE) { - return TerminateProcess(service->process_handle, exitcode); + return TerminateProcess(process_handle, exitcode); } return 0; } /* Simulate a Control-C event to our console (shared with the app). */ -int kill_console(nssm_service_t *service) { +int kill_console(nssm_service_t *service, kill_t *k) { unsigned long ret; if (! service) return 1; @@ -202,7 +196,7 @@ int kill_console(nssm_service_t *service) { if (! imports.AttachConsole) return 4; /* Try to attach to the process's console. */ - if (! imports.AttachConsole(service->pid)) { + if (! imports.AttachConsole(k->pid)) { ret = GetLastError(); switch (ret) { @@ -257,6 +251,26 @@ void kill_process_tree(nssm_service_t *service, unsigned long pid, unsigned long _sntprintf_s(code, _countof(code), _TRUNCATE, _T("%lu"), exitcode); log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_KILLING, service->name, pid_string, code, 0); + /* We will need a process handle in order to call TerminateProcess() later. */ + HANDLE process_handle = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION | PROCESS_VM_READ | PROCESS_TERMINATE, false, pid); + if (process_handle) { + /* Kill this process first, then its descendents. */ + TCHAR ppid_string[16]; + _sntprintf_s(ppid_string, _countof(ppid_string), _TRUNCATE, _T("%lu"), ppid); + log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_KILL_PROCESS_TREE, pid_string, ppid_string, service->name, 0); + if (! kill_process(service, process_handle, pid, exitcode)) { + /* Maybe it already died. */ + unsigned long ret; + if (! GetExitCodeProcess(process_handle, &ret) || ret == STILL_ACTIVE) { + if (service->stop_method & NSSM_STOP_METHOD_TERMINATE) log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_TERMINATEPROCESS_FAILED, pid_string, service->name, error_string(GetLastError()), 0); + else log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_PROCESS_STILL_ACTIVE, service->name, pid_string, NSSM, NSSM_REG_STOP_METHOD_SKIP, 0); + } + } + + CloseHandle(process_handle); + } + else log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OPENPROCESS_FAILED, pid_string, service->name, error_string(GetLastError()), 0); + /* Get a snapshot of all processes in the system. */ HANDLE snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); if (! snapshot) { @@ -291,25 +305,4 @@ void kill_process_tree(nssm_service_t *service, unsigned long pid, unsigned long } CloseHandle(snapshot); - - /* We will need a process handle in order to call TerminateProcess() later. */ - HANDLE process_handle = OpenProcess(SYNCHRONIZE | PROCESS_QUERY_INFORMATION | PROCESS_VM_READ | PROCESS_TERMINATE, false, pid); - if (! process_handle) { - log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OPENPROCESS_FAILED, pid_string, service->name, error_string(GetLastError()), 0); - return; - } - - TCHAR ppid_string[16]; - _sntprintf_s(ppid_string, _countof(ppid_string), _TRUNCATE, _T("%lu"), ppid); - log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_KILL_PROCESS_TREE, pid_string, ppid_string, service->name, 0); - if (! kill_process(service, process_handle, pid, exitcode)) { - /* Maybe it already died. */ - unsigned long ret; - if (! GetExitCodeProcess(process_handle, &ret) || ret == STILL_ACTIVE) { - if (service->stop_method & NSSM_STOP_METHOD_TERMINATE) log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_TERMINATEPROCESS_FAILED, pid_string, service->name, error_string(GetLastError()), 0); - else log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_PROCESS_STILL_ACTIVE, service->name, pid_string, NSSM, NSSM_REG_STOP_METHOD_SKIP, 0); - } - } - - CloseHandle(process_handle); }