mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-12 02:31:05 +00:00
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
from common import *
|
|
|
|
from trezor.crypto import hashlib
|
|
|
|
|
|
class TestCryptoRipemd160(unittest.TestCase):
|
|
|
|
# vectors from http://homes.esat.kuleuven.be/~bosselae/ripemd160.html
|
|
vectors = [
|
|
(b'', '9c1185a5c5e9fc54612808977ee8f548b2258d31'),
|
|
(b'a', '0bdc9d2d256b3ee9daae347be6f4dc835a467ffe'),
|
|
(b'abc', '8eb208f7e05d987a9b044a8e98c6b087f15a0bfc'),
|
|
(b'message digest', '5d0689ef49d2fae572b881b123a85ffa21595f36'),
|
|
(b'abcdefghijklmnopqrstuvwxyz', 'f71c27109c692c1b56bbdceb5b9d2865b3708dbc'),
|
|
(b'abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq', '12a053384a9c0c88e405a06c27dcf49ada62eb2b'),
|
|
(b'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789', 'b0e20b6e3116640286ed3a87a5713079b21f5189'),
|
|
(b'12345678901234567890123456789012345678901234567890123456789012345678901234567890', '9b752e45573d4b39f4dbd3323cab82bf63326bfb'),
|
|
]
|
|
|
|
def test_digest(self):
|
|
for b, d in self.vectors:
|
|
self.assertEqual(hashlib.ripemd160(b).digest(), unhexlify(d))
|
|
|
|
def test_update(self):
|
|
for b, d in self.vectors:
|
|
x = hashlib.ripemd160()
|
|
x.update(b)
|
|
self.assertEqual(x.digest(), unhexlify(d))
|
|
|
|
x = hashlib.ripemd160()
|
|
for i in range(8):
|
|
x.update(b'1234567890')
|
|
self.assertEqual(x.digest(), unhexlify('9b752e45573d4b39f4dbd3323cab82bf63326bfb'))
|
|
|
|
def test_digest_multi(self):
|
|
x = hashlib.ripemd160()
|
|
d0 = x.digest()
|
|
d1 = x.digest()
|
|
d2 = x.digest()
|
|
self.assertEqual(d0, d1)
|
|
self.assertEqual(d0, d2)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|