mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-18 13:38:12 +00:00
41bf9201b5
As the connected Trezors can be left in various weird states, trezorctl list may not always return what is expected or even fail. If it failed TREZOR_PATH was set to empty string which means random device got selected. For now let's avoid using incorrect device by powering down the other usb port. Fix log lines ordering.
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
import datetime
|
|
import sys
|
|
import time
|
|
from subprocess import run
|
|
|
|
|
|
class Device:
|
|
def __init__(self, uhub_location, device_port):
|
|
self.uhub_location = uhub_location
|
|
self.device_port = device_port
|
|
|
|
@staticmethod
|
|
def log(msg):
|
|
print(msg, flush=True, file=sys.stderr)
|
|
|
|
def run_trezorctl(self, cmd: str, **kwargs):
|
|
full_cmd = "trezorctl "
|
|
full_cmd += cmd
|
|
self.log("[software/trezorctl] Running '{}'".format(full_cmd))
|
|
return run(full_cmd, shell=True, check=True, **kwargs)
|
|
|
|
def check_model(self, model=None):
|
|
res = self.run_trezorctl("list", capture_output=True, text=True).stdout
|
|
self.log(res)
|
|
self.run_trezorctl("get-features | grep version")
|
|
lines = res.splitlines()
|
|
if len(lines) != 1:
|
|
raise RuntimeError("{} trezors connected".format(len(lines)))
|
|
if model and model not in lines[0]:
|
|
raise RuntimeError(
|
|
"invalid trezor model connected (expected {})".format(model)
|
|
)
|
|
return lines[0].split()[0]
|
|
|
|
def reboot(self):
|
|
self.power_off()
|
|
self.power_on()
|
|
|
|
def power_on(self):
|
|
self.now()
|
|
self.log("[hardware/usb] Turning power on...")
|
|
run(
|
|
"uhubctl -l {} -p {} -a on".format(self.uhub_location, self.device_port),
|
|
shell=True,
|
|
check=True,
|
|
)
|
|
self.wait(3)
|
|
|
|
def power_off(self):
|
|
self.now()
|
|
self.log("[hardware/usb] Turning power off...")
|
|
run(
|
|
"uhubctl -l {} -p {} -r 100 -a off".format(
|
|
self.uhub_location, self.device_port
|
|
),
|
|
shell=True,
|
|
check=True,
|
|
)
|
|
self.wait(3)
|
|
|
|
def touch(self, location, action):
|
|
raise NotImplementedError
|
|
|
|
@staticmethod
|
|
def wait(seconds):
|
|
Device.now()
|
|
Device.log("[software] Waiting for {} seconds...".format(seconds))
|
|
time.sleep(seconds)
|
|
|
|
@staticmethod
|
|
def now():
|
|
Device.log("\n[timestamp] {}".format(datetime.datetime.now()))
|