[ros-diffs] [navaraf] 13418: Dynamically reallocate the buffer for PnP event if it's too small.

Dynamically reallocate the buffer for PnP event if it's too small.
Modified: trunk/reactos/services/umpnpmgr/umpnpmgr.c
  _____

Modified: trunk/reactos/services/umpnpmgr/umpnpmgr.c
--- trunk/reactos/services/umpnpmgr/umpnpmgr.c	2005-02-05 08:43:52 UTC
(rev 13417)
+++ trunk/reactos/services/umpnpmgr/umpnpmgr.c	2005-02-05 08:44:49 UTC
(rev 13418)
@@ -65,10 +65,12 @@

PnpEventThread(LPVOID lpParameter)
 {
   PPLUGPLAY_EVENT_BLOCK PnpEvent;
+  ULONG PnpEventSize;
   NTSTATUS Status;
   RPC_STATUS RpcStatus;

-  PnpEvent = HeapAlloc(GetProcessHeap(), 0, 1024);
+  PnpEventSize = 0x1000;
+  PnpEvent = HeapAlloc(GetProcessHeap(), 0, PnpEventSize);
   if (PnpEvent == NULL)
     return ERROR_OUTOFMEMORY;

@@ -76,10 +78,17 @@
   {
     DPRINT("Calling NtGetPlugPlayEvent()\n");

-    ZeroMemory(PnpEvent, 1024);
-
     /* Wait for the next pnp event */
     Status = NtGetPlugPlayEvent(0, 0, PnpEvent, 1024);
+    /* Resize the buffer for the PnP event if it's too small. */
+    if (Status == STATUS_BUFFER_TOO_SMALL)
+    {
+      PnpEventSize += 0x400;
+      PnpEvent = HeapReAlloc(GetProcessHeap(), 0, PnpEvent,
PnpEventSize);
+      if (PnpEvent == NULL)
+        return ERROR_OUTOFMEMORY;
+      continue;
+    }
     if (!NT_SUCCESS(Status))
     {
       DPRINT("NtPlugPlayEvent() failed (Status %lx)\n", Status);
@@ -87,7 +96,7 @@
     }

DPRINT("Received PnP Event\n");
-    if (UuidEqual((UUID*)&PnpEvent->EventGuid,
(UUID*)&GUID_DEVICE_ARRIVAL, &RpcStatus))
+    if (UuidEqual(&PnpEvent->EventGuid, (UUID*)&GUID_DEVICE_ARRIVAL,
&RpcStatus))
     {
       DPRINT1("Device arrival event: %S\n",
PnpEvent->TargetDevice.DeviceIds);
     }
@@ -98,7 +107,6 @@

/* FIXME: Process the pnp event */

-
     /* Dequeue the current pnp event and signal the next one */
     NtPlugPlayControl(PLUGPLAY_USER_RESPONSE, NULL, 0);
   }