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 "extensions/renderer/messaging_bindings.h" | 5 #include "extensions/renderer/messaging_bindings.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include <map> | 9 #include <map> |
10 #include <string> | 10 #include <string> |
11 | 11 |
12 #include "base/bind.h" | 12 #include "base/bind.h" |
13 #include "base/bind_helpers.h" | 13 #include "base/bind_helpers.h" |
14 #include "base/callback.h" | 14 #include "base/callback.h" |
15 #include "base/callback_helpers.h" | 15 #include "base/callback_helpers.h" |
16 #include "base/guid.h" | |
16 #include "base/lazy_instance.h" | 17 #include "base/lazy_instance.h" |
17 #include "base/message_loop/message_loop.h" | 18 #include "base/message_loop/message_loop.h" |
18 #include "base/metrics/histogram_macros.h" | 19 #include "base/metrics/histogram_macros.h" |
19 #include "base/values.h" | 20 #include "base/values.h" |
20 #include "content/public/child/v8_value_converter.h" | 21 #include "content/public/child/v8_value_converter.h" |
21 #include "content/public/common/child_process_host.h" | 22 #include "content/public/common/child_process_host.h" |
22 #include "content/public/renderer/render_frame.h" | 23 #include "content/public/renderer/render_frame.h" |
23 #include "content/public/renderer/render_thread.h" | 24 #include "content/public/renderer/render_thread.h" |
24 #include "extensions/common/api/messaging/message.h" | 25 #include "extensions/common/api/messaging/message.h" |
26 #include "extensions/common/api/messaging/port_id.h" | |
25 #include "extensions/common/extension_messages.h" | 27 #include "extensions/common/extension_messages.h" |
26 #include "extensions/common/manifest_handlers/externally_connectable.h" | 28 #include "extensions/common/manifest_handlers/externally_connectable.h" |
27 #include "extensions/renderer/extension_frame_helper.h" | 29 #include "extensions/renderer/extension_frame_helper.h" |
28 #include "extensions/renderer/extension_port.h" | 30 #include "extensions/renderer/extension_port.h" |
29 #include "extensions/renderer/gc_callback.h" | 31 #include "extensions/renderer/gc_callback.h" |
30 #include "extensions/renderer/script_context.h" | 32 #include "extensions/renderer/script_context.h" |
31 #include "extensions/renderer/script_context_set.h" | 33 #include "extensions/renderer/script_context_set.h" |
32 #include "extensions/renderer/v8_helpers.h" | 34 #include "extensions/renderer/v8_helpers.h" |
33 #include "third_party/WebKit/public/web/WebDocument.h" | 35 #include "third_party/WebKit/public/web/WebDocument.h" |
34 #include "third_party/WebKit/public/web/WebLocalFrame.h" | 36 #include "third_party/WebKit/public/web/WebLocalFrame.h" |
(...skipping 13 matching lines...) Expand all Loading... | |
48 namespace extensions { | 50 namespace extensions { |
49 | 51 |
50 using v8_helpers::ToV8String; | 52 using v8_helpers::ToV8String; |
51 | 53 |
52 namespace { | 54 namespace { |
53 | 55 |
54 // A global map between ScriptContext and MessagingBindings. | 56 // A global map between ScriptContext and MessagingBindings. |
55 base::LazyInstance<std::map<ScriptContext*, MessagingBindings*>> | 57 base::LazyInstance<std::map<ScriptContext*, MessagingBindings*>> |
56 g_messaging_map = LAZY_INSTANCE_INITIALIZER; | 58 g_messaging_map = LAZY_INSTANCE_INITIALIZER; |
57 | 59 |
58 void HasMessagePort(int global_port_id, | 60 void HasMessagePort(const PortId& port_id, |
59 bool* has_port, | 61 bool* has_port, |
60 ScriptContext* script_context) { | 62 ScriptContext* script_context) { |
61 if (*has_port) | 63 if (*has_port) |
62 return; // Stop checking if the port was found. | 64 return; // Stop checking if the port was found. |
63 | 65 |
64 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; | 66 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; |
65 DCHECK(bindings); | 67 DCHECK(bindings); |
66 if (bindings->GetPortWithGlobalId(global_port_id)) | 68 // No need for |=; we know this is false right now from above. |
67 *has_port = true; | 69 *has_port = bindings->GetPortWithId(port_id) != nullptr; |
68 } | 70 } |
69 | 71 |
70 void DispatchOnConnectToScriptContext( | 72 void DispatchOnConnectToScriptContext( |
71 int global_target_port_id, | 73 const PortId& target_port_id, |
72 const std::string& channel_name, | 74 const std::string& channel_name, |
73 const ExtensionMsg_TabConnectionInfo* source, | 75 const ExtensionMsg_TabConnectionInfo* source, |
74 const ExtensionMsg_ExternalConnectionInfo& info, | 76 const ExtensionMsg_ExternalConnectionInfo& info, |
75 const std::string& tls_channel_id, | 77 const std::string& tls_channel_id, |
76 bool* port_created, | 78 bool* port_created, |
77 ScriptContext* script_context) { | 79 ScriptContext* script_context) { |
78 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; | 80 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; |
79 DCHECK(bindings); | 81 DCHECK(bindings); |
80 | 82 |
81 int opposite_port_id = global_target_port_id ^ 1; | 83 if (bindings->context_id() == target_port_id.context_id) |
82 if (bindings->GetPortWithGlobalId(opposite_port_id)) | |
83 return; // The channel was opened by this same context; ignore it. | 84 return; // The channel was opened by this same context; ignore it. |
robwu
2016/12/02 11:58:54
I'd extend this comment with the remark that this
Devlin
2016/12/02 17:16:32
Done.
| |
84 | 85 |
85 ExtensionPort* port = | 86 ExtensionPort* port = bindings->CreateNewPortWithId(target_port_id); |
86 bindings->CreateNewPortWithGlobalId(global_target_port_id); | |
87 int local_port_id = port->local_id(); | |
88 // Remove the port. | 87 // Remove the port. |
89 base::ScopedClosureRunner remove_port( | 88 base::ScopedClosureRunner remove_port( |
90 base::Bind(&MessagingBindings::RemovePortWithLocalId, | 89 base::Bind(&MessagingBindings::RemovePortWithJsId, bindings->GetWeakPtr(), |
91 bindings->GetWeakPtr(), local_port_id)); | 90 port->js_id())); |
92 | 91 |
93 v8::Isolate* isolate = script_context->isolate(); | 92 v8::Isolate* isolate = script_context->isolate(); |
94 v8::HandleScope handle_scope(isolate); | 93 v8::HandleScope handle_scope(isolate); |
95 | 94 |
96 | 95 |
97 const std::string& source_url_spec = info.source_url.spec(); | 96 const std::string& source_url_spec = info.source_url.spec(); |
98 std::string target_extension_id = script_context->GetExtensionID(); | 97 std::string target_extension_id = script_context->GetExtensionID(); |
99 const Extension* extension = script_context->extension(); | 98 const Extension* extension = script_context->extension(); |
100 | 99 |
101 v8::Local<v8::Value> tab = v8::Null(isolate); | 100 v8::Local<v8::Value> tab = v8::Null(isolate); |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
134 !ToV8String(isolate, info.source_id.c_str(), &v8_source_id) || | 133 !ToV8String(isolate, info.source_id.c_str(), &v8_source_id) || |
135 !ToV8String(isolate, target_extension_id.c_str(), | 134 !ToV8String(isolate, target_extension_id.c_str(), |
136 &v8_target_extension_id) || | 135 &v8_target_extension_id) || |
137 !ToV8String(isolate, source_url_spec.c_str(), &v8_source_url_spec)) { | 136 !ToV8String(isolate, source_url_spec.c_str(), &v8_source_url_spec)) { |
138 NOTREACHED() << "dispatchOnConnect() passed non-string argument"; | 137 NOTREACHED() << "dispatchOnConnect() passed non-string argument"; |
139 return; | 138 return; |
140 } | 139 } |
141 | 140 |
142 v8::Local<v8::Value> arguments[] = { | 141 v8::Local<v8::Value> arguments[] = { |
143 // portId | 142 // portId |
144 v8::Integer::New(isolate, local_port_id), | 143 v8::Integer::New(isolate, port->js_id()), |
145 // channelName | 144 // channelName |
146 v8_channel_name, | 145 v8_channel_name, |
147 // sourceTab | 146 // sourceTab |
148 tab, | 147 tab, |
149 // source_frame_id | 148 // source_frame_id |
150 v8::Integer::New(isolate, source->frame_id), | 149 v8::Integer::New(isolate, source->frame_id), |
151 // guestProcessId | 150 // guestProcessId |
152 guest_process_id, | 151 guest_process_id, |
153 // guestRenderFrameRoutingId | 152 // guestRenderFrameRoutingId |
154 guest_render_frame_routing_id, | 153 guest_render_frame_routing_id, |
(...skipping 16 matching lines...) Expand all Loading... | |
171 bool used = retval.As<v8::Boolean>()->Value(); | 170 bool used = retval.As<v8::Boolean>()->Value(); |
172 *port_created |= used; | 171 *port_created |= used; |
173 if (used) // Port was used; don't remove it. | 172 if (used) // Port was used; don't remove it. |
174 remove_port.ReplaceClosure(base::Closure()); | 173 remove_port.ReplaceClosure(base::Closure()); |
175 } else { | 174 } else { |
176 LOG(ERROR) << "Empty return value from dispatchOnConnect."; | 175 LOG(ERROR) << "Empty return value from dispatchOnConnect."; |
177 } | 176 } |
178 } | 177 } |
179 | 178 |
180 void DeliverMessageToScriptContext(const Message& message, | 179 void DeliverMessageToScriptContext(const Message& message, |
181 int global_target_port_id, | 180 const PortId& target_port_id, |
182 ScriptContext* script_context) { | 181 ScriptContext* script_context) { |
183 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; | 182 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; |
184 DCHECK(bindings); | 183 DCHECK(bindings); |
185 ExtensionPort* port = bindings->GetPortWithGlobalId(global_target_port_id); | 184 ExtensionPort* port = bindings->GetPortWithId(target_port_id); |
186 if (!port) | 185 if (!port) |
187 return; | 186 return; |
188 | 187 |
189 v8::Isolate* isolate = script_context->isolate(); | 188 v8::Isolate* isolate = script_context->isolate(); |
190 v8::HandleScope handle_scope(isolate); | 189 v8::HandleScope handle_scope(isolate); |
191 | 190 |
192 v8::Local<v8::Value> port_id_handle = | 191 v8::Local<v8::Value> port_id_handle = |
193 v8::Integer::New(isolate, port->local_id()); | 192 v8::Integer::New(isolate, port->js_id()); |
194 | 193 |
195 v8::Local<v8::String> v8_data; | 194 v8::Local<v8::String> v8_data; |
196 if (!ToV8String(isolate, message.data.c_str(), &v8_data)) | 195 if (!ToV8String(isolate, message.data.c_str(), &v8_data)) |
197 return; | 196 return; |
198 std::vector<v8::Local<v8::Value>> arguments; | 197 std::vector<v8::Local<v8::Value>> arguments; |
199 arguments.push_back(v8_data); | 198 arguments.push_back(v8_data); |
200 arguments.push_back(port_id_handle); | 199 arguments.push_back(port_id_handle); |
201 | 200 |
202 std::unique_ptr<blink::WebScopedUserGesture> web_user_gesture; | 201 std::unique_ptr<blink::WebScopedUserGesture> web_user_gesture; |
203 std::unique_ptr<blink::WebScopedWindowFocusAllowedIndicator> | 202 std::unique_ptr<blink::WebScopedWindowFocusAllowedIndicator> |
204 allow_window_focus; | 203 allow_window_focus; |
205 if (message.user_gesture) { | 204 if (message.user_gesture) { |
206 web_user_gesture.reset( | 205 web_user_gesture.reset( |
207 new blink::WebScopedUserGesture(script_context->web_frame())); | 206 new blink::WebScopedUserGesture(script_context->web_frame())); |
208 | 207 |
209 if (script_context->web_frame()) { | 208 if (script_context->web_frame()) { |
210 blink::WebDocument document = script_context->web_frame()->document(); | 209 blink::WebDocument document = script_context->web_frame()->document(); |
211 allow_window_focus.reset(new blink::WebScopedWindowFocusAllowedIndicator( | 210 allow_window_focus.reset(new blink::WebScopedWindowFocusAllowedIndicator( |
212 &document)); | 211 &document)); |
213 } | 212 } |
214 } | 213 } |
215 | 214 |
216 script_context->module_system()->CallModuleMethodSafe( | 215 script_context->module_system()->CallModuleMethodSafe( |
217 "messaging", "dispatchOnMessage", &arguments); | 216 "messaging", "dispatchOnMessage", &arguments); |
218 } | 217 } |
219 | 218 |
220 void DispatchOnDisconnectToScriptContext(int global_port_id, | 219 void DispatchOnDisconnectToScriptContext(const PortId& port_id, |
221 const std::string& error_message, | 220 const std::string& error_message, |
222 ScriptContext* script_context) { | 221 ScriptContext* script_context) { |
223 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; | 222 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; |
224 DCHECK(bindings); | 223 DCHECK(bindings); |
225 ExtensionPort* port = bindings->GetPortWithGlobalId(global_port_id); | 224 ExtensionPort* port = bindings->GetPortWithId(port_id); |
226 if (!port) | 225 if (!port) |
227 return; | 226 return; |
228 | 227 |
229 v8::Isolate* isolate = script_context->isolate(); | 228 v8::Isolate* isolate = script_context->isolate(); |
230 v8::HandleScope handle_scope(isolate); | 229 v8::HandleScope handle_scope(isolate); |
231 | 230 |
232 std::vector<v8::Local<v8::Value>> arguments; | 231 std::vector<v8::Local<v8::Value>> arguments; |
233 arguments.push_back(v8::Integer::New(isolate, port->local_id())); | 232 arguments.push_back(v8::Integer::New(isolate, port->js_id())); |
234 v8::Local<v8::String> v8_error_message; | 233 v8::Local<v8::String> v8_error_message; |
235 if (!error_message.empty()) | 234 if (!error_message.empty()) |
236 ToV8String(isolate, error_message.c_str(), &v8_error_message); | 235 ToV8String(isolate, error_message.c_str(), &v8_error_message); |
237 if (!v8_error_message.IsEmpty()) { | 236 if (!v8_error_message.IsEmpty()) { |
238 arguments.push_back(v8_error_message); | 237 arguments.push_back(v8_error_message); |
239 } else { | 238 } else { |
240 arguments.push_back(v8::Null(isolate)); | 239 arguments.push_back(v8::Null(isolate)); |
241 } | 240 } |
242 | 241 |
243 script_context->module_system()->CallModuleMethodSafe( | 242 script_context->module_system()->CallModuleMethodSafe( |
244 "messaging", "dispatchOnDisconnect", &arguments); | 243 "messaging", "dispatchOnDisconnect", &arguments); |
245 } | 244 } |
246 | 245 |
247 } // namespace | 246 } // namespace |
248 | 247 |
249 MessagingBindings::MessagingBindings(ScriptContext* context) | 248 MessagingBindings::MessagingBindings(ScriptContext* context) |
250 : ObjectBackedNativeHandler(context), weak_ptr_factory_(this) { | 249 : ObjectBackedNativeHandler(context), |
250 context_id_(base::GenerateGUID()), | |
251 weak_ptr_factory_(this) { | |
251 g_messaging_map.Get()[context] = this; | 252 g_messaging_map.Get()[context] = this; |
252 RouteFunction("CloseChannel", base::Bind(&MessagingBindings::CloseChannel, | 253 RouteFunction("CloseChannel", base::Bind(&MessagingBindings::CloseChannel, |
253 base::Unretained(this))); | 254 base::Unretained(this))); |
254 RouteFunction("PostMessage", base::Bind(&MessagingBindings::PostMessage, | 255 RouteFunction("PostMessage", base::Bind(&MessagingBindings::PostMessage, |
255 base::Unretained(this))); | 256 base::Unretained(this))); |
256 // TODO(fsamuel, kalman): Move BindToGC out of messaging natives. | 257 // TODO(fsamuel, kalman): Move BindToGC out of messaging natives. |
257 RouteFunction("BindToGC", base::Bind(&MessagingBindings::BindToGC, | 258 RouteFunction("BindToGC", base::Bind(&MessagingBindings::BindToGC, |
258 base::Unretained(this))); | 259 base::Unretained(this))); |
259 RouteFunction("OpenChannelToExtension", "runtime.connect", | 260 RouteFunction("OpenChannelToExtension", "runtime.connect", |
260 base::Bind(&MessagingBindings::OpenChannelToExtension, | 261 base::Bind(&MessagingBindings::OpenChannelToExtension, |
(...skipping 18 matching lines...) Expand all Loading... | |
279 ports_created_in_unload_); | 280 ports_created_in_unload_); |
280 UMA_HISTOGRAM_COUNTS_1000( | 281 UMA_HISTOGRAM_COUNTS_1000( |
281 "Extensions.Messaging.ExtensionPortsCreated.Normal", | 282 "Extensions.Messaging.ExtensionPortsCreated.Normal", |
282 ports_created_normal_); | 283 ports_created_normal_); |
283 } | 284 } |
284 } | 285 } |
285 | 286 |
286 // static | 287 // static |
287 void MessagingBindings::ValidateMessagePort( | 288 void MessagingBindings::ValidateMessagePort( |
288 const ScriptContextSet& context_set, | 289 const ScriptContextSet& context_set, |
289 int global_port_id, | 290 const PortId& port_id, |
290 content::RenderFrame* render_frame) { | 291 content::RenderFrame* render_frame) { |
291 int routing_id = render_frame->GetRoutingID(); | 292 int routing_id = render_frame->GetRoutingID(); |
292 | 293 |
293 bool has_port = false; | 294 bool has_port = false; |
294 context_set.ForEach(render_frame, | 295 context_set.ForEach(render_frame, |
295 base::Bind(&HasMessagePort, global_port_id, &has_port)); | 296 base::Bind(&HasMessagePort, port_id, &has_port)); |
296 | 297 |
297 // A reply is only sent if the port is missing, because the port is assumed to | 298 // A reply is only sent if the port is missing, because the port is assumed to |
298 // exist unless stated otherwise. | 299 // exist unless stated otherwise. |
299 if (!has_port) { | 300 if (!has_port) { |
300 content::RenderThread::Get()->Send( | 301 content::RenderThread::Get()->Send( |
301 new ExtensionHostMsg_CloseMessagePort(routing_id, | 302 new ExtensionHostMsg_CloseMessagePort(routing_id, port_id, false)); |
302 global_port_id, false)); | |
303 } | 303 } |
304 } | 304 } |
305 | 305 |
306 // static | 306 // static |
307 void MessagingBindings::DispatchOnConnect( | 307 void MessagingBindings::DispatchOnConnect( |
308 const ScriptContextSet& context_set, | 308 const ScriptContextSet& context_set, |
309 int target_port_id, | 309 const PortId& target_port_id, |
310 const std::string& channel_name, | 310 const std::string& channel_name, |
311 const ExtensionMsg_TabConnectionInfo& source, | 311 const ExtensionMsg_TabConnectionInfo& source, |
312 const ExtensionMsg_ExternalConnectionInfo& info, | 312 const ExtensionMsg_ExternalConnectionInfo& info, |
313 const std::string& tls_channel_id, | 313 const std::string& tls_channel_id, |
314 content::RenderFrame* restrict_to_render_frame) { | 314 content::RenderFrame* restrict_to_render_frame) { |
315 DCHECK(!target_port_id.is_opener); | |
315 int routing_id = restrict_to_render_frame | 316 int routing_id = restrict_to_render_frame |
316 ? restrict_to_render_frame->GetRoutingID() | 317 ? restrict_to_render_frame->GetRoutingID() |
317 : MSG_ROUTING_NONE; | 318 : MSG_ROUTING_NONE; |
318 bool port_created = false; | 319 bool port_created = false; |
319 context_set.ForEach( | 320 context_set.ForEach( |
320 info.target_id, restrict_to_render_frame, | 321 info.target_id, restrict_to_render_frame, |
321 base::Bind(&DispatchOnConnectToScriptContext, target_port_id, | 322 base::Bind(&DispatchOnConnectToScriptContext, target_port_id, |
322 channel_name, &source, info, tls_channel_id, &port_created)); | 323 channel_name, &source, info, tls_channel_id, &port_created)); |
323 // Note: |restrict_to_render_frame| may have been deleted at this point! | 324 // Note: |restrict_to_render_frame| may have been deleted at this point! |
324 | 325 |
325 if (port_created) { | 326 if (port_created) { |
326 content::RenderThread::Get()->Send( | 327 content::RenderThread::Get()->Send( |
327 new ExtensionHostMsg_OpenMessagePort(routing_id, target_port_id)); | 328 new ExtensionHostMsg_OpenMessagePort(routing_id, target_port_id)); |
328 } else { | 329 } else { |
329 content::RenderThread::Get()->Send(new ExtensionHostMsg_CloseMessagePort( | 330 content::RenderThread::Get()->Send(new ExtensionHostMsg_CloseMessagePort( |
330 routing_id, target_port_id, false)); | 331 routing_id, target_port_id, false)); |
331 } | 332 } |
332 } | 333 } |
333 | 334 |
334 // static | 335 // static |
335 void MessagingBindings::DeliverMessage( | 336 void MessagingBindings::DeliverMessage( |
336 const ScriptContextSet& context_set, | 337 const ScriptContextSet& context_set, |
337 int target_port_id, | 338 const PortId& target_port_id, |
338 const Message& message, | 339 const Message& message, |
339 content::RenderFrame* restrict_to_render_frame) { | 340 content::RenderFrame* restrict_to_render_frame) { |
340 context_set.ForEach( | 341 context_set.ForEach( |
341 restrict_to_render_frame, | 342 restrict_to_render_frame, |
342 base::Bind(&DeliverMessageToScriptContext, message, target_port_id)); | 343 base::Bind(&DeliverMessageToScriptContext, message, target_port_id)); |
343 } | 344 } |
344 | 345 |
345 // static | 346 // static |
346 void MessagingBindings::DispatchOnDisconnect( | 347 void MessagingBindings::DispatchOnDisconnect( |
347 const ScriptContextSet& context_set, | 348 const ScriptContextSet& context_set, |
348 int port_id, | 349 const PortId& port_id, |
349 const std::string& error_message, | 350 const std::string& error_message, |
350 content::RenderFrame* restrict_to_render_frame) { | 351 content::RenderFrame* restrict_to_render_frame) { |
351 context_set.ForEach( | 352 context_set.ForEach( |
352 restrict_to_render_frame, | 353 restrict_to_render_frame, |
353 base::Bind(&DispatchOnDisconnectToScriptContext, port_id, error_message)); | 354 base::Bind(&DispatchOnDisconnectToScriptContext, port_id, error_message)); |
354 } | 355 } |
355 | 356 |
356 ExtensionPort* MessagingBindings::GetPortWithGlobalId(int id) { | 357 ExtensionPort* MessagingBindings::GetPortWithId(const PortId& id) { |
357 for (const auto& key_value : ports_) { | 358 for (const auto& key_value : ports_) { |
358 if (key_value.second->global_id() == id) | 359 if (key_value.second->id() == id) |
359 return key_value.second.get(); | 360 return key_value.second.get(); |
360 } | 361 } |
361 return nullptr; | 362 return nullptr; |
362 } | 363 } |
363 | 364 |
364 ExtensionPort* MessagingBindings::CreateNewPortWithGlobalId(int global_id) { | 365 ExtensionPort* MessagingBindings::CreateNewPortWithId(const PortId& id) { |
365 int local_id = GetNextLocalId(); | 366 int js_id = GetNextJsId(); |
366 ExtensionPort* port = | 367 auto port = base::MakeUnique<ExtensionPort>(context(), id, js_id); |
367 ports_ | 368 return ports_.insert(std::make_pair(js_id, std::move(port))) |
368 .insert(std::make_pair( | 369 .first->second.get(); |
369 local_id, base::MakeUnique<ExtensionPort>(context(), local_id))) | |
370 .first->second.get(); | |
371 port->SetGlobalId(global_id); | |
372 return port; | |
373 } | 370 } |
374 | 371 |
375 void MessagingBindings::RemovePortWithLocalId(int local_id) { | 372 void MessagingBindings::RemovePortWithJsId(int js_id) { |
376 ports_.erase(local_id); | 373 ports_.erase(js_id); |
377 } | 374 } |
378 | 375 |
379 base::WeakPtr<MessagingBindings> MessagingBindings::GetWeakPtr() { | 376 base::WeakPtr<MessagingBindings> MessagingBindings::GetWeakPtr() { |
380 return weak_ptr_factory_.GetWeakPtr(); | 377 return weak_ptr_factory_.GetWeakPtr(); |
381 } | 378 } |
382 | 379 |
383 void MessagingBindings::PostMessage( | 380 void MessagingBindings::PostMessage( |
384 const v8::FunctionCallbackInfo<v8::Value>& args) { | 381 const v8::FunctionCallbackInfo<v8::Value>& args) { |
385 // Arguments are (int32_t port_id, string message). | 382 // Arguments are (int32_t port_id, string message). |
386 CHECK(args.Length() == 2); | 383 CHECK(args.Length() == 2); |
387 CHECK(args[0]->IsInt32()); | 384 CHECK(args[0]->IsInt32()); |
388 CHECK(args[1]->IsString()); | 385 CHECK(args[1]->IsString()); |
389 | 386 |
390 int port_id = args[0].As<v8::Int32>()->Value(); | 387 int js_port_id = args[0].As<v8::Int32>()->Value(); |
391 auto iter = ports_.find(port_id); | 388 auto iter = ports_.find(js_port_id); |
392 if (iter != ports_.end()) { | 389 if (iter != ports_.end()) { |
393 iter->second->PostExtensionMessage(base::MakeUnique<Message>( | 390 iter->second->PostExtensionMessage(base::MakeUnique<Message>( |
394 *v8::String::Utf8Value(args[1]), | 391 *v8::String::Utf8Value(args[1]), |
395 blink::WebUserGestureIndicator::isProcessingUserGesture())); | 392 blink::WebUserGestureIndicator::isProcessingUserGesture())); |
396 } | 393 } |
397 } | 394 } |
398 | 395 |
399 void MessagingBindings::CloseChannel( | 396 void MessagingBindings::CloseChannel( |
400 const v8::FunctionCallbackInfo<v8::Value>& args) { | 397 const v8::FunctionCallbackInfo<v8::Value>& args) { |
401 // Arguments are (int32_t port_id, bool force_close). | 398 // Arguments are (int32_t port_id, bool force_close). |
402 CHECK_EQ(2, args.Length()); | 399 CHECK_EQ(2, args.Length()); |
403 CHECK(args[0]->IsInt32()); | 400 CHECK(args[0]->IsInt32()); |
404 CHECK(args[1]->IsBoolean()); | 401 CHECK(args[1]->IsBoolean()); |
405 | 402 |
406 int port_id = args[0].As<v8::Int32>()->Value(); | 403 int js_port_id = args[0].As<v8::Int32>()->Value(); |
407 bool force_close = args[1].As<v8::Boolean>()->Value(); | 404 bool force_close = args[1].As<v8::Boolean>()->Value(); |
408 ClosePort(port_id, force_close); | 405 ClosePort(js_port_id, force_close); |
409 } | 406 } |
410 | 407 |
411 void MessagingBindings::BindToGC( | 408 void MessagingBindings::BindToGC( |
412 const v8::FunctionCallbackInfo<v8::Value>& args) { | 409 const v8::FunctionCallbackInfo<v8::Value>& args) { |
413 CHECK(args.Length() == 3); | 410 CHECK(args.Length() == 3); |
414 CHECK(args[0]->IsObject()); | 411 CHECK(args[0]->IsObject()); |
415 CHECK(args[1]->IsFunction()); | 412 CHECK(args[1]->IsFunction()); |
416 CHECK(args[2]->IsInt32()); | 413 CHECK(args[2]->IsInt32()); |
417 int local_port_id = args[2].As<v8::Int32>()->Value(); | 414 int js_port_id = args[2].As<v8::Int32>()->Value(); |
418 base::Closure fallback = base::Bind(&base::DoNothing); | 415 base::Closure fallback = base::Bind(&base::DoNothing); |
419 if (local_port_id >= 0) { | 416 if (js_port_id >= 0) { |
420 // TODO(robwu): Falling back to closing the port shouldn't be needed. If | 417 // TODO(robwu): Falling back to closing the port shouldn't be needed. If |
421 // the script context is destroyed, then the frame has navigated. But that | 418 // the script context is destroyed, then the frame has navigated. But that |
422 // is already detected by the browser, so this logic is redundant. Remove | 419 // is already detected by the browser, so this logic is redundant. Remove |
423 // this fallback (and move BindToGC out of messaging because it is also | 420 // this fallback (and move BindToGC out of messaging because it is also |
424 // used in other places that have nothing to do with messaging...). | 421 // used in other places that have nothing to do with messaging...). |
425 fallback = base::Bind(&MessagingBindings::ClosePort, | 422 fallback = base::Bind(&MessagingBindings::ClosePort, |
426 weak_ptr_factory_.GetWeakPtr(), local_port_id, | 423 weak_ptr_factory_.GetWeakPtr(), js_port_id, |
427 false /* force_close */); | 424 false /* force_close */); |
428 } | 425 } |
429 // Destroys itself when the object is GC'd or context is invalidated. | 426 // Destroys itself when the object is GC'd or context is invalidated. |
430 new GCCallback(context(), args[0].As<v8::Object>(), | 427 new GCCallback(context(), args[0].As<v8::Object>(), |
431 args[1].As<v8::Function>(), fallback); | 428 args[1].As<v8::Function>(), fallback); |
432 } | 429 } |
433 | 430 |
434 void MessagingBindings::OpenChannelToExtension( | 431 void MessagingBindings::OpenChannelToExtension( |
435 const v8::FunctionCallbackInfo<v8::Value>& args) { | 432 const v8::FunctionCallbackInfo<v8::Value>& args) { |
436 content::RenderFrame* render_frame = context()->GetRenderFrame(); | 433 content::RenderFrame* render_frame = context()->GetRenderFrame(); |
437 if (!render_frame) | 434 if (!render_frame) |
438 return; | 435 return; |
439 | 436 |
440 // The Javascript code should validate/fill the arguments. | 437 // The Javascript code should validate/fill the arguments. |
441 CHECK_EQ(args.Length(), 3); | 438 CHECK_EQ(args.Length(), 3); |
442 CHECK(args[0]->IsString()); | 439 CHECK(args[0]->IsString()); |
443 CHECK(args[1]->IsString()); | 440 CHECK(args[1]->IsString()); |
444 CHECK(args[2]->IsBoolean()); | 441 CHECK(args[2]->IsBoolean()); |
445 | 442 |
446 int local_id = GetNextLocalId(); | 443 int js_id = GetNextJsId(); |
447 ports_[local_id] = base::MakeUnique<ExtensionPort>(context(), local_id); | 444 PortId port_id(context_id_, js_id, true); |
445 ports_[js_id] = base::MakeUnique<ExtensionPort>(context(), port_id, js_id); | |
448 | 446 |
449 ExtensionMsg_ExternalConnectionInfo info; | 447 ExtensionMsg_ExternalConnectionInfo info; |
450 // For messaging APIs, hosted apps should be considered a web page so hide | 448 // For messaging APIs, hosted apps should be considered a web page so hide |
451 // its extension ID. | 449 // its extension ID. |
452 const Extension* extension = context()->extension(); | 450 const Extension* extension = context()->extension(); |
453 if (extension && !extension->is_hosted_app()) | 451 if (extension && !extension->is_hosted_app()) |
454 info.source_id = extension->id(); | 452 info.source_id = extension->id(); |
455 | 453 |
456 info.target_id = *v8::String::Utf8Value(args[0]); | 454 info.target_id = *v8::String::Utf8Value(args[0]); |
457 info.source_url = context()->url(); | 455 info.source_url = context()->url(); |
458 std::string channel_name = *v8::String::Utf8Value(args[1]); | 456 std::string channel_name = *v8::String::Utf8Value(args[1]); |
459 // TODO(devlin): Why is this not part of info? | 457 // TODO(devlin): Why is this not part of info? |
460 bool include_tls_channel_id = | 458 bool include_tls_channel_id = |
461 args.Length() > 2 ? args[2]->BooleanValue() : false; | 459 args.Length() > 2 ? args[2]->BooleanValue() : false; |
462 | 460 |
463 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); | 461 { |
464 DCHECK(frame_helper); | 462 SCOPED_UMA_HISTOGRAM_TIMER( |
465 | 463 "Extensions.Messaging.GetPortIdSyncTime.Extension"); |
466 blink::WebLocalFrame* web_frame = render_frame->GetWebFrame(); | 464 render_frame->Send(new ExtensionHostMsg_OpenChannelToExtension( |
467 // If the frame is unloading, we need to assign the port id synchronously to | 465 render_frame->GetRoutingID(), info, channel_name, |
468 // avoid dropping the message on the floor; see crbug.com/660706. | 466 include_tls_channel_id, port_id)); |
469 // TODO(devlin): Investigate whether we need to continue supporting this long- | |
470 // term, and, if so, find an alternative that doesn't require synchronous | |
471 // IPCs. | |
472 if (!web_frame->document().isNull() && | |
473 (web_frame->document().unloadStartedDoNotUse() || | |
474 web_frame->document().processingBeforeUnloadDoNotUse())) { | |
475 ports_created_in_before_unload_ += | |
476 web_frame->document().processingBeforeUnloadDoNotUse() ? 1 : 0; | |
477 ports_created_in_unload_ += | |
478 web_frame->document().unloadStartedDoNotUse() ? 1 : 0; | |
479 int global_id = frame_helper->RequestSyncPortId(info, channel_name, | |
480 include_tls_channel_id); | |
481 ports_[local_id]->SetGlobalId(global_id); | |
482 } else { | |
483 ++ports_created_normal_; | |
484 frame_helper->RequestPortId( | |
485 info, channel_name, include_tls_channel_id, | |
486 base::Bind(&MessagingBindings::SetGlobalPortId, | |
487 weak_ptr_factory_.GetWeakPtr(), local_id)); | |
488 } | 467 } |
489 | 468 |
490 args.GetReturnValue().Set(static_cast<int32_t>(local_id)); | 469 args.GetReturnValue().Set(static_cast<int32_t>(js_id)); |
491 } | 470 } |
492 | 471 |
493 void MessagingBindings::OpenChannelToNativeApp( | 472 void MessagingBindings::OpenChannelToNativeApp( |
494 const v8::FunctionCallbackInfo<v8::Value>& args) { | 473 const v8::FunctionCallbackInfo<v8::Value>& args) { |
495 // The Javascript code should validate/fill the arguments. | 474 // The Javascript code should validate/fill the arguments. |
496 CHECK_EQ(args.Length(), 1); | 475 CHECK_EQ(args.Length(), 1); |
497 CHECK(args[0]->IsString()); | 476 CHECK(args[0]->IsString()); |
498 // This should be checked by our function routing code. | 477 // This should be checked by our function routing code. |
499 CHECK(context()->GetAvailability("runtime.connectNative").is_available()); | 478 CHECK(context()->GetAvailability("runtime.connectNative").is_available()); |
500 | 479 |
501 content::RenderFrame* render_frame = context()->GetRenderFrame(); | 480 content::RenderFrame* render_frame = context()->GetRenderFrame(); |
502 if (!render_frame) | 481 if (!render_frame) |
503 return; | 482 return; |
504 | 483 |
505 std::string native_app_name = *v8::String::Utf8Value(args[0]); | 484 std::string native_app_name = *v8::String::Utf8Value(args[0]); |
506 | 485 |
507 int local_id = GetNextLocalId(); | 486 int js_id = GetNextJsId(); |
508 ports_[local_id] = base::MakeUnique<ExtensionPort>(context(), local_id); | 487 PortId port_id(context_id_, js_id, true); |
488 ports_[js_id] = base::MakeUnique<ExtensionPort>(context(), port_id, js_id); | |
509 | 489 |
510 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); | 490 { |
511 DCHECK(frame_helper); | 491 SCOPED_UMA_HISTOGRAM_TIMER( |
512 frame_helper->RequestNativeAppPortId( | 492 "Extensions.Messaging.GetPortIdSyncTime.NativeApp"); |
513 native_app_name, | 493 render_frame->Send(new ExtensionHostMsg_OpenChannelToNativeApp( |
514 base::Bind(&MessagingBindings::SetGlobalPortId, | 494 render_frame->GetRoutingID(), native_app_name, port_id)); |
515 weak_ptr_factory_.GetWeakPtr(), local_id)); | 495 } |
516 args.GetReturnValue().Set(static_cast<int32_t>(local_id)); | 496 |
497 args.GetReturnValue().Set(static_cast<int32_t>(js_id)); | |
517 } | 498 } |
518 | 499 |
519 void MessagingBindings::OpenChannelToTab( | 500 void MessagingBindings::OpenChannelToTab( |
520 const v8::FunctionCallbackInfo<v8::Value>& args) { | 501 const v8::FunctionCallbackInfo<v8::Value>& args) { |
521 content::RenderFrame* render_frame = context()->GetRenderFrame(); | 502 content::RenderFrame* render_frame = context()->GetRenderFrame(); |
522 if (!render_frame) | 503 if (!render_frame) |
523 return; | 504 return; |
524 | 505 |
525 // tabs_custom_bindings.js unwraps arguments to tabs.connect/sendMessage and | 506 // tabs_custom_bindings.js unwraps arguments to tabs.connect/sendMessage and |
526 // passes them to OpenChannelToTab, in the following order: | 507 // passes them to OpenChannelToTab, in the following order: |
527 // - |tab_id| - Positive number that specifies the destination of the channel. | 508 // - |tab_id| - Positive number that specifies the destination of the channel. |
528 // - |frame_id| - Target frame(s) in the tab where onConnect is dispatched: | 509 // - |frame_id| - Target frame(s) in the tab where onConnect is dispatched: |
529 // -1 for all frames, 0 for the main frame, >0 for a child frame. | 510 // -1 for all frames, 0 for the main frame, >0 for a child frame. |
530 // - |extension_id| - ID of the initiating extension. | 511 // - |extension_id| - ID of the initiating extension. |
531 // - |channel_name| - A user-defined channel name. | 512 // - |channel_name| - A user-defined channel name. |
532 CHECK(args.Length() == 4); | 513 CHECK(args.Length() == 4); |
533 CHECK(args[0]->IsInt32()); | 514 CHECK(args[0]->IsInt32()); |
534 CHECK(args[1]->IsInt32()); | 515 CHECK(args[1]->IsInt32()); |
535 CHECK(args[2]->IsString()); | 516 CHECK(args[2]->IsString()); |
536 CHECK(args[3]->IsString()); | 517 CHECK(args[3]->IsString()); |
537 | 518 |
538 int local_id = GetNextLocalId(); | 519 int js_id = GetNextJsId(); |
539 ports_[local_id] = base::MakeUnique<ExtensionPort>(context(), local_id); | 520 PortId port_id(context_id_, js_id, true); |
521 ports_[js_id] = base::MakeUnique<ExtensionPort>(context(), port_id, js_id); | |
540 | 522 |
541 ExtensionMsg_TabTargetConnectionInfo info; | 523 ExtensionMsg_TabTargetConnectionInfo info; |
542 info.tab_id = args[0]->Int32Value(); | 524 info.tab_id = args[0]->Int32Value(); |
543 info.frame_id = args[1]->Int32Value(); | 525 info.frame_id = args[1]->Int32Value(); |
544 // TODO(devlin): Why is this not part of info? | 526 // TODO(devlin): Why is this not part of info? |
545 std::string extension_id = *v8::String::Utf8Value(args[2]); | 527 std::string extension_id = *v8::String::Utf8Value(args[2]); |
546 std::string channel_name = *v8::String::Utf8Value(args[3]); | 528 std::string channel_name = *v8::String::Utf8Value(args[3]); |
547 | 529 |
548 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); | 530 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); |
549 DCHECK(frame_helper); | 531 DCHECK(frame_helper); |
550 frame_helper->RequestTabPortId( | |
551 info, extension_id, channel_name, | |
552 base::Bind(&MessagingBindings::SetGlobalPortId, | |
553 weak_ptr_factory_.GetWeakPtr(), local_id)); | |
554 | 532 |
555 args.GetReturnValue().Set(static_cast<int32_t>(local_id)); | 533 { |
534 SCOPED_UMA_HISTOGRAM_TIMER("Extensions.Messaging.GetPortIdSyncTime.Tab"); | |
535 render_frame->Send(new ExtensionHostMsg_OpenChannelToTab( | |
536 render_frame->GetRoutingID(), info, extension_id, channel_name, | |
537 port_id)); | |
538 } | |
539 | |
540 args.GetReturnValue().Set(static_cast<int32_t>(js_id)); | |
556 } | 541 } |
557 | 542 |
558 void MessagingBindings::ClosePort(int local_port_id, bool force_close) { | 543 void MessagingBindings::ClosePort(int js_port_id, bool force_close) { |
559 // TODO(robwu): Merge this logic with CloseChannel once the TODO in BindToGC | 544 // TODO(robwu): Merge this logic with CloseChannel once the TODO in BindToGC |
560 // has been addressed. | 545 // has been addressed. |
561 auto iter = ports_.find(local_port_id); | 546 auto iter = ports_.find(js_port_id); |
562 if (iter != ports_.end()) { | 547 if (iter != ports_.end()) { |
563 std::unique_ptr<ExtensionPort> port = std::move(iter->second); | 548 std::unique_ptr<ExtensionPort> port = std::move(iter->second); |
564 ports_.erase(iter); | 549 ports_.erase(iter); |
565 port->Close(force_close); | 550 port->Close(force_close); |
566 // If the port hasn't been initialized, we can't delete it just yet, because | |
567 // we need to wait for it to send any pending messages. This is important | |
568 // to support the flow: | |
569 // var port = chrome.runtime.connect(); | |
570 // port.postMessage({message}); | |
571 // port.disconnect(); | |
572 if (!port->initialized()) | |
573 disconnected_ports_[port->local_id()] = std::move(port); | |
574 } | 551 } |
575 } | 552 } |
576 | 553 |
577 void MessagingBindings::SetGlobalPortId(int local_id, int global_id) { | 554 int MessagingBindings::GetNextJsId() { return next_js_id_++; } |
578 auto iter = ports_.find(local_id); | |
579 if (iter != ports_.end()) { | |
580 iter->second->SetGlobalId(global_id); | |
581 return; | |
582 } | |
583 | |
584 iter = disconnected_ports_.find(local_id); | |
585 DCHECK(iter != disconnected_ports_.end()); | |
586 iter->second->SetGlobalId(global_id); | |
587 // Setting the global id dispatches pending messages, so we can delete the | |
588 // port now. | |
589 disconnected_ports_.erase(iter); | |
590 } | |
591 | |
592 int MessagingBindings::GetNextLocalId() { return next_local_id_++; } | |
593 | 555 |
594 } // namespace extensions | 556 } // namespace extensions |
OLD | NEW |