gui.cpp 20 KB

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