# This file is part of the Trezor project. # # Copyright (C) 2012-2019 SatoshiLabs and contributors # # This library is free software: you can redistribute it and/or modify # it under the terms of the GNU Lesser General Public License version 3 # as published by the Free Software Foundation. # # This library is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Lesser General Public License for more details. # # You should have received a copy of the License along with this library. # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>. import pytest import shamir_mnemonic as shamir from trezorlib import device, messages from trezorlib.debuglink import TrezorClientDebugLink as Client from trezorlib.exceptions import TrezorFailure from trezorlib.messages import ButtonRequestType as B from ..common import ( MNEMONIC12, MNEMONIC_SLIP39_ADVANCED_20, MNEMONIC_SLIP39_BASIC_20_3of6, read_and_confirm_mnemonic, ) def click_info_button(debug): """Click Shamir backup info button and return back.""" debug.press_info() yield # Info screen with text debug.press_yes() @pytest.mark.skip_t1 # TODO we want this for t1 too @pytest.mark.setup_client(needs_backup=True, mnemonic=MNEMONIC12) def test_backup_bip39(client: Client): assert client.features.needs_backup is True mnemonic = None def input_flow(): nonlocal mnemonic yield # Confirm Backup client.debug.press_yes() # Mnemonic phrases mnemonic = yield from read_and_confirm_mnemonic(client.debug) yield # Confirm success client.debug.press_yes() yield # Backup is done client.debug.press_yes() with client: client.set_input_flow(input_flow) client.set_expected_responses( [ messages.ButtonRequest(code=B.ResetDevice), messages.ButtonRequest(code=B.ResetDevice), messages.ButtonRequest(code=B.Success), messages.ButtonRequest(code=B.Success), messages.Success, messages.Features, ] ) device.backup(client) assert mnemonic == MNEMONIC12 client.init_device() assert client.features.initialized is True assert client.features.needs_backup is False assert client.features.unfinished_backup is False assert client.features.no_backup is False assert client.features.backup_type is messages.BackupType.Bip39 @pytest.mark.skip_t1 @pytest.mark.setup_client(needs_backup=True, mnemonic=MNEMONIC_SLIP39_BASIC_20_3of6) @pytest.mark.parametrize( "click_info", [True, False], ids=["click_info", "no_click_info"] ) def test_backup_slip39_basic(client: Client, click_info: bool): assert client.features.needs_backup is True mnemonics = [] def input_flow(): yield # Checklist client.debug.press_yes() if click_info: yield from click_info_button(client.debug) yield # Number of shares (5) client.debug.press_yes() yield # Checklist client.debug.press_yes() if click_info: yield from click_info_button(client.debug) yield # Threshold (3) client.debug.press_yes() yield # Checklist client.debug.press_yes() yield # Confirm show seeds client.debug.press_yes() # Mnemonic phrases for _ in range(5): # Phrase screen mnemonic = yield from read_and_confirm_mnemonic(client.debug) mnemonics.append(mnemonic) yield # Confirm continue to next client.debug.press_yes() yield # Confirm backup client.debug.press_yes() with client: client.set_input_flow(input_flow) client.set_expected_responses( [messages.ButtonRequest(code=B.ResetDevice)] * (8 if click_info else 6) # intro screens (and optional info) + [ messages.ButtonRequest(code=B.ResetDevice), messages.ButtonRequest(code=B.Success), ] * 5 # individual shares + [ messages.ButtonRequest(code=B.Success), messages.Success, messages.Features, ] ) device.backup(client) client.init_device() assert client.features.initialized is True assert client.features.needs_backup is False assert client.features.unfinished_backup is False assert client.features.no_backup is False assert client.features.backup_type is messages.BackupType.Slip39_Basic expected_ms = shamir.combine_mnemonics(MNEMONIC_SLIP39_BASIC_20_3of6) actual_ms = shamir.combine_mnemonics(mnemonics[:3]) assert expected_ms == actual_ms @pytest.mark.skip_t1 @pytest.mark.setup_client(needs_backup=True, mnemonic=MNEMONIC_SLIP39_ADVANCED_20) @pytest.mark.parametrize( "click_info", [True, False], ids=["click_info", "no_click_info"] ) def test_backup_slip39_advanced(client: Client, click_info: bool): assert client.features.needs_backup is True mnemonics = [] def input_flow(): yield # Checklist client.debug.press_yes() if click_info: yield from click_info_button(client.debug) yield # Set and confirm group count client.debug.press_yes() yield # Checklist client.debug.press_yes() if click_info: yield from click_info_button(client.debug) yield # Set and confirm group threshold client.debug.press_yes() yield # Checklist client.debug.press_yes() for _ in range(5): # for each of 5 groups if click_info: yield from click_info_button(client.debug) yield # Set & Confirm number of shares client.debug.press_yes() if click_info: yield from click_info_button(client.debug) yield # Set & confirm share threshold value client.debug.press_yes() yield # Confirm show seeds client.debug.press_yes() # Mnemonic phrases for _ in range(5): for _ in range(5): # Phrase screen mnemonic = yield from read_and_confirm_mnemonic(client.debug) mnemonics.append(mnemonic) yield # Confirm continue to next client.debug.press_yes() yield # Confirm backup client.debug.press_yes() with client: client.set_input_flow(input_flow) client.set_expected_responses( [messages.ButtonRequest(code=B.ResetDevice)] * (8 if click_info else 6) # intro screens (and optional info) + [ (click_info, messages.ButtonRequest(code=B.ResetDevice)), messages.ButtonRequest(code=B.ResetDevice), (click_info, messages.ButtonRequest(code=B.ResetDevice)), messages.ButtonRequest(code=B.ResetDevice), ] * 5 # group thresholds (and optional info) + [ messages.ButtonRequest(code=B.ResetDevice), messages.ButtonRequest(code=B.Success), ] * 25 # individual shares + [ messages.ButtonRequest(code=B.Success), messages.Success, messages.Features, ] ) device.backup(client) client.init_device() assert client.features.initialized is True assert client.features.needs_backup is False assert client.features.unfinished_backup is False assert client.features.no_backup is False assert client.features.backup_type is messages.BackupType.Slip39_Advanced expected_ms = shamir.combine_mnemonics(MNEMONIC_SLIP39_ADVANCED_20) actual_ms = shamir.combine_mnemonics( mnemonics[:3] + mnemonics[5:8] + mnemonics[10:13] ) assert expected_ms == actual_ms # we only test this with bip39 because the code path is always the same @pytest.mark.setup_client(no_backup=True) def test_no_backup_fails(client: Client): client.ensure_unlocked() assert client.features.initialized is True assert client.features.no_backup is True assert client.features.needs_backup is False # backup attempt should fail because no_backup=True with pytest.raises(TrezorFailure, match=r".*Seed already backed up"): device.backup(client) # we only test this with bip39 because the code path is always the same @pytest.mark.setup_client(needs_backup=True) def test_interrupt_backup_fails(client: Client): client.ensure_unlocked() assert client.features.initialized is True assert client.features.needs_backup is True assert client.features.unfinished_backup is False assert client.features.no_backup is False # start backup client.call_raw(messages.BackupDevice()) # interupt backup by sending initialize client.init_device() # check that device state is as expected assert client.features.initialized is True assert client.features.needs_backup is False assert client.features.unfinished_backup is True assert client.features.no_backup is False # Second attempt at backup should fail with pytest.raises(TrezorFailure, match=r".*Seed already backed up"): device.backup(client) # we only test this with bip39 because the code path is always the same @pytest.mark.setup_client(uninitialized=True) def test_no_backup_show_entropy_fails(client: Client): with pytest.raises( TrezorFailure, match=r".*Can't show internal entropy when backup is skipped" ): device.reset( client, display_random=True, strength=128, passphrase_protection=False, pin_protection=False, label="test", language="en-US", no_backup=True, )