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 or --address on
   6 // --port and requests URLs specified on the command line.
   7 //
   8 // For example:
   9 //  quic_client --port=6122 /index.html /favicon.ico
  10
  11 #include "base/at_exit.h"
  12 #include "base/command_line.h"
  13 #include "base/logging.h"
  14 #include "base/strings/string_number_conversions.h"
  15 #include "net/base/ip_endpoint.h"
  16 #include "net/quic/quic_protocol.h"
  17 #include "net/tools/quic/quic_client.h"
  18
  19 int32 FLAGS_port = 6121;
  20 std::string FLAGS_address = "127.0.0.1";
  21 std::string FLAGS_hostname = "localhost";
  22
  23 int main(int argc, char *argv[]) {
  24   CommandLine::Init(argc, argv);
  25   CommandLine* line = CommandLine::ForCurrentProcess();
  26   if (line->HasSwitch("port")) {
  27     int port;
  28     if (base::StringToInt(line->GetSwitchValueASCII("port"), &port)) {
  29       FLAGS_port = port;
  30     }
  31   }
  32   if (line->HasSwitch("address")) {
  33     FLAGS_address = line->GetSwitchValueASCII("address");
  34   }
  35   if (line->HasSwitch("hostname")) {
  36     FLAGS_hostname = line->GetSwitchValueASCII("hostname");
  37   }
  38   LOG(INFO) << "server port: " << FLAGS_port
  39             << " address: " << FLAGS_address
  40             << " hostname: " << FLAGS_hostname;
  41
  42   base::AtExitManager exit_manager;
  43
  44   net::IPAddressNumber addr;
  45   CHECK(net::ParseIPLiteralToNumber(FLAGS_address, &addr));
  46   // TODO(rjshade): Set version on command line.
  47   net::tools::QuicClient client(
  48       net::IPEndPoint(addr, FLAGS_port), FLAGS_hostname, net::QuicVersionMax());
  49
  50   client.Initialize();
  51
  52   if (!client.Connect()) return 1;
  53
  54   client.SendRequestsAndWaitForResponse(line->GetArgs());
  55   return 0;
  56 }