2014-02-28 01:10:42 +01:00
|
|
|
/*
|
2014-03-30 12:09:21 +02:00
|
|
|
* nghttp2 - HTTP/2 C Library
|
2014-02-28 01:10:42 +01:00
|
|
|
*
|
|
|
|
* Copyright (c) 2014 Tatsuhiro Tsujikawa
|
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
|
|
* a copy of this software and associated documentation files (the
|
|
|
|
* "Software"), to deal in the Software without restriction, including
|
|
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
|
|
* the following conditions:
|
|
|
|
*
|
|
|
|
* The above copyright notice and this permission notice shall be
|
|
|
|
* included in all copies or substantial portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
|
|
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
|
|
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
|
|
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
*/
|
|
|
|
#include "h2load.h"
|
|
|
|
|
|
|
|
#include <getopt.h>
|
|
|
|
#include <signal.h>
|
2015-05-13 15:30:35 +02:00
|
|
|
#ifdef HAVE_NETINET_IN_H
|
2014-02-28 01:10:42 +01:00
|
|
|
#include <netinet/in.h>
|
2015-05-13 15:30:35 +02:00
|
|
|
#endif // HAVE_NETINET_IN_H
|
2014-02-28 01:10:42 +01:00
|
|
|
#include <netinet/tcp.h>
|
2015-03-26 11:53:42 +01:00
|
|
|
#include <sys/stat.h>
|
2015-05-13 15:30:35 +02:00
|
|
|
#ifdef HAVE_FCNTL_H
|
2015-03-26 11:53:42 +01:00
|
|
|
#include <fcntl.h>
|
2015-05-13 15:30:35 +02:00
|
|
|
#endif // HAVE_FCNTL_H
|
2014-02-28 01:10:42 +01:00
|
|
|
|
|
|
|
#include <cstdio>
|
|
|
|
#include <cassert>
|
|
|
|
#include <cstdlib>
|
|
|
|
#include <iostream>
|
2015-01-31 10:13:07 +01:00
|
|
|
#include <iomanip>
|
2014-10-20 22:59:55 +02:00
|
|
|
#include <fstream>
|
2014-02-28 01:10:42 +01:00
|
|
|
#include <chrono>
|
2014-03-03 17:12:11 +01:00
|
|
|
#include <thread>
|
2014-05-26 16:29:28 +02:00
|
|
|
#include <future>
|
2014-02-28 01:10:42 +01:00
|
|
|
|
|
|
|
#ifdef HAVE_SPDYLAY
|
|
|
|
#include <spdylay/spdylay.h>
|
|
|
|
#endif // HAVE_SPDYLAY
|
|
|
|
|
|
|
|
#include <openssl/err.h>
|
2014-08-02 03:11:45 +02:00
|
|
|
#include <openssl/conf.h>
|
2014-02-28 01:10:42 +01:00
|
|
|
|
|
|
|
#include "http-parser/http_parser.h"
|
|
|
|
|
|
|
|
#include "h2load_http2_session.h"
|
|
|
|
#ifdef HAVE_SPDYLAY
|
|
|
|
#include "h2load_spdy_session.h"
|
|
|
|
#endif // HAVE_SPDYLAY
|
2014-03-04 13:29:42 +01:00
|
|
|
#include "ssl.h"
|
2014-02-28 01:10:42 +01:00
|
|
|
#include "http2.h"
|
|
|
|
#include "util.h"
|
2015-02-05 15:21:53 +01:00
|
|
|
#include "template.h"
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2015-03-26 11:53:42 +01:00
|
|
|
#ifndef O_BINARY
|
|
|
|
#define O_BINARY (0)
|
|
|
|
#endif // O_BINARY
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
using namespace nghttp2;
|
|
|
|
|
|
|
|
namespace h2load {
|
|
|
|
|
|
|
|
Config::Config()
|
2015-03-26 11:53:42 +01:00
|
|
|
: data_length(-1), addrs(nullptr), nreqs(1), nclients(1), nthreads(1),
|
2015-04-27 14:23:01 +02:00
|
|
|
max_concurrent_streams(-1), window_bits(30), connection_window_bits(30),
|
2015-07-16 19:35:03 +02:00
|
|
|
rate(0), nconns(0), no_tls_proto(PROTO_HTTP2), data_fd(-1), port(0),
|
|
|
|
default_port(0), verbose(false), current_worker(0) {}
|
2015-03-26 11:53:42 +01:00
|
|
|
|
|
|
|
Config::~Config() {
|
|
|
|
freeaddrinfo(addrs);
|
2014-11-27 15:39:04 +01:00
|
|
|
|
2015-03-26 11:53:42 +01:00
|
|
|
if (data_fd != -1) {
|
|
|
|
close(data_fd);
|
|
|
|
}
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2015-07-14 20:16:37 +02:00
|
|
|
bool Config::is_rate_mode() { return (this->rate != 0); }
|
2015-07-13 20:28:15 +02:00
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
Config config;
|
|
|
|
|
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
void debug(const char *format, ...) {
|
|
|
|
if (config.verbose) {
|
2014-02-28 01:10:42 +01:00
|
|
|
fprintf(stderr, "[DEBUG] ");
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
vfprintf(stderr, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
namespace {
|
|
|
|
void debug_nextproto_error() {
|
|
|
|
#ifdef HAVE_SPDYLAY
|
|
|
|
debug("no supported protocol was negotiated, expected: %s, "
|
|
|
|
"spdy/2, spdy/3, spdy/3.1\n",
|
|
|
|
NGHTTP2_PROTO_VERSION_ID);
|
|
|
|
#else // !HAVE_SPDYLAY
|
|
|
|
debug("no supported protocol was negotiated, expected: %s\n",
|
|
|
|
NGHTTP2_PROTO_VERSION_ID);
|
|
|
|
#endif // !HAVE_SPDYLAY
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2015-03-26 11:53:42 +01:00
|
|
|
RequestStat::RequestStat() : data_offset(0), completed(false) {}
|
2015-01-31 10:13:07 +01:00
|
|
|
|
|
|
|
Stats::Stats(size_t req_todo)
|
|
|
|
: req_todo(0), req_started(0), req_done(0), req_success(0),
|
|
|
|
req_status_success(0), req_failed(0), req_error(0), bytes_total(0),
|
|
|
|
bytes_head(0), bytes_body(0), status(), req_stats(req_todo) {}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
Stream::Stream() : status_success(-1) {}
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
namespace {
|
|
|
|
void writecb(struct ev_loop *loop, ev_io *w, int revents) {
|
|
|
|
auto client = static_cast<Client *>(w->data);
|
2015-01-04 17:56:02 +01:00
|
|
|
auto rv = client->do_write();
|
|
|
|
if (rv == Client::ERR_CONNECT_FAIL) {
|
|
|
|
client->disconnect();
|
|
|
|
rv = client->connect();
|
|
|
|
if (rv != 0) {
|
|
|
|
client->fail();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if (rv != 0) {
|
2014-12-27 18:59:06 +01:00
|
|
|
client->fail();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2015-02-10 16:44:30 +01:00
|
|
|
namespace {
|
|
|
|
void readcb(struct ev_loop *loop, ev_io *w, int revents) {
|
|
|
|
auto client = static_cast<Client *>(w->data);
|
|
|
|
if (client->do_read() != 0) {
|
|
|
|
client->fail();
|
|
|
|
return;
|
|
|
|
}
|
2015-03-28 11:00:59 +01:00
|
|
|
writecb(loop, &client->wev, revents);
|
|
|
|
// client->disconnect() and client->fail() may be called
|
2015-02-10 16:44:30 +01:00
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-06-22 04:33:16 +02:00
|
|
|
Client::Client(Worker *worker, size_t req_todo)
|
2014-12-27 18:59:06 +01:00
|
|
|
: worker(worker), ssl(nullptr), next_addr(config.addrs), reqidx(0),
|
2015-05-03 09:47:32 +02:00
|
|
|
state(CLIENT_IDLE), first_byte_received(false), req_todo(req_todo),
|
|
|
|
req_started(0), req_done(0), fd(-1) {
|
2014-12-27 18:59:06 +01:00
|
|
|
ev_io_init(&wev, writecb, 0, EV_WRITE);
|
|
|
|
ev_io_init(&rev, readcb, 0, EV_READ);
|
|
|
|
|
|
|
|
wev.data = this;
|
|
|
|
rev.data = this;
|
|
|
|
}
|
2014-11-27 15:39:04 +01:00
|
|
|
|
|
|
|
Client::~Client() { disconnect(); }
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
int Client::do_read() { return readfn(*this); }
|
|
|
|
int Client::do_write() { return writefn(*this); }
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
int Client::connect() {
|
2015-04-30 00:29:37 +02:00
|
|
|
record_start_time(&worker->stats);
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
while (next_addr) {
|
|
|
|
auto addr = next_addr;
|
|
|
|
next_addr = next_addr->ai_next;
|
|
|
|
fd = util::create_nonblock_socket(addr->ai_family);
|
|
|
|
if (fd == -1) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (config.scheme == "https") {
|
|
|
|
ssl = SSL_new(worker->ssl_ctx);
|
2014-03-15 07:32:38 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
auto config = worker->config;
|
2014-03-15 07:32:38 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
if (!util::numeric_host(config->host.c_str())) {
|
|
|
|
SSL_set_tlsext_host_name(ssl, config->host.c_str());
|
|
|
|
}
|
2014-03-15 07:32:38 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
SSL_set_fd(ssl, fd);
|
|
|
|
SSL_set_connect_state(ssl);
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
auto rv = ::connect(fd, addr->ai_addr, addr->ai_addrlen);
|
|
|
|
if (rv != 0 && errno != EINPROGRESS) {
|
|
|
|
if (ssl) {
|
|
|
|
SSL_free(ssl);
|
|
|
|
ssl = nullptr;
|
|
|
|
}
|
|
|
|
close(fd);
|
|
|
|
fd = -1;
|
|
|
|
continue;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
break;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
if (fd == -1) {
|
2014-02-28 01:10:42 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
writefn = &Client::connected;
|
|
|
|
|
|
|
|
ev_io_set(&rev, fd, EV_READ);
|
|
|
|
ev_io_set(&wev, fd, EV_WRITE);
|
|
|
|
|
|
|
|
ev_io_start(worker->loop, &wev);
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::fail() {
|
2014-02-28 01:10:42 +01:00
|
|
|
process_abandoned_streams();
|
2014-06-26 16:20:12 +02:00
|
|
|
|
|
|
|
disconnect();
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::disconnect() {
|
2014-02-28 01:10:42 +01:00
|
|
|
streams.clear();
|
|
|
|
session.reset();
|
|
|
|
state = CLIENT_IDLE;
|
2014-12-27 18:59:06 +01:00
|
|
|
ev_io_stop(worker->loop, &wev);
|
|
|
|
ev_io_stop(worker->loop, &rev);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (ssl) {
|
2014-02-28 01:10:42 +01:00
|
|
|
SSL_set_shutdown(ssl, SSL_RECEIVED_SHUTDOWN);
|
2015-01-09 01:15:01 +01:00
|
|
|
ERR_clear_error();
|
2014-02-28 01:10:42 +01:00
|
|
|
SSL_shutdown(ssl);
|
|
|
|
SSL_free(ssl);
|
|
|
|
ssl = nullptr;
|
|
|
|
}
|
2014-11-27 15:39:04 +01:00
|
|
|
if (fd != -1) {
|
2014-02-28 01:10:42 +01:00
|
|
|
shutdown(fd, SHUT_WR);
|
|
|
|
close(fd);
|
2014-12-27 18:59:06 +01:00
|
|
|
fd = -1;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::submit_request() {
|
2015-01-31 10:13:07 +01:00
|
|
|
auto req_stat = &worker->stats.req_stats[worker->stats.req_started++];
|
|
|
|
session->submit_request(req_stat);
|
2014-06-22 04:33:16 +02:00
|
|
|
++req_started;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::process_abandoned_streams() {
|
2014-06-22 04:33:16 +02:00
|
|
|
auto req_abandoned = req_todo - req_done;
|
|
|
|
|
|
|
|
worker->stats.req_failed += req_abandoned;
|
|
|
|
worker->stats.req_error += req_abandoned;
|
|
|
|
worker->stats.req_done += req_abandoned;
|
2014-11-06 16:28:10 +01:00
|
|
|
|
|
|
|
req_done = req_todo;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::report_progress() {
|
2015-07-13 20:28:15 +02:00
|
|
|
if (!worker->config->is_rate_mode() && worker->id == 0 &&
|
2014-11-27 15:39:04 +01:00
|
|
|
worker->stats.req_done % worker->progress_interval == 0) {
|
2014-02-28 01:10:42 +01:00
|
|
|
std::cout << "progress: "
|
|
|
|
<< worker->stats.req_done * 100 / worker->stats.req_todo
|
2014-11-27 15:39:04 +01:00
|
|
|
<< "% done" << std::endl;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-08 13:24:24 +01:00
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
void print_server_tmp_key(SSL *ssl) {
|
2015-06-17 11:24:14 +02:00
|
|
|
// libressl does not have SSL_get_server_tmp_key
|
|
|
|
#if OPENSSL_VERSION_NUMBER >= 0x10002000L && defined(SSL_get_server_tmp_key)
|
2014-11-08 13:24:24 +01:00
|
|
|
EVP_PKEY *key;
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (!SSL_get_server_tmp_key(ssl, &key)) {
|
2014-11-08 13:24:24 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2015-02-06 15:27:15 +01:00
|
|
|
auto key_del = defer(EVP_PKEY_free, key);
|
2014-11-08 13:24:24 +01:00
|
|
|
|
|
|
|
std::cout << "Server Temp Key: ";
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
switch (EVP_PKEY_id(key)) {
|
2014-11-08 13:24:24 +01:00
|
|
|
case EVP_PKEY_RSA:
|
|
|
|
std::cout << "RSA " << EVP_PKEY_bits(key) << " bits" << std::endl;
|
|
|
|
break;
|
|
|
|
case EVP_PKEY_DH:
|
|
|
|
std::cout << "DH " << EVP_PKEY_bits(key) << " bits" << std::endl;
|
|
|
|
break;
|
|
|
|
case EVP_PKEY_EC: {
|
|
|
|
auto ec = EVP_PKEY_get1_EC_KEY(key);
|
2015-02-06 15:27:15 +01:00
|
|
|
auto ec_del = defer(EC_KEY_free, ec);
|
2014-11-08 13:24:24 +01:00
|
|
|
auto nid = EC_GROUP_get_curve_name(EC_KEY_get0_group(ec));
|
|
|
|
auto cname = EC_curve_nid2nist(nid);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (!cname) {
|
2014-11-08 13:24:24 +01:00
|
|
|
cname = OBJ_nid2sn(nid);
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
std::cout << "ECDH " << cname << " " << EVP_PKEY_bits(key) << " bits"
|
|
|
|
<< std::endl;
|
2014-11-08 13:24:24 +01:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif // OPENSSL_VERSION_NUMBER >= 0x10002000L
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::report_tls_info() {
|
|
|
|
if (worker->id == 0 && !worker->tls_info_report_done) {
|
2014-11-08 13:24:24 +01:00
|
|
|
worker->tls_info_report_done = true;
|
|
|
|
auto cipher = SSL_get_current_cipher(ssl);
|
2015-06-28 09:37:17 +02:00
|
|
|
std::cout << "Protocol: " << ssl::get_tls_protocol(ssl) << "\n"
|
2014-11-08 13:24:24 +01:00
|
|
|
<< "Cipher: " << SSL_CIPHER_get_name(cipher) << std::endl;
|
|
|
|
print_server_tmp_key(ssl);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::terminate_session() { session->terminate(); }
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::on_request(int32_t stream_id) { streams[stream_id] = Stream(); }
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
void Client::on_header(int32_t stream_id, const uint8_t *name, size_t namelen,
|
|
|
|
const uint8_t *value, size_t valuelen) {
|
2014-02-28 01:10:42 +01:00
|
|
|
auto itr = streams.find(stream_id);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (itr == std::end(streams)) {
|
2014-02-28 01:10:42 +01:00
|
|
|
return;
|
|
|
|
}
|
2014-11-27 15:39:04 +01:00
|
|
|
auto &stream = (*itr).second;
|
|
|
|
if (stream.status_success == -1 && namelen == 7 &&
|
2015-02-20 15:57:40 +01:00
|
|
|
util::streq_l(":status", name, namelen)) {
|
2014-02-28 01:10:42 +01:00
|
|
|
int status = 0;
|
2014-11-27 15:39:04 +01:00
|
|
|
for (size_t i = 0; i < valuelen; ++i) {
|
|
|
|
if ('0' <= value[i] && value[i] <= '9') {
|
2014-02-28 01:10:42 +01:00
|
|
|
status *= 10;
|
|
|
|
status += value[i] - '0';
|
2014-11-27 15:39:04 +01:00
|
|
|
if (status > 999) {
|
2014-02-28 01:10:42 +01:00
|
|
|
stream.status_success = 0;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (status >= 200 && status < 300) {
|
2014-02-28 01:10:42 +01:00
|
|
|
++worker->stats.status[2];
|
|
|
|
stream.status_success = 1;
|
2014-11-27 15:39:04 +01:00
|
|
|
} else if (status < 400) {
|
2014-02-28 01:10:42 +01:00
|
|
|
++worker->stats.status[3];
|
|
|
|
stream.status_success = 1;
|
2014-11-27 15:39:04 +01:00
|
|
|
} else if (status < 600) {
|
2014-02-28 01:10:42 +01:00
|
|
|
++worker->stats.status[status / 100];
|
|
|
|
stream.status_success = 0;
|
|
|
|
} else {
|
|
|
|
stream.status_success = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-01-31 10:13:07 +01:00
|
|
|
void Client::on_stream_close(int32_t stream_id, bool success,
|
|
|
|
RequestStat *req_stat) {
|
2015-01-31 15:54:03 +01:00
|
|
|
req_stat->stream_close_time = std::chrono::steady_clock::now();
|
2015-01-31 10:13:07 +01:00
|
|
|
if (success) {
|
|
|
|
req_stat->completed = true;
|
|
|
|
++worker->stats.req_success;
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
++worker->stats.req_done;
|
2014-06-22 04:33:16 +02:00
|
|
|
++req_done;
|
2014-11-27 15:39:04 +01:00
|
|
|
if (success && streams[stream_id].status_success == 1) {
|
2015-01-31 10:13:07 +01:00
|
|
|
++worker->stats.req_status_success;
|
2014-02-28 01:10:42 +01:00
|
|
|
} else {
|
|
|
|
++worker->stats.req_failed;
|
|
|
|
}
|
|
|
|
report_progress();
|
|
|
|
streams.erase(stream_id);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (req_done == req_todo) {
|
2014-11-06 16:28:10 +01:00
|
|
|
terminate_session();
|
2014-02-28 01:10:42 +01:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (req_started < req_todo) {
|
2014-02-28 01:10:42 +01:00
|
|
|
submit_request();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-03-28 12:13:37 +01:00
|
|
|
int Client::connection_made() {
|
2014-12-27 18:59:06 +01:00
|
|
|
if (ssl) {
|
|
|
|
report_tls_info();
|
|
|
|
|
|
|
|
const unsigned char *next_proto = nullptr;
|
|
|
|
unsigned int next_proto_len;
|
|
|
|
SSL_get0_next_proto_negotiated(ssl, &next_proto, &next_proto_len);
|
|
|
|
for (int i = 0; i < 2; ++i) {
|
|
|
|
if (next_proto) {
|
|
|
|
if (util::check_h2_is_selected(next_proto, next_proto_len)) {
|
2015-02-05 15:21:53 +01:00
|
|
|
session = make_unique<Http2Session>(this);
|
2015-05-21 17:48:32 +02:00
|
|
|
break;
|
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
#ifdef HAVE_SPDYLAY
|
2015-05-21 17:48:32 +02:00
|
|
|
else {
|
2014-12-27 18:59:06 +01:00
|
|
|
auto spdy_version =
|
|
|
|
spdylay_npn_get_version(next_proto, next_proto_len);
|
|
|
|
if (spdy_version) {
|
2015-02-05 15:21:53 +01:00
|
|
|
session = make_unique<SpdySession>(this, spdy_version);
|
2015-05-21 17:48:32 +02:00
|
|
|
break;
|
2014-12-27 18:59:06 +01:00
|
|
|
}
|
|
|
|
}
|
2015-05-21 17:48:32 +02:00
|
|
|
#endif // HAVE_SPDYLAY
|
|
|
|
|
|
|
|
next_proto = nullptr;
|
|
|
|
break;
|
2014-12-27 18:59:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
#if OPENSSL_VERSION_NUMBER >= 0x10002000L
|
|
|
|
SSL_get0_alpn_selected(ssl, &next_proto, &next_proto_len);
|
|
|
|
#else // OPENSSL_VERSION_NUMBER < 0x10002000L
|
|
|
|
break;
|
|
|
|
#endif // OPENSSL_VERSION_NUMBER < 0x10002000L
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!next_proto) {
|
|
|
|
debug_nextproto_error();
|
|
|
|
fail();
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
switch (config.no_tls_proto) {
|
|
|
|
case Config::PROTO_HTTP2:
|
2015-02-05 15:21:53 +01:00
|
|
|
session = make_unique<Http2Session>(this);
|
2014-12-27 18:59:06 +01:00
|
|
|
break;
|
|
|
|
#ifdef HAVE_SPDYLAY
|
|
|
|
case Config::PROTO_SPDY2:
|
2015-02-05 15:21:53 +01:00
|
|
|
session = make_unique<SpdySession>(this, SPDYLAY_PROTO_SPDY2);
|
2014-12-27 18:59:06 +01:00
|
|
|
break;
|
|
|
|
case Config::PROTO_SPDY3:
|
2015-02-05 15:21:53 +01:00
|
|
|
session = make_unique<SpdySession>(this, SPDYLAY_PROTO_SPDY3);
|
2014-12-27 18:59:06 +01:00
|
|
|
break;
|
|
|
|
case Config::PROTO_SPDY3_1:
|
2015-02-05 15:21:53 +01:00
|
|
|
session = make_unique<SpdySession>(this, SPDYLAY_PROTO_SPDY3_1);
|
2014-12-27 18:59:06 +01:00
|
|
|
break;
|
|
|
|
#endif // HAVE_SPDYLAY
|
|
|
|
default:
|
|
|
|
// unreachable
|
|
|
|
assert(0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
state = CLIENT_CONNECTED;
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
session->on_connect();
|
|
|
|
|
2015-04-30 00:29:37 +02:00
|
|
|
record_connect_time(&worker->stats);
|
2015-05-03 09:47:32 +02:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
auto nreq =
|
|
|
|
std::min(req_todo - req_started, (size_t)config.max_concurrent_streams);
|
2014-06-22 04:33:16 +02:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
for (; nreq > 0; --nreq) {
|
2014-02-28 01:10:42 +01:00
|
|
|
submit_request();
|
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
signal_write();
|
|
|
|
|
|
|
|
return 0;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
int Client::on_read(const uint8_t *data, size_t len) {
|
|
|
|
auto rv = session->on_read(data, len);
|
|
|
|
if (rv != 0) {
|
2014-02-28 01:10:42 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2015-01-05 17:24:03 +01:00
|
|
|
worker->stats.bytes_total += len;
|
2014-12-27 18:59:06 +01:00
|
|
|
signal_write();
|
|
|
|
return 0;
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
int Client::on_write() {
|
|
|
|
if (session->on_write() != 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
int Client::read_clear() {
|
2015-06-21 07:32:47 +02:00
|
|
|
uint8_t buf[8_k];
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
for (;;) {
|
|
|
|
ssize_t nread;
|
|
|
|
while ((nread = read(fd, buf, sizeof(buf))) == -1 && errno == EINTR)
|
|
|
|
;
|
|
|
|
if (nread == -1) {
|
|
|
|
if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
|
|
|
return 0;
|
|
|
|
}
|
2015-01-10 08:02:15 +01:00
|
|
|
return -1;
|
2014-12-27 18:59:06 +01:00
|
|
|
}
|
2014-06-22 04:33:16 +02:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
if (nread == 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
2014-06-22 04:33:16 +02:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
if (on_read(buf, nread) != 0) {
|
|
|
|
return -1;
|
2014-06-22 04:33:16 +02:00
|
|
|
}
|
2015-05-03 09:47:32 +02:00
|
|
|
|
2015-04-30 00:29:37 +02:00
|
|
|
if (!first_byte_received) {
|
|
|
|
first_byte_received = true;
|
2015-05-04 04:48:21 +02:00
|
|
|
record_ttfb(&worker->stats);
|
2015-05-03 09:47:32 +02:00
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
return 0;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
int Client::write_clear() {
|
|
|
|
for (;;) {
|
|
|
|
if (wb.rleft() > 0) {
|
|
|
|
ssize_t nwrite;
|
2015-02-06 15:40:34 +01:00
|
|
|
while ((nwrite = write(fd, wb.pos, wb.rleft())) == -1 && errno == EINTR)
|
2014-12-27 18:59:06 +01:00
|
|
|
;
|
|
|
|
if (nwrite == -1) {
|
|
|
|
if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
|
|
|
ev_io_start(worker->loop, &wev);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
wb.drain(nwrite);
|
|
|
|
continue;
|
|
|
|
}
|
2015-02-11 13:42:11 +01:00
|
|
|
wb.reset();
|
2014-12-27 18:59:06 +01:00
|
|
|
if (on_write() != 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (wb.rleft() == 0) {
|
|
|
|
break;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
ev_io_stop(worker->loop, &wev);
|
|
|
|
|
|
|
|
return 0;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
int Client::connected() {
|
2015-01-04 17:56:02 +01:00
|
|
|
if (!util::check_socket_connected(fd)) {
|
|
|
|
return ERR_CONNECT_FAIL;
|
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
ev_io_start(worker->loop, &rev);
|
|
|
|
ev_io_stop(worker->loop, &wev);
|
|
|
|
|
|
|
|
if (ssl) {
|
|
|
|
readfn = &Client::tls_handshake;
|
|
|
|
writefn = &Client::tls_handshake;
|
|
|
|
|
|
|
|
return do_write();
|
|
|
|
}
|
|
|
|
|
|
|
|
readfn = &Client::read_clear;
|
|
|
|
writefn = &Client::write_clear;
|
|
|
|
|
2015-03-28 12:13:37 +01:00
|
|
|
if (connection_made() != 0) {
|
2014-12-27 18:59:06 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
int Client::tls_handshake() {
|
2015-01-09 01:10:59 +01:00
|
|
|
ERR_clear_error();
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
auto rv = SSL_do_handshake(ssl);
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
if (rv == 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
if (rv < 0) {
|
|
|
|
auto err = SSL_get_error(ssl, rv);
|
|
|
|
switch (err) {
|
|
|
|
case SSL_ERROR_WANT_READ:
|
|
|
|
ev_io_stop(worker->loop, &wev);
|
|
|
|
return 0;
|
|
|
|
case SSL_ERROR_WANT_WRITE:
|
|
|
|
ev_io_start(worker->loop, &wev);
|
|
|
|
return 0;
|
|
|
|
default:
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ev_io_stop(worker->loop, &wev);
|
|
|
|
|
|
|
|
readfn = &Client::read_tls;
|
|
|
|
writefn = &Client::write_tls;
|
|
|
|
|
2015-03-28 12:13:37 +01:00
|
|
|
if (connection_made() != 0) {
|
2014-12-27 18:59:06 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int Client::read_tls() {
|
2015-06-21 07:32:47 +02:00
|
|
|
uint8_t buf[8_k];
|
2015-01-09 01:10:59 +01:00
|
|
|
|
|
|
|
ERR_clear_error();
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
for (;;) {
|
|
|
|
auto rv = SSL_read(ssl, buf, sizeof(buf));
|
|
|
|
|
|
|
|
if (rv == 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (rv < 0) {
|
|
|
|
auto err = SSL_get_error(ssl, rv);
|
|
|
|
switch (err) {
|
|
|
|
case SSL_ERROR_WANT_READ:
|
|
|
|
return 0;
|
|
|
|
case SSL_ERROR_WANT_WRITE:
|
2015-01-09 01:10:59 +01:00
|
|
|
// renegotiation started
|
|
|
|
return -1;
|
2014-04-21 14:35:45 +02:00
|
|
|
default:
|
2014-12-27 18:59:06 +01:00
|
|
|
return -1;
|
2014-04-21 14:35:45 +02:00
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
if (on_read(buf, rv) != 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
2015-05-03 09:47:32 +02:00
|
|
|
|
2015-04-30 00:29:37 +02:00
|
|
|
if (!first_byte_received) {
|
|
|
|
first_byte_received = true;
|
2015-05-04 04:48:21 +02:00
|
|
|
record_ttfb(&worker->stats);
|
2015-05-03 09:47:32 +02:00
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
int Client::write_tls() {
|
2015-01-09 01:10:59 +01:00
|
|
|
ERR_clear_error();
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
for (;;) {
|
|
|
|
if (wb.rleft() > 0) {
|
2015-02-06 15:40:34 +01:00
|
|
|
auto rv = SSL_write(ssl, wb.pos, wb.rleft());
|
2014-12-27 18:59:06 +01:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (rv == 0) {
|
2014-12-27 18:59:06 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (rv < 0) {
|
|
|
|
auto err = SSL_get_error(ssl, rv);
|
|
|
|
switch (err) {
|
|
|
|
case SSL_ERROR_WANT_READ:
|
2015-01-09 01:10:59 +01:00
|
|
|
// renegotiation started
|
|
|
|
return -1;
|
2014-12-27 18:59:06 +01:00
|
|
|
case SSL_ERROR_WANT_WRITE:
|
|
|
|
ev_io_start(worker->loop, &wev);
|
|
|
|
return 0;
|
|
|
|
default:
|
|
|
|
return -1;
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
wb.drain(rv);
|
|
|
|
|
|
|
|
continue;
|
|
|
|
}
|
2015-02-11 13:42:11 +01:00
|
|
|
wb.reset();
|
2014-12-27 18:59:06 +01:00
|
|
|
if (on_write() != 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
if (wb.rleft() == 0) {
|
|
|
|
break;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
|
|
|
|
ev_io_stop(worker->loop, &wev);
|
|
|
|
|
|
|
|
return 0;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2015-01-31 10:13:07 +01:00
|
|
|
void Client::record_request_time(RequestStat *req_stat) {
|
|
|
|
req_stat->request_time = std::chrono::steady_clock::now();
|
|
|
|
}
|
|
|
|
|
2015-04-30 00:29:37 +02:00
|
|
|
void Client::record_start_time(Stats *stat) {
|
|
|
|
stat->start_times.push_back(std::chrono::steady_clock::now());
|
|
|
|
}
|
|
|
|
|
|
|
|
void Client::record_connect_time(Stats *stat) {
|
|
|
|
stat->connect_times.push_back(std::chrono::steady_clock::now());
|
|
|
|
}
|
|
|
|
|
2015-05-04 04:48:21 +02:00
|
|
|
void Client::record_ttfb(Stats *stat) {
|
|
|
|
stat->ttfbs.push_back(std::chrono::steady_clock::now());
|
2015-04-30 00:29:37 +02:00
|
|
|
}
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
void Client::signal_write() { ev_io_start(worker->loop, &wev); }
|
|
|
|
|
|
|
|
Worker::Worker(uint32_t id, SSL_CTX *ssl_ctx, size_t req_todo, size_t nclients,
|
|
|
|
Config *config)
|
2015-01-31 10:13:07 +01:00
|
|
|
: stats(req_todo), loop(ev_loop_new(0)), ssl_ctx(ssl_ctx), config(config),
|
|
|
|
id(id), tls_info_report_done(false) {
|
2014-12-27 18:59:06 +01:00
|
|
|
stats.req_todo = req_todo;
|
|
|
|
progress_interval = std::max((size_t)1, req_todo / 10);
|
|
|
|
|
|
|
|
auto nreqs_per_client = req_todo / nclients;
|
|
|
|
auto nreqs_rem = req_todo % nclients;
|
|
|
|
|
|
|
|
for (size_t i = 0; i < nclients; ++i) {
|
|
|
|
auto req_todo = nreqs_per_client;
|
|
|
|
if (nreqs_rem > 0) {
|
|
|
|
++req_todo;
|
|
|
|
--nreqs_rem;
|
|
|
|
}
|
2015-02-05 15:21:53 +01:00
|
|
|
clients.push_back(make_unique<Client>(this, req_todo));
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-12-27 18:59:06 +01:00
|
|
|
Worker::~Worker() {
|
|
|
|
// first clear clients so that io watchers are stopped before
|
|
|
|
// destructing ev_loop.
|
|
|
|
clients.clear();
|
|
|
|
ev_loop_destroy(loop);
|
|
|
|
}
|
|
|
|
|
|
|
|
void Worker::run() {
|
|
|
|
for (auto &client : clients) {
|
|
|
|
if (client->connect() != 0) {
|
|
|
|
std::cerr << "client could not connect to host" << std::endl;
|
|
|
|
client->fail();
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-12-27 18:59:06 +01:00
|
|
|
ev_run(loop, 0);
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2015-01-31 10:13:07 +01:00
|
|
|
namespace {
|
2015-05-04 04:48:21 +02:00
|
|
|
// Returns percentage of number of samples within mean +/- sd.
|
|
|
|
template <typename Duration>
|
|
|
|
double within_sd(const std::vector<Duration> &samples, const Duration &mean,
|
|
|
|
const Duration &sd) {
|
|
|
|
if (samples.size() == 0) {
|
|
|
|
return 0.0;
|
|
|
|
}
|
|
|
|
auto lower = mean - sd;
|
|
|
|
auto upper = mean + sd;
|
|
|
|
auto m = std::count_if(
|
|
|
|
std::begin(samples), std::end(samples),
|
|
|
|
[&lower, &upper](const Duration &t) { return lower <= t && t <= upper; });
|
|
|
|
return (m / static_cast<double>(samples.size())) * 100;
|
2015-01-31 10:13:07 +01:00
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
|
|
|
namespace {
|
2015-05-04 04:48:21 +02:00
|
|
|
// Computes statistics using |samples|. The min, max, mean, sd, and
|
|
|
|
// percentage of number of samples within mean +/- sd are computed.
|
|
|
|
template <typename Duration>
|
|
|
|
TimeStat<Duration> compute_time_stat(const std::vector<Duration> &samples) {
|
2015-06-09 17:28:27 +02:00
|
|
|
if (samples.empty()) {
|
2015-05-04 04:48:21 +02:00
|
|
|
return {Duration::zero(), Duration::zero(), Duration::zero(),
|
|
|
|
Duration::zero(), 0.0};
|
|
|
|
}
|
|
|
|
// standard deviation calculated using Rapid calculation method:
|
|
|
|
// http://en.wikipedia.org/wiki/Standard_deviation#Rapid_calculation_methods
|
|
|
|
double a = 0, q = 0;
|
2015-01-31 10:13:07 +01:00
|
|
|
size_t n = 0;
|
2015-05-04 04:48:21 +02:00
|
|
|
int64_t sum = 0;
|
|
|
|
auto res = TimeStat<Duration>{Duration::max(), Duration::min()};
|
|
|
|
for (const auto &t : samples) {
|
|
|
|
++n;
|
|
|
|
res.min = std::min(res.min, t);
|
|
|
|
res.max = std::max(res.max, t);
|
|
|
|
sum += t.count();
|
2015-01-31 10:13:07 +01:00
|
|
|
|
2015-05-04 04:48:21 +02:00
|
|
|
auto na = a + (t.count() - a) / n;
|
|
|
|
q += (t.count() - a) * (t.count() - na);
|
|
|
|
a = na;
|
|
|
|
}
|
2015-04-30 00:29:37 +02:00
|
|
|
|
2015-06-09 17:29:03 +02:00
|
|
|
assert(n > 0);
|
2015-05-04 04:48:21 +02:00
|
|
|
res.mean = Duration(sum / n);
|
|
|
|
res.sd = Duration(static_cast<typename Duration::rep>(sqrt(q / n)));
|
|
|
|
res.within_sd = within_sd(samples, res.mean, res.sd);
|
2015-04-30 00:29:37 +02:00
|
|
|
|
2015-05-04 04:48:21 +02:00
|
|
|
return res;
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
TimeStats
|
|
|
|
process_time_stats(const std::vector<std::unique_ptr<Worker>> &workers) {
|
|
|
|
size_t nrequest_times = 0, nttfb_times = 0;
|
|
|
|
for (const auto &w : workers) {
|
|
|
|
nrequest_times += w->stats.req_stats.size();
|
|
|
|
nttfb_times += w->stats.ttfbs.size();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<std::chrono::microseconds> request_times;
|
|
|
|
request_times.reserve(nrequest_times);
|
|
|
|
std::vector<std::chrono::microseconds> connect_times, ttfb_times;
|
|
|
|
connect_times.reserve(nttfb_times);
|
|
|
|
ttfb_times.reserve(nttfb_times);
|
2015-01-31 10:13:07 +01:00
|
|
|
|
|
|
|
for (const auto &w : workers) {
|
|
|
|
for (const auto &req_stat : w->stats.req_stats) {
|
|
|
|
if (!req_stat.completed) {
|
|
|
|
continue;
|
|
|
|
}
|
2015-05-04 04:48:21 +02:00
|
|
|
request_times.push_back(
|
|
|
|
std::chrono::duration_cast<std::chrono::microseconds>(
|
|
|
|
req_stat.stream_close_time - req_stat.request_time));
|
2015-04-30 00:29:37 +02:00
|
|
|
}
|
2015-05-04 04:48:21 +02:00
|
|
|
|
|
|
|
const auto &stat = w->stats;
|
|
|
|
// rule out cases where we started but didn't connect or get the
|
|
|
|
// first byte (errors). We will get connect event before FFTB.
|
|
|
|
assert(stat.start_times.size() >= stat.ttfbs.size());
|
|
|
|
assert(stat.connect_times.size() >= stat.ttfbs.size());
|
|
|
|
for (size_t i = 0; i < stat.ttfbs.size(); ++i) {
|
|
|
|
connect_times.push_back(
|
|
|
|
std::chrono::duration_cast<std::chrono::microseconds>(
|
|
|
|
stat.connect_times[i] - stat.start_times[i]));
|
|
|
|
|
|
|
|
ttfb_times.push_back(
|
|
|
|
std::chrono::duration_cast<std::chrono::microseconds>(
|
|
|
|
stat.ttfbs[i] - stat.start_times[i]));
|
2015-01-31 10:13:07 +01:00
|
|
|
}
|
|
|
|
}
|
2015-05-04 04:48:21 +02:00
|
|
|
|
|
|
|
return {compute_time_stat(request_times), compute_time_stat(connect_times),
|
|
|
|
compute_time_stat(ttfb_times)};
|
2015-01-31 10:13:07 +01:00
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
void resolve_host() {
|
2014-02-28 01:10:42 +01:00
|
|
|
int rv;
|
|
|
|
addrinfo hints, *res;
|
|
|
|
|
|
|
|
memset(&hints, 0, sizeof(hints));
|
|
|
|
hints.ai_family = AF_UNSPEC;
|
|
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
|
|
hints.ai_protocol = 0;
|
|
|
|
hints.ai_flags = AI_ADDRCONFIG;
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
rv = getaddrinfo(config.host.c_str(), util::utos(config.port).c_str(), &hints,
|
|
|
|
&res);
|
|
|
|
if (rv != 0) {
|
|
|
|
std::cerr << "getaddrinfo() failed: " << gai_strerror(rv) << std::endl;
|
2014-02-28 01:10:42 +01:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
2014-11-27 15:39:04 +01:00
|
|
|
if (res == nullptr) {
|
2014-02-28 01:10:42 +01:00
|
|
|
std::cerr << "No address returned" << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
config.addrs = res;
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-03-14 15:15:01 +01:00
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
std::string get_reqline(const char *uri, const http_parser_url &u) {
|
2014-03-14 15:15:01 +01:00
|
|
|
std::string reqline;
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (util::has_uri_field(u, UF_PATH)) {
|
2014-03-14 15:15:01 +01:00
|
|
|
reqline = util::get_uri_field(uri, u, UF_PATH);
|
|
|
|
} else {
|
|
|
|
reqline = "/";
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (util::has_uri_field(u, UF_QUERY)) {
|
2014-03-14 15:15:01 +01:00
|
|
|
reqline += "?";
|
|
|
|
reqline += util::get_uri_field(uri, u, UF_QUERY);
|
|
|
|
}
|
|
|
|
|
|
|
|
return reqline;
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
int client_select_next_proto_cb(SSL *ssl, unsigned char **out,
|
|
|
|
unsigned char *outlen, const unsigned char *in,
|
|
|
|
unsigned int inlen, void *arg) {
|
2014-12-15 14:51:34 +01:00
|
|
|
if (util::select_h2(const_cast<const unsigned char **>(out), outlen, in,
|
|
|
|
inlen)) {
|
2014-02-28 01:10:42 +01:00
|
|
|
return SSL_TLSEXT_ERR_OK;
|
|
|
|
}
|
|
|
|
#ifdef HAVE_SPDYLAY
|
2014-12-15 14:51:34 +01:00
|
|
|
if (spdylay_select_next_protocol(out, outlen, in, inlen) > 0) {
|
2014-02-28 01:10:42 +01:00
|
|
|
return SSL_TLSEXT_ERR_OK;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
return SSL_TLSEXT_ERR_NOACK;
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-10-22 15:42:40 +02:00
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
template <typename Iterator>
|
|
|
|
std::vector<std::string> parse_uris(Iterator first, Iterator last) {
|
2014-10-22 15:42:40 +02:00
|
|
|
std::vector<std::string> reqlines;
|
|
|
|
|
|
|
|
// First URI is treated specially. We use scheme, host and port of
|
|
|
|
// this URI and ignore those in the remaining URIs if present.
|
|
|
|
http_parser_url u;
|
|
|
|
memset(&u, 0, sizeof(u));
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (first == last) {
|
2014-10-22 15:42:40 +02:00
|
|
|
std::cerr << "no URI available" << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
auto uri = (*first).c_str();
|
|
|
|
++first;
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (http_parser_parse_url(uri, strlen(uri), 0, &u) != 0 ||
|
|
|
|
!util::has_uri_field(u, UF_SCHEMA) || !util::has_uri_field(u, UF_HOST)) {
|
2014-10-22 15:42:40 +02:00
|
|
|
std::cerr << "invalid URI: " << uri << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
config.scheme = util::get_uri_field(uri, u, UF_SCHEMA);
|
|
|
|
config.host = util::get_uri_field(uri, u, UF_HOST);
|
|
|
|
config.default_port = util::get_default_port(uri, u);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (util::has_uri_field(u, UF_PORT)) {
|
2014-10-22 15:42:40 +02:00
|
|
|
config.port = u.port;
|
|
|
|
} else {
|
|
|
|
config.port = config.default_port;
|
|
|
|
}
|
|
|
|
|
|
|
|
reqlines.push_back(get_reqline(uri, u));
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
for (; first != last; ++first) {
|
2014-10-22 15:42:40 +02:00
|
|
|
http_parser_url u;
|
|
|
|
memset(&u, 0, sizeof(u));
|
|
|
|
|
|
|
|
auto uri = (*first).c_str();
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (http_parser_parse_url(uri, strlen(uri), 0, &u) != 0) {
|
2014-10-22 15:42:40 +02:00
|
|
|
std::cerr << "invalid URI: " << uri << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
reqlines.push_back(get_reqline(uri, u));
|
|
|
|
}
|
|
|
|
|
|
|
|
return reqlines;
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-10-22 15:51:00 +02:00
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
std::vector<std::string> read_uri_from_file(std::istream &infile) {
|
2014-10-22 15:51:00 +02:00
|
|
|
std::vector<std::string> uris;
|
|
|
|
std::string line_uri;
|
2014-11-27 15:39:04 +01:00
|
|
|
while (std::getline(infile, line_uri)) {
|
2014-10-22 15:51:00 +02:00
|
|
|
uris.push_back(line_uri);
|
|
|
|
}
|
|
|
|
|
|
|
|
return uris;
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2015-07-13 20:28:15 +02:00
|
|
|
namespace {
|
|
|
|
// Called every second when rate mode is being used
|
|
|
|
void second_timeout_cb(EV_P_ ev_timer *w, int revents) {
|
|
|
|
auto config = static_cast<Config *>(w->data);
|
|
|
|
|
|
|
|
auto nclients_per_worker = config->rate;
|
|
|
|
auto nreqs_per_worker = config->max_concurrent_streams * config->rate;
|
|
|
|
|
2015-07-16 19:51:41 +02:00
|
|
|
if (config->current_worker >= std::max((ssize_t)0, (config->seconds - 1))) {
|
2015-07-13 20:28:15 +02:00
|
|
|
nclients_per_worker = config->rate + config->conns_remainder;
|
|
|
|
nreqs_per_worker = (int)config->max_concurrent_streams *
|
|
|
|
(config->rate + config->conns_remainder);
|
|
|
|
ev_timer_stop(config->rate_loop, w);
|
|
|
|
}
|
|
|
|
|
2015-07-14 20:16:37 +02:00
|
|
|
config->workers.push_back(
|
|
|
|
make_unique<Worker>(config->current_worker, config->ssl_ctx,
|
|
|
|
nreqs_per_worker, nclients_per_worker, config));
|
2015-07-13 20:28:15 +02:00
|
|
|
|
|
|
|
config->current_worker++;
|
|
|
|
|
|
|
|
config->workers.back()->run();
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
void print_version(std::ostream &out) {
|
2014-02-28 01:10:42 +01:00
|
|
|
out << "h2load nghttp2/" NGHTTP2_VERSION << std::endl;
|
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
void print_usage(std::ostream &out) {
|
2015-01-09 16:37:42 +01:00
|
|
|
out << R"(Usage: h2load [OPTIONS]... [URI]...
|
2014-03-13 17:06:47 +01:00
|
|
|
benchmarking tool for HTTP/2 and SPDY server)" << std::endl;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
|
|
|
namespace {
|
2014-11-27 15:39:04 +01:00
|
|
|
void print_help(std::ostream &out) {
|
2014-02-28 01:10:42 +01:00
|
|
|
print_usage(out);
|
2014-03-13 17:06:47 +01:00
|
|
|
|
|
|
|
out << R"(
|
2015-01-15 16:00:38 +01:00
|
|
|
<URI> Specify URI to access. Multiple URIs can be specified.
|
|
|
|
URIs are used in this order for each client. All URIs
|
|
|
|
are used, then first URI is used and then 2nd URI, and
|
|
|
|
so on. The scheme, host and port in the subsequent
|
|
|
|
URIs, if present, are ignored. Those in the first URI
|
|
|
|
are used solely.
|
2014-03-13 17:06:47 +01:00
|
|
|
Options:
|
2015-01-15 16:00:38 +01:00
|
|
|
-n, --requests=<N>
|
|
|
|
Number of requests.
|
|
|
|
Default: )" << config.nreqs << R"(
|
|
|
|
-c, --clients=<N>
|
|
|
|
Number of concurrent clients.
|
|
|
|
Default: )" << config.nclients << R"(
|
|
|
|
-t, --threads=<N>
|
|
|
|
Number of native threads.
|
|
|
|
Default: )" << config.nthreads << R"(
|
2014-10-22 00:29:36 +02:00
|
|
|
-i, --input-file=<FILE>
|
2015-05-12 16:24:18 +02:00
|
|
|
Path of a file with multiple URIs are separated by EOLs.
|
2015-01-15 16:00:38 +01:00
|
|
|
This option will disable URIs getting from command-line.
|
|
|
|
If '-' is given as <FILE>, URIs will be read from stdin.
|
|
|
|
URIs are used in this order for each client. All URIs
|
|
|
|
are used, then first URI is used and then 2nd URI, and
|
|
|
|
so on. The scheme, host and port in the subsequent
|
|
|
|
URIs, if present, are ignored. Those in the first URI
|
|
|
|
are used solely.
|
2014-03-14 15:15:01 +01:00
|
|
|
-m, --max-concurrent-streams=(auto|<N>)
|
2015-01-15 16:00:38 +01:00
|
|
|
Max concurrent streams to issue per session. If "auto"
|
|
|
|
is given, the number of given URIs is used.
|
|
|
|
Default: auto
|
2014-03-13 17:06:47 +01:00
|
|
|
-w, --window-bits=<N>
|
2015-01-15 16:00:38 +01:00
|
|
|
Sets the stream level initial window size to (2**<N>)-1.
|
|
|
|
For SPDY, 2**<N> is used instead.
|
2015-04-27 14:23:01 +02:00
|
|
|
Default: )" << config.window_bits << R"(
|
2014-03-13 17:06:47 +01:00
|
|
|
-W, --connection-window-bits=<N>
|
2015-01-15 16:00:38 +01:00
|
|
|
Sets the connection level initial window size to
|
|
|
|
(2**<N>)-1. For SPDY, if <N> is strictly less than 16,
|
|
|
|
this option is ignored. Otherwise 2**<N> is used for
|
|
|
|
SPDY.
|
2015-04-27 14:23:01 +02:00
|
|
|
Default: )" << config.connection_window_bits << R"(
|
2014-10-21 23:25:38 +02:00
|
|
|
-H, --header=<HEADER>
|
2015-01-15 16:00:38 +01:00
|
|
|
Add/Override a header to the requests.
|
2015-07-08 16:14:41 +02:00
|
|
|
--ciphers=<SUITE>
|
|
|
|
Set allowed cipher list. The format of the string is
|
|
|
|
described in OpenSSL ciphers(1).
|
2014-04-21 14:35:45 +02:00
|
|
|
-p, --no-tls-proto=<PROTOID>
|
2015-01-15 16:00:38 +01:00
|
|
|
Specify ALPN identifier of the protocol to be used when
|
|
|
|
accessing http URI without SSL/TLS.)";
|
2015-07-13 20:28:15 +02:00
|
|
|
|
2014-04-21 14:35:45 +02:00
|
|
|
#ifdef HAVE_SPDYLAY
|
|
|
|
out << R"(
|
2015-01-15 16:00:38 +01:00
|
|
|
Available protocols: spdy/2, spdy/3, spdy/3.1 and )";
|
|
|
|
#else // !HAVE_SPDYLAY
|
2014-04-21 14:35:45 +02:00
|
|
|
out << R"(
|
2015-01-15 16:00:38 +01:00
|
|
|
Available protocol: )";
|
2014-04-21 14:35:45 +02:00
|
|
|
#endif // !HAVE_SPDYLAY
|
|
|
|
out << NGHTTP2_CLEARTEXT_PROTO_VERSION_ID << R"(
|
2015-07-14 20:16:37 +02:00
|
|
|
Default: )"
|
|
|
|
<< NGHTTP2_CLEARTEXT_PROTO_VERSION_ID << R"(
|
2015-03-26 11:53:42 +01:00
|
|
|
-d, --data=<FILE>
|
|
|
|
Post FILE to server. The request method is changed to
|
|
|
|
POST.
|
2015-07-13 20:28:15 +02:00
|
|
|
-r, --rate=<N>
|
|
|
|
Specified the fixed rate at which connections are
|
|
|
|
created. The rate must be a positive integer,
|
|
|
|
representing the number of connections to be made per
|
|
|
|
second. When the rate is 0, the program will run as it
|
|
|
|
normally does, creating connections at whatever variable
|
|
|
|
rate it wants. The default value for this option is 0.
|
|
|
|
-C, --num-conns=<N>
|
|
|
|
Specifies the total number of connections to create. The
|
|
|
|
total number of connections must be a positive integer.
|
|
|
|
On each connection, '-m' requests are made. The test
|
|
|
|
stops once as soon as the N connections have either
|
|
|
|
completed or failed. When the number of connections is
|
|
|
|
0, the program will run as it normally does, creating as
|
|
|
|
many connections as it needs in order to make the '-n'
|
2015-07-16 19:43:50 +02:00
|
|
|
requests specified. The default value for this option is
|
2015-07-13 20:28:15 +02:00
|
|
|
0. The '-n' option is not required if the '-C' option
|
|
|
|
is being used.
|
2015-01-15 16:00:38 +01:00
|
|
|
-v, --verbose
|
|
|
|
Output debug information.
|
|
|
|
--version Display version information and exit.
|
2015-07-14 20:16:37 +02:00
|
|
|
-h, --help Display this help and exit.)"
|
|
|
|
<< std::endl;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
int main(int argc, char **argv) {
|
2015-05-22 17:23:38 +02:00
|
|
|
#ifndef NOTHREADS
|
|
|
|
ssl::LibsslGlobalLock lock;
|
|
|
|
#endif // NOTHREADS
|
|
|
|
SSL_load_error_strings();
|
|
|
|
SSL_library_init();
|
|
|
|
OpenSSL_add_all_algorithms();
|
|
|
|
OPENSSL_config(nullptr);
|
|
|
|
|
2015-03-26 11:53:42 +01:00
|
|
|
std::string datafile;
|
2014-11-27 15:39:04 +01:00
|
|
|
while (1) {
|
2014-04-30 09:08:21 +02:00
|
|
|
static int flag = 0;
|
2014-02-28 01:10:42 +01:00
|
|
|
static option long_options[] = {
|
2014-11-27 15:39:04 +01:00
|
|
|
{"requests", required_argument, nullptr, 'n'},
|
|
|
|
{"clients", required_argument, nullptr, 'c'},
|
2015-03-26 11:53:42 +01:00
|
|
|
{"data", required_argument, nullptr, 'd'},
|
2014-11-27 15:39:04 +01:00
|
|
|
{"threads", required_argument, nullptr, 't'},
|
|
|
|
{"max-concurrent-streams", required_argument, nullptr, 'm'},
|
|
|
|
{"window-bits", required_argument, nullptr, 'w'},
|
|
|
|
{"connection-window-bits", required_argument, nullptr, 'W'},
|
|
|
|
{"input-file", required_argument, nullptr, 'i'},
|
|
|
|
{"header", required_argument, nullptr, 'H'},
|
|
|
|
{"no-tls-proto", required_argument, nullptr, 'p'},
|
|
|
|
{"verbose", no_argument, nullptr, 'v'},
|
|
|
|
{"help", no_argument, nullptr, 'h'},
|
|
|
|
{"version", no_argument, &flag, 1},
|
2015-07-08 16:14:41 +02:00
|
|
|
{"ciphers", required_argument, &flag, 2},
|
2015-07-13 20:28:15 +02:00
|
|
|
{"rate", required_argument, nullptr, 'r'},
|
|
|
|
{"num-conns", required_argument, nullptr, 'C'},
|
2014-11-27 15:39:04 +01:00
|
|
|
{nullptr, 0, nullptr, 0}};
|
2014-02-28 01:10:42 +01:00
|
|
|
int option_index = 0;
|
2015-07-14 20:16:37 +02:00
|
|
|
auto c = getopt_long(argc, argv, "hvW:c:d:m:n:p:t:w:H:i:r:C:", long_options,
|
|
|
|
&option_index);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (c == -1) {
|
2014-02-28 01:10:42 +01:00
|
|
|
break;
|
|
|
|
}
|
2014-11-27 15:39:04 +01:00
|
|
|
switch (c) {
|
2014-02-28 01:10:42 +01:00
|
|
|
case 'n':
|
|
|
|
config.nreqs = strtoul(optarg, nullptr, 10);
|
|
|
|
break;
|
|
|
|
case 'c':
|
|
|
|
config.nclients = strtoul(optarg, nullptr, 10);
|
|
|
|
break;
|
2015-03-26 11:53:42 +01:00
|
|
|
case 'd':
|
|
|
|
datafile = optarg;
|
|
|
|
break;
|
2014-02-28 01:10:42 +01:00
|
|
|
case 't':
|
2014-04-23 17:47:26 +02:00
|
|
|
#ifdef NOTHREADS
|
2014-11-27 15:39:04 +01:00
|
|
|
std::cerr << "-t: WARNING: Threading disabled at build time, "
|
|
|
|
<< "no threads created." << std::endl;
|
2014-04-23 17:47:26 +02:00
|
|
|
#else
|
2014-03-03 17:12:11 +01:00
|
|
|
config.nthreads = strtoul(optarg, nullptr, 10);
|
2014-05-14 16:22:23 +02:00
|
|
|
#endif // NOTHREADS
|
2014-02-28 01:10:42 +01:00
|
|
|
break;
|
|
|
|
case 'm':
|
2014-11-27 15:39:04 +01:00
|
|
|
if (util::strieq("auto", optarg)) {
|
2014-03-14 15:15:01 +01:00
|
|
|
config.max_concurrent_streams = -1;
|
|
|
|
} else {
|
|
|
|
config.max_concurrent_streams = strtoul(optarg, nullptr, 10);
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
break;
|
|
|
|
case 'w':
|
|
|
|
case 'W': {
|
|
|
|
errno = 0;
|
|
|
|
char *endptr = nullptr;
|
|
|
|
auto n = strtoul(optarg, &endptr, 10);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (errno == 0 && *endptr == '\0' && n < 31) {
|
|
|
|
if (c == 'w') {
|
2014-02-28 01:10:42 +01:00
|
|
|
config.window_bits = n;
|
|
|
|
} else {
|
|
|
|
config.connection_window_bits = n;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
std::cerr << "-" << static_cast<char>(c)
|
|
|
|
<< ": specify the integer in the range [0, 30], inclusive"
|
|
|
|
<< std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
2014-10-18 00:25:59 +02:00
|
|
|
case 'H': {
|
|
|
|
char *header = optarg;
|
|
|
|
// Skip first possible ':' in the header name
|
2014-11-27 15:39:04 +01:00
|
|
|
char *value = strchr(optarg + 1, ':');
|
|
|
|
if (!value || (header[0] == ':' && header + 1 == value)) {
|
|
|
|
std::cerr << "-H: invalid header: " << optarg << std::endl;
|
2014-10-18 00:25:59 +02:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
*value = 0;
|
|
|
|
value++;
|
2014-11-27 15:39:04 +01:00
|
|
|
while (isspace(*value)) {
|
|
|
|
value++;
|
|
|
|
}
|
|
|
|
if (*value == 0) {
|
2014-10-18 00:25:59 +02:00
|
|
|
// This could also be a valid case for suppressing a header
|
|
|
|
// similar to curl
|
|
|
|
std::cerr << "-H: invalid header - value missing: " << optarg
|
|
|
|
<< std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
// Note that there is no processing currently to handle multiple
|
|
|
|
// message-header fields with the same field name
|
|
|
|
config.custom_headers.emplace_back(header, value);
|
2014-10-22 15:53:54 +02:00
|
|
|
util::inp_strlower(config.custom_headers.back().name);
|
2014-10-18 00:25:59 +02:00
|
|
|
break;
|
|
|
|
}
|
2014-10-22 00:29:36 +02:00
|
|
|
case 'i': {
|
|
|
|
config.ifile = std::string(optarg);
|
|
|
|
break;
|
|
|
|
}
|
2014-04-21 14:35:45 +02:00
|
|
|
case 'p':
|
2014-11-27 15:39:04 +01:00
|
|
|
if (util::strieq(NGHTTP2_CLEARTEXT_PROTO_VERSION_ID, optarg)) {
|
2014-04-21 14:35:45 +02:00
|
|
|
config.no_tls_proto = Config::PROTO_HTTP2;
|
|
|
|
#ifdef HAVE_SPDYLAY
|
2014-11-27 15:39:04 +01:00
|
|
|
} else if (util::strieq("spdy/2", optarg)) {
|
2014-04-21 14:35:45 +02:00
|
|
|
config.no_tls_proto = Config::PROTO_SPDY2;
|
2014-11-27 15:39:04 +01:00
|
|
|
} else if (util::strieq("spdy/3", optarg)) {
|
2014-04-21 14:35:45 +02:00
|
|
|
config.no_tls_proto = Config::PROTO_SPDY3;
|
2014-11-27 15:39:04 +01:00
|
|
|
} else if (util::strieq("spdy/3.1", optarg)) {
|
2014-04-21 14:35:45 +02:00
|
|
|
config.no_tls_proto = Config::PROTO_SPDY3_1;
|
|
|
|
#endif // HAVE_SPDYLAY
|
|
|
|
} else {
|
|
|
|
std::cerr << "-p: unsupported protocol " << optarg << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
break;
|
2015-07-13 20:28:15 +02:00
|
|
|
case 'r':
|
|
|
|
config.rate = strtoul(optarg, nullptr, 10);
|
|
|
|
if (config.rate <= 0) {
|
|
|
|
std::cerr << "-r: the rate at which connections are made "
|
2015-07-14 20:16:37 +02:00
|
|
|
<< "must be positive." << std::endl;
|
2015-07-13 20:28:15 +02:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 'C':
|
|
|
|
config.nconns = strtoul(optarg, nullptr, 10);
|
|
|
|
if (config.nconns <= 0) {
|
|
|
|
std::cerr << "-C: the total number of connections made "
|
2015-07-14 20:16:37 +02:00
|
|
|
<< "must be positive." << std::endl;
|
2015-07-13 20:28:15 +02:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
break;
|
2014-02-28 01:10:42 +01:00
|
|
|
case 'v':
|
|
|
|
config.verbose = true;
|
|
|
|
break;
|
|
|
|
case 'h':
|
|
|
|
print_help(std::cout);
|
|
|
|
exit(EXIT_SUCCESS);
|
|
|
|
case '?':
|
|
|
|
util::show_candidates(argv[optind - 1], long_options);
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
case 0:
|
2014-11-27 15:39:04 +01:00
|
|
|
switch (flag) {
|
2014-02-28 01:10:42 +01:00
|
|
|
case 1:
|
|
|
|
// version option
|
|
|
|
print_version(std::cout);
|
|
|
|
exit(EXIT_SUCCESS);
|
2015-07-08 16:14:41 +02:00
|
|
|
case 2:
|
|
|
|
// ciphers option
|
|
|
|
config.ciphers = optarg;
|
|
|
|
break;
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (argc == optind) {
|
|
|
|
if (config.ifile.empty()) {
|
2014-10-22 00:29:36 +02:00
|
|
|
std::cerr << "no URI or input file given" << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.nreqs == 0) {
|
2014-02-28 01:10:42 +01:00
|
|
|
std::cerr << "-n: the number of requests must be strictly greater than 0."
|
|
|
|
<< std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.max_concurrent_streams == 0) {
|
2014-02-28 01:10:42 +01:00
|
|
|
std::cerr << "-m: the max concurrent streams must be strictly greater "
|
2014-11-27 15:39:04 +01:00
|
|
|
<< "than 0." << std::endl;
|
2014-02-28 01:10:42 +01:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.nthreads == 0) {
|
2014-03-03 17:12:11 +01:00
|
|
|
std::cerr << "-t: the number of threads must be strictly greater than 0."
|
|
|
|
<< std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.nreqs < config.nclients) {
|
2014-02-28 01:10:42 +01:00
|
|
|
std::cerr << "-n, -c: the number of requests must be greater than or "
|
2014-11-27 15:39:04 +01:00
|
|
|
<< "equal to the concurrent clients." << std::endl;
|
2014-02-28 01:10:42 +01:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2015-03-26 11:57:37 +01:00
|
|
|
if (config.nclients < config.nthreads) {
|
|
|
|
std::cerr << "-c, -t: the number of client must be greater than or equal "
|
|
|
|
"to the number of threads." << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.nthreads > std::thread::hardware_concurrency()) {
|
2014-03-03 17:12:11 +01:00
|
|
|
std::cerr << "-t: warning: the number of threads is greater than hardware "
|
2014-11-27 15:39:04 +01:00
|
|
|
<< "cores." << std::endl;
|
2014-03-03 17:12:11 +01:00
|
|
|
}
|
|
|
|
|
2015-07-13 20:28:15 +02:00
|
|
|
if (config.nconns < 0) {
|
|
|
|
std::cerr << "-C: the total number of connections made "
|
2015-07-14 20:16:37 +02:00
|
|
|
<< "cannot be negative." << std::endl;
|
2015-07-13 20:28:15 +02:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (config.rate < 0) {
|
|
|
|
std::cerr << "-r: the rate at which connections are made "
|
2015-07-14 20:16:37 +02:00
|
|
|
<< "cannot be negative." << std::endl;
|
2015-07-13 20:28:15 +02:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (config.rate != 0 && config.nthreads != 1) {
|
|
|
|
std::cerr << "-r, -t: warning: the -t option will be ignored when the -r "
|
|
|
|
<< "option is in use." << std::endl;
|
|
|
|
}
|
|
|
|
|
2015-03-26 11:53:42 +01:00
|
|
|
if (!datafile.empty()) {
|
|
|
|
config.data_fd = open(datafile.c_str(), O_RDONLY | O_BINARY);
|
|
|
|
if (config.data_fd == -1) {
|
|
|
|
std::cerr << "-d: Could not open file " << datafile << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
struct stat data_stat;
|
|
|
|
if (fstat(config.data_fd, &data_stat) == -1) {
|
|
|
|
std::cerr << "-d: Could not stat file " << datafile << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
config.data_length = data_stat.st_size;
|
|
|
|
}
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
struct sigaction act;
|
|
|
|
memset(&act, 0, sizeof(struct sigaction));
|
|
|
|
act.sa_handler = SIG_IGN;
|
|
|
|
sigaction(SIGPIPE, &act, nullptr);
|
2014-03-04 13:29:42 +01:00
|
|
|
|
2014-03-14 15:15:01 +01:00
|
|
|
auto ssl_ctx = SSL_CTX_new(SSLv23_client_method());
|
2014-11-27 15:39:04 +01:00
|
|
|
if (!ssl_ctx) {
|
2014-03-14 15:15:01 +01:00
|
|
|
std::cerr << "Failed to create SSL_CTX: "
|
|
|
|
<< ERR_error_string(ERR_get_error(), nullptr) << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
2014-12-12 17:37:57 +01:00
|
|
|
|
2015-06-22 16:26:45 +02:00
|
|
|
auto ssl_opts = (SSL_OP_ALL & ~SSL_OP_DONT_INSERT_EMPTY_FRAGMENTS) |
|
|
|
|
SSL_OP_NO_SSLv2 | SSL_OP_NO_SSLv3 | SSL_OP_NO_COMPRESSION |
|
|
|
|
SSL_OP_NO_SESSION_RESUMPTION_ON_RENEGOTIATION;
|
|
|
|
|
|
|
|
SSL_CTX_set_options(ssl_ctx, ssl_opts);
|
2014-12-12 17:37:57 +01:00
|
|
|
SSL_CTX_set_mode(ssl_ctx, SSL_MODE_AUTO_RETRY);
|
|
|
|
SSL_CTX_set_mode(ssl_ctx, SSL_MODE_RELEASE_BUFFERS);
|
|
|
|
|
2015-07-08 16:14:41 +02:00
|
|
|
const char *ciphers;
|
|
|
|
if (config.ciphers.empty()) {
|
|
|
|
ciphers = ssl::DEFAULT_CIPHER_LIST;
|
|
|
|
} else {
|
|
|
|
ciphers = config.ciphers.c_str();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (SSL_CTX_set_cipher_list(ssl_ctx, ciphers) == 0) {
|
|
|
|
std::cerr << "SSL_CTX_set_cipher_list with " << ciphers
|
|
|
|
<< " failed: " << ERR_error_string(ERR_get_error(), nullptr)
|
|
|
|
<< std::endl;
|
2014-12-12 17:37:57 +01:00
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
SSL_CTX_set_next_proto_select_cb(ssl_ctx, client_select_next_proto_cb,
|
|
|
|
nullptr);
|
2014-03-14 15:15:01 +01:00
|
|
|
|
2014-11-21 17:13:18 +01:00
|
|
|
#if OPENSSL_VERSION_NUMBER >= 0x10002000L
|
|
|
|
auto proto_list = util::get_default_alpn();
|
|
|
|
#ifdef HAVE_SPDYLAY
|
|
|
|
static const char spdy_proto_list[] = "\x8spdy/3.1\x6spdy/3\x6spdy/2";
|
2015-02-06 13:35:03 +01:00
|
|
|
std::copy_n(spdy_proto_list, sizeof(spdy_proto_list) - 1,
|
|
|
|
std::back_inserter(proto_list));
|
2014-11-21 17:13:18 +01:00
|
|
|
#endif // HAVE_SPDYLAY
|
|
|
|
SSL_CTX_set_alpn_protos(ssl_ctx, proto_list.data(), proto_list.size());
|
|
|
|
#endif // OPENSSL_VERSION_NUMBER >= 0x10002000L
|
|
|
|
|
2014-10-22 15:42:40 +02:00
|
|
|
std::vector<std::string> reqlines;
|
2014-10-20 22:59:55 +02:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.ifile.empty()) {
|
2014-10-22 15:42:40 +02:00
|
|
|
std::vector<std::string> uris;
|
2014-10-22 14:53:14 +02:00
|
|
|
std::copy(&argv[optind], &argv[argc], std::back_inserter(uris));
|
2014-10-22 15:42:40 +02:00
|
|
|
reqlines = parse_uris(std::begin(uris), std::end(uris));
|
2014-10-22 00:29:36 +02:00
|
|
|
} else {
|
2014-10-22 15:42:40 +02:00
|
|
|
std::vector<std::string> uris;
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.ifile == "-") {
|
2014-10-22 15:51:00 +02:00
|
|
|
uris = read_uri_from_file(std::cin);
|
|
|
|
} else {
|
|
|
|
std::ifstream infile(config.ifile);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (!infile) {
|
2014-10-22 15:51:00 +02:00
|
|
|
std::cerr << "cannot read input file: " << config.ifile << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
uris = read_uri_from_file(infile);
|
2014-10-22 14:53:14 +02:00
|
|
|
}
|
2014-10-22 00:29:36 +02:00
|
|
|
|
2014-10-22 15:42:40 +02:00
|
|
|
reqlines = parse_uris(std::begin(uris), std::end(uris));
|
2014-10-20 22:59:55 +02:00
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.max_concurrent_streams == -1) {
|
2014-03-14 15:15:01 +01:00
|
|
|
config.max_concurrent_streams = reqlines.size();
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2015-07-15 18:51:33 +02:00
|
|
|
|
2015-07-15 01:41:29 +02:00
|
|
|
// if not in rate mode and -C is set, warn that we are ignoring it
|
|
|
|
if (!config.is_rate_mode() && config.nconns != 0) {
|
|
|
|
std::cerr << "-C: warning: This option can only be used with -r, and"
|
|
|
|
<< " will be ignored otherwise." << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
ssize_t n_time = 0;
|
2015-07-15 18:51:33 +02:00
|
|
|
ssize_t c_time = 0;
|
2015-07-15 01:41:29 +02:00
|
|
|
// only care about n_time and c_time in rate mode
|
|
|
|
if (config.is_rate_mode() && config.max_concurrent_streams != 0) {
|
2015-07-15 18:51:33 +02:00
|
|
|
n_time = (int)config.nreqs /
|
2015-07-15 01:41:29 +02:00
|
|
|
((int)config.rate * (int)config.max_concurrent_streams);
|
|
|
|
c_time = (int)config.nconns / (int)config.rate;
|
|
|
|
}
|
|
|
|
// check to see if the two ways of determining test time conflict
|
|
|
|
if (config.is_rate_mode() && (int)config.max_concurrent_streams != 0 &&
|
|
|
|
(n_time != c_time) && config.nreqs != 1 && config.nconns != 0) {
|
|
|
|
if ((int)config.nreqs < config.nconns) {
|
|
|
|
std::cerr << "-C, -n: warning: two different test times are being set. "
|
|
|
|
<< std::endl;
|
2015-07-15 18:51:33 +02:00
|
|
|
std::cerr << "The test will run for " << c_time << " seconds."
|
|
|
|
<< std::endl;
|
|
|
|
} else {
|
2015-07-15 01:41:29 +02:00
|
|
|
std::cout << "-C, -n: warning: two different test times are being set. "
|
|
|
|
<< std::endl;
|
|
|
|
std::cout << "The smaller of the two will be chosen and the test will "
|
2015-07-15 18:51:33 +02:00
|
|
|
<< "run for " << std::min(n_time, c_time) << " seconds."
|
|
|
|
<< std::endl;
|
2015-07-15 01:41:29 +02:00
|
|
|
}
|
|
|
|
}
|
2015-07-15 18:51:33 +02:00
|
|
|
|
2014-03-14 15:15:01 +01:00
|
|
|
Headers shared_nva;
|
|
|
|
shared_nva.emplace_back(":scheme", config.scheme);
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.port != config.default_port) {
|
2014-03-14 15:15:01 +01:00
|
|
|
shared_nva.emplace_back(":authority",
|
2014-10-22 15:42:40 +02:00
|
|
|
config.host + ":" + util::utos(config.port));
|
2014-02-28 01:10:42 +01:00
|
|
|
} else {
|
2014-03-14 15:15:01 +01:00
|
|
|
shared_nva.emplace_back(":authority", config.host);
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2015-03-26 11:53:42 +01:00
|
|
|
shared_nva.emplace_back(":method", config.data_fd == -1 ? "GET" : "POST");
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
// list overridalbe headers
|
|
|
|
auto override_hdrs =
|
2015-02-05 15:39:36 +01:00
|
|
|
make_array<std::string>(":authority", ":host", ":method", ":scheme");
|
2014-10-21 23:25:38 +02:00
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
for (auto &kv : config.custom_headers) {
|
|
|
|
if (std::find(std::begin(override_hdrs), std::end(override_hdrs),
|
|
|
|
kv.name) != std::end(override_hdrs)) {
|
2014-10-21 23:25:38 +02:00
|
|
|
// override header
|
2014-11-27 15:39:04 +01:00
|
|
|
for (auto &nv : shared_nva) {
|
|
|
|
if ((nv.name == ":authority" && kv.name == ":host") ||
|
|
|
|
(nv.name == kv.name)) {
|
2014-10-22 15:53:54 +02:00
|
|
|
nv.value = kv.value;
|
2014-10-18 00:25:59 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// add additional headers
|
2014-10-22 15:53:54 +02:00
|
|
|
shared_nva.push_back(kv);
|
2014-10-18 00:25:59 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
for (auto &req : reqlines) {
|
2014-03-14 15:15:01 +01:00
|
|
|
// For nghttp2
|
|
|
|
std::vector<nghttp2_nv> nva;
|
2014-02-28 01:10:42 +01:00
|
|
|
|
2014-03-14 15:15:01 +01:00
|
|
|
nva.push_back(http2::make_nv_ls(":path", req));
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
for (auto &nv : shared_nva) {
|
2014-04-03 04:22:11 +02:00
|
|
|
nva.push_back(http2::make_nv(nv.name, nv.value, false));
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
2014-03-14 15:15:01 +01:00
|
|
|
|
|
|
|
config.nva.push_back(std::move(nva));
|
|
|
|
|
|
|
|
// For spdylay
|
2014-11-27 15:39:04 +01:00
|
|
|
std::vector<const char *> cva;
|
2014-03-14 15:15:01 +01:00
|
|
|
|
|
|
|
cva.push_back(":path");
|
|
|
|
cva.push_back(req.c_str());
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
for (auto &nv : shared_nva) {
|
|
|
|
if (nv.name == ":authority") {
|
2014-03-14 15:15:01 +01:00
|
|
|
cva.push_back(":host");
|
|
|
|
} else {
|
2014-04-03 04:22:11 +02:00
|
|
|
cva.push_back(nv.name.c_str());
|
2014-03-14 15:15:01 +01:00
|
|
|
}
|
2014-04-03 04:22:11 +02:00
|
|
|
cva.push_back(nv.value.c_str());
|
2014-03-14 15:15:01 +01:00
|
|
|
}
|
|
|
|
cva.push_back(":version");
|
|
|
|
cva.push_back("HTTP/1.1");
|
|
|
|
cva.push_back(nullptr);
|
|
|
|
|
|
|
|
config.nv.push_back(std::move(cva));
|
2014-02-28 01:10:42 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
resolve_host();
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
if (config.nclients == 1) {
|
2014-06-06 16:11:08 +02:00
|
|
|
config.nthreads = 1;
|
|
|
|
}
|
|
|
|
|
2014-03-03 17:12:11 +01:00
|
|
|
size_t nreqs_per_thread = config.nreqs / config.nthreads;
|
|
|
|
ssize_t nreqs_rem = config.nreqs % config.nthreads;
|
|
|
|
|
|
|
|
size_t nclients_per_thread = config.nclients / config.nthreads;
|
|
|
|
ssize_t nclients_rem = config.nclients % config.nthreads;
|
2014-02-28 01:10:42 +01:00
|
|
|
|
|
|
|
std::cout << "starting benchmark..." << std::endl;
|
|
|
|
|
|
|
|
auto start = std::chrono::steady_clock::now();
|
2014-03-03 17:12:11 +01:00
|
|
|
|
2015-07-13 20:28:15 +02:00
|
|
|
// if not in rate mode, continue making workers and clients normally
|
|
|
|
if (!config.is_rate_mode()) {
|
|
|
|
config.workers.reserve(config.nthreads);
|
2014-10-05 07:25:15 +02:00
|
|
|
#ifndef NOTHREADS
|
2015-07-13 20:28:15 +02:00
|
|
|
std::vector<std::future<void>> futures;
|
|
|
|
for (size_t i = 0; i < config.nthreads - 1; ++i) {
|
|
|
|
auto nreqs = nreqs_per_thread + (nreqs_rem-- > 0);
|
|
|
|
auto nclients = nclients_per_thread + (nclients_rem-- > 0);
|
|
|
|
std::cout << "spawning thread #" << i << ": " << nclients
|
|
|
|
<< " concurrent clients, " << nreqs << " total requests"
|
|
|
|
<< std::endl;
|
|
|
|
config.workers.push_back(
|
|
|
|
make_unique<Worker>(i, ssl_ctx, nreqs, nclients, &config));
|
|
|
|
auto &worker = config.workers.back();
|
|
|
|
futures.push_back(
|
|
|
|
std::async(std::launch::async, [&worker]() { worker->run(); }));
|
|
|
|
}
|
2014-10-05 07:25:15 +02:00
|
|
|
#endif // NOTHREADS
|
|
|
|
|
2015-07-13 20:28:15 +02:00
|
|
|
auto nreqs_last = nreqs_per_thread + (nreqs_rem-- > 0);
|
|
|
|
auto nclients_last = nclients_per_thread + (nclients_rem-- > 0);
|
|
|
|
std::cout << "spawning thread #" << (config.nthreads - 1) << ": "
|
|
|
|
<< nclients_last << " concurrent clients, " << nreqs_last
|
|
|
|
<< " total requests" << std::endl;
|
2015-07-14 20:16:37 +02:00
|
|
|
config.workers.push_back(make_unique<Worker>(
|
|
|
|
config.nthreads - 1, ssl_ctx, nreqs_last, nclients_last, &config));
|
2015-07-13 20:28:15 +02:00
|
|
|
config.workers.back()->run();
|
2014-03-03 17:12:11 +01:00
|
|
|
|
2014-10-05 07:25:15 +02:00
|
|
|
#ifndef NOTHREADS
|
2015-07-13 20:28:15 +02:00
|
|
|
for (auto &fut : futures) {
|
|
|
|
fut.get();
|
|
|
|
}
|
2014-10-05 07:25:15 +02:00
|
|
|
#endif // NOTHREADS
|
2015-07-15 18:51:33 +02:00
|
|
|
} //! config.is_rate_mode()
|
2015-07-13 20:28:15 +02:00
|
|
|
// if in rate mode, create a new worker each second
|
|
|
|
else {
|
|
|
|
// set various config values
|
|
|
|
if ((int)config.nreqs < config.nconns) {
|
|
|
|
config.seconds = c_time;
|
2015-07-14 20:16:37 +02:00
|
|
|
} else if (config.nconns == 0) {
|
2015-07-13 20:28:15 +02:00
|
|
|
config.seconds = n_time;
|
2015-07-14 20:16:37 +02:00
|
|
|
} else {
|
2015-07-16 20:02:01 +02:00
|
|
|
config.seconds = std::min(n_time, c_time);
|
2015-07-13 20:28:15 +02:00
|
|
|
}
|
2015-07-16 20:02:01 +02:00
|
|
|
config.workers.reserve(config.seconds);
|
2015-07-13 20:28:15 +02:00
|
|
|
|
|
|
|
config.conns_remainder = config.nconns % config.rate;
|
|
|
|
|
|
|
|
// config.seconds must be positive or else an exception is thrown
|
|
|
|
if (config.seconds <= 0) {
|
|
|
|
std::cerr << "Test cannot be run with current option values."
|
|
|
|
<< " Please look at documentation for -r option for"
|
|
|
|
<< " more information." << std::endl;
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
config.current_worker = 0;
|
|
|
|
|
|
|
|
config.ssl_ctx = ssl_ctx;
|
|
|
|
|
|
|
|
// create timer that will go off every second
|
|
|
|
ev_timer timeout_watcher;
|
|
|
|
|
|
|
|
// create loop for running the timer
|
|
|
|
struct ev_loop *rate_loop = EV_DEFAULT;
|
|
|
|
|
|
|
|
config.rate_loop = rate_loop;
|
|
|
|
|
|
|
|
// giving the second_timeout_cb access to config
|
|
|
|
timeout_watcher.data = &config;
|
|
|
|
|
|
|
|
ev_init(&timeout_watcher, second_timeout_cb);
|
|
|
|
timeout_watcher.repeat = 1.;
|
|
|
|
ev_timer_again(rate_loop, &timeout_watcher);
|
|
|
|
ev_run(rate_loop, 0);
|
|
|
|
} // end rate mode section
|
2014-03-03 17:12:11 +01:00
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
auto end = std::chrono::steady_clock::now();
|
2015-01-31 10:13:07 +01:00
|
|
|
auto duration =
|
|
|
|
std::chrono::duration_cast<std::chrono::microseconds>(end - start);
|
|
|
|
|
|
|
|
Stats stats(0);
|
2015-07-13 20:28:15 +02:00
|
|
|
for (const auto &w : config.workers) {
|
2015-01-31 10:13:07 +01:00
|
|
|
const auto &s = w->stats;
|
|
|
|
|
|
|
|
stats.req_todo += s.req_todo;
|
|
|
|
stats.req_started += s.req_started;
|
|
|
|
stats.req_done += s.req_done;
|
|
|
|
stats.req_success += s.req_success;
|
|
|
|
stats.req_status_success += s.req_status_success;
|
|
|
|
stats.req_failed += s.req_failed;
|
|
|
|
stats.req_error += s.req_error;
|
|
|
|
stats.bytes_total += s.bytes_total;
|
|
|
|
stats.bytes_head += s.bytes_head;
|
|
|
|
stats.bytes_body += s.bytes_body;
|
|
|
|
|
|
|
|
for (size_t i = 0; i < stats.status.size(); ++i) {
|
|
|
|
stats.status[i] += s.status[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-07-13 20:28:15 +02:00
|
|
|
auto ts = process_time_stats(config.workers);
|
2014-02-28 01:10:42 +01:00
|
|
|
|
|
|
|
// Requests which have not been issued due to connection errors, are
|
|
|
|
// counted towards req_failed and req_error.
|
2015-01-31 10:13:07 +01:00
|
|
|
auto req_not_issued =
|
|
|
|
stats.req_todo - stats.req_status_success - stats.req_failed;
|
|
|
|
stats.req_failed += req_not_issued;
|
|
|
|
stats.req_error += req_not_issued;
|
|
|
|
|
|
|
|
// UI is heavily inspired by weighttp[1] and wrk[2]
|
|
|
|
//
|
|
|
|
// [1] https://github.com/lighttpd/weighttp
|
|
|
|
// [2] https://github.com/wg/wrk
|
2015-07-13 20:28:15 +02:00
|
|
|
double rps = 0;
|
2015-01-31 10:13:07 +01:00
|
|
|
int64_t bps = 0;
|
|
|
|
if (duration.count() > 0) {
|
|
|
|
auto secd = static_cast<double>(duration.count()) / (1000 * 1000);
|
|
|
|
rps = stats.req_success / secd;
|
|
|
|
bps = stats.bytes_total / secd;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::cout << R"(
|
|
|
|
finished in )" << util::format_duration(duration) << ", " << rps << " req/s, "
|
|
|
|
<< util::utos_with_funit(bps) << R"(B/s
|
|
|
|
requests: )" << stats.req_todo << " total, " << stats.req_started
|
|
|
|
<< " started, " << stats.req_done << " done, "
|
|
|
|
<< stats.req_status_success << " succeeded, " << stats.req_failed
|
|
|
|
<< " failed, " << stats.req_error << R"( errored
|
|
|
|
status codes: )" << stats.status[2] << " 2xx, " << stats.status[3] << " 3xx, "
|
|
|
|
<< stats.status[4] << " 4xx, " << stats.status[5] << R"( 5xx
|
|
|
|
traffic: )" << stats.bytes_total << " bytes total, " << stats.bytes_head
|
|
|
|
<< " bytes headers, " << stats.bytes_body << R"( bytes data
|
|
|
|
min max mean sd +/- sd
|
2015-05-04 04:48:21 +02:00
|
|
|
time for request: )" << std::setw(10) << util::format_duration(ts.request.min)
|
|
|
|
<< " " << std::setw(10) << util::format_duration(ts.request.max)
|
|
|
|
<< " " << std::setw(10) << util::format_duration(ts.request.mean)
|
|
|
|
<< " " << std::setw(10) << util::format_duration(ts.request.sd)
|
|
|
|
<< std::setw(9) << util::dtos(ts.request.within_sd) << "%"
|
2015-05-03 09:47:32 +02:00
|
|
|
<< "\ntime for connect: " << std::setw(10)
|
2015-05-04 04:48:21 +02:00
|
|
|
<< util::format_duration(ts.connect.min) << " " << std::setw(10)
|
|
|
|
<< util::format_duration(ts.connect.max) << " " << std::setw(10)
|
|
|
|
<< util::format_duration(ts.connect.mean) << " " << std::setw(10)
|
|
|
|
<< util::format_duration(ts.connect.sd) << std::setw(9)
|
|
|
|
<< util::dtos(ts.connect.within_sd) << "%"
|
2015-05-03 09:47:32 +02:00
|
|
|
<< "\ntime to 1st byte: " << std::setw(10)
|
2015-05-04 04:48:21 +02:00
|
|
|
<< util::format_duration(ts.ttfb.min) << " " << std::setw(10)
|
|
|
|
<< util::format_duration(ts.ttfb.max) << " " << std::setw(10)
|
|
|
|
<< util::format_duration(ts.ttfb.mean) << " " << std::setw(10)
|
|
|
|
<< util::format_duration(ts.ttfb.sd) << std::setw(9)
|
|
|
|
<< util::dtos(ts.ttfb.within_sd) << "%" << std::endl;
|
2014-06-19 17:00:06 +02:00
|
|
|
|
|
|
|
SSL_CTX_free(ssl_ctx);
|
|
|
|
|
2014-02-28 01:10:42 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace h2load
|
|
|
|
|
2014-11-27 15:39:04 +01:00
|
|
|
int main(int argc, char **argv) { return h2load::main(argc, argv); }
|