feat(core/tools): make combine_firmware script a bit smarter and more opinionated

core/Makefile

@@ -321,10 +321,10 @@ sizecheck: ## check sizes of binary files
 
 combine: ## combine boardloader + bootloader + prodtest into one combined image
 	./tools/combine_firmware \
-		$(BOARDLOADER_START) $(BOARDLOADER_BUILD_DIR)/boardloader.bin \
-		$(BOOTLOADER_START) $(BOOTLOADER_BUILD_DIR)/bootloader.bin \
-		$(PRODTEST_START) $(PRODTEST_BUILD_DIR)/prodtest.bin \
-		> $(PRODTEST_BUILD_DIR)/combined.bin
+		$(BOARDLOADER_BUILD_DIR)/boardloader.bin \
+		$(BOOTLOADER_BUILD_DIR)/bootloader.bin \
+		$(PRODTEST_BUILD_DIR)/prodtest.bin \
+		$(PRODTEST_BUILD_DIR)/combined.bin
 
 upload: ## upload firmware using trezorctl
 	trezorctl firmware_update -f $(FIRMWARE_BUILD_DIR)/firmware.bin

core/tools/combine_firmware

@@ -1,26 +1,73 @@
 #!/usr/bin/env python3
+from __future__ import annotations
+
+import datetime
+import io
 import sys
+from pathlib import Path
+
+import click
+
+
+BOARDLOADER_START = 0x08000000
+BOARDLOADER_END = 0x0800C000
+BOOTLOADER_START = 0x08020000
+FIRMWARE_START = 0x08040000
+
+
+@click.command()
+@click.argument(
+    "boardloader",
+    type=click.Path(exists=True, dir_okay=False, readable=True, path_type=Path),
+)
+@click.argument(
+    "bootloader",
+    type=click.Path(exists=True, dir_okay=False, readable=True, path_type=Path),
+)
+@click.argument(
+    "firmware",
+    type=click.Path(exists=True, dir_okay=False, readable=True, path_type=Path),
+)
+@click.argument(
+    "outfile",
+    type=click.Path(dir_okay=False, writable=True, path_type=Path),
+    required=False,
+)
+def main(
+    boardloader: Path, bootloader: Path, firmware: Path, outfile: Path | None
+) -> None:
+    if outfile is None:
+        today = datetime.date.today().strftime(r"%Y-%m-%d")
+        outfile = Path(f"combined-{today}.bin")
+
+    offset = BOARDLOADER_START
+    out_bytes = io.BytesIO()
 
+    # write boardloader
+    offset += out_bytes.write(boardloader.read_bytes())
+    if offset > BOARDLOADER_END:
+        raise Exception("Boardloader too big")
 
-def pairwise(iterable):
-    a = iter(iterable)
-    return zip(a, a)
+    # zero-pad until next section:
+    offset += out_bytes.write(b"\x00" * (BOOTLOADER_START - offset))
+    assert offset == BOOTLOADER_START
 
+    # write bootlaoder
+    offset += out_bytes.write(bootloader.read_bytes())
+    if offset > FIRMWARE_START:
+        raise Exception("Bootloader too big")
 
-files = sys.argv[1:]
-files = list(pairwise(files))
+    # zero-pad until next section:
+    offset += out_bytes.write(b"\x00" * (FIRMWARE_START - offset))
+    assert offset == FIRMWARE_START
 
-offset = int(files[0][0], 16)
+    # write firmware
+    offset += out_bytes.write(firmware.read_bytes())
 
-out = bytearray()
+    # write out contents
+    click.echo(f"Writing {outfile} ({offset - BOARDLOADER_START} bytes)")
+    outfile.write_bytes(out_bytes.getvalue())
 
-for addr, fn in files:
-    addr = int(addr, 16) - offset
-    data = open(fn, "rb").read()
-    if len(out) < addr:
-        out += b"\x00" * (addr - len(out))
-    if len(out) != addr:
-        raise Exception("Alignment failed")
-    out += data
 
-sys.stdout.buffer.write(out)
+if __name__ == "__main__":
+    main()