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
static BOOLEAN	g_IsReactOS

Function Documentation

_IRQL_requires_max_()

static _Must_inspect_result_ _IRQL_requires_max_ ( DISPATCH_LEVEL  )

static

Definition at line 57 of file MmReservedMapping.c.

{
    BOOLEAN Valid = TRUE;
#if defined(_M_IX86) || defined(_M_AMD64)
    PUCHAR CurrentAddress;
    ULONGLONG PteValue, ExpectedValue;
    ULONG i;
 
#ifdef _M_AMD64
    if (skip(GetNTVersion() < _WIN32_WINNT_WIN10, 
             "Win10 1607+ breaks these next tests.\n"))
        return Valid;
#endif
 
    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);
 
    if (skip(GetNTVersion() < _WIN32_WINNT_WIN10, 
             "DVRT (Win10 1607+) breaks these next tests.\n"))
        return Valid;
 
#ifdef _M_AMD64
    if (GetNTVersion() >= _WIN32_WINNT_WIN8)
        ExpectedValue = ((PoolTag & ~1ULL) + 1) << 28;
    else if (GetNTVersion() >= _WIN32_WINNT_VISTA)
#else
    if (GetNTVersion() >= _WIN32_WINNT_VISTA)
#endif
        ExpectedValue = ((PoolTag & ~1ULL) + 1) << 32;
    else
        ExpectedValue = PoolTag & ~1ULL;
 
    Valid = Valid &&
            ok(PteValue == ExpectedValue,
               "PTE for %p contains 0x%I64x, expected 0x%I64x\n",
               CurrentAddress, PteValue, ExpectedValue);
    CurrentAddress = (PUCHAR)BaseAddress - 2 * PAGE_SIZE;
    PteValue = GET_PTE_VALUE(CurrentAddress);
 
#ifdef _M_AMD64
    if (GetNTVersion() >= _WIN32_WINNT_WIN8)
    {
        ExpectedValue = (TotalPtes + 2) << 28;
    }
    else if (g_IsReactOS || GetNTVersion() >= _WIN32_WINNT_VISTA)
#else
    if (g_IsReactOS || GetNTVersion() >= _WIN32_WINNT_VISTA)
#endif
    {
        /* On ReactOS and on Vista+ the size is stored in
         * the NextEntry field of a MMPTE_LIST structure */
        ExpectedValue = ((ULONG64)TotalPtes + 2) << 32;
    }
    else
    {
        /* On Windows 2003 the size is shifted by 1 bit only */
        ExpectedValue = (TotalPtes + 2) * 2;
    }
    Valid = Valid &&
            ok(PteValue == ExpectedValue,
               "PTE for %p contains 0x%I64x, expected 0x%I64x\n",
               CurrentAddress, PteValue, ExpectedValue);
#endif
 
    return Valid;
}

START_TEST()

START_TEST

(

MmReservedMapping

)

Definition at line 331 of file MmReservedMapping.c.

{
    PVOID Mapping;
 
    g_IsPae = ExIsProcessorFeaturePresent(PF_PAE_ENABLED);
    g_IsReactOS = *(PULONG)(KI_USER_SHARED_DATA + PAGE_SIZE - sizeof(ULONG)) == 0x8eac705;
    if (!g_IsReactOS)
        trace("Not ReactOS\n");
 
    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');
    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 167 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;
            ok_eq_ulong(*(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"))
    {
        if (GetNTVersion() >= _WIN32_WINNT_WIN8)
            ok_eq_pointer(BaseAddress, (PVOID)ALIGN_DOWN_BY((PUCHAR)Mapping + sizeof(ULONG), 16));
        else
            ok_eq_pointer(BaseAddress, (PUCHAR)Mapping + sizeof(ULONG));
 
        ok_bool_true(ValidateMapping(BaseAddress, TotalPtes, PoolTag, 1, MdlPages),
                     "ValidateMapping returned");
 
        KmtStartSeh()
            *(volatile ULONG *)BaseAddress = 0x01234567;
            ok_eq_ulong(*(volatile ULONG *)BaseAddress, 0x01234567);
        KmtEndSeh(STATUS_SUCCESS);
 
        MmUnmapReservedMapping(BaseAddress,
                               PoolTag,
                               Mdl);
 
        ok_bool_true(ValidateMapping(Mapping, TotalPtes, PoolTag, 0, NULL),
                     "ValidateMapping returned");
    }
 
    MmFreePagesFromMdl(Mdl);
    ExFreePool(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 *)((PUCHAR)BaseAddress + i * PAGE_SIZE)) = 0x01234567 + i;
                ok_eq_ulong(*((volatile ULONG *)((PUCHAR)BaseAddress + i * PAGE_SIZE)), 0x01234567 + i);
            KmtEndSeh(STATUS_SUCCESS);
        }
 
        MmUnmapReservedMapping(BaseAddress,
                               PoolTag,
                               Mdl);
 
        ok_bool_true(ValidateMapping(Mapping, TotalPtes, PoolTag, 0, NULL),
                     "ValidateMapping returned");
    }
 
    MmFreePagesFromMdl(Mdl);
    ExFreePool(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);
    ExFreePool(Mdl);
}

Variable Documentation

g_IsPae

BOOLEAN g_IsPae

static

Definition at line 11 of file MmReservedMapping.c.

Referenced by START_TEST().

g_IsReactOS

BOOLEAN g_IsReactOS

static

Definition at line 12 of file MmReservedMapping.c.

Referenced by _IRQL_requires_max_(), and START_TEST().