arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-06-26 01:42:34 +00:00
Code Issues Releases Wiki Activity

firmware_update command

Browse Source
This commit is contained in:
slush0 2013-10-12 17:41:55 +02:00
parent 02f6dd9681
commit 0b54b50ba2
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
trezorlib/client.py
View File

@ -127,10 +127,10 @@ class TrezorClient(object):
if isinstance(resp, proto.Failure): if isinstance(resp, proto.Failure):
self.message_func(resp.message) self.message_func(resp.message)
if resp.code == 4: if resp.code == proto.Failure_ActionCancelled:
raise CallException("Action cancelled by user") raise CallException("Action cancelled by user")
elif resp.code == 6: elif resp.code == proto.Failure_PinInvalid:
raise PinException("PIN is invalid") raise PinException("PIN is invalid")
raise CallException(resp.code, resp.message) raise CallException(resp.code, resp.message)
@ -259,3 +259,16 @@ class TrezorClient(object):
resp = self.call(proto.LoadDevice(seed=seed, pin=pin)) resp = self.call(proto.LoadDevice(seed=seed, pin=pin))
self.init_device() self.init_device()
return isinstance(resp, proto.Success) return isinstance(resp, proto.Success)
def firmware_update(self, fp, force=False):
if self.features.bootloader_mode == False:
raise Exception("Device must be in bootloader mode")
resp = self.call(proto.FirmwareUpdate(force=force, payload=fp.read()))
if isinstance(resp, proto.Success):
return True
elif isinstance(resp, proto.Failure) and resp.code == proto.Failure_FirmwareDataIncompatibility:
return False
raise Exception("Unexpected result " % resp)