1c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Copyright (c) 2012 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" 1490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "base/strings/string_number_conversions.h" 15c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/base/ip_endpoint.h" 1690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "net/tools/quic/quic_in_memory_cache.h" 17c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/tools/quic/quic_server.h" 18c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 19c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// The port the quic server will listen on. 2090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 21c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)int32 FLAGS_port = 6121; 22c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 23c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)int main(int argc, char *argv[]) { 2446d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) base::CommandLine::Init(argc, argv); 2546d4c2bc3267f3f028f39e7e311b0f89aba2e4fdTorne (Richard Coles) base::CommandLine* line = base::CommandLine::ForCurrentProcess(); 265d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 275d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) logging::LoggingSettings settings; 285d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG; 295d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) CHECK(logging::InitLogging(settings)); 305d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 314e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) if (line->HasSwitch("h") || line->HasSwitch("help")) { 324e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const char* help_str = 334e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "Usage: quic_server [options]\n" 344e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "\n" 354e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "Options:\n" 364e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "-h, --help show this help message and exit\n" 374e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "--port=<port> specify the port to listen on\n" 384e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "--quic_in_memory_cache_dir directory containing response data\n" 394e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) " to load\n"; 404e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) std::cout << help_str; 414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) exit(0); 424e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) } 434e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 4490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (line->HasSwitch("quic_in_memory_cache_dir")) { 4590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) net::tools::FLAGS_quic_in_memory_cache_dir = 4690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) line->GetSwitchValueASCII("quic_in_memory_cache_dir"); 4790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 4890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 4990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (line->HasSwitch("port")) { 5090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) int port; 5190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (base::StringToInt(line->GetSwitchValueASCII("port"), &port)) { 5290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) FLAGS_port = port; 5390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 5490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 55c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 56c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::AtExitManager exit_manager; 57c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 58c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::IPAddressNumber ip; 59c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) CHECK(net::ParseIPLiteralToNumber("::", &ip)); 60c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 61c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::tools::QuicServer server; 62c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 63c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!server.Listen(net::IPEndPoint(ip, FLAGS_port))) { 64c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return 1; 65c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 66c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 67c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) while (1) { 68c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) server.WaitForEvents(); 69c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 70c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 71c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return 0; 72c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 73