Update path of checkdeps to buildtools checkout
[chromium-blink-merge.git] / extensions / browser / api / sockets_udp / udp_socket_event_dispatcher.cc
blobc457ec893e2f188827a2231f650335fe7a2b48f8
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.
5 #include "extensions/browser/api/sockets_udp/udp_socket_event_dispatcher.h"
7 #include "extensions/browser/api/socket/udp_socket.h"
8 #include "extensions/browser/event_router.h"
9 #include "extensions/browser/extensions_browser_client.h"
10 #include "net/base/net_errors.h"
12 namespace extensions {
13 namespace core_api {
15 using content::BrowserThread;
17 static base::LazyInstance<
18 BrowserContextKeyedAPIFactory<UDPSocketEventDispatcher> > g_factory =
19 LAZY_INSTANCE_INITIALIZER;
21 // static
22 BrowserContextKeyedAPIFactory<UDPSocketEventDispatcher>*
23 UDPSocketEventDispatcher::GetFactoryInstance() {
24 return g_factory.Pointer();
27 // static
28 UDPSocketEventDispatcher* UDPSocketEventDispatcher::Get(
29 content::BrowserContext* context) {
30 DCHECK_CURRENTLY_ON(BrowserThread::UI);
32 return BrowserContextKeyedAPIFactory<UDPSocketEventDispatcher>::Get(context);
35 UDPSocketEventDispatcher::UDPSocketEventDispatcher(
36 content::BrowserContext* context)
37 : thread_id_(Socket::kThreadId), browser_context_(context) {
38 ApiResourceManager<ResumableUDPSocket>* manager =
39 ApiResourceManager<ResumableUDPSocket>::Get(browser_context_);
40 DCHECK(manager)
41 << "There is no socket manager. "
42 "If this assertion is failing during a test, then it is likely that "
43 "TestExtensionSystem is failing to provide an instance of "
44 "ApiResourceManager<ResumableUDPSocket>.";
45 sockets_ = manager->data_;
48 UDPSocketEventDispatcher::~UDPSocketEventDispatcher() {}
50 UDPSocketEventDispatcher::ReceiveParams::ReceiveParams() {}
52 UDPSocketEventDispatcher::ReceiveParams::~ReceiveParams() {}
54 void UDPSocketEventDispatcher::OnSocketBind(const std::string& extension_id,
55 int socket_id) {
56 OnSocketResume(extension_id, socket_id);
59 void UDPSocketEventDispatcher::OnSocketResume(const std::string& extension_id,
60 int socket_id) {
61 DCHECK_CURRENTLY_ON(thread_id_);
63 ReceiveParams params;
64 params.thread_id = thread_id_;
65 params.browser_context_id = browser_context_;
66 params.extension_id = extension_id;
67 params.sockets = sockets_;
68 params.socket_id = socket_id;
70 StartReceive(params);
73 /* static */
74 void UDPSocketEventDispatcher::StartReceive(const ReceiveParams& params) {
75 DCHECK_CURRENTLY_ON(params.thread_id);
77 ResumableUDPSocket* socket =
78 params.sockets->Get(params.extension_id, params.socket_id);
79 if (socket == NULL) {
80 // This can happen if the socket is closed while our callback is active.
81 return;
83 DCHECK(params.extension_id == socket->owner_extension_id())
84 << "Socket has wrong owner.";
86 // Don't start another read if the socket has been paused.
87 if (socket->paused())
88 return;
90 int buffer_size = (socket->buffer_size() <= 0 ? 4096 : socket->buffer_size());
91 socket->RecvFrom(
92 buffer_size,
93 base::Bind(&UDPSocketEventDispatcher::ReceiveCallback, params));
96 /* static */
97 void UDPSocketEventDispatcher::ReceiveCallback(
98 const ReceiveParams& params,
99 int bytes_read,
100 scoped_refptr<net::IOBuffer> io_buffer,
101 const std::string& address,
102 int port) {
103 DCHECK_CURRENTLY_ON(params.thread_id);
105 // If |bytes_read| == 0, the message contained no data.
106 // If |bytes_read| < 0, there was a network error, and |bytes_read| is a value
107 // from "net::ERR_".
109 if (bytes_read >= 0) {
110 // Dispatch "onReceive" event.
111 sockets_udp::ReceiveInfo receive_info;
112 receive_info.socket_id = params.socket_id;
113 receive_info.data = std::string(io_buffer->data(), bytes_read);
114 receive_info.remote_address = address;
115 receive_info.remote_port = port;
116 scoped_ptr<base::ListValue> args =
117 sockets_udp::OnReceive::Create(receive_info);
118 scoped_ptr<Event> event(
119 new Event(sockets_udp::OnReceive::kEventName, args.Pass()));
120 PostEvent(params, event.Pass());
122 // Post a task to delay the read until the socket is available, as
123 // calling StartReceive at this point would error with ERR_IO_PENDING.
124 BrowserThread::PostTask(
125 params.thread_id,
126 FROM_HERE,
127 base::Bind(&UDPSocketEventDispatcher::StartReceive, params));
128 } else if (bytes_read == net::ERR_IO_PENDING) {
129 // This happens when resuming a socket which already had an
130 // active "recv" callback.
131 } else {
132 // Dispatch "onReceiveError" event but don't start another read to avoid
133 // potential infinite reads if we have a persistent network error.
134 sockets_udp::ReceiveErrorInfo receive_error_info;
135 receive_error_info.socket_id = params.socket_id;
136 receive_error_info.result_code = bytes_read;
137 scoped_ptr<base::ListValue> args =
138 sockets_udp::OnReceiveError::Create(receive_error_info);
139 scoped_ptr<Event> event(
140 new Event(sockets_udp::OnReceiveError::kEventName, args.Pass()));
141 PostEvent(params, event.Pass());
143 // Since we got an error, the socket is now "paused" until the application
144 // "resumes" it.
145 ResumableUDPSocket* socket =
146 params.sockets->Get(params.extension_id, params.socket_id);
147 if (socket) {
148 socket->set_paused(true);
153 /* static */
154 void UDPSocketEventDispatcher::PostEvent(const ReceiveParams& params,
155 scoped_ptr<Event> event) {
156 DCHECK_CURRENTLY_ON(params.thread_id);
158 BrowserThread::PostTask(BrowserThread::UI,
159 FROM_HERE,
160 base::Bind(&DispatchEvent,
161 params.browser_context_id,
162 params.extension_id,
163 base::Passed(event.Pass())));
166 /*static*/
167 void UDPSocketEventDispatcher::DispatchEvent(void* browser_context_id,
168 const std::string& extension_id,
169 scoped_ptr<Event> event) {
170 DCHECK_CURRENTLY_ON(BrowserThread::UI);
172 content::BrowserContext* context =
173 reinterpret_cast<content::BrowserContext*>(browser_context_id);
174 if (!extensions::ExtensionsBrowserClient::Get()->IsValidContext(context))
175 return;
176 EventRouter* router = EventRouter::Get(context);
177 if (router)
178 router->DispatchEventToExtension(extension_id, event.Pass());
181 } // namespace core_api
182 } // namespace extensions