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 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_method_timeout(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 check_io(TCHAR *name, TCHAR *buffer, size_t bufsize, unsigned long control) {\r
  78   if (! SendMessage(GetDlgItem(tablist[NSSM_TAB_IO], control), WM_GETTEXTLENGTH, 0, 0)) return;\r
  79   if (GetDlgItemText(tablist[NSSM_TAB_IO], control, buffer, (int) bufsize)) return;\r
  80   popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, name);\r
  81   ZeroMemory(buffer, bufsize);\r
  82 }\r
  83 \r
  84 /* Install the service. */\r
  85 int install(HWND window) {\r
  86   if (! window) return 1;\r
  87 \r
  88   nssm_service_t *service = alloc_nssm_service();\r
  89   if (service) {\r
  90     set_nssm_service_defaults(service);\r
  91 \r
  92     /* Get service name. */\r
  93     if (! GetDlgItemText(window, IDC_NAME, service->name, sizeof(service->name))) {\r
  94       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_SERVICE_NAME);\r
  95       cleanup_nssm_service(service);\r
  96       return 2;\r
  97     }\r
  98 \r
  99     /* Get executable name */\r
 100     if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_PATH, service->exe, sizeof(service->exe))) {\r
 101       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_PATH);\r
 102       return 3;\r
 103     }\r
 104 \r
 105     /* Get startup directory. */\r
 106     if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_DIR, service->dir, sizeof(service->dir))) {\r
 107       _sntprintf_s(service->dir, _countof(service->dir), _TRUNCATE, _T("%s"), service->exe);\r
 108       strip_basename(service->dir);\r
 109     }\r
 110 \r
 111     /* Get flags. */\r
 112     if (SendMessage(GetDlgItem(tablist[NSSM_TAB_APPLICATION], IDC_FLAGS), WM_GETTEXTLENGTH, 0, 0)) {\r
 113       if (! GetDlgItemText(tablist[NSSM_TAB_APPLICATION], IDC_FLAGS, service->flags, sizeof(service->flags))) {\r
 114         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_OPTIONS);\r
 115         return 4;\r
 116       }\r
 117     }\r
 118 \r
 119     /* Get stop method stuff. */\r
 120     check_stop_method(service, NSSM_STOP_METHOD_CONSOLE, IDC_METHOD_CONSOLE);\r
 121     check_stop_method(service, NSSM_STOP_METHOD_WINDOW, IDC_METHOD_WINDOW);\r
 122     check_stop_method(service, NSSM_STOP_METHOD_THREADS, IDC_METHOD_THREADS);\r
 123     check_stop_method(service, NSSM_STOP_METHOD_TERMINATE, IDC_METHOD_TERMINATE);\r
 124     check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, &service->kill_console_delay);\r
 125     check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, &service->kill_window_delay);\r
 126     check_method_timeout(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, &service->kill_threads_delay);\r
 127 \r
 128     /* Get exit action stuff. */\r
 129     check_method_timeout(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, &service->throttle_delay);\r
 130     HWND combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);\r
 131     service->default_exit_action = (unsigned long) SendMessage(combo, CB_GETCURSEL, 0, 0);\r
 132     if (service->default_exit_action == CB_ERR) service->default_exit_action = 0;\r
 133 \r
 134     /* Get I/O stuff. */\r
 135     check_io(_T("stdin"), service->stdin_path, sizeof(service->stdin_path), IDC_STDIN);\r
 136     check_io(_T("stdout"), service->stdout_path, sizeof(service->stdout_path), IDC_STDOUT);\r
 137     check_io(_T("stderr"), service->stderr_path, sizeof(service->stderr_path), IDC_STDERR);\r
 138     /* Override stdout and/or stderr. */\r
 139     if (SendDlgItemMessage(tablist[NSSM_TAB_IO], IDC_TRUNCATE, BM_GETCHECK, 0, 0) & BST_CHECKED) {\r
 140       if (service->stdout_path[0]) service->stdout_disposition = CREATE_ALWAYS;\r
 141       if (service->stderr_path[0]) service->stderr_disposition = CREATE_ALWAYS;\r
 142     }\r
 143 \r
 144     /* Get environment. */\r
 145     unsigned long envlen = (unsigned long) SendMessage(GetDlgItem(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT), WM_GETTEXTLENGTH, 0, 0);\r
 146     if (envlen) {\r
 147       TCHAR *env = (TCHAR *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, (envlen + 2) * sizeof(TCHAR));\r
 148       if (! env) {\r
 149         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("environment"), _T("install()"));\r
 150         cleanup_nssm_service(service);\r
 151         return 5;\r
 152       }\r
 153 \r
 154       if (! GetDlgItemText(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT, env, envlen + 1)) {\r
 155         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_ENVIRONMENT);\r
 156         HeapFree(GetProcessHeap(), 0, env);\r
 157         cleanup_nssm_service(service);\r
 158         return 5;\r
 159       }\r
 160 \r
 161       /* Strip CR and replace LF with NULL. */\r
 162       unsigned long newlen = 0;\r
 163       unsigned long i, j;\r
 164       for (i = 0; i < envlen; i++) if (env[i] != _T('\r')) newlen++;\r
 165       /* Must end with two NULLs. */\r
 166       newlen += 2;\r
 167 \r
 168       TCHAR *newenv = (TCHAR *) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, newlen * sizeof(TCHAR));\r
 169       if (! newenv) {\r
 170         HeapFree(GetProcessHeap(), 0, env);\r
 171         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("environment"), _T("install()"));\r
 172         cleanup_nssm_service(service);\r
 173         return 5;\r
 174       }\r
 175 \r
 176       for (i = 0, j = 0; i < envlen; i++) {\r
 177         if (env[i] == _T('\r')) continue;\r
 178         if (env[i] == _T('\n')) newenv[j] = _T('\0');\r
 179         else newenv[j] = env[i];\r
 180         j++;\r
 181       }\r
 182 \r
 183       HeapFree(GetProcessHeap(), 0, env);\r
 184       env = newenv;\r
 185       envlen = newlen;\r
 186 \r
 187       /* Test the environment is valid. */\r
 188       if (test_environment(env)) {\r
 189         popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INVALID_ENVIRONMENT);\r
 190         HeapFree(GetProcessHeap(), 0, env);\r
 191         cleanup_nssm_service(service);\r
 192         return 5;\r
 193       }\r
 194 \r
 195       if (SendDlgItemMessage(tablist[NSSM_TAB_ENVIRONMENT], IDC_ENVIRONMENT_REPLACE, BM_GETCHECK, 0, 0) & BST_CHECKED) {\r
 196         service->env = env;\r
 197         service->envlen = envlen;\r
 198       }\r
 199       else {\r
 200         service->env_extra = env;\r
 201         service->env_extralen = envlen;\r
 202       }\r
 203     }\r
 204   }\r
 205 \r
 206   /* See if it works. */\r
 207   switch (install_service(service)) {\r
 208     case 1:\r
 209       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("service"), _T("install()"));\r
 210       cleanup_nssm_service(service);\r
 211       return 1;\r
 212 \r
 213     case 2:\r
 214       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);\r
 215       cleanup_nssm_service(service);\r
 216       return 2;\r
 217 \r
 218     case 3:\r
 219       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_PATH_TOO_LONG, NSSM);\r
 220       cleanup_nssm_service(service);\r
 221       return 3;\r
 222 \r
 223     case 4:\r
 224       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_OUT_OF_MEMORY_FOR_IMAGEPATH);\r
 225       cleanup_nssm_service(service);\r
 226       return 4;\r
 227 \r
 228     case 5:\r
 229       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_INSTALL_SERVICE_FAILED);\r
 230       cleanup_nssm_service(service);\r
 231       return 5;\r
 232 \r
 233     case 6:\r
 234       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_CREATE_PARAMETERS_FAILED);\r
 235       cleanup_nssm_service(service);\r
 236       return 6;\r
 237   }\r
 238 \r
 239   popup_message(MB_OK, NSSM_MESSAGE_SERVICE_INSTALLED, service->name);\r
 240   cleanup_nssm_service(service);\r
 241   return 0;\r
 242 }\r
 243 \r
 244 /* Remove the service */\r
 245 int remove(HWND window) {\r
 246   if (! window) return 1;\r
 247 \r
 248   /* See if it works */\r
 249   nssm_service_t *service = alloc_nssm_service();\r
 250   if (service) {\r
 251     /* Get service name */\r
 252     if (! GetDlgItemText(window, IDC_NAME, service->name, sizeof(service->name))) {\r
 253       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_MISSING_SERVICE_NAME);\r
 254       cleanup_nssm_service(service);\r
 255       return 2;\r
 256     }\r
 257 \r
 258     /* Confirm */\r
 259     if (popup_message(MB_YESNO, NSSM_GUI_ASK_REMOVE_SERVICE, service->name) != IDYES) {\r
 260       cleanup_nssm_service(service);\r
 261       return 0;\r
 262     }\r
 263   }\r
 264 \r
 265   switch (remove_service(service)) {\r
 266     case 1:\r
 267       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_EVENT_OUT_OF_MEMORY, _T("service"), _T("remove()"));\r
 268       cleanup_nssm_service(service);\r
 269       return 1;\r
 270 \r
 271     case 2:\r
 272       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_MESSAGE_OPEN_SERVICE_MANAGER_FAILED);\r
 273       cleanup_nssm_service(service);\r
 274       return 2;\r
 275 \r
 276     case 3:\r
 277       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_SERVICE_NOT_INSTALLED);\r
 278       return 3;\r
 279       cleanup_nssm_service(service);\r
 280 \r
 281     case 4:\r
 282       popup_message(MB_OK | MB_ICONEXCLAMATION, NSSM_GUI_REMOVE_SERVICE_FAILED);\r
 283       cleanup_nssm_service(service);\r
 284       return 4;\r
 285   }\r
 286 \r
 287   popup_message(MB_OK, NSSM_MESSAGE_SERVICE_REMOVED, service->name);\r
 288   cleanup_nssm_service(service);\r
 289   return 0;\r
 290 }\r
 291 \r
 292 static TCHAR *browse_filter(int message) {\r
 293   switch (message) {\r
 294     case NSSM_GUI_BROWSE_FILTER_APPLICATIONS: return _T("*.exe;*.bat;*.cmd");\r
 295     case NSSM_GUI_BROWSE_FILTER_DIRECTORIES: return _T(".");\r
 296     case NSSM_GUI_BROWSE_FILTER_ALL_FILES: /* Fall through. */\r
 297     default: return _T("*.*");\r
 298   }\r
 299 }\r
 300 \r
 301 UINT_PTR CALLBACK browse_hook(HWND dlg, UINT message, WPARAM w, LPARAM l) {\r
 302   switch (message) {\r
 303     case WM_INITDIALOG:\r
 304       return 1;\r
 305   }\r
 306 \r
 307   return 0;\r
 308 }\r
 309 \r
 310 /* Browse for application */\r
 311 void browse(HWND window, TCHAR *current, unsigned long flags, ...) {\r
 312   if (! window) return;\r
 313 \r
 314   va_list arg;\r
 315   size_t bufsize = 256;\r
 316   size_t len = bufsize;\r
 317   int i;\r
 318 \r
 319   OPENFILENAME ofn;\r
 320   ZeroMemory(&ofn, sizeof(ofn));\r
 321   ofn.lStructSize = sizeof(ofn);\r
 322   ofn.lpstrFilter = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, bufsize * sizeof(TCHAR));\r
 323   /* XXX: Escaping nulls with FormatMessage is tricky */\r
 324   if (ofn.lpstrFilter) {\r
 325     ZeroMemory((void *) ofn.lpstrFilter, bufsize);\r
 326     len = 0;\r
 327     /* "Applications" + NULL + "*.exe" + NULL */\r
 328     va_start(arg, flags);\r
 329     while (i = va_arg(arg, int)) {\r
 330       TCHAR *localised = message_string(i);\r
 331       _sntprintf_s((TCHAR *) ofn.lpstrFilter + len, bufsize, _TRUNCATE, localised);\r
 332       len += _tcslen(localised) + 1;\r
 333       LocalFree(localised);\r
 334       TCHAR *filter = browse_filter(i);\r
 335       _sntprintf_s((TCHAR *) ofn.lpstrFilter + len, bufsize - len, _TRUNCATE, _T("%s"), filter);\r
 336       len += _tcslen(filter) + 1;\r
 337     }\r
 338     va_end(arg);\r
 339     /* Remainder of the buffer is already zeroed */\r
 340   }\r
 341   ofn.lpstrFile = new TCHAR[MAX_PATH];\r
 342   if (flags & OFN_NOVALIDATE) {\r
 343     /* Directory hack. */\r
 344     _sntprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, _T(":%s:"), message_string(NSSM_GUI_BROWSE_FILTER_DIRECTORIES));\r
 345   }\r
 346   else _sntprintf_s(ofn.lpstrFile, MAX_PATH, _TRUNCATE, _T("%s"), current);\r
 347   ofn.lpstrTitle = message_string(NSSM_GUI_BROWSE_TITLE);\r
 348   ofn.nMaxFile = MAX_PATH;\r
 349   ofn.Flags = OFN_EXPLORER | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST | flags;\r
 350 \r
 351   if (GetOpenFileName(&ofn)) {\r
 352     /* Directory hack. */\r
 353     if (flags & OFN_NOVALIDATE) strip_basename(ofn.lpstrFile);\r
 354     SendMessage(window, WM_SETTEXT, 0, (LPARAM) ofn.lpstrFile);\r
 355   }\r
 356   if (ofn.lpstrFilter) HeapFree(GetProcessHeap(), 0, (void *) ofn.lpstrFilter);\r
 357 \r
 358   delete[] ofn.lpstrFile;\r
 359 }\r
 360 \r
 361 INT_PTR CALLBACK tab_dlg(HWND tab, UINT message, WPARAM w, LPARAM l) {\r
 362   switch (message) {\r
 363     case WM_INITDIALOG:\r
 364       return 1;\r
 365 \r
 366     /* Button was pressed or control was controlled. */\r
 367     case WM_COMMAND:\r
 368       HWND dlg;\r
 369       TCHAR buffer[MAX_PATH];\r
 370 \r
 371       switch (LOWORD(w)) {\r
 372         /* Browse for application. */\r
 373         case IDC_BROWSE:\r
 374           dlg = GetDlgItem(tab, IDC_PATH);\r
 375           GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));\r
 376           browse(dlg, buffer, OFN_FILEMUSTEXIST, NSSM_GUI_BROWSE_FILTER_APPLICATIONS, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 377           /* Fill in startup directory if it wasn't already specified. */\r
 378           GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));\r
 379           if (! buffer[0]) {\r
 380             GetDlgItemText(tab, IDC_PATH, buffer, sizeof(buffer));\r
 381             strip_basename(buffer);\r
 382             SetDlgItemText(tab, IDC_DIR, buffer);\r
 383           }\r
 384           break;\r
 385 \r
 386           /* Browse for startup directory. */\r
 387         case IDC_BROWSE_DIR:\r
 388           dlg = GetDlgItem(tab, IDC_DIR);\r
 389           GetDlgItemText(tab, IDC_DIR, buffer, sizeof(buffer));\r
 390           browse(dlg, buffer, OFN_NOVALIDATE, NSSM_GUI_BROWSE_FILTER_DIRECTORIES, 0);\r
 391           break;\r
 392 \r
 393         /* Browse for stdin. */\r
 394         case IDC_BROWSE_STDIN:\r
 395           dlg = GetDlgItem(tab, IDC_STDIN);\r
 396           GetDlgItemText(tab, IDC_STDIN, buffer, sizeof(buffer));\r
 397           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 398           break;\r
 399 \r
 400         /* Browse for stdout. */\r
 401         case IDC_BROWSE_STDOUT:\r
 402           dlg = GetDlgItem(tab, IDC_STDOUT);\r
 403           GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));\r
 404           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 405           /* Fill in stderr if it wasn't already specified. */\r
 406           GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));\r
 407           if (! buffer[0]) {\r
 408             GetDlgItemText(tab, IDC_STDOUT, buffer, sizeof(buffer));\r
 409             SetDlgItemText(tab, IDC_STDERR, buffer);\r
 410           }\r
 411           break;\r
 412 \r
 413         /* Browse for stderr. */\r
 414         case IDC_BROWSE_STDERR:\r
 415           dlg = GetDlgItem(tab, IDC_STDERR);\r
 416           GetDlgItemText(tab, IDC_STDERR, buffer, sizeof(buffer));\r
 417           browse(dlg, buffer, 0, NSSM_GUI_BROWSE_FILTER_ALL_FILES, 0);\r
 418           break;\r
 419       }\r
 420       return 1;\r
 421   }\r
 422 \r
 423   return 0;\r
 424 }\r
 425 \r
 426 /* Install/remove dialogue callback */\r
 427 INT_PTR CALLBACK install_dlg(HWND window, UINT message, WPARAM w, LPARAM l) {\r
 428   switch (message) {\r
 429     /* Creating the dialogue */\r
 430     case WM_INITDIALOG:\r
 431       SetFocus(GetDlgItem(window, IDC_NAME));\r
 432 \r
 433       HWND tabs;\r
 434       HWND combo;\r
 435       tabs = GetDlgItem(window, IDC_TAB1);\r
 436       if (! tabs) return 0;\r
 437 \r
 438       /* Set up tabs. */\r
 439       TCITEM tab;\r
 440       ZeroMemory(&tab, sizeof(tab));\r
 441       tab.mask = TCIF_TEXT;\r
 442 \r
 443       /* Application tab. */\r
 444       tab.pszText = message_string(NSSM_GUI_TAB_APPLICATION);\r
 445       tab.cchTextMax = (int) _tcslen(tab.pszText);\r
 446       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_APPLICATION, (LPARAM) &tab);\r
 447       tablist[NSSM_TAB_APPLICATION] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPLICATION), window, tab_dlg);\r
 448       ShowWindow(tablist[NSSM_TAB_APPLICATION], SW_SHOW);\r
 449 \r
 450       /* Shutdown tab. */\r
 451       tab.pszText = message_string(NSSM_GUI_TAB_SHUTDOWN);\r
 452       tab.cchTextMax = (int) _tcslen(tab.pszText);\r
 453       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_SHUTDOWN, (LPARAM) &tab);\r
 454       tablist[NSSM_TAB_SHUTDOWN] = CreateDialog(0, MAKEINTRESOURCE(IDD_SHUTDOWN), window, tab_dlg);\r
 455       ShowWindow(tablist[NSSM_TAB_SHUTDOWN], SW_HIDE);\r
 456 \r
 457       /* Set defaults. */\r
 458       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_CONSOLE, BM_SETCHECK, BST_CHECKED, 0);\r
 459       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_CONSOLE, NSSM_KILL_CONSOLE_GRACE_PERIOD, 0);\r
 460       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_WINDOW, BM_SETCHECK, BST_CHECKED, 0);\r
 461       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_WINDOW, NSSM_KILL_WINDOW_GRACE_PERIOD, 0);\r
 462       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_THREADS, BM_SETCHECK, BST_CHECKED, 0);\r
 463       SetDlgItemInt(tablist[NSSM_TAB_SHUTDOWN], IDC_KILL_THREADS, NSSM_KILL_THREADS_GRACE_PERIOD, 0);\r
 464       SendDlgItemMessage(tablist[NSSM_TAB_SHUTDOWN], IDC_METHOD_TERMINATE, BM_SETCHECK, BST_CHECKED, 0);\r
 465 \r
 466       /* Restart tab. */\r
 467       tab.pszText = message_string(NSSM_GUI_TAB_EXIT);\r
 468       tab.cchTextMax = (int) _tcslen(tab.pszText);\r
 469       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_EXIT, (LPARAM) &tab);\r
 470       tablist[NSSM_TAB_EXIT] = CreateDialog(0, MAKEINTRESOURCE(IDD_APPEXIT), window, tab_dlg);\r
 471       ShowWindow(tablist[NSSM_TAB_EXIT], SW_HIDE);\r
 472 \r
 473       /* Set defaults. */\r
 474       SetDlgItemInt(tablist[NSSM_TAB_EXIT], IDC_THROTTLE, NSSM_RESET_THROTTLE_RESTART, 0);\r
 475       combo = GetDlgItem(tablist[NSSM_TAB_EXIT], IDC_APPEXIT);\r
 476       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_RESTART, (LPARAM) message_string(NSSM_GUI_EXIT_RESTART));\r
 477       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_IGNORE, (LPARAM) message_string(NSSM_GUI_EXIT_IGNORE));\r
 478       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_REALLY, (LPARAM) message_string(NSSM_GUI_EXIT_REALLY));\r
 479       SendMessage(combo, CB_INSERTSTRING, NSSM_EXIT_UNCLEAN, (LPARAM) message_string(NSSM_GUI_EXIT_UNCLEAN));\r
 480       SendMessage(combo, CB_SETCURSEL, NSSM_EXIT_RESTART, 0);\r
 481 \r
 482       /* I/O tab. */\r
 483       tab.pszText = message_string(NSSM_GUI_TAB_IO);\r
 484       tab.cchTextMax = (int) _tcslen(tab.pszText) + 1;\r
 485       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_IO, (LPARAM) &tab);\r
 486       tablist[NSSM_TAB_IO] = CreateDialog(0, MAKEINTRESOURCE(IDD_IO), window, tab_dlg);\r
 487       ShowWindow(tablist[NSSM_TAB_IO], SW_HIDE);\r
 488 \r
 489       /* Environment tab. */\r
 490       tab.pszText = message_string(NSSM_GUI_TAB_ENVIRONMENT);\r
 491       tab.cchTextMax = (int) _tcslen(tab.pszText) + 1;\r
 492       SendMessage(tabs, TCM_INSERTITEM, NSSM_TAB_ENVIRONMENT, (LPARAM) &tab);\r
 493       tablist[NSSM_TAB_ENVIRONMENT] = CreateDialog(0, MAKEINTRESOURCE(IDD_ENVIRONMENT), window, tab_dlg);\r
 494       ShowWindow(tablist[NSSM_TAB_ENVIRONMENT], SW_HIDE);\r
 495 \r
 496       selected_tab = 0;\r
 497 \r
 498       return 1;\r
 499 \r
 500     /* Tab change. */\r
 501     case WM_NOTIFY:\r
 502       NMHDR *notification;\r
 503 \r
 504       notification = (NMHDR *) l;\r
 505       switch (notification->code) {\r
 506         case TCN_SELCHANGE:\r
 507           HWND tabs;\r
 508           int selection;\r
 509 \r
 510           tabs = GetDlgItem(window, IDC_TAB1);\r
 511           if (! tabs) return 0;\r
 512 \r
 513           selection = (int) SendMessage(tabs, TCM_GETCURSEL, 0, 0);\r
 514           if (selection != selected_tab) {\r
 515             ShowWindow(tablist[selected_tab], SW_HIDE);\r
 516             /*\r
 517               XXX: Sets focus to the service name which isn't ideal but is\r
 518                    better than leaving it in another tab.\r
 519             */\r
 520             ShowWindow(tablist[selection], SW_SHOWDEFAULT);\r
 521             SetFocus(tablist[selection]);\r
 522             selected_tab = selection;\r
 523           }\r
 524           return 1;\r
 525       }\r
 526 \r
 527       return 0;\r
 528 \r
 529     /* Button was pressed or control was controlled */\r
 530     case WM_COMMAND:\r
 531       switch (LOWORD(w)) {\r
 532         /* OK button */\r
 533         case IDOK:\r
 534           if (! install(window)) PostQuitMessage(0);\r
 535           break;\r
 536 \r
 537         /* Cancel button */\r
 538         case IDCANCEL:\r
 539           DestroyWindow(window);\r
 540           break;\r
 541 \r
 542         /* Remove button */\r
 543         case IDC_REMOVE:\r
 544           if (! remove(window)) PostQuitMessage(0);\r
 545           break;\r
 546       }\r
 547       return 1;\r
 548 \r
 549     /* Window closing */\r
 550     case WM_CLOSE:\r
 551       DestroyWindow(window);\r
 552       return 0;\r
 553     case WM_DESTROY:\r
 554       PostQuitMessage(0);\r
 555   }\r
 556   return 0;\r
 557 }\r