1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 05:58:23 +00:00
trezor-firmware/tests/device_tests/test_autolock.py

170 lines
4.8 KiB
Python

#
# 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 time
import pytest
from trezorlib import device, messages, models
from trezorlib.debuglink import TrezorClientDebugLink as Client
from trezorlib.exceptions import TrezorFailure
from ..common import TEST_ADDRESS_N, get_test_address
PIN4 = "1234"
pytestmark = pytest.mark.setup_client(pin=PIN4)
def pin_request(client: Client):
return (
messages.PinMatrixRequest
if client.model is models.T1B1
else messages.ButtonRequest
)
def set_autolock_delay(client: Client, delay):
with client:
client.use_pin_sequence([PIN4])
client.set_expected_responses(
[
pin_request(client),
messages.ButtonRequest,
messages.Success,
messages.Features,
]
)
device.apply_settings(client, auto_lock_delay_ms=delay)
def test_apply_auto_lock_delay(client: Client):
set_autolock_delay(client, 10 * 1000)
time.sleep(0.1) # sleep less than auto-lock delay
with client:
# No PIN protection is required.
client.set_expected_responses([messages.Address])
get_test_address(client)
time.sleep(10.5) # sleep more than auto-lock delay
with client:
client.use_pin_sequence([PIN4])
client.set_expected_responses([pin_request(client), messages.Address])
get_test_address(client)
@pytest.mark.parametrize(
"seconds",
[
10, # 10 seconds, minimum
60, # 1 minute
123, # 2 minutes
3601, # 1 hour
7227, # 2 hours
536870, # 149 hours, maximum
],
)
def test_apply_auto_lock_delay_valid(client: Client, seconds):
set_autolock_delay(client, seconds * 1000)
assert client.features.auto_lock_delay_ms == seconds * 1000
def test_autolock_default_value(client: Client):
assert client.features.auto_lock_delay_ms is None
with client:
client.use_pin_sequence([PIN4])
device.apply_settings(client, label="pls unlock")
client.refresh_features()
assert client.features.auto_lock_delay_ms == 60 * 10 * 1000
@pytest.mark.parametrize(
"seconds",
[0, 1, 9, 536871, 2**22],
)
def test_apply_auto_lock_delay_out_of_range(client: Client, seconds):
with client:
client.use_pin_sequence([PIN4])
client.set_expected_responses(
[
pin_request(client),
messages.Failure(code=messages.FailureType.ProcessError),
]
)
delay = seconds * 1000
with pytest.raises(TrezorFailure):
device.apply_settings(client, auto_lock_delay_ms=delay)
@pytest.mark.models("core")
def test_autolock_cancels_ui(client: Client):
set_autolock_delay(client, 10 * 1000)
resp = client.call_raw(
messages.GetAddress(
coin_name="Testnet",
address_n=TEST_ADDRESS_N,
show_display=True,
script_type=messages.InputScriptType.SPENDADDRESS,
)
)
assert isinstance(resp, messages.ButtonRequest)
# send an ack, do not read response
client._raw_write(messages.ButtonAck())
# sleep more than auto-lock delay
time.sleep(10.5)
resp = client._raw_read()
assert isinstance(resp, messages.Failure)
assert resp.code == messages.FailureType.ActionCancelled
def test_autolock_ignores_initialize(client: Client):
set_autolock_delay(client, 10 * 1000)
assert client.features.unlocked is True
start = time.monotonic()
while time.monotonic() - start < 11:
# init_device should always work even if locked
client.init_device()
time.sleep(0.1)
# after 11 seconds we are definitely locked
assert client.features.unlocked is False
def test_autolock_ignores_getaddress(client: Client):
set_autolock_delay(client, 10 * 1000)
assert client.features.unlocked is True
start = time.monotonic()
# let's continue for 8 seconds to give a little leeway to the slow CI
while time.monotonic() - start < 8:
get_test_address(client)
time.sleep(0.1)
# sleep 3 more seconds to wait for autolock
time.sleep(3)
# after 11 seconds we are definitely locked
client.refresh_features()
assert client.features.unlocked is False