OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #include <stdio.h> | 5 #include <stdio.h> |
6 | 6 |
7 #include "base/at_exit.h" | 7 #include "base/at_exit.h" |
8 #include "base/command_line.h" | 8 #include "base/command_line.h" |
9 #include "base/file_path.h" | 9 #include "base/file_path.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 #include "base/message_loop.h" | 11 #include "base/message_loop.h" |
12 #include "base/process_util.h" | |
13 #include "base/string_number_conversions.h" | |
14 #include "base/test/test_timeouts.h" | 12 #include "base/test/test_timeouts.h" |
15 #include "base/utf_string_conversions.h" | 13 #include "base/utf_string_conversions.h" |
16 #include "net/test/local_sync_test_server.h" | |
17 #include "net/test/python_utils.h" | |
18 #include "net/test/test_server.h" | 14 #include "net/test/test_server.h" |
19 | 15 |
20 static void PrintUsage() { | 16 static void PrintUsage() { |
21 printf("run_testserver --doc-root=relpath\n" | 17 printf("run_testserver --doc-root=relpath\n" |
22 " [--http|--https|--ws|--wss|--ftp|--sync]\n" | 18 " [--http|--https|--ws|--wss|--ftp]\n" |
23 " [--ssl-cert=ok|mismatched-name|expired]\n" | 19 " [--ssl-cert=ok|mismatched-name|expired]\n"); |
24 " [--port=<port>] [--xmpp-port=<xmpp_port>]\n"); | |
25 printf("(NOTE: relpath should be relative to the 'src' directory.\n"); | 20 printf("(NOTE: relpath should be relative to the 'src' directory.\n"); |
26 printf(" --port and --xmpp-port only work with the --sync flag.)\n"); | |
27 } | |
28 | |
29 // Launches the chromiumsync_test script, testing the --sync functionality. | |
30 static bool RunSyncTest() { | |
31 if (!net::TestServer::SetPythonPathStatic()) { | |
32 LOG(ERROR) << "Error trying to set python path. Exiting."; | |
33 return false; | |
34 } | |
35 | |
36 FilePath sync_test_path; | |
37 if (!net::TestServer::GetTestServerDirectory(&sync_test_path)) { | |
38 LOG(ERROR) << "Error trying to get python test server path."; | |
39 return false; | |
40 } | |
41 | |
42 sync_test_path = | |
43 sync_test_path.Append(FILE_PATH_LITERAL("chromiumsync_test.py")); | |
44 | |
45 CommandLine python_command(CommandLine::NO_PROGRAM); | |
46 if (!GetPythonCommand(&python_command)) { | |
47 LOG(ERROR) << "Could not get python runtime command."; | |
48 return false; | |
49 } | |
50 | |
51 python_command.AppendArgPath(sync_test_path); | |
52 if (!base::LaunchProcess(python_command, base::LaunchOptions(), NULL)) { | |
53 LOG(ERROR) << "Failed to launch test script."; | |
54 return false; | |
55 } | |
56 return true; | |
57 } | |
58 | |
59 // Gets a port value from the switch with name |switch_name| and writes it to | |
60 // |port|. Returns true if successful and false otherwise. | |
61 static bool GetPortFromSwitch(const std::string& switch_name, uint16* port) { | |
62 DCHECK(port != NULL) << "|port| is NULL"; | |
63 CommandLine* command_line = CommandLine::ForCurrentProcess(); | |
64 int port_int = 0; | |
65 if (command_line->HasSwitch(switch_name)) { | |
66 std::string port_str = command_line->GetSwitchValueASCII(switch_name); | |
67 if (!base::StringToInt(port_str, &port_int)) { | |
68 LOG(WARNING) << "Could not extract port from switch " << switch_name; | |
69 return false; | |
70 } | |
71 } | |
72 *port = static_cast<uint16>(port_int); | |
73 return true; | |
74 } | 21 } |
75 | 22 |
76 int main(int argc, const char* argv[]) { | 23 int main(int argc, const char* argv[]) { |
77 base::AtExitManager at_exit_manager; | 24 base::AtExitManager at_exit_manager; |
78 MessageLoopForIO message_loop; | 25 MessageLoopForIO message_loop; |
79 | 26 |
80 // Process command line | 27 // Process command line |
81 CommandLine::Init(argc, argv); | 28 CommandLine::Init(argc, argv); |
82 CommandLine* command_line = CommandLine::ForCurrentProcess(); | 29 CommandLine* command_line = CommandLine::ForCurrentProcess(); |
83 | 30 |
84 if (!logging::InitLogging( | 31 if (!logging::InitLogging( |
85 FILE_PATH_LITERAL("testserver.log"), | 32 FILE_PATH_LITERAL("testserver.log"), |
86 logging::LOG_TO_BOTH_FILE_AND_SYSTEM_DEBUG_LOG, | 33 logging::LOG_TO_BOTH_FILE_AND_SYSTEM_DEBUG_LOG, |
87 logging::LOCK_LOG_FILE, | 34 logging::LOCK_LOG_FILE, |
88 logging::APPEND_TO_OLD_LOG_FILE, | 35 logging::APPEND_TO_OLD_LOG_FILE, |
89 logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS)) { | 36 logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS)) { |
90 printf("Error: could not initialize logging. Exiting.\n"); | 37 printf("Error: could not initialize logging. Exiting.\n"); |
91 return -1; | 38 return -1; |
92 } | 39 } |
93 | 40 |
94 TestTimeouts::Initialize(); | 41 TestTimeouts::Initialize(); |
95 | 42 |
96 if (command_line->GetSwitches().empty() || | 43 if (command_line->GetSwitches().empty() || |
97 command_line->HasSwitch("help") || | 44 command_line->HasSwitch("help")) { |
98 ((command_line->HasSwitch("port") || | |
99 command_line->HasSwitch("xmpp-port")) && | |
100 !command_line->HasSwitch("sync"))) { | |
101 PrintUsage(); | 45 PrintUsage(); |
102 return -1; | 46 return -1; |
103 } | 47 } |
104 | 48 |
105 net::TestServer::Type server_type; | 49 net::TestServer::Type server_type; |
106 if (command_line->HasSwitch("http")) { | 50 if (command_line->HasSwitch("http")) { |
107 server_type = net::TestServer::TYPE_HTTP; | 51 server_type = net::TestServer::TYPE_HTTP; |
108 } else if (command_line->HasSwitch("https")) { | 52 } else if (command_line->HasSwitch("https")) { |
109 server_type = net::TestServer::TYPE_HTTPS; | 53 server_type = net::TestServer::TYPE_HTTPS; |
110 } else if (command_line->HasSwitch("ws")) { | 54 } else if (command_line->HasSwitch("ws")) { |
111 server_type = net::TestServer::TYPE_WS; | 55 server_type = net::TestServer::TYPE_WS; |
112 } else if (command_line->HasSwitch("wss")) { | 56 } else if (command_line->HasSwitch("wss")) { |
113 server_type = net::TestServer::TYPE_WSS; | 57 server_type = net::TestServer::TYPE_WSS; |
114 } else if (command_line->HasSwitch("ftp")) { | 58 } else if (command_line->HasSwitch("ftp")) { |
115 server_type = net::TestServer::TYPE_FTP; | 59 server_type = net::TestServer::TYPE_FTP; |
116 } else if (command_line->HasSwitch("sync")) { | |
117 server_type = net::TestServer::TYPE_SYNC; | |
118 } else if (command_line->HasSwitch("sync-test")) { | |
119 return RunSyncTest() ? 0 : -1; | |
120 } else { | 60 } else { |
121 // If no scheme switch is specified, select http or https scheme. | 61 // If no scheme switch is specified, select http or https scheme. |
122 // TODO(toyoshim): Remove this estimation. | 62 // TODO(toyoshim): Remove this estimation. |
123 if (command_line->HasSwitch("ssl-cert")) | 63 if (command_line->HasSwitch("ssl-cert")) |
124 server_type = net::TestServer::TYPE_HTTPS; | 64 server_type = net::TestServer::TYPE_HTTPS; |
125 else | 65 else |
126 server_type = net::TestServer::TYPE_HTTP; | 66 server_type = net::TestServer::TYPE_HTTP; |
127 } | 67 } |
128 | 68 |
129 net::TestServer::SSLOptions ssl_options; | 69 net::TestServer::SSLOptions ssl_options; |
(...skipping 13 matching lines...) Expand all Loading... |
143 ssl_options.server_certificate = | 83 ssl_options.server_certificate = |
144 net::TestServer::SSLOptions::CERT_EXPIRED; | 84 net::TestServer::SSLOptions::CERT_EXPIRED; |
145 } else { | 85 } else { |
146 printf("Error: --ssl-cert has invalid value %s\n", cert_option.c_str()); | 86 printf("Error: --ssl-cert has invalid value %s\n", cert_option.c_str()); |
147 PrintUsage(); | 87 PrintUsage(); |
148 return -1; | 88 return -1; |
149 } | 89 } |
150 } | 90 } |
151 | 91 |
152 FilePath doc_root = command_line->GetSwitchValuePath("doc-root"); | 92 FilePath doc_root = command_line->GetSwitchValuePath("doc-root"); |
153 if ((server_type != net::TestServer::TYPE_SYNC) && doc_root.empty()) { | 93 if (doc_root.empty()) { |
154 printf("Error: --doc-root must be specified\n"); | 94 printf("Error: --doc-root must be specified\n"); |
155 PrintUsage(); | 95 PrintUsage(); |
156 return -1; | 96 return -1; |
157 } | 97 } |
158 | 98 |
159 scoped_ptr<net::TestServer> test_server; | 99 scoped_ptr<net::TestServer> test_server; |
160 if (net::TestServer::UsingSSL(server_type)) { | 100 if (net::TestServer::UsingSSL(server_type)) { |
161 test_server.reset(new net::TestServer(server_type, ssl_options, doc_root)); | 101 test_server.reset(new net::TestServer(server_type, ssl_options, doc_root)); |
162 } else if (server_type == net::TestServer::TYPE_SYNC) { | |
163 uint16 port = 0; | |
164 uint16 xmpp_port = 0; | |
165 if (!GetPortFromSwitch("port", &port) || | |
166 !GetPortFromSwitch("xmpp-port", &xmpp_port)) { | |
167 printf("Error: Could not extract --port and/or --xmpp-port.\n"); | |
168 return -1; | |
169 } | |
170 test_server.reset(new net::LocalSyncTestServer(port, xmpp_port)); | |
171 } else { | 102 } else { |
172 test_server.reset(new net::TestServer(server_type, | 103 test_server.reset(new net::TestServer(server_type, |
173 net::TestServer::kLocalhost, | 104 net::TestServer::kLocalhost, |
174 doc_root)); | 105 doc_root)); |
175 } | 106 } |
176 | 107 |
177 if (!test_server->Start()) { | 108 if (!test_server->Start()) { |
178 printf("Error: failed to start test server. Exiting.\n"); | 109 printf("Error: failed to start test server. Exiting.\n"); |
179 return -1; | 110 return -1; |
180 } | 111 } |
181 | 112 |
182 if (!file_util::DirectoryExists(test_server->document_root())) { | 113 if (!file_util::DirectoryExists(test_server->document_root())) { |
183 printf("Error: invalid doc root: \"%s\" does not exist!\n", | 114 printf("Error: invalid doc root: \"%s\" does not exist!\n", |
184 UTF16ToUTF8(test_server->document_root().LossyDisplayName()).c_str()); | 115 UTF16ToUTF8(test_server->document_root().LossyDisplayName()).c_str()); |
185 return -1; | 116 return -1; |
186 } | 117 } |
187 | 118 |
188 printf("testserver running at %s (type ctrl+c to exit)\n", | 119 printf("testserver running at %s (type ctrl+c to exit)\n", |
189 test_server->host_port_pair().ToString().c_str()); | 120 test_server->host_port_pair().ToString().c_str()); |
190 | 121 |
191 message_loop.Run(); | 122 message_loop.Run(); |
192 return 0; | 123 return 0; |
193 } | 124 } |
OLD | NEW |