arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-15 02:48:11 +00:00
Code Issues Releases Wiki Activity
e196413bb7
trezor-firmware/tests/input_flows.py
obrusvit af6240b52c refactor(core): model_r confirm_summary 
- altcoin_tx_summary removed and replaced with confirm_summary

[no changelog]
2024-12-04 14:47:53 +01:00

2300 lines
75 KiB
Python
Raw Blame History

"""
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.
"""
from __future__ import annotations
import time
from typing import Callable, Generator
from trezorlib import messages
from trezorlib.debuglink import DebugLink, LayoutContent, LayoutType
from trezorlib.debuglink import TrezorClientDebugLink as Client
from trezorlib.debuglink import multipage_content
from . import buttons
from . import translations as TR
from .common import (
BRGeneratorType,
check_pin_backoff_time,
click_info_button_mercury,
click_info_button_tt,
click_through,
get_text_possible_pagination,
read_and_confirm_mnemonic,
swipe_if_necessary,
)
from .input_flows_helpers import BackupFlow, EthereumFlow, PinFlow, RecoveryFlow
B = messages.ButtonRequestType
class InputFlowBase:
def __init__(self, client: Client):
self.client = client
self.debug: DebugLink = client.debug
self.PIN = PinFlow(self.client)
self.REC = RecoveryFlow(self.client)
self.BAK = BackupFlow(self.client)
self.ETH = EthereumFlow(self.client)
def get(self) -> Callable[[], BRGeneratorType]:
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.client.layout_type is LayoutType.TT:
return self.input_flow_tt
elif self.client.layout_type is LayoutType.TR:
return self.input_flow_tr
elif self.client.layout_type is LayoutType.Mercury:
return self.input_flow_t3t1
else:
raise ValueError("Unknown model")
def input_flow_tt(self) -> BRGeneratorType:
"""Special for TT"""
raise NotImplementedError
def input_flow_tr(self) -> BRGeneratorType:
"""Special for TR"""
raise NotImplementedError
def input_flow_t3t1(self) -> BRGeneratorType:
"""Special for T3T1"""
raise NotImplementedError
def text_content(self) -> str:
return self.debug.read_layout().text_content()
def main_component(self) -> str:
return self.debug.read_layout().main_component()
def all_components(self) -> list[str]:
return self.debug.read_layout().all_components()
def title(self) -> str:
return self.debug.read_layout().title()
class InputFlowNewCodeMismatch(InputFlowBase):
def __init__(
self,
client: Client,
first_code: str,
second_code: str,
what: str,
):
super().__init__(client)
self.first_code = first_code
self.second_code = second_code
self.what = what
def input_flow_common(self) -> BRGeneratorType:
assert (yield).name == f"set_{self.what}"
self.debug.press_yes()
if self.client.layout_type is LayoutType.TR:
layout = self.debug.read_layout()
if "PinKeyboard" not in layout.all_components():
yield from swipe_if_necessary(self.debug) # code info
self.debug.press_yes()
def input_two_different_pins() -> BRGeneratorType:
yield from self.PIN.setup_new_pin(
self.first_code, self.second_code, what=self.what
)
yield from input_two_different_pins()
assert (yield).name == f"{self.what}_mismatch" # PIN mismatch
self.debug.press_yes() # try again
yield from input_two_different_pins()
assert (yield).name == f"{self.what}_mismatch" # PIN mismatch
self.debug.press_yes() # try again
assert (yield).name == "pin_device" # 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) -> BRGeneratorType:
yield # do you want to change pin?
self.debug.press_yes()
yield # enter current pin
self.debug.input(self.current_pin)
yield from self.PIN.setup_new_pin(self.new_pin_1, 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) -> BRGeneratorType:
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) -> BRGeneratorType:
"""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) -> BRGeneratorType:
# 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.read_layout()
self.debug.press_yes()
br = yield
assert br.pages is not None
for i in range(br.pages):
layout = self.debug.read_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) -> BRGeneratorType:
# confirm address
yield
self.debug.press_yes()
# paginate through the whole message
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()
# confirm message
yield
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
# 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.read_layout()
self.debug.press_yes()
br = yield
# assert br.pages is not None
for i in range(br.pages or 1):
layout = self.debug.read_layout()
layouts.append(layout)
if br.pages and i < br.pages - 1:
self.debug.swipe_up()
self.message_read = multipage_content(layouts)
self.debug.press_yes()
class InputFlowSignMessageInfo(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> BRGeneratorType:
yield
# signing address/message info
self.debug.click(buttons.CORNER_BUTTON)
self.debug.click(buttons.CORNER_BUTTON)
self.debug.press_no()
self.debug.synchronize_at("IconDialog")
# address mismatch?
self.debug.press_no()
yield
self.debug.press_yes()
yield
# going back to the signing address
self.debug.press_no()
yield
self.debug.press_no()
# address mismatch?
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
yield
# show address/message info
self.debug.click(buttons.CORNER_BUTTON)
self.debug.click(buttons.VERTICAL_MENU[0])
self.debug.click(buttons.CORNER_BUTTON)
self.debug.click(buttons.VERTICAL_MENU[1])
# address mismatch?
self.debug.swipe_up()
yield
class InputFlowShowAddressQRCode(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> BRGeneratorType:
yield
self.debug.click(buttons.CORNER_BUTTON)
# synchronize; TODO get rid of this once we have single-global-layout
self.debug.synchronize_at("SimplePage")
self.debug.swipe_left()
self.debug.swipe_right()
self.debug.swipe_left()
self.debug.click(buttons.CORNER_BUTTON)
self.debug.press_no()
self.debug.press_no()
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
# Find out the page-length of the address
br = yield
if br.pages is not None:
address_swipes = br.pages - 1
else:
address_swipes = 0
for _ in range(address_swipes):
self.debug.press_right()
# 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
for _ in range(address_swipes):
self.debug.press_right()
self.debug.press_middle()
def input_flow_t3t1(self) -> BRGeneratorType:
yield
self.debug.click(buttons.CORNER_BUTTON)
# synchronize; TODO get rid of this once we have single-global-layout
self.debug.synchronize_at("VerticalMenu")
# menu
self.debug.click(buttons.VERTICAL_MENU[0])
self.debug.synchronize_at("Qr")
# qr code
self.debug.click(buttons.CORNER_BUTTON)
# menu
self.debug.click(buttons.VERTICAL_MENU[1])
# address details
self.debug.click(buttons.CORNER_BUTTON)
# menu
self.debug.click(buttons.VERTICAL_MENU[2])
# cancel
self.debug.swipe_up()
# really cancel
self.debug.click(buttons.CORNER_BUTTON)
# menu
layout = self.debug.click(buttons.CORNER_BUTTON)
while "PromptScreen" not in layout.all_components():
layout = self.debug.swipe_up()
self.debug.synchronize_at("PromptScreen")
# tap to confirm
self.debug.click(buttons.TAP_TO_CONFIRM)
class InputFlowShowAddressQRCodeCancel(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> BRGeneratorType:
yield
self.debug.click(buttons.CORNER_BUTTON)
# synchronize; TODO get rid of this once we have single-global-layout
self.debug.synchronize_at("SimplePage")
self.debug.swipe_left()
self.debug.click(buttons.CORNER_BUTTON)
self.debug.press_no()
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
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
# Clicking right twice, as some languages can have two pages
self.debug.press_right()
self.debug.press_right()
def input_flow_t3t1(self) -> BRGeneratorType:
yield
self.debug.click(buttons.CORNER_BUTTON)
# synchronize; TODO get rid of this once we have single-global-layout
self.debug.synchronize_at("VerticalMenu")
# menu
self.debug.click(buttons.VERTICAL_MENU[0])
self.debug.synchronize_at("Qr")
# qr code
self.debug.click(buttons.CORNER_BUTTON)
# menu
self.debug.click(buttons.VERTICAL_MENU[1])
# address details
self.debug.click(buttons.CORNER_BUTTON)
# menu
self.debug.click(buttons.VERTICAL_MENU[2])
# cancel
self.debug.swipe_up()
self.debug.synchronize_at("PromptScreen")
# really cancel
self.debug.click(buttons.TAP_TO_CONFIRM)
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 _assert_xpub_title(self, title: str, xpub_num: int) -> None:
expected_title = f"MULTISIG XPUB #{xpub_num + 1}"
assert expected_title in title
if self.index == xpub_num:
assert TR.address__title_yours in title
else:
assert TR.address__title_cosigner in title
def input_flow_tt(self) -> BRGeneratorType:
yield # multisig address warning
self.debug.press_yes()
yield # show address
layout = self.debug.read_layout()
assert TR.address__title_receive_address in layout.title()
assert "(MULTISIG)" in layout.title()
assert layout.text_content().replace(" ", "") == self.address
self.debug.click(buttons.CORNER_BUTTON)
assert "Qr" in self.all_components()
layout = self.debug.swipe_left()
# address details
assert "Multisig 2 of 3" in layout.screen_content()
assert TR.address_details__derivation_path in layout.screen_content()
# Three xpub pages with the same testing logic
for xpub_num in range(3):
layout = self.debug.swipe_left()
self._assert_xpub_title(layout.title(), xpub_num)
content = layout.text_content().replace(" ", "")
assert self.xpubs[xpub_num] in content
self.debug.click(buttons.CORNER_BUTTON)
# show address
self.debug.press_no()
# address mismatch
self.debug.press_no()
# show address
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
yield # multisig address warning
self.debug.press_middle()
yield # show address
layout = self.debug.read_layout()
assert TR.address__title_receive_address in layout.title()
assert "(MULTISIG)" in layout.title()
assert layout.text_content().replace(" ", "") == self.address
self.debug.press_right()
assert "Qr" in self.all_components()
layout = self.debug.press_right()
# 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):
layout = self.debug.press_right()
self._assert_xpub_title(layout.title(), xpub_num)
xpub_part_1 = layout.text_content().replace(" ", "")
# Press "SHOW MORE"
layout = self.debug.press_middle()
xpub_part_2 = layout.text_content().replace(" ", "")
# Go back
self.debug.press_left()
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()
def input_flow_t3t1(self) -> BRGeneratorType:
yield # multisig address warning
self.debug.click(buttons.CORNER_BUTTON)
self.debug.synchronize_at("VerticalMenu")
self.debug.click(buttons.VERTICAL_MENU[1])
yield # show address
layout = self.debug.read_layout()
assert TR.address__title_receive_address in layout.title()
assert layout.text_content().replace(" ", "") == self.address
self.debug.click(buttons.CORNER_BUTTON)
assert "VerticalMenu" in self.all_components()
# menu
self.debug.click(buttons.VERTICAL_MENU[0])
self.debug.synchronize_at("Qr")
# qr code
assert "Qr" in self.all_components()
self.debug.click(buttons.CORNER_BUTTON)
# menu
assert "VerticalMenu" in self.all_components()
self.debug.click(buttons.VERTICAL_MENU[1])
layout = self.debug.synchronize_at("AddressDetails")
# address details
assert "Multisig 2 of 3" in layout.screen_content()
assert TR.address_details__derivation_path in layout.screen_content()
# three xpub pages with the same testing logic
for _xpub_num in range(3):
layout = self.debug.swipe_left()
layout = self.debug.swipe_left()
self.debug.click(buttons.CORNER_BUTTON)
layout = self.debug.synchronize_at("VerticalMenu")
# menu
self.debug.click(buttons.VERTICAL_MENU[2])
# cancel
self.debug.swipe_up()
# really cancel
self.debug.click(buttons.CORNER_BUTTON)
layout = self.debug.synchronize_at("VerticalMenu")
# menu
self.debug.click(buttons.CORNER_BUTTON)
layout = self.debug.synchronize_at("Paragraphs")
# address
while "PromptScreen" not in layout.all_components():
layout = self.debug.swipe_up()
self.debug.synchronize_at("PromptScreen")
# tap to confirm
self.debug.press_yes()
class InputFlowShowXpubQRCode(InputFlowBase):
def __init__(self, client: Client, passphrase: bool = False):
super().__init__(client)
self.passphrase = passphrase
def input_flow_tt(self) -> BRGeneratorType:
if self.passphrase:
yield
self.debug.press_yes()
yield
self.debug.press_yes()
br = yield
layout = self.debug.read_layout()
if "coinjoin" in layout.title().lower() or br.code == B.UnknownDerivationPath:
self.debug.press_yes()
br = yield
self.debug.click(buttons.CORNER_BUTTON)
# synchronize; TODO get rid of this once we have single-global-layout
self.debug.synchronize_at("SimplePage")
self.debug.swipe_left()
self.debug.swipe_right()
self.debug.swipe_left()
self.debug.click(buttons.CORNER_BUTTON)
self.debug.press_no()
self.debug.press_no()
for _ in range(br.pages - 1):
self.debug.swipe_up()
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
if self.passphrase:
yield
self.debug.press_right()
yield
self.debug.press_right()
br = yield
layout = self.debug.read_layout()
if "coinjoin" in layout.title().lower() or br.code == B.UnknownDerivationPath:
self.debug.press_yes()
br = yield
# Go into details
self.debug.press_right()
# Go through details and back
self.debug.press_right()
self.debug.press_right()
self.debug.press_right()
self.debug.press_left()
self.debug.press_left()
assert br.pages is not None
for _ in range(br.pages - 1):
self.debug.press_right()
# Confirm
self.debug.press_middle()
def input_flow_t3t1(self) -> BRGeneratorType:
if self.passphrase:
yield
self.debug.press_yes()
yield
self.debug.press_yes()
br = yield
layout = self.debug.read_layout()
if "coinjoin" in layout.title().lower() or br.code == B.UnknownDerivationPath:
self.debug.press_yes()
br = yield
layout = self.debug.read_layout()
assert layout.title() in (TR.address__public_key, "XPUB")
self.debug.click(buttons.CORNER_BUTTON)
assert "VerticalMenu" in self.all_components()
# menu
self.debug.click(buttons.VERTICAL_MENU[0])
self.debug.synchronize_at("Qr")
# qr code
assert "Qr" in self.all_components()
self.debug.click(buttons.CORNER_BUTTON)
# menu
assert "VerticalMenu" in self.all_components()
self.debug.click(buttons.VERTICAL_MENU[1])
layout = self.debug.synchronize_at("AddressDetails")
# address details
assert TR.address_details__derivation_path in layout.screen_content()
self.debug.click(buttons.CORNER_BUTTON)
layout = self.debug.synchronize_at("VerticalMenu")
# menu
self.debug.click(buttons.VERTICAL_MENU[2])
# cancel
self.debug.swipe_up()
# really cancel
self.debug.click(buttons.CORNER_BUTTON)
layout = self.debug.synchronize_at("VerticalMenu")
# menu
self.debug.click(buttons.CORNER_BUTTON)
layout = self.debug.synchronize_at("Paragraphs")
# address
while "PromptScreen" not in layout.all_components():
layout = self.debug.swipe_up()
self.debug.synchronize_at("PromptScreen")
# tap to confirm
self.debug.press_yes()
class InputFlowPaymentRequestDetails(InputFlowBase):
def __init__(self, client: Client, outputs: list[messages.TxOutputType]):
super().__init__(client)
self.outputs = outputs
def input_flow_tt(self) -> BRGeneratorType:
yield # request to see details
self.debug.read_layout()
self.debug.press_info()
yield # confirm first output
assert self.outputs[0].address[:16] in self.text_content() # type: ignore
self.debug.press_yes()
yield # confirm first output
self.debug.read_layout()
self.debug.press_yes()
yield # confirm second output
assert self.outputs[1].address[:16] in self.text_content() # type: ignore
self.debug.press_yes()
yield # confirm second output
self.debug.read_layout()
self.debug.press_yes()
yield # confirm transaction
self.debug.press_yes()
yield # confirm transaction
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
yield # request to see details
self.debug.read_layout()
self.debug.press_info()
yield # confirm first output
assert self.outputs[0].address[:16] in self.text_content() # type: ignore
self.debug.swipe_up()
yield # confirm first output
self.debug.read_layout()
self.debug.swipe_up()
yield # confirm second output
assert self.outputs[1].address[:16] in self.text_content() # type: ignore
self.debug.swipe_up()
yield # confirm second output
self.debug.read_layout()
self.debug.swipe_up()
yield # confirm transaction
self.debug.swipe_up()
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]) -> BRGeneratorType:
for expected in screens:
br = yield
assert br.code == expected
self.debug.press_yes()
self.finished = True
def input_flow_tt(self) -> BRGeneratorType:
screens = [
B.ConfirmOutput,
B.ConfirmOutput,
B.FeeOverThreshold,
B.SignTx,
]
yield from self.go_through_all_screens(screens)
def input_flow_tr(self) -> BRGeneratorType:
screens = [
B.ConfirmOutput,
B.ConfirmOutput,
B.FeeOverThreshold,
B.SignTx,
]
yield from self.go_through_all_screens(screens)
def input_flow_t3t1(self) -> BRGeneratorType:
screens = [
B.ConfirmOutput,
B.ConfirmOutput,
B.FeeOverThreshold,
B.SignTx,
]
for expected in screens:
br = yield
assert br.code == expected
self.debug.swipe_up()
if br.code == B.SignTx:
self.debug.press_yes()
self.finished = True
def sign_tx_go_to_info(client: Client) -> Generator[None, messages.ButtonRequest, str]:
yield # confirm output
client.debug.read_layout()
client.debug.press_yes()
yield # confirm output
client.debug.read_layout()
client.debug.press_yes()
yield # confirm transaction
client.debug.read_layout()
client.debug.press_info()
layout = client.debug.read_layout()
content = layout.text_content()
client.debug.click(buttons.CORNER_BUTTON)
return content
def sign_tx_go_to_info_t3t1(
client: Client, multi_account: bool = False
) -> Generator[None, messages.ButtonRequest, str]:
yield # confirm output
client.debug.read_layout()
client.debug.swipe_up()
yield # confirm output
client.debug.read_layout()
client.debug.swipe_up()
if multi_account:
yield
client.debug.read_layout()
client.debug.swipe_up()
yield # confirm transaction
client.debug.read_layout()
client.debug.click(buttons.CORNER_BUTTON)
client.debug.synchronize_at("VerticalMenu")
client.debug.click(buttons.VERTICAL_MENU[0])
layout = client.debug.read_layout()
content = layout.text_content()
client.debug.click(buttons.CORNER_BUTTON)
client.debug.synchronize_at("VerticalMenu")
client.debug.click(buttons.VERTICAL_MENU[1])
layout = client.debug.read_layout()
content += " " + layout.text_content()
client.debug.click(buttons.CORNER_BUTTON)
client.debug.click(buttons.CORNER_BUTTON)
return content
def sign_tx_go_to_info_tr(
client: Client,
) -> Generator[None, messages.ButtonRequest, str]:
yield # confirm address
client.debug.read_layout()
client.debug.press_yes() # CONTINUE
yield # confirm amount
client.debug.read_layout()
client.debug.press_yes() # CONFIRM
screen_texts: list[str] = []
yield # confirm total
layout = client.debug.read_layout()
if "multiple accounts" in layout.text_content().lower():
client.debug.press_middle()
yield
layout = client.debug.press_right()
screen_texts.append(layout.visible_screen())
layout = client.debug.press_right()
screen_texts.append(layout.visible_screen())
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, title_path: str) -> None:
assert TR.translate(title_path) in content
assert "Legacy #6" in content
assert TR.confirm_total__fee_rate in content
assert "71.56 sat" in content
def input_flow_tt(self) -> BRGeneratorType:
content = yield from sign_tx_go_to_info(self.client)
self.assert_content(content, "confirm_total__sending_from_account")
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
content = yield from sign_tx_go_to_info_tr(self.client)
print("content", content)
self.assert_content(content, "confirm_total__title_sending_from")
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
content = yield from sign_tx_go_to_info_t3t1(self.client)
self.assert_content(content, "confirm_total__sending_from_account")
self.debug.swipe_up()
self.debug.press_yes()
class InputFlowSignTxInformationMixed(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def assert_content(self, content: str, title_path: str) -> None:
assert TR.translate(title_path) in content
assert TR.bitcoin__multiple_accounts in content
assert TR.confirm_total__fee_rate in content
assert "18.33 sat" in content
def input_flow_tt(self) -> BRGeneratorType:
# multiple accounts warning
yield
self.debug.press_yes()
content = yield from sign_tx_go_to_info(self.client)
self.assert_content(content, "confirm_total__sending_from_account")
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
# multiple accounts warning
yield
self.debug.press_yes()
content = yield from sign_tx_go_to_info_tr(self.client)
self.assert_content(content, "confirm_total__title_sending_from")
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
content = yield from sign_tx_go_to_info_t3t1(self.client, multi_account=True)
self.assert_content(content, "confirm_total__sending_from_account")
self.debug.swipe_up()
self.debug.press_yes()
class InputFlowSignTxInformationCancel(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> BRGeneratorType:
yield from sign_tx_go_to_info(self.client)
self.debug.press_no()
def input_flow_tr(self) -> BRGeneratorType:
yield from sign_tx_go_to_info_tr(self.client)
self.debug.press_left()
def input_flow_t3t1(self) -> BRGeneratorType:
yield from sign_tx_go_to_info_t3t1(self.client)
self.debug.click(buttons.CORNER_BUTTON)
self.debug.click(buttons.VERTICAL_MENU[2])
self.debug.synchronize_at("PromptScreen")
self.debug.click(buttons.TAP_TO_CONFIRM)
class InputFlowSignTxInformationReplacement(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> BRGeneratorType:
yield # confirm txid
self.debug.press_yes()
yield # confirm address
self.debug.press_yes()
# go back to address
yield
self.debug.press_no()
# confirm address
self.debug.press_yes()
# confirm amount
self.debug.press_yes()
yield # transaction summary, press info
self.debug.click(buttons.CORNER_BUTTON)
self.debug.click(buttons.CORNER_BUTTON)
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
yield # confirm txid
self.debug.press_right()
self.debug.press_right()
yield # modify amount - address
self.debug.press_right()
self.debug.press_right()
yield # modify amount - amount
self.debug.press_right()
yield # modify fee
self.debug.press_right()
self.debug.press_right()
self.debug.press_right()
input_flow_t3t1 = input_flow_tt
def lock_time_input_flow_tt(
debug: DebugLink,
layout_assert_func: Callable[[DebugLink, messages.ButtonRequest], None],
double_confirm: bool = False,
) -> BRGeneratorType:
yield # confirm output
debug.read_layout()
debug.press_yes()
yield # confirm output
debug.read_layout()
debug.press_yes()
br = yield # confirm locktime
layout_assert_func(debug, br)
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, messages.ButtonRequest], None],
) -> BRGeneratorType:
yield # confirm address
debug.read_layout()
debug.press_yes()
yield # confirm amount
debug.read_layout()
debug.press_yes()
br = yield # confirm locktime
layout_assert_func(debug, br)
debug.press_yes()
yield # confirm transaction
debug.press_yes()
def lock_time_input_flow_t3t1(
debug: DebugLink,
layout_assert_func: Callable[[DebugLink, messages.ButtonRequest], None],
double_confirm: bool = False,
) -> BRGeneratorType:
yield # confirm output
debug.read_layout()
debug.swipe_up()
yield # confirm output
debug.read_layout()
debug.swipe_up()
br = yield # confirm locktime
layout_assert_func(debug, br)
debug.press_yes()
yield # confirm transaction
debug.swipe_up()
debug.press_yes()
if double_confirm:
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 assert_func(self, debug: DebugLink, br: messages.ButtonRequest) -> None:
layout_text = get_text_possible_pagination(debug, br)
assert TR.bitcoin__locktime_set_to_blockheight in layout_text
assert self.block_height in layout_text
def input_flow_tt(self) -> BRGeneratorType:
yield from lock_time_input_flow_tt(
self.debug, self.assert_func, double_confirm=True
)
def input_flow_tr(self) -> BRGeneratorType:
yield from lock_time_input_flow_tr(self.debug, self.assert_func)
def input_flow_t3t1(self) -> BRGeneratorType:
yield from lock_time_input_flow_t3t1(
self.debug, self.assert_func, double_confirm=True
)
class InputFlowLockTimeDatetime(InputFlowBase):
def __init__(self, client: Client, lock_time_str: str):
super().__init__(client)
self.lock_time_str = lock_time_str
def assert_func(self, debug: DebugLink, br: messages.ButtonRequest) -> None:
layout_text = get_text_possible_pagination(debug, br)
assert TR.bitcoin__locktime_set_to in layout_text
assert self.lock_time_str.replace(" ", "") in layout_text.replace(" ", "")
def input_flow_tt(self) -> BRGeneratorType:
yield from lock_time_input_flow_tt(self.debug, self.assert_func)
def input_flow_tr(self) -> BRGeneratorType:
yield from lock_time_input_flow_tr(self.debug, self.assert_func)
def input_flow_t3t1(self) -> BRGeneratorType:
yield from lock_time_input_flow_t3t1(self.debug, self.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.client.layout_type in (LayoutType.TT, LayoutType.Mercury):
self.debug.click(self.SHOW_MORE)
elif self.client.layout_type is LayoutType.TR:
self.debug.press_right()
else:
raise NotImplementedError
def input_flow_common(self) -> BRGeneratorType:
"""Triggers show more wherever possible"""
yield # confirm address
self.debug.press_yes()
yield # confirm domain
self.debug.read_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.read_layout()
self._confirm_show_more()
yield # confirm message.from
self.debug.read_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.read_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) -> BRGeneratorType:
"""Clicks cancelling button"""
yield # confirm address
self.debug.press_yes()
yield # confirm domain
self.debug.press_no()
class InputFlowEthereumSignTxShowFeeInfo(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> BRGeneratorType:
yield from self.ETH.confirm_tx(info=True)
class InputFlowEthereumSignTxGoBackFromSummary(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> BRGeneratorType:
yield from self.ETH.confirm_tx(go_back_from_summary=True)
class InputFlowEthereumSignTxDataSkip(InputFlowBase):
def __init__(self, client: Client, cancel: bool = False):
super().__init__(client)
self.cancel = cancel
def input_flow_common(self) -> BRGeneratorType:
yield from self.ETH.confirm_data()
yield from self.ETH.confirm_tx(cancel=self.cancel)
class InputFlowEthereumSignTxDataScrollDown(InputFlowBase):
def __init__(self, client: Client, cancel: bool = False):
super().__init__(client)
self.cancel = cancel
def input_flow_common(self) -> BRGeneratorType:
# this flow will not test for the cancel case,
# because once we enter the "view all data",
# the only way to cancel is by going back to the 1st page view
# but that case would be covered by InputFlowEthereumSignTxDataGoBack
assert not self.cancel
yield from self.ETH.confirm_data(info=True)
yield from self.ETH.paginate_data()
yield from self.ETH.confirm_tx()
class InputFlowEthereumSignTxDataGoBack(InputFlowBase):
def __init__(self, client: Client, cancel: bool = False):
super().__init__(client)
self.cancel = cancel
def input_flow_common(self) -> BRGeneratorType:
yield from self.ETH.confirm_data(info=True)
yield from self.ETH.paginate_data_go_back()
if self.cancel:
yield from self.ETH.confirm_data(cancel=True)
else:
yield from self.ETH.confirm_data()
yield from self.ETH.confirm_tx()
class InputFlowEthereumSignTxStaking(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> BRGeneratorType:
yield from self.ETH.confirm_tx_staking(info=True)
def get_mnemonic_and_confirm_success(
debug: DebugLink,
) -> Generator[None, "messages.ButtonRequest", str]:
# mnemonic phrases
mnemonic = yield from read_and_confirm_mnemonic(debug)
is_slip39 = len(mnemonic.split()) in (20, 33)
if debug.layout_type in (LayoutType.TT, LayoutType.TR) or is_slip39:
br = yield # confirm recovery share 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) -> BRGeneratorType:
# 1. Backup intro
# 2. Backup warning
yield from click_through(self.debug, screens=2, 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_tt(self) -> BRGeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Backup intro
# 4. Confirm warning
yield from click_through(self.debug, screens=4, code=B.ResetDevice)
# mnemonic phrases and rest
self.mnemonic = yield from get_mnemonic_and_confirm_success(self.debug)
def input_flow_tr(self) -> BRGeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Backup intro
# 4. Confirm warning
yield from click_through(self.debug, screens=4, code=B.ResetDevice)
# mnemonic phrases and rest
self.mnemonic = yield from get_mnemonic_and_confirm_success(self.debug)
def input_flow_t3t1(self) -> BRGeneratorType:
# 1. Confirm Reset
# 2. Wallet created
# 3. Backup your seed
# 4. Backup intro
# 5. Confirm warning
yield from click_through(self.debug, screens=5, 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) -> BRGeneratorType:
br = yield # Confirm Reset
assert br.code == B.ResetDevice
self.debug.press_yes()
yield from self.PIN.setup_new_pin("654")
if self.debug.layout_type is LayoutType.Mercury:
br = yield # Wallet created
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()
br = yield # Backup intro
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) -> BRGeneratorType:
screens = 5 if self.debug.layout_type is LayoutType.Mercury else 4
# 1. Confirm Reset
# 1a. (T3T1) Walet Creation done
# 2. Confirm backup prompt
# 3. Backup your seed
# 4. Confirm warning
yield from click_through(self.debug, screens=screens, 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_N_shares(
debug: DebugLink,
n: int,
) -> Generator[None, "messages.ButtonRequest", list[str]]:
mnemonics: list[str] = []
for _ in range(n):
# 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, repeated: bool = False):
super().__init__(client)
self.mnemonics: list[str] = []
self.click_info = click_info
self.repeated = repeated
def input_flow_tt(self) -> BRGeneratorType:
if self.repeated:
assert (yield).name == "confirm_repeated_backup"
self.debug.press_yes()
assert (yield).name == "backup_intro"
self.debug.press_yes()
assert (yield).name == "slip39_checklist"
self.debug.press_yes()
assert (yield).name == "slip39_shares"
if self.click_info:
br = yield from click_info_button_tt(self.debug)
assert br.name == "slip39_shares"
self.debug.press_yes()
assert (yield).name == "slip39_checklist"
self.debug.press_yes()
assert (yield).name == "slip39_threshold"
if self.click_info:
br = yield from click_info_button_tt(self.debug)
assert br.name == "slip39_threshold"
self.debug.press_yes()
assert (yield).name == "slip39_checklist"
self.debug.press_yes()
assert (yield).name == "backup_warning"
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, 5)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
if self.repeated:
# intro confirmation screen
yield
self.debug.press_yes()
yield # 1. Backup intro
self.debug.press_yes()
yield # 2. Checklist
self.debug.press_yes()
yield # 2.5 Number of shares info
self.debug.press_yes()
yield # 3. Number of shares (5)
self.debug.input("5")
yield # 4. Checklist
self.debug.press_yes()
yield # 4.5 Threshold info
self.debug.press_yes()
yield # 5. Threshold (3)
self.debug.input("3")
yield # 6. Checklist
self.debug.press_yes()
yield # 7. Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, 5)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
if self.repeated:
# intro confirmation screen
assert (yield).name == "confirm_repeated_backup"
self.debug.press_yes()
assert (yield).name == "backup_intro"
self.debug.swipe_up()
assert (yield).name == "slip39_checklist"
self.debug.swipe_up()
assert (yield).name == "slip39_shares"
if self.click_info:
click_info_button_mercury(self.debug)
self.debug.swipe_up()
assert (yield).name == "slip39_checklist"
self.debug.swipe_up()
assert (yield).name == "slip39_threshold"
if self.click_info:
click_info_button_mercury(self.debug)
self.debug.swipe_up()
assert (yield).name == "slip39_checklist"
self.debug.swipe_up()
assert (yield).name == "backup_warning"
self.debug.swipe_up()
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, 5)
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) -> BRGeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Backup intro
# 4. Confirm warning
# 5. shares info
# 6. Set & Confirm number of shares
# 7. threshold info
# 8. Set & confirm threshold value
# 9. Confirm show seeds
yield from click_through(self.debug, screens=9, code=B.ResetDevice)
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, 5)
br = yield # safety warning
assert br.code == B.Success
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
yield # Confirm Reset
self.debug.press_yes()
yield # Backup your seed
self.debug.press_yes()
yield # Backup intro
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_N_shares(self.debug, 5)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
# 1. Confirm Reset
# 2. Wallet Created
# 3. Backup your seed
# 4. Backup intro
# 5. Set & Confirm number of shares
# 6. threshold info
# 7. Set & confirm threshold value
# 8. Confirm show seeds
# 9. Warning
# 10. Instructions
yield from click_through(self.debug, screens=10, code=B.ResetDevice)
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, 5)
br = yield # success screen
assert br.code == B.Success
self.debug.press_yes()
class InputFlowSlip39CustomBackup(InputFlowBase):
def __init__(self, client: Client, share_count: int, repeated: bool = False):
super().__init__(client)
self.mnemonics: list[str] = []
self.share_count = share_count
self.repeated = repeated
def input_flow_tt(self) -> BRGeneratorType:
if self.repeated:
yield
self.debug.press_yes()
if self.share_count > 1:
yield # Checklist
self.debug.press_yes()
else:
yield # Backup intro
self.debug.press_yes()
yield # Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, self.share_count)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
if self.repeated:
yield
self.debug.press_yes()
if self.share_count > 1:
yield # Checklist
self.debug.press_yes()
else:
yield # Backup intro
self.debug.press_yes()
yield # Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, self.share_count)
br = yield # Confirm backup
assert br.code == B.Success
self.debug.press_yes()
def input_flow_t3t1(self) -> BRGeneratorType:
if self.repeated:
yield
self.debug.press_yes()
if self.share_count > 1:
yield # Checklist
self.debug.press_yes()
else:
yield # Backup intro
self.debug.press_yes()
yield # Confirm show seeds
self.debug.press_yes()
# Mnemonic phrases
self.mnemonics = yield from load_N_shares(self.debug, self.share_count)
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) -> BRGeneratorType:
assert (yield).name == "backup_intro"
self.debug.press_yes()
assert (yield).name == "slip39_checklist"
self.debug.press_yes()
assert (yield).name == "slip39_groups"
if self.click_info:
br = yield from click_info_button_tt(self.debug)
assert br.name == "slip39_groups"
self.debug.press_yes()
assert (yield).name == "slip39_checklist"
self.debug.press_yes()
assert (yield).name == "slip39_group_threshold"
if self.click_info:
br = yield from click_info_button_tt(self.debug)
assert br.name == "slip39_group_threshold"
self.debug.press_yes()
assert (yield).name == "slip39_checklist"
self.debug.press_yes()
for _ in range(5): # for each of 5 groups
assert (yield).name == "slip39_shares"
if self.click_info:
br = yield from click_info_button_tt(self.debug)
assert br.name == "slip39_shares"
self.debug.press_yes()
assert (yield).name == "slip39_threshold"
if self.click_info:
br = yield from click_info_button_tt(self.debug)
assert br.name == "slip39_threshold"
self.debug.press_yes()
assert (yield).name == "backup_warning"
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) -> BRGeneratorType:
yield # 1. Backup intro
self.debug.press_yes()
yield # 2. Checklist
self.debug.press_yes()
yield # 3. Set and confirm group count
self.debug.input("5")
yield # 4. Checklist
self.debug.press_yes()
yield # 5. Set and confirm group threshold
self.debug.input("3")
yield # 6. 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()
def input_flow_t3t1(self) -> BRGeneratorType:
assert (yield).name == "backup_intro"
self.debug.swipe_up()
assert (yield).name == "slip39_checklist"
self.debug.swipe_up()
assert (yield).name == "slip39_groups"
if self.click_info:
click_info_button_mercury(self.debug)
self.debug.swipe_up()
assert (yield).name == "slip39_checklist"
self.debug.swipe_up()
assert (yield).name == "slip39_group_threshold"
if self.click_info:
click_info_button_mercury(self.debug)
self.debug.swipe_up()
assert (yield).name == "slip39_checklist"
self.debug.swipe_up()
for _i in range(5): # for each of 5 groups
assert (yield).name == "slip39_shares"
if self.click_info:
click_info_button_mercury(self.debug)
self.debug.swipe_up()
assert (yield).name == "slip39_threshold"
if self.click_info:
click_info_button_mercury(self.debug)
self.debug.swipe_up()
assert (yield).name == "backup_warning"
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) -> BRGeneratorType:
# 1. Confirm Reset
# 2. Backup your seed
# 3. Backup intro
# 4. Confirm warning
# 5. shares info
# 6. Set & Confirm number of groups
# 7. threshold info
# 8. Set & confirm group threshold value
# 9-18: for each of 5 groups:
# 1. Set & Confirm number of shares
# 2. Set & confirm share threshold value
# 19. Confirm show seeds
yield from click_through(self.debug, screens=19, 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) -> BRGeneratorType:
yield # Wallet backup
self.debug.press_yes()
yield # Wallet creation
self.debug.press_yes()
yield # Backup intro
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 input_flow_t3t1(self) -> BRGeneratorType:
# 1. Confirm Reset
# 2. Wallet Created
# 3. Prompt Backup
# 4. Backup intro
# 5. Confirm warning
# 6. shares info
# 7. Set & Confirm number of groups
# 8. threshold info
# 9. Set & confirm group threshold value
# 10-19: for each of 5 groups:
# 1. Set & Confirm number of shares
# 2. Set & confirm share threshold value
# 20. Confirm show seeds
yield from click_through(self.debug, screens=20, 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()
class InputFlowBip39RecoveryDryRun(InputFlowBase):
def __init__(self, client: Client, mnemonic: list[str], mismatch: bool = False):
super().__init__(client)
self.mnemonic = mnemonic
self.mismatch = mismatch
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_dry_run()
yield from self.REC.setup_bip39_recovery(len(self.mnemonic))
yield from self.REC.input_mnemonic(self.mnemonic)
if self.mismatch:
yield from self.REC.warning_bip39_dryrun_mismatch()
else:
yield from self.REC.success_bip39_dry_run_valid()
class InputFlowBip39RecoveryDryRunInvalid(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.invalid_mnemonic = ["stick"] * 12
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_dry_run()
yield from self.REC.setup_bip39_recovery(len(self.invalid_mnemonic))
yield from self.REC.input_mnemonic(self.invalid_mnemonic)
yield from self.REC.warning_invalid_recovery_seed()
yield
self.client.cancel()
class InputFlowBip39Recovery(InputFlowBase):
def __init__(self, client: Client, mnemonic: list[str], pin: str | None = None):
super().__init__(client)
self.mnemonic = mnemonic
self.pin = pin
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
if self.pin is not None:
yield from self.PIN.setup_new_pin(self.pin)
yield from self.REC.setup_bip39_recovery(len(self.mnemonic))
yield from self.REC.input_mnemonic(self.mnemonic)
yield from self.REC.success_wallet_recovered()
class InputFlowSlip39AdvancedRecoveryDryRun(InputFlowBase):
def __init__(self, client: Client, shares: list[str], mismatch: bool = False):
super().__init__(client)
self.shares = shares
self.mismatch = mismatch
self.word_count = len(shares[0].split(" "))
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_dry_run()
yield from self.REC.setup_slip39_recovery(self.word_count)
yield from self.REC.input_all_slip39_shares(self.shares, has_groups=True)
if self.mismatch:
yield from self.REC.warning_slip39_dryrun_mismatch()
else:
yield from self.REC.success_slip39_dryrun_valid()
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
self.word_count = len(shares[0].split(" "))
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
yield from self.REC.setup_slip39_recovery(self.word_count)
yield from self.REC.input_all_slip39_shares(
self.shares, has_groups=True, click_info=self.click_info
)
yield from self.REC.success_wallet_recovered()
class InputFlowSlip39AdvancedRecoveryAbort(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
if self.client.layout_type in (LayoutType.TT, LayoutType.Mercury):
yield from self.REC.input_number_of_words(20)
yield from self.REC.abort_recovery(True)
class InputFlowSlip39AdvancedRecoveryNoAbort(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
self.word_count = len(shares[0].split(" "))
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
if self.client.layout_type in (LayoutType.TT, LayoutType.Mercury):
yield from self.REC.input_number_of_words(self.word_count)
yield from self.REC.abort_recovery(False)
else:
yield from self.REC.abort_recovery(False)
yield from self.REC.tr_recovery_homescreen()
yield from self.REC.input_number_of_words(self.word_count)
yield from self.REC.enter_any_share()
yield from self.REC.input_all_slip39_shares(self.shares, has_groups=True)
yield from self.REC.success_wallet_recovered()
class InputFlowSlip39AdvancedRecoveryThresholdReached(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) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
yield from self.REC.setup_slip39_recovery(len(self.first_share))
yield from self.REC.input_mnemonic(self.first_share)
yield from self.REC.success_share_group_entered()
yield from self.REC.success_more_shares_needed()
yield from self.REC.input_mnemonic(self.second_share)
yield from self.REC.warning_group_threshold_reached()
yield
self.client.cancel()
class InputFlowSlip39AdvancedRecoveryShareAlreadyEntered(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) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
yield from self.REC.setup_slip39_recovery(len(self.first_share))
yield from self.REC.input_mnemonic(self.first_share)
yield from self.REC.success_share_group_entered()
yield from self.REC.success_more_shares_needed()
yield from self.REC.input_mnemonic(self.second_share)
yield from self.REC.warning_share_already_entered()
yield
self.client.cancel()
class InputFlowSlip39BasicRecoveryDryRun(InputFlowBase):
def __init__(
self,
client: Client,
shares: list[str],
mismatch: bool = False,
unlock_repeated_backup=False,
):
super().__init__(client)
self.shares = shares
self.mismatch = mismatch
self.unlock_repeated_backup = unlock_repeated_backup
self.word_count = len(shares[0].split(" "))
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_dry_run()
if self.unlock_repeated_backup:
yield from self.REC.setup_repeated_backup_recovery(self.word_count)
else:
yield from self.REC.setup_slip39_recovery(self.word_count)
yield from self.REC.input_all_slip39_shares(self.shares)
if self.mismatch:
yield from self.REC.warning_slip39_dryrun_mismatch()
elif not self.unlock_repeated_backup:
yield from self.REC.success_slip39_dryrun_valid()
class InputFlowSlip39BasicRecovery(InputFlowBase):
def __init__(self, client: Client, shares: list[str], pin: str | None = None):
super().__init__(client)
self.shares = shares
self.pin = pin
self.word_count = len(shares[0].split(" "))
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
if self.pin is not None:
yield from self.PIN.setup_new_pin(self.pin)
yield from self.REC.setup_slip39_recovery(self.word_count)
yield from self.REC.input_all_slip39_shares(self.shares)
yield from self.REC.success_wallet_recovered()
class InputFlowSlip39BasicRecoveryAbort(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
if self.client.layout_type in (LayoutType.TT, LayoutType.Mercury):
yield from self.REC.input_number_of_words(20)
yield from self.REC.abort_recovery(True)
class InputFlowSlip39BasicRecoveryAbortBetweenShares(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.first_share = shares[0].split(" ")
self.word_count = len(self.first_share)
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
if self.client.layout_type in (LayoutType.TT, LayoutType.Mercury):
yield from self.REC.input_number_of_words(20)
else:
yield from self.REC.tr_recovery_homescreen()
yield from self.REC.input_number_of_words(self.word_count)
yield from self.REC.enter_any_share()
yield from self.REC.input_mnemonic(self.first_share)
yield from self.REC.abort_recovery_between_shares()
class InputFlowSlip39BasicRecoveryNoAbort(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
self.word_count = len(shares[0].split(" "))
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
if self.client.layout_type in (LayoutType.TT, LayoutType.Mercury):
yield from self.REC.input_number_of_words(self.word_count)
yield from self.REC.abort_recovery(False)
else:
yield from self.REC.abort_recovery(False)
yield from self.REC.tr_recovery_homescreen()
yield from self.REC.input_number_of_words(self.word_count)
yield from self.REC.enter_any_share()
yield from self.REC.input_all_slip39_shares(self.shares)
yield from self.REC.success_wallet_recovered()
class InputFlowSlip39BasicRecoveryInvalidFirstShare(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
self.first_invalid = ["slush"] * 20
self.second_invalid = ["slush"] * 33
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
yield from self.REC.setup_slip39_recovery(len(self.first_invalid))
yield from self.REC.input_mnemonic(self.first_invalid)
yield from self.REC.warning_invalid_recovery_share()
yield from self.REC.setup_slip39_recovery(len(self.second_invalid))
yield from self.REC.input_mnemonic(self.second_invalid)
yield from self.REC.warning_invalid_recovery_share()
yield
self.client.cancel()
class InputFlowSlip39BasicRecoveryInvalidSecondShare(InputFlowBase):
def __init__(self, client: Client, shares: list[str]):
super().__init__(client)
self.shares = shares
self.first_share = shares[0].split(" ")
self.invalid_share = self.first_share[:3] + ["slush"] * 17
self.second_share = shares[1].split(" ")
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
yield from self.REC.setup_slip39_recovery(len(self.first_share))
yield from self.REC.input_mnemonic(self.first_share)
yield from self.REC.success_more_shares_needed(2)
yield from self.REC.input_mnemonic(self.invalid_share)
yield from self.REC.warning_invalid_recovery_share()
yield from self.REC.input_mnemonic(self.second_share)
yield from self.REC.success_more_shares_needed(1)
yield
self.client.cancel()
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
# Invalid share - just enough words to trigger the warning
self.modified_share = share[:nth_word] + [self.share[-1]]
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
yield from self.REC.setup_slip39_recovery(len(self.share))
yield from self.REC.input_mnemonic(self.share)
yield from self.REC.success_more_shares_needed()
yield from self.REC.input_mnemonic(self.modified_share)
yield from self.REC.warning_share_from_another_shamir()
yield
self.client.cancel()
class InputFlowSlip39BasicRecoverySameShare(InputFlowBase):
def __init__(self, client: Client, share: list[str]):
super().__init__(client)
self.share = share
# Second duplicate share - only 4 words are needed to verify it
self.duplicate_share = self.share[:4]
def input_flow_common(self) -> BRGeneratorType:
yield from self.REC.confirm_recovery()
yield from self.REC.setup_slip39_recovery(len(self.share))
yield from self.REC.input_mnemonic(self.share)
yield from self.REC.success_more_shares_needed()
yield from self.REC.input_mnemonic(self.duplicate_share)
yield from self.REC.warning_share_already_entered()
yield
self.client.cancel()
class InputFlowResetSkipBackup(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> BRGeneratorType:
yield from self.BAK.confirm_new_wallet()
yield # Skip Backup
assert TR.backup__new_wallet_successfully_created in self.text_content()
self.debug.press_no()
yield # Confirm skip backup
assert TR.backup__want_to_skip in self.text_content()
self.debug.press_no()
def input_flow_tr(self) -> BRGeneratorType:
yield from self.BAK.confirm_new_wallet()
yield # Skip Backup
assert TR.backup__new_wallet_created in self.text_content()
self.debug.press_right()
self.debug.press_no()
yield # Confirm skip backup
assert TR.backup__want_to_skip in self.text_content()
self.debug.press_no()
def input_flow_t3t1(self) -> BRGeneratorType:
yield from self.BAK.confirm_new_wallet()
yield # Skip Backup
assert TR.backup__new_wallet_created in self.text_content()
self.debug.swipe_up()
yield
self.debug.click(buttons.CORNER_BUTTON)
self.debug.synchronize_at("VerticalMenu")
self.debug.click(buttons.VERTICAL_MENU[0])
self.debug.swipe_up()
self.debug.synchronize_at("PromptScreen")
self.debug.click(buttons.TAP_TO_CONFIRM)
class InputFlowConfirmAllWarnings(InputFlowBase):
def __init__(self, client: Client):
super().__init__(client)
def input_flow_tt(self) -> BRGeneratorType:
br = yield
while True:
# wait for homescreen to go away
self.debug.read_layout()
self.client.ui._default_input_flow(br)
br = yield
def input_flow_tr(self) -> BRGeneratorType:
return self.input_flow_tt()
def input_flow_t3t1(self) -> BRGeneratorType:
br = yield
while True:
# wait for homescreen to go away
# probably won't be needed after https://github.com/trezor/trezor-firmware/pull/3686
self.debug.read_layout()
# Paginating (going as further as possible) and pressing Yes
if br.pages is not None:
for _ in range(br.pages - 1):
self.debug.swipe_up()
layout = self.debug.read_layout()
text = layout.text_content().lower()
# hi priority warning
hi_prio = (
TR.ethereum__unknown_contract_address,
TR.addr_mismatch__wrong_derivation_path,
TR.send__receiving_to_multisig,
"witness path",
"certificate path",
"pool owner staking path",
"using different paths for different xpubs",
)
if any(needle.lower() in text for needle in hi_prio):
self.debug.click(buttons.CORNER_BUTTON)
self.debug.synchronize_at("VerticalMenu")
self.debug.click(buttons.VERTICAL_MENU[1])
elif "PromptScreen" in layout.all_components():
self.debug.press_yes()
elif "SwipeContent" in layout.all_components():
self.debug.swipe_up()
else:
self.debug.press_yes()
br = yield
class InputFlowFidoConfirm(InputFlowBase):
def __init__(self, client: Client, cancel: bool = False):
super().__init__(client)
self.cancel = cancel
def input_flow_tt(self) -> BRGeneratorType:
while True:
yield
self.debug.press_yes()
def input_flow_tr(self) -> BRGeneratorType:
yield from self.input_flow_tt()
def input_flow_t3t1(self) -> BRGeneratorType:
while True:
yield
self.debug.swipe_up()
self.debug.click(buttons.TAP_TO_CONFIRM)
Reference in New Issue View Git Blame Copy Permalink