| // 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. |
| |
| #ifndef COMPONENTS_UI_DEVTOOLS_DEVTOOLS_SERVER_H_ |
| #define COMPONENTS_UI_DEVTOOLS_DEVTOOLS_SERVER_H_ |
| |
| #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 "components/ui_devtools/string_util.h" |
| #include "services/network/public/cpp/server/http_server.h" |
| |
| namespace ui_devtools { |
| |
| class UI_DEVTOOLS_EXPORT UiDevToolsServer |
| : public network::server::HttpServer::Delegate { |
| public: |
| // 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 String& message); |
| |
| int port() const { return port_; } |
| |
| private: |
| 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_; |
| |
| SEQUENCE_CHECKER(devtools_server_sequence_); |
| base::WeakPtrFactory<UiDevToolsServer> weak_ptr_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(UiDevToolsServer); |
| }; |
| |
| } // namespace ui_devtools |
| |
| #endif // COMPONENTS_UI_DEVTOOLS_DEVTOOLS_SERVER_H_ |