forked from wolfSSL/wolfssl
		
	
		
			
				
	
	
		
			873 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			873 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* client.c
 | 
						|
 *
 | 
						|
 * Copyright (C) 2006-2014 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
 | 
						|
 */
 | 
						|
 | 
						|
#ifdef HAVE_CONFIG_H
 | 
						|
        #include <config.h>
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(CYASSL_MDK_ARM)
 | 
						|
        #include <stdio.h>
 | 
						|
        #include <string.h>
 | 
						|
 | 
						|
        #if defined(CYASSL_MDK5)
 | 
						|
            #include "cmsis_os.h"
 | 
						|
            #include "rl_fs.h" 
 | 
						|
            #include "rl_net.h" 
 | 
						|
        #else
 | 
						|
            #include "rtl.h"
 | 
						|
        #endif
 | 
						|
 | 
						|
        #include "cyassl_MDK_ARM.h"
 | 
						|
#endif
 | 
						|
 | 
						|
#include <cyassl/ctaocrypt/settings.h>
 | 
						|
 | 
						|
#if !defined(CYASSL_TRACK_MEMORY) && !defined(NO_MAIN_DRIVER)
 | 
						|
    /* in case memory tracker wants stats */
 | 
						|
    #define CYASSL_TRACK_MEMORY
 | 
						|
#endif
 | 
						|
 | 
						|
#include <cyassl/ssl.h>
 | 
						|
 | 
						|
#include <cyassl/test.h>
 | 
						|
 | 
						|
#include "examples/client/client.h"
 | 
						|
 | 
						|
 | 
						|
