1116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch// Copyright 2014 The Chromium Authors. All rights reserved.
2c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
3c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// found in the LICENSE file.
4c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)//
5c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// A binary wrapper for QuicServer.  It listens forever on --port
6c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// (default 6121) until it's killed or ctrl-cd to death.
7c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
84e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include <iostream>
94e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
10c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/at_exit.h"
11c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/basictypes.h"
12c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/command_line.h"
135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "base/logging.h"
14116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "base/run_loop.h"
1590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "base/strings/string_number_conversions.h"
16c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/base/ip_endpoint.h"
17116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "net/quic/quic_in_memory_cache.h"
18116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "net/quic/quic_protocol.h"
19116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "net/quic/quic_server.h"
20c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
21c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// The port the quic server will listen on.
22c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)int32 FLAGS_port = 6121;
23c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
24c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)int main(int argc, char *argv[]) {
2546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)  base::CommandLine::Init(argc, argv);
2646d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles)  base::CommandLine* line = base::CommandLine::ForCurrentProcess();
275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  logging::LoggingSettings settings;
295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG;
305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  CHECK(logging::InitLogging(settings));
315d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
324e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  if (line->HasSwitch("h") || line->HasSwitch("help")) {
334e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    const char* help_str =
344e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        "Usage: quic_server [options]\n"
354e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        "\n"
364e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        "Options:\n"
374e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        "-h, --help                  show this help message and exit\n"
384e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        "--port=<port>               specify the port to listen on\n"
394e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        "--quic_in_memory_cache_dir  directory containing response data\n"
404e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)        "                            to load\n";
414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    std::cout << help_str;
424e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)    exit(0);
434e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)  }
444e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)
4590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  if (line->HasSwitch("quic_in_memory_cache_dir")) {
46116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch    net::g_quic_in_memory_cache_dir =
47116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch        line->GetSwitchValueNative("quic_in_memory_cache_dir");
4890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  }
4990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)
5090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  if (line->HasSwitch("port")) {
5190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)    int port;
5290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)    if (base::StringToInt(line->GetSwitchValueASCII("port"), &port)) {
5390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)      FLAGS_port = port;
5490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)    }
5590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)  }
56c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
57c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  base::AtExitManager exit_manager;
58c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
59116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch  base::MessageLoopForIO message_loop;
60116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch
61c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  net::IPAddressNumber ip;
62c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  CHECK(net::ParseIPLiteralToNumber("::", &ip));
63c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
64116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch  net::QuicConfig config;
65116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch  config.SetDefaults();
66c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
67116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch  net::QuicServer server(config, net::QuicSupportedVersions());
68116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch
69116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch  int rc = server.Listen(net::IPEndPoint(ip, FLAGS_port));
70116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch  if (rc < 0) {
71c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)    return 1;
72c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  }
73c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
74116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch  base::RunLoop().Run();
75c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)
76c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)  return 0;
77c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)}
78