78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from flipper.app import App
|
||
|
|
from flipper.utils.fff import FlipperFormatFile
|
||
|
|
from os.path import basename, join, exists
|
||
|
|
from os import makedirs
|
||
|
|
import shutil
|
||
|
|
import zlib
|
||
|
|
|
||
|
|
|
||
|
|
class Main(App):
|
||
|
|
def init(self):
|
||
|
|
self.subparsers = self.parser.add_subparsers(help="sub-command help")
|
||
|
|
|
||
|
|
# generate
|
||
|
|
self.parser_generate = self.subparsers.add_parser(
|
||
|
|
"generate", help="Generate update description file"
|
||
|
|
)
|
||
|
|
|
||
|
|
self.parser_generate.add_argument("-d", dest="directory", required=True)
|
||
|
|
self.parser_generate.add_argument("-v", dest="version", required=True)
|
||
|
|
self.parser_generate.add_argument("-t", dest="target", required=True)
|
||
|
|
self.parser_generate.add_argument("-dfu", dest="dfu", required=True)
|
||
|
|
self.parser_generate.add_argument("-stage", dest="stage", required=True)
|
||
|
|
self.parser_generate.add_argument("-radio", dest="radiobin", required=False)
|
||
|
|
self.parser_generate.add_argument(
|
||
|
|
"-radioaddr", dest="radioaddr", required=False
|
||
|
|
)
|
||
|
|
|
||
|
|
self.parser_generate.set_defaults(func=self.generate)
|
||
|
|
|
||
|
|
def generate(self):
|
||
|
|
stage_basename = basename(self.args.stage)
|
||
|
|
dfu_basename = basename(self.args.dfu)
|
||
|
|
|
||
|
|
if not exists(self.args.directory):
|
||
|
|
makedirs(self.args.directory)
|
||
|
|
|
||
|
|
shutil.copyfile(self.args.stage, join(self.args.directory, stage_basename))
|
||
|
|
shutil.copyfile(self.args.dfu, join(self.args.directory, dfu_basename))
|
||
|
|
|
||
|
|
file = FlipperFormatFile()
|
||
|
|
file.setHeader("Flipper firmware upgrade configuration", 1)
|
||
|
|
file.writeKey("Info", self.args.version)
|
||
|
|
file.writeKey("Target", self.args.target[1:]) # dirty 'f' strip
|
||
|
|
file.writeKey("Loader", stage_basename)
|
||
|
|
file.writeComment("little-endian hex!")
|
||
|
|
file.writeKey("Loader CRC", self.int2ffhex(self.crc(self.args.stage)))
|
||
|
|
file.writeKey("Firmware", dfu_basename)
|
||
|
|
file.writeKey("Radio", self.args.radiobin or "")
|
||
|
|
file.writeKey("Radio address", self.int2ffhex(self.args.radioaddr or 0))
|
||
|
|
file.save("%s/update.fuf" % self.args.directory)
|
||
|
|
|
||
|
|
return 0
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def int2ffhex(value: int):
|
||
|
|
hexstr = "%08X" % value
|
||
|
|
return " ".join(list(Main.batch(hexstr, 2))[::-1])
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def crc(fileName):
|
||
|
|
prev = 0
|
||
|
|
with open(fileName, "rb") as file:
|
||
|
|
for eachLine in file:
|
||
|
|
prev = zlib.crc32(eachLine, prev)
|
||
|
|
return prev & 0xFFFFFFFF
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def batch(iterable, n=1):
|
||
|
|
l = len(iterable)
|
||
|
|
for ndx in range(0, l, n):
|
||
|
|
yield iterable[ndx : min(ndx + n, l)]
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
Main()()
|