arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-30 09:11:07 +00:00
Code Issues Releases Wiki Activity

fix(tools): Strip old header in firmware_hash.py.

Browse Source
This commit is contained in:
Andrew Kozlik 2022-05-11 10:46:51 +02:00 committed by Andrew Kozlik
parent 55bfab3976
commit 11cb171e1a
1 changed files with 27 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tools/firmware_hash.py
View File

@ -6,24 +6,42 @@ from hashlib import blake2s
FILE_T1 = "legacy/firmware/trezor.bin"
FILE_T2 = "core/build/firmware/firmware.bin"
T1_HEADER_MAGIC_OLD = b"TRZR"
T1_HEADER_OLD_SIZE = 256
T2_HEADER_MAGIC_VENDOR = b"TRZV"
SIZE_T1 = (7 * 128 + 64) * 1024
SIZE_T2 = 13 * 128 * 1024
for filename, size in ((FILE_T1, SIZE_T1), (FILE_T2, SIZE_T2)):
if len(sys.argv) > 2:
filenames = sys.argv[2:]
elif len(sys.argv) == 2:
filenames = (FILE_T1, FILE_T2)
else:
print(f"Usage: {sys.argv[0]} HEX_CHALLENGE [FILE]...")
print(f" HEX_CHALLENGE: a 0-32 byte challenge in hexadecimal")
exit(1)
for filename in filenames:
try:
data = open(filename, "rb").read()
except FileNotFoundError:
print(f"{filename} not found")
continue
if len(data) > size:
raise ValueError(filename, "too big")
data = data + b"\xff" * (size - len(data))
if len(sys.argv) > 1:
challenge = bytes.fromhex(sys.argv[1])
firmware_hash = blake2s(data, key=challenge).hexdigest()
offset = 0
if data[:4] == T2_HEADER_MAGIC_VENDOR:
size = SIZE_T2
else:
firmware_hash = blake2s(data).hexdigest()
size = SIZE_T1
if data[:4] == T1_HEADER_MAGIC_OLD:
offset = T1_HEADER_OLD_SIZE
if len(data) - offset > size:
raise ValueError(filename, "too big")
data = data[offset:] + b"\xff" * (size - len(data) + offset)
challenge = bytes.fromhex(sys.argv[1])
firmware_hash = blake2s(data, key=challenge).hexdigest()
print(f"{filename}: {firmware_hash}")