Changeset 19826 in vbox for trunk/src/VBox/Runtime/r0drv

Timestamp:

May 19, 2009 2:26:00 PM (16 years ago)

Author:

vboxsync

Message:

Implemented a non-blocking RTMpPokeCpu for Windows hosts.

File:

• trunk/src/VBox/Runtime/r0drv/nt/mp-r0drv-nt.cpp (modified) (1 diff)

trunk/src/VBox/Runtime/r0drv/nt/mp-r0drv-nt.cpp

@@ -330 +330 @@
 }
 
+static KDPC aPokeDpcs[MAXIMUM_PROCESSORS] = {0};
+static bool fPokeDPCsInitialized = false;
+
+static VOID rtMpNtPokeCpuDummy(IN PKDPC Dpc, IN PVOID DeferredContext, IN PVOID SystemArgument1, IN PVOID SystemArgument2)
+{
+    NOREF(Dpc);
+    NOREF(DeferredContext);
+    NOREF(SystemArgument1);
+    NOREF(SystemArgument2);
+}
+
+
+RTDECL(int) RTMpPokeCpu(RTCPUID idCpu)
+{
+    if (!RTMpIsCpuOnline(idCpu))
+        return !RTMpIsCpuPossible(idCpu)
+              ? VERR_CPU_NOT_FOUND
+              : VERR_CPU_OFFLINE;
+
+    if (!fPokeDPCsInitialized)
+    {
+        for (unsigned i = 0; i < RT_ELEMENTS(aPokeDpcs); i++)
+        {
+            KeInitializeDpc(&aPokeDpcs[i], rtmpNtDPCWrapper, NULL);
+            KeSetImportanceDpc(&aPokeDpcs[i], HighImportance);
+            KeSetTargetProcessorDpc(&aPokeDpcs[i], (int)i);
+        }
+        fPokeDPCsInitialized = true;
+    }
+
+    /* Raise the IRQL to DISPATCH_LEVEL so we can't be rescheduled to another cpu.
+     * KeInsertQueueDpc must also be executed at IRQL >= DISPATCH_LEVEL.
+     */
+    KIRQL oldIrql;
+    KeRaiseIrql(DISPATCH_LEVEL, &oldIrql);
+
+    /* Assuming here that high importance DPCs will be delivered immediately; or at least an IPI will be sent immediately.
+     * Todo: verify!
+     */
+    BOOLEAN ret = KeInsertQueueDpc(&aPokeDpcs[idCpu], 0, 0);
+    Assert(ret);
+
+    KeLowerIrql(oldIrql);
+    return VINF_SUCCESS;
+}