// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
-// A binary wrapper for QuicClient. Connects to --hostname via --address
-// on --port and requests URLs specified on the command line.
+// A binary wrapper for QuicClient.
+// Connects to a host using QUIC, and sends requests to the provided URLS.
//
-// For example:
+// Example usage:
// quic_client --address=127.0.0.1 --port=6122 --hostname=www.google.com
// http://www.google.com/index.html http://www.google.com/favicon.ico
#include "base/logging.h"
#include "base/strings/string_number_conversions.h"
#include "net/base/ip_endpoint.h"
+#include "net/base/privacy_mode.h"
#include "net/quic/quic_protocol.h"
+#include "net/quic/quic_server_id.h"
+#include "net/tools/epoll_server/epoll_server.h"
#include "net/tools/quic/quic_client.h"
-int32 FLAGS_port = 6121;
std::string FLAGS_address = "127.0.0.1";
+// The IP or hostname the quic client will connect to.
std::string FLAGS_hostname = "localhost";
+// The port the quic client will connect to.
+int32 FLAGS_port = 6121;
+// Check the certificates using proof verifier.
+bool FLAGS_secure = false;
+// QUIC version to speak, e.g. 21. Default value of 0 means 'use the latest
+// version'.
+int32 FLAGS_quic_version = 0;
+// Size of flow control receive window to advertize to the peer.
+int32 FLAGS_flow_control_window_bytes = 10 * 1024 * 1024; // 10 Mb
int main(int argc, char *argv[]) {
- CommandLine::Init(argc, argv);
- CommandLine* line = CommandLine::ForCurrentProcess();
+ base::CommandLine::Init(argc, argv);
+ base::CommandLine* line = base::CommandLine::ForCurrentProcess();
+ const base::CommandLine::StringVector& urls = line->GetArgs();
logging::LoggingSettings settings;
settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG;
CHECK(logging::InitLogging(settings));
- if (line->HasSwitch("h") || line->HasSwitch("help")) {
+ if (line->HasSwitch("h") || line->HasSwitch("help") || urls.empty()) {
const char* help_str =
- "Usage: quic_client [options]\n"
+ "Usage: quic_client [options] <url> ...\n"
"\n"
+ "At least one <url> with scheme must be provided "
+ "(e.g. http://www.google.com/)\n\n"
"Options:\n"
"-h, --help show this help message and exit\n"
"--port=<port> specify the port to connect to\n"
"--address=<address> specify the IP address to connect to\n"
- "--host=<host> specify the SNI hostname to use\n";
+ "--host=<host> specify the SNI hostname to use\n"
+ "--secure check certificates\n"
+ "--quic-version=<quic version> specify QUIC version to speak\n"
+ "--flow-control-window-bytes=<bytes> specify size of flow control "
+ "receive window to advertize to the peer\n";
std::cout << help_str;
exit(0);
}
if (line->HasSwitch("hostname")) {
FLAGS_hostname = line->GetSwitchValueASCII("hostname");
}
+ if (line->HasSwitch("secure")) {
+ FLAGS_secure = true;
+ }
+ if (line->HasSwitch("quic-version")) {
+ int quic_version;
+ if (base::StringToInt(line->GetSwitchValueASCII("quic-version"),
+ &quic_version)) {
+ FLAGS_quic_version = quic_version;
+ }
+ }
+ if (line->HasSwitch("flow-control-window-bytes")) {
+ int flow_control_window_bytes;
+ if (base::StringToInt(
+ line->GetSwitchValueASCII("flow-control-window-bytes"),
+ &flow_control_window_bytes)) {
+ FLAGS_flow_control_window_bytes = flow_control_window_bytes;
+ }
+ }
VLOG(1) << "server port: " << FLAGS_port
<< " address: " << FLAGS_address
- << " hostname: " << FLAGS_hostname;
+ << " hostname: " << FLAGS_hostname
+ << " secure: " << FLAGS_secure
+ << " quic-version: " << FLAGS_quic_version;
base::AtExitManager exit_manager;
+ // Determine IP address to connect to from supplied hostname.
net::IPAddressNumber addr;
CHECK(net::ParseIPLiteralToNumber(FLAGS_address, &addr));
- // TODO(rjshade): Set version on command line.
+
+ // Populate version vector with all versions if none specified.
+ net::QuicVersionVector versions;
+ if (FLAGS_quic_version == 0) {
+ versions = net::QuicSupportedVersions();
+ } else {
+ versions.push_back(static_cast<net::QuicVersion>(FLAGS_quic_version));
+ }
+
+ // Build the client, and try to connect.
+ VLOG(1) << "Conecting to " << FLAGS_hostname << ":" << FLAGS_port
+ << " with supported versions "
+ << QuicVersionVectorToString(versions);
+ net::EpollServer epoll_server;
+ net::QuicConfig config;
+
+ // The default flow control window of 16 Kb is too small for practical
+ // purposes. Set it to the specified value, which has a large default.
+ config.SetInitialFlowControlWindowToSend(
+ FLAGS_flow_control_window_bytes);
+ config.SetInitialStreamFlowControlWindowToSend(
+ FLAGS_flow_control_window_bytes);
+ config.SetInitialSessionFlowControlWindowToSend(
+ FLAGS_flow_control_window_bytes);
+
net::tools::QuicClient client(
- net::IPEndPoint(addr, FLAGS_port), FLAGS_hostname,
- net::QuicSupportedVersions(), true);
+ net::IPEndPoint(addr, FLAGS_port),
+ net::QuicServerId(FLAGS_hostname, FLAGS_port, FLAGS_secure,
+ net::PRIVACY_MODE_DISABLED),
+ versions, true, config, &epoll_server);
client.Initialize();
- if (!client.Connect()) return 1;
+ if (!client.Connect()) {
+ LOG(ERROR) << "Client failed to connect to host: "
+ << FLAGS_hostname << ":" << FLAGS_port;
+ return 1;
+ }
- client.SendRequestsAndWaitForResponse(line->GetArgs());
+ // Send a GET request for each supplied url.
+ client.SendRequestsAndWaitForResponse(urls);
return 0;
}