arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-04-15 06:45:59 +00:00
Code Issues Releases Wiki Activity

add FirmwareErase message

Browse Source
This commit is contained in:
Pavol Rusnak 2013-10-22 11:29:33 +02:00
parent f863e1e602
commit c01830462a
3 changed files with 110 additions and 69 deletions

10
protobuf/trezor.proto
View File

@ -18,7 +18,8 @@ enum MessageType {
MessageType_Failure = 3;
MessageType_ChangePin = 4;
MessageType_WipeDevice = 5;
MessageType_FirmwareUpdate = 6;
MessageType_FirmwareErase = 6;
MessageType_FirmwareUpload = 7;
MessageType_GetEntropy = 9;
MessageType_Entropy = 10;
MessageType_GetMasterPublicKey = 11;
@ -71,7 +72,7 @@ enum FailureType {
Failure_PinExpected = 5;
Failure_PinCancelled = 6;
Failure_PinInvalid = 7;
Failure_FirmwareDataIncompatibility = 99;
Failure_FirmwareError = 99;
}
// Specifies which script will be used for given transaction output.
@ -349,7 +350,10 @@ message TxOutput {
// Bootloader messages
//
message FirmwareUpdate {
message FirmwareErase {
}
message FirmwareUpload {
required bytes payload = 1 [(binary) = true]; // Firmware to flash into device
}

8
trezorlib/client.py
View File

@ -264,11 +264,15 @@ class TrezorClient(object):
if self.features.bootloader_mode == False:
raise Exception("Device must be in bootloader mode")
resp = self.call(proto.FirmwareUpdate(payload=fp.read()))
resp = self.call(proto.FirmwareErase())
if isinstance(resp, proto.Failure) and resp.code == proto.Failure_FirmwareError:
return False
resp = self.call(proto.FirmwareUpload(payload=fp.read()))
if isinstance(resp, proto.Success):
return True
elif isinstance(resp, proto.Failure) and resp.code == proto.Failure_FirmwareDataIncompatibility:
elif isinstance(resp, proto.Failure) and resp.code == proto.Failure_FirmwareError:
return False
raise Exception("Unexpected result " % resp)

161
trezorlib/trezor_pb2.py
View File

File diff suppressed because one or more lines are too long