1
0
Fork 0
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

145 lines
3.0 KiB

/*
* Copyright (c) 2022 Rockchip Electronics Co. Ltd.
*/
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include "test_ae.h"
#include "test_otp_key_crypto.h"
#include "test_throughput.h"
#include "test_cipher.h"
#include "test_crypto_mem.h"
#include "test_hash.h"
#include "test_random.h"
#include "test_stress.h"
#include "test_multi.h"
#include "test_rsa.h"
enum {
OPTION_TOP = 0,
ALL,
CIPHER,
AEAD,
HASH,
HMAC,
SETKEY,
OTPKEY,
MEM,
RANDOM,
THROUGHPUT,
STRESS,
MULTI,
RSA,
OPTION_BUTT,
};
static void guide(void)
{
printf("\n######## rkcrypto api test ########\n");
printf("%s\n", RK_CRYPTO_API_FULL_VERSION);
printf("Entry one parameter as follow:\n");
printf("\t-all Function of all ciphers\n");
printf("\t-cipher Function of cipher\n");
printf("\t-aead Function of aead\n");
printf("\t-hash Function of hash\n");
printf("\t-hmac Function of hmac\n");
printf("\t-rsa Function of rsa\n");
printf("\t-setkey Function of setkey. NOTE: it will write key to OTP area.\n");
printf("\t-otpkey Function of otpkey\n");
printf("\t-mem Maximum buffer size requested by crypto mem alloc, test until alloc failed\n");
printf("\t-random Function of get random\n");
printf("\t-throughput Throughput of all ciphers, MB/s\n");
printf("\t-stress [cnt] stress cnt times of all cipher/hash/hmac\n");
}
int main(int argc, char *argv[])
{
int opt;
int option_index = 0;
int stress_cnt = 1;
int verbose = 1;
static struct option long_options[] = {
{"all", 0, NULL, ALL},
{"cipher", 0, NULL, CIPHER},
{"aead", 0, NULL, AEAD},
{"hash", 0, NULL, HASH},
{"hmac", 0, NULL, HMAC},
{"setkey", 0, NULL, SETKEY},
{"otpkey", 0, NULL, OTPKEY},
{"mem", 0, NULL, MEM},
{"random", 0, NULL, RANDOM},
{"throughput", 0, NULL, THROUGHPUT},
{"stress", 1, NULL, STRESS},
{"multi", 0, NULL, MULTI},
{"rsa", 0, NULL, RSA},
{NULL, 0, NULL, 0},
};
if (argc < 2)
guide();
while ((opt = getopt_long_only(argc, argv, "", long_options, &option_index)) != -1) {
switch (opt) {
case ALL:
test_cipher(verbose);
test_ae(verbose);
test_hash(verbose);
test_hmac(verbose);
test_rsa(verbose);
test_write_otp_key();
test_otp_key();
test_crypto_mem();
break;
case CIPHER:
test_cipher(verbose);
break;
case AEAD:
test_ae(verbose);
break;
case HASH:
test_hash(verbose);
break;
case HMAC:
test_hmac(verbose);
break;
case RSA:
test_rsa(verbose);
break;
case SETKEY:
test_write_otp_key();
break;
case OTPKEY:
test_otp_key();
break;
case MEM:
test_crypto_mem();
break;
case RANDOM:
test_random();
break;
case THROUGHPUT:
test_throughput();
break;
case STRESS:
stress_cnt = atoi(optarg);
stress_test(stress_cnt);
break;
case MULTI:
test_multi();
break;
case '?':
guide();
break;
default:
break;
}
}
printf("\n######## Test done ########\n");
return 0;
}