123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219 |
- /* benchmark main.c
- *
- * Copyright (C) 2006-2022 wolfSSL Inc.
- *
- * This file is part of wolfSSL.
- *
- * wolfSSL is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * wolfSSL is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA
- */
- /* ESP-IDF */
- #include <esp_log.h>
- #include "sdkconfig.h"
- /* wolfSSL */
- #include <user_settings.h>
- #ifndef WOLFSSL_ESPIDF
- #warning "problem with wolfSSL user_settings. Check components/wolfssl/include"
- #endif
- #include <wolfssl/wolfcrypt/settings.h>
- #include <wolfssl/wolfcrypt/types.h>
- #include <wolfcrypt/benchmark/benchmark.h>
- /* check BENCH_ARGV in sdkconfig to determine need to set WOLFSSL_BENCH_ARGV */
- #ifdef CONFIG_BENCH_ARGV
- #define WOLFSSL_BENCH_ARGV CONFIG_BENCH_ARGV
- #define WOLFSSL_BENCH_ARGV_MAX_ARGUMENTS 22 /* arbitrary number of max args */
- #endif
- /*
- ** the wolfssl component can be installed in either:
- **
- ** - the ESP-IDF component directory
- **
- ** ** OR **
- **
- ** - the local project component directory
- **
- ** it is not recommended to install in both.
- **
- */
- #include "main.h"
- static const char* const TAG = "wolfssl_benchmark";
- #if defined(WOLFSSL_ESPWROOM32SE) && defined(HAVE_PK_CALLBACKS) \
- && defined(WOLFSSL_ATECC508A)
- #include "wolfssl/wolfcrypt/port/atmel/atmel.h"
- /* when you need to use a custom slot allocation, */
- /* enable the definition CUSTOM_SLOT_ALLOCAION. */
- #if defined(CUSTOM_SLOT_ALLOCATION)
- static byte mSlotList[ATECC_MAX_SLOT];
- /* initialize slot array */
- void my_atmel_slotInit()
- {
- int i;
- for(i = 0;i < ATECC_MAX_SLOT;i++) {
- mSlotList[i] = ATECC_INVALID_SLOT;
- }
- }
- /* allocate slot depending on slotType */
- int my_atmel_alloc(int slotType)
- {
- int i, slot = -1;
- switch(slotType){
- case ATMEL_SLOT_ENCKEY:
- slot = 4;
- break;
- case ATMEL_SLOT_DEVICE:
- slot = 0;
- break;
- case ATMEL_SLOT_ECDHE:
- slot = 0;
- break;
- case ATMEL_SLOT_ECDHE_ENC:
- slot = 4;
- break;
- case ATMEL_SLOT_ANY:
- for(i = 0;i < ATECC_MAX_SLOT;i++){
- if(mSlotList[i] == ATECC_INVALID_SLOT){
- slot = i;
- break;
- }
- }
- }
- return slot;
- }
- /* free slot array */
- void my_atmel_free(int slotId)
- {
- if(slotId >= 0 && slotId < ATECC_MAX_SLOT){
- mSlotList[slotId] = ATECC_INVALID_SLOT;
- }
- }
- #endif /* CUSTOM_SLOT_ALLOCATION */
- #endif /* WOLFSSL_ESPWROOM32SE && HAVE_PK_CALLBACK && WOLFSSL_ATECC508A */
- /* the following are needed by benchmark.c with args */
- #ifdef WOLFSSL_BENCH_ARGV
- char* __argv[WOLFSSL_BENCH_ARGV_MAX_ARGUMENTS];
- int construct_argv()
- {
- int cnt = 0;
- int i = 0;
- int len = 0;
- char *_argv; /* buffer for copying the string */
- char *ch; /* char pointer to trace the string */
- char buff[16] = { 0 }; /* buffer for a argument copy */
- ESP_LOGI(TAG, "construct_argv arg:%s\n", CONFIG_BENCH_ARGV);
- len = strlen(CONFIG_BENCH_ARGV);
- _argv = (char*)malloc(len + 1);
- if (!_argv) {
- return -1;
- }
- memset(_argv, 0, len + 1);
- memcpy(_argv, CONFIG_BENCH_ARGV, len);
- _argv[len] = '\0';
- ch = _argv;
- __argv[cnt] = malloc(10);
- sprintf(__argv[cnt], "benchmark");
- __argv[cnt][9] = '\0';
- cnt = 1;
- while (*ch != '\0') {
- /* check that we don't overflow manual arg assembly */
- if (cnt >= (WOLFSSL_BENCH_ARGV_MAX_ARGUMENTS)) {
- ESP_LOGE(TAG, "Abort construct_argv;"
- "Reached maximum defined arguments = %d",
- WOLFSSL_BENCH_ARGV_MAX_ARGUMENTS);
- break;
- }
- /* skip white-space */
- while (*ch == ' ') { ++ch; }
- memset(buff, 0, sizeof(buff));
- /* copy each args into buffer */
- i = 0;
- while ((*ch != ' ') && (*ch != '\0') && (i < 16)) {
- buff[i] = *ch;
- ++i;
- ++ch;
- }
- /* copy the string into argv */
- __argv[cnt] = (char*)malloc(i + 1);
- memset(__argv[cnt], 0, i + 1);
- memcpy(__argv[cnt], buff, i + 1);
- /* next args */
- ++cnt;
- }
- free(_argv);
- return (cnt);
- }
- #endif
- /* entry point */
- void app_main(void)
- {
- ESP_LOGI(TAG, "app_main CONFIG_BENCH_ARGV = %s", WOLFSSL_BENCH_ARGV);
- /* when using atecc608a on esp32-wroom-32se */
- #if defined(WOLFSSL_ESPWROOM32SE) && defined(HAVE_PK_CALLBACKS) \
- && defined(WOLFSSL_ATECC508A)
- #if defined(CUSTOM_SLOT_ALLOCATION)
- my_atmel_slotInit();
- /* to register the callback, it needs to be initialized. */
- if ((wolfCrypt_Init()) != 0) {
- ESP_LOGE(TAG, "wolfCrypt_Init failed");
- return;
- }
- atmel_set_slot_allocator(my_atmel_alloc, my_atmel_free);
- #endif
- #endif
- #ifdef NO_CRYPT_BENCHMARK
- ESP_LOGI(TAG, "NO_CRYPT_BENCHMARK defined, skipping wolf_benchmark_task")
- #else
- /* although wolfCrypt_Init() may be explicitly called above,
- ** note it is still always called in wolf_benchmark_task.
- */
- wolf_benchmark_task();
- /* wolfCrypt_Cleanup should always be called at completion,
- ** and is called in wolf_benchmark_task().
- */
- /* after the test, we'll just wait */
- while (1) {
- /* nothing */
- }
- #endif /* NO_CRYPT_BENCHMARK */
- } /* main */
|