Compiler food.

[nssm.git] / event.cpp

diff --git a/event.cpp b/event.cpp
index bd43be3..995b7af 100644 (file)
--- a/event.cpp
+++ b/event.cpp
@@ -1,15 +1,34 @@
 #include "nssm.h"\r
 \r
-static char error_message[65535];\r
+#define NSSM_ERROR_BUFSIZE 65535\r
+unsigned long tls_index;\r
 \r
 /* Convert error code to error string - must call LocalFree() on return value */\r
 char *error_string(unsigned long error) {\r
-  if (! FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (char *) &error_message, sizeof(error_message), 0)) {\r
-    if (_snprintf(error_message, sizeof(error_message), "system error %lu", error) < 0) return 0;\r
+  /* Thread-safe buffer */\r
+  char *error_message = (char *) TlsGetValue(tls_index);\r
+  if (! error_message) {\r
+    error_message = (char *) LocalAlloc(LPTR, NSSM_ERROR_BUFSIZE);\r
+    if (! error_message) return "<out of memory for error message>";\r
+    TlsSetValue(tls_index, (void *) error_message);\r
+  }\r
+\r
+  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
+    if (_snprintf_s(error_message, NSSM_ERROR_BUFSIZE, _TRUNCATE, "system error %lu", error) < 0) return 0;\r
   }\r
   return error_message;\r
 }\r
 \r
+/* Convert message code to format string */\r
+char *message_string(unsigned long error) {\r
+  char *ret;\r
+  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
+    ret = (char *) HeapAlloc(GetProcessHeap(), 0, 32);\r
+    if (_snprintf_s(ret, NSSM_ERROR_BUFSIZE, _TRUNCATE, "system error %lu", error) < 0) return 0;\r
+  }\r
+  return ret;\r
+}\r
+\r
 /* Log a message to the Event Log */\r
 void log_event(unsigned short type, unsigned long id, ...) {\r
   va_list arg;\r
@@ -31,3 +50,42 @@ void log_event(unsigned short type, unsigned long id, ...) {
   /* Close event log */\r
   DeregisterEventSource(handle);\r
 }\r
+\r
+/* Log a message to the console */\r
+void print_message(FILE *file, unsigned long id, ...) {\r
+  va_list arg;\r
+\r
+  char *format = message_string(id);\r
+  if (! format) return;\r
+\r
+  va_start(arg, id);\r
+  vfprintf(file, format, arg);\r
+  va_end(arg);\r
+\r
+  LocalFree(format);\r
+}\r
+\r
+/* Show a GUI dialogue */\r
+int popup_message(unsigned int type, unsigned long id, ...) {\r
+  va_list arg;\r
+\r
+  char *format = message_string(id);\r
+  if (! format) {\r
+    return MessageBox(0, "Message %lu was supposed to go here!", NSSM, MB_OK | MB_ICONEXCLAMATION);\r
+  }\r
+\r
+  char blurb[256];\r
+  va_start(arg, id);\r
+  if (vsnprintf_s(blurb, sizeof(blurb), _TRUNCATE, format, arg) < 0) {\r
+    va_end(arg);\r
+    LocalFree(format);\r
+    return MessageBox(0, "Message %lu was supposed to go here!", NSSM, MB_OK | MB_ICONEXCLAMATION);\r
+  }\r
+  va_end(arg);\r
+\r
+  int ret = MessageBox(0, blurb, NSSM, type);\r
+\r
+  LocalFree(format);\r
+\r
+  return ret;\r
+}\r