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