virtmem.c File Reference

#include <k32.h>
#include <debug.h>

Include dependency graph for virtmem.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
LPVOID NTAPI	VirtualAllocEx (IN HANDLE hProcess, IN LPVOID lpAddress, IN SIZE_T dwSize, IN DWORD flAllocationType, IN DWORD flProtect)
LPVOID NTAPI	VirtualAlloc (IN LPVOID lpAddress, IN SIZE_T dwSize, IN DWORD flAllocationType, IN DWORD flProtect)
BOOL NTAPI	VirtualFreeEx (IN HANDLE hProcess, IN LPVOID lpAddress, IN SIZE_T dwSize, IN DWORD dwFreeType)
BOOL NTAPI	VirtualFree (IN LPVOID lpAddress, IN SIZE_T dwSize, IN DWORD dwFreeType)
BOOL NTAPI	VirtualProtect (IN LPVOID lpAddress, IN SIZE_T dwSize, IN DWORD flNewProtect, OUT PDWORD lpflOldProtect)
BOOL NTAPI	VirtualProtectEx (IN HANDLE hProcess, IN LPVOID lpAddress, IN SIZE_T dwSize, IN DWORD flNewProtect, OUT PDWORD lpflOldProtect)
BOOL NTAPI	VirtualLock (IN LPVOID lpAddress, IN SIZE_T dwSize)
SIZE_T NTAPI	VirtualQuery (IN LPCVOID lpAddress, OUT PMEMORY_BASIC_INFORMATION lpBuffer, IN SIZE_T dwLength)
SIZE_T NTAPI	VirtualQueryEx (IN HANDLE hProcess, IN LPCVOID lpAddress, OUT PMEMORY_BASIC_INFORMATION lpBuffer, IN SIZE_T dwLength)
BOOL NTAPI	VirtualUnlock (IN LPVOID lpAddress, IN SIZE_T dwSize)
UINT WINAPI	GetWriteWatch (IN DWORD dwFlags, IN PVOID lpBaseAddress, IN SIZE_T dwRegionSize, IN PVOID *lpAddresses, OUT PULONG_PTR lpdwCount, OUT PULONG lpdwGranularity)
UINT WINAPI	ResetWriteWatch (IN LPVOID lpBaseAddress, IN SIZE_T dwRegionSize)
BOOL WINAPI	AllocateUserPhysicalPages (IN HANDLE hProcess, IN PULONG_PTR NumberOfPages, OUT PULONG_PTR UserPfnArray)
BOOL WINAPI	FreeUserPhysicalPages (IN HANDLE hProcess, IN PULONG_PTR NumberOfPages, IN PULONG_PTR PageArray)
BOOL WINAPI	MapUserPhysicalPages (IN PVOID VirtualAddress, IN ULONG_PTR NumberOfPages, OUT PULONG_PTR PageArray OPTIONAL)
BOOL WINAPI	MapUserPhysicalPagesScatter (IN PVOID *VirtualAddresses, IN ULONG_PTR NumberOfPages, OUT PULONG_PTR PageArray OPTIONAL)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 13 of file virtmem.c.

Function Documentation

AllocateUserPhysicalPages()

BOOL WINAPI AllocateUserPhysicalPages	(	IN HANDLE	hProcess,
		IN PULONG_PTR	NumberOfPages,
		OUT PULONG_PTR	UserPfnArray
	)

Definition at line 338 of file virtmem.c.

{
    NTSTATUS Status;
 
    Status = NtAllocateUserPhysicalPages(hProcess, NumberOfPages, UserPfnArray);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    return TRUE;
}

FreeUserPhysicalPages()

BOOL WINAPI FreeUserPhysicalPages	(	IN HANDLE	hProcess,
		IN PULONG_PTR	NumberOfPages,
		IN PULONG_PTR	PageArray
	)

Definition at line 359 of file virtmem.c.

{
    NTSTATUS Status;
 
    Status = NtFreeUserPhysicalPages(hProcess, NumberOfPages, PageArray);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    return TRUE;
}

