diff --git a/fw.py b/fw.py index 03f089a..a40d9f0 100755 --- a/fw.py +++ b/fw.py @@ -27,6 +27,12 @@ def main(cmd): fw2 = Firmware.load(argv[3]) fw.compare(fw2) + if cmd == 'enc': + fw.write() + + if cmd == 'dec': + fw.write_raw() + if cmd == 'search': fw.search(argv[3].encode()) diff --git a/uvk5.py b/uvk5.py index 9c70f9d..f45a75f 100755 --- a/uvk5.py +++ b/uvk5.py @@ -153,6 +153,13 @@ class Firmware(bytearray): os.write(1, encrypted) + def write_raw(self, path=None): + if path: + pass + else: + os.write(1, self) + + class FirmwareModifiable(Firmware): def apply_mods(self, mod_names): for mod in mod_names: