Changeset 13495 in vbox

Timestamp:

Oct 22, 2008 2:51:14 PM (16 years ago)

Author:

vboxsync

Message:

Register power state change callback handler.

File:

• trunk/src/VBox/HostDrivers/Support/win/SUPDrv-win.cpp (modified) (4 diffs)

trunk/src/VBox/HostDrivers/Support/win/SUPDrv-win.cpp

@@ -78 +78 @@
 static int                 VBoxDrvNtDeviceControlSlow(PSUPDRVDEVEXT pDevExt, PSUPDRVSESSION pSession, PIRP pIrp, PIO_STACK_LOCATION pStack);
 static NTSTATUS _stdcall   VBoxDrvNtInternalDeviceControl(PDEVICE_OBJECT pDevObj, PIRP pIrp);
+static VOID     _stdcall   VBoxPowerDispatchCallback(PVOID pCallbackContext, PVOID pArgument1, PVOID pArgument2);
 static NTSTATUS _stdcall   VBoxDrvNtNotSupportedStub(PDEVICE_OBJECT pDevObj, PIRP pIrp);
 static NTSTATUS            VBoxDrvNtErr2NtStatus(int rc);
@@ -142 +143 @@
                     pDrvObj->MajorFunction[IRP_MJ_READ]                     = VBoxDrvNtNotSupportedStub;
                     pDrvObj->MajorFunction[IRP_MJ_WRITE]                    = VBoxDrvNtNotSupportedStub;
+
                     /* more? */
+
+                    /* Register ourselves for power state changes. */
+                    UNICODE_STRING      CallbackName;
+                    OBJECT_ATTRIBUTES   Attr;
+
+                    RtlInitUnicodeString(&CallbackName, L"\\Callback\\PowerState");
+                        InitializeObjectAttributes(&Attr, &CallbackName, OBJ_CASE_INSENSITIVE, NULL, NULL);
+
+                        rc = ExCreateCallback(&pDevExt->pObjPowerCallback, &Attr, TRUE, TRUE);
+                    if (rc == STATUS_SUCCESS)
+                                pDevExt->hPowerCallback = ExRegisterCallback(pDevExt->pObjPowerCallback, VBoxPowerDispatchCallback, pDevObj);
+
                     dprintf(("VBoxDrv::DriverEntry   returning STATUS_SUCCESS\n"));
                     return STATUS_SUCCESS;
@@ -184 +198 @@
 
     dprintf(("VBoxDrvNtUnload at irql %d\n", KeGetCurrentIrql()));
+
+    /* Clean up the power callback registration. */
+    if (pDevExt->hPowerCallback)
+        ExUnregisterCallback(pDevExt->hPowerCallback);
+    if (pDevExt->pObjPowerCallback)
+        ObDereferenceObject(pDevExt->pObjPowerCallback);
 
     /*
@@ -500 +520 @@
 
 /**
+ * ExRegisterCallback handler for power events
+ *
+ * @param   pCallbackContext    User supplied parameter (pDevObj)
+ * @param   pArgument1          First argument
+ * @param   pArgument2          Second argument
+ */
+VOID _stdcall VBoxPowerDispatchCallback(PVOID pCallbackContext, PVOID pArgument1, PVOID pArgument2)
+{
+    PDEVICE_OBJECT pDevObj = (PDEVICE_OBJECT)pCallbackContext;
+
+    dprintf(("VBoxPowerDispatchCallback: %x %x\n", pArgument1, pArgument2));
+
+    /* Power change imminent? */
+    if ((unsigned)pArgument1 == PO_CB_SYSTEM_STATE_LOCK)
+    {
+        if ((unsigned)pArgument2 == 0)
+            dprintf(("VBoxPowerDispatchCallback: about to go into suspend mode!\n"));
+        else
+            dprintf(("VBoxPowerDispatchCallback: resumed!\n"));
+    }
+}
+
+
+/**
  * Initializes any OS specific object creator fields.
  */