Changeset 2016 in vbox for trunk/src/VBox/Additions/common

Timestamp:

Apr 10, 2007 4:22:01 PM (18 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

20304

Message:

Windows guest library: allocate the event structure from pool.

File:

• trunk/src/VBox/Additions/common/VBoxGuestLib/SysHlp.cpp (modified) (3 diffs)

trunk/src/VBox/Additions/common/VBoxGuestLib/SysHlp.cpp

@@ -89 +89 @@
     IO_STATUS_BLOCK ioStatusBlock;
 
-    KEVENT event;
-    KeInitializeEvent (&event, NotificationEvent, FALSE);
+    KEVENT *pEvent;
+#ifdef KEVENT_STACK_ALLOC
+    KEVENT eventAlloc;
+    pEvent = &eventAlloc;
+#else
+    pEvent = (KEVENT *)ExAllocatePool (NonPagedPool, sizeof (KEVENT));
+    if (!pEvent) return VERR_NO_MEMORY;
+#endif /* KEVENT_STACK_ALLOC */
+    KeInitializeEvent (pEvent, NotificationEvent, FALSE);
 
     PIRP irp = IoBuildDeviceIoControlRequest (u32Function,
@@ -99 +106 @@
                                               cbData,
                                               FALSE,
-                                              &event,
+                                              pEvent,
                                               &ioStatusBlock);
     if (irp == NULL)
     {
         Log(("vbglDriverIOCtl: IoBuildDeviceIoControlRequest failed\n"));
+#ifndef KEVENT_STACK_ALLOC
+        ExFreePool (pEvent);
+#endif /* KEVENT_STACK_ALLOC */
         return VERR_NO_MEMORY;
     }
@@ -109 +119 @@
     NTSTATUS rc = IoCallDriver (pDriver->pDeviceObject, irp);
 
+    if (rc == STATUS_PENDING)
+    {
+        Log(("vbglDriverIOCtl: STATUS_PENDING\n"));
+        rc = KeWaitForSingleObject(pEvent,
+                                   Executive,
+                                   KernelMode,
+                                   FALSE,
+                                   NULL);
+
+        rc = ioStatusBlock.Status;                     
+    }    
+
     if (!NT_SUCCESS(rc))
         Log(("vbglDriverIOCtl: IoCallDriver failed with ntstatus=%x\n", rc));
+
+#ifndef KEVENT_STACK_ALLOC
+    ExFreePool (pEvent);
+#endif /* KEVENT_STACK_ALLOC */
 
     return NT_SUCCESS(rc)? VINF_SUCCESS: VERR_VBGL_IOCTL_FAILED;