|  | /* | 
|  | * Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved. | 
|  | * | 
|  | * Licensed under the OpenSSL license (the "License").  You may not use | 
|  | * this file except in compliance with the License.  You can obtain a copy | 
|  | * in the file LICENSE in the source distribution or at | 
|  | * https://www.openssl.org/source/license.html | 
|  | */ | 
|  |  | 
|  | #define NO_SHUTDOWN | 
|  |  | 
|  | #include <stdio.h> | 
|  | #include <stdlib.h> | 
|  | #include <string.h> | 
|  |  | 
|  | #include <openssl/opensslconf.h> | 
|  |  | 
|  | #ifndef OPENSSL_NO_SOCK | 
|  |  | 
|  | #define USE_SOCKETS | 
|  | #include "apps.h" | 
|  | #include <openssl/x509.h> | 
|  | #include <openssl/ssl.h> | 
|  | #include <openssl/pem.h> | 
|  | #include "s_apps.h" | 
|  | #include <openssl/err.h> | 
|  | #if !defined(OPENSSL_SYS_MSDOS) | 
|  | # include OPENSSL_UNISTD | 
|  | #endif | 
|  |  | 
|  | #undef ioctl | 
|  | #define ioctl ioctlsocket | 
|  |  | 
|  | #define SSL_CONNECT_NAME        "localhost:4433" | 
|  |  | 
|  | /* no default cert. */ | 
|  | /* | 
|  | * #define TEST_CERT "client.pem" | 
|  | */ | 
|  |  | 
|  | #undef min | 
|  | #undef max | 
|  | #define min(a,b) (((a) < (b)) ? (a) : (b)) | 
|  | #define max(a,b) (((a) > (b)) ? (a) : (b)) | 
|  |  | 
|  | #undef SECONDS | 
|  | #define SECONDS 30 | 
|  | #define SECONDSSTR "30" | 
|  |  | 
|  | static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx); | 
|  |  | 
|  | static const char fmt_http_get_cmd[] = "GET %s HTTP/1.0\r\n\r\n"; | 
|  |  | 
|  | typedef enum OPTION_choice { | 
|  | OPT_ERR = -1, OPT_EOF = 0, OPT_HELP, | 
|  | OPT_CONNECT, OPT_CIPHER, OPT_CERT, OPT_KEY, OPT_CAPATH, | 
|  | OPT_CAFILE, OPT_NOCAPATH, OPT_NOCAFILE, OPT_NEW, OPT_REUSE, OPT_BUGS, | 
|  | OPT_VERIFY, OPT_TIME, OPT_SSL3, | 
|  | OPT_WWW | 
|  | } OPTION_CHOICE; | 
|  |  | 
|  | const OPTIONS s_time_options[] = { | 
|  | {"help", OPT_HELP, '-', "Display this summary"}, | 
|  | {"connect", OPT_CONNECT, 's', | 
|  | "Where to connect as post:port (default is " SSL_CONNECT_NAME ")"}, | 
|  | {"cipher", OPT_CIPHER, 's', "Cipher to use, see 'openssl ciphers'"}, | 
|  | {"cert", OPT_CERT, '<', "Cert file to use, PEM format assumed"}, | 
|  | {"key", OPT_KEY, '<', "File with key, PEM; default is -cert file"}, | 
|  | {"CApath", OPT_CAPATH, '/', "PEM format directory of CA's"}, | 
|  | {"cafile", OPT_CAFILE, '<', "PEM format file of CA's"}, | 
|  | {"no-CAfile", OPT_NOCAFILE, '-', | 
|  | "Do not load the default certificates file"}, | 
|  | {"no-CApath", OPT_NOCAPATH, '-', | 
|  | "Do not load certificates from the default certificates directory"}, | 
|  | {"new", OPT_NEW, '-', "Just time new connections"}, | 
|  | {"reuse", OPT_REUSE, '-', "Just time connection reuse"}, | 
|  | {"bugs", OPT_BUGS, '-', "Turn on SSL bug compatibility"}, | 
|  | {"verify", OPT_VERIFY, 'p', | 
|  | "Turn on peer certificate verification, set depth"}, | 
|  | {"time", OPT_TIME, 'p', "Seconds to collect data, default " SECONDSSTR}, | 
|  | {"www", OPT_WWW, 's', "Fetch specified page from the site"}, | 
|  | #ifndef OPENSSL_NO_SSL3 | 
|  | {"ssl3", OPT_SSL3, '-', "Just use SSLv3"}, | 
|  | #endif | 
|  | {NULL} | 
|  | }; | 
|  |  | 
|  | #define START   0 | 
|  | #define STOP    1 | 
|  |  | 
|  | static double tm_Time_F(int s) | 
|  | { | 
|  | return app_tminterval(s, 1); | 
|  | } | 
|  |  | 
|  | int s_time_main(int argc, char **argv) | 
|  | { | 
|  | char buf[1024 * 8]; | 
|  | SSL *scon = NULL; | 
|  | SSL_CTX *ctx = NULL; | 
|  | const SSL_METHOD *meth = NULL; | 
|  | char *CApath = NULL, *CAfile = NULL, *cipher = NULL, *www_path = NULL; | 
|  | char *host = SSL_CONNECT_NAME, *certfile = NULL, *keyfile = NULL, *prog; | 
|  | double totalTime = 0.0; | 
|  | int noCApath = 0, noCAfile = 0; | 
|  | int maxtime = SECONDS, nConn = 0, perform = 3, ret = 1, i, st_bugs = 0; | 
|  | long bytes_read = 0, finishtime = 0; | 
|  | OPTION_CHOICE o; | 
|  | int max_version = 0, ver, buf_len; | 
|  | size_t buf_size; | 
|  |  | 
|  | meth = TLS_client_method(); | 
|  |  | 
|  | prog = opt_init(argc, argv, s_time_options); | 
|  | while ((o = opt_next()) != OPT_EOF) { | 
|  | switch (o) { | 
|  | case OPT_EOF: | 
|  | case OPT_ERR: | 
|  | opthelp: | 
|  | BIO_printf(bio_err, "%s: Use -help for summary.\n", prog); | 
|  | goto end; | 
|  | case OPT_HELP: | 
|  | opt_help(s_time_options); | 
|  | ret = 0; | 
|  | goto end; | 
|  | case OPT_CONNECT: | 
|  | host = opt_arg(); | 
|  | break; | 
|  | case OPT_REUSE: | 
|  | perform = 2; | 
|  | break; | 
|  | case OPT_NEW: | 
|  | perform = 1; | 
|  | break; | 
|  | case OPT_VERIFY: | 
|  | if (!opt_int(opt_arg(), &verify_args.depth)) | 
|  | goto opthelp; | 
|  | BIO_printf(bio_err, "%s: verify depth is %d\n", | 
|  | prog, verify_args.depth); | 
|  | break; | 
|  | case OPT_CERT: | 
|  | certfile = opt_arg(); | 
|  | break; | 
|  | case OPT_KEY: | 
|  | keyfile = opt_arg(); | 
|  | break; | 
|  | case OPT_CAPATH: | 
|  | CApath = opt_arg(); | 
|  | break; | 
|  | case OPT_CAFILE: | 
|  | CAfile = opt_arg(); | 
|  | break; | 
|  | case OPT_NOCAPATH: | 
|  | noCApath = 1; | 
|  | break; | 
|  | case OPT_NOCAFILE: | 
|  | noCAfile = 1; | 
|  | break; | 
|  | case OPT_CIPHER: | 
|  | cipher = opt_arg(); | 
|  | break; | 
|  | case OPT_BUGS: | 
|  | st_bugs = 1; | 
|  | break; | 
|  | case OPT_TIME: | 
|  | if (!opt_int(opt_arg(), &maxtime)) | 
|  | goto opthelp; | 
|  | break; | 
|  | case OPT_WWW: | 
|  | www_path = opt_arg(); | 
|  | buf_size = strlen(www_path) + sizeof(fmt_http_get_cmd) - 2;  /* 2 is for %s */ | 
|  | if (buf_size > sizeof(buf)) { | 
|  | BIO_printf(bio_err, "%s: -www option is too long\n", prog); | 
|  | goto end; | 
|  | } | 
|  | break; | 
|  | case OPT_SSL3: | 
|  | max_version = SSL3_VERSION; | 
|  | break; | 
|  | } | 
|  | } | 
|  | argc = opt_num_rest(); | 
|  | if (argc != 0) | 
|  | goto opthelp; | 
|  |  | 
|  | if (cipher == NULL) | 
|  | cipher = getenv("SSL_CIPHER"); | 
|  | if (cipher == NULL) { | 
|  | BIO_printf(bio_err, "No CIPHER specified\n"); | 
|  | goto end; | 
|  | } | 
|  |  | 
|  | if ((ctx = SSL_CTX_new(meth)) == NULL) | 
|  | goto end; | 
|  |  | 
|  | SSL_CTX_set_quiet_shutdown(ctx, 1); | 
|  | if (SSL_CTX_set_max_proto_version(ctx, max_version) == 0) | 
|  | goto end; | 
|  |  | 
|  | if (st_bugs) | 
|  | SSL_CTX_set_options(ctx, SSL_OP_ALL); | 
|  | if (!SSL_CTX_set_cipher_list(ctx, cipher)) | 
|  | goto end; | 
|  | if (!set_cert_stuff(ctx, certfile, keyfile)) | 
|  | goto end; | 
|  |  | 
|  | if (!ctx_set_verify_locations(ctx, CAfile, CApath, noCAfile, noCApath)) { | 
|  | ERR_print_errors(bio_err); | 
|  | goto end; | 
|  | } | 
|  | if (!(perform & 1)) | 
|  | goto next; | 
|  | printf("Collecting connection statistics for %d seconds\n", maxtime); | 
|  |  | 
|  | /* Loop and time how long it takes to make connections */ | 
|  |  | 
|  | bytes_read = 0; | 
|  | finishtime = (long)time(NULL) + maxtime; | 
|  | tm_Time_F(START); | 
|  | for (;;) { | 
|  | if (finishtime < (long)time(NULL)) | 
|  | break; | 
|  |  | 
|  | if ((scon = doConnection(NULL, host, ctx)) == NULL) | 
|  | goto end; | 
|  |  | 
|  | if (www_path != NULL) { | 
|  | buf_len = BIO_snprintf(buf, sizeof buf, | 
|  | fmt_http_get_cmd, www_path); | 
|  | if (SSL_write(scon, buf, buf_len) <= 0) | 
|  | goto end; | 
|  | while ((i = SSL_read(scon, buf, sizeof(buf))) > 0) | 
|  | bytes_read += i; | 
|  | } | 
|  | #ifdef NO_SHUTDOWN | 
|  | SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN); | 
|  | #else | 
|  | SSL_shutdown(scon); | 
|  | #endif | 
|  | BIO_closesocket(SSL_get_fd(scon)); | 
|  |  | 
|  | nConn += 1; | 
|  | if (SSL_session_reused(scon)) | 
|  | ver = 'r'; | 
|  | else { | 
|  | ver = SSL_version(scon); | 
|  | if (ver == TLS1_VERSION) | 
|  | ver = 't'; | 
|  | else if (ver == SSL3_VERSION) | 
|  | ver = '3'; | 
|  | else | 
|  | ver = '*'; | 
|  | } | 
|  | fputc(ver, stdout); | 
|  | fflush(stdout); | 
|  |  | 
|  | SSL_free(scon); | 
|  | scon = NULL; | 
|  | } | 
|  | totalTime += tm_Time_F(STOP); /* Add the time for this iteration */ | 
|  |  | 
|  | i = (int)((long)time(NULL) - finishtime + maxtime); | 
|  | printf | 
|  | ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", | 
|  | nConn, totalTime, ((double)nConn / totalTime), bytes_read); | 
|  | printf | 
|  | ("%d connections in %ld real seconds, %ld bytes read per connection\n", | 
|  | nConn, (long)time(NULL) - finishtime + maxtime, bytes_read / nConn); | 
|  |  | 
|  | /* | 
|  | * Now loop and time connections using the same session id over and over | 
|  | */ | 
|  |  | 
|  | next: | 
|  | if (!(perform & 2)) | 
|  | goto end; | 
|  | printf("\n\nNow timing with session id reuse.\n"); | 
|  |  | 
|  | /* Get an SSL object so we can reuse the session id */ | 
|  | if ((scon = doConnection(NULL, host, ctx)) == NULL) { | 
|  | BIO_printf(bio_err, "Unable to get connection\n"); | 
|  | goto end; | 
|  | } | 
|  |  | 
|  | if (www_path != NULL) { | 
|  | buf_len = BIO_snprintf(buf, sizeof buf, | 
|  | fmt_http_get_cmd, www_path); | 
|  | if (SSL_write(scon, buf, buf_len) <= 0) | 
|  | goto end; | 
|  | while (SSL_read(scon, buf, sizeof(buf)) > 0) | 
|  | continue; | 
|  | } | 
|  | #ifdef NO_SHUTDOWN | 
|  | SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN); | 
|  | #else | 
|  | SSL_shutdown(scon); | 
|  | #endif | 
|  | BIO_closesocket(SSL_get_fd(scon)); | 
|  |  | 
|  | nConn = 0; | 
|  | totalTime = 0.0; | 
|  |  | 
|  | finishtime = (long)time(NULL) + maxtime; | 
|  |  | 
|  | printf("starting\n"); | 
|  | bytes_read = 0; | 
|  | tm_Time_F(START); | 
|  |  | 
|  | for (;;) { | 
|  | if (finishtime < (long)time(NULL)) | 
|  | break; | 
|  |  | 
|  | if ((doConnection(scon, host, ctx)) == NULL) | 
|  | goto end; | 
|  |  | 
|  | if (www_path) { | 
|  | BIO_snprintf(buf, sizeof buf, "GET %s HTTP/1.0\r\n\r\n", | 
|  | www_path); | 
|  | if (SSL_write(scon, buf, strlen(buf)) <= 0) | 
|  | goto end; | 
|  | while ((i = SSL_read(scon, buf, sizeof(buf))) > 0) | 
|  | bytes_read += i; | 
|  | } | 
|  | #ifdef NO_SHUTDOWN | 
|  | SSL_set_shutdown(scon, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN); | 
|  | #else | 
|  | SSL_shutdown(scon); | 
|  | #endif | 
|  | BIO_closesocket(SSL_get_fd(scon)); | 
|  |  | 
|  | nConn += 1; | 
|  | if (SSL_session_reused(scon)) | 
|  | ver = 'r'; | 
|  | else { | 
|  | ver = SSL_version(scon); | 
|  | if (ver == TLS1_VERSION) | 
|  | ver = 't'; | 
|  | else if (ver == SSL3_VERSION) | 
|  | ver = '3'; | 
|  | else | 
|  | ver = '*'; | 
|  | } | 
|  | fputc(ver, stdout); | 
|  | fflush(stdout); | 
|  | } | 
|  | totalTime += tm_Time_F(STOP); /* Add the time for this iteration */ | 
|  |  | 
|  | printf | 
|  | ("\n\n%d connections in %.2fs; %.2f connections/user sec, bytes read %ld\n", | 
|  | nConn, totalTime, ((double)nConn / totalTime), bytes_read); | 
|  | printf | 
|  | ("%d connections in %ld real seconds, %ld bytes read per connection\n", | 
|  | nConn, (long)time(NULL) - finishtime + maxtime, bytes_read / nConn); | 
|  |  | 
|  | ret = 0; | 
|  |  | 
|  | end: | 
|  | SSL_free(scon); | 
|  | SSL_CTX_free(ctx); | 
|  | return (ret); | 
|  | } | 
|  |  | 
|  | /*- | 
|  | * doConnection - make a connection | 
|  | */ | 
|  | static SSL *doConnection(SSL *scon, const char *host, SSL_CTX *ctx) | 
|  | { | 
|  | BIO *conn; | 
|  | SSL *serverCon; | 
|  | int width, i; | 
|  | fd_set readfds; | 
|  |  | 
|  | if ((conn = BIO_new(BIO_s_connect())) == NULL) | 
|  | return (NULL); | 
|  |  | 
|  | BIO_set_conn_hostname(conn, host); | 
|  |  | 
|  | if (scon == NULL) | 
|  | serverCon = SSL_new(ctx); | 
|  | else { | 
|  | serverCon = scon; | 
|  | SSL_set_connect_state(serverCon); | 
|  | } | 
|  |  | 
|  | SSL_set_bio(serverCon, conn, conn); | 
|  |  | 
|  | /* ok, lets connect */ | 
|  | for (;;) { | 
|  | i = SSL_connect(serverCon); | 
|  | if (BIO_sock_should_retry(i)) { | 
|  | BIO_printf(bio_err, "DELAY\n"); | 
|  |  | 
|  | i = SSL_get_fd(serverCon); | 
|  | width = i + 1; | 
|  | FD_ZERO(&readfds); | 
|  | openssl_fdset(i, &readfds); | 
|  | /* | 
|  | * Note: under VMS with SOCKETSHR the 2nd parameter is currently | 
|  | * of type (int *) whereas under other systems it is (void *) if | 
|  | * you don't have a cast it will choke the compiler: if you do | 
|  | * have a cast then you can either go for (int *) or (void *). | 
|  | */ | 
|  | select(width, (void *)&readfds, NULL, NULL, NULL); | 
|  | continue; | 
|  | } | 
|  | break; | 
|  | } | 
|  | if (i <= 0) { | 
|  | BIO_printf(bio_err, "ERROR\n"); | 
|  | if (verify_args.error != X509_V_OK) | 
|  | BIO_printf(bio_err, "verify error:%s\n", | 
|  | X509_verify_cert_error_string(verify_args.error)); | 
|  | else | 
|  | ERR_print_errors(bio_err); | 
|  | if (scon == NULL) | 
|  | SSL_free(serverCon); | 
|  | return NULL; | 
|  | } | 
|  |  | 
|  | return serverCon; | 
|  | } | 
|  | #endif /* OPENSSL_NO_SOCK */ |