mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-25 08:58:14 +00:00
e073e619c9
isort set to skip the first necessary "from common import *" line. A better solution would be to get rid of the need of this import in the future. [no changelog]
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
from common import * # isort:skip
|
|
|
|
from trezor.crypto import hashlib
|
|
|
|
|
|
class TestCryptoSha512(unittest.TestCase):
|
|
|
|
# vectors from http://www.di-mgt.com.au/sha_testvectors.html
|
|
vectors = [
|
|
(
|
|
b"",
|
|
"cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e",
|
|
),
|
|
(
|
|
b"abc",
|
|
"ddaf35a193617abacc417349ae20413112e6fa4e89a97ea20a9eeee64b55d39a2192992a274fc1a836ba3c23a3feebbd454d4423643ce80e2a9ac94fa54ca49f",
|
|
),
|
|
(
|
|
b"abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq",
|
|
"204a8fc6dda82f0a0ced7beb8e08a41657c16ef468b228a8279be331a703c33596fd15c13b1b07f9aa1d3bea57789ca031ad85c7a71dd70354ec631238ca3445",
|
|
),
|
|
(
|
|
b"abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu",
|
|
"8e959b75dae313da8cf4f72814fc143f8f7779c6eb9f7fa17299aeadb6889018501d289e4900f7e4331b99dec4b5433ac7d329eeb6dd26545e96e55b874be909",
|
|
),
|
|
]
|
|
|
|
def test_digest(self):
|
|
for b, d in self.vectors:
|
|
self.assertEqual(hashlib.sha512(b).digest(), unhexlify(d))
|
|
|
|
def test_update(self):
|
|
for b, d in self.vectors:
|
|
x = hashlib.sha512()
|
|
x.update(b)
|
|
self.assertEqual(x.digest(), unhexlify(d))
|
|
|
|
x = hashlib.sha512()
|
|
for i in range(1000000):
|
|
x.update(b"a")
|
|
self.assertEqual(
|
|
x.digest(),
|
|
unhexlify(
|
|
"e718483d0ce769644e2e42c7bc15b4638e1f98b13b2044285632a803afa973ebde0ff244877ea60a4cb0432ce577c31beb009c5c2c49aa2e4eadb217ad8cc09b"
|
|
),
|
|
)
|
|
|
|
"""
|
|
x = hashlib.sha512()
|
|
for i in range(16777216):
|
|
x.update(b'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmno')
|
|
self.assertEqual(x.digest(), unhexlify('b47c933421ea2db149ad6e10fce6c7f93d0752380180ffd7f4629a712134831d77be6091b819ed352c2967a2e2d4fa5050723c9630691f1a05a7281dbe6c1086'))
|
|
"""
|
|
|
|
def test_digest_multi(self):
|
|
x = hashlib.sha512()
|
|
d0 = x.digest()
|
|
d1 = x.digest()
|
|
d2 = x.digest()
|
|
self.assertEqual(d0, d1)
|
|
self.assertEqual(d0, d2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|