winevt_c 0.4.1-x86-mingw32 → 0.4.2-x86-mingw32

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9524469dfab709e5ccab74919030550448efb1565a53a5841cfb76c0d611d9f5
-  data.tar.gz: 5c6ec4637fa2e018f0673518defef25cc9d5a32fac0bfa624c27ed71c624d8b4
+  metadata.gz: d2142e2c4d124e610434fa0103a8477e51c675d9fc561b02578671cc6fc98826
+  data.tar.gz: 3c5f32bbbeaa63746637a772299b42da1e54700185eeaee68d5404bc87c4538f
 SHA512:
-  metadata.gz: 189388df964b905a4ea276a5704b7deda991f35b8f4ada4591725ec99a50929545a68e4b69c95210f9ef621ef9bc420397de33ad5e99ccc4ef77becf56c15190
-  data.tar.gz: 567defa57d64a8941312641cae88a8672c1218739aa0b281d971eb694f6fafebba893faac22700f4c87bf34950d5c048527627393fe605ae07915d6168e7e0f4
+  metadata.gz: 7b5554ed7598743cfc42bff9d112e69308f1e6e263dbdaeffd21de64f9530a4a9cd58f502a2d1c1e972684c7139680825f34c06a72d857d48835e88fcfa3e489
+  data.tar.gz: 97c98e76a6f86156f6bc852405e9ec3e05135ebadd01a38a498d75bb5208fc57524e61606a0d81d76576b518571bc48279c380b75e31358693c4804ec8739e72

data/ext/winevt/extconf.rb

@@ -17,6 +17,7 @@ have_library("ole32")
 
 $LDFLAGS << " -lwevtapi -ladvapi32 -lole32"
 $CFLAGS << " -std=c99 -fPIC -fms-extensions "
+$CXXFLAGS << " -std=c++11 -fPIC -fms-extensions "
 # $CFLAGS << " -g -O0"
 
 create_makefile("winevt/winevt")

data/ext/winevt/winevt_bookmark.c

@@ -90,6 +90,8 @@ rb_winevt_bookmark_render(VALUE self)
 
   utf8str = rb_utf8_str_new_cstr(result);
   free_allocated_mbstr(result);
+  if (wResult != NULL)
+    free(wResult);
 
   return utf8str;
 }

data/ext/winevt/winevt_c.h

@@ -21,12 +21,20 @@
 #define EventBookMark(object) ((struct WinevtBookmark *)DATA_PTR(object))
 #define EventChannel(object) ((struct WinevtChannel *)DATA_PTR(object))
 
+#ifdef __cplusplus
+extern "C" {
+#endif /* __cplusplus */
+
 char* wstr_to_mbstr(UINT cp, const WCHAR *wstr, int clen);
 void free_allocated_mbstr(const char* str);
 WCHAR* render_event(EVT_HANDLE handle, DWORD flags);
 WCHAR* get_description(EVT_HANDLE handle);
 VALUE get_values(EVT_HANDLE handle);
 
+#ifdef __cplusplus
+}
+#endif /* __cplusplus */
+
 VALUE rb_cQuery;
 VALUE rb_cChannel;
 VALUE rb_cBookmark;

data/ext/winevt/winevt_query.c

