VMM/VMMR0: Preemption hooks implemented and enabled for Solaris and Linux hosts.

git-svn-id: https://www.virtualbox.org/svn/vbox/trunk@47989 cfe28804-0f27-0410-a406-dd0f0b0b656f

1 files changed, 16 insertions, 9 deletions

diff --git a/src/VBox/VMM/VMMAll/VMMAll.cpp b/src/VBox/VMM/VMMAll/VMMAll.cpp
index a891b7251c5..8a883796102 100644
--- a/src/VBox/VMM/VMMAll/VMMAll.cpp
+++ b/src/VBox/VMM/VMMAll/VMMAll.cpp
@@ -252,20 +252,27 @@ VMMDECL(PVMCPU) VMMGetCpu(PVM pVM)
     if (pVM->cCpus == 1)
         return &pVM->aCpus[0];
 
-    /* Search first by host cpu id (most common case)
+    /*
+     * Search first by host cpu id (most common case)
      * and then by native thread id (page fusion case).
      */
+    if (!RTThreadPreemptIsEnabled(NIL_RTTHREAD))
+    {
+        /** @todo r=ramshankar: This doesn't buy us anything in terms of performance
+         *        leaving it here for hysterical raisins and as a reference if we
+         *        implemented a hashing approach in the future. */
 
-    /* RTMpCpuId had better be cheap. */
-    RTCPUID idHostCpu = RTMpCpuId();
+        /* RTMpCpuId had better be cheap. */
+        RTCPUID idHostCpu = RTMpCpuId();
 
-    /** @todo optimize for large number of VCPUs when that becomes more common. */
-    for (VMCPUID idCpu = 0; idCpu < pVM->cCpus; idCpu++)
-    {
-        PVMCPU pVCpu = &pVM->aCpus[idCpu];
+        /** @todo optimize for large number of VCPUs when that becomes more common. */
+        for (VMCPUID idCpu = 0; idCpu < pVM->cCpus; idCpu++)
+        {
+            PVMCPU pVCpu = &pVM->aCpus[idCpu];
 
-        if (pVCpu->idHostCpu == idHostCpu)
-            return pVCpu;
+            if (pVCpu->idHostCpu == idHostCpu)
+                return pVCpu;
+        }
     }
 
     /* RTThreadGetNativeSelf had better be cheap. */