ut_subseed.py 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  1. #!/usr/bin/env python3
  2. """
  3. test/unit_tests_d/ut_subseed: subseed 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.subseed import SubSeedList,SubSeedIdxRange
  10. nSubseeds = SubSeedList.dfl_len
  11. def basic_ops():
  12. msg_r('Testing basic ops...')
  13. for a,b,c,d,e,f,h in (
  14. (8,'4710FBF0','0C1B0615','803B165C','2669AC64',256,'10L'),
  15. (6,'9D07ABBD','EBA9C33F','20787E6A','192E2AA2',192,'10L'),
  16. (4,'43670520','04A4CCB3','B5F21D7B','C1934CFF',128,'10L'),
  17. ):
  18. seed_bin = bytes.fromhex('deadbeef' * a)
  19. seed = Seed(seed_bin)
  20. assert seed.sid == b, seed.sid
  21. subseed = seed.subseed('2s')
  22. assert subseed.sid == c, subseed.sid
  23. subseed = seed.subseed('3')
  24. assert subseed.sid == d, subseed.sid
  25. subseed = seed.subseed_by_seed_id(e)
  26. assert subseed.bitlen == f, subseed.bitlen
  27. assert subseed.sid == e, subseed.sid
  28. assert subseed.idx == 10, subseed.idx
  29. assert subseed.ss_idx == h, subseed.ss_idx
  30. seed2 = Seed(seed_bin)
  31. ss2_list = seed2.subseeds
  32. seed2.subseeds._generate(1)
  33. assert len(ss2_list) == 1, len(ss2_list)
  34. seed2.subseeds._generate(1) # do nothing
  35. seed2.subseeds._generate(2) # append one item
  36. seed2.subseeds._generate(5)
  37. assert len(ss2_list) == 5, len(ss2_list)
  38. seed2.subseeds._generate(3) # do nothing
  39. assert len(ss2_list) == 5, len(ss2_list)
  40. seed2.subseeds._generate(10)
  41. assert len(ss2_list) == 10, len(ss2_list)
  42. assert seed.pfmt() == seed2.pfmt()
  43. s = seed.subseeds.format(1,nSubseeds)
  44. s_lines = s.strip().split('\n')
  45. assert len(s_lines) == nSubseeds + 4, s
  46. a = seed.subseed('2L').sid
  47. b = [e for e in s_lines if ' 2L:' in e][0].strip().split()[1]
  48. assert a == b, b
  49. c = seed.subseed('2').sid
  50. assert c == a, c
  51. a = seed.subseed('5S').sid
  52. b = [e for e in s_lines if ' 5S:' in e][0].strip().split()[3]
  53. assert a == b, b
  54. s = seed.subseeds.format(nSubseeds+1,nSubseeds+2)
  55. s_lines = s.strip().split('\n')
  56. assert len(s_lines) == 6, s
  57. ss_idx = str(nSubseeds+2) + 'S'
  58. a = seed.subseed(ss_idx).sid
  59. b = [e for e in s_lines if f' {ss_idx}:' in e][0].strip().split()[3]
  60. assert a == b, b
  61. s = seed.subseeds.format(1,10)
  62. s_lines = s.strip().split('\n')
  63. assert len(s_lines) == 14, s
  64. vmsg_r(f'\n{s}')
  65. msg('OK')
  66. def defaults_and_limits():
  67. msg_r('Testing defaults and limits...')
  68. seed_bin = bytes.fromhex('deadbeef' * 8)
  69. seed = Seed(seed_bin,nSubseeds=11)
  70. seed.subseeds._generate()
  71. ss = seed.subseeds
  72. assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
  73. assert len(ss) == 11, len(ss)
  74. seed = Seed(seed_bin)
  75. seed.subseeds._generate()
  76. ss = seed.subseeds
  77. assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
  78. assert len(ss) == nSubseeds, len(ss)
  79. seed = Seed(seed_bin)
  80. seed.subseed_by_seed_id('EEEEEEEE')
  81. ss = seed.subseeds
  82. assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
  83. assert len(ss) == nSubseeds, len(ss)
  84. seed = Seed(seed_bin)
  85. subseed = seed.subseed_by_seed_id('803B165C')
  86. assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
  87. assert subseed.sid == '803B165C', subseed.sid
  88. assert subseed.idx == 3, subseed.idx
  89. seed = Seed(seed_bin)
  90. subseed = seed.subseed_by_seed_id('803B165C',last_idx=1)
  91. assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
  92. assert subseed == None, subseed
  93. r = SubSeedIdxRange('1-5')
  94. r2 = SubSeedIdxRange(1,5)
  95. assert r2 == r, r2
  96. assert r == (r.first,r.last), r
  97. assert r.first == 1, r.first
  98. assert r.last == 5, r.last
  99. assert r.items == [1,2,3,4,5], r.items
  100. assert list(r.iterate()) == r.items, list(r.iterate())
  101. r = SubSeedIdxRange('22')
  102. r2 = SubSeedIdxRange(22,22)
  103. assert r2 == r, r2
  104. assert r == (r.first,r.last), r
  105. assert r.first == 22, r.first
  106. assert r.last == 22, r.last
  107. assert r.items == [22], r
  108. assert list(r.iterate()) == r.items, list(r.iterate())
  109. r = SubSeedIdxRange('3-3')
  110. assert r.items == [3], r.items
  111. r = SubSeedIdxRange(f'{nSubseeds-1}-{nSubseeds}')
  112. assert r.items == [nSubseeds-1,nSubseeds], r.items
  113. for n,e in enumerate(SubSeedIdxRange('1-5').iterate(),1):
  114. assert n == e, e
  115. assert n == 5, n
  116. msg('OK')
  117. def collisions():
  118. ss_count,ltr,last_sid,collisions_chk = (
  119. (SubSeedIdxRange.max_idx,'S','2788F26B',470),
  120. (49509,'L','8D1FE500',2)
  121. )[bool(opt.fast)]
  122. last_idx = str(ss_count) + ltr
  123. msg_r(f'Testing Seed ID collisions ({ss_count} subseed pairs)...')
  124. seed_bin = bytes.fromhex('12abcdef' * 8) # 95B3D78D
  125. seed = Seed(seed_bin)
  126. seed.subseeds._generate(ss_count)
  127. ss = seed.subseeds
  128. assert seed.subseed(last_idx).sid == last_sid, seed.subseed(last_idx).sid
  129. for sid in ss.data['long']:
  130. # msg(sid)
  131. assert sid not in ss.data['short']
  132. collisions = 0
  133. for k in ('short','long'):
  134. for sid in ss.data[k]:
  135. collisions += ss.data[k][sid][1]
  136. assert collisions == collisions_chk, collisions
  137. vmsg_r(f'\n{collisions} collisions, last_sid {last_sid}')
  138. msg('OK')
  139. basic_ops()
  140. defaults_and_limits()
  141. collisions()
  142. return True