cc9a9797556e3b7d126defaace44701401c740db
[openssl.git] / apps / s_time.c
1 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
2  * All rights reserved.
3  *
4  * This package is an SSL implementation written
5  * by Eric Young (eay@cryptsoft.com).
6  * The implementation was written so as to conform with Netscapes SSL.
7  *
8  * This library is free for commercial and non-commercial use as long as
9  * the following conditions are aheared to.  The following conditions
10  * apply to all code found in this distribution, be it the RC4, RSA,
11  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
12  * included with this distribution is covered by the same copyright terms
13  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
14  *
15  * Copyright remains Eric Young's, and as such any Copyright notices in
16  * the code are not to be removed.
17  * If this package is used in a product, Eric Young should be given attribution
18  * as the author of the parts of the library used.
19  * This can be in the form of a textual message at program startup or
20  * in documentation (online or textual) provided with the package.
21  *
22  * Redistribution and use in source and binary forms, with or without
23  * modification, are permitted provided that the following conditions
24  * are met:
25  * 1. Redistributions of source code must retain the copyright
26  *    notice, this list of conditions and the following disclaimer.
27  * 2. Redistributions in binary form must reproduce the above copyright
28  *    notice, this list of conditions and the following disclaimer in the
29  *    documentation and/or other materials provided with the distribution.
30  * 3. All advertising materials mentioning features or use of this software
31  *    must display the following acknowledgement:
32  *    "This product includes cryptographic software written by
33  *     Eric Young (eay@cryptsoft.com)"
34  *    The word 'cryptographic' can be left out if the rouines from the library
35  *    being used are not cryptographic related :-).
36  * 4. If you include any Windows specific code (or a derivative thereof) from
37  *    the apps directory (application code) you must include an acknowledgement:
38  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
39  *
40  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
41  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
43  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
44  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
45  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
46  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
48  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
49  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
50  * SUCH DAMAGE.
51  *
52  * The licence and distribution terms for any publically available version or
53  * derivative of this code cannot be changed.  i.e. this code cannot simply be
54  * copied and put under another distribution licence
55  * [including the GNU Public Licence.]
56  */
57
58 #define NO_SHUTDOWN
59
60 /* ----------------------------------------
61    s_time - SSL client connection timer program
62    Written and donated by Larry Streepy <streepy@healthcare.com>
63   -----------------------------------------*/
64
65 #include <stdio.h>
66 #include <stdlib.h>
67 #include <string.h>
68
69 #define USE_SOCKETS
70 #include "apps.h"
71 #include <openssl/x509.h>
72 #include <openssl/ssl.h>
73 #include <openssl/pem.h>
74 #include "s_apps.h"
75 #include <openssl/err.h>
76 #if !defined(OPENSSL_SYS_MSDOS)
77 # include OPENSSL_UNISTD
78 #endif
79
80 #undef ioctl
81 #define ioctl ioctlsocket
82
83 #define SSL_CONNECT_NAME        "localhost:4433"
84
85 /* no default cert. */
86 /*
87  * #define TEST_CERT "client.pem"
88  */
89
90 #undef BUFSIZZ
91 #define BUFSIZZ 1024*10
92
93 #define MYBUFSIZ 1024*8
94
95 #undef min
96 #undef max
97 #define min(a,b) (((a) < (b)) ? (a) : (b))
98 #define max(a,b) (((a) > (b)) ? (a) : (b))
99
100 #undef SECONDS
101 #define SECONDS 30
102 #define SECONDSSTR "30"
103
104 extern int verify_depth;
105 extern int verify_error;
106
107 static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx);
108
109 typedef enum OPTION_choice {
110     OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
111     OPT_CONNECT, OPT_CIPHER, OPT_CERT, OPT_KEY, OPT_CAPATH,
112     OPT_CAFILE, OPT_NOCAPATH, OPT_NOCAFILE, OPT_NEW, OPT_REUSE, OPT_BUGS,
113     OPT_VERIFY, OPT_TIME, OPT_SSL3,
114     OPT_WWW
115 } OPTION_CHOICE;
116
117 OPTIONS s_time_options[] = {
118     {"help", OPT_HELP, '-', "Display this summary"},
119     {"connect", OPT_CONNECT, 's',
120      "Where to connect as post:port (default is " SSL_CONNECT_NAME ")"},
121     {"cipher", OPT_CIPHER, 's', "Cipher to use, see 'openssl ciphers'"},
122     {"cert", OPT_CERT, '<', "Cert file to use, PEM format assumed"},
123     {"key", OPT_KEY, '<', "File with key, PEM; default is -cert file"},
124     {"CApath", OPT_CAPATH, '/', "PEM format directory of CA's"},
125     {"cafile", OPT_CAFILE, '<', "PEM format file of CA's"},
126     {"no-CAfile", OPT_NOCAFILE, '-',
127      "Do not load the default certificates file"},
128     {"no-CApath", OPT_NOCAPATH, '-',
129      "Do not load certificates from the default certificates directory"},
130     {"new", OPT_NEW, '-', "Just time new connections"},
131     {"reuse", OPT_REUSE, '-', "Just time connection reuse"},
132     {"bugs", OPT_BUGS, '-', "Turn on SSL bug compatibility"},
133     {"verify", OPT_VERIFY, 'p',
134      "Turn on peer certificate verification, set depth"},
135     {"time", OPT_TIME, 'p', "Sf seconds to collect data, default" SECONDSSTR},
136     {"www", OPT_WWW, 's', "Fetch specified page from the site"},
137 #ifndef OPENSSL_NO_SSL3
138     {"ssl3", OPT_SSL3, '-', "Just use SSLv3"},
139 #endif
140     {NULL}
141 };
142
143 #define START   0
144 #define STOP    1
145
146 static double tm_Time_F(int s)
147 {
148     return app_tminterval(s, 1);
149 }
150
151 int s_time_main(int argc, char **argv)
152 {
153     char buf[1024 * 8];
154     SSL *scon = NULL;
155     SSL_CTX *ctx = NULL;
156     const SSL_METHOD *meth = NULL;
157     char *CApath = NULL, *CAfile = NULL, *cipher = NULL, *www_path = NULL;
158     char *host = SSL_CONNECT_NAME, *certfile = NULL, *keyfile = NULL, *prog;
159     double totalTime = 0.0;
160     int noCApath = 0, noCAfile = 0;
161     int maxtime = SECONDS, nConn = 0, perform = 3, ret = 1, i, st_bugs =
162         0, ver;
163     long bytes_read = 0, finishtime = 0;
164     OPTION_CHOICE o;
165
166     meth = TLS_client_method();
167     verify_depth = 0;
168     verify_error = X509_V_OK;
169
170     prog = opt_init(argc, argv, s_time_options);
171     while ((o = opt_next()) != OPT_EOF) {
172         switch (o) {
173         case OPT_EOF:
174         case OPT_ERR:
175  opthelp:
176             BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
177             goto end;
178         case OPT_HELP:
179             opt_help(s_time_options);
180             ret = 0;
181             goto end;
182         case OPT_CONNECT:
183             host = opt_arg();
184             break;
185         case OPT_REUSE:
186             perform = 2;
187             break;
188         case OPT_NEW:
189             perform = 1;
190             break;
191         case OPT_VERIFY:
192             if (!opt_int(opt_arg(), &verify_depth))
193                 goto opthelp;
194             BIO_printf(bio_err, "%s: verify depth is %d\n",
195                        prog, verify_depth);
196             break;
197         case OPT_CERT:
198             certfile = opt_arg();
199             break;
200         case OPT_KEY:
201             keyfile = opt_arg();
202             break;
203         case OPT_CAPATH:
204             CApath = opt_arg();
205             break;
206         case OPT_CAFILE:
207             CAfile = opt_arg();
208             break;
209         case OPT_NOCAPATH:
210             noCApath = 1;
211             break;
212         case OPT_NOCAFILE:
213             noCAfile = 1;
214             break;
215         case OPT_CIPHER:
216             cipher = opt_arg();
217             break;
218         case OPT_BUGS:
219             st_bugs = 1;
220             break;
221         case OPT_TIME:
222             if (!opt_int(opt_arg(), &maxtime))
223                 goto opthelp;
224             break;
225         case OPT_WWW:
226             www_path = opt_arg();
227             if (strlen(www_path) > MYBUFSIZ - 100) {
228                 BIO_printf(bio_err, "%s: -www option too long\n", prog);
229                 goto end;
230             }
231             break;
232         case OPT_SSL3:
233 #ifndef OPENSSL_NO_SSL3
234             meth = SSLv3_client_method();
235 #endif
236             break;
237         }
238     }
239     argc = opt_num_rest();
240     if (argc != 0)
241         goto opthelp;
242
243     if (cipher == NULL)
244         cipher = getenv("SSL_CIPHER");
245     if (cipher == NULL) {
246         BIO_printf(bio_err, "No CIPHER specified\n");
247         goto end;
248     }
249
250     if ((ctx = SSL_CTX_new(meth)) == NULL)
251         goto end;
252
253     SSL_CTX_set_quiet_shutdown(ctx, 1);
254
255     if (st_bugs)
256         SSL_CTX_set_options(ctx, SSL_OP_ALL);
257     if (!SSL_CTX_set_cipher_list(ctx, cipher))
258         goto end;
259     if (!set_cert_stuff(ctx, certfile, keyfile))
260         goto end;
261
262     if (!ctx_set_verify_locations(ctx, CAfile, CApath, noCAfile, noCApath)) {
263         ERR_print_errors(bio_err);
264         goto end;
265     }
266     if (!(perform & 1))
267         goto next;
268     printf("Collecting connection statistics for %d seconds\n", maxtime);
269
270     /* Loop and time how long it takes to make connections */
271
272     bytes_read = 0;
273     finishtime = (long)time(NULL) + maxtime;
274     tm_Time_F(START);
275     for (;;) {
276         if (finishtime < (long)time(NULL))
277             break;
278
279         if ((scon = doConnection(NULL, host, ctx)) == NULL)
280             goto end;
281
282         if (www_path != NULL) {
283             BIO_snprintf(buf, sizeof buf, "GET %s HTTP/1.0\r\n\r\n",
284                          www_path);
285             if (SSL_write(scon, buf, strlen(buf)) <= 0)
286                 goto end;
287             while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
288                 bytes_read += i;
289         }
290 #ifdef NO_SHUTDOWN
291         SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
292 #else
293         SSL_shutdown(scon);
294 #endif
295         BIO_closesocket(SSL_get_fd(scon));
296
297         nConn += 1;
298         if (SSL_session_reused(scon))
299             ver = 'r';
300         else {
301             ver = SSL_version(scon);
302             if (ver == TLS1_VERSION)
303                 ver = 't';
304             else if (ver == SSL3_VERSION)
305                 ver = '3';
306             else
307                 ver = '*';
308         }
309         fputc(ver, stdout);
310         fflush(stdout);
311
312         SSL_free(scon);
313         scon = NULL;
314     }
315     totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
316
317     i = (int)((long)time(NULL) - finishtime + maxtime);
318     printf
319         ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n",
320          nConn, totalTime, ((double)nConn / totalTime), bytes_read);
321     printf
322         ("%d connections in %ld real seconds, %ld bytes read per connection\n",
323          nConn, (long)time(NULL) - finishtime + maxtime, bytes_read / nConn);
324
325     /*
326      * Now loop and time connections using the same session id over and over
327      */
328
329  next:
330     if (!(perform & 2))
331         goto end;
332     printf("\n\nNow timing with session id reuse.\n");
333
334     /* Get an SSL object so we can reuse the session id */
335     if ((scon = doConnection(NULL, host, ctx)) == NULL) {
336         BIO_printf(bio_err, "Unable to get connection\n");
337         goto end;
338     }
339
340     if (www_path != NULL) {
341         BIO_snprintf(buf, sizeof buf, "GET %s HTTP/1.0\r\n\r\n", www_path);
342         if (SSL_write(scon, buf, strlen(buf)) <= 0)
343             goto end;
344         while (SSL_read(scon, buf, sizeof(buf)) > 0)
345             continue;
346     }
347 #ifdef NO_SHUTDOWN
348     SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
349 #else
350     SSL_shutdown(scon);
351 #endif
352     BIO_closesocket(SSL_get_fd(scon));
353
354     nConn = 0;
355     totalTime = 0.0;
356
357     finishtime = (long)time(NULL) + maxtime;
358
359     printf("starting\n");
360     bytes_read = 0;
361     tm_Time_F(START);
362
363     for (;;) {
364         if (finishtime < (long)time(NULL))
365             break;
366
367         if ((doConnection(scon, host, ctx)) == NULL)
368             goto end;
369
370         if (www_path) {
371             BIO_snprintf(buf, sizeof buf, "GET %s HTTP/1.0\r\n\r\n",
372                          www_path);
373             if (SSL_write(scon, buf, strlen(buf)) <= 0)
374                 goto end;
375             while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
376                 bytes_read += i;
377         }
378 #ifdef NO_SHUTDOWN
379         SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
380 #else
381         SSL_shutdown(scon);
382 #endif
383         BIO_closesocket(SSL_get_fd(scon));
384
385         nConn += 1;
386         if (SSL_session_reused(scon))
387             ver = 'r';
388         else {
389             ver = SSL_version(scon);
390             if (ver == TLS1_VERSION)
391                 ver = 't';
392             else if (ver == SSL3_VERSION)
393                 ver = '3';
394             else
395                 ver = '*';
396         }
397         fputc(ver, stdout);
398         fflush(stdout);
399     }
400     totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
401
402     printf
403         ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n",
404          nConn, totalTime, ((double)nConn / totalTime), bytes_read);
405     printf
406         ("%d connections in %ld real seconds, %ld bytes read per connection\n",
407          nConn, (long)time(NULL) - finishtime + maxtime, bytes_read / nConn);
408
409     ret = 0;
410
411  end:
412     SSL_free(scon);
413     SSL_CTX_free(ctx);
414     return (ret);
415 }
416
417 /*-
418  * doConnection - make a connection
419  */
420 static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx)
421 {
422     BIO *conn;
423     SSL *serverCon;
424     int width, i;
425     fd_set readfds;
426
427     if ((conn = BIO_new(BIO_s_connect())) == NULL)
428         return (NULL);
429
430     BIO_set_conn_hostname(conn, host);
431
432     if (scon == NULL)
433         serverCon = SSL_new(ctx);
434     else {
435         serverCon = scon;
436         SSL_set_connect_state(serverCon);
437     }
438
439     SSL_set_bio(serverCon, conn, conn);
440
441     /* ok, lets connect */
442     for (;;) {
443         i = SSL_connect(serverCon);
444         if (BIO_sock_should_retry(i)) {
445             BIO_printf(bio_err, "DELAY\n");
446
447             i = SSL_get_fd(serverCon);
448             width = i + 1;
449             FD_ZERO(&readfds);
450             openssl_fdset(i, &readfds);
451             /*
452              * Note: under VMS with SOCKETSHR the 2nd parameter is currently
453              * of type (int *) whereas under other systems it is (void *) if
454              * you don't have a cast it will choke the compiler: if you do
455              * have a cast then you can either go for (int *) or (void *).
456              */
457             select(width, (void *)&readfds, NULL, NULL, NULL);
458             continue;
459         }
460         break;
461     }
462     if (i <= 0) {
463         BIO_printf(bio_err, "ERROR\n");
464         if (verify_error != X509_V_OK)
465             BIO_printf(bio_err, "verify error:%s\n",
466                        X509_verify_cert_error_string(verify_error));
467         else
468             ERR_print_errors(bio_err);
469         if (scon == NULL)
470             SSL_free(serverCon);
471         return NULL;
472     }
473
474     return serverCon;
475 }