unleashed-firmware/scripts/bin2dfu.py

72 lines
1.9 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
import os
import struct
from zlib import crc32
from flipper.app import App
class Main(App):
def init(self):
self.parser.add_argument("-i", "--input", help=".bin input path", required=True)
self.parser.add_argument(
"-o", "--output", help=".dfu output path", required=True
)
self.parser.add_argument(
"-a",
"--address",
help="Flash address",
type=lambda x: int(x, 0),
required=True,
)
self.parser.add_argument(
"-l", "--label", help="DFU Target label", required=True
)
self.parser.add_argument(
"--vid", help="USB Vendor ID", default=0x0483, type=lambda x: int(x, 0)
)
self.parser.add_argument(
"--pid", help="USB Product ID", default=0xDF11, type=lambda x: int(x, 0)
)
self.parser.set_defaults(func=self.convert)
def convert(self):
if not os.path.exists(self.args.input):
self.logger.error(f'"{self.args.input}" does not exist')
return 1
with open(self.args.input, mode="rb") as file:
bindata = file.read()
data = struct.pack("<II", self.args.address, len(bindata)) + bindata
# Target prefix
szTargetName = self.args.label.encode("ascii")
data = (
struct.pack("<6sBI255sII", b"Target", 0, 1, szTargetName, len(data), 1)
+ data
)
# Prefix
data = struct.pack("<5sBIB", b"DfuSe", 0x01, len(data) + 11, 1) + data
# Suffix
data += struct.pack(
"<HHHH3sB", 0xFFFF, self.args.pid, self.args.vid, 0x011A, b"UFD", 16
)
dwCRC = ~crc32(data) & 0xFFFFFFFF
data += struct.pack("<I", dwCRC)
with open(self.args.output, "wb") as file:
file.write(data)
return 0
if __name__ == "__main__":
Main()()