Changeset 70405 in vbox for trunk/src/VBox/Runtime/r3/win

Timestamp:

Jan 1, 2018 5:45:14 PM (7 years ago)

Author:

vboxsync

Message:

IPRT/time-win.cpp: Use RtlGetInterruptTimePrecise for RTTimeSystemNanoTS when available (based on RDTSC) and using KUSER_SHARED_DATA from nt.h instead of duplicating it and its address.

File:

• trunk/src/VBox/Runtime/r3/win/time-win.cpp (modified) (5 diffs)

trunk/src/VBox/Runtime/r3/win/time-win.cpp

@@ -5 +5 @@
 
 /*
- * Copyright (C) 2006-2017 Oracle Corporation
+ * Copyright (C) 2006-2018 Oracle Corporation
  *
  * This file is part of VirtualBox Open Source Edition (OSE), as
@@ -30 +30 @@
 *********************************************************************************************************************************/
 #define LOG_GROUP RTLOGGROUP_TIME
-#include <iprt/win/windows.h>
+#include <iprt/nt/nt-and-windows.h>
 
 #include <iprt/time.h>
@@ -53 +53 @@
 //#endif
 
-
-#ifdef USE_INTERRUPT_TIME
-
-typedef struct _MY_KSYSTEM_TIME
-{
-    ULONG LowPart;
-    LONG High1Time;
-    LONG High2Time;
-} MY_KSYSTEM_TIME;
-
-typedef struct _MY_KUSER_SHARED_DATA
-{
-    ULONG TickCountLowDeprecated;
-    ULONG TickCountMultiplier;
-    volatile MY_KSYSTEM_TIME InterruptTime;
-    /* The rest is not relevant. */
-} MY_KUSER_SHARED_DATA, *PMY_KUSER_SHARED_DATA;
-
-#endif /* USE_INTERRUPT_TIME */
 
 
@@ -112 +93 @@
     /*
      * Use interrupt time if we can (not possible on NT 3.1).
+     * Note! We cannot entirely depend on g_enmWinVer here as we're likely to
+     *       get called before IPRT is initialized.  Ditto g_hModNtDll.
      */
-    /** @todo use RtlGetInterruptTimePrecise if available (W10+). */
-
-    static int volatile g_fCanUseUserSharedData = -1;
-    int fCanUseUserSharedData = g_fCanUseUserSharedData;
-    if (fCanUseUserSharedData != -1)
+    static PFNRTLGETINTERRUPTTIMEPRECISE    s_pfnRtlGetInterruptTimePrecise = NULL;
+    static int volatile                     s_iCanUseUserSharedData         = -1;
+    int                                     iCanUseUserSharedData           = s_iCanUseUserSharedData;
+    if (iCanUseUserSharedData != -1)
     { /* likely */ }
     else
@@ -123 +105 @@
         /* We may be called before g_enmWinVer has been initialized. */
         if (g_enmWinVer != kRTWinOSType_UNKNOWN)
-            fCanUseUserSharedData = g_enmWinVer > kRTWinOSType_NT310;
+            iCanUseUserSharedData = g_enmWinVer > kRTWinOSType_NT310;
         else
         {
             DWORD dwVer = GetVersion();
-            fCanUseUserSharedData = (dwVer & 0xff) != 3 || ((dwVer >> 8) & 0xff) >= 50;
+            iCanUseUserSharedData = (dwVer & 0xff) != 3 || ((dwVer >> 8) & 0xff) >= 50;
         }
-        g_fCanUseUserSharedData = fCanUseUserSharedData;
+        if (iCanUseUserSharedData != 0)
+        {
+            FARPROC pfn = GetProcAddress(g_hModNtDll ? g_hModNtDll : GetModuleHandleW(L"ntdll"), "RtlGetInterruptTimePrecise");
+            if (pfn != NULL)
+            {
+                ASMAtomicWritePtr(&s_pfnRtlGetInterruptTimePrecise, pfn);
+                iCanUseUserSharedData = 42;
+            }
+        }
+        s_iCanUseUserSharedData = iCanUseUserSharedData;
     }
 
-    if (fCanUseUserSharedData != 0)
+    if (iCanUseUserSharedData != 0)
     {
-        PMY_KUSER_SHARED_DATA pUserSharedData = (PMY_KUSER_SHARED_DATA)(uintptr_t)0x7ffe0000;
         LARGE_INTEGER Time;
-        do
+        if (iCanUseUserSharedData == 42)
         {
-            Time.HighPart = pUserSharedData->InterruptTime.High1Time;
-            Time.LowPart  = pUserSharedData->InterruptTime.LowPart;
-        } while (pUserSharedData->InterruptTime.High2Time != Time.HighPart);
+            uint64_t iIgnored;
+            Time.QuadPart = s_pfnRtlGetInterruptTimePrecise(&iIgnored);
+        }
+        else
+        {
+            PKUSER_SHARED_DATA pUserSharedData = (PKUSER_SHARED_DATA)MM_SHARED_USER_DATA_VA;
+            do
+            {
+                Time.HighPart = pUserSharedData->InterruptTime.High1Time;
+                Time.LowPart  = pUserSharedData->InterruptTime.LowPart;
+            } while (pUserSharedData->InterruptTime.High2Time != Time.HighPart);
+        }
 
         return (uint64_t)Time.QuadPart * 100;