mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-10-19 14:30:31 +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]
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from common import * # isort:skip
|
|
|
|
from trezor.crypto.curve import curve25519
|
|
|
|
|
|
class TestCryptoCurve25519(unittest.TestCase):
|
|
|
|
vectors = [
|
|
(
|
|
"38c9d9b17911de26ed812f5cc19c0029e8d016bcbc6078bc9db2af33f1761e4a",
|
|
"311b6248af8dabec5cc81eac5bf229925f6d218a12e0547fb1856e015cc76f5d",
|
|
"a93dbdb23e5c99da743e203bd391af79f2b83fb8d0fd6ec813371c71f08f2d4d",
|
|
),
|
|
]
|
|
|
|
def test_generate_secret(self):
|
|
for _ in range(100):
|
|
sk = curve25519.generate_secret()
|
|
self.assertTrue(len(sk) == 32)
|
|
self.assertTrue(sk[0] & 7 == 0 and sk[31] & 128 == 0 and sk[31] & 64 == 64)
|
|
|
|
def test_multiply(self):
|
|
for sk, pk, session in self.vectors:
|
|
session2 = curve25519.multiply(unhexlify(sk), unhexlify(pk))
|
|
self.assertEqual(session2, unhexlify(session))
|
|
|
|
def test_multiply_random(self):
|
|
for _ in range(100):
|
|
sk1 = curve25519.generate_secret()
|
|
sk2 = curve25519.generate_secret()
|
|
pk1 = curve25519.publickey(sk1)
|
|
pk2 = curve25519.publickey(sk2)
|
|
session1 = curve25519.multiply(sk1, pk2)
|
|
session2 = curve25519.multiply(sk2, pk1)
|
|
self.assertEqual(session1, session2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|