mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-08-02 03:48:58 +00:00
device_tests: sample usage of input_flow
This commit is contained in:
matejcik
parent
0f7f694914
commit
886d4f18f4
@ -15,81 +15,88 @@
|
|||||||
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
|
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
|
||||||
|
|
||||||
import time
|
import time
|
||||||
|
from unittest import mock
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from mnemonic import Mnemonic
|
from mnemonic import Mnemonic
|
||||||
|
|
||||||
from trezorlib import device, messages as proto
|
from trezorlib import device, messages as proto
|
||||||
|
from trezorlib.messages import ButtonRequestType as B
|
||||||
|
|
||||||
from .common import TrezorTest, generate_entropy
|
from .common import TrezorTest, generate_entropy
|
||||||
|
|
||||||
|
EXTERNAL_ENTROPY = b"zlutoucky kun upel divoke ody" * 2
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skip_t1
|
@pytest.mark.skip_t1
|
||||||
class TestMsgResetDeviceT2(TrezorTest):
|
class TestMsgResetDeviceT2(TrezorTest):
|
||||||
def test_reset_device(self):
|
def test_reset_device(self):
|
||||||
|
words = []
|
||||||
|
strength = 128
|
||||||
|
|
||||||
|
def input_flow():
|
||||||
|
# Backup your seed
|
||||||
|
btn_code = yield
|
||||||
|
assert btn_code == B.ResetDevice
|
||||||
|
self.client.debug.press_yes()
|
||||||
|
|
||||||
|
# mnemonic phrases
|
||||||
|
btn_code = yield
|
||||||
|
assert btn_code == B.ResetDevice
|
||||||
|
# 12 words, 3 pages
|
||||||
|
for i in range(3):
|
||||||
|
time.sleep(0.3)
|
||||||
|
words.extend(self.client.debug.state().reset_word.split())
|
||||||
|
if i < 2:
|
||||||
|
self.client.debug.swipe_down()
|
||||||
|
else:
|
||||||
|
# last page is confirmation
|
||||||
|
self.client.debug.press_yes()
|
||||||
|
|
||||||
|
# check backup words
|
||||||
|
for _ in range(2):
|
||||||
|
time.sleep(0.3)
|
||||||
|
index = self.client.debug.state().reset_word_pos
|
||||||
|
self.client.debug.input(words[index])
|
||||||
|
|
||||||
|
# safety warning
|
||||||
|
btn_code = yield
|
||||||
|
assert btn_code == B.ResetDevice
|
||||||
|
self.client.debug.press_yes()
|
||||||
|
|
||||||
|
os_urandom = mock.Mock(return_value=EXTERNAL_ENTROPY)
|
||||||
|
with mock.patch("os.urandom", os_urandom), self.client:
|
||||||
|
self.client.set_expected_responses(
|
||||||
|
[
|
||||||
|
proto.EntropyRequest(),
|
||||||
|
proto.ButtonRequest(code=B.ResetDevice),
|
||||||
|
proto.ButtonRequest(code=B.ResetDevice),
|
||||||
|
proto.ButtonRequest(code=B.ResetDevice),
|
||||||
|
proto.Success(),
|
||||||
|
proto.Features(),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
self.client.set_input_flow(input_flow)
|
||||||
|
|
||||||
# No PIN, no passphrase, don't display random
|
# No PIN, no passphrase, don't display random
|
||||||
external_entropy = b"zlutoucky kun upel divoke ody" * 2
|
device.reset(
|
||||||
strength = 128
|
self.client,
|
||||||
ret = self.client.call_raw(
|
|
||||||
proto.ResetDevice(
|
|
||||||
display_random=False,
|
display_random=False,
|
||||||
strength=strength,
|
strength=strength,
|
||||||
passphrase_protection=False,
|
passphrase_protection=False,
|
||||||
pin_protection=False,
|
pin_protection=False,
|
||||||
label="test",
|
label="test",
|
||||||
)
|
language="english",
|
||||||
)
|
)
|
||||||
|
|
||||||
# Provide entropy
|
# generate mnemonic locally
|
||||||
assert isinstance(ret, proto.EntropyRequest)
|
internal_entropy = self.client.debug.state().reset_entropy
|
||||||
internal_entropy = self.client.debug.read_reset_entropy()
|
entropy = generate_entropy(strength, internal_entropy, EXTERNAL_ENTROPY)
|
||||||
ret = self.client.call_raw(proto.EntropyAck(entropy=external_entropy))
|
|
||||||
|
|
||||||
# Generate mnemonic locally
|
|
||||||
entropy = generate_entropy(strength, internal_entropy, external_entropy)
|
|
||||||
expected_mnemonic = Mnemonic("english").to_mnemonic(entropy)
|
expected_mnemonic = Mnemonic("english").to_mnemonic(entropy)
|
||||||
|
|
||||||
# Safety warning
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
ret = self.client.call_raw(proto.ButtonAck())
|
|
||||||
|
|
||||||
# List through mnemonic pages
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.transport.write(proto.ButtonAck())
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
words = []
|
|
||||||
time.sleep(1)
|
|
||||||
words.extend(self.client.debug.read_reset_word().split())
|
|
||||||
self.client.debug.swipe_down()
|
|
||||||
time.sleep(1)
|
|
||||||
words.extend(self.client.debug.read_reset_word().split())
|
|
||||||
self.client.debug.swipe_down()
|
|
||||||
time.sleep(1)
|
|
||||||
words.extend(self.client.debug.read_reset_word().split())
|
|
||||||
|
|
||||||
# Compare that device generated proper mnemonic for given entropies
|
# Compare that device generated proper mnemonic for given entropies
|
||||||
assert " ".join(words) == expected_mnemonic
|
assert " ".join(words) == expected_mnemonic
|
||||||
|
|
||||||
# Confirm the mnemonic
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
|
|
||||||
# Check mnemonic words
|
|
||||||
time.sleep(1)
|
|
||||||
index = self.client.debug.read_reset_word_pos()
|
|
||||||
self.client.debug.input(words[index])
|
|
||||||
time.sleep(1)
|
|
||||||
index = self.client.debug.read_reset_word_pos()
|
|
||||||
self.client.debug.input(words[index])
|
|
||||||
ret = self.client.transport.read()
|
|
||||||
|
|
||||||
# Safety warning
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
ret = self.client.call_raw(proto.ButtonAck())
|
|
||||||
assert isinstance(ret, proto.Success)
|
|
||||||
|
|
||||||
# Check if device is properly initialized
|
# Check if device is properly initialized
|
||||||
resp = self.client.call_raw(proto.Initialize())
|
resp = self.client.call_raw(proto.Initialize())
|
||||||
assert resp.initialized is True
|
assert resp.initialized is True
|
||||||
@ -98,84 +105,88 @@ class TestMsgResetDeviceT2(TrezorTest):
|
|||||||
assert resp.passphrase_protection is False
|
assert resp.passphrase_protection is False
|
||||||
|
|
||||||
def test_reset_device_pin(self):
|
def test_reset_device_pin(self):
|
||||||
|
words = []
|
||||||
|
strength = 128
|
||||||
|
|
||||||
|
def input_flow():
|
||||||
|
# Enter new PIN
|
||||||
|
yield
|
||||||
|
self.client.debug.input("654")
|
||||||
|
|
||||||
|
# Confirm PIN
|
||||||
|
yield
|
||||||
|
self.client.debug.input("654")
|
||||||
|
|
||||||
|
# Confirm entropy
|
||||||
|
btn_code = yield
|
||||||
|
assert btn_code == B.ResetDevice
|
||||||
|
self.client.debug.press_yes()
|
||||||
|
|
||||||
|
# Backup your seed
|
||||||
|
btn_code = yield
|
||||||
|
assert btn_code == B.ResetDevice
|
||||||
|
self.client.debug.press_yes()
|
||||||
|
|
||||||
|
# mnemonic phrases
|
||||||
|
btn_code = yield
|
||||||
|
assert btn_code == B.ResetDevice
|
||||||
|
# 12 words, 3 pages
|
||||||
|
for i in range(3):
|
||||||
|
time.sleep(0.3)
|
||||||
|
words.extend(self.client.debug.state().reset_word.split())
|
||||||
|
if i < 2:
|
||||||
|
self.client.debug.swipe_down()
|
||||||
|
else:
|
||||||
|
# last page is confirmation
|
||||||
|
self.client.debug.press_yes()
|
||||||
|
|
||||||
|
# check backup words
|
||||||
|
for _ in range(2):
|
||||||
|
time.sleep(0.3)
|
||||||
|
index = self.client.debug.state().reset_word_pos
|
||||||
|
self.client.debug.input(words[index])
|
||||||
|
|
||||||
|
# safety warning
|
||||||
|
btn_code = yield
|
||||||
|
assert btn_code == B.ResetDevice
|
||||||
|
self.client.debug.press_yes()
|
||||||
|
|
||||||
|
os_urandom = mock.Mock(return_value=EXTERNAL_ENTROPY)
|
||||||
|
with mock.patch("os.urandom", os_urandom), self.client:
|
||||||
|
self.client.set_expected_responses(
|
||||||
|
[
|
||||||
|
proto.ButtonRequest(code=B.Other),
|
||||||
|
proto.ButtonRequest(code=B.Other),
|
||||||
|
proto.ButtonRequest(code=B.ResetDevice),
|
||||||
|
proto.EntropyRequest(),
|
||||||
|
proto.ButtonRequest(code=B.ResetDevice),
|
||||||
|
proto.ButtonRequest(code=B.ResetDevice),
|
||||||
|
proto.ButtonRequest(code=B.ResetDevice),
|
||||||
|
proto.Success(),
|
||||||
|
proto.Features(),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
self.client.set_input_flow(input_flow)
|
||||||
|
|
||||||
# PIN, passphrase, display random
|
# PIN, passphrase, display random
|
||||||
external_entropy = b"zlutoucky kun upel divoke ody" * 2
|
device.reset(
|
||||||
strength = 128
|
self.client,
|
||||||
ret = self.client.call_raw(
|
|
||||||
proto.ResetDevice(
|
|
||||||
display_random=True,
|
display_random=True,
|
||||||
strength=strength,
|
strength=strength,
|
||||||
passphrase_protection=True,
|
passphrase_protection=True,
|
||||||
pin_protection=True,
|
pin_protection=True,
|
||||||
label="test",
|
label="test",
|
||||||
)
|
language="english",
|
||||||
)
|
)
|
||||||
|
|
||||||
# Enter PIN for first time
|
# generate mnemonic locally
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
internal_entropy = self.client.debug.state().reset_entropy
|
||||||
self.client.debug.input("654")
|
entropy = generate_entropy(strength, internal_entropy, EXTERNAL_ENTROPY)
|
||||||
ret = self.client.call_raw(proto.ButtonAck())
|
|
||||||
|
|
||||||
# Enter PIN for second time
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.debug.input("654")
|
|
||||||
ret = self.client.call_raw(proto.ButtonAck())
|
|
||||||
|
|
||||||
# Confirm entropy
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
ret = self.client.call_raw(proto.ButtonAck())
|
|
||||||
|
|
||||||
# Provide entropy
|
|
||||||
assert isinstance(ret, proto.EntropyRequest)
|
|
||||||
internal_entropy = self.client.debug.read_reset_entropy()
|
|
||||||
ret = self.client.call_raw(proto.EntropyAck(entropy=external_entropy))
|
|
||||||
|
|
||||||
# Generate mnemonic locally
|
|
||||||
entropy = generate_entropy(strength, internal_entropy, external_entropy)
|
|
||||||
expected_mnemonic = Mnemonic("english").to_mnemonic(entropy)
|
expected_mnemonic = Mnemonic("english").to_mnemonic(entropy)
|
||||||
|
|
||||||
# Safety warning
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
ret = self.client.call_raw(proto.ButtonAck())
|
|
||||||
|
|
||||||
# List through mnemonic pages
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.transport.write(proto.ButtonAck())
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
words = []
|
|
||||||
time.sleep(1)
|
|
||||||
words.extend(self.client.debug.read_reset_word().split())
|
|
||||||
self.client.debug.swipe_down()
|
|
||||||
time.sleep(1)
|
|
||||||
words.extend(self.client.debug.read_reset_word().split())
|
|
||||||
self.client.debug.swipe_down()
|
|
||||||
time.sleep(1)
|
|
||||||
words.extend(self.client.debug.read_reset_word().split())
|
|
||||||
|
|
||||||
# Compare that device generated proper mnemonic for given entropies
|
# Compare that device generated proper mnemonic for given entropies
|
||||||
assert " ".join(words) == expected_mnemonic
|
assert " ".join(words) == expected_mnemonic
|
||||||
|
|
||||||
# Confirm the mnemonic
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
|
|
||||||
# Check mnemonic words
|
|
||||||
time.sleep(1)
|
|
||||||
index = self.client.debug.read_reset_word_pos()
|
|
||||||
self.client.debug.input(words[index])
|
|
||||||
time.sleep(1)
|
|
||||||
index = self.client.debug.read_reset_word_pos()
|
|
||||||
self.client.debug.input(words[index])
|
|
||||||
ret = self.client.transport.read()
|
|
||||||
|
|
||||||
# Safety warning
|
|
||||||
assert isinstance(ret, proto.ButtonRequest)
|
|
||||||
self.client.debug.press_yes()
|
|
||||||
ret = self.client.call_raw(proto.ButtonAck())
|
|
||||||
assert isinstance(ret, proto.Success)
|
|
||||||
|
|
||||||
# Check if device is properly initialized
|
# Check if device is properly initialized
|
||||||
resp = self.client.call_raw(proto.Initialize())
|
resp = self.client.call_raw(proto.Initialize())
|
||||||
assert resp.initialized is True
|
assert resp.initialized is True
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user