9ce114dcb6fc22d213e38422dc9fc93b386665e6
[openssl.git] / apps / s_time.c
1 /* apps/s_time.c */
2 /* Copyright (C) 1995-1998 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 #define NO_SHUTDOWN
60
61 /*-----------------------------------------
62    s_time - 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 #define USE_SOCKETS
71 #include "apps.h"
72 #ifdef OPENSSL_NO_STDIO
73 #define APPS_WIN16
74 #endif
75 #include <openssl/x509.h>
76 #include <openssl/ssl.h>
77 #include <openssl/pem.h>
78 #include "s_apps.h"
79 #include <openssl/err.h>
80 #ifdef WIN32_STUFF
81 #include "winmain.h"
82 #include "wintext.h"
83 #endif
84 #if !defined(OPENSSL_SYS_MSDOS)
85 #include OPENSSL_UNISTD
86 #endif
87
88 #if !defined(OPENSSL_SYS_MSDOS) && (!defined(OPENSSL_SYS_VMS) || defined(__DECC)) && !defined(OPENSSL_SYS_MACOSX)
89 #define TIMES
90 #endif
91
92 #ifndef _IRIX
93 #include <time.h>
94 #endif
95 #ifdef TIMES
96 #include <sys/types.h>
97 #include <sys/times.h>
98 #endif
99
100 /* Depending on the VMS version, the tms structure is perhaps defined.
101    The __TMS macro will show if it was.  If it wasn't defined, we should
102    undefine TIMES, since that tells the rest of the program how things
103    should be handled.                           -- Richard Levitte */
104 #if defined(OPENSSL_SYS_VMS_DECC) && !defined(__TMS)
105 #undef TIMES
106 #endif
107
108 #ifndef TIMES
109 #include <sys/timeb.h>
110 #endif
111
112 #ifdef _AIX
113 #include <sys/select.h>
114 #endif
115
116 #if defined(sun) || defined(__ultrix)
117 #define _POSIX_SOURCE
118 #include <limits.h>
119 #include <sys/param.h>
120 #endif
121
122 /* The following if from times(3) man page.  It may need to be changed
123 */
124 #ifndef HZ
125 # ifdef _SC_CLK_TCK
126 #  define HZ ((double)sysconf(_SC_CLK_TCK))
127 # else
128 #  ifndef CLK_TCK
129 #   ifndef _BSD_CLK_TCK_ /* FreeBSD hack */
130 #    define HZ  100.0
131 #   else /* _BSD_CLK_TCK_ */
132 #    define HZ ((double)_BSD_CLK_TCK_)
133 #   endif
134 #  else /* CLK_TCK */
135 #   define HZ ((double)CLK_TCK)
136 #  endif
137 # endif
138 #endif
139
140 #undef PROG
141 #define PROG s_time_main
142
143 #undef ioctl
144 #define ioctl ioctlsocket
145
146 #define SSL_CONNECT_NAME        "localhost:4433"
147
148 /*#define TEST_CERT "client.pem" */ /* no default cert. */
149
150 #undef BUFSIZZ
151 #define BUFSIZZ 1024*10
152
153 #define min(a,b) (((a) < (b)) ? (a) : (b))
154 #define max(a,b) (((a) > (b)) ? (a) : (b))
155
156 #undef SECONDS
157 #define SECONDS 30
158 extern int verify_depth;
159 extern int verify_error;
160
161 static void s_time_usage(void);
162 static int parseArgs( int argc, char **argv );
163 static SSL *doConnection( SSL *scon );
164 static void s_time_init(void);
165
166 /***********************************************************************
167  * Static data declarations
168  */
169
170 /* static char *port=PORT_STR;*/
171 static char *host=SSL_CONNECT_NAME;
172 static char *t_cert_file=NULL;
173 static char *t_key_file=NULL;
174 static char *CApath=NULL;
175 static char *CAfile=NULL;
176 static char *tm_cipher=NULL;
177 static int tm_verify = SSL_VERIFY_NONE;
178 static int maxTime = SECONDS;
179 static SSL_CTX *tm_ctx=NULL;
180 static SSL_METHOD *s_time_meth=NULL;
181 static char *s_www_path=NULL;
182 static long bytes_read=0; 
183 static int st_bugs=0;
184 static int perform=0;
185 #ifdef FIONBIO
186 static int t_nbio=0;
187 #endif
188 #ifdef OPENSSL_SYS_WIN32
189 static int exitNow = 0;         /* Set when it's time to exit main */
190 #endif
191
192 static void s_time_init(void)
193         {
194         host=SSL_CONNECT_NAME;
195         t_cert_file=NULL;
196         t_key_file=NULL;
197         CApath=NULL;
198         CAfile=NULL;
199         tm_cipher=NULL;
200         tm_verify = SSL_VERIFY_NONE;
201         maxTime = SECONDS;
202         tm_ctx=NULL;
203         s_time_meth=NULL;
204         s_www_path=NULL;
205         bytes_read=0; 
206         st_bugs=0;
207         perform=0;
208
209 #ifdef FIONBIO
210         t_nbio=0;
211 #endif
212 #ifdef OPENSSL_SYS_WIN32
213         exitNow = 0;            /* Set when it's time to exit main */
214 #endif
215         }
216
217 /***********************************************************************
218  * usage - display usage message
219  */
220 static void s_time_usage(void)
221 {
222         static char umsg[] = "\
223 -time arg     - max number of seconds to collect data, default %d\n\
224 -verify arg   - turn on peer certificate verification, arg == depth\n\
225 -cert arg     - certificate file to use, PEM format assumed\n\
226 -key arg      - RSA file to use, PEM format assumed, key is in cert file\n\
227                 file if not specified by this option\n\
228 -CApath arg   - PEM format directory of CA's\n\
229 -CAfile arg   - PEM format file of CA's\n\
230 -cipher       - preferred cipher to use, play with 'openssl ciphers'\n\n";
231
232         printf( "usage: s_time <args>\n\n" );
233
234         printf("-connect host:port - host:port to connect to (default is %s)\n",SSL_CONNECT_NAME);
235 #ifdef FIONBIO
236         printf("-nbio         - Run with non-blocking IO\n");
237         printf("-ssl2         - Just use SSLv2\n");
238         printf("-ssl3         - Just use SSLv3\n");
239         printf("-bugs         - Turn on SSL bug compatibility\n");
240         printf("-new          - Just time new connections\n");
241         printf("-reuse        - Just time connection reuse\n");
242         printf("-www page     - Retrieve 'page' from the site\n");
243 #endif
244         printf( umsg,SECONDS );
245 }
246
247 /***********************************************************************
248  * parseArgs - Parse command line arguments and initialize data
249  *
250  * Returns 0 if ok, -1 on bad args
251  */
252 static int parseArgs(int argc, char **argv)
253 {
254     int badop = 0;
255
256     verify_depth=0;
257     verify_error=X509_V_OK;
258
259     argc--;
260     argv++;
261
262     while (argc >= 1) {
263         if (strcmp(*argv,"-connect") == 0)
264                 {
265                 if (--argc < 1) goto bad;
266                 host= *(++argv);
267                 }
268 #if 0
269         else if( strcmp(*argv,"-host") == 0)
270                 {
271                 if (--argc < 1) goto bad;
272                 host= *(++argv);
273                 }
274         else if( strcmp(*argv,"-port") == 0)
275                 {
276                 if (--argc < 1) goto bad;
277                 port= *(++argv);
278                 }
279 #endif
280         else if (strcmp(*argv,"-reuse") == 0)
281                 perform=2;
282         else if (strcmp(*argv,"-new") == 0)
283                 perform=1;
284         else if( strcmp(*argv,"-verify") == 0) {
285
286             tm_verify=SSL_VERIFY_PEER|SSL_VERIFY_CLIENT_ONCE;
287             if (--argc < 1) goto bad;
288             verify_depth=atoi(*(++argv));
289             BIO_printf(bio_err,"verify depth is %d\n",verify_depth);
290
291         } else if( strcmp(*argv,"-cert") == 0) {
292
293             if (--argc < 1) goto bad;
294             t_cert_file= *(++argv);
295
296         } else if( strcmp(*argv,"-key") == 0) {
297
298             if (--argc < 1) goto bad;
299             t_key_file= *(++argv);
300
301         } else if( strcmp(*argv,"-CApath") == 0) {
302
303             if (--argc < 1) goto bad;
304             CApath= *(++argv);
305
306         } else if( strcmp(*argv,"-CAfile") == 0) {
307
308             if (--argc < 1) goto bad;
309             CAfile= *(++argv);
310
311         } else if( strcmp(*argv,"-cipher") == 0) {
312
313             if (--argc < 1) goto bad;
314             tm_cipher= *(++argv);
315         }
316 #ifdef FIONBIO
317         else if(strcmp(*argv,"-nbio") == 0) {
318             t_nbio=1;
319         }
320 #endif
321         else if(strcmp(*argv,"-www") == 0)
322                 {
323                 if (--argc < 1) goto bad;
324                 s_www_path= *(++argv);
325                 }
326         else if(strcmp(*argv,"-bugs") == 0)
327             st_bugs=1;
328 #ifndef OPENSSL_NO_SSL2
329         else if(strcmp(*argv,"-ssl2") == 0)
330             s_time_meth=SSLv2_client_method();
331 #endif
332 #ifndef OPENSSL_NO_SSL3
333         else if(strcmp(*argv,"-ssl3") == 0)
334             s_time_meth=SSLv3_client_method();
335 #endif
336         else if( strcmp(*argv,"-time") == 0) {
337
338             if (--argc < 1) goto bad;
339             maxTime= atoi(*(++argv));
340         }
341         else {
342             BIO_printf(bio_err,"unknown option %s\n",*argv);
343             badop=1;
344             break;
345         }
346
347         argc--;
348         argv++;
349     }
350
351     if (perform == 0) perform=3;
352
353     if(badop) {
354 bad:
355                 s_time_usage();
356                 return -1;
357     }
358
359         return 0;                       /* Valid args */
360 }
361
362 /***********************************************************************
363  * TIME - time functions
364  */
365 #define START   0
366 #define STOP    1
367
368 static double tm_Time_F(int s)
369         {
370         static double ret;
371 #ifdef TIMES
372         static struct tms tstart,tend;
373
374         if(s == START) {
375                 times(&tstart);
376                 return(0);
377         } else {
378                 times(&tend);
379                 ret=((double)(tend.tms_utime-tstart.tms_utime))/HZ;
380                 return((ret == 0.0)?1e-6:ret);
381         }
382 #else /* !times() */
383         static struct timeb tstart,tend;
384         long i;
385
386         if(s == START) {
387                 ftime(&tstart);
388                 return(0);
389         } else {
390                 ftime(&tend);
391                 i=(long)tend.millitm-(long)tstart.millitm;
392                 ret=((double)(tend.time-tstart.time))+((double)i)/1000.0;
393                 return((ret == 0.0)?1e-6:ret);
394         }
395 #endif
396 }
397
398 /***********************************************************************
399  * MAIN - main processing area for client
400  *                      real name depends on MONOLITH
401  */
402 int MAIN(int, char **);
403
404 int MAIN(int argc, char **argv)
405         {
406         double totalTime = 0.0;
407         int nConn = 0;
408         SSL *scon=NULL;
409         long finishtime=0;
410         int ret=1,i;
411         MS_STATIC char buf[1024*8];
412         int ver;
413
414         apps_startup();
415         s_time_init();
416
417         if (bio_err == NULL)
418                 bio_err=BIO_new_fp(stderr,BIO_NOCLOSE);
419
420 #if !defined(OPENSSL_NO_SSL2) && !defined(OPENSSL_NO_SSL3)
421         s_time_meth=SSLv23_client_method();
422 #elif !defined(OPENSSL_NO_SSL3)
423         s_time_meth=SSLv3_client_method();
424 #elif !defined(OPENSSL_NO_SSL2)
425         s_time_meth=SSLv2_client_method();
426 #endif
427
428         /* parse the command line arguments */
429         if( parseArgs( argc, argv ) < 0 )
430                 goto end;
431
432         OpenSSL_add_ssl_algorithms();
433         if ((tm_ctx=SSL_CTX_new(s_time_meth)) == NULL) return(1);
434
435         SSL_CTX_set_quiet_shutdown(tm_ctx,1);
436
437         if (st_bugs) SSL_CTX_set_options(tm_ctx,SSL_OP_ALL);
438         SSL_CTX_set_cipher_list(tm_ctx,tm_cipher);
439         if(!set_cert_stuff(tm_ctx,t_cert_file,t_key_file)) 
440                 goto end;
441
442         SSL_load_error_strings();
443
444         if ((!SSL_CTX_load_verify_locations(tm_ctx,CAfile,CApath)) ||
445                 (!SSL_CTX_set_default_verify_paths(tm_ctx)))
446                 {
447                 /* BIO_printf(bio_err,"error setting default verify locations\n"); */
448                 ERR_print_errors(bio_err);
449                 /* goto end; */
450                 }
451
452         if (tm_cipher == NULL)
453                 tm_cipher = getenv("SSL_CIPHER");
454
455         if (tm_cipher == NULL ) {
456                 fprintf( stderr, "No CIPHER specified\n" );
457         }
458
459         if (!(perform & 1)) goto next;
460         printf( "Collecting connection statistics for %d seconds\n", maxTime );
461
462         /* Loop and time how long it takes to make connections */
463
464         bytes_read=0;
465         finishtime=(long)time(NULL)+maxTime;
466         tm_Time_F(START);
467         for (;;)
468                 {
469                 if (finishtime < time(NULL)) break;
470 #ifdef WIN32_STUFF
471
472                 if( flushWinMsgs(0) == -1 )
473                         goto end;
474
475                 if( waitingToDie || exitNow )           /* we're dead */
476                         goto end;
477 #endif
478
479                 if( (scon = doConnection( NULL )) == NULL )
480                         goto end;
481
482                 if (s_www_path != NULL)
483                         {
484                         sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
485                         SSL_write(scon,buf,strlen(buf));
486                         while ((i=SSL_read(scon,buf,sizeof(buf))) > 0)
487                                 bytes_read+=i;
488                         }
489
490 #ifdef NO_SHUTDOWN
491                 SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
492 #else
493                 SSL_shutdown(scon);
494 #endif
495                 SHUTDOWN2(SSL_get_fd(scon));
496
497                 nConn += 1;
498                 if (SSL_session_reused(scon))
499                         ver='r';
500                 else
501                         {
502                         ver=SSL_version(scon);
503                         if (ver == TLS1_VERSION)
504                                 ver='t';
505                         else if (ver == SSL3_VERSION)
506                                 ver='3';
507                         else if (ver == SSL2_VERSION)
508                                 ver='2';
509                         else
510                                 ver='*';
511                         }
512                 fputc(ver,stdout);
513                 fflush(stdout);
514
515                 SSL_free( scon );
516                 scon=NULL;
517                 }
518         totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
519
520         i=(int)(time(NULL)-finishtime+maxTime);
521         printf( "\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", nConn, totalTime, ((double)nConn/totalTime),bytes_read);
522         printf( "%d connections in %ld real seconds, %ld bytes read per connection\n",nConn,time(NULL)-finishtime+maxTime,bytes_read/nConn);
523
524         /* Now loop and time connections using the same session id over and over */
525
526 next:
527         if (!(perform & 2)) goto end;
528         printf( "\n\nNow timing with session id reuse.\n" );
529
530         /* Get an SSL object so we can reuse the session id */
531         if( (scon = doConnection( NULL )) == NULL )
532                 {
533                 fprintf( stderr, "Unable to get connection\n" );
534                 goto end;
535                 }
536
537         if (s_www_path != NULL)
538                 {
539                 sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
540                 SSL_write(scon,buf,strlen(buf));
541                 while (SSL_read(scon,buf,sizeof(buf)) > 0)
542                         ;
543                 }
544 #ifdef NO_SHUTDOWN
545         SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
546 #else
547         SSL_shutdown(scon);
548 #endif
549         SHUTDOWN2(SSL_get_fd(scon));
550
551         nConn = 0;
552         totalTime = 0.0;
553
554         finishtime=time(NULL)+maxTime;
555
556         printf( "starting\n" );
557         bytes_read=0;
558         tm_Time_F(START);
559                 
560         for (;;)
561                 {
562                 if (finishtime < time(NULL)) break;
563
564 #ifdef WIN32_STUFF
565                 if( flushWinMsgs(0) == -1 )
566                         goto end;
567
568                 if( waitingToDie || exitNow )   /* we're dead */
569                         goto end;
570 #endif
571
572                 if( (doConnection( scon )) == NULL )
573                         goto end;
574
575                 if (s_www_path)
576                         {
577                         sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
578                         SSL_write(scon,buf,strlen(buf));
579                         while ((i=SSL_read(scon,buf,sizeof(buf))) > 0)
580                                 bytes_read+=i;
581                         }
582
583 #ifdef NO_SHUTDOWN
584                 SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
585 #else
586                 SSL_shutdown(scon);
587 #endif
588                 SHUTDOWN2(SSL_get_fd(scon));
589         
590                 nConn += 1;
591                 if (SSL_session_reused(scon))
592                         ver='r';
593                 else
594                         {
595                         ver=SSL_version(scon);
596                         if (ver == TLS1_VERSION)
597                                 ver='t';
598                         else if (ver == SSL3_VERSION)
599                                 ver='3';
600                         else if (ver == SSL2_VERSION)
601                                 ver='2';
602                         else
603                                 ver='*';
604                         }
605                 fputc(ver,stdout);
606                 fflush(stdout);
607                 }
608         totalTime += tm_Time_F(STOP); /* Add the time for this iteration*/
609
610
611         printf( "\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", nConn, totalTime, ((double)nConn/totalTime),bytes_read);
612         printf( "%d connections in %ld real seconds, %ld bytes read per connection\n",nConn,time(NULL)-finishtime+maxTime,bytes_read/nConn);
613
614         ret=0;
615 end:
616         if (scon != NULL) SSL_free(scon);
617
618         if (tm_ctx != NULL)
619                 {
620                 SSL_CTX_free(tm_ctx);
621                 tm_ctx=NULL;
622                 }
623         apps_shutdown();
624         EXIT(ret);
625         }
626
627 /***********************************************************************
628  * doConnection - make a connection
629  * Args:
630  *              scon    = earlier ssl connection for session id, or NULL
631  * Returns:
632  *              SSL *   = the connection pointer.
633  */
634 static SSL *doConnection(SSL *scon)
635         {
636         BIO *conn;
637         SSL *serverCon;
638         int width, i;
639         fd_set readfds;
640
641         if ((conn=BIO_new(BIO_s_connect())) == NULL)
642                 return(NULL);
643
644 /*      BIO_set_conn_port(conn,port);*/
645         BIO_set_conn_hostname(conn,host);
646
647         if (scon == NULL)
648                 serverCon=SSL_new(tm_ctx);
649         else
650                 {
651                 serverCon=scon;
652                 SSL_set_connect_state(serverCon);
653                 }
654
655         SSL_set_bio(serverCon,conn,conn);
656
657 #if 0
658         if( scon != NULL )
659                 SSL_set_session(serverCon,SSL_get_session(scon));
660 #endif
661
662         /* ok, lets connect */
663         for(;;) {
664                 i=SSL_connect(serverCon);
665                 if (BIO_sock_should_retry(i))
666                         {
667                         BIO_printf(bio_err,"DELAY\n");
668
669                         i=SSL_get_fd(serverCon);
670                         width=i+1;
671                         FD_ZERO(&readfds);
672                         FD_SET(i,&readfds);
673                         /* Note: under VMS with SOCKETSHR the 2nd parameter
674                          * is currently of type (int *) whereas under other
675                          * systems it is (void *) if you don't have a cast it
676                          * will choke the compiler: if you do have a cast then
677                          * you can either go for (int *) or (void *).
678                          */
679                         select(width,(void *)&readfds,NULL,NULL,NULL);
680                         continue;
681                         }
682                 break;
683                 }
684         if(i <= 0)
685                 {
686                 BIO_printf(bio_err,"ERROR\n");
687                 if (verify_error != X509_V_OK)
688                         BIO_printf(bio_err,"verify error:%s\n",
689                                 X509_verify_cert_error_string(verify_error));
690                 else
691                         ERR_print_errors(bio_err);
692                 if (scon == NULL)
693                         SSL_free(serverCon);
694                 return NULL;
695                 }
696
697         return serverCon;
698         }
699
700