mirror of
				https://github.com/cuberite/polarssl.git
				synced 2025-10-31 03:30:35 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			910 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			910 lines
		
	
	
		
			28 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  *  SSL client with options
 | |
|  *
 | |
|  *  Copyright (C) 2006-2013, Brainspark B.V.
 | |
|  *
 | |
|  *  This file is part of PolarSSL (http://www.polarssl.org)
 | |
|  *  Lead Maintainer: Paul Bakker <polarssl_maintainer at polarssl.org>
 | |
|  *
 | |
|  *  All rights reserved.
 | |
|  *
 | |
|  *  This program 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.
 | |
|  *
 | |
|  *  This program 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.
 | |
|  */
 | |
| 
 | |
| #ifndef _CRT_SECURE_NO_DEPRECATE
 | |
| #define _CRT_SECURE_NO_DEPRECATE 1
 | |
| #endif
 | |
| 
 | |
| #if defined(_WIN32)
 | |
| #include <windows.h>
 | |
| #endif
 | |
| 
 | |
| #include <string.h>
 | |
| #include <stdlib.h>
 | |
| #include <stdio.h>
 | |
| 
 | |
| #include "polarssl/config.h"
 | |
| 
 | |
| #include "polarssl/net.h"
 | |
| #include "polarssl/ssl.h"
 | |
| #include "polarssl/entropy.h"
 | |
| #include "polarssl/ctr_drbg.h"
 | |
| #include "polarssl/certs.h"
 | |
| #include "polarssl/x509.h"
 | |
| #include "polarssl/error.h"
 | |
| 
 | |
| #if defined(POLARSSL_SSL_CACHE_C)
 | |
| #include "polarssl/ssl_cache.h"
 | |
| #endif
 | |
| 
 | |
| #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
 | |
| #include "polarssl/memory.h"
 | |
| #endif
 | |
| 
 | |
| #define DFL_SERVER_PORT         4433
 | |
| #define DFL_REQUEST_PAGE        "/"
 | |
| #define DFL_DEBUG_LEVEL         0
 | |
| #define DFL_CA_FILE             ""
 | |
| #define DFL_CA_PATH             ""
 | |
| #define DFL_CRT_FILE            ""
 | |
| #define DFL_KEY_FILE            ""
 | |
| #define DFL_PSK                 ""
 | |
| #define DFL_PSK_IDENTITY        "Client_identity"
 | |
| #define DFL_FORCE_CIPHER        0
 | |
| #define DFL_RENEGOTIATION       SSL_RENEGOTIATION_ENABLED
 | |
| #define DFL_ALLOW_LEGACY        SSL_LEGACY_NO_RENEGOTIATION
 | |
| #define DFL_MIN_VERSION         -1
 | |
| #define DFL_MAX_VERSION         -1
 | |
| #define DFL_AUTH_MODE           SSL_VERIFY_OPTIONAL
 | |
| #define DFL_MFL_CODE            SSL_MAX_FRAG_LEN_NONE
 | |
| #define DFL_TICKETS             SSL_SESSION_TICKETS_ENABLED
 | |
| 
 | |
| #define LONG_RESPONSE "<p>01-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n" \
 | |
|     "02-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
 | |
|     "03-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
 | |
|     "04-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
 | |
|     "05-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
 | |
|     "06-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
 | |
|     "07-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah</p>\r\n"
 | |
| 
 | |
| /* Uncomment LONG_RESPONSE at the end of HTTP_RESPONSE to test sending longer
 | |
|  * packets (for fragmentation purposes) */
 | |
| #define HTTP_RESPONSE \
 | |
|     "HTTP/1.0 200 OK\r\nContent-Type: text/html\r\n\r\n" \
 | |
|     "<h2>PolarSSL Test Server</h2>\r\n" \
 | |
|     "<p>Successful connection using: %s</p>\r\n" // LONG_RESPONSE
 | |
| 
 | |
| /*
 | |
|  * global options
 | |
|  */
 | |
| struct options
 | |
| {
 | |
|     int server_port;            /* port on which the ssl service runs       */
 | |
|     int debug_level;            /* level of debugging                       */
 | |
|     const char *ca_file;        /* the file with the CA certificate(s)      */
 | |
|     const char *ca_path;        /* the path with the CA certificate(s) reside */
 | |
|     const char *crt_file;       /* the file with the client certificate     */
 | |
|     const char *key_file;       /* the file with the client key             */
 | |
|     const char *psk;            /* the pre-shared key                       */
 | |
|     const char *psk_identity;   /* the pre-shared key identity              */
 | |
|     int force_ciphersuite[2];   /* protocol/ciphersuite to use, or all      */
 | |
|     int renegotiation;          /* enable / disable renegotiation           */
 | |
|     int allow_legacy;           /* allow legacy renegotiation               */
 | |
|     int min_version;            /* minimum protocol version accepted        */
 | |
|     int max_version;            /* maximum protocol version accepted        */
 | |
|     int auth_mode;              /* verify mode for connection               */
 | |
|     unsigned char mfl_code;     /* code for maximum fragment length         */
 | |
|     int tickets;                /* enable / disable session tickets         */
 | |
| } opt;
 | |
| 
 | |
| static void my_debug( void *ctx, int level, const char *str )
 | |
| {
 | |
|     if( level < opt.debug_level )
 | |
|     {
 | |
|         fprintf( (FILE *) ctx, "%s", str );
 | |
|         fflush(  (FILE *) ctx  );
 | |
|     }
 | |
| }
 | |
| 
 | |
| #if defined(POLARSSL_X509_PARSE_C)
 | |
| #if defined(POLARSSL_FS_IO)
 | |
| #define USAGE_IO \
 | |
|     "    ca_file=%%s          The single file containing the top-level CA(s) you fully trust\n" \
 | |
|     "                        default: \"\" (pre-loaded)\n" \
 | |
|     "    ca_path=%%s          The path containing the top-level CA(s) you fully trust\n" \
 | |
|     "                        default: \"\" (pre-loaded) (overrides ca_file)\n" \
 | |
|     "    crt_file=%%s         Your own cert and chain (in bottom to top order, top may be omitted)\n" \
 | |
|     "                        default: \"\" (pre-loaded)\n" \
 | |
|     "    key_file=%%s         default: \"\" (pre-loaded)\n"
 | |
| #else
 | |
| #define USAGE_IO \
 | |
|     "\n"                                                    \
 | |
|     "    No file operations available (POLARSSL_FS_IO not defined)\n" \
 | |
|     "\n"
 | |
| #endif /* POLARSSL_FS_IO */
 | |
| #else
 | |
| #define USAGE_IO ""
 | |
| #endif /* POLARSSL_X509_PARSE_C */
 | |
| 
 | |
| #if defined(POLARSSL_KEY_EXCHANGE_PSK_ENABLED)
 | |
| #define USAGE_PSK                                                   \
 | |
|     "    psk=%%s              default: \"\" (in hex, without 0x)\n" \
 | |
|     "    psk_identity=%%s     default: \"Client_identity\"\n"
 | |
| #else
 | |
| #define USAGE_PSK ""
 | |
| #endif /* POLARSSL_KEY_EXCHANGE_PSK_ENABLED */
 | |
| 
 | |
| #if defined(POLARSSL_SSL_SESSION_TICKETS)
 | |
| #define USAGE_TICKETS                                       \
 | |
|     "    tickets=%%d          default: 1 (enabled)\n"
 | |
| #else
 | |
| #define USAGE_TICKETS ""
 | |
| #endif /* POLARSSL_SSL_SESSION_TICKETS */
 | |
| 
 | |
| #if defined(POLARSSL_SSL_MAX_FRAGMENT_LENGTH)
 | |
| #define USAGE_MAX_FRAG_LEN                                      \
 | |
|     "    max_frag_len=%%d     default: 16384 (tls default)\n"   \
 | |
|     "                        options: 512, 1024, 2048, 4096\n"
 | |
| #else
 | |
| #define USAGE_MAX_FRAG_LEN ""
 | |
| #endif /* POLARSSL_SSL_MAX_FRAGMENT_LENGTH */
 | |
| 
 | |
| #define USAGE \
 | |
|     "\n usage: ssl_server2 param=<>...\n"                   \
 | |
|     "\n acceptable parameters:\n"                           \
 | |
|     "    server_port=%%d      default: 4433\n"              \
 | |
|     "    debug_level=%%d      default: 0 (disabled)\n"      \
 | |
|     USAGE_IO                                                \
 | |
|     "    request_page=%%s     default: \".\"\n"             \
 | |
|     "    renegotiation=%%d    default: 1 (enabled)\n"       \
 | |
|     USAGE_TICKETS                                           \
 | |
|     "    allow_legacy=%%d     default: 0 (disabled)\n"      \
 | |
|     "    min_version=%%s      default: \"ssl3\"\n"          \
 | |
|     "    max_version=%%s      default: \"tls1_2\"\n"        \
 | |
|     "    force_version=%%s    default: \"\" (none)\n"       \
 | |
|     "                        options: ssl3, tls1, tls1_1, tls1_2\n" \
 | |
|     "    auth_mode=%%s        default: \"optional\"\n"      \
 | |
|     "                        options: none, optional, required\n" \
 | |
|     USAGE_MAX_FRAG_LEN                                      \
 | |
|     USAGE_PSK                                               \
 | |
|     "\n"                                                    \
 | |
|     "    force_ciphersuite=<name>    default: all enabled\n"\
 | |
|     " acceptable ciphersuite names:\n"
 | |
| 
 | |
| #if !defined(POLARSSL_ENTROPY_C) ||  \
 | |
|     !defined(POLARSSL_SSL_TLS_C) || !defined(POLARSSL_SSL_SRV_C) || \
 | |
|     !defined(POLARSSL_NET_C) || !defined(POLARSSL_CTR_DRBG_C)
 | |
| int main( int argc, char *argv[] )
 | |
| {
 | |
|     ((void) argc);
 | |
|     ((void) argv);
 | |
| 
 | |
|     printf("POLARSSL_ENTROPY_C and/or "
 | |
|            "POLARSSL_SSL_TLS_C and/or POLARSSL_SSL_SRV_C and/or "
 | |
|            "POLARSSL_NET_C and/or POLARSSL_CTR_DRBG_C not defined.\n");
 | |
|     return( 0 );
 | |
| }
 | |
| #else
 | |
| int main( int argc, char *argv[] )
 | |
| {
 | |
|     int ret = 0, len, written, frags;
 | |
|     int listen_fd;
 | |
|     int client_fd = -1;
 | |
|     unsigned char buf[1024];
 | |
| #if defined(POLARSSL_KEY_EXCHANGE_PSK_ENABLED)
 | |
|     unsigned char psk[256];
 | |
|     size_t psk_len = 0;
 | |
| #endif
 | |
|     const char *pers = "ssl_server2";
 | |
| 
 | |
|     entropy_context entropy;
 | |
|     ctr_drbg_context ctr_drbg;
 | |
|     ssl_context ssl;
 | |
| #if defined(POLARSSL_X509_PARSE_C)
 | |
|     x509_cert cacert;
 | |
|     x509_cert srvcert;
 | |
|     pk_context pkey;
 | |
| #endif
 | |
| #if defined(POLARSSL_SSL_CACHE_C)
 | |
|     ssl_cache_context cache;
 | |
| #endif
 | |
| #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
 | |
|     unsigned char alloc_buf[100000];
 | |
| #endif
 | |
| 
 | |
|     int i;
 | |
|     char *p, *q;
 | |
|     const int *list;
 | |
| 
 | |
| #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C)
 | |
|     memory_buffer_alloc_init( alloc_buf, sizeof(alloc_buf) );
 | |
| #endif
 | |
| 
 | |
|     /*
 | |
|      * Make sure memory references are valid.
 | |
|      */
 | |
|     listen_fd = 0;
 | |
| #if defined(POLARSSL_X509_PARSE_C)
 | |
|     memset( &cacert, 0, sizeof( x509_cert ) );
 | |
|     memset( &srvcert, 0, sizeof( x509_cert ) );
 | |
|     pk_init( &pkey );
 | |
| #endif
 | |
| #if defined(POLARSSL_SSL_CACHE_C)
 | |
|     ssl_cache_init( &cache );
 | |
| #endif
 | |
| 
 | |
|     if( argc == 0 )
 | |
|     {
 | |
|     usage:
 | |
|         if( ret == 0 )
 | |
|             ret = 1;
 | |
| 
 | |
|         printf( USAGE );
 | |
| 
 | |
|         list = ssl_list_ciphersuites();
 | |
|         while( *list )
 | |
|         {
 | |
|             printf(" %-42s", ssl_get_ciphersuite_name( *list ) );
 | |
|             list++;
 | |
|             if( !*list )
 | |
|                 break;
 | |
|             printf(" %s\n", ssl_get_ciphersuite_name( *list ) );
 | |
|             list++;
 | |
|         }
 | |
|         printf("\n");
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
|     opt.server_port         = DFL_SERVER_PORT;
 | |
|     opt.debug_level         = DFL_DEBUG_LEVEL;
 | |
|     opt.ca_file             = DFL_CA_FILE;
 | |
|     opt.ca_path             = DFL_CA_PATH;
 | |
|     opt.crt_file            = DFL_CRT_FILE;
 | |
|     opt.key_file            = DFL_KEY_FILE;
 | |
|     opt.psk                 = DFL_PSK;
 | |
|     opt.psk_identity        = DFL_PSK_IDENTITY;
 | |
|     opt.force_ciphersuite[0]= DFL_FORCE_CIPHER;
 | |
|     opt.renegotiation       = DFL_RENEGOTIATION;
 | |
|     opt.allow_legacy        = DFL_ALLOW_LEGACY;
 | |
|     opt.min_version         = DFL_MIN_VERSION;
 | |
|     opt.max_version         = DFL_MAX_VERSION;
 | |
|     opt.auth_mode           = DFL_AUTH_MODE;
 | |
|     opt.mfl_code            = DFL_MFL_CODE;
 | |
|     opt.tickets             = DFL_TICKETS;
 | |
| 
 | |
|     for( i = 1; i < argc; i++ )
 | |
|     {
 | |
|         p = argv[i];
 | |
|         if( ( q = strchr( p, '=' ) ) == NULL )
 | |
|             goto usage;
 | |
|         *q++ = '\0';
 | |
| 
 | |
|         if( strcmp( p, "server_port" ) == 0 )
 | |
|         {
 | |
|             opt.server_port = atoi( q );
 | |
|             if( opt.server_port < 1 || opt.server_port > 65535 )
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "debug_level" ) == 0 )
 | |
|         {
 | |
|             opt.debug_level = atoi( q );
 | |
|             if( opt.debug_level < 0 || opt.debug_level > 65535 )
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "ca_file" ) == 0 )
 | |
|             opt.ca_file = q;
 | |
|         else if( strcmp( p, "ca_path" ) == 0 )
 | |
|             opt.ca_path = q;
 | |
|         else if( strcmp( p, "crt_file" ) == 0 )
 | |
|             opt.crt_file = q;
 | |
|         else if( strcmp( p, "key_file" ) == 0 )
 | |
|             opt.key_file = q;
 | |
|         else if( strcmp( p, "psk" ) == 0 )
 | |
|             opt.psk = q;
 | |
|         else if( strcmp( p, "psk_identity" ) == 0 )
 | |
|             opt.psk_identity = q;
 | |
|         else if( strcmp( p, "force_ciphersuite" ) == 0 )
 | |
|         {
 | |
|             opt.force_ciphersuite[0] = -1;
 | |
| 
 | |
|             opt.force_ciphersuite[0] = ssl_get_ciphersuite_id( q );
 | |
| 
 | |
|             if( opt.force_ciphersuite[0] <= 0 )
 | |
|             {
 | |
|                 ret = 2;
 | |
|                 goto usage;
 | |
|             }
 | |
|             opt.force_ciphersuite[1] = 0;
 | |
|         }
 | |
|         else if( strcmp( p, "renegotiation" ) == 0 )
 | |
|         {
 | |
|             opt.renegotiation = (atoi( q )) ? SSL_RENEGOTIATION_ENABLED :
 | |
|                                               SSL_RENEGOTIATION_DISABLED;
 | |
|         }
 | |
|         else if( strcmp( p, "allow_legacy" ) == 0 )
 | |
|         {
 | |
|             opt.allow_legacy = atoi( q );
 | |
|             if( opt.allow_legacy < 0 || opt.allow_legacy > 1 )
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "min_version" ) == 0 )
 | |
|         {
 | |
|             if( strcmp( q, "ssl3" ) == 0 )
 | |
|                 opt.min_version = SSL_MINOR_VERSION_0;
 | |
|             else if( strcmp( q, "tls1" ) == 0 )
 | |
|                 opt.min_version = SSL_MINOR_VERSION_1;
 | |
|             else if( strcmp( q, "tls1_1" ) == 0 )
 | |
|                 opt.min_version = SSL_MINOR_VERSION_2;
 | |
|             else if( strcmp( q, "tls1_2" ) == 0 )
 | |
|                 opt.min_version = SSL_MINOR_VERSION_3;
 | |
|             else
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "max_version" ) == 0 )
 | |
