| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "content/child/webmessageportchannel_impl.h" | 5 #include "content/child/webmessageportchannel_impl.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 #include <utility> | 8 #include <utility> |
| 9 | 9 |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| 11 #include "base/values.h" | |
| 12 #include "content/child/child_process.h" | 11 #include "content/child/child_process.h" |
| 13 #include "content/child/child_thread_impl.h" | 12 #include "content/child/child_thread_impl.h" |
| 14 #include "content/common/message_port_messages.h" | 13 #include "content/common/message_port_messages.h" |
| 15 #include "content/public/child/v8_value_converter.h" | |
| 16 #include "third_party/WebKit/public/platform/WebMessagePortChannelClient.h" | 14 #include "third_party/WebKit/public/platform/WebMessagePortChannelClient.h" |
| 17 #include "third_party/WebKit/public/platform/WebString.h" | 15 #include "third_party/WebKit/public/platform/WebString.h" |
| 18 #include "third_party/WebKit/public/web/WebSerializedScriptValue.h" | 16 #include "third_party/WebKit/public/web/WebSerializedScriptValue.h" |
| 19 #include "v8/include/v8.h" | 17 #include "v8/include/v8.h" |
| 20 | 18 |
| 21 using blink::WebMessagePortChannel; | 19 using blink::WebMessagePortChannel; |
| 22 using blink::WebMessagePortChannelArray; | 20 using blink::WebMessagePortChannelArray; |
| 23 using blink::WebMessagePortChannelClient; | 21 using blink::WebMessagePortChannelClient; |
| 24 using blink::WebString; | 22 using blink::WebString; |
| 25 | 23 |
| 26 namespace content { | 24 namespace content { |
| 27 | 25 |
| 28 WebMessagePortChannelImpl::WebMessagePortChannelImpl( | 26 WebMessagePortChannelImpl::WebMessagePortChannelImpl( |
| 29 const scoped_refptr<base::SingleThreadTaskRunner>& main_thread_task_runner) | 27 const scoped_refptr<base::SingleThreadTaskRunner>& main_thread_task_runner) |
| 30 : client_(NULL), | 28 : client_(NULL), |
| 31 route_id_(MSG_ROUTING_NONE), | 29 route_id_(MSG_ROUTING_NONE), |
| 32 message_port_id_(MSG_ROUTING_NONE), | 30 message_port_id_(MSG_ROUTING_NONE), |
| 33 send_messages_as_values_(false), | |
| 34 main_thread_task_runner_(main_thread_task_runner) { | 31 main_thread_task_runner_(main_thread_task_runner) { |
| 35 AddRef(); | 32 AddRef(); |
| 36 Init(); | 33 Init(); |
| 37 } | 34 } |
| 38 | 35 |
| 39 WebMessagePortChannelImpl::WebMessagePortChannelImpl( | 36 WebMessagePortChannelImpl::WebMessagePortChannelImpl( |
| 40 int route_id, | 37 int route_id, |
| 41 const TransferredMessagePort& port, | 38 int port_id, |
| 42 const scoped_refptr<base::SingleThreadTaskRunner>& main_thread_task_runner) | 39 const scoped_refptr<base::SingleThreadTaskRunner>& main_thread_task_runner) |
| 43 : client_(NULL), | 40 : client_(NULL), |
| 44 route_id_(route_id), | 41 route_id_(route_id), |
| 45 message_port_id_(port.id), | 42 message_port_id_(port_id), |
| 46 send_messages_as_values_(port.send_messages_as_values), | |
| 47 main_thread_task_runner_(main_thread_task_runner) { | 43 main_thread_task_runner_(main_thread_task_runner) { |
| 48 AddRef(); | 44 AddRef(); |
| 49 Init(); | 45 Init(); |
| 50 } | 46 } |
| 51 | 47 |
| 52 WebMessagePortChannelImpl::~WebMessagePortChannelImpl() { | 48 WebMessagePortChannelImpl::~WebMessagePortChannelImpl() { |
| 53 // If we have any queued messages with attached ports, manually destroy them. | 49 // If we have any queued messages with attached ports, manually destroy them. |
| 54 while (!message_queue_.empty()) { | 50 while (!message_queue_.empty()) { |
| 55 const WebMessagePortChannelArray& channel_array = | 51 const WebMessagePortChannelArray& channel_array = |
| 56 message_queue_.front().ports; | 52 message_queue_.front().ports; |
| (...skipping 21 matching lines...) Expand all Loading... |
| 78 new WebMessagePortChannelImpl(main_thread_task_runner); | 74 new WebMessagePortChannelImpl(main_thread_task_runner); |
| 79 | 75 |
| 80 impl1->Entangle(impl2); | 76 impl1->Entangle(impl2); |
| 81 impl2->Entangle(impl1); | 77 impl2->Entangle(impl1); |
| 82 | 78 |
| 83 *channel1 = impl1; | 79 *channel1 = impl1; |
| 84 *channel2 = impl2; | 80 *channel2 = impl2; |
| 85 } | 81 } |
| 86 | 82 |
| 87 // static | 83 // static |
| 88 std::vector<TransferredMessagePort> | 84 std::vector<int> |
| 89 WebMessagePortChannelImpl::ExtractMessagePortIDs( | 85 WebMessagePortChannelImpl::ExtractMessagePortIDs( |
| 90 std::unique_ptr<WebMessagePortChannelArray> channels) { | 86 std::unique_ptr<WebMessagePortChannelArray> channels) { |
| 91 std::vector<TransferredMessagePort> message_ports; | 87 std::vector<int> message_ports; |
| 92 if (channels) | 88 if (channels) |
| 93 message_ports = ExtractMessagePortIDs(*channels); | 89 message_ports = ExtractMessagePortIDs(*channels); |
| 94 return message_ports; | 90 return message_ports; |
| 95 } | 91 } |
| 96 | 92 |
| 97 // static | 93 // static |
| 98 std::vector<TransferredMessagePort> | 94 std::vector<int> |
| 99 WebMessagePortChannelImpl::ExtractMessagePortIDs( | 95 WebMessagePortChannelImpl::ExtractMessagePortIDs( |
| 100 const WebMessagePortChannelArray& channels) { | 96 const WebMessagePortChannelArray& channels) { |
| 101 std::vector<TransferredMessagePort> message_ports(channels.size()); | 97 std::vector<int> message_ports(channels.size()); |
| 102 for (size_t i = 0; i < channels.size(); ++i) { | 98 for (size_t i = 0; i < channels.size(); ++i) { |
| 103 WebMessagePortChannelImpl* webchannel = | 99 WebMessagePortChannelImpl* webchannel = |
| 104 static_cast<WebMessagePortChannelImpl*>(channels[i]); | 100 static_cast<WebMessagePortChannelImpl*>(channels[i]); |
| 105 // The message port ids might not be set up yet if this channel | 101 // The message port ids might not be set up yet if this channel |
| 106 // wasn't created on the main thread. | 102 // wasn't created on the main thread. |
| 107 DCHECK(webchannel->main_thread_task_runner_->BelongsToCurrentThread()); | 103 DCHECK(webchannel->main_thread_task_runner_->BelongsToCurrentThread()); |
| 108 message_ports[i].id = webchannel->message_port_id(); | 104 message_ports[i] = webchannel->message_port_id(); |
| 109 message_ports[i].send_messages_as_values = | |
| 110 webchannel->send_messages_as_values_; | |
| 111 webchannel->QueueMessages(); | 105 webchannel->QueueMessages(); |
| 112 DCHECK(message_ports[i].id != MSG_ROUTING_NONE); | 106 DCHECK(message_ports[i] != MSG_ROUTING_NONE); |
| 113 } | 107 } |
| 114 return message_ports; | 108 return message_ports; |
| 115 } | 109 } |
| 116 | 110 |
| 117 // static | 111 // static |
| 118 std::vector<TransferredMessagePort> | 112 std::vector<int> |
| 119 WebMessagePortChannelImpl::ExtractMessagePortIDsWithoutQueueing( | 113 WebMessagePortChannelImpl::ExtractMessagePortIDsWithoutQueueing( |
| 120 std::unique_ptr<WebMessagePortChannelArray> channels) { | 114 std::unique_ptr<WebMessagePortChannelArray> channels) { |
| 121 if (!channels) | 115 if (!channels) |
| 122 return std::vector<TransferredMessagePort>(); | 116 return std::vector<int>(); |
| 123 | 117 |
| 124 std::vector<TransferredMessagePort> message_ports(channels->size()); | 118 std::vector<int> message_ports(channels->size()); |
| 125 for (size_t i = 0; i < channels->size(); ++i) { | 119 for (size_t i = 0; i < channels->size(); ++i) { |
| 126 WebMessagePortChannelImpl* webchannel = | 120 WebMessagePortChannelImpl* webchannel = |
| 127 static_cast<WebMessagePortChannelImpl*>((*channels)[i]); | 121 static_cast<WebMessagePortChannelImpl*>((*channels)[i]); |
| 128 // The message port ids might not be set up yet if this channel | 122 // The message port ids might not be set up yet if this channel |
| 129 // wasn't created on the main thread. | 123 // wasn't created on the main thread. |
| 130 DCHECK(webchannel->main_thread_task_runner_->BelongsToCurrentThread()); | 124 DCHECK(webchannel->main_thread_task_runner_->BelongsToCurrentThread()); |
| 131 message_ports[i].id = webchannel->message_port_id(); | 125 message_ports[i] = webchannel->message_port_id(); |
| 132 message_ports[i].send_messages_as_values = | |
| 133 webchannel->send_messages_as_values_; | |
| 134 // Don't queue messages, but do increase the child processes ref-count to | 126 // Don't queue messages, but do increase the child processes ref-count to |
| 135 // ensure this child process stays alive long enough to receive all | 127 // ensure this child process stays alive long enough to receive all |
| 136 // in-flight messages. | 128 // in-flight messages. |
| 137 ChildProcess::current()->AddRefProcess(); | 129 ChildProcess::current()->AddRefProcess(); |
| 138 DCHECK(message_ports[i].id != MSG_ROUTING_NONE); | 130 DCHECK(message_ports[i] != MSG_ROUTING_NONE); |
| 139 } | 131 } |
| 140 return message_ports; | 132 return message_ports; |
| 141 } | 133 } |
| 142 | 134 |
| 143 // static | 135 // static |
| 144 WebMessagePortChannelArray WebMessagePortChannelImpl::CreatePorts( | 136 WebMessagePortChannelArray WebMessagePortChannelImpl::CreatePorts( |
| 145 const std::vector<TransferredMessagePort>& message_ports, | 137 const std::vector<int>& message_ports, |
| 146 const std::vector<int>& new_routing_ids, | 138 const std::vector<int>& new_routing_ids, |
| 147 const scoped_refptr<base::SingleThreadTaskRunner>& | 139 const scoped_refptr<base::SingleThreadTaskRunner>& |
| 148 main_thread_task_runner) { | 140 main_thread_task_runner) { |
| 149 DCHECK_EQ(message_ports.size(), new_routing_ids.size()); | 141 DCHECK_EQ(message_ports.size(), new_routing_ids.size()); |
| 150 WebMessagePortChannelArray channels(message_ports.size()); | 142 WebMessagePortChannelArray channels(message_ports.size()); |
| 151 for (size_t i = 0; i < message_ports.size() && i < new_routing_ids.size(); | 143 for (size_t i = 0; i < message_ports.size() && i < new_routing_ids.size(); |
| 152 ++i) { | 144 ++i) { |
| 153 channels[i] = new WebMessagePortChannelImpl( | 145 channels[i] = new WebMessagePortChannelImpl( |
| 154 new_routing_ids[i], message_ports[i], | 146 new_routing_ids[i], message_ports[i], |
| 155 main_thread_task_runner); | 147 main_thread_task_runner); |
| 156 } | 148 } |
| 157 return channels; | 149 return channels; |
| 158 } | 150 } |
| 159 | 151 |
| 160 void WebMessagePortChannelImpl::setClient(WebMessagePortChannelClient* client) { | 152 void WebMessagePortChannelImpl::setClient(WebMessagePortChannelClient* client) { |
| 161 // Must lock here since client_ is called on the main thread. | 153 // Must lock here since client_ is called on the main thread. |
| 162 base::AutoLock auto_lock(lock_); | 154 base::AutoLock auto_lock(lock_); |
| 163 client_ = client; | 155 client_ = client; |
| 164 } | 156 } |
| 165 | 157 |
| 166 void WebMessagePortChannelImpl::destroy() { | 158 void WebMessagePortChannelImpl::destroy() { |
| 167 setClient(NULL); | 159 setClient(NULL); |
| 168 | 160 |
| 169 // Release the object on the main thread, since the destructor might want to | 161 // Release the object on the main thread, since the destructor might want to |
| 170 // send an IPC, and that has to happen on the main thread. | 162 // send an IPC, and that has to happen on the main thread. |
| 171 main_thread_task_runner_->ReleaseSoon(FROM_HERE, this); | 163 main_thread_task_runner_->ReleaseSoon(FROM_HERE, this); |
| 172 } | 164 } |
| 173 | 165 |
| 174 void WebMessagePortChannelImpl::postMessage( | 166 void WebMessagePortChannelImpl::postMessage( |
| 175 const WebString& message_as_string, | 167 const WebString& message, |
| 176 WebMessagePortChannelArray* channels_ptr) { | 168 WebMessagePortChannelArray* channels_ptr) { |
| 177 MessagePortMessage message(message_as_string); | |
| 178 std::unique_ptr<WebMessagePortChannelArray> channels(channels_ptr); | 169 std::unique_ptr<WebMessagePortChannelArray> channels(channels_ptr); |
| 179 if (send_messages_as_values_) { | |
| 180 blink::WebSerializedScriptValue serialized_value = | |
| 181 blink::WebSerializedScriptValue::fromString(message_as_string); | |
| 182 v8::Local<v8::Value> v8_value = serialized_value.deserialize(); | |
| 183 std::unique_ptr<V8ValueConverter> converter(V8ValueConverter::create()); | |
| 184 converter->SetDateAllowed(true); | |
| 185 converter->SetRegExpAllowed(true); | |
| 186 std::unique_ptr<base::Value> message_as_value(converter->FromV8Value( | |
| 187 v8_value, v8::Isolate::GetCurrent()->GetCurrentContext())); | |
| 188 message = MessagePortMessage(std::move(message_as_value)); | |
| 189 } | |
| 190 if (!main_thread_task_runner_->BelongsToCurrentThread()) { | 170 if (!main_thread_task_runner_->BelongsToCurrentThread()) { |
| 191 main_thread_task_runner_->PostTask( | 171 main_thread_task_runner_->PostTask( |
| 192 FROM_HERE, base::Bind(&WebMessagePortChannelImpl::SendPostMessage, this, | 172 FROM_HERE, base::Bind(&WebMessagePortChannelImpl::SendPostMessage, this, |
| 193 message, base::Passed(std::move(channels)))); | 173 message, base::Passed(std::move(channels)))); |
| 194 } else { | 174 } else { |
| 195 SendPostMessage(message, std::move(channels)); | 175 SendPostMessage(message, std::move(channels)); |
| 196 } | 176 } |
| 197 } | 177 } |
| 198 | 178 |
| 199 void WebMessagePortChannelImpl::SendPostMessage( | 179 void WebMessagePortChannelImpl::SendPostMessage( |
| 200 const MessagePortMessage& message, | 180 const base::string16& message, |
| 201 std::unique_ptr<WebMessagePortChannelArray> channels) { | 181 std::unique_ptr<WebMessagePortChannelArray> channels) { |
| 202 IPC::Message* msg = new MessagePortHostMsg_PostMessage( | 182 IPC::Message* msg = new MessagePortHostMsg_PostMessage( |
| 203 message_port_id_, message, ExtractMessagePortIDs(std::move(channels))); | 183 message_port_id_, message, ExtractMessagePortIDs(std::move(channels))); |
| 204 Send(msg); | 184 Send(msg); |
| 205 } | 185 } |
| 206 | 186 |
| 207 bool WebMessagePortChannelImpl::tryGetMessage( | 187 bool WebMessagePortChannelImpl::tryGetMessage( |
| 208 WebString* message, | 188 WebString* message, |
| 209 WebMessagePortChannelArray& channels) { | 189 WebMessagePortChannelArray& channels) { |
| 210 base::AutoLock auto_lock(lock_); | 190 base::AutoLock auto_lock(lock_); |
| 211 if (message_queue_.empty()) | 191 if (message_queue_.empty()) |
| 212 return false; | 192 return false; |
| 213 | 193 |
| 214 const MessagePortMessage& data = message_queue_.front().message; | 194 *message = message_queue_.front().message; |
| 215 DCHECK(data.is_string() != data.is_value()); | |
| 216 if (data.is_value()) { | |
| 217 v8::HandleScope handle_scope(client_->scriptIsolate()); | |
| 218 v8::Context::Scope context_scope( | |
| 219 client_->scriptContextForMessageConversion()); | |
| 220 std::unique_ptr<V8ValueConverter> converter(V8ValueConverter::create()); | |
| 221 converter->SetDateAllowed(true); | |
| 222 converter->SetRegExpAllowed(true); | |
| 223 v8::Local<v8::Value> v8_value = converter->ToV8Value( | |
| 224 data.as_value(), client_->scriptContextForMessageConversion()); | |
| 225 blink::WebSerializedScriptValue serialized_value = | |
| 226 blink::WebSerializedScriptValue::serialize(v8_value); | |
| 227 *message = serialized_value.toString(); | |
| 228 } else { | |
| 229 *message = message_queue_.front().message.message_as_string; | |
| 230 } | |
| 231 channels = message_queue_.front().ports; | 195 channels = message_queue_.front().ports; |
| 232 message_queue_.pop(); | 196 message_queue_.pop(); |
| 233 return true; | 197 return true; |
| 234 } | 198 } |
| 235 | 199 |
| 236 void WebMessagePortChannelImpl::Init() { | 200 void WebMessagePortChannelImpl::Init() { |
| 237 if (!main_thread_task_runner_->BelongsToCurrentThread()) { | 201 if (!main_thread_task_runner_->BelongsToCurrentThread()) { |
| 238 main_thread_task_runner_->PostTask( | 202 main_thread_task_runner_->PostTask( |
| 239 FROM_HERE, base::Bind(&WebMessagePortChannelImpl::Init, this)); | 203 FROM_HERE, base::Bind(&WebMessagePortChannelImpl::Init, this)); |
| 240 return; | 204 return; |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 302 bool handled = true; | 266 bool handled = true; |
| 303 IPC_BEGIN_MESSAGE_MAP(WebMessagePortChannelImpl, message) | 267 IPC_BEGIN_MESSAGE_MAP(WebMessagePortChannelImpl, message) |
| 304 IPC_MESSAGE_HANDLER(MessagePortMsg_Message, OnMessage) | 268 IPC_MESSAGE_HANDLER(MessagePortMsg_Message, OnMessage) |
| 305 IPC_MESSAGE_HANDLER(MessagePortMsg_MessagesQueued, OnMessagesQueued) | 269 IPC_MESSAGE_HANDLER(MessagePortMsg_MessagesQueued, OnMessagesQueued) |
| 306 IPC_MESSAGE_UNHANDLED(handled = false) | 270 IPC_MESSAGE_UNHANDLED(handled = false) |
| 307 IPC_END_MESSAGE_MAP() | 271 IPC_END_MESSAGE_MAP() |
| 308 return handled; | 272 return handled; |
| 309 } | 273 } |
| 310 | 274 |
| 311 void WebMessagePortChannelImpl::OnMessage( | 275 void WebMessagePortChannelImpl::OnMessage( |
| 312 const MessagePortMessage& message, | 276 const base::string16& message, |
| 313 const std::vector<TransferredMessagePort>& sent_message_ports, | 277 const std::vector<int>& sent_message_ports, |
| 314 const std::vector<int>& new_routing_ids) { | 278 const std::vector<int>& new_routing_ids) { |
| 315 base::AutoLock auto_lock(lock_); | 279 base::AutoLock auto_lock(lock_); |
| 316 Message msg; | 280 Message msg; |
| 317 msg.message = message; | 281 msg.message = message; |
| 318 msg.ports = CreatePorts(sent_message_ports, new_routing_ids, | 282 msg.ports = CreatePorts(sent_message_ports, new_routing_ids, |
| 319 main_thread_task_runner_.get()); | 283 main_thread_task_runner_.get()); |
| 320 | 284 |
| 321 bool was_empty = message_queue_.empty(); | 285 bool was_empty = message_queue_.empty(); |
| 322 message_queue_.push(msg); | 286 message_queue_.push(msg); |
| 323 if (client_ && was_empty) | 287 if (client_ && was_empty) |
| 324 client_->messageAvailable(); | 288 client_->messageAvailable(); |
| 325 } | 289 } |
| 326 | 290 |
| 327 void WebMessagePortChannelImpl::OnMessagesQueued() { | 291 void WebMessagePortChannelImpl::OnMessagesQueued() { |
| 328 std::vector<QueuedMessage> queued_messages; | 292 std::vector<QueuedMessage> queued_messages; |
| 329 | 293 |
| 330 { | 294 { |
| 331 base::AutoLock auto_lock(lock_); | 295 base::AutoLock auto_lock(lock_); |
| 332 queued_messages.reserve(message_queue_.size()); | 296 queued_messages.reserve(message_queue_.size()); |
| 333 while (!message_queue_.empty()) { | 297 while (!message_queue_.empty()) { |
| 334 MessagePortMessage message = message_queue_.front().message; | 298 base::string16 message = message_queue_.front().message; |
| 335 std::vector<TransferredMessagePort> ports = | 299 std::vector<int> ports = |
| 336 ExtractMessagePortIDs(message_queue_.front().ports); | 300 ExtractMessagePortIDs(message_queue_.front().ports); |
| 337 queued_messages.push_back(std::make_pair(message, ports)); | 301 queued_messages.push_back(std::make_pair(message, ports)); |
| 338 message_queue_.pop(); | 302 message_queue_.pop(); |
| 339 } | 303 } |
| 340 } | 304 } |
| 341 | 305 |
| 342 Send(new MessagePortHostMsg_SendQueuedMessages( | 306 Send(new MessagePortHostMsg_SendQueuedMessages( |
| 343 message_port_id_, queued_messages)); | 307 message_port_id_, queued_messages)); |
| 344 | 308 |
| 345 message_port_id_ = MSG_ROUTING_NONE; | 309 message_port_id_ = MSG_ROUTING_NONE; |
| 346 | 310 |
| 347 Release(); | 311 Release(); |
| 348 ChildProcess::current()->ReleaseProcess(); | 312 ChildProcess::current()->ReleaseProcess(); |
| 349 } | 313 } |
| 350 | 314 |
| 351 WebMessagePortChannelImpl::Message::Message() {} | 315 WebMessagePortChannelImpl::Message::Message() {} |
| 352 | 316 |
| 353 WebMessagePortChannelImpl::Message::Message(const Message& other) = default; | 317 WebMessagePortChannelImpl::Message::Message(const Message& other) = default; |
| 354 | 318 |
| 355 WebMessagePortChannelImpl::Message::~Message() {} | 319 WebMessagePortChannelImpl::Message::~Message() {} |
| 356 | 320 |
| 357 } // namespace content | 321 } // namespace content |
| OLD | NEW |