| 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 the channel was opened by this same context, ignore it. This should only |
| 82 if (bindings->GetPortWithGlobalId(opposite_port_id)) | 84 // happen when messages are sent to an entire process (rather than a single |
| 85 // frame) as an optimization; otherwise the browser process filters this out. |
| 86 if (bindings->context_id() == target_port_id.context_id) |
| 83 return; // The channel was opened by this same context; ignore it. | 87 return; // The channel was opened by this same context; ignore it. |
| 84 | 88 |
| 85 ExtensionPort* port = | 89 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. | 90 // Remove the port. |
| 89 base::ScopedClosureRunner remove_port( | 91 base::ScopedClosureRunner remove_port( |
| 90 base::Bind(&MessagingBindings::RemovePortWithLocalId, | 92 base::Bind(&MessagingBindings::RemovePortWithJsId, bindings->GetWeakPtr(), |
| 91 bindings->GetWeakPtr(), local_port_id)); | 93 port->js_id())); |
| 92 | 94 |
| 93 v8::Isolate* isolate = script_context->isolate(); | 95 v8::Isolate* isolate = script_context->isolate(); |
| 94 v8::HandleScope handle_scope(isolate); | 96 v8::HandleScope handle_scope(isolate); |
| 95 | 97 |
| 96 | 98 |
| 97 const std::string& source_url_spec = info.source_url.spec(); | 99 const std::string& source_url_spec = info.source_url.spec(); |
| 98 std::string target_extension_id = script_context->GetExtensionID(); | 100 std::string target_extension_id = script_context->GetExtensionID(); |
| 99 const Extension* extension = script_context->extension(); | 101 const Extension* extension = script_context->extension(); |
| 100 | 102 |
| 101 v8::Local<v8::Value> tab = v8::Null(isolate); | 103 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) || | 136 !ToV8String(isolate, info.source_id.c_str(), &v8_source_id) || |
| 135 !ToV8String(isolate, target_extension_id.c_str(), | 137 !ToV8String(isolate, target_extension_id.c_str(), |
| 136 &v8_target_extension_id) || | 138 &v8_target_extension_id) || |
| 137 !ToV8String(isolate, source_url_spec.c_str(), &v8_source_url_spec)) { | 139 !ToV8String(isolate, source_url_spec.c_str(), &v8_source_url_spec)) { |
| 138 NOTREACHED() << "dispatchOnConnect() passed non-string argument"; | 140 NOTREACHED() << "dispatchOnConnect() passed non-string argument"; |
| 139 return; | 141 return; |
| 140 } | 142 } |
| 141 | 143 |
| 142 v8::Local<v8::Value> arguments[] = { | 144 v8::Local<v8::Value> arguments[] = { |
| 143 // portId | 145 // portId |
| 144 v8::Integer::New(isolate, local_port_id), | 146 v8::Integer::New(isolate, port->js_id()), |
| 145 // channelName | 147 // channelName |
| 146 v8_channel_name, | 148 v8_channel_name, |
| 147 // sourceTab | 149 // sourceTab |
| 148 tab, | 150 tab, |
| 149 // source_frame_id | 151 // source_frame_id |
| 150 v8::Integer::New(isolate, source->frame_id), | 152 v8::Integer::New(isolate, source->frame_id), |
| 151 // guestProcessId | 153 // guestProcessId |
| 152 guest_process_id, | 154 guest_process_id, |
| 153 // guestRenderFrameRoutingId | 155 // guestRenderFrameRoutingId |
| 154 guest_render_frame_routing_id, | 156 guest_render_frame_routing_id, |
| (...skipping 16 matching lines...) Expand all Loading... |
| 171 bool used = retval.As<v8::Boolean>()->Value(); | 173 bool used = retval.As<v8::Boolean>()->Value(); |
| 172 *port_created |= used; | 174 *port_created |= used; |
| 173 if (used) // Port was used; don't remove it. | 175 if (used) // Port was used; don't remove it. |
| 174 remove_port.ReplaceClosure(base::Closure()); | 176 remove_port.ReplaceClosure(base::Closure()); |
| 175 } else { | 177 } else { |
| 176 LOG(ERROR) << "Empty return value from dispatchOnConnect."; | 178 LOG(ERROR) << "Empty return value from dispatchOnConnect."; |
| 177 } | 179 } |
| 178 } | 180 } |
| 179 | 181 |
| 180 void DeliverMessageToScriptContext(const Message& message, | 182 void DeliverMessageToScriptContext(const Message& message, |
| 181 int global_target_port_id, | 183 const PortId& target_port_id, |
| 182 ScriptContext* script_context) { | 184 ScriptContext* script_context) { |
| 183 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; | 185 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; |
| 184 DCHECK(bindings); | 186 DCHECK(bindings); |
| 185 ExtensionPort* port = bindings->GetPortWithGlobalId(global_target_port_id); | 187 ExtensionPort* port = bindings->GetPortWithId(target_port_id); |
| 186 if (!port) | 188 if (!port) |
| 187 return; | 189 return; |
| 188 | 190 |
| 189 v8::Isolate* isolate = script_context->isolate(); | 191 v8::Isolate* isolate = script_context->isolate(); |
| 190 v8::HandleScope handle_scope(isolate); | 192 v8::HandleScope handle_scope(isolate); |
| 191 | 193 |
| 192 v8::Local<v8::Value> port_id_handle = | 194 v8::Local<v8::Value> port_id_handle = |
| 193 v8::Integer::New(isolate, port->local_id()); | 195 v8::Integer::New(isolate, port->js_id()); |
| 194 | 196 |
| 195 v8::Local<v8::String> v8_data; | 197 v8::Local<v8::String> v8_data; |
| 196 if (!ToV8String(isolate, message.data.c_str(), &v8_data)) | 198 if (!ToV8String(isolate, message.data.c_str(), &v8_data)) |
| 197 return; | 199 return; |
| 198 std::vector<v8::Local<v8::Value>> arguments; | 200 std::vector<v8::Local<v8::Value>> arguments; |
| 199 arguments.push_back(v8_data); | 201 arguments.push_back(v8_data); |
| 200 arguments.push_back(port_id_handle); | 202 arguments.push_back(port_id_handle); |
| 201 | 203 |
| 202 std::unique_ptr<blink::WebScopedUserGesture> web_user_gesture; | 204 std::unique_ptr<blink::WebScopedUserGesture> web_user_gesture; |
| 203 std::unique_ptr<blink::WebScopedWindowFocusAllowedIndicator> | 205 std::unique_ptr<blink::WebScopedWindowFocusAllowedIndicator> |
| 204 allow_window_focus; | 206 allow_window_focus; |
| 205 if (message.user_gesture) { | 207 if (message.user_gesture) { |
| 206 web_user_gesture.reset( | 208 web_user_gesture.reset( |
| 207 new blink::WebScopedUserGesture(script_context->web_frame())); | 209 new blink::WebScopedUserGesture(script_context->web_frame())); |
| 208 | 210 |
| 209 if (script_context->web_frame()) { | 211 if (script_context->web_frame()) { |
| 210 blink::WebDocument document = script_context->web_frame()->document(); | 212 blink::WebDocument document = script_context->web_frame()->document(); |
| 211 allow_window_focus.reset(new blink::WebScopedWindowFocusAllowedIndicator( | 213 allow_window_focus.reset(new blink::WebScopedWindowFocusAllowedIndicator( |
| 212 &document)); | 214 &document)); |
| 213 } | 215 } |
| 214 } | 216 } |
| 215 | 217 |
| 216 script_context->module_system()->CallModuleMethodSafe( | 218 script_context->module_system()->CallModuleMethodSafe( |
| 217 "messaging", "dispatchOnMessage", &arguments); | 219 "messaging", "dispatchOnMessage", &arguments); |
| 218 } | 220 } |
| 219 | 221 |
| 220 void DispatchOnDisconnectToScriptContext(int global_port_id, | 222 void DispatchOnDisconnectToScriptContext(const PortId& port_id, |
| 221 const std::string& error_message, | 223 const std::string& error_message, |
| 222 ScriptContext* script_context) { | 224 ScriptContext* script_context) { |
| 223 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; | 225 MessagingBindings* bindings = g_messaging_map.Get()[script_context]; |
| 224 DCHECK(bindings); | 226 DCHECK(bindings); |
| 225 ExtensionPort* port = bindings->GetPortWithGlobalId(global_port_id); | 227 ExtensionPort* port = bindings->GetPortWithId(port_id); |
| 226 if (!port) | 228 if (!port) |
| 227 return; | 229 return; |
| 228 | 230 |
| 229 v8::Isolate* isolate = script_context->isolate(); | 231 v8::Isolate* isolate = script_context->isolate(); |
| 230 v8::HandleScope handle_scope(isolate); | 232 v8::HandleScope handle_scope(isolate); |
| 231 | 233 |
| 232 std::vector<v8::Local<v8::Value>> arguments; | 234 std::vector<v8::Local<v8::Value>> arguments; |
| 233 arguments.push_back(v8::Integer::New(isolate, port->local_id())); | 235 arguments.push_back(v8::Integer::New(isolate, port->js_id())); |
| 234 v8::Local<v8::String> v8_error_message; | 236 v8::Local<v8::String> v8_error_message; |
| 235 if (!error_message.empty()) | 237 if (!error_message.empty()) |
| 236 ToV8String(isolate, error_message.c_str(), &v8_error_message); | 238 ToV8String(isolate, error_message.c_str(), &v8_error_message); |
| 237 if (!v8_error_message.IsEmpty()) { | 239 if (!v8_error_message.IsEmpty()) { |
| 238 arguments.push_back(v8_error_message); | 240 arguments.push_back(v8_error_message); |
| 239 } else { | 241 } else { |
| 240 arguments.push_back(v8::Null(isolate)); | 242 arguments.push_back(v8::Null(isolate)); |
| 241 } | 243 } |
| 242 | 244 |
| 243 script_context->module_system()->CallModuleMethodSafe( | 245 script_context->module_system()->CallModuleMethodSafe( |
| 244 "messaging", "dispatchOnDisconnect", &arguments); | 246 "messaging", "dispatchOnDisconnect", &arguments); |
| 245 } | 247 } |
| 246 | 248 |
| 247 } // namespace | 249 } // namespace |
| 248 | 250 |
| 249 MessagingBindings::MessagingBindings(ScriptContext* context) | 251 MessagingBindings::MessagingBindings(ScriptContext* context) |
| 250 : ObjectBackedNativeHandler(context), weak_ptr_factory_(this) { | 252 : ObjectBackedNativeHandler(context), |
| 253 context_id_(base::GenerateGUID()), |
| 254 weak_ptr_factory_(this) { |
| 251 g_messaging_map.Get()[context] = this; | 255 g_messaging_map.Get()[context] = this; |
| 252 RouteFunction("CloseChannel", base::Bind(&MessagingBindings::CloseChannel, | 256 RouteFunction("CloseChannel", base::Bind(&MessagingBindings::CloseChannel, |
| 253 base::Unretained(this))); | 257 base::Unretained(this))); |
| 254 RouteFunction("PostMessage", base::Bind(&MessagingBindings::PostMessage, | 258 RouteFunction("PostMessage", base::Bind(&MessagingBindings::PostMessage, |
| 255 base::Unretained(this))); | 259 base::Unretained(this))); |
| 256 // TODO(fsamuel, kalman): Move BindToGC out of messaging natives. | 260 // TODO(fsamuel, kalman): Move BindToGC out of messaging natives. |
| 257 RouteFunction("BindToGC", base::Bind(&MessagingBindings::BindToGC, | 261 RouteFunction("BindToGC", base::Bind(&MessagingBindings::BindToGC, |
| 258 base::Unretained(this))); | 262 base::Unretained(this))); |
| 259 RouteFunction("OpenChannelToExtension", "runtime.connect", | 263 RouteFunction("OpenChannelToExtension", "runtime.connect", |
| 260 base::Bind(&MessagingBindings::OpenChannelToExtension, | 264 base::Bind(&MessagingBindings::OpenChannelToExtension, |
| (...skipping 18 matching lines...) Expand all Loading... |
| 279 ports_created_in_unload_); | 283 ports_created_in_unload_); |
| 280 UMA_HISTOGRAM_COUNTS_1000( | 284 UMA_HISTOGRAM_COUNTS_1000( |
| 281 "Extensions.Messaging.ExtensionPortsCreated.Normal", | 285 "Extensions.Messaging.ExtensionPortsCreated.Normal", |
| 282 ports_created_normal_); | 286 ports_created_normal_); |
| 283 } | 287 } |
| 284 } | 288 } |
| 285 | 289 |
| 286 // static | 290 // static |
| 287 void MessagingBindings::ValidateMessagePort( | 291 void MessagingBindings::ValidateMessagePort( |
| 288 const ScriptContextSet& context_set, | 292 const ScriptContextSet& context_set, |
| 289 int global_port_id, | 293 const PortId& port_id, |
| 290 content::RenderFrame* render_frame) { | 294 content::RenderFrame* render_frame) { |
| 291 int routing_id = render_frame->GetRoutingID(); | 295 int routing_id = render_frame->GetRoutingID(); |
| 292 | 296 |
| 293 bool has_port = false; | 297 bool has_port = false; |
| 294 context_set.ForEach(render_frame, | 298 context_set.ForEach(render_frame, |
| 295 base::Bind(&HasMessagePort, global_port_id, &has_port)); | 299 base::Bind(&HasMessagePort, port_id, &has_port)); |
| 296 | 300 |
| 297 // A reply is only sent if the port is missing, because the port is assumed to | 301 // A reply is only sent if the port is missing, because the port is assumed to |
| 298 // exist unless stated otherwise. | 302 // exist unless stated otherwise. |
| 299 if (!has_port) { | 303 if (!has_port) { |
| 300 content::RenderThread::Get()->Send( | 304 content::RenderThread::Get()->Send( |
| 301 new ExtensionHostMsg_CloseMessagePort(routing_id, | 305 new ExtensionHostMsg_CloseMessagePort(routing_id, port_id, false)); |
| 302 global_port_id, false)); | |
| 303 } | 306 } |
| 304 } | 307 } |
| 305 | 308 |
| 306 // static | 309 // static |
| 307 void MessagingBindings::DispatchOnConnect( | 310 void MessagingBindings::DispatchOnConnect( |
| 308 const ScriptContextSet& context_set, | 311 const ScriptContextSet& context_set, |
| 309 int target_port_id, | 312 const PortId& target_port_id, |
| 310 const std::string& channel_name, | 313 const std::string& channel_name, |
| 311 const ExtensionMsg_TabConnectionInfo& source, | 314 const ExtensionMsg_TabConnectionInfo& source, |
| 312 const ExtensionMsg_ExternalConnectionInfo& info, | 315 const ExtensionMsg_ExternalConnectionInfo& info, |
| 313 const std::string& tls_channel_id, | 316 const std::string& tls_channel_id, |
| 314 content::RenderFrame* restrict_to_render_frame) { | 317 content::RenderFrame* restrict_to_render_frame) { |
| 318 DCHECK(!target_port_id.is_opener); |
| 315 int routing_id = restrict_to_render_frame | 319 int routing_id = restrict_to_render_frame |
| 316 ? restrict_to_render_frame->GetRoutingID() | 320 ? restrict_to_render_frame->GetRoutingID() |
| 317 : MSG_ROUTING_NONE; | 321 : MSG_ROUTING_NONE; |
| 318 bool port_created = false; | 322 bool port_created = false; |
| 319 context_set.ForEach( | 323 context_set.ForEach( |
| 320 info.target_id, restrict_to_render_frame, | 324 info.target_id, restrict_to_render_frame, |
| 321 base::Bind(&DispatchOnConnectToScriptContext, target_port_id, | 325 base::Bind(&DispatchOnConnectToScriptContext, target_port_id, |
| 322 channel_name, &source, info, tls_channel_id, &port_created)); | 326 channel_name, &source, info, tls_channel_id, &port_created)); |
| 323 // Note: |restrict_to_render_frame| may have been deleted at this point! | 327 // Note: |restrict_to_render_frame| may have been deleted at this point! |
| 324 | 328 |
| 325 if (port_created) { | 329 if (port_created) { |
| 326 content::RenderThread::Get()->Send( | 330 content::RenderThread::Get()->Send( |
| 327 new ExtensionHostMsg_OpenMessagePort(routing_id, target_port_id)); | 331 new ExtensionHostMsg_OpenMessagePort(routing_id, target_port_id)); |
| 328 } else { | 332 } else { |
| 329 content::RenderThread::Get()->Send(new ExtensionHostMsg_CloseMessagePort( | 333 content::RenderThread::Get()->Send(new ExtensionHostMsg_CloseMessagePort( |
| 330 routing_id, target_port_id, false)); | 334 routing_id, target_port_id, false)); |
| 331 } | 335 } |
| 332 } | 336 } |
| 333 | 337 |
| 334 // static | 338 // static |
| 335 void MessagingBindings::DeliverMessage( | 339 void MessagingBindings::DeliverMessage( |
| 336 const ScriptContextSet& context_set, | 340 const ScriptContextSet& context_set, |
| 337 int target_port_id, | 341 const PortId& target_port_id, |
| 338 const Message& message, | 342 const Message& message, |
| 339 content::RenderFrame* restrict_to_render_frame) { | 343 content::RenderFrame* restrict_to_render_frame) { |
| 340 context_set.ForEach( | 344 context_set.ForEach( |
| 341 restrict_to_render_frame, | 345 restrict_to_render_frame, |
| 342 base::Bind(&DeliverMessageToScriptContext, message, target_port_id)); | 346 base::Bind(&DeliverMessageToScriptContext, message, target_port_id)); |
| 343 } | 347 } |
| 344 | 348 |
| 345 // static | 349 // static |
| 346 void MessagingBindings::DispatchOnDisconnect( | 350 void MessagingBindings::DispatchOnDisconnect( |
| 347 const ScriptContextSet& context_set, | 351 const ScriptContextSet& context_set, |
| 348 int port_id, | 352 const PortId& port_id, |
| 349 const std::string& error_message, | 353 const std::string& error_message, |
| 350 content::RenderFrame* restrict_to_render_frame) { | 354 content::RenderFrame* restrict_to_render_frame) { |
| 351 context_set.ForEach( | 355 context_set.ForEach( |
| 352 restrict_to_render_frame, | 356 restrict_to_render_frame, |
| 353 base::Bind(&DispatchOnDisconnectToScriptContext, port_id, error_message)); | 357 base::Bind(&DispatchOnDisconnectToScriptContext, port_id, error_message)); |
| 354 } | 358 } |
| 355 | 359 |
| 356 ExtensionPort* MessagingBindings::GetPortWithGlobalId(int id) { | 360 ExtensionPort* MessagingBindings::GetPortWithId(const PortId& id) { |
| 357 for (const auto& key_value : ports_) { | 361 for (const auto& key_value : ports_) { |
| 358 if (key_value.second->global_id() == id) | 362 if (key_value.second->id() == id) |
| 359 return key_value.second.get(); | 363 return key_value.second.get(); |
| 360 } | 364 } |
| 361 return nullptr; | 365 return nullptr; |
| 362 } | 366 } |
| 363 | 367 |
| 364 ExtensionPort* MessagingBindings::CreateNewPortWithGlobalId(int global_id) { | 368 ExtensionPort* MessagingBindings::CreateNewPortWithId(const PortId& id) { |
| 365 int local_id = GetNextLocalId(); | 369 int js_id = GetNextJsId(); |
| 366 ExtensionPort* port = | 370 auto port = base::MakeUnique<ExtensionPort>(context(), id, js_id); |
| 367 ports_ | 371 return ports_.insert(std::make_pair(js_id, std::move(port))) |
| 368 .insert(std::make_pair( | 372 .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 } | 373 } |
| 374 | 374 |
| 375 void MessagingBindings::RemovePortWithLocalId(int local_id) { | 375 void MessagingBindings::RemovePortWithJsId(int js_id) { |
| 376 ports_.erase(local_id); | 376 ports_.erase(js_id); |
| 377 } | 377 } |
| 378 | 378 |
| 379 base::WeakPtr<MessagingBindings> MessagingBindings::GetWeakPtr() { | 379 base::WeakPtr<MessagingBindings> MessagingBindings::GetWeakPtr() { |
| 380 return weak_ptr_factory_.GetWeakPtr(); | 380 return weak_ptr_factory_.GetWeakPtr(); |
| 381 } | 381 } |
| 382 | 382 |
| 383 void MessagingBindings::PostMessage( | 383 void MessagingBindings::PostMessage( |
| 384 const v8::FunctionCallbackInfo<v8::Value>& args) { | 384 const v8::FunctionCallbackInfo<v8::Value>& args) { |
| 385 // Arguments are (int32_t port_id, string message). | 385 // Arguments are (int32_t port_id, string message). |
| 386 CHECK(args.Length() == 2); | 386 CHECK(args.Length() == 2); |
| 387 CHECK(args[0]->IsInt32()); | 387 CHECK(args[0]->IsInt32()); |
| 388 CHECK(args[1]->IsString()); | 388 CHECK(args[1]->IsString()); |
| 389 | 389 |
| 390 int port_id = args[0].As<v8::Int32>()->Value(); | 390 int js_port_id = args[0].As<v8::Int32>()->Value(); |
| 391 auto iter = ports_.find(port_id); | 391 auto iter = ports_.find(js_port_id); |
| 392 if (iter != ports_.end()) { | 392 if (iter != ports_.end()) { |
| 393 iter->second->PostExtensionMessage(base::MakeUnique<Message>( | 393 iter->second->PostExtensionMessage(base::MakeUnique<Message>( |
| 394 *v8::String::Utf8Value(args[1]), | 394 *v8::String::Utf8Value(args[1]), |
| 395 blink::WebUserGestureIndicator::isProcessingUserGesture())); | 395 blink::WebUserGestureIndicator::isProcessingUserGesture())); |
| 396 } | 396 } |
| 397 } | 397 } |
| 398 | 398 |
| 399 void MessagingBindings::CloseChannel( | 399 void MessagingBindings::CloseChannel( |
| 400 const v8::FunctionCallbackInfo<v8::Value>& args) { | 400 const v8::FunctionCallbackInfo<v8::Value>& args) { |
| 401 // Arguments are (int32_t port_id, bool force_close). | 401 // Arguments are (int32_t port_id, bool force_close). |
| 402 CHECK_EQ(2, args.Length()); | 402 CHECK_EQ(2, args.Length()); |
| 403 CHECK(args[0]->IsInt32()); | 403 CHECK(args[0]->IsInt32()); |
| 404 CHECK(args[1]->IsBoolean()); | 404 CHECK(args[1]->IsBoolean()); |
| 405 | 405 |
| 406 int port_id = args[0].As<v8::Int32>()->Value(); | 406 int js_port_id = args[0].As<v8::Int32>()->Value(); |
| 407 bool force_close = args[1].As<v8::Boolean>()->Value(); | 407 bool force_close = args[1].As<v8::Boolean>()->Value(); |
| 408 ClosePort(port_id, force_close); | 408 ClosePort(js_port_id, force_close); |
| 409 } | 409 } |
| 410 | 410 |
| 411 void MessagingBindings::BindToGC( | 411 void MessagingBindings::BindToGC( |
| 412 const v8::FunctionCallbackInfo<v8::Value>& args) { | 412 const v8::FunctionCallbackInfo<v8::Value>& args) { |
| 413 CHECK(args.Length() == 3); | 413 CHECK(args.Length() == 3); |
| 414 CHECK(args[0]->IsObject()); | 414 CHECK(args[0]->IsObject()); |
| 415 CHECK(args[1]->IsFunction()); | 415 CHECK(args[1]->IsFunction()); |
| 416 CHECK(args[2]->IsInt32()); | 416 CHECK(args[2]->IsInt32()); |
| 417 int local_port_id = args[2].As<v8::Int32>()->Value(); | 417 int js_port_id = args[2].As<v8::Int32>()->Value(); |
| 418 base::Closure fallback = base::Bind(&base::DoNothing); | 418 base::Closure fallback = base::Bind(&base::DoNothing); |
| 419 if (local_port_id >= 0) { | 419 if (js_port_id >= 0) { |
| 420 // TODO(robwu): Falling back to closing the port shouldn't be needed. If | 420 // 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 | 421 // 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 | 422 // 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 | 423 // this fallback (and move BindToGC out of messaging because it is also |
| 424 // used in other places that have nothing to do with messaging...). | 424 // used in other places that have nothing to do with messaging...). |
| 425 fallback = base::Bind(&MessagingBindings::ClosePort, | 425 fallback = base::Bind(&MessagingBindings::ClosePort, |
| 426 weak_ptr_factory_.GetWeakPtr(), local_port_id, | 426 weak_ptr_factory_.GetWeakPtr(), js_port_id, |
| 427 false /* force_close */); | 427 false /* force_close */); |
| 428 } | 428 } |
| 429 // Destroys itself when the object is GC'd or context is invalidated. | 429 // Destroys itself when the object is GC'd or context is invalidated. |
| 430 new GCCallback(context(), args[0].As<v8::Object>(), | 430 new GCCallback(context(), args[0].As<v8::Object>(), |
| 431 args[1].As<v8::Function>(), fallback); | 431 args[1].As<v8::Function>(), fallback); |
| 432 } | 432 } |
| 433 | 433 |
| 434 void MessagingBindings::OpenChannelToExtension( | 434 void MessagingBindings::OpenChannelToExtension( |
| 435 const v8::FunctionCallbackInfo<v8::Value>& args) { | 435 const v8::FunctionCallbackInfo<v8::Value>& args) { |
| 436 content::RenderFrame* render_frame = context()->GetRenderFrame(); | 436 content::RenderFrame* render_frame = context()->GetRenderFrame(); |
| 437 if (!render_frame) | 437 if (!render_frame) |
| 438 return; | 438 return; |
| 439 | 439 |
| 440 // The Javascript code should validate/fill the arguments. | 440 // The Javascript code should validate/fill the arguments. |
| 441 CHECK_EQ(args.Length(), 3); | 441 CHECK_EQ(args.Length(), 3); |
| 442 CHECK(args[0]->IsString()); | 442 CHECK(args[0]->IsString()); |
| 443 CHECK(args[1]->IsString()); | 443 CHECK(args[1]->IsString()); |
| 444 CHECK(args[2]->IsBoolean()); | 444 CHECK(args[2]->IsBoolean()); |
| 445 | 445 |
| 446 int local_id = GetNextLocalId(); | 446 int js_id = GetNextJsId(); |
| 447 ports_[local_id] = base::MakeUnique<ExtensionPort>(context(), local_id); | 447 PortId port_id(context_id_, js_id, true); |
| 448 ports_[js_id] = base::MakeUnique<ExtensionPort>(context(), port_id, js_id); |
| 448 | 449 |
| 449 ExtensionMsg_ExternalConnectionInfo info; | 450 ExtensionMsg_ExternalConnectionInfo info; |
| 450 // For messaging APIs, hosted apps should be considered a web page so hide | 451 // For messaging APIs, hosted apps should be considered a web page so hide |
| 451 // its extension ID. | 452 // its extension ID. |
| 452 const Extension* extension = context()->extension(); | 453 const Extension* extension = context()->extension(); |
| 453 if (extension && !extension->is_hosted_app()) | 454 if (extension && !extension->is_hosted_app()) |
| 454 info.source_id = extension->id(); | 455 info.source_id = extension->id(); |
| 455 | 456 |
| 456 info.target_id = *v8::String::Utf8Value(args[0]); | 457 info.target_id = *v8::String::Utf8Value(args[0]); |
| 457 info.source_url = context()->url(); | 458 info.source_url = context()->url(); |
| 458 std::string channel_name = *v8::String::Utf8Value(args[1]); | 459 std::string channel_name = *v8::String::Utf8Value(args[1]); |
| 459 // TODO(devlin): Why is this not part of info? | 460 // TODO(devlin): Why is this not part of info? |
| 460 bool include_tls_channel_id = | 461 bool include_tls_channel_id = |
| 461 args.Length() > 2 ? args[2]->BooleanValue() : false; | 462 args.Length() > 2 ? args[2]->BooleanValue() : false; |
| 462 | 463 |
| 463 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); | 464 { |
| 464 DCHECK(frame_helper); | 465 SCOPED_UMA_HISTOGRAM_TIMER( |
| 465 | 466 "Extensions.Messaging.GetPortIdSyncTime.Extension"); |
| 466 blink::WebLocalFrame* web_frame = render_frame->GetWebFrame(); | 467 render_frame->Send(new ExtensionHostMsg_OpenChannelToExtension( |
| 467 // If the frame is unloading, we need to assign the port id synchronously to | 468 render_frame->GetRoutingID(), info, channel_name, |
| 468 // avoid dropping the message on the floor; see crbug.com/660706. | 469 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 } | 470 } |
| 489 | 471 |
| 490 args.GetReturnValue().Set(static_cast<int32_t>(local_id)); | 472 args.GetReturnValue().Set(static_cast<int32_t>(js_id)); |
| 491 } | 473 } |
| 492 | 474 |
| 493 void MessagingBindings::OpenChannelToNativeApp( | 475 void MessagingBindings::OpenChannelToNativeApp( |
| 494 const v8::FunctionCallbackInfo<v8::Value>& args) { | 476 const v8::FunctionCallbackInfo<v8::Value>& args) { |
| 495 // The Javascript code should validate/fill the arguments. | 477 // The Javascript code should validate/fill the arguments. |
| 496 CHECK_EQ(args.Length(), 1); | 478 CHECK_EQ(args.Length(), 1); |
| 497 CHECK(args[0]->IsString()); | 479 CHECK(args[0]->IsString()); |
| 498 // This should be checked by our function routing code. | 480 // This should be checked by our function routing code. |
| 499 CHECK(context()->GetAvailability("runtime.connectNative").is_available()); | 481 CHECK(context()->GetAvailability("runtime.connectNative").is_available()); |
| 500 | 482 |
| 501 content::RenderFrame* render_frame = context()->GetRenderFrame(); | 483 content::RenderFrame* render_frame = context()->GetRenderFrame(); |
| 502 if (!render_frame) | 484 if (!render_frame) |
| 503 return; | 485 return; |
| 504 | 486 |
| 505 std::string native_app_name = *v8::String::Utf8Value(args[0]); | 487 std::string native_app_name = *v8::String::Utf8Value(args[0]); |
| 506 | 488 |
| 507 int local_id = GetNextLocalId(); | 489 int js_id = GetNextJsId(); |
| 508 ports_[local_id] = base::MakeUnique<ExtensionPort>(context(), local_id); | 490 PortId port_id(context_id_, js_id, true); |
| 491 ports_[js_id] = base::MakeUnique<ExtensionPort>(context(), port_id, js_id); |
| 509 | 492 |
| 510 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); | 493 { |
| 511 DCHECK(frame_helper); | 494 SCOPED_UMA_HISTOGRAM_TIMER( |
| 512 frame_helper->RequestNativeAppPortId( | 495 "Extensions.Messaging.GetPortIdSyncTime.NativeApp"); |
| 513 native_app_name, | 496 render_frame->Send(new ExtensionHostMsg_OpenChannelToNativeApp( |
| 514 base::Bind(&MessagingBindings::SetGlobalPortId, | 497 render_frame->GetRoutingID(), native_app_name, port_id)); |
| 515 weak_ptr_factory_.GetWeakPtr(), local_id)); | 498 } |
| 516 args.GetReturnValue().Set(static_cast<int32_t>(local_id)); | 499 |
| 500 args.GetReturnValue().Set(static_cast<int32_t>(js_id)); |
| 517 } | 501 } |
| 518 | 502 |
| 519 void MessagingBindings::OpenChannelToTab( | 503 void MessagingBindings::OpenChannelToTab( |
| 520 const v8::FunctionCallbackInfo<v8::Value>& args) { | 504 const v8::FunctionCallbackInfo<v8::Value>& args) { |
| 521 content::RenderFrame* render_frame = context()->GetRenderFrame(); | 505 content::RenderFrame* render_frame = context()->GetRenderFrame(); |
| 522 if (!render_frame) | 506 if (!render_frame) |
| 523 return; | 507 return; |
| 524 | 508 |
| 525 // tabs_custom_bindings.js unwraps arguments to tabs.connect/sendMessage and | 509 // tabs_custom_bindings.js unwraps arguments to tabs.connect/sendMessage and |
| 526 // passes them to OpenChannelToTab, in the following order: | 510 // passes them to OpenChannelToTab, in the following order: |
| 527 // - |tab_id| - Positive number that specifies the destination of the channel. | 511 // - |tab_id| - Positive number that specifies the destination of the channel. |
| 528 // - |frame_id| - Target frame(s) in the tab where onConnect is dispatched: | 512 // - |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. | 513 // -1 for all frames, 0 for the main frame, >0 for a child frame. |
| 530 // - |extension_id| - ID of the initiating extension. | 514 // - |extension_id| - ID of the initiating extension. |
| 531 // - |channel_name| - A user-defined channel name. | 515 // - |channel_name| - A user-defined channel name. |
| 532 CHECK(args.Length() == 4); | 516 CHECK(args.Length() == 4); |
| 533 CHECK(args[0]->IsInt32()); | 517 CHECK(args[0]->IsInt32()); |
| 534 CHECK(args[1]->IsInt32()); | 518 CHECK(args[1]->IsInt32()); |
| 535 CHECK(args[2]->IsString()); | 519 CHECK(args[2]->IsString()); |
| 536 CHECK(args[3]->IsString()); | 520 CHECK(args[3]->IsString()); |
| 537 | 521 |
| 538 int local_id = GetNextLocalId(); | 522 int js_id = GetNextJsId(); |
| 539 ports_[local_id] = base::MakeUnique<ExtensionPort>(context(), local_id); | 523 PortId port_id(context_id_, js_id, true); |
| 524 ports_[js_id] = base::MakeUnique<ExtensionPort>(context(), port_id, js_id); |
| 540 | 525 |
| 541 ExtensionMsg_TabTargetConnectionInfo info; | 526 ExtensionMsg_TabTargetConnectionInfo info; |
| 542 info.tab_id = args[0]->Int32Value(); | 527 info.tab_id = args[0]->Int32Value(); |
| 543 info.frame_id = args[1]->Int32Value(); | 528 info.frame_id = args[1]->Int32Value(); |
| 544 // TODO(devlin): Why is this not part of info? | 529 // TODO(devlin): Why is this not part of info? |
| 545 std::string extension_id = *v8::String::Utf8Value(args[2]); | 530 std::string extension_id = *v8::String::Utf8Value(args[2]); |
| 546 std::string channel_name = *v8::String::Utf8Value(args[3]); | 531 std::string channel_name = *v8::String::Utf8Value(args[3]); |
| 547 | 532 |
| 548 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); | 533 ExtensionFrameHelper* frame_helper = ExtensionFrameHelper::Get(render_frame); |
| 549 DCHECK(frame_helper); | 534 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 | 535 |
| 555 args.GetReturnValue().Set(static_cast<int32_t>(local_id)); | 536 { |
| 537 SCOPED_UMA_HISTOGRAM_TIMER("Extensions.Messaging.GetPortIdSyncTime.Tab"); |
| 538 render_frame->Send(new ExtensionHostMsg_OpenChannelToTab( |
| 539 render_frame->GetRoutingID(), info, extension_id, channel_name, |
| 540 port_id)); |
| 541 } |
| 542 |
| 543 args.GetReturnValue().Set(static_cast<int32_t>(js_id)); |
| 556 } | 544 } |
| 557 | 545 |
| 558 void MessagingBindings::ClosePort(int local_port_id, bool force_close) { | 546 void MessagingBindings::ClosePort(int js_port_id, bool force_close) { |
| 559 // TODO(robwu): Merge this logic with CloseChannel once the TODO in BindToGC | 547 // TODO(robwu): Merge this logic with CloseChannel once the TODO in BindToGC |
| 560 // has been addressed. | 548 // has been addressed. |
| 561 auto iter = ports_.find(local_port_id); | 549 auto iter = ports_.find(js_port_id); |
| 562 if (iter != ports_.end()) { | 550 if (iter != ports_.end()) { |
| 563 std::unique_ptr<ExtensionPort> port = std::move(iter->second); | 551 std::unique_ptr<ExtensionPort> port = std::move(iter->second); |
| 564 ports_.erase(iter); | 552 ports_.erase(iter); |
| 565 port->Close(force_close); | 553 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 } | 554 } |
| 575 } | 555 } |
| 576 | 556 |
| 577 void MessagingBindings::SetGlobalPortId(int local_id, int global_id) { | 557 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 | 558 |
| 594 } // namespace extensions | 559 } // namespace extensions |
| OLD | NEW |