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, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (TCHAR *) error_message, NSSM_ERROR_BUFSIZE, 0)) {\r
  18     if (_sntprintf_s(error_message, NSSM_ERROR_BUFSIZE, _TRUNCATE, _T("system error %lu"), error) < 0) return 0;\r
  19   }\r
  20   return error_message;\r
  21 }\r
  22 \r
  23 /* Convert message code to format string */\r
  24 TCHAR *message_string(unsigned long error) {\r
  25   TCHAR *ret;\r
  26   if (! FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) &ret, NSSM_ERROR_BUFSIZE, 0)) {\r
  27     ret = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, 32 * sizeof(TCHAR));\r
  28     if (_sntprintf_s(ret, NSSM_ERROR_BUFSIZE, _TRUNCATE, _T("system error %lu"), error) < 0) return 0;\r
  29   }\r
  30   return ret;\r
  31 }\r
  32 \r
  33 /* Log a message to the Event Log */\r
  34 void log_event(unsigned short type, unsigned long id, ...) {\r
  35   va_list arg;\r
  36   int count;\r
  37   TCHAR *s;\r
  38   TCHAR *strings[NSSM_NUM_EVENT_STRINGS];\r
  39 \r
  40   /* Open event log */\r
  41   HANDLE handle = RegisterEventSource(0, NSSM);\r
  42   if (! handle) return;\r
  43 \r
  44   /* Log it */\r
  45   count = 0;\r
  46   va_start(arg, id);\r
  47   while ((s = va_arg(arg, TCHAR *)) && count < NSSM_NUM_EVENT_STRINGS - 1) strings[count++] = s;\r
  48   strings[count] = 0;\r
  49   va_end(arg);\r
  50   ReportEvent(handle, type, 0, id, 0, count, 0, (const TCHAR **) strings, 0);\r
  51 \r
  52   /* Close event log */\r
  53   DeregisterEventSource(handle);\r
  54 }\r
  55 \r
  56 /* Log a message to the console */\r
  57 void print_message(FILE *file, unsigned long id, ...) {\r
  58   va_list arg;\r
  59 \r
  60   TCHAR *format = message_string(id);\r
  61   if (! format) return;\r
  62 \r
  63   va_start(arg, id);\r
  64   _vftprintf(file, format, arg);\r
  65   va_end(arg);\r
  66 \r
  67   LocalFree(format);\r
  68 }\r
  69 \r
  70 /* Show a GUI dialogue */\r
  71 int popup_message(unsigned int type, unsigned long id, ...) {\r
  72   va_list arg;\r
  73 \r
  74   TCHAR *format = message_string(id);\r
  75   if (! format) {\r
  76     return MessageBox(0, _T("Message %lu was supposed to go here!"), NSSM, MB_OK | MB_ICONEXCLAMATION);\r
  77   }\r
  78 \r
  79   TCHAR blurb[256];\r
  80   va_start(arg, id);\r
  81   if (_vsntprintf_s(blurb, sizeof(blurb), _TRUNCATE, format, arg) < 0) {\r
  82     va_end(arg);\r
  83     LocalFree(format);\r
  84     return MessageBox(0, _T("Message %lu was supposed to go here!"), NSSM, MB_OK | MB_ICONEXCLAMATION);\r
  85   }\r
  86   va_end(arg);\r
  87 \r
  88   int ret = MessageBox(0, blurb, NSSM, type);\r
  89 \r
  90   LocalFree(format);\r
  91 \r
  92   return ret;\r
  93 }\r