Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-10 06:25:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

[FREELDR] In PE loader, use cache to not load twice the start of the file

Browse source 
This is especially important in PXE boot, where a seek backwards requires reloading the file from start.
This commit is contained in:
Hervé Poussineau 2021-01-27 23:41:49 +01:00
parent 9d110db433
commit 5904361a54
1 changed files with 13 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
boot/freeldr/freeldr/lib/peloader.c
View file

@ -781,24 +781,19 @@ PeLdrLoadImage(
TRACE("Base PA: 0x%X, VA: 0x%X\n", PhysicalBase, VirtualBase);
/* Set to 0 position and fully load the file image */
Position.QuadPart = 0;
Status = ArcSeek(FileId, &Position, SeekAbsolute);
if (Status != ESUCCESS)
/* Copy headers from already read data */
RtlCopyMemory(PhysicalBase, HeadersBuffer, min(NtHeaders->OptionalHeader.SizeOfHeaders, sizeof(HeadersBuffer)));
/* If headers are quite big, request next bytes from file */
if (NtHeaders->OptionalHeader.SizeOfHeaders > sizeof(HeadersBuffer))
{
ERR("ArcSeek(File: '%s') failed. Status: 0x%lx\n", FileName, Status);
UiMessageBox("Error seeking the start of a file.");
ArcClose(FileId);
return FALSE;
}
Status = ArcRead(FileId, PhysicalBase, NtHeaders->OptionalHeader.SizeOfHeaders, &BytesRead);
if (Status != ESUCCESS)
{
ERR("ArcRead(File: '%s') failed. Status: %u\n", FileName, Status);
UiMessageBox("Error reading headers.");
ArcClose(FileId);
return FALSE;
Status = ArcRead(FileId, (PUCHAR)PhysicalBase + sizeof(HeadersBuffer), NtHeaders->OptionalHeader.SizeOfHeaders - sizeof(HeadersBuffer), &BytesRead);
if (Status != ESUCCESS)
{
ERR("ArcRead(File: '%s') failed. Status: %u\n", FileName, Status);
UiMessageBox("Error reading headers.");
ArcClose(FileId);
return FALSE;
}
}
/*
@ -843,7 +838,7 @@ PeLdrLoadImage(
if (SizeOfRawData != 0)
{
/* Seek to the correct position */
Position.LowPart = SectionHeader->PointerToRawData;
Position.QuadPart = SectionHeader->PointerToRawData;
Status = ArcSeek(FileId, &Position, SeekAbsolute);
TRACE("SH->VA: 0x%X\n", SectionHeader->VirtualAddress);