1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-25 07:48:10 +00:00
trezor-firmware/tests/input_flows.py
2023-07-17 15:30:00 +02:00

1800 lines
55 KiB
Python

"""
Central place for defining all input flows for the device tests.
Each model has potentially its own input flow, and in most cases
we need to distinguish between them. Doing it at one place
offers a better overview of the differences and makes it easier
to maintain. The whole `device_tests` folder can then focus
only on the actual tests and data-assertions, not on the lower-level
input flow details.
"""
import time
from typing import Callable, Generator, Optional
from trezorlib import messages
from trezorlib.debuglink import (
DebugLink,
LayoutContent,
TrezorClientDebugLink as Client,
multipage_content,
)
from . import buttons
from .common import (
check_pin_backoff_time,
click_info_button_tt,
click_through,
read_and_confirm_mnemonic,
recovery_enter_shares,
)
GeneratorType = Generator[None, messages.ButtonRequest, None]
B = messages.ButtonRequestType
def swipe_if_necessary(
debug: DebugLink, br_code: Optional[messages.ButtonRequestType] = None
) -> GeneratorType:
br = yield
if br_code is not None:
assert br.code == br_code
if br.pages is not None:
for _ in range(br.pages - 1):
debug.swipe_up()
class InputFlowBase:
def __init__(self, client: Client):
self.client = client
self.debug: DebugLink = client.debug
self.layout = client.debug.wait_layout
def model(self) -> Optional[str]:
return self.client.features.model
def get(self) -> Callable[[], GeneratorType]:
self.client.watch_layout(True)
# There could be one common input flow for all models
if hasattr(self, "input_flow_common"):
return getattr(self, "input_flow_common")
elif self.model() == "T":
return self.input_flow_tt
elif self.model() == "R":
return self.input_flow_tr
else:
raise ValueError("Unknown model")
def input_flow_tt(self) -> GeneratorType:
"""Special for TT"""
raise NotImplementedError
def input_flow_tr(self) -> GeneratorType:
"""Special for TR"""
raise NotImplementedError
class InputFlowSetupDevicePINWIpeCode(InputFlowBase):
def __init__(self, client: Client, pin: str, wipe_code: str):
super().__init__(client)
self.pin = pin
self.wipe_code = wipe_code
def input_flow_common(self) -> GeneratorType:
yield # do you want to set/change the wipe code?
self.debug.press_yes()
if self.debug.model == "R":
yield from swipe_if_necessary(self.debug) # wipe code info
self.debug.press_yes()
yield # enter current pin
self.debug.input(self.pin)
yield # enter new wipe code
self.debug.input(self.wipe_code)
yield # enter new wipe code again
self.debug.input(self.wipe_code)
yield # success
self.debug.press_yes()
class InputFlowNewCodeMismatch(InputFlowBase):
def __init__(
self,
client: Client,
first_code: str,
second_code: str,
):
super().__init__(client)
self.first_code = first_code
self.second_code = second_code
def input_flow_common(self) -> GeneratorType:
yield # do you want to set/change the pin/wipe code?
self.debug.press_yes()
if self.debug.model == "R":
yield from swipe_if_necessary(self.debug) # code info
self.debug.press_yes()
def input_two_different_pins():
yield # enter new PIN/wipe_code
self.debug.input(self.first_code)
if self.debug.model == "R":
yield # Please re-enter PIN to confirm
self.debug.press_yes()
yield # enter new PIN/wipe_code again (but different)
self.debug.input(self.second_code)
yield from input_two_different_pins()
yield # PIN mismatch
self.debug.press_yes() # try again
yield from input_two_different_pins()
yield # PIN mismatch
self.debug.press_yes() # try again
yield # PIN entry again
self.debug.press_no() # cancel
class InputFlowCodeChangeFail(InputFlowBase):
def __init__(
self, client: Client, current_pin: str, new_pin_1: str, new_pin_2: str
):
super().__init__(client)
self.current_pin = current_pin
self.new_pin_1 = new_pin_1
self.new_pin_2 = new_pin_2
def input_flow_common(self) -> GeneratorType:
yield # do you want to change pin?
self.debug.press_yes()
yield # enter current pin
self.debug.input(self.current_pin)
yield # enter new pin
self.debug.input(self.new_pin_1)
if self.debug.model == "R":
yield # Please re-enter PIN to confirm
self.debug.press_yes()
yield # enter new pin again (but different)
self.debug.input(self.new_pin_2)
yield # PIN mismatch
self.debug.press_yes() # try again
# failed retry
yield # enter current pin again
self.client.cancel()
class InputFlowWrongPIN(InputFlowBase):
def __init__(self, client: Client, wrong_pin: str):
super().__init__(client)
self.wrong_pin = wrong_pin
def input_flow_common(self) -> GeneratorType:
yield # do you want to change pin?
self.debug.press_yes()
yield # enter wrong current pin
self.debug.input(self.wrong_pin)
yield
self.debug.press_no()
class InputFlowPINBackoff(InputFlowBase):
def __init__(self, client: Client, wrong_pin: str, good_pin: str):
super().__init__(client)
self.wrong_pin = wrong_pin
self.good_pin = good_pin
def input_flow_common(self) -> GeneratorType:
"""Inputting some bad PINs and finally the correct one"""
yield # PIN entry
for attempt in range(3):
start = time.time()
self.debug.input(self.wrong_pin)
yield # PIN entry
check_pin_backoff_time(attempt, start)
self.debug.input(self.good_pin)
class InputFlowSignMessagePagination(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.message_read = ""
def input_flow_tt(self) -> GeneratorType:
# collect screen contents into `message_read`.
# Using a helper debuglink function to assemble the final text.
layouts: list[LayoutContent] = []
br = yield # confirm address
self.debug.wait_layout()
self.debug.press_yes()
br = yield
assert br.pages is not None
for i in range(br.pages):
layout = self.debug.wait_layout()
layouts.append(layout)
if i < br.pages - 1:
self.debug.swipe_up()
self.message_read = multipage_content(layouts)
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
# confirm address
yield
self.debug.press_yes()
br = yield
# TODO: try load the message_read the same way as in model T
if br.pages is not None:
for i in range(br.pages):
if i < br.pages - 1:
self.debug.swipe_up()
self.debug.press_yes()
class InputFlowShowAddressQRCode(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> GeneratorType:
yield
self.debug.click(buttons.CORNER_BUTTON, wait=True)
# synchronize; TODO get rid of this once we have single-global-layout
self.debug.synchronize_at("HorizontalPage")
self.debug.swipe_left(wait=True)
self.debug.swipe_right(wait=True)
self.debug.swipe_left(wait=True)
self.debug.click(buttons.CORNER_BUTTON, wait=True)
self.debug.press_no(wait=True)
self.debug.press_no(wait=True)
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield
# Go into details
self.debug.press_right()
# Go through details and back
self.debug.press_right()
self.debug.press_left()
self.debug.press_left()
# Confirm
self.debug.press_middle()
class InputFlowShowAddressQRCodeCancel(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> GeneratorType:
yield
self.debug.click(buttons.CORNER_BUTTON, wait=True)
# synchronize; TODO get rid of this once we have single-global-layout
self.debug.synchronize_at("HorizontalPage")
self.debug.swipe_left(wait=True)
self.debug.click(buttons.CORNER_BUTTON, wait=True)
self.debug.press_no(wait=True)
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield
# Go into details
self.debug.press_right()
# Go through details and back
self.debug.press_right()
self.debug.press_left()
self.debug.press_left()
# Cancel
self.debug.press_left()
# Confirm address mismatch
self.debug.press_right()
class InputFlowShowMultisigXPUBs(InputFlowBase):
def __init__(self, client: Client, address: str, xpubs: list[str], index: int):
super().__init__(client)
self.address = address
self.xpubs = xpubs
self.index = index
def input_flow_tt(self) -> GeneratorType:
yield # show address
layout = self.debug.wait_layout()
assert "RECEIVE ADDRESS\n(MULTISIG)" == layout.title()
assert layout.text_content().replace(" ", "") == self.address
self.debug.click(buttons.CORNER_BUTTON)
assert "Qr" in self.debug.wait_layout().all_components()
layout = self.debug.swipe_left(wait=True)
# address details
assert "Multisig 2 of 3" in layout.screen_content()
assert "Derivation path:" in layout.screen_content()
# Three xpub pages with the same testing logic
for xpub_num in range(3):
expected_title = f"MULTISIG XPUB #{xpub_num + 1}\n" + (
"(YOURS)" if self.index == xpub_num else "(COSIGNER)"
)
layout = self.debug.swipe_left(wait=True)
assert expected_title == layout.title()
content = layout.text_content().replace(" ", "")
assert self.xpubs[xpub_num] in content
self.debug.click(buttons.CORNER_BUTTON, wait=True)
# show address
self.debug.press_no(wait=True)
# address mismatch
self.debug.press_no(wait=True)
# show address
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield # show address
layout = self.debug.wait_layout()
assert "RECEIVE ADDRESS (MULTISIG)" in layout.title()
assert layout.text_content().replace(" ", "") == self.address
self.debug.press_right()
assert "Qr" in self.debug.wait_layout().all_components()
layout = self.debug.press_right(wait=True)
# address details
# TODO: locate it more precisely
assert "Multisig 2 of 3" in layout.json_str
# Three xpub pages with the same testing logic
for xpub_num in range(3):
expected_title = f"MULTISIG XPUB #{xpub_num + 1} " + (
"(YOURS)" if self.index == xpub_num else "(COSIGNER)"
)
layout = self.debug.press_right(wait=True)
assert expected_title in layout.title()
xpub_part_1 = layout.text_content().replace(" ", "")
# Press "SHOW MORE"
layout = self.debug.press_middle(wait=True)
xpub_part_2 = layout.text_content().replace(" ", "")
# Go back
self.debug.press_left(wait=True)
assert self.xpubs[xpub_num] == xpub_part_1 + xpub_part_2
for _ in range(5):
self.debug.press_left()
# show address
self.debug.press_left()
# address mismatch
self.debug.press_left()
# show address
self.debug.press_middle()
class InputFlowPaymentRequestDetails(InputFlowBase):
def __init__(self, client: Client, outputs: list[messages.TxOutputType]):
super().__init__(client)
self.outputs = outputs
def input_flow_tt(self) -> GeneratorType:
yield # request to see details
self.debug.wait_layout()
self.debug.press_info()
yield # confirm first output
assert self.outputs[0].address[:16] in self.layout().text_content() # type: ignore
self.debug.press_yes()
yield # confirm first output
self.debug.wait_layout()
self.debug.press_yes()
yield # confirm second output
assert self.outputs[1].address[:16] in self.layout().text_content() # type: ignore
self.debug.press_yes()
yield # confirm second output
self.debug.wait_layout()
self.debug.press_yes()
yield # confirm transaction
self.debug.press_yes()
yield # confirm transaction
self.debug.press_yes()
class InputFlowSignTxHighFee(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.finished = False
def go_through_all_screens(self, screens: list[B]) -> GeneratorType:
for expected in screens:
br = yield
assert br.code == expected
self.debug.press_yes()
self.finished = True
def input_flow_tt(self) -> GeneratorType:
screens = [
B.ConfirmOutput,
B.ConfirmOutput,
B.FeeOverThreshold,
B.SignTx,
]
yield from self.go_through_all_screens(screens)
def input_flow_tr(self) -> GeneratorType:
screens = [
B.ConfirmOutput,
B.FeeOverThreshold,
B.SignTx,
]
yield from self.go_through_all_screens(screens)
def sign_tx_go_to_info(client: Client) -> Generator[None, None, str]:
yield # confirm output
client.debug.wait_layout()
client.debug.press_yes()
yield # confirm output
client.debug.wait_layout()
client.debug.press_yes()
yield # confirm transaction
client.debug.wait_layout()
client.debug.press_info()
layout = client.debug.wait_layout()
content = layout.text_content().lower()
client.debug.click(buttons.CORNER_BUTTON, wait=True)
return content
def sign_tx_go_to_info_tr(
client: Client,
) -> Generator[None, None, str]:
yield # confirm output
client.debug.wait_layout()
client.debug.press_right() # CONTINUE
client.debug.wait_layout()
client.debug.press_right() # CONFIRM
screen_texts: list[str] = []
yield # confirm total
layout = client.debug.press_right(wait=True)
screen_texts.append(layout.text_content())
layout = client.debug.press_right(wait=True)
screen_texts.append(layout.text_content())
client.debug.press_left()
client.debug.press_left()
return "\n".join(screen_texts)
class InputFlowSignTxInformation(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def assert_content(self, content: str) -> None:
assert "sending from" in content
assert "legacy #6" in content
assert "fee rate" in content
assert "71.56 sat" in content
def input_flow_tt(self) -> GeneratorType:
content = yield from sign_tx_go_to_info(self.client)
self.assert_content(content)
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
content = yield from sign_tx_go_to_info_tr(self.client)
self.assert_content(content.lower())
self.debug.press_yes()
class InputFlowSignTxInformationMixed(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def assert_content(self, content: str) -> None:
assert "sending from" in content
assert "multiple accounts" in content
assert "fee rate" in content
assert "18.33 sat" in content
def input_flow_tt(self) -> GeneratorType:
content = yield from sign_tx_go_to_info(self.client)
self.assert_content(content)
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
content = yield from sign_tx_go_to_info_tr(self.client)
self.assert_content(content.lower())
self.debug.press_yes()
class InputFlowSignTxInformationCancel(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> GeneratorType:
yield from sign_tx_go_to_info(self.client)
self.debug.press_no()
def input_flow_tr(self) -> GeneratorType:
yield from sign_tx_go_to_info_tr(self.client)
self.debug.press_left()
class InputFlowSignTxInformationReplacement(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> GeneratorType:
yield # confirm txid
self.debug.press_yes()
yield # confirm address
self.debug.press_yes()
# go back to address
self.debug.press_no()
# confirm address
self.debug.press_yes()
yield # confirm amount
self.debug.press_yes()
yield # transaction summary, press info
self.debug.click(buttons.CORNER_BUTTON, wait=True)
self.debug.click(buttons.CORNER_BUTTON, wait=True)
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield # confirm txid
self.debug.press_right()
self.debug.press_right()
yield # confirm address
self.debug.press_right()
self.debug.press_right()
self.debug.press_right()
yield # confirm amount
self.debug.press_right()
self.debug.press_right()
self.debug.press_right()
def lock_time_input_flow_tt(
debug: DebugLink,
layout_assert_func: Callable[[DebugLink], None],
double_confirm: bool = False,
) -> GeneratorType:
yield # confirm output
debug.wait_layout()
debug.press_yes()
yield # confirm output
debug.wait_layout()
debug.press_yes()
yield # confirm locktime
layout_assert_func(debug)
debug.press_yes()
yield # confirm transaction
debug.press_yes()
if double_confirm:
yield # confirm transaction
debug.press_yes()
def lock_time_input_flow_tr(
debug: DebugLink, layout_assert_func: Callable[[DebugLink], None]
) -> GeneratorType:
yield # confirm output
debug.wait_layout()
debug.swipe_up()
debug.wait_layout()
debug.press_yes()
yield # confirm locktime
layout_assert_func(debug)
debug.press_yes()
yield # confirm transaction
debug.press_yes()
class InputFlowLockTimeBlockHeight(InputFlowBase):
def __init__(self, client: Client, block_height: str):
super().__init__(client)
self.block_height = block_height
def input_flow_tt(self) -> GeneratorType:
def assert_func(debug: DebugLink) -> None:
layout_text = debug.wait_layout().text_content()
assert "blockheight" in layout_text
assert self.block_height in layout_text
yield from lock_time_input_flow_tt(self.debug, assert_func, double_confirm=True)
def input_flow_tr(self) -> GeneratorType:
def assert_func(debug: DebugLink) -> None:
assert "blockheight" in debug.wait_layout().text_content()
debug.press_right()
assert self.block_height in debug.wait_layout().text_content()
yield from lock_time_input_flow_tr(self.debug, assert_func)
class InputFlowLockTimeDatetime(InputFlowBase):
def __init__(self, client: Client, lock_time_str: str):
super().__init__(client)
self.lock_time_str = lock_time_str
def input_flow_tt(self) -> GeneratorType:
def assert_func(debug: DebugLink):
layout_text = debug.wait_layout().text_content()
assert "Locktime" in layout_text
assert self.lock_time_str in layout_text
yield from lock_time_input_flow_tt(self.debug, assert_func)
def input_flow_tr(self) -> GeneratorType:
def assert_func(debug: DebugLink):
assert "Locktime" in debug.wait_layout().text_content()
debug.press_right()
assert self.lock_time_str in debug.wait_layout().text_content()
yield from lock_time_input_flow_tr(self.debug, assert_func)
class InputFlowEIP712ShowMore(InputFlowBase):
SHOW_MORE = (143, 167)
def __init__(self, client: Client):
super().__init__(client)
self.same_for_all_models = True
def _confirm_show_more(self) -> None:
"""Model-specific, either clicks a screen or presses a button."""
if self.model() == "T":
self.debug.click(self.SHOW_MORE)
elif self.model() == "R":
self.debug.press_right()
def input_flow_common(self) -> GeneratorType:
"""Triggers show more wherever possible"""
yield # confirm address
self.debug.press_yes()
yield # confirm domain
self.debug.wait_layout()
self._confirm_show_more()
# confirm domain properties
for _ in range(4):
yield from swipe_if_necessary(self.debug) # EIP712 DOMAIN
self.debug.press_yes()
yield # confirm message
self.debug.wait_layout()
self._confirm_show_more()
yield # confirm message.from
self.debug.wait_layout()
self._confirm_show_more()
# confirm message.from properties
for _ in range(2):
yield from swipe_if_necessary(self.debug)
self.debug.press_yes()
yield # confirm message.to
self.debug.wait_layout()
self._confirm_show_more()
# confirm message.to properties
for _ in range(2):
yield from swipe_if_necessary(self.debug)
self.debug.press_yes()
yield # confirm message.contents
self.debug.press_yes()
yield # confirm final hash
self.debug.press_yes()
class InputFlowEIP712Cancel(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> GeneratorType:
"""Clicks cancelling button"""
yield # confirm address
self.debug.press_yes()
yield # confirm domain
self.debug.press_no()
class InputFlowEthereumSignTxSkip(InputFlowBase):
def __init__(self, client: Client, cancel: bool = False):
super().__init__(client)
self.cancel = cancel
def input_flow_common(self) -> GeneratorType:
yield # confirm address
self.debug.press_yes()
yield # confirm amount
self.debug.wait_layout()
self.debug.press_yes()
yield # confirm data
if self.cancel:
self.debug.press_no()
else:
self.debug.press_yes()
yield # gas price
self.debug.press_yes()
yield # maximum fee
self.debug.press_yes()
yield # hold to confirm
self.debug.press_yes()
class InputFlowEthereumSignTxScrollDown(InputFlowBase):
SHOW_ALL = (143, 167)
def __init__(self, client: Client, cancel: bool = False):
super().__init__(client)
self.cancel = cancel
def input_flow_tt(self) -> GeneratorType:
yield # confirm address
self.debug.wait_layout()
self.debug.press_yes()
yield # confirm amount
self.debug.wait_layout()
self.debug.press_yes()
yield # confirm data
self.debug.wait_layout()
self.debug.click(self.SHOW_ALL)
br = yield # paginated data
assert br.pages is not None
for i in range(br.pages):
self.debug.wait_layout()
if i < br.pages - 1:
self.debug.swipe_up()
self.debug.press_yes()
yield # confirm data
if self.cancel:
self.debug.press_no()
else:
self.debug.press_yes()
yield # gas price
self.debug.press_yes()
yield # maximum fee
self.debug.press_yes()
yield # hold to confirm
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield # confirm address
self.debug.wait_layout()
self.debug.press_yes()
br = yield # paginated data
assert br.pages is not None
for _ in range(br.pages):
self.debug.wait_layout()
self.debug.swipe_up()
yield # confirm amount
self.debug.wait_layout()
self.debug.press_yes()
yield # confirm before send
if self.cancel:
self.debug.press_no()
else:
self.debug.press_yes()
class InputFlowEthereumSignTxGoBack(InputFlowBase):
SHOW_ALL = (143, 167)
GO_BACK = (16, 220)
def __init__(self, client: Client, cancel: bool = False):
super().__init__(client)
self.cancel = cancel
def input_flow_tt(self) -> GeneratorType:
br = yield # confirm address
self.debug.wait_layout()
self.debug.press_yes()
br = yield # confirm amount
self.debug.wait_layout()
self.debug.press_yes()
br = yield # confirm data
self.debug.wait_layout()
self.debug.click(self.SHOW_ALL)
br = yield # paginated data
assert br.pages is not None
for i in range(br.pages):
self.debug.wait_layout()
if i == 2:
self.debug.click(self.GO_BACK)
yield # confirm data
self.debug.wait_layout()
if self.cancel:
self.debug.press_no()
else:
self.debug.press_yes()
yield # confirm address
self.debug.wait_layout()
self.debug.press_yes()
yield # confirm amount
self.debug.wait_layout()
self.debug.press_yes()
yield # hold to confirm
self.debug.wait_layout()
self.debug.press_yes()
return
elif i < br.pages - 1:
self.debug.swipe_up()
def get_mnemonic_and_confirm_success(
debug: DebugLink,
) -> Generator[None, "messages.ButtonRequest", str]:
# mnemonic phrases
mnemonic = yield from read_and_confirm_mnemonic(debug)
br = yield # confirm recovery seed check
assert br.code == B.Success
debug.press_yes()
br = yield # confirm success
assert br.code == B.Success
debug.press_yes()
assert mnemonic is not None
return mnemonic
class InputFlowBip39Backup(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.mnemonic = None
def input_flow_common(self) -> GeneratorType:
# 1. Confirm Reset
yield from click_through(self.debug, screens=1, code=B.ResetDevice)
# mnemonic phrases and rest
self.mnemonic = yield from get_mnemonic_and_confirm_success(self.debug)
class InputFlowBip39ResetBackup(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.mnemonic = None
# NOTE: same as above, just two more YES
def input_flow_common(self) -> GeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Confirm warning
yield from click_through(self.debug, screens=3, code=B.ResetDevice)
# mnemonic phrases and rest
self.mnemonic = yield from get_mnemonic_and_confirm_success(self.debug)
class InputFlowBip39ResetPIN(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.mnemonic = None
def input_flow_common(self) -> GeneratorType:
br = yield # Confirm Reset
assert br.code == B.ResetDevice
self.debug.press_yes()
yield # Enter new PIN
self.debug.input("654")
if self.debug.model == "R":
yield # Re-enter PIN
self.debug.press_yes()
yield # Confirm PIN
self.debug.input("654")
br = yield # Confirm entropy
assert br.code == B.ResetDevice
self.debug.press_yes()
br = yield # Backup your seed
assert br.code == B.ResetDevice
self.debug.press_yes()
br = yield # Confirm warning
assert br.code == B.ResetDevice
self.debug.press_yes()
# mnemonic phrases
self.mnemonic = yield from read_and_confirm_mnemonic(self.debug)
br = yield # confirm recovery seed check
assert br.code == B.Success
self.debug.press_yes()
br = yield # confirm success
assert br.code == B.Success
self.debug.press_yes()
class InputFlowBip39ResetFailedCheck(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.mnemonic = None
def input_flow_common(self) -> GeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Confirm warning
yield from click_through(self.debug, screens=3, code=B.ResetDevice)
# mnemonic phrases, wrong answer
self.mnemonic = yield from read_and_confirm_mnemonic(
self.debug, choose_wrong=True
)
br = yield # warning screen
assert br.code == B.ResetDevice
self.debug.press_yes()
# mnemonic phrases
self.mnemonic = yield from read_and_confirm_mnemonic(self.debug)
br = yield # confirm recovery seed check
assert br.code == B.Success
self.debug.press_yes()
br = yield # confirm success
assert br.code == B.Success
self.debug.press_yes()
def load_5_shares(
debug: DebugLink,
) -> Generator[None, "messages.ButtonRequest", list[str]]:
mnemonics: list[str] = []
for _ in range(5):
# Phrase screen
mnemonic = yield from read_and_confirm_mnemonic(debug)
assert mnemonic is not None
mnemonics.append(mnemonic)
br = yield # Confirm continue to next
assert br.code == B.Success
debug.press_yes()
return mnemonics
class InputFlowSlip39BasicBackup(InputFlowBase):
def __init__(self, client: Client, click_info: bool):
super().__init__(client)
self.mnemonics: list[str] = []
self.click_info = click_info
def input_flow_tt(self) -> GeneratorType:
yield # 1. Checklist
self.debug.press_yes()
if self.click_info:
yield from click_info_button_tt(self.debug)
yield # 2. Number of shares (5)
self.debug.press_yes()
yield # 3. Checklist
self.debug.press_yes()
if self.click_info:
yield from click_info_button_tt(self.debug)
yield # 4. Threshold (3)
self.debug.press_yes()
yield # 5. Checklist
self.debug.press_yes()
yield # 6. Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_5_shares(self.debug)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield # 1. Checklist
self.debug.press_yes()
yield # 1.5 Number of shares info
self.debug.press_yes()
yield # 2. Number of shares (5)
self.debug.input("5")
yield # 3. Checklist
self.debug.press_yes()
yield # 3.5 Threshold info
self.debug.press_yes()
yield # 4. Threshold (3)
self.debug.input("3")
yield # 5. Checklist
self.debug.press_yes()
yield # 6. Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_5_shares(self.debug)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
class InputFlowSlip39BasicResetRecovery(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.mnemonics: list[str] = []
def input_flow_tt(self) -> GeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Confirm warning
# 4. shares info
# 5. Set & Confirm number of shares
# 6. threshold info
# 7. Set & confirm threshold value
# 8. Confirm show seeds
yield from click_through(self.debug, screens=8, code=B.ResetDevice)
# Mnemonic phrases
self.mnemonics = yield from load_5_shares(self.debug)
br = yield # safety warning
assert br.code == B.Success
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield # Confirm Reset
self.debug.press_yes()
yield # Backup your seed
self.debug.press_yes()
yield # Checklist
self.debug.press_yes()
yield # Number of shares info
self.debug.press_yes()
yield # Number of shares (5)
self.debug.input("5")
yield # Checklist
self.debug.press_yes()
yield # Threshold info
self.debug.press_yes()
yield # Threshold (3)
self.debug.input("3")
yield # Checklist
self.debug.press_yes()
yield # Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_5_shares(self.debug)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def load_5_groups_5_shares(
debug: DebugLink,
) -> Generator[None, "messages.ButtonRequest", list[str]]:
mnemonics: list[str] = []
for _g in range(5):
for _s in range(5):
# Phrase screen
mnemonic = yield from read_and_confirm_mnemonic(debug)
assert mnemonic is not None
mnemonics.append(mnemonic)
# Confirm continue to next
yield from swipe_if_necessary(debug, B.Success)
debug.press_yes()
return mnemonics
class InputFlowSlip39AdvancedBackup(InputFlowBase):
def __init__(self, client: Client, click_info: bool):
super().__init__(client)
self.mnemonics: list[str] = []
self.click_info = click_info
def input_flow_tt(self) -> GeneratorType:
yield # 1. Checklist
self.debug.press_yes()
if self.click_info:
yield from click_info_button_tt(self.debug)
yield # 2. Set and confirm group count
self.debug.press_yes()
yield # 3. Checklist
self.debug.press_yes()
if self.click_info:
yield from click_info_button_tt(self.debug)
yield # 4. Set and confirm group threshold
self.debug.press_yes()
yield # 5. Checklist
self.debug.press_yes()
for _ in range(5): # for each of 5 groups
if self.click_info:
yield from click_info_button_tt(self.debug)
yield # Set & Confirm number of shares
self.debug.press_yes()
if self.click_info:
yield from click_info_button_tt(self.debug)
yield # Set & confirm share threshold value
self.debug.press_yes()
yield # Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases - show & confirm shares for all groups
self.mnemonics = yield from load_5_groups_5_shares(self.debug)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield # 1. Checklist
self.debug.press_yes()
yield # 2. Set and confirm group count
self.debug.input("5")
yield # 3. Checklist
self.debug.press_yes()
yield # 4. Set and confirm group threshold
self.debug.input("3")
yield # 5. Checklist
self.debug.press_yes()
for _ in range(5): # for each of 5 groups
yield # Number of shares info
self.debug.press_yes()
yield # Number of shares (5)
self.debug.input("5")
yield # Threshold info
self.debug.press_yes()
yield # Threshold (3)
self.debug.input("3")
yield # Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases - show & confirm shares for all groups
self.mnemonics = yield from load_5_groups_5_shares(self.debug)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
class InputFlowSlip39AdvancedResetRecovery(InputFlowBase):
def __init__(self, client: Client, click_info: bool):
super().__init__(client)
self.mnemonics: list[str] = []
self.click_info = click_info
def input_flow_tt(self) -> GeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Confirm warning
# 4. shares info
# 5. Set & Confirm number of groups
# 6. threshold info
# 7. Set & confirm group threshold value
# 8-17: for each of 5 groups:
# 1. Set & Confirm number of shares
# 2. Set & confirm share threshold value
# 18. Confirm show seeds
yield from click_through(self.debug, screens=18, code=B.ResetDevice)
# Mnemonic phrases - show & confirm shares for all groups
self.mnemonics = yield from load_5_groups_5_shares(self.debug)
br = yield # safety warning
assert br.code == B.Success
self.debug.press_yes()
def input_flow_tr(self) -> GeneratorType:
yield # Wallet backup
self.debug.press_yes()
yield # Wallet creation
self.debug.press_yes()
yield # Checklist
self.debug.press_yes()
yield # Set and confirm group count
self.debug.input("5")
yield # Checklist
self.debug.press_yes()
yield # Set and confirm group threshold
self.debug.input("3")
yield # Checklist
self.debug.press_yes()
for _ in range(5): # for each of 5 groups
yield # Number of shares info
self.debug.press_yes()
yield # Number of shares (5)
self.debug.input("5")
yield # Threshold info
self.debug.press_yes()
yield # Threshold (3)
self.debug.input("3")
yield # Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases - show & confirm shares for all groups
self.mnemonics = yield from load_5_groups_5_shares(self.debug)
br = yield # safety warning
assert br.code == B.Success
self.debug.press_yes()
def enter_recovery_seed_dry_run_tt(
debug: DebugLink, mnemonic: list[str]
) -> GeneratorType:
yield
assert "check the recovery seed" in debug.wait_layout().text_content()
debug.click(buttons.OK)
yield
assert "number of words" in debug.wait_layout().text_content()
debug.click(buttons.OK)
yield
assert "SelectWordCount" in debug.wait_layout().all_components()
# click the correct number
word_option_offset = 6
word_options = (12, 18, 20, 24, 33)
index = word_option_offset + word_options.index(len(mnemonic))
debug.click(buttons.grid34(index % 3, index // 3))
yield
assert "Enter your backup" in debug.wait_layout().text_content()
debug.click(buttons.OK)
yield
for word in mnemonic:
assert debug.wait_layout().main_component() == "MnemonicKeyboard"
debug.input(word)
class InputFlowBip39RecoveryDryRun(InputFlowBase):
def __init__(self, client: Client, mnemonic: list[str]):
super().__init__(client)
self.mnemonic = mnemonic
def input_flow_tt(self) -> GeneratorType:
yield from enter_recovery_seed_dry_run_tt(self.debug, self.mnemonic)
yield
self.debug.wait_layout()
self.debug.click(buttons.OK)
def input_flow_tr(self) -> GeneratorType:
yield
assert "check the recovery seed" in self.layout().text_content()
self.debug.press_yes()
yield from enter_recovery_seed_tr(self.debug, self.mnemonic)
yield
self.debug.press_yes()
class InputFlowBip39RecoveryDryRunInvalid(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> GeneratorType:
mnemonic = ["stick"] * 12
yield from enter_recovery_seed_dry_run_tt(self.debug, mnemonic)
br = yield
assert br.code == messages.ButtonRequestType.Warning
assert "Invalid recovery seed" in self.layout().text_content()
self.debug.click(buttons.OK)
yield # retry screen
assert "number of words" in self.layout().text_content()
self.debug.click(buttons.CANCEL)
yield
assert "ABORT BACKUP CHECK" == self.layout().title()
self.debug.click(buttons.OK)
def input_flow_tr(self) -> GeneratorType:
yield
assert "check the recovery seed" in self.layout().text_content()
self.debug.press_right()
mnemonic = ["stick"] * 12
yield from enter_recovery_seed_tr(self.debug, mnemonic)
br = yield
assert br.code == messages.ButtonRequestType.Warning
assert "Invalid recovery seed" in self.layout().text_content()
self.debug.press_middle()
yield # retry screen
assert "number of words" in self.layout().text_content()
self.debug.press_left()
yield
assert "abort" in self.layout().text_content()
self.debug.press_right()
def bip39_recovery_possible_pin_tt(
debug: DebugLink, mnemonic: list[str], pin: Optional[str]
) -> GeneratorType:
yield
assert "By continuing you agree to" in debug.wait_layout().text_content()
debug.press_yes()
# PIN when requested
if pin is not None:
yield
assert "PinKeyboard" in debug.wait_layout().all_components()
debug.input(pin)
yield
assert "PinKeyboard" in debug.wait_layout().all_components()
debug.input(pin)
yield
assert "number of words" in debug.wait_layout().text_content()
debug.press_yes()
yield
assert "SelectWordCount" in debug.wait_layout().all_components()
debug.input(str(len(mnemonic)))
yield
assert "Enter your backup" in debug.wait_layout().text_content()
debug.press_yes()
yield
for word in mnemonic:
assert debug.wait_layout().main_component() == "MnemonicKeyboard"
debug.input(word)
yield
assert "Wallet recovered successfully" in debug.wait_layout().text_content()
debug.press_yes()
def bip39_recovery_possible_pin_tr(
debug: DebugLink, mnemonic: list[str], pin: Optional[str]
) -> GeneratorType:
yield
assert "By continuing you agree" in debug.wait_layout().text_content()
debug.press_right()
assert "trezor.io/tos" in debug.wait_layout().text_content()
debug.press_yes()
yield
assert "safe to eject" in debug.wait_layout().text_content()
debug.press_yes()
# PIN when requested
if pin is not None:
yield
debug.input("654")
yield
assert "re-enter PIN to confirm" in debug.wait_layout().text_content()
debug.press_right()
yield
debug.input("654")
yield from enter_recovery_seed_tr(debug, mnemonic)
yield
assert "Wallet recovered successfully" in debug.wait_layout().text_content()
debug.press_yes()
def enter_recovery_seed_tr(debug: DebugLink, mnemonic: list[str]) -> GeneratorType:
yield
assert "number of words" in debug.wait_layout().text_content()
debug.press_yes()
yield
assert "NUMBER OF WORDS" in debug.wait_layout().title()
debug.input(str(len(mnemonic)))
yield
assert "Enter your backup" in debug.wait_layout().text_content()
# Paginate to see info
debug.press_right()
debug.press_right()
debug.press_yes()
yield
for index, word in enumerate(mnemonic):
title = debug.wait_layout().title()
assert "WORD" in title
assert str(index + 1) in title
debug.input(word)
class InputFlowBip39RecoveryPIN(InputFlowBase):
def __init__(self, client: Client, mnemonic: list[str]):
super().__init__(client)
self.mnemonic = mnemonic
def input_flow_tt(self) -> GeneratorType:
yield from bip39_recovery_possible_pin_tt(self.debug, self.mnemonic, pin="654")
def input_flow_tr(self) -> GeneratorType:
yield from bip39_recovery_possible_pin_tr(self.debug, self.mnemonic, pin="654")
class InputFlowBip39RecoveryNoPIN(InputFlowBase):
def __init__(self, client: Client, mnemonic: list[str]):
super().__init__(client)
self.mnemonic = mnemonic
def input_flow_tt(self) -> GeneratorType:
yield from bip39_recovery_possible_pin_tt(self.debug, self.mnemonic, pin=None)
def input_flow_tr(self) -> GeneratorType:
yield from bip39_recovery_possible_pin_tr(self.debug, self.mnemonic, pin=None)
class InputFlowSlip39AdvancedRecoveryDryRun(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
def input_flow_common(self) -> GeneratorType:
yield # Confirm Dryrun
self.debug.press_yes()
# run recovery flow
yield from recovery_enter_shares(self.debug, self.shares, groups=True)
def confirm_recovery(debug: DebugLink) -> GeneratorType:
if debug.model == "T":
yield # Confirm Recovery
debug.press_yes()
elif debug.model == "R":
yield # Confirm Recovery
debug.press_right()
debug.press_yes()
yield # Safe to eject
debug.press_yes()
class InputFlowSlip39AdvancedRecovery(InputFlowBase):
def __init__(self, client: Client, shares: list[str], click_info: bool):
super().__init__(client)
self.shares = shares
self.click_info = click_info
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
yield from recovery_enter_shares(
self.debug, self.shares, groups=True, click_info=self.click_info
)
class InputFlowSlip39AdvancedRecoveryAbort(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
yield # Homescreen - abort process
self.debug.press_no()
yield # Homescreen - confirm abort
self.debug.press_yes()
class InputFlowSlip39AdvancedRecoveryNoAbort(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
yield # Homescreen - abort process
self.debug.press_no()
yield # Homescreen - go back to process
if self.debug.model == "R":
self.debug.press_right()
self.debug.press_no()
yield from recovery_enter_shares(self.debug, self.shares, groups=True)
class InputFlowSlip39AdvancedRecoveryTwoSharesWarning(InputFlowBase):
def __init__(self, client: Client, first_share: list[str], second_share: list[str]):
super().__init__(client)
self.first_share = first_share
self.second_share = second_share
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
yield from slip39_recovery_setup_and_first_share(self.debug, self.first_share)
yield # Continue to next share
self.debug.press_yes()
yield # Homescreen - next share
self.debug.press_yes()
yield # Enter next share
for word in self.second_share:
self.debug.input(word)
br = yield
assert br.code == messages.ButtonRequestType.Warning
self.debug.press_yes()
yield
self.client.cancel()
def slip39_recovery_possible_pin(
debug: DebugLink, shares: list[str], pin: Optional[str]
) -> GeneratorType:
yield # Confirm Recovery/Dryrun
if debug.model == "R" and "BACKUP CHECK" not in debug.wait_layout().title():
# dryruns do not have extra dialogs
debug.press_right()
debug.press_yes()
yield
debug.press_yes()
if pin is not None:
yield # Enter PIN
debug.input(pin)
if debug.model == "R":
yield # Reenter PIN
debug.press_yes()
yield # Enter PIN again
debug.input(pin)
# Proceed with recovery
yield from recovery_enter_shares(debug, shares)
class InputFlowSlip39BasicRecovery(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
def input_flow_common(self) -> GeneratorType:
yield from slip39_recovery_possible_pin(self.debug, self.shares, pin=None)
class InputFlowSlip39BasicRecoveryPIN(InputFlowBase):
def __init__(self, client: Client, shares: list[str], pin: str):
super().__init__(client)
self.shares = shares
self.pin = pin
def input_flow_common(self) -> GeneratorType:
yield from slip39_recovery_possible_pin(self.debug, self.shares, pin=self.pin)
class InputFlowSlip39BasicRecoveryAbort(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
yield # Homescreen - abort process
self.debug.press_no()
yield # Homescreen - confirm abort
self.debug.press_yes()
class InputFlowSlip39BasicRecoveryNoAbort(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
yield # Homescreen - abort process
self.debug.press_no()
yield # Homescreen - go back to process
if self.debug.model == "R":
self.debug.press_right()
self.debug.press_no()
# run recovery flow
yield from recovery_enter_shares(self.debug, self.shares)
def slip39_recovery_setup_and_first_share(
debug: DebugLink, first_share: list[str]
) -> GeneratorType:
yield # Homescreen - start process
debug.press_yes()
yield # Enter number of words
debug.input(str(len(first_share)))
yield # Homescreen - proceed to share entry
if debug.model == "R":
debug.press_right(wait=True)
debug.press_right(wait=True)
debug.press_yes()
yield # Enter first share
for word in first_share:
debug.input(word)
class InputFlowSlip39BasicRecoveryRetryFirst(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
first_share = ["slush"] * 20
yield from slip39_recovery_setup_and_first_share(self.debug, first_share)
br = yield # Invalid share
assert br.code == messages.ButtonRequestType.Warning
self.debug.press_yes()
first_share = ["slush"] * 33
yield from slip39_recovery_setup_and_first_share(self.debug, first_share)
br = yield # Invalid share
assert br.code == messages.ButtonRequestType.Warning
self.debug.press_yes()
yield # Homescreen
self.debug.press_no()
yield # Confirm abort
if self.debug.model == "R":
self.debug.press_right(wait=True)
self.debug.press_yes()
class InputFlowSlip39BasicRecoveryRetrySecond(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
# First valid share
first_share = self.shares[0].split(" ")
yield from slip39_recovery_setup_and_first_share(self.debug, first_share)
yield # More shares needed
self.debug.press_yes()
yield # Enter another share
invalid_share = first_share[:3] + ["slush"] * 17
for word in invalid_share:
self.debug.input(word)
br = yield # Invalid share
assert br.code == messages.ButtonRequestType.Warning
self.debug.press_yes()
yield # Proceed to next share
second_share = self.shares[1].split(" ")
for word in second_share:
self.debug.input(word)
yield # More shares needed
self.debug.press_no()
yield # Confirm abort
if self.debug.model == "R":
self.debug.press_right(wait=True)
self.debug.press_yes()
class InputFlowSlip39BasicRecoveryWrongNthWord(InputFlowBase):
def __init__(self, client: Client, share: list[str], nth_word: int):
super().__init__(client)
self.share = share
self.nth_word = nth_word
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
# First complete share
yield from slip39_recovery_setup_and_first_share(self.debug, self.share)
yield # Continue to next share
self.debug.press_yes()
yield # Enter next share
for i, word in enumerate(self.share):
if i < self.nth_word:
self.debug.input(word)
else:
self.debug.input(self.share[-1])
break
br = yield
assert br.code == messages.ButtonRequestType.Warning
self.debug.press_yes()
yield
self.client.cancel()
class InputFlowSlip39BasicRecoverySameShare(InputFlowBase):
def __init__(self, client: Client, first_share: list[str], second_share: list[str]):
super().__init__(client)
self.first_share = first_share
self.second_share = second_share
def input_flow_common(self) -> GeneratorType:
yield from confirm_recovery(self.debug)
# First complete share
yield from slip39_recovery_setup_and_first_share(self.debug, self.first_share)
yield # Continue to next share
self.debug.press_yes()
yield # Enter next share
for word in self.second_share:
self.debug.input(word)
br = yield
assert br.code == messages.ButtonRequestType.Warning
self.debug.press_yes()
yield
self.client.cancel()
class InputFlowResetSkipBackup(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> GeneratorType:
yield # Confirm Recovery
if self.debug.model == "R":
self.debug.press_right()
self.debug.press_yes()
yield # Skip Backup
if self.debug.model == "R":
self.debug.press_right()
self.debug.press_no()
yield # Confirm skip backup
self.debug.press_no()