console.cpp 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. #include "nssm.h"
  2. /* See if we were launched from a console window. */
  3. bool check_console() {
  4. /* If we're running in a service context there will be no console window. */
  5. HWND console = GetConsoleWindow();
  6. if (! console) return false;
  7. unsigned long pid;
  8. if (! GetWindowThreadProcessId(console, &pid)) return false;
  9. /*
  10. If the process associated with the console window handle is the same as
  11. this process, we were not launched from an existing console. The user
  12. probably double-clicked our executable.
  13. */
  14. if (GetCurrentProcessId() != pid) return true;
  15. /* We close our new console so that subsequent messages appear in a popup. */
  16. FreeConsole();
  17. return false;
  18. }
  19. /* Helpers for drawing the banner. */
  20. static inline void block(unsigned int a, short x, short y, unsigned long n) {
  21. HANDLE h = GetStdHandle(STD_OUTPUT_HANDLE);
  22. TCHAR s = _T(' ');
  23. unsigned long out;
  24. COORD c = { x, y };
  25. FillConsoleOutputAttribute(h, a, n, c, &out);
  26. FillConsoleOutputCharacter(h, s, n, c, &out);
  27. }
  28. static inline void R(short x, short y, unsigned long n) {
  29. block(BACKGROUND_RED | BACKGROUND_INTENSITY, x, y, n);
  30. }
  31. static inline void r(short x, short y, unsigned long n) {
  32. block(BACKGROUND_RED, x, y, n);
  33. }
  34. static inline void b(short x, short y, unsigned long n) {
  35. block(0, x, y, n);
  36. }
  37. void alloc_console(nssm_service_t *service) {
  38. if (service->no_console) return;
  39. AllocConsole();
  40. /* Disable accidental closure. */
  41. HWND window = GetConsoleWindow();
  42. HMENU menu = GetSystemMenu(window, false);
  43. EnableMenuItem(menu, SC_CLOSE, MF_GRAYED);
  44. /* Set a title like "[NSSM] Jenkins" */
  45. TCHAR displayname[SERVICE_NAME_LENGTH];
  46. unsigned long len = _countof(displayname);
  47. SC_HANDLE services = open_service_manager(SC_MANAGER_CONNECT);
  48. if (services) {
  49. if (! GetServiceDisplayName(services, service->name, displayname, &len)) ZeroMemory(displayname, sizeof(displayname));
  50. CloseServiceHandle(services);
  51. }
  52. if (! displayname[0]) _sntprintf_s(displayname, _countof(displayname), _TRUNCATE, _T("%s"), service->name);
  53. TCHAR title[65535];
  54. _sntprintf_s(title, _countof(title), _TRUNCATE, _T("[%s] %s"), NSSM, displayname);
  55. SetConsoleTitle(title);
  56. /* Draw the NSSM logo on the console window. */
  57. short y = 0;
  58. b(0, y, 80);
  59. y++;
  60. b(0, y, 80);
  61. y++;
  62. b(0, y, 80);
  63. y++;
  64. b(0, y, 80);
  65. y++;
  66. b(0, y, 80);
  67. r(18, y, 5); r(28, y, 4); r(41, y, 4); r(68, y, 1);
  68. R(6, y, 5); R(19, y, 4); R(29, y, 1); R(32, y, 3); R(42, y, 1); R(45, y, 3); R(52, y, 5); R(69, y, 4);
  69. y++;
  70. b(0, y, 80);
  71. r(8, y, 4); r(20, y, 1); r(28, y, 1); r(33, y, 3); r(41, y, 1); r(46, y, 3); r (57, y, 1);
  72. R(9, y, 2); R(21, y, 1); R(27, y, 1); R(34, y, 1); R(40, y, 1); R(47, y, 1); R(54, y, 3); R(68, y, 3);
  73. y++;
  74. b(0, y, 80);
  75. r(12, y, 1); r(20, y, 1); r(26, y, 1); r(34, y, 2); r(39, y, 1); r(47, y, 2); r(67, y, 2);
  76. R(9, y, 3); R(21, y, 1); R(27, y, 1); R(40, y, 1); R(54, y, 1); R(56, y, 2); R(67, y, 1); R(69, y, 2);
  77. y++;
  78. b(0, y, 80);
  79. r(9, y, 1); r(20, y, 1); r(26, y, 1); r (35, y, 1); r(39, y, 1); r(48, y, 1); r(58, y, 1);
  80. R(10, y, 3); R(21, y, 1); R(27, y, 1); R(40, y, 1); R(54, y, 1); R(56, y, 2); R(67, y, 1); R(69, y, 2);
  81. y++;
  82. b(0, y, 80);
  83. r(9, y, 1); r(56, y, 1); r(66, y, 2);
  84. R(11, y, 3); R(21, y, 1); R(26, y, 2); R(39, y, 2); R(54, y, 1); R(57, y, 2); R(69, y, 2);
  85. y++;
  86. b(0, y, 80);
  87. r(9, y, 1); r(26, y, 1); r(39, y, 1); r(59, y, 1);
  88. R(12, y, 3); R(21, y, 1); R(27, y, 2); R(40, y, 2); R(54, y, 1); R(57, y, 2); R(66, y, 1); R(69, y, 2);
  89. y++;
  90. b(0, y, 80);
  91. r(9, y, 1); r(12, y, 4); r(30, y, 1); r(43, y, 1); r(57, y, 1); r(65, y, 2);
  92. R(13, y, 2); R(21, y, 1); R(27, y, 3); R(40, y, 3); R(54, y, 1); R(58, y, 2); R(69, y, 2);
  93. y++;
  94. b(0, y, 80);
  95. r(9, y, 1); r(13, y, 4); r(27, y, 7); r(40, y, 7);
  96. R(14, y, 2); R(21, y, 1); R(28, y, 5); R(41, y, 5); R(54, y, 1); R(58, y, 2); R(65, y, 1); R(69, y, 2);
  97. y++;
  98. b(0, y, 80);
  99. r(9, y, 1); r(60, y, 1); r(65, y, 1);
  100. R(14, y, 3); R(21, y, 1); R(29, y, 6); R(42, y, 6); R(54, y, 1); R(58, y, 2); R(69, y, 2);
  101. y++;
  102. b(0, y, 80);
  103. r(9, y, 1); r(31, y, 1); r(44, y, 1); r(58, y, 1); r(64, y, 1);
  104. R(15, y, 3); R(21, y, 1); R(32, y, 4); R(45, y, 4); R(54, y, 1); R(59, y, 2); R(69, y, 2);
  105. y++;
  106. b(0, y, 80);
  107. r(9, y, 1); r(33, y, 1); r(46, y, 1); r(61, y, 1); r(64, y, 1);
  108. R(16, y, 3); R(21, y, 1); R(34, y, 2); R(47, y, 2); R(54, y, 1); R(59, y, 2); R(69, y, 2);
  109. y++;
  110. b(0, y, 80);
  111. r(9, y, 1); r(16, y, 4); r(36, y, 1); r(49, y, 1); r(59, y, 1); r(63, y, 1);
  112. R(17, y, 2); R(21, y, 1); R(34, y, 2); R(47, y, 2); R(54, y, 1); R(60, y, 2); R(69, y, 2);
  113. y++;
  114. b(0, y, 80);
  115. r(9, y, 1); r(17, y, 4); r(26, y, 1); r(36, y, 1); r(39, y, 1); r(49, y, 1);
  116. R(18, y, 2); R(21, y, 1); R(35, y, 1); R(48, y, 1); R(54, y, 1); R(60, y, 2); R(63, y, 1); R(69, y, 2);
  117. y++;
  118. b(0, y, 80);
  119. r(26, y, 2); r(39, y, 2); r(63, y, 1);
  120. R(9, y, 1); R(18, y, 4); R(35, y, 1); R(48, y, 1); R(54, y, 1); R(60, y, 3); R(69, y, 2);
  121. y++;
  122. b(0, y, 80);
  123. r(34, y, 1); r(47, y, 1); r(60, y, 1);
  124. R(9, y, 1); R(19, y, 3); R(26, y, 2); R(35, y, 1); R(39, y, 2); R(48, y, 1); R(54, y, 1); R(61, y, 2); R(69, y, 2);
  125. y++;
  126. b(0, y, 80);
  127. r(8, y, 1); r(35, y, 1); r(48, y, 1); r(62, y, 1); r(71, y, 1);
  128. R(9, y, 1); R(20, y, 2); R(26, y, 3); R(34, y, 1); R(39, y, 3); R(47, y, 1); R(54, y, 1); R(61, y, 1); R(69, y, 2);
  129. y++;
  130. b(0, y, 80);
  131. r(11, y, 1); r(26, y, 1); r(28, y, 5); r(39, y, 1); r(41, y, 5); r(51, y, 7); r(61, y, 1); r(66, y, 8);
  132. R(7, y, 4); R(21, y, 1); R(29, y, 1); R(33, y, 1); R(42, y, 1); R(46, y, 1); R(52, y, 5); R(67, y, 7);
  133. y++;
  134. b(0, y, 80);
  135. y++;
  136. b(0, y, 80);
  137. y++;
  138. }