|         {
 | |
|             if( strcmp( q, "ssl3" ) == 0 )
 | |
|                 opt.max_version = SSL_MINOR_VERSION_0;
 | |
|             else if( strcmp( q, "tls1" ) == 0 )
 | |
|                 opt.max_version = SSL_MINOR_VERSION_1;
 | |
|             else if( strcmp( q, "tls1_1" ) == 0 )
 | |
|                 opt.max_version = SSL_MINOR_VERSION_2;
 | |
|             else if( strcmp( q, "tls1_2" ) == 0 )
 | |
|                 opt.max_version = SSL_MINOR_VERSION_3;
 | |
|             else
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "force_version" ) == 0 )
 | |
|         {
 | |
|             if( strcmp( q, "ssl3" ) == 0 )
 | |
|             {
 | |
|                 opt.min_version = SSL_MINOR_VERSION_0;
 | |
|                 opt.max_version = SSL_MINOR_VERSION_0;
 | |
|             }
 | |
|             else if( strcmp( q, "tls1" ) == 0 )
 | |
|             {
 | |
|                 opt.min_version = SSL_MINOR_VERSION_1;
 | |
|                 opt.max_version = SSL_MINOR_VERSION_1;
 | |
|             }
 | |
|             else if( strcmp( q, "tls1_1" ) == 0 )
 | |
|             {
 | |
|                 opt.min_version = SSL_MINOR_VERSION_2;
 | |
|                 opt.max_version = SSL_MINOR_VERSION_2;
 | |
|             }
 | |
|             else if( strcmp( q, "tls1_2" ) == 0 )
 | |
|             {
 | |
|                 opt.min_version = SSL_MINOR_VERSION_3;
 | |
|                 opt.max_version = SSL_MINOR_VERSION_3;
 | |
|             }
 | |
|             else
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "auth_mode" ) == 0 )
 | |
