reactos/modules/rostests/kmtests/example/GuardedMemory.c
Thomas Faber a9ba7a3f9e
[KMTESTS] Change license to LGPL and update to SPDX headers.
Permission for this change was given by all authors
(who are now mentioned in the copyright lines).
2018-02-18 11:22:09 +01:00

63 lines
1.5 KiB
C

/*
* PROJECT: ReactOS kernel-mode tests
* LICENSE: LGPL-2.1+ (https://spdx.org/licenses/LGPL-2.1+)
* PURPOSE: Kernel-Mode Test Suite Guarded Memory example test
* COPYRIGHT: Copyright 2011-2018 Thomas Faber <thomas.faber@reactos.org>
*/
#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);
}