def test_signtx(self):
self.setup_mnemonic_pin_passphrase()
inp1 = proto_types.TxInputType(address_n=[0], # 14LmW5k4ssUrtbAB4255zdqv3b4w1TuX9e
prev_hash=binascii.unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
prev_index=0,
)
out1 = proto_types.TxOutputType(address='1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1',
amount=390000 - 10000,
script_type=proto_types.PAYTOADDRESS,
)
tx_responses = [
proto.TxRequest(request_type=proto_types.TXINPUT, details=proto_types.TxRequestDetailsType(request_index=0)),
proto.TxRequest(request_type=proto_types.TXMETA, details=proto_types.TxRequestDetailsType(tx_hash=binascii.unhexlify("d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882"))),
proto.TxRequest(request_type=proto_types.TXINPUT, details=proto_types.TxRequestDetailsType(request_index=0, tx_hash=binascii.unhexlify("d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882"))),
proto.TxRequest(request_type=proto_types.TXINPUT, details=proto_types.TxRequestDetailsType(request_index=1, tx_hash=binascii.unhexlify("d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882"))),
proto.TxRequest(request_type=proto_types.TXOUTPUT, details=proto_types.TxRequestDetailsType(request_index=0, tx_hash=binascii.unhexlify("d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882"))),
proto.TxRequest(request_type=proto_types.TXOUTPUT, details=proto_types.TxRequestDetailsType(request_index=0)),
proto.ButtonRequest(code=proto_types.ButtonRequest_ConfirmOutput),
proto.ButtonRequest(code=proto_types.ButtonRequest_SignTx),
proto.TxRequest(request_type=proto_types.TXINPUT, details=proto_types.TxRequestDetailsType(request_index=0)),
proto.TxRequest(request_type=proto_types.TXOUTPUT, details=proto_types.TxRequestDetailsType(request_index=0)),
proto.TxRequest(request_type=proto_types.TXOUTPUT, details=proto_types.TxRequestDetailsType(request_index=0)),
proto.TxRequest(request_type=proto_types.TXFINISHED),
]
self.client.clear_session()
with self.client:
# Pin & Passphrase are needed after device is locked
self.client.set_expected_responses([
proto.PinMatrixRequest(),
proto.PassphraseRequest(),
] + tx_responses)
self.client.sign_tx('Bitcoin', [inp1, ], [out1, ])
with self.client:
# Pin & Passphrase not needed, since they're cached, and the device is unlocked
self.client.set_expected_responses([
] + tx_responses)
self.client.sign_tx('Bitcoin', [inp1, ], [out1, ])
self.client.clear_session()
with self.client:
# Pin & Passphrase needed again after session is cleared, and the device is locked
self.client.set_expected_responses([
proto.PinMatrixRequest(),
proto.PassphraseRequest(),
] + tx_responses)
self.client.sign_tx('Bitcoin', [inp1, ], [out1, ])
def test_clearsession(self):
self.setup_mnemonic_pin_passphrase()
with self.client:
self.client.set_expected_responses([
proto.ButtonRequest(code=proto_types.ButtonRequest_Ping),
proto.PassphraseRequest(),
proto.Success()
])
res = self.client.ping('random data',
button_protection=True,
pin_protection=True,
passphrase_protection=True)
self.assertEqual(res, 'random data')
with self.client:
# pin and passphrase are cached
self.client.set_expected_responses([
proto.ButtonRequest(code=proto_types.ButtonRequest_Ping),
proto.Success()
])
res = self.client.ping('random data',
button_protection=True,
pin_protection=True,
passphrase_protection=True)
self.assertEqual(res, 'random data')
self.client.clear_session()
# session cache is cleared
with self.client:
self.client.set_expected_responses([
proto.ButtonRequest(code=proto_types.ButtonRequest_Ping),
proto.PinMatrixRequest(),
proto.PassphraseRequest(),
proto.Success()
])
res = self.client.ping('random data',
button_protection=True,
pin_protection=True,
passphrase_protection=True)
self.assertEqual(res, 'random data')
with self.client:
# pin and passphrase are cached
self.client.set_expected_responses([
proto.ButtonRequest(code=proto_types.ButtonRequest_Ping),
proto.Success()
])
res = self.client.ping('random data',
button_protection=True,
pin_protection=True,
passphrase_protection=True)
self.assertEqual(res, 'random data')
def test_get_address(self):
with self.client:
self.setup_mnemonic_pin_passphrase()
self.client.clear_session()
self.client.set_expected_responses([proto.PinMatrixRequest(),
proto.PassphraseRequest(),
proto.Address()])
self.client.get_address('Bitcoin', [])
def test_get_public_key(self):
with self.client:
self.setup_mnemonic_pin_passphrase()
self.client.clear_session()
self.client.set_expected_responses([proto.PinMatrixRequest(),
proto.PassphraseRequest(),
proto.PublicKey()])
self.client.get_public_node([])
def test_ping(self):
with self.client:
self.setup_mnemonic_pin_passphrase()
self.client.clear_session()
self.client.set_expected_responses([proto.ButtonRequest(),
proto.PinMatrixRequest(),
proto.PassphraseRequest(),
proto.Success()])
self.client.ping('msg', True, True, True)
def test_sign_message(self):
with self.client:
self.setup_mnemonic_pin_passphrase()
self.client.clear_session()
self.client.set_expected_responses([proto.ButtonRequest(),
proto.PinMatrixRequest(),
proto.PassphraseRequest(),
proto.MessageSignature()])
self.client.sign_message('Bitcoin', [], 'testing message')
def test_ping_format_specifier_sanitize(self):
self.setup_mnemonic_pin_passphrase()
self.client.clear_session()
with self.client:
self.client.set_expected_responses([
proto.ButtonRequest(code=proto_types.ButtonRequest_Ping),
proto.PinMatrixRequest(),
proto.PassphraseRequest(),
proto.ButtonRequest(code=proto_types.ButtonRequest_Other),
proto.Success()
])
res = self.client.ping('%s%x%n%p', button_protection=True, pin_protection=True, passphrase_protection=True)
self.assertEqual(res, '%s%x%n%p')
