1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// 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/strings/string_piece_forward.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 TracingAgent;
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, base::StringPiece message);
int port() const { return port_; }
TracingAgent* tracing_agent() { return tracing_agent_; }
void set_tracing_agent(TracingAgent* agent) { tracing_agent_ = agent; }
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, 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_;
TracingAgent* tracing_agent_ = nullptr;
// The server (owned by Chrome for now)
static UiDevToolsServer* devtools_server_;
SEQUENCE_CHECKER(devtools_server_sequence_);
base::WeakPtrFactory<UiDevToolsServer> weak_ptr_factory_{this};
DISALLOW_COPY_AND_ASSIGN(UiDevToolsServer);
};
} // namespace ui_devtools
#endif // COMPONENTS_UI_DEVTOOLS_DEVTOOLS_SERVER_H_
|
