RtlFls.c File Reference

#include "precomp.h"

Include dependency graph for RtlFls.c:

Go to the source code of this file.

Classes
struct	_FLS_CALLBACK_INFO
struct	_RTL_FLS_DATA_VISTA
struct	_RTLP_FLS_VALUE_ARRAY
struct	_RTL_FLS_DATA_WIN10

Macros
#define	INIT_FN_PTR(func)

Typedefs
typedef struct _FLS_CALLBACK_INFO	FLS_CALLBACK_INFO
typedef struct _FLS_CALLBACK_INFO *	PFLS_CALLBACK_INFO
typedef NTSTATUS NTAPI	FN_RtlFlsAlloc(_In_ PFLS_CALLBACK_FUNCTION Callback, _Out_ PULONG FlsIndex)
typedef NTSTATUS NTAPI	FN_RtlFlsAllocEx(_In_ PFLS_CALLBACK_FUNCTION Callback, _Out_ PULONG, _Out_ PULONG FlsIndex)
typedef NTSTATUS NTAPI	FN_RtlFlsFree(_In_ ULONG FlsIndex)
typedef NTSTATUS NTAPI	FN_RtlFlsGetValue(_In_ ULONG FlsIndex, _Out_ PVOID *FlsData)
typedef PVOID WINAPI	FN_RtlFlsGetValue2(_In_ ULONG FlsIndex)
typedef NTSTATUS NTAPI	FN_RtlFlsSetValue(_In_ ULONG FlsIndex, _In_ PVOID FlsData)
typedef NTSTATUS NTAPI	FN_RtlProcessFlsData_Vista(_In_ PRTL_FLS_DATA FlsData)
typedef NTSTATUS NTAPI	FN_RtlProcessFlsData_Win10(_In_ HANDLE ProcessHandle, _Out_ PVOID *FlsData)
typedef struct _RTL_FLS_DATA_VISTA	RTL_FLS_DATA_VISTA
typedef struct _RTL_FLS_DATA_VISTA *	PRTL_FLS_DATA_VISTA
typedef struct _RTLP_FLS_VALUE_ARRAY	RTLP_FLS_VALUE_ARRAY
typedef struct _RTLP_FLS_VALUE_ARRAY *	PRTLP_FLS_VALUE_ARRAY
typedef struct _RTL_FLS_DATA_WIN10	RTL_FLS_DATA_WIN10
typedef struct _RTL_FLS_DATA_WIN10 *	PRTL_FLS_DATA_WIN10

Functions
static void	InitFunctionPtrs (void)
void NTAPI	FlsTestCallback (PVOID lpFlsData)
void	Test_RtlFlsAllocFree (void)
void	Test_RtlFlsAllocEx (void)
void	Test_RtlFlsGetSetValue (void)
void	Test_RtlFlsGetValue2 (void)
void	Test_RtlProcessFlsData_Vista (void)
void	Test_RtlProcessFlsData_Win10 (void)
void	Test_RtlProcessFlsData (void)
static DWORD WINAPI	TestThreadProc (LPVOID lpParameter)
	START_TEST (RtlFls)

Variables
FN_RtlFlsAlloc *	pRtlFlsAlloc
FN_RtlFlsAllocEx *	pRtlFlsAllocEx
FN_RtlFlsFree *	pRtlFlsFree
FN_RtlFlsGetValue *	pRtlFlsGetValue
FN_RtlFlsGetValue2 *	pRtlFlsGetValue2
FN_RtlFlsSetValue *	pRtlFlsSetValue
FN_RtlProcessFlsData_Vista *	pRtlProcessFlsData
static ULONG	g_SharedFlsIndex1
static ULONG	g_SharedFlsIndex2
static ULONG	g_SharedFlsIndex3
static ULONG	FlsCallbackCount = 0

Macro Definition Documentation

INIT_FN_PTR

#define INIT_FN_PTR

(

func

)