@@ -203,7 +203,6 @@ rb_winevt_query_seek(VALUE self, VALUE bookmark_or_flag)
 {
   struct WinevtQuery *winevtQuery;
   struct WinevtBookmark *winevtBookmark = NULL;
-  DWORD status;
   DWORD flag;
 
   switch (TYPE(bookmark_or_flag)) {

data/ext/winevt/winevt_subscribe.c

@@ -78,7 +78,6 @@ rb_winevt_subscribe_subscribe(int argc, VALUE *argv, VALUE self)
   VALUE wpathBuf, wqueryBuf;
   PWSTR path, query;
   DWORD status = ERROR_SUCCESS;
-  struct WinevtBookmark *winevtBookmark;
   struct WinevtSubscribe *winevtSubscribe;
 
   hSignalEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

data/ext/winevt/{winevt_utils.c → winevt_utils.cpp}

@@ -1,13 +1,14 @@
 #include <winevt_c.h>
 #include <sddl.h>
 #include <stdlib.h>
+#include <string>
 
 char*
 wstr_to_mbstr(UINT cp, const WCHAR *wstr, int clen)
 {
     char *ptr;
     int len = WideCharToMultiByte(cp, 0, wstr, clen, NULL, 0, NULL, NULL);
-    if (!(ptr = xmalloc(len))) return 0;
+    if (!(ptr = static_cast<char *>(xmalloc(len)))) return 0;
     WideCharToMultiByte(cp, 0, wstr, clen, ptr, len, NULL, NULL);
 
     return ptr;
@@ -25,14 +26,14 @@ WCHAR* render_event(EVT_HANDLE handle, DWORD flags)
   ULONG      bufferSize = 0;
   ULONG      bufferSizeNeeded = 0;
   ULONG      status, count;
-  static WCHAR* result = L"";
+  static WCHAR* result;
   LPTSTR     msgBuf;
 
   do {
     if (bufferSizeNeeded > bufferSize) {
       free(buffer);
       bufferSize = bufferSizeNeeded;
-      buffer = xmalloc(bufferSize);
+      buffer = static_cast<WCHAR *>(xmalloc(bufferSize));
       if (buffer == NULL) {
         status = ERROR_OUTOFMEMORY;
         bufferSize = 0;
@@ -77,13 +78,23 @@ WCHAR* render_event(EVT_HANDLE handle, DWORD flags)
   return result;
 }
 
+static std::wstring guid_to_wstr(const GUID& guid) {
+  LPOLESTR p = NULL;
+  if (FAILED(StringFromCLSID(guid, &p))) {
+    return NULL;
+  }
+  std::wstring s(p);
+  CoTaskMemFree(p);
+  return s;
+}
+
 VALUE get_values(EVT_HANDLE handle)
 {
-  PWSTR buffer = NULL;
+  std::wstring buffer;
   ULONG bufferSize = 0;
   ULONG bufferSizeNeeded = 0;
   DWORD status, propCount = 0;
-  char *result = "";
+  char *result;
   LPTSTR msgBuf;
   WCHAR* tmpWChar = NULL;
   VALUE userValues = rb_ary_new();
@@ -96,10 +107,9 @@ VALUE get_values(EVT_HANDLE handle)
 
   do {
     if (bufferSizeNeeded > bufferSize) {
-      free(buffer);
       bufferSize = bufferSizeNeeded;
-      buffer = xmalloc(bufferSize);
-      if (buffer == NULL) {
+      buffer.resize(bufferSize);
+      if (buffer.c_str() == NULL) {
         status = ERROR_OUTOFMEMORY;
         bufferSize = 0;
         rb_raise(rb_eWinevtQueryError, "Out of memory");
@@ -110,8 +120,8 @@ VALUE get_values(EVT_HANDLE handle)
     if (EvtRender(renderContext,
                   handle,
                   EvtRenderEventValues,
-                  bufferSize,
-                  buffer,
+                  buffer.size(),
+                  &buffer[0],
                   &bufferSizeNeeded,
                   &propCount) != FALSE) {
       status = ERROR_SUCCESS;
@@ -135,7 +145,7 @@ VALUE get_values(EVT_HANDLE handle)
     rb_raise(rb_eWinevtQueryError, "ErrorCode: %d\nError: %s\n", status, RSTRING_PTR(errmsg));
   }
 
-  PEVT_VARIANT pRenderedValues = (PEVT_VARIANT)buffer;
+  PEVT_VARIANT pRenderedValues = (PEVT_VARIANT)buffer.c_str();
   LARGE_INTEGER timestamp;
   SYSTEMTIME st;
   FILETIME ft;
@@ -205,16 +215,16 @@ VALUE get_values(EVT_HANDLE handle)
       rb_ary_push(userValues, rb_utf8_str_new_cstr(result));
       break;
     case EvtVarTypeBoolean:
-      result = pRenderedValues[i].BooleanVal ? "true" : "false";
+      result = const_cast<char *>(pRenderedValues[i].BooleanVal ? "true" : "false");
       rb_ary_push(userValues, rb_utf8_str_new_cstr(result));
       break;
     case EvtVarTypeGuid:
       if (pRenderedValues[i].GuidVal != NULL) {
-        StringFromCLSID(pRenderedValues[i].GuidVal, &tmpWChar);
-        result = wstr_to_mbstr(CP_UTF8, tmpWChar, -1);
+        const GUID guid = *pRenderedValues[i].GuidVal;
+        std::wstring wstr = guid_to_wstr(guid);
+        result = wstr_to_mbstr(CP_UTF8, wstr.c_str(), -1);
         rb_ary_push(userValues, rb_utf8_str_new_cstr(result));
         free_allocated_mbstr(result);
-        CoTaskMemFree(tmpWChar);
       } else {
         rb_ary_push(userValues, rb_utf8_str_new_cstr("?"));
       }
@@ -284,28 +294,22 @@ VALUE get_values(EVT_HANDLE handle)
     }
   }
 
-  if (buffer)
-    xfree(buffer);
-
   if (renderContext)
     EvtClose(renderContext);
 
   return userValues;
 }
 
-static WCHAR* get_message(EVT_HANDLE hMetadata, EVT_HANDLE handle)
+static std::wstring get_message(EVT_HANDLE hMetadata, EVT_HANDLE handle)
 {
 #define BUFSIZE 4096
-  static WCHAR* result = L"";
+  std::wstring result;
   ULONG  status;
   ULONG bufferSizeNeeded = 0;
   LPVOID lpMsgBuf;
-  WCHAR*     prevBuffer;
-  WCHAR     *message;
-  WCHAR     *reallocatedMessage;
+  std::wstring message(BUFSIZE, '\0');
 
-  message = (WCHAR *)xmalloc(sizeof(WCHAR) * BUFSIZE);
-  if (!EvtFormatMessage(hMetadata, handle, 0xffffffff, 0, NULL, EvtFormatMessageEvent, BUFSIZE, message, &bufferSizeNeeded)) {
+  if (!EvtFormatMessage(hMetadata, handle, 0xffffffff, 0, NULL, EvtFormatMessageEvent, message.size(), &message[0], &bufferSizeNeeded)) {
     status = GetLastError();
 
     if (status != ERROR_EVT_UNRESOLVED_VALUE_INSERT) {
@@ -331,7 +335,7 @@ static WCHAR* get_message(EVT_HANDLE hMetadata, EVT_HANDLE handle)
                          MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US),
                          (WCHAR *) &lpMsgBuf, 0, NULL);
 
-        result = _wcsdup((WCHAR *)lpMsgBuf);
+        result = (WCHAR *)lpMsgBuf;
         LocalFree(lpMsgBuf);
 
         goto cleanup;
@@ -344,14 +348,9 @@ static WCHAR* get_message(EVT_HANDLE hMetadata, EVT_HANDLE handle)
     }
 
     if (status == ERROR_INSUFFICIENT_BUFFER) {
-      prevBuffer = message;
-      reallocatedMessage = (WCHAR *)realloc(prevBuffer, sizeof(WCHAR) * bufferSizeNeeded);
-      if (reallocatedMessage == NULL) {
-        rb_raise(rb_eWinevtQueryError, "Reallocation failed.");
-      }
-      message = reallocatedMessage;
+      message.resize(bufferSizeNeeded);
 
-      if(!EvtFormatMessage(hMetadata, handle, 0xffffffff, 0, NULL, EvtFormatMessageEvent, bufferSizeNeeded, message, &bufferSizeNeeded)) {
+      if(!EvtFormatMessage(hMetadata, handle, 0xffffffff, 0, NULL, EvtFormatMessageEvent, message.size(), &message[0], &bufferSizeNeeded)) {
         status = GetLastError();
 
         if (status != ERROR_EVT_UNRESOLVED_VALUE_INSERT) {
@@ -377,7 +376,7 @@ static WCHAR* get_message(EVT_HANDLE hMetadata, EVT_HANDLE handle)
                              MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US),
                              (WCHAR *) &lpMsgBuf, 0, NULL);
 
-            result = _wcsdup((WCHAR *)lpMsgBuf);
+            result = (WCHAR *)lpMsgBuf;
             LocalFree(lpMsgBuf);
 
             goto cleanup;
@@ -389,14 +388,11 @@ static WCHAR* get_message(EVT_HANDLE hMetadata, EVT_HANDLE handle)
     }
   }
 
-  result = _wcsdup(message);
+  result = message;
 
 cleanup:
 
-  if (message)
-    xfree(message);
-
-  return result;
+  return std::wstring(result);
 
 #undef BUFSIZE
 }
@@ -404,14 +400,13 @@ cleanup:
 WCHAR* get_description(EVT_HANDLE handle)
 {
 #define BUFSIZE 4096
-  WCHAR      buffer[BUFSIZE];
+  std::wstring buffer(BUFSIZE, '\0');
   ULONG      bufferSize = 0;
   ULONG      bufferSizeNeeded = 0;
   ULONG      status, count;
-  static WCHAR *result = L"";
-  LPTSTR     msgBuf = "";
+  std::wstring result;
+  LPTSTR     msgBuf;
   EVT_HANDLE hMetadata = NULL;
-  PEVT_VARIANT values = NULL;
 
   static PCWSTR eventProperties[] = {L"Event/System/Provider/@Name"};
   EVT_HANDLE renderContext = EvtCreateRenderContext(1, eventProperties, EvtRenderContextValues);
@@ -422,8 +417,8 @@ WCHAR* get_description(EVT_HANDLE handle)
   if (EvtRender(renderContext,
                 handle,
                 EvtRenderEventValues,
-                _countof(buffer),
-                buffer,
+                buffer.size(),
+                &buffer[0],
                 &bufferSizeNeeded,
                 &count) != FALSE) {
     status = ERROR_SUCCESS;
@@ -447,7 +442,7 @@ WCHAR* get_description(EVT_HANDLE handle)
   }
 
   // Obtain buffer as EVT_VARIANT pointer. To avoid ErrorCide 87 in EvtRender.
-  values = (PEVT_VARIANT)buffer;
+  const PEVT_VARIANT values = reinterpret_cast<PEVT_VARIANT>(const_cast<WCHAR *>(buffer.c_str()));
 
   // Open publisher metadata
   hMetadata = EvtOpenPublisherMetadata(NULL, values[0].StringVal, NULL, MAKELCID(MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), SORT_DEFAULT), 0);
@@ -457,7 +452,7 @@ WCHAR* get_description(EVT_HANDLE handle)
     goto cleanup;
   }
 
-  result = _wcsdup(get_message(hMetadata, handle));
+  result = get_message(hMetadata, handle);
 
 #undef BUFSIZE
 
@@ -469,5 +464,5 @@ cleanup:
   if (hMetadata)
     EvtClose(hMetadata);
 
-  return result;
+  return const_cast<WCHAR *>(result.c_str());
 }

data/lib/winevt/version.rb

@@ -1,3 +1,3 @@
 module Winevt
-  VERSION = "0.4.1"
+  VERSION = "0.4.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: winevt_c
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.2
 platform: x86-mingw32
 authors:
 - Hiroshi Hatake
@@ -111,7 +111,7 @@ files:
 - ext/winevt/winevt_channel.c
 - ext/winevt/winevt_query.c
 - ext/winevt/winevt_subscribe.c
-- ext/winevt/winevt_utils.c
+- ext/winevt/winevt_utils.cpp
 - lib/winevt.rb
 - lib/winevt/2.4/winevt.so
 - lib/winevt/2.5/winevt.so