Changeset 57922 in vbox

Timestamp:

Sep 28, 2015 2:49:09 AM (9 years ago)

Author:

vboxsync

Message:

iprt/nt: Forgot to commit RTNtCurrentPeb/Teb fixes for direnum-r3-nt.cpp.

Location:

trunk/include/iprt/nt

Files:

• nt-and-windows.h (modified) (1 diff)
• nt.h (modified) (1 diff)

trunk/include/iprt/nt/nt-and-windows.h

@@ -37 +37 @@
 
 #define IPRT_NT_USE_WINTERNL
+#define IPRT_NT_HAVE_CURRENT_TEB_MACRO
 #define WIN32_NO_STATUS
 #include <Windows.h>

trunk/include/iprt/nt/nt.h

@@ -1358 +1358 @@
 typedef PTEB_COMMON PTEB;
 
-#define RTNtCurrentTeb()        ((PTEB)NtCurrentTeb())
-#define RTNtCurrentPeb()        (RTNtCurrentTeb()->ProcessEnvironmentBlock)
-#define NtCurrentPeb()          RTNtCurrentPeb()
-#define RTNtCurrentThreadId()   ((uint32_t)(uintptr_t)RTNtCurrentTeb()->ClientId.UniqueThread)
+#if !defined(NtCurrentTeb) && !defined(IPRT_NT_HAVE_CURRENT_TEB_MACRO)
+# ifdef RT_ARCH_X86
+DECL_FORCE_INLINE(PTEB)     RTNtCurrentTeb(void) { return (PTEB)__readfsdword(RT_OFFSETOF(TEB_COMMON, NtTib.Self)); }
+DECL_FORCE_INLINE(PPEB)     RTNtCurrentPeb(void) { return (PPEB)__readfsdword(RT_OFFSETOF(TEB_COMMON, ProcessEnvironmentBlock)); }
+DECL_FORCE_INLINE(uint32_t) RTNtCurrentThreadId(void) { return __readfsdword(RT_OFFSETOF(TEB_COMMON, ClientId.UniqueThread)); }
+# elif defined(RT_ARCH_AMD64)
+DECL_FORCE_INLINE(PTEB)     RTNtCurrentTeb(void) { return (PTEB)__readgsqword(RT_OFFSETOF(TEB_COMMON, NtTib.Self)); }
+DECL_FORCE_INLINE(PPEB)     RTNtCurrentPeb(void) { return (PPEB)__readgsqword(RT_OFFSETOF(TEB_COMMON, ProcessEnvironmentBlock)); }
+DECL_FORCE_INLINE(uint32_t) RTNtCurrentThreadId(void) { return (uint32_t)__readgsqword(RT_OFFSETOF(TEB_COMMON, ClientId.UniqueThread)); }
+# else
+#  error "Port me"
+# endif
+#else
+# define RTNtCurrentTeb()        ((PTEB)NtCurrentTeb())
+# define RTNtCurrentPeb()        (RTNtCurrentTeb()->ProcessEnvironmentBlock)
+# define RTNtCurrentThreadId()   ((uint32_t)(uintptr_t)RTNtCurrentTeb()->ClientId.UniqueThread)
+#endif
+#define NtCurrentPeb()           RTNtCurrentPeb()
+
 
 /** @} */