ut_gen.py 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. #!/usr/bin/env python3
  2. """
  3. test.unit_tests_d.ut_gen: key/address generation unit tests for the MMGen suite
  4. """
  5. from mmgen.common import *
  6. from mmgen.protocol import init_proto
  7. from mmgen.key import PrivKey
  8. from mmgen.addr import MMGenAddrType
  9. from mmgen.addrgen import KeyGenerator,AddrGenerator
  10. from mmgen.keygen import get_backends
  11. # TODO: add viewkey checks
  12. vectors = { # from tooltest2
  13. 'btc': ( (
  14. '5HwzecKMWD82ppJK3qMKpC7ohXXAwcyAN5VgdJ9PLFaAzpBG4sX',
  15. '1C5VPtgq9xQ6AcTgMAR3J6GDrs72HC4pS1',
  16. 'legacy'
  17. ), (
  18. 'KwojSzt1VvW343mQfWQi3J537siAt5ktL2qbuCg1ZyKR8BLQ6UJm',
  19. '1Kz9fVSUMshzPejpzW9D95kScgA3rY6QxF',
  20. 'compressed'
  21. ), (
  22. 'KwojSzt1VvW343mQfWQi3J537siAt5ktL2qbuCg1ZyKR8BLQ6UJm',
  23. '3AhjTiWHhVJAi1s5CfKMcLzYps12x3gZhg',
  24. 'segwit'
  25. ), (
  26. 'KwojSzt1VvW343mQfWQi3J537siAt5ktL2qbuCg1ZyKR8BLQ6UJm',
  27. 'bc1q6pqnfwwakuuejpm9w52ds342f9d5u36v0qnz7c',
  28. 'bech32' ),
  29. ),
  30. 'eth': ( (
  31. '0000000000000000000000000000000000000000000000000000000000000001',
  32. '7e5f4552091a69125d5dfcb7b8c2659029395bdf',
  33. 'ethereum',
  34. ), ),
  35. 'xmr': ( (
  36. '0000000000000000000000000000000000000000000000000000000000000001',
  37. '42nsXK8WbVGTNayQ6Kjw5UdgqbQY5KCCufdxdCgF7NgTfjC69Mna7DJSYyie77hZTQ8H92G2HwgFhgEUYnDzrnLnQdF28r3',
  38. 'monero',
  39. ), ),
  40. 'zec': ( (
  41. 'SKxny894fJe2rmZjeuoE6GVfNkWoXfPp8337VrLLNWG56FjqVUYR',
  42. 'zceQDpyNwek7dKqF5ZuFGj7YrNVxh7X1aPkrVxDLVxWSiZAFDEuy5C7XNV8VhyZ3ghTPQ61xjCGiyLT3wqpiN1Yi6mdmaCq',
  43. 'zcash_z',
  44. ), ),
  45. }
  46. def do_test(proto,wif,addr_chk,addr_type,internal_keccak):
  47. if internal_keccak:
  48. opt.use_internal_keccak_module = True
  49. add_msg = ' (internal keccak module)'
  50. else:
  51. add_msg = ''
  52. at = MMGenAddrType(proto,addr_type)
  53. privkey = PrivKey(proto,wif=wif)
  54. for n,backend in enumerate(get_backends(at.pubkey_type)):
  55. kg = KeyGenerator(proto,at.pubkey_type,n+1)
  56. qmsg(blue(f' Testing backend {backend!r} for addr type {addr_type!r}{add_msg}'))
  57. data = kg.gen_data(privkey)
  58. for k,v in data._asdict().items():
  59. if v and k in ('pubkey','viewkey_bytes'):
  60. qmsg(f' {k+":":19} {v.hex()}')
  61. ag = AddrGenerator(proto,addr_type)
  62. addr = ag.to_addr(data)
  63. qmsg(f' addr: {addr}\n')
  64. assert addr == addr_chk, f'{addr} != {addr_chk}'
  65. opt.use_internal_keccak_module = False
  66. def do_tests(coin,internal_keccak=False):
  67. proto = init_proto(coin)
  68. for wif,addr,addr_type in vectors[coin]:
  69. do_test(proto,wif,addr,addr_type,internal_keccak)
  70. return True
  71. class unit_tests:
  72. altcoin_deps = ('eth','xmr','zec')
  73. def btc(self,name,ut):
  74. return do_tests('btc')
  75. def eth(self,name,ut):
  76. do_tests('eth')
  77. return do_tests('eth',internal_keccak=True)
  78. def xmr(self,name,ut):
  79. if not opt.fast:
  80. do_tests('xmr')
  81. return do_tests('xmr',internal_keccak=True)
  82. def zec(self,name,ut):
  83. return do_tests('zec')