gui.cpp

   1 #include "nssm.h"\r
   2 \r
   3 static enum { NSSM_TAB_APPLICATION, NSSM_TAB_DETAILS, NSSM_TAB_SHUTDOWN, NSSM_TAB_EXIT, NSSM_TAB_IO, NSSM_TAB_ROTATION, NSSM_TAB_ENVIRONMENT, NSSM_NUM_TABS };\r
   4 static HWND tablist[NSSM_NUM_TABS];\r
   5 static int selected_tab;\r
   6 \r
   7 int nssm_gui(int resource, TCHAR *name) {\r
   8   /* Create window */\r
   9   HWND dlg = CreateDialog(0, MAKEINTRESOURCE(resource), 0, install_dlg);\r
  10   if (! dlg) {\r
  11     popup_message(MB_OK, NSSM_GUI_CREATEDIALOG_FAILED, error_string(GetLastError()));\r
  12     return 1;\r
  13   }\r
  14 \r
  15   /* Display the window */\r
  16   centre_window(dlg);\r
  17   ShowWindow(dlg, SW_SHOW);\r
  18 \r
  19   /* Set service name if given */\r
  20   if (name) {\r
  21     SetDlgItemText(dlg, IDC_NAME, name);\r
  22     /* No point making user click remove if the name is already entered */\r
  23     if (resource == IDD_REMOVE) {\r
  24       HWND button = GetDlgItem(dlg, IDC_REMOVE);\r
  25       if (button) {\r
  26         SendMessage(button, WM_LBUTTONDOWN, 0, 0);\r
  27         SendMessage(button, WM_LBUTTONUP, 0, 0);\r
  28       }\r
  29     }\r
  30   }\r
  31 \r
  32   /* Go! */\r
  33   MSG message;\r
  34   while (GetMessage(&message, 0, 0, 0)) {\r
  35     if (IsDialogMessage(dlg, &message)) continue;\r
  36     TranslateMessage(&message);\r
  37     DispatchMessage(&message);\r
  38   }\r
  39 \r
  40   return (int) message.wParam;\r
  41 }\r
  42 \r
  43 void centre_window(HWND window) {\r
  44   HWND desktop;\r
  45   RECT size, desktop_size;\r
  46   unsigned long x, y;\r
  47 \r
  48   if (! window) return;\r
  49 \r
  50   /* Find window size */\r
  51   if (! GetWindowRect(window, &size)) return;\r
  52   \r
  53   /* Find desktop window */\r
  54   desktop = GetDesktopWindow();\r
  55   if (! desktop) return;\r
  56 \r
  57   /* Find desktop window size */\r
  58   if (! GetWindowRect(desktop, &desktop_size)) return;\r
  59 \r
  60   /* Centre window */\r
  61   x = (desktop_size.right - size.right) / 2;\r
  62   y = (desktop_size.bottom - size.bottom) / 2;\r
  63   MoveWindow(window, x, y, size.right - size.left, size.bottom - size.top, 0);\r
  64 }\r
  65 \r
  66 static inline void check_stop_method(nssm_service_t *service, unsigned long method, unsigned long control) {\r
  67   if (SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], control, BM_GETCHECK, 0, 0) & BST_CHECKED) return;\r
  68   service->stop_method &= ~method;\r
  69 }\r
  70 \r
  71 static inline void check_number(HWND tab, unsigned long control, unsigned long *timeout) {\r
  72   BOOL translated;\r
  73   unsigned long configured = GetDlgItemInt(tab, control, &translated, 0);\r
  74   if (translated) *timeout = configured;\r
  75 }\r
  76 \r
  77 static inline void set_timeout_enabled(unsigned long control, unsigned long dependent) {\r
  78   unsigned char enabled = 0;\r
  79   if (SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], control, BM_GETCHECK, 0, 0) & BST_CHECKED) enabled = 1;\r
  80   EnableWindow(GetDlgItem(tablist[NSSM_TAB_SHUTDOWN], dependent), enabled);\r
  81 }\r
  82 \r
  83 static inline void set_rotation_enabled(unsigned char enabled) {\r
  84   EnableWindow(GetDlgItem(tablist[NSSM_TAB_ROTATION], IDC_ROTATE_SECONDS), enabled);\r
  85   EnableWindow(GetDlgItem(tablist[NSSM_TAB_ROTATION], IDC_ROTATE_BYTES_LOW), enabled);\r
  86 }\r
  87 \r
  88 static inline void check_io(TCHAR *name, TCHAR *buffer, unsigned long len, unsigned long control) {\r
  89   if (! SendMessage(GetDlgItem(tablist[NSSM_TAB_IO], control), WM_GETTEXTLENGTH, 0, 0)) return;\r
  90   if (GetDlgItemText(tablist[NSSM_TAB_IO], control, buffer, (int) len)) return;\r
  91   popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, name);\r
  92   ZeroMemory(buffer, len * sizeof(TCHAR));\r
  93 }\r
  94 \r
  95 /* Install the service. */\r
  96 int install(HWND window) {\r
  97   if (! window) return 1;\r
  98 \r
  99   nssm_service_t *service = alloc_nssm_service();\r
 100   if (service) {\r
 101     set_nssm_service_defaults(service);\r
 102 \r
 103     /* Get service name. */\r
 104     if (! GetDlgItemText(window, IDC_NAME, service->name, _countof(service->name))) {\r
 105       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_SERVICE_NAME);\r
 106       cleanup_nssm_service(service);\r
 107       return 2;\r
 108     }\r
 109 \r
 110     /* Get executable name */\r
 111     if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_PATH, service->exe, _countof(service->exe))) {\r
 112       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_PATH);\r
 113       return 3;\r
 114     }\r
 115 \r
 116     /* Get startup directory. */\r
 117     if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_DIR, service->dir, _countof(service->dir))) {\r
 118       _sntprintf_s(service->dir, _countof(service->dir), _TRUNCATE, _T("%s"), service->exe);\r
 119       strip_basename(service->dir);\r
 120     }\r
 121 \r
 122     /* Get flags. */\r
 123     if (SendMessage(GetDlgItem(tablist[NSSM_TAB_APPLICATION], IDC_FLAGS), WM_GETTEXTLENGTH, 0, 0)) {\r
 124       if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_FLAGS, service->flags, _countof(service->flags))) {\r
 125         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_OPTIONS);\r
 126         return 4;\r
 127       }\r
 128     }\r
 129 \r
 130     /* Get details. */\r
 131     if (SendMessage(GetDlgItem(tablist[NSSM_TAB_DETAILS], IDC_DISPLAYNAME), WM_GETTEXTLENGTH, 0, 0)) {\r
 132       if (! GetDlgItemText(tablist[NSSM_TAB_DETAILS], IDC_DISPLAYNAME, service->displayname, _countof(service->displayname))) {\r
 133         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_DISPLAYNAME);\r
 134         return 5;\r
 135       }\r
 136     }\r
 137 \r
 138     if (SendMessage(GetDlgItem(tablist[NSSM_TAB_DETAILS], IDC_DESCRIPTION), WM_GETTEXTLENGTH, 0, 0)) {\r
 139       if (! GetDlgItemText(tablist[NSSM_TAB_DETAILS], IDC_DESCRIPTION, service->description, _countof(service->description))) {\r
 140         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_DESCRIPTION);\r
 141         return 5;\r
 142       }\r
 143     }\r
 144 \r
 145     HWND combo = GetDlgItem(tablist[NSSM_TAB_DETAILS], IDC_STARTUP);\r
 146     service->startup = (unsigned long) SendMessage(combo, CB_GETCURSEL, 0, 0);\r
 147     if (service->startup == CB_ERR) service->startup = 0;\r
 148 \r
 149     /* Get stop method stuff. */\r
 150     check_stop_method(service, NSSM_STOP_METHOD_CONSOLE, IDC_METHOD_CONSOLE);\r
 151     check_stop_method(service, NSSM_STOP_METHOD_WINDOW, IDC_METHOD_WINDOW);\r
 152     check_stop_method(service, NSSM_STOP_METHOD_THREADS, IDC_METHOD_THREADS);\r
 153     check_stop_method(service, NSSM_STOP_METHOD_TERMINATE, IDC_METHOD_TERMINATE);\r
 154     check_number(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, &service->kill_console_delay);\r
 155     check_number(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, &service->kill_window_delay);\r
 156     check_number(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, &service->kill_threads_delay);\r
 157 \r
 158     /* Get exit action stuff. */\r
 159     check_number(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, &service->throttle_delay);\r
 160     combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);\r
 161     service->default_exit_action = (unsigned long) SendMessage(combo, CB_GETCURSEL, 0, 0);\r
 162     if (service->default_exit_action == CB_ERR) service->default_exit_action = 0;\r
 163 \r
 164     /* Get I/O stuff. */\r
 165     check_io(_T("stdin"), service->stdin_path, _countof(service->stdin_path), IDC_STDIN);\r
 166     check_io(_T("stdout"), service->stdout_path, _countof(service->stdout_path), IDC_STDOUT);\r
 167     check_io(_T("stderr"), service->stderr_path, _countof(service->stderr_path), IDC_STDERR);\r
 168 \r
 169     /* Override stdout and/or stderr. */\r
 170     if (SendDlgItemMessage(tablist[NSSM_TAB_ROTATION], IDC_TRUNCATE, BM_GETCHECK, 0, 0) & BST_CHECKED) {\r
 171       if (service->stdout_path[0]) service->stdout_disposition = CREATE_ALWAYS;\r
 172       if (service->stderr_path[0]) service->stderr_disposition = CREATE_ALWAYS;\r
 173     }\r
 174 \r
 175     /* Get rotation stuff. */\r
 176     if (SendDlgItemMessage(tablist[NSSM_TAB_ROTATION], IDC_ROTATE, BM_GETCHECK, 0, 0) & BST_CHECKED) {\r
 177       service->rotate_files = true;\r
 178       check_number(tablist[NSSM_TAB_ROTATION], IDC_ROTATE_SECONDS, &service->rotate_seconds);\r
 179       check_number(tablist[NSSM_TAB_ROTATION], IDC_ROTATE_BYTES_LOW, &service->rotate_bytes_low);\r
 180     }\r
 181 \r
 182     /* Get environment. */\r
 183     unsigned long envlen = (unsigned long) SendMessage(GetDlgItem(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT), WM_GETTEXTLENGTH, 0, 0);\r
 184     if (envlen) {\r
 185       TCHAR *env = (TCHAR *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, (envlen + 2) * sizeof(TCHAR));\r
 186       if (! env) {\r
 187         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("environment"), _T("install()"));\r
 188         cleanup_nssm_service(service);\r
 189         return 5;\r
 190       }\r
 191 \r
 192       if (! GetDlgItemText(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT, env, envlen + 1)) {\r
 193         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_ENVIRONMENT);\r
 194         HeapFree(GetProcessHeap(), 0, env);\r
 195         cleanup_nssm_service(service);\r
 196         return 5;\r
 197       }\r
 198 \r
 199       /* Strip CR and replace LF with NULL. */\r
 200       unsigned long newlen = 0;\r
 201       unsigned long i, j;\r
 202       for (i = 0; i < envlen; i++) if (env[i] != _T('\r')) newlen++;\r
 203       /* Must end with two NULLs. */\r
 204       newlen += 2;\r
 205 \r
 206       TCHAR *newenv = (TCHAR *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, newlen * sizeof(TCHAR));\r
 207       if (! newenv) {\r
 208         HeapFree(GetProcessHeap(), 0, env);\r
 209         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("environment"), _T("install()"));\r
 210         cleanup_nssm_service(service);\r
 211         return 5;\r
 212       }\r
 213 \r
 214       for (i = 0, j = 0; i < envlen; i++) {\r
 215         if (env[i] == _T('\r')) continue;\r
 216         if (env[i] == _T('\n')) newenv[j] = _T('\0');\r
 217         else newenv[j] = env[i];\r
 218         j++;\r
 219       }\r
 220 \r
 221       HeapFree(GetProcessHeap(), 0, env);\r
 222       env = newenv;\r
 223       envlen = newlen;\r
 224 \r
 225       /* Test the environment is valid. */\r
 226       if (test_environment(env)) {\r
 227         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_ENVIRONMENT);\r
 228         HeapFree(GetProcessHeap(), 0, env);\r
 229         cleanup_nssm_service(service);\r
 230         return 5;\r
 231       }\r
 232 \r
 233       if (SendDlgItemMessage(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT_REPLACE, BM_GETCHECK, 0, 0) & BST_CHECKED) {\r
 234         service->env = env;\r
 235         service->envlen = envlen;\r
 236       }\r
 237       else {\r
 238         service->env_extra = env;\r
 239         service->env_extralen = envlen;\r
 240       }\r
 241     }\r
 242   }\r
 243 \r
 244   /* See if it works. */\r
 245   switch (install_service(service)) {\r
 246     case 1:\r
 247       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("service"), _T("install()"));\r
 248       cleanup_nssm_service(service);\r
 249       return 1;\r
 250 \r
 251     case 2:\r
 252       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);\r
 253       cleanup_nssm_service(service);\r
 254       return 2;\r
 255 \r
 256     case 3:\r
 257       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, NSSM);\r
 258       cleanup_nssm_service(service);\r
 259       return 3;\r
 260 \r
 261     case 4:\r
 262       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_OUT_OF_MEMORY_FOR_IMAGEPATH);\r
 263       cleanup_nssm_service(service);\r
 264       return 4;\r
 265 \r
 266     case 5:\r
 267       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INSTALL_SERVICE_FAILED);\r
 268       cleanup_nssm_service(service);\r
 269       return 5;\r
 270 \r
 271     case 6:\r
 272       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_CREATE_PARAMETERS_FAILED);\r
 273       cleanup_nssm_service(service);\r
 274       return 6;\r
 275   }\r
 276 \r
 277   popup_message(MB_OK, NSSM_MESSAGE_SERVICE_INSTALLED, service->name);\r
 278   cleanup_nssm_service(service);\r
 279   return 0;\r
 280 }\r
 281 \r
 282 /* Remove the service */\r
 283 int remove(HWND window) {\r
 284   if (! window) return 1;\r
 285 \r
 286   /* See if it works */\r
 287   nssm_service_t *service = alloc_nssm_service();\r
 288   if (service) {\r
 289     /* Get service name */\r
 290     if (! GetDlgItemText(window, IDC_NAME, service->name, _countof(service->name))) {\r
 291       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_SERVICE_NAME);\r
 292       cleanup_nssm_service(service);\r
 293       return 2;\r
 294     }\r
 295 \r
 296     /* Confirm */\r
 297     if (popup_message(MB_YESNO, NSSM_GUI_ASK_REMOVE_SERVICE, service->name) != IDYES) {\r
 298       cleanup_nssm_service(service);\r
 299       return 0;\r
 300     }\r
 301   }\r
 302 \r
 303   switch (remove_service(service)) {\r
 304     case 1:\r
 305       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("service"), _T("remove()"));\r
 306       cleanup_nssm_service(service);\r
 307       return 1;\r
 308 \r
 309     case 2:\r
 310       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);\r
 311       cleanup_nssm_service(service);\r
 312       return 2;\r
 313 \r
 314     case 3:\r
 315       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_SERVICE_NOT_INSTALLED);\r
 316       return 3;\r
 317       cleanup_nssm_service(service);\r
 318 \r
 319     case 4:\r
 320       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_REMOVE_SERVICE_FAILED);\r
 321       cleanup_nssm_service(service);\r
 322       return 4;\r
 323   }\r
 324 \r
 325   popup_message(MB_OK, NSSM_MESSAGE_SERVICE_REMOVED, service->name);\r
 326   cleanup_nssm_service(service);\r
 327   return 0;\r
 328 }\r
 329 \r
 330 static TCHAR *browse_filter(int message) {\r
 331   switch (message) {\r
 332     case NSSM_GUI_BROWSE_FILTER_APPLICATIONS: return _T("*.exe;*.bat;*.cmd");\r
 333     case NSSM_GUI_BROWSE_FILTER_DIRECTORIES: return _T(".");\r
 334     case NSSM_GUI_BROWSE_FILTER_ALL_FILES: /* Fall through. */\r
 335     default: return _T("*.*");\r
 336   }\r
 337 }\r
 338 \r
 339 UINT_PTR CALLBACK browse_hook(HWND dlg, UINT message, WPARAM w, LPARAM l) {\r
 340   switch (message) {\r
 341     case WM_INITDIALOG:\r
 342       return 1;\r
 343   }\r
 344 \r
 345   return 0;\r
 346 }\r
 347 \r
 348 /* Browse for application */\r
 349 void browse(HWND window, TCHAR *current, unsigned long flags, ...) {\r
 350   if (! window) return;\r
 351 \r
 352   va_list arg;\r
 353   size_t bufsize = 256;\r
 354   size_t len = bufsize;\r
 355   int i;\r
 356 \r
 357   OPENFILENAME ofn;\r
 358   ZeroMemory(&ofn, sizeof(ofn));\r
 359   ofn.lStructSize = sizeof(ofn);\r
 360   ofn.lpstrFilter = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, bufsize * sizeof(TCHAR));\r
 361   /* XXX: Escaping nulls with FormatMessage is tricky */\r
 362   if (ofn.lpstrFilter) {\r
 363     ZeroMemory((void *) ofn.lpstrFilter, bufsize);\r
 364     len = 0;\r
 365     /* "Applications" + NULL + "*.exe" + NULL */\r
 366     va_start(arg, flags);\r
 367     while (i = va_arg(arg, int)) {\r
 368       TCHAR *localised = message_string(i);\r
 369       _sntprintf_s((TCHAR *) ofn.lpstrFilter + len, bufsize, _TRUNCATE, localised);\r
 370       len += _tcslen(localised) + 1;\r
 371       LocalFree(localised);\r
 372       TCHAR *filter = browse_filter(i);\r
 373       _sntprintf_s((TCHAR *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, _T("%s"), filter);\r
 374       len += _tcslen(filter) + 1;\r
 375     }\r
 376     va_end(arg);\r
 377     /* Remainder of the buffer is already zeroed */\r
 378   }\r
 379   ofn.lpstrFile = new TCHAR[MAX_PATH];\r
 380   if (flags & OFN_NOVALIDATE) {\r
 381     /* Directory hack. */\r
 382     _sntprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, _T(":%s:"), message_string(NSSM_GUI_BROWSE_FILTER_DIRECTORIES));\r
 383   }\r
 384   else _sntprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, _T("%s"), current);\r
 385   ofn.lpstrTitle = message_string(NSSM_GUI_BROWSE_TITLE);\r
 386   ofn.nMaxFile = MAX_PATH;\r
 387   ofn.Flags = OFN_EXPLORER | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST | flags;\r
 388 \r
 389   if (GetOpenFileName(&ofn)) {\r
 390     /* Directory hack. */\r
 391     if (flags & OFN_NOVALIDATE) strip_basename(ofn.lpstrFile);\r
 392     SendMessage(window, WM_SETTEXT, 0, (LPARAM) ofn.lpstrFile);\r
 393   }\r
 394   if (ofn.lpstrFilter) HeapFree(GetProcessHeap(), 0, (void *) ofn.lpstrFilter);\r
 395 \r
 396   delete[] ofn.lpstrFile;\r
 397 }\r
 398 \r
 399 INT_PTR CALLBACK tab_dlg(HWND tab, UINT message, WPARAM w, LPARAM l) {\r
 400   switch (message) {\r
 401     case WM_INITDIALOG:\r
 402       return 1;\r
 403 \r
 404     /* Button was pressed or control was controlled. */\r
 405     case WM_COMMAND:\r
 406       HWND dlg;\r
 407       TCHAR buffer[MAX_PATH];\r
 408       unsigned char enabled;\r
 409 \r
 410       switch (LOWORD(w)) {\r
 411         /* Browse for application. */\r
 412         case IDC_BROWSE:\r
 413           dlg = GetDlgItem(tab, IDC_PATH);\r
 414           GetDlgItemText(tab, IDC_PATH, buffer, _countof(buffer));\r
 415           browse(dlg, buffer, OFN_FILEMUSTEXIST, NSSM_GUI_BROWSE_FILTER_APPLICATIONS, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 416           /* Fill in startup directory if it wasn't already specified. */\r
 417           GetDlgItemText(tab, IDC_DIR, buffer, _countof(buffer));\r
 418           if (! buffer[0]) {\r
 419             GetDlgItemText(tab, IDC_PATH, buffer, _countof(buffer));\r
 420             strip_basename(buffer);\r
 421             SetDlgItemText(tab, IDC_DIR, buffer);\r
 422           }\r
 423           break;\r
 424 \r
 425         /* Browse for startup directory. */\r
 426         case IDC_BROWSE_DIR:\r
 427           dlg = GetDlgItem(tab, IDC_DIR);\r
 428           GetDlgItemText(tab, IDC_DIR, buffer, _countof(buffer));\r
 429           browse(dlg, buffer, OFN_NOVALIDATE, NSSM_GUI_BROWSE_FILTER_DIRECTORIES, 0);\r
 430           break;\r
 431 \r
 432         /* Shutdown methods. */\r
 433         case IDC_METHOD_CONSOLE:\r
 434           set_timeout_enabled(LOWORD(w), IDC_KILL_CONSOLE);\r
 435           break;\r
 436 \r
 437         case IDC_METHOD_WINDOW:\r
 438           set_timeout_enabled(LOWORD(w), IDC_KILL_WINDOW);\r
 439           break;\r
 440 \r
 441         case IDC_METHOD_THREADS:\r
 442           set_timeout_enabled(LOWORD(w), IDC_KILL_THREADS);\r
 443           break;\r
 444 \r
 445         /* Browse for stdin. */\r
 446         case IDC_BROWSE_STDIN:\r
 447           dlg = GetDlgItem(tab, IDC_STDIN);\r
 448           GetDlgItemText(tab, IDC_STDIN, buffer, _countof(buffer));\r
 449           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 450           break;\r
 451 \r
 452         /* Browse for stdout. */\r
 453         case IDC_BROWSE_STDOUT:\r
 454           dlg = GetDlgItem(tab, IDC_STDOUT);\r
 455           GetDlgItemText(tab, IDC_STDOUT, buffer, _countof(buffer));\r
 456           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 457           /* Fill in stderr if it wasn't already specified. */\r
 458           GetDlgItemText(tab, IDC_STDERR, buffer, _countof(buffer));\r
 459           if (! buffer[0]) {\r
 460             GetDlgItemText(tab, IDC_STDOUT, buffer, _countof(buffer));\r
 461             SetDlgItemText(tab, IDC_STDERR, buffer);\r
 462           }\r
 463           break;\r
 464 \r
 465         /* Browse for stderr. */\r
 466         case IDC_BROWSE_STDERR:\r
 467           dlg = GetDlgItem(tab, IDC_STDERR);\r
 468           GetDlgItemText(tab, IDC_STDERR, buffer, _countof(buffer));\r
 469           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 470           break;\r
 471 \r
 472         /* Rotation. */\r
 473         case IDC_ROTATE:\r
 474           if (SendDlgItemMessage(tab, LOWORD(w), BM_GETCHECK, 0, 0) & BST_CHECKED) enabled = 1;\r
 475           else enabled = 0;\r
 476           set_rotation_enabled(enabled);\r
 477           break;\r
 478       }\r
 479       return 1;\r
 480   }\r
 481 \r
 482   return 0;\r
 483 }\r
 484 \r
 485 /* Install/remove dialogue callback */\r
 486 INT_PTR CALLBACK install_dlg(HWND window, UINT message, WPARAM w, LPARAM l) {\r
 487   switch (message) {\r
 488     /* Creating the dialogue */\r
 489     case WM_INITDIALOG:\r
 490       SetFocus(GetDlgItem(window, IDC_NAME));\r
 491 \r
 492       HWND tabs;\r
 493       HWND combo;\r
 494       tabs = GetDlgItem(window, IDC_TAB1);\r
 495       if (! tabs) return 0;\r
 496 \r
 497       /* Set up tabs. */\r
 498       TCITEM tab;\r
 499       ZeroMemory(&tab, sizeof(tab));\r
 500       tab.mask = TCIF_TEXT;\r
 501 \r
 502       /* Application tab. */\r
 503       tab.pszText = message_string(NSSM_GUI_TAB_APPLICATION);\r
 504       tab.cchTextMax = (int) _tcslen(tab.pszText);\r
 505       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_APPLICATION, (LPARAM) &tab);\r
 506       tablist[NSSM_TAB_APPLICATION] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPLICATION), window, tab_dlg);\r
 507       ShowWindow(tablist[NSSM_TAB_APPLICATION], SW_SHOW);\r
 508 \r
 509       /* Details tab. */\r
 510       tab.pszText = message_string(NSSM_GUI_TAB_DETAILS);\r
 511       tab.cchTextMax = (int) _tcslen(tab.pszText);\r
 512       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_DETAILS, (LPARAM) &tab);\r
 513       tablist[NSSM_TAB_DETAILS] = CreateDialog(0, MAKEINTRESOURCE(IDD_DETAILS), window, tab_dlg);\r
 514       ShowWindow(tablist[NSSM_TAB_DETAILS], SW_HIDE);\r
 515 \r
 516       /* Set defaults. */\r
 517       combo = GetDlgItem(tablist[NSSM_TAB_DETAILS], IDC_STARTUP);\r
 518       SendMessage(combo, CB_INSERTSTRING, NSSM_STARTUP_AUTOMATIC, (LPARAM) message_string(NSSM_GUI_STARTUP_AUTOMATIC));\r
 519       SendMessage(combo, CB_INSERTSTRING, NSSM_STARTUP_DELAYED, (LPARAM) message_string(NSSM_GUI_STARTUP_DELAYED));\r
 520       SendMessage(combo, CB_INSERTSTRING, NSSM_STARTUP_MANUAL, (LPARAM) message_string(NSSM_GUI_STARTUP_MANUAL));\r
 521       SendMessage(combo, CB_INSERTSTRING, NSSM_STARTUP_DISABLED, (LPARAM) message_string(NSSM_GUI_STARTUP_DISABLED));\r
 522       SendMessage(combo, CB_SETCURSEL, NSSM_STARTUP_AUTOMATIC, 0);\r
 523 \r
 524       /* Shutdown tab. */\r
 525       tab.pszText = message_string(NSSM_GUI_TAB_SHUTDOWN);\r
 526       tab.cchTextMax = (int) _tcslen(tab.pszText);\r
 527       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_SHUTDOWN, (LPARAM) &tab);\r
 528       tablist[NSSM_TAB_SHUTDOWN] = CreateDialog(0, MAKEINTRESOURCE(IDD_SHUTDOWN), window, tab_dlg);\r
 529       ShowWindow(tablist[NSSM_TAB_SHUTDOWN], SW_HIDE);\r
 530 \r
 531       /* Set defaults. */\r
 532       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_CONSOLE, BM_SETCHECK, BST_CHECKED, 0);\r
 533       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, NSSM_KILL_CONSOLE_GRACE_PERIOD, 0);\r
 534       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_WINDOW, BM_SETCHECK, BST_CHECKED, 0);\r
 535       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, NSSM_KILL_WINDOW_GRACE_PERIOD, 0);\r
 536       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_THREADS, BM_SETCHECK, BST_CHECKED, 0);\r
 537       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, NSSM_KILL_THREADS_GRACE_PERIOD, 0);\r
 538       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_TERMINATE, BM_SETCHECK, BST_CHECKED, 0);\r
 539 \r
 540       /* Restart tab. */\r
 541       tab.pszText = message_string(NSSM_GUI_TAB_EXIT);\r
 542       tab.cchTextMax = (int) _tcslen(tab.pszText);\r
 543       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_EXIT, (LPARAM) &tab);\r
 544       tablist[NSSM_TAB_EXIT] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPEXIT), window, tab_dlg);\r
 545       ShowWindow(tablist[NSSM_TAB_EXIT], SW_HIDE);\r
 546 \r
 547       /* Set defaults. */\r
 548       SetDlgItemInt(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, NSSM_RESET_THROTTLE_RESTART, 0);\r
 549       combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);\r
 550       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_RESTART, (LPARAM) message_string(NSSM_GUI_EXIT_RESTART));\r
 551       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_IGNORE, (LPARAM) message_string(NSSM_GUI_EXIT_IGNORE));\r
 552       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_REALLY, (LPARAM) message_string(NSSM_GUI_EXIT_REALLY));\r
 553       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_UNCLEAN, (LPARAM) message_string(NSSM_GUI_EXIT_UNCLEAN));\r
 554       SendMessage(combo, CB_SETCURSEL, NSSM_EXIT_RESTART, 0);\r
 555 \r
 556       /* I/O tab. */\r
 557       tab.pszText = message_string(NSSM_GUI_TAB_IO);\r
 558       tab.cchTextMax = (int) _tcslen(tab.pszText) + 1;\r
 559       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_IO, (LPARAM) &tab);\r
 560       tablist[NSSM_TAB_IO] = CreateDialog(0, MAKEINTRESOURCE(IDD_IO), window, tab_dlg);\r
 561       ShowWindow(tablist[NSSM_TAB_IO], SW_HIDE);\r
 562 \r
 563       /* Rotation tab. */\r
 564       tab.pszText = message_string(NSSM_GUI_TAB_ROTATION);\r
 565       tab.cchTextMax = (int) _tcslen(tab.pszText) + 1;\r
 566       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_ROTATION, (LPARAM) &tab);\r
 567       tablist[NSSM_TAB_ROTATION] = CreateDialog(0, MAKEINTRESOURCE(IDD_ROTATION), window, tab_dlg);\r
 568       ShowWindow(tablist[NSSM_TAB_ROTATION], SW_HIDE);\r
 569 \r
 570       /* Set defaults. */\r
 571       SetDlgItemInt(tablist[NSSM_TAB_ROTATION], IDC_ROTATE_SECONDS, 0, 0);\r
 572       SetDlgItemInt(tablist[NSSM_TAB_ROTATION], IDC_ROTATE_BYTES_LOW, 0, 0);\r
 573       set_rotation_enabled(0);\r
 574 \r
 575       /* Environment tab. */\r
 576       tab.pszText = message_string(NSSM_GUI_TAB_ENVIRONMENT);\r
 577       tab.cchTextMax = (int) _tcslen(tab.pszText) + 1;\r
 578       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_ENVIRONMENT, (LPARAM) &tab);\r
 579       tablist[NSSM_TAB_ENVIRONMENT] = CreateDialog(0, MAKEINTRESOURCE(IDD_ENVIRONMENT), window, tab_dlg);\r
 580       ShowWindow(tablist[NSSM_TAB_ENVIRONMENT], SW_HIDE);\r
 581 \r
 582       selected_tab = 0;\r
 583 \r
 584       return 1;\r
 585 \r
 586     /* Tab change. */\r
 587     case WM_NOTIFY:\r
 588       NMHDR *notification;\r
 589 \r
 590       notification = (NMHDR *) l;\r
 591       switch (notification->code) {\r
 592         case TCN_SELCHANGE:\r
 593           HWND tabs;\r
 594           int selection;\r
 595 \r
 596           tabs = GetDlgItem(window, IDC_TAB1);\r
 597           if (! tabs) return 0;\r
 598 \r
 599           selection = (int) SendMessage(tabs, TCM_GETCURSEL, 0, 0);\r
 600           if (selection != selected_tab) {\r
 601             ShowWindow(tablist[selected_tab], SW_HIDE);\r
 602             /*\r
 603               XXX: Sets focus to the service name which isn't ideal but is\r
 604                    better than leaving it in another tab.\r
 605             */\r
 606             ShowWindow(tablist[selection], SW_SHOWDEFAULT);\r
 607             SetFocus(tablist[selection]);\r
 608             selected_tab = selection;\r
 609           }\r
 610           return 1;\r
 611       }\r
 612 \r
 613       return 0;\r
 614 \r
 615     /* Button was pressed or control was controlled */\r
 616     case WM_COMMAND:\r
 617       switch (LOWORD(w)) {\r
 618         /* OK button */\r
 619         case IDOK:\r
 620           if (! install(window)) PostQuitMessage(0);\r
 621           break;\r
 622 \r
 623         /* Cancel button */\r
 624         case IDCANCEL:\r
 625           DestroyWindow(window);\r
 626           break;\r
 627 \r
 628         /* Remove button */\r
 629         case IDC_REMOVE:\r
 630           if (! remove(window)) PostQuitMessage(0);\r
 631           break;\r
 632       }\r
 633       return 1;\r
 634 \r
 635     /* Window closing */\r
 636     case WM_CLOSE:\r
 637       DestroyWindow(window);\r
 638       return 0;\r
 639     case WM_DESTROY:\r
 640       PostQuitMessage(0);\r
 641   }\r
 642   return 0;\r
 643 }\r