[tfaber] 69168: [KMTESTS:MM] - Add a test for MmAllocateMappingAddress/MmMapLockedPagesWithReservedMapping CORE-10147
10 Sep
2015
10 Sep
'15
9:58 a.m.
Author: tfaber
Date: Thu Sep 10 09:58:02 2015
New Revision: 69168
URL: http://svn.reactos.org/svn/reactos?rev=69168&view=rev
Log:
[KMTESTS:MM]
- Add a test for MmAllocateMappingAddress/MmMapLockedPagesWithReservedMapping
CORE-10147
Added:
trunk/rostests/kmtests/ntos_mm/MmReservedMapping.c (with props)
Modified:
trunk/rostests/kmtests/CMakeLists.txt
trunk/rostests/kmtests/kmtest_drv/testlist.c
Modified: trunk/rostests/kmtests/CMakeLists.txt
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/CMakeLists.txt?re…
==============================================================================
--- trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] (original)
+++ trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] Thu Sep 10 09:58:02 2015
@@ -65,6 +65,7 @@
ntos_ke/KeSpinLock.c
ntos_ke/KeTimer.c
ntos_mm/MmMdl.c
+ ntos_mm/MmReservedMapping.c
ntos_mm/MmSection.c
ntos_mm/ZwAllocateVirtualMemory.c
ntos_mm/ZwCreateSection.c
Modified: trunk/rostests/kmtests/kmtest_drv/testlist.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/kmtest_drv/testli…
==============================================================================
--- trunk/rostests/kmtests/kmtest_drv/testlist.c [iso-8859-1] (original)
+++ trunk/rostests/kmtests/kmtest_drv/testlist.c [iso-8859-1] Thu Sep 10 09:58:02
2015
@@ -41,6 +41,7 @@
KMT_TESTFUNC Test_KernelType;
KMT_TESTFUNC Test_MmMdl;
KMT_TESTFUNC Test_MmSection;
+KMT_TESTFUNC Test_MmReservedMapping;
KMT_TESTFUNC Test_NpfsConnect;
KMT_TESTFUNC Test_NpfsCreate;
KMT_TESTFUNC Test_NpfsFileInfo;
@@ -101,6 +102,7 @@
{ "-KernelType", Test_KernelType },
{ "MmMdl", Test_MmMdl },
{ "MmSection", Test_MmSection },
+ { "MmReservedMapping", Test_MmReservedMapping },
{ "NpfsConnect", Test_NpfsConnect },
{ "NpfsCreate", Test_NpfsCreate },
{ "NpfsFileInfo", Test_NpfsFileInfo },
Added: trunk/rostests/kmtests/ntos_mm/MmReservedMapping.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/ntos_mm/MmReserve…
==============================================================================
--- trunk/rostests/kmtests/ntos_mm/MmReservedMapping.c (added)
+++ trunk/rostests/kmtests/ntos_mm/MmReservedMapping.c [iso-8859-1] Thu Sep 10 09:58:02
2015
@@ -0,0 +1,228 @@
+/*
+ * PROJECT: ReactOS kernel-mode tests
+ * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
+ * PURPOSE: Kernel-Mode Test Suite Reserved Mapping test
+ * PROGRAMMER: Thomas Faber <thomas.faber(a)reactos.org>
+ */
+
+#include <kmt_test.h>
+
+#ifdef _M_IX86
+
+#define PTE_BASE 0xC0000000
+#define MiAddressToPte(x) \
+ ((PMMPTE)(((((ULONG)(x)) >> 12) << 2) + PTE_BASE))
+#define MiPteToAddress(_Pte) ((PVOID)((ULONG)(_Pte) << 10))
+
+#elif defined(_M_AMD64)
+
+#define PTI_SHIFT 12L
+#define PTE_BASE 0xFFFFF68000000000ULL
+PMMPTE
+FORCEINLINE
+_MiAddressToPte(PVOID Address)
+{
+ ULONG64 Offset = (ULONG64)Address >> (PTI_SHIFT - 3);
+ Offset &= 0xFFFFFFFFFULL << 3;
+ return (PMMPTE)(PTE_BASE + Offset);
+}
+#define MiAddressToPte(x) _MiAddressToPte((PVOID)(x))
+
+#endif
+
+static
+BOOLEAN
+ValidateMapping(
+ _In_ PVOID BaseAddress,
+ _In_ ULONG TotalPtes,
+ _In_ ULONG ValidPtes,
+ _In_ PPFN_NUMBER Pfns)
+{
+ BOOLEAN Valid = TRUE;
+#if defined(_M_IX86) || defined(_M_AMD64)
+ PMMPTE PointerPte;
+ ULONG i;
+
+ PointerPte = MiAddressToPte(BaseAddress);
+ for (i = 0; i < ValidPtes; i++)
+ {
+ Valid = Valid &&
+ ok(PointerPte[i].u.Hard.Valid == 1,
+ "[%lu] PTE %p is not valid\n", i, &PointerPte[i]);
+
+ Valid = Valid &&
+ ok(PointerPte[i].u.Hard.PageFrameNumber == Pfns[i],
+ "[%lu] PTE %p has PFN %Ix, expected %Ix\n",
+ i, &PointerPte[i], PointerPte[i].u.Hard.PageFrameNumber, Pfns[i]);
+ }
+ for (; i < TotalPtes; i++)
+ {
+ Valid = Valid &&
+ ok_eq_hex(PointerPte[i].u.Long, 0UL);
+ }
+ Valid = Valid &&
+ ok_eq_tag(PointerPte[-1].u.Long, 'MRmK' & ~1);
+ Valid = Valid &&
+ ok_eq_ulong(PointerPte[-2].u.Long, (TotalPtes + 2) * 2);
+#endif
+
+ return Valid;
+}
+
+static
+VOID
+TestMap(
+ _In_ PVOID Mapping)
+{
+ PMDL Mdl;
+ PHYSICAL_ADDRESS ZeroPhysical;
+ PHYSICAL_ADDRESS MaxPhysical;
+ PVOID BaseAddress;
+ PPFN_NUMBER MdlPages;
+ ULONG i;
+
+ ZeroPhysical.QuadPart = 0;
+ MaxPhysical.QuadPart = 0xffffffffffffffffLL;
+
+ /* Create a one-page MDL and map it */
+ Mdl = MmAllocatePagesForMdlEx(ZeroPhysical,
+ MaxPhysical,
+ ZeroPhysical,
+ PAGE_SIZE,
+ MmCached,
+ 0);
+ if (skip(Mdl != NULL, "No MDL\n"))
+ {
+ return;
+ }
+
+ MdlPages = (PVOID)(Mdl + 1);
+
+ BaseAddress = MmMapLockedPagesWithReservedMapping(Mapping,
+ 'MRmK',
+ Mdl,
+ MmCached);
+ if (BaseAddress)
+ {
+ ok_eq_pointer(BaseAddress, Mapping);
+
+ ok_bool_true(ValidateMapping(BaseAddress, 10, 1, MdlPages),
+ "ValidateMapping returned");
+
+ KmtStartSeh()
+ *(volatile ULONG *)BaseAddress = 0x01234567;
+ KmtEndSeh(STATUS_SUCCESS);
+
+ MmUnmapReservedMapping(BaseAddress,
+ 'MRmK',
+ Mdl);
+
+ ok_bool_true(ValidateMapping(Mapping, 10, 0, NULL),
+ "ValidateMapping returned");
+ }
+
+ MmFreePagesFromMdl(Mdl);
+
+ /* Map all pages */
+ Mdl = MmAllocatePagesForMdlEx(ZeroPhysical,
+ MaxPhysical,
+ ZeroPhysical,
+ 10 * PAGE_SIZE,
+ MmCached,
+ 0);
+ if (skip(Mdl != NULL, "No MDL\n"))
+ {
+ return;
+ }
+
+ MdlPages = (PVOID)(Mdl + 1);
+
+ BaseAddress = MmMapLockedPagesWithReservedMapping(Mapping,
+ 'MRmK',
+ Mdl,
+ MmCached);
+ if (BaseAddress)
+ {
+ ok_eq_pointer(BaseAddress, Mapping);
+
+ ok_bool_true(ValidateMapping(BaseAddress, 10, 10, MdlPages),
+ "ValidateMapping returned");
+
+ for (i = 0; i < 10; i++)
+ {
+ KmtStartSeh()
+ *((volatile ULONG *)BaseAddress + i * PAGE_SIZE / sizeof(ULONG)) =
0x01234567;
+ KmtEndSeh(STATUS_SUCCESS);
+ }
+
+ MmUnmapReservedMapping(BaseAddress,
+ 'MRmK',
+ Mdl);
+
+ ok_bool_true(ValidateMapping(Mapping, 10, 0, NULL),
+ "ValidateMapping returned");
+ }
+
+ MmFreePagesFromMdl(Mdl);
+
+ /* Try to map more pages than we reserved */
+ Mdl = MmAllocatePagesForMdlEx(ZeroPhysical,
+ MaxPhysical,
+ ZeroPhysical,
+ 11 * PAGE_SIZE,
+ MmCached,
+ 0);
+ if (skip(Mdl != NULL, "No MDL\n"))
+ {
+ return;
+ }
+
+ BaseAddress = MmMapLockedPagesWithReservedMapping(Mapping,
+ 'MRmK',
+ Mdl,
+ MmCached);
+ ok_eq_pointer(BaseAddress, NULL);
+ if (BaseAddress)
+ {
+ MmUnmapReservedMapping(BaseAddress,
+ 'MRmK',
+ Mdl);
+ }
+
+ MmFreePagesFromMdl(Mdl);
+}
+
+START_TEST(MmReservedMapping)
+{
+ PVOID Mapping;
+
+ /* 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, 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, 0, NULL),
+ "ValidateMapping returned");
+
+ /* PAGE_FAULT_IN_NONPAGED_AREA can't be caught with SEH */
+ if (0)
+ {
+ (void)*(volatile UCHAR *)Mapping;
+ }
+
+ TestMap(Mapping);
+
+ MmFreeMappingAddress(Mapping, 'MRmK');
+ }
+}
Propchange: trunk/rostests/kmtests/ntos_mm/MmReservedMapping.c
------------------------------------------------------------------------------
svn:eol-style = native
3525
days inactive
3525
days old
0 comments
1 participants
participants (1)
-
tfaber＠svn.reactos.org