Author: tfaber Date: Fri Jun 26 07:31:01 2015 New Revision: 68271
URL: http://svn.reactos.org/svn/reactos?rev=68271&view=rev Log: [KMTESTS:MM] - Add a test for MmAllocatePagesForMdl
Added: trunk/rostests/kmtests/ntos_mm/MmMdl.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?rev... ============================================================================== --- trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] (original) +++ trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] Fri Jun 26 07:31:01 2015 @@ -64,6 +64,7 @@ ntos_ke/KeProcessor.c ntos_ke/KeSpinLock.c ntos_ke/KeTimer.c + ntos_mm/MmMdl.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/testlis... ============================================================================== --- trunk/rostests/kmtests/kmtest_drv/testlist.c [iso-8859-1] (original) +++ trunk/rostests/kmtests/kmtest_drv/testlist.c [iso-8859-1] Fri Jun 26 07:31:01 2015 @@ -39,6 +39,7 @@ KMT_TESTFUNC Test_KeSpinLock; KMT_TESTFUNC Test_KeTimer; KMT_TESTFUNC Test_KernelType; +KMT_TESTFUNC Test_MmMdl; KMT_TESTFUNC Test_MmSection; KMT_TESTFUNC Test_NpfsConnect; KMT_TESTFUNC Test_NpfsCreate; @@ -98,6 +99,7 @@ { "KeSpinLock", Test_KeSpinLock }, { "KeTimer", Test_KeTimer }, { "-KernelType", Test_KernelType }, + { "MmMdl", Test_MmMdl }, { "MmSection", Test_MmSection }, { "NpfsConnect", Test_NpfsConnect }, { "NpfsCreate", Test_NpfsCreate },
Added: trunk/rostests/kmtests/ntos_mm/MmMdl.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/ntos_mm/MmMdl.c?re... ============================================================================== --- trunk/rostests/kmtests/ntos_mm/MmMdl.c (added) +++ trunk/rostests/kmtests/ntos_mm/MmMdl.c [iso-8859-1] Fri Jun 26 07:31:01 2015 @@ -0,0 +1,168 @@ +/* + * PROJECT: ReactOS kernel-mode tests + * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory + * PURPOSE: Kernel-Mode Test Suite MDL test + * PROGRAMMER: Thomas Faber thomas.faber@reactos.org + */ + +#include <kmt_test.h> + +static +VOID +TestMmAllocatePagesForMdl(VOID) +{ + PMDL Mdl; + PHYSICAL_ADDRESS LowAddress; + PHYSICAL_ADDRESS HighAddress; + PHYSICAL_ADDRESS SkipBytes; + PVOID SystemVa; + PMDL Mdls[32]; + PVOID SystemVas[32]; + ULONG i; + + LowAddress.QuadPart = 0; + HighAddress.QuadPart = -1; + SkipBytes.QuadPart = 0; + /* simple allocate/free */ + Mdl = MmAllocatePagesForMdl(LowAddress, + HighAddress, + SkipBytes, + 2 * 1024 * 1024); + ok(Mdl != NULL, "MmAllocatePagesForMdl failed\n"); + if (skip(Mdl != NULL, "No Mdl\n")) + return; + ok(MmGetMdlByteCount(Mdl) == 2 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdl)); + ok(!(Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + MmFreePagesFromMdl(Mdl); + + /* Now map/unmap it */ + Mdl = MmAllocatePagesForMdl(LowAddress, + HighAddress, + SkipBytes, + 2 * 1024 * 1024); + ok(Mdl != NULL, "MmAllocatePagesForMdl failed\n"); + if (skip(Mdl != NULL, "No Mdl\n")) + return; + ok(MmGetMdlByteCount(Mdl) == 2 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdl)); + ok(!(Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + SystemVa = MmMapLockedPagesSpecifyCache(Mdl, + KernelMode, + MmCached, + NULL, + FALSE, + NormalPagePriority); + ok(SystemVa != NULL, "MmMapLockedPagesSpecifyCache failed\n"); + if (!skip(SystemVa != NULL, "No system VA\n")) + { + ok(MmGetMdlByteCount(Mdl) == 2 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p, System VA: %p\n", MmGetMdlVirtualAddress(Mdl), SystemVa); + ok(Mdl->MappedSystemVa == SystemVa, "MappedSystemVa: %p, System VA: %p\n", Mdl->MappedSystemVa, SystemVa); + ok((Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + MmUnmapLockedPages(SystemVa, Mdl); + } + ok(MmGetMdlByteCount(Mdl) == 2 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdl)); + ok(!(Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + MmFreePagesFromMdl(Mdl); + + /* Now map it, and free without unmapping */ + Mdl = MmAllocatePagesForMdl(LowAddress, + HighAddress, + SkipBytes, + 2 * 1024 * 1024); + ok(Mdl != NULL, "MmAllocatePagesForMdl failed\n"); + if (skip(Mdl != NULL, "No Mdl\n")) + return; + ok(MmGetMdlByteCount(Mdl) == 2 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdl)); + ok(!(Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + SystemVa = MmMapLockedPagesSpecifyCache(Mdl, + KernelMode, + MmCached, + NULL, + FALSE, + NormalPagePriority); + ok(SystemVa != NULL, "MmMapLockedPagesSpecifyCache failed\n"); + ok(MmGetMdlByteCount(Mdl) == 2 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p, System VA: %p\n", MmGetMdlVirtualAddress(Mdl), SystemVa); + ok(Mdl->MappedSystemVa == SystemVa, "MappedSystemVa: %p, System VA: %p\n", Mdl->MappedSystemVa, SystemVa); + ok((Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + MmFreePagesFromMdl(Mdl); + + /* try to allocate 2 GB -- should succeed but not map */ + Mdl = MmAllocatePagesForMdl(LowAddress, + HighAddress, + SkipBytes, + 2UL * 1024 * 1024 * 1024); + ok(Mdl != NULL, "MmAllocatePagesForMdl failed for 2 GB\n"); + if (Mdl != NULL) + { + ok(MmGetMdlByteCount(Mdl) != 2UL * 1024 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdl)); + ok(!(Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + SystemVa = MmMapLockedPagesSpecifyCache(Mdl, + KernelMode, + MmCached, + NULL, + FALSE, + NormalPagePriority); + ok(SystemVa == NULL, "MmMapLockedPagesSpecifyCache succeeded for 2 GB\n"); + if (SystemVa != NULL) + MmUnmapLockedPages(SystemVa, Mdl); + ok(MmGetMdlByteCount(Mdl) != 2UL * 1024 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdl)); + ok(MmGetMdlVirtualAddress(Mdl) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdl)); + ok(!(Mdl->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdl->MdlFlags); + MmFreePagesFromMdl(Mdl); + } + + /* now allocate and map 32 MB Mdls until we fail */ + for (i = 0; i < sizeof(Mdls) / sizeof(Mdls[0]); i++) + { + Mdls[i] = MmAllocatePagesForMdl(LowAddress, + HighAddress, + SkipBytes, + 32 * 1024 * 1024); + if (Mdls[i] == NULL) + { + trace("MmAllocatePagesForMdl failed with i = %lu\n", i); + break; + } + ok(MmGetMdlVirtualAddress(Mdls[i]) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdls[i])); + ok(!(Mdls[i]->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdls[i]->MdlFlags); + SystemVas[i] = MmMapLockedPagesSpecifyCache(Mdls[i], + KernelMode, + MmCached, + NULL, + FALSE, + NormalPagePriority); + if (SystemVas[i] == NULL) + { + ok(MmGetMdlByteCount(Mdls[i]) <= 32 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdls[i])); + ok(MmGetMdlVirtualAddress(Mdls[i]) == NULL, "Virtual address: %p\n", MmGetMdlVirtualAddress(Mdls[i])); + ok(!(Mdls[i]->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdls[i]->MdlFlags); + trace("MmMapLockedPagesSpecifyCache failed with i = %lu\n", i); + break; + } + ok(MmGetMdlByteCount(Mdls[i]) == 32 * 1024 * 1024, "Byte count: %lu\n", MmGetMdlByteCount(Mdls[i])); + ok(MmGetMdlVirtualAddress(Mdls[i]) == NULL, "Virtual address: %p, System VA: %p\n", MmGetMdlVirtualAddress(Mdls[i]), SystemVas[i]); + ok(Mdls[i]->MappedSystemVa == SystemVas[i], "MappedSystemVa: %p\n", Mdls[i]->MappedSystemVa, SystemVas[i]); + ok((Mdls[i]->MdlFlags & MDL_MAPPED_TO_SYSTEM_VA), "MdlFlags: %lx\n", Mdls[i]->MdlFlags); + } + for (i = 0; i < sizeof(Mdls) / sizeof(Mdls[0]); i++) + { + if (Mdls[i] == NULL) + break; + if (SystemVas[i] != NULL) + MmUnmapLockedPages(SystemVas[i], Mdls[i]); + MmFreePagesFromMdl(Mdls[i]); + if (SystemVas[i] == NULL) + break; + } +} + +START_TEST(MmMdl) +{ + TestMmAllocatePagesForMdl(); +}
Propchange: trunk/rostests/kmtests/ntos_mm/MmMdl.c ------------------------------------------------------------------------------ svn:eol-style = native
-
tfaber＠svn.reactos.org