+++ /dev/null
-/*
- * PROJECT: ReactOS api tests
- * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
- * PURPOSE: Test for RtlImageRvaToVa
- * PROGRAMMERS: Thomas Faber <thomas.faber@reactos.org>
- */
-
-#include <apitest.h>
-#include <ndk/mmfuncs.h>
-#include <ndk/rtlfuncs.h>
-
-static
-PVOID
-AllocateGuarded(
- _In_ SIZE_T SizeRequested)
-{
- NTSTATUS Status;
- SIZE_T Size = PAGE_ROUND_UP(SizeRequested + PAGE_SIZE);
- PVOID VirtualMemory = NULL;
- PCHAR StartOfBuffer;
-
- Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_RESERVE, PAGE_NOACCESS);
-
- if (!NT_SUCCESS(Status))
- return NULL;
-
- Size -= PAGE_SIZE;
- if (Size)
- {
- Status = NtAllocateVirtualMemory(NtCurrentProcess(), &VirtualMemory, 0, &Size, MEM_COMMIT, PAGE_READWRITE);
- if (!NT_SUCCESS(Status))
- {
- Size = 0;
- Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
- ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
- return NULL;
- }
- }
-
- StartOfBuffer = VirtualMemory;
- StartOfBuffer += Size - SizeRequested;
-
- return StartOfBuffer;
-}
-
-static
-VOID
-FreeGuarded(
- _In_ PVOID Pointer)
-{
- NTSTATUS Status;
- PVOID VirtualMemory = (PVOID)PAGE_ROUND_DOWN((SIZE_T)Pointer);
- SIZE_T Size = 0;
-
- Status = NtFreeVirtualMemory(NtCurrentProcess(), &VirtualMemory, &Size, MEM_RELEASE);
- ok(Status == STATUS_SUCCESS, "Status = %lx\n", Status);
-}
-
-START_TEST(RtlImageRvaToVa)
-{
- PIMAGE_NT_HEADERS NtHeader;
- PIMAGE_SECTION_HEADER Section;
- ULONG NumberOfSections;
- ULONG ImageSize;
- PUCHAR BaseAddress;
- PVOID Va;
- PIMAGE_SECTION_HEADER OutSection;
- IMAGE_SECTION_HEADER DummySection;
- struct
- {
- ULONG Rva;
- ULONG VaOffset;
- ULONG SectionIndex;
- } Tests[] =
- {
- { 0, 0, 0 },
- { 0xfff, 0, 0 },
- { 0x1000, 0x3000, 0 },
- { 0x1001, 0x3001, 0 },
- { 0x1fff, 0x3fff, 0 },
- { 0x2000, 0, 0 },
- { 0x2fff, 0, 0 },
- { 0x3000, 0x4000, 3 },
- { 0x3fff, 0x4fff, 3 },
- { 0x4000, 0x5000, 3 },
- { 0x4fff, 0x5fff, 3 },
- { 0x5000, 0, 0 },
- { 0x5fff, 0, 0 },
- { 0x6000, 0, 0 },
- { 0x6fff, 0, 0 },
- { 0x7000, 0x7000, 5 },
- { 0x7fff, 0x7fff, 5 },
- { 0x8000, 0x9000, 7 },
- { 0x8fff, 0x9fff, 7 },
- { 0x9000, 0x8000, 6 },
- { 0x9fff, 0x8fff, 6 },
- };
- ULONG i;
-
- NumberOfSections = 8;
- ImageSize = FIELD_OFFSET(IMAGE_NT_HEADERS, OptionalHeader.DataDirectory) +
- NumberOfSections * sizeof(IMAGE_SECTION_HEADER);
- NtHeader = AllocateGuarded(ImageSize);
- if (!NtHeader)
- {
- skip("Could not allocate %lu bytes\n", ImageSize);
- return;
- }
-
- RtlFillMemory(NtHeader, ImageSize, 0xDD);
- NtHeader->FileHeader.NumberOfSections = NumberOfSections;
- NtHeader->FileHeader.SizeOfOptionalHeader = FIELD_OFFSET(IMAGE_OPTIONAL_HEADER, DataDirectory);
- Section = (PIMAGE_SECTION_HEADER)((PUCHAR)&NtHeader->OptionalHeader +
- NtHeader->FileHeader.SizeOfOptionalHeader);
- Section[0].VirtualAddress = 0x1000;
- Section[0].Misc.VirtualSize = 0x1000;
- Section[0].SizeOfRawData = 0x1000;
- Section[0].PointerToRawData = 0x3000;
- Section[1].VirtualAddress = 0x2000;
- Section[1].Misc.VirtualSize = 0;
- Section[1].SizeOfRawData = 0;
- Section[1].PointerToRawData = 0x4000;
- Section[2].VirtualAddress = 0x2000;
- Section[2].Misc.VirtualSize = 0x1000;
- Section[2].SizeOfRawData = 0;
- Section[2].PointerToRawData = 0x4000;
- Section[3].VirtualAddress = 0x3000;
- Section[3].Misc.VirtualSize = 0x1000;
- Section[3].SizeOfRawData = 0x2000;
- Section[3].PointerToRawData = 0x4000;
- Section[4].VirtualAddress = 0x4000;
- Section[4].Misc.VirtualSize = 0x2000;
- Section[4].SizeOfRawData = 0x1000;
- Section[4].PointerToRawData = 0x6000;
- Section[5].VirtualAddress = 0x7000;
- Section[5].Misc.VirtualSize = 0x1000;
- Section[5].SizeOfRawData = 0x1000;
- Section[5].PointerToRawData = 0x7000;
- Section[6].VirtualAddress = 0x9000;
- Section[6].Misc.VirtualSize = 0x1000;
- Section[6].SizeOfRawData = 0x1000;
- Section[6].PointerToRawData = 0x8000;
- Section[7].VirtualAddress = 0x8000;
- Section[7].Misc.VirtualSize = 0x1000;
- Section[7].SizeOfRawData = 0x1000;
- Section[7].PointerToRawData = 0x9000;
- DummySection.VirtualAddress = 0xf000;
- DummySection.Misc.VirtualSize = 0xf000;
- DummySection.SizeOfRawData = 0xf000;
- DummySection.PointerToRawData = 0xf000;
-
- BaseAddress = (PUCHAR)0x2000000;
-
- StartSeh()
- RtlImageRvaToVa(NULL, NULL, 0, NULL);
- EndSeh(STATUS_ACCESS_VIOLATION);
-
- Va = RtlImageRvaToVa(NtHeader, NULL, 0, NULL);
- ok(Va == NULL, "Va = %p\n", Va);
-
- Va = RtlImageRvaToVa(NtHeader, BaseAddress, 0, NULL);
- ok(Va == NULL, "Va = %p\n", Va);
-
- OutSection = NULL;
- Va = RtlImageRvaToVa(NtHeader, BaseAddress, 0, &OutSection);
- ok(Va == NULL, "Va = %p\n", Va);
- ok(OutSection == NULL, "OutSection = %p\n", OutSection);
-
- OutSection = (PVOID)1;
- StartSeh()
- RtlImageRvaToVa(NtHeader, BaseAddress, 0, &OutSection);
- EndSeh(STATUS_ACCESS_VIOLATION);
-
- for (i = 0; i < RTL_NUMBER_OF(Tests); i++)
- {
- /* Section = not specified */
- StartSeh()
- Va = RtlImageRvaToVa(NtHeader, BaseAddress, Tests[i].Rva, NULL);
- EndSeh(STATUS_SUCCESS);
- if (Tests[i].VaOffset == 0)
- ok(Va == NULL, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
- else
- ok(Va == BaseAddress + Tests[i].VaOffset, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
-
- /* Section = NULL */
- OutSection = NULL;
- StartSeh()
- Va = RtlImageRvaToVa(NtHeader, BaseAddress, Tests[i].Rva, &OutSection);
- EndSeh(STATUS_SUCCESS);
- if (Tests[i].VaOffset == 0)
- {
- ok(Va == NULL, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
- ok(OutSection == NULL, "[0x%lx] OutSection = %p (%Id)\n", Tests[i].Rva, OutSection, OutSection - Section);
- }
- else
- {
- ok(Va == BaseAddress + Tests[i].VaOffset, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
- ok(OutSection == &Section[Tests[i].SectionIndex], "[0x%lx] OutSection = %p (%Id)\n", Tests[i].Rva, OutSection, OutSection - Section);
- }
-
- /* Section = first section */
- OutSection = Section;
- StartSeh()
- Va = RtlImageRvaToVa(NtHeader, BaseAddress, Tests[i].Rva, &OutSection);
- EndSeh(STATUS_SUCCESS);
- if (Tests[i].VaOffset == 0)
- {
- ok(Va == NULL, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
- ok(OutSection == Section, "[0x%lx] OutSection = %p (%Id)\n", Tests[i].Rva, OutSection, OutSection - Section);
- }
- else
- {
- ok(Va == BaseAddress + Tests[i].VaOffset, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
- ok(OutSection == &Section[Tests[i].SectionIndex], "[0x%lx] OutSection = %p (%Id)\n", Tests[i].Rva, OutSection, OutSection - Section);
- }
-
- /* Section = dummy section */
- OutSection = &DummySection;
- StartSeh()
- Va = RtlImageRvaToVa(NtHeader, BaseAddress, Tests[i].Rva, &OutSection);
- EndSeh(STATUS_SUCCESS);
- if (Tests[i].VaOffset == 0)
- {
- ok(Va == NULL, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
- ok(OutSection == &DummySection, "[0x%lx] OutSection = %p (%Id)\n", Tests[i].Rva, OutSection, OutSection - Section);
- }
- else
- {
- ok(Va == BaseAddress + Tests[i].VaOffset, "[0x%lx] Va = %p\n", Tests[i].Rva, Va);
- ok(OutSection == &Section[Tests[i].SectionIndex], "[0x%lx] OutSection = %p (%Id)\n", Tests[i].Rva, OutSection, OutSection - Section);
- }
- }
-
- FreeGuarded(NtHeader);
-}