RtlInitializeNtUserPfn.c File Reference

#include "precomp.h"

Include dependency graph for RtlInitializeNtUserPfn.c:

Go to the source code of this file.

Typedefs
typedef NTSTATUS NTAPI	FN_RtlInitializeNtUserPfn(PVOID *ProcsA, SIZE_T SizeOfProcsA, PVOID *ProcsW, SIZE_T SizeOfProcsW, PVOID *OtherProcs, SIZE_T SizeOfOtherProcs)

Functions
	START_TEST (RtlInitializeNtUserPfn)

Variables
FN_RtlInitializeNtUserPfn *	pRtlInitializeNtUserPfn

Typedef Documentation

FN_RtlInitializeNtUserPfn

typedef NTSTATUS NTAPI FN_RtlInitializeNtUserPfn(PVOID *ProcsA, SIZE_T SizeOfProcsA, PVOID *ProcsW, SIZE_T SizeOfProcsW, PVOID *OtherProcs, SIZE_T SizeOfOtherProcs)

Definition at line 13 of file RtlInitializeNtUserPfn.c.

Function Documentation

START_TEST()

START_TEST

(

RtlInitializeNtUserPfn

)

Definition at line 23 of file RtlInitializeNtUserPfn.c.

{
    HINSTANCE hinstNtdll = GetModuleHandleW(L"ntdll.dll");
    pRtlInitializeNtUserPfn = (FN_RtlInitializeNtUserPfn*)GetProcAddress(hinstNtdll, "RtlInitializeNtUserPfn");
    if (pRtlInitializeNtUserPfn == NULL)
    {
        skip("RtlInitializeNtUserPfn is not available\n");
        return;
    }
 
    NTSTATUS Status;
    PVOID ProcsA[24] = { 0 };
    PVOID ProcsW[24] = { 0 };
    PVOID OtherProcs[11] = { 0 };
    ULONG Count1, Count2;
 
    Status = pRtlInitializeNtUserPfn(NULL, 1, NULL, 0, NULL, 0);
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(NULL, 0, NULL, 1, NULL, 0);
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(NULL, 0, NULL, 0, NULL, 1);
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(ProcsA, 0, ProcsW, 0, OtherProcs, 1);
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
 
    printf("NTDDI version: %lx\n", GetNTDDIVersion());
 
    /* OS version specific count of A/W functions */
    Count1 = (GetNTDDIVersion() >= NTDDI_WIN10) ? 24 : 23;
 
    /* OS version specific count of other functions */
    Count2 = (GetNTDDIVersion() >= NTDDI_WIN11) ? 18 : 11;
 
    /* Try different variants of too large sizes */
    Status = pRtlInitializeNtUserPfn(ProcsA, (Count1 + 1) * sizeof(PVOID), ProcsW, Count1 * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, (Count1 + 1) * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(ProcsA, (Count1 + 1) * sizeof(PVOID), ProcsW, (Count1 + 1) * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, Count1 * sizeof(PVOID), OtherProcs, (Count2 + 1) * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
 
    if (GetNTVersion() < _WIN32_WINNT_WIN10)
    {
        /* Try different variants of too small sizes (these succeed on Windows 10) */
        Status = pRtlInitializeNtUserPfn(ProcsA, (Count1 - 1) * sizeof(PVOID), ProcsW, Count1 * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
        ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
        Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, (Count1 - 1) * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
        ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
        Status = pRtlInitializeNtUserPfn(ProcsA, (Count1 - 1) * sizeof(PVOID), ProcsW, (Count1 - 1) * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
        ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
        Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, Count1 * sizeof(PVOID), OtherProcs, (Count2 - 1) * sizeof(PVOID));
        ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    }
 
    /* Try unaligned sizes */
    Status = pRtlInitializeNtUserPfn(ProcsA, (Count1 * sizeof(PVOID)) - 1, ProcsW, Count1 * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, (Count1 * sizeof(PVOID)) - 1, OtherProcs, Count2 * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
    Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, Count1 * sizeof(PVOID), OtherProcs, (Count2 * sizeof(PVOID)) - 1);
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
 
    /* Pass the expected sizes */
    Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, Count1 * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_SUCCESS);
 
    /* Try to initialize again */
    Status = pRtlInitializeNtUserPfn(ProcsA, Count1 * sizeof(PVOID), ProcsW, Count1 * sizeof(PVOID), OtherProcs, Count2 * sizeof(PVOID));
    ok_ntstatus(Status, STATUS_INVALID_PARAMETER);
}

Variable Documentation

pRtlInitializeNtUserPfn

FN_RtlInitializeNtUserPfn* pRtlInitializeNtUserPfn

Definition at line 21 of file RtlInitializeNtUserPfn.c.

Referenced by InitFunctionPtrs(), START_TEST(), and test_user_procs().