support v2 firmware in trezorctl

2017-06-20 17:17:33 +02:00 · 2017-06-20 17:17:33 +02:00 · e435920d6a
parent 609b8d4a4a
commit e435920d6a
1 changed files with 2 additions and 2 deletions
--- a/4
+++ b/4
@ -370,9 +370,9 @@ class Commands(object):
            return self.firmware_update(args)

        if not args.skip_check:
-            if fp[:8] == b'54525a52':
+            if fp[:8] == b'54525a52' or fp[:8] == b'54525a56':
                fp = binascii.unhexlify(fp)
-            if fp[:4] != b'TRZR':
+            if fp[:4] != b'TRZR' and fp[:4] != b'TRZV':
                raise CallException(types.Failure_FirmwareError, "TREZOR firmware header expected")

        print("Please confirm action on device...")