3 static enum { NSSM_TAB_APPLICATION, NSSM_TAB_SHUTDOWN, NSSM_TAB_EXIT, NSSM_TAB_IO, NSSM_NUM_TABS };
\r
4 static HWND tablist[NSSM_NUM_TABS];
\r
5 static int selected_tab;
\r
7 static void strip_basename(char *buffer) {
\r
8 size_t len = strlen(buffer);
\r
10 for (i = len; i && buffer[i] != '\\' && buffer[i] != '/'; i--);
\r
12 if (i && buffer[i-1] == ':') i++;
\r
16 int nssm_gui(int resource, char *name) {
\r
18 HWND dlg = CreateDialog(0, MAKEINTRESOURCE(resource), 0, install_dlg);
\r
20 popup_message(MB_OK, NSSM_GUI_CREATEDIALOG_FAILED, error_string(GetLastError()));
\r
24 /* Display the window */
\r
26 ShowWindow(dlg, SW_SHOW);
\r
28 /* Set service name if given */
\r
30 SetDlgItemText(dlg, IDC_NAME, name);
\r
31 /* No point making user click remove if the name is already entered */
\r
32 if (resource == IDD_REMOVE) {
\r
33 HWND button = GetDlgItem(dlg, IDC_REMOVE);
\r
35 SendMessage(button, WM_LBUTTONDOWN, 0, 0);
\r
36 SendMessage(button, WM_LBUTTONUP, 0, 0);
\r
43 while (GetMessage(&message, 0, 0, 0)) {
\r
44 if (IsDialogMessage(dlg, &message)) continue;
\r
45 TranslateMessage(&message);
\r
46 DispatchMessage(&message);
\r
49 return (int) message.wParam;
\r
52 void centre_window(HWND window) {
\r
54 RECT size, desktop_size;
\r
57 if (! window) return;
\r
59 /* Find window size */
\r
60 if (! GetWindowRect(window, &size)) return;
\r
62 /* Find desktop window */
\r
63 desktop = GetDesktopWindow();
\r
64 if (! desktop) return;
\r
66 /* Find desktop window size */
\r
67 if (! GetWindowRect(desktop, &desktop_size)) return;
\r
70 x = (desktop_size.right - size.right) / 2;
\r
71 y = (desktop_size.bottom - size.bottom) / 2;
\r
72 MoveWindow(window, x, y, size.right - size.left, size.bottom - size.top, 0);
\r
75 static inline void check_stop_method(nssm_service_t *service, unsigned long method, unsigned long control) {
\r
76 if (SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], control, BM_GETCHECK, 0, 0) & BST_CHECKED) return;
\r
77 service->stop_method &= ~method;
\r
80 static inline void check_method_timeout(HWND tab, unsigned long control, unsigned long *timeout) {
\r
82 unsigned long configured = GetDlgItemInt(tab, control, &translated, 0);
\r
83 if (translated) *timeout = configured;
\r
86 static inline void check_io(char *name, char *buffer, size_t bufsize, unsigned long control) {
\r
87 if (! SendMessage(GetDlgItem(tablist[NSSM_TAB_IO], control), WM_GETTEXTLENGTH, 0, 0)) return;
\r
88 if (GetDlgItemText(tablist[NSSM_TAB_IO], control, buffer, (int) bufsize)) return;
\r
89 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, name);
\r
90 ZeroMemory(buffer, bufsize);
\r
93 /* Install the service. */
\r
94 int install(HWND window) {
\r
95 if (! window) return 1;
\r
97 nssm_service_t *service = alloc_nssm_service();
\r
99 /* Get service name. */
\r
100 if (! GetDlgItemText(window, IDC_NAME, service->name, sizeof(service->name))) {
\r
101 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_SERVICE_NAME);
\r
102 cleanup_nssm_service(service);
\r
106 /* Get executable name */
\r
107 if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_PATH, service->exe, sizeof(service->exe))) {
\r
108 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_PATH);
\r
112 /* Get startup directory. */
\r
113 if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_DIR, service->dir, sizeof(service->dir))) {
\r
114 memmove(service->dir, service->exe, sizeof(service->dir));
\r
115 strip_basename(service->dir);
\r
119 if (SendMessage(GetDlgItem(tablist[NSSM_TAB_APPLICATION], IDC_FLAGS), WM_GETTEXTLENGTH, 0, 0)) {
\r
120 if (! GetDlgItemText(window, IDC_FLAGS, service->flags, sizeof(service->flags))) {
\r
121 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_OPTIONS);
\r
126 /* Get stop method stuff. */
\r
127 service->stop_method = ~0;
\r
128 check_stop_method(service, NSSM_STOP_METHOD_CONSOLE, IDC_METHOD_CONSOLE);
\r
129 check_stop_method(service, NSSM_STOP_METHOD_WINDOW, IDC_METHOD_WINDOW);
\r
130 check_stop_method(service, NSSM_STOP_METHOD_THREADS, IDC_METHOD_THREADS);
\r
131 check_stop_method(service, NSSM_STOP_METHOD_TERMINATE, IDC_METHOD_TERMINATE);
\r
132 check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, &service->kill_console_delay);
\r
133 check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, &service->kill_window_delay);
\r
134 check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, &service->kill_threads_delay);
\r
136 /* Get exit action stuff. */
\r
137 check_method_timeout(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, &service->throttle_delay);
\r
138 HWND combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);
\r
139 service->default_exit_action = (unsigned long) SendMessage(combo, CB_GETCURSEL, 0, 0);
\r
140 if (service->default_exit_action == CB_ERR) service->default_exit_action = 0;
\r
142 /* Get I/O stuff. */
\r
143 check_io("stdin", service->stdin_path, sizeof(service->stdin_path), IDC_STDIN);
\r
144 check_io("stdout", service->stdout_path, sizeof(service->stdout_path), IDC_STDOUT);
\r
145 check_io("stderr", service->stderr_path, sizeof(service->stderr_path), IDC_STDERR);
\r
148 /* See if it works. */
\r
149 switch (install_service(service)) {
\r
151 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, "service", "install()");
\r
152 cleanup_nssm_service(service);
\r
156 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);
\r
157 cleanup_nssm_service(service);
\r
161 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, NSSM);
\r
162 cleanup_nssm_service(service);
\r
166 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_OUT_OF_MEMORY_FOR_IMAGEPATH);
\r
167 cleanup_nssm_service(service);
\r
171 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INSTALL_SERVICE_FAILED);
\r
172 cleanup_nssm_service(service);
\r
176 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_CREATE_PARAMETERS_FAILED);
\r
177 cleanup_nssm_service(service);
\r
181 popup_message(MB_OK, NSSM_MESSAGE_SERVICE_INSTALLED, service->name);
\r
182 cleanup_nssm_service(service);
\r
186 /* Remove the service */
\r
187 int remove(HWND window) {
\r
188 if (! window) return 1;
\r
190 /* See if it works */
\r
191 nssm_service_t *service = alloc_nssm_service();
\r
193 /* Get service name */
\r
194 if (! GetDlgItemText(window, IDC_NAME, service->name, sizeof(service->name))) {
\r
195 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_SERVICE_NAME);
\r
196 cleanup_nssm_service(service);
\r
201 if (popup_message(MB_YESNO, NSSM_GUI_ASK_REMOVE_SERVICE, service->name) != IDYES) {
\r
202 cleanup_nssm_service(service);
\r
207 switch (remove_service(service)) {
\r
209 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, "service", "remove()");
\r
210 cleanup_nssm_service(service);
\r
214 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);
\r
215 cleanup_nssm_service(service);
\r
219 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_SERVICE_NOT_INSTALLED);
\r
221 cleanup_nssm_service(service);
\r
224 popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_REMOVE_SERVICE_FAILED);
\r
225 cleanup_nssm_service(service);
\r
229 popup_message(MB_OK, NSSM_MESSAGE_SERVICE_REMOVED, service->name);
\r
230 cleanup_nssm_service(service);
\r
234 static char *browse_filter(int message) {
\r
236 case NSSM_GUI_BROWSE_FILTER_APPLICATIONS: return "*.exe;*.bat;*.cmd";
\r
237 case NSSM_GUI_BROWSE_FILTER_DIRECTORIES: return ".";
\r
238 case NSSM_GUI_BROWSE_FILTER_ALL_FILES: /* Fall through. */
\r
239 default: return "*.*";
\r
243 UINT_PTR CALLBACK browse_hook(HWND dlg, UINT message, WPARAM w, LPARAM l) {
\r
245 case WM_INITDIALOG:
\r
252 /* Browse for application */
\r
253 void browse(HWND window, char *current, unsigned long flags, ...) {
\r
254 if (! window) return;
\r
257 size_t bufsize = 256;
\r
258 size_t len = bufsize;
\r
262 ZeroMemory(&ofn, sizeof(ofn));
\r
263 ofn.lStructSize = sizeof(ofn);
\r
264 ofn.lpstrFilter = (char *) HeapAlloc(GetProcessHeap(), 0, bufsize);
\r
265 /* XXX: Escaping nulls with FormatMessage is tricky */
\r
266 if (ofn.lpstrFilter) {
\r
267 ZeroMemory((void *) ofn.lpstrFilter, bufsize);
\r
269 /* "Applications" + NULL + "*.exe" + NULL */
\r
270 va_start(arg, flags);
\r
271 while (i = va_arg(arg, int)) {
\r
272 char *localised = message_string(i);
\r
273 _snprintf_s((char *) ofn.lpstrFilter + len, bufsize, _TRUNCATE, localised);
\r
274 len += strlen(localised) + 1;
\r
275 LocalFree(localised);
\r
276 char *filter = browse_filter(i);
\r
277 _snprintf_s((char *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, "%s", filter);
\r
278 len += strlen(filter) + 1;
\r
281 /* Remainder of the buffer is already zeroed */
\r
283 ofn.lpstrFile = new char[MAX_PATH];
\r
284 if (flags & OFN_NOVALIDATE) {
\r
285 /* Directory hack. */
\r
286 _snprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, ":%s:", message_string(NSSM_GUI_BROWSE_FILTER_DIRECTORIES));
\r
288 else _snprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, "%s", current);
\r
289 ofn.lpstrTitle = message_string(NSSM_GUI_BROWSE_TITLE);
\r
290 ofn.nMaxFile = MAX_PATH;
\r
291 ofn.Flags = OFN_EXPLORER | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST | flags;
\r
293 if (GetOpenFileName(&ofn)) {
\r
294 /* Directory hack. */
\r
295 if (flags & OFN_NOVALIDATE) strip_basename(ofn.lpstrFile);
\r
296 SendMessage(window, WM_SETTEXT, 0, (LPARAM) ofn.lpstrFile);
\r
298 if (ofn.lpstrFilter) HeapFree(GetProcessHeap(), 0, (void *) ofn.lpstrFilter);
\r
300 delete[] ofn.lpstrFile;
\r
303 INT_PTR CALLBACK tab_dlg(HWND tab, UINT message, WPARAM w, LPARAM l) {
\r
305 case WM_INITDIALOG:
\r
308 /* Button was pressed or control was controlled. */
\r
311 char buffer[MAX_PATH];
\r
313 switch (LOWORD(w)) {
\r
314 /* Browse for application. */
\r
316 dlg = GetDlgItem(tab, IDC_PATH);
\r
317 GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));
\r
318 browse(dlg, buffer, OFN_FILEMUSTEXIST, NSSM_GUI_BROWSE_FILTER_APPLICATIONS, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);
\r
319 /* Fill in startup directory if it wasn't already specified. */
\r
320 GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));
\r
322 GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));
\r
323 strip_basename(buffer);
\r
324 SetDlgItemText(tab, IDC_DIR, buffer);
\r
328 /* Browse for startup directory. */
\r
329 case IDC_BROWSE_DIR:
\r
330 dlg = GetDlgItem(tab, IDC_DIR);
\r
331 GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));
\r
332 browse(dlg, buffer, OFN_NOVALIDATE, NSSM_GUI_BROWSE_FILTER_DIRECTORIES, 0);
\r
335 /* Browse for stdin. */
\r
336 case IDC_BROWSE_STDIN:
\r
337 dlg = GetDlgItem(tab, IDC_STDIN);
\r
338 GetDlgItemText(tab, IDC_STDIN, buffer, sizeof(buffer));
\r
339 browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);
\r
342 /* Browse for stdout. */
\r
343 case IDC_BROWSE_STDOUT:
\r
344 dlg = GetDlgItem(tab, IDC_STDOUT);
\r
345 GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));
\r
346 browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);
\r
347 /* Fill in stderr if it wasn't already specified. */
\r
348 GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));
\r
350 GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));
\r
351 SetDlgItemText(tab, IDC_STDERR, buffer);
\r
355 /* Browse for stderr. */
\r
356 case IDC_BROWSE_STDERR:
\r
357 dlg = GetDlgItem(tab, IDC_STDERR);
\r
358 GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));
\r
359 browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);
\r
368 /* Install/remove dialogue callback */
\r
369 INT_PTR CALLBACK install_dlg(HWND window, UINT message, WPARAM w, LPARAM l) {
\r
371 /* Creating the dialogue */
\r
372 case WM_INITDIALOG:
\r
375 tabs = GetDlgItem(window, IDC_TAB1);
\r
376 if (! tabs) return 0;
\r
380 ZeroMemory(&tab, sizeof(tab));
\r
381 tab.mask = TCIF_TEXT;
\r
383 /* Application tab. */
\r
384 tab.pszText = message_string(NSSM_GUI_TAB_APPLICATION);
\r
385 tab.cchTextMax = (int) strlen(tab.pszText);
\r
386 SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_APPLICATION, (LPARAM) &tab);
\r
387 tablist[NSSM_TAB_APPLICATION] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPLICATION), window, tab_dlg);
\r
388 ShowWindow(tablist[NSSM_TAB_APPLICATION], SW_SHOW);
\r
390 /* Shutdown tab. */
\r
391 tab.pszText = message_string(NSSM_GUI_TAB_SHUTDOWN);
\r
392 tab.cchTextMax = (int) strlen(tab.pszText);
\r
393 SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_SHUTDOWN, (LPARAM) &tab);
\r
394 tablist[NSSM_TAB_SHUTDOWN] = CreateDialog(0, MAKEINTRESOURCE(IDD_SHUTDOWN), window, tab_dlg);
\r
395 ShowWindow(tablist[NSSM_TAB_SHUTDOWN], SW_HIDE);
\r
397 /* Set defaults. */
\r
398 SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_CONSOLE, BM_SETCHECK, BST_CHECKED, 0);
\r
399 SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, NSSM_KILL_CONSOLE_GRACE_PERIOD, 0);
\r
400 SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_WINDOW, BM_SETCHECK, BST_CHECKED, 0);
\r
401 SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, NSSM_KILL_WINDOW_GRACE_PERIOD, 0);
\r
402 SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_THREADS, BM_SETCHECK, BST_CHECKED, 0);
\r
403 SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, NSSM_KILL_THREADS_GRACE_PERIOD, 0);
\r
404 SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_TERMINATE, BM_SETCHECK, BST_CHECKED, 0);
\r
407 tab.pszText = message_string(NSSM_GUI_TAB_EXIT);
\r
408 tab.cchTextMax = (int) strlen(tab.pszText);
\r
409 SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_EXIT, (LPARAM) &tab);
\r
410 tablist[NSSM_TAB_EXIT] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPEXIT), window, tab_dlg);
\r
411 ShowWindow(tablist[NSSM_TAB_EXIT], SW_HIDE);
\r
413 /* Set defaults. */
\r
414 SetDlgItemInt(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, NSSM_RESET_THROTTLE_RESTART, 0);
\r
415 combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);
\r
416 SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_RESTART, (LPARAM) message_string(NSSM_GUI_EXIT_RESTART));
\r
417 SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_IGNORE, (LPARAM) message_string(NSSM_GUI_EXIT_IGNORE));
\r
418 SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_REALLY, (LPARAM) message_string(NSSM_GUI_EXIT_REALLY));
\r
419 SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_UNCLEAN, (LPARAM) message_string(NSSM_GUI_EXIT_UNCLEAN));
\r
420 SendMessage(combo, CB_SETCURSEL, NSSM_EXIT_RESTART, 0);
\r
423 tab.pszText = message_string(NSSM_GUI_TAB_IO);
\r
424 tab.cchTextMax = (int) strlen(tab.pszText) + 1;
\r
425 SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_IO, (LPARAM) &tab);
\r
426 tablist[NSSM_TAB_IO] = CreateDialog(0, MAKEINTRESOURCE(IDD_IO), window, tab_dlg);
\r
427 ShowWindow(tablist[NSSM_TAB_IO], SW_HIDE);
\r
435 NMHDR *notification;
\r
437 notification = (NMHDR *) l;
\r
438 switch (notification->code) {
\r
439 case TCN_SELCHANGE:
\r
443 tabs = GetDlgItem(window, IDC_TAB1);
\r
444 if (! tabs) return 0;
\r
446 selection = (int) SendMessage(tabs, TCM_GETCURSEL, 0, 0);
\r
447 if (selection != selected_tab) {
\r
448 ShowWindow(tablist[selected_tab], SW_HIDE);
\r
450 XXX: Sets focus to the service name which isn't ideal but is
\r
451 better than leaving it in another tab.
\r
453 ShowWindow(tablist[selection], SW_SHOWDEFAULT);
\r
454 SetFocus(tablist[selection]);
\r
455 selected_tab = selection;
\r
462 /* Button was pressed or control was controlled */
\r
464 switch (LOWORD(w)) {
\r
467 if (! install(window)) PostQuitMessage(0);
\r
470 /* Cancel button */
\r
472 DestroyWindow(window);
\r
475 /* Remove button */
\r
477 if (! remove(window)) PostQuitMessage(0);
\r
482 /* Window closing */
\r
484 DestroyWindow(window);
\r
487 PostQuitMessage(0);
\r