[rharabien] 52619: [WIN32CSR] - Fix CSR hard error messages. Parameters are now converted to ANSI string which are expected by message format string.

Author: rharabien
Date: Sun Jul 10 22:40:43 2011
New Revision: 52619

URL: http://svn.reactos.org/svn/reactos?rev=52619&view=rev
Log:
[WIN32CSR]
- Fix CSR hard error messages. Parameters are now converted to ANSI string which are expected by message format string.

Modified:
    trunk/reactos/subsystems/win32/csrss/win32csr/harderror.c

Modified: trunk/reactos/subsystems/win32/csrss/win32csr/harderror.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/subsystems/win32/csrss/win3...
==============================================================================
--- trunk/reactos/subsystems/win32/csrss/win32csr/harderror.c [iso-8859-1] (original)
+++ trunk/reactos/subsystems/win32/csrss/win32csr/harderror.c [iso-8859-1] Sun Jul 10 22:40:43 2011
@@ -106,6 +106,25 @@
     return STATUS_SUCCESS;
 }

+static
+VOID
+CsrpFreeStringParameters(
+    IN OUT PULONG_PTR Parameters,
+    IN PHARDERROR_MSG HardErrorMessage)
+{
+    ULONG nParam;
+
+    /* Loop all parameters */
+    for (nParam = 0; nParam < HardErrorMessage->NumberOfParameters; nParam++)
+    {
+        /* Check if the current parameter is a string */
+        if (HardErrorMessage->UnicodeStringParameterMask & (1 << nParam) && Parameters[nParam])
+        {
+            /* Free the string buffer */
+            RtlFreeHeap(RtlGetProcessHeap(), 0, (PVOID)Parameters[nParam]);
+        }
+    }
+}

static
 NTSTATUS
