/*shell.c * * Copyright (C) 2006-2020 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 */ /*** tiny Shell for wolfSSL apps ***/ #ifdef HAVE_CONFIG_H #include #endif #include "wolfssl/internal.h" #include #if defined(WOLFSSL_MDK_ARM) || defined(WOLFSSL_KEIL_TCP_NET) #include #include #include #if defined(WOLFSSL_MDK5) || defined(WOLFSSL_KEIL_TCP_NET) #include "cmsis_os.h" #include "rl_fs.h" #else #include #endif #endif #ifdef WOLFSSL_KEIL_TCP_NET #include "wolfssl/test.h" #else typedef struct func_args { int argc; char** argv; int return_code; } func_args; #endif #if defined(WOLFSSL_CMSIS_RTOS) #define HAVE_KEIL_RTX #endif #ifdef NO_ECHOCLIENT #define echoclient_test command_not_found #endif #ifdef NO_ECHOSERVER #define echoserver_test command_not_found #endif #ifdef NO_SIMPLE_CLIENT #define client_test command_not_found #endif #ifdef NO_SIMPLE_SERVER #define server_test command_not_found #endif #ifdef NO_CRYPT_BENCHMARK #define benchmark_test command_not_found #endif #ifdef NO_CRYPT_TEST #define ctaocrypt_test command_not_found #endif #ifndef WOLFSSL_KEIL_TCP_NET #define ipaddr_comm command_not_found #endif #if !defined(HAVE_KEIL_RTX) #define stack_comm command_not_found #endif #if !defined(DEBUG_WOLFSSL) #define dbg_comm command_not_found #endif void command_not_found(void *argv) { printf("Command not found\n") ; } extern void echoclient_test(void *args) ; extern void echoserver_test(void *args) ; extern void benchmark_test(void *args) ; extern void wolfcrypt_test(void *args) ; extern void client_test(void *args) ; extern void server_test(void *args) ; extern void kill_task(void *args) ; extern void ipaddr_comm(void *args) ; extern void stack_comm(void *args) ; extern void for_command(void *args) ; extern void dbg_comm(void *arg) ; extern void help_comm(void *arg) ; #if !defined(NO_CRYPT_TEST) #ifndef NO_MD5 extern void md5_test(void *arg) ; #endif #ifdef WOLFSSL_MD2 extern void md2_test(void *arg) ; #endif #ifndef NO_MD4 extern void md4_test(void *arg) ; #endif extern void sha_test(void *arg) ; #ifndef NO_SHA256 extern void sha256_test(void *arg) ; #endif #ifdef WOLFSSL_SHA384 extern void sha384_test(void *arg) ; #endif #ifdef WOLFSSL_SHA512 extern void sha512_test(void *arg) ; #endif #ifdef WOLFSSL_RIPEMD extern void ripemd_test(void *arg) ; #endif #ifndef NO_HMAC #ifndef NO_MD5 extern void hmac_md5_test(void *arg) ; #endif extern void hmac_sha_test(void *arg) ; #ifndef NO_SHA256 extern void hmac_sha256_test(void *arg) ; #endif #ifdef WOLFSSL_SHA384 extern void hmac_sha384_test(void *arg) ; #endif #endif #ifndef NO_RC4 extern void arc4_test(void *arg) ; #endif #ifndef NO_HC128 extern void hc128_test(void *arg) ; #endif #ifndef NO_RABBIT extern void rabbit_test(void *arg) ; #endif #ifndef NO_DES3 extern void des_test(void *arg) ; extern void des3_test(void *arg) ; #endif #ifndef NO_AES extern void aes_test(void *arg) ; #ifdef HAVE_AESGCM extern void aesgcm_test(void *arg) ; #endif #ifdef HAVE_AESCCM extern void aesccm_test(void *arg) ; #endif #endif #ifdef HAVE_CAMELLIA extern void camellia_test(void *arg) ; #endif extern void random_test(void *arg) ; #ifndef NO_RSA extern void rsa_test(void *arg) ; #endif #ifndef NO_DH extern void dh_test(void *arg) ; #endif #ifndef NO_DSA extern void dsa_test(void *arg) ; #endif #ifndef NO_PWDBASED extern void pwdbased_test(void *arg) ; #endif #ifdef OPENSSL_EXTRA extern void openssl_test(void *arg) ; #endif #ifdef HAVE_ECC extern void ecc_test(void *arg) ; #endif #endif /* NO_CRYPT_TEST */ static struct { const char *command ; void (*func)(void *args) ; } commandTable[] = { "echoclient", echoclient_test, "echoserver", echoserver_test, "benchmark", benchmark_test, "test", wolfcrypt_test, "client", client_test, "server", server_test, "ipaddr", ipaddr_comm, /* TBD */ "stack", stack_comm, /* On/Off check stack size */ "for", for_command, /* iterate next command X times */ "debug", dbg_comm, /* On/Off debug message */ "help", help_comm, /* Brief description about the commands */ /** short name **/ "ec", echoclient_test, "es", echoserver_test, "bm", benchmark_test, "te", wolfcrypt_test, "cl", client_test, "sv", server_test, "ip", ipaddr_comm, "st", stack_comm, "dbg", dbg_comm, "?", help_comm, /*** test suites ****/ #if !defined(NO_CRYPT_TEST) #ifndef NO_MD5 "md5", md5_test, #endif #ifdef WOLFSSL_MD2 "md2", md2_test, #endif #ifndef NO_MD4 "md4", md4_test, #endif "sha", sha_test, #ifndef NO_SHA256 "sha256", sha256_test, #endif #ifdef WOLFSSL_SHA384 "sha384", sha384_test, #endif #ifdef WOLFSSL_SHA512 "sha512", sha512_test, #endif #ifdef WOLFSSL_RIPEMD "ripemd", ripemd_test, #endif #ifndef NO_HMAC #ifndef NO_MD5 "hmac_md5", hmac_md5_test, #endif "hmac_sha", hmac_sha_test, #ifndef NO_SHA256 "hmac_sha256", hmac_sha256_test, #endif #ifdef WOLFSSL_SHA384 "hmac_sha384", hmac_sha384_test, #endif #endif #ifndef NO_RC4 "arc4", arc4_test, #endif #ifndef NO_HC128 "hc128", hc128_test, #endif #ifndef NO_RABBIT "rabbit", rabbit_test, #endif #ifndef NO_DES3 "des", des_test, "des3", des3_test, #endif #ifndef NO_AES "aes", aes_test, #ifdef HAVE_AESGCM "aesgcm", aesgcm_test, #endif #ifdef HAVE_AESCCM "aesccm", aesccm_test, #endif #endif #ifdef HAVE_CAMELLIA "camellia", camellia_test, #endif "random", random_test, #ifndef NO_RSA "rsa", rsa_test, #endif #ifndef NO_DH "dh", dh_test, #endif #ifndef NO_DSA "dsa", dsa_test, #endif #ifndef NO_PWDBASED "pwdbased", pwdbased_test, #endif #ifdef OPENSSL_EXTRA "openssl", openssl_test, #endif #ifdef HAVE_ECC "ecc", ecc_test, #endif #endif /* NO_CRYPT_TEST */ "", NULL } ; enum jobtype { FORGROUND, BACKGROUND } ; #define IF_DELIMITER(ch) ((ch) == ' ' || (ch) == '\n') static int BackGround = 0 ; /* 1: background job is running */ char * wolfssl_fgets ( char * str, int num, FILE * f ) { int i ; for(i = 0 ; i< num ; i++) { while((str[i] = getchar()) == 0) { #if defined (HAVE_KEIL_RTX) && !defined(WOLFSSL_CMSIS_RTOS) os_tsk_pass (); #else osThreadYield (); #endif } if(str[i] == '\n' || str[i] == '\012' || str[i] == '\015') { putchar('\n') ; str[i++] = '\n' ; str[i] = '\0' ; break ; } else if(str[i] == '\010') { /* BS */ if(i) { /* erace one char */ putchar('\010') ; putchar(' ') ; putchar('\010') ; i = (i>0 ? (i-2) : -1 ) ; continue ; } } else if(str[i] == '\033' || str[i] == '\004' ) { /* ESC or ^D */ str[i] = '\0' ; return(0) ; } putchar(str[i]) ; } return(str) ; } /******* Get Command Line *****************************/ static int getline(char * line, int sz, func_args *args, int*bf_flg) { char * ret ; int i ; #define MAXARGS 10 #define MAXARGLEN 30 static char *argv[MAXARGS] ; args->argv = argv ; putchar('>') ; fflush(stdout) ; ret = wolfssl_fgets(line, sz, stdin) ; #define SHELL_ERROR_FGETS -102 if(ret != line) return(SHELL_ERROR_FGETS) ; if(line[strlen(line)-2] == '&') { (*bf_flg) = BACKGROUND ; line[strlen(line)-2] = '\n' ; } else { (*bf_flg) = FORGROUND ; } args->argc = 0 ; for(i=0; iargv[args->argc] = &(line[i]) ; while(!IF_DELIMITER(line[i])) i++ ; args->argc++ ; if(line[i] == '\n') { line[i] = '\0' ; break ; } else { line[i] = '\0' ; } } return i ; } /************* Embedded Shell Commands **********************************/ #define IP_SIZE 16 #ifdef WOLFSSL_KEIL_TCP_NET static void ipaddr_comm(void *args) { if(((func_args *)args)->argc == 1) { printf("IP addr: %s, port %d\n", wolfSSLIP, wolfSSLPort) ; } else { if(BackGround != 0) { printf("Cannot change IP addr while background server is running\n") ; } else if(((func_args *)args)->argc == 3 && ((func_args *)args)->argv[1][0] == '-'&& ((func_args *)args)->argv[1][1] == 'a' ) { /* strcpy(yasslIP, ((func_args *)args)->argv[2]) ; */ } else if(((func_args *)args)->argc == 3 && ((func_args *)args)->argv[1][0] == '-' && ((func_args *)args)->argv[1][1] == 'p' ) { /* yasslPort = atoi(((func_args *)args)->argv[2]) ; */ } else printf("Invalid argument\n") ; } } #endif #if defined(HAVE_KEIL_RTX) static int stack_ck = 0 ; void stack_comm(void *args) { if(stack_ck) { printf("Stack Check: Off\n") ; stack_ck = 0 ; } else { printf("Stack Check: On\n") ; stack_ck = 1 ; } } #define FILL_PATTERN 0xa596695a void stack_fill(char * stack, int size) { int i ; if(stack_ck == 0)return ; for(i=1; iargc == 1) { printf("For %d times\n", for_iteration) ; } else if(((func_args *)args)->argc == 2) { for_iteration = atoi(((func_args *)args)->argv[1]) ; } else printf("Invalid argument\n") ; } #if defined(DEBUG_WOLFSSL) static int wolfsslDebug = 1 ; static void dbg_comm(void *args) { if(wolfsslDebug == 1) { wolfsslDebug = 0 ; printf("Turning OFF Debug message\n") ; wolfSSL_Debugging_OFF() ; } else { wolfasslDebug = 1 ; printf("Turning ON Debug message\n") ; wolfSSL_Debugging_ON() ; } } #endif static void help_comm(void *args) { static char *commands[] = { "test", "benchmark", "echoserver& : simple echo server in background mode", "echoclient : simple echo client followed by any input string, or \"quit\", \"break\"", "server& : simple server in background mode", "client : simple client", "client -g -v [0123] -h xxx.xxx.xxx.xxx -p 443 : usage example", "server/client -h : help for server/client command", "help", "" } ; int i ; printf("Commands:\n") ; for(i=0; commands[i][0] ; i++) printf(" %s\n", commands[i]) ; } #define BG_JOB_STACK_SIZE 16000 #if (!defined(NO_SIMPLE_SERVER) && !defined(NO_ECHOSERVER)) && \ defined(HAVE_KEIL_RTX) #if !defined(WOLFSSL_CMSIS_RTOS) static char bg_job_stack[BG_JOB_STACK_SIZE] ; #endif #endif #define COMMAND_STACK_SIZE 24000 #if defined(HAVE_KEIL_RTX) && !defined(WOLFSSL_CMSIS_RTOS) static char command_stack[COMMAND_STACK_SIZE] ; #endif #if defined(HAVE_KEIL_RTX) || defined(WOLFSSL_CMSIS_RTOS) static wolfSSL_Mutex command_mutex ; #endif void exit_command(void) { printf("Command Aborted\n") ; #ifdef WOLFSSL_CMSIS_RTOS osThreadTerminate(osThreadGetId()) ; #else os_tsk_delete_self() ; #endif } /*********** Invoke Foreground Command *********************/ static void command_invoke(void const *args) { void (*func)(void const * ) ; int i,iteration ; func = (void(*)(void const *))((func_args *)args)->argv[0] ; #if defined(HAVE_KEIL_RTX) wc_LockMutex((wolfSSL_Mutex *)&command_mutex) ; #endif iteration = for_iteration ; for(i=0; i< iteration; i++) { if(iteration > 1) printf("--- Start for %d ---->\n", i) ; #if defined(HAVE_KEIL_RTX) && !defined(WOLFSSL_CMSIS_RTOS) stack_fill(command_stack, COMMAND_STACK_SIZE) ; #endif func(args) ; /* invoke command */ #if defined(HAVE_KEIL_RTX)&& !defined(WOLFSSL_CMSIS_RTOS) stack_check(command_stack, COMMAND_STACK_SIZE) ; #endif } if(iteration > 1) for_iteration = 1 ; osDelay(20000) ; #ifdef HAVE_KEIL_RTX wc_UnLockMutex((wolfSSL_Mutex *)&command_mutex) ; #ifdef WOLFSSL_CMSIS_RTOS osThreadTerminate(osThreadGetId()) ; #else os_tsk_delete_self() ; #endif #endif } #if defined(HAVE_KEIL_RTX) || defined(WOLFSSL_CMSIS_RTOS) /******* Invoke Background Job *******************************/ static void bg_job_invoke(void const *args) { void (*func)(void const * ) ; BackGround = 1 ; #if defined(HAVE_KEIL_RTX)&& !defined(WOLFSSL_CMSIS_RTOS) stack_fill(bg_job_stack, BG_JOB_STACK_SIZE) ; #endif func = (void(*)(void const *))((func_args *)args)->argv[0] ; func(args) ; /* invoke command */ #if defined(HAVE_KEIL_RTX) && !defined(WOLFSSL_CMSIS_RTOS) stack_check(bg_job_stack, BG_JOB_STACK_SIZE) ; #endif osDelay(20000) ; BackGround = 0 ; #ifdef WOLFSSL_CMSIS_RTOS osThreadTerminate(osThreadGetId()) ; #else os_tsk_delete_self() ; ; #endif } #endif #define LINESIZE 100 static char line[LINESIZE] ; #if defined(WOLFSSL_CMSIS_RTOS) osThreadDef (command_invoke, osPriorityAboveNormal , 1, COMMAND_STACK_SIZE) ; osThreadDef (bg_job_invoke, osPriorityNormal , 1 , BG_JOB_STACK_SIZE) ; #endif /********* SHEULL MAIN LOOP ***********************************/ void shell_main(void *arg) { int i ; func_args args ; int bf_flg ; #if defined(WOLFSSL_CMSIS_RTOS) osThreadId cmd ; #endif i = BackGround ; /* Dummy for avoiding warning: BackGround is defined but not used. */ #if defined(HAVE_KEIL_RTX) wc_InitMutex(&command_mutex) ; #endif help_comm(NULL) ; printf("Starting Shell\n") ; while(1) { if(getline(line, LINESIZE, &args, &bf_flg) > 0) { for(i=0; commandTable[i].func != NULL; i++) { if(strcmp(commandTable[i].command, args.argv[0]) == 0) { args.argv[0] = (char *) commandTable[i].func ; if(bf_flg == FORGROUND) { #if defined(HAVE_KEIL_RTX) && !defined(WOLFSSL_CMSIS_RTOS) wc_UnLockMutex((wolfSSL_Mutex *)&command_mutex) ; os_tsk_create_user_ex( (void(*)(void *))&command_invoke, 7, command_stack, COMMAND_STACK_SIZE, &args) ; os_tsk_pass (); #else #if defined(WOLFSSL_CMSIS_RTOS) wc_UnLockMutex((wolfSSL_Mutex *)&command_mutex) ; cmd = osThreadCreate (osThread (command_invoke) , &args); if(cmd == NULL) { printf("Cannon create command thread\n") ; } osThreadYield (); #else command_invoke(&args) ; #endif #endif #ifdef HAVE_KEIL_RTX wc_LockMutex((wolfSSL_Mutex *)&command_mutex) ; #endif } else { #if (!defined(NO_SIMPLE_SERVER) && \ !defined(NO_ECHOSERVER)) && \ defined(HAVE_KEIL_RTX) if(BackGround != 0) { printf("Multiple background servers not supported.\n") ; } else { printf("\"%s\" is running with the background mode.\n", commandTable[i].command) ; #if defined(HAVE_KEIL_RTX) && !defined(WOLFSSL_CMSIS_RTOS) os_tsk_create_user_ex( (void(*)(void *))&bg_job_invoke, 6, bg_job_stack, BG_JOB_STACK_SIZE, &args) ; #else osThreadCreate (osThread (bg_job_invoke), &args); osDelay (500) ; #endif } #else printf("Invalid Command: no background job\n") ; #endif } break ; } } if(commandTable[i].func == NULL) printf("Command not found\n") ; } } }