You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
153 lines
4.7 KiB
153 lines
4.7 KiB
# 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>.
|
|
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from trezorlib import device, exceptions, messages, models
|
|
from trezorlib.debuglink import TrezorClientDebugLink as Client
|
|
|
|
from ...common import MNEMONIC12
|
|
from ...input_flows import (
|
|
InputFlowBip39RecoveryDryRun,
|
|
InputFlowBip39RecoveryDryRunInvalid,
|
|
)
|
|
|
|
|
|
def do_recover_legacy(client: Client, mnemonic: list[str], **kwargs: Any):
|
|
def input_callback(_):
|
|
word, pos = client.debug.read_recovery_word()
|
|
if pos != 0 and pos is not None:
|
|
word = mnemonic[pos - 1]
|
|
mnemonic[pos - 1] = None
|
|
assert word is not None
|
|
|
|
return word
|
|
|
|
ret = device.recover(
|
|
client,
|
|
dry_run=True,
|
|
word_count=len(mnemonic),
|
|
type=messages.RecoveryDeviceType.ScrambledWords,
|
|
input_callback=input_callback,
|
|
**kwargs
|
|
)
|
|
# if the call succeeded, check that all words have been used
|
|
assert all(m is None for m in mnemonic)
|
|
return ret
|
|
|
|
|
|
def do_recover_core(client: Client, mnemonic: list[str], mismatch: bool = False):
|
|
with client:
|
|
client.watch_layout()
|
|
IF = InputFlowBip39RecoveryDryRun(client, mnemonic, mismatch=mismatch)
|
|
client.set_input_flow(IF.get())
|
|
return device.recover(client, dry_run=True)
|
|
|
|
|
|
def do_recover(client: Client, mnemonic: list[str], mismatch: bool = False):
|
|
if client.model is models.T1B1:
|
|
return do_recover_legacy(client, mnemonic)
|
|
else:
|
|
return do_recover_core(client, mnemonic, mismatch)
|
|
|
|
|
|
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
|
|
def test_dry_run(client: Client):
|
|
ret = do_recover(client, MNEMONIC12.split(" "))
|
|
assert isinstance(ret, messages.Success)
|
|
|
|
|
|
@pytest.mark.setup_client(mnemonic=MNEMONIC12)
|
|
def test_seed_mismatch(client: Client):
|
|
with pytest.raises(
|
|
exceptions.TrezorFailure, match="does not match the one in the device"
|
|
):
|
|
do_recover(client, ["all"] * 12, mismatch=True)
|
|
|
|
|
|
@pytest.mark.skip_t2t1
|
|
@pytest.mark.skip_t2b1
|
|
@pytest.mark.skip_t3t1
|
|
def test_invalid_seed_t1(client: Client):
|
|
with pytest.raises(exceptions.TrezorFailure, match="Invalid seed"):
|
|
do_recover(client, ["stick"] * 12)
|
|
|
|
|
|
@pytest.mark.skip_t1b1
|
|
def test_invalid_seed_core(client: Client):
|
|
with client:
|
|
client.watch_layout()
|
|
IF = InputFlowBip39RecoveryDryRunInvalid(client)
|
|
client.set_input_flow(IF.get())
|
|
with pytest.raises(exceptions.Cancelled):
|
|
return device.recover(client, dry_run=True)
|
|
|
|
|
|
@pytest.mark.setup_client(uninitialized=True)
|
|
def test_uninitialized(client: Client):
|
|
with pytest.raises(exceptions.TrezorFailure, match="not initialized"):
|
|
do_recover(client, ["all"] * 12)
|
|
|
|
|
|
DRY_RUN_ALLOWED_FIELDS = (
|
|
"dry_run",
|
|
"word_count",
|
|
"enforce_wordlist",
|
|
"type",
|
|
"show_tutorial",
|
|
)
|
|
|
|
|
|
def _make_bad_params():
|
|
"""Generate a list of field names that must NOT be set on a dry-run message,
|
|
and default values of the appropriate type.
|
|
"""
|
|
for field in messages.RecoveryDevice.FIELDS.values():
|
|
# language is not supported anymore:
|
|
if field.name == "language":
|
|
continue
|
|
|
|
if field.name in DRY_RUN_ALLOWED_FIELDS:
|
|
continue
|
|
|
|
if "int" in field.type:
|
|
yield field.name, 1
|
|
elif field.type == "bool":
|
|
yield field.name, True
|
|
elif field.type == "string":
|
|
yield field.name, "test"
|
|
else:
|
|
# Someone added a field to RecoveryDevice of a type that has no assigned
|
|
# default value. This test must be fixed.
|
|
raise RuntimeError("unknown field in RecoveryDevice")
|
|
|
|
|
|
@pytest.mark.parametrize("field_name, field_value", _make_bad_params())
|
|
def test_bad_parameters(client: Client, field_name: str, field_value: Any):
|
|
msg = messages.RecoveryDevice(
|
|
dry_run=True,
|
|
word_count=12,
|
|
enforce_wordlist=True,
|
|
type=messages.RecoveryDeviceType.ScrambledWords,
|
|
)
|
|
setattr(msg, field_name, field_value)
|
|
with pytest.raises(
|
|
exceptions.TrezorFailure, match="Forbidden field set in dry-run"
|
|
):
|
|
client.call(msg)
|