1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
# Import libnacl libs
import libnacl.kx
# Import python libs
import unittest
class TestKX(unittest.TestCase):
'''
'''
def test_exchange_key(self):
'''
'''
msg = b'You\'ve got two empty halves of coconut and you\'re bangin\' \'em together.'
ad = b'A Duck!'
# Make Bob and Alice Exchange Keys
bob = libnacl.kx.ExchangeKey()
alice = libnacl.kx.ExchangeKey()
# Encrypt with bob as clientm alic as server
bob_ctxt = bob.encrypt_client(alice.kx_pk, msg, ad)
self.assertNotEqual(msg, bob_ctxt)
bclear = alice.decrypt_server(bob.kx_pk, bob_ctxt, len(ad))
self.assertEqual(ad, bob_ctxt[:len(ad)])
self.assertEqual(msg, bclear)
alice_ctxt = alice.encrypt_server(bob.kx_pk, msg, ad)
aclear = bob.decrypt_client(alice.kx_pk, alice_ctxt, len(ad))
self.assertEqual(ad, alice_ctxt[:len(ad)])
self.assertEqual(msg, aclear)
self.assertNotEqual(msg, alice_ctxt)
# Encrypt with Alice as client bob as server
alice_ctxt = alice.encrypt_client(bob.kx_pk, msg, ad)
self.assertNotEqual(msg, alice_ctxt)
bclear = bob.decrypt_server(alice.kx_pk, alice_ctxt, len(ad))
self.assertEqual(ad, alice_ctxt[:len(ad)])
self.assertEqual(msg, bclear)
|
