VirtualMemory.c File Reference

#include <ntoskrnl.h>
#include "regtests.h"

Include dependency graph for VirtualMemory.c:

Go to the source code of this file.

Macros
#define	TestProcessHandle   (HANDLE) 1
#define	TestProcessObject   (PVOID) 0x2
#define	TestBaseAddress   (PVOID) 0x1000
#define	TestNumberOfBytesToLock   0x2000
#define	TestMdl   (PMDL) 0xD0000000

Functions
static VOID NTAPI	MockExFreePool (PVOID Block)
static PMDL NTAPI	MockMmCreateMdl (PMDL Mdl, PVOID Base, ULONG Length)
static VOID NTAPI	MockMmProbeAndLockPages (PMDL Mdl, KPROCESSOR_MODE AccessMode, LOCK_OPERATION Operation)
static VOID FASTCALL	MockObDereferenceObject (PVOID Object)
static NTSTATUS NTAPI	MockObReferenceObjectByHandle (HANDLE Handle, ACCESS_MASK DesiredAccess, POBJECT_TYPE ObjectType, KPROCESSOR_MODE AccessMode, PVOID *Object, POBJECT_HANDLE_INFORMATION HandleInformation)
static void	RunTest ()

Variables
static BOOLEAN	MockExFreePoolCalled = FALSE
static BOOLEAN	MockMmCreateMdlCalled = FALSE
static BOOLEAN	MockMmProbeAndLockPagesCalled = FALSE
static BOOLEAN	MockObDereferenceObjectCalled = FALSE
static BOOLEAN	MockObReferenceObjectByHandleCalled = FALSE

Macro Definition Documentation

TestBaseAddress

#define TestBaseAddress   (PVOID) 0x1000

Definition at line 15 of file VirtualMemory.c.

TestMdl

#define TestMdl   (PMDL) 0xD0000000

Definition at line 17 of file VirtualMemory.c.

TestNumberOfBytesToLock

#define TestNumberOfBytesToLock   0x2000

Definition at line 16 of file VirtualMemory.c.

TestProcessHandle

#define TestProcessHandle   (HANDLE) 1

Definition at line 13 of file VirtualMemory.c.

TestProcessObject

#define TestProcessObject   (PVOID) 0x2

Definition at line 14 of file VirtualMemory.c.

Function Documentation

MockExFreePool()

static VOID NTAPI MockExFreePool ( PVOID  Block )

static

Definition at line 22 of file VirtualMemory.c.

{
  _AssertFalse(MockExFreePoolCalled);
  _AssertEqualValue(TestMdl, Block);
  MockExFreePoolCalled = TRUE;
}

Referenced by RunTest().

MockMmCreateMdl()

static PMDL NTAPI MockMmCreateMdl ( PMDL  Mdl, PVOID  Base, ULONG  Length  )

static

Definition at line 32 of file VirtualMemory.c.

{
  _AssertFalse(MockMmCreateMdlCalled);
  _AssertEqualValue(TestBaseAddress, Base);
  _AssertEqualValue(TestNumberOfBytesToLock, Length);
  MockMmCreateMdlCalled = TRUE;
  return TestMdl;
}

Referenced by RunTest().

MockMmProbeAndLockPages()

static VOID NTAPI MockMmProbeAndLockPages ( PMDL  Mdl, KPROCESSOR_MODE  AccessMode, LOCK_OPERATION  Operation  )

static

Definition at line 46 of file VirtualMemory.c.

{
  _AssertFalse(MockMmProbeAndLockPagesCalled);
  _AssertEqualValue(TestMdl, Mdl);
  _AssertEqualValue(UserMode, AccessMode);
  _AssertEqualValue(IoWriteAccess, Operation);
  MockMmProbeAndLockPagesCalled = TRUE;
}

Referenced by RunTest().

MockObDereferenceObject()

static VOID FASTCALL MockObDereferenceObject ( PVOID  Object )

static

Definition at line 60 of file VirtualMemory.c.

{
  _AssertFalse(MockObDereferenceObjectCalled);
  _AssertEqualValue(TestProcessObject, Object);
  MockObDereferenceObjectCalled = TRUE;
}

Referenced by RunTest().

MockObReferenceObjectByHandle()

static NTSTATUS NTAPI MockObReferenceObjectByHandle ( HANDLE  Handle, ACCESS_MASK  DesiredAccess, POBJECT_TYPE  ObjectType, KPROCESSOR_MODE  AccessMode, PVOID *  Object, POBJECT_HANDLE_INFORMATION  HandleInformation  )

static

Definition at line 70 of file VirtualMemory.c.

{
  _AssertFalse(MockObReferenceObjectByHandleCalled);
  _AssertEqualValue(TestProcessHandle, Handle);
  _AssertEqualValue(PROCESS_VM_WRITE, DesiredAccess);
  _AssertEqualValue(NULL, ObjectType);
  _AssertEqualValue(UserMode, AccessMode);
  _AssertNotEqualValue(NULL, Object);
  _AssertEqualValue(NULL, HandleInformation);
  *Object = TestProcessObject;
  MockObReferenceObjectByHandleCalled = TRUE;
  return STATUS_SUCCESS;
}

Referenced by RunTest().

RunTest()

static void RunTest ( )

static

Definition at line 89 of file VirtualMemory.c.

{
  ULONG NumberOfBytesLocked;
  NTSTATUS status;
 
  status = MiLockVirtualMemory(TestProcessHandle,
    TestBaseAddress,
    TestNumberOfBytesToLock,
    &NumberOfBytesLocked,
    MockObReferenceObjectByHandle,
    MockMmCreateMdl,
    MockObDereferenceObject,
    MockMmProbeAndLockPages,
    MockExFreePool);
  _AssertEqualValue(STATUS_SUCCESS, status);
  _AssertTrue(MockObReferenceObjectByHandleCalled);
  _AssertTrue(MockMmCreateMdlCalled);
  _AssertTrue(MockMmProbeAndLockPagesCalled);
  _AssertTrue(MockExFreePoolCalled);
  _AssertTrue(MockObDereferenceObjectCalled);
  _AssertEqualValue(TestNumberOfBytesToLock, NumberOfBytesLocked);
}

Variable Documentation

MockExFreePoolCalled

BOOLEAN MockExFreePoolCalled = FALSE

static

Definition at line 19 of file VirtualMemory.c.

Referenced by MockExFreePool(), and RunTest().

MockMmCreateMdlCalled

BOOLEAN MockMmCreateMdlCalled = FALSE

static

Definition at line 29 of file VirtualMemory.c.

Referenced by MockMmCreateMdl(), and RunTest().

MockMmProbeAndLockPagesCalled

BOOLEAN MockMmProbeAndLockPagesCalled = FALSE

static

Definition at line 43 of file VirtualMemory.c.

Referenced by MockMmProbeAndLockPages(), and RunTest().

MockObDereferenceObjectCalled

BOOLEAN MockObDereferenceObjectCalled = FALSE

static

Definition at line 57 of file VirtualMemory.c.

Referenced by MockObDereferenceObject(), and RunTest().

MockObReferenceObjectByHandleCalled

BOOLEAN MockObReferenceObjectByHandleCalled = FALSE

static

Definition at line 67 of file VirtualMemory.c.

Referenced by MockObReferenceObjectByHandle(), and RunTest().