#include "nssm.h"\r
\r
+static enum { NSSM_TAB_APPLICATION, NSSM_TAB_SHUTDOWN, NSSM_TAB_EXIT, NSSM_TAB_IO, NSSM_NUM_TABS };\r
+static HWND tablist[NSSM_NUM_TABS];\r
+static int selected_tab;\r
+\r
static void strip_basename(char *buffer) {\r
size_t len = strlen(buffer);\r
size_t i;\r
MoveWindow(window, x, y, size.right - size.left, size.bottom - size.top, 0);\r
}\r
\r
-/* Install the service */\r
+static inline void check_stop_method(nssm_service_t *service, unsigned long method, unsigned long control) {\r
+ if (SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], control, BM_GETCHECK, 0, 0) & BST_CHECKED) return;\r
+ service->stop_method &= ~method;\r
+}\r
+\r
+static inline void check_method_timeout(HWND tab, unsigned long control, unsigned long *timeout) {\r
+ BOOL translated;\r
+ unsigned long configured = GetDlgItemInt(tab, control, &translated, 0);\r
+ if (translated) *timeout = configured;\r
+}\r
+\r
+static inline void check_io(char *name, char *buffer, size_t bufsize, unsigned long control) {\r
+ if (! SendMessage(GetDlgItem(tablist[NSSM_TAB_IO], control), WM_GETTEXTLENGTH, 0, 0)) return;\r
+ if (GetDlgItemText(tablist[NSSM_TAB_IO], control, buffer, (int) bufsize)) return;\r
+ popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, name);\r
+ ZeroMemory(buffer, bufsize);\r
+}\r
+\r
+/* Install the service. */\r
int install(HWND window) {\r
if (! window) return 1;\r
\r
}\r
\r
/* Get executable name */\r
- if (! GetDlgItemText(window, IDC_PATH, service->exe, sizeof(service->exe))) {\r
+ if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_PATH, service->exe, sizeof(service->exe))) {\r
popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_PATH);\r
return 3;\r
}\r
- \r
- /* Get flags */\r
- if (SendMessage(GetDlgItem(window, IDC_FLAGS), WM_GETTEXTLENGTH, 0, 0)) {\r
+\r
+ /* Get startup directory. */\r
+ if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_DIR, service->dir, sizeof(service->dir))) {\r
+ memmove(service->dir, service->exe, sizeof(service->dir));\r
+ strip_basename(service->dir);\r
+ }\r
+\r
+ /* Get flags. */\r
+ if (SendMessage(GetDlgItem(tablist[NSSM_TAB_APPLICATION], IDC_FLAGS), WM_GETTEXTLENGTH, 0, 0)) {\r
if (! GetDlgItemText(window, IDC_FLAGS, service->flags, sizeof(service->flags))) {\r
popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_OPTIONS);\r
return 4;\r
}
}\r
\r
- memmove(service->dir, service->exe, strlen(service->exe));\r
- strip_basename(service->dir);\r
+ /* Get stop method stuff. */\r
+ service->stop_method = ~0;\r
+ check_stop_method(service, NSSM_STOP_METHOD_CONSOLE, IDC_METHOD_CONSOLE);\r
+ check_stop_method(service, NSSM_STOP_METHOD_WINDOW, IDC_METHOD_WINDOW);\r
+ check_stop_method(service, NSSM_STOP_METHOD_THREADS, IDC_METHOD_THREADS);\r
+ check_stop_method(service, NSSM_STOP_METHOD_TERMINATE, IDC_METHOD_TERMINATE);\r
+ check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, &service->kill_console_delay);\r
+ check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, &service->kill_window_delay);\r
+ check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, &service->kill_threads_delay);\r
+\r
+ /* Get exit action stuff. */\r
+ check_method_timeout(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, &service->throttle_delay);\r
+ HWND combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);\r
+ service->default_exit_action = (unsigned long) SendMessage(combo, CB_GETCURSEL, 0, 0);\r
+ if (service->default_exit_action == CB_ERR) service->default_exit_action = 0;\r
+\r
+ /* Get I/O stuff. */\r
+ check_io("stdin", service->stdin_path, sizeof(service->stdin_path), IDC_STDIN);\r
+ check_io("stdout", service->stdout_path, sizeof(service->stdout_path), IDC_STDOUT);\r
+ check_io("stderr", service->stderr_path, sizeof(service->stderr_path), IDC_STDERR);\r
}\r
\r
/* See if it works. */\r
return 0;\r
}\r
\r
+static char *browse_filter(int message) {\r
+ switch (message) {\r
+ case NSSM_GUI_BROWSE_FILTER_APPLICATIONS: return "*.exe;*.bat;*.cmd";\r
+ case NSSM_GUI_BROWSE_FILTER_DIRECTORIES: return ".";\r
+ case NSSM_GUI_BROWSE_FILTER_ALL_FILES: /* Fall through. */\r
+ default: return "*.*";\r
+ }\r
+}\r
+\r
+UINT_PTR CALLBACK browse_hook(HWND dlg, UINT message, WPARAM w, LPARAM l) {\r
+ switch (message) {\r
+ case WM_INITDIALOG:\r
+ return 1;\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
/* Browse for application */\r
-void browse(HWND window) {\r
+void browse(HWND window, char *current, unsigned long flags, ...) {\r
if (! window) return;\r
\r
+ va_list arg;\r
size_t bufsize = 256;\r
size_t len = bufsize;\r
+ int i;\r
+\r
OPENFILENAME ofn;\r
ZeroMemory(&ofn, sizeof(ofn));\r
ofn.lStructSize = sizeof(ofn);\r
/* XXX: Escaping nulls with FormatMessage is tricky */\r
if (ofn.lpstrFilter) {\r
ZeroMemory((void *) ofn.lpstrFilter, bufsize);\r
- char *localised = message_string(NSSM_GUI_BROWSE_FILTER_APPLICATIONS);\r
- _snprintf_s((char *) ofn.lpstrFilter, bufsize, _TRUNCATE, localised);\r
+ len = 0;\r
/* "Applications" + NULL + "*.exe" + NULL */\r
- len = strlen(localised) + 1;\r
- LocalFree(localised);\r
- _snprintf_s((char *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, "*.exe");\r
- /* "All files" + NULL + "*.*" + NULL */\r
- len += 6;\r
- localised = message_string(NSSM_GUI_BROWSE_FILTER_ALL_FILES);\r
- _snprintf_s((char *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, localised);\r
- len += strlen(localised) + 1;\r
- LocalFree(localised);\r
- _snprintf_s((char *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, "*.*");\r
+ va_start(arg, flags);\r
+ while (i = va_arg(arg, int)) {\r
+ char *localised = message_string(i);\r
+ _snprintf_s((char *) ofn.lpstrFilter + len, bufsize, _TRUNCATE, localised);\r
+ len += strlen(localised) + 1;\r
+ LocalFree(localised);\r
+ char *filter = browse_filter(i);\r
+ _snprintf_s((char *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, "%s", filter);\r
+ len += strlen(filter) + 1;\r
+ }\r
+ va_end(arg);\r
/* Remainder of the buffer is already zeroed */\r
}\r
ofn.lpstrFile = new char[MAX_PATH];\r
- ofn.lpstrFile[0] = '\0';\r
+ if (flags & OFN_NOVALIDATE) {\r
+ /* Directory hack. */\r
+ _snprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, ":%s:", message_string(NSSM_GUI_BROWSE_FILTER_DIRECTORIES));\r
+ }\r
+ else _snprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, "%s", current);\r
ofn.lpstrTitle = message_string(NSSM_GUI_BROWSE_TITLE);\r
ofn.nMaxFile = MAX_PATH;\r
- ofn.Flags = OFN_EXPLORER | OFN_PATHMUSTEXIST | OFN_HIDEREADONLY;\r
+ ofn.Flags = OFN_EXPLORER | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST | flags;\r
\r
if (GetOpenFileName(&ofn)) {\r
+ /* Directory hack. */\r
+ if (flags & OFN_NOVALIDATE) strip_basename(ofn.lpstrFile);\r
SendMessage(window, WM_SETTEXT, 0, (LPARAM) ofn.lpstrFile);\r
}\r
if (ofn.lpstrFilter) HeapFree(GetProcessHeap(), 0, (void *) ofn.lpstrFilter);\r
delete[] ofn.lpstrFile;\r
}\r
\r
+INT_PTR CALLBACK tab_dlg(HWND tab, UINT message, WPARAM w, LPARAM l) {\r
+ switch (message) {\r
+ case WM_INITDIALOG:\r
+ return 1;\r
+\r
+ /* Button was pressed or control was controlled. */\r
+ case WM_COMMAND:\r
+ HWND dlg;\r
+ char buffer[MAX_PATH];\r
+\r
+ switch (LOWORD(w)) {\r
+ /* Browse for application. */\r
+ case IDC_BROWSE:\r
+ dlg = GetDlgItem(tab, IDC_PATH);\r
+ GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));\r
+ browse(dlg, buffer, OFN_FILEMUSTEXIST, NSSM_GUI_BROWSE_FILTER_APPLICATIONS, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
+ /* Fill in startup directory if it wasn't already specified. */\r
+ GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));\r
+ if (! buffer[0]) {\r
+ GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));\r
+ strip_basename(buffer);\r
+ SetDlgItemText(tab, IDC_DIR, buffer);\r
+ }\r
+ break;\r
+\r
+ /* Browse for startup directory. */\r
+ case IDC_BROWSE_DIR:\r
+ dlg = GetDlgItem(tab, IDC_DIR);\r
+ GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));\r
+ browse(dlg, buffer, OFN_NOVALIDATE, NSSM_GUI_BROWSE_FILTER_DIRECTORIES, 0);\r
+ break;\r
+\r
+ /* Browse for stdin. */\r
+ case IDC_BROWSE_STDIN:\r
+ dlg = GetDlgItem(tab, IDC_STDIN);\r
+ GetDlgItemText(tab, IDC_STDIN, buffer, sizeof(buffer));\r
+ browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
+ break;\r
+\r
+ /* Browse for stdout. */\r
+ case IDC_BROWSE_STDOUT:\r
+ dlg = GetDlgItem(tab, IDC_STDOUT);\r
+ GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));\r
+ browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
+ /* Fill in stderr if it wasn't already specified. */\r
+ GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));\r
+ if (! buffer[0]) {\r
+ GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));\r
+ SetDlgItemText(tab, IDC_STDERR, buffer);\r
+ }\r
+ break;\r
+\r
+ /* Browse for stderr. */\r
+ case IDC_BROWSE_STDERR:\r
+ dlg = GetDlgItem(tab, IDC_STDERR);\r
+ GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));\r
+ browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
+ break;\r
+ }\r
+ return 1;\r
+ }\r
+\r
+ return 0;\r
+}\r
+\r
/* Install/remove dialogue callback */\r
INT_PTR CALLBACK install_dlg(HWND window, UINT message, WPARAM w, LPARAM l) {\r
switch (message) {\r
/* Creating the dialogue */\r
case WM_INITDIALOG:\r
+ HWND tabs;\r
+ HWND combo;\r
+ tabs = GetDlgItem(window, IDC_TAB1);\r
+ if (! tabs) return 0;\r
+\r
+ /* Set up tabs. */\r
+ TCITEM tab;\r
+ ZeroMemory(&tab, sizeof(tab));\r
+ tab.mask = TCIF_TEXT;\r
+\r
+ /* Application tab. */\r
+ tab.pszText = message_string(NSSM_GUI_TAB_APPLICATION);\r
+ tab.cchTextMax = (int) strlen(tab.pszText);\r
+ SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_APPLICATION, (LPARAM) &tab);\r
+ tablist[NSSM_TAB_APPLICATION] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPLICATION), window, tab_dlg);\r
+ ShowWindow(tablist[NSSM_TAB_APPLICATION], SW_SHOW);\r
+\r
+ /* Shutdown tab. */\r
+ tab.pszText = message_string(NSSM_GUI_TAB_SHUTDOWN);\r
+ tab.cchTextMax = (int) strlen(tab.pszText);\r
+ SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_SHUTDOWN, (LPARAM) &tab);\r
+ tablist[NSSM_TAB_SHUTDOWN] = CreateDialog(0, MAKEINTRESOURCE(IDD_SHUTDOWN), window, tab_dlg);\r
+ ShowWindow(tablist[NSSM_TAB_SHUTDOWN], SW_HIDE);\r
+\r
+ /* Set defaults. */\r
+ SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_CONSOLE, BM_SETCHECK, BST_CHECKED, 0);\r
+ SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, NSSM_KILL_CONSOLE_GRACE_PERIOD, 0);\r
+ SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_WINDOW, BM_SETCHECK, BST_CHECKED, 0);\r
+ SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, NSSM_KILL_WINDOW_GRACE_PERIOD, 0);\r
+ SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_THREADS, BM_SETCHECK, BST_CHECKED, 0);\r
+ SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, NSSM_KILL_THREADS_GRACE_PERIOD, 0);\r
+ SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_TERMINATE, BM_SETCHECK, BST_CHECKED, 0);\r
+\r
+ /* Restart tab. */\r
+ tab.pszText = message_string(NSSM_GUI_TAB_EXIT);\r
+ tab.cchTextMax = (int) strlen(tab.pszText);\r
+ SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_EXIT, (LPARAM) &tab);\r
+ tablist[NSSM_TAB_EXIT] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPEXIT), window, tab_dlg);\r
+ ShowWindow(tablist[NSSM_TAB_EXIT], SW_HIDE);\r
+\r
+ /* Set defaults. */\r
+ SetDlgItemInt(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, NSSM_RESET_THROTTLE_RESTART, 0);\r
+ combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);\r
+ SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_RESTART, (LPARAM) message_string(NSSM_GUI_EXIT_RESTART));\r
+ SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_IGNORE, (LPARAM) message_string(NSSM_GUI_EXIT_IGNORE));\r
+ SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_REALLY, (LPARAM) message_string(NSSM_GUI_EXIT_REALLY));\r
+ SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_UNCLEAN, (LPARAM) message_string(NSSM_GUI_EXIT_UNCLEAN));\r
+ SendMessage(combo, CB_SETCURSEL, NSSM_EXIT_RESTART, 0);\r
+\r
+ /* I/O tab. */\r
+ tab.pszText = message_string(NSSM_GUI_TAB_IO);\r
+ tab.cchTextMax = (int) strlen(tab.pszText) + 1;\r
+ SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_IO, (LPARAM) &tab);\r
+ tablist[NSSM_TAB_IO] = CreateDialog(0, MAKEINTRESOURCE(IDD_IO), window, tab_dlg);\r
+ ShowWindow(tablist[NSSM_TAB_IO], SW_HIDE);\r
+\r
+ selected_tab = 0;\r
+\r
return 1;\r
\r
+ /* Tab change. */\r
+ case WM_NOTIFY:\r
+ NMHDR *notification;\r
+\r
+ notification = (NMHDR *) l;\r
+ switch (notification->code) {\r
+ case TCN_SELCHANGE:\r
+ HWND tabs;\r
+ int selection;\r
+\r
+ tabs = GetDlgItem(window, IDC_TAB1);\r
+ if (! tabs) return 0;\r
+\r
+ selection = (int) SendMessage(tabs, TCM_GETCURSEL, 0, 0);\r
+ if (selection != selected_tab) {\r
+ ShowWindow(tablist[selected_tab], SW_HIDE);\r
+ /*\r
+ XXX: Sets focus to the service name which isn't ideal but is\r
+ better than leaving it in another tab.\r
+ */\r
+ ShowWindow(tablist[selection], SW_SHOWDEFAULT);\r
+ SetFocus(tablist[selection]);\r
+ selected_tab = selection;\r
+ }\r
+ return 1;\r
+ }\r
+\r
+ return 0;\r
+\r
/* Button was pressed or control was controlled */\r
case WM_COMMAND:\r
switch (LOWORD(w)) {\r
/* OK button */\r
- case IDC_OK:\r
+ case IDOK:\r
if (! install(window)) PostQuitMessage(0);\r
break;\r
\r
DestroyWindow(window);\r
break;\r
\r
- /* Browse button */\r
- case IDC_BROWSE:\r
- browse(GetDlgItem(window, IDC_PATH));\r
- break;\r
-\r
/* Remove button */\r
case IDC_REMOVE:\r
if (! remove(window)) PostQuitMessage(0);\r
git://git.iain.cx/iain / nssm.git / blobdiff