|         {
 | |
|             if( strcmp( q, "none" ) == 0 )
 | |
|                 opt.auth_mode = SSL_VERIFY_NONE;
 | |
|             else if( strcmp( q, "optional" ) == 0 )
 | |
|                 opt.auth_mode = SSL_VERIFY_OPTIONAL;
 | |
|             else if( strcmp( q, "required" ) == 0 )
 | |
|                 opt.auth_mode = SSL_VERIFY_REQUIRED;
 | |
|             else
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "max_frag_len" ) == 0 )
 | |
|         {
 | |
|             if( strcmp( q, "512" ) == 0 )
 | |
|                 opt.mfl_code = SSL_MAX_FRAG_LEN_512;
 | |
|             else if( strcmp( q, "1024" ) == 0 )
 | |
|                 opt.mfl_code = SSL_MAX_FRAG_LEN_1024;
 | |
|             else if( strcmp( q, "2048" ) == 0 )
 | |
|                 opt.mfl_code = SSL_MAX_FRAG_LEN_2048;
 | |
|             else if( strcmp( q, "4096" ) == 0 )
 | |
|                 opt.mfl_code = SSL_MAX_FRAG_LEN_4096;
 | |
|             else
 | |
|                 goto usage;
 | |
|         }
 | |
|         else if( strcmp( p, "tickets" ) == 0 )
 | |
