apitest_guard.h File Reference

#include <ndk/mmfuncs.h>
#include <ndk/psfuncs.h>

Include dependency graph for apitest_guard.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
static PVOID	AllocateGuarded (_In_ SIZE_T SizeRequested)
static VOID	FreeGuarded (_In_ PVOID Pointer)
static VOID	MakeReadOnly (PVOID Pointer, SIZE_T SizeRequested)

Function Documentation

AllocateGuarded()

static PVOID AllocateGuarded ( _In_ SIZE_T  SizeRequested )

inlinestatic

Definition at line 10 of file apitest_guard.h.

{
    NTSTATUS Status;
    SIZE_T Size = PAGE_ROUND_UP(SizeRequested + PAGE_SIZE);
    PVOID VirtualMemory = NULL;
    PCHAR StartOfBuffer;
 
    Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_RESERVE, PAGE_NOACCESS);
 
    if (!NT_SUCCESS(Status))
        return NULL;
 
    Size -= PAGE_SIZE;
    if (Size)
    {
        Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_COMMIT, PAGE_READWRITE);
        if (!NT_SUCCESS(Status))
        {
            Size = 0;
            Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
            ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
            return NULL;
        }
    }
 
    StartOfBuffer = VirtualMemory;
    StartOfBuffer += Size - SizeRequested;
 
    return StartOfBuffer;
}

Referenced by MakeAcl(), START_TEST(), Test_RtlAreBitsClear(), Test_RtlAreBitsSet(), Test_RtlClearAllBits(), Test_RtlClearBits(), Test_RtlFindClearBits(), Test_RtlFindClearBitsAndSet(), Test_RtlFindNextForwardRunClear(), Test_RtlFindSetBits(), Test_RtlFindSetBitsAndClear(), Test_RtlNumberOfClearBits(), Test_RtlNumberOfSetBits(), Test_RtlSetAllBits(), Test_RtlSetBits(), TestGetUserObjectInfoA(), and TestGetUserObjectInfoW().

FreeGuarded()

static VOID FreeGuarded ( _In_ PVOID  Pointer )

inlinestatic

Definition at line 45 of file apitest_guard.h.

{
    NTSTATUS Status;
    PVOID VirtualMemory = (PVOID)PAGE_ROUND_DOWN((SIZE_T)Pointer);
    SIZE_T Size = 0;
 
    Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
    ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
}

Referenced by START_TEST(), Test_RtlAreBitsClear(), Test_RtlAreBitsSet(), Test_RtlClearAllBits(), Test_RtlClearBits(), Test_RtlFindClearBits(), Test_RtlFindClearBitsAndSet(), Test_RtlFindNextForwardRunClear(), Test_RtlFindSetBits(), Test_RtlFindSetBitsAndClear(), Test_RtlNumberOfClearBits(), Test_RtlNumberOfSetBits(), Test_RtlSetAllBits(), Test_RtlSetBits(), TestGetUserObjectInfoA(), and TestGetUserObjectInfoW().

MakeReadOnly()

static VOID MakeReadOnly ( PVOID  Pointer, SIZE_T  SizeRequested  )

inlinestatic

Definition at line 59 of file apitest_guard.h.

{
    NTSTATUS Status;
    SIZE_T Size = PAGE_ROUND_UP(SizeRequested);
    PVOID VirtualMemory = (PVOID)PAGE_ROUND_DOWN((SIZE_T)Pointer);
 
    if (Size)
    {
        Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_COMMIT, PAGE_READWRITE);
        if (!NT_SUCCESS(Status))
        {
            Size = 0;
            Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
            ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
        }
    }
}

Referenced by START_TEST().