You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.7 KiB
49 lines
1.7 KiB
8 years ago
|
from common import *
|
||
9 years ago
|
|
||
9 years ago
|
from trezor.crypto import hashlib
|
||
9 years ago
|
|
||
7 years ago
|
|
||
9 years ago
|
class TestCryptoSha256(unittest.TestCase):
|
||
|
|
||
|
# vectors from http://www.di-mgt.com.au/sha_testvectors.html
|
||
7 years ago
|
vectors = [
|
||
|
(b'', 'e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'),
|
||
|
(b'abc', 'ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad'),
|
||
|
(b'abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq', '248d6a61d20638b8e5c026930c3e6039a33ce45964ff2167f6ecedd419db06c1'),
|
||
|
(b'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu', 'cf5b16a778af8380036ce59e7b0492370b249b11e8f07a51afac45037afee9d1'),
|
||
|
]
|
||
9 years ago
|
|
||
|
def test_digest(self):
|
||
7 years ago
|
for b, d in self.vectors:
|
||
|
self.assertEqual(hashlib.sha256(b).digest(), unhexlify(d))
|
||
9 years ago
|
|
||
|
def test_update(self):
|
||
7 years ago
|
for b, d in self.vectors:
|
||
|
x = hashlib.sha256()
|
||
|
x.update(b)
|
||
|
self.assertEqual(x.digest(), unhexlify(d))
|
||
9 years ago
|
|
||
9 years ago
|
x = hashlib.sha256()
|
||
9 years ago
|
for i in range(1000000):
|
||
|
x.update(b'a')
|
||
9 years ago
|
self.assertEqual(x.digest(), unhexlify('cdc76e5c9914fb9281a1c7e284d73e67f1809a48a497200e046d39ccc7112cd0'))
|
||
9 years ago
|
|
||
|
'''
|
||
9 years ago
|
x = hashlib.sha256()
|
||
9 years ago
|
for i in range(16777216):
|
||
|
x.update(b'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmno')
|
||
9 years ago
|
self.assertEqual(x.digest(), unhexlify('50e72a0e26442fe2552dc3938ac58658228c0cbfb1d2ca872ae435266fcd055e'))
|
||
9 years ago
|
'''
|
||
9 years ago
|
|
||
9 years ago
|
def test_digest_multi(self):
|
||
9 years ago
|
x = hashlib.sha256()
|
||
9 years ago
|
d0 = x.digest()
|
||
|
d1 = x.digest()
|
||
|
d2 = x.digest()
|
||
|
self.assertEqual(d0, d1)
|
||
|
self.assertEqual(d0, d2)
|
||
|
|
||
7 years ago
|
|
||
9 years ago
|
if __name__ == '__main__':
|
||
|
unittest.main()
|