ut_seedsplit.py 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. #!/usr/bin/env python3
  2. """
  3. test/unit_tests_d/ut_seedsplit: seed splitting unit test for the MMGen suite
  4. """
  5. from mmgen.common import *
  6. class unit_test(object):
  7. def run_test(self,name,ut):
  8. from mmgen.seed import Seed
  9. from mmgen.seedsplit import SeedShareList,SeedShareIdx
  10. g.debug_subseed = bool(opt.verbose)
  11. def basic_ops(master_idx):
  12. test_data = {
  13. 'default': (
  14. (8,'4710FBF0','B3D9411B','2670E83D','D1FC57ED','AE49CABE','63FFBA62',0,0),
  15. (6,'9D07ABBD','AF5DC2F6','1A3BBDAC','2548AEE9','B94F7450','1F4E5A12',0,0),
  16. (4,'43670520','1F72C066','E5AA8DA1','A33966A0','D2BCE0A5','A568C315',0,0),
  17. ),
  18. 'φυβαρ': (
  19. (8,'4710FBF0','269D658C','9D25889E','6D730ECB','C61A963F','9FE99C05',0,0),
  20. (6,'9D07ABBD','4998B33E','F00CE041','C612BEE5','35CD3675','41B3BE61',0,0),
  21. (4,'43670520','77140076','EA82CB30','80F7AEDE','D168D768','77BE57AA',0,0),
  22. )
  23. }
  24. test_data_master = {
  25. '1': {
  26. 'default': (
  27. (8,'4710FBF0','6AE6177F','AC12090C','6AE6177F','3E87A907','7D1FEA56','BFEBFFFF','629A9808'),
  28. (4,'43670520','6739535C','ABF4DD38','6739535C','778E9C60','89CBCFD2','689FABF5','70BED76B'),
  29. ),
  30. 'φυβαρ': (
  31. (8,'4710FBF0','6AE6177F','AC5FA32E','6AE6177F','9777A750','C7CF2AFC','035AAACB','C777FBE4'),
  32. (4,'43670520','6739535C','37EBA2F5','6739535C','927549D2','29BADEE7','9CA73A03','313F5528'))
  33. },
  34. '5': {
  35. 'default': (
  36. (8,'4710FBF0','5EFAC3D6','B489167D','5EFAC3D6','BB004DC5','1A0381C0','4EA182E3','547FB2DC'),
  37. (4,'43670520','EE93DB0E','44962A7D','EE93DB0E','07339882','376A05B1','CE51D022','00149CA3'),
  38. ),
  39. 'φυβαρ': (
  40. (8,'4710FBF0','5EFAC3D6','A6E27EE3','5EFAC3D6','32C24668','B4C54297','1EC9B71B','8C5C6B1C'),
  41. (4,'43670520','EE93DB0E','B584E963','EE93DB0E','4BEA2AB2','4BEA65C7','140FC43F','BBD19461'))
  42. }
  43. }
  44. if master_idx:
  45. test_data = test_data_master[str(master_idx)]
  46. for id_str in (None,'default','φυβαρ'):
  47. msg_r(f'Testing basic ops (id_str={id_str!r}, master_idx={master_idx})...')
  48. vmsg('')
  49. for a,b,c,d,e,f,h,i,p in test_data[id_str if id_str is not None else 'default']:
  50. seed_bin = bytes.fromhex('deadbeef' * a)
  51. seed = Seed(seed_bin)
  52. assert seed.sid == b, seed.sid
  53. for share_count,j,k,l,m in ((2,c,c,d,i),(5,e,f,h,p)):
  54. shares = seed.split(share_count,id_str,master_idx)
  55. A = len(shares)
  56. assert A == share_count, A
  57. s = shares.format()
  58. vmsg_r(f'\n{s}')
  59. assert len(s.strip().split('\n')) == share_count+6, s
  60. if master_idx:
  61. A = shares.get_share_by_idx(1,base_seed=False).sid
  62. B = shares.get_share_by_seed_id(j,base_seed=False).sid
  63. assert A == B == m, A
  64. A = shares.get_share_by_idx(1,base_seed=True).sid
  65. B = shares.get_share_by_seed_id(j,base_seed=True).sid
  66. assert A == B == j, A
  67. A = shares.get_share_by_idx(share_count-1,base_seed=True).sid
  68. B = shares.get_share_by_seed_id(k,base_seed=True).sid
  69. assert A == B == k, A
  70. A = shares.get_share_by_idx(share_count).sid
  71. B = shares.get_share_by_seed_id(l).sid
  72. assert A == B == l, A
  73. A = shares.join().sid
  74. assert A == b, A
  75. if master_idx:
  76. slist = [shares.get_share_by_idx(i+1,base_seed=True) for i in range(len(shares))]
  77. A = Seed.join_shares(slist,master_idx,id_str).sid
  78. assert A == b, A
  79. msg('OK')
  80. def defaults_and_limits():
  81. msg_r('Testing defaults and limits...')
  82. seed_bin = bytes.fromhex('deadbeef' * 8)
  83. seed = Seed(seed_bin)
  84. shares = seed.split(SeedShareIdx.max_val)
  85. s = shares.format()
  86. # vmsg_r(f'\n{s}')
  87. assert len(s.strip().split('\n')) == 1030, s
  88. A = shares.get_share_by_idx(1024).sid
  89. B = shares.get_share_by_seed_id('4BA23728').sid
  90. assert A == '4BA23728', A
  91. assert B == '4BA23728', B
  92. A = shares.join().sid
  93. B = seed.sid
  94. assert A == B, A
  95. msg('OK')
  96. def collisions(seed_hex,ss_count,last_sid,collisions_chk,master_idx):
  97. msg_r(f'Testing Seed ID collisions ({ss_count} seed shares, master_idx={master_idx})...')
  98. vmsg('')
  99. seed_bin = bytes.fromhex(seed_hex)
  100. seed = Seed(seed_bin)
  101. SeedShareIdx.max_val = ss_count
  102. shares = seed.split(ss_count,master_idx=master_idx)
  103. A = shares.get_share_by_idx(ss_count).sid
  104. B = shares.get_share_by_seed_id(last_sid).sid
  105. assert A == last_sid, A
  106. assert B == last_sid, B
  107. assert shares.nonce_start == 0, shares.nonce_start
  108. collisions = 0
  109. for sid in shares.data['long']:
  110. collisions += shares.data['long'][sid][1]
  111. assert collisions == collisions_chk, collisions
  112. vmsg_r(f'{collisions} collisions, last_sid {last_sid}')
  113. msg('OK')
  114. def last_share_collisions():
  115. msg_r('Testing last share collisions with shortened Seed IDs')
  116. vmsg('')
  117. seed_bin = bytes.fromhex('2eadbeef'*8)
  118. seed = Seed(seed_bin)
  119. ssm_save = SeedShareIdx.max_val
  120. ssm = SeedShareIdx.max_val = 2048
  121. shares = SeedShareList(seed,count=ssm,id_str='foo',master_idx=1,debug_last_share=True)
  122. lsid = shares.last_share.sid
  123. collisions = shares.data['long'][lsid][1]
  124. assert collisions == 2, collisions
  125. assert lsid == 'B5B8AD09', lsid
  126. SeedShareIdx.max_val = ssm_save
  127. vmsg_r(f'{collisions} collisions, last_share sid {lsid}')
  128. msg('..OK')
  129. basic_ops(master_idx=None)
  130. basic_ops(master_idx=1)
  131. basic_ops(master_idx=5)
  132. defaults_and_limits()
  133. last_share_collisions()
  134. collisions('1dabcdef'*4,65535,'B5CBCE0A',3,master_idx=None)
  135. collisions('18abcdef'*4,65535,'FF03CE82',3,master_idx=1)
  136. return True