event.cpp

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