Support Transferable objects

content/worker/webworkerclient_proxy.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/worker/webworkerclient_proxy.h"
 
 #include "base/command_line.h"
 #include "base/message_loop.h"
 #include "content/common/content_switches.h"
 #include "content/common/file_system/file_system_dispatcher.h"
 #include "content/common/file_system/webfilesystem_callback_dispatcher.h"
 #include "content/common/webmessageportchannel_impl.h"
 #include "content/common/worker_messages.h"
 // TODO(jam): uncomment this and WebWorkerClientProxy::createWorker when the
 // renderer worker code moves to content. This code isn't used now since we
 // don't support nested workers anyways.
 //#include "content/renderer/webworker_proxy.h"
 #include "content/worker/webworker_stub_base.h"
 #include "content/worker/worker_devtools_agent.h"
 #include "content/worker/worker_thread.h"
 #include "content/worker/worker_webapplicationcachehost_impl.h"
 #include "ipc/ipc_logging.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebDocument.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFileSystemCallbacks.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebSecurityOrigin.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebSerializedScriptValue.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebString.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebURL.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebWorker.h"
 
 using WebKit::WebApplicationCacheHost;
 using WebKit::WebFrame;
 using WebKit::WebMessagePortChannel;
-using WebKit::WebMessagePortChannelArray;
+#ifdef WebTransferableReceipt_h
+using WebKit::WebMessagePortReceipt;
+#endif
+using WebKit::WebTransferableReceiptArray;
+using WebKit::WebSerializedScriptValue;
 using WebKit::WebSecurityOrigin;
 using WebKit::WebString;
 using WebKit::WebWorker;
 using WebKit::WebWorkerClient;
 
 // How long to wait for worker to finish after it's been told to terminate.
 #define kMaxTimeForRunawayWorkerMs 3000
 
 WebWorkerClientProxy::WebWorkerClientProxy(int route_id,
                                            WebWorkerStubBase* stub)
     : route_id_(route_id),
       appcache_host_id_(0),
       stub_(stub),
       ALLOW_THIS_IN_INITIALIZER_LIST(kill_process_factory_(this)),
       devtools_agent_(NULL) {
 }
 
 WebWorkerClientProxy::~WebWorkerClientProxy() {
 }
 
 void WebWorkerClientProxy::postMessageToWorkerObject(
-    const WebString& message,
-    const WebMessagePortChannelArray& channels) {
-  std::vector<int> message_port_ids(channels.size());
-  std::vector<int> routing_ids(channels.size());
-  for (size_t i = 0; i < channels.size(); ++i) {
-    WebMessagePortChannelImpl* webchannel =
-        static_cast<WebMessagePortChannelImpl*>(channels[i]);
-    message_port_ids[i] = webchannel->message_port_id();
-    webchannel->QueueMessages();
-    DCHECK(message_port_ids[i] != MSG_ROUTING_NONE);
-    routing_ids[i] = MSG_ROUTING_NONE;
+    const WebString& raw_message,
+    const WebTransferableReceiptArray& receipts) {
+  size_t unflattenable_count;
+#ifndef WebTransferableReceipt_h
+  WebString message = raw_message;
+  unflattenable_count = receipts.size();
+#else
+  WebString message = WebSerializedScriptValue::flattenReceiptList(
+      raw_message, receipts, &unflattenable_count);
+#endif
+  std::vector<int> message_port_ids(unflattenable_count);
+  std::vector<int> routing_ids(unflattenable_count);
+  size_t next_unflattenable = 0;
+  for (size_t i = 0; i < receipts.size(); ++i) {
+#ifndef WebTransferableReceipt_h
+    {
+#else
+    if (!WebSerializedScriptValue::isFlattenable(receipts[i])) {
+      DCHECK(toWebMessagePortReceipt(receipts[i]));
+#endif
+      DCHECK(next_unflattenable < unflattenable_count);
+      WebMessagePortChannelImpl* webchannel =
+#ifndef WebTransferableReceipt_h
+          static_cast<WebMessagePortChannelImpl*>(receipts[i]);
+#else
+          static_cast<WebMessagePortChannelImpl*>(
+              static_cast<WebMessagePortReceipt*>(receipts[i])->channel());
+#endif
+      message_port_ids[next_unflattenable] = webchannel->message_port_id();
+      webchannel->QueueMessages();
+      CHECK(message_port_ids[next_unflattenable] != MSG_ROUTING_NONE);
+      routing_ids[next_unflattenable] = MSG_ROUTING_NONE;
+      ++next_unflattenable;
+    }
   }
+  DCHECK(next_unflattenable == unflattenable_count);
 
   Send(new WorkerMsg_PostMessage(
       route_id_, message, message_port_ids, routing_ids));
 }
 
 void WebWorkerClientProxy::postExceptionToWorkerObject(
     const WebString& error_message,
     int line_number,
     const WebString& source_url) {
   Send(new WorkerHostMsg_PostExceptionToWorkerObject(
@@ skipping 122 matching lines @@
 
   // This shuts down the process cleanly from the perspective of the browser
   // process, and avoids the crashed worker infobar from appearing to the new
   // page. It's ok to post several of theese, because the first executed task
   // will exit the message loop and subsequent ones won't be executed.
   MessageLoop::current()->PostDelayedTask(FROM_HERE,
       kill_process_factory_.NewRunnableMethod(
           &WebWorkerClientProxy::workerContextDestroyed),
           kMaxTimeForRunawayWorkerMs);
 }