Value:

    p##func = (void*)GetProcAddress(hNtdll, #func); \
    if (p##func == NULL) \
    { \
        trace("%s not found\n", #func); \
    }

Typedef Documentation

FLS_CALLBACK_INFO

typedef struct _FLS_CALLBACK_INFO FLS_CALLBACK_INFO

FN_RtlFlsAlloc

typedef NTSTATUS NTAPI FN_RtlFlsAlloc(_In_ PFLS_CALLBACK_FUNCTION Callback, _Out_ PULONG FlsIndex)

Definition at line 26 of file RtlFls.c.

FN_RtlFlsAllocEx

typedef NTSTATUS NTAPI FN_RtlFlsAllocEx(_In_ PFLS_CALLBACK_FUNCTION Callback, _Out_ PULONG, _Out_ PULONG FlsIndex)

Definition at line 33 of file RtlFls.c.

FN_RtlFlsFree

typedef NTSTATUS NTAPI FN_RtlFlsFree(_In_ ULONG FlsIndex)

Definition at line 41 of file RtlFls.c.

FN_RtlFlsGetValue

typedef NTSTATUS NTAPI FN_RtlFlsGetValue(_In_ ULONG FlsIndex, _Out_ PVOID *FlsData)

Definition at line 47 of file RtlFls.c.

FN_RtlFlsGetValue2

typedef PVOID WINAPI FN_RtlFlsGetValue2(_In_ ULONG FlsIndex)

Definition at line 54 of file RtlFls.c.

FN_RtlFlsSetValue

typedef NTSTATUS NTAPI FN_RtlFlsSetValue(_In_ ULONG FlsIndex, _In_ PVOID FlsData)

Definition at line 60 of file RtlFls.c.

FN_RtlProcessFlsData_Vista

typedef NTSTATUS NTAPI FN_RtlProcessFlsData_Vista(_In_ PRTL_FLS_DATA FlsData)

Definition at line 67 of file RtlFls.c.

FN_RtlProcessFlsData_Win10

typedef NTSTATUS NTAPI FN_RtlProcessFlsData_Win10(_In_ HANDLE ProcessHandle, _Out_ PVOID *FlsData)

Definition at line 73 of file RtlFls.c.

PFLS_CALLBACK_INFO

typedef struct _FLS_CALLBACK_INFO * PFLS_CALLBACK_INFO

PRTL_FLS_DATA_VISTA

typedef struct _RTL_FLS_DATA_VISTA * PRTL_FLS_DATA_VISTA

PRTL_FLS_DATA_WIN10

typedef struct _RTL_FLS_DATA_WIN10 * PRTL_FLS_DATA_WIN10

PRTLP_FLS_VALUE_ARRAY

typedef struct _RTLP_FLS_VALUE_ARRAY * PRTLP_FLS_VALUE_ARRAY

RTL_FLS_DATA_VISTA

typedef struct _RTL_FLS_DATA_VISTA RTL_FLS_DATA_VISTA

RTL_FLS_DATA_WIN10

typedef struct _RTL_FLS_DATA_WIN10 RTL_FLS_DATA_WIN10

RTLP_FLS_VALUE_ARRAY

typedef struct _RTLP_FLS_VALUE_ARRAY RTLP_FLS_VALUE_ARRAY

Function Documentation

FlsTestCallback()

void NTAPI FlsTestCallback

(

PVOID

lpFlsData

)

Definition at line 131 of file RtlFls.c.

{
    (void)lpFlsData;
    FlsCallbackCount++;
}

Referenced by START_TEST(), and Test_RtlFlsAllocFree().

InitFunctionPtrs()

static void InitFunctionPtrs ( void  )

static

Definition at line 89 of file RtlFls.c.

{
    HMODULE hNtdll = GetModuleHandleA("ntdll.dll");
 
#define INIT_FN_PTR(func) \
    p##func = (void*)GetProcAddress(hNtdll, #func); \
    if (p##func == NULL) \
    { \
        trace("%s not found\n", #func); \
    }
 
    INIT_FN_PTR(RtlFlsAlloc);
    INIT_FN_PTR(RtlFlsAllocEx);
    INIT_FN_PTR(RtlFlsFree);
    INIT_FN_PTR(RtlFlsGetValue);
    INIT_FN_PTR(RtlFlsGetValue2);
    INIT_FN_PTR(RtlFlsSetValue);
    INIT_FN_PTR(RtlProcessFlsData);
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

RtlFls

)

Definition at line 655 of file RtlFls.c.

{
    HANDLE hThread;
 
    InitFunctionPtrs();
 
    if (GetNTVersion() >= _WIN32_WINNT_VISTA)
    {
        ok(pRtlFlsAlloc != NULL, "pRtlFlsAlloc is NULL\n");
        ok(pRtlFlsFree != NULL, "pRtlFlsFree is NULL\n");
        ok(pRtlProcessFlsData != NULL, "pRtlProcessFlsData is NULL\n");
    }
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        ok(pRtlFlsGetValue != NULL, "pRtlFlsGetValue is NULL\n");
        ok(pRtlFlsSetValue != NULL, "pRtlFlsSetValue is NULL\n");
    }
 
    /* Allocate 3 global FLS indices, one without, one with callback, one with bad callback throwing an exception */
    g_SharedFlsIndex1 = FlsAlloc(NULL);
    ok((g_SharedFlsIndex1 >= 1) && (g_SharedFlsIndex1 <= 6), "Unexpected first FLS index: %lu\n", g_SharedFlsIndex1);
    FlsSetValue(g_SharedFlsIndex1, UlongToPtr(0xdead0001));
    g_SharedFlsIndex2 = FlsAlloc(FlsTestCallback);
    ok((g_SharedFlsIndex2 >= 2) && (g_SharedFlsIndex2 <= 7), "Unexpected first FLS index: %lu\n", g_SharedFlsIndex2);
    FlsSetValue(g_SharedFlsIndex2, UlongToPtr(0xdead0002));
    g_SharedFlsIndex3 = FlsAlloc(FlsBadCallback);
    ok((g_SharedFlsIndex3 >= 3) && (g_SharedFlsIndex3 <= 8), "Unexpected first FLS index: %lu\n", g_SharedFlsIndex3);
    FlsSetValue(g_SharedFlsIndex3, UlongToPtr(0xdead0003));
 
    /* Create a test thread to test FLS in a different thread context */
    hThread = CreateThread(NULL, 0, TestThreadProc, NULL, 0, NULL);
    WaitForSingleObject(hThread, INFINITE);
    CloseHandle(hThread);
 
    /* Check the shared indices */
    if (GetNTVersion() < _WIN32_WINNT_WIN10)
    {
        PPEB peb = NtCurrentPeb();
        PTEB teb = NtCurrentTeb();;
        PFLS_CALLBACK_INFO callbackInfo = (PFLS_CALLBACK_INFO)peb->FlsCallback;
        ok_eq_pointer(callbackInfo[g_SharedFlsIndex1].Callback, NULL);
        ok_eq_pointer(callbackInfo[g_SharedFlsIndex2].Callback, NULL);
        PRTL_FLS_DATA_VISTA flsData = (PRTL_FLS_DATA_VISTA)teb->FlsData;
        ok_eq_pointer(flsData->Data[g_SharedFlsIndex1], NULL);
        ok_eq_pointer(flsData->Data[g_SharedFlsIndex2], NULL);
        ok_eq_pointer(flsData->Data[g_SharedFlsIndex3], NULL);
    }
 
    ok_eq_pointer(FlsGetValue(g_SharedFlsIndex1), NULL);
    ok_eq_pointer(FlsGetValue(g_SharedFlsIndex2), NULL);
    ok_eq_pointer(FlsGetValue(g_SharedFlsIndex3), NULL);
}

Test_RtlFlsAllocEx()

void Test_RtlFlsAllocEx

(

void

)

Definition at line 399 of file RtlFls.c.

{
    if (pRtlFlsAllocEx == NULL)
    {
        skip("RtlFlsAllocEx not available\n");
        return;
    }
 
    // TODO
}

Referenced by TestThreadProc().

Test_RtlFlsAllocFree()

void Test_RtlFlsAllocFree

(

void

)

Definition at line 186 of file RtlFls.c.

{
    PTEB teb = NtCurrentTeb();
    PPEB peb = NtCurrentPeb();
    ULONG flsIndex1, flsIndex2, flsIndexFirst;
    NTSTATUS status;
 
    if ((pRtlFlsAlloc == NULL) || (pRtlFlsFree == NULL))
    {
        skip("RtlFlsAlloc or RtlFlsFree not available\n");
        return;
    }
 
    /* We expect an access violation, if the 2nd parameter is NULL */
    StartSeh()
        status = pRtlFlsAlloc(NULL, NULL);
    EndSeh(STATUS_ACCESS_VIOLATION);
 
    /* Allocate first index without a callback */
    status = pRtlFlsAlloc(NULL, &flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok(flsIndex1 >= 4 && flsIndex1 <= 9, "Unexpected first FLS index: %lu\n", flsIndex1);
    flsIndexFirst = flsIndex1;
 
    /* Allocate 2nd index with a callback */
    status = pRtlFlsAlloc(FlsTestCallback, &flsIndex2);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(flsIndex2, flsIndexFirst + 1);
 
    /* Try to free invalid indices */
    status = pRtlFlsFree(0);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
    status = pRtlFlsFree(5000);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
 
    /* Try to free an index that was not allocated */
    status = pRtlFlsFree(flsIndex2 + 1);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
 
    /* Delete first FLS slot, should not trigger callback */
    status = pRtlFlsFree(flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(FlsCallbackCount, 0);
 
    /* Double free slot 1 */
    status = pRtlFlsFree(flsIndex1);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
 
    /* Allocate index without a callback and set the value */
    status = pRtlFlsAlloc(NULL, &flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(flsIndex1, (GetNTVersion() >= _WIN32_WINNT_WIN10) ? flsIndex2 + 1 : flsIndexFirst);
    FlsSetValue(flsIndex1, (PVOID)0x1234);
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
        ok_eq_pointer(flsData->ValueArrays[0]->Elements[flsIndex1 - 1], (PVOID)0x1234);
    }
    else
    {
        PRTL_FLS_DATA_VISTA flsData = (PRTL_FLS_DATA_VISTA)teb->FlsData;
        ok_eq_pointer(flsData->Data[flsIndex1], (PVOID)0x1234);
        ok_eq_ulong(peb->FlsHighIndex, flsIndex2);
        PFLS_CALLBACK_INFO callbackInfo = (PFLS_CALLBACK_INFO)peb->FlsCallback;
        ok((callbackInfo == NULL) || (callbackInfo[flsIndex1].Callback == NULL), "Unexpected callback\n");
    }
 
    /* Delete flsIndex1, should not trigger callback */
    status = pRtlFlsFree(flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(FlsCallbackCount, 0);
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
        ok_eq_pointer(flsData->ValueArrays[0]->Elements[flsIndex1 - 1], NULL);
    }
    else
    {
        PRTL_FLS_DATA_VISTA flsData = (PRTL_FLS_DATA_VISTA)teb->FlsData;
        ok_eq_pointer(flsData->Data[flsIndex1 - 1], NULL);
    }
 
    /* Delete flsIndex2, should not trigger callback */
    status = pRtlFlsFree(flsIndex2);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(FlsCallbackCount, 0);
 
    /* Allocate index with a callback and set the value */
    status = pRtlFlsAlloc(FlsTestCallback, &flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(flsIndex1, flsIndexFirst);
    FlsSetValue(flsIndex1, (PVOID)(ULONG_PTR)0x1234);
 
    if (GetNTVersion() < _WIN32_WINNT_WIN10)
    {
        PFLS_CALLBACK_INFO callbackInfo = (PFLS_CALLBACK_INFO)peb->FlsCallback;
        ok(callbackInfo != NULL, "peb->FlsCallback is NULL. Bye.\n");
        ok_eq_pointer(callbackInfo[flsIndex1].Callback, FlsTestCallback);
    }
 
    /* Delete FLS slot, this should now trigger callback */
    status = pRtlFlsFree(flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(FlsCallbackCount, 1);
    FlsCallbackCount = 0;
 
    if (GetNTVersion() < _WIN32_WINNT_WIN10)
    {
        PFLS_CALLBACK_INFO callbackInfo = (PFLS_CALLBACK_INFO)peb->FlsCallback;
        ok_eq_pointer(callbackInfo[flsIndex1].Callback, NULL);
    }
 
    /* Allocate index with a callback and set the value, then set it to NULL */
    status = pRtlFlsAlloc(FlsTestCallback, &flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(flsIndex1, flsIndexFirst);
    FlsSetValue(flsIndex1, (PVOID)0x1234);
    FlsSetValue(flsIndex1, NULL);
 
    /* Delete FLS slot, should not trigger callback */
    status = pRtlFlsFree(flsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(FlsCallbackCount, 0);
 
    /* Test how many slots we can allocate */
    ULONG NumberAllocated = 0;
    while (status = pRtlFlsAlloc(NULL, &flsIndex2), NT_SUCCESS(status))
    {
        ok(flsIndex2 == flsIndexFirst + NumberAllocated, "Invalid index: %lu != %lu + %lu\n", flsIndex2, flsIndexFirst, NumberAllocated);
        NumberAllocated++;
    }
 
    ok_ntstatus(status, STATUS_NO_MEMORY);
 
    /* Final index is not overwritten */
    ok(flsIndex2 == flsIndexFirst + NumberAllocated - 1, "Invalid index: %lu != %lu + %lu - 1\n", flsIndex2, flsIndexFirst, NumberAllocated);
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        ok_eq_ulong(NumberAllocated, 4080 - flsIndexFirst);
    }
    else
    {
        ok_eq_ulong(NumberAllocated, 128 - flsIndexFirst);
    }
 
    /* Free all allocated slots */
    for (ULONG i = 0; i < NumberAllocated; i++)
    {
        status = pRtlFlsFree(flsIndexFirst + i);
        ok_ntstatus(status, STATUS_SUCCESS);
    }
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        /* All arrays are still allocated */
        PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
        for (ULONG i = 0; i < 8; i++)
        {
            ok(flsData->ValueArrays[i] != NULL, "flsData->ValueArrays[%lu] is NULL\n", i);
        }
    }
    else
    {
        ok_eq_ulong(peb->FlsHighIndex, 127);
    }
 
    /* Set the 3 shared indices */
    FlsSetValue(g_SharedFlsIndex1, UlongToPtr(0xF0001));
    FlsSetValue(g_SharedFlsIndex2, UlongToPtr(0xF0002));
    FlsSetValue(g_SharedFlsIndex3, UlongToPtr(0xF0003));
 
    /* Free the first 2 shared indices */
    status = pRtlFlsFree(g_SharedFlsIndex1);
    ok_ntstatus(status, STATUS_SUCCESS);
    status = pRtlFlsFree(g_SharedFlsIndex2);
    ok_ntstatus(status, STATUS_SUCCESS);
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
        ok_eq_pointer(flsData->ValueArrays[0]->Elements[g_SharedFlsIndex1 - 1], NULL);
        ok_eq_pointer(flsData->ValueArrays[0]->Elements[g_SharedFlsIndex2 - 1], NULL);
    }
    else
    {
        PRTL_FLS_DATA_VISTA flsData = (PRTL_FLS_DATA_VISTA)teb->FlsData;
        ok_eq_pointer(flsData->Data[g_SharedFlsIndex1], NULL);
        ok_eq_pointer(flsData->Data[g_SharedFlsIndex2], NULL);
    }
 
    //PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
    //PVOID* Element = &flsData->ValueArrays[0]->Elements[g_SharedFlsIndex3 - 1];
    //PRTL_FLS_DATA_WIN10 flsData2 = CONTAINING_RECORD(flsData->ListEntry.Blink, RTL_FLS_DATA_WIN10, ListEntry);
    //PVOID* Element2 = &flsData2->ValueArrays[0]->Elements[g_SharedFlsIndex3 - 1];
 
#if defined(_M_IX86) || defined(_M_AMD64)
    // The 3rd shared index has a bad callback and needs SEH handling */
    _SEH2_TRY
    {
        pRtlFlsFree(g_SharedFlsIndex3);
    }
    _SEH2_EXCEPT(FlsBadCallbackFilter(_SEH2_GetExceptionInformation()))
    {
        ok(FALSE, "Unexpected call to exception handler\n");
    }
    _SEH2_END;
    ok_eq_ulong(FlsTestCallbackFilterCallCount, 2);
#endif
}

Referenced by TestThreadProc().

Test_RtlFlsGetSetValue()

void Test_RtlFlsGetSetValue

(

void

)

Definition at line 410 of file RtlFls.c.

{
    PTEB teb = NtCurrentTeb();
    PVOID value;
    BOOL ret;
    NTSTATUS status;
 
    /* Test getting index 0 */
    SetLastError(0xdeadbeef);
    value = FlsGetValue(0);
    ok_eq_pointer(value, NULL);
    ok_eq_ulong(GetLastError(), ERROR_INVALID_PARAMETER);
 
    /* Test setting index 0 */
    SetLastError(0xdeadbeef);
    ret = FlsSetValue(0, (PVOID)0x1234);
    ok_eq_bool(ret, FALSE);
    ok_eq_ulong(GetLastError(), ERROR_INVALID_PARAMETER);
 
    /* Test getting index 5000 */
    SetLastError(0xdeadbeef);
    value = FlsGetValue(5000);
    ok_eq_pointer(value, NULL);
    ok_eq_ulong(GetLastError(), ERROR_INVALID_PARAMETER);
 
    /* Test setting index 5000 */
    SetLastError(0xdeadbeef);
    ret = FlsSetValue(5000, (PVOID)0x1234);
    ok_eq_bool(ret, FALSE);
    ok_eq_ulong(GetLastError(), ERROR_INVALID_PARAMETER);
 
    /* Test setting index 77 (not allocated) */
    SetLastError(0xdeadbeef);
    ret = FlsSetValue(77, (PVOID)0x1234);
    ok_eq_bool(ret, TRUE);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    /* Test getting index 77 (not allocated) */
    SetLastError(0xdeadbeef);
    value = FlsGetValue(77);
    ok_eq_pointer(value, (PVOID)0x1234);
    ok_eq_ulong(GetLastError(), ERROR_SUCCESS);
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
        ok_eq_pointer(flsData->ValueArrays[2]->Elements[28], (PVOID)0x1234);
    }
    else
    {
        PRTL_FLS_DATA_VISTA flsData = (PRTL_FLS_DATA_VISTA)teb->FlsData;
        ok_eq_pointer(flsData->Data[77], (PVOID)0x1234);
    }
 
    /* Free index 77 (fails, as it was never allocated) */
    ret = FlsFree(77);
    ok_eq_bool(ret, FALSE);
 
    if ((pRtlFlsGetValue == NULL) || (pRtlFlsSetValue == NULL))
    {
        skip("RtlFlsGetValue or pRtlFlsSetValue not available\n");
        return;
    }
 
    /* Test setting index 0 */
    SetLastError(0xdeadbeef);
    status = pRtlFlsSetValue(0, (PVOID)0x1234);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    /* Test getting index 0 */
    SetLastError(0xdeadbeef);
    value = (PVOID)(ULONG_PTR)0xDEADBEEFDEADBEEFull;
    status = pRtlFlsGetValue(0, &value);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
    ok_eq_pointer(value, (PVOID)(ULONG_PTR)0xDEADBEEFDEADBEEFull);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    /* Test setting index 5000 */
    SetLastError(0xdeadbeef);
    status = pRtlFlsSetValue(5000, (PVOID)0x1234);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    /* Test getting index 5000 */
    SetLastError(0xdeadbeef);
    value = (PVOID)(ULONG_PTR)0xDEADBEEFDEADBEEFull;
    status = pRtlFlsGetValue(5000, &value);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
    ok_eq_pointer(value, (PVOID)(ULONG_PTR)0xDEADBEEFDEADBEEFull);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    /* Test getting index 99 (not allocated, not set) */
    SetLastError(0xdeadbeef);
    value = (PVOID)(ULONG_PTR)0xDEADBEEFDEADBEEFull;
    status = pRtlFlsGetValue(99, &value);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_pointer(value, NULL);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    /* Test setting index 99 (not allocated) */
    SetLastError(0xdeadbeef);
    status = pRtlFlsSetValue(99, (PVOID)(ULONG_PTR)0x1234);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    /* Test getting index 99 (not allocated) */
    SetLastError(0xdeadbeef);
    value = (PVOID)(ULONG_PTR)0xDEADBEEFDEADBEEFull;
    status = pRtlFlsGetValue(99, &value);
    ok_ntstatus(status, STATUS_SUCCESS);
    ok_eq_pointer(value, (PVOID)(ULONG_PTR)0x1234);
    ok_eq_ulong(GetLastError(), 0xdeadbeef);
 
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
        ok_eq_pointer(flsData->ValueArrays[2]->Elements[50], (PVOID)0x1234);
    }
    else
    {
        PRTL_FLS_DATA_VISTA flsData = (PRTL_FLS_DATA_VISTA)teb->FlsData;
        ok_eq_pointer(flsData->Data[99], (PVOID)(ULONG_PTR)0x1234);
    }
 
    /* Free index 99 (fails, as it was never allocated) */
    status = pRtlFlsFree(99);
    ok_ntstatus(status, STATUS_INVALID_PARAMETER);
}

Referenced by TestThreadProc().

Test_RtlFlsGetValue2()

void Test_RtlFlsGetValue2

(

void

)

Definition at line 540 of file RtlFls.c.

{
    if (pRtlFlsGetValue2 == NULL)
    {
        skip("RtlFlsGetValue2 not available\n");
        return;
    }
 
    // TODO
}

Referenced by TestThreadProc().

Test_RtlProcessFlsData()

void Test_RtlProcessFlsData

(

void

)

Definition at line 575 of file RtlFls.c.

{
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        Test_RtlProcessFlsData_Win10();
    }
    else
    {
        Test_RtlProcessFlsData_Vista();
    }
}

Referenced by TestThreadProc().

Test_RtlProcessFlsData_Vista()

void Test_RtlProcessFlsData_Vista

(

void

)

Definition at line 551 of file RtlFls.c.

{
    if (pRtlProcessFlsData == NULL)
    {
        skip("RtlProcessFlsData not available\n");
        return;
    }
 
    // TODO
}

Referenced by Test_RtlProcessFlsData().

Test_RtlProcessFlsData_Win10()

void Test_RtlProcessFlsData_Win10

(

void

)

Definition at line 562 of file RtlFls.c.

{
    FN_RtlProcessFlsData_Win10* pRtlProcessFlsData_Win10 = (FN_RtlProcessFlsData_Win10*)pRtlProcessFlsData;
    if (pRtlProcessFlsData == NULL)
    {
        skip("RtlProcessFlsData not available\n");
        return;
    }
 
    // TODO
    DBG_UNREFERENCED_LOCAL_VARIABLE(pRtlProcessFlsData_Win10);
}

Referenced by Test_RtlProcessFlsData().

TestThreadProc()

static DWORD WINAPI TestThreadProc ( LPVOID  lpParameter )

static

Definition at line 587 of file RtlFls.c.

{
    PTEB teb = NtCurrentTeb();
    PPEB peb = NtCurrentPeb();
 
    /* Test the initial state */
    if (GetNTVersion() >= _WIN32_WINNT_WIN10)
    {
        /* Windows 10 doesn't use the PEB fields anymore */
        ok_eq_pointer(peb->FlsCallback, NULL);
        ok_eq_pointer(peb->FlsBitmap, NULL);
        ok_eq_ulong(peb->FlsBitmapBits[0], 0);
        ok_eq_ulong(peb->FlsBitmapBits[1], 0);
        ok_eq_ulong(peb->FlsBitmapBits[2], 0);
        ok_eq_ulong(peb->FlsBitmapBits[3], 0);
        ok_eq_ulong(peb->FlsHighIndex, 0);
        ok_eq_pointer(peb->FlsListHead.Flink, NULL);
        ok_eq_pointer(peb->FlsListHead.Blink, NULL);
 
        ok(teb->FlsData != NULL, "teb->FlsData is NULL\n");
        if (teb->FlsData != NULL)
        {
            PRTL_FLS_DATA_WIN10 flsData = (PRTL_FLS_DATA_WIN10)teb->FlsData;
            ok(flsData->ListEntry.Flink != NULL, "flsData->ListEntry.Flink is NULL\n");
            ok(flsData->ListEntry.Blink != NULL, "flsData->ListEntry.Blink is NULL\n");
 
            /* Windows 10 a 2-level array table */
            ok(flsData->ValueArrays[0] != NULL, "flsData->ValueArrays[0] is NULL\n");
            ok(flsData->ValueArrays[1] == NULL, "flsData->ValueArrays[1] is not NULL\n");
            ok(flsData->ValueArrays[0]->Unknown1 == 0, "flsData->ValueArrays[0]->Unknown1 is not 0\n");
            ok(flsData->ValueArrays[0]->Unknown2 == 0, "flsData->ValueArrays[0]->Unknown2 is not 0\n");
            ok(flsData->ValueArrays[0]->Elements[0] != NULL, "flsData->ValueArrays[0]->Elements[0] is NULL\n");
        }
    }
    else
    {
        ok(peb->FlsCallback != NULL, "peb->FlsCallback is NULL\n");
        ok(peb->FlsBitmap != NULL, "peb->FlsBitmap is NULL\n");
        ok(peb->FlsBitmapBits[0] > 0 && peb->FlsBitmapBits[0] < 128,
           "Unexpected value for peb->FlsBitmapBits[0]: 0x%lx\n", peb->FlsBitmapBits[0]);
        ok_eq_ulong(peb->FlsBitmapBits[1], 0);
        ok_eq_ulong(peb->FlsBitmapBits[2], 0);
        ok_eq_ulong(peb->FlsBitmapBits[3], 0);
        ok(peb->FlsHighIndex > 0 && peb->FlsHighIndex < 128,
           "Unexpected value for peb->FlsHighIndex: 0x%lx\n", peb->FlsHighIndex);
        ok(peb->FlsListHead.Flink != NULL, "peb->FlsListHead.Flink is NULL\n");
        ok(peb->FlsListHead.Blink != NULL, "peb->FlsListHead.Blink is NULL\n");
        todo_ros ok(teb->FlsData != NULL, "teb->FlsData == %p\n", teb->FlsData);
        if (teb->FlsData != NULL)
        {
            PRTL_FLS_DATA_VISTA flsData = (PRTL_FLS_DATA_VISTA)teb->FlsData;
            ok(flsData->ListEntry.Flink != NULL, "flsData->ListEntry.Flink is NULL\n");
            ok(flsData->ListEntry.Blink != NULL, "flsData->ListEntry.Blink is NULL\n");
            ok_eq_pointer(flsData->ListEntry.Flink, &peb->FlsListHead);
            ok(flsData->Data[127] == NULL, "flsData->Data[127] is not NULL\n");
        }
    }
 
    Test_RtlFlsAllocFree();
    Test_RtlFlsAllocEx();
    Test_RtlFlsGetSetValue();
    Test_RtlFlsGetValue2();
    Test_RtlProcessFlsData();
 
    return 0;
}

Referenced by START_TEST().

Variable Documentation

FlsCallbackCount

ULONG FlsCallbackCount = 0

static

Definition at line 129 of file RtlFls.c.

Referenced by FlsTestCallback(), and Test_RtlFlsAllocFree().

g_SharedFlsIndex1

ULONG g_SharedFlsIndex1

static

Definition at line 85 of file RtlFls.c.

Referenced by START_TEST(), and Test_RtlFlsAllocFree().

g_SharedFlsIndex2

ULONG g_SharedFlsIndex2

static

Definition at line 86 of file RtlFls.c.

Referenced by START_TEST(), and Test_RtlFlsAllocFree().

g_SharedFlsIndex3

ULONG g_SharedFlsIndex3

static

Definition at line 87 of file RtlFls.c.

Referenced by START_TEST(), and Test_RtlFlsAllocFree().

pRtlFlsAlloc

FN_RtlFlsAlloc* pRtlFlsAlloc

Definition at line 77 of file RtlFls.c.

Referenced by START_TEST(), test_FiberLocalStorage(), and Test_RtlFlsAllocFree().

pRtlFlsAllocEx

FN_RtlFlsAllocEx* pRtlFlsAllocEx

Definition at line 78 of file RtlFls.c.

Referenced by Test_RtlFlsAllocEx().

pRtlFlsFree

FN_RtlFlsFree* pRtlFlsFree

Definition at line 79 of file RtlFls.c.

Referenced by START_TEST(), test_FiberLocalStorage(), Test_RtlFlsAllocFree(), and Test_RtlFlsGetSetValue().

pRtlFlsGetValue

FN_RtlFlsGetValue* pRtlFlsGetValue

Definition at line 80 of file RtlFls.c.

Referenced by START_TEST(), test_FiberLocalStorage(), and Test_RtlFlsGetSetValue().

pRtlFlsGetValue2

FN_RtlFlsGetValue2* pRtlFlsGetValue2

Definition at line 81 of file RtlFls.c.

Referenced by Test_RtlFlsGetValue2().

pRtlFlsSetValue

FN_RtlFlsSetValue* pRtlFlsSetValue

Definition at line 82 of file RtlFls.c.

Referenced by START_TEST(), test_FiberLocalStorage(), and Test_RtlFlsGetSetValue().

pRtlProcessFlsData

FN_RtlProcessFlsData_Vista* pRtlProcessFlsData

Definition at line 83 of file RtlFls.c.

Referenced by START_TEST(), test_FiberLocalStorage(), Test_RtlProcessFlsData_Vista(), and Test_RtlProcessFlsData_Win10().