MmReservedMapping.c File Reference

#include <kmt_test.h>

Include dependency graph for MmReservedMapping.c:

Go to the source code of this file.

Functions
static _Must_inspect_result_	_IRQL_requires_max_ (DISPATCH_LEVEL)
static VOID	TestMap (_In_ PVOID Mapping, _In_ ULONG TotalPtes, _In_ ULONG PoolTag)
	START_TEST (MmReservedMapping)

Variables
static BOOLEAN	g_IsPae

Function Documentation

_IRQL_requires_max_()

static _Must_inspect_result_ _IRQL_requires_max_ ( DISPATCH_LEVEL  )

static

Definition at line 56 of file MmReservedMapping.c.

{
    BOOLEAN Valid = TRUE;
#if defined(_M_IX86) || defined(_M_AMD64)
    PUCHAR CurrentAddress;
    ULONGLONG PteValue;
    ULONG i;
 
    for (i = 0; i < ValidPtes; i++)
    {
        CurrentAddress = (PUCHAR)BaseAddress + i * PAGE_SIZE;
        PteValue = GET_PTE_VALUE(CurrentAddress);
        Valid = Valid &&
                ok(PTE_IS_VALID(PteValue),
                   "[%lu] PTE for %p is not valid (0x%I64x)\n",
                   i, CurrentAddress, PteValue);
 
        Valid = Valid &&
                ok(PTE_GET_PFN(PteValue) == Pfns[i],
                   "[%lu] PTE for %p has PFN %Ix, expected %Ix\n",
                   i, CurrentAddress, PTE_GET_PFN(PteValue), Pfns[i]);
    }
    for (; i < TotalPtes; i++)
    {
        CurrentAddress = (PUCHAR)BaseAddress + i * PAGE_SIZE;
        PteValue = GET_PTE_VALUE(CurrentAddress);
        Valid = Valid &&
                ok(PteValue == 0,
                   "[%lu] PTE for %p is nonzero (0x%I64x)\n",
                   i, CurrentAddress, PteValue);
    }
    CurrentAddress = (PUCHAR)BaseAddress - 1 * PAGE_SIZE;
    PteValue = GET_PTE_VALUE(CurrentAddress);
    Valid = Valid &&
            ok(PteValue == (PoolTag & ~1ULL),
               "PTE for %p contains 0x%I64x, expected %x\n",
               CurrentAddress, PteValue, PoolTag & ~1);
    CurrentAddress = (PUCHAR)BaseAddress - 2 * PAGE_SIZE;
    PteValue = GET_PTE_VALUE(CurrentAddress);
    Valid = Valid &&
            ok(PteValue == (TotalPtes + 2) * 2,
               "PTE for %p contains 0x%I64x, expected %x\n",
               CurrentAddress, PteValue, (TotalPtes + 2) * 2);
#endif
 
    return Valid;
}

START_TEST()

START_TEST

(

MmReservedMapping

)

Definition at line 279 of file MmReservedMapping.c.

