mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-25 08:58:14 +00:00
e073e619c9
isort set to skip the first necessary "from common import *" line. A better solution would be to get rid of the need of this import in the future. [no changelog]
220 lines
8.0 KiB
Python
220 lines
8.0 KiB
Python
from common import * # isort:skip
|
|
|
|
from trezor.crypto import bip39
|
|
from trezor.enums import AmountUnit, OutputScriptType
|
|
from trezor.enums.RequestType import TXFINISHED, TXINPUT, TXMETA, TXOUTPUT
|
|
from trezor.messages import (
|
|
PrevInput,
|
|
PrevOutput,
|
|
PrevTx,
|
|
SignTx,
|
|
TxAckInput,
|
|
TxAckInputWrapper,
|
|
TxAckOutput,
|
|
TxAckOutputWrapper,
|
|
TxAckPrevInput,
|
|
TxAckPrevInputWrapper,
|
|
TxAckPrevMeta,
|
|
TxAckPrevOutput,
|
|
TxAckPrevOutputWrapper,
|
|
TxInput,
|
|
TxOutput,
|
|
TxRequest,
|
|
TxRequestDetailsType,
|
|
TxRequestSerializedType,
|
|
)
|
|
from trezor.utils import chunks
|
|
|
|
from apps.bitcoin.keychain import _get_schemas_for_coin
|
|
from apps.bitcoin.sign_tx import bitcoinlike, helpers
|
|
from apps.common import coins
|
|
from apps.common.keychain import Keychain
|
|
|
|
EMPTY_SERIALIZED = TxRequestSerializedType(serialized_tx=bytearray())
|
|
|
|
|
|
@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
|
|
class TestSignTx_GRS(unittest.TestCase):
|
|
# pylint: disable=C0301
|
|
|
|
def test_one_one_fee(self):
|
|
# http://groestlsight.groestlcoin.org/tx/f56521b17b828897f72b30dd21b0192fd942342e89acbb06abf1d446282c30f5
|
|
# ptx1: http://groestlsight.groestlcoin.org/api/tx/cb74c8478c5814742c87cffdb4a21231869888f8042fb07a90e015a9db1f9d4a
|
|
|
|
coin = coins.by_name("Groestlcoin")
|
|
|
|
ptx1 = PrevTx(
|
|
version=1,
|
|
lock_time=2160993,
|
|
inputs_count=1,
|
|
outputs_count=1,
|
|
extra_data_len=0,
|
|
)
|
|
pinp1 = PrevInput(
|
|
script_sig=unhexlify(
|
|
"48304502210096a287593b1212a188e778596eb8ecd4cc169b93a4d115226460d8e3deae431c02206c78ec09b3df977f04a6df5eb53181165c4ea5a0b35f826551349130f879d6b8012102cf5126ff54e38a80a919579d7091cafe24840eab1d30fe2b4d59bdd9d267cad8"
|
|
),
|
|
prev_hash=unhexlify(
|
|
"7dc74a738c50c2ae1228ce9890841e5355fd6d7f2c1367e0a74403ab60db3224"
|
|
),
|
|
prev_index=0,
|
|
sequence=4294967294,
|
|
)
|
|
pout1 = PrevOutput(
|
|
script_pubkey=unhexlify(
|
|
"76a914172b4e06e9b7881a48d2ee8062b495d0b2517fe888ac"
|
|
),
|
|
amount=210016,
|
|
)
|
|
|
|
inp1 = TxInput(
|
|
address_n=[
|
|
44 | 0x80000000,
|
|
17 | 0x80000000,
|
|
0 | 0x80000000,
|
|
0,
|
|
2,
|
|
], # FXHDsC5ZqWQHkDmShzgRVZ1MatpWhwxTAA
|
|
prev_hash=unhexlify(
|
|
"cb74c8478c5814742c87cffdb4a21231869888f8042fb07a90e015a9db1f9d4a"
|
|
),
|
|
prev_index=0,
|
|
amount=210016,
|
|
)
|
|
out1 = TxOutput(
|
|
address="FtM4zAn9aVYgHgxmamWBgWPyZsb6RhvkA9",
|
|
amount=210016 - 192,
|
|
script_type=OutputScriptType.PAYTOADDRESS,
|
|
address_n=[],
|
|
)
|
|
tx = SignTx(
|
|
coin_name="Groestlcoin",
|
|
version=1,
|
|
lock_time=0,
|
|
inputs_count=1,
|
|
outputs_count=1,
|
|
)
|
|
|
|
# precomputed tx weight is 768
|
|
fee_rate = 192 / (768 / 4)
|
|
|
|
messages = [
|
|
None,
|
|
TxRequest(
|
|
request_type=TXINPUT,
|
|
details=TxRequestDetailsType(request_index=0, tx_hash=None),
|
|
serialized=EMPTY_SERIALIZED,
|
|
),
|
|
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
|
|
TxRequest(
|
|
request_type=TXOUTPUT,
|
|
details=TxRequestDetailsType(request_index=0, tx_hash=None),
|
|
serialized=EMPTY_SERIALIZED,
|
|
),
|
|
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
|
|
helpers.UiConfirmOutput(out1, coin, AmountUnit.BITCOIN, 0, False),
|
|
True,
|
|
helpers.UiConfirmTotal(
|
|
210016, 192, fee_rate, coin, AmountUnit.BITCOIN, inp1.address_n[:3]
|
|
),
|
|
True,
|
|
TxRequest(
|
|
request_type=TXINPUT,
|
|
details=TxRequestDetailsType(request_index=0, tx_hash=None),
|
|
serialized=EMPTY_SERIALIZED,
|
|
),
|
|
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
|
|
TxRequest(
|
|
request_type=TXMETA,
|
|
details=TxRequestDetailsType(
|
|
request_index=None,
|
|
tx_hash=unhexlify(
|
|
"cb74c8478c5814742c87cffdb4a21231869888f8042fb07a90e015a9db1f9d4a"
|
|
),
|
|
),
|
|
serialized=EMPTY_SERIALIZED,
|
|
),
|
|
TxAckPrevMeta(tx=ptx1),
|
|
TxRequest(
|
|
request_type=TXINPUT,
|
|
details=TxRequestDetailsType(
|
|
request_index=0,
|
|
tx_hash=unhexlify(
|
|
"cb74c8478c5814742c87cffdb4a21231869888f8042fb07a90e015a9db1f9d4a"
|
|
),
|
|
),
|
|
serialized=EMPTY_SERIALIZED,
|
|
),
|
|
TxAckPrevInput(tx=TxAckPrevInputWrapper(input=pinp1)),
|
|
TxRequest(
|
|
request_type=TXOUTPUT,
|
|
details=TxRequestDetailsType(
|
|
request_index=0,
|
|
tx_hash=unhexlify(
|
|
"cb74c8478c5814742c87cffdb4a21231869888f8042fb07a90e015a9db1f9d4a"
|
|
),
|
|
),
|
|
serialized=EMPTY_SERIALIZED,
|
|
),
|
|
TxAckPrevOutput(tx=TxAckPrevOutputWrapper(output=pout1)),
|
|
# ButtonRequest(code=ButtonRequest_ConfirmOutput),
|
|
# ButtonRequest(code=ButtonRequest_SignTx),
|
|
TxRequest(
|
|
request_type=TXINPUT,
|
|
details=TxRequestDetailsType(request_index=0, tx_hash=None),
|
|
serialized=TxRequestSerializedType(
|
|
signature_index=None,
|
|
signature=None,
|
|
serialized_tx=unhexlify("0100000001"),
|
|
),
|
|
),
|
|
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
|
|
TxRequest(
|
|
request_type=TXOUTPUT,
|
|
details=TxRequestDetailsType(request_index=0, tx_hash=None),
|
|
serialized=EMPTY_SERIALIZED,
|
|
),
|
|
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
|
|
TxRequest(
|
|
request_type=TXOUTPUT,
|
|
details=TxRequestDetailsType(request_index=0, tx_hash=None),
|
|
serialized=TxRequestSerializedType(
|
|
signature_index=0,
|
|
signature=unhexlify(
|
|
"304402201fb96d20d0778f54520ab59afe70d5fb20e500ecc9f02281cf57934e8029e8e10220383d5a3e80f2e1eb92765b6da0f23d454aecbd8236f083d483e9a74302368761"
|
|
),
|
|
serialized_tx=unhexlify(
|
|
"4a9d1fdba915e0907ab02f04f88898863112a2b4fdcf872c7414588c47c874cb000000006a47304402201fb96d20d0778f54520ab59afe70d5fb20e500ecc9f02281cf57934e8029e8e10220383d5a3e80f2e1eb92765b6da0f23d454aecbd8236f083d483e9a7430236876101210331693756f749180aeed0a65a0fab0625a2250bd9abca502282a4cf0723152e67ffffffff01"
|
|
),
|
|
),
|
|
),
|
|
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
|
|
TxRequest(
|
|
request_type=TXFINISHED,
|
|
details=TxRequestDetailsType(),
|
|
serialized=TxRequestSerializedType(
|
|
signature_index=None,
|
|
signature=None,
|
|
serialized_tx=unhexlify(
|
|
"a0330300000000001976a914fe40329c95c5598ac60752a5310b320cb52d18e688ac00000000"
|
|
),
|
|
),
|
|
),
|
|
]
|
|
|
|
seed = bip39.seed(" ".join(["all"] * 12), "")
|
|
ns = _get_schemas_for_coin(coin)
|
|
keychain = Keychain(seed, coin.curve_name, ns)
|
|
signer = bitcoinlike.Bitcoinlike(tx, keychain, coin, None).signer()
|
|
for request, expected_response in chunks(messages, 2):
|
|
response = signer.send(request)
|
|
if isinstance(response, tuple):
|
|
_, response = response
|
|
self.assertEqual(response, expected_response)
|
|
with self.assertRaises(StopIteration):
|
|
signer.send(None)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|