Author: tfaber Date: Mon Nov 16 09:48:39 2015 New Revision: 69895
URL: http://svn.reactos.org/svn/reactos?rev=69895&view=rev Log: [NTDLL_APITEST] - Add a test for RtlImageRvaToVa CORE-10523
Added: trunk/rostests/apitests/ntdll/RtlImageRvaToVa.c (with props) Modified: trunk/rostests/apitests/ntdll/CMakeLists.txt 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] Mon Nov 16 09:48:39 2015 @@ -31,6 +31,7 @@ RtlGetFullPathName_UstrEx.c RtlGetLengthWithoutTrailingPathSeperators.c RtlGetLongestNtPathLength.c + RtlImageRvaToVa.c RtlInitializeBitMap.c RtlMemoryStream.c RtlReAllocateHeap.c
Added: trunk/rostests/apitests/ntdll/RtlImageRvaToVa.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ntdll/RtlImageRva... ============================================================================== --- trunk/rostests/apitests/ntdll/RtlImageRvaToVa.c (added) +++ trunk/rostests/apitests/ntdll/RtlImageRvaToVa.c [iso-8859-1] Mon Nov 16 09:48:39 2015 @@ -0,0 +1,235 @@ +/* + * 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); +}
Propchange: trunk/rostests/apitests/ntdll/RtlImageRvaToVa.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] Mon Nov 16 09:48:39 2015 @@ -35,6 +35,7 @@ extern void func_RtlGetFullPathName_UstrEx(void); extern void func_RtlGetLengthWithoutTrailingPathSeperators(void); extern void func_RtlGetLongestNtPathLength(void); +extern void func_RtlImageRvaToVa(void); extern void func_RtlInitializeBitMap(void); extern void func_RtlMemoryStream(void); extern void func_RtlReAllocateHeap(void); @@ -75,6 +76,7 @@ { "RtlGetFullPathName_UstrEx", func_RtlGetFullPathName_UstrEx }, { "RtlGetLengthWithoutTrailingPathSeperators", func_RtlGetLengthWithoutTrailingPathSeperators }, { "RtlGetLongestNtPathLength", func_RtlGetLongestNtPathLength }, + { "RtlImageRvaToVa", func_RtlImageRvaToVa }, { "RtlInitializeBitMap", func_RtlInitializeBitMap }, { "RtlMemoryStream", func_RtlMemoryStream }, { "RtlReAllocateHeap", func_RtlReAllocateHeap },
-
tfaber＠svn.reactos.org