[ros-diffs] [hpoussin] 31091: Fix some possible problems

Author: hpoussin
Date: Sat Dec  8 22:38:39 2007
New Revision: 31091

URL: http://svn.reactos.org/svn/reactos?rev=31091&view=rev
Log:
Fix some possible problems

Modified:
    trunk/reactos/base/setup/usetup/interface/devinst.c

Modified: trunk/reactos/base/setup/usetup/interface/devinst.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/base/setup/usetup/interfac…
==============================================================================
--- trunk/reactos/base/setup/usetup/interface/devinst.c (original)
+++ trunk/reactos/base/setup/usetup/interface/devinst.c Sat Dec  8 22:38:39 2007
@@ -222,7 +222,7 @@
         NtClose(hDeviceKey);
         return;
     }
-    else
+    else if (pPartialInformation)
     {
         for (HardwareID = (LPCWSTR)pPartialInformation->Data;
              (PUCHAR)HardwareID < pPartialInformation->Data +
pPartialInformation->DataLength
@@ -277,7 +277,7 @@
             DPRINT1("NtQueryValueKey() failed with status 0x%08x\n", Status);
             return;
         }
-        else
+        else if (pPartialInformation)
         {
             for (HardwareID = (LPCWSTR)pPartialInformation->Data;
                  (PUCHAR)HardwareID < pPartialInformation->Data +
pPartialInformation->DataLength
@@ -347,7 +347,8 @@
         if (Status == STATUS_BUFFER_TOO_SMALL)
         {
             PnpEventSize += 0x400;
-            PnpEvent = (PPLUGPLAY_EVENT_BLOCK)RtlReAllocateHeap(ProcessHeap, 0, PnpEvent,
PnpEventSize);
+            RtlFreeHeap(ProcessHeap, 0, PnpEvent);
+            PnpEvent = (PPLUGPLAY_EVENT_BLOCK)RtlAllocateHeap(ProcessHeap, 0,
PnpEventSize);
             if (PnpEvent == NULL)
             {
                 NtClose(hEnum);