lywsvip/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2026-07-06 18:54:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

[KMTESTS:EX]

Browse Source 
- Test that quota allocations keep their QUOTA_POOL_MASK bit in POOL_HEADER

svn path=/trunk/; revision=60175
This commit is contained in:
Thomas Faber
2013-09-16 19:02:28 +00:00
parent 3aad455588
commit 4dcd167f3e
2 changed files with 28 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
rostests/kmtests/include/kmt_test.h
View File

@@ -131,6 +131,7 @@ extern PDRIVER_OBJECT KmtDriverObject;
VOID KmtSetIrql(IN KIRQL NewIrql);
BOOLEAN KmtAreInterruptsEnabled(VOID);
ULONG KmtGetPoolTag(PVOID Memory);
USHORT KmtGetPoolType(PVOID Memory);
#elif defined KMT_USER_MODE
DWORD KmtRunKernelTest(IN PCSTR TestName);
@@ -334,6 +335,20 @@ ULONG KmtGetPoolTag(PVOID Memory)
return Header->PoolTag;
}
USHORT KmtGetPoolType(PVOID Memory)
{
PPOOL_HEADER Header;
/* it's not so easy for allocations of PAGE_SIZE */
if (((ULONG_PTR)Memory & (PAGE_SIZE - 1)) == 0)
return 0;
Header = Memory;
Header--;
return Header->PoolType;
}
INT __cdecl KmtVSNPrintF(PSTR Buffer, SIZE_T BufferMaxLength, PCSTR Format, va_list Arguments) KMT_FORMAT(ms_printf, 3, 0);
#elif defined KMT_USER_MODE
static PKMT_RESULTBUFFER KmtAllocateResultBuffer(SIZE_T ResultBufferSize)

15
rostests/kmtests/ntos_ex/ExPools.c
View File

@@ -10,6 +10,11 @@
#define NDEBUG
#include <debug.h>
#define TAG_POOLTEST 'tstP'
#define BASE_POOL_TYPE_MASK 1
#define QUOTA_POOL_MASK 8
static
LONG
GetRefCount(
@@ -19,8 +24,6 @@ GetRefCount(
return Header->PointerCount;
}
#define TAG_POOLTEST 'tstP'
static VOID PoolsTest(VOID)
{
PVOID Ptr;
@@ -190,6 +193,7 @@ TestPoolQuota(VOID)
PVOID Memory;
LONG InitialRefCount;
LONG RefCount;
USHORT PoolType;
NTSTATUS ExceptionStatus;
InitialRefCount = GetRefCount(Process);
@@ -212,6 +216,13 @@ TestPoolQuota(VOID)
StoredProcess = ((PVOID *)((ULONG_PTR)Memory + 2 * sizeof(LIST_ENTRY)))[-1];
ok_eq_pointer(StoredProcess, Process);
/* Pool type should have QUOTA_POOL_MASK set */
PoolType = KmtGetPoolType(Memory);
ok(PoolType != 0, "PoolType is 0\n");
PoolType--;
ok(PoolType & QUOTA_POOL_MASK, "PoolType = %x\n", PoolType);
ok((PoolType & BASE_POOL_TYPE_MASK) == PagedPool, "PoolType = %x\n", PoolType);
ExFreePoolWithTag(Memory, 'tQmK');
RefCount = GetRefCount(Process);
ok_eq_long(RefCount, InitialRefCount);