FLS.c File Reference

#include "precomp.h"
#include <ndk/pstypes.h>
#include <ndk/rtlfuncs.h>
#include <pseh/pseh2.h>

Include dependency graph for FLS.c:

Go to the source code of this file.

Classes
struct	_FLS_CALLBACK_INFO

Macros
#define	NtCurrentPeb()   (NtCurrentTeb()->ProcessEnvironmentBlock)
#define	WINVER_2003   0x0502
#define	ok_fls   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : ok_fls_
#define	X(f)   p##f = (void*)GetProcAddress(hKernel32, #f);

Typedefs
typedef struct _FLS_CALLBACK_INFO	FLS_CALLBACK_INFO
typedef struct _FLS_CALLBACK_INFO *	PFLS_CALLBACK_INFO

Functions
static	DWORD (WINAPI *pFlsAlloc)(PFLS_CALLBACK_FUNCTION)
static	BOOL (WINAPI *pFlsFree)(DWORD)
static	PVOID (WINAPI *pFlsGetValue)(DWORD)
VOID WINAPI	FlsCallback1 (PVOID lpFlsData)
VOID WINAPI	FlsCallback2 (PVOID lpFlsData)
VOID WINAPI	FlsCallback3 (PVOID lpFlsData)
void	ok_fls_ (DWORD dwIndex, PVOID pValue, PFLS_CALLBACK_FUNCTION lpCallback)
static VOID	init_funcs (void)
	START_TEST (FLS)

Variables
static	PVOID
static DWORD	g_WinVersion = 0
PVOID	g_FlsData1 = NULL
LONG	g_FlsCalled1 = 0
PVOID	g_FlsData2 = NULL
LONG	g_FlsCalled2 = 0
PVOID	g_FlsData3 = NULL
LONG	g_FlsCalled3 = 0
BOOL	g_FlsExcept3 = FALSE

Macro Definition Documentation

NtCurrentPeb

#define NtCurrentPeb

(

)

(NtCurrentTeb()->ProcessEnvironmentBlock)

Definition at line 22 of file FLS.c.

ok_fls

#define ok_fls   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : ok_fls_

Definition at line 119 of file FLS.c.

WINVER_2003

#define WINVER_2003   0x0502

Definition at line 23 of file FLS.c.

X

#define X

(

f

)

p##f = (void*)GetProcAddress(hKernel32, #f);

Typedef Documentation

FLS_CALLBACK_INFO

typedef struct _FLS_CALLBACK_INFO FLS_CALLBACK_INFO

PFLS_CALLBACK_INFO

typedef struct _FLS_CALLBACK_INFO * PFLS_CALLBACK_INFO

Function Documentation

BOOL()

static BOOL ( WINAPI *  pFlsFree )

static

DWORD()

static DWORD ( WINAPI *  pFlsAlloc )

static

FlsCallback1()

VOID WINAPI FlsCallback1

(

PVOID

lpFlsData

)

Definition at line 35 of file FLS.c.

{
    ok(lpFlsData == g_FlsData1, "Expected g_FlsData1(%p), got %p\n", g_FlsData1, lpFlsData);
    InterlockedIncrement(&g_FlsCalled1);
}

Referenced by START_TEST().

FlsCallback2()

VOID WINAPI FlsCallback2

(

PVOID

lpFlsData

)

Definition at line 41 of file FLS.c.

{
    ok(lpFlsData == g_FlsData2, "Expected g_FlsData2(%p), got %p\n", g_FlsData2, lpFlsData);
    InterlockedIncrement(&g_FlsCalled2);
}

Referenced by START_TEST().

FlsCallback3()

VOID WINAPI FlsCallback3

(

PVOID

lpFlsData

)

Definition at line 47 of file FLS.c.

{
    ok(lpFlsData == g_FlsData3, "Expected g_FlsData3(%p), got %p\n", g_FlsData3, lpFlsData);
 
    if (g_WinVersion <= WINVER_2003)
        ok(pRtlIsCriticalSectionLockedByThread(NtCurrentPeb()->FastPebLock), "Expected lock on PEB\n");
    InterlockedIncrement(&g_FlsCalled3);
    if (g_FlsExcept3)
    {
        RaiseException(ERROR_INVALID_PARAMETER, EXCEPTION_NONCONTINUABLE, 0, NULL);
    }
}

Referenced by ok_fls_(), and START_TEST().

init_funcs()

static VOID init_funcs ( void  )

static

Definition at line 121 of file FLS.c.

{
    HMODULE hKernel32 = GetModuleHandleA("kernel32.dll");
    HMODULE hNTDLL = GetModuleHandleA("ntdll.dll");
 
#define X(f) p##f = (void*)GetProcAddress(hKernel32, #f);
    X(FlsAlloc);
    X(FlsFree);
    X(FlsGetValue);
    X(FlsSetValue);
#undef X
    pRtlIsCriticalSectionLockedByThread = (void*)GetProcAddress(hNTDLL, "RtlIsCriticalSectionLockedByThread");
}

