mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-10 09:39:00 +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]
367 lines
16 KiB
Python
367 lines
16 KiB
Python
from common import * # isort:skip
|
|
|
|
from trezor import wire
|
|
from trezor.enums import CardanoCertificateType
|
|
from trezor.messages import CardanoPoolParametersType, CardanoTxCertificate
|
|
|
|
from apps.common.paths import HARDENED
|
|
|
|
if not utils.BITCOIN_ONLY:
|
|
from apps.cardano import certificates
|
|
from apps.cardano.helpers import network_ids, protocol_magics
|
|
from apps.cardano.helpers.account_path_check import AccountPathChecker
|
|
|
|
|
|
@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
|
|
class TestCardanoCertificate(unittest.TestCase):
|
|
def test_validate_certificate(self):
|
|
valid_test_vectors = [
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
key_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DELEGATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DELEGATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DELEGATION,
|
|
key_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
key_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_POOL_REGISTRATION,
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
metadata=None,
|
|
),
|
|
),
|
|
]
|
|
|
|
invalid_test_vectors = [
|
|
# STAKE_REGISTRATION neither path or script_hash is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
),
|
|
# STAKE_REGISTRATION both path and script_hash are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
# STAKE_REGISTRATION both script_hash and key_hash are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
key_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
# STAKE_REGISTRATION pool is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
# STAKE_REGISTRATION pool parameters are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_REGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
),
|
|
),
|
|
# STAKE_DELEGATION neither path or script_hash is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DELEGATION,
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
# STAKE_DELEGATION both path and script_hash are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DELEGATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
# STAKE_DELEGATION both script_hash and key_hash are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DELEGATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
key_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
# STAKE_DELEGATION pool parameters are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DELEGATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
),
|
|
),
|
|
# STAKE_DEREGISTRATION neither path or script_hash is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
),
|
|
# STAKE_DEREGISTRATION both path and script_hash are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
# STAKE_DEREGISTRATION both script_hash and key_hash are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
key_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
),
|
|
# STAKE_DEREGISTRATION pool is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
),
|
|
# STAKE_DEREGISTRATION pool parameters are set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_DEREGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
),
|
|
),
|
|
# STAKE_POOL_REGISTRATION pool parameters are not set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_POOL_REGISTRATION,
|
|
),
|
|
# STAKE_POOL_REGISTRATION path is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_POOL_REGISTRATION,
|
|
path=[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0],
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
),
|
|
),
|
|
# STAKE_POOL_REGISTRATION script hash is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_POOL_REGISTRATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
),
|
|
),
|
|
# STAKE_POOL_REGISTRATION key hash is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_POOL_REGISTRATION,
|
|
key_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
),
|
|
),
|
|
# STAKE_POOL_REGISTRATION pool is set
|
|
CardanoTxCertificate(
|
|
type=CardanoCertificateType.STAKE_POOL_REGISTRATION,
|
|
script_hash=unhexlify(
|
|
"29fb5fd4aa8cadd6705acc8263cee0fc62edca5ac38db593fec2f9fd"
|
|
),
|
|
pool=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
pool_parameters=CardanoPoolParametersType(
|
|
pool_id=unhexlify(
|
|
"f61c42cbf7c8c53af3f520508212ad3e72f674f957fe23ff0acb4973"
|
|
),
|
|
vrf_key_hash=unhexlify(
|
|
"198890ad6c92e80fbdab554dda02da9fb49d001bbd96181f3e07f7a6ab0d0640"
|
|
),
|
|
pledge=500000000,
|
|
cost=340000000,
|
|
margin_numerator=1,
|
|
margin_denominator=2,
|
|
reward_account="stake1uya87zwnmax0v6nnn8ptqkl6ydx4522kpsc3l3wmf3yswygwx45el",
|
|
owners_count=1,
|
|
relays_count=1,
|
|
),
|
|
),
|
|
]
|
|
|
|
for certificate in valid_test_vectors:
|
|
certificates.validate(
|
|
certificate,
|
|
protocol_magics.MAINNET,
|
|
network_ids.MAINNET,
|
|
AccountPathChecker(),
|
|
)
|
|
|
|
for certificate in invalid_test_vectors:
|
|
with self.assertRaises(wire.ProcessError):
|
|
certificates.validate(
|
|
certificate,
|
|
protocol_magics.MAINNET,
|
|
network_ids.MAINNET,
|
|
AccountPathChecker(),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|