{
    PVOID Mapping;
 
    g_IsPae = ExIsProcessorFeaturePresent(PF_PAE_ENABLED);
 
    pMmAllocatePagesForMdlEx = KmtGetSystemRoutineAddress(L"MmAllocatePagesForMdlEx");
 
    /* one byte - single page */
    Mapping = MmAllocateMappingAddress(1, 'MRmK');
    ok(Mapping != NULL, "MmAllocateMappingAddress failed\n");
    if (!skip(Mapping != NULL, "No mapping\n"))
    {
        ok_bool_true(ValidateMapping(Mapping, 1, 'MRmK', 0, NULL),
                     "ValidateMapping returned");
 
        MmFreeMappingAddress(Mapping, 'MRmK');
    }
 
    /* 10 pages */
    Mapping = MmAllocateMappingAddress(10 * PAGE_SIZE, 'MRmK' & ~1);
    ok(Mapping != NULL, "MmAllocateMappingAddress failed\n");
    if (!skip(Mapping != NULL, "No mapping\n"))
    {
        ok_bool_true(ValidateMapping(Mapping, 10, 'MRmK', 0, NULL),
                     "ValidateMapping returned");
 
        /* PAGE_FAULT_IN_NONPAGED_AREA can't be caught with SEH */
        if (0)
        {
            (void)*(volatile UCHAR *)Mapping;
        }
 
        TestMap(Mapping, 10, 'MRmK');
 
        MmFreeMappingAddress(Mapping, 'MRmK');
    }
 
    /* PoolTag = 0 */
    Mapping = MmAllocateMappingAddress(1, 0);
    ok(Mapping == NULL, "MmAllocateMappingAddress failed\n");
    if (Mapping != NULL)
    {
        MmFreeMappingAddress(Mapping, 0);
    }
 
    /* PoolTag = 1 */
    Mapping = MmAllocateMappingAddress(1, 1);
    ok(Mapping != NULL, "MmAllocateMappingAddress failed\n");
    if (Mapping != NULL)
    {
        ok_bool_true(ValidateMapping(Mapping, 1, 1, 0, NULL),
                     "ValidateMapping returned");
 
        TestMap(Mapping, 1, 1);
 
        MmFreeMappingAddress(Mapping, 1);
    }
 
    /* Free an unaligned address */
    Mapping = MmAllocateMappingAddress(PAGE_SIZE, 'MRmK');
    ok(Mapping != NULL, "MmAllocateMappingAddress failed\n");
    if (Mapping != NULL)
    {
        ok_bool_true(ValidateMapping(Mapping, 1, 'MRmK', 0, NULL),
                     "ValidateMapping returned");
 
        TestMap(Mapping, 1, 'MRmK');
 
        MmFreeMappingAddress((PUCHAR)Mapping + sizeof(ULONG), 'MRmK');
    }
}

TestMap()

static VOID TestMap ( _In_ PVOID  Mapping, _In_ ULONG  TotalPtes, _In_ ULONG  PoolTag  )

static

Definition at line 124 of file MmReservedMapping.c.

