2020-09-03 09:11:21 +00:00
|
|
|
from .device import Device
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TrezorT(Device):
|
|
|
|
|
def update_firmware(self, file=None):
|
|
|
|
|
if not file:
|
|
|
|
|
raise ValueError(
|
|
|
|
|
"Uploading production firmware will replace the bootloader, it is not allowed!"
|
|
|
|
|
)
|
|
|
|
|
|
2021-02-15 12:23:23 +00:00
|
|
|
# reset to enter bootloader again
|
|
|
|
|
self.power_off()
|
|
|
|
|
self.power_on()
|
|
|
|
|
|
2021-01-23 22:03:56 +00:00
|
|
|
self.wait(5)
|
2021-02-15 12:23:23 +00:00
|
|
|
self.check_model("Trezor T bootloader")
|
|
|
|
|
|
2021-01-23 22:03:56 +00:00
|
|
|
self.run_trezorctl("device wipe --bootloader || true")
|
|
|
|
|
self.wait(5)
|
|
|
|
|
self.power_off()
|
|
|
|
|
self.power_on()
|
|
|
|
|
|
2020-09-03 09:11:21 +00:00
|
|
|
self.wait(5)
|
2021-01-21 23:18:34 +00:00
|
|
|
self.log("[software] Updating the firmware to {}".format(file))
|
2020-12-04 16:16:57 +00:00
|
|
|
self.run_trezorctl("firmware-update -s -f {}".format(file))
|
2020-09-03 09:11:21 +00:00
|
|
|
|
2020-12-04 16:16:57 +00:00
|
|
|
# after firmware-update finishes wait for reboot
|
|
|
|
|
self.wait(15)
|
2021-01-21 23:18:34 +00:00
|
|
|
return self.check_model("Trezor T")
|