GetWriteWatch()

UINT WINAPI GetWriteWatch	(	IN DWORD	dwFlags,
		IN PVOID	lpBaseAddress,
		IN SIZE_T	dwRegionSize,
		IN PVOID *	lpAddresses,
		OUT PULONG_PTR	lpdwCount,
		OUT PULONG	lpdwGranularity
	)

Definition at line 286 of file virtmem.c.

{
    NTSTATUS Status;
 
    Status = NtGetWriteWatch(GetCurrentProcess(),
                             dwFlags,
                             lpBaseAddress,
                             dwRegionSize,
                             lpAddresses,
                             lpdwCount,
                             lpdwGranularity);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return -1;
    }
 
    return 0;
}

MapUserPhysicalPages()

BOOL WINAPI MapUserPhysicalPages	(	IN PVOID	VirtualAddress,
		IN ULONG_PTR	NumberOfPages,
		OUT PULONG_PTR PageArray	OPTIONAL
	)

Definition at line 380 of file virtmem.c.

{
    NTSTATUS Status;
 
    Status = NtMapUserPhysicalPages(VirtualAddress, NumberOfPages, PageArray);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    return TRUE;
}

MapUserPhysicalPagesScatter()

BOOL WINAPI MapUserPhysicalPagesScatter	(	IN PVOID *	VirtualAddresses,
		IN ULONG_PTR	NumberOfPages,
		OUT PULONG_PTR PageArray	OPTIONAL
	)

Definition at line 401 of file virtmem.c.

{
    NTSTATUS Status;
 
    Status = NtMapUserPhysicalPagesScatter(VirtualAddresses,
                                           NumberOfPages,
                                           PageArray);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    return TRUE;
}

ResetWriteWatch()

UINT WINAPI ResetWriteWatch	(	IN LPVOID	lpBaseAddress,
		IN SIZE_T	dwRegionSize
	)

Definition at line 316 of file virtmem.c.

{
    NTSTATUS Status;
 
    Status = NtResetWriteWatch(NtCurrentProcess(),
                               lpBaseAddress,
                               dwRegionSize);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return -1;
    }
 
    return 0;
}

VirtualAlloc()

LPVOID NTAPI VirtualAlloc	(	IN LPVOID	lpAddress,
		IN SIZE_T	dwSize,
		IN DWORD	flAllocationType,
		IN DWORD	flProtect
	)

Definition at line 65 of file virtmem.c.

{
    /* Call the extended API */
    return VirtualAllocEx(GetCurrentProcess(),
                          lpAddress,
                          dwSize,
                          flAllocationType,
                          flProtect);
}

Referenced by allocate_block(), CKsAllocator::Commit(), copy(), ATL::CWndProcThunk::CWndProcThunk(), main(), replace(), setup_thiscall_wrappers(), START_TEST(), StdMemAllocator_Alloc(), test__strupr(), test_CompareStringW(), test_CreateFileMapping_protection(), Test_InstructionFaults(), test_Loader(), test_mapping(), test_MapViewOfFile(), test_NtAreMappedFilesTheSame(), test_query_process_vm(), test_read_write(), test_VirtualAlloc(), test_VirtualAlloc_protection(), test_VirtualAllocEx(), test_VirtualProtect(), test_write_watch(), test_WriteFileGather(), test_ws_functions(), VirtualTest(), wave_in_test_device(), and wave_out_test_device().

VirtualAllocEx()

LPVOID NTAPI VirtualAllocEx	(	IN HANDLE	hProcess,
		IN LPVOID	lpAddress,
		IN SIZE_T	dwSize,
		IN DWORD	flAllocationType,
		IN DWORD	flProtect
	)

Definition at line 23 of file virtmem.c.

