gui.cpp

   1 #include "nssm.h"\r
   2 \r
   3 static enum { NSSM_TAB_APPLICATION, NSSM_TAB_SHUTDOWN, NSSM_TAB_EXIT, NSSM_TAB_IO, NSSM_TAB_ENVIRONMENT, NSSM_NUM_TABS };\r
   4 static HWND tablist[NSSM_NUM_TABS];\r
   5 static int selected_tab;\r
   6 \r
   7 static void strip_basename(char *buffer) {\r
   8   size_t len = strlen(buffer);\r
   9   size_t i;\r
  10   for (i = len; i && buffer[i] != '\\' && buffer[i] != '/'; i--);\r
  11   /* X:\ is OK. */\r
  12   if (i && buffer[i-1] == ':') i++;\r
  13   buffer[i] = '\0';\r
  14 }\r
  15 \r
  16 int nssm_gui(int resource, char *name) {\r
  17   /* Create window */\r
  18   HWND dlg = CreateDialog(0, MAKEINTRESOURCE(resource), 0, install_dlg);\r
  19   if (! dlg) {\r
  20     popup_message(MB_OK, NSSM_GUI_CREATEDIALOG_FAILED, error_string(GetLastError()));\r
  21     return 1;\r
  22   }\r
  23 \r
  24   /* Display the window */\r
  25   centre_window(dlg);\r
  26   ShowWindow(dlg, SW_SHOW);\r
  27 \r
  28   /* Set service name if given */\r
  29   if (name) {\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
  34       if (button) {\r
  35         SendMessage(button, WM_LBUTTONDOWN, 0, 0);\r
  36         SendMessage(button, WM_LBUTTONUP, 0, 0);\r
  37       }\r
  38     }\r
  39   }\r
  40 \r
  41   /* Go! */\r
  42   MSG message;\r
  43   while (GetMessage(&message, 0, 0, 0)) {\r
  44     if (IsDialogMessage(dlg, &message)) continue;\r
  45     TranslateMessage(&message);\r
  46     DispatchMessage(&message);\r
  47   }\r
  48 \r
  49   return (int) message.wParam;\r
  50 }\r
  51 \r
  52 void centre_window(HWND window) {\r
  53   HWND desktop;\r
  54   RECT size, desktop_size;\r
  55   unsigned long x, y;\r
  56 \r
  57   if (! window) return;\r
  58 \r
  59   /* Find window size */\r
  60   if (! GetWindowRect(window, &size)) return;\r
  61   \r
  62   /* Find desktop window */\r
  63   desktop = GetDesktopWindow();\r
  64   if (! desktop) return;\r
  65 \r
  66   /* Find desktop window size */\r
  67   if (! GetWindowRect(desktop, &desktop_size)) return;\r
  68 \r
  69   /* Centre window */\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
  73 }\r
  74 \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
  78 }\r
  79 \r
  80 static inline void check_method_timeout(HWND tab, unsigned long control, unsigned long *timeout) {\r
  81   BOOL translated;\r
  82   unsigned long configured = GetDlgItemInt(tab, control, &translated, 0);\r
  83   if (translated) *timeout = configured;\r
  84 }\r
  85 \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
  91 }\r
  92 \r
  93 /* Install the service. */\r
  94 int install(HWND window) {\r
  95   if (! window) return 1;\r
  96 \r
  97   nssm_service_t *service = alloc_nssm_service();\r
  98   if (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
 103       return 2;\r
 104     }\r
 105 \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
 109       return 3;\r
 110     }\r
 111 \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
 116     }\r
 117 \r
 118     /* Get flags. */\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
 122         return 4;\r
 123       }\r
 124     }\r
 125 \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
 135 \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
 141 \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
 146 \r
 147     /* Get environment. */\r
 148     unsigned long envlen = (unsigned long) SendMessage(GetDlgItem(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT), WM_GETTEXTLENGTH, 0, 0);\r
 149     if (envlen) {\r
 150       char *env = (char *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, envlen + 2);\r
 151       if (! env) {\r
 152         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, "environment", "install()");\r
 153         cleanup_nssm_service(service);\r
 154         return 5;\r
 155       }\r
 156 \r
 157       if (! GetDlgItemText(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT, env, envlen + 1)) {\r
 158         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_ENVIRONMENT);\r
 159         HeapFree(GetProcessHeap(), 0, env);\r
 160         cleanup_nssm_service(service);\r
 161         return 5;\r
 162       }\r
 163 \r
 164       /* Strip CR and replace LF with NULL. */\r
 165       unsigned long newlen = 0;\r
 166       unsigned long i, j;\r
 167       for (i = 0; i < envlen; i++) if (env[i] != '\r') newlen++;\r
 168       /* Must end with two NULLs. */\r
 169       newlen++;\r
 170 \r
 171       char *newenv = (char *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, newlen);\r
 172       if (! newenv) {\r
 173         HeapFree(GetProcessHeap(), 0, env);\r
 174         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, "environment", "install()");\r
 175         cleanup_nssm_service(service);\r
 176         return 5;\r
 177       }\r
 178 \r
 179       for (i = 0, j = 0; i < envlen; i++) {\r
 180         if (env[i] == '\r') continue;\r
 181         if (env[i] == '\n') newenv[j] = '\0';\r
 182         else newenv[j] = env[i];\r
 183         j++;\r
 184       }\r
 185 \r
 186       HeapFree(GetProcessHeap(), 0, env);\r
 187       env = newenv;\r
 188       envlen = newlen;\r
 189 \r
 190       /* Test the environment is valid. */\r
 191       char path[MAX_PATH];\r
 192       GetModuleFileName(0, path, sizeof(path));\r
 193       STARTUPINFO si;\r
 194       ZeroMemory(&si, sizeof(si));\r
 195       si.cb = sizeof(si);\r
 196       PROCESS_INFORMATION pi;\r
 197       ZeroMemory(&pi, sizeof(pi));\r
 198 \r
 199       if (! CreateProcess(0, path, 0, 0, 0, CREATE_SUSPENDED, env, 0, &si, &pi)) {\r
 200         unsigned long error = GetLastError();\r
 201         if (error == ERROR_INVALID_PARAMETER) {\r
 202           popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_ENVIRONMENT);\r
 203           HeapFree(GetProcessHeap(), 0, env);\r
 204           envlen = 0;\r
 205         }\r
 206         cleanup_nssm_service(service);\r
 207         return 5;\r
 208       }\r
 209       TerminateProcess(pi.hProcess, 0);\r
 210 \r
 211       if (SendDlgItemMessage(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT_REPLACE, BM_GETCHECK, 0, 0) & BST_CHECKED) {\r
 212         service->env = env;\r
 213         service->envlen = envlen;\r
 214       }\r
 215       else {\r
 216         service->env_extra = env;\r
 217         service->env_extralen = envlen;\r
 218       }\r
 219     }\r
 220   }\r
 221 \r
 222   /* See if it works. */\r
 223   switch (install_service(service)) {\r
 224     case 1:\r
 225       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, "service", "install()");\r
 226       cleanup_nssm_service(service);\r
 227       return 1;\r
 228 \r
 229     case 2:\r
 230       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);\r
 231       cleanup_nssm_service(service);\r
 232       return 2;\r
 233 \r
 234     case 3:\r
 235       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, NSSM);\r
 236       cleanup_nssm_service(service);\r
 237       return 3;\r
 238 \r
 239     case 4:\r
 240       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_OUT_OF_MEMORY_FOR_IMAGEPATH);\r
 241       cleanup_nssm_service(service);\r
 242       return 4;\r
 243 \r
 244     case 5:\r
 245       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INSTALL_SERVICE_FAILED);\r
 246       cleanup_nssm_service(service);\r
 247       return 5;\r
 248 \r
 249     case 6:\r
 250       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_CREATE_PARAMETERS_FAILED);\r
 251       cleanup_nssm_service(service);\r
 252       return 6;\r
 253   }\r
 254 \r
 255   popup_message(MB_OK, NSSM_MESSAGE_SERVICE_INSTALLED, service->name);\r
 256   cleanup_nssm_service(service);\r
 257   return 0;\r
 258 }\r
 259 \r
 260 /* Remove the service */\r
 261 int remove(HWND window) {\r
 262   if (! window) return 1;\r
 263 \r
 264   /* See if it works */\r
 265   nssm_service_t *service = alloc_nssm_service();\r
 266   if (service) {\r
 267     /* Get service name */\r
 268     if (! GetDlgItemText(window, IDC_NAME, service->name, sizeof(service->name))) {\r
 269       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_SERVICE_NAME);\r
 270       cleanup_nssm_service(service);\r
 271       return 2;\r
 272     }\r
 273 \r
 274     /* Confirm */\r
 275     if (popup_message(MB_YESNO, NSSM_GUI_ASK_REMOVE_SERVICE, service->name) != IDYES) {\r
 276       cleanup_nssm_service(service);\r
 277       return 0;\r
 278     }\r
 279   }\r
 280 \r
 281   switch (remove_service(service)) {\r
 282     case 1:\r
 283       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, "service", "remove()");\r
 284       cleanup_nssm_service(service);\r
 285       return 1;\r
 286 \r
 287     case 2:\r
 288       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);\r
 289       cleanup_nssm_service(service);\r
 290       return 2;\r
 291 \r
 292     case 3:\r
 293       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_SERVICE_NOT_INSTALLED);\r
 294       return 3;\r
 295       cleanup_nssm_service(service);\r
 296 \r
 297     case 4:\r
 298       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_REMOVE_SERVICE_FAILED);\r
 299       cleanup_nssm_service(service);\r
 300       return 4;\r
 301   }\r
 302 \r
 303   popup_message(MB_OK, NSSM_MESSAGE_SERVICE_REMOVED, service->name);\r
 304   cleanup_nssm_service(service);\r
 305   return 0;\r
 306 }\r
 307 \r
 308 static char *browse_filter(int message) {\r
 309   switch (message) {\r
 310     case NSSM_GUI_BROWSE_FILTER_APPLICATIONS: return "*.exe;*.bat;*.cmd";\r
 311     case NSSM_GUI_BROWSE_FILTER_DIRECTORIES: return ".";\r
 312     case NSSM_GUI_BROWSE_FILTER_ALL_FILES: /* Fall through. */\r
 313     default: return "*.*";\r
 314   }\r
 315 }\r
 316 \r
 317 UINT_PTR CALLBACK browse_hook(HWND dlg, UINT message, WPARAM w, LPARAM l) {\r
 318   switch (message) {\r
 319     case WM_INITDIALOG:\r
 320       return 1;\r
 321   }\r
 322 \r
 323   return 0;\r
 324 }\r
 325 \r
 326 /* Browse for application */\r
 327 void browse(HWND window, char *current, unsigned long flags, ...) {\r
 328   if (! window) return;\r
 329 \r
 330   va_list arg;\r
 331   size_t bufsize = 256;\r
 332   size_t len = bufsize;\r
 333   int i;\r
 334 \r
 335   OPENFILENAME ofn;\r
 336   ZeroMemory(&ofn, sizeof(ofn));\r
 337   ofn.lStructSize = sizeof(ofn);\r
 338   ofn.lpstrFilter = (char *) HeapAlloc(GetProcessHeap(), 0, bufsize);\r
 339   /* XXX: Escaping nulls with FormatMessage is tricky */\r
 340   if (ofn.lpstrFilter) {\r
 341     ZeroMemory((void *) ofn.lpstrFilter, bufsize);\r
 342     len = 0;\r
 343     /* "Applications" + NULL + "*.exe" + NULL */\r
 344     va_start(arg, flags);\r
 345     while (i = va_arg(arg, int)) {\r
 346       char *localised = message_string(i);\r
 347       _snprintf_s((char *) ofn.lpstrFilter + len, bufsize, _TRUNCATE, localised);\r
 348       len += strlen(localised) + 1;\r
 349       LocalFree(localised);\r
 350       char *filter = browse_filter(i);\r
 351       _snprintf_s((char *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, "%s", filter);\r
 352       len += strlen(filter) + 1;\r
 353     }\r
 354     va_end(arg);\r
 355     /* Remainder of the buffer is already zeroed */\r
 356   }\r
 357   ofn.lpstrFile = new char[MAX_PATH];\r
 358   if (flags & OFN_NOVALIDATE) {\r
 359     /* Directory hack. */\r
 360     _snprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, ":%s:", message_string(NSSM_GUI_BROWSE_FILTER_DIRECTORIES));\r
 361   }\r
 362   else _snprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, "%s", current);\r
 363   ofn.lpstrTitle = message_string(NSSM_GUI_BROWSE_TITLE);\r
 364   ofn.nMaxFile = MAX_PATH;\r
 365   ofn.Flags = OFN_EXPLORER | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST | flags;\r
 366 \r
 367   if (GetOpenFileName(&ofn)) {\r
 368     /* Directory hack. */\r
 369     if (flags & OFN_NOVALIDATE) strip_basename(ofn.lpstrFile);\r
 370     SendMessage(window, WM_SETTEXT, 0, (LPARAM) ofn.lpstrFile);\r
 371   }\r
 372   if (ofn.lpstrFilter) HeapFree(GetProcessHeap(), 0, (void *) ofn.lpstrFilter);\r
 373 \r
 374   delete[] ofn.lpstrFile;\r
 375 }\r
 376 \r
 377 INT_PTR CALLBACK tab_dlg(HWND tab, UINT message, WPARAM w, LPARAM l) {\r
 378   switch (message) {\r
 379     case WM_INITDIALOG:\r
 380       return 1;\r
 381 \r
 382     /* Button was pressed or control was controlled. */\r
 383     case WM_COMMAND:\r
 384       HWND dlg;\r
 385       char buffer[MAX_PATH];\r
 386 \r
 387       switch (LOWORD(w)) {\r
 388         /* Browse for application. */\r
 389         case IDC_BROWSE:\r
 390           dlg = GetDlgItem(tab, IDC_PATH);\r
 391           GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));\r
 392           browse(dlg, buffer, OFN_FILEMUSTEXIST, NSSM_GUI_BROWSE_FILTER_APPLICATIONS, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 393           /* Fill in startup directory if it wasn't already specified. */\r
 394           GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));\r
 395           if (! buffer[0]) {\r
 396             GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));\r
 397             strip_basename(buffer);\r
 398             SetDlgItemText(tab, IDC_DIR, buffer);\r
 399           }\r
 400           break;\r
 401 \r
 402           /* Browse for startup directory. */\r
 403         case IDC_BROWSE_DIR:\r
 404           dlg = GetDlgItem(tab, IDC_DIR);\r
 405           GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));\r
 406           browse(dlg, buffer, OFN_NOVALIDATE, NSSM_GUI_BROWSE_FILTER_DIRECTORIES, 0);\r
 407           break;\r
 408 \r
 409         /* Browse for stdin. */\r
 410         case IDC_BROWSE_STDIN:\r
 411           dlg = GetDlgItem(tab, IDC_STDIN);\r
 412           GetDlgItemText(tab, IDC_STDIN, buffer, sizeof(buffer));\r
 413           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 414           break;\r
 415 \r
 416         /* Browse for stdout. */\r
 417         case IDC_BROWSE_STDOUT:\r
 418           dlg = GetDlgItem(tab, IDC_STDOUT);\r
 419           GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));\r
 420           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 421           /* Fill in stderr if it wasn't already specified. */\r
 422           GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));\r
 423           if (! buffer[0]) {\r
 424             GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));\r
 425             SetDlgItemText(tab, IDC_STDERR, buffer);\r
 426           }\r
 427           break;\r
 428 \r
 429         /* Browse for stderr. */\r
 430         case IDC_BROWSE_STDERR:\r
 431           dlg = GetDlgItem(tab, IDC_STDERR);\r
 432           GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));\r
 433           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 434           break;\r
 435       }\r
 436       return 1;\r
 437   }\r
 438 \r
 439   return 0;\r
 440 }\r
 441 \r
 442 /* Install/remove dialogue callback */\r
 443 INT_PTR CALLBACK install_dlg(HWND window, UINT message, WPARAM w, LPARAM l) {\r
 444   switch (message) {\r
 445     /* Creating the dialogue */\r
 446     case WM_INITDIALOG:\r
 447       HWND tabs;\r
 448       HWND combo;\r
 449       tabs = GetDlgItem(window, IDC_TAB1);\r
 450       if (! tabs) return 0;\r
 451 \r
 452       /* Set up tabs. */\r
 453       TCITEM tab;\r
 454       ZeroMemory(&tab, sizeof(tab));\r
 455       tab.mask = TCIF_TEXT;\r
 456 \r
 457       /* Application tab. */\r
 458       tab.pszText = message_string(NSSM_GUI_TAB_APPLICATION);\r
 459       tab.cchTextMax = (int) strlen(tab.pszText);\r
 460       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_APPLICATION, (LPARAM) &tab);\r
 461       tablist[NSSM_TAB_APPLICATION] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPLICATION), window, tab_dlg);\r
 462       ShowWindow(tablist[NSSM_TAB_APPLICATION], SW_SHOW);\r
 463 \r
 464       /* Shutdown tab. */\r
 465       tab.pszText = message_string(NSSM_GUI_TAB_SHUTDOWN);\r
 466       tab.cchTextMax = (int) strlen(tab.pszText);\r
 467       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_SHUTDOWN, (LPARAM) &tab);\r
 468       tablist[NSSM_TAB_SHUTDOWN] = CreateDialog(0, MAKEINTRESOURCE(IDD_SHUTDOWN), window, tab_dlg);\r
 469       ShowWindow(tablist[NSSM_TAB_SHUTDOWN], SW_HIDE);\r
 470 \r
 471       /* Set defaults. */\r
 472       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_CONSOLE, BM_SETCHECK, BST_CHECKED, 0);\r
 473       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, NSSM_KILL_CONSOLE_GRACE_PERIOD, 0);\r
 474       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_WINDOW, BM_SETCHECK, BST_CHECKED, 0);\r
 475       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, NSSM_KILL_WINDOW_GRACE_PERIOD, 0);\r
 476       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_THREADS, BM_SETCHECK, BST_CHECKED, 0);\r
 477       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, NSSM_KILL_THREADS_GRACE_PERIOD, 0);\r
 478       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_TERMINATE, BM_SETCHECK, BST_CHECKED, 0);\r
 479 \r
 480       /* Restart tab. */\r
 481       tab.pszText = message_string(NSSM_GUI_TAB_EXIT);\r
 482       tab.cchTextMax = (int) strlen(tab.pszText);\r
 483       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_EXIT, (LPARAM) &tab);\r
 484       tablist[NSSM_TAB_EXIT] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPEXIT), window, tab_dlg);\r
 485       ShowWindow(tablist[NSSM_TAB_EXIT], SW_HIDE);\r
 486 \r
 487       /* Set defaults. */\r
 488       SetDlgItemInt(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, NSSM_RESET_THROTTLE_RESTART, 0);\r
 489       combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);\r
 490       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_RESTART, (LPARAM) message_string(NSSM_GUI_EXIT_RESTART));\r
 491       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_IGNORE, (LPARAM) message_string(NSSM_GUI_EXIT_IGNORE));\r
 492       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_REALLY, (LPARAM) message_string(NSSM_GUI_EXIT_REALLY));\r
 493       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_UNCLEAN, (LPARAM) message_string(NSSM_GUI_EXIT_UNCLEAN));\r
 494       SendMessage(combo, CB_SETCURSEL, NSSM_EXIT_RESTART, 0);\r
 495 \r
 496       /* I/O tab. */\r
 497       tab.pszText = message_string(NSSM_GUI_TAB_IO);\r
 498       tab.cchTextMax = (int) strlen(tab.pszText) + 1;\r
 499       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_IO, (LPARAM) &tab);\r
 500       tablist[NSSM_TAB_IO] = CreateDialog(0, MAKEINTRESOURCE(IDD_IO), window, tab_dlg);\r
 501       ShowWindow(tablist[NSSM_TAB_IO], SW_HIDE);\r
 502 \r
 503       /* Environment tab. */\r
 504       tab.pszText = message_string(NSSM_GUI_TAB_ENVIRONMENT);\r
 505       tab.cchTextMax = (int) strlen(tab.pszText) + 1;\r
 506       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_ENVIRONMENT, (LPARAM) &tab);\r
 507       tablist[NSSM_TAB_ENVIRONMENT] = CreateDialog(0, MAKEINTRESOURCE(IDD_ENVIRONMENT), window, tab_dlg);\r
 508       ShowWindow(tablist[NSSM_TAB_ENVIRONMENT], SW_HIDE);\r
 509 \r
 510       selected_tab = 0;\r
 511 \r
 512       return 1;\r
 513 \r
 514     /* Tab change. */\r
 515     case WM_NOTIFY:\r
 516       NMHDR *notification;\r
 517 \r
 518       notification = (NMHDR *) l;\r
 519       switch (notification->code) {\r
 520         case TCN_SELCHANGE:\r
 521           HWND tabs;\r
 522           int selection;\r
 523 \r
 524           tabs = GetDlgItem(window, IDC_TAB1);\r
 525           if (! tabs) return 0;\r
 526 \r
 527           selection = (int) SendMessage(tabs, TCM_GETCURSEL, 0, 0);\r
 528           if (selection != selected_tab) {\r
 529             ShowWindow(tablist[selected_tab], SW_HIDE);\r
 530             /*\r
 531               XXX: Sets focus to the service name which isn't ideal but is\r
 532                    better than leaving it in another tab.\r
 533             */\r
 534             ShowWindow(tablist[selection], SW_SHOWDEFAULT);\r
 535             SetFocus(tablist[selection]);\r
 536             selected_tab = selection;\r
 537           }\r
 538           return 1;\r
 539       }\r
 540 \r
 541       return 0;\r
 542 \r
 543     /* Button was pressed or control was controlled */\r
 544     case WM_COMMAND:\r
 545       switch (LOWORD(w)) {\r
 546         /* OK button */\r
 547         case IDOK:\r
 548           if (! install(window)) PostQuitMessage(0);\r
 549           break;\r
 550 \r
 551         /* Cancel button */\r
 552         case IDCANCEL:\r
 553           DestroyWindow(window);\r
 554           break;\r
 555 \r
 556         /* Remove button */\r
 557         case IDC_REMOVE:\r
 558           if (! remove(window)) PostQuitMessage(0);\r
 559           break;\r
 560       }\r
 561       return 1;\r
 562 \r
 563     /* Window closing */\r
 564     case WM_CLOSE:\r
 565       DestroyWindow(window);\r
 566       return 0;\r
 567     case WM_DESTROY:\r
 568       PostQuitMessage(0);\r
 569   }\r
 570   return 0;\r
 571 }\r