X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=process.cpp;h=f060d725992a13dcff36241883e0d9f26eba73aa;hb=636b16d3702049145143511c6461ff20bf1f245d;hp=b63cc4134276e38acd8aceb9e31c0ca56eaeb42c;hpb=9884e231415258809dfb1ef117b6bc9a5b011514;p=nssm.git diff --git a/process.cpp b/process.cpp index b63cc41..f060d72 100644 --- a/process.cpp +++ b/process.cpp @@ -1,5 +1,10 @@ #include "nssm.h" +extern imports_t imports; +extern unsigned long kill_console_delay; +extern unsigned long kill_window_delay; +extern unsigned long kill_threads_delay; + int get_process_creation_time(HANDLE process_handle, FILETIME *ft) { FILETIME creation_time, exit_time, kernel_time, user_time; @@ -39,7 +44,7 @@ int check_parent(char *service_name, PROCESSENTRY32 *pe, unsigned long ppid, FIL HANDLE process_handle = OpenProcess(PROCESS_QUERY_INFORMATION, false, pe->th32ProcessID); if (! process_handle) { char pid_string[16]; - _snprintf(pid_string, sizeof(pid_string), "%d", pe->th32ProcessID); + _snprintf_s(pid_string, sizeof(pid_string), _TRUNCATE, "%lu", pe->th32ProcessID); log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OPENPROCESS_FAILED, pid_string, service_name, error_string(GetLastError()), 0); return 2; } @@ -134,7 +139,7 @@ int kill_threads(char *service_name, kill_t *k) { } /* Give the process a chance to die gracefully. */ -int kill_process(char *service_name, HANDLE process_handle, unsigned long pid, unsigned long exitcode) { +int kill_process(char *service_name, unsigned long stop_method, HANDLE process_handle, unsigned long pid, unsigned long exitcode) { /* Shouldn't happen. */ if (! pid) return 1; if (! process_handle) return 1; @@ -146,14 +151,21 @@ int kill_process(char *service_name, HANDLE process_handle, unsigned long pid, u kill_t k = { pid, exitcode, 0 }; + /* Try to send a Control-C event to the console. */ + if (stop_method & NSSM_STOP_METHOD_CONSOLE) { + if (! kill_console(service_name, process_handle, pid)) return 1; + } + /* Try to post messages to the windows belonging to the given process ID. If the process is a console application it won't have any windows so there's no guarantee of success. */ - EnumWindows((WNDENUMPROC) kill_window, (LPARAM) &k); - if (k.signalled) { - if (! WaitForSingleObject(process_handle, NSSM_KILL_WINDOW_GRACE_PERIOD)) return 1; + if (stop_method & NSSM_STOP_METHOD_WINDOW) { + EnumWindows((WNDENUMPROC) kill_window, (LPARAM) &k); + if (k.signalled) { + if (! WaitForSingleObject(process_handle, kill_window_delay)) return 1; + } } /* @@ -161,21 +173,81 @@ int kill_process(char *service_name, HANDLE process_handle, unsigned long pid, u process. Console applications might have them (but probably won't) so there's still no guarantee of success. */ - if (kill_threads(service_name, &k)) { - if (! WaitForSingleObject(process_handle, NSSM_KILL_THREADS_GRACE_PERIOD)) return 1; + if (stop_method & NSSM_STOP_METHOD_THREADS) { + if (kill_threads(service_name, &k)) { + if (! WaitForSingleObject(process_handle, kill_threads_delay)) return 1; + } } /* We tried being nice. Time for extreme prejudice. */ - return TerminateProcess(process_handle, exitcode); + if (stop_method & NSSM_STOP_METHOD_TERMINATE) { + return TerminateProcess(process_handle, exitcode); + } + + return 0; } -void kill_process_tree(char *service_name, unsigned long pid, unsigned long exitcode, unsigned long ppid, FILETIME *parent_creation_time, FILETIME *parent_exit_time) { +/* Simulate a Control-C event to our console (shared with the app). */ +int kill_console(char *service_name, HANDLE process_handle, unsigned long pid) { + unsigned long ret; + + /* Check we loaded AttachConsole(). */ + if (! imports.AttachConsole) return 4; + + /* Try to attach to the process's console. */ + if (! imports.AttachConsole(pid)) { + ret = GetLastError(); + + switch (ret) { + case ERROR_INVALID_HANDLE: + /* The app doesn't have a console. */ + return 1; + + case ERROR_GEN_FAILURE: + /* The app already exited. */ + return 2; + + case ERROR_ACCESS_DENIED: + default: + /* We already have a console. */ + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_ATTACHCONSOLE_FAILED, service_name, error_string(ret), 0); + return 3; + } + } + + /* Ignore the event ourselves. */ + ret = 0; + if (! SetConsoleCtrlHandler(0, TRUE)) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_SETCONSOLECTRLHANDLER_FAILED, service_name, error_string(GetLastError()), 0); + ret = 4; + } + + /* Send the event. */ + if (! ret) { + if (! GenerateConsoleCtrlEvent(CTRL_C_EVENT, 0)) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_GENERATECONSOLECTRLEVENT_FAILED, service_name, error_string(GetLastError()), 0); + ret = 5; + } + } + + /* Detach from the console. */ + if (! FreeConsole()) { + log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_FREECONSOLE_FAILED, service_name, error_string(GetLastError()), 0); + } + + /* Wait for process to exit. */ + if (WaitForSingleObject(process_handle, kill_console_delay)) return 6; + + return ret; +} + +void kill_process_tree(char *service_name, unsigned long stop_method, unsigned long pid, unsigned long exitcode, unsigned long ppid, FILETIME *parent_creation_time, FILETIME *parent_exit_time) { /* Shouldn't happen unless the service failed to start. */ if (! pid) return; char pid_string[16], code[16]; - _snprintf(pid_string, sizeof(pid_string), "%d", pid); - _snprintf(code, sizeof(code), "%d", exitcode); + _snprintf_s(pid_string, sizeof(pid_string), _TRUNCATE, "%lu", pid); + _snprintf_s(code, sizeof(code), _TRUNCATE, "%lu", exitcode); log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_KILLING, service_name, pid_string, code, 0); /* Get a snapshot of all processes in the system. */ @@ -196,7 +268,7 @@ void kill_process_tree(char *service_name, unsigned long pid, unsigned long exit } /* This is a child of the doomed process so kill it. */ - if (! check_parent(service_name, &pe, pid, parent_creation_time, parent_exit_time)) kill_process_tree(service_name, pe.th32ProcessID, exitcode, ppid, parent_creation_time, parent_exit_time); + if (! check_parent(service_name, &pe, pid, parent_creation_time, parent_exit_time)) kill_process_tree(service_name, stop_method, pe.th32ProcessID, exitcode, ppid, parent_creation_time, parent_exit_time); while (true) { /* Try to get the next process. */ @@ -208,7 +280,7 @@ void kill_process_tree(char *service_name, unsigned long pid, unsigned long exit return; } - if (! check_parent(service_name, &pe, pid, parent_creation_time, parent_exit_time)) kill_process_tree(service_name, pe.th32ProcessID, exitcode, ppid, parent_creation_time, parent_exit_time); + if (! check_parent(service_name, &pe, pid, parent_creation_time, parent_exit_time)) kill_process_tree(service_name, stop_method, pe.th32ProcessID, exitcode, ppid, parent_creation_time, parent_exit_time); } CloseHandle(snapshot); @@ -221,12 +293,15 @@ void kill_process_tree(char *service_name, unsigned long pid, unsigned long exit } char ppid_string[16]; - _snprintf(ppid_string, sizeof(ppid_string), "%d", ppid); + _snprintf_s(ppid_string, sizeof(ppid_string), _TRUNCATE, "%lu", ppid); log_event(EVENTLOG_INFORMATION_TYPE, NSSM_EVENT_KILL_PROCESS_TREE, pid_string, ppid_string, service_name, 0); - if (! kill_process(service_name, process_handle, pid, exitcode)) { + if (! kill_process(service_name, stop_method, process_handle, pid, exitcode)) { /* Maybe it already died. */ unsigned long ret; - if (! GetExitCodeProcess(process_handle, &ret) || ret == STILL_ACTIVE) log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_TERMINATEPROCESS_FAILED, pid_string, service_name, error_string(GetLastError()), 0); + if (! GetExitCodeProcess(process_handle, &ret) || ret == STILL_ACTIVE) { + if (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);