{
    NTSTATUS Status;
 
    /* Make sure the address is within the granularity of the system (64K) */
    if ((lpAddress != NULL) &&
        (lpAddress < UlongToPtr(BaseStaticServerData->SysInfo.AllocationGranularity)))
    {
        /* Fail the call */
        SetLastError(ERROR_INVALID_PARAMETER);
        return NULL;
    }
 
    /* Allocate the memory */
    Status = NtAllocateVirtualMemory(hProcess,
                                     &lpAddress,
                                     0,
                                     &dwSize,
                                     flAllocationType,
                                     flProtect);
 
    /* Check for status */
    if (!NT_SUCCESS(Status))
    {
        /* We failed */
        BaseSetLastNTError(Status);
        return NULL;
    }
 
    /* Return the allocated address */
    return lpAddress;
}

Referenced by _DoDLLInjection(), test_ExitProcess(), and VirtualAlloc().

VirtualFree()

BOOL NTAPI VirtualFree	(	IN LPVOID	lpAddress,
		IN SIZE_T	dwSize,
		IN DWORD	dwFreeType
	)

Definition at line 119 of file virtmem.c.

{
    /* Call the extended API */
    return VirtualFreeEx(GetCurrentProcess(),
                         lpAddress,
                         dwSize,
                         dwFreeType);
}

Referenced by allocate_block(), copy(), CKsAllocator::FreeMediaSamples(), main(), replace(), START_TEST(), StdMemAllocator_Free(), test__strupr(), test_CompareStringW(), Test_InstructionFaults(), test_MapViewOfFile(), test_NtAreMappedFilesTheSame(), test_OpenProcess(), test_query_process_vm(), test_read_write(), test_VirtualAlloc(), test_VirtualAlloc_protection(), test_VirtualAllocEx(), test_VirtualProtect(), test_write_watch(), test_WriteFileGather(), test_ws_functions(), wave_in_test_device(), wave_out_test_device(), and ATL::CWndProcThunk::~CWndProcThunk().

VirtualFreeEx()

BOOL NTAPI VirtualFreeEx	(	IN HANDLE	hProcess,
		IN LPVOID	lpAddress,
		IN SIZE_T	dwSize,
		IN DWORD	dwFreeType
	)

Definition at line 83 of file virtmem.c.

{
    NTSTATUS Status;
 
    /* Validate size and flags */
    if (!(dwSize) || !(dwFreeType & MEM_RELEASE))
    {
        /* Free the memory */
        Status = NtFreeVirtualMemory(hProcess,
                                     &lpAddress,
                                     &dwSize,
                                     dwFreeType);
        if (!NT_SUCCESS(Status))
        {
            /* We failed */
            BaseSetLastNTError(Status);
            return FALSE;
        }
 
        /* Return success */
        return TRUE;
    }
 
    /* Invalid combo */
    BaseSetLastNTError(STATUS_INVALID_PARAMETER);
    return FALSE;
}

Referenced by test_ExitProcess(), and VirtualFree().

VirtualLock()

BOOL NTAPI VirtualLock	(	IN LPVOID	lpAddress,
		IN SIZE_T	dwSize
	)

Definition at line 183 of file virtmem.c.

