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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#!/usr/bin/env python3
import crypto_aead
class DecryptionError(Exception):
def __init__(self):
super().__init__('crypto_aead_decrypt did not recover the plaintext')
MAX_MESSAGE_LENGTH = 32
MAX_ASSOCIATED_DATA_LENGTH = 32
def print_bstr(output, label, buf):
print('{l} = {b}'.format(l=label, b=buf.hex().upper()), file=output)
def generate_test_vectors():
count = 1
filename = 'LWC_AEAD_KAT_{key}_{npub}.txt'.format(
key=crypto_aead.KEYBYTES*8, npub=crypto_aead.NPUBBYTES*8
)
npub = bytes(range(crypto_aead.NPUBBYTES))
key = bytes(range(crypto_aead.KEYBYTES))
with open(filename, 'w') as output:
for mlen in range(MAX_MESSAGE_LENGTH+1):
for adlen in range(MAX_ASSOCIATED_DATA_LENGTH+1):
msg = bytes(range(mlen))
ad = bytes(range(adlen))
print('Count = {c}'.format(c=count), file=output)
count += 1
print_bstr(output, 'Key', key)
print_bstr(output, 'Nonce', npub)
print_bstr(output, 'PT', msg)
print_bstr(output, 'AD', ad)
ct = crypto_aead.encrypt(msg, ad, npub, key)
print_bstr(output, 'CT', ct)
msg2 = crypto_aead.decrypt(ct, ad, npub, key)
if msg != msg2:
raise DecryptionError()
print(file=output)
if __name__ == '__main__':
generate_test_vectors()
|