|         {
 | |
|             opt.tickets = atoi( q );
 | |
|             if( opt.tickets < 0 || opt.tickets > 1 )
 | |
|                 goto usage;
 | |
|         }
 | |
|         else
 | |
|             goto usage;
 | |
|     }
 | |
| 
 | |
|     if( opt.force_ciphersuite[0] > 0 )
 | |
|     {
 | |
|         const ssl_ciphersuite_t *ciphersuite_info;
 | |
|         ciphersuite_info = ssl_ciphersuite_from_id( opt.force_ciphersuite[0] );
 | |
| 
 | |
|         if( opt.max_version != -1 &&
 | |
|             ciphersuite_info->min_minor_ver > opt.max_version )
 | |
|         {
 | |
|             printf("forced ciphersuite not allowed with this protocol version\n");
 | |
|             ret = 2;
 | |
|             goto usage;
 | |
|         }
 | |
|         if( opt.min_version != -1 &&
 | |
|             ciphersuite_info->max_minor_ver < opt.min_version )
 | |
|         {
 | |
|             printf("forced ciphersuite not allowed with this protocol version\n");
 | |
|             ret = 2;
 | |
|             goto usage;
 | |
|         }
 | |
|         if( opt.max_version > ciphersuite_info->max_minor_ver )
 | |
|             opt.max_version = ciphersuite_info->max_minor_ver;
 | |
|         if( opt.min_version < ciphersuite_info->min_minor_ver )
 | |
|             opt.min_version = ciphersuite_info->min_minor_ver;
 | |
|     }
 | |
