mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-16 03:18:09 +00:00
254 lines
12 KiB
Python
254 lines
12 KiB
Python
import unittest
|
|
import common
|
|
import binascii
|
|
|
|
import trezorlib.messages_pb2 as proto
|
|
import trezorlib.types_pb2 as proto_types
|
|
|
|
class TestSignTx(common.TrezorTest):
|
|
|
|
def test_simplesigntx(self):
|
|
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
|
|
# input 0: 0.0039 BTC
|
|
|
|
inp1 = proto_types.TxInputType(address_n=[0], # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
|
|
# amount=390000,
|
|
prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
|
|
prev_index=0,
|
|
)
|
|
|
|
out1 = proto_types.TxOutputType(address='1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1',
|
|
amount=390000 - 10000,
|
|
script_type=proto_types.PAYTOADDRESS,
|
|
)
|
|
|
|
tx = self.client.simple_sign_tx('Bitcoin', [inp1, ], [out1, ])
|
|
self.assertEqual(binascii.hexlify(tx.serialized_tx), '010000000182488650ef25a58fef6788bd71b8212038d7f2bbe4750bc7bcb44701e85ef6d5000000006b4830450221009a0b7be0d4ed3146ee262b42202841834698bb3ee39c24e7437df208b8b7077102202b79ab1e7736219387dffe8d615bbdba87e11477104b867ef47afed1a5ede7810121023230848585885f63803a0a8aecdd6538792d5c539215c91698e315bf0253b43dffffffff0160cc0500000000001976a914de9b2a8da088824e8fe51debea566617d851537888ac00000000')
|
|
|
|
'''
|
|
def test_fee_too_high(self):
|
|
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
|
|
# input 0: 0.0039 BTC
|
|
|
|
inp1 = proto_types.TxInputType(address_n=[0], # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
|
|
# amount=390000,
|
|
prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
|
|
prev_index=0,
|
|
)
|
|
|
|
out1 = proto_types.TxOutputType(address='1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1',
|
|
amount=390000 - 50000,
|
|
script_type=proto_types.PAYTOADDRESS,
|
|
)
|
|
|
|
tx = self.client.simple_sign_tx('Bitcoin', [inp1, ], [out1, ])
|
|
# FIXME - Check if "Fee too high" appeared on display
|
|
'''
|
|
|
|
|
|
def test_estimate_size(self):
|
|
inp1 = proto_types.TxInputType(address_n=[0], # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
|
|
# amount=390000,
|
|
prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
|
|
prev_index=0,
|
|
)
|
|
|
|
out1 = proto_types.TxOutputType(address='1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1',
|
|
amount=390000 - 10000,
|
|
script_type=proto_types.PAYTOADDRESS,
|
|
)
|
|
|
|
|
|
est_size = self.client.estimate_tx_size('Bitcoin', [inp1, ], [out1, ])
|
|
self.assertEqual(est_size, 194)
|
|
|
|
tx = self.client.simple_sign_tx('Bitcoin', [inp1, ], [out1, ])
|
|
real_size = len(tx.serialized_tx)
|
|
|
|
self.assertGreaterEqual(est_size, real_size)
|
|
|
|
'''
|
|
def test_simplesigntx_testnet(self):
|
|
self.client.load_device_by_xprv('xprv9s21ZrQH143K3zttRjiQmYwyugvd13pnd2VzefWrfSouRfnj5oSkJgBQXxtn18E9mqrDop7fQ8Xnb9JCLPE4vghzhpU4dT33ZJ7frjzTEW8',
|
|
'', False, 'testnet')
|
|
|
|
inp1 = proto_types.TxInputType(address_n=[6], # mo8uUSFJULCMA4neRS9aS9jiXZ1N72FSLK
|
|
# amount=390000,
|
|
prev_hash=binascii.unhexlify('d83b27f16ce5069e0c8e4a02813f252500e257744d5b00c9b6128be7189117b1'),
|
|
prev_index=0,
|
|
)
|
|
|
|
out1 = proto_types.TxOutputType(address='mjKKH3Dk95VMbdNnDQYHZXoQ9QwuCZocwb',
|
|
amount=80085000,
|
|
script_type=proto_types.PAYTOADDRESS,
|
|
)
|
|
|
|
rawtx = {'d83b27f16ce5069e0c8e4a02813f252500e257744d5b00c9b6128be7189117b1': '01000000013b21cc65080c57793d0e47045a24d8e92262dc47efdc425fd5cad9a25e928f6c000000006b483045022100bde591f2c997bafa8388916663b148f4093914851a33a9903da69ad97afa6f470220138c6ff11321339974bac9c0992d7b9d72aef0c2d098f26267ec9f05d532c859012103edcc8dc5cac7dca6ed191d812621fb300863fea0dd5d14180b482b917a35acc4ffffffff020800c604000000001976a91453958011070469e2ef5e1115f34f509717d6884288acf8c99502000000001976a9141e2ba9407a6920246d0f345beecb89ed47c99a7788ac00000000'}
|
|
tx = self.client.simple_sign_tx('Testnet', [inp1, ], [out1, ])
|
|
print binascii.hexlify(tx.serialized_tx)
|
|
self.assertEqual(binascii.hexlify(tx.serialized_tx), '010000000182488650ef25a58fef6788bd71b8212038d7f2bbe4750bc7bcb44701e85ef6d5000000006b4830450221009a0b7be0d4ed3146ee262b42202841834698bb3ee39c24e7437df208b8b7077102202b79ab1e7736219387dffe8d615bbdba87e11477104b867ef47afed1a5ede7810121023230848585885f63803a0a8aecdd6538792d5c539215c91698e315bf0253b43dffffffff0160cc0500000000001976a914de9b2a8da088824e8fe51debea566617d851537888ac00000000')
|
|
|
|
def test_signtx(self):
|
|
expected_tx = '01000000012de70f7d6ffed0db70f8882f3fca90db9bb09f0e99bce27468c23d3c994fcd56' \
|
|
'010000008b4830450221009b985e14d53cfeed3496846db6ddaa77a0206138d0df4c2ccd3b' \
|
|
'759e91bae3e1022004c76e10f99ccac8ced761719181a96bae25a74829eab3ecb8f29eb07f' \
|
|
'e18f7e01410436ae8595f03a7324d1d1482ede8560a4508c767fbc662559482d5759b32209' \
|
|
'a62964699995f6e018cfbeb7a71a66d4c64fa38875d79ead0a9ac66f59c1c8b3a3ffffffff' \
|
|
'0250c30000000000001976a91444ce5c6789b0bb0e8a9ab9a4769fe181cb274c4688aca086' \
|
|
'0100000000001976a9149e03078026388661b197129a43f0f64f88379ce688ac00000000'
|
|
|
|
inp1 = proto.TxInput(index=0,
|
|
address_n=[1, 0],
|
|
amount=200000, # 0.002 BTC
|
|
prev_hash=binascii.unhexlify('56cd4f993c3dc26874e2bc990e9fb09bdb90ca3f2f88f870dbd0fe6f7d0fe72d'),
|
|
prev_index=1,
|
|
#script_sig=
|
|
)
|
|
|
|
out1 = proto.TxOutput(index=0,
|
|
address='1GnnT11aZeH6QZCtT7EjCvRF3EXHoY3owE',
|
|
address_n=[0, 1],
|
|
amount=50000, # 0.0005 BTC
|
|
script_type=proto.PAYTOADDRESS,
|
|
#script_args=
|
|
)
|
|
|
|
out2 = proto.TxOutput(index=1,
|
|
address='1FQVPnjrbkPWeA8poUoEnX9U3n9DyhAVtv',
|
|
#address_n=[],
|
|
amount=100000, # 0.001 BTC
|
|
script_type=proto.PAYTOADDRESS,
|
|
#script_args=
|
|
)
|
|
|
|
print binascii.hexlify(self.client.sign_tx([inp1], [out1, out2])[1])
|
|
'''
|
|
'''
|
|
def test_workflow(self):
|
|
inp1 = proto.TxInput(index=0,
|
|
address_n=[1,0],
|
|
amount=100000000,
|
|
prev_hash='prevhash1234354346543456543654',
|
|
prev_index=11,
|
|
)
|
|
|
|
inp2 = proto.TxInput(index=1,
|
|
address_n=[2,0],
|
|
amount=100000000,
|
|
prev_hash='prevhash2222254346543456543654',
|
|
prev_index=11,
|
|
)
|
|
|
|
out1 = proto.TxOutput(index=0,
|
|
address='1BitkeyP2nDd5oa647AjvBbbwST54W5Zmx',
|
|
amount=100000000,
|
|
script_type=proto.PAYTOADDRESS,
|
|
)
|
|
|
|
out2 = proto.TxOutput(index=1,
|
|
address='1BitkeyP2nDd5oa647AjvBbbwST54W5Zmx',
|
|
#address_n=[],
|
|
amount=100000000,
|
|
script_type=proto.PAYTOADDRESS,
|
|
#script_args=
|
|
)
|
|
|
|
serialized = ''
|
|
|
|
# Prepare and send initial message
|
|
tx = proto.SignTx()
|
|
tx.algo = proto.ELECTRUM
|
|
tx.random = self.client._get_local_entropy()
|
|
tx.inputs_count = 2
|
|
tx.outputs_count = 2
|
|
|
|
res = self.client.call(tx)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXINPUT)
|
|
self.assertEqual(res.request_index, 0)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertEqual(res.serialized_tx, '')
|
|
|
|
# FIRST SIGNATURE
|
|
res = self.client.call(inp1)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXINPUT)
|
|
self.assertEqual(res.request_index, 1)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertEqual(res.serialized_tx, '')
|
|
|
|
res = self.client.call(inp2)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXOUTPUT)
|
|
self.assertEqual(res.request_index, 0)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertEqual(res.serialized_tx, '')
|
|
|
|
res = self.client.call(out1)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXOUTPUT)
|
|
self.assertEqual(res.request_index, 1)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertEqual(res.serialized_tx, '')
|
|
|
|
res = self.client.call(out2)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXINPUT)
|
|
self.assertEqual(res.request_index, 0)
|
|
self.assertNotEqual(res.signature, '')
|
|
self.assertNotEqual(res.serialized_tx, '')
|
|
serialized += res.serialized_tx
|
|
|
|
# SECOND SIGNATURE
|
|
res = self.client.call(inp1)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXINPUT)
|
|
self.assertEqual(res.request_index, 1)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertEqual(res.serialized_tx, '')
|
|
|
|
res = self.client.call(inp2)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXOUTPUT)
|
|
self.assertEqual(res.request_index, 0)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertEqual(res.serialized_tx, '')
|
|
|
|
res = self.client.call(out1)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXOUTPUT)
|
|
self.assertEqual(res.request_index, 1)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertEqual(res.serialized_tx, '')
|
|
|
|
res = self.client.call(out2)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXOUTPUT)
|
|
self.assertEqual(res.request_index, 0)
|
|
self.assertNotEqual(res.signature, '')
|
|
self.assertNotEqual(res.serialized_tx, '')
|
|
serialized += res.serialized_tx
|
|
|
|
# FINALIZING OUTPUTS
|
|
res = self.client.call(out1)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXOUTPUT)
|
|
self.assertEqual(res.request_index, 1)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertNotEqual(res.serialized_tx, '')
|
|
serialized += res.serialized_tx
|
|
|
|
res = self.client.call(out2)
|
|
self.assertIsInstance(res, proto.TxRequest)
|
|
self.assertEqual(res.request_type, proto.TXOUTPUT)
|
|
self.assertEqual(res.request_index, -1)
|
|
self.assertEqual(res.signature, '')
|
|
self.assertNotEqual(res.serialized_tx, '')
|
|
serialized += res.serialized_tx
|
|
|
|
print binascii.hexlify(serialized)
|
|
'''
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|