Changeset 45264 in vbox for trunk/src/VBox/Runtime/testcase

Timestamp:

Mar 31, 2013 12:40:20 PM (12 years ago)

Author:

vboxsync

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

84653

Message:

tstR0ThreadPreemptionDriver.cpp: Start a bunch of threads to make sure the CPU will require preemption.

File:

• trunk/src/VBox/Runtime/testcase/tstR0ThreadPreemptionDriver.cpp (modified) (4 diffs)

trunk/src/VBox/Runtime/testcase/tstR0ThreadPreemptionDriver.cpp

@@ -30 +30 @@
 #include <iprt/initterm.h>
 
+#include <iprt/asm.h>
+#include <iprt/cpuset.h>
 #include <iprt/err.h>
 #include <iprt/path.h>
@@ -42 +44 @@
 #endif
 
+/*******************************************************************************
+*   Global Variables                                                           *
+*******************************************************************************/
+static bool volatile g_fTerminate = false;
+
+
+/**
+ * Try make sure all online CPUs will be engaged.
+ */
+static DECLCALLBACK(int) MyThreadProc(RTTHREAD hSelf, void *pvCpuIdx)
+{
+    RTCPUSET Affinity;
+    RTCpuSetEmpty(&Affinity);
+    RTCpuSetAddByIndex(&Affinity, (intptr_t)pvCpuIdx);
+    RTThreadSetAffinity(&Affinity); /* ignore return code as it's not supported on all hosts. */
+
+    while (!g_fTerminate)
+        RTThreadSleep(50);
+
+    return VINF_SUCCESS;
+}
+
 
 int main(int argc, char **argv)
@@ -141 +165 @@
      * Stay in ring-0 until preemption is pending.
      */
+    RTTHREAD ahThreads[RTCPUSET_MAX_CPUS];
+    uint32_t cThreads = RTMpGetCount();
+    RTCPUSET OnlineSet;
+    RTMpGetOnlineSet(&OnlineSet);
+    for (uint32_t i = 0; i < RT_ELEMENTS(ahThreads); i++)
+    {
+        ahThreads[i] = NIL_RTTHREAD;
+        if (RTCpuSetIsMemberByIndex(&OnlineSet, i))
+            RTThreadCreateF(&ahThreads[i], MyThreadProc, (void *)(uintptr_t)i, 0, RTTHREADTYPE_DEFAULT,
+                            RTTHREADFLAGS_WAITABLE, "cpu=%u", i);
+    }
+
 RTThreadSleep(250); /** @todo fix GIP initialization? */
+
     RTTestSub(hTest, "Pending Preemption");
     for (int i = 0; ; i++)
@@ -163 +200 @@
     }
 
+    ASMAtomicWriteBool(&g_fTerminate, true);
+    for (uint32_t i = 0; i < RT_ELEMENTS(ahThreads); i++)
+        if (ahThreads[i] != NIL_RTTHREAD)
+            RTThreadWait(ahThreads[i], 5000, NULL);
+
     /*
      * Test nested RTThreadPreemptDisable calls.