granicus.if.org Git - apache/blob - support/ab.c

   1 /* Licensed to the Apache Software Foundation (ASF) under one or more
   2  * contributor license agreements.  See the NOTICE file distributed with
   3  * this work for additional information regarding copyright ownership.
   4  * The ASF licenses this file to You under the Apache License, Version 2.0
   5  * (the "License"); you may not use this file except in compliance with
   6  * the License.  You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 /*
  18    ** This program is based on ZeusBench V1.0 written by Adam Twiss
  19    ** which is Copyright (c) 1996 by Zeus Technology Ltd. http://www.zeustech.net/
  20    **
  21    ** This software is provided "as is" and any express or implied waranties,
  22    ** including but not limited to, the implied warranties of merchantability and
  23    ** fitness for a particular purpose are disclaimed.  In no event shall
  24    ** Zeus Technology Ltd. be liable for any direct, indirect, incidental, special,
  25    ** exemplary, or consequential damaged (including, but not limited to,
  26    ** procurement of substitute good or services; loss of use, data, or profits;
  27    ** or business interruption) however caused and on theory of liability.  Whether
  28    ** in contract, strict liability or tort (including negligence or otherwise)
  29    ** arising in any way out of the use of this software, even if advised of the
  30    ** possibility of such damage.
  31    **
  32  */
  33
  34 /*
  35    ** HISTORY:
  36    **    - Originally written by Adam Twiss <adam@zeus.co.uk>, March 1996
  37    **      with input from Mike Belshe <mbelshe@netscape.com> and
  38    **      Michael Campanella <campanella@stevms.enet.dec.com>
  39    **    - Enhanced by Dean Gaudet <dgaudet@apache.org>, November 1997
  40    **    - Cleaned up by Ralf S. Engelschall <rse@apache.org>, March 1998
  41    **    - POST and verbosity by Kurt Sussman <kls@merlot.com>, August 1998
  42    **    - HTML table output added by David N. Welton <davidw@prosa.it>, January 1999
  43    **    - Added Cookie, Arbitrary header and auth support. <dirkx@webweaving.org>, April 1999
  44    ** Version 1.3d
  45    **    - Increased version number - as some of the socket/error handling has
  46    **      fundamentally changed - and will give fundamentally different results
  47    **      in situations where a server is dropping requests. Therefore you can
  48    **      no longer compare results of AB as easily. Hence the inc of the version.
  49    **      They should be closer to the truth though. Sander & <dirkx@covalent.net>, End 2000.
  50    **    - Fixed proxy functionality, added median/mean statistics, added gnuplot
  51    **      output option, added _experimental/rudimentary_ SSL support. Added
  52    **      confidence guestimators and warnings. Sander & <dirkx@covalent.net>, End 2000
  53    **    - Fixed serious int overflow issues which would cause realistic (longer
  54    **      than a few minutes) run's to have wrong (but believable) results. Added
  55    **      trapping of connection errors which influenced measurements.
  56    **      Contributed by Sander Temme, Early 2001
  57    ** Version 1.3e
  58    **    - Changed timeout behavour during write to work whilst the sockets
  59    **      are filling up and apr_write() does writes a few - but not all.
  60    **      This will potentially change results. <dirkx@webweaving.org>, April 2001
  61    ** Version 2.0.36-dev
  62    **    Improvements to concurrent processing:
  63    **      - Enabled non-blocking connect()s.
  64    **      - Prevent blocking calls to apr_socket_recv() (thereby allowing AB to
  65    **        manage its entire set of socket descriptors).
  66    **      - Any error returned from apr_socket_recv() that is not EAGAIN or EOF
  67    **        is now treated as fatal.
  68    **      Contributed by Aaron Bannert, April 24, 2002
  69    **
  70    ** Version 2.0.36-2
  71    **     Internalized the version string - this string is part
  72    **     of the Agent: header and the result output.
  73    **
  74    ** Version 2.0.37-dev
  75    **     Adopted SSL code by Madhu Mathihalli <madhusudan_mathihalli@hp.com>
  76    **     [PATCH] ab with SSL support  Posted Wed, 15 Aug 2001 20:55:06 GMT
  77    **     Introduces four 'if (int == value)' tests per non-ssl request.
  78    **
  79    ** Version 2.0.40-dev
  80    **     Switched to the new abstract pollset API, allowing ab to
  81    **     take advantage of future apr_pollset_t scalability improvements.
  82    **     Contributed by Brian Pane, August 31, 2002
  83    **
  84    ** Version 2.3
  85    **     SIGINT now triggers output_results().
  86    **     Contributed by colm, March 30, 2006
  87    **/
  88
  89 /* Note: this version string should start with \d+[\d\.]* and be a valid
  90  * string for an HTTP Agent: header when prefixed with 'ApacheBench/'.
  91  * It should reflect the version of AB - and not that of the apache server
  92  * it happens to accompany. And it should be updated or changed whenever
  93  * the results are no longer fundamentally comparable to the results of
  94  * a previous version of ab. Either due to a change in the logic of
  95  * ab - or to due to a change in the distribution it is compiled with
  96  * (such as an APR change in for example blocking).
  97  */
  98 #define AP_AB_BASEREVISION "2.3"
  99
 100 /*
 101  * BUGS:
 102  *
 103  * - uses strcpy/etc.
 104  * - has various other poor buffer attacks related to the lazy parsing of
 105  *   response headers from the server
 106  * - doesn't implement much of HTTP/1.x, only accepts certain forms of
 107  *   responses
 108  * - (performance problem) heavy use of strstr shows up top in profile
 109  *   only an issue for loopback usage
 110  */
 111
 112 /*  -------------------------------------------------------------------- */
 113
 114 #if 'A' != 0x41
 115 /* Hmmm... This source code isn't being compiled in ASCII.
 116  * In order for data that flows over the network to make
 117  * sense, we need to translate to/from ASCII.
 118  */
 119 #define NOT_ASCII
 120 #endif
 121
 122 /* affects include files on Solaris */
 123 #define BSD_COMP
 124
 125 #include "apr.h"
 126 #include "apr_signal.h"
 127 #include "apr_strings.h"
 128 #include "apr_network_io.h"
 129 #include "apr_file_io.h"
 130 #include "apr_time.h"
 131 #include "apr_getopt.h"
 132 #include "apr_general.h"
 133 #include "apr_lib.h"
 134 #include "apr_portable.h"
 135 #include "ap_release.h"
 136 #include "apr_poll.h"
 137
 138 #define APR_WANT_STRFUNC
 139 #include "apr_want.h"
 140
 141 #include "apr_base64.h"
 142 #ifdef NOT_ASCII
 143 #include "apr_xlate.h"
 144 #endif
 145 #if APR_HAVE_STDIO_H
 146 #include <stdio.h>
 147 #endif
 148 #if APR_HAVE_STDLIB_H
 149 #include <stdlib.h>
 150 #endif
 151 #if APR_HAVE_UNISTD_H
 152 #include <unistd.h> /* for getpid() */
 153 #endif
 154
 155 #if !defined(WIN32) && !defined(NETWARE)
 156 #include "ap_config_auto.h"
 157 #endif
 158
 159 #if defined(HAVE_SSLC)
 160
 161 /* Libraries for RSA SSL-C */
 162 #include <rsa.h>
 163 #include <x509.h>
 164 #include <pem.h>
 165 #include <err.h>
 166 #include <ssl.h>
 167 #include <r_rand.h>
 168 #include <sslc.h>
 169 #define USE_SSL
 170 #define RSAREF
 171 #define SK_NUM(x) sk_num(x)
 172 #define SK_VALUE(x,y) sk_value(x,y)
 173 typedef STACK X509_STACK_TYPE;
 174
 175 #elif defined(HAVE_OPENSSL)
 176
 177 /* Libraries on most systems.. */
 178 #include <openssl/rsa.h>
 179 #include <openssl/crypto.h>
 180 #include <openssl/x509.h>
 181 #include <openssl/pem.h>
 182 #include <openssl/err.h>
 183 #include <openssl/ssl.h>
 184 #include <openssl/rand.h>
 185 #define USE_SSL
 186 #define SK_NUM(x) sk_X509_num(x)
 187 #define SK_VALUE(x,y) sk_X509_value(x,y)
 188 typedef STACK_OF(X509) X509_STACK_TYPE;
 189
 190 #endif
 191
 192 #if defined(USE_SSL)
 193 #if (OPENSSL_VERSION_NUMBER >= 0x00909000)
 194 #define AB_SSL_METHOD_CONST const
 195 #else
 196 #define AB_SSL_METHOD_CONST
 197 #endif
 198 #if (OPENSSL_VERSION_NUMBER >= 0x0090707f)
 199 #define AB_SSL_CIPHER_CONST const
 200 #else
 201 #define AB_SSL_CIPHER_CONST
 202 #endif
 203 #endif
 204
 205 #include <math.h>
 206 #if APR_HAVE_CTYPE_H
 207 #include <ctype.h>
 208 #endif
 209 #if APR_HAVE_LIMITS_H
 210 #include <limits.h>
 211 #endif
 212
 213 /* ------------------- DEFINITIONS -------------------------- */
 214
 215 #ifndef LLONG_MAX
 216 #define AB_MAX APR_INT64_C(0x7fffffffffffffff)
 217 #else
 218 #define AB_MAX LLONG_MAX
 219 #endif
 220
 221 /* maximum number of requests on a time limited test */
 222 #define MAX_REQUESTS (INT_MAX > 50000 ? 50000 : INT_MAX)
 223
 224 /* connection state
 225  * don't add enums or rearrange or otherwise change values without
 226  * visiting set_conn_state()
 227  */
 228 typedef enum {
 229     STATE_UNCONNECTED = 0,
 230     STATE_CONNECTING,           /* TCP connect initiated, but we don't
 231                                  * know if it worked yet
 232                                  */
 233     STATE_CONNECTED,            /* we know TCP connect completed */
 234     STATE_READ
 235 } connect_state_e;
 236
 237 #define CBUFFSIZE (2048)
 238
 239 struct connection {
 240     apr_pool_t *ctx;
 241     apr_socket_t *aprsock;
 242     apr_pollfd_t pollfd;
 243     int state;
 244     apr_size_t read;            /* amount of bytes read */
 245     apr_size_t bread;           /* amount of body read */
 246     apr_size_t rwrite, rwrote;  /* keep pointers in what we write - across
 247                                  * EAGAINs */
 248     apr_size_t length;          /* Content-Length value used for keep-alive */
 249     char cbuff[CBUFFSIZE];      /* a buffer to store server response header */
 250     int cbx;                    /* offset in cbuffer */
 251     int keepalive;              /* non-zero if a keep-alive request */
 252     int gotheader;              /* non-zero if we have the entire header in
 253                                  * cbuff */
 254     apr_time_t start,           /* Start of connection */
 255                connect,         /* Connected, start writing */
 256                endwrite,        /* Request written */
 257                beginread,       /* First byte of input */
 258                done;            /* Connection closed */
 259
 260     int socknum;
 261 #ifdef USE_SSL
 262     SSL *ssl;
 263 #endif
 264 };
 265
 266 struct data {
 267     apr_time_t starttime;         /* start time of connection */
 268     apr_interval_time_t waittime; /* between request and reading response */
 269     apr_interval_time_t ctime;    /* time to connect */
 270     apr_interval_time_t time;     /* time for connection */
 271 };
 272
 273 #define ap_min(a,b) ((a)<(b))?(a):(b)
 274 #define ap_max(a,b) ((a)>(b))?(a):(b)
 275 #define ap_round_ms(a) ((apr_time_t)((a) + 500)/1000)
 276 #define ap_double_ms(a) ((double)(a)/1000.0)
 277 #define MAX_CONCURRENCY 20000
 278
 279 /* --------------------- GLOBALS ---------------------------- */
 280
 281 int verbosity = 0;      /* no verbosity by default */
 282 int recverrok = 0;      /* ok to proceed after socket receive errors */
 283 enum {NO_METH = 0, GET, HEAD, PUT, POST} method = NO_METH;
 284 const char *method_str[] = {"bug", "GET", "HEAD", "PUT", "POST"};
 285 const char *method_str_pasttense[] = {"bug", "bug", "bug", "PUT", "POSTed"};
 286 int send_body = 0;      /* non-zero if sending body with request */
 287 int requests = 1;       /* Number of requests to make */
 288 int heartbeatres = 100; /* How often do we say we're alive */
 289 int concurrency = 1;    /* Number of multiple requests to make */
 290 int percentile = 1;     /* Show percentile served */
 291 int confidence = 1;     /* Show confidence estimator and warnings */
 292 int tlimit = 0;         /* time limit in secs */
 293 int keepalive = 0;      /* try and do keepalive connections */
 294 int windowsize = 0;     /* we use the OS default window size */
 295 char servername[1024];  /* name that server reports */
 296 char *hostname;         /* host name from URL */
 297 char *host_field;       /* value of "Host:" header field */
 298 char *path;             /* path name */
 299 char postfile[1024];    /* name of file containing post data */
 300 char *postdata;         /* *buffer containing data from postfile */
 301 apr_size_t postlen = 0; /* length of data to be POSTed */
 302 char content_type[1024];/* content type to put in POST header */
 303 char *cookie,           /* optional cookie line */
 304      *auth,             /* optional (basic/uuencoded) auhentication */
 305      *hdrs;             /* optional arbitrary headers */
 306 apr_port_t port;        /* port number */
 307 char proxyhost[1024];   /* proxy host name */
 308 int proxyport = 0;      /* proxy port */
 309 char *connecthost;
 310 apr_port_t connectport;
 311 char *gnuplot;          /* GNUplot file */
 312 char *csvperc;          /* CSV Percentile file */
 313 char url[1024];
 314 char * fullurl, * colonhost;
 315 int isproxy = 0;
 316 apr_interval_time_t aprtimeout = apr_time_from_sec(30); /* timeout value */
 317
 318 /* overrides for ab-generated common headers */
 319 int opt_host = 0;       /* was an optional "Host:" header specified? */
 320 int opt_useragent = 0;  /* was an optional "User-Agent:" header specified? */
 321 int opt_accept = 0;     /* was an optional "Accept:" header specified? */
 322  /*
 323   * XXX - this is now a per read/write transact type of value
 324   */
 325
 326 int use_html = 0;       /* use html in the report */
 327 const char *tablestring;
 328 const char *trstring;
 329 const char *tdstring;
 330
 331 apr_size_t doclen = 0;     /* the length the document should be */
 332 apr_int64_t totalread = 0;    /* total number of bytes read */
 333 apr_int64_t totalbread = 0;   /* totoal amount of entity body read */
 334 apr_int64_t totalposted = 0;  /* total number of bytes posted, inc. headers */
 335 int started = 0;           /* number of requests started, so no excess */
 336 int done = 0;              /* number of requests we have done */
 337 int doneka = 0;            /* number of keep alive connections done */
 338 int good = 0, bad = 0;     /* number of good and bad requests */
 339 int epipe = 0;             /* number of broken pipe writes */
 340 int err_length = 0;        /* requests failed due to response length */
 341 int err_conn = 0;          /* requests failed due to connection drop */
 342 int err_recv = 0;          /* requests failed due to broken read */
 343 int err_except = 0;        /* requests failed due to exception */
 344 int err_response = 0;      /* requests with invalid or non-200 response */
 345
 346 #ifdef USE_SSL
 347 int is_ssl;
 348 SSL_CTX *ssl_ctx;
 349 char *ssl_cipher = NULL;
 350 char *ssl_info = NULL;
 351 BIO *bio_out,*bio_err;
 352 #endif
 353
 354 apr_time_t start, lasttime, stoptime;
 355
 356 /* global request (and its length) */
 357 char _request[2048];
 358 char *request = _request;
 359 apr_size_t reqlen;
 360
 361 /* one global throw-away buffer to read stuff into */
 362 char buffer[8192];
 363
 364 /* interesting percentiles */
 365 int percs[] = {50, 66, 75, 80, 90, 95, 98, 99, 100};
 366
 367 struct connection *con;     /* connection array */
 368 struct data *stats;         /* data for each request */
 369 apr_pool_t *cntxt;
 370
 371 apr_pollset_t *readbits;
 372
 373 apr_sockaddr_t *destsa;
 374
 375 #ifdef NOT_ASCII
 376 apr_xlate_t *from_ascii, *to_ascii;
 377 #endif
 378
 379 static void write_request(struct connection * c);
 380 static void close_connection(struct connection * c);
 381
 382 /* --------------------------------------------------------- */
 383
 384 /* simple little function to write an error string and exit */
 385
 386 static void err(char *s)
 387 {
 388     fprintf(stderr, "%s\n", s);
 389     if (done)
 390         printf("Total of %d requests completed\n" , done);
 391     exit(1);
 392 }
 393
 394 /* simple little function to write an APR error string and exit */
 395
 396 static void apr_err(char *s, apr_status_t rv)
 397 {
 398     char buf[120];
 399
 400     fprintf(stderr,
 401         "%s: %s (%d)\n",
 402         s, apr_strerror(rv, buf, sizeof buf), rv);
 403     if (done)
 404         printf("Total of %d requests completed\n" , done);
 405     exit(rv);
 406 }
 407
 408 static void set_polled_events(struct connection *c, apr_int16_t new_reqevents)
 409 {
 410     apr_status_t rv;
 411
 412     if (c->pollfd.reqevents != new_reqevents) {
 413         if (c->pollfd.reqevents != 0) {
 414             rv = apr_pollset_remove(readbits, &c->pollfd);
 415             if (rv != APR_SUCCESS) {
 416                 apr_err("apr_pollset_remove()", rv);
 417             }
 418         }
 419
 420         if (new_reqevents != 0) {
 421             c->pollfd.reqevents = new_reqevents;
 422             rv = apr_pollset_add(readbits, &c->pollfd);
 423             if (rv != APR_SUCCESS) {
 424                 apr_err("apr_pollset_add()", rv);
 425             }
 426         }
 427     }
 428 }
 429
 430 static void set_conn_state(struct connection *c, connect_state_e new_state)
 431 {
 432     apr_int16_t events_by_state[] = {
 433         0,           /* for STATE_UNCONNECTED */
 434         APR_POLLOUT, /* for STATE_CONNECTING */
 435         APR_POLLIN,  /* for STATE_CONNECTED; we don't poll in this state,
 436                       * so prepare for polling in the following state --
 437                       * STATE_READ
 438                       */
 439         APR_POLLIN   /* for STATE_READ */
 440     };
 441
 442     c->state = new_state;
 443
 444     set_polled_events(c, events_by_state[new_state]);
 445 }
 446
 447 /* --------------------------------------------------------- */
 448 /* write out request to a connection - assumes we can write
 449  * (small) request out in one go into our new socket buffer
 450  *
 451  */
 452 #ifdef USE_SSL
 453 static long ssl_print_cb(BIO *bio,int cmd,const char *argp,int argi,long argl,long ret)
 454 {
 455     BIO *out;
 456
 457     out=(BIO *)BIO_get_callback_arg(bio);
 458     if (out == NULL) return(ret);
 459
 460     if (cmd == (BIO_CB_READ|BIO_CB_RETURN)) {
 461         BIO_printf(out,"read from %p [%p] (%d bytes => %ld (0x%lX))\n",
 462                    bio, argp, argi, ret, ret);
 463         BIO_dump(out,(char *)argp,(int)ret);
 464         return(ret);
 465     }
 466     else if (cmd == (BIO_CB_WRITE|BIO_CB_RETURN)) {
 467         BIO_printf(out,"write to %p [%p] (%d bytes => %ld (0x%lX))\n",
 468                    bio, argp, argi, ret, ret);
 469         BIO_dump(out,(char *)argp,(int)ret);
 470     }
 471     return ret;
 472 }
 473
 474 static void ssl_state_cb(const SSL *s, int w, int r)
 475 {
 476     if (w & SSL_CB_ALERT) {
 477         BIO_printf(bio_err, "SSL/TLS Alert [%s] %s:%s\n",
 478                    (w & SSL_CB_READ ? "read" : "write"),
 479                    SSL_alert_type_string_long(r),
 480                    SSL_alert_desc_string_long(r));
 481     } else if (w & SSL_CB_LOOP) {
 482         BIO_printf(bio_err, "SSL/TLS State [%s] %s\n",
 483                    (SSL_in_connect_init((SSL*)s) ? "connect" : "-"),
 484                    SSL_state_string_long(s));
 485     } else if (w & (SSL_CB_HANDSHAKE_START|SSL_CB_HANDSHAKE_DONE)) {
 486         BIO_printf(bio_err, "SSL/TLS Handshake [%s] %s\n",
 487                    (w & SSL_CB_HANDSHAKE_START ? "Start" : "Done"),
 488                    SSL_state_string_long(s));
 489     }
 490 }
 491
 492 #ifndef RAND_MAX
 493 #define RAND_MAX INT_MAX
 494 #endif
 495
 496 static int ssl_rand_choosenum(int l, int h)
 497 {
 498     int i;
 499     char buf[50];
 500
 501     srand((unsigned int)time(NULL));
 502     apr_snprintf(buf, sizeof(buf), "%.0f",
 503                  (((double)(rand()%RAND_MAX)/RAND_MAX)*(h-l)));
 504     i = atoi(buf)+1;
 505     if (i < l) i = l;
 506     if (i > h) i = h;
 507     return i;
 508 }
 509
 510 static void ssl_rand_seed(void)
 511 {
 512     int nDone = 0;
 513     int n, l;
 514     time_t t;
 515     pid_t pid;
 516     unsigned char stackdata[256];
 517
 518     /*
 519      * seed in the current time (usually just 4 bytes)
 520      */
 521     t = time(NULL);
 522     l = sizeof(time_t);
 523     RAND_seed((unsigned char *)&t, l);
 524     nDone += l;
 525
 526     /*
 527      * seed in the current process id (usually just 4 bytes)
 528      */
 529     pid = getpid();
 530     l = sizeof(pid_t);
 531     RAND_seed((unsigned char *)&pid, l);
 532     nDone += l;
 533
 534     /*
 535      * seed in some current state of the run-time stack (128 bytes)
 536      */
 537     n = ssl_rand_choosenum(0, sizeof(stackdata)-128-1);
 538     RAND_seed(stackdata+n, 128);
 539     nDone += 128;
 540 }
 541
 542 static int ssl_print_connection_info(BIO *bio, SSL *ssl)
 543 {
 544     AB_SSL_CIPHER_CONST SSL_CIPHER *c;
 545     int alg_bits,bits;
 546
 547     c = SSL_get_current_cipher(ssl);
 548     BIO_printf(bio,"Cipher Suite Protocol   :%s\n", SSL_CIPHER_get_version(c));
 549     BIO_printf(bio,"Cipher Suite Name       :%s\n",SSL_CIPHER_get_name(c));
 550
 551     bits = SSL_CIPHER_get_bits(c,&alg_bits);
 552     BIO_printf(bio,"Cipher Suite Cipher Bits:%d (%d)\n",bits,alg_bits);
 553
 554     return(1);
 555 }
 556
 557 static void ssl_print_cert_info(BIO *bio, X509 *cert)
 558 {
 559     X509_NAME *dn;
 560     char buf[1024];
 561
 562     BIO_printf(bio, "Certificate version: %ld\n", X509_get_version(cert)+1);
 563     BIO_printf(bio,"Valid from: ");
 564     ASN1_UTCTIME_print(bio, X509_get_notBefore(cert));
 565     BIO_printf(bio,"\n");
 566
 567     BIO_printf(bio,"Valid to  : ");
 568     ASN1_UTCTIME_print(bio, X509_get_notAfter(cert));
 569     BIO_printf(bio,"\n");
 570
 571     BIO_printf(bio,"Public key is %d bits\n",
 572                EVP_PKEY_bits(X509_get_pubkey(cert)));
 573
 574     dn = X509_get_issuer_name(cert);
 575     X509_NAME_oneline(dn, buf, sizeof(buf));
 576     BIO_printf(bio,"The issuer name is %s\n", buf);
 577
 578     dn=X509_get_subject_name(cert);
 579     X509_NAME_oneline(dn, buf, sizeof(buf));
 580     BIO_printf(bio,"The subject name is %s\n", buf);
 581
 582     /* dump the extension list too */
 583     BIO_printf(bio, "Extension Count: %d\n", X509_get_ext_count(cert));
 584 }
 585
 586 static void ssl_print_info(struct connection *c)
 587 {
 588     X509_STACK_TYPE *sk;
 589     X509 *cert;
 590     int count;
 591
 592     BIO_printf(bio_err, "\n");
 593     sk = SSL_get_peer_cert_chain(c->ssl);
 594     if ((count = SK_NUM(sk)) > 0) {
 595         int i;
 596         for (i=1; i<count; i++) {
 597             cert = (X509 *)SK_VALUE(sk, i);
 598             ssl_print_cert_info(bio_out, cert);
 599     }
 600     }
 601     cert = SSL_get_peer_certificate(c->ssl);
 602     if (cert == NULL) {
 603         BIO_printf(bio_out, "Anon DH\n");
 604     } else {
 605         BIO_printf(bio_out, "Peer certificate\n");
 606         ssl_print_cert_info(bio_out, cert);
 607         X509_free(cert);
 608     }
 609     ssl_print_connection_info(bio_err,c->ssl);
 610     SSL_SESSION_print(bio_err, SSL_get_session(c->ssl));
 611     }
 612
 613 static void ssl_proceed_handshake(struct connection *c)
 614 {
 615     int do_next = 1;
 616
 617     while (do_next) {
 618         int ret, ecode;
 619
 620         ret = SSL_do_handshake(c->ssl);
 621         ecode = SSL_get_error(c->ssl, ret);
 622
 623         switch (ecode) {
 624         case SSL_ERROR_NONE:
 625             if (verbosity >= 2)
 626                 ssl_print_info(c);
 627             if (ssl_info == NULL) {
 628                 AB_SSL_CIPHER_CONST SSL_CIPHER *ci;
 629                 X509 *cert;
 630                 int sk_bits, pk_bits, swork;
 631
 632                 ci = SSL_get_current_cipher(c->ssl);
 633                 sk_bits = SSL_CIPHER_get_bits(ci, &swork);
 634                 cert = SSL_get_peer_certificate(c->ssl);
 635                 if (cert)
 636                     pk_bits = EVP_PKEY_bits(X509_get_pubkey(cert));
 637                 else
 638                     pk_bits = 0;  /* Anon DH */
 639
 640                 ssl_info = malloc(128);
 641                 apr_snprintf(ssl_info, 128, "%s,%s,%d,%d",
 642                              SSL_CIPHER_get_version(ci),
 643                              SSL_CIPHER_get_name(ci),
 644                              pk_bits, sk_bits);
 645             }
 646             write_request(c);
 647             do_next = 0;
 648             break;
 649         case SSL_ERROR_WANT_READ:
 650             set_polled_events(c, APR_POLLIN);
 651             do_next = 0;
 652             break;
 653         case SSL_ERROR_WANT_WRITE:
 654             /* Try again */
 655             do_next = 1;
 656             break;
 657         case SSL_ERROR_WANT_CONNECT:
 658         case SSL_ERROR_SSL:
 659         case SSL_ERROR_SYSCALL:
 660             /* Unexpected result */
 661             BIO_printf(bio_err, "SSL handshake failed (%d).\n", ecode);
 662             ERR_print_errors(bio_err);
 663             close_connection(c);
 664             do_next = 0;
 665             break;
 666         }
 667     }
 668 }
 669
 670 #endif /* USE_SSL */
 671
 672 static void write_request(struct connection * c)
 673 {
 674     do {
 675         apr_time_t tnow;
 676         apr_size_t l = c->rwrite;
 677         apr_status_t e = APR_SUCCESS; /* prevent gcc warning */
 678
 679         tnow = lasttime = apr_time_now();
 680
 681         /*
 682          * First time round ?
 683          */
 684         if (c->rwrite == 0) {
 685             apr_socket_timeout_set(c->aprsock, 0);
 686             c->connect = tnow;
 687             c->rwrote = 0;
 688             c->rwrite = reqlen;
 689             if (send_body)
 690                 c->rwrite += postlen;
 691         }
 692         else if (tnow > c->connect + aprtimeout) {
 693             printf("Send request timed out!\n");
 694             close_connection(c);
 695             return;
 696         }
 697
 698 #ifdef USE_SSL
 699         if (c->ssl) {
 700             apr_size_t e_ssl;
 701             e_ssl = SSL_write(c->ssl,request + c->rwrote, l);
 702             if (e_ssl != l) {
 703                 BIO_printf(bio_err, "SSL write failed - closing connection\n");
 704                 ERR_print_errors(bio_err);
 705                 close_connection (c);
 706                 return;
 707             }
 708             l = e_ssl;
 709             e = APR_SUCCESS;
 710         }
 711         else
 712 #endif
 713             e = apr_socket_send(c->aprsock, request + c->rwrote, &l);
 714
 715         if (e != APR_SUCCESS && !APR_STATUS_IS_EAGAIN(e)) {
 716             epipe++;
 717             printf("Send request failed!\n");
 718             close_connection(c);
 719             return;
 720         }
 721         totalposted += l;
 722         c->rwrote += l;
 723         c->rwrite -= l;
 724     } while (c->rwrite);
 725
 726     c->endwrite = lasttime = apr_time_now();
 727     set_conn_state(c, STATE_READ);
 728 }
 729
 730 /* --------------------------------------------------------- */
 731
 732 /* calculate and output results */
 733
 734 static int compradre(struct data * a, struct data * b)
 735 {
 736     if ((a->ctime) < (b->ctime))
 737         return -1;
 738     if ((a->ctime) > (b->ctime))
 739         return +1;
 740     return 0;
 741 }
 742
 743 static int comprando(struct data * a, struct data * b)
 744 {
 745     if ((a->time) < (b->time))
 746         return -1;
 747     if ((a->time) > (b->time))
 748         return +1;
 749     return 0;
 750 }
 751
 752 static int compri(struct data * a, struct data * b)
 753 {
 754     apr_interval_time_t p = a->time - a->ctime;
 755     apr_interval_time_t q = b->time - b->ctime;
 756     if (p < q)
 757         return -1;
 758     if (p > q)
 759         return +1;
 760     return 0;
 761 }
 762
 763 static int compwait(struct data * a, struct data * b)
 764 {
 765     if ((a->waittime) < (b->waittime))
 766         return -1;
 767     if ((a->waittime) > (b->waittime))
 768         return 1;
 769     return 0;
 770 }
 771
 772 static void output_results(int sig)
 773 {
 774     double timetaken;
 775
 776     if (sig) {
 777         lasttime = apr_time_now();  /* record final time if interrupted */
 778     }
 779     timetaken = (double) (lasttime - start) / APR_USEC_PER_SEC;
 780
 781     printf("\n\n");
 782     printf("Server Software:        %s\n", servername);
 783     printf("Server Hostname:        %s\n", hostname);
 784     printf("Server Port:            %hu\n", port);
 785 #ifdef USE_SSL
 786     if (is_ssl && ssl_info) {
 787         printf("SSL/TLS Protocol:       %s\n", ssl_info);
 788     }
 789 #endif
 790     printf("\n");
 791     printf("Document Path:          %s\n", path);
 792     printf("Document Length:        %" APR_SIZE_T_FMT " bytes\n", doclen);
 793     printf("\n");
 794     printf("Concurrency Level:      %d\n", concurrency);
 795     printf("Time taken for tests:   %.3f seconds\n", timetaken);
 796     printf("Complete requests:      %d\n", done);
 797     printf("Failed requests:        %d\n", bad);
 798     if (bad)
 799         printf("   (Connect: %d, Receive: %d, Length: %d, Exceptions: %d)\n",
 800             err_conn, err_recv, err_length, err_except);
 801     printf("Write errors:           %d\n", epipe);
 802     if (err_response)
 803         printf("Non-2xx responses:      %d\n", err_response);
 804     if (keepalive)
 805         printf("Keep-Alive requests:    %d\n", doneka);
 806     printf("Total transferred:      %" APR_INT64_T_FMT " bytes\n", totalread);
 807     if (send_body)
 808         printf("Total %s:%*s          %" APR_INT64_T_FMT "\n",
 809                method_str_pasttense[method],
 810                7 - strlen(method_str_pasttense[method]),
 811                " ",
 812                totalposted);
 813     printf("HTML transferred:       %" APR_INT64_T_FMT " bytes\n", totalbread);
 814
 815     /* avoid divide by zero */
 816     if (timetaken && done) {
 817         printf("Requests per second:    %.2f [#/sec] (mean)\n",
 818                (double) done / timetaken);
 819         printf("Time per request:       %.3f [ms] (mean)\n",
 820                (double) concurrency * timetaken * 1000 / done);
 821         printf("Time per request:       %.3f [ms] (mean, across all concurrent requests)\n",
 822                (double) timetaken * 1000 / done);
 823         printf("Transfer rate:          %.2f [Kbytes/sec] received\n",
 824                (double) totalread / 1024 / timetaken);
 825         if (send_body) {
 826             printf("                        %.2f kb/s sent\n",
 827                (double) totalposted / timetaken / 1024);
 828             printf("                        %.2f kb/s total\n",
 829                (double) (totalread + totalposted) / timetaken / 1024);
 830         }
 831     }
 832
 833     if (done > 0) {
 834         /* work out connection times */
 835         int i;
 836         apr_time_t totalcon = 0, total = 0, totald = 0, totalwait = 0;
 837         apr_time_t meancon, meantot, meand, meanwait;
 838         apr_interval_time_t mincon = AB_MAX, mintot = AB_MAX, mind = AB_MAX,
 839                             minwait = AB_MAX;
 840         apr_interval_time_t maxcon = 0, maxtot = 0, maxd = 0, maxwait = 0;
 841         apr_interval_time_t mediancon = 0, mediantot = 0, mediand = 0, medianwait = 0;
 842         double sdtot = 0, sdcon = 0, sdd = 0, sdwait = 0;
 843
 844         for (i = 0; i < done; i++) {
 845             struct data *s = &stats[i];
 846             mincon = ap_min(mincon, s->ctime);
 847             mintot = ap_min(mintot, s->time);
 848             mind = ap_min(mind, s->time - s->ctime);
 849             minwait = ap_min(minwait, s->waittime);
 850
 851             maxcon = ap_max(maxcon, s->ctime);
 852             maxtot = ap_max(maxtot, s->time);
 853             maxd = ap_max(maxd, s->time - s->ctime);
 854             maxwait = ap_max(maxwait, s->waittime);
 855
 856             totalcon += s->ctime;
 857             total += s->time;
 858             totald += s->time - s->ctime;
 859             totalwait += s->waittime;
 860         }
 861         meancon = totalcon / done;
 862         meantot = total / done;
 863         meand = totald / done;
 864         meanwait = totalwait / done;
 865
 866         /* calculating the sample variance: the sum of the squared deviations, divided by n-1 */
 867         for (i = 0; i < done; i++) {
 868             struct data *s = &stats[i];
 869             double a;
 870             a = ((double)s->time - meantot);
 871             sdtot += a * a;
 872             a = ((double)s->ctime - meancon);
 873             sdcon += a * a;
 874             a = ((double)s->time - (double)s->ctime - meand);
 875             sdd += a * a;
 876             a = ((double)s->waittime - meanwait);
 877             sdwait += a * a;
 878         }
 879
 880         sdtot = (done > 1) ? sqrt(sdtot / (done - 1)) : 0;
 881         sdcon = (done > 1) ? sqrt(sdcon / (done - 1)) : 0;
 882         sdd = (done > 1) ? sqrt(sdd / (done - 1)) : 0;
 883         sdwait = (done > 1) ? sqrt(sdwait / (done - 1)) : 0;
 884
 885         /*
 886          * XXX: what is better; this hideous cast of the compradre function; or
 887          * the four warnings during compile ? dirkx just does not know and
 888          * hates both/
 889          */
 890         qsort(stats, done, sizeof(struct data),
 891               (int (*) (const void *, const void *)) compradre);
 892         if ((done > 1) && (done % 2))
 893             mediancon = (stats[done / 2].ctime + stats[done / 2 + 1].ctime) / 2;
 894         else
 895             mediancon = stats[done / 2].ctime;
 896
 897         qsort(stats, done, sizeof(struct data),
 898               (int (*) (const void *, const void *)) compri);
 899         if ((done > 1) && (done % 2))
 900             mediand = (stats[done / 2].time + stats[done / 2 + 1].time \
 901             -stats[done / 2].ctime - stats[done / 2 + 1].ctime) / 2;
 902         else
 903             mediand = stats[done / 2].time - stats[done / 2].ctime;
 904
 905         qsort(stats, done, sizeof(struct data),
 906               (int (*) (const void *, const void *)) compwait);
 907         if ((done > 1) && (done % 2))
 908             medianwait = (stats[done / 2].waittime + stats[done / 2 + 1].waittime) / 2;
 909         else
 910             medianwait = stats[done / 2].waittime;
 911
 912         qsort(stats, done, sizeof(struct data),
 913               (int (*) (const void *, const void *)) comprando);
 914         if ((done > 1) && (done % 2))
 915             mediantot = (stats[done / 2].time + stats[done / 2 + 1].time) / 2;
 916         else
 917             mediantot = stats[done / 2].time;
 918
 919         printf("\nConnection Times (ms)\n");
 920         /*
 921          * Reduce stats from apr time to milliseconds
 922          */
 923         mincon     = ap_round_ms(mincon);
 924         mind       = ap_round_ms(mind);
 925         minwait    = ap_round_ms(minwait);
 926         mintot     = ap_round_ms(mintot);
 927         meancon    = ap_round_ms(meancon);
 928         meand      = ap_round_ms(meand);
 929         meanwait   = ap_round_ms(meanwait);
 930         meantot    = ap_round_ms(meantot);
 931         mediancon  = ap_round_ms(mediancon);
 932         mediand    = ap_round_ms(mediand);
 933         medianwait = ap_round_ms(medianwait);
 934         mediantot  = ap_round_ms(mediantot);
 935         maxcon     = ap_round_ms(maxcon);
 936         maxd       = ap_round_ms(maxd);
 937         maxwait    = ap_round_ms(maxwait);
 938         maxtot     = ap_round_ms(maxtot);
 939         sdcon      = ap_double_ms(sdcon);
 940         sdd        = ap_double_ms(sdd);
 941         sdwait     = ap_double_ms(sdwait);
 942         sdtot      = ap_double_ms(sdtot);
 943
 944         if (confidence) {
 945 #define CONF_FMT_STRING "%5" APR_TIME_T_FMT " %4" APR_TIME_T_FMT " %5.1f %6" APR_TIME_T_FMT " %7" APR_TIME_T_FMT "\n"
 946             printf("              min  mean[+/-sd] median   max\n");
 947             printf("Connect:    " CONF_FMT_STRING,
 948                    mincon, meancon, sdcon, mediancon, maxcon);
 949             printf("Processing: " CONF_FMT_STRING,
 950                    mind, meand, sdd, mediand, maxd);
 951             printf("Waiting:    " CONF_FMT_STRING,
 952                    minwait, meanwait, sdwait, medianwait, maxwait);
 953             printf("Total:      " CONF_FMT_STRING,
 954                    mintot, meantot, sdtot, mediantot, maxtot);
 955 #undef CONF_FMT_STRING
 956
 957 #define     SANE(what,mean,median,sd) \
 958               { \
 959                 double d = (double)mean - median; \
 960                 if (d < 0) d = -d; \
 961                 if (d > 2 * sd ) \
 962                     printf("ERROR: The median and mean for " what " are more than twice the standard\n" \
 963                            "       deviation apart. These results are NOT reliable.\n"); \
 964                 else if (d > sd ) \
 965                     printf("WARNING: The median and mean for " what " are not within a normal deviation\n" \
 966                            "        These results are probably not that reliable.\n"); \
 967             }
 968             SANE("the initial connection time", meancon, mediancon, sdcon);
 969             SANE("the processing time", meand, mediand, sdd);
 970             SANE("the waiting time", meanwait, medianwait, sdwait);
 971             SANE("the total time", meantot, mediantot, sdtot);
 972         }
 973         else {
 974             printf("              min   avg   max\n");
 975 #define CONF_FMT_STRING "%5" APR_TIME_T_FMT " %5" APR_TIME_T_FMT "%5" APR_TIME_T_FMT "\n"
 976             printf("Connect:    " CONF_FMT_STRING, mincon, meancon, maxcon);
 977             printf("Processing: " CONF_FMT_STRING, mintot - mincon,
 978                                                    meantot - meancon,
 979                                                    maxtot - maxcon);
 980             printf("Total:      " CONF_FMT_STRING, mintot, meantot, maxtot);
 981 #undef CONF_FMT_STRING
 982         }
 983
 984
 985         /* Sorted on total connect times */
 986         if (percentile && (done > 1)) {
 987             printf("\nPercentage of the requests served within a certain time (ms)\n");
 988             for (i = 0; i < sizeof(percs) / sizeof(int); i++) {
 989                 if (percs[i] <= 0)
 990                     printf(" 0%%  <0> (never)\n");
 991                 else if (percs[i] >= 100)
 992                     printf(" 100%%  %5" APR_TIME_T_FMT " (longest request)\n",
 993                            ap_round_ms(stats[done - 1].time));
 994                 else
 995                     printf("  %d%%  %5" APR_TIME_T_FMT "\n", percs[i],
 996                            ap_round_ms(stats[(int) (done * percs[i] / 100)].time));
 997             }
 998         }
 999         if (csvperc) {
1000             FILE *out = fopen(csvperc, "w");
1001             if (!out) {
1002                 perror("Cannot open CSV output file");
1003                 exit(1);
1004             }
1005             fprintf(out, "" "Percentage served" "," "Time in ms" "\n");
1006             for (i = 0; i < 100; i++) {
1007                 double t;
1008                 if (i == 0)
1009                     t = ap_double_ms(stats[0].time);
1010                 else if (i == 100)
1011                     t = ap_double_ms(stats[done - 1].time);
1012                 else
1013                     t = ap_double_ms(stats[(int) (0.5 + done * i / 100.0)].time);
1014                 fprintf(out, "%d,%.3f\n", i, t);
1015             }
1016             fclose(out);
1017         }
1018         if (gnuplot) {
1019             FILE *out = fopen(gnuplot, "w");
1020             char tmstring[APR_CTIME_LEN];
1021             if (!out) {
1022                 perror("Cannot open gnuplot output file");
1023                 exit(1);
1024             }
1025             fprintf(out, "starttime\tseconds\tctime\tdtime\tttime\twait\n");
1026             for (i = 0; i < done; i++) {
1027                 (void) apr_ctime(tmstring, stats[i].starttime);
1028                 fprintf(out, "%s\t%" APR_TIME_T_FMT "\t%" APR_TIME_T_FMT
1029                                "\t%" APR_TIME_T_FMT "\t%" APR_TIME_T_FMT
1030                                "\t%" APR_TIME_T_FMT "\n", tmstring,
1031                         apr_time_sec(stats[i].starttime),
1032                         ap_round_ms(stats[i].ctime),
1033                         ap_round_ms(stats[i].time - stats[i].ctime),
1034                         ap_round_ms(stats[i].time),
1035                         ap_round_ms(stats[i].waittime));
1036             }
1037             fclose(out);
1038         }
1039     }
1040
1041     if (sig) {
1042         exit(1);
1043     }
1044 }
1045
1046 /* --------------------------------------------------------- */
1047
1048 /* calculate and output results in HTML  */
1049
1050 static void output_html_results(void)
1051 {
1052     double timetaken = (double) (lasttime - start) / APR_USEC_PER_SEC;
1053
1054     printf("\n\n<table %s>\n", tablestring);
1055     printf("<tr %s><th colspan=2 %s>Server Software:</th>"
1056        "<td colspan=2 %s>%s</td></tr>\n",
1057        trstring, tdstring, tdstring, servername);
1058     printf("<tr %s><th colspan=2 %s>Server Hostname:</th>"
1059        "<td colspan=2 %s>%s</td></tr>\n",
1060        trstring, tdstring, tdstring, hostname);
1061     printf("<tr %s><th colspan=2 %s>Server Port:</th>"
1062        "<td colspan=2 %s>%hu</td></tr>\n",
1063        trstring, tdstring, tdstring, port);
1064     printf("<tr %s><th colspan=2 %s>Document Path:</th>"
1065        "<td colspan=2 %s>%s</td></tr>\n",
1066        trstring, tdstring, tdstring, path);
1067     printf("<tr %s><th colspan=2 %s>Document Length:</th>"
1068        "<td colspan=2 %s>%" APR_SIZE_T_FMT " bytes</td></tr>\n",
1069        trstring, tdstring, tdstring, doclen);
1070     printf("<tr %s><th colspan=2 %s>Concurrency Level:</th>"
1071        "<td colspan=2 %s>%d</td></tr>\n",
1072        trstring, tdstring, tdstring, concurrency);
1073     printf("<tr %s><th colspan=2 %s>Time taken for tests:</th>"
1074        "<td colspan=2 %s>%.3f seconds</td></tr>\n",
1075        trstring, tdstring, tdstring, timetaken);
1076     printf("<tr %s><th colspan=2 %s>Complete requests:</th>"
1077        "<td colspan=2 %s>%d</td></tr>\n",
1078        trstring, tdstring, tdstring, done);
1079     printf("<tr %s><th colspan=2 %s>Failed requests:</th>"
1080        "<td colspan=2 %s>%d</td></tr>\n",
1081        trstring, tdstring, tdstring, bad);
1082     if (bad)
1083         printf("<tr %s><td colspan=4 %s >   (Connect: %d, Length: %d, Exceptions: %d)</td></tr>\n",
1084            trstring, tdstring, err_conn, err_length, err_except);
1085     if (err_response)
1086         printf("<tr %s><th colspan=2 %s>Non-2xx responses:</th>"
1087            "<td colspan=2 %s>%d</td></tr>\n",
1088            trstring, tdstring, tdstring, err_response);
1089     if (keepalive)
1090         printf("<tr %s><th colspan=2 %s>Keep-Alive requests:</th>"
1091            "<td colspan=2 %s>%d</td></tr>\n",
1092            trstring, tdstring, tdstring, doneka);
1093     printf("<tr %s><th colspan=2 %s>Total transferred:</th>"
1094        "<td colspan=2 %s>%" APR_INT64_T_FMT " bytes</td></tr>\n",
1095        trstring, tdstring, tdstring, totalread);
1096     if (send_body)
1097         printf("<tr %s><th colspan=2 %s>Total %s:</th>"
1098            "<td colspan=2 %s>%" APR_INT64_T_FMT "</td></tr>\n",
1099            trstring, tdstring,
1100            method_str_pasttense[method],
1101            tdstring, totalposted);
1102     printf("<tr %s><th colspan=2 %s>HTML transferred:</th>"
1103        "<td colspan=2 %s>%" APR_INT64_T_FMT " bytes</td></tr>\n",
1104        trstring, tdstring, tdstring, totalbread);
1105
1106     /* avoid divide by zero */
1107     if (timetaken) {
1108         printf("<tr %s><th colspan=2 %s>Requests per second:</th>"
1109            "<td colspan=2 %s>%.2f</td></tr>\n",
1110            trstring, tdstring, tdstring, (double) done * 1000 / timetaken);
1111         printf("<tr %s><th colspan=2 %s>Transfer rate:</th>"
1112            "<td colspan=2 %s>%.2f kb/s received</td></tr>\n",
1113            trstring, tdstring, tdstring, (double) totalread / timetaken);
1114         if (send_body) {
1115             printf("<tr %s><td colspan=2 %s>&nbsp;</td>"
1116                "<td colspan=2 %s>%.2f kb/s sent</td></tr>\n",
1117                trstring, tdstring, tdstring,
1118                (double) totalposted / timetaken);
1119             printf("<tr %s><td colspan=2 %s>&nbsp;</td>"
1120                "<td colspan=2 %s>%.2f kb/s total</td></tr>\n",
1121                trstring, tdstring, tdstring,
1122                (double) (totalread + totalposted) / timetaken);
1123         }
1124     }
1125     {
1126         /* work out connection times */
1127         int i;
1128         apr_interval_time_t totalcon = 0, total = 0;
1129         apr_interval_time_t mincon = AB_MAX, mintot = AB_MAX;
1130         apr_interval_time_t maxcon = 0, maxtot = 0;
1131
1132         for (i = 0; i < done; i++) {
1133             struct data *s = &stats[i];
1134             mincon = ap_min(mincon, s->ctime);
1135             mintot = ap_min(mintot, s->time);
1136             maxcon = ap_max(maxcon, s->ctime);
1137             maxtot = ap_max(maxtot, s->time);
1138             totalcon += s->ctime;
1139             total    += s->time;
1140         }
1141         /*
1142          * Reduce stats from apr time to milliseconds
1143          */
1144         mincon   = ap_round_ms(mincon);
1145         mintot   = ap_round_ms(mintot);
1146         maxcon   = ap_round_ms(maxcon);
1147         maxtot   = ap_round_ms(maxtot);
1148         totalcon = ap_round_ms(totalcon);
1149         total    = ap_round_ms(total);
1150
1151         if (done > 0) { /* avoid division by zero (if 0 done) */
1152             printf("<tr %s><th %s colspan=4>Connnection Times (ms)</th></tr>\n",
1153                trstring, tdstring);
1154             printf("<tr %s><th %s>&nbsp;</th> <th %s>min</th>   <th %s>avg</th>   <th %s>max</th></tr>\n",
1155                trstring, tdstring, tdstring, tdstring, tdstring);
1156             printf("<tr %s><th %s>Connect:</th>"
1157                "<td %s>%5" APR_TIME_T_FMT "</td>"
1158                "<td %s>%5" APR_TIME_T_FMT "</td>"
1159                "<td %s>%5" APR_TIME_T_FMT "</td></tr>\n",
1160                trstring, tdstring, tdstring, mincon, tdstring, totalcon / done, tdstring, maxcon);
1161             printf("<tr %s><th %s>Processing:</th>"
1162                "<td %s>%5" APR_TIME_T_FMT "</td>"
1163                "<td %s>%5" APR_TIME_T_FMT "</td>"
1164                "<td %s>%5" APR_TIME_T_FMT "</td></tr>\n",
1165                trstring, tdstring, tdstring, mintot - mincon, tdstring,
1166                (total / done) - (totalcon / done), tdstring, maxtot - maxcon);
1167             printf("<tr %s><th %s>Total:</th>"
1168                "<td %s>%5" APR_TIME_T_FMT "</td>"
1169                "<td %s>%5" APR_TIME_T_FMT "</td>"
1170                "<td %s>%5" APR_TIME_T_FMT "</td></tr>\n",
1171                trstring, tdstring, tdstring, mintot, tdstring, total / done, tdstring, maxtot);
1172         }
1173         printf("</table>\n");
1174     }
1175 }
1176
1177 /* --------------------------------------------------------- */
1178
1179 /* start asnchronous non-blocking connection */
1180
1181 static void start_connect(struct connection * c)
1182 {
1183     apr_status_t rv;
1184
1185     if (!(started < requests))
1186     return;
1187
1188     c->read = 0;
1189     c->bread = 0;
1190     c->keepalive = 0;
1191     c->cbx = 0;
1192     c->gotheader = 0;
1193     c->rwrite = 0;
1194     if (c->ctx)
1195         apr_pool_clear(c->ctx);
1196     else
1197         apr_pool_create(&c->ctx, cntxt);
1198
1199     if ((rv = apr_socket_create(&c->aprsock, destsa->family,
1200                 SOCK_STREAM, 0, c->ctx)) != APR_SUCCESS) {
1201     apr_err("socket", rv);
1202     }
1203
1204     c->pollfd.desc_type = APR_POLL_SOCKET;
1205     c->pollfd.desc.s = c->aprsock;
1206     c->pollfd.reqevents = 0;
1207     c->pollfd.client_data = c;
1208
1209     if ((rv = apr_socket_opt_set(c->aprsock, APR_SO_NONBLOCK, 1))
1210          != APR_SUCCESS) {
1211         apr_err("socket nonblock", rv);
1212     }
1213
1214     if (windowsize != 0) {
1215         rv = apr_socket_opt_set(c->aprsock, APR_SO_SNDBUF,
1216                                 windowsize);
1217         if (rv != APR_SUCCESS && rv != APR_ENOTIMPL) {
1218             apr_err("socket send buffer", rv);
1219         }
1220         rv = apr_socket_opt_set(c->aprsock, APR_SO_RCVBUF,
1221                                 windowsize);
1222         if (rv != APR_SUCCESS && rv != APR_ENOTIMPL) {
1223             apr_err("socket receive buffer", rv);
1224         }
1225     }
1226
1227     c->start = lasttime = apr_time_now();
1228 #ifdef USE_SSL
1229     if (is_ssl) {
1230         BIO *bio;
1231         apr_os_sock_t fd;
1232
1233         if ((c->ssl = SSL_new(ssl_ctx)) == NULL) {
1234             BIO_printf(bio_err, "SSL_new failed.\n");
1235             ERR_print_errors(bio_err);
1236             exit(1);
1237         }
1238         ssl_rand_seed();
1239         apr_os_sock_get(&fd, c->aprsock);
1240         bio = BIO_new_socket(fd, BIO_NOCLOSE);
1241         SSL_set_bio(c->ssl, bio, bio);
1242         SSL_set_connect_state(c->ssl);
1243         if (verbosity >= 4) {
1244             BIO_set_callback(bio, ssl_print_cb);
1245             BIO_set_callback_arg(bio, (void *)bio_err);
1246         }
1247     } else {
1248         c->ssl = NULL;
1249     }
1250 #endif
1251     if ((rv = apr_socket_connect(c->aprsock, destsa)) != APR_SUCCESS) {
1252         if (APR_STATUS_IS_EINPROGRESS(rv)) {
1253             set_conn_state(c, STATE_CONNECTING);
1254             c->rwrite = 0;
1255             return;
1256         }
1257         else {
1258             set_conn_state(c, STATE_UNCONNECTED);
1259             apr_socket_close(c->aprsock);
1260             err_conn++;
1261             if (bad++ > 10) {
1262                 fprintf(stderr,
1263                    "\nTest aborted after 10 failures\n\n");
1264                 apr_err("apr_socket_connect()", rv);
1265             }
1266
1267             start_connect(c);
1268             return;
1269         }
1270     }
1271
1272     /* connected first time */
1273     set_conn_state(c, STATE_CONNECTED);
1274     started++;
1275 #ifdef USE_SSL
1276     if (c->ssl) {
1277         ssl_proceed_handshake(c);
1278     } else
1279 #endif
1280     {
1281         write_request(c);
1282     }
1283 }
1284
1285 /* --------------------------------------------------------- */
1286
1287 /* close down connection and save stats */
1288
1289 static void close_connection(struct connection * c)
1290 {
1291     if (c->read == 0 && c->keepalive) {
1292         /*
1293          * server has legitimately shut down an idle keep alive request
1294          */
1295         if (good)
1296             good--;     /* connection never happened */
1297     }
1298     else {
1299         if (good == 1) {
1300             /* first time here */
1301             doclen = c->bread;
1302         }
1303         else if (c->bread != doclen) {
1304             bad++;
1305             err_length++;
1306         }
1307         /* save out time */
1308         if (done < requests) {
1309             struct data *s = &stats[done++];
1310             c->done      = lasttime = apr_time_now();
1311             s->starttime = c->start;
1312             s->ctime     = ap_max(0, c->connect - c->start);
1313             s->time      = ap_max(0, c->done - c->start);
1314             s->waittime  = ap_max(0, c->beginread - c->endwrite);
1315             if (heartbeatres && !(done % heartbeatres)) {
1316                 fprintf(stderr, "Completed %d requests\n", done);
1317                 fflush(stderr);
1318             }
1319         }
1320     }
1321
1322     set_conn_state(c, STATE_UNCONNECTED);
1323 #ifdef USE_SSL
1324     if (c->ssl) {
1325         SSL_shutdown(c->ssl);
1326         SSL_free(c->ssl);
1327         c->ssl = NULL;
1328     }
1329 #endif
1330     apr_socket_close(c->aprsock);
1331
1332     /* connect again */
1333     start_connect(c);
1334     return;
1335 }
1336
1337 /* --------------------------------------------------------- */
1338
1339 /* read data from connection */
1340
1341 static void read_connection(struct connection * c)
1342 {
1343     apr_size_t r;
1344     apr_status_t status;
1345     char *part;
1346     char respcode[4];       /* 3 digits and null */
1347
1348     r = sizeof(buffer);
1349 #ifdef USE_SSL
1350     if (c->ssl) {
1351         status = SSL_read(c->ssl, buffer, r);
1352         if (status <= 0) {
1353             int scode = SSL_get_error(c->ssl, status);
1354
1355             if (scode == SSL_ERROR_ZERO_RETURN) {
1356                 /* connection closed cleanly: */
1357                 good++;
1358                 close_connection(c);
1359             }
1360             else if (scode != SSL_ERROR_WANT_WRITE
1361                      && scode != SSL_ERROR_WANT_READ) {
1362                 /* some fatal error: */
1363                 c->read = 0;
1364                 BIO_printf(bio_err, "SSL read failed - closing connection\n");
1365                 ERR_print_errors(bio_err);
1366                 close_connection(c);
1367             }
1368             return;
1369         }
1370         r = status;
1371     }
1372     else
1373 #endif
1374     {
1375         status = apr_socket_recv(c->aprsock, buffer, &r);
1376         if (APR_STATUS_IS_EAGAIN(status))
1377             return;
1378         else if (r == 0 && APR_STATUS_IS_EOF(status)) {
1379             good++;
1380             close_connection(c);
1381             return;
1382         }
1383         /* catch legitimate fatal apr_socket_recv errors */
1384         else if (status != APR_SUCCESS) {
1385             err_recv++;
1386             if (recverrok) {
1387                 bad++;
1388                 close_connection(c);
1389                 if (verbosity >= 1) {
1390                     char buf[120];
1391                     fprintf(stderr,"%s: %s (%d)\n", "apr_socket_recv", apr_strerror(status, buf, sizeof buf), status);
1392                 }
1393                 return;
1394             } else {
1395                 apr_err("apr_socket_recv", status);
1396             }
1397         }
1398     }
1399
1400     totalread += r;
1401     if (c->read == 0) {
1402         c->beginread = apr_time_now();
1403     }
1404     c->read += r;
1405
1406
1407     if (!c->gotheader) {
1408         char *s;
1409         int l = 4;
1410         apr_size_t space = CBUFFSIZE - c->cbx - 1; /* -1 allows for \0 term */
1411         int tocopy = (space < r) ? space : r;
1412 #ifdef NOT_ASCII
1413         apr_size_t inbytes_left = space, outbytes_left = space;
1414
1415         status = apr_xlate_conv_buffer(from_ascii, buffer, &inbytes_left,
1416                            c->cbuff + c->cbx, &outbytes_left);
1417         if (status || inbytes_left || outbytes_left) {
1418             fprintf(stderr, "only simple translation is supported (%d/%" APR_SIZE_T_FMT
1419                             "/%" APR_SIZE_T_FMT ")\n", status, inbytes_left, outbytes_left);
1420             exit(1);
1421         }
1422 #else
1423         memcpy(c->cbuff + c->cbx, buffer, space);
1424 #endif              /* NOT_ASCII */
1425         c->cbx += tocopy;
1426         space -= tocopy;
1427         c->cbuff[c->cbx] = 0;   /* terminate for benefit of strstr */
1428         if (verbosity >= 2) {
1429             printf("LOG: header received:\n%s\n", c->cbuff);
1430         }
1431         s = strstr(c->cbuff, "\r\n\r\n");
1432         /*
1433          * this next line is so that we talk to NCSA 1.5 which blatantly
1434          * breaks the http specifaction
1435          */
1436         if (!s) {
1437             s = strstr(c->cbuff, "\n\n");
1438             l = 2;
1439         }
1440
1441         if (!s) {
1442             /* read rest next time */
1443             if (space) {
1444                 return;
1445             }
1446             else {
1447             /* header is in invalid or too big - close connection */
1448                 set_conn_state(c, STATE_UNCONNECTED);
1449                 apr_socket_close(c->aprsock);
1450                 err_response++;
1451                 if (bad++ > 10) {
1452                     err("\nTest aborted after 10 failures\n\n");
1453                 }
1454                 start_connect(c);
1455             }
1456         }
1457         else {
1458             /* have full header */
1459             if (!good) {
1460                 /*
1461                  * this is first time, extract some interesting info
1462                  */
1463                 char *p, *q;
1464                 p = strstr(c->cbuff, "Server:");
1465                 q = servername;
1466                 if (p) {
1467                     p += 8;
1468                     while (*p > 32)
1469                     *q++ = *p++;
1470                 }
1471                 *q = 0;
1472             }
1473             /*
1474              * XXX: this parsing isn't even remotely HTTP compliant... but in
1475              * the interest of speed it doesn't totally have to be, it just
1476              * needs to be extended to handle whatever servers folks want to
1477              * test against. -djg
1478              */
1479
1480             /* check response code */
1481             part = strstr(c->cbuff, "HTTP");    /* really HTTP/1.x_ */
1482             if (part && strlen(part) > strlen("HTTP/1.x_")) {
1483                 strncpy(respcode, (part + strlen("HTTP/1.x_")), 3);
1484                 respcode[3] = '\0';
1485             }
1486             else {
1487                 strcpy(respcode, "500");
1488             }
1489
1490             if (respcode[0] != '2') {
1491                 err_response++;
1492                 if (verbosity >= 2)
1493                     printf("WARNING: Response code not 2xx (%s)\n", respcode);
1494             }
1495             else if (verbosity >= 3) {
1496                 printf("LOG: Response code = %s\n", respcode);
1497             }
1498             c->gotheader = 1;
1499             *s = 0;     /* terminate at end of header */
1500             if (keepalive &&
1501             (strstr(c->cbuff, "Keep-Alive")
1502              || strstr(c->cbuff, "keep-alive"))) {  /* for benefit of MSIIS */
1503                 char *cl;
1504                 cl = strstr(c->cbuff, "Content-Length:");
1505                 /* handle NCSA, which sends Content-length: */
1506                 if (!cl)
1507                     cl = strstr(c->cbuff, "Content-length:");
1508                 if (cl) {
1509                     c->keepalive = 1;
1510                     /* response to HEAD doesn't have entity body */
1511                     c->length = method != HEAD ? atoi(cl + 16) : 0;
1512                 }
1513                 /* The response may not have a Content-Length header */
1514                 if (!cl) {
1515                     c->keepalive = 1;
1516                     c->length = 0;
1517                 }
1518             }
1519             c->bread += c->cbx - (s + l - c->cbuff) + r - tocopy;
1520             totalbread += c->bread;
1521         }
1522     }
1523     else {
1524         /* outside header, everything we have read is entity body */
1525         c->bread += r;
1526         totalbread += r;
1527     }
1528
1529     if (c->keepalive && (c->bread >= c->length)) {
1530         /* finished a keep-alive connection */
1531         good++;
1532         /* save out time */
1533         if (good == 1) {
1534             /* first time here */
1535             doclen = c->bread;
1536         }
1537         else if (c->bread != doclen) {
1538             bad++;
1539             err_length++;
1540         }
1541         if (done < requests) {
1542             struct data *s = &stats[done++];
1543             doneka++;
1544             c->done      = apr_time_now();
1545             s->starttime = c->start;
1546             s->ctime     = ap_max(0, c->connect - c->start);
1547             s->time      = ap_max(0, c->done - c->start);
1548             s->waittime  = ap_max(0, c->beginread - c->endwrite);
1549             if (heartbeatres && !(done % heartbeatres)) {
1550                 fprintf(stderr, "Completed %d requests\n", done);
1551                 fflush(stderr);
1552             }
1553         }
1554         c->keepalive = 0;
1555         c->length = 0;
1556         c->gotheader = 0;
1557         c->cbx = 0;
1558         c->read = c->bread = 0;
1559         /* zero connect time with keep-alive */
1560         c->start = c->connect = lasttime = apr_time_now();
1561         write_request(c);
1562     }
1563 }
1564
1565 /* --------------------------------------------------------- */
1566
1567 /* run the tests */
1568
1569 static void test(void)
1570 {
1571     apr_time_t stoptime;
1572     apr_int16_t rtnev;
1573     apr_status_t rv;
1574     int i;
1575     apr_status_t status;
1576     int snprintf_res = 0;
1577 #ifdef NOT_ASCII
1578     apr_size_t inbytes_left, outbytes_left;
1579 #endif
1580
1581     if (isproxy) {
1582         connecthost = apr_pstrdup(cntxt, proxyhost);
1583         connectport = proxyport;
1584     }
1585     else {
1586         connecthost = apr_pstrdup(cntxt, hostname);
1587         connectport = port;
1588     }
1589
1590     if (!use_html) {
1591         printf("Benchmarking %s ", hostname);
1592     if (isproxy)
1593         printf("[through %s:%d] ", proxyhost, proxyport);
1594     printf("(be patient)%s",
1595            (heartbeatres ? "\n" : "..."));
1596     fflush(stdout);
1597     }
1598
1599     con = calloc(concurrency, sizeof(struct connection));
1600
1601     stats = calloc(requests, sizeof(struct data));
1602
1603     if ((status = apr_pollset_create(&readbits, concurrency, cntxt,
1604                                      APR_POLLSET_NOCOPY)) != APR_SUCCESS) {
1605         apr_err("apr_pollset_create failed", status);
1606     }
1607
1608     /* add default headers if necessary */
1609     if (!opt_host) {
1610         /* Host: header not overridden, add default value to hdrs */
1611         hdrs = apr_pstrcat(cntxt, hdrs, "Host: ", host_field, colonhost, "\r\n", NULL);
1612     }
1613     else {
1614         /* Header overridden, no need to add, as it is already in hdrs */
1615     }
1616
1617     if (!opt_useragent) {
1618         /* User-Agent: header not overridden, add default value to hdrs */
1619         hdrs = apr_pstrcat(cntxt, hdrs, "User-Agent: ApacheBench/", AP_AB_BASEREVISION, "\r\n", NULL);
1620     }
1621     else {
1622         /* Header overridden, no need to add, as it is already in hdrs */
1623     }
1624
1625     if (!opt_accept) {
1626         /* Accept: header not overridden, add default value to hdrs */
1627         hdrs = apr_pstrcat(cntxt, hdrs, "Accept: */*\r\n", NULL);
1628     }
1629     else {
1630         /* Header overridden, no need to add, as it is already in hdrs */
1631     }
1632
1633     /* setup request */
1634     if (!send_body) {
1635         snprintf_res = apr_snprintf(request, sizeof(_request),
1636             "%s %s HTTP/1.0\r\n"
1637             "%s" "%s" "%s"
1638             "%s" "\r\n",
1639             method_str[method],
1640             (isproxy) ? fullurl : path,
1641             keepalive ? "Connection: Keep-Alive\r\n" : "",
1642             cookie, auth, hdrs);
1643     }
1644     else {
1645         snprintf_res = apr_snprintf(request,  sizeof(_request),
1646             "%s %s HTTP/1.0\r\n"
1647             "%s" "%s" "%s"
1648             "Content-length: %" APR_SIZE_T_FMT "\r\n"
1649             "Content-type: %s\r\n"
1650             "%s"
1651             "\r\n",
1652             method_str[method],
1653             (isproxy) ? fullurl : path,
1654             keepalive ? "Connection: Keep-Alive\r\n" : "",
1655             cookie, auth,
1656             postlen,
1657             (content_type[0]) ? content_type : "text/plain", hdrs);
1658     }
1659     if (snprintf_res >= sizeof(_request)) {
1660         err("Request too long\n");
1661     }
1662
1663     if (verbosity >= 2)
1664         printf("INFO: %s header == \n---\n%s\n---\n",
1665                method_str[method], request);
1666
1667     reqlen = strlen(request);
1668
1669     /*
1670      * Combine headers and (optional) post file into one continuous buffer
1671      */
1672     if (send_body) {
1673         char *buff = malloc(postlen + reqlen + 1);
1674         if (!buff) {
1675             fprintf(stderr, "error creating request buffer: out of memory\n");
1676             return;
1677         }
1678         strcpy(buff, request);
1679         memcpy(buff + reqlen, postdata, postlen);
1680         request = buff;
1681     }
1682
1683 #ifdef NOT_ASCII
1684     inbytes_left = outbytes_left = reqlen;
1685     status = apr_xlate_conv_buffer(to_ascii, request, &inbytes_left,
1686                    request, &outbytes_left);
1687     if (status || inbytes_left || outbytes_left) {
1688         fprintf(stderr, "only simple translation is supported (%d/%"
1689                         APR_SIZE_T_FMT "/%" APR_SIZE_T_FMT ")\n",
1690                         status, inbytes_left, outbytes_left);
1691         exit(1);
1692     }
1693 #endif              /* NOT_ASCII */
1694
1695     /* This only needs to be done once */
1696     if ((rv = apr_sockaddr_info_get(&destsa, connecthost, APR_UNSPEC, connectport, 0, cntxt))
1697        != APR_SUCCESS) {
1698         char buf[120];
1699         apr_snprintf(buf, sizeof(buf),
1700                  "apr_sockaddr_info_get() for %s", connecthost);
1701         apr_err(buf, rv);
1702     }
1703
1704     /* ok - lets start */
1705     start = lasttime = apr_time_now();
1706     stoptime = tlimit ? (start + apr_time_from_sec(tlimit)) : AB_MAX;
1707
1708 #ifdef SIGINT
1709     /* Output the results if the user terminates the run early. */
1710     apr_signal(SIGINT, output_results);
1711 #endif
1712
1713     /* initialise lots of requests */
1714     for (i = 0; i < concurrency; i++) {
1715         con[i].socknum = i;
1716         start_connect(&con[i]);
1717     }
1718
1719     do {
1720         apr_int32_t n;
1721         const apr_pollfd_t *pollresults;
1722
1723         n = concurrency;
1724         do {
1725             status = apr_pollset_poll(readbits, aprtimeout, &n, &pollresults);
1726         } while (APR_STATUS_IS_EINTR(status));
1727         if (status != APR_SUCCESS)
1728             apr_err("apr_poll", status);
1729
1730         if (!n) {
1731             err("\nServer timed out\n\n");
1732         }
1733
1734         for (i = 0; i < n; i++) {
1735             const apr_pollfd_t *next_fd = &(pollresults[i]);
1736             struct connection *c;
1737
1738             c = next_fd->client_data;
1739
1740             /*
1741              * If the connection isn't connected how can we check it?
1742              */
1743             if (c->state == STATE_UNCONNECTED)
1744                 continue;
1745
1746             rtnev = next_fd->rtnevents;
1747
1748 #ifdef USE_SSL
1749             if (c->state == STATE_CONNECTED && c->ssl && SSL_in_init(c->ssl)) {
1750                 ssl_proceed_handshake(c);
1751                 continue;
1752             }
1753 #endif
1754
1755             /*
1756              * Notes: APR_POLLHUP is set after FIN is received on some
1757              * systems, so treat that like APR_POLLIN so that we try to read
1758              * again.
1759              *
1760              * Some systems return APR_POLLERR with APR_POLLHUP.  We need to
1761              * call read_connection() for APR_POLLHUP, so check for
1762              * APR_POLLHUP first so that a closed connection isn't treated
1763              * like an I/O error.  If it is, we never figure out that the
1764              * connection is done and we loop here endlessly calling
1765              * apr_poll().
1766              */
1767             if ((rtnev & APR_POLLIN) || (rtnev & APR_POLLPRI) || (rtnev & APR_POLLHUP))
1768                 read_connection(c);
1769             if ((rtnev & APR_POLLERR) || (rtnev & APR_POLLNVAL)) {
1770                 bad++;
1771                 err_except++;
1772                 /* avoid apr_poll/EINPROGRESS loop on HP-UX, let recv discover ECONNREFUSED */
1773                 if (c->state == STATE_CONNECTING) {
1774                     read_connection(c);
1775                 }
1776                 else {
1777                     start_connect(c);
1778                 }
1779                 continue;
1780             }
1781             if (rtnev & APR_POLLOUT) {
1782                 if (c->state == STATE_CONNECTING) {
1783                     rv = apr_socket_connect(c->aprsock, destsa);
1784                     if (rv != APR_SUCCESS) {
1785                         apr_socket_close(c->aprsock);
1786                         err_conn++;
1787                         if (bad++ > 10) {
1788                             fprintf(stderr,
1789                                     "\nTest aborted after 10 failures\n\n");
1790                             apr_err("apr_socket_connect()", rv);
1791                         }
1792                         set_conn_state(c, STATE_UNCONNECTED);
1793                         start_connect(c);
1794                         continue;
1795                     }
1796                     else {
1797                         set_conn_state(c, STATE_CONNECTED);
1798                         started++;
1799 #ifdef USE_SSL
1800                         if (c->ssl)
1801                             ssl_proceed_handshake(c);
1802                         else
1803 #endif
1804                         write_request(c);
1805                     }
1806                 }
1807                 else {
1808                     write_request(c);
1809                 }
1810             }
1811         }
1812     } while (lasttime < stoptime && done < requests);
1813
1814     if (heartbeatres)
1815         fprintf(stderr, "Finished %d requests\n", done);
1816     else
1817         printf("..done\n");
1818
1819     if (use_html)
1820         output_html_results();
1821     else
1822         output_results(0);
1823 }
1824
1825 /* ------------------------------------------------------- */
1826
1827 /* display copyright information */
1828 static void copyright(void)
1829 {
1830     if (!use_html) {
1831         printf("This is ApacheBench, Version %s\n", AP_AB_BASEREVISION " <$Revision$>");
1832         printf("Copyright 1996 Adam Twiss, Zeus Technology Ltd, http://www.zeustech.net/\n");
1833         printf("Licensed to The Apache Software Foundation, http://www.apache.org/\n");
1834         printf("\n");
1835     }
1836     else {
1837         printf("<p>\n");
1838         printf(" This is ApacheBench, Version %s <i>&lt;%s&gt;</i><br>\n", AP_AB_BASEREVISION, "$Revision$");
1839         printf(" Copyright 1996 Adam Twiss, Zeus Technology Ltd, http://www.zeustech.net/<br>\n");
1840         printf(" Licensed to The Apache Software Foundation, http://www.apache.org/<br>\n");
1841         printf("</p>\n<p>\n");
1842     }
1843 }
1844
1845 /* display usage information */
1846 static void usage(const char *progname)
1847 {
1848     fprintf(stderr, "Usage: %s [options] [http"
1849 #ifdef USE_SSL
1850         "[s]"
1851 #endif
1852         "://]hostname[:port]/path\n", progname);
1853 /* 80 column ruler:  ********************************************************************************
1854  */
1855     fprintf(stderr, "Options are:\n");
1856     fprintf(stderr, "    -n requests     Number of requests to perform\n");
1857     fprintf(stderr, "    -c concurrency  Number of multiple requests to make\n");
1858     fprintf(stderr, "    -t timelimit    Seconds to max. wait for responses\n");
1859     fprintf(stderr, "    -b windowsize   Size of TCP send/receive buffer, in bytes\n");
1860     fprintf(stderr, "    -p postfile     File containing data to POST. Remember also to set -T\n");
1861     fprintf(stderr, "    -u putfile      File containing data to PUT. Remember also to set -T\n");
1862     fprintf(stderr, "    -T content-type Content-type header for POSTing, eg.\n");
1863     fprintf(stderr, "                    'application/x-www-form-urlencoded'\n");
1864     fprintf(stderr, "                    Default is 'text/plain'\n");
1865     fprintf(stderr, "    -v verbosity    How much troubleshooting info to print\n");
1866     fprintf(stderr, "    -w              Print out results in HTML tables\n");
1867     fprintf(stderr, "    -i              Use HEAD instead of GET\n");
1868     fprintf(stderr, "    -x attributes   String to insert as table attributes\n");
1869     fprintf(stderr, "    -y attributes   String to insert as tr attributes\n");
1870     fprintf(stderr, "    -z attributes   String to insert as td or th attributes\n");
1871     fprintf(stderr, "    -C attribute    Add cookie, eg. 'Apache=1234. (repeatable)\n");
1872     fprintf(stderr, "    -H attribute    Add Arbitrary header line, eg. 'Accept-Encoding: gzip'\n");
1873     fprintf(stderr, "                    Inserted after all normal header lines. (repeatable)\n");
1874     fprintf(stderr, "    -A attribute    Add Basic WWW Authentication, the attributes\n");
1875     fprintf(stderr, "                    are a colon separated username and password.\n");
1876     fprintf(stderr, "    -P attribute    Add Basic Proxy Authentication, the attributes\n");
1877     fprintf(stderr, "                    are a colon separated username and password.\n");
1878     fprintf(stderr, "    -X proxy:port   Proxyserver and port number to use\n");
1879     fprintf(stderr, "    -V              Print version number and exit\n");
1880     fprintf(stderr, "    -k              Use HTTP KeepAlive feature\n");
1881     fprintf(stderr, "    -d              Do not show percentiles served table.\n");
1882     fprintf(stderr, "    -S              Do not show confidence estimators and warnings.\n");
1883     fprintf(stderr, "    -g filename     Output collected data to gnuplot format file.\n");
1884     fprintf(stderr, "    -e filename     Output CSV file with percentages served\n");
1885     fprintf(stderr, "    -r              Don't exit on socket receive errors.\n");
1886     fprintf(stderr, "    -h              Display usage information (this message)\n");
1887 #ifdef USE_SSL
1888     fprintf(stderr, "    -Z ciphersuite  Specify SSL/TLS cipher suite (See openssl ciphers)\n");
1889     fprintf(stderr, "    -f protocol     Specify SSL/TLS protocol (SSL2, SSL3, TLS1, or ALL)\n");
1890 #endif
1891     exit(EINVAL);
1892 }
1893
1894 /* ------------------------------------------------------- */
1895
1896 /* split URL into parts */
1897
1898 static int parse_url(char *url)
1899 {
1900     char *cp;
1901     char *h;
1902     char *scope_id;
1903     apr_status_t rv;
1904
1905     /* Save a copy for the proxy */
1906     fullurl = apr_pstrdup(cntxt, url);
1907
1908     if (strlen(url) > 7 && strncmp(url, "http://", 7) == 0) {
1909         url += 7;
1910 #ifdef USE_SSL
1911         is_ssl = 0;
1912 #endif
1913     }
1914     else
1915 #ifdef USE_SSL
1916     if (strlen(url) > 8 && strncmp(url, "https://", 8) == 0) {
1917         url += 8;
1918         is_ssl = 1;
1919     }
1920 #else
1921     if (strlen(url) > 8 && strncmp(url, "https://", 8) == 0) {
1922         fprintf(stderr, "SSL not compiled in; no https support\n");
1923         exit(1);
1924     }
1925 #endif
1926
1927     if ((cp = strchr(url, '/')) == NULL)
1928         return 1;
1929     h = apr_pstrmemdup(cntxt, url, cp - url);
1930     rv = apr_parse_addr_port(&hostname, &scope_id, &port, h, cntxt);
1931     if (rv != APR_SUCCESS || !hostname || scope_id) {
1932         return 1;
1933     }
1934     path = apr_pstrdup(cntxt, cp);
1935     *cp = '\0';
1936     if (*url == '[') {      /* IPv6 numeric address string */
1937         host_field = apr_psprintf(cntxt, "[%s]", hostname);
1938     }
1939     else {
1940         host_field = hostname;
1941     }
1942
1943     if (port == 0) {        /* no port specified */
1944 #ifdef USE_SSL
1945         if (is_ssl)
1946             port = 443;
1947         else
1948 #endif
1949         port = 80;
1950     }
1951
1952     if ((
1953 #ifdef USE_SSL
1954          is_ssl && (port != 443)) || (!is_ssl &&
1955 #endif
1956          (port != 80)))
1957     {
1958         colonhost = apr_psprintf(cntxt,":%d",port);
1959     } else
1960         colonhost = "";
1961     return 0;
1962 }
1963
1964 /* ------------------------------------------------------- */
1965
1966 /* read data to POST/PUT from file, save contents and length */
1967
1968 static apr_status_t open_postfile(const char *pfile)
1969 {
1970     apr_file_t *postfd;
1971     apr_finfo_t finfo;
1972     apr_status_t rv;
1973     char errmsg[120];
1974
1975     rv = apr_file_open(&postfd, pfile, APR_READ, APR_OS_DEFAULT, cntxt);
1976     if (rv != APR_SUCCESS) {
1977         fprintf(stderr, "ab: Could not open POST data file (%s): %s\n", pfile,
1978                 apr_strerror(rv, errmsg, sizeof errmsg));
1979         return rv;
1980     }
1981
1982     rv = apr_file_info_get(&finfo, APR_FINFO_NORM, postfd);
1983     if (rv != APR_SUCCESS) {
1984         fprintf(stderr, "ab: Could not stat POST data file (%s): %s\n", pfile,
1985                 apr_strerror(rv, errmsg, sizeof errmsg));
1986         return rv;
1987     }
1988     postlen = (apr_size_t)finfo.size;
1989     postdata = malloc(postlen);
1990     if (!postdata) {
1991         fprintf(stderr, "ab: Could not allocate POST data buffer\n");
1992         return APR_ENOMEM;
1993     }
1994     rv = apr_file_read_full(postfd, postdata, postlen, NULL);
1995     if (rv != APR_SUCCESS) {
1996         fprintf(stderr, "ab: Could not read POST data file: %s\n",
1997                 apr_strerror(rv, errmsg, sizeof errmsg));
1998         return rv;
1999     }
2000     apr_file_close(postfd);
2001     return APR_SUCCESS;
2002 }
2003
2004 /* ------------------------------------------------------- */
2005
2006 /* sort out command-line args and call test */
2007 int main(int argc, const char * const argv[])
2008 {
2009     int l;
2010     char tmp[1024];
2011     apr_status_t status;
2012     apr_getopt_t *opt;
2013     const char *optarg;
2014     char c;
2015 #ifdef USE_SSL
2016     AB_SSL_METHOD_CONST SSL_METHOD *meth = SSLv23_client_method();
2017 #endif
2018
2019     /* table defaults  */
2020     tablestring = "";
2021     trstring = "";
2022     tdstring = "bgcolor=white";
2023     cookie = "";
2024     auth = "";
2025     proxyhost[0] = '\0';
2026     hdrs = "";
2027
2028     apr_app_initialize(&argc, &argv, NULL);
2029     atexit(apr_terminate);
2030     apr_pool_create(&cntxt, NULL);
2031
2032 #ifdef NOT_ASCII
2033     status = apr_xlate_open(&to_ascii, "ISO-8859-1", APR_DEFAULT_CHARSET, cntxt);
2034     if (status) {
2035         fprintf(stderr, "apr_xlate_open(to ASCII)->%d\n", status);
2036         exit(1);
2037     }
2038     status = apr_xlate_open(&from_ascii, APR_DEFAULT_CHARSET, "ISO-8859-1", cntxt);
2039     if (status) {
2040         fprintf(stderr, "apr_xlate_open(from ASCII)->%d\n", status);
2041         exit(1);
2042     }
2043     status = apr_base64init_ebcdic(to_ascii, from_ascii);
2044     if (status) {
2045         fprintf(stderr, "apr_base64init_ebcdic()->%d\n", status);
2046         exit(1);
2047     }
2048 #endif
2049
2050     apr_getopt_init(&opt, cntxt, argc, argv);
2051     while ((status = apr_getopt(opt, "n:c:t:b:T:p:u:v:rkVhwix:y:z:C:H:P:A:g:X:de:Sq"
2052 #ifdef USE_SSL
2053             "Z:f:"
2054 #endif
2055             ,&c, &optarg)) == APR_SUCCESS) {
2056         switch (c) {
2057             case 'n':
2058                 requests = atoi(optarg);
2059                 if (requests <= 0) {
2060                     err("Invalid number of requests\n");
2061                 }
2062                 break;
2063             case 'k':
2064                 keepalive = 1;
2065                 break;
2066             case 'q':
2067                 heartbeatres = 0;
2068                 break;
2069             case 'c':
2070                 concurrency = atoi(optarg);
2071                 break;
2072             case 'b':
2073                 windowsize = atoi(optarg);
2074                 break;
2075             case 'i':
2076                 if (method != NO_METH)
2077                     err("Cannot mix HEAD with other methods\n");
2078                 method = HEAD;
2079                 break;
2080             case 'g':
2081                 gnuplot = strdup(optarg);
2082                 break;
2083             case 'd':
2084                 percentile = 0;
2085                 break;
2086             case 'e':
2087                 csvperc = strdup(optarg);
2088                 break;
2089             case 'S':
2090                 confidence = 0;
2091                 break;
2092             case 'p':
2093                 if (method != NO_METH)
2094                     err("Cannot mix POST with other methods\n");
2095                 if ((status = open_postfile(optarg)) != APR_SUCCESS) {
2096                     exit(1);
2097                 }
2098                 method = POST;
2099                 send_body = 1;
2100                 break;
2101             case 'u':
2102                 if (method != NO_METH)
2103                     err("Cannot mix PUT with other methods\n");
2104                 if ((status = open_postfile(optarg)) != APR_SUCCESS) {
2105                     exit(1);
2106                 }
2107                 method = PUT;
2108                 send_body = 1;
2109                 break;
2110             case 'r':
2111                 recverrok = 1;
2112                 break;
2113             case 'v':
2114                 verbosity = atoi(optarg);
2115                 break;
2116             case 't':
2117                 tlimit = atoi(optarg);
2118                 requests = MAX_REQUESTS;    /* need to size data array on
2119                                              * something */
2120                 break;
2121             case 'T':
2122                 strcpy(content_type, optarg);
2123                 break;
2124             case 'C':
2125                 cookie = apr_pstrcat(cntxt, "Cookie: ", optarg, "\r\n", NULL);
2126                 break;
2127             case 'A':
2128                 /*
2129                  * assume username passwd already to be in colon separated form.
2130                  * Ready to be uu-encoded.
2131                  */
2132                 while (apr_isspace(*optarg))
2133                     optarg++;
2134                 if (apr_base64_encode_len(strlen(optarg)) > sizeof(tmp)) {
2135                     err("Authentication credentials too long\n");
2136                 }
2137                 l = apr_base64_encode(tmp, optarg, strlen(optarg));
2138                 tmp[l] = '\0';
2139
2140                 auth = apr_pstrcat(cntxt, auth, "Authorization: Basic ", tmp,
2141                                        "\r\n", NULL);
2142                 break;
2143             case 'P':
2144                 /*
2145                  * assume username passwd already to be in colon separated form.
2146                  */
2147                 while (apr_isspace(*optarg))
2148                 optarg++;
2149                 if (apr_base64_encode_len(strlen(optarg)) > sizeof(tmp)) {
2150                     err("Proxy credentials too long\n");
2151                 }
2152                 l = apr_base64_encode(tmp, optarg, strlen(optarg));
2153                 tmp[l] = '\0';
2154
2155                 auth = apr_pstrcat(cntxt, auth, "Proxy-Authorization: Basic ",
2156                                        tmp, "\r\n", NULL);
2157                 break;
2158             case 'H':
2159                 hdrs = apr_pstrcat(cntxt, hdrs, optarg, "\r\n", NULL);
2160                 /*
2161                  * allow override of some of the common headers that ab adds
2162                  */
2163                 if (strncasecmp(optarg, "Host:", 5) == 0) {
2164                     opt_host = 1;
2165                 } else if (strncasecmp(optarg, "Accept:", 7) == 0) {
2166                     opt_accept = 1;
2167                 } else if (strncasecmp(optarg, "User-Agent:", 11) == 0) {
2168                     opt_useragent = 1;
2169                 }
2170                 break;
2171             case 'w':
2172                 use_html = 1;
2173                 break;
2174                 /*
2175                  * if any of the following three are used, turn on html output
2176                  * automatically
2177                  */
2178             case 'x':
2179                 use_html = 1;
2180                 tablestring = optarg;
2181                 break;
2182             case 'X':
2183                 {
2184                     char *p;
2185                     /*
2186                      * assume proxy-name[:port]
2187                      */
2188                     if ((p = strchr(optarg, ':'))) {
2189                         *p = '\0';
2190                         p++;
2191                         proxyport = atoi(p);
2192                     }
2193                     strcpy(proxyhost, optarg);
2194                     isproxy = 1;
2195                 }
2196                 break;
2197             case 'y':
2198                 use_html = 1;
2199                 trstring = optarg;
2200                 break;
2201             case 'z':
2202                 use_html = 1;
2203                 tdstring = optarg;
2204                 break;
2205             case 'h':
2206                 usage(argv[0]);
2207                 break;
2208             case 'V':
2209                 copyright();
2210                 return 0;
2211 #ifdef USE_SSL
2212             case 'Z':
2213                 ssl_cipher = strdup(optarg);
2214                 break;
2215             case 'f':
2216                 if (strncasecmp(optarg, "ALL", 3) == 0) {
2217                     meth = SSLv23_client_method();
2218                 } else if (strncasecmp(optarg, "SSL2", 4) == 0) {
2219                     meth = SSLv2_client_method();
2220                 } else if (strncasecmp(optarg, "SSL3", 4) == 0) {
2221                     meth = SSLv3_client_method();
2222                 } else if (strncasecmp(optarg, "TLS1", 4) == 0) {
2223                     meth = TLSv1_client_method();
2224                 }
2225                 break;
2226 #endif
2227         }
2228     }
2229
2230     if (opt->ind != argc - 1) {
2231         fprintf(stderr, "%s: wrong number of arguments\n", argv[0]);
2232         usage(argv[0]);
2233     }
2234
2235     if (method == NO_METH) {
2236         method = GET;
2237     }
2238
2239     if (parse_url(apr_pstrdup(cntxt, opt->argv[opt->ind++]))) {
2240         fprintf(stderr, "%s: invalid URL\n", argv[0]);
2241         usage(argv[0]);
2242     }
2243
2244     if ((concurrency < 0) || (concurrency > MAX_CONCURRENCY)) {
2245         fprintf(stderr, "%s: Invalid Concurrency [Range 0..%d]\n",
2246                 argv[0], MAX_CONCURRENCY);
2247         usage(argv[0]);
2248     }
2249
2250     if (concurrency > requests) {
2251         fprintf(stderr, "%s: Cannot use concurrency level greater than "
2252                 "total number of requests\n", argv[0]);
2253         usage(argv[0]);
2254     }
2255
2256     if ((heartbeatres) && (requests > 150)) {
2257         heartbeatres = requests / 10;   /* Print line every 10% of requests */
2258         if (heartbeatres < 100)
2259             heartbeatres = 100; /* but never more often than once every 100
2260                                  * connections. */
2261     }
2262     else
2263         heartbeatres = 0;
2264
2265 #ifdef USE_SSL
2266 #ifdef RSAREF
2267     R_malloc_init();
2268 #else
2269     CRYPTO_malloc_init();
2270 #endif
2271     SSL_load_error_strings();
2272     SSL_library_init();
2273     bio_out=BIO_new_fp(stdout,BIO_NOCLOSE);
2274     bio_err=BIO_new_fp(stderr,BIO_NOCLOSE);
2275
2276     if (!(ssl_ctx = SSL_CTX_new(meth))) {
2277         BIO_printf(bio_err, "Could not initialize SSL Context.\n");
2278         ERR_print_errors(bio_err);
2279         exit(1);
2280     }
2281     SSL_CTX_set_options(ssl_ctx, SSL_OP_ALL);
2282     if (ssl_cipher != NULL) {
2283         if (!SSL_CTX_set_cipher_list(ssl_ctx, ssl_cipher)) {
2284             fprintf(stderr, "error setting cipher list [%s]\n", ssl_cipher);
2285         ERR_print_errors_fp(stderr);
2286         exit(1);
2287     }
2288     }
2289     if (verbosity >= 3) {
2290         SSL_CTX_set_info_callback(ssl_ctx, ssl_state_cb);
2291     }
2292 #endif
2293 #ifdef SIGPIPE
2294     apr_signal(SIGPIPE, SIG_IGN);       /* Ignore writes to connections that
2295                                          * have been closed at the other end. */
2296 #endif
2297     copyright();
2298     test();
2299     apr_pool_destroy(cntxt);
2300
2301     return 0;
2302 }