#ifdef CYASSL_CALLBACKS
 | 
						|
    int handShakeCB(HandShakeInfo*);
 | 
						|
    int timeoutCB(TimeoutInfo*);
 | 
						|
    Timeval timeout;
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
static void NonBlockingSSL_Connect(CYASSL* ssl)
 | 
						|
{
 | 
						|
#ifndef CYASSL_CALLBACKS
 | 
						|
    int ret = CyaSSL_connect(ssl);
 | 
						|
#else
 | 
						|
    int ret = CyaSSL_connect_ex(ssl, handShakeCB, timeoutCB, timeout);
 | 
						|
#endif
 | 
						|
    int error = CyaSSL_get_error(ssl, 0);
 | 
						|
    SOCKET_T sockfd = (SOCKET_T)CyaSSL_get_fd(ssl);
 | 
						|
    int select_ret;
 | 
						|
 | 
						|
    while (ret != SSL_SUCCESS && (error == SSL_ERROR_WANT_READ ||
 | 
						|
                                  error == SSL_ERROR_WANT_WRITE)) {
 | 
						|
        int currTimeout = 1;
 | 
						|
 | 
						|
        if (error == SSL_ERROR_WANT_READ)
 | 
						|
            printf("... client would read block\n");
 | 
						|
        else
 | 
						|
            printf("... client would write block\n");
 | 
						|
 | 
						|
#ifdef CYASSL_DTLS
 | 
						|
        currTimeout = CyaSSL_dtls_get_current_timeout(ssl);
 | 
						|
#endif
 | 
						|
        select_ret = tcp_select(sockfd, currTimeout);
 | 
						|
 | 
						|
        if ((select_ret == TEST_RECV_READY) ||
 | 
						|
                                        (select_ret == TEST_ERROR_READY)) {
 | 
						|
            #ifndef CYASSL_CALLBACKS
 | 
						|
                    ret = CyaSSL_connect(ssl);
 | 
						|
            #else
 | 
						|
                ret = CyaSSL_connect_ex(ssl,handShakeCB,timeoutCB,timeout);
 | 
						|
            #endif
 | 
						|
            error = CyaSSL_get_error(ssl, 0);
 | 
						|
        }
 | 
						|
        else if (select_ret == TEST_TIMEOUT && !CyaSSL_dtls(ssl)) {
 | 
						|
            error = SSL_ERROR_WANT_READ;
 | 
						|
        }
 | 
						|
#ifdef CYASSL_DTLS
 | 
						|
        else if (select_ret == TEST_TIMEOUT && CyaSSL_dtls(ssl) &&
 | 
						|
                                            CyaSSL_dtls_got_timeout(ssl) >= 0) {
 | 
						|
            error = SSL_ERROR_WANT_READ;
 | 
						|
        }
 | 
						|
#endif
 | 
						|
        else {
 | 
						|
            error = SSL_FATAL_ERROR;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    if (ret != SSL_SUCCESS)
 | 
						|
        err_sys("SSL_connect failed");
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static void Usage(void)
 | 
						|
{
 | 
						|
    printf("client "    LIBCYASSL_VERSION_STRING
 | 
						|
           " NOTE: All files relative to CyaSSL home dir\n");
 | 
						|
    printf("-?          Help, print this usage\n");
 | 
						|
    printf("-h <host>   Host to connect to, default %s\n", yasslIP);
 | 
						|
    printf("-p <num>    Port to connect on, not 0, default %d\n", yasslPort);
 | 
						|
    printf("-v <num>    SSL version [0-3], SSLv3(0) - TLS1.2(3)), default %d\n",
 | 
						|
                                 CLIENT_DEFAULT_VERSION);
 | 
						|
    printf("-l <str>    Cipher list\n");
 | 
						|
    printf("-c <file>   Certificate file,           default %s\n", cliCert);
 | 
						|
    printf("-k <file>   Key file,                   default %s\n", cliKey);
 | 
						|
    printf("-A <file>   Certificate Authority file, default %s\n", caCert);
 | 
						|
    printf("-b <num>    Benchmark <num> connections and print stats\n");
 | 
						|
    printf("-s          Use pre Shared keys\n");
 | 
						|
    printf("-t          Track CyaSSL memory use\n");
 | 
						|
    printf("-d          Disable peer checks\n");
 | 
						|
    printf("-D          Override Date Errors example\n");
 | 
						|
    printf("-g          Send server HTTP GET\n");
 | 
						|
    printf("-u          Use UDP DTLS,"
 | 
						|
           " add -v 2 for DTLSv1 (default), -v 3 for DTLSv1.2\n");
 | 
						|
    printf("-m          Match domain name in cert\n");
 | 
						|
    printf("-N          Use Non-blocking sockets\n");
 | 
						|
    printf("-r          Resume session\n");
 | 
						|
    printf("-f          Fewer packets/group messages\n");
 | 
						|
    printf("-x          Disable client cert/key loading\n");
 | 
						|
#ifdef SHOW_SIZES
 | 
						|
    printf("-z          Print structure sizes\n");
 | 
						|
#endif
 | 
						|
#ifdef HAVE_SNI
 | 
						|
    printf("-S <str>    Use Host Name Indication\n");
 | 
						|
#endif
 | 
						|
#ifdef HAVE_MAX_FRAGMENT
 | 
						|
    printf("-L <num>    Use Maximum Fragment Length [1-5]\n");
 | 
						|
#endif
 | 
						|
#ifdef HAVE_TRUNCATED_HMAC
 | 
						|
    printf("-T          Use Truncated HMAC\n");
 | 
						|
#endif
 | 
						|
#ifdef HAVE_OCSP
 | 
						|
    printf("-o          Perform OCSP lookup on peer certificate\n");
 | 
						|
    printf("-O <url>    Perform OCSP lookup using <url> as responder\n");
 | 
						|
#endif
 | 
						|
#ifdef ATOMIC_USER
 | 
						|
    printf("-U          Atomic User Record Layer Callbacks\n");
 | 
						|
#endif
 | 
						|
#ifdef HAVE_PK_CALLBACKS 
 | 
						|
    printf("-P          Public Key Callbacks\n");
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
THREAD_RETURN CYASSL_THREAD client_test(void* args)
 | 
						|
{
 | 
						|
    SOCKET_T sockfd = 0;
 | 
						|
 | 
						|
    CYASSL_METHOD*  method  = 0;
 | 
						|
    CYASSL_CTX*     ctx     = 0;
 | 
						|
    CYASSL*         ssl     = 0;
 | 
						|
    
 | 
						|
    CYASSL*         sslResume = 0;
 | 
						|
    CYASSL_SESSION* session = 0;
 | 
						|
    char         resumeMsg[] = "resuming cyassl!";
 | 
						|
    int          resumeSz    = sizeof(resumeMsg);
 | 
						|
 | 
						|
    char msg[32] = "hello cyassl!";   /* GET may make bigger */
 | 
						|
    char reply[80];
 | 
						|
    int  input;
 | 
						|
    int  msgSz = (int)strlen(msg);
 | 
						|
 | 
						|
    word16 port   = yasslPort;
 | 
						|
    char* host   = (char*)yasslIP;
 | 
						|
    const char* domain = "www.yassl.com";
 | 
						|
 | 
						|
    int    ch;
 | 
						|
    int    version = CLIENT_INVALID_VERSION;
 | 
						|
    int    usePsk   = 0;
 | 
						|
    int    sendGET  = 0;
 | 
						|
    int    benchmark = 0;
 | 
						|
    int    doDTLS    = 0;
 | 
						|
    int    matchName = 0;
 | 
						|
    int    doPeerCheck = 1;
 | 
						|
    int    nonBlocking = 0;
 | 
						|
    int    resumeSession = 0;
 | 
						|
    int    trackMemory   = 0;
 | 
						|
    int    useClientCert = 1;
 | 
						|
    int    fewerPackets  = 0;
 | 
						|
    int    atomicUser    = 0;
 | 
						|
    int    pkCallbacks   = 0;
 | 
						|
    int    overrideDateErrors = 0;
 | 
						|
    char*  cipherList = NULL;
 | 
						|
    const char* verifyCert = caCert;
 | 
						|
    const char* ourCert    = cliCert;
 | 
						|
    const char* ourKey     = cliKey;
 | 
						|
 | 
						|
#ifdef HAVE_SNI
 | 
						|
    char*  sniHostName = NULL;
 | 
						|
#endif
 | 
						|
#ifdef HAVE_MAX_FRAGMENT
 | 
						|
    byte maxFragment = 0;
 | 
						|
#endif
 | 
						|
#ifdef HAVE_TRUNCATED_HMAC
 | 
						|
    byte  truncatedHMAC = 0;
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
#ifdef HAVE_OCSP
 | 
						|
    int    useOcsp  = 0;
 | 
						|
    char*  ocspUrl  = NULL;
 | 
						|
#endif
 | 
						|
 | 
						|
    int     argc = ((func_args*)args)->argc;
 | 
						|
    char**  argv = ((func_args*)args)->argv;
 | 
						|
 | 
						|
    ((func_args*)args)->return_code = -1; /* error state */
 | 
						|
 | 
						|
#ifdef NO_RSA
 | 
						|
    verifyCert = (char*)eccCert;
 | 
						|
    ourCert    = (char*)cliEccCert;
 | 
						|
    ourKey     = (char*)cliEccKey;
 | 
						|
#endif
 | 
						|
    (void)resumeSz;
 | 
						|
    (void)session;
 | 
						|
    (void)sslResume;
 | 
						|
    (void)trackMemory;
 | 
						|
    (void)atomicUser;
 | 
						|
    (void)pkCallbacks;
 | 
						|
 | 
						|
    StackTrap();
 | 
						|
 | 
						|
    while ((ch = mygetopt(argc, argv,
 | 
						|
                          "?gdDusmNrtfxUPh:p:v:l:A:c:k:b:zS:L:ToO:")) != -1) {
 | 
						|
        switch (ch) {
 | 
						|
            case '?' :
 | 
						|
                Usage();
 | 
						|
                exit(EXIT_SUCCESS);
 | 
						|
 | 
						|
            case 'g' :
 | 
						|
                sendGET = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'd' :
 | 
						|
                doPeerCheck = 0;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'D' :
 | 
						|
                overrideDateErrors = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'u' :
 | 
						|
                doDTLS  = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 's' :
 | 
						|
                usePsk = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 't' :
 | 
						|
            #ifdef USE_CYASSL_MEMORY
 | 
						|
                trackMemory = 1;
 | 
						|
            #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'm' :
 | 
						|
                matchName = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'x' :
 | 
						|
                useClientCert = 0;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'f' :
 | 
						|
                fewerPackets = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'U' :
 | 
						|
            #ifdef ATOMIC_USER
 | 
						|
                atomicUser = 1;
 | 
						|
            #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'P' :
 | 
						|
            #ifdef HAVE_PK_CALLBACKS 
 | 
						|
                pkCallbacks = 1;
 | 
						|
            #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'h' :
 | 
						|
                host   = myoptarg;
 | 
						|
                domain = myoptarg;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'p' :
 | 
						|
                port = (word16)atoi(myoptarg);
 | 
						|
                #if !defined(NO_MAIN_DRIVER) || defined(USE_WINDOWS_API)
 | 
						|
                    if (port == 0)
 | 
						|
                        err_sys("port number cannot be 0");
 | 
						|
                #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'v' :
 | 
						|
                version = atoi(myoptarg);
 | 
						|
                if (version < 0 || version > 3) {
 | 
						|
                    Usage();
 | 
						|
                    exit(MY_EX_USAGE);
 | 
						|
                }
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'l' :
 | 
						|
                cipherList = myoptarg;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'A' :
 | 
						|
                verifyCert = myoptarg;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'c' :
 | 
						|
                ourCert = myoptarg;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'k' :
 | 
						|
                ourKey = myoptarg;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'b' :
 | 
						|
                benchmark = atoi(myoptarg);
 | 
						|
                if (benchmark < 0 || benchmark > 1000000) {
 | 
						|
                    Usage();
 | 
						|
                    exit(MY_EX_USAGE);
 | 
						|
                }
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'N' :
 | 
						|
                nonBlocking = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'r' :
 | 
						|
                resumeSession = 1;
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'z' :
 | 
						|
                #ifndef CYASSL_LEANPSK
 | 
						|
                    CyaSSL_GetObjectSize();
 | 
						|
                #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'S' :
 | 
						|
                #ifdef HAVE_SNI
 | 
						|
                    sniHostName = myoptarg;
 | 
						|
                #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'L' :
 | 
						|
                #ifdef HAVE_MAX_FRAGMENT
 | 
						|
                    maxFragment = atoi(myoptarg);
 | 
						|
                    if (maxFragment < CYASSL_MFL_2_9 ||
 | 
						|
                                                maxFragment > CYASSL_MFL_2_13) {
 | 
						|
                        Usage();
 | 
						|
                        exit(MY_EX_USAGE);
 | 
						|
                    }
 | 
						|
                #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'T' :
 | 
						|
                #ifdef HAVE_TRUNCATED_HMAC
 | 
						|
                    truncatedHMAC = 1;
 | 
						|
                #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'o' :
 | 
						|
                #ifdef HAVE_OCSP
 | 
						|
                    useOcsp = 1;
 | 
						|
                #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            case 'O' :
 | 
						|
                #ifdef HAVE_OCSP
 | 
						|
                    useOcsp = 1;
 | 
						|
                    ocspUrl = myoptarg;
 | 
						|
                #endif
 | 
						|
                break;
 | 
						|
 | 
						|
            default:
 | 
						|
                Usage();
 | 
						|
                exit(MY_EX_USAGE);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    myoptind = 0;      /* reset for test cases */
 | 
						|
 | 
						|
    /* sort out DTLS versus TLS versions */
 | 
						|
    if (version == CLIENT_INVALID_VERSION) {
 | 
						|
        if (doDTLS)
 | 
						|
            version = CLIENT_DTLS_DEFAULT_VERSION;
 | 
						|
        else
 | 
						|
            version = CLIENT_DEFAULT_VERSION;
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        if (doDTLS) {
 | 
						|
            if (version == 3)
 | 
						|
                version = -2;
 | 
						|
            else
 | 
						|
                version = -1;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
#ifdef USE_CYASSL_MEMORY
 | 
						|
    if (trackMemory)
 | 
						|
        InitMemoryTracker(); 
 | 
						|
#endif
 | 
						|
 | 
						|
    switch (version) {
 | 
						|
#ifndef NO_OLD_TLS
 | 
						|
        case 0:
 | 
						|
            method = CyaSSLv3_client_method();
 | 
						|
            break;
 | 
						|
                
 | 
						|
                
 | 
						|
    #ifndef NO_TLS
 | 
						|
        case 1:
 | 
						|
            method = CyaTLSv1_client_method();
 | 
						|
            break;
 | 
						|
 | 
						|
        case 2:
 | 
						|
            method = CyaTLSv1_1_client_method();
 | 
						|
            break;
 | 
						|
    #endif /* NO_TLS */
 | 
						|
                
 | 
						|
#endif  /* NO_OLD_TLS */
 | 
						|
                
 | 
						|
#ifndef NO_TLS
 | 
						|
        case 3:
 | 
						|
            method = CyaTLSv1_2_client_method();
 | 
						|
            break;
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef CYASSL_DTLS
 | 
						|
        case -1:
 | 
						|
            method = CyaDTLSv1_client_method();
 | 
						|
            break;
 | 
						|
 | 
						|
        case -2:
 | 
						|
            method = CyaDTLSv1_2_client_method();
 | 
						|
            break;
 | 
						|
#endif
 | 
						|
 | 
						|
        default:
 | 
						|
            err_sys("Bad SSL version");
 | 
						|
            break;
 | 
						|
    }
 | 
						|
 | 
						|
    if (method == NULL)
 | 
						|
        err_sys("unable to get method");
 | 
						|
 | 
						|
    ctx = CyaSSL_CTX_new(method);
 | 
						|
    if (ctx == NULL)
 | 
						|
        err_sys("unable to get ctx");
 | 
						|
 | 
						|
    if (cipherList)
 | 
						|
        if (CyaSSL_CTX_set_cipher_list(ctx, cipherList) != SSL_SUCCESS)
 | 
						|
            err_sys("client can't set cipher list 1");
 | 
						|
 | 
						|
#ifdef CYASSL_LEANPSK
 | 
						|
    usePsk = 1;
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(NO_RSA) && !defined(HAVE_ECC)
 | 
						|
    usePsk = 1;
 | 
						|
#endif
 | 
						|
 | 
						|
    if (fewerPackets)
 | 
						|
        CyaSSL_CTX_set_group_messages(ctx);
 | 
						|
 | 
						|
    if (usePsk) {
 | 
						|
#ifndef NO_PSK
 | 
						|
        CyaSSL_CTX_set_psk_client_callback(ctx, my_psk_client_cb);
 | 
						|
        if (cipherList == NULL) {
 | 
						|
            const char *defaultCipherList;
 | 
						|
            #ifdef HAVE_NULL_CIPHER
 | 
						|
                defaultCipherList = "PSK-NULL-SHA256";
 | 
						|
            #else
 | 
						|
                defaultCipherList = "PSK-AES128-CBC-SHA256";
 | 
						|
            #endif
 | 
						|
            if (CyaSSL_CTX_set_cipher_list(ctx,defaultCipherList) !=SSL_SUCCESS)
 | 
						|
                err_sys("client can't set cipher list 2");
 | 
						|
        }
 | 
						|
#endif
 | 
						|
        useClientCert = 0;
 | 
						|
    }
 | 
						|
 | 
						|
#if defined(OPENSSL_EXTRA) || defined(HAVE_WEBSERVER)
 | 
						|
    CyaSSL_CTX_set_default_passwd_cb(ctx, PasswordCallBack);
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(CYASSL_SNIFFER) && !defined(HAVE_NTRU) && !defined(HAVE_ECC)
 | 
						|
    if (cipherList == NULL) {
 | 
						|
        /* don't use EDH, can't sniff tmp keys */
 | 
						|
        if (CyaSSL_CTX_set_cipher_list(ctx, "AES256-SHA256") != SSL_SUCCESS) {
 | 
						|
            err_sys("client can't set cipher list 3");
 | 
						|
        }
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_OCSP
 | 
						|
    if (useOcsp) {
 | 
						|
        if (ocspUrl != NULL) {
 | 
						|
            CyaSSL_CTX_SetOCSP_OverrideURL(ctx, ocspUrl);
 | 
						|
            CyaSSL_CTX_EnableOCSP(ctx, CYASSL_OCSP_NO_NONCE
 | 
						|
                                                    | CYASSL_OCSP_URL_OVERRIDE);
 | 
						|
        }
 | 
						|
        else
 | 
						|
            CyaSSL_CTX_EnableOCSP(ctx, CYASSL_OCSP_NO_NONCE);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef USER_CA_CB
 | 
						|
    CyaSSL_CTX_SetCACb(ctx, CaCb);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef VERIFY_CALLBACK
 | 
						|
    CyaSSL_CTX_set_verify(ctx, SSL_VERIFY_PEER, myVerify);
 | 
						|
#endif
 | 
						|
#if !defined(NO_FILESYSTEM) && !defined(NO_CERTS)
 | 
						|
    if (useClientCert){
 | 
						|
        if (CyaSSL_CTX_use_certificate_chain_file(ctx, ourCert) != SSL_SUCCESS)
 | 
						|
            err_sys("can't load client cert file, check file and run from"
 | 
						|
                    " CyaSSL home dir");
 | 
						|
 | 
						|
        if (CyaSSL_CTX_use_PrivateKey_file(ctx, ourKey, SSL_FILETYPE_PEM)
 | 
						|
                                         != SSL_SUCCESS)
 | 
						|
            err_sys("can't load client private key file, check file and run "
 | 
						|
                    "from CyaSSL home dir");
 | 
						|
    }
 | 
						|
 | 
						|
    if (!usePsk) {
 | 
						|
        if (CyaSSL_CTX_load_verify_locations(ctx, verifyCert, 0) != SSL_SUCCESS)
 | 
						|
                err_sys("can't load ca file, Please run from CyaSSL home dir");
 | 
						|
    }
 | 
						|
#endif
 | 
						|
#if !defined(NO_CERTS)
 | 
						|
    if (!usePsk && doPeerCheck == 0)
 | 
						|
        CyaSSL_CTX_set_verify(ctx, SSL_VERIFY_NONE, 0);
 | 
						|
    if (!usePsk && overrideDateErrors == 1)
 | 
						|
        CyaSSL_CTX_set_verify(ctx, SSL_VERIFY_PEER, myDateCb);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_CAVIUM
 | 
						|
    CyaSSL_CTX_UseCavium(ctx, CAVIUM_DEV_ID);
 | 
						|
#endif
 | 
						|
 | 
						|
#ifdef HAVE_SNI
 | 
						|
    if (sniHostName)
 | 
						|
        if (CyaSSL_CTX_UseSNI(ctx, 0, sniHostName, XSTRLEN(sniHostName))
 | 
						|
                                                                 != SSL_SUCCESS)
 | 
						|
            err_sys("UseSNI failed");
 | 
						|
#endif
 | 
						|
#ifdef HAVE_MAX_FRAGMENT
 | 
						|
    if (maxFragment)
 | 
						|
        if (CyaSSL_CTX_UseMaxFragment(ctx, maxFragment) != SSL_SUCCESS)
 | 
						|
            err_sys("UseMaxFragment failed");
 | 
						|
#endif
 | 
						|
#ifdef HAVE_TRUNCATED_HMAC
 | 
						|
    if (truncatedHMAC)
 | 
						|
        if (CyaSSL_CTX_UseTruncatedHMAC(ctx) != SSL_SUCCESS)
 | 
						|
            err_sys("UseTruncatedHMAC failed");
 | 
						|
#endif
 | 
						|
 | 
						|
    if (benchmark) {
 | 
						|
        /* time passed in number of connects give average */
 | 
						|
        int times = benchmark;
 | 
						|
        int i = 0;
 | 
						|
 | 
						|
        double start = current_time(), avg;
 | 
						|
 | 
						|
        for (i = 0; i < times; i++) {
 | 
						|
            tcp_connect(&sockfd, host, port, doDTLS);
 | 
						|
 | 
						|
            ssl = CyaSSL_new(ctx);
 | 
						|
            CyaSSL_set_fd(ssl, sockfd);
 | 
						|
            if (CyaSSL_connect(ssl) != SSL_SUCCESS)
 | 
						|
                err_sys("SSL_connect failed");
 | 
						|
 | 
						|
            CyaSSL_shutdown(ssl);
 | 
						|
            CyaSSL_free(ssl);
 | 
						|
            CloseSocket(sockfd);
 | 
						|
        }
 | 
						|
        avg = current_time() - start;
 | 
						|
        avg /= times;
 | 
						|
        avg *= 1000;   /* milliseconds */
 | 
						|
        printf("CyaSSL_connect avg took: %8.3f milliseconds\n", avg);
 | 
						|
 | 
						|
        CyaSSL_CTX_free(ctx);
 | 
						|
        ((func_args*)args)->return_code = 0;
 | 
						|
 | 
						|
        exit(EXIT_SUCCESS);
 | 
						|
    }
 | 
						|
    
 | 
						|
    #if defined(CYASSL_MDK_ARM)
 | 
						|
    CyaSSL_CTX_set_verify(ctx, SSL_VERIFY_NONE, 0);
 | 
						|
    #endif
 | 
						|
    
 | 
						|
    ssl = CyaSSL_new(ctx);
 | 
						|
 | 
						|
    if (ssl == NULL)
 | 
						|
        err_sys("unable to get SSL object");
 | 
						|
    if (doDTLS) {
 | 
						|
        SOCKADDR_IN_T addr;
 | 
						|
        build_addr(&addr, host, port, 1);
 | 
						|
        CyaSSL_dtls_set_peer(ssl, &addr, sizeof(addr));
 | 
						|
        tcp_socket(&sockfd, 1);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
        tcp_connect(&sockfd, host, port, 0);
 | 
						|
    }
 | 
						|
 | 
						|
#ifdef HAVE_POLY1305
 | 
						|
    /* use old poly to connect with google server */
 | 
						|
    if (!XMEMCMP(domain, "www.google.com", 15)) {
 | 
						|
        if (CyaSSL_use_old_poly(ssl, 1) != 0)
 | 
						|
            err_sys("unable to set to old poly");
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    CyaSSL_set_fd(ssl, sockfd);
 | 
						|
#ifdef HAVE_CRL
 | 
						|
    if (CyaSSL_EnableCRL(ssl, CYASSL_CRL_CHECKALL) != SSL_SUCCESS)
 | 
						|
        err_sys("can't enable crl check");
 | 
						|
    if (CyaSSL_LoadCRL(ssl, crlPemDir, SSL_FILETYPE_PEM, 0) != SSL_SUCCESS)
 | 
						|
        err_sys("can't load crl, check crlfile and date validity");
 | 
						|
    if (CyaSSL_SetCRL_Cb(ssl, CRL_CallBack) != SSL_SUCCESS)
 | 
						|
        err_sys("can't set crl callback");
 | 
						|
#endif
 | 
						|
#ifdef ATOMIC_USER
 | 
						|
    if (atomicUser)
 | 
						|
        SetupAtomicUser(ctx, ssl);
 | 
						|
#endif
 | 
						|
#ifdef HAVE_PK_CALLBACKS
 | 
						|
    if (pkCallbacks)
 | 
						|
        SetupPkCallbacks(ctx, ssl);
 | 
						|
#endif
 | 
						|
    if (matchName && doPeerCheck)
 | 
						|
        CyaSSL_check_domain_name(ssl, domain);
 | 
						|
#ifndef CYASSL_CALLBACKS
 | 
						|
    if (nonBlocking) {
 | 
						|
        CyaSSL_set_using_nonblock(ssl, 1);
 | 
						|
        tcp_set_nonblocking(&sockfd);
 | 
						|
        NonBlockingSSL_Connect(ssl);
 | 
						|
    }
 | 
						|
    else if (CyaSSL_connect(ssl) != SSL_SUCCESS) {
 | 
						|
        /* see note at top of README */
 | 
						|
        int  err = CyaSSL_get_error(ssl, 0);
 | 
						|
        char buffer[CYASSL_MAX_ERROR_SZ];
 | 
						|
        printf("err = %d, %s\n", err,
 | 
						|
                                CyaSSL_ERR_error_string(err, buffer));
 | 
						|
        err_sys("SSL_connect failed");
 | 
						|
        /* if you're getting an error here  */
 | 
						|
    }
 | 
						|
#else
 | 
						|
    timeout.tv_sec  = 2;
 | 
						|
    timeout.tv_usec = 0;
 | 
						|
    NonBlockingSSL_Connect(ssl);  /* will keep retrying on timeout */
 | 
						|
#endif
 | 
						|
    showPeer(ssl);
 | 
						|
 | 
						|
    if (sendGET) {
 | 
						|
        printf("SSL connect ok, sending GET...\n");
 | 
						|
        msgSz = 28;
 | 
						|
        strncpy(msg, "GET /index.html HTTP/1.0\r\n\r\n", msgSz);
 | 
						|
        msg[msgSz] = '\0';
 | 
						|
    }
 | 
						|
    if (CyaSSL_write(ssl, msg, msgSz) != msgSz)
 | 
						|
        err_sys("SSL_write failed");
 | 
						|
 | 
						|
    input = CyaSSL_read(ssl, reply, sizeof(reply)-1);
 | 
						|
    if (input > 0) {
 | 
						|
        reply[input] = 0;
 | 
						|
        printf("Server response: %s\n", reply);
 | 
						|
 | 
						|
        if (sendGET) {  /* get html */
 | 
						|
            while (1) {
 | 
						|
                input = CyaSSL_read(ssl, reply, sizeof(reply)-1);
 | 
						|
                if (input > 0) {
 | 
						|
                    reply[input] = 0;
 | 
						|
                    printf("%s\n", reply);
 | 
						|
                }
 | 
						|
                else
 | 
						|
                    break;
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
    else if (input < 0) {
 | 
						|
        int readErr = CyaSSL_get_error(ssl, 0);
 | 
						|
        if (readErr != SSL_ERROR_WANT_READ)
 | 
						|
            err_sys("CyaSSL_read failed");
 | 
						|
    }
 | 
						|
 | 
						|
#ifndef NO_SESSION_CACHE
 | 
						|
    if (resumeSession) {
 | 
						|
        if (doDTLS) {
 | 
						|
            strncpy(msg, "break", 6);
 | 
						|
            msgSz = (int)strlen(msg);
 | 
						|
            /* try to send session close */
 | 
						|
            CyaSSL_write(ssl, msg, msgSz);
 | 
						|
        }
 | 
						|
        session   = CyaSSL_get_session(ssl);
 | 
						|
        sslResume = CyaSSL_new(ctx);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    if (doDTLS == 0)            /* don't send alert after "break" command */
 | 
						|
        CyaSSL_shutdown(ssl);  /* echoserver will interpret as new conn */
 | 
						|
#ifdef ATOMIC_USER
 | 
						|
    if (atomicUser)
 | 
						|
        FreeAtomicUser(ssl);
 | 
						|
#endif
 | 
						|
    CyaSSL_free(ssl);
 | 
						|
    CloseSocket(sockfd);
 | 
						|
 | 
						|
#ifndef NO_SESSION_CACHE
 | 
						|
    if (resumeSession) {
 | 
						|
        if (doDTLS) {
 | 
						|
            SOCKADDR_IN_T addr;
 | 
						|
            #ifdef USE_WINDOWS_API 
 | 
						|
                Sleep(500);
 | 
						|
            #else
 | 
						|
                sleep(1);
 | 
						|
            #endif
 | 
						|
            build_addr(&addr, host, port, 1);
 | 
						|
            CyaSSL_dtls_set_peer(sslResume, &addr, sizeof(addr));
 | 
						|
            tcp_socket(&sockfd, 1);
 | 
						|
        }
 | 
						|
        else {
 | 
						|
            tcp_connect(&sockfd, host, port, 0);
 | 
						|
        }
 | 
						|
        CyaSSL_set_fd(sslResume, sockfd);
 | 
						|
        CyaSSL_set_session(sslResume, session);
 | 
						|
       
 | 
						|
        showPeer(sslResume);
 | 
						|
#ifndef CYASSL_CALLBACKS
 | 
						|
        if (nonBlocking) {
 | 
						|
            CyaSSL_set_using_nonblock(sslResume, 1);
 | 
						|
            tcp_set_nonblocking(&sockfd);
 | 
						|
            NonBlockingSSL_Connect(sslResume);
 | 
						|
        }
 | 
						|
        else if (CyaSSL_connect(sslResume) != SSL_SUCCESS)
 | 
						|
            err_sys("SSL resume failed");
 | 
						|
#else
 | 
						|
        timeout.tv_sec  = 2;
 | 
						|
        timeout.tv_usec = 0;
 | 
						|
        NonBlockingSSL_Connect(ssl);  /* will keep retrying on timeout */
 | 
						|
#endif
 | 
						|
 | 
						|
        if (CyaSSL_session_reused(sslResume))
 | 
						|
            printf("reused session id\n");
 | 
						|
        else
 | 
						|
            printf("didn't reuse session id!!!\n");
 | 
						|
 | 
						|
        if (CyaSSL_write(sslResume, resumeMsg, resumeSz) != resumeSz)
 | 
						|
            err_sys("SSL_write failed");
 | 
						|
 | 
						|
        if (nonBlocking) {
 | 
						|
            /* give server a chance to bounce a message back to client */
 | 
						|
            #ifdef USE_WINDOWS_API
 | 
						|
                Sleep(500);
 | 
						|
            #else
 | 
						|
                sleep(1);
 | 
						|
            #endif
 | 
						|
        }
 | 
						|
 | 
						|
        input = CyaSSL_read(sslResume, reply, sizeof(reply)-1);
 | 
						|
        if (input > 0) {
 | 
						|
            reply[input] = 0;
 | 
						|
            printf("Server resume response: %s\n", reply);
 | 
						|
        }
 | 
						|
 | 
						|
        /* try to send session break */
 | 
						|
        CyaSSL_write(sslResume, msg, msgSz); 
 | 
						|
 | 
						|
        CyaSSL_shutdown(sslResume);
 | 
						|
        CyaSSL_free(sslResume);
 | 
						|
        CloseSocket(sockfd);
 | 
						|
    }
 | 
						|
#endif /* NO_SESSION_CACHE */
 | 
						|
 | 
						|
    CyaSSL_CTX_free(ctx);
 | 
						|
 | 
						|
    ((func_args*)args)->return_code = 0;
 | 
						|
 | 
						|
#ifdef USE_CYASSL_MEMORY
 | 
						|
    if (trackMemory)
 | 
						|
        ShowMemoryTracker();
 | 
						|
#endif /* USE_CYASSL_MEMORY */
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* so overall tests can pull in test function */
 | 
						|
#ifndef NO_MAIN_DRIVER
 | 
						|
 | 
						|
    int main(int argc, char** argv)
 | 
						|
    {
 | 
						|
        func_args args;
 | 
						|
 | 
						|
#ifdef HAVE_CAVIUM
 | 
						|
        int ret = OpenNitroxDevice(CAVIUM_DIRECT, CAVIUM_DEV_ID);
 | 
						|
        if (ret != 0)
 | 
						|
            err_sys("Cavium OpenNitroxDevice failed");
 | 
						|
#endif /* HAVE_CAVIUM */
 | 
						|
 | 
						|
        StartTCP();
 | 
						|
 | 
						|
        args.argc = argc;
 | 
						|
        args.argv = argv;
 | 
						|
 | 
						|
        CyaSSL_Init();
 | 
						|
#if defined(DEBUG_CYASSL) && !defined(CYASSL_MDK_SHELL) && !defined(STACK_TRAP)
 | 
						|
        CyaSSL_Debugging_ON();
 | 
						|
#endif
 | 
						|
        if (CurrentDir("client"))
 | 
						|
            ChangeDirBack(2);
 | 
						|
        else if (CurrentDir("Debug") || CurrentDir("Release"))
 | 
						|
            ChangeDirBack(3);
 | 
						|
  
 | 
						|
#ifdef HAVE_STACK_SIZE
 | 
						|
        StackSizeCheck(&args, client_test);
 | 
						|
#else 
 | 
						|
        client_test(&args);
 | 
						|
#endif
 | 
						|
        CyaSSL_Cleanup();
 | 
						|
 | 
						|
#ifdef HAVE_CAVIUM
 | 
						|
        CspShutdown(CAVIUM_DEV_ID);
 | 
						|
#endif
 | 
						|
        return args.return_code;
 | 
						|
    }
 | 
						|
 | 
						|
    int myoptind = 0;
 | 
						|
    char* myoptarg = NULL;
 | 
						|
 | 
						|
#endif /* NO_MAIN_DRIVER */
 | 
						|
 | 
						|
 | 
						|
 | 
						|
#ifdef CYASSL_CALLBACKS
 | 
						|
 | 
						|
    int handShakeCB(HandShakeInfo* info)
 | 
						|
    {
 | 
						|
        (void)info;
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
 | 
						|
 | 
						|
    int timeoutCB(TimeoutInfo* info)
 | 
						|
    {
 | 
						|
        (void)info;
 | 
						|
        return 0;
 | 
						|
    }
 | 
						|
 | 
						|
#endif
 | 
						|
 |