| 
 | |
| #if defined(POLARSSL_KEY_EXCHANGE_PSK_ENABLED)
 | |
|     /*
 | |
|      * Unhexify the pre-shared key if any is given
 | |
|      */
 | |
|     if( strlen( opt.psk ) )
 | |
|     {
 | |
|         unsigned char c;
 | |
|         size_t j;
 | |
| 
 | |
|         if( strlen( opt.psk ) % 2 != 0 )
 | |
|         {
 | |
|             printf("pre-shared key not valid hex\n");
 | |
|             goto exit;
 | |
|         }
 | |
| 
 | |
|         psk_len = strlen( opt.psk ) / 2;
 | |
| 
 | |
|         for( j = 0; j < strlen( opt.psk ); j += 2 )
 | |
|         {
 | |
|             c = opt.psk[j];
 | |
|             if( c >= '0' && c <= '9' )
 | |
|                 c -= '0';
 | |
|             else if( c >= 'a' && c <= 'f' )
 | |
|                 c -= 'a' - 10;
 | |
|             else if( c >= 'A' && c <= 'F' )
 | |
|                 c -= 'A' - 10;
 | |
|             else
 | |
|             {
 | |
|                 printf("pre-shared key not valid hex\n");
 | |
|                 goto exit;
 | |
|             }
 | |
|             psk[ j / 2 ] = c << 4;
 | |
| 
 | |
|             c = opt.psk[j + 1];
 | |
|             if( c >= '0' && c <= '9' )
 | |
|                 c -= '0';
 | |
|             else if( c >= 'a' && c <= 'f' )
 | |
|                 c -= 'a' - 10;
 | |
|             else if( c >= 'A' && c <= 'F' )
 | |
|                 c -= 'A' - 10;
 | |
|             else
 | |
|             {
 | |
|                 printf("pre-shared key not valid hex\n");
 | |
|                 goto exit;
 | |
|             }
 | |
|             psk[ j / 2 ] |= c;
 | |
|         }
 | |
|     }
 | |
| #endif /* POLARSSL_KEY_EXCHANGE_PSK_ENABLED */
 | |
| 
 | |
|     /*
 | |
|      * 0. Initialize the RNG and the session data
 | |
|      */
 | |
|     printf( "\n  . Seeding the random number generator..." );
 | |
|     fflush( stdout );
 | |
| 
 | |
|     entropy_init( &entropy );
 | |
|     if( ( ret = ctr_drbg_init( &ctr_drbg, entropy_func, &entropy,
 | |
|                                (const unsigned char *) pers,
 | |
|                                strlen( pers ) ) ) != 0 )
 | |
