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