apps/s_time.c

   1 /* apps/s_time.c */
   2 /* Copyright (C) 1995-1997 Eric Young (eay@cryptsoft.com)
   3  * All rights reserved.
   4  *
   5  * This package is an SSL implementation written
   6  * by Eric Young (eay@cryptsoft.com).
   7  * The implementation was written so as to conform with Netscapes SSL.
   8  *
   9  * This library is free for commercial and non-commercial use as long as
  10  * the following conditions are aheared to.  The following conditions
  11  * apply to all code found in this distribution, be it the RC4, RSA,
  12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
  13  * included with this distribution is covered by the same copyright terms
  14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  15  *
  16  * Copyright remains Eric Young's, and as such any Copyright notices in
  17  * the code are not to be removed.
  18  * If this package is used in a product, Eric Young should be given attribution
  19  * as the author of the parts of the library used.
  20  * This can be in the form of a textual message at program startup or
  21  * in documentation (online or textual) provided with the package.
  22  *
  23  * Redistribution and use in source and binary forms, with or without
  24  * modification, are permitted provided that the following conditions
  25  * are met:
  26  * 1. Redistributions of source code must retain the copyright
  27  *    notice, this list of conditions and the following disclaimer.
  28  * 2. Redistributions in binary form must reproduce the above copyright
  29  *    notice, this list of conditions and the following disclaimer in the
  30  *    documentation and/or other materials provided with the distribution.
  31  * 3. All advertising materials mentioning features or use of this software
  32  *    must display the following acknowledgement:
  33  *    "This product includes cryptographic software written by
  34  *     Eric Young (eay@cryptsoft.com)"
  35  *    The word 'cryptographic' can be left out if the rouines from the library
  36  *    being used are not cryptographic related :-).
  37  * 4. If you include any Windows specific code (or a derivative thereof) from
  38  *    the apps directory (application code) you must include an acknowledgement:
  39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  40  *
  41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51  * SUCH DAMAGE.
  52  *
  53  * The licence and distribution terms for any publically available version or
  54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
  55  * copied and put under another distribution licence
  56  * [including the GNU Public Licence.]
  57  */
  58
  59 #undef NO_SHUTDOWN
  60
  61 /*-----------------------------------------
  62    cntime - SSL client connection timer program
  63    Written and donated by Larry Streepy <streepy@healthcare.com>
  64   -----------------------------------------*/
  65
  66 #include <stdio.h>
  67 #include <stdlib.h>
  68 #include <string.h>
  69
  70 #ifdef WIN16
  71 #define APPS_WIN16
  72 #endif
  73 #include "x509.h"
  74 #include "ssl.h"
  75 #include "pem.h"
  76 #define USE_SOCKETS
  77 #include "apps.h"
  78 #include "s_apps.h"
  79 #include "err.h"
  80 #ifdef WIN32_STUFF
  81 #include "winmain.h"
  82 #include "wintext.h"
  83 #endif
  84
  85 #ifndef MSDOS
  86 #define TIMES
  87 #endif
  88
  89 #ifndef VMS
  90 #ifndef _IRIX
  91 #include <time.h>
  92 #endif
  93 #ifdef TIMES
  94 #include <sys/types.h>
  95 #include <sys/times.h>
  96 #endif
  97 #else /* VMS */
  98 #include <types.h>
  99 struct tms {
 100         time_t tms_utime;
 101         time_t tms_stime;
 102         time_t tms_uchild;      /* I dunno...  */
 103         time_t tms_uchildsys;   /* so these names are a guess :-) */
 104         }
 105 #endif
 106 #ifndef TIMES
 107 #include <sys/timeb.h>
 108 #endif
 109
 110 #ifdef _AIX
 111 #include <sys/select.h>
 112 #endif
 113
 114 #ifdef sun
 115 #include <limits.h>
 116 #include <sys/param.h>
 117 #endif
 118
 119 /* The following if from times(3) man page.  It may need to be changed
 120 */
 121 #ifndef HZ
 122 #ifndef CLK_TCK
 123 #ifndef VMS
 124 #define HZ      100.0
 125 #else /* VMS */
 126 #define HZ      100.0
 127 #endif
 128 #else /* CLK_TCK */
 129 #define HZ ((double)CLK_TCK)
 130 #endif
 131 #endif
 132
 133 #undef PROG
 134 #define PROG s_time_main
 135
 136 #define ioctl ioctlsocket
 137
 138 #define SSL_CONNECT_NAME        "localhost:4433"
 139
 140 /*#define TEST_CERT "client.pem" */ /* no default cert. */
 141
 142 #undef BUFSIZZ
 143 #define BUFSIZZ 1024*10
 144
 145 #define min(a,b) (((a) < (b)) ? (a) : (b))
 146 #define max(a,b) (((a) > (b)) ? (a) : (b))
 147
 148 #undef SECONDS
 149 #define SECONDS 30
 150 extern int verify_depth;
 151 extern int verify_error;
 152
 153 #ifndef NOPROTO
 154 static void s_time_usage(void);
 155 static int parseArgs( int argc, char **argv );
 156 static SSL *doConnection( SSL *scon );
 157 #else
 158 static void s_time_usage();
 159 static int parseArgs();
 160 static SSL *doConnection();
 161 #endif
 162
 163
 164 /***********************************************************************
 165  * Static data declarations
 166  */
 167
 168 /* static char *port=PORT_STR;*/
 169 static char *host=SSL_CONNECT_NAME;
 170 static char *t_cert_file=NULL;
 171 static char *t_key_file=NULL;
 172 static char *CApath=NULL;
 173 static char *CAfile=NULL;
 174 static char *tm_cipher=NULL;
 175 static int tm_verify = SSL_VERIFY_NONE;
 176 static int maxTime = SECONDS;
 177 static SSL_CTX *tm_ctx=NULL;
 178 static SSL_METHOD *s_time_meth=NULL;
 179 static char *s_www_path=NULL;
 180 static long bytes_read=0;
 181 static int st_bugs=0;
 182 static int perform=0;
 183
 184 #ifdef FIONBIO
 185 static int t_nbio=0;
 186 #endif
 187
 188 #ifdef WIN32
 189 static int exitNow = 0;         /* Set when it's time to exit main */
 190 #endif
 191
 192 /***********************************************************************
 193  * usage - display usage message
 194  */
 195 static void s_time_usage()
 196 {
 197         static char umsg[] = "\
 198 -time arg     - max number of seconds to collect data, default %d\n\
 199 -verify arg   - turn on peer certificate verification, arg == depth\n\
 200 -cert arg     - certificate file to use, PEM format assumed\n\
 201 -key arg      - RSA file to use, PEM format assumed, in cert file if\n\
 202                 not specified but cert fill is.\n\
 203 -CApath arg   - PEM format directory of CA's\n\
 204 -CAfile arg   - PEM format file of CA's\n\
 205 -cipher       - prefered cipher to use, play with 'ssleay ciphers'\n\n";
 206
 207         printf( "usage: client <args>\n\n" );
 208
 209         printf("-connect host:port - host:port to connect to (default is %s)\n",SSL_CONNECT_NAME);
 210 #ifdef FIONBIO
 211         printf("-nbio         - Run with non-blocking IO\n");
 212         printf("-ssl2         - Just use SSLv2\n");
 213         printf("-ssl3         - Just use SSLv3\n");
 214         printf("-bugs         - Turn on SSL bug compatability\n");
 215         printf("-new          - Just time new connections\n");
 216         printf("-reuse        - Just time connection reuse\n");
 217         printf("-www page     - Retrieve 'page' from the site\n");
 218 #endif
 219         printf( umsg,SECONDS );
 220 }
 221
 222 /***********************************************************************
 223  * parseArgs - Parse command line arguments and initialize data
 224  *
 225  * Returns 0 if ok, -1 on bad args
 226  */
 227 static int parseArgs(argc,argv)
 228 int argc;
 229 char **argv;
 230 {
 231     int badop = 0;
 232
 233     verify_depth=0;
 234     verify_error=X509_V_OK;
 235 #ifdef FIONBIO
 236     t_nbio=0;
 237 #endif
 238
 239         apps_startup();
 240
 241         if (bio_err == NULL)
 242                 bio_err=BIO_new_fp(stderr,BIO_NOCLOSE);
 243
 244     argc--;
 245     argv++;
 246
 247     while (argc >= 1) {
 248         if (strcmp(*argv,"-connect") == 0)
 249                 {
 250                 if (--argc < 1) goto bad;
 251                 host= *(++argv);
 252                 }
 253 #if 0
 254         else if( strcmp(*argv,"-host") == 0)
 255                 {
 256                 if (--argc < 1) goto bad;
 257                 host= *(++argv);
 258                 }
 259         else if( strcmp(*argv,"-port") == 0)
 260                 {
 261                 if (--argc < 1) goto bad;
 262                 port= *(++argv);
 263                 }
 264 #endif
 265         else if (strcmp(*argv,"-reuse") == 0)
 266                 perform=2;
 267         else if (strcmp(*argv,"-new") == 0)
 268                 perform=1;
 269         else if( strcmp(*argv,"-verify") == 0) {
 270
 271             tm_verify=SSL_VERIFY_PEER|SSL_VERIFY_CLIENT_ONCE;
 272             if (--argc < 1) goto bad;
 273             verify_depth=atoi(*(++argv));
 274             BIO_printf(bio_err,"verify depth is %d\n",verify_depth);
 275
 276         } else if( strcmp(*argv,"-cert") == 0) {
 277
 278             if (--argc < 1) goto bad;
 279             t_cert_file= *(++argv);
 280
 281         } else if( strcmp(*argv,"-key") == 0) {
 282
 283             if (--argc < 1) goto bad;
 284             t_key_file= *(++argv);
 285
 286         } else if( strcmp(*argv,"-CApath") == 0) {
 287
 288             if (--argc < 1) goto bad;
 289             CApath= *(++argv);
 290
 291         } else if( strcmp(*argv,"-CAfile") == 0) {
 292
 293             if (--argc < 1) goto bad;
 294             CAfile= *(++argv);
 295
 296         } else if( strcmp(*argv,"-cipher") == 0) {
 297
 298             if (--argc < 1) goto bad;
 299             tm_cipher= *(++argv);
 300         }
 301 #ifdef FIONBIO
 302         else if(strcmp(*argv,"-nbio") == 0) {
 303             t_nbio=1;
 304         }
 305 #endif
 306         else if(strcmp(*argv,"-www") == 0)
 307                 {
 308                 if (--argc < 1) goto bad;
 309                 s_www_path= *(++argv);
 310                 }
 311         else if(strcmp(*argv,"-bugs") == 0)
 312             st_bugs=1;
 313 #ifndef NO_SSL2
 314         else if(strcmp(*argv,"-ssl2") == 0)
 315             s_time_meth=SSLv2_client_method();
 316 #endif
 317 #ifndef NO_SSL3
 318         else if(strcmp(*argv,"-ssl3") == 0)
 319             s_time_meth=SSLv3_client_method();
 320 #endif
 321         else if( strcmp(*argv,"-time") == 0) {
 322
 323             if (--argc < 1) goto bad;
 324             maxTime= atoi(*(++argv));
 325         }
 326         else {
 327             BIO_printf(bio_err,"unknown option %s\n",*argv);
 328             badop=1;
 329             break;
 330         }
 331
 332         argc--;
 333         argv++;
 334     }
 335
 336     if (perform == 0) perform=3;
 337
 338     if(badop) {
 339 bad:
 340                 s_time_usage();
 341                 return -1;
 342     }
 343
 344         return 0;                       /* Valid args */
 345 }
 346
 347 /***********************************************************************
 348  * TIME - time functions
 349  */
 350 #define START   0
 351 #define STOP    1
 352
 353 static double tm_Time_F(s)
 354 int s;
 355         {
 356         static double ret;
 357 #ifdef TIMES
 358         static struct tms tstart,tend;
 359
 360         if(s == START) {
 361                 times(&tstart);
 362                 return(0);
 363         } else {
 364                 times(&tend);
 365                 ret=((double)(tend.tms_utime-tstart.tms_utime))/HZ;
 366                 return((ret == 0.0)?1e-6:ret);
 367         }
 368 #else /* !times() */
 369         static struct timeb tstart,tend;
 370         long i;
 371
 372         if(s == START) {
 373                 ftime(&tstart);
 374                 return(0);
 375         } else {
 376                 ftime(&tend);
 377                 i=(long)tend.millitm-(long)tstart.millitm;
 378                 ret=((double)(tend.time-tstart.time))+((double)i)/1000.0;
 379                 return((ret == 0.0)?1e-6:ret);
 380         }
 381 #endif
 382 }
 383
 384 /***********************************************************************
 385  * MAIN - main processing area for client
 386  *                      real name depends on MONOLITH
 387  */
 388 int
 389 MAIN(argc,argv)
 390 int argc;
 391 char **argv;
 392         {
 393         double totalTime = 0.0;
 394         int nConn = 0;
 395         SSL *scon=NULL;
 396         long finishtime=0;
 397         int ret=1,i;
 398         MS_STATIC char buf[1024*8];
 399
 400 #if !defined(NO_SSL2) && !defined(NO_SSL3)
 401         s_time_meth=SSLv23_client_method();
 402 #elif !defined(NO_SSL3)
 403         s_time_meth=SSLv3_client_method();
 404 #elif !defined(NO_SSL2)
 405         s_time_meth=SSLv2_client_method();
 406 #endif
 407
 408         /* parse the command line arguments */
 409         if( parseArgs( argc, argv ) < 0 )
 410                 goto end;
 411
 412         SSLeay_add_ssl_algorithms();
 413         if ((tm_ctx=SSL_CTX_new(s_time_meth)) == NULL) return(1);
 414
 415         if (st_bugs) SSL_CTX_set_options(tm_ctx,SSL_OP_ALL);
 416         SSL_CTX_set_cipher_list(tm_ctx,tm_cipher);
 417         if(!set_cert_stuff(tm_ctx,t_cert_file,t_key_file))
 418                 goto end;
 419
 420         SSL_load_error_strings();
 421
 422         if ((!SSL_CTX_load_verify_locations(tm_ctx,CAfile,CApath)) ||
 423                 (!SSL_CTX_set_default_verify_paths(tm_ctx)))
 424                 {
 425                 BIO_printf(bio_err,"error seting default verify locations\n");
 426                 ERR_print_errors(bio_err);
 427                 goto end;
 428                 }
 429
 430         if (tm_cipher == NULL)
 431                 tm_cipher = getenv("SSL_CIPHER");
 432
 433         if (tm_cipher == NULL ) {
 434                 fprintf( stderr, "No CIPHER specified\n" );
 435 /*              EXIT(1); */
 436         }
 437
 438         if (!(perform & 1)) goto next;
 439         printf( "Collecting connection statistics for %d seconds\n", maxTime );
 440
 441         /* Loop and time how long it takes to make connections */
 442
 443         bytes_read=0;
 444         finishtime=(long)time(NULL)+maxTime;
 445         tm_Time_F(START);
 446         for (;;)
 447                 {
 448                 if (finishtime < time(NULL)) break;
 449 #ifdef WIN32_STUFF
 450
 451                 if( flushWinMsgs(0) == -1 )
 452                         goto end;
 453
 454                 if( waitingToDie || exitNow )           /* we're dead */
 455                         goto end;
 456 #endif
 457
 458                 if( (scon = doConnection( NULL )) == NULL )
 459                         goto end;
 460
 461                 if (s_www_path != NULL)
 462                         {
 463                         sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
 464                         SSL_write(scon,buf,strlen(buf));
 465                         while ((i=SSL_read(scon,buf,sizeof(buf))) > 0)
 466                                 bytes_read+=i;
 467                         }
 468
 469 #ifdef NO_SHUTDOWN
 470                 SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
 471 #else
 472                 SSL_shutdown(scon);
 473 #endif
 474                 SHUTDOWN(SSL_get_fd(scon));
 475
 476                 nConn += 1;
 477                 fputc(SSL_session_reused(scon)?'r':
 478                         (SSL_version(scon))+'0', stdout );
 479                 fflush(stdout);
 480
 481                 SSL_free( scon );
 482                 scon=NULL;
 483                 }
 484         totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
 485
 486         i=(int)(time(NULL)-finishtime+maxTime);
 487         printf( "\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", nConn, totalTime, ((double)nConn/totalTime),bytes_read);
 488         printf( "%d connections in %ld real seconds, %ld bytes read per connection\n",nConn,time(NULL)-finishtime+maxTime,bytes_read/nConn);
 489
 490         /* Now loop and time connections using the same session id over and over */
 491
 492 next:
 493         if (!(perform & 2)) goto end;
 494         printf( "\n\nNow timing with session id reuse.\n" );
 495
 496         /* Get an SSL object so we can reuse the session id */
 497         if( (scon = doConnection( NULL )) == NULL )
 498                 {
 499                 fprintf( stderr, "Unable to get connection\n" );
 500                 goto end;
 501                 }
 502
 503         if (s_www_path != NULL)
 504                 {
 505                 sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
 506                 SSL_write(scon,buf,strlen(buf));
 507                 while (SSL_read(scon,buf,sizeof(buf)) > 0)
 508                         ;
 509                 }
 510 #ifdef NO_SHUTDOWN
 511         SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
 512 #else
 513         SSL_shutdown(scon);
 514 #endif
 515         SHUTDOWN(SSL_get_fd(scon));
 516
 517         nConn = 0;
 518         totalTime = 0.0;
 519
 520         finishtime=time(NULL)+maxTime;
 521
 522         printf( "starting\n" );
 523         bytes_read=0;
 524         tm_Time_F(START);
 525
 526         for (;;)
 527                 {
 528                 if (finishtime < time(NULL)) break;
 529
 530 #ifdef WIN32_STUFF
 531                 if( flushWinMsgs(0) == -1 )
 532                         goto end;
 533
 534                 if( waitingToDie || exitNow )   /* we're dead */
 535                         goto end;
 536 #endif
 537
 538                 if( (doConnection( scon )) == NULL )
 539                         goto end;
 540
 541                 if (s_www_path)
 542                         {
 543                         sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
 544                         SSL_write(scon,buf,strlen(buf));
 545                         while ((i=SSL_read(scon,buf,sizeof(buf))) > 0)
 546                                 bytes_read+=i;
 547                         }
 548
 549 #ifdef NO_SHUTDOWN
 550                 SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
 551 #else
 552                 SSL_shutdown(scon);
 553 #endif
 554                 SHUTDOWN(SSL_get_fd(scon));
 555
 556                 nConn += 1;
 557                 fputc(SSL_session_reused(scon)?'r':
 558                         (SSL_version(scon))+'0', stdout );
 559                 fflush(stdout);
 560                 }
 561         totalTime += tm_Time_F(STOP); /* Add the time for this iteration*/
 562
 563
 564         printf( "\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", nConn, totalTime, ((double)nConn/totalTime),bytes_read);
 565         printf( "%d connections in %ld real seconds, %ld bytes read per connection\n",nConn,time(NULL)-finishtime+maxTime,bytes_read/nConn);
 566
 567         ret=0;
 568 end:
 569         if (scon != NULL) SSL_free(scon);
 570
 571         if (tm_ctx != NULL)
 572                 {
 573                 SSL_CTX_free(tm_ctx);
 574                 tm_ctx=NULL;
 575                 }
 576         EXIT(ret);
 577         }
 578
 579 /***********************************************************************
 580  * doConnection - make a connection
 581  * Args:
 582  *              scon    = earlier ssl connection for session id, or NULL
 583  * Returns:
 584  *              SSL *   = the connection pointer.
 585  */
 586 static SSL *
 587 doConnection(scon)
 588 SSL *scon;
 589         {
 590         BIO *conn;
 591         SSL *serverCon;
 592         int width, i;
 593         fd_set readfds;
 594
 595         if ((conn=BIO_new(BIO_s_connect())) == NULL)
 596                 return(NULL);
 597
 598 /*      BIO_set_port(conn,port);*/
 599         BIO_set_hostname(conn,host);
 600
 601         if (scon == NULL)
 602                 serverCon=(SSL *)SSL_new(tm_ctx);
 603         else
 604                 {
 605                 serverCon=scon;
 606                 SSL_set_connect_state(serverCon);
 607                 }
 608
 609         SSL_set_bio(serverCon,conn,conn);
 610
 611 #if 0
 612         if( scon != NULL )
 613                 SSL_set_session(serverCon,SSL_get_session(scon));
 614 #endif
 615
 616         /* ok, lets connect */
 617         for(;;) {
 618                 i=SSL_connect(serverCon);
 619                 if (BIO_sock_should_retry(i))
 620                         {
 621                         BIO_printf(bio_err,"DELAY\n");
 622
 623                         i=SSL_get_fd(serverCon);
 624                         width=i+1;
 625                         FD_ZERO(&readfds);
 626                         FD_SET(i,&readfds);
 627                         select(width,&readfds,NULL,NULL,NULL);
 628                         continue;
 629                         }
 630                 break;
 631                 }
 632         if(i <= 0)
 633                 {
 634                 BIO_printf(bio_err,"ERROR\n");
 635                 if (verify_error != X509_V_OK)
 636                         BIO_printf(bio_err,"verify error:%s\n",
 637                                 X509_verify_cert_error_string(verify_error));
 638                 else
 639                         ERR_print_errors(bio_err);
 640                 if (scon == NULL)
 641                         SSL_free(serverCon);
 642                 return NULL;
 643                 }
 644
 645         return serverCon;
 646         }
 647
 648