| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // | 4 // |
| 5 // A binary wrapper for QuicServer. It listens forever on --port | 5 // A binary wrapper for QuicServer. It listens forever on --port |
| 6 // (default 6121) until it's killed or ctrl-cd to death. | 6 // (default 6121) until it's killed or ctrl-cd to death. |
| 7 | 7 |
| 8 #include <iostream> | 8 #include <iostream> |
| 9 | 9 |
| 10 #include "base/at_exit.h" | 10 #include "base/at_exit.h" |
| 11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
| 12 #include "base/command_line.h" | 12 #include "base/command_line.h" |
| 13 #include "base/logging.h" | 13 #include "base/logging.h" |
| 14 #include "base/run_loop.h" |
| 14 #include "base/strings/string_number_conversions.h" | 15 #include "base/strings/string_number_conversions.h" |
| 15 #include "net/base/ip_endpoint.h" | 16 #include "net/base/ip_endpoint.h" |
| 16 #include "net/tools/quic/quic_in_memory_cache.h" | 17 #include "net/quic/quic_in_memory_cache.h" |
| 17 #include "net/tools/quic/quic_server.h" | 18 #include "net/quic/quic_protocol.h" |
| 19 #include "net/quic/quic_server.h" |
| 18 | 20 |
| 19 // The port the quic server will listen on. | 21 // The port the quic server will listen on. |
| 20 | |
| 21 int32 FLAGS_port = 6121; | 22 int32 FLAGS_port = 6121; |
| 22 | 23 |
| 23 int main(int argc, char *argv[]) { | 24 int main(int argc, char *argv[]) { |
| 24 base::CommandLine::Init(argc, argv); | 25 base::CommandLine::Init(argc, argv); |
| 25 base::CommandLine* line = base::CommandLine::ForCurrentProcess(); | 26 base::CommandLine* line = base::CommandLine::ForCurrentProcess(); |
| 26 | 27 |
| 27 logging::LoggingSettings settings; | 28 logging::LoggingSettings settings; |
| 28 settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG; | 29 settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG; |
| 29 CHECK(logging::InitLogging(settings)); | 30 CHECK(logging::InitLogging(settings)); |
| 30 | 31 |
| 31 if (line->HasSwitch("h") || line->HasSwitch("help")) { | 32 if (line->HasSwitch("h") || line->HasSwitch("help")) { |
| 32 const char* help_str = | 33 const char* help_str = |
| 33 "Usage: quic_server [options]\n" | 34 "Usage: quic_server [options]\n" |
| 34 "\n" | 35 "\n" |
| 35 "Options:\n" | 36 "Options:\n" |
| 36 "-h, --help show this help message and exit\n" | 37 "-h, --help show this help message and exit\n" |
| 37 "--port=<port> specify the port to listen on\n" | 38 "--port=<port> specify the port to listen on\n" |
| 38 "--quic_in_memory_cache_dir directory containing response data\n" | 39 "--quic_in_memory_cache_dir directory containing response data\n" |
| 39 " to load\n"; | 40 " to load\n"; |
| 40 std::cout << help_str; | 41 std::cout << help_str; |
| 41 exit(0); | 42 exit(0); |
| 42 } | 43 } |
| 43 | 44 |
| 44 if (line->HasSwitch("quic_in_memory_cache_dir")) { | 45 if (line->HasSwitch("quic_in_memory_cache_dir")) { |
| 45 net::tools::FLAGS_quic_in_memory_cache_dir = | 46 net::g_quic_in_memory_cache_dir = |
| 46 line->GetSwitchValueASCII("quic_in_memory_cache_dir"); | 47 line->GetSwitchValueNative("quic_in_memory_cache_dir"); |
| 47 } | 48 } |
| 48 | 49 |
| 49 if (line->HasSwitch("port")) { | 50 if (line->HasSwitch("port")) { |
| 50 int port; | 51 int port; |
| 51 if (base::StringToInt(line->GetSwitchValueASCII("port"), &port)) { | 52 if (base::StringToInt(line->GetSwitchValueASCII("port"), &port)) { |
| 52 FLAGS_port = port; | 53 FLAGS_port = port; |
| 53 } | 54 } |
| 54 } | 55 } |
| 55 | 56 |
| 56 base::AtExitManager exit_manager; | 57 base::AtExitManager exit_manager; |
| 57 | 58 |
| 59 base::MessageLoopForIO message_loop; |
| 60 |
| 58 net::IPAddressNumber ip; | 61 net::IPAddressNumber ip; |
| 59 CHECK(net::ParseIPLiteralToNumber("::", &ip)); | 62 CHECK(net::ParseIPLiteralToNumber("::", &ip)); |
| 60 | 63 |
| 61 net::tools::QuicServer server; | 64 net::QuicConfig config; |
| 65 config.SetDefaults(); |
| 62 | 66 |
| 63 if (!server.Listen(net::IPEndPoint(ip, FLAGS_port))) { | 67 net::QuicServer server(config, net::QuicSupportedVersions()); |
| 68 |
| 69 int rc = server.Listen(net::IPEndPoint(ip, FLAGS_port)); |
| 70 if (rc < 0) { |
| 64 return 1; | 71 return 1; |
| 65 } | 72 } |
| 66 | 73 |
| 67 while (1) { | 74 base::RunLoop().Run(); |
| 68 server.WaitForEvents(); | |
| 69 } | |
| 70 | 75 |
| 71 return 0; | 76 return 0; |
| 72 } | 77 } |
| OLD | NEW |