src/apps/wallet/sign_tx: refactor Zcash, add WIP ZIP243; use ensure instead of assert where possible
Pavol Rusnak
6 years ago
parent
b2327c37b7
commit
e16a8fc8c9
@ -0,0 +1,184 @@
|
||||
from common import *
|
||||
from trezor.messages import InputScriptType
|
||||
from trezor.messages.SignTx import SignTx
|
||||
from trezor.messages.TxInputType import TxInputType
|
||||
from trezor.messages.TxOutputBinType import TxOutputBinType
|
||||
|
||||
from apps.common import coins
|
||||
from apps.wallet.sign_tx.zcash import Zip143
|
||||
|
||||
|
||||
# test vectors inspired from https://github.com/zcash-hackworks/zcash-test-vectors/blob/master/zip_0143.py
|
||||
class TestZcashZip143(unittest.TestCase):
|
||||
|
||||
VECTORS = [
|
||||
{
|
||||
"expiry": 71895707,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 35268204,
|
||||
"prevout": [
|
||||
"702c35a67cd7364d3fab552fb349e35c15c50250453fd18f7b855992632e2c76",
|
||||
4025613248,
|
||||
],
|
||||
"pubkeyhash": "4d9cafb657677f2321fc538e367767dbdf551539",
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"sequence": 1999822371,
|
||||
}
|
||||
],
|
||||
"lock_time": 452079490,
|
||||
"outputs": [
|
||||
{"script_pubkey": "06535251635252", "amount": 1246336469307855}
|
||||
],
|
||||
"version": 3,
|
||||
"version_group_id": 0x3C48270,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"bd4318eecf841a0cf01c2be532cf4bc3303e881e2aface159f1882f153152688",
|
||||
"sequence_hash": b"9ac6a31952ff626bf5a0a30d3d8ac63a0d4298d33d7bc38854bfa5860695e30a",
|
||||
"outputs_hash": b"d0cadf116b4441f5e1e17814908dee509ec262a79f3c88f7f3389e8200658992",
|
||||
"preimage_hash": b"287146efaa30b3c1c7bd1a72308c46205712e6944780fdc5a7f91f477fddd55b",
|
||||
},
|
||||
{
|
||||
"expiry": 231041495,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 39263472,
|
||||
"prevout": [
|
||||
"76647d2be4c2cd6b3d17d6870971d7a098baf72c6f6f1214cf1faae488bd7de2",
|
||||
1547817817,
|
||||
],
|
||||
"pubkeyhash": "9f5d230603ce57a0e8b31a8c9b6c983ad5b00cd5",
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"sequence": 3973122135,
|
||||
},
|
||||
{
|
||||
"amount": 57533728,
|
||||
"prevout": [
|
||||
"cccc0df65a04943ad5cbc13f295f000fe056c40b2d88f27dc34cfeb803be3483",
|
||||
3053054889,
|
||||
],
|
||||
"pubkeyhash": "b5e71ef1df5ed3a2589607ca58ed19634f07fb4f",
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"sequence": 3932380530,
|
||||
},
|
||||
],
|
||||
"lock_time": 3087412294,
|
||||
"outputs": [
|
||||
{"script_pubkey": "03ac6552", "amount": 546412698509744},
|
||||
{"script_pubkey": "00", "amount": 166856241017532},
|
||||
],
|
||||
"version": 3,
|
||||
"version_group_id": 0x3C48270,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"8e286c6c0dde3119271c9c1398ef46614b0253c502b00a3691cec2e9047da35b",
|
||||
"sequence_hash": b"58477fd9ecd5faf3e08159e0ab5fdaab66cab364d081498ddcef41de0af3624e",
|
||||
"outputs_hash": b"c518797fc6f2c08fc22aa3f66122047b360e1db4df5c3feb28573c00cdf45fa1",
|
||||
"preimage_hash": b"b71358954aca8ffe1d3f1b5707ed6c082ed2b35e35e293b897bfc12ae7c85396",
|
||||
},
|
||||
{
|
||||
"expiry": 186996458,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 14267260,
|
||||
"prevout": [
|
||||
"6c6fae359f645c276891c0dcab3faf187700c082dc477740fb3f2cd7bb59fb35",
|
||||
1290359941,
|
||||
],
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"pubkeyhash": "0873d1baed6c8696e4bd2b26755692b4d8050086",
|
||||
"sequence": 1230917966,
|
||||
}
|
||||
],
|
||||
"lock_time": 1520002857,
|
||||
"outputs": [],
|
||||
"version": 3,
|
||||
"version_group_id": 0x3C48270,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"445bc6328cd33b3c86259953dd674bded341ff1e1104dc21856919e9761036dd",
|
||||
"sequence_hash": b"42e1d5c2636f165afaa954afa6d7a50779eb145e947bf668f1a40dd771c711fc",
|
||||
"outputs_hash": b"869eda84eecf7257f9979a4848bbf52f4969a5736594ab7ba41452e7bb906824",
|
||||
"preimage_hash": b"19ea4ce2467cb5cdcc4562ce81b22006f1fb55189ea93ccc7a3bc78ff5d97a93",
|
||||
},
|
||||
{
|
||||
"expiry": 254788522,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 36100600,
|
||||
"prevout": [
|
||||
"e818f9057c5abaaa2e5c15b94945cd424c28a5fa385dadfe4907b274d842707d",
|
||||
1517971891,
|
||||
],
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"pubkeyhash": "80b6c35a9b2efd77dbfd5d5d3bb1fd03ff40f182",
|
||||
"sequence": 3833577708,
|
||||
},
|
||||
{
|
||||
"amount": 71238918,
|
||||
"prevout": [
|
||||
"7350d1014670212efe81fb7c73e8450df814ef6232f7490f63ccf07480f884a6",
|
||||
687648622,
|
||||
],
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"pubkeyhash": "fb8c27a442afe0f0b39a3875823c4893fe8f8550",
|
||||
"sequence": 4190617831,
|
||||
},
|
||||
],
|
||||
"lock_time": 1557067344,
|
||||
"outputs": [
|
||||
{"script_pubkey": "076a53655151516a", "amount": 470086065540185}
|
||||
],
|
||||
"version": 3,
|
||||
"version_group_id": 0x3C48270,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"509abdfafcc75265037f1ce6a4658ac9ecadd7b82378c3fbaeb48ab437ff6898",
|
||||
"sequence_hash": b"2b13f671cd1a9aa04c1e250eef74a316d7d2b049360d20604514ddc2dfacfd23",
|
||||
"outputs_hash": b"4f01b8785e80779290aa86c16b24952f9b7f8bc09da44e68f760ab1920ab8f2a",
|
||||
"preimage_hash": b"d0b92233e185a69f320f18bfacff7e5dcc5f0ed02af2f3ecf59359853e105e59",
|
||||
},
|
||||
]
|
||||
|
||||
def test_zip143(self):
|
||||
coin = coins.by_name("Zcash")
|
||||
|
||||
for v in self.VECTORS:
|
||||
tx = SignTx(
|
||||
coin_name="Zcash",
|
||||
inputs_count=len(v["inputs"]),
|
||||
outputs_count=len(v["outputs"]),
|
||||
version=v["version"],
|
||||
lock_time=v["lock_time"],
|
||||
expiry=v["expiry"],
|
||||
overwintered=(v["version"] >= 3),
|
||||
version_group_id=v["version_group_id"],
|
||||
)
|
||||
zip143 = Zip143()
|
||||
for i in v["inputs"]:
|
||||
txi = TxInputType()
|
||||
txi.amount = i["amount"]
|
||||
txi.prev_hash = unhexlify(i["prevout"][0])
|
||||
txi.prev_index = i["prevout"][1]
|
||||
txi.script_type = i["script_type"]
|
||||
txi.sequence = i["sequence"]
|
||||
zip143.add_prevouts(txi)
|
||||
zip143.add_sequence(txi)
|
||||
for o in v["outputs"]:
|
||||
txo = TxOutputBinType()
|
||||
txo.amount = o["amount"]
|
||||
txo.script_pubkey = unhexlify(o["script_pubkey"])
|
||||
zip143.add_output(txo)
|
||||
|
||||
self.assertEqual(hexlify(zip143.get_prevouts_hash()), v["prevouts_hash"])
|
||||
self.assertEqual(hexlify(zip143.get_sequence_hash()), v["sequence_hash"])
|
||||
self.assertEqual(hexlify(zip143.get_outputs_hash()), v["outputs_hash"])
|
||||
self.assertEqual(
|
||||
hexlify(
|
||||
zip143.preimage_hash(
|
||||
coin, tx, txi, unhexlify(i["pubkeyhash"]), v["hash_type"]
|
||||
)
|
||||
),
|
||||
v["preimage_hash"],
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@ -0,0 +1,184 @@
|
||||
from common import *
|
||||
from trezor.messages import InputScriptType
|
||||
from trezor.messages.SignTx import SignTx
|
||||
from trezor.messages.TxInputType import TxInputType
|
||||
from trezor.messages.TxOutputBinType import TxOutputBinType
|
||||
|
||||
from apps.common import coins
|
||||
from apps.wallet.sign_tx.zcash import Zip243
|
||||
|
||||
|
||||
# test vectors inspired from https://github.com/zcash-hackworks/zcash-test-vectors/blob/master/zip_0243.py
|
||||
class TestZcashZip243(unittest.TestCase):
|
||||
|
||||
VECTORS = [
|
||||
{
|
||||
"expiry": 71895707,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 35268204,
|
||||
"prevout": [
|
||||
"702c35a67cd7364d3fab552fb349e35c15c50250453fd18f7b855992632e2c76",
|
||||
4025613248,
|
||||
],
|
||||
"pubkeyhash": "4d9cafb657677f2321fc538e367767dbdf551539",
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"sequence": 1999822371,
|
||||
}
|
||||
],
|
||||
"lock_time": 452079490,
|
||||
"outputs": [
|
||||
{"script_pubkey": "06535251635252", "amount": 1246336469307855}
|
||||
],
|
||||
"version": 4,
|
||||
"version_group_id": 0x892F2085,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"bd4318eecf841a0cf01c2be532cf4bc3303e881e2aface159f1882f153152688",
|
||||
"sequence_hash": b"9ac6a31952ff626bf5a0a30d3d8ac63a0d4298d33d7bc38854bfa5860695e30a",
|
||||
"outputs_hash": b"d0cadf116b4441f5e1e17814908dee509ec262a79f3c88f7f3389e8200658992",
|
||||
"preimage_hash": b"80b3cdfbefc9cea6818e148b5af6133dfac7d9ff8435b256ef1958b906b3647f",
|
||||
},
|
||||
{
|
||||
"expiry": 231041495,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 39263472,
|
||||
"prevout": [
|
||||
"76647d2be4c2cd6b3d17d6870971d7a098baf72c6f6f1214cf1faae488bd7de2",
|
||||
1547817817,
|
||||
],
|
||||
"pubkeyhash": "9f5d230603ce57a0e8b31a8c9b6c983ad5b00cd5",
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"sequence": 3973122135,
|
||||
},
|
||||
{
|
||||
"amount": 57533728,
|
||||
"prevout": [
|
||||
"cccc0df65a04943ad5cbc13f295f000fe056c40b2d88f27dc34cfeb803be3483",
|
||||
3053054889,
|
||||
],
|
||||
"pubkeyhash": "b5e71ef1df5ed3a2589607ca58ed19634f07fb4f",
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"sequence": 3932380530,
|
||||
},
|
||||
],
|
||||
"lock_time": 3087412294,
|
||||
"outputs": [
|
||||
{"script_pubkey": "03ac6552", "amount": 546412698509744},
|
||||
{"script_pubkey": "00", "amount": 166856241017532},
|
||||
],
|
||||
"version": 4,
|
||||
"version_group_id": 0x892F2085,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"8e286c6c0dde3119271c9c1398ef46614b0253c502b00a3691cec2e9047da35b",
|
||||
"sequence_hash": b"58477fd9ecd5faf3e08159e0ab5fdaab66cab364d081498ddcef41de0af3624e",
|
||||
"outputs_hash": b"c518797fc6f2c08fc22aa3f66122047b360e1db4df5c3feb28573c00cdf45fa1",
|
||||
"preimage_hash": b"25769ee5a943f0e1fd165c2b049de2732ea88f059658f63932f12b16087337fd",
|
||||
},
|
||||
{
|
||||
"expiry": 186996458,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 14267260,
|
||||
"prevout": [
|
||||
"6c6fae359f645c276891c0dcab3faf187700c082dc477740fb3f2cd7bb59fb35",
|
||||
1290359941,
|
||||
],
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"pubkeyhash": "0873d1baed6c8696e4bd2b26755692b4d8050086",
|
||||
"sequence": 1230917966,
|
||||
}
|
||||
],
|
||||
"lock_time": 1520002857,
|
||||
"outputs": [],
|
||||
"version": 4,
|
||||
"version_group_id": 0x892F2085,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"445bc6328cd33b3c86259953dd674bded341ff1e1104dc21856919e9761036dd",
|
||||
"sequence_hash": b"42e1d5c2636f165afaa954afa6d7a50779eb145e947bf668f1a40dd771c711fc",
|
||||
"outputs_hash": b"869eda84eecf7257f9979a4848bbf52f4969a5736594ab7ba41452e7bb906824",
|
||||
"preimage_hash": b"e2d5a735f8f4c3f6e382405a214daa520a7f80200c177d8daafe1c0344db9dd3",
|
||||
},
|
||||
{
|
||||
"expiry": 254788522,
|
||||
"inputs": [
|
||||
{
|
||||
"amount": 36100600,
|
||||
"prevout": [
|
||||
"e818f9057c5abaaa2e5c15b94945cd424c28a5fa385dadfe4907b274d842707d",
|
||||
1517971891,
|
||||
],
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"pubkeyhash": "80b6c35a9b2efd77dbfd5d5d3bb1fd03ff40f182",
|
||||
"sequence": 3833577708,
|
||||
},
|
||||
{
|
||||
"amount": 71238918,
|
||||
"prevout": [
|
||||
"7350d1014670212efe81fb7c73e8450df814ef6232f7490f63ccf07480f884a6",
|
||||
687648622,
|
||||
],
|
||||
"script_type": InputScriptType.SPENDADDRESS,
|
||||
"pubkeyhash": "fb8c27a442afe0f0b39a3875823c4893fe8f8550",
|
||||
"sequence": 4190617831,
|
||||
},
|
||||
],
|
||||
"lock_time": 1557067344,
|
||||
"outputs": [
|
||||
{"script_pubkey": "076a53655151516a", "amount": 470086065540185}
|
||||
],
|
||||
"version": 4,
|
||||
"version_group_id": 0x892F2085,
|
||||
"hash_type": 1,
|
||||
"prevouts_hash": b"509abdfafcc75265037f1ce6a4658ac9ecadd7b82378c3fbaeb48ab437ff6898",
|
||||
"sequence_hash": b"2b13f671cd1a9aa04c1e250eef74a316d7d2b049360d20604514ddc2dfacfd23",
|
||||
"outputs_hash": b"4f01b8785e80779290aa86c16b24952f9b7f8bc09da44e68f760ab1920ab8f2a",
|
||||
"preimage_hash": b"74908bfb31d1ce68f1b82c12e96fb7799669293d92da84a689a1dd1f9ed74d8d",
|
||||
},
|
||||
]
|
||||
|
||||
def test_zip243(self):
|
||||
coin = coins.by_name("Zcash")
|
||||
|
||||
for v in self.VECTORS:
|
||||
tx = SignTx(
|
||||
coin_name="Zcash",
|
||||
inputs_count=len(v["inputs"]),
|
||||
outputs_count=len(v["outputs"]),
|
||||
version=v["version"],
|
||||
lock_time=v["lock_time"],
|
||||
expiry=v["expiry"],
|
||||
overwintered=(v["version"] >= 3),
|
||||
version_group_id=v["version_group_id"],
|
||||
)
|
||||
zip243 = Zip243()
|
||||
for i in v["inputs"]:
|
||||
txi = TxInputType()
|
||||
txi.amount = i["amount"]
|
||||
txi.prev_hash = unhexlify(i["prevout"][0])
|
||||
txi.prev_index = i["prevout"][1]
|
||||
txi.script_type = i["script_type"]
|
||||
txi.sequence = i["sequence"]
|
||||
zip243.add_prevouts(txi)
|
||||
zip243.add_sequence(txi)
|
||||
for o in v["outputs"]:
|
||||
txo = TxOutputBinType()
|
||||
txo.amount = o["amount"]
|
||||
txo.script_pubkey = unhexlify(o["script_pubkey"])
|
||||
zip243.add_output(txo)
|
||||
|
||||
self.assertEqual(hexlify(zip243.get_prevouts_hash()), v["prevouts_hash"])
|
||||
self.assertEqual(hexlify(zip243.get_sequence_hash()), v["sequence_hash"])
|
||||
self.assertEqual(hexlify(zip243.get_outputs_hash()), v["outputs_hash"])
|
||||
self.assertEqual(
|
||||
hexlify(
|
||||
zip243.preimage_hash(
|
||||
coin, tx, txi, unhexlify(i["pubkeyhash"]), v["hash_type"]
|
||||
)
|
||||
),
|
||||
v["preimage_hash"],
|
||||
)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@ -1 +1 @@
|
||||
Subproject commit 41e4a84b5b01d03e980f84fab29c8f0b0ec948f5
|
||||
Subproject commit ea0262266d9c42315617e9abbcf0b539c373ebc6
|
||||
Loading…
Reference in new issue