tyler.durden/audk
1
0
Fork 0
mirror of https://github.com/acidanthera/audk.git synced 2025-09-23 17:57:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

ImageTool: Check HeaderInfo

Browse Source
This commit is contained in:
Marvin Häuser 2023-06-13 01:25:08 +02:00 committed by MikhailKrichanov
parent 31c0046cbe
commit f374c858df
1 changed files with 38 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
BaseTools/ImageTool/Image.c
View File

@ -92,6 +92,28 @@ CheckToolImageSegmentInfo (
return true;
}
static
bool
CheckToolImageHeaderInfo (
const image_tool_header_info_t *HeaderInfo,
const image_tool_segment_info_t *SegmentInfo,
uint32_t ImageSize
)
{
if (SegmentInfo->Segments[0].ImageAddress > HeaderInfo->EntryPointAddress ||
HeaderInfo->EntryPointAddress > ImageSize) {
DEBUG_RAISE ();
return false;
}
if (!IS_ALIGNED (HeaderInfo->BaseAddress, SegmentInfo->SegmentAlignment)) {
DEBUG_RAISE ();
return false;
}
return true;
}
const image_tool_segment_t *
ImageGetSegmentByAddress (
uint32_t *Address,
@ -296,6 +318,16 @@ CheckToolImage (
return false;
}
Result = CheckToolImageHeaderInfo (
&Image->HeaderInfo,
&Image->SegmentInfo,
ImageSize
);
if (!Result) {
DEBUG_RAISE ();
return false;
}
Result = CheckToolImageRelocInfo (Image, ImageSize);
if (!Result) {
DEBUG_RAISE ();
@ -371,10 +403,15 @@ ToolImageRelocate (
uint32_t RelocTarget32;
uint64_t RelocTarget64;
if (!IS_ALIGNED (BaseAddress, Image->SegmentInfo.SegmentAlignment)) {
DEBUG_RAISE ();
return false;
}
Adjust = BaseAddress - Image->HeaderInfo.BaseAddress;
if (Adjust == 0) {
return TRUE;
return true;
}
for (Index = 0; Index < Image->RelocInfo.NumRelocs; ++Index) {