event.cpp

   1 #include "nssm.h"\r
   2 \r
   3 #define NSSM_ERROR_BUFSIZE 65535\r
   4 #define NSSM_NUM_EVENT_STRINGS 16\r
   5 unsigned long tls_index;\r
   6 \r
   7 /* Convert error code to error string - must call LocalFree() on return value */\r
   8 TCHAR *error_string(unsigned long error) {\r
   9   /* Thread-safe buffer */\r
  10   TCHAR *error_message = (TCHAR *) TlsGetValue(tls_index);\r
  11   if (! error_message) {\r
  12     error_message = (TCHAR *) LocalAlloc(LPTR, NSSM_ERROR_BUFSIZE);\r
  13     if (! error_message) return _T("<out of memory for error message>");\r
  14     TlsSetValue(tls_index, (void *) error_message);\r
  15   }\r
  16 \r
  17   if (! FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, GetUserDefaultLangID(), (TCHAR *) error_message, NSSM_ERROR_BUFSIZE, 0)) {\r
  18     if (! FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, 0, (TCHAR *) error_message, NSSM_ERROR_BUFSIZE, 0)) {\r
  19       if (_sntprintf_s(error_message, NSSM_ERROR_BUFSIZE, _TRUNCATE, _T("system error %lu"), error) < 0) return 0;\r
  20     }\r
  21   }\r
  22   return error_message;\r
  23 }\r
  24 \r
  25 /* Convert message code to format string */\r
  26 TCHAR *message_string(unsigned long error) {\r
  27   TCHAR *ret;\r
  28   if (! FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, GetUserDefaultLangID(), (LPTSTR) &ret, NSSM_ERROR_BUFSIZE, 0)) {\r
  29     if (! FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, 0, (LPTSTR) &ret, NSSM_ERROR_BUFSIZE, 0)) {\r
  30       ret = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, 32 * sizeof(TCHAR));\r
  31       if (_sntprintf_s(ret, NSSM_ERROR_BUFSIZE, _TRUNCATE, _T("system error %lu"), error) < 0) return 0;\r
  32     }\r
  33   }\r
  34   return ret;\r
  35 }\r
  36 \r
  37 /* Log a message to the Event Log */\r
  38 void log_event(unsigned short type, unsigned long id, ...) {\r
  39   va_list arg;\r
  40   int count;\r
  41   TCHAR *s;\r
  42   TCHAR *strings[NSSM_NUM_EVENT_STRINGS];\r
  43 \r
  44   /* Open event log */\r
  45   HANDLE handle = RegisterEventSource(0, NSSM);\r
  46   if (! handle) return;\r
  47 \r
  48   /* Log it */\r
  49   count = 0;\r
  50   va_start(arg, id);\r
  51   while ((s = va_arg(arg, TCHAR *)) && count < NSSM_NUM_EVENT_STRINGS - 1) strings[count++] = s;\r
  52   strings[count] = 0;\r
  53   va_end(arg);\r
  54   ReportEvent(handle, type, 0, id, 0, count, 0, (const TCHAR **) strings, 0);\r
  55 \r
  56   /* Close event log */\r
  57   DeregisterEventSource(handle);\r
  58 }\r
  59 \r
  60 /* Log a message to the console */\r
  61 void print_message(FILE *file, unsigned long id, ...) {\r
  62   va_list arg;\r
  63 \r
  64   TCHAR *format = message_string(id);\r
  65   if (! format) return;\r
  66 \r
  67   va_start(arg, id);\r
  68   _vftprintf(file, format, arg);\r
  69   va_end(arg);\r
  70 \r
  71   LocalFree(format);\r
  72 }\r
  73 \r
  74 /* Show a GUI dialogue */\r
  75 int popup_message(unsigned int type, unsigned long id, ...) {\r
  76   va_list arg;\r
  77 \r
  78   TCHAR *format = message_string(id);\r
  79   if (! format) {\r
  80     return MessageBox(0, _T("The message which was supposed to go here is missing!"), NSSM, MB_OK | MB_ICONEXCLAMATION);\r
  81   }\r
  82 \r
  83   TCHAR blurb[1024];\r
  84   va_start(arg, id);\r
  85   if (_vsntprintf_s(blurb, _countof(blurb), _TRUNCATE, format, arg) < 0) {\r
  86     va_end(arg);\r
  87     LocalFree(format);\r
  88     return MessageBox(0, _T("The message which was supposed to go here is too big!"), NSSM, MB_OK | MB_ICONEXCLAMATION);\r
  89   }\r
  90   va_end(arg);\r
  91 \r
  92   int ret = MessageBox(0, blurb, NSSM, type);\r
  93 \r
  94   LocalFree(format);\r
  95 \r
  96   return ret;\r
  97 }\r