1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-24 08:28:12 +00:00
trezor-firmware/python/tests/test_btc.py
2023-08-16 13:29:51 +02:00

237 lines
10 KiB
Python

# This file is part of the Trezor project.
#
# Copyright (C) 2012-2022 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 json
from decimal import Decimal
from trezorlib import btc
# https://btc1.trezor.io/api/tx-specific/f5e735549daeb480d4348f2574b8967a4f149715edb220a742d8bb654d668348
TX_JSON_BIG = """
{
"txid": "f5e735549daeb480d4348f2574b8967a4f149715edb220a742d8bb654d668348",
"hash": "6cc9b72db1440ce2de78a46c1fe6f8979807eddb14ab6e532947c83937520569",
"version": 2,
"size": 931,
"vsize": 529,
"weight": 2113,
"locktime": 620109,
"vin": [
{
"txid": "a1291df0b8ef7cce44d5ae7f31e50ef85aedc129cfcb5e1a82ab5ac917c39733",
"vout": 4,
"scriptSig": {
"asm": "001476e4db8a1d5c4c238775df63899f71dfda0197f4",
"hex": "16001476e4db8a1d5c4c238775df63899f71dfda0197f4"
},
"txinwitness": [
"30440220395aeb2327c8860f42c671dec6807c962f73a67db8f7106691bc1d02095f075e0220471812cf1830b769246ddbeb9d99d034b2119415291fa6fe8fb33d6f39858e4f01",
"030c9daf8f58ccd1733de05574964eaf2810c5d6e2375dee0f49603151cf589e1d"
],
"sequence": 4294967294
},
{
"txid": "05a049179ee576e3c3d6ae71011e08066aa6a588af31d6d61332215c65e8eebd",
"vout": 3,
"scriptSig": {
"asm": "0014004bebeb0b79c94754f5458a3ed8a293370df7fb",
"hex": "160014004bebeb0b79c94754f5458a3ed8a293370df7fb"
},
"txinwitness": [
"3044022029283ef96d3def47843716e64606604044125c1642732102533076c9b1958ce9022071d0fb9504fe47b215ebc320113ec77936f143659b9fcdf4f5c9e872fdfec9b601",
"025389a45a00c6d57e6a61258e95d6f7413f2c23274f7f82a7164ec040f7bf73ab"
],
"sequence": 4294967294
},
{
"txid": "05a049179ee576e3c3d6ae71011e08066aa6a588af31d6d61332215c65e8eebd",
"vout": 7,
"scriptSig": {
"asm": "0014d07cba9645f2814af5ed30aef767234ddda9ac13",
"hex": "160014d07cba9645f2814af5ed30aef767234ddda9ac13"
},
"txinwitness": [
"3044022068dc9e81b98036014adfc6d6235a66caaf12a9060243fb768abd161024275443022075e89fa03d6a5f5d21ce778e97de521c5bb51e6eaa0f9eccd92f487d22410ed801",
"03d79e1063b8b5bd169ac5b2791b57e19f43160c7e553baa0b243cb666d30c19f4"
],
"sequence": 4294967294
},
{
"txid": "05a049179ee576e3c3d6ae71011e08066aa6a588af31d6d61332215c65e8eebd",
"vout": 8,
"scriptSig": {
"asm": "0014ae471e13840e2b5dcdec9c4a9b54ba11fe7b665c",
"hex": "160014ae471e13840e2b5dcdec9c4a9b54ba11fe7b665c"
},
"txinwitness": [
"304402206d2ddc00b36896e6cbd50dfae25e8ae81f0fbf902325f34a2c831a04503cd58202202fe96b43daa1281c96a5c9507a01956cfb0d4b94b5406a94d6de96c3b82bdf7d01",
"03a261fa4d379512d74d8eb14a7f797f694bbbda5ff3dcc05a08a55be7dd2b2e00"
],
"sequence": 4294967294
},
{
"txid": "dc27d76fa566e14e97709c59e0b1c5e1b7628c96f6dc8cbca5816a381a4234cb",
"vout": 14,
"scriptSig": {
"asm": "0014995cd732c74a3446f1ad35bafbd8d6f79828ef04",
"hex": "160014995cd732c74a3446f1ad35bafbd8d6f79828ef04"
},
"txinwitness": [
"304402202659160053eee86b6f27be4e82cb7f390597e923ae0f9d17b8e3864f6ae17a2602200468b59da0b6c1b1edd848dbbba802be74228b43dda9de5e9ad36b1e6558950101",
"0393c79d23ae89f461f039b895dfd6f365ef5cc89f1a3ef030382f80c2cbd84caa"
],
"sequence": 4294967294
}
],
"vout": [
{
"value": 0.42651353,
"n": 0,
"scriptPubKey": {
"asm": "OP_HASH160 3a530afe02afe8dd7bc1b7d731550eec4f442666 OP_EQUAL",
"hex": "a9143a530afe02afe8dd7bc1b7d731550eec4f44266687",
"reqSigs": 1,
"type": "scripthash",
"addresses": [
"371QbbRvKoqgSxfUm9Ly6qHoBJzuaBHDCT"
]
}
},
{
"value": 22.63125,
"n": 1,
"scriptPubKey": {
"asm": "OP_HASH160 01a331e12b1d6789c8109d86f27e6fdd6105b194 OP_EQUAL",
"hex": "a91401a331e12b1d6789c8109d86f27e6fdd6105b19487",
"reqSigs": 1,
"type": "scripthash",
"addresses": [
"31qg6iFQUxSdxyTk1RJyKmVPNvz7XV5s5c"
]
}
}
],
"hex": "020000000001053397c317c95aab821a5ecbcf29c1ed5af80ee5317faed544ce7cefb8f01d29a1040000001716001476e4db8a1d5c4c238775df63899f71dfda0197f4feffffffbdeee8655c213213d6d631af88a5a66a06081e0171aed6c3e376e59e1749a0050300000017160014004bebeb0b79c94754f5458a3ed8a293370df7fbfeffffffbdeee8655c213213d6d631af88a5a66a06081e0171aed6c3e376e59e1749a0050700000017160014d07cba9645f2814af5ed30aef767234ddda9ac13feffffffbdeee8655c213213d6d631af88a5a66a06081e0171aed6c3e376e59e1749a0050800000017160014ae471e13840e2b5dcdec9c4a9b54ba11fe7b665cfeffffffcb34421a386a81a5bc8cdcf6968c62b7e1c5b1e0599c70974ee166a56fd727dc0e00000017160014995cd732c74a3446f1ad35bafbd8d6f79828ef04feffffff02d9ce8a020000000017a9143a530afe02afe8dd7bc1b7d731550eec4f44266687088ce4860000000017a91401a331e12b1d6789c8109d86f27e6fdd6105b19487024730440220395aeb2327c8860f42c671dec6807c962f73a67db8f7106691bc1d02095f075e0220471812cf1830b769246ddbeb9d99d034b2119415291fa6fe8fb33d6f39858e4f0121030c9daf8f58ccd1733de05574964eaf2810c5d6e2375dee0f49603151cf589e1d02473044022029283ef96d3def47843716e64606604044125c1642732102533076c9b1958ce9022071d0fb9504fe47b215ebc320113ec77936f143659b9fcdf4f5c9e872fdfec9b60121025389a45a00c6d57e6a61258e95d6f7413f2c23274f7f82a7164ec040f7bf73ab02473044022068dc9e81b98036014adfc6d6235a66caaf12a9060243fb768abd161024275443022075e89fa03d6a5f5d21ce778e97de521c5bb51e6eaa0f9eccd92f487d22410ed8012103d79e1063b8b5bd169ac5b2791b57e19f43160c7e553baa0b243cb666d30c19f40247304402206d2ddc00b36896e6cbd50dfae25e8ae81f0fbf902325f34a2c831a04503cd58202202fe96b43daa1281c96a5c9507a01956cfb0d4b94b5406a94d6de96c3b82bdf7d012103a261fa4d379512d74d8eb14a7f797f694bbbda5ff3dcc05a08a55be7dd2b2e000247304402202659160053eee86b6f27be4e82cb7f390597e923ae0f9d17b8e3864f6ae17a2602200468b59da0b6c1b1edd848dbbba802be74228b43dda9de5e9ad36b1e6558950101210393c79d23ae89f461f039b895dfd6f365ef5cc89f1a3ef030382f80c2cbd84caa4d760900",
"blockhash": "000000000000000000031d097eeb2fe33a4c6fbf2dcebcdbd49051a4d7e37390",
"confirmations": 13,
"time": 1583317141,
"blocktime": 1583317141
}
"""
# https://btc1.trezor.io/api/tx-specific/317f8a6e343384bd7d1a06ca25407d991ad3fc956e4ebedc66e1ec3b2ed9ccc6
TX_JSON_COINBASE = """
{
"txid": "317f8a6e343384bd7d1a06ca25407d991ad3fc956e4ebedc66e1ec3b2ed9ccc6",
"hash": "a55b6b60791108e5bd6b014a3527be285867744d58baba4ca7bcc9ea713533a3",
"version": 2,
"size": 342,
"vsize": 315,
"weight": 1260,
"locktime": 0,
"vin": [
{
"coinbase": "03737609040a995f5e626a30332f4254432e434f4d2ffabe6d6db1d4bc579e23d79b7a494282f8189a5e99567ce755a1dd25e1964fcda592ac21080000007296cd102f0461a3aee6070000000000",
"sequence": 4294967295
}
],
"vout": [
{
"value": 12.56294353,
"n": 0,
"scriptPubKey": {
"asm": "0 97cfc76442fe717f2a3f0cc9c175f7561b661997",
"hex": "001497cfc76442fe717f2a3f0cc9c175f7561b661997",
"reqSigs": 1,
"type": "witness_v0_keyhash",
"addresses": [
"bc1qjl8uwezzlech723lpnyuza0h2cdkvxvh54v3dn"
]
}
},
{
"value": 0,
"n": 1,
"scriptPubKey": {
"asm": "OP_RETURN aa21a9edcc94907af04544c15ac0bd15520012b98edbb04c604b20f43306382831850da9",
"hex": "6a24aa21a9edcc94907af04544c15ac0bd15520012b98edbb04c604b20f43306382831850da9",
"type": "nulldata"
}
},
{
"value": 0,
"n": 2,
"scriptPubKey": {
"asm": "OP_RETURN 52534b424c4f434b3a738dd5336a7d4869904cb27f7d5b5afd534c360b6d0b9c45e2fafc2a00210406",
"hex": "6a2952534b424c4f434b3a738dd5336a7d4869904cb27f7d5b5afd534c360b6d0b9c45e2fafc2a00210406",
"type": "nulldata"
}
},
{
"value": 0,
"n": 3,
"scriptPubKey": {
"asm": "OP_RETURN b9e11b6df9e2d9e640ea4dc449b522003347a2c0aa421b4128f4644560c508e78a433684",
"hex": "6a24b9e11b6df9e2d9e640ea4dc449b522003347a2c0aa421b4128f4644560c508e78a433684",
"type": "nulldata"
}
}
],
"hex": "020000000001010000000000000000000000000000000000000000000000000000000000000000ffffffff4e03737609040a995f5e626a30332f4254432e434f4d2ffabe6d6db1d4bc579e23d79b7a494282f8189a5e99567ce755a1dd25e1964fcda592ac21080000007296cd102f0461a3aee6070000000000ffffffff04d187e14a0000000016001497cfc76442fe717f2a3f0cc9c175f7561b6619970000000000000000266a24aa21a9edcc94907af04544c15ac0bd15520012b98edbb04c604b20f43306382831850da900000000000000002b6a2952534b424c4f434b3a738dd5336a7d4869904cb27f7d5b5afd534c360b6d0b9c45e2fafc2a002104060000000000000000266a24b9e11b6df9e2d9e640ea4dc449b522003347a2c0aa421b4128f4644560c508e78a4336840120000000000000000000000000000000000000000000000000000000000000000000000000",
"blockhash": "00000000000000000008f5eb1ce2bac08ecd50041fa162e45470dda519fc159d",
"confirmations": 1,
"time": 1583323402,
"blocktime": 1583323402
}
"""
def test_from_json():
tx_dict = json.loads(TX_JSON_BIG, parse_float=Decimal)
tx = btc.from_json(tx_dict)
assert tx.version == 2
assert tx.lock_time == 620109
assert len(tx.inputs) == 5
assert len(tx.bin_outputs) == 2
assert sum(o.amount for o in tx.bin_outputs) == 2305776353
for v, i in zip(tx_dict["vin"], tx.inputs):
assert i.prev_hash.hex() == v["txid"]
assert i.prev_index == v["vout"]
assert i.script_sig.hex() == v["scriptSig"]["hex"]
assert i.sequence == v["sequence"]
for v, o in zip(tx_dict["vout"], tx.bin_outputs):
assert o.amount == int(Decimal(v["value"]) * (10**8))
assert o.script_pubkey.hex() == v["scriptPubKey"]["hex"]
def test_coinbase_from_json():
tx_dict = json.loads(TX_JSON_COINBASE, parse_float=Decimal)
tx = btc.from_json(tx_dict)
assert tx.version == 2
assert tx.lock_time == 0
assert len(tx.inputs) == 1
assert len(tx.bin_outputs) == 4
assert sum(o.amount for o in tx.bin_outputs) == 1256294353
coinbase = tx.inputs[0]
assert coinbase.prev_hash == b"\x00" * 32
assert coinbase.prev_index == 2**32 - 1
assert coinbase.script_sig.hex() == tx_dict["vin"][0]["coinbase"]