diff options
Diffstat (limited to 'test/modes_internal_test.c')
-rw-r--r-- | test/modes_internal_test.c | 33 |
1 files changed, 11 insertions, 22 deletions
diff --git a/test/modes_internal_test.c b/test/modes_internal_test.c index b3a2b102a4..ea463abae5 100644 --- a/test/modes_internal_test.c +++ b/test/modes_internal_test.c @@ -1,5 +1,5 @@ /* - * Copyright 2016 The OpenSSL Project Authors. All Rights Reserved. + * Copyright 2016-2017 The OpenSSL Project Authors. All Rights Reserved. * * Licensed under the OpenSSL license (the "License"). You may not use * this file except in compliance with the License. You can obtain a copy @@ -949,33 +949,22 @@ static void benchmark_gcm128(const unsigned char *K, size_t Klen, #endif } -int test_main(int argc, char **argv) +int setup_tests(void) { - int result = 0; - int iter_argv; - int benchmark = 0; - - for (iter_argv = 1; iter_argv < argc; iter_argv++) { - if (strcmp(argv[iter_argv], "-b") == 0) - benchmark = 1; - else if (strcmp(argv[iter_argv], "-h") == 0) - goto help; + if (test_has_option("-h")) { + printf("-h\tThis help\n"); + printf("-b\tBenchmark gcm128 in addition to the tests\n"); + return 1; } ADD_ALL_TESTS(test_cts128, OSSL_NELEM(cts128_vectors)); ADD_ALL_TESTS(test_cts128_nist, OSSL_NELEM(cts128_vectors)); ADD_ALL_TESTS(test_gcm128, OSSL_NELEM(gcm128_vectors)); + return 1; +} - result = run_tests(argv[0]); - - if (benchmark) +void cleanup_tests(void) +{ + if (test_has_option("-b")) benchmark_gcm128(K1, sizeof(K1), IV1, sizeof(IV1)); - - return result; - - help: - printf("-h\tThis help\n"); - printf("-b\tBenchmark gcm128 in addition to the tests\n"); - - return 0; } |