sha256test.py 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. #!/usr/bin/env python3
  2. import sys,os,hashlib
  3. from binascii import hexlify
  4. from mmgen.sha256 import Sha256
  5. random_rounds = int(sys.argv[1]) if len(sys.argv) == 2 else 500
  6. def msg(s): sys.stderr.write(s)
  7. def green(s): return '\033[32;1m' + s + '\033[0m'
  8. def compare_hashes(dlen,data):
  9. sha2 = hashlib.sha256(data).hexdigest().encode()
  10. # msg('Dlen {:<5} {}\r'.format(dlen,sha2))
  11. my_sha2 = Sha256(data).hexdigest()
  12. assert my_sha2 == sha2,'Hashes do not match!'
  13. def test_K():
  14. msg('Testing generated constants: ')
  15. K_ref = [1116352408,1899447441,-1245643825,-373957723,961987163,1508970993,-1841331548,-1424204075,-670586216,310598401,607225278,1426881987,1925078388,-2132889090,-1680079193,-1046744716,-459576895,-272742522,264347078,604807628,770255983,1249150122,1555081692,1996064986,-1740746414,-1473132947,-1341970488,-1084653625,-958395405,-710438585,113926993,338241895,666307205,773529912,1294757372,1396182291,1695183700,1986661051,-2117940946,-1838011259,-1564481375,-1474664885,-1035236496,-949202525,-778901479,-694614492,-200395387,275423344,430227734,506948616,659060556,883997877,958139571,1322822218,1537002063,1747873779,1955562222,2024104815,-2067236844,-1933114872,-1866530822,-1538233109,-1090935817,-965641998]
  16. def toSigned32(n): return ((n & 0xffffffff) ^ 0x80000000) - 0x80000000
  17. K_sig = [toSigned32(n) for n in Sha256.K]
  18. assert K_sig == K_ref,'Generated constants in K[] differ from reference value'
  19. msg('OK\n')
  20. def test_ref():
  21. inputs = (
  22. '','x','xa','the','the quick','the quick brown fox',
  23. '\x00','\x00\x00','\x00'*256,'\x00'*512,'\x00'*511,'\x00'*513,
  24. '\x0f','\x0f\x0f','\x0f'*256,'\x0f'*512,'\x0f'*511,'\x0f'*513,
  25. '\x0f\x0d','\x0e\x0e'*256,'\x00\x0f'*512,'\x0e\x0f'*511,'\x0a\x0d'*513
  26. )
  27. for i,data in enumerate(inputs):
  28. msg('\rTesting reference input data: {:4}/{} '.format(i+1,len(inputs)))
  29. compare_hashes(len(data),data.encode())
  30. msg('OK\n')
  31. def test_random(rounds):
  32. for i in range(rounds):
  33. if i+1 in (1,rounds) or not (i+1) % 10:
  34. msg('\rTesting random input data: {:4}/{} '.format(i+1,rounds))
  35. dlen = int(hexlify(os.urandom(4)),16) >> 18
  36. compare_hashes(dlen,os.urandom(dlen))
  37. msg('OK\n')
  38. msg(green('Testing MMGen implementation of Sha256()\n'))
  39. test_K()
  40. test_ref()
  41. test_random(random_rounds)