Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "sky/services/inspector/inspector_frontend_impl.h" | 5 #include "sky/services/inspector/inspector_frontend_impl.h" |
| 6 | 6 |
| 7 #include "base/lazy_instance.h" | |
| 8 #include "net/server/http_server.h" | |
| 9 #include "net/socket/tcp_server_socket.h" | |
| 10 | |
| 11 namespace sky { | 7 namespace sky { |
| 12 namespace inspector { | 8 namespace inspector { |
| 13 namespace { | |
| 14 const int kNotConnected = -1; | |
| 15 static base::LazyInstance<std::map<int, InspectorFrontendImpl*>> g_servers = | |
| 16 LAZY_INSTANCE_INITIALIZER; | |
| 17 } | |
| 18 | 9 |
| 19 InspectorFrontendImpl::InspectorFrontendImpl() | 10 InspectorFrontendImpl::InspectorFrontendImpl(Delegate* delegate) |
| 20 : connection_id_(kNotConnected) { | 11 : delegate_(delegate) { |
| 12 delegate_->Register(this); | |
| 21 } | 13 } |
| 22 | 14 |
| 23 InspectorFrontendImpl::~InspectorFrontendImpl() { | 15 InspectorFrontendImpl::~InspectorFrontendImpl() { |
| 24 StopListening(); | 16 delegate_->Unregister(this); |
| 25 } | 17 } |
| 26 | 18 |
| 27 void InspectorFrontendImpl::OnConnect(int connection_id) { | 19 void InspectorFrontendImpl::OnShutdown() { |
| 20 client()->OnDisconnect(); | |
|
Aaron Boodman
2014/11/11 16:12:50
The client will find out when this end of the pipe
eseidel
2014/11/11 18:15:26
I believe this represents that the Web Inspector h
| |
| 21 delete this; | |
| 28 } | 22 } |
| 29 | 23 |
| 30 void InspectorFrontendImpl::OnHttpRequest( | 24 void InspectorFrontendImpl::OnConnectionError() { |
| 31 int connection_id, const net::HttpServerRequestInfo& info) { | 25 delete this; |
| 32 web_server_->Send500(connection_id, "websockets protocol only"); | |
| 33 } | |
| 34 | |
| 35 void InspectorFrontendImpl::OnWebSocketRequest( | |
| 36 int connection_id, const net::HttpServerRequestInfo& info) { | |
| 37 if (connection_id_ != kNotConnected) { | |
| 38 web_server_->Close(connection_id); | |
| 39 return; | |
| 40 } | |
| 41 web_server_->AcceptWebSocket(connection_id, info); | |
| 42 connection_id_ = connection_id; | |
| 43 client()->OnConnect(); | |
| 44 } | |
| 45 | |
| 46 void InspectorFrontendImpl::OnWebSocketMessage( | |
| 47 int connection_id, const std::string& data) { | |
| 48 DCHECK_EQ(connection_id, connection_id_); | |
| 49 client()->OnMessage(data); | |
| 50 } | |
| 51 | |
| 52 void InspectorFrontendImpl::OnClose(int connection_id) { | |
| 53 if (connection_id != connection_id_) | |
| 54 return; | |
| 55 connection_id_ = kNotConnected; | |
| 56 client()->OnDisconnect(); | |
| 57 } | |
| 58 | |
| 59 void InspectorFrontendImpl::Listen(int32_t port) { | |
| 60 Register(port); | |
| 61 scoped_ptr<net::ServerSocket> server_socket( | |
| 62 new net::TCPServerSocket(NULL, net::NetLog::Source())); | |
| 63 server_socket->ListenWithAddressAndPort("0.0.0.0", port, 1); | |
| 64 web_server_.reset(new net::HttpServer(server_socket.Pass(), this)); | |
| 65 } | |
| 66 | |
| 67 void InspectorFrontendImpl::StopListening() { | |
| 68 if (!web_server_) | |
| 69 return; | |
| 70 web_server_.reset(); | |
| 71 Unregister(); | |
| 72 } | |
| 73 | |
| 74 void InspectorFrontendImpl::Register(int port) { | |
| 75 auto& servers = g_servers.Get(); | |
| 76 auto iter = servers.find(port); | |
| 77 if (iter != servers.end()) | |
| 78 iter->second->StopListening(); | |
| 79 DCHECK(servers.find(port) == servers.end()); | |
| 80 servers[port] = this; | |
| 81 port_ = port; | |
| 82 } | |
| 83 | |
| 84 void InspectorFrontendImpl::Unregister() { | |
| 85 DCHECK(g_servers.Get().find(port_)->second == this); | |
| 86 g_servers.Get().erase(port_); | |
| 87 port_ = kNotConnected; | |
| 88 } | 26 } |
| 89 | 27 |
| 90 void InspectorFrontendImpl::SendMessage(const mojo::String& message) { | 28 void InspectorFrontendImpl::SendMessage(const mojo::String& message) { |
| 91 if (connection_id_ == kNotConnected) | 29 delegate_->SendMessage(message); |
| 92 return; | |
| 93 web_server_->SendOverWebSocket(connection_id_, message); | |
| 94 } | 30 } |
| 95 | 31 |
| 96 } // namespace inspector | 32 } // namespace inspector |
| 97 } // namespace sky | 33 } // namespace sky |
| OLD | NEW |