[PATCH 1/1] advapi32: Avoid that invalid pointers cause crashes.

[Haoyang Chen]

From: Haoyang Chen

---
 dlls/advapi32/eventlog.c       | 21 ++++++++++++++++-----
 dlls/advapi32/tests/eventlog.c |  6 ++++++
 2 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/dlls/advapi32/eventlog.c b/dlls/advapi32/eventlog.c
index 58db53f5536..4b9f059ddc8 100644
--- a/dlls/advapi32/eventlog.c
+++ b/dlls/advapi32/eventlog.c
@@ -32,6 +32,7 @@
 #include "evntprov.h"

#include "wine/debug.h"
+#include "wine/exception.h"

#include "advapi32_misc.h"

@@ -601,7 +602,7 @@ BOOL WINAPI ReportEventA ( HANDLE hEventLog, WORD wType, WORD wCategory, DWORD d
     LPWSTR *wideStrArray;
     UNICODE_STRING str;
     UINT i;
-    BOOL ret;
+    BOOL ret = FALSE;

FIXME("(%p,0x%04x,0x%04x,0x%08lx,%p,0x%04x,0x%08lx,%p,%p): stub\n", hEventLog,
           wType, wCategory, dwEventID, lpUserSid, wNumStrings, dwDataSize, lpStrings, lpRawData);
@@ -610,12 +611,22 @@ BOOL WINAPI ReportEventA ( HANDLE hEventLog, WORD wType, WORD wCategory, DWORD d
     if (!lpStrings) return TRUE;

wideStrArray = malloc(sizeof(WCHAR *) * wNumStrings);
-    for (i = 0; i < wNumStrings; i++)
+    __TRY
+    {
+        for (i = 0; i < wNumStrings; i++)
+        {
+            RtlCreateUnicodeStringFromAsciiz(&str, lpStrings[i]);
+            wideStrArray[i] = str.Buffer;
+        }
+        ret = TRUE;
+    }
+    __EXCEPT_PAGE_FAULT
     {
-        RtlCreateUnicodeStringFromAsciiz(&str, lpStrings[i]);
-        wideStrArray[i] = str.Buffer;
+        SetLastError(ERROR_INVALID_PARAMETER);
     }
-    ret = ReportEventW(hEventLog, wType, wCategory, dwEventID, lpUserSid,
+    __ENDTRY
+
+    if (ret) ret = ReportEventW(hEventLog, wType, wCategory, dwEventID, lpUserSid,
                        wNumStrings, dwDataSize, (LPCWSTR *)wideStrArray, lpRawData);
     for (i = 0; i < wNumStrings; i++)
         free(wideStrArray[i]);
diff --git a/dlls/advapi32/tests/eventlog.c b/dlls/advapi32/tests/eventlog.c
index 5649ca37b3b..be4d75d2b01 100644
--- a/dlls/advapi32/tests/eventlog.c
+++ b/dlls/advapi32/tests/eventlog.c
@@ -829,6 +829,12 @@ static void test_readwrite(void)
         }
     }

+    SetLastError(0xdeadbeef);
+    buf = (void*)0xdeadbeef;
+    ret = ReportEventA(handle, 0, 0, 0, NULL, 1, 0, buf, NULL);
+    ok(!ret, "Expected failure\n");
+    ok(GetLastError() == ERROR_INVALID_PARAMETER, "Expected ERROR_INVALID_PARAMETER, but got: %lx\n",  GetLastError());
+
     SetLastError(0xdeadbeef);
     ret = ReportEventA(handle, 0x20, 0, 0, NULL, 0, 0, NULL, NULL);
     if (!ret && GetLastError() == ERROR_CRC)