| OLD | NEW |
|---|
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "net/test/spawned_test_server/spawner_communicator.h" | |
| 6 | |
| 7 #include "base/json/json_reader.h" | |
| 8 #include "base/logging.h" | |
| 9 #include "base/strings/stringprintf.h" | |
| 10 #include "base/supports_user_data.h" | |
| 11 #include "base/test/test_timeouts.h" | |
| 12 #include "base/time/time.h" | |
| 13 #include "base/values.h" | |
| 14 #include "build/build_config.h" | |
| 15 #include "net/base/elements_upload_data_stream.h" | |
| 16 #include "net/base/net_util.h" | |
| 17 #include "net/base/request_priority.h" | |
| 18 #include "net/base/upload_bytes_element_reader.h" | |
| 19 #include "net/http/http_response_headers.h" | |
| 20 #include "net/url_request/url_request_test_util.h" | |
| 21 #include "url/gurl.h" | |
| 22 | |
| 23 namespace net { | |
| 24 | |
| 25 namespace { | |
| 26 | |
| 27 GURL GenerateSpawnerCommandURL(const std::string& command, uint16 port) { | |
| 28 // Always performs HTTP request for sending command to the spawner server. | |
| 29 return GURL(base::StringPrintf("%s:%u/%s", "http://127.0.0.1", port, | |
| 30 command.c_str())); | |
| 31 } | |
| 32 | |
| 33 int kBufferSize = 2048; | |
| 34 | |
| 35 // A class to hold all data needed to send a command to spawner server. | |
| 36 class SpawnerRequestData : public base::SupportsUserData::Data { | |
| 37 public: | |
| 38 SpawnerRequestData(int id, int* result_code, std::string* data_received) | |
| 39 : request_id_(id), | |
| 40 buf_(new IOBuffer(kBufferSize)), | |
| 41 result_code_(result_code), | |
| 42 data_received_(data_received), | |
| 43 response_started_count_(0) { | |
| 44 DCHECK(result_code); | |
| 45 *result_code_ = OK; | |
| 46 DCHECK(data_received); | |
| 47 data_received_->clear(); | |
| 48 } | |
| 49 | |
| 50 ~SpawnerRequestData() override {} | |
| 51 | |
| 52 bool DoesRequestIdMatch(int request_id) const { | |
| 53 return request_id_ == request_id; | |
| 54 } | |
| 55 | |
| 56 IOBuffer* buf() const { return buf_.get(); } | |
| 57 | |
| 58 bool IsResultOK() const { return *result_code_ == OK; } | |
| 59 | |
| 60 void ClearReceivedData() { data_received_->clear(); } | |
| 61 | |
| 62 void SetResultCode(int result_code) { *result_code_ = result_code; } | |
| 63 | |
| 64 void IncreaseResponseStartedCount() { response_started_count_++; } | |
| 65 | |
| 66 int response_started_count() const { return response_started_count_; } | |
| 67 | |
| 68 // Write data read from URLRequest::Read() to |data_received_|. Returns true | |
| 69 // if |num_bytes| is great than 0. |num_bytes| is 0 for EOF, < 0 on errors. | |
| 70 bool ConsumeBytesRead(int num_bytes) { | |
| 71 // Error while reading, or EOF. | |
| 72 if (num_bytes <= 0) | |
| 73 return false; | |
| 74 | |
| 75 data_received_->append(buf_->data(), num_bytes); | |
| 76 return true; | |
| 77 } | |
| 78 | |
| 79 private: | |
| 80 // Unique ID for the current request. | |
| 81 int request_id_; | |
| 82 | |
| 83 // Buffer that URLRequest writes into. | |
| 84 scoped_refptr<IOBuffer> buf_; | |
| 85 | |
| 86 // Holds the error condition that was hit on the current request, or OK. | |
| 87 int* result_code_; | |
| 88 | |
| 89 // Data received from server; | |
| 90 std::string* data_received_; | |
| 91 | |
| 92 // Used to track how many times the OnResponseStarted get called after | |
| 93 // sending a command to spawner server. | |
| 94 int response_started_count_; | |
| 95 | |
| 96 DISALLOW_COPY_AND_ASSIGN(SpawnerRequestData); | |
| 97 }; | |
| 98 | |
| 99 } // namespace | |
| 100 | |
| 101 SpawnerCommunicator::SpawnerCommunicator(uint16 port) | |
| 102 : io_thread_("spawner_communicator"), | |
| 103 event_(false, false), | |
| 104 port_(port), | |
| 105 next_id_(0), | |
| 106 is_running_(false), | |
| 107 weak_factory_(this) {} | |
| 108 | |
| 109 SpawnerCommunicator::~SpawnerCommunicator() { | |
| 110 DCHECK(!is_running_); | |
| 111 } | |
| 112 | |
| 113 void SpawnerCommunicator::WaitForResponse() { | |
| 114 DCHECK_NE(base::MessageLoop::current(), io_thread_.message_loop()); | |
| 115 event_.Wait(); | |
| 116 event_.Reset(); | |
| 117 } | |
| 118 | |
| 119 void SpawnerCommunicator::StartIOThread() { | |
| 120 DCHECK_NE(base::MessageLoop::current(), io_thread_.message_loop()); | |
| 121 if (is_running_) | |
| 122 return; | |
| 123 | |
| 124 allowed_port_.reset(new ScopedPortException(port_)); | |
| 125 base::Thread::Options options; | |
| 126 options.message_loop_type = base::MessageLoop::TYPE_IO; | |
| 127 is_running_ = io_thread_.StartWithOptions(options); | |
| 128 DCHECK(is_running_); | |
| 129 } | |
| 130 | |
| 131 void SpawnerCommunicator::Shutdown() { | |
| 132 DCHECK_NE(base::MessageLoop::current(), io_thread_.message_loop()); | |
| 133 DCHECK(is_running_); | |
| 134 // The request and its context should be created and destroyed only on the | |
| 135 // IO thread. | |
| 136 DCHECK(!cur_request_.get()); | |
| 137 DCHECK(!context_.get()); | |
| 138 is_running_ = false; | |
| 139 io_thread_.Stop(); | |
| 140 allowed_port_.reset(); | |
| 141 } | |
| 142 | |
| 143 void SpawnerCommunicator::SendCommandAndWaitForResult( | |
| 144 const std::string& command, | |
| 145 const std::string& post_data, | |
| 146 int* result_code, | |
| 147 std::string* data_received) { | |
| 148 if (!result_code || !data_received) | |
| 149 return; | |
| 150 // Start the communicator thread to talk to test server spawner. | |
| 151 StartIOThread(); | |
| 152 DCHECK(io_thread_.message_loop()); | |
| 153 | |
| 154 // Since the method will be blocked until SpawnerCommunicator gets result | |
| 155 // from the spawner server or timed-out. It's safe to use base::Unretained | |
| 156 // when using base::Bind. | |
| 157 io_thread_.message_loop()->PostTask(FROM_HERE, base::Bind( | |
| 158 &SpawnerCommunicator::SendCommandAndWaitForResultOnIOThread, | |
| 159 base::Unretained(this), command, post_data, result_code, data_received)); | |
| 160 WaitForResponse(); | |
| 161 } | |
| 162 | |
| 163 void SpawnerCommunicator::SendCommandAndWaitForResultOnIOThread( | |
| 164 const std::string& command, | |
| 165 const std::string& post_data, | |
| 166 int* result_code, | |
| 167 std::string* data_received) { | |
| 168 base::MessageLoop* loop = io_thread_.message_loop(); | |
| 169 DCHECK(loop); | |
| 170 DCHECK_EQ(base::MessageLoop::current(), loop); | |
| 171 | |
| 172 // Prepare the URLRequest for sending the command. | |
| 173 DCHECK(!cur_request_.get()); | |
| 174 context_.reset(new TestURLRequestContext); | |
| 175 cur_request_ = context_->CreateRequest( | |
| 176 GenerateSpawnerCommandURL(command, port_), DEFAULT_PRIORITY, this, NULL); | |
| 177 DCHECK(cur_request_); | |
| 178 int current_request_id = ++next_id_; | |
| 179 SpawnerRequestData* data = new SpawnerRequestData(current_request_id, | |
| 180 result_code, | |
| 181 data_received); | |
| 182 DCHECK(data); | |
| 183 cur_request_->SetUserData(this, data); | |
| 184 | |
| 185 if (post_data.empty()) { | |
| 186 cur_request_->set_method("GET"); | |
| 187 } else { | |
| 188 cur_request_->set_method("POST"); | |
| 189 scoped_ptr<UploadElementReader> reader( | |
| 190 UploadOwnedBytesElementReader::CreateWithString(post_data)); | |
| 191 cur_request_->set_upload( | |
| 192 ElementsUploadDataStream::CreateWithReader(reader.Pass(), 0)); | |
| 193 HttpRequestHeaders headers; | |
| 194 headers.SetHeader(HttpRequestHeaders::kContentType, | |
| 195 "application/json"); | |
| 196 cur_request_->SetExtraRequestHeaders(headers); | |
| 197 } | |
| 198 | |
| 199 // Post a task to timeout this request if it takes too long. | |
| 200 base::MessageLoop::current()->PostDelayedTask( | |
| 201 FROM_HERE, | |
| 202 base::Bind(&SpawnerCommunicator::OnTimeout, | |
| 203 weak_factory_.GetWeakPtr(), | |
| 204 current_request_id), | |
| 205 TestTimeouts::action_max_timeout()); | |
| 206 | |
| 207 // Start the request. | |
| 208 cur_request_->Start(); | |
| 209 } | |
| 210 | |
| 211 void SpawnerCommunicator::OnTimeout(int id) { | |
| 212 // Timeout tasks may outlive the URLRequest they reference. Make sure it | |
| 213 // is still applicable. | |
| 214 if (!cur_request_.get()) | |
| 215 return; | |
| 216 SpawnerRequestData* data = | |
| 217 static_cast<SpawnerRequestData*>(cur_request_->GetUserData(this)); | |
| 218 DCHECK(data); | |
| 219 | |
| 220 if (!data->DoesRequestIdMatch(id)) | |
| 221 return; | |
| 222 // Set the result code and cancel the timed-out task. | |
| 223 data->SetResultCode(ERR_TIMED_OUT); | |
| 224 cur_request_->Cancel(); | |
| 225 OnSpawnerCommandCompleted(cur_request_.get()); | |
| 226 } | |
| 227 | |
| 228 void SpawnerCommunicator::OnSpawnerCommandCompleted(URLRequest* request) { | |
| 229 if (!cur_request_.get()) | |
| 230 return; | |
| 231 DCHECK_EQ(request, cur_request_.get()); | |
| 232 SpawnerRequestData* data = | |
| 233 static_cast<SpawnerRequestData*>(cur_request_->GetUserData(this)); | |
| 234 DCHECK(data); | |
| 235 | |
| 236 // If request is faild,return the error code. | |
| 237 if (!cur_request_->status().is_success()) | |
| 238 data->SetResultCode(cur_request_->status().error()); | |
| 239 | |
| 240 if (!data->IsResultOK()) { | |
| 241 LOG(ERROR) << "request failed, status: " | |
| 242 << static_cast<int>(request->status().status()) | |
| 243 << ", error: " << request->status().error(); | |
| 244 // Clear the buffer of received data if any net error happened. | |
| 245 data->ClearReceivedData(); | |
| 246 } else { | |
| 247 DCHECK_EQ(1, data->response_started_count()); | |
| 248 } | |
| 249 | |
| 250 // Clear current request to indicate the completion of sending a command | |
| 251 // to spawner server and getting the result. | |
| 252 cur_request_.reset(); | |
| 253 context_.reset(); | |
| 254 // Invalidate the weak pointers on the IO thread. | |
| 255 weak_factory_.InvalidateWeakPtrs(); | |
| 256 | |
| 257 // Wakeup the caller in user thread. | |
| 258 event_.Signal(); | |
| 259 } | |
| 260 | |
| 261 void SpawnerCommunicator::ReadResult(URLRequest* request) { | |
| 262 DCHECK_EQ(request, cur_request_.get()); | |
| 263 SpawnerRequestData* data = | |
| 264 static_cast<SpawnerRequestData*>(cur_request_->GetUserData(this)); | |
| 265 DCHECK(data); | |
| 266 | |
| 267 IOBuffer* buf = data->buf(); | |
| 268 // Read as many bytes as are available synchronously. | |
| 269 while (true) { | |
| 270 int num_bytes; | |
| 271 if (!request->Read(buf, kBufferSize, &num_bytes)) { | |
| 272 // Check whether the read failed synchronously. | |
| 273 if (!request->status().is_io_pending()) | |
| 274 OnSpawnerCommandCompleted(request); | |
| 275 return; | |
| 276 } | |
| 277 if (!data->ConsumeBytesRead(num_bytes)) { | |
| 278 OnSpawnerCommandCompleted(request); | |
| 279 return; | |
| 280 } | |
| 281 } | |
| 282 } | |
| 283 | |
| 284 void SpawnerCommunicator::OnResponseStarted(URLRequest* request) { | |
| 285 DCHECK_EQ(request, cur_request_.get()); | |
| 286 SpawnerRequestData* data = | |
| 287 static_cast<SpawnerRequestData*>(cur_request_->GetUserData(this)); | |
| 288 DCHECK(data); | |
| 289 | |
| 290 data->IncreaseResponseStartedCount(); | |
| 291 | |
| 292 if (!request->status().is_success()) { | |
| 293 OnSpawnerCommandCompleted(request); | |
| 294 return; | |
| 295 } | |
| 296 | |
| 297 // Require HTTP responses to have a success status code. | |
| 298 if (request->GetResponseCode() != 200) { | |
| 299 LOG(ERROR) << "Spawner server returned bad status: " | |
| 300 << request->response_headers()->GetStatusLine(); | |
| 301 data->SetResultCode(ERR_FAILED); | |
| 302 request->Cancel(); | |
| 303 OnSpawnerCommandCompleted(request); | |
| 304 return; | |
| 305 } | |
| 306 | |
| 307 ReadResult(request); | |
| 308 } | |
| 309 | |
| 310 void SpawnerCommunicator::OnReadCompleted(URLRequest* request, int num_bytes) { | |
| 311 if (!cur_request_.get()) | |
| 312 return; | |
| 313 DCHECK_EQ(request, cur_request_.get()); | |
| 314 SpawnerRequestData* data = | |
| 315 static_cast<SpawnerRequestData*>(cur_request_->GetUserData(this)); | |
| 316 DCHECK(data); | |
| 317 | |
| 318 if (data->ConsumeBytesRead(num_bytes)) { | |
| 319 // Keep reading. | |
| 320 ReadResult(request); | |
| 321 } else { | |
| 322 OnSpawnerCommandCompleted(request); | |
| 323 } | |
| 324 } | |
| 325 | |
| 326 bool SpawnerCommunicator::StartServer(const std::string& arguments, | |
| 327 uint16* port) { | |
| 328 *port = 0; | |
| 329 // Send the start command to spawner server to start the Python test server | |
| 330 // on remote machine. | |
| 331 std::string server_return_data; | |
| 332 int result_code; | |
| 333 SendCommandAndWaitForResult("start", arguments, &result_code, | |
| 334 &server_return_data); | |
| 335 if (OK != result_code || server_return_data.empty()) | |
| 336 return false; | |
| 337 | |
| 338 // Check whether the data returned from spawner server is JSON-formatted. | |
| 339 scoped_ptr<base::Value> value(base::JSONReader::Read(server_return_data)); | |
| 340 if (!value.get() || !value->IsType(base::Value::TYPE_DICTIONARY)) { | |
| 341 LOG(ERROR) << "Invalid server data: " << server_return_data.c_str(); | |
| 342 return false; | |
| 343 } | |
| 344 | |
| 345 // Check whether spawner server returns valid data. | |
| 346 base::DictionaryValue* server_data = | |
| 347 static_cast<base::DictionaryValue*>(value.get()); | |
| 348 std::string message; | |
| 349 if (!server_data->GetString("message", &message) || message != "started") { | |
| 350 LOG(ERROR) << "Invalid message in server data: "; | |
| 351 return false; | |
| 352 } | |
| 353 int int_port; | |
| 354 if (!server_data->GetInteger("port", &int_port) || int_port <= 0 || | |
| 355 int_port > kuint16max) { | |
| 356 LOG(ERROR) << "Invalid port value: " << int_port; | |
| 357 return false; | |
| 358 } | |
| 359 *port = static_cast<uint16>(int_port); | |
| 360 return true; | |
| 361 } | |
| 362 | |
| 363 bool SpawnerCommunicator::StopServer() { | |
| 364 // It's OK to stop the SpawnerCommunicator without starting it. Some tests | |
| 365 // have test server on their test fixture but do not actually use it. | |
| 366 if (!is_running_) | |
| 367 return true; | |
| 368 | |
| 369 // When the test is done, ask the test server spawner to kill the test server | |
| 370 // on the remote machine. | |
| 371 std::string server_return_data; | |
| 372 int result_code; | |
| 373 SendCommandAndWaitForResult("kill", "", &result_code, &server_return_data); | |
| 374 Shutdown(); | |
| 375 if (OK != result_code || server_return_data != "killed") | |
| 376 return false; | |
| 377 return true; | |
| 378 } | |
| 379 | |
| 380 } // namespace net | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 992733002:
Remove //net (except for Android test stuff) and sdch (Closed)
Base URL: git@github.com:domokit/mojo.git@master