Brendan Shanks : ntdll/tests: Test that threads have unique TEBs.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 6d4ec1255acceec7152ed98764ee29991ac04f10
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=6d4ec1255acceec7152ed9876...

Author: Brendan Shanks <bshanks(a)codeweavers.com>
Date:   Mon Apr 18 11:25:10 2022 -0700

ntdll/tests: Test that threads have unique TEBs.

Signed-off-by: Brendan Shanks <bshanks(a)codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>

---

 dlls/ntdll/tests/thread.c | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/dlls/ntdll/tests/thread.c b/dlls/ntdll/tests/thread.c
index a41b1f2cc6a..7a47486a3b6 100644
--- a/dlls/ntdll/tests/thread.c
+++ b/dlls/ntdll/tests/thread.c
@@ -118,9 +118,45 @@ static void test_dbg_hidden_thread_creation(void)
     CloseHandle( thread );
 }
 
+static void CALLBACK test_unique_teb_proc(void *param)
+{
+    TEB **teb = param;
+    *teb = NtCurrentTeb();
+}
+
+static void test_unique_teb(void)
+{
+    HANDLE threads[2];
+    TEB *teb1, *teb2;
+    NTSTATUS status;
+
+    if (!pNtCreateThreadEx)
+    {
+        win_skip( "NtCreateThreadEx is not available.\n" );
+        return;
+    }
+
+    status = pNtCreateThreadEx( &threads[0], THREAD_ALL_ACCESS, NULL, GetCurrentProcess(), test_unique_teb_proc,
+                                &teb1, 0, 0, 0, 0, NULL );
+    ok( status == STATUS_SUCCESS, "Got unexpected status %#lx.\n", status );
+
+    status = pNtCreateThreadEx( &threads[1], THREAD_ALL_ACCESS, NULL, GetCurrentProcess(), test_unique_teb_proc,
+                                &teb2, 0, 0, 0, 0, NULL );
+    ok( status == STATUS_SUCCESS, "Got unexpected status %#lx.\n", status );
+
+    WaitForMultipleObjects( 2, threads, TRUE, INFINITE );
+    CloseHandle( threads[0] );
+    CloseHandle( threads[1] );
+
+    ok( NtCurrentTeb() != teb1, "Multiple threads have TEB %p.\n", teb1 );
+    ok( NtCurrentTeb() != teb2, "Multiple threads have TEB %p.\n", teb2 );
+    ok( teb1 != teb2, "Multiple threads have TEB %p.\n", teb1 );
+}
+
 START_TEST(thread)
 {
     init_function_pointers();
 
     test_dbg_hidden_thread_creation();
+    test_unique_teb();
 }