X-Git-Url: http://git.iain.cx/?p=nssm.git;a=blobdiff_plain;f=event.cpp;h=b44482c2ab2fc2fdcccfcf05f7a0efe426fed394;hp=7f2b9f4f3363f2ae1feab6c3ee3b96099dcd94d6;hb=HEAD;hpb=d007ff22cecf2ce85cc68a1de81a010d4a65badc

diff --git a/event.cpp b/event.cpp
index 7f2b9f4..b44482c 100644
--- a/event.cpp
+++ b/event.cpp
@@ -1,41 +1,111 @@
 #include "nssm.h"
 
+#define NSSM_SOURCE _T("nssm")
+#define NSSM_ERROR_BUFSIZE 65535
+#define NSSM_NUM_EVENT_STRINGS 16
+unsigned long tls_index;
+
 /* Convert error code to error string - must call LocalFree() on return value */
-char *error_string(unsigned long error) {
-  char *message;
-  if (! FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (char *) &message, 0, 0)) return 0;
-  return message;
+TCHAR *error_string(unsigned long error) {
+  /* Thread-safe buffer */
+  TCHAR *error_message = (TCHAR *) TlsGetValue(tls_index);
+  if (! error_message) {
+    error_message = (TCHAR *) LocalAlloc(LPTR, NSSM_ERROR_BUFSIZE);
+    if (! error_message) return _T("<out of memory for error message>");
+    TlsSetValue(tls_index, (void *) error_message);
+  }
+
+  if (! FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, GetUserDefaultLangID(), (TCHAR *) error_message, NSSM_ERROR_BUFSIZE, 0)) {
+    if (! FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, 0, (TCHAR *) error_message, NSSM_ERROR_BUFSIZE, 0)) {
+      if (_sntprintf_s(error_message, NSSM_ERROR_BUFSIZE, _TRUNCATE, _T("system error %lu"), error) < 0) return 0;
+    }
+  }
+  return error_message;
+}
+
+/* Convert message code to format string */
+TCHAR *message_string(unsigned long error) {
+  TCHAR *ret;
+  if (! FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, GetUserDefaultLangID(), (LPTSTR) &ret, NSSM_ERROR_BUFSIZE, 0)) {
+    if (! FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_HMODULE | FORMAT_MESSAGE_IGNORE_INSERTS, 0, error, 0, (LPTSTR) &ret, NSSM_ERROR_BUFSIZE, 0)) {
+      ret = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, 32 * sizeof(TCHAR));
+      if (_sntprintf_s(ret, NSSM_ERROR_BUFSIZE, _TRUNCATE, _T("system error %lu"), error) < 0) return 0;
+    }
+  }
+  return ret;
 }
 
 /* Log a message to the Event Log */
-void eventprintf(unsigned short type, char *format, ...) {
-  char message[4096];
-  char *strings[2];
-  int n, size;
+void log_event(unsigned short type, unsigned long id, ...) {
   va_list arg;
+  int count;
+  TCHAR *s;
+  TCHAR *strings[NSSM_NUM_EVENT_STRINGS];
 
-  /* Construct the message */
-  size = sizeof(message);
-  va_start(arg, format);
-  n = _vsnprintf(message, size, format, arg);
-  va_end(arg);
-
-  /* Check success */
-  if (n < 0 || n >= size) return;
-
-  /* Construct strings array */
-  strings[0] = message;
-  strings[1] = 0;
-    
   /* Open event log */
-  HANDLE handle = RegisterEventSource(0, TEXT(NSSM));
+  HANDLE handle = RegisterEventSource(0, NSSM_SOURCE);
   if (! handle) return;
 
   /* Log it */
-  if (! ReportEvent(handle, type, 0, 0, 0, 1, 0, (const char **) strings, 0)) {
-    printf("ReportEvent(): %s\n", error_string(GetLastError()));
-  }
+  count = 0;
+  va_start(arg, id);
+  while ((s = va_arg(arg, TCHAR *)) && count < NSSM_NUM_EVENT_STRINGS - 1) strings[count++] = s;
+  strings[count] = 0;
+  va_end(arg);
+  ReportEvent(handle, type, 0, id, 0, count, 0, (const TCHAR **) strings, 0);
 
   /* Close event log */
   DeregisterEventSource(handle);
 }
+
+/* Log a message to the console */
+void print_message(FILE *file, unsigned long id, ...) {
+  va_list arg;
+
+  TCHAR *format = message_string(id);
+  if (! format) return;
+
+  va_start(arg, id);
+  _vftprintf(file, format, arg);
+  va_end(arg);
+
+  LocalFree(format);
+}
+
+/* Show a GUI dialogue */
+int popup_message(HWND owner, unsigned int type, unsigned long id, ...) {
+  va_list arg;
+
+  TCHAR *format = message_string(id);
+  if (! format) {
+    return MessageBox(0, _T("The message which was supposed to go here is missing!"), NSSM, MB_OK | MB_ICONEXCLAMATION);
+  }
+
+  TCHAR blurb[NSSM_ERROR_BUFSIZE];
+  va_start(arg, id);
+  if (_vsntprintf_s(blurb, _countof(blurb), _TRUNCATE, format, arg) < 0) {
+    va_end(arg);
+    LocalFree(format);
+    return MessageBox(0, _T("The message which was supposed to go here is too big!"), NSSM, MB_OK | MB_ICONEXCLAMATION);
+  }
+  va_end(arg);
+
+  MSGBOXPARAMS params;
+  ZeroMemory(&params, sizeof(params));
+  params.cbSize = sizeof(params);
+  params.hInstance = GetModuleHandle(0);
+  params.hwndOwner = owner;
+  params.lpszText = blurb;
+  params.lpszCaption = NSSM;
+  params.dwStyle = type;
+  if (type == MB_OK) {
+    params.dwStyle |= MB_USERICON;
+    params.lpszIcon = MAKEINTRESOURCE(IDI_NSSM);
+  }
+
+  int ret = MessageBoxIndirect(&params);
+
+  LocalFree(format);
+
+  return ret;
+}