BaseTools: Fixed '-D EDK2_REDUCE_FW_SIZE' build option.

acidanthera · Nov 3, 2023 · 3dff9d7 · 3dff9d7
1 parent 70265b6
commit 3dff9d7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/BaseTools/ImageTool/PeEmit.c b/BaseTools/ImageTool/PeEmit.c
@@ -588,6 +588,12 @@ ToolImageEmitPeFile (
   SizeOfPeHeaders      = SectionHeadersOffset + SectionHeadersSize;
   SizeOfHeaders        = sizeof (mDosHdr) + SizeOfPeHeaders;
   AlignedSizeOfHeaders = ALIGN_VALUE (SizeOfHeaders, FileAlignment);
+  //
+  // Necessary adjustment for '-D EDK2_REDUCE_FW_SIZE' build option.
+  //
+  if (AlignedSizeOfHeaders < Image->SegmentInfo.Segments[0].ImageAddress) {
+    AlignedSizeOfHeaders = (uint32_t)Image->SegmentInfo.Segments[0].ImageAddress;
+  }
 
   PeOffset = ImageToolBufferAppendReserve (
                Buffer,
@@ -623,7 +629,7 @@ ToolImageEmitPeFile (
   PeHdr->ImageBase           = (EFI_IMAGE_NT_BASE_ADDRESS)Image->HeaderInfo.BaseAddress;
   PeHdr->SectionAlignment    = Image->SegmentInfo.SegmentAlignment;
   PeHdr->FileAlignment       = FileAlignment;
-  PeHdr->SizeOfImage         = ALIGN_VALUE (SizeOfHeaders, PeHdr->SectionAlignment);
+  PeHdr->SizeOfImage         = AlignedSizeOfHeaders;
   PeHdr->SizeOfHeaders       = AlignedSizeOfHeaders;
   PeHdr->Subsystem           = Image->HeaderInfo.Subsystem;
   PeHdr->NumberOfRvaAndSizes = EFI_IMAGE_NUMBER_OF_DIRECTORY_ENTRIES;