Referenced by START_TEST().

ok_fls_()

void ok_fls_	(	DWORD	dwIndex,
		PVOID	pValue,
		PFLS_CALLBACK_FUNCTION	lpCallback
	)

Definition at line 67 of file FLS.c.

{
    PFLS_CALLBACK_INFO FlsCallback;
    PVOID* FlsData;
    PVOID gotValue;
 
    FlsCallback = (PFLS_CALLBACK_INFO)NtCurrentPeb()->FlsCallback;
    FlsData = (PVOID*)NtCurrentTeb()->FlsData;
 
    winetest_ok(FlsData != NULL, "Expected FlsData\n");
    winetest_ok(FlsCallback != NULL, "Expected FlsCallback\n");
 
    if (FlsData == NULL || FlsCallback == NULL)
    {
        winetest_skip("Unable to continue test\n");
        return;
    }
 
    if (g_WinVersion <= WINVER_2003)
    {
        winetest_ok(NtCurrentPeb()->FlsCallback[dwIndex] == lpCallback,
                    "Expected NtCurrentPeb()->FlsCallback[%lu] to be %p, was %p\n",
                    dwIndex,
                    lpCallback,
                    NtCurrentPeb()->FlsCallback[dwIndex]);
    }
    else
    {
        winetest_ok(FlsCallback[dwIndex].lpCallback == lpCallback,
                    "Expected FlsCallback[%lu].lpCallback to be %p, was %p\n",
                    dwIndex,
                    lpCallback,
                    FlsCallback[dwIndex].lpCallback);
        if (lpCallback != &FlsCallback3 || !g_FlsExcept3)
        {
            winetest_ok(FlsCallback[dwIndex].Unknown == NULL,
                        "Expected FlsCallback[%lu].Unknown to be %p, was %p\n",
                        dwIndex,
                        NULL,
                        FlsCallback[dwIndex].Unknown);
        }
    }
    winetest_ok(FlsData[dwIndex + 2] == pValue,
                "Expected FlsData[%lu + 2] to be %p, was %p\n",
                dwIndex,
                pValue,
                FlsData[dwIndex + 2]);
 
    gotValue = pFlsGetValue(dwIndex);
    winetest_ok(gotValue == pValue, "Expected FlsGetValue(%lu) to be %p, was %p\n", dwIndex, pValue, gotValue);
}

PVOID()

static PVOID ( WINAPI *  pFlsGetValue )

static

START_TEST()

START_TEST

(

FLS

)

Definition at line 137 of file FLS.c.

