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 |