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