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 3 // found in the LICENSE file. 4 5 #include "extensions/browser/api/sockets_tcp_server/tcp_server_socket_event_dispatcher.h" 6 7 #include "extensions/browser/api/socket/tcp_socket.h" 8 #include "extensions/browser/event_router.h" 9 #include "extensions/browser/extensions_browser_client.h" 10 #include "net/base/net_errors.h" 11 12 namespace extensions { 13 namespace core_api { 14 15 using content::BrowserThread; 16 17 static base::LazyInstance< 18 BrowserContextKeyedAPIFactory<TCPServerSocketEventDispatcher> > g_factory = 19 LAZY_INSTANCE_INITIALIZER; 20 21 // static 22 BrowserContextKeyedAPIFactory<TCPServerSocketEventDispatcher>* 23 TCPServerSocketEventDispatcher::GetFactoryInstance() { 24 return g_factory.Pointer(); 25 } 26 27 // static 28 TCPServerSocketEventDispatcher* TCPServerSocketEventDispatcher::Get( 29 content::BrowserContext* context) { 30 DCHECK_CURRENTLY_ON(BrowserThread::UI); 31 32 return BrowserContextKeyedAPIFactory<TCPServerSocketEventDispatcher>::Get( 33 context); 34 } 35 36 TCPServerSocketEventDispatcher::TCPServerSocketEventDispatcher( 37 content::BrowserContext* context) 38 : thread_id_(Socket::kThreadId), browser_context_(context) { 39 ApiResourceManager<ResumableTCPServerSocket>* server_manager = 40 ApiResourceManager<ResumableTCPServerSocket>::Get(browser_context_); 41 DCHECK(server_manager) 42 << "There is no server socket manager. " 43 "If this assertion is failing during a test, then it is likely that " 44 "TestExtensionSystem is failing to provide an instance of " 45 "ApiResourceManager<ResumableTCPServerSocket>."; 46 server_sockets_ = server_manager->data_; 47 48 ApiResourceManager<ResumableTCPSocket>* client_manager = 49 ApiResourceManager<ResumableTCPSocket>::Get(browser_context_); 50 DCHECK(client_manager) 51 << "There is no client socket manager. " 52 "If this assertion is failing during a test, then it is likely that " 53 "TestExtensionSystem is failing to provide an instance of " 54 "ApiResourceManager<ResumableTCPSocket>."; 55 client_sockets_ = client_manager->data_; 56 } 57 58 TCPServerSocketEventDispatcher::~TCPServerSocketEventDispatcher() {} 59 60 TCPServerSocketEventDispatcher::AcceptParams::AcceptParams() {} 61 62 TCPServerSocketEventDispatcher::AcceptParams::~AcceptParams() {} 63 64 void TCPServerSocketEventDispatcher::OnServerSocketListen( 65 const std::string& extension_id, 66 int socket_id) { 67 DCHECK_CURRENTLY_ON(thread_id_); 68 69 StartSocketAccept(extension_id, socket_id); 70 } 71 72 void TCPServerSocketEventDispatcher::OnServerSocketResume( 73 const std::string& extension_id, 74 int socket_id) { 75 DCHECK_CURRENTLY_ON(thread_id_); 76 77 StartSocketAccept(extension_id, socket_id); 78 } 79 80 void TCPServerSocketEventDispatcher::StartSocketAccept( 81 const std::string& extension_id, 82 int socket_id) { 83 DCHECK_CURRENTLY_ON(thread_id_); 84 85 AcceptParams params; 86 params.thread_id = thread_id_; 87 params.browser_context_id = browser_context_; 88 params.extension_id = extension_id; 89 params.server_sockets = server_sockets_; 90 params.client_sockets = client_sockets_; 91 params.socket_id = socket_id; 92 93 StartAccept(params); 94 } 95 96 // static 97 void TCPServerSocketEventDispatcher::StartAccept(const AcceptParams& params) { 98 DCHECK_CURRENTLY_ON(params.thread_id); 99 100 ResumableTCPServerSocket* socket = 101 params.server_sockets->Get(params.extension_id, params.socket_id); 102 if (!socket) { 103 // This can happen if the socket is closed while our callback is active. 104 return; 105 } 106 DCHECK(params.extension_id == socket->owner_extension_id()) 107 << "Socket has wrong owner."; 108 109 // Don't start another accept if the socket has been paused. 110 if (socket->paused()) 111 return; 112 113 socket->Accept( 114 base::Bind(&TCPServerSocketEventDispatcher::AcceptCallback, params)); 115 } 116 117 // static 118 void TCPServerSocketEventDispatcher::AcceptCallback( 119 const AcceptParams& params, 120 int result_code, 121 net::TCPClientSocket* socket) { 122 DCHECK_CURRENTLY_ON(params.thread_id); 123 124 if (result_code >= 0) { 125 ResumableTCPSocket* client_socket = 126 new ResumableTCPSocket(socket, params.extension_id, true); 127 client_socket->set_paused(true); 128 int client_socket_id = params.client_sockets->Add(client_socket); 129 130 // Dispatch "onAccept" event. 131 sockets_tcp_server::AcceptInfo accept_info; 132 accept_info.socket_id = params.socket_id; 133 accept_info.client_socket_id = client_socket_id; 134 scoped_ptr<base::ListValue> args = 135 sockets_tcp_server::OnAccept::Create(accept_info); 136 scoped_ptr<Event> event( 137 new Event(sockets_tcp_server::OnAccept::kEventName, args.Pass())); 138 PostEvent(params, event.Pass()); 139 140 // Post a task to delay the "accept" until the socket is available, as 141 // calling StartAccept at this point would error with ERR_IO_PENDING. 142 BrowserThread::PostTask( 143 params.thread_id, 144 FROM_HERE, 145 base::Bind(&TCPServerSocketEventDispatcher::StartAccept, params)); 146 } else { 147 // Dispatch "onAcceptError" event but don't start another accept to avoid 148 // potential infinite "accepts" if we have a persistent network error. 149 sockets_tcp_server::AcceptErrorInfo accept_error_info; 150 accept_error_info.socket_id = params.socket_id; 151 accept_error_info.result_code = result_code; 152 scoped_ptr<base::ListValue> args = 153 sockets_tcp_server::OnAcceptError::Create(accept_error_info); 154 scoped_ptr<Event> event( 155 new Event(sockets_tcp_server::OnAcceptError::kEventName, args.Pass())); 156 PostEvent(params, event.Pass()); 157 158 // Since we got an error, the socket is now "paused" until the application 159 // "resumes" it. 160 ResumableTCPServerSocket* socket = 161 params.server_sockets->Get(params.extension_id, params.socket_id); 162 if (socket) { 163 socket->set_paused(true); 164 } 165 } 166 } 167 168 // static 169 void TCPServerSocketEventDispatcher::PostEvent(const AcceptParams& params, 170 scoped_ptr<Event> event) { 171 DCHECK_CURRENTLY_ON(params.thread_id); 172 173 BrowserThread::PostTask(BrowserThread::UI, 174 FROM_HERE, 175 base::Bind(&DispatchEvent, 176 params.browser_context_id, 177 params.extension_id, 178 base::Passed(event.Pass()))); 179 } 180 181 // static 182 void TCPServerSocketEventDispatcher::DispatchEvent( 183 void* browser_context_id, 184 const std::string& extension_id, 185 scoped_ptr<Event> event) { 186 DCHECK_CURRENTLY_ON(BrowserThread::UI); 187 188 content::BrowserContext* context = 189 reinterpret_cast<content::BrowserContext*>(browser_context_id); 190 if (!extensions::ExtensionsBrowserClient::Get()->IsValidContext(context)) 191 return; 192 EventRouter* router = EventRouter::Get(context); 193 if (router) 194 router->DispatchEventToExtension(extension_id, event.Pass()); 195 } 196 197 } // namespace core_api 198 } // namespace extensions 199