This commit was generated by cvs2svn to track changes on a CVS vendor
[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    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 NO_STDIO
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 static void s_time_init(void);
158 #else
159 static void s_time_usage();
160 static int parseArgs();
161 static SSL *doConnection();
162 static void s_time_init();
163 #endif
164
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 WIN32
189 static int exitNow = 0;         /* Set when it's time to exit main */
190 #endif
191
192 static void s_time_init()
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 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()
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, in cert file if\n\
227                 not specified but cert fill is.\n\
228 -CApath arg   - PEM format directory of CA's\n\
229 -CAfile arg   - PEM format file of CA's\n\
230 -cipher       - prefered cipher to use, play with 'ssleay ciphers'\n\n";
231
232         printf( "usage: client <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 compatability\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(argc,argv)
253 int argc;
254 char **argv;
255 {
256     int badop = 0;
257
258     verify_depth=0;
259     verify_error=X509_V_OK;
260 #ifdef FIONBIO
261     t_nbio=0;
262 #endif
263
264         apps_startup();
265         s_time_init();
266
267         if (bio_err == NULL)
268                 bio_err=BIO_new_fp(stderr,BIO_NOCLOSE);
269
270     argc--;
271     argv++;
272
273     while (argc >= 1) {
274         if (strcmp(*argv,"-connect") == 0)
275                 {
276                 if (--argc < 1) goto bad;
277                 host= *(++argv);
278                 }
279 #if 0
280         else if( strcmp(*argv,"-host") == 0)
281                 {
282                 if (--argc < 1) goto bad;
283                 host= *(++argv);
284                 }
285         else if( strcmp(*argv,"-port") == 0)
286                 {
287                 if (--argc < 1) goto bad;
288                 port= *(++argv);
289                 }
290 #endif
291         else if (strcmp(*argv,"-reuse") == 0)
292                 perform=2;
293         else if (strcmp(*argv,"-new") == 0)
294                 perform=1;
295         else if( strcmp(*argv,"-verify") == 0) {
296
297             tm_verify=SSL_VERIFY_PEER|SSL_VERIFY_CLIENT_ONCE;
298             if (--argc < 1) goto bad;
299             verify_depth=atoi(*(++argv));
300             BIO_printf(bio_err,"verify depth is %d\n",verify_depth);
301
302         } else if( strcmp(*argv,"-cert") == 0) {
303
304             if (--argc < 1) goto bad;
305             t_cert_file= *(++argv);
306
307         } else if( strcmp(*argv,"-key") == 0) {
308
309             if (--argc < 1) goto bad;
310             t_key_file= *(++argv);
311
312         } else if( strcmp(*argv,"-CApath") == 0) {
313
314             if (--argc < 1) goto bad;
315             CApath= *(++argv);
316
317         } else if( strcmp(*argv,"-CAfile") == 0) {
318
319             if (--argc < 1) goto bad;
320             CAfile= *(++argv);
321
322         } else if( strcmp(*argv,"-cipher") == 0) {
323
324             if (--argc < 1) goto bad;
325             tm_cipher= *(++argv);
326         }
327 #ifdef FIONBIO
328         else if(strcmp(*argv,"-nbio") == 0) {
329             t_nbio=1;
330         }
331 #endif
332         else if(strcmp(*argv,"-www") == 0)
333                 {
334                 if (--argc < 1) goto bad;
335                 s_www_path= *(++argv);
336                 }
337         else if(strcmp(*argv,"-bugs") == 0)
338             st_bugs=1;
339 #ifndef NO_SSL2
340         else if(strcmp(*argv,"-ssl2") == 0)
341             s_time_meth=SSLv2_client_method();
342 #endif
343 #ifndef NO_SSL3
344         else if(strcmp(*argv,"-ssl3") == 0)
345             s_time_meth=SSLv3_client_method();
346 #endif
347         else if( strcmp(*argv,"-time") == 0) {
348
349             if (--argc < 1) goto bad;
350             maxTime= atoi(*(++argv));
351         }
352         else {
353             BIO_printf(bio_err,"unknown option %s\n",*argv);
354             badop=1;
355             break;
356         }
357
358         argc--;
359         argv++;
360     }
361
362     if (perform == 0) perform=3;
363
364     if(badop) {
365 bad:
366                 s_time_usage();
367                 return -1;
368     }
369
370         return 0;                       /* Valid args */
371 }
372
373 /***********************************************************************
374  * TIME - time functions
375  */
376 #define START   0
377 #define STOP    1
378
379 static double tm_Time_F(s)
380 int s;
381         {
382         static double ret;
383 #ifdef TIMES
384         static struct tms tstart,tend;
385
386         if(s == START) {
387                 times(&tstart);
388                 return(0);
389         } else {
390                 times(&tend);
391                 ret=((double)(tend.tms_utime-tstart.tms_utime))/HZ;
392                 return((ret == 0.0)?1e-6:ret);
393         }
394 #else /* !times() */
395         static struct timeb tstart,tend;
396         long i;
397
398         if(s == START) {
399                 ftime(&tstart);
400                 return(0);
401         } else {
402                 ftime(&tend);
403                 i=(long)tend.millitm-(long)tstart.millitm;
404                 ret=((double)(tend.time-tstart.time))+((double)i)/1000.0;
405                 return((ret == 0.0)?1e-6:ret);
406         }
407 #endif
408 }
409
410 /***********************************************************************
411  * MAIN - main processing area for client
412  *                      real name depends on MONOLITH
413  */
414 int
415 MAIN(argc,argv)
416 int argc;
417 char **argv;
418         {
419         double totalTime = 0.0;
420         int nConn = 0;
421         SSL *scon=NULL;
422         long finishtime=0;
423         int ret=1,i;
424         MS_STATIC char buf[1024*8];
425         int ver;
426
427 #if !defined(NO_SSL2) && !defined(NO_SSL3)
428         s_time_meth=SSLv23_client_method();
429 #elif !defined(NO_SSL3)
430         s_time_meth=SSLv3_client_method();
431 #elif !defined(NO_SSL2)
432         s_time_meth=SSLv2_client_method();
433 #endif
434
435         /* parse the command line arguments */
436         if( parseArgs( argc, argv ) < 0 )
437                 goto end;
438
439         SSLeay_add_ssl_algorithms();
440         if ((tm_ctx=SSL_CTX_new(s_time_meth)) == NULL) return(1);
441
442         SSL_CTX_set_quiet_shutdown(tm_ctx,1);
443
444         if (st_bugs) SSL_CTX_set_options(tm_ctx,SSL_OP_ALL);
445         SSL_CTX_set_cipher_list(tm_ctx,tm_cipher);
446         if(!set_cert_stuff(tm_ctx,t_cert_file,t_key_file)) 
447                 goto end;
448
449         SSL_load_error_strings();
450
451         if ((!SSL_CTX_load_verify_locations(tm_ctx,CAfile,CApath)) ||
452                 (!SSL_CTX_set_default_verify_paths(tm_ctx)))
453                 {
454                 /* BIO_printf(bio_err,"error seting default verify locations\n"); */
455                 ERR_print_errors(bio_err);
456                 /* goto end; */
457                 }
458
459         if (tm_cipher == NULL)
460                 tm_cipher = getenv("SSL_CIPHER");
461
462         if (tm_cipher == NULL ) {
463                 fprintf( stderr, "No CIPHER specified\n" );
464 /*              EXIT(1); */
465         }
466
467         if (!(perform & 1)) goto next;
468         printf( "Collecting connection statistics for %d seconds\n", maxTime );
469
470         /* Loop and time how long it takes to make connections */
471
472         bytes_read=0;
473         finishtime=(long)time(NULL)+maxTime;
474         tm_Time_F(START);
475         for (;;)
476                 {
477                 if (finishtime < time(NULL)) break;
478 #ifdef WIN32_STUFF
479
480                 if( flushWinMsgs(0) == -1 )
481                         goto end;
482
483                 if( waitingToDie || exitNow )           /* we're dead */
484                         goto end;
485 #endif
486
487                 if( (scon = doConnection( NULL )) == NULL )
488                         goto end;
489
490                 if (s_www_path != NULL)
491                         {
492                         sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
493                         SSL_write(scon,buf,strlen(buf));
494                         while ((i=SSL_read(scon,buf,sizeof(buf))) > 0)
495                                 bytes_read+=i;
496                         }
497
498 #ifdef NO_SHUTDOWN
499                 SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
500 #else
501                 SSL_shutdown(scon);
502 #endif
503                 SHUTDOWN2(SSL_get_fd(scon));
504
505                 nConn += 1;
506                 if (SSL_session_reused(scon))
507                         ver='r';
508                 else
509                         {
510                         ver=SSL_version(scon);
511                         if (ver == TLS1_VERSION)
512                                 ver='t';
513                         else if (ver == SSL3_VERSION)
514                                 ver='3';
515                         else if (ver == SSL2_VERSION)
516                                 ver='2';
517                         else
518                                 ver='*';
519                         }
520                 fputc(ver,stdout);
521                 fflush(stdout);
522
523                 SSL_free( scon );
524                 scon=NULL;
525                 }
526         totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
527
528         i=(int)(time(NULL)-finishtime+maxTime);
529         printf( "\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", nConn, totalTime, ((double)nConn/totalTime),bytes_read);
530         printf( "%d connections in %ld real seconds, %ld bytes read per connection\n",nConn,time(NULL)-finishtime+maxTime,bytes_read/nConn);
531
532         /* Now loop and time connections using the same session id over and over */
533
534 next:
535         if (!(perform & 2)) goto end;
536         printf( "\n\nNow timing with session id reuse.\n" );
537
538         /* Get an SSL object so we can reuse the session id */
539         if( (scon = doConnection( NULL )) == NULL )
540                 {
541                 fprintf( stderr, "Unable to get connection\n" );
542                 goto end;
543                 }
544
545         if (s_www_path != NULL)
546                 {
547                 sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
548                 SSL_write(scon,buf,strlen(buf));
549                 while (SSL_read(scon,buf,sizeof(buf)) > 0)
550                         ;
551                 }
552 #ifdef NO_SHUTDOWN
553         SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
554 #else
555         SSL_shutdown(scon);
556 #endif
557         SHUTDOWN2(SSL_get_fd(scon));
558
559         nConn = 0;
560         totalTime = 0.0;
561
562         finishtime=time(NULL)+maxTime;
563
564         printf( "starting\n" );
565         bytes_read=0;
566         tm_Time_F(START);
567                 
568         for (;;)
569                 {
570                 if (finishtime < time(NULL)) break;
571
572 #ifdef WIN32_STUFF
573                 if( flushWinMsgs(0) == -1 )
574                         goto end;
575
576                 if( waitingToDie || exitNow )   /* we're dead */
577                         goto end;
578 #endif
579
580                 if( (doConnection( scon )) == NULL )
581                         goto end;
582
583                 if (s_www_path)
584                         {
585                         sprintf(buf,"GET %s HTTP/1.0\r\n\r\n",s_www_path);
586                         SSL_write(scon,buf,strlen(buf));
587                         while ((i=SSL_read(scon,buf,sizeof(buf))) > 0)
588                                 bytes_read+=i;
589                         }
590
591 #ifdef NO_SHUTDOWN
592                 SSL_set_shutdown(scon,SSL_SENT_SHUTDOWN|SSL_RECEIVED_SHUTDOWN);
593 #else
594                 SSL_shutdown(scon);
595 #endif
596                 SHUTDOWN2(SSL_get_fd(scon));
597         
598                 nConn += 1;
599                 if (SSL_session_reused(scon))
600                         ver='r';
601                 else
602                         {
603                         ver=SSL_version(scon);
604                         if (ver == TLS1_VERSION)
605                                 ver='t';
606                         else if (ver == SSL3_VERSION)
607                                 ver='3';
608                         else if (ver == SSL2_VERSION)
609                                 ver='2';
610                         else
611                                 ver='*';
612                         }
613                 fputc(ver,stdout);
614                 fflush(stdout);
615                 }
616         totalTime += tm_Time_F(STOP); /* Add the time for this iteration*/
617
618
619         printf( "\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", nConn, totalTime, ((double)nConn/totalTime),bytes_read);
620         printf( "%d connections in %ld real seconds, %ld bytes read per connection\n",nConn,time(NULL)-finishtime+maxTime,bytes_read/nConn);
621
622         ret=0;
623 end:
624         if (scon != NULL) SSL_free(scon);
625
626         if (tm_ctx != NULL)
627                 {
628                 SSL_CTX_free(tm_ctx);
629                 tm_ctx=NULL;
630                 }
631         EXIT(ret);
632         }
633
634 /***********************************************************************
635  * doConnection - make a connection
636  * Args:
637  *              scon    = earlier ssl connection for session id, or NULL
638  * Returns:
639  *              SSL *   = the connection pointer.
640  */
641 static SSL *
642 doConnection(scon)
643 SSL *scon;
644         {
645         BIO *conn;
646         SSL *serverCon;
647         int width, i;
648         fd_set readfds;
649
650         if ((conn=BIO_new(BIO_s_connect())) == NULL)
651                 return(NULL);
652
653 /*      BIO_set_conn_port(conn,port);*/
654         BIO_set_conn_hostname(conn,host);
655
656         if (scon == NULL)
657                 serverCon=(SSL *)SSL_new(tm_ctx);
658         else
659                 {
660                 serverCon=scon;
661                 SSL_set_connect_state(serverCon);
662                 }
663
664         SSL_set_bio(serverCon,conn,conn);
665
666 #if 0
667         if( scon != NULL )
668                 SSL_set_session(serverCon,SSL_get_session(scon));
669 #endif
670
671         /* ok, lets connect */
672         for(;;) {
673                 i=SSL_connect(serverCon);
674                 if (BIO_sock_should_retry(i))
675                         {
676                         BIO_printf(bio_err,"DELAY\n");
677
678                         i=SSL_get_fd(serverCon);
679                         width=i+1;
680                         FD_ZERO(&readfds);
681                         FD_SET(i,&readfds);
682                         select(width,&readfds,NULL,NULL,NULL);
683                         continue;
684                         }
685                 break;
686                 }
687         if(i <= 0)
688                 {
689                 BIO_printf(bio_err,"ERROR\n");
690                 if (verify_error != X509_V_OK)
691                         BIO_printf(bio_err,"verify error:%s\n",
692                                 X509_verify_cert_error_string(verify_error));
693                 else
694                         ERR_print_errors(bio_err);
695                 if (scon == NULL)
696                         SSL_free(serverCon);
697                 return NULL;
698                 }
699
700         return serverCon;
701         }
702
703