{
    RTL_OSVERSIONINFOW rtlinfo = { sizeof(rtlinfo) };
    DWORD dwIndex1, dwIndex2, dwIndex3, dwErr;
    BOOL bRet;
 
    init_funcs();
    if (!pFlsAlloc || !pFlsFree || !pFlsGetValue || !pFlsSetValue)
    {
        skip("Fls functions not available\n");
        return;
    }
    if (!pRtlIsCriticalSectionLockedByThread)
    {
        skip("RtlIsCriticalSectionLockedByThread function not available\n");
        return;
    }
 
    RtlGetVersion(&rtlinfo);
    g_WinVersion = (rtlinfo.dwMajorVersion << 8) | rtlinfo.dwMinorVersion;
 
    dwIndex1 = pFlsAlloc(FlsCallback1);
    ok(dwIndex1 != FLS_OUT_OF_INDEXES, "Unable to allocate FLS index\n");
    dwIndex2 = pFlsAlloc(FlsCallback2);
    ok(dwIndex2 != FLS_OUT_OF_INDEXES, "Unable to allocate FLS index\n");
    ok(dwIndex1 != dwIndex2, "Expected different indexes, got %lu\n", dwIndex1);
 
    dwIndex3 = pFlsAlloc(FlsCallback3);
    ok(dwIndex3 != FLS_OUT_OF_INDEXES, "Unable to allocate FLS index\n");
    ok(dwIndex1 != dwIndex3, "Expected different indexes, got %lu\n", dwIndex1);
 
    if (dwIndex1 == FLS_OUT_OF_INDEXES || dwIndex2 == FLS_OUT_OF_INDEXES || dwIndex3 == FLS_OUT_OF_INDEXES)
    {
        skip("Unable to continue test\n");
        return;
    }
 
    ok_fls(dwIndex1, g_FlsData1, &FlsCallback1);
    ok_fls(dwIndex2, g_FlsData2, &FlsCallback2);
    ok_fls(dwIndex3, g_FlsData3, &FlsCallback3);
 
    g_FlsData1 = (PVOID)0x123456;
    ok(pFlsSetValue(dwIndex1, g_FlsData1), "FlsSetValue(%lu, %p) failed\n", dwIndex1, g_FlsData1);
 
    ok_fls(dwIndex1, g_FlsData1, &FlsCallback1);
    ok_fls(dwIndex2, g_FlsData2, &FlsCallback2);
    ok_fls(dwIndex3, g_FlsData3, &FlsCallback3);
 
    ok_int(g_FlsCalled1, 0);
    ok_int(g_FlsCalled2, 0);
    ok_int(g_FlsCalled3, 0);
 
    g_FlsData2 = (PVOID)0x9876112;
    ok(pFlsSetValue(dwIndex2, g_FlsData2), "FlsSetValue(%lu, %p) failed\n", dwIndex2, g_FlsData2);
 
    ok_fls(dwIndex1, g_FlsData1, &FlsCallback1);
    ok_fls(dwIndex2, g_FlsData2, &FlsCallback2);
    ok_fls(dwIndex3, g_FlsData3, &FlsCallback3);
 
 
    ok_int(g_FlsCalled1, 0);
    ok_int(g_FlsCalled2, 0);
    ok_int(g_FlsCalled3, 0);
 
    g_FlsData3 = (PVOID)0x98762;
    ok(pFlsSetValue(dwIndex3, g_FlsData3), "FlsSetValue(%lu, %p) failed\n", dwIndex3, g_FlsData3);
 
    ok_fls(dwIndex1, g_FlsData1, &FlsCallback1);
    ok_fls(dwIndex2, g_FlsData2, &FlsCallback2);
    ok_fls(dwIndex3, g_FlsData3, &FlsCallback3);
 
    ok_int(g_FlsCalled1, 0);
    ok_int(g_FlsCalled2, 0);
    ok_int(g_FlsCalled3, 0);
 
    ok(pFlsFree(dwIndex1) == TRUE, "FlsFree(%lu) failed\n", dwIndex1);
    g_FlsData1 = NULL;
 
    ok_fls(dwIndex1, g_FlsData1, NULL);
    ok_fls(dwIndex2, g_FlsData2, &FlsCallback2);
    ok_fls(dwIndex3, g_FlsData3, &FlsCallback3);
 
    ok_int(g_FlsCalled1, 1);
    ok_int(g_FlsCalled2, 0);
    ok_int(g_FlsCalled3, 0);
 
    g_FlsExcept3 = TRUE;
    _SEH2_TRY
    {
        bRet = pFlsFree(dwIndex3);
        dwErr = GetLastError();
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        bRet = 12345;
        dwErr = 0xdeaddead;
    }
    _SEH2_END;
    ok(pRtlIsCriticalSectionLockedByThread(NtCurrentPeb()->FastPebLock) == FALSE, "Expected no lock on PEB\n");
 
    ok(bRet == 12345, "FlsFree(%lu) should have failed, got %u\n", dwIndex3, bRet);
    ok(dwErr == 0xdeaddead, "Expected GetLastError() to be 0xdeaddead, was %lx\n", dwErr);
 
    ok_fls(dwIndex1, g_FlsData1, NULL);
    ok_fls(dwIndex2, g_FlsData2, &FlsCallback2);
    ok_fls(dwIndex3, g_FlsData3, &FlsCallback3);
 
    ok_int(g_FlsCalled1, 1);
    ok_int(g_FlsCalled2, 0);
    ok_int(g_FlsCalled3, 1);
}

Variable Documentation

g_FlsCalled1

LONG g_FlsCalled1 = 0

Definition at line 27 of file FLS.c.

Referenced by FlsCallback1(), and START_TEST().

g_FlsCalled2

LONG g_FlsCalled2 = 0

Definition at line 29 of file FLS.c.

Referenced by FlsCallback2(), and START_TEST().

g_FlsCalled3

LONG g_FlsCalled3 = 0

Definition at line 31 of file FLS.c.

Referenced by FlsCallback3(), and START_TEST().

g_FlsData1

PVOID g_FlsData1 = NULL

Definition at line 26 of file FLS.c.

Referenced by FlsCallback1(), and START_TEST().

g_FlsData2

PVOID g_FlsData2 = NULL

Definition at line 28 of file FLS.c.

Referenced by FlsCallback2(), and START_TEST().

g_FlsData3

PVOID g_FlsData3 = NULL

Definition at line 30 of file FLS.c.

Referenced by FlsCallback3(), and START_TEST().

g_FlsExcept3

BOOL g_FlsExcept3 = FALSE

Definition at line 32 of file FLS.c.

Referenced by FlsCallback3(), ok_fls_(), and START_TEST().

g_WinVersion

DWORD g_WinVersion = 0

static

Definition at line 25 of file FLS.c.

Referenced by FlsCallback3(), ok_fls_(), and START_TEST().

PVOID

PVOID

Definition at line 18 of file FLS.c.

Referenced by START_TEST().