477 lines
11 KiB
C
477 lines
11 KiB
C
/* testsuite.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
|
|
*/
|
|
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <wolfssl/wolfcrypt/settings.h>
|
|
|
|
#include <wolfssl/ssl.h>
|
|
#include <wolfssl/test.h>
|
|
#include <wolfcrypt/test/test.h>
|
|
|
|
|
|
#ifndef SINGLE_THREADED
|
|
|
|
#ifdef OPENSSL_EXTRA
|
|
#include <wolfssl/openssl/ssl.h>
|
|
#endif
|
|
#include <wolfssl/wolfcrypt/sha256.h>
|
|
#include <wolfssl/wolfcrypt/ecc.h>
|
|
|
|
#include <examples/echoclient/echoclient.h>
|
|
#include <examples/echoserver/echoserver.h>
|
|
#include <examples/server/server.h>
|
|
#include <examples/client/client.h>
|
|
|
|
|
|
#ifndef NO_SHA256
|
|
void file_test(const char* file, byte* hash);
|
|
#endif
|
|
|
|
#if !defined(NO_WOLFSSL_SERVER) && !defined(NO_WOLFSSL_CLIENT)
|
|
|
|
#ifdef HAVE_STACK_SIZE
|
|
static THREAD_RETURN simple_test(func_args*);
|
|
#else
|
|
static void simple_test(func_args*);
|
|
#endif
|
|
|
|
enum {
|
|
NUMARGS = 3
|
|
};
|
|
|
|
static const char *outputName;
|
|
#endif
|
|
|
|
int myoptind = 0;
|
|
char* myoptarg = NULL;
|
|
|
|
#ifndef NO_TESTSUITE_MAIN_DRIVER
|
|
|
|
static int testsuite_test(int argc, char** argv);
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
return testsuite_test(argc, argv);
|
|
}
|
|
|
|
#endif /* NO_TESTSUITE_MAIN_DRIVER */
|
|
|
|
#ifdef HAVE_STACK_SIZE
|
|
static void *echoclient_test_wrapper(void* args) {
|
|
echoclient_test(args);
|
|
|
|
#if defined(HAVE_ECC) && defined(FP_ECC) && defined(HAVE_THREAD_LS)
|
|
wc_ecc_fp_free(); /* free per thread cache */
|
|
#endif
|
|
|
|
return (void *)0;
|
|
}
|
|
#endif
|
|
|
|
int testsuite_test(int argc, char** argv)
|
|
{
|
|
#if !defined(NO_WOLFSSL_SERVER) && !defined(NO_WOLFSSL_CLIENT)
|
|
func_args server_args;
|
|
|
|
tcp_ready ready;
|
|
THREAD_TYPE serverThread;
|
|
|
|
#ifndef USE_WINDOWS_API
|
|
char tempName[] = "/tmp/output-XXXXXX";
|
|
int len = 18;
|
|
int num = 6;
|
|
#else
|
|
char tempName[] = "fnXXXXXX";
|
|
int len = 8;
|
|
int num = 6;
|
|
#endif
|
|
#ifdef HAVE_STACK_SIZE
|
|
void *serverThreadStackContext = 0;
|
|
#endif
|
|
|
|
#ifdef HAVE_WNR
|
|
if (wc_InitNetRandom(wnrConfig, NULL, 5000) != 0) {
|
|
err_sys("Whitewood netRandom global config failed");
|
|
return -1237;
|
|
}
|
|
#endif /* HAVE_WNR */
|
|
|
|
StartTCP();
|
|
|
|
server_args.argc = argc;
|
|
server_args.argv = argv;
|
|
|
|
wolfSSL_Init();
|
|
#if defined(DEBUG_WOLFSSL) && !defined(HAVE_VALGRIND)
|
|
wolfSSL_Debugging_ON();
|
|
#endif
|
|
|
|
#if !defined(WOLFSSL_TIRTOS)
|
|
ChangeToWolfRoot();
|
|
#endif
|
|
|
|
#ifdef WOLFSSL_TIRTOS
|
|
fdOpenSession(Task_self());
|
|
#endif
|
|
|
|
server_args.signal = &ready;
|
|
InitTcpReady(&ready);
|
|
|
|
#ifndef NO_CRYPT_TEST
|
|
/* wc_ test */
|
|
#ifdef HAVE_STACK_SIZE
|
|
StackSizeCheck(&server_args, wolfcrypt_test);
|
|
#else
|
|
wolfcrypt_test(&server_args);
|
|
#endif
|
|
if (server_args.return_code != 0) return server_args.return_code;
|
|
#endif
|
|
|
|
/* Simple wolfSSL client server test */
|
|
#ifdef HAVE_STACK_SIZE
|
|
StackSizeCheck(&server_args, (THREAD_RETURN (*)(void *))simple_test);
|
|
#else
|
|
simple_test(&server_args);
|
|
#endif
|
|
if (server_args.return_code != 0) return server_args.return_code;
|
|
/* Echo input wolfSSL client server test */
|
|
#ifdef HAVE_STACK_SIZE
|
|
StackSizeCheck_launch(&server_args, echoserver_test, &serverThread, &serverThreadStackContext);
|
|
#else
|
|
start_thread(echoserver_test, &server_args, &serverThread);
|
|
#endif
|
|
wait_tcp_ready(&server_args);
|
|
{
|
|
func_args echo_args;
|
|
char* myArgv[NUMARGS];
|
|
|
|
char arg[3][32];
|
|
|
|
myArgv[0] = arg[0];
|
|
myArgv[1] = arg[1];
|
|
myArgv[2] = arg[2];
|
|
|
|
echo_args.argc = 3;
|
|
echo_args.argv = myArgv;
|
|
|
|
/* Create unique file name */
|
|
outputName = mymktemp(tempName, len, num);
|
|
if (outputName == NULL) {
|
|
printf("Could not create unique file name");
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
strcpy(arg[0], "testsuite");
|
|
strcpy(arg[1], "input");
|
|
strcpy(arg[2], outputName);
|
|
|
|
/* Share the signal, it has the new port number in it. */
|
|
echo_args.signal = server_args.signal;
|
|
|
|
/* make sure OK */
|
|
|
|
#ifdef HAVE_STACK_SIZE
|
|
fputs("echoclient_test #1: ", stdout);
|
|
StackSizeCheck(&echo_args, echoclient_test_wrapper);
|
|
#else
|
|
echoclient_test(&echo_args);
|
|
#endif
|
|
if (echo_args.return_code != 0) return echo_args.return_code;
|
|
|
|
#ifdef WOLFSSL_DTLS
|
|
wait_tcp_ready(&server_args);
|
|
#endif
|
|
/* send quit to echoserver */
|
|
echo_args.argc = 2;
|
|
strcpy(echo_args.argv[1], "quit");
|
|
|
|
#ifdef HAVE_STACK_SIZE
|
|
fputs("echoclient_test #2: ", stdout);
|
|
StackSizeCheck(&echo_args, echoclient_test_wrapper);
|
|
#else
|
|
echoclient_test(&echo_args);
|
|
#endif
|
|
if (echo_args.return_code != 0) return echo_args.return_code;
|
|
#ifdef HAVE_STACK_SIZE
|
|
fputs("reaping echoserver_test: ", stdout);
|
|
StackSizeCheck_reap(serverThread, serverThreadStackContext);
|
|
#else
|
|
join_thread(serverThread);
|
|
#endif
|
|
if (server_args.return_code != 0) return server_args.return_code;
|
|
}
|
|
|
|
/* show ciphers */
|
|
{
|
|
char ciphers[WOLFSSL_CIPHER_LIST_MAX_SIZE];
|
|
XMEMSET(ciphers, 0, sizeof(ciphers));
|
|
wolfSSL_get_ciphers(ciphers, sizeof(ciphers)-1);
|
|
printf("ciphers = %s\n", ciphers);
|
|
}
|
|
|
|
/* validate output equals input */
|
|
{
|
|
#ifndef NO_SHA256
|
|
byte input[WC_SHA256_DIGEST_SIZE];
|
|
byte output[WC_SHA256_DIGEST_SIZE];
|
|
|
|
file_test("input", input);
|
|
file_test(outputName, output);
|
|
#endif
|
|
remove(outputName);
|
|
#ifndef NO_SHA256
|
|
if (memcmp(input, output, sizeof(input)) != 0)
|
|
return EXIT_FAILURE;
|
|
#endif
|
|
}
|
|
|
|
wolfSSL_Cleanup();
|
|
FreeTcpReady(&ready);
|
|
|
|
#ifdef WOLFSSL_TIRTOS
|
|
fdCloseSession(Task_self());
|
|
#endif
|
|
|
|
#ifdef HAVE_WNR
|
|
if (wc_FreeNetRandom() < 0)
|
|
err_sys("Failed to free netRandom context");
|
|
#endif /* HAVE_WNR */
|
|
|
|
printf("\nAll tests passed!\n");
|
|
|
|
#else
|
|
(void)argc;
|
|
(void)argv;
|
|
#endif /* !NO_WOLFSSL_SERVER && !NO_WOLFSSL_CLIENT */
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
#if !defined(NO_WOLFSSL_SERVER) && !defined(NO_WOLFSSL_CLIENT)
|
|
#ifdef HAVE_STACK_SIZE
|
|
static THREAD_RETURN simple_test(func_args* args)
|
|
#else
|
|
static void simple_test(func_args* args)
|
|
#endif
|
|
{
|
|
THREAD_TYPE serverThread;
|
|
|
|
int i;
|
|
|
|
func_args svrArgs;
|
|
char *svrArgv[9];
|
|
char argvs[9][32];
|
|
|
|
func_args cliArgs;
|
|
char *cliArgv[NUMARGS];
|
|
char argvc[3][32];
|
|
|
|
for (i = 0; i < 9; i++)
|
|
svrArgv[i] = argvs[i];
|
|
for (i = 0; i < 3; i++)
|
|
cliArgv[i] = argvc[i];
|
|
|
|
strcpy(argvs[0], "SimpleServer");
|
|
svrArgs.argc = 1;
|
|
svrArgs.argv = svrArgv;
|
|
svrArgs.return_code = 0;
|
|
#if !defined(USE_WINDOWS_API) && !defined(WOLFSSL_SNIFFER) && \
|
|
!defined(WOLFSSL_TIRTOS)
|
|
strcpy(argvs[svrArgs.argc++], "-p");
|
|
strcpy(argvs[svrArgs.argc++], "0");
|
|
#endif
|
|
/* Set the last arg later, when it is known. */
|
|
|
|
args->return_code = 0;
|
|
svrArgs.signal = args->signal;
|
|
start_thread(server_test, &svrArgs, &serverThread);
|
|
wait_tcp_ready(&svrArgs);
|
|
|
|
/* Setting the actual port number. */
|
|
strcpy(argvc[0], "SimpleClient");
|
|
cliArgs.argv = cliArgv;
|
|
cliArgs.return_code = 0;
|
|
#ifndef USE_WINDOWS_API
|
|
cliArgs.argc = NUMARGS;
|
|
strcpy(argvc[1], "-p");
|
|
snprintf(argvc[2], sizeof(argvc[2]), "%d", svrArgs.signal->port);
|
|
#else
|
|
cliArgs.argc = 1;
|
|
#endif
|
|
|
|
client_test(&cliArgs);
|
|
if (cliArgs.return_code != 0) {
|
|
args->return_code = cliArgs.return_code;
|
|
#ifdef HAVE_STACK_SIZE
|
|
return (THREAD_RETURN)0;
|
|
#else
|
|
return;
|
|
#endif
|
|
}
|
|
join_thread(serverThread);
|
|
if (svrArgs.return_code != 0) args->return_code = svrArgs.return_code;
|
|
#ifdef HAVE_STACK_SIZE
|
|
return (THREAD_RETURN)0;
|
|
#endif
|
|
}
|
|
#endif /* !NO_WOLFSSL_SERVER && !NO_WOLFSSL_CLIENT */
|
|
|
|
|
|
void wait_tcp_ready(func_args* args)
|
|
{
|
|
#if defined(_POSIX_THREADS) && !defined(__MINGW32__)
|
|
pthread_mutex_lock(&args->signal->mutex);
|
|
|
|
if (!args->signal->ready)
|
|
pthread_cond_wait(&args->signal->cond, &args->signal->mutex);
|
|
args->signal->ready = 0; /* reset */
|
|
|
|
pthread_mutex_unlock(&args->signal->mutex);
|
|
#else
|
|
(void)args;
|
|
#endif
|
|
}
|
|
|
|
|
|
void start_thread(THREAD_FUNC fun, func_args* args, THREAD_TYPE* thread)
|
|
{
|
|
#if defined(_POSIX_THREADS) && !defined(__MINGW32__)
|
|
pthread_create(thread, 0, fun, args);
|
|
return;
|
|
#elif defined(WOLFSSL_TIRTOS)
|
|
/* Initialize the defaults and set the parameters. */
|
|
Task_Params taskParams;
|
|
Task_Params_init(&taskParams);
|
|
taskParams.arg0 = (UArg)args;
|
|
taskParams.stackSize = 65535;
|
|
*thread = Task_create((Task_FuncPtr)fun, &taskParams, NULL);
|
|
if (*thread == NULL) {
|
|
printf("Failed to create new Task\n");
|
|
}
|
|
Task_yield();
|
|
#else
|
|
*thread = (THREAD_TYPE)_beginthreadex(0, 0, fun, args, 0, 0);
|
|
#endif
|
|
}
|
|
|
|
|
|
void join_thread(THREAD_TYPE thread)
|
|
{
|
|
#if defined(_POSIX_THREADS) && !defined(__MINGW32__)
|
|
pthread_join(thread, 0);
|
|
#elif defined(WOLFSSL_TIRTOS)
|
|
while(1) {
|
|
if (Task_getMode(thread) == Task_Mode_TERMINATED) {
|
|
Task_sleep(5);
|
|
break;
|
|
}
|
|
Task_yield();
|
|
}
|
|
#else
|
|
int res = WaitForSingleObject((HANDLE)thread, INFINITE);
|
|
assert(res == WAIT_OBJECT_0);
|
|
res = CloseHandle((HANDLE)thread);
|
|
assert(res);
|
|
(void)res; /* Suppress un-used variable warning */
|
|
#endif
|
|
}
|
|
|
|
|
|
#ifndef NO_SHA256
|
|
void file_test(const char* file, byte* check)
|
|
{
|
|
FILE* f;
|
|
int i = 0, j, ret;
|
|
wc_Sha256 sha256;
|
|
byte buf[1024];
|
|
byte shasum[WC_SHA256_DIGEST_SIZE];
|
|
|
|
ret = wc_InitSha256(&sha256);
|
|
if (ret != 0) {
|
|
printf("Can't wc_InitSha256 %d\n", ret);
|
|
return;
|
|
}
|
|
if( !( f = fopen( file, "rb" ) )) {
|
|
printf("Can't open %s\n", file);
|
|
return;
|
|
}
|
|
while( ( i = (int)fread(buf, 1, sizeof(buf), f )) > 0 ) {
|
|
ret = wc_Sha256Update(&sha256, buf, i);
|
|
if (ret != 0) {
|
|
printf("Can't wc_Sha256Update %d\n", ret);
|
|
fclose(f);
|
|
return;
|
|
}
|
|
}
|
|
|
|
ret = wc_Sha256Final(&sha256, shasum);
|
|
wc_Sha256Free(&sha256);
|
|
|
|
if (ret != 0) {
|
|
printf("Can't wc_Sha256Final %d\n", ret);
|
|
fclose(f);
|
|
return;
|
|
}
|
|
|
|
XMEMCPY(check, shasum, sizeof(shasum));
|
|
|
|
for(j = 0; j < WC_SHA256_DIGEST_SIZE; ++j )
|
|
printf( "%02x", shasum[j] );
|
|
|
|
printf(" %s\n", file);
|
|
|
|
fclose(f);
|
|
}
|
|
#endif
|
|
|
|
#else /* SINGLE_THREADED */
|
|
|
|
|
|
int myoptind = 0;
|
|
char* myoptarg = NULL;
|
|
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
func_args server_args;
|
|
|
|
server_args.argc = argc;
|
|
server_args.argv = argv;
|
|
|
|
wolfSSL_Init();
|
|
ChangeToWolfRoot();
|
|
|
|
wolfcrypt_test(&server_args);
|
|
if (server_args.return_code != 0) return server_args.return_code;
|
|
|
|
wolfSSL_Cleanup();
|
|
printf("\nAll tests passed!\n");
|
|
|
|
EXIT_TEST(EXIT_SUCCESS);
|
|
}
|
|
|
|
|
|
#endif /* SINGLE_THREADED */
|
|
|