2016-04-20 17:40:08 +00:00
|
|
|
import sys
|
|
|
|
sys.path.append('..')
|
|
|
|
sys.path.append('../lib')
|
|
|
|
import unittest
|
2016-04-26 23:08:39 +00:00
|
|
|
from ubinascii import unhexlify
|
2016-04-20 17:40:08 +00:00
|
|
|
|
|
|
|
from trezor.crypto import hashlib
|
|
|
|
|
|
|
|
class TestCryptoSha3_256(unittest.TestCase):
|
|
|
|
|
|
|
|
# vectors from http://www.di-mgt.com.au/sha_testvectors.html
|
|
|
|
|
|
|
|
def test_digest(self):
|
2016-04-26 23:08:39 +00:00
|
|
|
self.assertEqual(hashlib.sha3_256(b'').digest(), unhexlify('a7ffc6f8bf1ed76651c14756a061d662f580ff4de43b49fa82d80a4b80f8434a'))
|
|
|
|
self.assertEqual(hashlib.sha3_256(b'abc').digest(), unhexlify('3a985da74fe225b2045c172d6bd390bd855f086e3e9d525b46bfe24511431532'))
|
|
|
|
self.assertEqual(hashlib.sha3_256(b'abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq').digest(), unhexlify('41c0dba2a9d6240849100376a8235e2c82e1b9998a999e21db32dd97496d3376'))
|
|
|
|
self.assertEqual(hashlib.sha3_256(b'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu').digest(), unhexlify('916f6061fe879741ca6469b43971dfdb28b1a32dc36cb3254e812be27aad1d18'))
|
2016-04-20 17:40:08 +00:00
|
|
|
|
|
|
|
def test_update(self):
|
|
|
|
x = hashlib.sha3_256()
|
2016-04-26 23:08:39 +00:00
|
|
|
self.assertEqual(x.digest(), unhexlify('a7ffc6f8bf1ed76651c14756a061d662f580ff4de43b49fa82d80a4b80f8434a'))
|
2016-04-20 17:40:08 +00:00
|
|
|
|
|
|
|
x = hashlib.sha3_256()
|
|
|
|
x.update(b'abc')
|
2016-04-26 23:08:39 +00:00
|
|
|
self.assertEqual(x.digest(), unhexlify('3a985da74fe225b2045c172d6bd390bd855f086e3e9d525b46bfe24511431532'))
|
2016-04-20 17:40:08 +00:00
|
|
|
|
|
|
|
x = hashlib.sha3_256()
|
|
|
|
x.update(b'abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq')
|
2016-04-26 23:08:39 +00:00
|
|
|
self.assertEqual(x.digest(), unhexlify('41c0dba2a9d6240849100376a8235e2c82e1b9998a999e21db32dd97496d3376'))
|
2016-04-20 17:40:08 +00:00
|
|
|
|
|
|
|
x = hashlib.sha3_256()
|
|
|
|
x.update(b'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu')
|
2016-04-26 23:08:39 +00:00
|
|
|
self.assertEqual(x.digest(), unhexlify('916f6061fe879741ca6469b43971dfdb28b1a32dc36cb3254e812be27aad1d18'))
|
2016-04-20 17:40:08 +00:00
|
|
|
|
|
|
|
x = hashlib.sha3_256()
|
|
|
|
for i in range(1000000):
|
|
|
|
x.update(b'a')
|
2016-04-26 23:08:39 +00:00
|
|
|
self.assertEqual(x.digest(), unhexlify('5c8875ae474a3634ba4fd55ec85bffd661f32aca75c6d699d0cdcb6c115891c1'))
|
2016-04-20 17:40:08 +00:00
|
|
|
|
|
|
|
'''
|
|
|
|
x = hashlib.sha3_256()
|
|
|
|
for i in range(16777216):
|
|
|
|
x.update(b'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmno')
|
2016-04-26 23:08:39 +00:00
|
|
|
self.assertEqual(x.digest(), unhexlify('ecbbc42cbf296603acb2c6bc0410ef4378bafb24b710357f12df607758b33e2b'))
|
2016-04-20 17:40:08 +00:00
|
|
|
'''
|
|
|
|
|
|
|
|
def test_digest_multi(self):
|
|
|
|
x = hashlib.sha3_256()
|
|
|
|
d0 = x.digest()
|
|
|
|
d1 = x.digest()
|
|
|
|
d2 = x.digest()
|
|
|
|
self.assertEqual(d0, d1)
|
|
|
|
self.assertEqual(d0, d2)
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|