[tkreuzer] 58613: [NTDLL_APITEST] Add more tests for NtAllocateVirtualMemory and NtMapViewOfSection
27 Mar
2013
27 Mar
'13
7:10 p.m.
Author: tkreuzer
Date: Wed Mar 27 19:10:51 2013
New Revision: 58613
URL: http://svn.reactos.org/svn/reactos?rev=58613&view=rev
Log:
[NTDLL_APITEST]
Add more tests for NtAllocateVirtualMemory and NtMapViewOfSection
Added:
trunk/rostests/apitests/ntdll/NtMapViewOfSection.c (with props)
trunk/rostests/apitests/testdata/test.dll (with props)
Modified:
trunk/rostests/apitests/ntdll/CMakeLists.txt
trunk/rostests/apitests/ntdll/NtAllocateVirtualMemory.c
trunk/rostests/apitests/ntdll/testlist.c
Modified: trunk/rostests/apitests/ntdll/CMakeLists.txt
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/CMakeLists…
==============================================================================
--- trunk/rostests/apitests/ntdll/CMakeLists.txt [iso-8859-1] (original)
+++ trunk/rostests/apitests/ntdll/CMakeLists.txt [iso-8859-1] Wed Mar 27 19:10:51 2013
@@ -3,6 +3,7 @@
LdrEnumResources.c
NtAllocateVirtualMemory.c
NtFreeVirtualMemory.c
+ NtMapViewOfSection.c
NtQuerySystemEnvironmentValue.c
RtlBitmap.c
RtlDetermineDosPathNameType.c
Modified: trunk/rostests/apitests/ntdll/NtAllocateVirtualMemory.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/NtAllocate…
==============================================================================
--- trunk/rostests/apitests/ntdll/NtAllocateVirtualMemory.c [iso-8859-1] (original)
+++ trunk/rostests/apitests/ntdll/NtAllocateVirtualMemory.c [iso-8859-1] Wed Mar 27
19:10:51 2013
@@ -9,6 +9,7 @@
#include <wine/test.h>
#include <ndk/rtlfuncs.h>
#include <ndk/mmfuncs.h>
+#include <pseh/pseh2.h>
static PVOID Allocations[4096] = { NULL };
static ULONG CurrentAllocation = 0;
@@ -198,6 +199,275 @@
return TRUE;
}
+VOID
+CheckSize(ULONG_PTR Base, SIZE_T InSize, SIZE_T ExpectedSize)
+{
+ NTSTATUS Status;
+ PVOID BaseAddress;
+ SIZE_T Size;
+
+ /* Reserve memory */
+ BaseAddress = UlongToPtr(Base);
+ Size = InSize;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_RESERVE,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+ ok(BaseAddress == UlongToPtr(Base & ~0xFFFF), "Got back wrong base address:
%p\n", BaseAddress);
+ ok(Size == ExpectedSize, "Alloc of 0x%Ix: got back wrong size: 0x%Ix, expected
0x%Ix\n", InSize, Size, ExpectedSize);
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(NT_SUCCESS(Status), "NtFreeVirtualMemory failed!\n");
+}
+
+VOID
+CheckAlignment()
+{
+ NTSTATUS Status;
+ PVOID BaseAddress;
+ SIZE_T Size;
+
+ CheckSize(0x50000000, 0x0001, 0x1000);
+ CheckSize(0x50008000, 0x0001, 0x9000);
+ CheckSize(0x50000010, 0x1000, 0x2000);
+ CheckSize(0x50010000, 0x2000, 0x2000);
+ CheckSize(0x5000FFFF, 0x3000, 0x13000);
+ CheckSize(0x50001010, 0x7000, 0x9000);
+ CheckSize(0x50001010, 0xC000, 0xe000);
+
+ /* Reserve memory not aligned to allocation granularity */
+ BaseAddress = UlongToPtr(0x50001010);
+ Size = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_RESERVE,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+ ok(BaseAddress == UlongToPtr(0x50000000), "Got back wrong base address:
%p", BaseAddress);
+ ok(Size == 0x3000, "Got back wrong size: 0x%Ix", Size);
+
+ /* Try to reserve again in the same 64k region */
+ BaseAddress = UlongToPtr(0x50008000);
+ Size = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_RESERVE,
+ PAGE_NOACCESS);
+ ok(!NT_SUCCESS(Status), "NtAllocateVirtualMemory should fail!\n");
+
+ /* Commit memory */
+ BaseAddress = UlongToPtr(0x50002000);
+ Size = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_COMMIT,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+ ok(BaseAddress == UlongToPtr(0x50002000), "Got back wrong base address:
%p", BaseAddress);
+ ok(Size == 0x1000, "Got back wrong size: 0x%Ix", Size);
+
+ /* Commit memory */
+ BaseAddress = UlongToPtr(0x50003000);
+ Size = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_COMMIT,
+ PAGE_NOACCESS);
+ ok(!NT_SUCCESS(Status), "NtAllocateVirtualMemory should fail!\n");
+
+ /* Try to release memory in a different 64k region */
+ BaseAddress = UlongToPtr(0x50010000);
+ Size = 0x1000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(!NT_SUCCESS(Status), "NtFreeVirtualMemory should fail!\n");
+
+ /* Release the memory in the same 64k region at a different address */
+ BaseAddress = UlongToPtr(0x50008000);
+ Size = 0x1000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(!NT_SUCCESS(Status), "NtFreeVirtualMemory failed!\n");
+
+ /* Release the memory at the correct address but with wrong size */
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x4000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(!NT_SUCCESS(Status), "NtFreeVirtualMemory should fail!\n");
+
+ /* Release the memory */
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x3000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(NT_SUCCESS(Status), "NtFreeVirtualMemory failed!\n");
+
+ /* Reserve and commit at once */
+ BaseAddress = UlongToPtr(0x50004080);
+ Size = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_RESERVE | MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+ ok(BaseAddress == UlongToPtr(0x50000000), "Got back wrong base address:
%p", BaseAddress);
+ ok(Size == 0x6000, "Got back wrong size: 0x%Ix", Size);
+
+ _SEH2_TRY
+ {
+ *(int*)UlongToPtr(BaseAddress) = 1;
+ *(int*)UlongToPtr(0x50004080) = 1;
+ }
+ _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+ {
+ ok(0, "Got exception\n");
+ }
+ _SEH2_END;
+
+ /* Release the memory */
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(NT_SUCCESS(Status), "NtFreeVirtualMemory failed!\n");
+
+}
+
+static
+VOID
+CheckAdjacentVADs()
+{
+ NTSTATUS Status;
+ PVOID BaseAddress;
+ SIZE_T Size;
+
+ /* Reserve a full 64k region */
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x10000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_RESERVE,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+ if (!NT_SUCCESS(Status))
+ return;
+
+ /* Reserve another 64k region, but with 64k between */
+ BaseAddress = UlongToPtr(0x50020000);
+ Size = 0x10000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_RESERVE,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+ if (!NT_SUCCESS(Status))
+ return;
+
+ /* Try to free the whole at once */
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x30000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(!NT_SUCCESS(Status), "NtFreeVirtualMemory should fail!\n");
+
+ /* Reserve the part in the middle */
+ BaseAddress = UlongToPtr(0x50010000);
+ Size = 0x10000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_RESERVE,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+
+ /* Try to commit memory covering 2 allocations */
+ BaseAddress = UlongToPtr(0x50004000);
+ Size = 0x10000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_COMMIT,
+ PAGE_NOACCESS);
+ ok(!NT_SUCCESS(Status), "NtAllocateVirtualMemory should fail!\n");
+
+ /* Allocate a page */
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+
+ /* Allocate another page */
+ BaseAddress = UlongToPtr(0x50002000);
+ Size = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_COMMIT,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+
+ _SEH2_TRY
+ {
+ *(int*)UlongToPtr(0x50000000) = 1;
+ //*(int*)UlongToPtr(0x50002000) = 1;
+ }
+ _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+ {
+ ok(0, "Got exception\n");
+ }
+ _SEH2_END;
+
+ /* Allocate 3 pages, on top of the previous 2 */
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x3000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &Size,
+ MEM_COMMIT,
+ PAGE_NOACCESS);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed!\n");
+
+ /* Try to free the whole region at once */
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x30000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(!NT_SUCCESS(Status), "NtFreeVirtualMemory should fail!\n");
+
+ BaseAddress = UlongToPtr(0x50000000);
+ Size = 0x10000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(NT_SUCCESS(Status), "NtFreeVirtualMemory failed!\n");
+
+ BaseAddress = UlongToPtr(0x50010000);
+ Size = 0x10000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(NT_SUCCESS(Status), "NtFreeVirtualMemory failed!\n");
+
+ BaseAddress = UlongToPtr(0x50020000);
+ Size = 0x10000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &Size,
MEM_RELEASE);
+ ok(NT_SUCCESS(Status), "NtFreeVirtualMemory failed!\n");
+
+}
+
#define RUNS 32
START_TEST(NtAllocateVirtualMemory)
@@ -205,6 +475,9 @@
PVOID Mem1, Mem2;
SIZE_T Size1, Size2;
ULONG i;
+
+ CheckAlignment();
+ CheckAdjacentVADs();
Size1 = 32;
Mem1 = Allocate(Size1);
Added: trunk/rostests/apitests/ntdll/NtMapViewOfSection.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/NtMapViewO…
==============================================================================
--- trunk/rostests/apitests/ntdll/NtMapViewOfSection.c (added)
+++ trunk/rostests/apitests/ntdll/NtMapViewOfSection.c [iso-8859-1] Wed Mar 27 19:10:51
2013
@@ -1,0 +1,390 @@
+
+#define WIN32_NO_STATUS
+#include <wine/test.h>
+#include <ndk/ntndk.h>
+#include <pseh/pseh2.h>
+
+NTSYSAPI
+NTSTATUS
+NTAPI
+NtMapViewOfSection(
+ HANDLE SectionHandle,
+ HANDLE ProcessHandle,
+ PVOID *BaseAddress,
+ ULONG_PTR ZeroBits,
+ SIZE_T CommitSize,
+ PLARGE_INTEGER SectionOffset,
+ PSIZE_T ViewSize,
+ SECTION_INHERIT InheritDisposition,
+ ULONG AllocationType,
+ ULONG Protect);
+
+void
+Test_PageFileSection(void)
+{
+ NTSTATUS Status;
+ HANDLE SectionHandle;
+ LARGE_INTEGER MaximumSize, SectionOffset;
+ PVOID BaseAddress;
+ SIZE_T ViewSize;
+
+ /* Create a page file backed section */
+ MaximumSize.QuadPart = 0x20000;
+ Status = NtCreateSection(&SectionHandle,
+ SECTION_ALL_ACCESS,
+ NULL,
+ &MaximumSize,
+ PAGE_READWRITE,
+ SEC_COMMIT,
+ NULL);
+ ok(NT_SUCCESS(Status), "NtCreateSection failed with Status %lx\n",
Status);
+ if (!NT_SUCCESS(Status))
+ return;
+
+ /* Try to map a page at an address that is not 64k aligned */
+ BaseAddress = (PVOID)0x30001000;
+ SectionOffset.QuadPart = 0;
+ ViewSize = 0x1000;
+ Status = NtMapViewOfSection(SectionHandle,
+ NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ 0,
+ &SectionOffset,
+ &ViewSize,
+ ViewShare,
+ 0,
+ PAGE_READWRITE);
+ ok(Status == STATUS_MAPPED_ALIGNMENT,
+ "NtMapViewOfSection returned wrong Status %lx\n", Status);
+
+ /* Try to map a page with execute rights */
+ BaseAddress = (PVOID)0x30000000;
+ SectionOffset.QuadPart = 0;
+ ViewSize = 0x1000;
+ Status = NtMapViewOfSection(SectionHandle,
+ NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ 0,
+ &SectionOffset,
+ &ViewSize,
+ ViewShare,
+ 0,
+ PAGE_EXECUTE_READWRITE);
+ ok(Status == STATUS_SECTION_PROTECTION,
+ "NtMapViewOfSection returned wrong Status %lx\n", Status);
+
+ /* Map 2 pages, not comitting them */
+ BaseAddress = (PVOID)0x30000000;
+ SectionOffset.QuadPart = 0;
+ ViewSize = 0x2000;
+ Status = NtMapViewOfSection(SectionHandle,
+ NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ 0,
+ &SectionOffset,
+ &ViewSize,
+ ViewShare,
+ 0,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtMapViewOfSection failed with Status %lx\n",
Status);
+ if (!NT_SUCCESS(Status))
+ return;
+
+ /* Commit a page in the section */
+ BaseAddress = (PVOID)0x30000000;
+ ViewSize = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &ViewSize,
+ MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed with Status %lx\n",
Status);
+
+ /* Try to commit a range larger than the section */
+ BaseAddress = (PVOID)0x30000000;
+ ViewSize = 0x3000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &ViewSize,
+ MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(Status == STATUS_CONFLICTING_ADDRESSES,
+ "NtAllocateVirtualMemory failed with wrong Status %lx\n", Status);
+
+ /* Try to commit a page after the section */
+ BaseAddress = (PVOID)0x30002000;
+ ViewSize = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &ViewSize,
+ MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(!NT_SUCCESS(Status), "NtAllocateVirtualMemory Should fail\n");
+
+ /* Try to allocate a page after the section */
+ BaseAddress = (PVOID)0x30002000;
+ ViewSize = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &ViewSize,
+ MEM_RESERVE | MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(!NT_SUCCESS(Status), "NtAllocateVirtualMemory should fail\n");
+
+ /* Need to go to next 64k boundary */
+ BaseAddress = (PVOID)0x30010000;
+ ViewSize = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &ViewSize,
+ MEM_RESERVE | MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed with Status %lx\n",
Status);
+ if (!NT_SUCCESS(Status))
+ return;
+
+ /* Free the allocation */
+ BaseAddress = (PVOID)0x30010000;
+ ViewSize = 0x1000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ &ViewSize,
+ MEM_RELEASE);
+ ok(NT_SUCCESS(Status), "NtFreeVirtualMemory failed with Status %lx\n",
Status);
+
+ /* Free the section mapping */
+ BaseAddress = (PVOID)0x30000000;
+ ViewSize = 0x1000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ &ViewSize,
+ MEM_RELEASE);
+ ok(Status == STATUS_UNABLE_TO_DELETE_SECTION,
+ "NtFreeVirtualMemory failed with wrong Status %lx\n", Status);
+
+ /* Commit a page in the section */
+ BaseAddress = (PVOID)0x30001000;
+ ViewSize = 0x1000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &ViewSize,
+ MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed with Status %lx\n",
Status);
+
+ /* Try to decommit the page */
+ BaseAddress = (PVOID)0x30001000;
+ ViewSize = 0x1000;
+ Status = NtFreeVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ &ViewSize,
+ MEM_DECOMMIT);
+ ok(Status == STATUS_UNABLE_TO_DELETE_SECTION,
+ "NtFreeVirtualMemory failed with wrong Status %lx\n", Status);
+
+ BaseAddress = (PVOID)0x40000000;
+ SectionOffset.QuadPart = 0;
+ ViewSize = 0x1000;
+ Status = NtMapViewOfSection(SectionHandle,
+ NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ 0,
+ &SectionOffset,
+ &ViewSize,
+ ViewShare,
+ 0,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtMapViewOfSection failed with Status %lx\n",
Status);
+ if (!NT_SUCCESS(Status))
+ return;
+
+ ok(BaseAddress == (PVOID)0x40000000, "Invalid BaseAddress: %p",
BaseAddress);
+
+ BaseAddress = (PVOID)0x40080000;
+ SectionOffset.QuadPart = 0x10000;
+ ViewSize = 0x1000;
+ Status = NtMapViewOfSection(SectionHandle,
+ NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ 0,
+ &SectionOffset,
+ &ViewSize,
+ ViewShare,
+ 0,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtMapViewOfSection failed with Status %lx\n",
Status);
+ if (!NT_SUCCESS(Status))
+ return;
+
+ ok(BaseAddress == (PVOID)0x40080000, "Invalid BaseAddress: %p",
BaseAddress);
+
+ /* Commit a page in the section */
+ BaseAddress = (PVOID)0x40000000;
+ Status = NtAllocateVirtualMemory(NtCurrentProcess(),
+ &BaseAddress,
+ 0,
+ &ViewSize,
+ MEM_COMMIT,
+ PAGE_READWRITE);
+ ok(NT_SUCCESS(Status), "NtAllocateVirtualMemory failed with Status %lx\n",
Status);
+ if (!NT_SUCCESS(Status))
+ return;
+
+}
+
+void
+Test_ImageSection(void)
+{
+ UNICODE_STRING FileName;
+ NTSTATUS Status;
+ OBJECT_ATTRIBUTES FileObjectAttributes;
+ IO_STATUS_BLOCK IoStatusBlock;
+ HANDLE FileHandle, DataSectionHandle, ImageSectionHandle;
+ PVOID DataBase, ImageBase;
+ SIZE_T ViewSize;
+
+ if (!RtlDosPathNameToNtPathName_U(L"testdata\\test.dll",
+ &FileName,
+ NULL,
+ NULL))
+ {
+ ok(0, "RtlDosPathNameToNtPathName_U failed\n");
+ return;
+ }
+
+ InitializeObjectAttributes(&FileObjectAttributes,
+ &FileName,
+ 0,
+ NULL,
+ NULL);
+
+ Status = NtOpenFile(&FileHandle,
+ GENERIC_READ|GENERIC_WRITE|SYNCHRONIZE,
+ &FileObjectAttributes,
+ &IoStatusBlock,
+ FILE_SHARE_READ,
+ FILE_SYNCHRONOUS_IO_NONALERT);
+ ok(Status == STATUS_SUCCESS, "NtOpenFile failed, Status 0x%lx\n", Status);
+
+ /* Create a data section with write access */
+ Status = NtCreateSection(&DataSectionHandle,
+ SECTION_ALL_ACCESS, // DesiredAccess
+ NULL, // ObjectAttributes
+ NULL, // MaximumSize
+ PAGE_READWRITE, // SectionPageProtection
+ SEC_COMMIT, // AllocationAttributes
+ FileHandle);
+ ok(Status == STATUS_SUCCESS, "NtCreateSection failed, Status 0x%lx\n",
Status);
+
+ /* Map the data section */
+ DataBase = NULL;
+ ViewSize = 0;
+ Status = NtMapViewOfSection(DataSectionHandle,
+ NtCurrentProcess(),
+ &DataBase,
+ 0,
+ 0,
+ NULL,
+ &ViewSize,
+ ViewShare,
+ 0,
+ PAGE_READWRITE);
+ ok(Status == STATUS_SUCCESS, "NtMapViewOfSection failed, Status 0x%lx\n",
Status);
+
+ /* Check the original data */
+ ok(*(ULONG*)DataBase == 0x00905a4d, "Header not ok\n");
+
+ /* Now modify the data in the data section */
+ *(ULONG*)DataBase = 0xdeadbabe;
+
+ /* Check the data */
+ ok(*(ULONG*)DataBase == 0xdeadbabe, "Header not ok\n");
+
+ /* Now try to create an image section */
+ Status = NtCreateSection(&ImageSectionHandle,
+ SECTION_ALL_ACCESS, // DesiredAccess
+ NULL, // ObjectAttributes
+ NULL, // MaximumSize
+ PAGE_READWRITE, // SectionPageProtection
+ SEC_IMAGE, // AllocationAttributes
+ FileHandle);
+ ok(Status == STATUS_INVALID_IMAGE_NOT_MZ, "NtCreateSection failed, Status
0x%lx\n", Status);
+
+ /* Restore the original data */
+ *(ULONG*)DataBase = 0x00905a4d;
+
+ /* Try to create an image section again */
+ Status = NtCreateSection(&ImageSectionHandle,
+ SECTION_ALL_ACCESS, // DesiredAccess
+ NULL, // ObjectAttributes
+ NULL, // MaximumSize
+ PAGE_READWRITE, // SectionPageProtection
+ SEC_IMAGE, // AllocationAttributes
+ FileHandle);
+ ok(Status == STATUS_SUCCESS, "NtCreateSection failed, Status 0x%lx\n",
Status);
+
+ /* Map the image section */
+ ImageBase = NULL;
+ ViewSize = 0;
+ Status = NtMapViewOfSection(ImageSectionHandle,
+ NtCurrentProcess(),
+ &ImageBase,
+ 0,
+ 0,
+ NULL,
+ &ViewSize,
+ ViewShare,
+ 0,
+ PAGE_READONLY);
+ ok(Status == STATUS_SUCCESS, "NtMapViewOfSection failed, Status 0x%lx\n",
Status);
+
+ /* Check the data */
+ ok(*(ULONG*)DataBase == 0x00905a4d, "Header not ok\n");
+ ok(*(ULONG*)ImageBase == 0x00905a4d, "Header not ok\n");
+
+ /* Now modify the data again */
+ *(ULONG*)DataBase = 0xdeadbabe;
+
+ /* Check the data */
+ ok(*(ULONG*)DataBase == 0xdeadbabe, "Header not ok\n");
+ ok(*(ULONG*)ImageBase == 0x00905a4d, "Data should not be synced!\n");
+
+ /* Flush the view */
+ ViewSize = 0x1000;
+ Status = NtFlushVirtualMemory(NtCurrentProcess(),
+ &DataBase,
+ &ViewSize,
+ &IoStatusBlock);
+ ok(Status == STATUS_SUCCESS, "NtFlushVirtualMemory failed, Status 0x%lx\n",
Status);
+
+ /* Check the data again */
+ ok(*(ULONG*)ImageBase == 0x00905a4d, "Data should not be synced!\n");
+
+ /* Restore the original data */
+ *(ULONG*)DataBase = 0x00905a4d;
+ ok(*(ULONG*)DataBase == 0x00905a4d, "Header not ok\n");
+
+ /* Cleanup */
+ NtUnmapViewOfSection(NtCurrentProcess(), ImageBase);
+ NtUnmapViewOfSection(NtCurrentProcess(), DataBase);
+ NtClose(ImageSectionHandle);
+ NtClose(DataSectionHandle);
+ NtClose(FileHandle);
+}
+
+START_TEST(NtMapViewOfSection)
+{
+ //Test_PageFileSection();
+ Test_ImageSection();
+}
Propchange: trunk/rostests/apitests/ntdll/NtMapViewOfSection.c
------------------------------------------------------------------------------
svn:eol-style = native
Modified: trunk/rostests/apitests/ntdll/testlist.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/testlist.c…
==============================================================================
--- trunk/rostests/apitests/ntdll/testlist.c [iso-8859-1] (original)
+++ trunk/rostests/apitests/ntdll/testlist.c [iso-8859-1] Wed Mar 27 19:10:51 2013
@@ -6,6 +6,7 @@
extern void func_LdrEnumResources(void);
extern void func_NtAllocateVirtualMemory(void);
extern void func_NtFreeVirtualMemory(void);
+extern void func_NtMapViewOfSection(void);
extern void func_NtQuerySystemEnvironmentValue(void);
extern void func_NtSystemInformation(void);
extern void func_RtlBitmap(void);
@@ -26,6 +27,7 @@
{ "LdrEnumResources", func_LdrEnumResources },
{ "NtAllocateVirtualMemory", func_NtAllocateVirtualMemory },
{ "NtFreeVirtualMemory", func_NtFreeVirtualMemory },
+ { "NtMapViewOfSection", func_NtMapViewOfSection },
{ "NtQuerySystemEnvironmentValue", func_NtQuerySystemEnvironmentValue },
{ "NtSystemInformation", func_NtSystemInformation },
{ "RtlBitmapApi", func_RtlBitmap },
Added: trunk/rostests/apitests/testdata/test.dll
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/testdata/test.dl…
==============================================================================
Binary file - no diff available.
Propchange: trunk/rostests/apitests/testdata/test.dll
------------------------------------------------------------------------------
svn:mime-type = application/octet-stream
4251
days inactive
4251
days old
0 comments
1 participants
participants (1)
-
tkreuzer＠svn.reactos.org