/* * PROJECT: ReactOS API Tests * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory * PURPOSE: Test for NtMapViewOfSection * PROGRAMMERS: Timo Kreuzer * Thomas Faber */ #include "precomp.h" void Test_PageFileSection(void) { NTSTATUS Status; HANDLE SectionHandle; LARGE_INTEGER MaximumSize, SectionOffset; PVOID BaseAddress, BaseAddress2; SIZE_T ViewSize; ULONG OldProtect; /* Create a page file backed section with SEC_COMMIT */ MaximumSize.QuadPart = 0x20000; Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_COMMIT, NULL); ok_ntstatus(Status, STATUS_SUCCESS); 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_ntstatus(Status, STATUS_MAPPED_ALIGNMENT); /* 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_ntstatus(Status, STATUS_SECTION_PROTECTION); /* Try to map 2 pages with MEM_COMMIT */ BaseAddress = (PVOID)0x30000000; SectionOffset.QuadPart = 0; ViewSize = 0x2000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, PAGE_SIZE, &SectionOffset, &ViewSize, ViewShare, MEM_COMMIT, PAGE_READWRITE); ok_ntstatus(Status, STATUS_INVALID_PARAMETER_9); /* Try to map 1 page, with free base address and zero bits compatible with 64k granularity */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = 0x1000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 10, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); { ULONG_PTR gran = 64 * 1024; ULONG_PTR ZeroBits = 11; ok_hex(gran, 0x10000); gran <<= ZeroBits; ok_hex(gran, 0x8000000); gran >>= ZeroBits; ok_hex(gran, 0x10000); ok_hex((gran << ZeroBits) >> ZeroBits, gran); } /* Try to map 1 page, with free base address and zero bits incompatible with 64k granularity */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = 0x1000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 11, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_NO_MEMORY); /* Try to map 1 page, with base address and zero bits being compatible */ BaseAddress = (PVOID)0x30000000; SectionOffset.QuadPart = 0; ViewSize = 0x1000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 2, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); /* Try to map 1 page, with base address and zero bits being incompatible */ BaseAddress = (PVOID)0x30000000; SectionOffset.QuadPart = 0; ViewSize = 0x1000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 3, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_INVALID_PARAMETER_4); /* Map 2 pages, without MEM_COMMIT */ BaseAddress = (PVOID)0x30000000; SectionOffset.QuadPart = 0; ViewSize = 0x2000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); /* We must be able to access the memory */ _SEH2_TRY { *(PULONG)BaseAddress = 1; } _SEH2_EXCEPT(1) { ok(FALSE, "Got an exception\n"); } _SEH2_END; /* Commit a page in the section */ BaseAddress = (PVOID)0x30000000; ViewSize = 0x1000; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress, 0, &ViewSize, MEM_COMMIT, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); /* Try to decommit a page in the section */ Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &ViewSize, MEM_DECOMMIT); ok_ntstatus(Status, STATUS_UNABLE_TO_DELETE_SECTION); /* 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_ntstatus(Status, STATUS_CONFLICTING_ADDRESSES); /* Try to commit a page after the section */ BaseAddress = (PVOID)0x30002000; ViewSize = 0x1000; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress, 0, &ViewSize, MEM_COMMIT, PAGE_READWRITE); ok_ntstatus(Status, STATUS_CONFLICTING_ADDRESSES); /* 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_ntstatus(Status, STATUS_CONFLICTING_ADDRESSES); /* 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_ntstatus(Status, STATUS_SUCCESS); 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); /* Try to release the section mapping with NtFreeVirtualMemory */ BaseAddress = (PVOID)0x30000000; ViewSize = 0x1000; Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &ViewSize, MEM_RELEASE); ok_ntstatus(Status, STATUS_UNABLE_TO_DELETE_SECTION); /* Commit a page in the section */ BaseAddress = (PVOID)0x30001000; ViewSize = 0x1000; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress, 0, &ViewSize, MEM_COMMIT, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); /* Try to decommit the page */ BaseAddress = (PVOID)0x30001000; ViewSize = 0x1000; Status = NtFreeVirtualMemory(NtCurrentProcess(), &BaseAddress, &ViewSize, MEM_DECOMMIT); ok_ntstatus(Status, STATUS_UNABLE_TO_DELETE_SECTION); BaseAddress = UlongToPtr(0x40000000); SectionOffset.QuadPart = 0; ViewSize = 0x1000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); if (!NT_SUCCESS(Status)) return; ok(BaseAddress == UlongToPtr(0x40000000), "Invalid BaseAddress: %p\n", BaseAddress); BaseAddress = (PVOID)0x40080000; SectionOffset.QuadPart = 0x10000; ViewSize = 0x1000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); ok(BaseAddress == (PVOID)0x40080000, "Invalid BaseAddress: %p\n", BaseAddress); /* Commit a page in the section */ BaseAddress = (PVOID)0x40000000; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress, 0, &ViewSize, MEM_COMMIT, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); /* Close the mapping */ Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); BaseAddress = (PVOID)0x30000000; Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); Status = NtClose(SectionHandle); ok_ntstatus(Status, STATUS_SUCCESS); /* Create a page file backed section, but only reserved */ MaximumSize.QuadPart = 0x20000; Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_RESERVE, NULL); ok_ntstatus(Status, STATUS_SUCCESS); /* Try to map 1 page, passing MEM_RESERVE */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = PAGE_SIZE; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, PAGE_SIZE, &SectionOffset, &ViewSize, ViewShare, MEM_RESERVE, PAGE_READWRITE); ok_ntstatus(Status, STATUS_INVALID_PARAMETER_9); /* Try to map 1 page using MEM_COMMIT */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = PAGE_SIZE; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, PAGE_SIZE, &SectionOffset, &ViewSize, ViewShare, MEM_COMMIT, PAGE_READWRITE); ok_ntstatus(Status, STATUS_INVALID_PARAMETER_9); /* Map 2 pages, but commit 1 */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = 2 * PAGE_SIZE; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, PAGE_SIZE, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); /* We must be able to access the 1st page */ Status = STATUS_SUCCESS; _SEH2_TRY { *(PUCHAR)BaseAddress = 1; } _SEH2_EXCEPT(1) { Status = _SEH2_GetExceptionCode(); } _SEH2_END; ok_ntstatus(Status, STATUS_SUCCESS); /* We must not be able to access the 2nd page */ Status = STATUS_SUCCESS; _SEH2_TRY { *((PUCHAR)BaseAddress + PAGE_SIZE) = 1; } _SEH2_EXCEPT(1) { Status = _SEH2_GetExceptionCode(); } _SEH2_END; ok_ntstatus(Status, STATUS_ACCESS_VIOLATION); /* Map the 2 pages again into a different memory location */ BaseAddress2 = NULL; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress2, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); /* Commit a the 2nd page in the 2nd memory location */ BaseAddress2 = (PUCHAR)BaseAddress2 + PAGE_SIZE; ViewSize = PAGE_SIZE; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress2, 0, &ViewSize, MEM_COMMIT, PAGE_READONLY); ok_ntstatus(Status, STATUS_SUCCESS); /* Try to commit again (the already committed page) */ Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress2, 0, &ViewSize, MEM_COMMIT, PAGE_READONLY); ok_ntstatus(Status, STATUS_SUCCESS); /* We must be able to access the memory in the 2nd page of the 1st memory location */ Status = STATUS_SUCCESS; _SEH2_TRY { *((PUCHAR)BaseAddress + PAGE_SIZE) = 2; } _SEH2_EXCEPT(1) { Status = _SEH2_GetExceptionCode(); } _SEH2_END; ok_ntstatus(Status, STATUS_SUCCESS); ok(*(PULONG)BaseAddress2 == 2, "Value in memory was wrong\n"); /* Close the mapping */ Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); Status = NtUnmapViewOfSection(NtCurrentProcess(), (PUCHAR)BaseAddress2 - PAGE_SIZE); ok_ntstatus(Status, STATUS_SUCCESS); Status = NtClose(SectionHandle); ok_ntstatus(Status, STATUS_SUCCESS); /* Try to create a 512 GB page file backed section with committed pages */ MaximumSize.QuadPart = 0x8000000000; Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_COMMIT, NULL); ok_ntstatus(Status, STATUS_COMMITMENT_LIMIT); /* Try to create a huge page file backed section with PAGE_NOACCESS protection */ MaximumSize.QuadPart = 0x8000000000; Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_NOACCESS, SEC_COMMIT, NULL); ok_ntstatus(Status, STATUS_INVALID_PAGE_PROTECTION); /* Try to create a very huge page file backed section, but only reserved */ MaximumSize.QuadPart = 0x80000000000; Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_RESERVE, NULL); #ifdef _WIN64 ok_ntstatus(Status, STATUS_INSUFFICIENT_RESOURCES); #else /* WoW64 returns STATUS_INSUFFICIENT_RESOURCES */ ok((Status == STATUS_INSUFFICIENT_RESOURCES) || (Status == STATUS_SECTION_TOO_BIG), "got wrong Status: 0x%lx\n", Status); #endif /* Try to create a even huger page file backed section, but only reserved */ MaximumSize.QuadPart = 0x800000000000; Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_RESERVE, NULL); ok_ntstatus(Status, STATUS_SECTION_TOO_BIG); /* Create a 8 GB page file backed section, but only reserved */ MaximumSize.QuadPart = 0x200000000; Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_RESERVE, NULL); ok_ntstatus(Status, STATUS_SUCCESS); /* Pass a too large region size */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = MAXULONG_PTR; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); #ifdef _WIN64 ok_ntstatus(Status, STATUS_INVALID_PARAMETER_3); #else /* WoW64 returns STATUS_INVALID_PARAMETER_4 */ ok((Status == STATUS_INVALID_PARAMETER_4) || (Status == STATUS_INVALID_PARAMETER_3), "got wrong Status: 0x%lx\n", Status); #endif /* Pass 0 region size */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = 0; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); #ifdef _WIN64 ok_ntstatus(Status, STATUS_SUCCESS); ok(ViewSize == 0x200000000, "wrong ViewSize: 0x%Ix\n", ViewSize); #else /* WoW64 returns STATUS_NO_MEMORY */ ok((Status == STATUS_NO_MEMORY) || (Status == STATUS_INVALID_VIEW_SIZE), "got wrong Status: 0x%lx\n", Status); ok(ViewSize == 0, "wrong ViewSize: 0x%Ix\n", ViewSize); #endif /* Map with PAGE_NOACCESS */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = 0x20000000; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_NOACCESS); ok_ntstatus(Status, STATUS_SUCCESS); /* Try to change protection to read/write */ ViewSize = 0x1000; OldProtect = -1; BaseAddress2 = BaseAddress; Status = NtProtectVirtualMemory(NtCurrentProcess(), &BaseAddress2, &ViewSize, PAGE_READWRITE, &OldProtect); ok_ntstatus(Status, STATUS_SECTION_PROTECTION); // Windows 2003 returns bogus //ok(OldProtect == PAGE_READWRITE, "Wrong protection returned: %u\n", OldProtect); /* Test read access */ Status = STATUS_SUCCESS; _SEH2_TRY { (void)(*(volatile char*)BaseAddress2); } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { Status = _SEH2_GetExceptionCode(); } _SEH2_END; ok_ntstatus(Status, STATUS_ACCESS_VIOLATION); /* Try to change protection to read/write */ ViewSize = 0x1000; OldProtect = -1; BaseAddress2 = BaseAddress; Status = NtProtectVirtualMemory(NtCurrentProcess(), &BaseAddress2, &ViewSize, PAGE_READWRITE, &OldProtect); ok_ntstatus(Status, STATUS_SECTION_PROTECTION); #ifdef _WIN64 ok(OldProtect == 0, "Wrong protection returned: 0x%lx\n", OldProtect); #else // Windows 2003 returns bogus #endif /* Try to change protection to readonly */ ViewSize = 0x1000; OldProtect = -1; BaseAddress2 = BaseAddress; Status = NtProtectVirtualMemory(NtCurrentProcess(), &BaseAddress2, &ViewSize, PAGE_READONLY, &OldProtect); ok_ntstatus(Status, STATUS_SECTION_PROTECTION); #ifdef _WIN64 //ok(OldProtect == 0, "Wrong protection returned: 0x%lx\n", OldProtect); #else // Windows 2003 returns bogus #endif /* Commit a page */ ViewSize = 0x1000; BaseAddress2 = BaseAddress; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress2, 0, &ViewSize, MEM_COMMIT, PAGE_READONLY); ok_ntstatus(Status, STATUS_SUCCESS); ok(BaseAddress2 == BaseAddress, "Invalid base address: %p\n", BaseAddress2); /* Commit the page again */ ViewSize = 0x1000; BaseAddress2 = BaseAddress; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress2, 0, &ViewSize, MEM_COMMIT, PAGE_READONLY); ok_ntstatus(Status, STATUS_SUCCESS); ok(BaseAddress2 == BaseAddress, "Invalid base address: %p\n", BaseAddress2); /* Test read access */ Status = STATUS_SUCCESS; _SEH2_TRY { (void)(*(volatile char*)BaseAddress2); } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { Status = _SEH2_GetExceptionCode(); } _SEH2_END; ok_ntstatus(Status, STATUS_SUCCESS); /* Test write access */ Status = STATUS_SUCCESS; _SEH2_TRY { *(char*)BaseAddress2 = 1; } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { Status = _SEH2_GetExceptionCode(); } _SEH2_END; ok_ntstatus(Status, STATUS_ACCESS_VIOLATION); /* Update protection to PAGE_READWRITE */ ViewSize = 0x1000; BaseAddress2 = BaseAddress; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress2, 0, &ViewSize, MEM_COMMIT, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); /* Test write access */ Status = STATUS_SUCCESS; _SEH2_TRY { *(char*)BaseAddress2 = 1; } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { Status = _SEH2_GetExceptionCode(); } _SEH2_END; ok_ntstatus(Status, STATUS_SUCCESS); /* Update protection to PAGE_EXECUTE_READWRITE (1 page) */ ViewSize = 0x1000; BaseAddress2 = BaseAddress; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress2, 0, &ViewSize, MEM_COMMIT, PAGE_EXECUTE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); Status = NtClose(SectionHandle); ok_ntstatus(Status, STATUS_SUCCESS); } void Test_ImageSection(void) { UNICODE_STRING FileName; NTSTATUS Status; OBJECT_ATTRIBUTES FileObjectAttributes; IO_STATUS_BLOCK IoStatusBlock; WCHAR TestDllPath[MAX_PATH]; HANDLE FileHandle, DataSectionHandle, ImageSectionHandle; PVOID DataBase, ImageBase; SIZE_T ViewSize; GetModuleFileNameW(NULL, TestDllPath, RTL_NUMBER_OF(TestDllPath)); wcsrchr(TestDllPath, L'\\')[1] = UNICODE_NULL; StringCbCatW(TestDllPath, sizeof(TestDllPath), L"testdata\\test.dll"); if (!RtlDosPathNameToNtPathName_U(TestDllPath, &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_ntstatus(Status, STATUS_SUCCESS); if (!NT_SUCCESS(Status)) { skip("Failed to open file\n"); return; } /* 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_ntstatus(Status, STATUS_SUCCESS); if (!NT_SUCCESS(Status)) { skip("Failed to create data section\n"); NtClose(FileHandle); return; } /* Map the data section as flat mapping */ DataBase = NULL; ViewSize = 0; Status = NtMapViewOfSection(DataSectionHandle, NtCurrentProcess(), &DataBase, 0, 0, NULL, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); //ok(ViewSize == 0x3f95cc48, "ViewSize wrong: 0x%lx\n"); if (!NT_SUCCESS(Status)) { skip("Failed to map view of data section\n"); NtClose(DataSectionHandle); NtClose(FileHandle); return; } /* Check the original data */ ok(*(ULONG*)DataBase == 0x00905a4d, "Header not ok\n"); /* Modify the PE header (but do not flush!) */ *(ULONG*)DataBase = 0xdeadbabe; ok(*(ULONG*)DataBase == 0xdeadbabe, "Header not ok\n"); /* Modify data in the .data section (but do not flush!) */ ok(*(ULONG*)((PUCHAR)DataBase + 0x800) == 0x12345678, "Data in .data section invalid: 0x%lx!\n", *(ULONG*)((PUCHAR)DataBase + 0x800)); *(ULONG*)((PUCHAR)DataBase + 0x800) = 0x87654321; /* Now try to create an image section (should fail) */ Status = NtCreateSection(&ImageSectionHandle, SECTION_ALL_ACCESS, // DesiredAccess NULL, // ObjectAttributes NULL, // MaximumSize PAGE_READWRITE, // SectionPageProtection SEC_IMAGE, // AllocationAttributes FileHandle); ok_ntstatus(Status, STATUS_INVALID_IMAGE_NOT_MZ); if (NT_SUCCESS(Status)) NtClose(ImageSectionHandle); /* Restore the original header */ *(ULONG*)DataBase = 0x00905a4d; /* Modify data in the .data section (but do not flush!) */ ok_hex(*(ULONG*)((PUCHAR)DataBase + 0x800), 0x87654321); *(ULONG*)((PUCHAR)DataBase + 0x800) = 0xdeadbabe; /* 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_ntstatus(Status, STATUS_SUCCESS); if (!NT_SUCCESS(Status)) { skip("Failed to create image section\n"); NtClose(DataSectionHandle); NtClose(FileHandle); return; } /* Map the image section */ ImageBase = NULL; ViewSize = 0; Status = NtMapViewOfSection(ImageSectionHandle, NtCurrentProcess(), &ImageBase, 0, // ZeroBits 0, // CommitSize NULL, // SectionOffset &ViewSize, ViewShare, 0, // AllocationType PAGE_READONLY); #ifdef _M_IX86 ok_ntstatus(Status, STATUS_SUCCESS); #else ok_ntstatus(Status, STATUS_IMAGE_MACHINE_TYPE_MISMATCH); #endif if (!NT_SUCCESS(Status)) { skip("Failed to map view of image section\n"); NtClose(ImageSectionHandle); NtClose(DataSectionHandle); NtClose(FileHandle); return; } /* Check the header */ ok(*(ULONG*)DataBase == 0x00905a4d, "Header not ok\n"); ok(*(ULONG*)ImageBase == 0x00905a4d, "Header not ok\n"); /* Check the data section. Either of these can be present! */ ok((*(ULONG*)((PUCHAR)ImageBase + 0x80000) == 0x87654321) || (*(ULONG*)((PUCHAR)ImageBase + 0x80000) == 0x12345678), "Wrong value in data section: 0x%lx!\n", *(ULONG*)((PUCHAR)ImageBase + 0x80000)); /* Now modify the data again */ *(ULONG*)DataBase = 0xdeadbabe; *(ULONG*)((PUCHAR)DataBase + 0x800) = 0xf00dada; /* Check the data */ ok(*(ULONG*)DataBase == 0xdeadbabe, "Header not ok\n"); ok(*(ULONG*)ImageBase == 0x00905a4d, "Data should not be synced!\n"); ok((*(ULONG*)((PUCHAR)ImageBase + 0x80000) == 0x87654321) || (*(ULONG*)((PUCHAR)ImageBase + 0x80000) == 0x12345678), "Wrong value in data section: 0x%lx!\n", *(ULONG*)((PUCHAR)ImageBase + 0x80000)); /* Flush the view */ ViewSize = 0x1000; Status = NtFlushVirtualMemory(NtCurrentProcess(), &DataBase, &ViewSize, &IoStatusBlock); ok_ntstatus(Status, STATUS_SUCCESS); /* Check the data again */ ok(*(ULONG*)ImageBase == 0x00905a4d, "Data should not be synced!\n"); ok((*(ULONG*)((PUCHAR)ImageBase + 0x80000) == 0x87654321) || (*(ULONG*)((PUCHAR)ImageBase + 0x80000) == 0x12345678), "Wrong value in data section: 0x%lx!\n", *(ULONG*)((PUCHAR)ImageBase + 0x80000)); /* Restore the original header */ *(ULONG*)DataBase = 0x00905a4d; ok(*(ULONG*)DataBase == 0x00905a4d, "Header not ok\n"); /* Close the image mapping */ NtUnmapViewOfSection(NtCurrentProcess(), ImageBase); NtClose(ImageSectionHandle); /* Create an image section again */ Status = NtCreateSection(&ImageSectionHandle, SECTION_ALL_ACCESS, // DesiredAccess NULL, // ObjectAttributes NULL, // MaximumSize PAGE_READWRITE, // SectionPageProtection SEC_IMAGE, // AllocationAttributes FileHandle); ok_ntstatus(Status, STATUS_SUCCESS); if (!NT_SUCCESS(Status)) { skip("Failed to create image section\n"); NtClose(DataSectionHandle); NtClose(FileHandle); return; } /* Map the image section again */ ImageBase = NULL; ViewSize = 0; Status = NtMapViewOfSection(ImageSectionHandle, NtCurrentProcess(), &ImageBase, 0, 0, NULL, &ViewSize, ViewShare, 0, PAGE_READONLY); #ifdef _M_IX86 ok_ntstatus(Status, STATUS_SUCCESS); #else ok_ntstatus(Status, STATUS_IMAGE_MACHINE_TYPE_MISMATCH); #endif if (!NT_SUCCESS(Status)) { skip("Failed to map view of image section\n"); NtClose(ImageSectionHandle); NtClose(DataSectionHandle); NtClose(FileHandle); return; } // This one doesn't always work, needs investigation /* Check the .data section again */ //ok(*(ULONG*)((PUCHAR)ImageBase + 0x80000) == 0xf00dada, // "Data should be synced: 0x%lx!\n", *(ULONG*)((PUCHAR)ImageBase + 0x80000)); /* Restore the original data */ *(ULONG*)((PUCHAR)DataBase + 0x800) = 0x12345678; /* Close the data mapping */ NtUnmapViewOfSection(NtCurrentProcess(), DataBase); NtClose(DataSectionHandle); /* Try to allocate memory inside the image mapping */ DataBase = (PUCHAR)ImageBase + 0x20000; ViewSize = 0x1000; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &DataBase, 0, &ViewSize, MEM_RESERVE, PAGE_NOACCESS); ok_ntstatus(Status, STATUS_CONFLICTING_ADDRESSES); /* Cleanup */ NtClose(FileHandle); NtClose(ImageSectionHandle); NtUnmapViewOfSection(NtCurrentProcess(), ImageBase); } void Test_ImageSection2(void) { UNICODE_STRING FileName; NTSTATUS Status; OBJECT_ATTRIBUTES FileObjectAttributes; IO_STATUS_BLOCK IoStatusBlock; HANDLE FileHandle, ImageSectionHandle; PVOID ImageBase, BaseAddress; SIZE_T ViewSize; LARGE_INTEGER MaximumSize, SectionOffset; if (!RtlDosPathNameToNtPathName_U(L"testdata\\nvoglv32.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_ntstatus(Status, STATUS_SUCCESS); printf("Opened file with handle %p\n", FileHandle); /* Create a data section with write access */ MaximumSize.QuadPart = 0x20000; Status = NtCreateSection(&ImageSectionHandle, SECTION_ALL_ACCESS, // DesiredAccess NULL, // ObjectAttributes &MaximumSize, // MaximumSize PAGE_READWRITE, // SectionPageProtection SEC_IMAGE, // AllocationAttributes FileHandle); ok_ntstatus(Status, STATUS_SUCCESS); printf("Created image section with handle %p\n", ImageSectionHandle); //system("PAUSE"); /* Map the image section */ ImageBase = NULL; ViewSize = 0x0000; SectionOffset.QuadPart = 0x00000; Status = NtMapViewOfSection(ImageSectionHandle, NtCurrentProcess(), &ImageBase, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); ok_ntstatus(Status, STATUS_SUCCESS); printf("Mapped image section at %p, value in text section: %lx\n", ImageBase, *((ULONG*)((PCHAR)ImageBase + 0x1196))); system("PAUSE"); /* Try to allocate a page after the section */ BaseAddress = (PUCHAR)ImageBase + 0x10000; ViewSize = 0x1000; Status = NtAllocateVirtualMemory(NtCurrentProcess(), &BaseAddress, 0, &ViewSize, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE); printf("allocation status: %lx\n", Status); system("PAUSE"); } // doesn't work with WoW64! void Test_BasedSection(void) { NTSTATUS Status; HANDLE SectionHandle1, SectionHandle2; LARGE_INTEGER MaximumSize, SectionOffset; PVOID BaseAddress1, BaseAddress2; SIZE_T ViewSize; /* Create a based section with SEC_COMMIT */ MaximumSize.QuadPart = 0x1000; Status = NtCreateSection(&SectionHandle1, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_COMMIT | SEC_BASED, NULL); ok_ntstatus(Status, STATUS_SUCCESS); /* Map the 1st section */ BaseAddress1 = NULL; SectionOffset.QuadPart = 0; ViewSize = 0; Status = NtMapViewOfSection(SectionHandle1, NtCurrentProcess(), &BaseAddress1, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); #if 0 // WOW64? ok_ntstatus(Status, STATUS_CONFLICTING_ADDRESSES); #else ok_ntstatus(Status, STATUS_SUCCESS); #endif /* Create a 2nd based section with SEC_COMMIT */ MaximumSize.QuadPart = 0x1000; Status = NtCreateSection(&SectionHandle2, SECTION_ALL_ACCESS, NULL, &MaximumSize, PAGE_READWRITE, SEC_COMMIT | SEC_BASED, NULL); ok_ntstatus(Status, STATUS_SUCCESS);// /* Map the 2nd section */ BaseAddress2 = NULL; SectionOffset.QuadPart = 0; ViewSize = 0; Status = NtMapViewOfSection(SectionHandle2, NtCurrentProcess(), &BaseAddress2, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); #if 0 // WOW64? ok_ntstatus(Status, STATUS_CONFLICTING_ADDRESSES); #else ok_ntstatus(Status, STATUS_SUCCESS); ok((ULONG_PTR)BaseAddress2 < (ULONG_PTR)BaseAddress1, "Invalid addresses: BaseAddress1=%p, BaseAddress2=%p\n", BaseAddress1, BaseAddress2); ok(((ULONG_PTR)BaseAddress1 - (ULONG_PTR)BaseAddress2) == 0x10000, "Invalid addresses: BaseAddress1=%p, BaseAddress2=%p\n", BaseAddress1, BaseAddress2); #endif } #define BYTES4(x) x, x, x, x #define BYTES8(x) BYTES4(x), BYTES4(x) #define BYTES16(x) BYTES8(x), BYTES8(x) #define BYTES32(x) BYTES16(x), BYTES16(x) #define BYTES64(x) BYTES32(x), BYTES32(x) #define BYTES128(x) BYTES64(x), BYTES64(x) #define BYTES256(x) BYTES128(x), BYTES128(x) #define BYTES512(x) BYTES256(x), BYTES256(x) #define BYTES1024(x) BYTES512(x), BYTES512(x) static struct _MY_IMAGE_FILE { IMAGE_DOS_HEADER doshdr; WORD stub[32]; IMAGE_NT_HEADERS32 nthdrs; IMAGE_SECTION_HEADER text_header; IMAGE_SECTION_HEADER rossym_header; IMAGE_SECTION_HEADER rsrc_header; IMAGE_SECTION_HEADER clc_header; BYTE pad[488]; BYTE text_data[0x400]; BYTE rossym_data[0x400]; BYTE rsrc_data[0x400]; BYTE clc_data[0x1000]; } ImageFile = { /* IMAGE_DOS_HEADER */ { IMAGE_DOS_SIGNATURE, 144, 3, 0, 4, 0, 0xFFFF, 0, 0xB8, 0, 0, 0, 0x40, 0, { 0 }, 0, 0, { 0 }, 0x80 }, /* binary to print "This program cannot be run in DOS mode." */ { 0x1F0E, 0x0EBA, 0xB400, 0xCD09, 0xB821, 0x4C01, 0x21CD, 0x6854, 0x7369, 0x7020, 0x6F72, 0x7267, 0x6D61, 0x6320, 0x6E61, 0x6F6E, 0x2074, 0x6562, 0x7220, 0x6E75, 0x6920, 0x206E, 0x4F44, 0x2053, 0x6F6D, 0x6564, 0x0D2E, 0x0A0D, 0x0024, 0x0000, 0x0000, 0x0000 }, /* IMAGE_NT_HEADERS32 */ { IMAGE_NT_SIGNATURE, /* Signature */ /* IMAGE_FILE_HEADER */ { IMAGE_FILE_MACHINE_I386, /* Machine */ 4, /* NumberOfSections */ 0x47EFDF09, /* TimeDateStamp */ 0, /* PointerToSymbolTable */ 0, /* NumberOfSymbols */ 0xE0, /* SizeOfOptionalHeader */ IMAGE_FILE_32BIT_MACHINE | IMAGE_FILE_LOCAL_SYMS_STRIPPED | IMAGE_FILE_LINE_NUMS_STRIPPED | IMAGE_FILE_EXECUTABLE_IMAGE | IMAGE_FILE_DLL, /* Characteristics */ }, /* IMAGE_OPTIONAL_HEADER32 */ { IMAGE_NT_OPTIONAL_HDR32_MAGIC, /* Magic */ 8, /* MajorLinkerVersion */ 0, /* MinorLinkerVersion */ 0x400, /* SizeOfCode */ 0x000, /* SizeOfInitializedData */ 0, /* SizeOfUninitializedData */ 0x2000, /* AddressOfEntryPoint */ 0x2000, /* BaseOfCode */ 0x0000, /* BaseOfData */ 0x400000, /* ImageBase */ 0x2000, /* SectionAlignment */ 0x200, /* FileAlignment */ 4, /* MajorOperatingSystemVersion */ 0, /* MinorOperatingSystemVersion */ 0, /* MajorImageVersion */ 0, /* MinorImageVersion */ 4, /* MajorSubsystemVersion */ 0, /* MinorSubsystemVersion */ 0, /* Win32VersionValue */ 0xa000, /* SizeOfImage */ 0x400, /* SizeOfHeaders */ 0x0, /* CheckSum */ IMAGE_SUBSYSTEM_WINDOWS_CUI, /* Subsystem */ IMAGE_DLLCHARACTERISTICS_DYNAMIC_BASE | IMAGE_DLLCHARACTERISTICS_NO_SEH | IMAGE_DLLCHARACTERISTICS_NX_COMPAT, /* DllCharacteristics */ 0x100000, /* SizeOfStackReserve */ 0x1000, /* SizeOfStackCommit */ 0x100000, /* SizeOfHeapReserve */ 0x1000, /* SizeOfHeapCommit */ 0, /* LoaderFlags */ 0x10, /* NumberOfRvaAndSizes */ /* IMAGE_DATA_DIRECTORY */ { { 0 }, /* Export Table */ { 0 }, /* Import Table */ { 0 }, /* Resource Table */ { 0 }, /* Exception Table */ { 0 }, /* Certificate Table */ { 0 }, /* Base Relocation Table */ { 0 }, /* Debug */ { 0 }, /* Copyright */ { 0 }, /* Global Ptr */ { 0 }, /* TLS Table */ { 0 }, /* Load Config Table */ { 0 }, /* Bound Import */ { 0 }, /* IAT */ { 0 }, /* Delay Import Descriptor */ { 0 }, /* CLI Header */ { 0 } /* Reserved */ } } }, /* IMAGE_SECTION_HEADER */ { ".text", /* Name */ { 0x394 }, /* Misc.VirtualSize */ 0x2000, /* VirtualAddress */ 0x400, /* SizeOfRawData */ 0x400, /* PointerToRawData */ 0, /* PointerToRelocations */ 0, /* PointerToLinenumbers */ 0, /* NumberOfRelocations */ 0, /* NumberOfLinenumbers */ IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_EXECUTE | IMAGE_SCN_CNT_CODE, /* Characteristics */ }, /* IMAGE_SECTION_HEADER */ { ".rossym", /* Name */ { 0x100 }, /* Misc.VirtualSize */ 0x4000, /* VirtualAddress */ 0x400, /* SizeOfRawData */ 0x800, /* PointerToRawData */ 0, /* PointerToRelocations */ 0, /* PointerToLinenumbers */ 0, /* NumberOfRelocations */ 0, /* NumberOfLinenumbers */ /* CORE-8384 */ IMAGE_SCN_MEM_READ | IMAGE_SCN_TYPE_NOLOAD, /* Characteristics */ }, /* IMAGE_SECTION_HEADER */ { ".rsrc", /* Name */ { 0x100 }, /* Misc.VirtualSize */ 0x6000, /* VirtualAddress */ 0x400, /* SizeOfRawData */ 0xC00, /* PointerToRawData */ 0, /* PointerToRelocations */ 0, /* PointerToLinenumbers */ 0, /* NumberOfRelocations */ 0, /* NumberOfLinenumbers */ IMAGE_SCN_MEM_READ | IMAGE_SCN_CNT_INITIALIZED_DATA, /* Characteristics */ }, /* IMAGE_SECTION_HEADER */ { ".clc", /* Name */ { 0x2000 }, /* Misc.VirtualSize */ 0x8000, /* VirtualAddress */ 0x1000, /* SizeOfRawData */ 0x1000, /* PointerToRawData */ 0, /* PointerToRelocations */ 0, /* PointerToLinenumbers */ 0, /* NumberOfRelocations */ 0, /* NumberOfLinenumbers */ /* CORE-12582 */ IMAGE_SCN_MEM_WRITE | IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_EXECUTE | IMAGE_SCN_CNT_UNINITIALIZED_DATA | IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_CNT_CODE, /* Characteristics */ }, /* fill */ { 0 }, /* text */ { 0xc3, 0 }, /* rossym */ { 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, BYTES8(0xaa), BYTES16(0xbb), BYTES32(0xcc), BYTES64(0xdd), BYTES64(0xee), BYTES64(0xff), }, /* rsrc */ { BYTES128(0xee), BYTES128(0x55), BYTES128(0xee), BYTES128(0x11), BYTES128(0xff), BYTES128(0x00), BYTES128(0x00), BYTES128(0xdd), }, /* clc */ { BYTES512(0x11), BYTES512(0x22), BYTES512(0x33), BYTES512(0x44), BYTES512(0x55), BYTES512(0x66), BYTES512(0x77), BYTES512(0x88), }, }; C_ASSERT(FIELD_OFFSET(struct _MY_IMAGE_FILE, text_data) == 0x400); C_ASSERT(FIELD_OFFSET(struct _MY_IMAGE_FILE, rossym_data) == 0x800); C_ASSERT(FIELD_OFFSET(struct _MY_IMAGE_FILE, rsrc_data) == 0xc00); C_ASSERT(FIELD_OFFSET(struct _MY_IMAGE_FILE, clc_data) == 0x1000); static void Test_SectionContents(BOOL Relocate) { NTSTATUS Status; WCHAR TempPath[MAX_PATH]; WCHAR FileName[MAX_PATH]; HANDLE Handle; HANDLE SectionHandle; LARGE_INTEGER SectionOffset; PVOID BaseAddress; SIZE_T ViewSize; ULONG Written; ULONG Length; BOOL Success; Length = GetTempPathW(MAX_PATH, TempPath); ok(Length != 0, "GetTempPathW failed with %lu\n", GetLastError()); Length = GetTempFileNameW(TempPath, L"nta", 0, FileName); ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError()); Handle = CreateFileW(FileName, FILE_ALL_ACCESS, 0, NULL, CREATE_ALWAYS, 0, NULL); if (Handle == INVALID_HANDLE_VALUE) { skip("Failed to create temp file %ls, error %lu\n", FileName, GetLastError()); return; } if (Relocate) { ok((ULONG_PTR)GetModuleHandle(NULL) <= 0x80000000, "Module at %p\n", GetModuleHandle(NULL)); ImageFile.nthdrs.OptionalHeader.ImageBase = (ULONG)(ULONG_PTR)GetModuleHandle(NULL); } else { ImageFile.nthdrs.OptionalHeader.ImageBase = 0xe400000; } Success = WriteFile(Handle, &ImageFile, sizeof(ImageFile), &Written, NULL); ok(Success == TRUE, "WriteFile failed with %lu\n", GetLastError()); ok(Written == sizeof(ImageFile), "WriteFile wrote %lu bytes\n", Written); Status = NtCreateSection(&SectionHandle, SECTION_ALL_ACCESS, NULL, NULL, PAGE_EXECUTE_READWRITE, SEC_IMAGE, Handle); ok_ntstatus(Status, STATUS_SUCCESS); if (NT_SUCCESS(Status)) { /* Map the section with */ BaseAddress = NULL; SectionOffset.QuadPart = 0; ViewSize = 0; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, &SectionOffset, &ViewSize, ViewShare, 0, PAGE_READWRITE); if (Relocate) ok_ntstatus(Status, STATUS_IMAGE_NOT_AT_BASE); else ok_ntstatus(Status, STATUS_SUCCESS); if (NT_SUCCESS(Status)) { PUCHAR Bytes = BaseAddress; #define TEST_BYTE(n, v) StartSeh() ok_hex(Bytes[n], v); EndSeh(STATUS_SUCCESS); #define TEST_WRITE(n) StartSeh() *(volatile UCHAR *)&Bytes[n] = Bytes[n]; EndSeh(STATUS_SUCCESS); #define TEST_NOWRITE(n) StartSeh() *(volatile UCHAR *)&Bytes[n] = Bytes[n]; EndSeh(STATUS_ACCESS_VIOLATION); TEST_NOWRITE(0x2000); TEST_BYTE(0x2000, 0xc3); TEST_BYTE(0x2001, 0x00); TEST_NOWRITE(0x4000); TEST_BYTE(0x4000, 0x01); TEST_BYTE(0x4001, 0x23); TEST_BYTE(0x4007, 0xef); TEST_BYTE(0x4008, 0xaa); TEST_BYTE(0x4010, 0xbb); TEST_BYTE(0x4020, 0xcc); TEST_BYTE(0x4040, 0xdd); TEST_BYTE(0x4080, 0xee); TEST_BYTE(0x40c0, 0xff); TEST_BYTE(0x40ff, 0xff); TEST_BYTE(0x4100, 0x00); TEST_BYTE(0x41ff, 0x00); TEST_NOWRITE(0x6000); TEST_BYTE(0x6000, 0xee); TEST_BYTE(0x60ff, 0x55); TEST_BYTE(0x6100, 0xee); TEST_BYTE(0x63ff, 0xdd); TEST_BYTE(0x6400, 0x00); TEST_WRITE(0x8000); TEST_BYTE(0x8000, 0x11); TEST_BYTE(0x8400, 0x33); Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); } Status = NtClose(SectionHandle); ok_ntstatus(Status, STATUS_SUCCESS); } CloseHandle(Handle); DeleteFileW(FileName); } static void Test_EmptyFile(VOID) { NTSTATUS Status; WCHAR TempPath[MAX_PATH]; WCHAR FileName[MAX_PATH]; HANDLE Handle; HANDLE SectionHandle; ULONG Length; Length = GetTempPathW(MAX_PATH, TempPath); ok(Length != 0, "GetTempPathW failed with %lu\n", GetLastError()); Length = GetTempFileNameW(TempPath, L"nta", 0, FileName); ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError()); Handle = CreateFileW(FileName, FILE_ALL_ACCESS, 0, NULL, CREATE_ALWAYS, 0, NULL); if (Handle == INVALID_HANDLE_VALUE) { skip("Failed to create temp file %ls, error %lu\n", FileName, GetLastError()); return; } Status = NtCreateSection(&SectionHandle, STANDARD_RIGHTS_REQUIRED | SECTION_QUERY | SECTION_MAP_READ, 0, 0, PAGE_READONLY, SEC_COMMIT, Handle); ok_ntstatus(Status, STATUS_MAPPED_FILE_SIZE_ZERO); if (NT_SUCCESS(Status)) NtClose(SectionHandle); Status = NtCreateSection(&SectionHandle, STANDARD_RIGHTS_REQUIRED | SECTION_QUERY | SECTION_MAP_READ, 0, 0, PAGE_READONLY, SEC_IMAGE, Handle); ok_ntstatus(Status, STATUS_INVALID_FILE_FOR_SECTION); if (NT_SUCCESS(Status)) NtClose(SectionHandle); CloseHandle(Handle); DeleteFileW(FileName); } // CORE-11206 static void Test_Truncate(VOID) { WCHAR TempPath[MAX_PATH]; WCHAR FileName[MAX_PATH]; NTSTATUS Status; SIZE_T ViewSize = 0; HANDLE Handle; HANDLE SectionHandle; ULONG Length; BOOL Success; DWORD Written, Error; VOID* BaseAddress; Length = GetTempPathW(MAX_PATH, TempPath); ok(Length != 0, "GetTempPathW failed with %lu\n", GetLastError()); Length = GetTempFileNameW(TempPath, L"nta", 0, FileName); ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError()); Handle = CreateFileW(FileName, FILE_ALL_ACCESS, 0, NULL, CREATE_ALWAYS, 0, NULL); Success = WriteFile(Handle, "TESTDATA", 8, &Written, NULL); ok(Success == TRUE, "WriteFile failed with %lu\n", GetLastError()); ok(Written == 8, "WriteFile wrote %lu bytes\n", Written); Written = SetFilePointer(Handle, 6, NULL, FILE_BEGIN); ok(Written == 6, "SetFilePointer returned %lu bytes\n", Written); Success = SetEndOfFile(Handle); ok(Success == TRUE, "SetEndOfFile failed with %lu\n", GetLastError()); Status = NtCreateSection(&SectionHandle, STANDARD_RIGHTS_REQUIRED | SECTION_QUERY | SECTION_MAP_READ, 0, 0, PAGE_READONLY, SEC_COMMIT, Handle); ok_ntstatus(Status, STATUS_SUCCESS); BaseAddress = NULL; ViewSize = 0; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, 0, &ViewSize, ViewShare, 0, PAGE_READONLY); ok_ntstatus(Status, STATUS_SUCCESS); if (BaseAddress) { // First we test data that was truncated even before the file mapping was opened Length = strlen((char*)BaseAddress); ok(Length == 6, "Old data was not properly erased! (Length=%lu)\n", Length); } // Now we truncate the file on disk some more Written = SetFilePointer(Handle, 4, NULL, FILE_BEGIN); ok(Written == 4, "SetFilePointer returned %lu bytes\n", Written); Success = SetEndOfFile(Handle); Error = GetLastError(); ok(Success == FALSE, "SetEndOfFile succeeded\n"); ok(Error == ERROR_USER_MAPPED_FILE, "SetEndOfFile did not set error to ERROR_USER_MAPPED_FILE (%lu)\n", Error); if (BaseAddress) { Length = strlen((char*)BaseAddress); ok(Length == 6, "Length should not have changed! (Length=%lu)\n", Length); } // Unmap and set the end shorter. Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); Success = CloseHandle(SectionHandle); ok(Success == TRUE, "CloseHandle failed with %lu\n", GetLastError()); Written = SetFilePointer(Handle, 4, NULL, FILE_BEGIN); ok(Written == 4, "SetFilePointer returned %lu bytes\n", Written); Success = SetEndOfFile(Handle); ok(Success == TRUE, "SetEndOfFile failed with %lu\n", GetLastError()); Status = NtCreateSection(&SectionHandle, STANDARD_RIGHTS_REQUIRED | SECTION_QUERY | SECTION_MAP_READ, 0, 0, PAGE_READONLY, SEC_COMMIT, Handle); ok_ntstatus(Status, STATUS_SUCCESS); BaseAddress = NULL; ViewSize = 0; Status = NtMapViewOfSection(SectionHandle, NtCurrentProcess(), &BaseAddress, 0, 0, 0, &ViewSize, ViewShare, 0, PAGE_READONLY); ok_ntstatus(Status, STATUS_SUCCESS); // CLEANUP Status = NtUnmapViewOfSection(NtCurrentProcess(), BaseAddress); ok_ntstatus(Status, STATUS_SUCCESS); Success = CloseHandle(SectionHandle); ok(Success == TRUE, "CloseHandle failed with %lu\n", GetLastError()); Success = CloseHandle(Handle); ok(Success == TRUE, "CloseHandle failed with %lu\n", GetLastError()); Success = DeleteFileW(FileName); ok(Success == TRUE, "DeleteFileW failed with %lu\n", GetLastError()); } START_TEST(NtMapViewOfSection) { Test_PageFileSection(); Test_ImageSection(); Test_BasedSection(); Test_SectionContents(FALSE); Test_SectionContents(TRUE); Test_EmptyFile(); Test_Truncate(); }