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 if not utils.BITCOIN_ONLY: 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 @unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin") 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"53a12bca557c27defa366c2b4c0e46ede01f81ef3dd3aa3750db62a5505c1d06", }, { "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"d1bc60986cc5c4d57f91002e48459a50f72bdb96b8f5889cf8f467a7d968b97c", }, { "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"7536cdb202a30bf09c45e1a2f775c4efd41b9e67557b62abe12b64d367a2316e", }, { "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"df1359c0aacc05e88ae6bbecfc54fef50bd31f21f543c49f95a321dd835263be", }, # "Test vector 3" from https://github.com/zcash/zips/blob/master/zip-0243.rst { "expiry": 0x0004b048, "inputs": [ { "amount": 0x02faf080, "prevout": [ "d9042195d9a1b65b2f1f79d68ceb1a5ea6459c9651a6ad4dc1f465824785c6a8", 1, ], "script_type": InputScriptType.SPENDADDRESS, "pubkeyhash": "507173527b4c3318a2aecd793bf1cfed705950cf", "sequence": 0xfffffffe, } ], "lock_time": 0x0004b029, "outputs": [ { "script_pubkey": "76a9148132712c3ff19f3a151234616777420a6d7ef22688ac", "amount": 0x02625a00, }, { "script_pubkey": "76a9145453e4698f02a38abdaa521cd1ff2dee6fac187188ac", "amount": 0x0098958b, }, ], "version": 4, "version_group_id": 0x892f2085, "hash_type": 1, "prevouts_hash": b"fae31b8dec7b0b77e2c8d6b6eb0e7e4e55abc6574c26dd44464d9408a8e33f11", "sequence_hash": b"6c80d37f12d89b6f17ff198723e7db1247c4811d1a695d74d930f99e98418790", "outputs_hash": b"d2b04118469b7810a0d1cc59568320aad25a84f407ecac40b4f605a4e6868454", "preimage_hash": b"f3148f80dfab5e573d5edfe7a850f5fd39234f80b5429d3a57edcc11e34c585b", }, ] 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"], version_group_id=v["version_group_id"], ) zip243 = Zip243(0x76b809bb) # Sapling 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()