reactos/rostests/kmtests/example/GuardedMemory.c
Thomas Faber 2b17fa159d [KMTESTS]
- Add support for guarded allocations, that can be used to detect buffer overruns
- Add GuardedMemory example test
- Add MmSection test with some basic parameter checks for MmCreateSection
- Fix some x64 build problems
- Update readme

svn path=/trunk/; revision=53687
2011-09-11 11:22:00 +00:00

64 lines
1.5 KiB
C

/*
* PROJECT: ReactOS kernel-mode tests
* LICENSE: GPLv2+ - See COPYING in the top level directory
* PURPOSE: Kernel-Mode Test Suite Guarded Memory example test
* PROGRAMMER: Thomas Faber <thfabba@gmx.de>
*/
#include <kmt_test.h>
#ifdef __GNUC__
#pragma GCC diagnostic ignored "-Wnonnull"
#endif /* defined __GNUC__ */
START_TEST(GuardedMemory)
{
NTSTATUS Status;
SIZE_T Size = 123;
PCHAR *Buffer;
/* access some invalid memory to test SEH */
Status = STATUS_SUCCESS;
_SEH2_TRY
{
RtlFillMemory(NULL, 1, 0);
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
Status = _SEH2_GetExceptionCode();
} _SEH2_END;
ok_eq_hex(Status, STATUS_ACCESS_VIOLATION);
/* get guarded mem */
Buffer = KmtAllocateGuarded(Size);
if (skip(Buffer != NULL, "Failed to allocate guarded memory\n"))
return;
/* access to guarded mem should be fine */
Status = STATUS_SUCCESS;
_SEH2_TRY
{
RtlFillMemory(Buffer, Size, 0);
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
Status = _SEH2_GetExceptionCode();
} _SEH2_END;
ok_eq_hex(Status, STATUS_SUCCESS);
/* access one byte behind guarded mem must cause an access violation! */
Status = STATUS_SUCCESS;
_SEH2_TRY
{
RtlFillMemory(Buffer + Size, 1, 0);
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
Status = _SEH2_GetExceptionCode();
} _SEH2_END;
ok_eq_hex(Status, STATUS_ACCESS_VIOLATION);
KmtFreeGuarded(Buffer);
}