Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-02-24 01:15:09 +00:00
Code Issues Projects Releases Packages Wiki Activity

[KMTESTS:EX]

Browse source 
- Add test for ExAllocatePoolWithQuotaTag

svn path=/trunk/; revision=60174
This commit is contained in:
Thomas Faber 2013-09-16 18:15:02 +00:00
parent db00a7c83c
commit 3aad455588
1 changed files with 83 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

83
rostests/kmtests/ntos_ex/ExPools.c
View file

@ -10,6 +10,15 @@
#define NDEBUG
#include <debug.h>
static
LONG
GetRefCount(
_In_ PVOID Object)
{
POBJECT_HEADER Header = OBJECT_TO_OBJECT_HEADER(Object);
return Header->PointerCount;
}
#define TAG_POOLTEST 'tstP'
static VOID PoolsTest(VOID)
@ -172,9 +181,83 @@ TestPoolTags(VOID)
ExFreePoolWithTag(Memory, 'MyTa');
}
static
VOID
TestPoolQuota(VOID)
{
PEPROCESS Process = PsGetCurrentProcess();
PEPROCESS StoredProcess;
PVOID Memory;
LONG InitialRefCount;
LONG RefCount;
NTSTATUS ExceptionStatus;
InitialRefCount = GetRefCount(Process);
/* We get some memory from this function, and it's properly aligned.
* Also, it takes a reference to the process, and releases it on free */
Memory = ExAllocatePoolWithQuotaTag(PagedPool | POOL_QUOTA_FAIL_INSTEAD_OF_RAISE,
sizeof(LIST_ENTRY),
'tQmK');
ok(Memory != NULL, "ExAllocatePoolWithQuotaTag returned NULL\n");
if (!skip(Memory != NULL, "No memory\n"))
{
ok((ULONG_PTR)Memory % sizeof(LIST_ENTRY) == 0,
"Allocation %p is badly aligned\n",
Memory);
RefCount = GetRefCount(Process);
ok_eq_long(RefCount, InitialRefCount + 1);
/* A pointer to the process is found right before the next pool header */
StoredProcess = ((PVOID *)((ULONG_PTR)Memory + 2 * sizeof(LIST_ENTRY)))[-1];
ok_eq_pointer(StoredProcess, Process);
ExFreePoolWithTag(Memory, 'tQmK');
RefCount = GetRefCount(Process);
ok_eq_long(RefCount, InitialRefCount);
}
/* Large allocations are page-aligned, don't reference the process */
Memory = ExAllocatePoolWithQuotaTag(PagedPool | POOL_QUOTA_FAIL_INSTEAD_OF_RAISE,
PAGE_SIZE,
'tQmK');
ok(Memory != NULL, "ExAllocatePoolWithQuotaTag returned NULL\n");
if (!skip(Memory != NULL, "No memory\n"))
{
ok((ULONG_PTR)Memory % PAGE_SIZE == 0,
"Allocation %p is badly aligned\n",
Memory);
RefCount = GetRefCount(Process);
ok_eq_long(RefCount, InitialRefCount);
ExFreePoolWithTag(Memory, 'tQmK');
RefCount = GetRefCount(Process);
ok_eq_long(RefCount, InitialRefCount);
}
/* Function raises by default */
KmtStartSeh()
Memory = ExAllocatePoolWithQuotaTag(PagedPool,
0x7FFFFFFF,
'tQmK');
if (Memory)
ExFreePoolWithTag(Memory, 'tQmK');
KmtEndSeh(STATUS_INSUFFICIENT_RESOURCES);
/* Function returns NULL with POOL_QUOTA_FAIL_INSTEAD_OF_RAISE */
KmtStartSeh()
Memory = ExAllocatePoolWithQuotaTag(PagedPool | POOL_QUOTA_FAIL_INSTEAD_OF_RAISE,
0x7FFFFFFF,
'tQmK');
ok(Memory == NULL, "Successfully got 2GB block: %p\n", Memory);
if (Memory)
ExFreePoolWithTag(Memory, 'tQmK');
KmtEndSeh(STATUS_SUCCESS);
}
START_TEST(ExPools)
{
PoolsTest();
PoolsCorruption();
TestPoolTags();
TestPoolQuota();
}