diff --git a/examples/spdycat.cc b/examples/spdycat.cc index 01349a7..f54be6d 100644 --- a/examples/spdycat.cc +++ b/examples/spdycat.cc @@ -66,22 +66,46 @@ struct Config { bool remote_name; bool verbose; bool get_assets; + bool stat; int spdy_version; int timeout; std::string certfile; std::string keyfile; int window_bits; Config():null_out(false), remote_name(false), verbose(false), - get_assets(false), spdy_version(-1), timeout(-1), window_bits(-1) + get_assets(false), stat(false), + spdy_version(-1), timeout(-1), window_bits(-1) {} }; +struct RequestStat { + timeval on_syn_stream_time; + timeval on_syn_reply_time; + timeval on_complete_time; + RequestStat() + { + on_syn_stream_time.tv_sec = -1; + on_syn_stream_time.tv_usec = -1; + on_syn_reply_time.tv_sec = -1; + on_syn_reply_time.tv_usec = -1; + on_complete_time.tv_sec = -1; + on_complete_time.tv_usec = -1; + } +}; + +void record_time(timeval *tv) +{ + gettimeofday(tv, 0); +} + struct Request { uri::UriStruct us; spdylay_gzip *inflater; HtmlParser *html_parser; // Recursion level: 0: first entity, 1: entity linked from first entity int level; + RequestStat stat; + std::string status; Request(const uri::UriStruct& us, int level = 0) : us(us), inflater(0), html_parser(0), level(level) {} @@ -114,6 +138,30 @@ struct Request { fin); return rv; } + + void record_syn_stream_time() + { + record_time(&stat.on_syn_stream_time); + } + + void record_syn_reply_time() + { + record_time(&stat.on_syn_reply_time); + } + + void record_complete_time() + { + record_time(&stat.on_complete_time); + } +}; + +struct SessionStat { + timeval on_handshake_time; + SessionStat() + { + on_handshake_time.tv_sec = -1; + on_handshake_time.tv_usec = -1; + } }; struct SpdySession { @@ -127,6 +175,7 @@ struct SpdySession { size_t complete; std::string hostport; Spdylay *sc; + SessionStat stat; SpdySession():complete(0) {} ~SpdySession() { @@ -165,6 +214,10 @@ struct SpdySession { return true; } } + void record_handshake_time() + { + record_time(&stat.on_handshake_time); + } }; Config config; @@ -251,6 +304,7 @@ void check_stream_id(spdylay_session *session, Request *req = (Request*)spdylay_session_get_stream_user_data(session, stream_id); spdySession->streams[stream_id] = req; + req->record_syn_stream_time(); } void on_ctrl_send_callback2 @@ -260,16 +314,9 @@ void on_ctrl_send_callback2 if(type == SPDYLAY_SYN_STREAM) { check_stream_id(session, type, frame, user_data); } -} - -void on_ctrl_send_callback3 -(spdylay_session *session, spdylay_frame_type type, spdylay_frame *frame, - void *user_data) -{ - if(type == SPDYLAY_SYN_STREAM) { - check_stream_id(session, type, frame, user_data); + if(config.verbose) { + on_ctrl_send_callback(session, type, frame, user_data); } - on_ctrl_send_callback(session, type, frame, user_data); } void check_response_header @@ -294,6 +341,8 @@ void check_response_header for(size_t i = 0; nv[i]; i += 2) { if(strcmp("content-encoding", nv[i]) == 0) { gzip = util::strieq("gzip", nv[i+1]) || util::strieq("deflate", nv[i+1]); + } else if(strcmp(":status", nv[i]) == 0) { + req->status = nv[i+1]; } } if(gzip) { @@ -312,15 +361,16 @@ void on_ctrl_recv_callback2 (spdylay_session *session, spdylay_frame_type type, spdylay_frame *frame, void *user_data) { + if(type == SPDYLAY_SYN_REPLY) { + Request *req = (Request*)spdylay_session_get_stream_user_data + (session, frame->syn_reply.stream_id); + assert(req); + req->record_syn_reply_time(); + } check_response_header(session, type, frame, user_data); -} - -void on_ctrl_recv_callback3 -(spdylay_session *session, spdylay_frame_type type, spdylay_frame *frame, - void *user_data) -{ - check_response_header(session, type, frame, user_data); - on_ctrl_recv_callback(session, type, frame, user_data); + if(config.verbose) { + on_ctrl_recv_callback(session, type, frame, user_data); + } } void on_stream_close_callback @@ -332,6 +382,7 @@ void on_stream_close_callback spdySession->streams.find(stream_id); if(itr != spdySession->streams.end()) { update_html_parser(spdySession, (*itr).second, 0, 0, 1); + (*itr).second->record_complete_time(); ++spdySession->complete; if(spdySession->all_requests_processed()) { spdylay_submit_goaway(session, SPDYLAY_GOAWAY_OK); @@ -339,6 +390,52 @@ void on_stream_close_callback } } +int64_t time_delta(const timeval& a, const timeval& b) +{ + int64_t res; + res = a.tv_usec - b.tv_usec; + int off = 0; + if(res < 0) { + res += 1000000; + off = -1; + } + res += (a.tv_sec + off - b.tv_sec)*1000000; + return res/1000; +} + +void print_stats(const SpdySession& spdySession) +{ + std::cout << "***** Statistics *****" << std::endl; + for(size_t i = 0; i < spdySession.reqvec.size(); ++i) { + const Request *req = spdySession.reqvec[i]; + std::cout << "#" << i+1 << ": " << uri::construct(req->us) << std::endl; + std::cout << " Status: " << req->status << std::endl; + std::cout << " Delta (ms) from SSL/TLS handshake(SYN_STREAM):" + << std::endl; + if(req->stat.on_syn_reply_time.tv_sec >= 0) { + std::cout << " SYN_REPLY: " + << time_delta(req->stat.on_syn_reply_time, + spdySession.stat.on_handshake_time) + << "(" + << time_delta(req->stat.on_syn_reply_time, + req->stat.on_syn_stream_time) + << ")" + << std::endl; + } + if(req->stat.on_complete_time.tv_sec >= 0) { + std::cout << " Completed: " + << time_delta(req->stat.on_complete_time, + spdySession.stat.on_handshake_time) + << "(" + << time_delta(req->stat.on_complete_time, + req->stat.on_syn_stream_time) + << ")" + << std::endl; + } + std::cout << std::endl; + } +} + int communicate(const std::string& host, uint16_t port, SpdySession& spdySession, const spdylay_session_callbacks *callbacks) @@ -386,6 +483,7 @@ int communicate(const std::string& host, uint16_t port, if(ssl_handshake(ssl, fd) == -1) { return -1; } + spdySession.record_handshake_time(); make_non_block(fd); set_tcp_nodelay(fd); int spdy_version = spdylay_npn_get_version( @@ -459,6 +557,9 @@ int communicate(const std::string& host, uint16_t port, SSL_CTX_free(ssl_ctx); shutdown(fd, SHUT_WR); close(fd); + if(config.stat) { + print_stats(spdySession); + } return ok ? 0 : -1; } @@ -469,17 +570,14 @@ int run(char **uris, int n) callbacks.send_callback = send_callback; callbacks.recv_callback = recv_callback; callbacks.on_stream_close_callback = on_stream_close_callback; + callbacks.on_ctrl_recv_callback = on_ctrl_recv_callback2; + callbacks.on_ctrl_send_callback = on_ctrl_send_callback2; if(config.verbose) { - callbacks.on_ctrl_recv_callback = on_ctrl_recv_callback3; callbacks.on_data_recv_callback = on_data_recv_callback; - callbacks.on_ctrl_send_callback = on_ctrl_send_callback3; callbacks.on_invalid_ctrl_recv_callback = on_invalid_ctrl_recv_callback; callbacks.on_ctrl_recv_parse_error_callback = on_ctrl_recv_parse_error_callback; callbacks.on_unknown_ctrl_recv_callback = on_unknown_ctrl_recv_callback; - } else { - callbacks.on_ctrl_recv_callback = on_ctrl_recv_callback2; - callbacks.on_ctrl_send_callback = on_ctrl_send_callback2; } callbacks.on_data_chunk_recv_callback = on_data_chunk_recv_callback; ssl_debug = config.verbose; @@ -515,7 +613,7 @@ int run(char **uris, int n) void print_usage(std::ostream& out) { - out << "Usage: spdycat [-Oanv23] [-t ] [-w ] [--cert=]\n" + out << "Usage: spdycat [-Oansv23] [-t ] [-w ] [--cert=]\n" << " [--key=] ..." << std::endl; } @@ -542,6 +640,7 @@ void print_help(std::ostream& out) << " resource. Only links whose origins are the\n" << " same with the linking resource will be\n" << " downloaded.\n" + << " -s, --stat Print statistics.\n" << " --cert= Use the specified client certificate file.\n" << " The file must be in PEM format.\n" << " --key= Use the client private key file. The file\n" @@ -562,13 +661,15 @@ int main(int argc, char **argv) {"timeout", required_argument, 0, 't' }, {"window-bits", required_argument, 0, 'w' }, {"get-assets", no_argument, 0, 'a' }, + {"stat", no_argument, 0, 's' }, {"cert", required_argument, &flag, 1 }, {"key", required_argument, &flag, 2 }, {"help", no_argument, 0, 'h' }, {0, 0, 0, 0 } }; int option_index = 0; - int c = getopt_long(argc, argv, "Oanhv23t:w:", long_options, &option_index); + int c = getopt_long(argc, argv, "Oanhv23st:w:", long_options, + &option_index); if(c == -1) { break; } @@ -615,6 +716,9 @@ int main(int argc, char **argv) << std::endl; #endif // !HAVE_LIBXML2 break; + case 's': + config.stat = true; + break; case '?': exit(EXIT_FAILURE); case 0: