307 lines
7.4 KiB
C
307 lines
7.4 KiB
C
/* testsuite.c
|
|
*
|
|
* Copyright (C) 2006-2013 wolfSSL Inc.
|
|
*
|
|
* This file is part of CyaSSL.
|
|
*
|
|
* CyaSSL 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.
|
|
*
|
|
* CyaSSL 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <cyassl/openssl/ssl.h>
|
|
#include <cyassl/test.h>
|
|
#include <cyassl/ctaocrypt/sha256.h>
|
|
|
|
#include "ctaocrypt/test/test.h"
|
|
|
|
#ifdef SINGLE_THREADED
|
|
#error testsuite needs threads to run, please run ctaocrypt/test, \
|
|
and the examples/ individually
|
|
#endif
|
|
|
|
#include "examples/echoclient/echoclient.h"
|
|
#include "examples/echoserver/echoserver.h"
|
|
#include "examples/server/server.h"
|
|
#include "examples/client/client.h"
|
|
#include "ctaocrypt/test/test.h"
|
|
|
|
|
|
void file_test(const char* file, byte* hash);
|
|
|
|
void simple_test(func_args*);
|
|
|
|
enum {
|
|
NUMARGS = 3
|
|
};
|
|
|
|
|
|
int myoptind = 0;
|
|
char* myoptarg = NULL;
|
|
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
func_args server_args;
|
|
|
|
tcp_ready ready;
|
|
THREAD_TYPE serverThread;
|
|
|
|
#ifdef HAVE_CAVIUM
|
|
int ret = OpenNitroxDevice(CAVIUM_DIRECT, CAVIUM_DEV_ID);
|
|
if (ret != 0)
|
|
err_sys("Cavium OpenNitroxDevice failed");
|
|
#endif /* HAVE_CAVIUM */
|
|
|
|
StartTCP();
|
|
|
|
server_args.argc = argc;
|
|
server_args.argv = argv;
|
|
|
|
CyaSSL_Init();
|
|
#if defined(DEBUG_CYASSL) && !defined(HAVE_VALGRIND)
|
|
CyaSSL_Debugging_ON();
|
|
#endif
|
|
|
|
if (CurrentDir("testsuite"))
|
|
ChangeDirBack(1);
|
|
else if (CurrentDir("build")) /* Xcode->Preferences->Locations->Build */
|
|
ChangeDirBack(2); /* Location "Place build product in locations
|
|
specified by targets", uses build/Debug */
|
|
server_args.signal = &ready;
|
|
InitTcpReady(&ready);
|
|
|
|
/* CTaoCrypt test */
|
|
ctaocrypt_test(&server_args);
|
|
if (server_args.return_code != 0) return server_args.return_code;
|
|
|
|
/* Simple CyaSSL client server test */
|
|
simple_test(&server_args);
|
|
if (server_args.return_code != 0) return server_args.return_code;
|
|
|
|
/* Echo input yaSSL client server test */
|
|
start_thread(echoserver_test, &server_args, &serverThread);
|
|
wait_tcp_ready(&server_args);
|
|
{
|
|
func_args echo_args;
|
|
char* myArgv[NUMARGS];
|
|
|
|
char argc0[32];
|
|
char argc1[32];
|
|
char argc2[32];
|
|
|
|
myArgv[0] = argc0;
|
|
myArgv[1] = argc1;
|
|
myArgv[2] = argc2;
|
|
|
|
echo_args.argc = NUMARGS;
|
|
echo_args.argv = myArgv;
|
|
|
|
strcpy(echo_args.argv[0], "echoclient");
|
|
strcpy(echo_args.argv[1], "input");
|
|
strcpy(echo_args.argv[2], "output");
|
|
remove("output");
|
|
|
|
/* Share the signal, it has the new port number in it. */
|
|
echo_args.signal = server_args.signal;
|
|
|
|
/* make sure OK */
|
|
echoclient_test(&echo_args);
|
|
if (echo_args.return_code != 0) return echo_args.return_code;
|
|
|
|
#ifdef CYASSL_DTLS
|
|
wait_tcp_ready(&server_args);
|
|
#endif
|
|
/* send quit to echoserver */
|
|
echo_args.argc = 2;
|
|
strcpy(echo_args.argv[1], "quit");
|
|
|
|
echoclient_test(&echo_args);
|
|
if (echo_args.return_code != 0) return echo_args.return_code;
|
|
join_thread(serverThread);
|
|
if (server_args.return_code != 0) return server_args.return_code;
|
|
}
|
|
|
|
/* validate output equals input */
|
|
{
|
|
byte input[SHA256_DIGEST_SIZE];
|
|
byte output[SHA256_DIGEST_SIZE];
|
|
|
|
file_test("input", input);
|
|
file_test("output", output);
|
|
if (memcmp(input, output, sizeof(input)) != 0)
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
CyaSSL_Cleanup();
|
|
FreeTcpReady(&ready);
|
|
|
|
#ifdef HAVE_CAVIUM
|
|
CspShutdown(CAVIUM_DEV_ID);
|
|
#endif
|
|
printf("\nAll tests passed!\n");
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
void simple_test(func_args* args)
|
|
{
|
|
THREAD_TYPE serverThread;
|
|
|
|
func_args svrArgs;
|
|
char *svrArgv[NUMARGS];
|
|
char argc0s[32];
|
|
char argc1s[32];
|
|
char argc2s[32];
|
|
|
|
func_args cliArgs;
|
|
char *cliArgv[NUMARGS];
|
|
char argc0c[32];
|
|
char argc1c[32];
|
|
char argc2c[32];
|
|
|
|
svrArgv[0] = argc0s;
|
|
svrArgv[1] = argc1s;
|
|
svrArgv[2] = argc2s;
|
|
cliArgv[0] = argc0c;
|
|
cliArgv[1] = argc1c;
|
|
cliArgv[2] = argc2c;
|
|
|
|
svrArgs.argc = 1;
|
|
svrArgs.argv = svrArgv;
|
|
svrArgs.return_code = 0;
|
|
cliArgs.argc = 1;
|
|
cliArgs.argv = cliArgv;
|
|
cliArgs.return_code = 0;
|
|
|
|
strcpy(svrArgs.argv[0], "SimpleServer");
|
|
#ifndef USE_WINDOWS_API
|
|
svrArgs.argc = NUMARGS;
|
|
strcpy(svrArgs.argv[1], "-p");
|
|
strcpy(svrArgs.argv[2], "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(cliArgs.argv[0], "SimpleClient");
|
|
#ifndef USE_WINDOWS_API
|
|
cliArgs.argc = NUMARGS;
|
|
strcpy(cliArgs.argv[1], "-p");
|
|
snprintf(cliArgs.argv[2], sizeof(argc2c), "%d", svrArgs.signal->port);
|
|
#endif
|
|
|
|
client_test(&cliArgs);
|
|
if (cliArgs.return_code != 0) {
|
|
args->return_code = cliArgs.return_code;
|
|
return;
|
|
}
|
|
join_thread(serverThread);
|
|
if (svrArgs.return_code != 0) args->return_code = svrArgs.return_code;
|
|
}
|
|
|
|
|
|
void wait_tcp_ready(func_args* args)
|
|
{
|
|
#ifdef _POSIX_THREADS
|
|
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);
|
|
#endif
|
|
}
|
|
|
|
|
|
void start_thread(THREAD_FUNC fun, func_args* args, THREAD_TYPE* thread)
|
|
{
|
|
#ifdef _POSIX_THREADS
|
|
pthread_create(thread, 0, fun, args);
|
|
return;
|
|
#else
|
|
*thread = (THREAD_TYPE)_beginthreadex(0, 0, fun, args, 0, 0);
|
|
#endif
|
|
}
|
|
|
|
|
|
void join_thread(THREAD_TYPE thread)
|
|
{
|
|
#ifdef _POSIX_THREADS
|
|
pthread_join(thread, 0);
|
|
#else
|
|
int res = WaitForSingleObject(thread, INFINITE);
|
|
assert(res == WAIT_OBJECT_0);
|
|
res = CloseHandle(thread);
|
|
assert(res);
|
|
#endif
|
|
}
|
|
|
|
|
|
void InitTcpReady(tcp_ready* ready)
|
|
{
|
|
ready->ready = 0;
|
|
ready->port = 0;
|
|
#ifdef _POSIX_THREADS
|
|
pthread_mutex_init(&ready->mutex, 0);
|
|
pthread_cond_init(&ready->cond, 0);
|
|
#endif
|
|
}
|
|
|
|
|
|
void FreeTcpReady(tcp_ready* ready)
|
|
{
|
|
#ifdef _POSIX_THREADS
|
|
pthread_mutex_destroy(&ready->mutex);
|
|
pthread_cond_destroy(&ready->cond);
|
|
#endif
|
|
}
|
|
|
|
|
|
void file_test(const char* file, byte* check)
|
|
{
|
|
FILE* f;
|
|
int i = 0, j;
|
|
Sha256 sha256;
|
|
byte buf[1024];
|
|
byte shasum[SHA256_DIGEST_SIZE];
|
|
|
|
InitSha256(&sha256);
|
|
if( !( f = fopen( file, "rb" ) )) {
|
|
printf("Can't open %s\n", file);
|
|
return;
|
|
}
|
|
while( ( i = (int)fread(buf, 1, sizeof(buf), f )) > 0 )
|
|
Sha256Update(&sha256, buf, i);
|
|
|
|
Sha256Final(&sha256, shasum);
|
|
memcpy(check, shasum, sizeof(shasum));
|
|
|
|
for(j = 0; j < SHA256_DIGEST_SIZE; ++j )
|
|
printf( "%02x", shasum[j] );
|
|
|
|
printf(" %s\n", file);
|
|
|
|
fclose(f);
|
|
}
|
|
|
|
|