winevt_c 0.4.1-x64-mingw32 → 0.4.2-x64-mingw32

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 88141775c03683c6fd5da7e76e8bf0be595695c645714261e02e61ff81af9ec8
-  data.tar.gz: fc22c0af77c78433a14a880c67bd45410b051780617305f3dc00006d03cbe810
+  metadata.gz: 65a174665abedea769521e68fa74e406202eb8fca415751f908ff8ff50e66b4c
+  data.tar.gz: 16ab7d798c5c7a33166bc1b1a641ad00310dc9e343ae3cf43b2ba01ba9451d08
 SHA512:
-  metadata.gz: 28d084d5623e96846f77f91bdbbfc15f6883f0196d17ec64a2a10ebab1f830361e0fd023cbe4c205b62cbd821d0d5e23815dd076a732ed22922c92668f100be7
-  data.tar.gz: 63404d30eb3eec15bd635921035ad24783cbd8856ba4b2098c37f9b17b2160eeca7fdc83a57fc5e1641619b3b1894b615aa4e299db20a5d85369c3eb85d29f3f
+  metadata.gz: 42bbc71c339e706db12973a53871c493f1a814285b4be132fada95e6f210d8e8be661f7597b5dc059dbd98d1db78cb259414fe1d058361cc91d2b665b22ac1c9
+  data.tar.gz: 718ed7191272f1cda5ea4c15ee60a2048eba448b5819d211a619dd5e7f27575d13d40989f18032c316d0b8694622676c4303f5ce98ba53c39405e49bee52daf6

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: x64-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