Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-04-28 01:11:35 +00:00
Code Issues Projects Releases Packages Wiki Activity

[RTL] Fix RtlImageDirectoryEntryToData to support both 32 and 64 bit images

Browse source
This commit is contained in:
Timo Kreuzer 2020-10-04 20:07:34 +02:00
parent cac7b003c0
commit 883ae6d93b
1 changed files with 29 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
sdk/lib/rtl/image.c
View file

@ -289,17 +289,38 @@ RtlImageDirectoryEntryToData(
if (NtHeader == NULL)
return NULL;
if (Directory >= SWAPD(NtHeader->OptionalHeader.NumberOfRvaAndSizes))
return NULL;
if (NtHeader->OptionalHeader.Magic == IMAGE_NT_OPTIONAL_HDR64_MAGIC)
{
PIMAGE_OPTIONAL_HEADER64 OptionalHeader = (PIMAGE_OPTIONAL_HEADER64)&NtHeader->OptionalHeader;
Va = SWAPD(NtHeader->OptionalHeader.DataDirectory[Directory].VirtualAddress);
if (Va == 0)
return NULL;
if (Directory >= SWAPD(OptionalHeader->NumberOfRvaAndSizes))
return NULL;
*Size = SWAPD(NtHeader->OptionalHeader.DataDirectory[Directory].Size);
Va = SWAPD(OptionalHeader->DataDirectory[Directory].VirtualAddress);
if (Va == 0)
return NULL;
if (MappedAsImage || Va < SWAPD(NtHeader->OptionalHeader.SizeOfHeaders))
return (PVOID)((ULONG_PTR)BaseAddress + Va);
*Size = SWAPD(OptionalHeader->DataDirectory[Directory].Size);
if (MappedAsImage || Va < SWAPD(OptionalHeader->SizeOfHeaders))
return (PVOID)((ULONG_PTR)BaseAddress + Va);
}
else
{
PIMAGE_OPTIONAL_HEADER32 OptionalHeader = (PIMAGE_OPTIONAL_HEADER32)&NtHeader->OptionalHeader;
if (Directory >= SWAPD(OptionalHeader->NumberOfRvaAndSizes))
return NULL;
Va = SWAPD(OptionalHeader->DataDirectory[Directory].VirtualAddress);
if (Va == 0)
return NULL;
*Size = SWAPD(OptionalHeader->DataDirectory[Directory].Size);
if (MappedAsImage || Va < SWAPD(OptionalHeader->SizeOfHeaders))
return (PVOID)((ULONG_PTR)BaseAddress + Va);
}
/* Image mapped as ordinary file, we must find raw pointer */
return RtlImageRvaToVa(NtHeader, BaseAddress, Va, NULL);