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
3 // found in the LICENSE file.
5 #include "net/test/spawned_test_server/local_test_server.h"
7 #include "base/command_line.h"
8 #include "base/json/json_reader.h"
9 #include "base/logging.h"
10 #include "base/path_service.h"
11 #include "base/process/kill.h"
12 #include "base/strings/string_number_conversions.h"
13 #include "base/values.h"
14 #include "net/base/host_port_pair.h"
15 #include "net/base/net_errors.h"
16 #include "net/test/python_utils.h"
23 bool AppendArgumentFromJSONValue(const std::string& key,
24 const base::Value& value_node,
25 base::CommandLine* command_line) {
26 std::string argument_name = "--" + key;
27 switch (value_node.GetType()) {
28 case base::Value::TYPE_NULL:
29 command_line->AppendArg(argument_name);
31 case base::Value::TYPE_INTEGER: {
33 bool result = value_node.GetAsInteger(&value);
35 command_line->AppendArg(argument_name + "=" + base::IntToString(value));
38 case base::Value::TYPE_STRING: {
40 bool result = value_node.GetAsString(&value);
41 if (!result || value.empty())
43 command_line->AppendArg(argument_name + "=" + value);
46 case base::Value::TYPE_BOOLEAN:
47 case base::Value::TYPE_DOUBLE:
48 case base::Value::TYPE_LIST:
49 case base::Value::TYPE_DICTIONARY:
50 case base::Value::TYPE_BINARY:
52 NOTREACHED() << "improper json type";
60 LocalTestServer::LocalTestServer(Type type,
61 const std::string& host,
62 const base::FilePath& document_root)
63 : BaseTestServer(type, host) {
64 if (!Init(document_root))
68 LocalTestServer::LocalTestServer(Type type,
69 const SSLOptions& ssl_options,
70 const base::FilePath& document_root)
71 : BaseTestServer(type, ssl_options) {
72 if (!Init(document_root))
76 LocalTestServer::~LocalTestServer() {
80 bool LocalTestServer::GetTestServerPath(base::FilePath* testserver_path) const {
81 base::FilePath testserver_dir;
82 if (!PathService::Get(base::DIR_SOURCE_ROOT, &testserver_dir)) {
83 LOG(ERROR) << "Failed to get DIR_SOURCE_ROOT";
86 testserver_dir = testserver_dir.Append(FILE_PATH_LITERAL("net"))
87 .Append(FILE_PATH_LITERAL("tools"))
88 .Append(FILE_PATH_LITERAL("testserver"));
89 *testserver_path = testserver_dir.Append(FILE_PATH_LITERAL("testserver.py"));
93 bool LocalTestServer::Start() {
94 return StartInBackground() && BlockUntilStarted();
97 bool LocalTestServer::StartInBackground() {
98 // Get path to Python server script.
99 base::FilePath testserver_path;
100 if (!GetTestServerPath(&testserver_path))
103 if (!SetPythonPath())
106 if (!LaunchPython(testserver_path))
112 bool LocalTestServer::BlockUntilStarted() {
113 if (!WaitToStart()) {
118 return SetupWhenServerStarted();
121 bool LocalTestServer::Stop() {
122 CleanUpWhenStoppingServer();
124 if (!process_handle_)
128 // This kills all the processes in the job object.
132 // First check if the process has already terminated.
133 bool ret = base::WaitForSingleProcess(process_handle_, base::TimeDelta());
135 ret = base::KillProcess(process_handle_, 1, true);
139 base::CloseProcessHandle(process_handle_);
140 process_handle_ = base::kNullProcessHandle;
142 VLOG(1) << "Kill failed?";
148 bool LocalTestServer::Init(const base::FilePath& document_root) {
149 if (document_root.IsAbsolute())
152 // At this point, the port that the test server will listen on is unknown.
153 // The test server will listen on an ephemeral port, and write the port
154 // number out over a pipe that this TestServer object will read from. Once
155 // that is complete, the host port pair will contain the actual port.
157 process_handle_ = base::kNullProcessHandle;
159 base::FilePath src_dir;
160 if (!PathService::Get(base::DIR_SOURCE_ROOT, &src_dir))
162 SetResourcePath(src_dir.Append(document_root),
163 src_dir.AppendASCII("net")
166 .AppendASCII("certificates"));
170 bool LocalTestServer::SetPythonPath() const {
171 base::FilePath third_party_dir;
172 if (!PathService::Get(base::DIR_SOURCE_ROOT, &third_party_dir)) {
173 LOG(ERROR) << "Failed to get DIR_SOURCE_ROOT";
176 third_party_dir = third_party_dir.AppendASCII("third_party");
178 // For simplejson. (simplejson, unlike all the other Python modules
179 // we include, doesn't have an extra 'simplejson' directory, so we
180 // need to include its parent directory, i.e. third_party_dir).
181 AppendToPythonPath(third_party_dir);
183 AppendToPythonPath(third_party_dir.AppendASCII("tlslite"));
185 third_party_dir.AppendASCII("pyftpdlib").AppendASCII("src"));
187 third_party_dir.AppendASCII("pywebsocket").AppendASCII("src"));
189 // Locate the Python code generated by the protocol buffers compiler.
190 base::FilePath pyproto_dir;
191 if (!GetPyProtoPath(&pyproto_dir)) {
192 LOG(WARNING) << "Cannot find pyproto dir for generated code. "
193 << "Testserver features that rely on it will not work";
196 AppendToPythonPath(pyproto_dir);
201 bool LocalTestServer::AddCommandLineArguments(
202 base::CommandLine* command_line) const {
203 base::DictionaryValue arguments_dict;
204 if (!GenerateArguments(&arguments_dict))
207 // Serialize the argument dictionary into CommandLine.
208 for (base::DictionaryValue::Iterator it(arguments_dict); !it.IsAtEnd();
210 const base::Value& value = it.value();
211 const std::string& key = it.key();
213 // Add arguments from a list.
214 if (value.IsType(base::Value::TYPE_LIST)) {
215 const base::ListValue* list = NULL;
216 if (!value.GetAsList(&list) || !list || list->empty())
218 for (base::ListValue::const_iterator list_it = list->begin();
219 list_it != list->end(); ++list_it) {
220 if (!AppendArgumentFromJSONValue(key, *(*list_it), command_line))
223 } else if (!AppendArgumentFromJSONValue(key, value, command_line)) {
228 // Append the appropriate server type argument.
230 case TYPE_HTTP: // The default type is HTTP, no argument required.
233 command_line->AppendArg("--https");
237 command_line->AppendArg("--websocket");
240 command_line->AppendArg("--ftp");
243 command_line->AppendArg("--tcp-echo");
246 command_line->AppendArg("--udp-echo");
248 case TYPE_BASIC_AUTH_PROXY:
249 command_line->AppendArg("--basic-auth-proxy");