# 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 from trezorlib import btc, device, messages, misc from trezorlib.exceptions import TrezorFailure from ..common import MNEMONIC12 from ..tx_cache import TxCache from .signtx import request_finished, request_input, request_meta, request_output B = messages.ButtonRequestType TXHASH_d5f65e = bytes.fromhex( "d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882" ) PIN4 = "1234" @pytest.mark.skip_t2 class TestProtectionLevels: @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_initialize(self, client): with client: client.set_expected_responses([messages.Features]) client.init_device() @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_apply_settings(self, client): with client: client.use_pin_sequence([PIN4]) client.set_expected_responses( [ messages.PinMatrixRequest, messages.ButtonRequest, messages.Success, messages.Features, ] ) # TrezorClient reinitializes device device.apply_settings(client, label="nazdar") @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_change_pin(self, client): with client: client.use_pin_sequence([PIN4, PIN4, PIN4]) client.set_expected_responses( [ messages.ButtonRequest, messages.PinMatrixRequest, messages.PinMatrixRequest, messages.PinMatrixRequest, messages.Success, messages.Features, ] ) device.change_pin(client) def test_ping(self, client): with client: client.set_expected_responses([messages.ButtonRequest, messages.Success]) client.ping("msg", True) @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_get_entropy(self, client): with client: client.set_expected_responses([messages.ButtonRequest, messages.Entropy]) misc.get_entropy(client, 10) @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_get_public_key(self, client): with client: client.use_pin_sequence([PIN4]) client.set_expected_responses( [ messages.PinMatrixRequest, messages.PassphraseRequest, messages.PublicKey, ] ) btc.get_public_node(client, []) @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_get_address(self, client): with client: client.use_pin_sequence([PIN4]) client.set_expected_responses( [ messages.PinMatrixRequest, messages.PassphraseRequest, messages.Address, ] ) btc.get_address(client, "Bitcoin", []) @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_wipe_device(self, client): with client: client.set_expected_responses( [messages.ButtonRequest, messages.Success, messages.Features] ) device.wipe(client) @pytest.mark.setup_client(uninitialized=True) def test_reset_device(self, client): with client: client.set_expected_responses( [messages.ButtonRequest] + [messages.EntropyRequest] + [messages.ButtonRequest] * 24 + [messages.Success, messages.Features] ) device.reset(client, False, 128, True, False, "label", "en-US") with pytest.raises(TrezorFailure): # This must fail, because device is already initialized # Using direct call because `device.reset` has its own check client.call( messages.ResetDevice( display_random=False, strength=128, passphrase_protection=True, pin_protection=False, label="label", language="en-US", ) ) @pytest.mark.setup_client(uninitialized=True) def test_recovery_device(self, client): client.use_mnemonic(MNEMONIC12) with client: client.set_expected_responses( [messages.ButtonRequest] + [messages.WordRequest] * 24 + [messages.Success, messages.Features] ) device.recover( client, 12, False, False, "label", "en-US", client.mnemonic_callback ) with pytest.raises(TrezorFailure): # This must fail, because device is already initialized # Using direct call because `device.reset` has its own check client.call( messages.RecoveryDevice( word_count=12, passphrase_protection=False, pin_protection=False, label="label", language="en-US", ) ) @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_sign_message(self, client): with client: client.use_pin_sequence([PIN4]) client.set_expected_responses( [ messages.ButtonRequest, messages.PinMatrixRequest, messages.PassphraseRequest, messages.MessageSignature, ] ) btc.sign_message(client, "Bitcoin", [], "testing message") @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_verify_message(self, client): with client: client.set_expected_responses( [messages.ButtonRequest, messages.ButtonRequest, messages.Success] ) btc.verify_message( client, "Bitcoin", "14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e", bytes.fromhex( "209e23edf0e4e47ff1dec27f32cd78c50e74ef018ee8a6adf35ae17c7a9b0dd96f48b493fd7dbab03efb6f439c6383c9523b3bbc5f1a7d158a6af90ab154e9be80" ), "This is an example of a signed message.", ) @pytest.mark.setup_client(pin=PIN4, passphrase=True) def test_signtx(self, client): inp1 = messages.TxInputType( address_n=[0], # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e prev_hash=TXHASH_d5f65e, prev_index=0, amount=390000, ) out1 = messages.TxOutputType( address="1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1", amount=390000 - 10000, script_type=messages.OutputScriptType.PAYTOADDRESS, ) with client: client.use_pin_sequence([PIN4]) client.set_expected_responses( [ messages.PinMatrixRequest, messages.PassphraseRequest, request_input(0), request_output(0), messages.ButtonRequest(code=B.ConfirmOutput), messages.ButtonRequest(code=B.SignTx), request_input(0), request_meta(TXHASH_d5f65e), request_input(0, TXHASH_d5f65e), request_input(1, TXHASH_d5f65e), request_output(0, TXHASH_d5f65e), request_input(0), request_output(0), request_output(0), request_finished(), ] ) btc.sign_tx(client, "Bitcoin", [inp1], [out1], prev_txes=TxCache("Bitcoin")) # def test_firmware_erase(self): # pass # def test_firmware_upload(self): # pass @pytest.mark.setup_client(pin=PIN4) def test_unlocked(self, client): assert client.features.unlocked is False with client: client.use_pin_sequence([PIN4]) client.set_expected_responses([messages.PinMatrixRequest, messages.Address]) btc.get_address(client, "Testnet", [0]) client.init_device() assert client.features.unlocked is True with client: client.set_expected_responses([messages.Address]) btc.get_address(client, "Testnet", [0]) @pytest.mark.setup_client(passphrase=True) def test_passphrase_cached(self, client): with client: client.set_expected_responses( [messages.PassphraseRequest, messages.Address] ) btc.get_address(client, "Testnet", [0]) with client: client.set_expected_responses([messages.Address]) btc.get_address(client, "Testnet", [0])