Chromium Code Reviews Chromium Code Reviews
Issue 2374513002:
Add ui devtools server (Closed)
| Index: ui/devtools/devtools_server.cc |
| diff --git a/ui/devtools/devtools_server.cc b/ui/devtools/devtools_server.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..d4d616a13ee067c1941054272b10e4f43e079483 |
| --- /dev/null |
| +++ b/ui/devtools/devtools_server.cc |
| @@ -0,0 +1,152 @@ |
| +// 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 "ui/devtools/devtools_server.h" |
| + |
| +#include "base/bind.h" |
| +#include "base/bind_helpers.h" |
| +#include "base/format_macros.h" |
| +#include "base/strings/string_number_conversions.h" |
| +#include "base/strings/stringprintf.h" |
| +#include "base/values.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="; |
| +} |
| + |
| +UiDevToolsServer::UiDevToolsServer() |
| + : thread_(new base::Thread("UiDevToolsServerThread")) {} |
| + |
| +UiDevToolsServer::~UiDevToolsServer() {} |
| + |
| +void UiDevToolsServer::Start(const std::string& address_string, uint16_t port) { |
| + if (thread_ && thread_->IsRunning()) { |
|
sadrul
2016/10/17 16:05:08
No {}
Sarmad Hashmi
2016/10/17 16:58:21
Done.
|
| + return; |
| + } |
| + // Start IO thread upon which all the methods will run |
| + base::Thread::Options options; |
| + options.message_loop_type = base::MessageLoop::TYPE_IO; |
| + if (thread_->StartWithOptions(options)) { |
| + thread_->task_runner()->PostTask( |
| + FROM_HERE, base::Bind(&UiDevToolsServer::StartServer, |
| + base::Unretained(this), address_string, port)); |
| + } |
| +} // namespace |
|
sadrul
2016/10/17 16:05:08
// namespace should move up above (in line 25)
Sarmad Hashmi
2016/10/17 16:58:21
Done.
|
| + |
| +void UiDevToolsServer::StartServer(const std::string& address_string, |
| + uint16_t port) { |
| + std::unique_ptr<net::ServerSocket> socket( |
| + new net::TCPServerSocket(nullptr, net::NetLog::Source())); |
| + if (socket->ListenWithAddressAndPort(address_string, port, 1) != net::OK) { |
| + return; |
|
sadrul
2016/10/17 16:05:08
No {}
The '1' in the param-list is a little bit c
Sarmad Hashmi
2016/10/17 16:58:21
Done.
|
| + } |
| + server_.reset(new net::HttpServer(std::move(socket), this)); |
|
sadrul
2016/10/17 16:05:08
server_ = base::MakeUnique<net::HttpServer>(std::m
Sarmad Hashmi
2016/10/17 16:58:21
Done.
|
| +} |
| + |
| +void UiDevToolsServer::SendOverWebSocket(int connection_id, |
| + const String& message) { |
| + thread_->task_runner()->PostTask( |
| + FROM_HERE, |
| + base::Bind(&net::HttpServer::SendOverWebSocket, |
| + base::Unretained(server_.get()), connection_id, message)); |
| +} |
| + |
| +void UiDevToolsServer::AttachClient(std::unique_ptr<UiDevToolsClient> client) { |
| + clients_.push_back(std::move(client)); |
| +} |
| + |
| +// HttpServer::Delegate Implementation |
| +void UiDevToolsServer::OnConnect(int connection_id) { |
| + NOTIMPLEMENTED(); |
| +} |
| + |
| +void UiDevToolsServer::OnHttpRequest(int connection_id, |
| + const net::HttpServerRequestInfo& info) { |
| + // Display a simple html page with all the clients and the corresponding |
| + // devtools links |
| + // TODO(mhashmi): Remove and display all clients under chrome://inspect/#other |
| + if (info.path.empty() || info.path == "/") { |
| + std::string clientHTML = "<html>"; |
| + clientHTML += |
| + "<h3>Copy paste the corresponding links in your browser to inspect " |
| + "them:</h3>"; |
| + for (ClientsList::size_type i = 0; i != clients_.size(); i++) { |
| + net::IPEndPoint ip; |
| + server_->GetLocalAddress(&ip); |
| + clientHTML += base::StringPrintf( |
| + "<p><strong>%s</strong> (%s%s/%" PRIuS ")</p>", |
| + clients_[i]->name_.c_str(), kChromeDeveloperToolsPrefix, |
| + ip.ToString().c_str(), i); |
| + } |
| + clientHTML += "</html>"; |
| + thread_->task_runner()->PostTask( |
| + FROM_HERE, |
| + base::Bind(&net::HttpServer::Send200, base::Unretained(server_.get()), |
| + connection_id, clientHTML, "text/html")); |
| + } |
| +} |
| + |
| +void UiDevToolsServer::OnWebSocketRequest( |
| + int connection_id, |
| + const net::HttpServerRequestInfo& info) { |
| + size_t target_id; |
| + 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->connection_id = connection_id; |
| + connections_[connection_id] = client; |
| + 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); |
| + if (it == connections_.end()) |
|
sadrul
2016/10/17 16:05:08
Should this be a DCHECK() that it != connections_.
Sarmad Hashmi
2016/10/17 16:58:21
Done. Makes sense since a connection_id that never
|
| + return; |
| + |
| + UiDevToolsClient* client = it->second; |
| + if (!client) { |
| + // Client shut down, close the connection |
| + connections_.erase(it); |
| + thread_->task_runner()->PostTask( |
| + FROM_HERE, base::Bind(&net::HttpServer::Close, |
| + base::Unretained(server_.get()), connection_id)); |
| + return; |
| + } |
| + |
| + 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()) |
|
sadrul
2016/10/17 16:05:08
Should this be a DCHECK that it != connections_.en
Sarmad Hashmi
2016/10/17 16:58:21
Done.
|
| + return; |
| + UiDevToolsClient* client = it->second; |
| + if (client) |
| + client->connection_id = UiDevToolsClient::kNotConnected; |
| + connections_.erase(it); |
| +} |
| + |
| +} // namespace devtools |
| +} // namespace ui |
