diff options
author | jaseg <git@jaseg.de> | 2020-12-21 16:26:57 +0100 |
---|---|---|
committer | jaseg <git@jaseg.de> | 2020-12-21 16:26:57 +0100 |
commit | 7b85ba8d4fb34e76d34a2d581e89e856aa471cf5 (patch) | |
tree | 1d5b1817578a5e588469b2203ed394ac07f7b3bf /fw/tools/crypto_test_runner.py | |
parent | 7c0a0f40e98df545386e2620c5b08cff7c29141f (diff) | |
download | minikbd-7b85ba8d4fb34e76d34a2d581e89e856aa471cf5.tar.gz minikbd-7b85ba8d4fb34e76d34a2d581e89e856aa471cf5.tar.bz2 minikbd-7b85ba8d4fb34e76d34a2d581e89e856aa471cf5.zip |
Move fw into direct subdir
Diffstat (limited to 'fw/tools/crypto_test_runner.py')
-rw-r--r-- | fw/tools/crypto_test_runner.py | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/fw/tools/crypto_test_runner.py b/fw/tools/crypto_test_runner.py new file mode 100644 index 0000000..34c8b59 --- /dev/null +++ b/fw/tools/crypto_test_runner.py @@ -0,0 +1,46 @@ +#!/usr/bin/env python3 +import subprocess +from os import path +import binascii +import re + +import presig_gen + +def do_test(domain, value, height, root_key, binary, expect_fail=False): + auth = presig_gen.gen_at_height(domain, value, height, root_key) + auth = binascii.hexlify(auth).decode() + + output = subprocess.check_output([binary, auth]) + *lines, rc_line = output.decode().splitlines() + rc = int(re.match('^rc=(\d+)$', rc_line).group(1)) + assert expect_fail == (rc == 0) + +def run_tests(root_key, max_height, binary): + for domain, value in { + 'all': 'all', + 'vendor': presig_gen.TEST_VENDOR, + 'series': presig_gen.TEST_SERIES, + 'country': presig_gen.TEST_COUNTRY, + 'region': presig_gen.TEST_REGION, + }.items(): + for height in range(max_height): + do_test(domain, value, height, root_key, binary) + do_test(domain, 'fail', height, root_key, binary, expect_fail=True) + do_test('fail', 'fail', height, root_key, binary, expect_fail=True) + do_test('', '', height, root_key, binary, expect_fail=True) + do_test(domain, value, max_height, root_key, binary, expect_fail=True) + do_test(domain, value, max_height+1, root_key, binary, expect_fail=True) + +if __name__ == '__main__': + import argparse + parser = argparse.ArgumentParser() + parser.add_argument('keyfile', help='Root key file') + parser.add_argument('max_height', type=int, default=8, nargs='?', help='Height of generated prekeys') + default_binary = path.abspath(path.join(path.dirname(__file__), '../build/tools/crypto_test')) + parser.add_argument('binary', default=default_binary, nargs='?', help='crypto_test binary to use') + args = parser.parse_args() + + with open(args.keyfile, 'r') as f: + root_key = binascii.unhexlify(f.read().strip()) + + run_tests(root_key, args.max_height, args.binary) |