mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-21 23:18:13 +00:00
python: move json_to_tx to btc.from_json, add test cases
This commit is contained in:
matejcik
parent
b3569a53e6
commit
5a526888cf
@ -14,10 +14,43 @@
|
||||
# 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>.
|
||||
|
||||
from decimal import Decimal
|
||||
|
||||
from . import messages
|
||||
from .tools import CallException, expect, normalize_nfc, session
|
||||
|
||||
|
||||
def from_json(json_dict):
|
||||
def make_input(vin):
|
||||
i = messages.TxInputType()
|
||||
if "coinbase" in vin:
|
||||
i.prev_hash = b"\0" * 32
|
||||
i.prev_index = 0xFFFFFFFF # signed int -1
|
||||
i.script_sig = bytes.fromhex(vin["coinbase"])
|
||||
i.sequence = vin["sequence"]
|
||||
|
||||
else:
|
||||
i.prev_hash = bytes.fromhex(vin["txid"])
|
||||
i.prev_index = vin["vout"]
|
||||
i.script_sig = bytes.fromhex(vin["scriptSig"]["hex"])
|
||||
i.sequence = vin["sequence"]
|
||||
|
||||
return i
|
||||
|
||||
def make_bin_output(vout):
|
||||
o = messages.TxOutputBinType()
|
||||
o.amount = int(Decimal(vout["value"]) * (10 ** 8))
|
||||
o.script_pubkey = bytes.fromhex(vout["scriptPubKey"]["hex"])
|
||||
return o
|
||||
|
||||
t = messages.TransactionType()
|
||||
t.version = json_dict["version"]
|
||||
t.lock_time = json_dict.get("locktime")
|
||||
t.inputs = [make_input(vin) for vin in json_dict["vin"]]
|
||||
t.bin_outputs = [make_bin_output(vout) for vout in json_dict["vout"]]
|
||||
return t
|
||||
|
||||
|
||||
@expect(messages.PublicKey)
|
||||
def get_public_node(
|
||||
client,
|
||||
|
||||
237
python/tests/test_btc.py
Normal file
237
python/tests/test_btc.py
Normal file
@ -0,0 +1,237 @@
|
||||
# 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 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"]
|
||||
@ -23,8 +23,7 @@ from pathlib import Path
|
||||
import click
|
||||
import requests
|
||||
|
||||
from trezorlib import messages, protobuf
|
||||
from trezorlib.tx_api import json_to_tx
|
||||
from trezorlib import btc, messages, protobuf
|
||||
|
||||
REPOSITORY_ROOT = Path(__file__).parent.parent
|
||||
TOOLS_PATH = REPOSITORY_ROOT / "common" / "tools"
|
||||
@ -45,7 +44,7 @@ def _get_blockbooks():
|
||||
if not coin.get("blockbook"):
|
||||
continue
|
||||
|
||||
res[coin["coin_name"].lower()] = coin["blockbook"][0] + "/api/tx/{}"
|
||||
res[coin["coin_name"].lower()] = coin["blockbook"][0] + "/api/tx-specific/{}"
|
||||
return res
|
||||
|
||||
|
||||
@ -86,7 +85,7 @@ def cli(tx, coin_name):
|
||||
Without URL, default blockbook server will be used:
|
||||
./tests/tx_cache.py bcash bc37c28dfb467d2ecb50261387bf752a3977d7e5337915071bb4151e6b711a78
|
||||
It is also possible to specify URL explicitly:
|
||||
./tests/tx_cache.py bcash https://bch1.trezor.io/api/tx/bc37c28dfb467d2ecb50261387bf752a3977d7e5337915071bb4151e6b711a78
|
||||
./tests/tx_cache.py bcash https://bch1.trezor.io/api/tx-specific/bc37c28dfb467d2ecb50261387bf752a3977d7e5337915071bb4151e6b711a78
|
||||
|
||||
The transaction will be parsed into Trezor format and saved in
|
||||
tests/txcache/<COIN_NAME>/<TXHASH>.json. Note that only Bitcoin-compatible fields
|
||||
@ -109,7 +108,7 @@ def cli(tx, coin_name):
|
||||
click.echo(f"Fetching from {tx_url}...")
|
||||
try:
|
||||
tx_src = requests.get(tx_url).json(parse_float=Decimal)
|
||||
tx_proto = json_to_tx(tx_src)
|
||||
tx_proto = btc.from_json(tx_src)
|
||||
tx_dict = protobuf.to_dict(tx_proto)
|
||||
tx_json = json.dumps(tx_dict, sort_keys=True, indent=2) + "\n"
|
||||
except Exception as e:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user