mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-22 14:28:07 +00:00
112 lines
4.0 KiB
Python
112 lines
4.0 KiB
Python
|
from common import unittest, H_
|
||
|
|
|
||
|
|
from trezor.messages.AuthorizeCoinJoin import AuthorizeCoinJoin
|
||
|
|
from trezor.messages.GetOwnershipProof import GetOwnershipProof
|
||
|
|
from trezor.messages.SignTx import SignTx
|
||
|
|
from trezor.messages import InputScriptType
|
||
|
|
|
||
|
|
from apps.common import coins
|
||
|
|
from apps.bitcoin.authorization import CoinJoinAuthorization
|
||
|
|
|
||
|
|
_ROUND_ID_LEN = 32
|
||
|
|
|
||
|
|
|
||
|
|
class TestAuthorization(unittest.TestCase):
|
||
|
|
|
||
|
|
coin = coins.by_name('Bitcoin')
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.msg_auth = AuthorizeCoinJoin(
|
||
|
|
coordinator="www.example.com",
|
||
|
|
max_total_fee=40000,
|
||
|
|
fee_per_anonymity=0.003 * 10**9,
|
||
|
|
address_n=[H_(84), H_(0), H_(0)],
|
||
|
|
coin_name=self.coin.coin_name,
|
||
|
|
script_type=InputScriptType.SPENDWITNESS,
|
||
|
|
)
|
||
|
|
|
||
|
|
self.authorization = CoinJoinAuthorization(self.msg_auth, None, self.coin)
|
||
|
|
|
||
|
|
def test_ownership_proof_account_depth_mismatch(self):
|
||
|
|
# Account depth mismatch.
|
||
|
|
msg = GetOwnershipProof(
|
||
|
|
address_n=[H_(84), H_(0), H_(0), 1],
|
||
|
|
coin_name=self.coin.coin_name,
|
||
|
|
script_type=InputScriptType.SPENDWITNESS,
|
||
|
|
user_confirmation=True,
|
||
|
|
commitment_data=b"www.example.com" + int.to_bytes(1, _ROUND_ID_LEN, "big"),
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertFalse(self.authorization.check_get_ownership_proof(msg))
|
||
|
|
|
||
|
|
def test_ownership_proof_account_path_mismatch(self):
|
||
|
|
# Account path mismatch.
|
||
|
|
msg = GetOwnershipProof(
|
||
|
|
address_n=[H_(49), H_(0), H_(0), 1, 2],
|
||
|
|
coin_name=self.coin.coin_name,
|
||
|
|
script_type=InputScriptType.SPENDWITNESS,
|
||
|
|
user_confirmation=True,
|
||
|
|
commitment_data=b"www.example.com" + int.to_bytes(1, _ROUND_ID_LEN, "big"),
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertFalse(self.authorization.check_get_ownership_proof(msg))
|
||
|
|
|
||
|
|
def test_ownership_proof_coordinator_mismatch(self):
|
||
|
|
# Coordinator name mismatch.
|
||
|
|
msg = GetOwnershipProof(
|
||
|
|
address_n=[H_(84), H_(0), H_(0), 1, 2],
|
||
|
|
coin_name=self.coin.coin_name,
|
||
|
|
script_type=InputScriptType.SPENDWITNESS,
|
||
|
|
user_confirmation=True,
|
||
|
|
commitment_data=b"www.example.org" + int.to_bytes(1, _ROUND_ID_LEN, "big"),
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertFalse(self.authorization.check_get_ownership_proof(msg))
|
||
|
|
|
||
|
|
def test_ownership_proof_wrong_round_id(self):
|
||
|
|
# Wrong round ID length.
|
||
|
|
msg = GetOwnershipProof(
|
||
|
|
address_n=[H_(84), H_(0), H_(0), 1, 2],
|
||
|
|
coin_name=self.coin.coin_name,
|
||
|
|
script_type=InputScriptType.SPENDWITNESS,
|
||
|
|
user_confirmation=True,
|
||
|
|
commitment_data=b"www.example.com" + int.to_bytes(1, _ROUND_ID_LEN - 1, "big"),
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertFalse(self.authorization.check_get_ownership_proof(msg))
|
||
|
|
|
||
|
|
msg = GetOwnershipProof(
|
||
|
|
address_n=[H_(84), H_(0), H_(0), 1, 2],
|
||
|
|
coin_name=self.coin.coin_name,
|
||
|
|
script_type=InputScriptType.SPENDWITNESS,
|
||
|
|
user_confirmation=True,
|
||
|
|
commitment_data=b"www.example.com" + int.to_bytes(1, _ROUND_ID_LEN + 1, "big"),
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertFalse(self.authorization.check_get_ownership_proof(msg))
|
||
|
|
|
||
|
|
def test_authorize_ownership_proof(self):
|
||
|
|
|
||
|
|
msg = GetOwnershipProof(
|
||
|
|
address_n=[H_(84), H_(0), H_(0), 1, 2],
|
||
|
|
coin_name=self.coin.coin_name,
|
||
|
|
script_type=InputScriptType.SPENDWITNESS,
|
||
|
|
user_confirmation=True,
|
||
|
|
commitment_data=b"www.example.com" + int.to_bytes(1, _ROUND_ID_LEN, "big"),
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertTrue(self.authorization.check_get_ownership_proof(msg))
|
||
|
|
|
||
|
|
def test_approve_sign_tx(self):
|
||
|
|
|
||
|
|
msg = SignTx(outputs_count=10, inputs_count=21, coin_name=self.coin.coin_name, lock_time=0)
|
||
|
|
|
||
|
|
self.assertTrue(self.authorization.approve_sign_tx(msg, 10000))
|
||
|
|
self.assertTrue(self.authorization.approve_sign_tx(msg, 20000))
|
||
|
|
self.assertFalse(self.authorization.approve_sign_tx(msg, 10001))
|
||
|
|
self.assertTrue(self.authorization.approve_sign_tx(msg, 10000))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|