|     {
 | |
|         printf( " failed\n  ! ctr_drbg_init returned -0x%x\n", -ret );
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
|     printf( " ok\n" );
 | |
| 
 | |
| #if defined(POLARSSL_X509_PARSE_C)
 | |
|     /*
 | |
|      * 1.1. Load the trusted CA
 | |
|      */
 | |
|     printf( "  . Loading the CA root certificate ..." );
 | |
|     fflush( stdout );
 | |
| 
 | |
| #if defined(POLARSSL_FS_IO)
 | |
|     if( strlen( opt.ca_path ) )
 | |
|         ret = x509parse_crtpath( &cacert, opt.ca_path );
 | |
|     else if( strlen( opt.ca_file ) )
 | |
|         ret = x509parse_crtfile( &cacert, opt.ca_file );
 | |
|     else 
 | |
| #endif
 | |
| #if defined(POLARSSL_CERTS_C)
 | |
|         ret = x509parse_crt( &cacert, (const unsigned char *) test_ca_crt,
 | |
|                 strlen( test_ca_crt ) );
 | |
| #else
 | |
|     {
 | |
|         ret = 1;
 | |
|         printf("POLARSSL_CERTS_C not defined.");
 | |
|     }
 | |
| #endif
 | |
|     if( ret < 0 )
 | |
|     {
 | |
|         printf( " failed\n  !  x509parse_crt returned -0x%x\n\n", -ret );
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
|     printf( " ok (%d skipped)\n", ret );
 | |
| 
 | |
|     /*
 | |
|      * 1.2. Load own certificate and private key
 | |
|      */
 | |
|     printf( "  . Loading the server cert. and key..." );
 | |
|     fflush( stdout );
 | |
| 
 | |
| #if defined(POLARSSL_FS_IO)
 | |
|     if( strlen( opt.crt_file ) )
 | |
|         ret = x509parse_crtfile( &srvcert, opt.crt_file );
 | |
|     else 
 | |
| #endif
 | |
| #if defined(POLARSSL_CERTS_C)
 | |
|         ret = x509parse_crt( &srvcert, (const unsigned char *) test_srv_crt,
 | |
|                 strlen( test_srv_crt ) );
 | |
| #else
 | |
|     {
 | |
|         ret = 1;
 | |
|         printf("POLARSSL_CERTS_C not defined.");
 | |
|     }
 | |
| #endif
 | |
|     if( ret != 0 )
 | |
|     {
 | |
|         printf( " failed\n  !  x509parse_crt returned -0x%x\n\n", -ret );
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
| #if defined(POLARSSL_FS_IO)
 | |
|     if( strlen( opt.key_file ) )
 | |
|         ret = x509parse_keyfile( &pkey, opt.key_file, "" );
 | |
|     else
 | |
| #endif
 | |
| #if defined(POLARSSL_CERTS_C)
 | |
|         ret = x509parse_key( &pkey, (const unsigned char *) test_srv_key,
 | |
|                 strlen( test_srv_key ), NULL, 0 );
 | |
| #else
 | |
|     {
 | |
|         ret = 1;
 | |
|         printf("POLARSSL_CERTS_C not defined.");
 | |
|     }
 | |
| #endif
 | |
|     if( ret != 0 )
 | |
|     {
 | |
|         printf( " failed\n  !  x509parse_key returned -0x%x\n\n", -ret );
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
|     printf( " ok\n" );
 | |
| #endif /* POLARSSL_X509_PARSE_C */
 | |
| 
 | |
|     /*
 | |
|      * 2. Setup the listening TCP socket
 | |
|      */
 | |
|     printf( "  . Bind on tcp://localhost:%-4d/ ...", opt.server_port );
 | |
|     fflush( stdout );
 | |
| 
 | |
|     if( ( ret = net_bind( &listen_fd, NULL, opt.server_port ) ) != 0 )
 | |
|     {
 | |
|         printf( " failed\n  ! net_bind returned -0x%x\n\n", -ret );
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
|     printf( " ok\n" );
 | |
| 
 | |
|     /*
 | |
|      * 3. Setup stuff
 | |
|      */
 | |
|     printf( "  . Setting up the SSL/TLS structure..." );
 | |
|     fflush( stdout );
 | |
| 
 | |
|     if( ( ret = ssl_init( &ssl ) ) != 0 )
 | |
|     {
 | |
|         printf( " failed\n  ! ssl_init returned -0x%x\n\n", -ret );
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
|     ssl_set_endpoint( &ssl, SSL_IS_SERVER );
 | |
|     ssl_set_authmode( &ssl, opt.auth_mode );
 | |
| 
 | |
| #if defined(POLARSSL_SSL_MAX_FRAGMENT_LENGTH)
 | |
|     ssl_set_max_frag_len( &ssl, opt.mfl_code );
 | |
| #endif
 | |
| 
 | |
|     ssl_set_rng( &ssl, ctr_drbg_random, &ctr_drbg );
 | |
|     ssl_set_dbg( &ssl, my_debug, stdout );
 | |
| 
 | |
| #if defined(POLARSSL_SSL_CACHE_C)
 | |
|     ssl_set_session_cache( &ssl, ssl_cache_get, &cache,
 | |
|                                  ssl_cache_set, &cache );
 | |
| #endif
 | |
| 
 | |
| #if defined(POLARSSL_SSL_SESSION_TICKETS)
 | |
|     ssl_set_session_tickets( &ssl, opt.tickets );
 | |
| #endif
 | |
| 
 | |
|     if( opt.force_ciphersuite[0] != DFL_FORCE_CIPHER )
 | |
|         ssl_set_ciphersuites( &ssl, opt.force_ciphersuite );
 | |
| 
 | |
|     ssl_set_renegotiation( &ssl, opt.renegotiation );
 | |
|     ssl_legacy_renegotiation( &ssl, opt.allow_legacy );
 | |
| 
 | |
| #if defined(POLARSSL_X509_PARSE_C)
 | |
|     ssl_set_ca_chain( &ssl, &cacert, NULL, NULL );
 | |
|     ssl_set_own_cert( &ssl, &srvcert, &pkey );
 | |
| #endif
 | |
| 
 | |
| #if defined(POLARSSL_KEY_EXCHANGE_PSK_ENABLED)
 | |
|     ssl_set_psk( &ssl, psk, psk_len, (const unsigned char *) opt.psk_identity,
 | |
|                  strlen( opt.psk_identity ) );
 | |
| #endif
 | |
| 
 | |
| #if defined(POLARSSL_DHM_C)
 | |
|     /*
 | |
|      * Use different group than default DHM group
 | |
|      */
 | |
|     ssl_set_dh_param( &ssl, POLARSSL_DHM_RFC5114_MODP_2048_P,
 | |
|                             POLARSSL_DHM_RFC5114_MODP_2048_G );
 | |
| #endif
 | |
| 
 | |
|     if( opt.min_version != -1 )
 | |
|         ssl_set_min_version( &ssl, SSL_MAJOR_VERSION_3, opt.min_version );
 | |
| 
 | |
|     if( opt.max_version != -1 )
 | |
|         ssl_set_max_version( &ssl, SSL_MAJOR_VERSION_3, opt.max_version );
 | |
| 
 | |
|     printf( " ok\n" );
 | |
| 
 | |
| reset:
 | |
| #ifdef POLARSSL_ERROR_C
 | |
|     if( ret != 0 )
 | |
|     {
 | |
|         char error_buf[100];
 | |
|         polarssl_strerror( ret, error_buf, 100 );
 | |
|         printf("Last error was: %d - %s\n\n", ret, error_buf );
 | |
|     }
 | |
| #endif
 | |
| 
 | |
|     if( client_fd != -1 )
 | |
|         net_close( client_fd );
 | |
| 
 | |
|     ssl_session_reset( &ssl );
 | |
| 
 | |
|     /*
 | |
|      * 3. Wait until a client connects
 | |
|      */
 | |
| #if defined(_WIN32_WCE)
 | |
|     {
 | |
|         SHELLEXECUTEINFO sei;
 | |
| 
 | |
|         ZeroMemory( &sei, sizeof( SHELLEXECUTEINFO ) );
 | |
| 
 | |
|         sei.cbSize = sizeof( SHELLEXECUTEINFO );
 | |
|         sei.fMask = 0;
 | |
|         sei.hwnd = 0;
 | |
|         sei.lpVerb = _T( "open" );
 | |
|         sei.lpFile = _T( "https://localhost:4433/" );
 | |
|         sei.lpParameters = NULL;
 | |
|         sei.lpDirectory = NULL;
 | |
|         sei.nShow = SW_SHOWNORMAL;
 | |
| 
 | |
|         ShellExecuteEx( &sei );
 | |
|     }
 | |
| #elif defined(_WIN32)
 | |
|     ShellExecute( NULL, "open", "https://localhost:4433/",
 | |
|                   NULL, NULL, SW_SHOWNORMAL );
 | |
| #endif
 | |
| 
 | |
|     client_fd = -1;
 | |
| 
 | |
|     printf( "  . Waiting for a remote connection ..." );
 | |
|     fflush( stdout );
 | |
| 
 | |
|     if( ( ret = net_accept( listen_fd, &client_fd, NULL ) ) != 0 )
 | |
|     {
 | |
|         printf( " failed\n  ! net_accept returned -0x%x\n\n", -ret );
 | |
|         goto exit;
 | |
|     }
 | |
| 
 | |
|     ssl_set_bio( &ssl, net_recv, &client_fd,
 | |
|                        net_send, &client_fd );
 | |
| 
 | |
|     printf( " ok\n" );
 | |
| 
 | |
|     /*
 | |
|      * 4. Handshake
 | |
|      */
 | |
|     printf( "  . Performing the SSL/TLS handshake..." );
 | |
|     fflush( stdout );
 | |
| 
 | |
|     while( ( ret = ssl_handshake( &ssl ) ) != 0 )
 | |
|     {
 | |
|         if( ret != POLARSSL_ERR_NET_WANT_READ && ret != POLARSSL_ERR_NET_WANT_WRITE )
 | |
|         {
 | |
|             printf( " failed\n  ! ssl_handshake returned -0x%x\n\n", -ret );
 | |
|             goto reset;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     printf( " ok\n    [ Ciphersuite is %s ]\n",
 | |
|             ssl_get_ciphersuite( &ssl ) );
 | |
| 
 | |
| #if defined(POLARSSL_X509_PARSE_C)
 | |
|     /*
 | |
|      * 5. Verify the server certificate
 | |
|      */
 | |
|     printf( "  . Verifying peer X.509 certificate..." );
 | |
| 
 | |
|     if( ( ret = ssl_get_verify_result( &ssl ) ) != 0 )
 | |
|     {
 | |
|         printf( " failed\n" );
 | |
| 
 | |
|         if( !ssl_get_peer_cert( &ssl ) )
 | |
|             printf( "  ! no client certificate sent\n" );
 | |
| 
 | |
|         if( ( ret & BADCERT_EXPIRED ) != 0 )
 | |
|             printf( "  ! client certificate has expired\n" );
 | |
| 
 | |
|         if( ( ret & BADCERT_REVOKED ) != 0 )
 | |
|             printf( "  ! client certificate has been revoked\n" );
 | |
| 
 | |
|         if( ( ret & BADCERT_NOT_TRUSTED ) != 0 )
 | |
|             printf( "  ! self-signed or not signed by a trusted CA\n" );
 | |
| 
 | |
|         printf( "\n" );
 | |
|     }
 | |
|     else
 | |
|         printf( " ok\n" );
 | |
| 
 | |
|     if( ssl_get_peer_cert( &ssl ) )
 | |
|     {
 | |
|         printf( "  . Peer certificate information    ...\n" );
 | |
|         x509parse_cert_info( (char *) buf, sizeof( buf ) - 1, "      ",
 | |
|                              ssl_get_peer_cert( &ssl ) );
 | |
|         printf( "%s\n", buf );
 | |
|     }
 | |
| #endif /* POLARSSL_X509_PARSE_C */
 | |
| 
 | |
|     /*
 | |
|      * 6. Read the HTTP Request
 | |
|      */
 | |
|     printf( "  < Read from client:" );
 | |
|     fflush( stdout );
 | |
| 
 | |
|     do
 | |
|     {
 | |
|         len = sizeof( buf ) - 1;
 | |
|         memset( buf, 0, sizeof( buf ) );
 | |
|         ret = ssl_read( &ssl, buf, len );
 | |
| 
 | |
|         if( ret == POLARSSL_ERR_NET_WANT_READ || ret == POLARSSL_ERR_NET_WANT_WRITE )
 | |
|             continue;
 | |
| 
 | |
|         if( ret <= 0 )
 | |
|         {
 | |
|             switch( ret )
 | |
|             {
 | |
|                 case POLARSSL_ERR_SSL_PEER_CLOSE_NOTIFY:
 | |
|                     printf( " connection was closed gracefully\n" );
 | |
|                     break;
 | |
| 
 | |
|                 case POLARSSL_ERR_NET_CONN_RESET:
 | |
|                     printf( " connection was reset by peer\n" );
 | |
|                     break;
 | |
| 
 | |
|                 default:
 | |
|                     printf( " ssl_read returned -0x%x\n", -ret );
 | |
|                     break;
 | |
|             }
 | |
| 
 | |
|             break;
 | |
|         }
 | |
| 
 | |
|         len = ret;
 | |
|         printf( " %d bytes read\n\n%s\n", len, (char *) buf );
 | |
| 
 | |
|         if( memcmp( buf, "SERVERQUIT", 10 ) == 0 )
 | |
|         {
 | |
|             ret = 0;
 | |
|             goto exit;
 | |
|         }
 | |
| 
 | |
|         if( ret > 0 )
 | |
|             break;
 | |
|     }
 | |
|     while( 1 );
 | |
| 
 | |
|     /*
 | |
|      * 7. Write the 200 Response
 | |
|      */
 | |
|     printf( "  > Write to client:" );
 | |
|     fflush( stdout );
 | |
| 
 | |
|     len = sprintf( (char *) buf, HTTP_RESPONSE,
 | |
|                    ssl_get_ciphersuite( &ssl ) );
 | |
| 
 | |
|     for( written = 0, frags = 0; written < len; written += ret, frags++ )
 | |
|     {
 | |
|         while( ( ret = ssl_write( &ssl, buf + written, len - written ) ) <= 0 )
 | |
|         {
 | |
|             if( ret == POLARSSL_ERR_NET_CONN_RESET )
 | |
|             {
 | |
|                 printf( " failed\n  ! peer closed the connection\n\n" );
 | |
|                 goto reset;
 | |
|             }
 | |
| 
 | |
|             if( ret != POLARSSL_ERR_NET_WANT_READ && ret != POLARSSL_ERR_NET_WANT_WRITE )
 | |
|             {
 | |
|                 printf( " failed\n  ! ssl_write returned %d\n\n", ret );
 | |
|                 goto exit;
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     buf[written] = '\0';
 | |
|     printf( " %d bytes written in %d fragments\n\n%s\n", written, frags, (char *) buf );
 | |
| 
 | |
|     ret = 0;
 | |
|     goto reset;
 | |
| 
 | |
| exit:
 | |
| 
 | |
| #ifdef POLARSSL_ERROR_C
 | |
|     if( ret != 0 )
 | |
|     {
 | |
|         char error_buf[100];
 | |
|         polarssl_strerror( ret, error_buf, 100 );
 | |
|         printf("Last error was: -0x%X - %s\n\n", -ret, error_buf );
 | |
|     }
 | |
| #endif
 | |
| 
 | |
|     net_close( client_fd );
 | |
| #if defined(POLARSSL_X509_PARSE_C)
 | |
|     x509_free( &srvcert );
 | |
|     x509_free( &cacert );
 | |
|     pk_free( &pkey );
 | |
| #endif
 | |
| 
 | |
|     ssl_free( &ssl );
 | |
| 
 | |
| #if defined(POLARSSL_SSL_CACHE_C)
 | |
|     ssl_cache_free( &cache );
 | |
| #endif
 | |
| 
 | |
| #if defined(POLARSSL_MEMORY_BUFFER_ALLOC_C) && defined(POLARSSL_MEMORY_DEBUG)
 | |
|     memory_buffer_alloc_status();
 | |
| #endif
 | |
| 
 | |
| #if defined(_WIN32)
 | |
|     printf( "  + Press Enter to exit this program.\n" );
 | |
|     fflush( stdout ); getchar();
 | |
| #endif
 | |
| 
 | |
|     // Shell can not handle large exit numbers -> 1 for errors
 | |
|     if( ret < 0 )
 | |
|         ret = 1;
 | |
| 
 | |
|     return( ret );
 | |
| }
 | |
| #endif /* POLARSSL_BIGNUM_C && POLARSSL_ENTROPY_C && POLARSSL_SSL_TLS_C &&
 | |
|           POLARSSL_SSL_SRV_C && POLARSSL_NET_C && POLARSSL_RSA_C &&
 | |
|           POLARSSL_CTR_DRBG_C */
 | 
