implicit_tls.c File Reference

#include "precomp.h"
#include <pseh/pseh2.h>

Include dependency graph for implicit_tls.c:

Go to the source code of this file.

Macros
#define	TLS_VECTOR_MAX_SIZE   512

Functions
DWORD WINAPI	AuxThread0Proc (_In_ LPVOID pParameter)
DWORD WINAPI	AuxThread1Proc (_In_ LPVOID pParameter)
	START_TEST (implicit_tls)

Variables
WCHAR	dllpath_implicit_tls [MAX_PATH]
HANDLE	ThreadEvent
ULONG	ExtantTlsEntryCount = 0

Macro Definition Documentation

TLS_VECTOR_MAX_SIZE

#define TLS_VECTOR_MAX_SIZE   512

Definition at line 12 of file implicit_tls.c.

Function Documentation

AuxThread0Proc()

DWORD WINAPI AuxThread0Proc

(

_In_ LPVOID

pParameter

)

Definition at line 21 of file implicit_tls.c.

{
    PVOID* TlsVector;
    PTEB Teb = NtCurrentTeb();
    int i = 0;
    WaitForSingleObject(ThreadEvent, INFINITE);
    TlsVector = Teb->ThreadLocalStoragePointer;
    while (TlsVector[i] && !((ULONG_PTR)TlsVector[i] % 4))
    {
        ++i;
    }
    WaitForSingleObject(ThreadEvent, INFINITE);
    ok(i == TLS_VECTOR_MAX_SIZE + ExtantTlsEntryCount, "ThreadLocalStoragePointer length is %d, expected length %lu\n", i, TLS_VECTOR_MAX_SIZE + ExtantTlsEntryCount);
    return 0;
}

Referenced by START_TEST().

AuxThread1Proc()

DWORD WINAPI AuxThread1Proc

(

_In_ LPVOID

pParameter

)

Definition at line 40 of file implicit_tls.c.

{
    PVOID* TlsVector;
    PTEB Teb = NtCurrentTeb();
    TlsVector = Teb->ThreadLocalStoragePointer;
    ok(TlsVector[1] != NULL, "ThreadLocalStoragePointer index 1 is NULL; failed to initialize other thread with TLS entries present\n");
    return 0;
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

implicit_tls

)

Definition at line 51 of file implicit_tls.c.

{
    PVOID* TlsVector;
    WCHAR workdir[MAX_PATH];
    WCHAR duplicatepath[MAX_PATH];
    WCHAR basedllname[MAX_PATH];
    HMODULE DllAddr[TLS_VECTOR_MAX_SIZE];
    BOOL IsSuccess;
    DWORD Length;
    HANDLE AuxThread0, AuxThread1;
    int i;
    PTEB Teb = NtCurrentTeb();
    ULONG TlsIdx0Value;
    ULONG TlsIdx0ValueAfter;
 
    IsSuccess = GetTempPathW(_countof(workdir), workdir);
    ok(IsSuccess, "GetTempPathW error: %lu\n", GetLastError());
 
    Length = GetTempFileNameW(workdir, L"implicit_tls", 0, dllpath_implicit_tls);
    ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError());
 
    IsSuccess = CopyFileW(L"implicit_tls.dll", dllpath_implicit_tls, FALSE);
    ok(IsSuccess, "CopyFileW failed with %lu\n", GetLastError());
 
    ThreadEvent = CreateEventA(NULL, FALSE, FALSE, NULL);
    ok(ThreadEvent != INVALID_HANDLE_VALUE, "CreateEventA failed with %lu\n", GetLastError());
 
    AuxThread0 = CreateThread(NULL, 0, AuxThread0Proc, NULL, 0, NULL);
    ok(AuxThread0 != NULL, "CreateThread failed with %lu\n", GetLastError());
 
    TlsVector = Teb->ThreadLocalStoragePointer;
    PULONG_PTR ModuleHandle = TlsVector[0];
    #if defined(_MSC_VER)
    #pragma warning( disable : 4311)
    #endif
    *ModuleHandle = (ULONG)GetModuleHandleA(NULL) + 3;
 
    TlsIdx0Value = *(PULONG)TlsVector[0];
    while (TlsVector[ExtantTlsEntryCount] && !((ULONG_PTR)TlsVector[ExtantTlsEntryCount] % 4))
    {
        ++ExtantTlsEntryCount;
    }
    for (i = 0; i < TLS_VECTOR_MAX_SIZE; i++)
    {
        StringCchPrintfW(basedllname, MAX_PATH, L"basedllname_%d", i);
        Length = GetTempFileNameW(workdir, basedllname, 0, duplicatepath);
        ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError());
        IsSuccess = CopyFileW(dllpath_implicit_tls, duplicatepath, FALSE);
        ok(IsSuccess, "CopyFileW failed with %lu\n", GetLastError());
        DllAddr[i] = LoadLibraryW(duplicatepath);
        ok(DllAddr[i] != NULL, "LoadLibraryW failed with %lu\n", GetLastError());
 
        if (i == TLS_VECTOR_MAX_SIZE / 2)
        {
            AuxThread1 = CreateThread(NULL, 0, AuxThread1Proc, NULL, 0, NULL);
            ok(AuxThread1 != NULL, "CreateThread failed with %lu\n", GetLastError());
        }
    }
 
    TlsVector = Teb->ThreadLocalStoragePointer;
    ok(TlsVector[1] != NULL, "ThreadLocalStoragePointer index 1 is NULL; Implicit TLS unavailable\n");
 
    SetEvent(ThreadEvent);
 
    TlsIdx0ValueAfter = *(PULONG)TlsVector[0];
 
    ok(TlsIdx0Value == TlsIdx0ValueAfter, "Value in TLS index 0 corrupted by DLL loads; expected %lu and got %lu\n", TlsIdx0Value, TlsIdx0ValueAfter);
 
    i = 0;
    while (TlsVector[i] && !((ULONG_PTR)TlsVector[i] % 4))
    {
         ++i;
    }
    ok(i == TLS_VECTOR_MAX_SIZE + ExtantTlsEntryCount, "ThreadLocalStoragePointer length is %d, expected length %lu\n", i, TLS_VECTOR_MAX_SIZE + ExtantTlsEntryCount);
 
    SetEvent(ThreadEvent);
 
    for (i = 0; i < TLS_VECTOR_MAX_SIZE; i++)
    {
        StringCchPrintfW(basedllname, MAX_PATH, L"basedllname_%d", i);
        Length = GetTempFileNameW(workdir, basedllname, 0, duplicatepath);
        ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError());
        IsSuccess = DeleteFileW(duplicatepath);
        ok(IsSuccess, "DeleteFileW failed with %lu\n", GetLastError());
        FreeLibrary(DllAddr[i]);
    }
 
    DeleteFileW(dllpath_implicit_tls);
}

Variable Documentation

dllpath_implicit_tls

WCHAR dllpath_implicit_tls[MAX_PATH]

Definition at line 14 of file implicit_tls.c.

Referenced by START_TEST().

ExtantTlsEntryCount

ULONG ExtantTlsEntryCount = 0

Definition at line 18 of file implicit_tls.c.

Referenced by AuxThread0Proc(), and START_TEST().

ThreadEvent

HANDLE ThreadEvent

Definition at line 16 of file implicit_tls.c.

Referenced by AuxThread0Proc(), and START_TEST().