mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-10-20 14:59:06 +00:00
3ab036a1d1
Add additional device tests for GRS Add P2TR GetAddress for GRS mainnet and testnet Add SignTX spending taproot UTXO (The resulting transaction from the SignTx device test was successfully mined on Testnet.)
301 lines
9.3 KiB
Python
301 lines
9.3 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.exceptions import TrezorFailure
|
|
from trezorlib.messages import SafetyCheckLevel
|
|
from trezorlib.tools import parse_path
|
|
|
|
from ... import bip32
|
|
|
|
|
|
def getmultisig(chain, nr, xpubs):
|
|
return messages.MultisigRedeemScriptType(
|
|
nodes=[bip32.deserialize(xpub) for xpub in xpubs],
|
|
address_n=[chain, nr],
|
|
signatures=[b"", b"", b""],
|
|
m=2,
|
|
)
|
|
|
|
|
|
def test_btc(client):
|
|
assert (
|
|
btc.get_address(client, "Bitcoin", parse_path("m/44'/0'/0'/0/0"))
|
|
== "1JAd7XCBzGudGpJQSDSfpmJhiygtLQWaGL"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Bitcoin", parse_path("m/44'/0'/0'/0/1"))
|
|
== "1GWFxtwWmNVqotUPXLcKVL2mUKpshuJYo"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Bitcoin", parse_path("m/44'/0'/0'/1/0"))
|
|
== "1DyHzbQUoQEsLxJn6M7fMD8Xdt1XvNiwNE"
|
|
)
|
|
|
|
|
|
@pytest.mark.altcoin
|
|
def test_ltc(client):
|
|
assert (
|
|
btc.get_address(client, "Litecoin", parse_path("m/44'/2'/0'/0/0"))
|
|
== "LcubERmHD31PWup1fbozpKuiqjHZ4anxcL"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Litecoin", parse_path("m/44'/2'/0'/0/1"))
|
|
== "LVWBmHBkCGNjSPHucvL2PmnuRAJnucmRE6"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Litecoin", parse_path("m/44'/2'/0'/1/0"))
|
|
== "LWj6ApswZxay4cJEJES2sGe7fLMLRvvv8h"
|
|
)
|
|
|
|
|
|
def test_tbtc(client):
|
|
assert (
|
|
btc.get_address(client, "Testnet", parse_path("m/44'/1'/0'/0/0"))
|
|
== "mvbu1Gdy8SUjTenqerxUaZyYjmveZvt33q"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Testnet", parse_path("m/44'/1'/0'/0/1"))
|
|
== "mopZWqZZyQc3F2Sy33cvDtJchSAMsnLi7b"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Testnet", parse_path("m/44'/1'/0'/1/0"))
|
|
== "mm6kLYbGEL1tGe4ZA8xacfgRPdW1NLjCbZ"
|
|
)
|
|
|
|
|
|
@pytest.mark.altcoin
|
|
def test_bch(client):
|
|
assert (
|
|
btc.get_address(client, "Bcash", parse_path("44'/145'/0'/0/0"))
|
|
== "bitcoincash:qr08q88p9etk89wgv05nwlrkm4l0urz4cyl36hh9sv"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Bcash", parse_path("44'/145'/0'/0/1"))
|
|
== "bitcoincash:qr23ajjfd9wd73l87j642puf8cad20lfmqdgwvpat4"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Bcash", parse_path("44'/145'/0'/1/0"))
|
|
== "bitcoincash:qzc5q87w069lzg7g3gzx0c8dz83mn7l02scej5aluw"
|
|
)
|
|
|
|
|
|
@pytest.mark.altcoin
|
|
def test_grs(client):
|
|
assert (
|
|
btc.get_address(client, "Groestlcoin", parse_path("44'/17'/0'/0/0"))
|
|
== "Fj62rBJi8LvbmWu2jzkaUX1NFXLEqDLoZM"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Groestlcoin", parse_path("44'/17'/0'/1/0"))
|
|
== "FmRaqvVBRrAp2Umfqx9V1ectZy8gw54QDN"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Groestlcoin", parse_path("44'/17'/0'/1/1"))
|
|
== "Fmhtxeh7YdCBkyQF7AQG4QnY8y3rJg89di"
|
|
)
|
|
|
|
|
|
@pytest.mark.altcoin
|
|
def test_tgrs(client):
|
|
assert (
|
|
btc.get_address(client, "Groestlcoin Testnet", parse_path("44'/1'/0'/0/0"))
|
|
== "mvbu1Gdy8SUjTenqerxUaZyYjmvedc787y"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Groestlcoin Testnet", parse_path("44'/1'/0'/1/0"))
|
|
== "mm6kLYbGEL1tGe4ZA8xacfgRPdW1LMq8cN"
|
|
)
|
|
assert (
|
|
btc.get_address(client, "Groestlcoin Testnet", parse_path("44'/1'/0'/1/1"))
|
|
== "mjXZwmEi1z1MzveZrKUAo4DBgbdq6ZhGD6"
|
|
)
|
|
|
|
|
|
@pytest.mark.altcoin
|
|
def test_elements(client):
|
|
assert (
|
|
btc.get_address(client, "Elements", parse_path("m/44'/1'/0'/0/0"))
|
|
== "2dpWh6jbhAowNsQ5agtFzi7j6nKscj6UnEr"
|
|
)
|
|
|
|
|
|
@pytest.mark.multisig
|
|
def test_multisig(client):
|
|
xpubs = []
|
|
for n in range(1, 4):
|
|
node = btc.get_public_node(client, parse_path(f"44'/0'/{n}'"))
|
|
xpubs.append(node.xpub)
|
|
|
|
for nr in range(1, 4):
|
|
assert (
|
|
btc.get_address(
|
|
client,
|
|
"Bitcoin",
|
|
parse_path(f"44'/0'/{nr}'/0/0"),
|
|
show_display=(nr == 1),
|
|
multisig=getmultisig(0, 0, xpubs=xpubs),
|
|
)
|
|
== "3Pdz86KtfJBuHLcSv4DysJo4aQfanTqCzG"
|
|
)
|
|
assert (
|
|
btc.get_address(
|
|
client,
|
|
"Bitcoin",
|
|
parse_path(f"44'/0'/{nr}'/1/0"),
|
|
show_display=(nr == 1),
|
|
multisig=getmultisig(1, 0, xpubs=xpubs),
|
|
)
|
|
== "36gP3KVx1ooStZ9quZDXbAF3GCr42b2zzd"
|
|
)
|
|
|
|
|
|
@pytest.mark.multisig
|
|
@pytest.mark.parametrize("show_display", (True, False))
|
|
def test_multisig_missing(client, show_display):
|
|
# Multisig with global suffix specification.
|
|
# Use account numbers 1, 2 and 3 to create a valid multisig,
|
|
# but not containing the keys from account 0 used below.
|
|
nodes = [
|
|
btc.get_public_node(client, parse_path(f"44'/0'/{i}'")).node
|
|
for i in range(1, 4)
|
|
]
|
|
multisig1 = messages.MultisigRedeemScriptType(
|
|
nodes=nodes, address_n=[0, 0], signatures=[b"", b"", b""], m=2
|
|
)
|
|
|
|
# Multisig with per-node suffix specification.
|
|
node = btc.get_public_node(
|
|
client, parse_path("44h/0h/0h/0"), coin_name="Bitcoin"
|
|
).node
|
|
|
|
multisig2 = messages.MultisigRedeemScriptType(
|
|
pubkeys=[
|
|
messages.HDNodePathType(node=node, address_n=[1]),
|
|
messages.HDNodePathType(node=node, address_n=[2]),
|
|
messages.HDNodePathType(node=node, address_n=[3]),
|
|
],
|
|
signatures=[b"", b"", b""],
|
|
m=2,
|
|
)
|
|
|
|
for multisig in (multisig1, multisig2):
|
|
with pytest.raises(TrezorFailure):
|
|
btc.get_address(
|
|
client,
|
|
"Bitcoin",
|
|
parse_path("44'/0'/0'/0/0"),
|
|
show_display=show_display,
|
|
multisig=multisig,
|
|
)
|
|
|
|
|
|
@pytest.mark.altcoin
|
|
@pytest.mark.multisig
|
|
def test_bch_multisig(client):
|
|
xpubs = []
|
|
for n in range(1, 4):
|
|
node = btc.get_public_node(
|
|
client, parse_path(f"44'/145'/{n}'"), coin_name="Bcash"
|
|
)
|
|
xpubs.append(node.xpub)
|
|
|
|
for nr in range(1, 4):
|
|
assert (
|
|
btc.get_address(
|
|
client,
|
|
"Bcash",
|
|
parse_path(f"44'/145'/{nr}'/0/0"),
|
|
show_display=(nr == 1),
|
|
multisig=getmultisig(0, 0, xpubs=xpubs),
|
|
)
|
|
== "bitcoincash:pqguz4nqq64jhr5v3kvpq4dsjrkda75hwy86gq0qzw"
|
|
)
|
|
assert (
|
|
btc.get_address(
|
|
client,
|
|
"Bcash",
|
|
parse_path(f"44'/145'/{nr}'/1/0"),
|
|
show_display=(nr == 1),
|
|
multisig=getmultisig(1, 0, xpubs=xpubs),
|
|
)
|
|
== "bitcoincash:pp6kcpkhua7789g2vyj0qfkcux3yvje7euhyhltn0a"
|
|
)
|
|
|
|
|
|
def test_public_ckd(client):
|
|
node = btc.get_public_node(client, parse_path("m/44'/0'/0'")).node
|
|
node_sub1 = btc.get_public_node(client, parse_path("m/44'/0'/0'/1/0")).node
|
|
node_sub2 = bip32.public_ckd(node, [1, 0])
|
|
|
|
assert node_sub1.chain_code == node_sub2.chain_code
|
|
assert node_sub1.public_key == node_sub2.public_key
|
|
|
|
address1 = btc.get_address(client, "Bitcoin", parse_path("m/44'/0'/0'/1/0"))
|
|
address2 = bip32.get_address(node_sub2, 0)
|
|
|
|
assert address2 == "1DyHzbQUoQEsLxJn6M7fMD8Xdt1XvNiwNE"
|
|
assert address1 == address2
|
|
|
|
|
|
def test_invalid_path(client):
|
|
with pytest.raises(TrezorFailure, match="Forbidden key path"):
|
|
# slip44 id mismatch
|
|
btc.get_address(
|
|
client, "Bitcoin", parse_path("m/44'/111'/0'/0/0"), show_display=True
|
|
)
|
|
|
|
|
|
def test_unknown_path(client):
|
|
UNKNOWN_PATH = parse_path("m/44'/9'/0'/0/0")
|
|
with client:
|
|
client.set_expected_responses([messages.Failure])
|
|
|
|
with pytest.raises(TrezorFailure, match="Forbidden key path"):
|
|
# account number is too high
|
|
btc.get_address(client, "Bitcoin", UNKNOWN_PATH, show_display=True)
|
|
|
|
# disable safety checks
|
|
device.apply_settings(client, safety_checks=SafetyCheckLevel.PromptTemporarily)
|
|
|
|
with client:
|
|
client.set_expected_responses(
|
|
[
|
|
messages.ButtonRequest(
|
|
code=messages.ButtonRequestType.UnknownDerivationPath
|
|
),
|
|
messages.ButtonRequest(code=messages.ButtonRequestType.Address),
|
|
messages.Address,
|
|
]
|
|
)
|
|
# try again with a warning
|
|
btc.get_address(client, "Bitcoin", UNKNOWN_PATH, show_display=True)
|
|
|
|
with client:
|
|
# no warning is displayed when the call is silent
|
|
client.set_expected_responses([messages.Address])
|
|
btc.get_address(client, "Bitcoin", UNKNOWN_PATH, show_display=False)
|
|
|
|
|
|
@pytest.mark.altcoin
|
|
def test_crw(client):
|
|
assert (
|
|
btc.get_address(client, "Crown", parse_path("44'/72'/0'/0/0"))
|
|
== "CRWYdvZM1yXMKQxeN3hRsAbwa7drfvTwys48"
|
|
)
|