gui.cpp 30 KB

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