nina-fw/combine.py
2023-12-14 12:14:39 -05:00

66 lines
1.9 KiB
Python

#!/usr/bin/env python
import re
import sys
def extract_firmware_version():
with open('main/CommandHandler.cpp', 'r') as file:
for line in file:
if 'const char FIRMWARE_VERSION[6] = ' in line:
# The line format is `const char FIRMWARE_VERSION[6] = "1.7.6";`
# Split by double quote and get the second element
version = line.split('"')[1]
return version
booloaderData = open("build/bootloader/bootloader.bin", "rb").read()
partitionData = open("build/partitions.bin", "rb").read()
appData = open("build/nina-fw.bin", "rb").read()
# remove everything between certificate markers to save space. There might be comments and other information.
certsData = b""
with open("certificates/data/roots.pem", "rb") as certs_file:
in_cert = False
for line in certs_file:
if line.startswith(b"-----BEGIN CERTIFICATE-----"):
in_cert = True
if in_cert:
certsData += line
if line.startswith(b"-----END CERTIFICATE-----"):
in_cert = False
# calculate the output binary size, app offset
outputSize = 0x30000 + len(appData)
if outputSize % 1024:
outputSize += 1024 - (outputSize % 1024)
# allocate and init to 0xff
outputData = bytearray(b"\xff") * outputSize
# copy data: bootloader, partitions, app
for i in range(0, len(booloaderData)):
outputData[0x1000 + i] = booloaderData[i]
for i in range(0, len(partitionData)):
outputData[0x8000 + i] = partitionData[i]
for i in range(0, len(appData)):
outputData[0x30000 + i] = appData[i]
for i in range(0, len(certsData)):
outputData[0x10000 + i] = certsData[i]
# zero terminate the pem file
outputData[0x10000 + len(certsData)] = 0
version = extract_firmware_version()
outputFilename = f"NINA_W102-{version}.bin"
if len(sys.argv) > 1:
outputFilename = sys.argv[1]
# write out
with open(outputFilename, "w+b") as f:
f.seek(0)
f.write(outputData)