Changeset 81107 in vbox for trunk/src/VBox/ValidationKit/common

Timestamp:

Oct 4, 2019 1:41:17 AM (5 years ago)

Author:

vboxsync

Message:

common/utils.py: Added getPresentCpuCount method. bugref:9501

File:

• trunk/src/VBox/ValidationKit/common/utils.py (modified) (1 diff)

trunk/src/VBox/ValidationKit/common/utils.py

@@ -300 +300 @@
     return sVersion;
 
+def getPresentCpuCount():
+    """
+    Gets the number of CPUs present in the system.
+
+    This differs from multiprocessor.cpu_count() and os.cpu_count() on windows in
+    that we return the active count rather than the maximum count.  If we don't,
+    we will end up thinking testboxmem1 has 512 CPU threads, which it doesn't and
+    never will have.
+
+    @todo This is probably not exactly what we get on non-windows...
+    """
+
+    if getHostOs() == 'win':
+        fnGetActiveProcessorCount = getattr(ctypes.windll.kernel32, 'GetActiveProcessorCount', None);
+        if fnGetActiveProcessorCount:
+            cCpus = fnGetActiveProcessorCount(ctypes.c_ushort(0xffff));
+            if cCpus > 0:
+                return cCpus;
+
+    import multiprocessor;
+    return multiprocessor.cpu_count();
+
+
 #
 # File system.