mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-18 04:18:10 +00:00
250 lines
8.9 KiB
Python
250 lines
8.9 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 pytest
|
|
|
|
from trezorlib import btc, device, messages
|
|
from trezorlib.debuglink import TrezorClientDebugLink as Client
|
|
from trezorlib.exceptions import TrezorFailure
|
|
from trezorlib.tools import H_, parse_path
|
|
|
|
from ...tx_cache import TxCache
|
|
from ..signtx import request_finished, request_input, request_meta, request_output
|
|
|
|
B = messages.ButtonRequestType
|
|
TX_CACHE_MAINNET = TxCache("Bitcoin")
|
|
TX_CACHE_TESTNET = TxCache("Testnet")
|
|
TX_CACHE_BCASH = TxCache("Bcash")
|
|
|
|
TXHASH_8cc1f4 = bytes.fromhex(
|
|
"8cc1f4adf7224ce855cf535a5104594a0004cb3b640d6714fdb00b9128832dd5"
|
|
)
|
|
TXHASH_a5cd2a = bytes.fromhex(
|
|
"a5cd2a706d680587e572df16a8ce5233139a094ebbd148cc66a8004dcc88819c"
|
|
)
|
|
TXHASH_d5f65e = bytes.fromhex(
|
|
"d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882"
|
|
)
|
|
TXHASH_fa80a9 = bytes.fromhex(
|
|
"fa80a9949f1094119195064462f54d0e0eabd3139becd4514ae635b8c7fe3a46"
|
|
)
|
|
TXHASH_5dfd1b = bytes.fromhex(
|
|
"5dfd1b037633adc7f84a17b2df31c9994fe50b3ab3e246c44c4ceff3d326f62e"
|
|
)
|
|
|
|
|
|
# Adapted from TestMsgSigntx.test_one_one_fee,
|
|
# only changed the coin from Bitcoin to Litecoin.
|
|
# Litecoin does not have strong replay protection using SIGHASH_FORKID,
|
|
# spending from Bitcoin path should fail.
|
|
@pytest.mark.altcoin
|
|
def test_invalid_path_fail(client: Client):
|
|
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
|
|
# input 0: 0.0039 BTC
|
|
|
|
inp1 = messages.TxInputType(
|
|
address_n=parse_path("m/44h/0h/0h/0/0"),
|
|
amount=390_000,
|
|
prev_hash=TXHASH_d5f65e,
|
|
prev_index=0,
|
|
)
|
|
|
|
# address is converted from 1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1 by changing the version
|
|
out1 = messages.TxOutputType(
|
|
address="LfWz9wLHmqU9HoDkMg9NqbRosrHvEixeVZ",
|
|
amount=390_000 - 10_000,
|
|
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
|
)
|
|
|
|
with pytest.raises(TrezorFailure) as exc:
|
|
btc.sign_tx(client, "Litecoin", [inp1], [out1], prev_txes=TX_CACHE_MAINNET)
|
|
|
|
assert exc.value.code == messages.FailureType.DataError
|
|
assert exc.value.message.endswith("Forbidden key path")
|
|
|
|
|
|
# Adapted from TestMsgSigntx.test_one_one_fee,
|
|
# only changed the coin from Bitcoin to Litecoin and set safety checks to prompt.
|
|
# Litecoin does not have strong replay protection using SIGHASH_FORKID, but
|
|
# spending from Bitcoin path should pass with safety checks set to prompt.
|
|
@pytest.mark.altcoin
|
|
def test_invalid_path_prompt(client: Client):
|
|
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
|
|
# input 0: 0.0039 BTC
|
|
|
|
inp1 = messages.TxInputType(
|
|
address_n=parse_path("m/44h/0h/0h/0/0"),
|
|
amount=390_000,
|
|
prev_hash=TXHASH_d5f65e,
|
|
prev_index=0,
|
|
)
|
|
|
|
# address is converted from 1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1 by changing the version
|
|
out1 = messages.TxOutputType(
|
|
address="LfWz9wLHmqU9HoDkMg9NqbRosrHvEixeVZ",
|
|
amount=390_000 - 10_000,
|
|
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
|
)
|
|
|
|
device.apply_settings(
|
|
client, safety_checks=messages.SafetyCheckLevel.PromptTemporarily
|
|
)
|
|
|
|
btc.sign_tx(client, "Litecoin", [inp1], [out1], prev_txes=TX_CACHE_MAINNET)
|
|
|
|
|
|
# Adapted from TestMsgSigntx.test_one_one_fee,
|
|
# only changed the coin from Bitcoin to Bcash.
|
|
# Bcash does have strong replay protection using SIGHASH_FORKID,
|
|
# spending from Bitcoin path should work.
|
|
@pytest.mark.altcoin
|
|
def test_invalid_path_pass_forkid(client: Client):
|
|
# tx: 8cc1f4adf7224ce855cf535a5104594a0004cb3b640d6714fdb00b9128832dd5
|
|
# input 0: 0.0039 BTC
|
|
|
|
inp1 = messages.TxInputType(
|
|
address_n=parse_path("m/44h/0h/0h/0/0"),
|
|
amount=390_000,
|
|
prev_hash=TXHASH_8cc1f4,
|
|
prev_index=0,
|
|
)
|
|
|
|
# address is converted from 1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1 to cashaddr format
|
|
out1 = messages.TxOutputType(
|
|
address="bitcoincash:qr0fk25d5zygyn50u5w7h6jkvctas52n0qxff9ja6r",
|
|
amount=390_000 - 10_000,
|
|
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
|
)
|
|
|
|
btc.sign_tx(client, "Bcash", [inp1], [out1], prev_txes=TX_CACHE_BCASH)
|
|
|
|
|
|
def test_attack_path_segwit(client: Client):
|
|
# Scenario: The attacker falsely claims that the transaction uses Testnet paths to
|
|
# avoid the path warning dialog, but in step6_sign_segwit_inputs() uses Bitcoin paths
|
|
# to get a valid signature.
|
|
|
|
device.apply_settings(
|
|
client, safety_checks=messages.SafetyCheckLevel.PromptTemporarily
|
|
)
|
|
|
|
inp1 = messages.TxInputType(
|
|
# The actual input that the attcker wants to get signed.
|
|
address_n=parse_path("m/84h/0h/0h/0/0"),
|
|
amount=9_426,
|
|
prev_hash=TXHASH_fa80a9,
|
|
prev_index=0,
|
|
script_type=messages.InputScriptType.SPENDWITNESS,
|
|
)
|
|
inp2 = messages.TxInputType(
|
|
# The actual input that the attcker wants to get signed.
|
|
# We need this one to be from a different account, so that the match checker
|
|
# allows the transaction to pass.
|
|
address_n=parse_path("m/84h/0h/1h/0/1"),
|
|
amount=7_086,
|
|
prev_hash=TXHASH_5dfd1b,
|
|
prev_index=0,
|
|
script_type=messages.InputScriptType.SPENDWITNESS,
|
|
)
|
|
|
|
out1 = messages.TxOutputType(
|
|
# Attacker's Mainnet address encoded as Testnet.
|
|
address="tb1q694ccp5qcc0udmfwgp692u2s2hjpq5h407urtu",
|
|
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
|
amount=9_426 + 7_086 - 500,
|
|
)
|
|
|
|
attack_count = 6
|
|
|
|
def attack_processor(msg):
|
|
nonlocal attack_count
|
|
# Make the inputs look like they are coming from Testnet paths until we reach the
|
|
# signing phase.
|
|
if attack_count > 0 and msg.tx.inputs and msg.tx.inputs[0] in (inp1, inp2):
|
|
attack_count -= 1
|
|
msg.tx.inputs[0].address_n[1] = H_(1)
|
|
|
|
return msg
|
|
|
|
with client:
|
|
client.set_filter(messages.TxAck, attack_processor)
|
|
client.set_expected_responses(
|
|
[
|
|
# Step: process inputs
|
|
request_input(0),
|
|
# Attacker bypasses warning about non-standard path.
|
|
request_input(1),
|
|
# Attacker bypasses warning about non-standard path.
|
|
# Step: approve outputs
|
|
request_output(0),
|
|
messages.ButtonRequest(code=B.ConfirmOutput),
|
|
messages.ButtonRequest(code=B.SignTx),
|
|
# Step: verify inputs
|
|
request_input(0),
|
|
request_meta(TXHASH_fa80a9),
|
|
request_input(0, TXHASH_fa80a9),
|
|
request_output(0, TXHASH_fa80a9),
|
|
request_output(1, TXHASH_fa80a9),
|
|
request_input(1),
|
|
request_meta(TXHASH_5dfd1b),
|
|
request_input(0, TXHASH_5dfd1b),
|
|
request_output(0, TXHASH_5dfd1b),
|
|
request_output(1, TXHASH_5dfd1b),
|
|
# Step: serialize inputs
|
|
request_input(0),
|
|
request_input(1),
|
|
# Step: serialize outputs
|
|
request_output(0),
|
|
# Step: sign segwit inputs
|
|
request_input(0),
|
|
# Trezor must warn about non-standard path before signing.
|
|
messages.ButtonRequest(code=B.UnknownDerivationPath),
|
|
request_input(1),
|
|
# Trezor must warn about non-standard path before signing.
|
|
messages.ButtonRequest(code=B.UnknownDerivationPath),
|
|
request_finished(),
|
|
]
|
|
)
|
|
|
|
btc.sign_tx(client, "Testnet", [inp1, inp2], [out1], prev_txes=TX_CACHE_MAINNET)
|
|
|
|
|
|
@pytest.mark.skip_t1(reason="T1 only prevents using paths known to be altcoins")
|
|
def test_invalid_path_fail_asap(client: Client):
|
|
inp1 = messages.TxInputType(
|
|
address_n=parse_path("m/0"),
|
|
amount=4_977_040,
|
|
prev_hash=TXHASH_a5cd2a,
|
|
prev_index=0,
|
|
script_type=messages.InputScriptType.SPENDWITNESS,
|
|
sequence=4_294_967_293,
|
|
)
|
|
|
|
out1 = messages.TxOutputType(
|
|
address_n=parse_path("m/84h/0h/0h/1/0"),
|
|
amount=4_977_040,
|
|
script_type=messages.OutputScriptType.PAYTOWITNESS,
|
|
)
|
|
|
|
with client:
|
|
client.set_expected_responses(
|
|
[request_input(0), messages.Failure(code=messages.FailureType.DataError)]
|
|
)
|
|
try:
|
|
btc.sign_tx(client, "Testnet", [inp1], [out1], prev_txes=TX_CACHE_TESTNET)
|
|
except TrezorFailure:
|
|
pass
|