src/net/tools/quic/quic_client_bin.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 // A binary wrapper for QuicClient.  Connects to --hostname via --address
   6 // on --port and requests URLs specified on the command line.
   7 //
   8 // For example:
   9 //  quic_client --address=127.0.0.1 --port=6122 --hostname=www.google.com
  10 //      http://www.google.com/index.html http://www.google.com/favicon.ico
  11
  12 #include <iostream>
  13
  14 #include "base/at_exit.h"
  15 #include "base/command_line.h"
  16 #include "base/logging.h"
  17 #include "base/strings/string_number_conversions.h"
  18 #include "net/base/ip_endpoint.h"
  19 #include "net/quic/quic_protocol.h"
  20 #include "net/tools/quic/quic_client.h"
  21
  22 int32 FLAGS_port = 6121;
  23 std::string FLAGS_address = "127.0.0.1";
  24 std::string FLAGS_hostname = "localhost";
  25
  26 int main(int argc, char *argv[]) {
  27   CommandLine::Init(argc, argv);
  28   CommandLine* line = CommandLine::ForCurrentProcess();
  29
  30   logging::LoggingSettings settings;
  31   settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG;
  32   CHECK(logging::InitLogging(settings));
  33
  34   if (line->HasSwitch("h") || line->HasSwitch("help")) {
  35     const char* help_str =
  36         "Usage: quic_client [options]\n"
  37         "\n"
  38         "Options:\n"
  39         "-h, --help                  show this help message and exit\n"
  40         "--port=<port>               specify the port to connect to\n"
  41         "--address=<address>         specify the IP address to connect to\n"
  42         "--host=<host>               specify the SNI hostname to use\n";
  43     std::cout << help_str;
  44     exit(0);
  45   }
  46   if (line->HasSwitch("port")) {
  47     int port;
  48     if (base::StringToInt(line->GetSwitchValueASCII("port"), &port)) {
  49       FLAGS_port = port;
  50     }
  51   }
  52   if (line->HasSwitch("address")) {
  53     FLAGS_address = line->GetSwitchValueASCII("address");
  54   }
  55   if (line->HasSwitch("hostname")) {
  56     FLAGS_hostname = line->GetSwitchValueASCII("hostname");
  57   }
  58   VLOG(1) << "server port: " << FLAGS_port
  59           << " address: " << FLAGS_address
  60           << " hostname: " << FLAGS_hostname;
  61
  62   base::AtExitManager exit_manager;
  63
  64   net::IPAddressNumber addr;
  65   CHECK(net::ParseIPLiteralToNumber(FLAGS_address, &addr));
  66   // TODO(rjshade): Set version on command line.
  67   net::tools::QuicClient client(
  68       net::IPEndPoint(addr, FLAGS_port), FLAGS_hostname,
  69       net::QuicSupportedVersions(), true);
  70
  71   client.Initialize();
  72
  73   if (!client.Connect()) return 1;
  74
  75   client.SendRequestsAndWaitForResponse(line->GetArgs());
  76   return 0;
  77 }