17dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch// Copyright 2014 The Chromium Authors. All rights reserved. 27dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch// Use of this source code is governed by a BSD-style license that can be 37dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch// found in the LICENSE file. 47dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 57dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "extensions/browser/api/sockets_tcp_server/tcp_server_socket_event_dispatcher.h" 67dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 77dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "base/lazy_instance.h" 87dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "extensions/browser/api/socket/tcp_socket.h" 97dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "extensions/browser/event_router.h" 107dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "extensions/browser/extensions_browser_client.h" 117dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "net/base/net_errors.h" 127dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 137dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochnamespace extensions { 147dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochnamespace core_api { 157dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 167dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochusing content::BrowserThread; 177dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 187dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochstatic base::LazyInstance< 197dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch BrowserContextKeyedAPIFactory<TCPServerSocketEventDispatcher> > g_factory = 207dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch LAZY_INSTANCE_INITIALIZER; 217dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 227dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch// static 237dbb3d5cf0c15f500944d211057644d6a2f37371Ben MurdochBrowserContextKeyedAPIFactory<TCPServerSocketEventDispatcher>* 247dbb3d5cf0c15f500944d211057644d6a2f37371Ben MurdochTCPServerSocketEventDispatcher::GetFactoryInstance() { 257dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch return g_factory.Pointer(); 267dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 277dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 287dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch// static 297dbb3d5cf0c15f500944d211057644d6a2f37371Ben MurdochTCPServerSocketEventDispatcher* TCPServerSocketEventDispatcher::Get( 307dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch content::BrowserContext* context) { 317dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch DCHECK_CURRENTLY_ON(BrowserThread::UI); 327dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 337dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch return BrowserContextKeyedAPIFactory<TCPServerSocketEventDispatcher>::Get( 347dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch context); 357dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 367dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 377dbb3d5cf0c15f500944d211057644d6a2f37371Ben MurdochTCPServerSocketEventDispatcher::TCPServerSocketEventDispatcher( 387dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch content::BrowserContext* context) 397dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch : thread_id_(Socket::kThreadId), browser_context_(context) { 403240926e260ce088908e02ac07a6cf7b0c0cbf44Ben Murdoch ApiResourceManager<ResumableTCPServerSocket>* server_manager = 413240926e260ce088908e02ac07a6cf7b0c0cbf44Ben Murdoch ApiResourceManager<ResumableTCPServerSocket>::Get(browser_context_); 423240926e260ce088908e02ac07a6cf7b0c0cbf44Ben Murdoch DCHECK(server_manager) 433240926e260ce088908e02ac07a6cf7b0c0cbf44Ben Murdoch << "There is no server socket manager. " 447dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch "If this assertion is failing during a test, then it is likely that " 457dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch "TestExtensionSystem is failing to provide an instance of " 467dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch "ApiResourceManager<ResumableTCPServerSocket>."; 477dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch server_sockets_ = server_manager->data_; 487dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 497dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch ApiResourceManager<ResumableTCPSocket>* client_manager = 507dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch ApiResourceManager<ResumableTCPSocket>::Get(browser_context_); 517dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch DCHECK(client_manager) 527dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch << "There is no client socket manager. " 537dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch "If this assertion is failing during a test, then it is likely that " 547dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch "TestExtensionSystem is failing to provide an instance of " 557dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch "ApiResourceManager<ResumableTCPSocket>."; 567dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch client_sockets_ = client_manager->data_; 577dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 587dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 597dbb3d5cf0c15f500944d211057644d6a2f37371Ben MurdochTCPServerSocketEventDispatcher::~TCPServerSocketEventDispatcher() {} 607dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 617dbb3d5cf0c15f500944d211057644d6a2f37371Ben MurdochTCPServerSocketEventDispatcher::AcceptParams::AcceptParams() {} 627dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 637dbb3d5cf0c15f500944d211057644d6a2f37371Ben MurdochTCPServerSocketEventDispatcher::AcceptParams::~AcceptParams() {} 647dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 657dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochvoid TCPServerSocketEventDispatcher::OnServerSocketListen( 667dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch const std::string& extension_id, 677dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch int socket_id) { 687dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch DCHECK_CURRENTLY_ON(thread_id_); 69 70 StartSocketAccept(extension_id, socket_id); 71} 72 73void TCPServerSocketEventDispatcher::OnServerSocketResume( 74 const std::string& extension_id, 75 int socket_id) { 76 DCHECK_CURRENTLY_ON(thread_id_); 77 78 StartSocketAccept(extension_id, socket_id); 79} 80 81void TCPServerSocketEventDispatcher::StartSocketAccept( 82 const std::string& extension_id, 83 int socket_id) { 84 DCHECK_CURRENTLY_ON(thread_id_); 85 86 AcceptParams params; 87 params.thread_id = thread_id_; 88 params.browser_context_id = browser_context_; 89 params.extension_id = extension_id; 90 params.server_sockets = server_sockets_; 91 params.client_sockets = client_sockets_; 92 params.socket_id = socket_id; 93 94 StartAccept(params); 95} 96 97// static 98void TCPServerSocketEventDispatcher::StartAccept(const AcceptParams& params) { 99 DCHECK_CURRENTLY_ON(params.thread_id); 100 101 ResumableTCPServerSocket* socket = 102 params.server_sockets->Get(params.extension_id, params.socket_id); 103 if (!socket) { 104 // This can happen if the socket is closed while our callback is active. 105 return; 106 } 107 DCHECK(params.extension_id == socket->owner_extension_id()) 108 << "Socket has wrong owner."; 109 110 // Don't start another accept if the socket has been paused. 111 if (socket->paused()) 112 return; 113 114 socket->Accept( 115 base::Bind(&TCPServerSocketEventDispatcher::AcceptCallback, params)); 116} 117 118// static 119void TCPServerSocketEventDispatcher::AcceptCallback( 120 const AcceptParams& params, 121 int result_code, 122 net::TCPClientSocket* socket) { 123 DCHECK_CURRENTLY_ON(params.thread_id); 124 125 if (result_code >= 0) { 126 ResumableTCPSocket* client_socket = 127 new ResumableTCPSocket(socket, params.extension_id, true); 128 client_socket->set_paused(true); 129 int client_socket_id = params.client_sockets->Add(client_socket); 130 131 // Dispatch "onAccept" event. 132 sockets_tcp_server::AcceptInfo accept_info; 133 accept_info.socket_id = params.socket_id; 134 accept_info.client_socket_id = client_socket_id; 135 scoped_ptr<base::ListValue> args = 136 sockets_tcp_server::OnAccept::Create(accept_info); 137 scoped_ptr<Event> event( 138 new Event(sockets_tcp_server::OnAccept::kEventName, args.Pass())); 139 PostEvent(params, event.Pass()); 140 141 // Post a task to delay the "accept" until the socket is available, as 142 // calling StartAccept at this point would error with ERR_IO_PENDING. 143 BrowserThread::PostTask( 144 params.thread_id, 145 FROM_HERE, 146 base::Bind(&TCPServerSocketEventDispatcher::StartAccept, params)); 147 } else { 148 // Dispatch "onAcceptError" event but don't start another accept to avoid 149 // potential infinite "accepts" if we have a persistent network error. 150 sockets_tcp_server::AcceptErrorInfo accept_error_info; 151 accept_error_info.socket_id = params.socket_id; 152 accept_error_info.result_code = result_code; 153 scoped_ptr<base::ListValue> args = 154 sockets_tcp_server::OnAcceptError::Create(accept_error_info); 155 scoped_ptr<Event> event( 156 new Event(sockets_tcp_server::OnAcceptError::kEventName, args.Pass())); 157 PostEvent(params, event.Pass()); 158 159 // Since we got an error, the socket is now "paused" until the application 160 // "resumes" it. 161 ResumableTCPServerSocket* socket = 162 params.server_sockets->Get(params.extension_id, params.socket_id); 163 if (socket) { 164 socket->set_paused(true); 165 } 166 } 167} 168 169// static 170void TCPServerSocketEventDispatcher::PostEvent(const AcceptParams& params, 171 scoped_ptr<Event> event) { 172 DCHECK_CURRENTLY_ON(params.thread_id); 173 174 BrowserThread::PostTask(BrowserThread::UI, 175 FROM_HERE, 176 base::Bind(&DispatchEvent, 177 params.browser_context_id, 178 params.extension_id, 179 base::Passed(event.Pass()))); 180} 181 182// static 183void TCPServerSocketEventDispatcher::DispatchEvent( 184 void* browser_context_id, 185 const std::string& extension_id, 186 scoped_ptr<Event> event) { 187 DCHECK_CURRENTLY_ON(BrowserThread::UI); 188 189 content::BrowserContext* context = 190 reinterpret_cast<content::BrowserContext*>(browser_context_id); 191 if (!extensions::ExtensionsBrowserClient::Get()->IsValidContext(context)) 192 return; 193 EventRouter* router = EventRouter::Get(context); 194 if (router) 195 router->DispatchEventToExtension(extension_id, event.Pass()); 196} 197 198} // namespace core_api 199} // namespace extensions 200