{
    NTSTATUS Status;
    SIZE_T RegionSize = dwSize;
    PVOID BaseAddress = lpAddress;
 
    /* Lock the memory */
    Status = NtLockVirtualMemory(NtCurrentProcess(),
                                 &BaseAddress,
                                 &RegionSize,
                                 MAP_PROCESS);
    if (!NT_SUCCESS(Status))
    {
        /* We failed */
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    /* Return success */
    return TRUE;
}

Referenced by test_ws_functions().

VirtualProtect()

BOOL NTAPI VirtualProtect	(	IN LPVOID	lpAddress,
		IN SIZE_T	dwSize,
		IN DWORD	flNewProtect,
		OUT PDWORD	lpflOldProtect
	)

Definition at line 135 of file virtmem.c.

{
    /* Call the extended API */
    return VirtualProtectEx(GetCurrentProcess(),
                            lpAddress,
                            dwSize,
                            flNewProtect,
                            lpflOldProtect);
}

Referenced by __write_memory(), _resetstkoflw(), allocate_block(), hex2mem(), hook_WaitForInputIdle(), main(), RedirectIat(), test__strupr(), test_CompareStringW(), test_CreateFileMapping_protection(), test_mapping(), test_read_write(), test_VirtualAlloc(), test_VirtualAllocEx(), test_VirtualProtect(), test_write_watch(), UnProtect::UnProtect(), wave_in_test_device(), wave_out_test_device(), and UnProtect::~UnProtect().

VirtualProtectEx()

BOOL NTAPI VirtualProtectEx	(	IN HANDLE	hProcess,
		IN LPVOID	lpAddress,
		IN SIZE_T	dwSize,
		IN DWORD	flNewProtect,
		OUT PDWORD	lpflOldProtect
	)

Definition at line 153 of file virtmem.c.

{
    NTSTATUS Status;
 
    /* Change the protection */
    Status = NtProtectVirtualMemory(hProcess,
                                    &lpAddress,
                                    &dwSize,
                                    flNewProtect,
                                    (PULONG)lpflOldProtect);
    if (!NT_SUCCESS(Status))
    {
        /* We failed */
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    /* Return success */
    return TRUE;
}

Referenced by test_ExitProcess(), test_VirtualAllocEx(), and VirtualProtect().

VirtualQuery()

SIZE_T NTAPI VirtualQuery	(	IN LPCVOID	lpAddress,
		OUT PMEMORY_BASIC_INFORMATION	lpBuffer,
		IN SIZE_T	dwLength
	)

Definition at line 211 of file virtmem.c.

{
    /* Call the extended API */
    return VirtualQueryEx(NtCurrentProcess(),
                          lpAddress,
                          lpBuffer,
                          dwLength);
}

Referenced by __write_memory(), _module_name_from_addr(), create_stream_from_map(), KmtUserCallbackThread(), test_CreateFileMapping_protection(), test_dib_bits_access(), test_dibsections(), test_image_mapping(), test_IME(), test_Loader(), test_mapping(), test_MapViewOfFile(), test_section_access(), test_VirtualAlloc(), test_VirtualAlloc_protection(), test_VirtualProtect(), and test_write_watch().

VirtualQueryEx()

SIZE_T NTAPI VirtualQueryEx	(	IN HANDLE	hProcess,
		IN LPCVOID	lpAddress,
		OUT PMEMORY_BASIC_INFORMATION	lpBuffer,
		IN SIZE_T	dwLength
	)

Definition at line 227 of file virtmem.c.

{
    NTSTATUS Status;
    SIZE_T ResultLength;
 
    /* Query basic information */
    Status = NtQueryVirtualMemory(hProcess,
                                  (LPVOID)lpAddress,
                                  MemoryBasicInformation,
                                  lpBuffer,
                                  dwLength,
                                  &ResultLength);
    if (!NT_SUCCESS(Status))
    {
        /* We failed */
        BaseSetLastNTError(Status);
        return 0;
    }
 
    /* Return the length returned */
    return ResultLength;
}

Referenced by fetch_memory64_info(), get_shiminfo(), test_ExitProcess(), test_OpenProcess(), test_section_access(), test_VirtualAllocEx(), and VirtualQuery().

VirtualUnlock()

BOOL NTAPI VirtualUnlock	(	IN LPVOID	lpAddress,
		IN SIZE_T	dwSize
	)

Definition at line 258 of file virtmem.c.

{
    NTSTATUS Status;
    SIZE_T RegionSize = dwSize;
    PVOID BaseAddress = lpAddress;
 
    /* Lock the memory */
    Status = NtUnlockVirtualMemory(NtCurrentProcess(),
                                   &BaseAddress,
                                   &RegionSize,
                                   MAP_PROCESS);
    if (!NT_SUCCESS(Status))
    {
        /* We failed */
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    /* Return success */
    return TRUE;
}