console.cpp

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