coldcard/firmware
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle corrupt DFU files (bad magic)

Browse Source
This commit is contained in:
Peter D. Gray 2023-12-08 14:34:10 -05:00
parent 33cc1f0b2d
commit d330c09e60
No known key found for this signature in database
GPG Key ID: A2DCD558C2BE5D7C
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
shared/actions.py
View File

@ -237,18 +237,20 @@ async def microsd_upgrade(menu, label, item):
failed = None
with CardSlot(force_vdisk=force_vdisk) as card:
with card.open(fn, 'rb') as fp:
offset, size = dfu_parse(fp)
try:
offset, size = dfu_parse(fp)
# we also put a copy of special signed heaer at the end of the flash
# read just the signature header
hdr = bytearray(FW_HEADER_SIZE)
fp.seek(offset + FW_HEADER_OFFSET)
rv = fp.readinto(hdr)
assert rv == FW_HEADER_SIZE
# read just the signature header
hdr = bytearray(FW_HEADER_SIZE)
fp.seek(offset + FW_HEADER_OFFSET)
rv = fp.readinto(hdr)
assert rv == FW_HEADER_SIZE
# check header values
failed = check_firmware_hdr(hdr, size)
# check header values
failed = check_firmware_hdr(hdr, size)
except Exception as exc:
# recover from garbage DFU files.
failed = "Failed: " + str(exc)
if not failed:
# copy binary into PSRAM