@@ -115,62 +134,82 @@
     IN PHARDERROR_MSG HardErrorMessage,
     HANDLE hProcess)
 {
-    ULONG nParam, UnicodeStringParameterMask, Size = 0;
-    NTSTATUS Status;
-    UNICODE_STRING TempStringU;
-    CHAR *ParamString;
-
-    UnicodeStringParameterMask = HardErrorMessage->UnicodeStringParameterMask;
+    ULONG nParam, Size = 0;
+    NTSTATUS Status = STATUS_SUCCESS;
+    UNICODE_STRING TempStringU, ParamStringU;
+    ANSI_STRING TempStringA;

/* Read all strings from client space */
-    for (nParam = 0;
-            nParam < HardErrorMessage->NumberOfParameters;
-            nParam++, UnicodeStringParameterMask >>= 1)
+    for (nParam = 0; nParam < HardErrorMessage->NumberOfParameters; nParam++)
     {
         Parameters[nParam] = 0;

/* Check if the current parameter is a unicode string */
-        if (UnicodeStringParameterMask & 0x01)
+        if (HardErrorMessage->UnicodeStringParameterMask & (1 << nParam))
         {
             /* Read the UNICODE_STRING from the process memory */
             Status = NtReadVirtualMemory(hProcess,
                                          (PVOID)HardErrorMessage->Parameters[nParam],
-                                         &TempStringU,
-                                         sizeof(TempStringU),
+                                         &ParamStringU,
+                                         sizeof(ParamStringU),
                                          NULL);

-            if (!NT_SUCCESS(Status)) return Status;
+            if (!NT_SUCCESS(Status))
+                break;

/* Allocate a buffer for the string */
-            ParamString = RtlAllocateHeap(RtlGetProcessHeap(),
-                                          HEAP_ZERO_MEMORY,
-                                          TempStringU.Length + sizeof(WCHAR));
-
-            if (!ParamString)
+            TempStringU.MaximumLength = ParamStringU.Length;
+            TempStringU.Length = ParamStringU.Length;
+            TempStringU.Buffer = RtlAllocateHeap(RtlGetProcessHeap(),
+                                                 HEAP_ZERO_MEMORY,
+                                                 TempStringU.MaximumLength);
+
+            if (!TempStringU.Buffer)
             {
-                DPRINT1("Cannot allocate memory %d\n", TempStringU.Length);
-                return STATUS_NO_MEMORY;
+                DPRINT1("Cannot allocate memory %u\n", TempStringU.MaximumLength);
+                Status = STATUS_NO_MEMORY;
             }

/* Read the string buffer from the process memory */
             Status = NtReadVirtualMemory(hProcess,
+                                         ParamStringU.Buffer,
                                          TempStringU.Buffer,
-                                         ParamString,
-                                         TempStringU.Length,
+                                         ParamStringU.Length,
                                          NULL);
             if (!NT_SUCCESS(Status))
             {
                 DPRINT1("NtReadVirtualMemory failed with code: %lx\n", Status);
-                RtlFreeHeap(RtlGetProcessHeap(), 0, ParamString);
-                return Status;
+                RtlFreeHeap(RtlGetProcessHeap(), 0, TempStringU.Buffer);
+                break;
             }

-            /* Zero terminate the string */
-            ParamString[TempStringU.Length] = 0;
-            ParamString[TempStringU.Length + 1] = 0;
-            DPRINT("ParamString='%S'\n", ParamString);
-
-            Parameters[nParam] = (ULONG_PTR)ParamString;
+            DPRINT("ParamString='%wZ'\n", &TempStringU);
+
+            /* Allocate a buffer for converted to ANSI string */
+            TempStringA.MaximumLength = RtlUnicodeStringToAnsiSize(&TempStringU);
+            TempStringA.Buffer = RtlAllocateHeap(RtlGetProcessHeap(),
+                                                 HEAP_ZERO_MEMORY,
+                                                 TempStringA.MaximumLength);
+
+            if (!TempStringA.Buffer)
+            {
+                DPRINT1("Cannot allocate memory %u\n", TempStringA.MaximumLength);
+                RtlFreeHeap(RtlGetProcessHeap(), 0, TempStringU.Buffer);
+                Status = STATUS_NO_MEMORY;
+                break;
+            }
+
+            /* Convert string to ANSI and free temporary buffer */
+            Status = RtlUnicodeStringToAnsiString(&TempStringA, &TempStringU, FALSE);
+            RtlFreeHeap(RtlGetProcessHeap(), 0, TempStringU.Buffer);
+            if (!NT_SUCCESS(Status))
+            {
+                RtlFreeHeap(RtlGetProcessHeap(), 0, TempStringA.Buffer);
+                break;
+            }
+
+            /* Note: RtlUnicodeStringToAnsiString returns NULL terminated string */
+            Parameters[nParam] = (ULONG_PTR)TempStringA.Buffer;
             Size += TempStringU.Length;
         }
         else
@@ -180,34 +219,15 @@
         }
     }

+    if (!NT_SUCCESS(Status))
+    {
+        CsrpFreeStringParameters(Parameters, HardErrorMessage);
+        return Status;
+    }
+
     *SizeOfAllUnicodeStrings = Size;
-    return STATUS_SUCCESS;
+    return Status;
 }
-
-static
-VOID
-CsrpFreeStringParameters(
-    IN OUT PULONG_PTR Parameters,
-    IN PHARDERROR_MSG HardErrorMessage)
-{
-    ULONG nParam, UnicodeStringParameterMask;
-
-    UnicodeStringParameterMask = HardErrorMessage->UnicodeStringParameterMask;
-
-    /* Loop all parameters */
-    for (nParam = 0;
-            nParam < HardErrorMessage->NumberOfParameters;
-            nParam++, UnicodeStringParameterMask >>= 1)
-    {
-        /* Check if the current parameter is a string */
-        if (UnicodeStringParameterMask & 0x01)
-        {
-            /* Free the string buffer */
-            RtlFreeHeap(RtlGetProcessHeap(), 0, (PVOID)Parameters[nParam]);
-        }
-    }
-}
-

static
 NTSTATUS