blob: 12877d13b8b823b0d74bb2882124d2bf3583b261 [file] [log] [blame]
// 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 <vector>
#include "base/compiler_specific.h"
#include "base/memory/weak_ptr.h"
#include "base/single_thread_task_runner.h"
#include "base/threading/thread.h"
#include "components/ui_devtools/DOM.h"
#include "components/ui_devtools/Forward.h"
#include "components/ui_devtools/Protocol.h"
#include "components/ui_devtools/devtools_client.h"
#include "components/ui_devtools/devtools_export.h"
#include "services/network/public/cpp/server/http_server.h"
namespace ui_devtools {
class UI_DEVTOOLS_EXPORT UiDevToolsServer
: public network::server::HttpServer::Delegate {
// Network tags to be used for the UI and the Viz devtools servers.
static const net::NetworkTrafficAnnotationTag kUIDevtoolsServerTag;
static const net::NetworkTrafficAnnotationTag kVizDevtoolsServerTag;
~UiDevToolsServer() override;
// Returns an empty unique_ptr if ui devtools flag isn't enabled or if a
// server instance has already been created.
static std::unique_ptr<UiDevToolsServer> CreateForViews(
network::mojom::NetworkContext* network_context,
int port);
// Assumes that the devtools flag is enabled, and was checked when the socket
// was created.
static std::unique_ptr<UiDevToolsServer> CreateForViz(
network::mojom::TCPServerSocketPtr server_socket,
int port);
// Creates a TCPServerSocket to be used by a UiDevToolsServer.
static void CreateTCPServerSocket(
network::mojom::TCPServerSocketRequest server_socket_request,
network::mojom::NetworkContext* network_context,
int port,
net::NetworkTrafficAnnotationTag tag,
network::mojom::NetworkContext::CreateTCPServerSocketCallback callback);
// Returns a list of attached UiDevToolsClient name + URL
using NameUrlPair = std::pair<std::string, std::string>;
static std::vector<NameUrlPair> GetClientNamesAndUrls();
// Returns true if UI Devtools is enabled by the given commandline switch.
static bool IsUiDevToolsEnabled(const char* enable_devtools_flag);
// Returns the port number specified by a command line flag. If a number is
// not specified as a command line argument, returns the |default_port|.
static int GetUiDevToolsPort(const char* enable_devtools_flag,
int default_port);
void AttachClient(std::unique_ptr<UiDevToolsClient> client);
void SendOverWebSocket(int connection_id, const protocol::String& message);
int port() const { return port_; }
UiDevToolsServer(int port, const net::NetworkTrafficAnnotationTag tag);
void MakeServer(network::mojom::TCPServerSocketPtr server_socket,
int result,
const base::Optional<net::IPEndPoint>& local_addr);
// HttpServer::Delegate
void OnConnect(int connection_id) override;
void OnHttpRequest(
int connection_id,
const network::server::HttpServerRequestInfo& info) override;
void OnWebSocketRequest(
int connection_id,
const network::server::HttpServerRequestInfo& info) override;
void OnWebSocketMessage(int connection_id, const std::string& data) override;
void OnClose(int connection_id) override;
using ClientsList = std::vector<std::unique_ptr<UiDevToolsClient>>;
using ConnectionsMap = std::map<uint32_t, UiDevToolsClient*>;
ClientsList clients_;
ConnectionsMap connections_;
std::unique_ptr<network::server::HttpServer> server_;
// The port the devtools server listens on
const int port_;
const net::NetworkTrafficAnnotationTag tag_;
// The server (owned by Chrome for now)
static UiDevToolsServer* devtools_server_;
base::WeakPtrFactory<UiDevToolsServer> weak_ptr_factory_;
} // namespace ui_devtools