crypto/openssl/apps/s_time.c

   1 /*
   2  * Copyright 1995-2021 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 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13
  14 #include <openssl/opensslconf.h>
  15
  16 #ifndef OPENSSL_NO_SOCK
  17
  18 #include "apps.h"
  19 #include "progs.h"
  20 #include <openssl/x509.h>
  21 #include <openssl/ssl.h>
  22 #include <openssl/pem.h>
  23 #include "s_apps.h"
  24 #include <openssl/err.h>
  25 #include <internal/sockets.h>
  26 #if !defined(OPENSSL_SYS_MSDOS)
  27 # include OPENSSL_UNISTD
  28 #endif
  29
  30 #define SSL_CONNECT_NAME        "localhost:4433"
  31
  32 #define SECONDS 30
  33 #define SECONDSSTR "30"
  34
  35 static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx);
  36
  37 /*
  38  * Define a HTTP get command globally.
  39  * Also define the size of the command, this is two bytes less than
  40  * the size of the string because the %s is replaced by the URL.
  41  */
  42 static const char fmt_http_get_cmd[] = "GET %s HTTP/1.0\r\n\r\n";
  43 static const size_t fmt_http_get_cmd_size = sizeof(fmt_http_get_cmd) - 2;
  44
  45 typedef enum OPTION_choice {
  46     OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
  47     OPT_CONNECT, OPT_CIPHER, OPT_CIPHERSUITES, OPT_CERT, OPT_NAMEOPT, OPT_KEY,
  48     OPT_CAPATH, OPT_CAFILE, OPT_NOCAPATH, OPT_NOCAFILE, OPT_NEW, OPT_REUSE,
  49     OPT_BUGS, OPT_VERIFY, OPT_TIME, OPT_SSL3,
  50     OPT_WWW
  51 } OPTION_CHOICE;
  52
  53 const OPTIONS s_time_options[] = {
  54     {"help", OPT_HELP, '-', "Display this summary"},
  55     {"connect", OPT_CONNECT, 's',
  56      "Where to connect as post:port (default is " SSL_CONNECT_NAME ")"},
  57     {"cipher", OPT_CIPHER, 's', "TLSv1.2 and below cipher list to be used"},
  58     {"ciphersuites", OPT_CIPHERSUITES, 's',
  59      "Specify TLSv1.3 ciphersuites to be used"},
  60     {"cert", OPT_CERT, '<', "Cert file to use, PEM format assumed"},
  61     {"nameopt", OPT_NAMEOPT, 's', "Various certificate name options"},
  62     {"key", OPT_KEY, '<', "File with key, PEM; default is -cert file"},
  63     {"CApath", OPT_CAPATH, '/', "PEM format directory of CA's"},
  64     {"cafile", OPT_CAFILE, '<', "PEM format file of CA's"},
  65     {"CAfile", OPT_CAFILE, '<', "PEM format file of CA's"},
  66     {"no-CAfile", OPT_NOCAFILE, '-',
  67      "Do not load the default certificates file"},
  68     {"no-CApath", OPT_NOCAPATH, '-',
  69      "Do not load certificates from the default certificates directory"},
  70     {"new", OPT_NEW, '-', "Just time new connections"},
  71     {"reuse", OPT_REUSE, '-', "Just time connection reuse"},
  72     {"bugs", OPT_BUGS, '-', "Turn on SSL bug compatibility"},
  73     {"verify", OPT_VERIFY, 'p',
  74      "Turn on peer certificate verification, set depth"},
  75     {"time", OPT_TIME, 'p', "Seconds to collect data, default " SECONDSSTR},
  76     {"www", OPT_WWW, 's', "Fetch specified page from the site"},
  77 #ifndef OPENSSL_NO_SSL3
  78     {"ssl3", OPT_SSL3, '-', "Just use SSLv3"},
  79 #endif
  80     {NULL}
  81 };
  82
  83 #define START   0
  84 #define STOP    1
  85
  86 static double tm_Time_F(int s)
  87 {
  88     return app_tminterval(s, 1);
  89 }
  90
  91 int s_time_main(int argc, char **argv)
  92 {
  93     char buf[1024 * 8];
  94     SSL *scon = NULL;
  95     SSL_CTX *ctx = NULL;
  96     const SSL_METHOD *meth = NULL;
  97     char *CApath = NULL, *CAfile = NULL, *cipher = NULL, *ciphersuites = NULL;
  98     char *www_path = NULL;
  99     char *host = SSL_CONNECT_NAME, *certfile = NULL, *keyfile = NULL, *prog;
 100     double totalTime = 0.0;
 101     int noCApath = 0, noCAfile = 0;
 102     int maxtime = SECONDS, nConn = 0, perform = 3, ret = 1, i, st_bugs = 0;
 103     long bytes_read = 0, finishtime = 0;
 104     OPTION_CHOICE o;
 105     int max_version = 0, ver, buf_len;
 106     size_t buf_size;
 107
 108     meth = TLS_client_method();
 109
 110     prog = opt_init(argc, argv, s_time_options);
 111     while ((o = opt_next()) != OPT_EOF) {
 112         switch (o) {
 113         case OPT_EOF:
 114         case OPT_ERR:
 115  opthelp:
 116             BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
 117             goto end;
 118         case OPT_HELP:
 119             opt_help(s_time_options);
 120             ret = 0;
 121             goto end;
 122         case OPT_CONNECT:
 123             host = opt_arg();
 124             break;
 125         case OPT_REUSE:
 126             perform = 2;
 127             break;
 128         case OPT_NEW:
 129             perform = 1;
 130             break;
 131         case OPT_VERIFY:
 132             if (!opt_int(opt_arg(), &verify_args.depth))
 133                 goto opthelp;
 134             BIO_printf(bio_err, "%s: verify depth is %d\n",
 135                        prog, verify_args.depth);
 136             break;
 137         case OPT_CERT:
 138             certfile = opt_arg();
 139             break;
 140         case OPT_NAMEOPT:
 141             if (!set_nameopt(opt_arg()))
 142                 goto end;
 143             break;
 144         case OPT_KEY:
 145             keyfile = opt_arg();
 146             break;
 147         case OPT_CAPATH:
 148             CApath = opt_arg();
 149             break;
 150         case OPT_CAFILE:
 151             CAfile = opt_arg();
 152             break;
 153         case OPT_NOCAPATH:
 154             noCApath = 1;
 155             break;
 156         case OPT_NOCAFILE:
 157             noCAfile = 1;
 158             break;
 159         case OPT_CIPHER:
 160             cipher = opt_arg();
 161             break;
 162         case OPT_CIPHERSUITES:
 163             ciphersuites = 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) + fmt_http_get_cmd_size;
 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
 192     if ((ctx = SSL_CTX_new(meth)) == NULL)
 193         goto end;
 194
 195     SSL_CTX_set_mode(ctx, SSL_MODE_AUTO_RETRY);
 196     SSL_CTX_set_quiet_shutdown(ctx, 1);
 197     if (SSL_CTX_set_max_proto_version(ctx, max_version) == 0)
 198         goto end;
 199
 200     if (st_bugs)
 201         SSL_CTX_set_options(ctx, SSL_OP_ALL);
 202     if (cipher != NULL && !SSL_CTX_set_cipher_list(ctx, cipher))
 203         goto end;
 204     if (ciphersuites != NULL && !SSL_CTX_set_ciphersuites(ctx, ciphersuites))
 205         goto end;
 206     if (!set_cert_stuff(ctx, certfile, keyfile))
 207         goto end;
 208
 209     if (!ctx_set_verify_locations(ctx, CAfile, CApath, noCAfile, noCApath)) {
 210         ERR_print_errors(bio_err);
 211         goto end;
 212     }
 213     if (!(perform & 1))
 214         goto next;
 215     printf("Collecting connection statistics for %d seconds\n", maxtime);
 216
 217     /* Loop and time how long it takes to make connections */
 218
 219     bytes_read = 0;
 220     finishtime = (long)time(NULL) + maxtime;
 221     tm_Time_F(START);
 222     for (;;) {
 223         if (finishtime < (long)time(NULL))
 224             break;
 225
 226         if ((scon = doConnection(NULL, host, ctx)) == NULL)
 227             goto end;
 228
 229         if (www_path != NULL) {
 230             buf_len = BIO_snprintf(buf, sizeof(buf), fmt_http_get_cmd,
 231                                    www_path);
 232             if (buf_len <= 0 || SSL_write(scon, buf, buf_len) <= 0)
 233                 goto end;
 234             while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
 235                 bytes_read += i;
 236         }
 237         SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
 238         BIO_closesocket(SSL_get_fd(scon));
 239
 240         nConn += 1;
 241         if (SSL_session_reused(scon)) {
 242             ver = 'r';
 243         } else {
 244             ver = SSL_version(scon);
 245             if (ver == TLS1_VERSION)
 246                 ver = 't';
 247             else if (ver == SSL3_VERSION)
 248                 ver = '3';
 249             else
 250                 ver = '*';
 251         }
 252         fputc(ver, stdout);
 253         fflush(stdout);
 254
 255         SSL_free(scon);
 256         scon = NULL;
 257     }
 258     totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
 259
 260     i = (int)((long)time(NULL) - finishtime + maxtime);
 261     printf
 262         ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n",
 263          nConn, totalTime, ((double)nConn / totalTime), bytes_read);
 264     printf
 265         ("%d connections in %ld real seconds, %ld bytes read per connection\n",
 266          nConn, (long)time(NULL) - finishtime + maxtime,
 267          nConn > 0 ? bytes_read / nConn : 0l);
 268
 269     /*
 270      * Now loop and time connections using the same session id over and over
 271      */
 272
 273  next:
 274     if (!(perform & 2))
 275         goto end;
 276     printf("\n\nNow timing with session id reuse.\n");
 277
 278     /* Get an SSL object so we can reuse the session id */
 279     if ((scon = doConnection(NULL, host, ctx)) == NULL) {
 280         BIO_printf(bio_err, "Unable to get connection\n");
 281         goto end;
 282     }
 283
 284     if (www_path != NULL) {
 285         buf_len = BIO_snprintf(buf, sizeof(buf), fmt_http_get_cmd, www_path);
 286         if (buf_len <= 0 || SSL_write(scon, buf, buf_len) <= 0)
 287             goto end;
 288         while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
 289             continue;
 290     }
 291     SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
 292     BIO_closesocket(SSL_get_fd(scon));
 293
 294     nConn = 0;
 295     totalTime = 0.0;
 296
 297     finishtime = (long)time(NULL) + maxtime;
 298
 299     printf("starting\n");
 300     bytes_read = 0;
 301     tm_Time_F(START);
 302
 303     for (;;) {
 304         if (finishtime < (long)time(NULL))
 305             break;
 306
 307         if ((doConnection(scon, host, ctx)) == NULL)
 308             goto end;
 309
 310         if (www_path != NULL) {
 311             buf_len = BIO_snprintf(buf, sizeof(buf), fmt_http_get_cmd,
 312                                    www_path);
 313             if (buf_len <= 0 || SSL_write(scon, buf, buf_len) <= 0)
 314                 goto end;
 315             while ((i = SSL_read(scon, buf, sizeof(buf))) > 0)
 316                 bytes_read += i;
 317         }
 318         SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
 319         BIO_closesocket(SSL_get_fd(scon));
 320
 321         nConn += 1;
 322         if (SSL_session_reused(scon)) {
 323             ver = 'r';
 324         } else {
 325             ver = SSL_version(scon);
 326             if (ver == TLS1_VERSION)
 327                 ver = 't';
 328             else if (ver == SSL3_VERSION)
 329                 ver = '3';
 330             else
 331                 ver = '*';
 332         }
 333         fputc(ver, stdout);
 334         fflush(stdout);
 335     }
 336     totalTime += tm_Time_F(STOP); /* Add the time for this iteration */
 337
 338     printf
 339         ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n",
 340          nConn, totalTime, ((double)nConn / totalTime), bytes_read);
 341     printf
 342         ("%d connections in %ld real seconds, %ld bytes read per connection\n",
 343          nConn, (long)time(NULL) - finishtime + maxtime, bytes_read / nConn);
 344
 345     ret = 0;
 346
 347  end:
 348     SSL_free(scon);
 349     SSL_CTX_free(ctx);
 350     return ret;
 351 }
 352
 353 /*-
 354  * doConnection - make a connection
 355  */
 356 static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx)
 357 {
 358     BIO *conn;
 359     SSL *serverCon;
 360     int i;
 361
 362     if ((conn = BIO_new(BIO_s_connect())) == NULL)
 363         return NULL;
 364
 365     BIO_set_conn_hostname(conn, host);
 366     BIO_set_conn_mode(conn, BIO_SOCK_NODELAY);
 367
 368     if (scon == NULL)
 369         serverCon = SSL_new(ctx);
 370     else {
 371         serverCon = scon;
 372         SSL_set_connect_state(serverCon);
 373     }
 374
 375     SSL_set_bio(serverCon, conn, conn);
 376
 377     /* ok, lets connect */
 378     i = SSL_connect(serverCon);
 379     if (i <= 0) {
 380         BIO_printf(bio_err, "ERROR\n");
 381         if (verify_args.error != X509_V_OK)
 382             BIO_printf(bio_err, "verify error:%s\n",
 383                        X509_verify_cert_error_string(verify_args.error));
 384         else
 385             ERR_print_errors(bio_err);
 386         if (scon == NULL)
 387             SSL_free(serverCon);
 388         return NULL;
 389     }
 390
 391 #if defined(SOL_SOCKET) && defined(SO_LINGER)
 392     {
 393         struct linger no_linger;
 394         int fd;
 395
 396         no_linger.l_onoff  = 1;
 397         no_linger.l_linger = 0;
 398         fd = SSL_get_fd(serverCon);
 399         if (fd >= 0)
 400             (void)setsockopt(fd, SOL_SOCKET, SO_LINGER, (char*)&no_linger,
 401                              sizeof(no_linger));
 402     }
 403 #endif
 404
 405     return serverCon;
 406 }
 407 #endif /* OPENSSL_NO_SOCK */