de25319929dd7def0e562421d8c48d175e15fcb5
[oweals/openssl.git] / apps / s_time.c
1 /*
2  * Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 #define NO_SHUTDOWN
11
12 #include <stdio.h>
13 #include <stdlib.h>
14 #include <string.h>
15
16 #include <openssl/opensslconf.h>
17
18 #ifndef OPENSSL_NO_SOCK
19
20 #define USE_SOCKETS
21 #include "apps.h"
22 #include <openssl/x509.h>
23 #include <openssl/ssl.h>
24 #include <openssl/pem.h>
25 #include "s_apps.h"
26 #include <openssl/err.h>
27 #if !defined(OPENSSL_SYS_MSDOS)
28 # include OPENSSL_UNISTD
29 #endif
30
31 #undef ioctl
32 #define ioctl ioctlsocket
33
34 #define SSL_CONNECT_NAME        "localhost:4433"
35
36 /* no default cert. */
37 /*
38  * #define TEST_CERT "client.pem"
39  */
40
41 #undef BUFSIZZ
42 #define BUFSIZZ 1024*10
43
44 #undef min
45 #undef max
46 #define min(a,b) (((a) < (b)) ? (a) : (b))
47 #define max(a,b) (((a) > (b)) ? (a) : (b))
48
49 #undef SECONDS
50 #define SECONDS 30
51 #define SECONDSSTR "30"
52
53 static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx);
54
55 static const char fmt_http_get_cmd[] = "GET %s HTTP/1.0\r\n\r\n";
56
57 typedef enum OPTION_choice {
58     OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
59     OPT_CONNECT, OPT_CIPHER, OPT_CERT, OPT_KEY, OPT_CAPATH,
60     OPT_CAFILE, OPT_NOCAPATH, OPT_NOCAFILE, OPT_NEW, OPT_REUSE, OPT_BUGS,
61     OPT_VERIFY, OPT_TIME, OPT_SSL3,
62     OPT_WWW
63 } OPTION_CHOICE;
64
65 OPTIONS s_time_options[] = {
66     {"help", OPT_HELP, '-', "Display this summary"},
67     {"connect", OPT_CONNECT, 's',
68      "Where to connect as post:port (default is " SSL_CONNECT_NAME ")"},
69     {"cipher", OPT_CIPHER, 's', "Cipher to use, see 'openssl ciphers'"},
70     {"cert", OPT_CERT, '<', "Cert file to use, PEM format assumed"},
71     {"key", OPT_KEY, '<', "File with key, PEM; default is -cert file"},
72     {"CApath", OPT_CAPATH, '/', "PEM format directory of CA's"},
73     {"cafile", OPT_CAFILE, '<', "PEM format file of CA's"},
74     {"no-CAfile", OPT_NOCAFILE, '-',
75      "Do not load the default certificates file"},
76     {"no-CApath", OPT_NOCAPATH, '-',
77      "Do not load certificates from the default certificates directory"},
78     {"new", OPT_NEW, '-', "Just time new connections"},
79     {"reuse", OPT_REUSE, '-', "Just time connection reuse"},
80     {"bugs", OPT_BUGS, '-', "Turn on SSL bug compatibility"},
81     {"verify", OPT_VERIFY, 'p',
82      "Turn on peer certificate verification, set depth"},
83     {"time", OPT_TIME, 'p', "Seconds to collect data, default " SECONDSSTR},
84     {"www", OPT_WWW, 's', "Fetch specified page from the site"},
85 #ifndef OPENSSL_NO_SSL3
86     {"ssl3", OPT_SSL3, '-', "Just use SSLv3"},
87 #endif
88     {NULL}
89 };
90
91 #define START   0
92 #define STOP    1
93
94 static double tm_Time_F(int s)
95 {
96     return app_tminterval(s, 1);
97 }
98
99 int s_time_main(int argc, char **argv)
100 {
101     char buf[1024 * 8];
102     SSL *scon = NULL;
103     SSL_CTX *ctx = NULL;
104     const SSL_METHOD *meth = NULL;
105     char *CApath = NULL, *CAfile = NULL, *cipher = NULL, *www_path = NULL;
106     char *host = SSL_CONNECT_NAME, *certfile = NULL, *keyfile = NULL, *prog;
107     double totalTime = 0.0;
108     int noCApath = 0, noCAfile = 0;
109     int maxtime = SECONDS, nConn = 0, perform = 3, ret = 1, i, st_bugs = 0;
110     long bytes_read = 0, finishtime = 0;
111     OPTION_CHOICE o;
112     int max_version = 0, ver, buf_len;
113     size_t buf_size;
114
115     meth = TLS_client_method();
116
117     prog = opt_init(argc, argv, s_time_options);
118     while ((o = opt_next()) != OPT_EOF) {
119         switch (o) {
120         case OPT_EOF:
121         case OPT_ERR:
122  opthelp:
123             BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
124             goto end;
125         case OPT_HELP:
126             opt_help(s_time_options);
127             ret = 0;
128             goto end;
129         case OPT_CONNECT:
130             host = opt_arg();
131             break;
132         case OPT_REUSE:
133             perform = 2;
134             break;
135         case OPT_NEW:
136             perform = 1;
137             break;
138         case OPT_VERIFY:
139             if (!opt_int(opt_arg(), &verify_args.depth))
140                 goto opthelp;
141             BIO_printf(bio_err, "%s: verify depth is %d\n",
142                        prog, verify_args.depth);
143             break;
144         case OPT_CERT:
145             certfile = opt_arg();
146             break;
147         case OPT_KEY:
148             keyfile = opt_arg();
149             break;
150         case OPT_CAPATH:
151             CApath = opt_arg();
152             break;
153         case OPT_CAFILE:
154             CAfile = opt_arg();
155             break;
156         case OPT_NOCAPATH:
157             noCApath = 1;
158             break;
159         case OPT_NOCAFILE:
160             noCAfile = 1;
161             break;
162         case OPT_CIPHER:
163             cipher = opt_arg();
164             break;
165         case OPT_BUGS:
166             st_bugs = 1;
167             break;
168         case OPT_TIME:
169             if (!opt_int(opt_arg(), &maxtime))
170                 goto opthelp;
171             break;
172         case OPT_WWW:
173             www_path = opt_arg();
174             buf_size = strlen(www_path) + sizeof(fmt_http_get_cmd) - 2;  /* 2 is for %s */
175             if (buf_size > sizeof(buf)) {
176                 BIO_printf(bio_err, "%s: -www option is too long\n", prog);
177                 goto end;
178             }
179             break;
180         case OPT_SSL3:
181             max_version = SSL3_VERSION;
182             break;
183         }
184     }
185     argc = opt_num_rest();
186     if (argc != 0)
187         goto opthelp;
188
189     if (cipher == NULL)
190         cipher = getenv("SSL_CIPHER");
191     if (cipher == NULL) {
192         BIO_printf(bio_err, "No CIPHER specified\n");
193         goto end;
194     }
195
196     if ((ctx = SSL_CTX_new(meth)) == NULL)
197         goto end;
198
199     SSL_CTX_set_quiet_shutdown(ctx, 1);
200     if (SSL_CTX_set_max_proto_version(ctx, max_version) == 0)
201         goto end;
202
203     if (st_bugs)
204         SSL_CTX_set_options(ctx, SSL_OP_ALL);
205     if (!SSL_CTX_set_cipher_list(ctx, cipher))
206         goto end;
207     if (!set_cert_stuff(ctx, certfile, keyfile))
208         goto end;
209
210     if (!ctx_set_verify_locations(ctx, CAfile, CApath, noCAfile, noCApath)) {
211         ERR_print_errors(bio_err);
212         goto end;
213     }
214     if (!(perform & 1))
215         goto next;
216     printf("Collecting connection statistics for %d seconds\n", maxtime);
217
218     /* Loop and time how long it takes to make connections */
219
220     bytes_read = 0;
221     finishtime = (long)time(NULL) + maxtime;
222     tm_Time_F(START);
223     for (;;) {
224         if (finishtime < (long)time(NULL))
225             break;
226
227         if ((scon = doConnection(NULL, host, ctx)) == NULL)
228             goto end;
229
230         if (www_path != NULL) {
231             buf_len = BIO_snprintf(buf, sizeof buf,
232                                    fmt_http_get_cmd, www_path);
233             if (SSL_write(scon, buf, buf_len) <= 0)
234                 goto end;
235             while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
236                 bytes_read += i;
237         }
238 #ifdef NO_SHUTDOWN
239         SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
240 #else
241         SSL_shutdown(scon);
242 #endif
243         BIO_closesocket(SSL_get_fd(scon));
244
245         nConn += 1;
246         if (SSL_session_reused(scon))
247             ver = 'r';
248         else {
249             ver = SSL_version(scon);
250             if (ver == TLS1_VERSION)
251                 ver = 't';
252             else if (ver == SSL3_VERSION)
253                 ver = '3';
254             else
255                 ver = '*';
256         }
257         fputc(ver, stdout);
258         fflush(stdout);
259
260         SSL_free(scon);
261         scon = NULL;
262     }
263     totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
264
265     i = (int)((long)time(NULL) - finishtime + maxtime);
266     printf
267         ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n",
268          nConn, totalTime, ((double)nConn / totalTime), bytes_read);
269     printf
270         ("%d connections in %ld real seconds, %ld bytes read per connection\n",
271          nConn, (long)time(NULL) - finishtime + maxtime, bytes_read / nConn);
272
273     /*
274      * Now loop and time connections using the same session id over and over
275      */
276
277  next:
278     if (!(perform & 2))
279         goto end;
280     printf("\n\nNow timing with session id reuse.\n");
281
282     /* Get an SSL object so we can reuse the session id */
283     if ((scon = doConnection(NULL, host, ctx)) == NULL) {
284         BIO_printf(bio_err, "Unable to get connection\n");
285         goto end;
286     }
287
288     if (www_path != NULL) {
289         buf_len = BIO_snprintf(buf, sizeof buf,
290                                fmt_http_get_cmd, www_path);
291         if (SSL_write(scon, buf, buf_len) <= 0)
292             goto end;
293         while (SSL_read(scon, buf, sizeof(buf)) > 0)
294             continue;
295     }
296 #ifdef NO_SHUTDOWN
297     SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
298 #else
299     SSL_shutdown(scon);
300 #endif
301     BIO_closesocket(SSL_get_fd(scon));
302
303     nConn = 0;
304     totalTime = 0.0;
305
306     finishtime = (long)time(NULL) + maxtime;
307
308     printf("starting\n");
309     bytes_read = 0;
310     tm_Time_F(START);
311
312     for (;;) {
313         if (finishtime < (long)time(NULL))
314             break;
315
316         if ((doConnection(scon, host, ctx)) == NULL)
317             goto end;
318
319         if (www_path) {
320             BIO_snprintf(buf, sizeof buf, "GET %s HTTP/1.0\r\n\r\n",
321                          www_path);
322             if (SSL_write(scon, buf, strlen(buf)) <= 0)
323                 goto end;
324             while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
325                 bytes_read += i;
326         }
327 #ifdef NO_SHUTDOWN
328         SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
329 #else
330         SSL_shutdown(scon);
331 #endif
332         BIO_closesocket(SSL_get_fd(scon));
333
334         nConn += 1;
335         if (SSL_session_reused(scon))
336             ver = 'r';
337         else {
338             ver = SSL_version(scon);
339             if (ver == TLS1_VERSION)
340                 ver = 't';
341             else if (ver == SSL3_VERSION)
342                 ver = '3';
343             else
344                 ver = '*';
345         }
346         fputc(ver, stdout);
347         fflush(stdout);
348     }
349     totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
350
351     printf
352         ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n",
353          nConn, totalTime, ((double)nConn / totalTime), bytes_read);
354     printf
355         ("%d connections in %ld real seconds, %ld bytes read per connection\n",
356          nConn, (long)time(NULL) - finishtime + maxtime, bytes_read / nConn);
357
358     ret = 0;
359
360  end:
361     SSL_free(scon);
362     SSL_CTX_free(ctx);
363     return (ret);
364 }
365
366 /*-
367  * doConnection - make a connection
368  */
369 static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx)
370 {
371     BIO *conn;
372     SSL *serverCon;
373     int width, i;
374     fd_set readfds;
375
376     if ((conn = BIO_new(BIO_s_connect())) == NULL)
377         return (NULL);
378
379     BIO_set_conn_hostname(conn, host);
380
381     if (scon == NULL)
382         serverCon = SSL_new(ctx);
383     else {
384         serverCon = scon;
385         SSL_set_connect_state(serverCon);
386     }
387
388     SSL_set_bio(serverCon, conn, conn);
389
390     /* ok, lets connect */
391     for (;;) {
392         i = SSL_connect(serverCon);
393         if (BIO_sock_should_retry(i)) {
394             BIO_printf(bio_err, "DELAY\n");
395
396             i = SSL_get_fd(serverCon);
397             width = i + 1;
398             FD_ZERO(&readfds);
399             openssl_fdset(i, &readfds);
400             /*
401              * Note: under VMS with SOCKETSHR the 2nd parameter is currently
402              * of type (int *) whereas under other systems it is (void *) if
403              * you don't have a cast it will choke the compiler: if you do
404              * have a cast then you can either go for (int *) or (void *).
405              */
406             select(width, (void *)&readfds, NULL, NULL, NULL);
407             continue;
408         }
409         break;
410     }
411     if (i <= 0) {
412         BIO_printf(bio_err, "ERROR\n");
413         if (verify_args.error != X509_V_OK)
414             BIO_printf(bio_err, "verify error:%s\n",
415                        X509_verify_cert_error_string(verify_args.error));
416         else
417             ERR_print_errors(bio_err);
418         if (scon == NULL)
419             SSL_free(serverCon);
420         return NULL;
421     }
422
423     return serverCon;
424 }
425 #endif /* OPENSSL_NO_SOCK */