{
    PMDL Mdl;
    PHYSICAL_ADDRESS ZeroPhysical;
    PHYSICAL_ADDRESS MaxPhysical;
    PVOID BaseAddress;
    PPFN_NUMBER MdlPages;
    ULONG i;
 
    if (skip(pMmAllocatePagesForMdlEx != NULL, "MmAllocatePagesForMdlEx unavailable\n"))
    {
        return;
    }
 
    ZeroPhysical.QuadPart = 0;
    MaxPhysical.QuadPart = 0xffffffffffffffffLL;
 
    /* Create a one-page MDL and map it */
    Mdl = pMmAllocatePagesForMdlEx(ZeroPhysical,
                                   MaxPhysical,
                                   ZeroPhysical,
                                   PAGE_SIZE,
                                   MmCached,
                                   0);
    if (skip(Mdl != NULL, "No MDL\n"))
    {
        return;
    }
 
    MdlPages = (PVOID)(Mdl + 1);
 
    BaseAddress = MmMapLockedPagesWithReservedMapping(Mapping,
                                                      PoolTag,
                                                      Mdl,
                                                      MmCached);
    ok(BaseAddress != NULL, "MmMapLockedPagesWithReservedMapping failed\n");
    if (!skip(BaseAddress != NULL, "Failed to map MDL\n"))
    {
        ok_eq_pointer(BaseAddress, Mapping);
 
        ok_bool_true(ValidateMapping(BaseAddress, TotalPtes, PoolTag, 1, MdlPages),
                     "ValidateMapping returned");
 
        KmtStartSeh()
            *(volatile ULONG *)BaseAddress = 0x01234567;
        KmtEndSeh(STATUS_SUCCESS);
 
        MmUnmapReservedMapping(BaseAddress,
                               PoolTag,
                               Mdl);
 
        ok_bool_true(ValidateMapping(Mapping, TotalPtes, PoolTag, 0, NULL),
                     "ValidateMapping returned");
    }
 
    /* Try again but at an unaligned address */
    BaseAddress = MmMapLockedPagesWithReservedMapping((PUCHAR)Mapping + sizeof(ULONG),
                                                      PoolTag,
                                                      Mdl,
                                                      MmCached);
    ok(BaseAddress != NULL, "MmMapLockedPagesWithReservedMapping failed\n");
    if (!skip(BaseAddress != NULL, "Failed to map MDL\n"))
    {
        ok_eq_pointer(BaseAddress, (PUCHAR)Mapping + sizeof(ULONG));
 
        ok_bool_true(ValidateMapping(BaseAddress, TotalPtes, PoolTag, 1, MdlPages),
                     "ValidateMapping returned");
 
        KmtStartSeh()
            *(volatile ULONG *)BaseAddress = 0x01234567;
        KmtEndSeh(STATUS_SUCCESS);
 
        MmUnmapReservedMapping(BaseAddress,
                               PoolTag,
                               Mdl);
 
        ok_bool_true(ValidateMapping(Mapping, TotalPtes, PoolTag, 0, NULL),
                     "ValidateMapping returned");
    }
 
    MmFreePagesFromMdl(Mdl);
 
    /* Map all pages */
    Mdl = pMmAllocatePagesForMdlEx(ZeroPhysical,
                                   MaxPhysical,
                                   ZeroPhysical,
                                   TotalPtes * PAGE_SIZE,
                                   MmCached,
                                   0);
    if (skip(Mdl != NULL, "No MDL\n"))
    {
        return;
    }
 
    MdlPages = (PVOID)(Mdl + 1);
 
    BaseAddress = MmMapLockedPagesWithReservedMapping(Mapping,
                                                      PoolTag,
                                                      Mdl,
                                                      MmCached);
    ok(BaseAddress != NULL, "MmMapLockedPagesWithReservedMapping failed\n");
    if (!skip(BaseAddress != NULL, "Failed to map MDL\n"))
    {
        ok_eq_pointer(BaseAddress, Mapping);
 
        ok_bool_true(ValidateMapping(BaseAddress, TotalPtes, PoolTag, TotalPtes, MdlPages),
                     "ValidateMapping returned");
 
        for (i = 0; i < TotalPtes; i++)
        {
            KmtStartSeh()
                *((volatile ULONG *)BaseAddress + i * PAGE_SIZE / sizeof(ULONG)) = 0x01234567;
            KmtEndSeh(STATUS_SUCCESS);
        }
 
        MmUnmapReservedMapping(BaseAddress,
                               PoolTag,
                               Mdl);
 
        ok_bool_true(ValidateMapping(Mapping, TotalPtes, PoolTag, 0, NULL),
                     "ValidateMapping returned");
    }
 
    MmFreePagesFromMdl(Mdl);
 
    /* Try to map more pages than we reserved */
    Mdl = pMmAllocatePagesForMdlEx(ZeroPhysical,
                                   MaxPhysical,
                                   ZeroPhysical,
                                   (TotalPtes + 1) * PAGE_SIZE,
                                   MmCached,
                                   0);
    if (skip(Mdl != NULL, "No MDL\n"))
    {
        return;
    }
 
    BaseAddress = MmMapLockedPagesWithReservedMapping(Mapping,
                                                      PoolTag,
                                                      Mdl,
                                                      MmCached);
    ok_eq_pointer(BaseAddress, NULL);
    if (BaseAddress)
    {
        MmUnmapReservedMapping(BaseAddress,
                               PoolTag,
                               Mdl);
    }
 
    MmFreePagesFromMdl(Mdl);
}

Variable Documentation

g_IsPae

BOOLEAN g_IsPae

static

Definition at line 11 of file MmReservedMapping.c.

Referenced by START_TEST().