| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 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 #include <locale> | 6 #include <locale> |
| 7 #include <string> | 7 #include <string> |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/at_exit.h" | 10 #include "base/at_exit.h" |
| 11 #include "base/bind.h" | 11 #include "base/bind.h" |
| 12 #include "base/callback.h" | 12 #include "base/callback.h" |
| 13 #include "base/command_line.h" | 13 #include "base/command_line.h" |
| 14 #include "base/files/file_path.h" | 14 #include "base/files/file_path.h" |
| 15 #include "base/lazy_instance.h" | 15 #include "base/lazy_instance.h" |
| 16 #include "base/logging.h" | 16 #include "base/logging.h" |
| 17 #include "base/memory/scoped_ptr.h" | 17 #include "base/memory/scoped_ptr.h" |
| 18 #include "base/message_loop/message_loop.h" | 18 #include "base/message_loop/message_loop.h" |
| 19 #include "base/run_loop.h" | 19 #include "base/run_loop.h" |
| 20 #include "base/strings/string_number_conversions.h" | 20 #include "base/strings/string_number_conversions.h" |
| 21 #include "base/strings/string_split.h" | 21 #include "base/strings/string_split.h" |
| 22 #include "base/strings/string_util.h" | 22 #include "base/strings/string_util.h" |
| 23 #include "base/strings/stringprintf.h" | 23 #include "base/strings/stringprintf.h" |
| 24 #include "base/synchronization/waitable_event.h" | 24 #include "base/synchronization/waitable_event.h" |
| 25 #include "base/thread_task_runner_handle.h" |
| 25 #include "base/threading/thread.h" | 26 #include "base/threading/thread.h" |
| 26 #include "base/threading/thread_local.h" | 27 #include "base/threading/thread_local.h" |
| 27 #include "chrome/test/chromedriver/logging.h" | 28 #include "chrome/test/chromedriver/logging.h" |
| 28 #include "chrome/test/chromedriver/net/port_server.h" | 29 #include "chrome/test/chromedriver/net/port_server.h" |
| 29 #include "chrome/test/chromedriver/server/http_handler.h" | 30 #include "chrome/test/chromedriver/server/http_handler.h" |
| 30 #include "chrome/test/chromedriver/version.h" | 31 #include "chrome/test/chromedriver/version.h" |
| 31 #include "net/base/ip_endpoint.h" | 32 #include "net/base/ip_endpoint.h" |
| 32 #include "net/base/net_errors.h" | 33 #include "net/base/net_errors.h" |
| 33 #include "net/server/http_server.h" | 34 #include "net/server/http_server.h" |
| 34 #include "net/server/http_server_request_info.h" | 35 #include "net/server/http_server_request_info.h" |
| (...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 129 | 130 |
| 130 handler->Handle(request, send_response_func); | 131 handler->Handle(request, send_response_func); |
| 131 } | 132 } |
| 132 | 133 |
| 133 void HandleRequestOnIOThread( | 134 void HandleRequestOnIOThread( |
| 134 const scoped_refptr<base::SingleThreadTaskRunner>& cmd_task_runner, | 135 const scoped_refptr<base::SingleThreadTaskRunner>& cmd_task_runner, |
| 135 const HttpRequestHandlerFunc& handle_request_on_cmd_func, | 136 const HttpRequestHandlerFunc& handle_request_on_cmd_func, |
| 136 const net::HttpServerRequestInfo& request, | 137 const net::HttpServerRequestInfo& request, |
| 137 const HttpResponseSenderFunc& send_response_func) { | 138 const HttpResponseSenderFunc& send_response_func) { |
| 138 cmd_task_runner->PostTask( | 139 cmd_task_runner->PostTask( |
| 139 FROM_HERE, | 140 FROM_HERE, base::Bind(handle_request_on_cmd_func, request, |
| 140 base::Bind(handle_request_on_cmd_func, | 141 base::Bind(&SendResponseOnCmdThread, |
| 141 request, | 142 base::ThreadTaskRunnerHandle::Get(), |
| 142 base::Bind(&SendResponseOnCmdThread, | 143 send_response_func))); |
| 143 base::MessageLoopProxy::current(), | |
| 144 send_response_func))); | |
| 145 } | 144 } |
| 146 | 145 |
| 147 base::LazyInstance<base::ThreadLocalPointer<HttpServer> > | 146 base::LazyInstance<base::ThreadLocalPointer<HttpServer> > |
| 148 lazy_tls_server = LAZY_INSTANCE_INITIALIZER; | 147 lazy_tls_server = LAZY_INSTANCE_INITIALIZER; |
| 149 | 148 |
| 150 void StopServerOnIOThread() { | 149 void StopServerOnIOThread() { |
| 151 // Note, |server| may be NULL. | 150 // Note, |server| may be NULL. |
| 152 HttpServer* server = lazy_tls_server.Pointer()->Get(); | 151 HttpServer* server = lazy_tls_server.Pointer()->Get(); |
| 153 lazy_tls_server.Pointer()->Set(NULL); | 152 lazy_tls_server.Pointer()->Set(NULL); |
| 154 delete server; | 153 delete server; |
| (...skipping 15 matching lines...) Expand all Loading... |
| 170 const std::vector<std::string>& whitelisted_ips, | 169 const std::vector<std::string>& whitelisted_ips, |
| 171 const std::string& url_base, | 170 const std::string& url_base, |
| 172 int adb_port, | 171 int adb_port, |
| 173 scoped_ptr<PortServer> port_server) { | 172 scoped_ptr<PortServer> port_server) { |
| 174 base::Thread io_thread("ChromeDriver IO"); | 173 base::Thread io_thread("ChromeDriver IO"); |
| 175 CHECK(io_thread.StartWithOptions( | 174 CHECK(io_thread.StartWithOptions( |
| 176 base::Thread::Options(base::MessageLoop::TYPE_IO, 0))); | 175 base::Thread::Options(base::MessageLoop::TYPE_IO, 0))); |
| 177 | 176 |
| 178 base::MessageLoop cmd_loop; | 177 base::MessageLoop cmd_loop; |
| 179 base::RunLoop cmd_run_loop; | 178 base::RunLoop cmd_run_loop; |
| 180 HttpHandler handler(cmd_run_loop.QuitClosure(), | 179 HttpHandler handler(cmd_run_loop.QuitClosure(), io_thread.task_runner(), |
| 181 io_thread.message_loop_proxy(), | 180 url_base, adb_port, port_server.Pass()); |
| 182 url_base, | |
| 183 adb_port, | |
| 184 port_server.Pass()); | |
| 185 HttpRequestHandlerFunc handle_request_func = | 181 HttpRequestHandlerFunc handle_request_func = |
| 186 base::Bind(&HandleRequestOnCmdThread, &handler, whitelisted_ips); | 182 base::Bind(&HandleRequestOnCmdThread, &handler, whitelisted_ips); |
| 187 | 183 |
| 188 io_thread.message_loop() | 184 io_thread.message_loop()->PostTask( |
| 189 ->PostTask(FROM_HERE, | 185 FROM_HERE, |
| 190 base::Bind(&StartServerOnIOThread, | 186 base::Bind(&StartServerOnIOThread, port, allow_remote, |
| 191 port, | 187 base::Bind(&HandleRequestOnIOThread, cmd_loop.task_runner(), |
| 192 allow_remote, | 188 handle_request_func))); |
| 193 base::Bind(&HandleRequestOnIOThread, | |
| 194 cmd_loop.message_loop_proxy(), | |
| 195 handle_request_func))); | |
| 196 // Run the command loop. This loop is quit after the response for a shutdown | 189 // Run the command loop. This loop is quit after the response for a shutdown |
| 197 // request is posted to the IO loop. After the command loop quits, a task | 190 // request is posted to the IO loop. After the command loop quits, a task |
| 198 // is posted to the IO loop to stop the server. Lastly, the IO thread is | 191 // is posted to the IO loop to stop the server. Lastly, the IO thread is |
| 199 // destroyed, which waits until all pending tasks have been completed. | 192 // destroyed, which waits until all pending tasks have been completed. |
| 200 // This assumes the response is sent synchronously as part of the IO task. | 193 // This assumes the response is sent synchronously as part of the IO task. |
| 201 cmd_run_loop.Run(); | 194 cmd_run_loop.Run(); |
| 202 io_thread.message_loop() | 195 io_thread.message_loop() |
| 203 ->PostTask(FROM_HERE, base::Bind(&StopServerOnIOThread)); | 196 ->PostTask(FROM_HERE, base::Bind(&StopServerOnIOThread)); |
| 204 } | 197 } |
| 205 | 198 |
| (...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 310 } | 303 } |
| 311 | 304 |
| 312 if (!InitLogging()) { | 305 if (!InitLogging()) { |
| 313 printf("Unable to initialize logging. Exiting...\n"); | 306 printf("Unable to initialize logging. Exiting...\n"); |
| 314 return 1; | 307 return 1; |
| 315 } | 308 } |
| 316 RunServer(port, allow_remote, whitelisted_ips, | 309 RunServer(port, allow_remote, whitelisted_ips, |
| 317 url_base, adb_port, port_server.Pass()); | 310 url_base, adb_port, port_server.Pass()); |
| 318 return 0; | 311 return 0; |
| 319 } | 312 } |
| OLD | NEW |