From e18e3c4305537ee18a0b61676b0a8f30efedac19 Mon Sep 17 00:00:00 2001
From: Luca Boccassi <bluca@debian.org>
Date: Tue, 23 May 2023 01:43:59 +0100
Subject: [PATCH] elf2efi: add parameter to increase reserved space for headers

When building a minimal empty addon it would not have enough
space to append sections. Add an option that will later be
used to reserve enough space.
---
 tools/elf2efi.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tools/elf2efi.py b/tools/elf2efi.py
index 6179ba82132..20d25321c4d 100755
--- a/tools/elf2efi.py
+++ b/tools/elf2efi.py
@@ -512,10 +512,11 @@ def elf2efi(args: argparse.Namespace):
     opt.SizeOfImage = align_to(
         sections[-1].VirtualAddress + sections[-1].VirtualSize, SECTION_ALIGNMENT
     )
+
     opt.SizeOfHeaders = align_to(
         PE_OFFSET
         + coff.SizeOfOptionalHeader
-        + sizeof(PeSection) * coff.NumberOfSections,
+        + sizeof(PeSection) * max(coff.NumberOfSections, args.minimum_sections),
         FILE_ALIGNMENT,
     )
     # DYNAMIC_BASE|NX_COMPAT|HIGH_ENTROPY_VA or DYNAMIC_BASE|NX_COMPAT
@@ -578,6 +579,12 @@ def main():
         type=argparse.FileType("wb"),
         help="Output PE/EFI file",
     )
+    parser.add_argument(
+        "--minimum-sections",
+        type=int,
+        default=0,
+        help="Minimum number of sections to leave space for",
+    )
 
     elf2efi(parser.parse_args())