| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/ui_devtools/devtools_server.h" |
| |
| #include "base/bind.h" |
| #include "base/bind_helpers.h" |
| #include "base/command_line.h" |
| #include "base/format_macros.h" |
| #include "base/memory/ptr_util.h" |
| #include "base/message_loop/message_loop.h" |
| #include "base/strings/string_number_conversions.h" |
| #include "base/strings/stringprintf.h" |
| #include "base/threading/thread_task_runner_handle.h" |
| #include "base/values.h" |
| #include "components/ui_devtools/switches.h" |
| #include "net/base/net_errors.h" |
| #include "net/log/net_log.h" |
| #include "net/server/http_server_request_info.h" |
| #include "net/socket/server_socket.h" |
| #include "net/socket/tcp_server_socket.h" |
| |
| namespace ui { |
| namespace devtools { |
| |
| namespace { |
| const char kChromeDeveloperToolsPrefix[] = |
| "chrome-devtools://devtools/bundled/inspector.html?ws="; |
| |
| bool IsUiDevToolsEnabled() { |
| return base::CommandLine::ForCurrentProcess()->HasSwitch(kEnableUiDevTools); |
| } |
| |
| int GetUiDevToolsPort() { |
| DCHECK(IsUiDevToolsEnabled()); |
| constexpr int kDefaultPort = 9223; |
| int port; |
| if (!base::StringToInt( |
| base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII( |
| kEnableUiDevTools), |
| &port)) |
| port = kDefaultPort; |
| return port; |
| } |
| |
| } // namespace |
| |
| UiDevToolsServer* UiDevToolsServer::devtools_server_ = nullptr; |
| |
| UiDevToolsServer::UiDevToolsServer( |
| scoped_refptr<base::SingleThreadTaskRunner> io_thread_task_runner) |
| : io_thread_task_runner_(io_thread_task_runner) { |
| DCHECK(!devtools_server_); |
| main_thread_task_runner_ = base::ThreadTaskRunnerHandle::Get(); |
| devtools_server_ = this; |
| if (io_thread_task_runner_) |
| return; |
| // If io_thread_task_runner not passed in, create an I/O thread |
| thread_.reset(new base::Thread("UiDevToolsServerThread")); |
| base::Thread::Options options; |
| options.message_loop_type = base::MessageLoop::TYPE_IO; |
| CHECK(thread_->StartWithOptions(options)); |
| io_thread_task_runner_ = thread_->task_runner(); |
| } |
| |
| UiDevToolsServer::~UiDevToolsServer() { |
| devtools_server_ = nullptr; |
| } |
| |
| // static |
| std::unique_ptr<UiDevToolsServer> UiDevToolsServer::Create( |
| scoped_refptr<base::SingleThreadTaskRunner> io_thread_task_runner) { |
| std::unique_ptr<UiDevToolsServer> server; |
| if (IsUiDevToolsEnabled() && !devtools_server_) { |
| // TODO(mhashmi): Change port if more than one inspectable clients |
| server.reset(new UiDevToolsServer(io_thread_task_runner)); |
| server->Start("0.0.0.0", GetUiDevToolsPort()); |
| } |
| return server; |
| } |
| |
| // static |
| std::vector<UiDevToolsServer::NameUrlPair> |
| UiDevToolsServer::GetClientNamesAndUrls() { |
| std::vector<NameUrlPair> pairs; |
| if (!devtools_server_) |
| return pairs; |
| |
| for (ClientsList::size_type i = 0; i != devtools_server_->clients_.size(); |
| i++) { |
| pairs.push_back(std::pair<std::string, std::string>( |
| devtools_server_->clients_[i]->name(), |
| base::StringPrintf("%s0.0.0.0:%d/%" PRIuS, kChromeDeveloperToolsPrefix, |
| GetUiDevToolsPort(), i))); |
| } |
| return pairs; |
| } |
| |
| void UiDevToolsServer::AttachClient(std::unique_ptr<UiDevToolsClient> client) { |
| clients_.push_back(std::move(client)); |
| } |
| |
| void UiDevToolsServer::SendOverWebSocket(int connection_id, |
| const String& message) { |
| io_thread_task_runner_->PostTask( |
| FROM_HERE, |
| base::Bind(&net::HttpServer::SendOverWebSocket, |
| base::Unretained(server_.get()), connection_id, message)); |
| } |
| |
| void UiDevToolsServer::Start(const std::string& address_string, uint16_t port) { |
| io_thread_task_runner_->PostTask( |
| FROM_HERE, base::Bind(&UiDevToolsServer::StartServer, |
| base::Unretained(this), address_string, port)); |
| } |
| |
| void UiDevToolsServer::StartServer(const std::string& address_string, |
| uint16_t port) { |
| DCHECK(!server_); |
| std::unique_ptr<net::ServerSocket> socket( |
| new net::TCPServerSocket(nullptr, net::NetLogSource())); |
| constexpr int kBacklog = 1; |
| if (socket->ListenWithAddressAndPort(address_string, port, kBacklog) != |
| net::OK) |
| return; |
| server_ = base::MakeUnique<net::HttpServer>(std::move(socket), this); |
| } |
| |
| // HttpServer::Delegate Implementation |
| void UiDevToolsServer::OnConnect(int connection_id) { |
| NOTIMPLEMENTED(); |
| } |
| |
| void UiDevToolsServer::OnHttpRequest(int connection_id, |
| const net::HttpServerRequestInfo& info) { |
| NOTIMPLEMENTED(); |
| } |
| |
| void UiDevToolsServer::OnWebSocketRequest( |
| int connection_id, |
| const net::HttpServerRequestInfo& info) { |
| size_t target_id = 0; |
| if (info.path.empty() || |
| !base::StringToSizeT(info.path.substr(1), &target_id) || |
| target_id > clients_.size()) |
| return; |
| |
| UiDevToolsClient* client = clients_[target_id].get(); |
| // Only one user can inspect the client at a time |
| if (client->connected()) |
| return; |
| client->set_connection_id(connection_id); |
| connections_[connection_id] = client; |
| io_thread_task_runner_->PostTask( |
| FROM_HERE, |
| base::Bind(&net::HttpServer::AcceptWebSocket, |
| base::Unretained(server_.get()), connection_id, info)); |
| } |
| |
| void UiDevToolsServer::OnWebSocketMessage(int connection_id, |
| const std::string& data) { |
| ConnectionsMap::iterator it = connections_.find(connection_id); |
| DCHECK(it != connections_.end()); |
| UiDevToolsClient* client = it->second; |
| DCHECK(client); |
| main_thread_task_runner_->PostTask( |
| FROM_HERE, |
| base::Bind(&UiDevToolsClient::Dispatch, base::Unretained(client), data)); |
| } |
| |
| void UiDevToolsServer::OnClose(int connection_id) { |
| ConnectionsMap::iterator it = connections_.find(connection_id); |
| if (it == connections_.end()) |
| return; |
| UiDevToolsClient* client = it->second; |
| DCHECK(client); |
| main_thread_task_runner_->PostTask( |
| FROM_HERE, |
| base::Bind(&UiDevToolsClient::Disconnect, base::Unretained(client))); |
| connections_.erase(it); |
| } |
| |
| } // namespace devtools |
| } // namespace ui |