ut_subseed.py 5.0 KB

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