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