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
|
#!/usr/bin/env python2
import binascii
import sha3
def main():
test_vectors = {
'Short': 'abc',
'Medium': 'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu',
'Long': 'a' * 1000000
}
hash_funcs = {'SHAKE128': sha3.SHAKE128, 'SHAKE256': sha3.SHAKE256}
output_lens = [31, 63]
for hash_name, hash_func in hash_funcs.items():
for vector_name, vector_value in test_vectors.items():
for output_len in output_lens:
print('%s with %s UTF-8 Input and %d bit Output: %s' % (
hash_name,
vector_name,
output_len * 8,
binascii.hexlify(hash_func(vector_value.encode(), output_len)).decode()
)
)
print('%s with %s UTF-16BE Input and %d bit Output: %s' % (
hash_name,
vector_name,
output_len * 8,
binascii.hexlify(hash_func(vector_value.encode('UTF-16BE'), output_len)).decode()
)
)
print('%s with %s UTF-16LE Input and %d bit Output: %s' % (
hash_name,
vector_name,
output_len * 8,
binascii.hexlify(hash_func(vector_value.encode('UTF-16LE'), output_len)).decode()
)
)
if ('__main__' == __name__):
main()
|
