mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-30 18:10:56 +00:00
31 lines
860 B
Python
31 lines
860 B
Python
from .device import Device
|
|
|
|
|
|
class TrezorCore(Device):
|
|
def update_firmware(self, file=None, model_name="Trezor T"):
|
|
if not file:
|
|
raise ValueError(
|
|
"Uploading production firmware will replace the bootloader, it is not allowed!"
|
|
)
|
|
|
|
# reset to enter bootloader again
|
|
self.power_off()
|
|
self.wait(5)
|
|
self.power_on()
|
|
|
|
self.wait(10)
|
|
self.check_model("bootloader")
|
|
|
|
self.run_trezorctl("device wipe --bootloader || true")
|
|
self.wait(5)
|
|
self.power_off()
|
|
self.power_on()
|
|
|
|
self.wait(5)
|
|
self.log(f"[software] Updating the firmware to {file}")
|
|
self.run_trezorctl(f"firmware-update -s -f {file}")
|
|
|
|
# after firmware-update finishes wait for reboot
|
|
self.wait(15)
|
|
return self.check_model(model_name)
|