ut_bip39.py 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. #!/usr/bin/env python3
  2. """
  3. test/unit_tests_d/ut_bip39: BIP39 unit test for the MMGen suite
  4. """
  5. from mmgen.common import *
  6. class unit_test(object):
  7. vectors = (
  8. ( "00000000000000000000000000000000",
  9. "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about"
  10. ),
  11. ( "7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f",
  12. "legal winner thank year wave sausage worth useful legal winner thank yellow"
  13. ),
  14. ( "80808080808080808080808080808080",
  15. "letter advice cage absurd amount doctor acoustic avoid letter advice cage above"
  16. ),
  17. ( "ffffffffffffffffffffffffffffffff",
  18. "zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo wrong"
  19. ),
  20. ( "000000000000000000000000000000000000000000000000",
  21. "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon agent"
  22. ),
  23. ( "7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f",
  24. "legal winner thank year wave sausage worth useful legal winner thank year wave sausage worth useful legal will"
  25. ),
  26. ( "808080808080808080808080808080808080808080808080",
  27. "letter advice cage absurd amount doctor acoustic avoid letter advice cage absurd amount doctor acoustic avoid letter always"
  28. ),
  29. ( "ffffffffffffffffffffffffffffffffffffffffffffffff",
  30. "zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo when"
  31. ),
  32. ( "0000000000000000000000000000000000000000000000000000000000000000",
  33. "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon art"
  34. ),
  35. ( "7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f7f",
  36. "legal winner thank year wave sausage worth useful legal winner thank year wave sausage worth useful legal winner thank year wave sausage worth title"
  37. ),
  38. ( "8080808080808080808080808080808080808080808080808080808080808080",
  39. "letter advice cage absurd amount doctor acoustic avoid letter advice cage absurd amount doctor acoustic avoid letter advice cage absurd amount doctor acoustic bless"
  40. ),
  41. ( "ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff",
  42. "zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo vote"
  43. ),
  44. ( "9e885d952ad362caeb4efe34a8e91bd2",
  45. "ozone drill grab fiber curtain grace pudding thank cruise elder eight picnic"
  46. ),
  47. ( "6610b25967cdcca9d59875f5cb50b0ea75433311869e930b",
  48. "gravity machine north sort system female filter attitude volume fold club stay feature office ecology stable narrow fog"
  49. ),
  50. ( "68a79eaca2324873eacc50cb9c6eca8cc68ea5d936f98787c60c7ebc74e6ce7c",
  51. "hamster diagram private dutch cause delay private meat slide toddler razor book happy fancy gospel tennis maple dilemma loan word shrug inflict delay length"
  52. ),
  53. ( "c0ba5a8e914111210f2bd131f3d5e08d",
  54. "scheme spot photo card baby mountain device kick cradle pact join borrow"
  55. ),
  56. ( "6d9be1ee6ebd27a258115aad99b7317b9c8d28b6d76431c3",
  57. "horn tenant knee talent sponsor spell gate clip pulse soap slush warm silver nephew swap uncle crack brave"
  58. ),
  59. ( "9f6a2878b2520799a44ef18bc7df394e7061a224d2c33cd015b157d746869863",
  60. "panda eyebrow bullet gorilla call smoke muffin taste mesh discover soft ostrich alcohol speed nation flash devote level hobby quick inner drive ghost inside"
  61. ),
  62. ( "23db8160a31d3e0dca3688ed941adbf3",
  63. "cat swing flag economy stadium alone churn speed unique patch report train"
  64. ),
  65. ( "8197a4a47f0425faeaa69deebc05ca29c0a5b5cc76ceacc0",
  66. "light rule cinnamon wrap drastic word pride squirrel upgrade then income fatal apart sustain crack supply proud access"
  67. ),
  68. ( "066dca1a2bb7e8a1db2832148ce9933eea0f3ac9548d793112d9a95c9407efad",
  69. "all hour make first leader extend hole alien behind guard gospel lava path output census museum junior mass reopen famous sing advance salt reform"
  70. ),
  71. ( "f30f8c1da665478f49b001d94c5fc452",
  72. "vessel ladder alter error federal sibling chat ability sun glass valve picture"
  73. ),
  74. ( "c10ec20dc3cd9f652c7fac2f1230f7a3c828389a14392f05",
  75. "scissors invite lock maple supreme raw rapid void congress muscle digital elegant little brisk hair mango congress clump"
  76. ),
  77. ( "f585c11aec520db57dd353c69554b21a89b20fb0650966fa0a9d6f74fd989d8f",
  78. "void come effort suffer camp survey warrior heavy shoot primary clutch crush open amazing screen patrol group space point ten exist slush involve unfold"
  79. )
  80. )
  81. def run_test(self,name,ut):
  82. msg_r('Testing BIP39 conversion routines...')
  83. qmsg('')
  84. from mmgen.bip39 import bip39
  85. b = bip39()
  86. b.check_wordlist()
  87. vmsg('')
  88. qmsg('Checking seed to mnemonic conversion:')
  89. for v in self.vectors:
  90. chk = tuple(v[1].split())
  91. vmsg(' '+v[1])
  92. res = b.fromhex( v[0] )
  93. assert res == chk, f'mismatch:\nres: {res}\nchk: {chk}'
  94. vmsg('')
  95. qmsg('Checking mnemonic to seed conversion:')
  96. for v in self.vectors:
  97. chk = v[0]
  98. vmsg(' '+chk)
  99. res = b.tohex( v[1].split() )
  100. assert res == chk, f'mismatch:\nres: {res}\nchk: {chk}'
  101. vmsg('')
  102. qmsg('Checking error handling:')
  103. good_mn = "zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo wrong".split()
  104. bad_len_mn = "zoo zoo zoo".split()
  105. bad_chksum_mn = "zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo".split()
  106. bad_word_mn = "admire zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo zoo".split()
  107. bad_seed = 'deadbeef'
  108. good_seed = 'deadbeef' * 4
  109. th = b.tohex
  110. fh = b.fromhex
  111. bad_data = (
  112. ('hex', 'AssertionError', 'not a hexadecimal',lambda:fh('xx')),
  113. ('seed len', 'AssertionError', 'invalid seed bit', lambda:fh(bad_seed)),
  114. ('mnemonic type', 'AssertionError', 'must be list', lambda:th('string')),
  115. ('arg (tostr=True)', 'AssertionError', "'tostr' must be", lambda:fh(good_seed,tostr=True)),
  116. ('pad len (fromhex)','AssertionError', "invalid 'pad' arg",lambda:fh(good_seed,pad=23)),
  117. ('pad len (tohex)', 'AssertionError', "invalid 'pad' arg",lambda:th(good_mn,pad=23)),
  118. ('word', 'MnemonicError', "not in the BIP39", lambda:th(bad_word_mn)),
  119. ('checksum', 'MnemonicError', "checksum", lambda:th(bad_chksum_mn)),
  120. ('seed phrase len', 'MnemonicError', "phrase len", lambda:th(bad_len_mn)),
  121. )
  122. ut.process_bad_data(bad_data)
  123. vmsg('')
  124. msg('OK')
  125. return True