Changeset 19227 in vbox for trunk/src/VBox/VMM/VMMR0/HWACCMR0.cpp

Timestamp:

Apr 28, 2009 8:50:31 AM (16 years ago)

Author:

vboxsync

Message:

Implemented VMMGetCpu in ring 0. (SMP guests)

File:

• trunk/src/VBox/VMM/VMMR0/HWACCMR0.cpp (modified) (3 diffs)

trunk/src/VBox/VMM/VMMR0/HWACCMR0.cpp

@@ -954 +954 @@
     ASMAtomicWriteBool(&pCpu->fInUse, true);
 
+    AssertMsg(pVCpu->hwaccm.s.idEnteredCpu == NIL_RTCPUID, ("%d", (int)pVCpu->hwaccm.s.idEnteredCpu));
+    pVCpu->hwaccm.s.idEnteredCpu = idCpu;
+
     pCtx = CPUMQueryGuestCtxPtr(pVCpu);
 
@@ -982 +985 @@
     if (RT_SUCCESS(rc))
     {
-        AssertMsg(pVCpu->hwaccm.s.idEnteredCpu == NIL_RTCPUID, ("%d", (int)pVCpu->hwaccm.s.idEnteredCpu));
-        pVCpu->hwaccm.s.idEnteredCpu = idCpu;
-
 #ifdef VBOX_WITH_2X_4GB_ADDR_SPACE
         PGMDynMapMigrateAutoSet(pVCpu);
 #endif
     }
+    else
+        pVCpu->hwaccm.s.idEnteredCpu = NIL_RTCPUID;
     return rc;
 }
@@ -1170 +1172 @@
 {
     return &HWACCMR0Globals.aCpuInfo[idCpu];
+}
+
+/**
+ * Returns the VMCPU of the current EMT thread.
+ *
+ * @param   pVM         The VM to operate on.
+ */
+VMMR0DECL(PVMCPU)  HWACCMR0GetVMCPU(PVM pVM)
+{
+    /* RTMpCpuId had better be cheap. */
+    RTCPUID idHostCpu = RTMpCpuId();
+
+    /** @todo optimize for large number of VCPUs when that becomes more common. */
+    for (unsigned idCpu=0;idCpu<pVM->cCPUs;idCpu++)
+    {
+        PVMCPU pVCpu = &pVM->aCpus[idCpu];
+
+        if (pVCpu->hwaccm.s.idEnteredCpu == idHostCpu)
+            return pVCpu;
+    }
+    AssertReleaseFailed();
+    return 0;
+}
+
+/**
+ * Returns the VMCPU id of the current EMT thread.
+ *
+ * @param   pVM         The VM to operate on.
+ */
+VMMR0DECL(VMCPUID) HWACCMR0GetVMCPUId(PVM pVM)
+{
+    PVMCPU pVCpu = HWACCMR0GetVMCPU(pVM);
+    if (pVCpu)
+        return pVCpu->idCpu;
+
+    AssertReleaseFailed();
+    return 0;
 }