}
/* 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;
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, NSSM_KILL_WINDOW_GRACE_PERIOD)) return 1;
+ }
}
/*
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, NSSM_KILL_THREADS_GRACE_PERIOD)) 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;
+
+ /* Try to attach to the process's console. */
+ if (! 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, NSSM_KILL_CONSOLE_GRACE_PERIOD)) 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;
}
/* 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. */
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);
char ppid_string[16];
_snprintf(ppid_string, sizeof(ppid_string), "%d", 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);