reactos/wine
1
0
Fork 0
mirror of https://github.com/reactos/wine.git synced 2025-02-23 06:12:31 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

kernel32/tests: Add thread tests for fiber-local storage.

Browse Source 
Signed-off-by: John Sheu <sheu@google.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
John Sheu 2016-08-02 15:55:44 -07:00 committed by Alexandre Julliard
parent ab11a5ae45
commit 4e8f43f423
1 changed files with 148 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

225
dlls/kernel32/tests/thread.c
View File

@ -1247,31 +1247,56 @@ static void test_RegisterWaitForSingleObject(void)
ok(ret, "UnregisterWait failed with error %d\n", GetLastError()); ok(ret, "UnregisterWait failed with error %d\n", GetLastError());
} }
static DWORD TLS_main; static DWORD LS_main;
static DWORD TLS_index0, TLS_index1; static DWORD LS_index0, LS_index1;
static DWORD LS_OutOfIndexesValue;
static DWORD WINAPI TLS_InheritanceProc(LPVOID p) /* Function pointers to the FLS/TLS functions to test in LS_ThreadProc() */
static DWORD (WINAPI *LS_AllocFunc)(void);
static PVOID (WINAPI *LS_GetValueFunc)(DWORD);
static BOOL (WINAPI *LS_SetValueFunc)(DWORD, PVOID);
static BOOL (WINAPI *LS_FreeFunc)(DWORD);
/* Names of the functions tested in LS_ThreadProc(), for error messages */
static const char* LS_AllocFuncName = "";
static const char* LS_GetValueFuncName = "";
static const char* LS_SetValueFuncName = "";
static const char* LS_FreeFuncName = "";
/* FLS entry points, dynamically loaded in platforms that support them */
static DWORD (WINAPI *pFlsAlloc)(PFLS_CALLBACK_FUNCTION);
static BOOL (WINAPI *pFlsFree)(DWORD);
static PVOID (WINAPI *pFlsGetValue)(DWORD);
static BOOL (WINAPI *pFlsSetValue)(DWORD,PVOID);
/* A thunk function to make FlsAlloc compatible with the signature of TlsAlloc */
static DWORD WINAPI FLS_AllocFuncThunk(void)
{ {
/* We should NOT inherit the TLS values from our parent or from the return pFlsAlloc(NULL);
}
static DWORD WINAPI LS_InheritanceProc(LPVOID p)
{
/* We should NOT inherit the FLS/TLS values from our parent or from the
main thread. */ main thread. */
LPVOID val; LPVOID val;
val = TlsGetValue(TLS_main); val = LS_GetValueFunc(LS_main);
ok(val == NULL, "TLS inheritance failed\n"); ok(val == NULL, "%s inheritance failed\n", LS_GetValueFuncName);
val = TlsGetValue(TLS_index0); val = LS_GetValueFunc(LS_index0);
ok(val == NULL, "TLS inheritance failed\n"); ok(val == NULL, "%s inheritance failed\n", LS_GetValueFuncName);
val = TlsGetValue(TLS_index1); val = LS_GetValueFunc(LS_index1);
ok(val == NULL, "TLS inheritance failed\n"); ok(val == NULL, "%s inheritance failed\n", LS_GetValueFuncName);
return 0; return 0;
} }
/* Basic TLS usage test. Make sure we can create slots and the values we /* Basic FLS/TLS usage test. Make sure we can create slots and the values we
store in them are separate among threads. Also test TLS value store in them are separate among threads. Also test FLS/TLS value
inheritance with TLS_InheritanceProc. */ inheritance with LS_InheritanceProc. */
static DWORD WINAPI TLS_ThreadProc(LPVOID p) static DWORD WINAPI LS_ThreadProc(LPVOID p)
{ {
LONG_PTR id = (LONG_PTR) p; LONG_PTR id = (LONG_PTR) p;
LPVOID val; LPVOID val;
@ -1279,80 +1304,80 @@ static DWORD WINAPI TLS_ThreadProc(LPVOID p)
if (sync_threads_and_run_one(0, id)) if (sync_threads_and_run_one(0, id))
{ {
TLS_index0 = TlsAlloc(); LS_index0 = LS_AllocFunc();
ok(TLS_index0 != TLS_OUT_OF_INDEXES, "TlsAlloc failed\n"); ok(LS_index0 != LS_OutOfIndexesValue, "%s failed\n", LS_AllocFuncName);
} }
resync_after_run(); resync_after_run();
if (sync_threads_and_run_one(1, id)) if (sync_threads_and_run_one(1, id))
{ {
TLS_index1 = TlsAlloc(); LS_index1 = LS_AllocFunc();
ok(TLS_index1 != TLS_OUT_OF_INDEXES, "TlsAlloc failed\n"); ok(LS_index1 != LS_OutOfIndexesValue, "%s failed\n", LS_AllocFuncName);
/* Slot indices should be different even if created in different /* Slot indices should be different even if created in different
threads. */ threads. */
ok(TLS_index0 != TLS_index1, "TlsAlloc failed\n"); ok(LS_index0 != LS_index1, "%s failed\n", LS_AllocFuncName);
/* Both slots should be initialized to NULL */ /* Both slots should be initialized to NULL */
val = TlsGetValue(TLS_index0); val = LS_GetValueFunc(LS_index0);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == NULL, "TLS slot not initialized correctly\n"); ok(val == NULL, "Slot not initialized correctly\n");
val = TlsGetValue(TLS_index1); val = LS_GetValueFunc(LS_index1);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == NULL, "TLS slot not initialized correctly\n"); ok(val == NULL, "Slot not initialized correctly\n");
} }
resync_after_run(); resync_after_run();
if (sync_threads_and_run_one(0, id)) if (sync_threads_and_run_one(0, id))
{ {
val = TlsGetValue(TLS_index0); val = LS_GetValueFunc(LS_index0);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == NULL, "TLS slot not initialized correctly\n"); ok(val == NULL, "Slot not initialized correctly\n");
val = TlsGetValue(TLS_index1); val = LS_GetValueFunc(LS_index1);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == NULL, "TLS slot not initialized correctly\n"); ok(val == NULL, "Slot not initialized correctly\n");
ret = TlsSetValue(TLS_index0, (LPVOID) 1); ret = LS_SetValueFunc(LS_index0, (LPVOID) 1);
ok(ret, "TlsSetValue failed\n"); ok(ret, "%s failed\n", LS_SetValueFuncName);
ret = TlsSetValue(TLS_index1, (LPVOID) 2); ret = LS_SetValueFunc(LS_index1, (LPVOID) 2);
ok(ret, "TlsSetValue failed\n"); ok(ret, "%s failed\n", LS_SetValueFuncName);
val = TlsGetValue(TLS_index0); val = LS_GetValueFunc(LS_index0);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == (LPVOID) 1, "TLS slot not initialized correctly\n"); ok(val == (LPVOID) 1, "Slot not initialized correctly\n");
val = TlsGetValue(TLS_index1); val = LS_GetValueFunc(LS_index1);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == (LPVOID) 2, "TLS slot not initialized correctly\n"); ok(val == (LPVOID) 2, "Slot not initialized correctly\n");
} }
resync_after_run(); resync_after_run();
if (sync_threads_and_run_one(1, id)) if (sync_threads_and_run_one(1, id))
{ {
val = TlsGetValue(TLS_index0); val = LS_GetValueFunc(LS_index0);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == NULL, "TLS slot not initialized correctly\n"); ok(val == NULL, "Slot not initialized correctly\n");
val = TlsGetValue(TLS_index1); val = LS_GetValueFunc(LS_index1);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == NULL, "TLS slot not initialized correctly\n"); ok(val == NULL, "Slot not initialized correctly\n");
ret = TlsSetValue(TLS_index0, (LPVOID) 3); ret = LS_SetValueFunc(LS_index0, (LPVOID) 3);
ok(ret, "TlsSetValue failed\n"); ok(ret, "%s failed\n", LS_SetValueFuncName);
ret = TlsSetValue(TLS_index1, (LPVOID) 4); ret = LS_SetValueFunc(LS_index1, (LPVOID) 4);
ok(ret, "TlsSetValue failed\n"); ok(ret, "%s failed\n", LS_SetValueFuncName);
val = TlsGetValue(TLS_index0); val = LS_GetValueFunc(LS_index0);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == (LPVOID) 3, "TLS slot not initialized correctly\n"); ok(val == (LPVOID) 3, "Slot not initialized correctly\n");
val = TlsGetValue(TLS_index1); val = LS_GetValueFunc(LS_index1);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == (LPVOID) 4, "TLS slot not initialized correctly\n"); ok(val == (LPVOID) 4, "Slot not initialized correctly\n");
} }
resync_after_run(); resync_after_run();
@ -1361,36 +1386,36 @@ static DWORD WINAPI TLS_ThreadProc(LPVOID p)
HANDLE thread; HANDLE thread;
DWORD waitret, tid; DWORD waitret, tid;
val = TlsGetValue(TLS_index0); val = LS_GetValueFunc(LS_index0);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == (LPVOID) 1, "TLS slot not initialized correctly\n"); ok(val == (LPVOID) 1, "Slot not initialized correctly\n");
val = TlsGetValue(TLS_index1); val = LS_GetValueFunc(LS_index1);
ok(GetLastError() == ERROR_SUCCESS, "TlsGetValue failed\n"); ok(GetLastError() == ERROR_SUCCESS, "%s failed\n", LS_GetValueFuncName);
ok(val == (LPVOID) 2, "TLS slot not initialized correctly\n"); ok(val == (LPVOID) 2, "Slot not initialized correctly\n");
thread = CreateThread(NULL, 0, TLS_InheritanceProc, 0, 0, &tid); thread = CreateThread(NULL, 0, LS_InheritanceProc, 0, 0, &tid);
ok(thread != NULL, "CreateThread failed\n"); ok(thread != NULL, "CreateThread failed\n");
waitret = WaitForSingleObject(thread, 60000); waitret = WaitForSingleObject(thread, 60000);
ok(waitret == WAIT_OBJECT_0, "WaitForSingleObject failed\n"); ok(waitret == WAIT_OBJECT_0, "WaitForSingleObject failed\n");
CloseHandle(thread); CloseHandle(thread);
ret = TlsFree(TLS_index0); ret = LS_FreeFunc(LS_index0);
ok(ret, "TlsFree failed\n"); ok(ret, "%s failed\n", LS_FreeFuncName);
} }
resync_after_run(); resync_after_run();
if (sync_threads_and_run_one(1, id)) if (sync_threads_and_run_one(1, id))
{ {
ret = TlsFree(TLS_index1); ret = LS_FreeFunc(LS_index1);
ok(ret, "TlsFree failed\n"); ok(ret, "%s failed\n", LS_FreeFuncName);
} }
resync_after_run(); resync_after_run();
return 0; return 0;
} }
static void test_TLS(void) static void run_LS_tests(void)
{ {
HANDLE threads[2]; HANDLE threads[2];
LONG_PTR i; LONG_PTR i;
@ -1399,17 +1424,17 @@ static void test_TLS(void)
init_thread_sync_helpers(); init_thread_sync_helpers();
/* Allocate a TLS slot in the main thread to test for inheritance. */ /* Allocate a slot in the main thread to test for inheritance. */
TLS_main = TlsAlloc(); LS_main = LS_AllocFunc();
ok(TLS_main != TLS_OUT_OF_INDEXES, "TlsAlloc failed\n"); ok(LS_main != LS_OutOfIndexesValue, "%s failed\n", LS_AllocFuncName);
suc = TlsSetValue(TLS_main, (LPVOID) 4114); suc = LS_SetValueFunc(LS_main, (LPVOID) 4114);
ok(suc, "TlsSetValue failed\n"); ok(suc, "%s failed\n", LS_SetValueFuncName);
for (i = 0; i < 2; ++i) for (i = 0; i < 2; ++i)
{ {
DWORD tid; DWORD tid;
threads[i] = CreateThread(NULL, 0, TLS_ThreadProc, (LPVOID) i, 0, &tid); threads[i] = CreateThread(NULL, 0, LS_ThreadProc, (LPVOID) i, 0, &tid);
ok(threads[i] != NULL, "CreateThread failed\n"); ok(threads[i] != NULL, "CreateThread failed\n");
} }
@ -1419,11 +1444,51 @@ static void test_TLS(void)
for (i = 0; i < 2; ++i) for (i = 0; i < 2; ++i)
CloseHandle(threads[i]); CloseHandle(threads[i]);
suc = TlsFree(TLS_main); suc = LS_FreeFunc(LS_main);
ok(suc, "TlsFree failed\n"); ok(suc, "%s failed\n", LS_FreeFuncName);
cleanup_thread_sync_helpers(); cleanup_thread_sync_helpers();
} }
static void test_TLS(void)
{
LS_OutOfIndexesValue = TLS_OUT_OF_INDEXES;
LS_AllocFunc = &TlsAlloc;
LS_GetValueFunc = &TlsGetValue;
LS_SetValueFunc = &TlsSetValue;
LS_FreeFunc = &TlsFree;
LS_AllocFuncName = "TlsAlloc";
LS_GetValueFuncName = "TlsGetValue";
LS_SetValueFuncName = "TlsSetValue";
LS_FreeFuncName = "TlsFree";
run_LS_tests();
}
static void test_FLS(void)
{
if (!pFlsAlloc || !pFlsFree || !pFlsGetValue || !pFlsSetValue)
{
win_skip("Fiber Local Storage not supported\n");
return;
}
LS_OutOfIndexesValue = FLS_OUT_OF_INDEXES;
LS_AllocFunc = &FLS_AllocFuncThunk;
LS_GetValueFunc = pFlsGetValue;
LS_SetValueFunc = pFlsSetValue;
LS_FreeFunc = pFlsFree;
LS_AllocFuncName = "FlsAlloc";
LS_GetValueFuncName = "FlsGetValue";
LS_SetValueFuncName = "FlsSetValue";
LS_FreeFuncName = "FlsFree";
run_LS_tests();
}
static void test_ThreadErrorMode(void) static void test_ThreadErrorMode(void)
{ {
DWORD oldmode; DWORD oldmode;
@ -1945,6 +2010,11 @@ static void init_funcs(void)
X(GetThreadGroupAffinity); X(GetThreadGroupAffinity);
X(SetThreadGroupAffinity); X(SetThreadGroupAffinity);
X(FlsAlloc);
X(FlsFree);
X(FlsSetValue);
X(FlsGetValue);
#undef X #undef X
#define X(f) p##f = (void*)GetProcAddress(ntdll, #f) #define X(f) p##f = (void*)GetProcAddress(ntdll, #f)
@ -2010,6 +2080,7 @@ START_TEST(thread)
test_QueueUserWorkItem(); test_QueueUserWorkItem();
test_RegisterWaitForSingleObject(); test_RegisterWaitForSingleObject();
test_TLS(); test_TLS();
test_FLS();
test_ThreadErrorMode(); test_ThreadErrorMode();
#if (defined(__GNUC__) && (defined(__i386__) || defined(__x86_64__))) || (defined(_MSC_VER) && defined(__i386__)) #if (defined(__GNUC__) && (defined(__i386__) || defined(__x86_64__))) || (defined(_MSC_VER) && defined(__i386__))
test_thread_fpu_cw(); test_thread_fpu_cw();