/*
 * 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);
}