Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-23 14:31:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

[KMTEST] Improve error output in ZwAllocateVirtualMemory test

Browse source
This commit is contained in:
Timo Kreuzer 2023-05-06 16:23:30 +03:00
parent 7c81367e9b
commit 7a8a3b7e91
2 changed files with 21 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modules/rostests/kmtests/include/kmt_test.h
View file

@ -282,16 +282,22 @@ VOID KmtFreeGuarded(PVOID Pointer);
_SEH2_TRY \
{
#define KmtEndSeh(ExpectedStatus) \
#define KmtEndSeh_(Line, ExpectedStatus) \
} \
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) \
{ \
ExceptionStatus = _SEH2_GetExceptionCode(); \
} \
_SEH2_END; \
ok_eq_hex(ExceptionStatus, (ExpectedStatus)); \
KmtOk((ExceptionStatus) == (ExpectedStatus), \
__FILE__ ":" #Line ":", \
"ExceptionStatus = 0x%08lx, expected " \
#ExpectedStatus " (0x%08lx)\n", \
ExceptionStatus, ExpectedStatus); \
}
#define KmtEndSeh(ExpectedStatus) KmtEndSeh_(__LINE__, ExpectedStatus)
#define KmtGetSystemOrEmbeddedRoutineAddress(RoutineName) \
p##RoutineName = KmtGetSystemRoutineAddress(L ## #RoutineName); \
if (!p##RoutineName) \

20
modules/rostests/kmtests/ntos_mm/ZwAllocateVirtualMemory.c
View file

@ -39,8 +39,8 @@ typedef struct _TEST_CONTEXT
else if (!NT_SUCCESS(Status)) \
ok_eq_pointer(BaseAddress, NULL); \
RegionSize = 0; \
Status = ZwFreeVirtualMemory(ProcessHandle, &BaseAddress, &RegionSize, MEM_RELEASE); \
if (FreeStatus != IGNORE) ok_eq_hex(Status, FreeStatus); \
Status2 = ZwFreeVirtualMemory(ProcessHandle, &BaseAddress, &RegionSize, MEM_RELEASE); \
if (FreeStatus != IGNORE) ok_eq_hex(Status2, FreeStatus); \
BaseAddress = NULL; \
RegionSize = DEFAULT_ALLOC_SIZE; \
} while (0) \
@ -67,38 +67,44 @@ CheckBuffer(PVOID Buffer, SIZE_T Size, UCHAR Value)
static
SIZE_T
CheckBufferRead(CONST VOID *Source, CONST VOID *Destination, SIZE_T Length, NTSTATUS ExpectedStatus)
CheckBufferRead_(ULONG Line, CONST VOID *Source, CONST VOID *Destination, SIZE_T Length, NTSTATUS ExpectedStatus)
{
SIZE_T Match = 0;
KmtStartSeh()
Match = RtlCompareMemory(Source, Destination, Length);
KmtEndSeh(ExpectedStatus);
KmtEndSeh_(Line, ExpectedStatus);
return Match;
}
#define CheckBufferRead(Source, Destination, Length, ExpectedStatus) \
CheckBufferRead_(__LINE__, Source, Destination, Length, ExpectedStatus)
static
VOID
CheckBufferReadWrite(PVOID Destination, CONST VOID *Source, SIZE_T Length, NTSTATUS ExpectedStatus)
CheckBufferReadWrite_(ULONG Line, PVOID Destination, CONST VOID *Source, SIZE_T Length, NTSTATUS ExpectedStatus)
{
//do a little bit of writing/reading to memory
SIZE_T Match = 0;
KmtStartSeh()
RtlCopyMemory(Destination, Source, Length);
KmtEndSeh(ExpectedStatus);
KmtEndSeh_(Line, ExpectedStatus);
Match = CheckBufferRead(Source, Destination, Length, ExpectedStatus);
if (ExpectedStatus == STATUS_SUCCESS) ok_eq_int(Match, Length);
if (ExpectedStatus == STATUS_SUCCESS && Match != Length) ok_eq_int(Line, 0);
}
#define CheckBufferReadWrite(Destination, Source, Length, ExpectedStatus) \
CheckBufferReadWrite_(__LINE__, Destination, Source, Length, ExpectedStatus)
static
VOID
SimpleErrorChecks(VOID)
{
NTSTATUS Status;
NTSTATUS Status, Status2;
PVOID Base = NULL;
SIZE_T RegionSize = DEFAULT_ALLOC_SIZE;