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

Timestamp:

Feb 28, 2018 10:44:38 AM (7 years ago)

Author:

vboxsync

Message:

IPRT: Added a RTThreadPoke implementation for windows that uses NtAlertThread.

Location:

trunk/src/VBox/Runtime/r3/win

Files:

• init-win.cpp (modified) (2 diffs)
• internal-r3-win.h (modified) (1 diff)
• thread-win.cpp (modified) (2 diffs)

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

@@ -84 +84 @@
 /** NtDuplicateToken (NT 3.51). */
 DECLHIDDEN(PFNNTDUPLICATETOKEN)             g_pfnNtDuplicateToken = NULL;
+/** NtAlertThread (NT 3.51). */
+decltype(NtAlertThread)                    *g_pfnNtAlertThread = NULL;
 
 /** Either ws2_32.dll (NT4+) or wsock32.dll (NT3.x). */
@@ -513 +515 @@
     g_pfnNtQueryFullAttributesFile = (PFNNTQUERYFULLATTRIBUTESFILE)GetProcAddress(g_hModNtDll, "NtQueryFullAttributesFile");
     g_pfnNtDuplicateToken          = (PFNNTDUPLICATETOKEN)GetProcAddress(         g_hModNtDll, "NtDuplicateToken");
-
+    g_pfnNtAlertThread             = (decltype(NtAlertThread) *)GetProcAddress(   g_hModNtDll, "NtAlertThread");
 
     /*

trunk/src/VBox/Runtime/r3/win/internal-r3-win.h

@@ -109 +109 @@
 typedef NTSTATUS (NTAPI *PFNNTDUPLICATETOKEN)(HANDLE, ACCESS_MASK, struct _OBJECT_ATTRIBUTES *, BOOLEAN, TOKEN_TYPE, PHANDLE);
 extern DECLHIDDEN(PFNNTDUPLICATETOKEN)             g_pfnNtDuplicateToken;
+#ifdef ___iprt_nt_nt_h___
+extern decltype(NtAlertThread)                    *g_pfnNtAlertThread;
+#endif
 
 extern DECLHIDDEN(HMODULE)                         g_hModWinSock;

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

@@ -45 +45 @@
 #include <iprt/mem.h>
 #include "internal/thread.h"
+#include "internal-r3-win.h"
 
 
@@ -400 +401 @@
 RT_EXPORT_SYMBOL(RTThreadGetNativeHandle);
 
+
+RTDECL(int) RTThreadPoke(RTTHREAD hThread)
+{
+    AssertReturn(hThread != RTThreadSelf(), VERR_INVALID_PARAMETER);
+    if (g_pfnNtAlertThread)
+    {
+        PRTTHREADINT pThread = rtThreadGet(hThread);
+        AssertReturn(pThread, VERR_INVALID_HANDLE);
+
+        NTSTATUS rcNt = g_pfnNtAlertThread((HANDLE)pThread->hThread);
+
+        rtThreadRelease(pThread);
+        if (NT_SUCCESS(rcNt))
+            return VINF_SUCCESS;
+        return RTErrConvertFromErrno(rcNt);
+    }
+    return VERR_NOT_IMPLEMENTED;
+}
+