Bug 911972 - MessagePort and MessageChannel in workers, r=smaug, r=bent

dom/messagechannel/SharedMessagePortMessage.cpp

@@ -0,0 +1,180 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "SharedMessagePortMessage.h"
+#include "MessagePort.h"
+#include "MessagePortChild.h"
+#include "MessagePortParent.h"
+#include "mozilla/dom/ipc/BlobChild.h"
+#include "mozilla/dom/ipc/BlobParent.h"
+#include "mozilla/dom/File.h"
+#include "mozilla/dom/PMessagePort.h"
+#include "mozilla/ipc/BackgroundChild.h"
+#include "mozilla/ipc/BackgroundParent.h"
+
+namespace mozilla {
+
+using namespace ipc;
+
+namespace dom {
+
+SharedMessagePortMessage::~SharedMessagePortMessage()
+{
+  if (!mData.IsEmpty()) {
+    FreeStructuredClone(mData, mClosure);
+  }
+}
+
+/* static */ void
+SharedMessagePortMessage::FromSharedToMessagesChild(
+                      MessagePortChild* aActor,
+                      const nsTArray<nsRefPtr<SharedMessagePortMessage>>& aData,
+                      nsTArray<MessagePortMessage>& aArray)
+{
+  MOZ_ASSERT(aActor);
+  MOZ_ASSERT(aArray.IsEmpty());
+  aArray.SetCapacity(aData.Length());
+
+  PBackgroundChild* backgroundManager = aActor->Manager();
+  MOZ_ASSERT(backgroundManager);
+
+  for (auto& data : aData) {
+    MessagePortMessage* message = aArray.AppendElement();
+    message->data().SwapElements(data->mData);
+
+    const nsTArray<nsRefPtr<BlobImpl>>& blobImpls =
+      data->mClosure.mBlobImpls;
+    if (!blobImpls.IsEmpty()) {
+      message->blobsChild().SetCapacity(blobImpls.Length());
+
+      for (uint32_t i = 0, len = blobImpls.Length(); i < len; ++i) {
+        PBlobChild* blobChild =
+          BackgroundChild::GetOrCreateActorForBlobImpl(backgroundManager,
+                                                       blobImpls[i]);
+        message->blobsChild().AppendElement(blobChild);
+      }
+    }
+
+    message->transferredPorts().AppendElements(
+      data->mClosure.mMessagePortIdentifiers);
+  }
+}
+
+/* static */ bool
+SharedMessagePortMessage::FromMessagesToSharedChild(
+                      nsTArray<MessagePortMessage>& aArray,
+                      FallibleTArray<nsRefPtr<SharedMessagePortMessage>>& aData)
+{
+  MOZ_ASSERT(aData.IsEmpty());
+
+  if (NS_WARN_IF(!aData.SetCapacity(aArray.Length(), mozilla::fallible))) {
+    return false;
+  }
+
+  for (auto& message : aArray) {
+    nsRefPtr<SharedMessagePortMessage> data = new SharedMessagePortMessage();
+
+    data->mData.SwapElements(message.data());
+
+    const nsTArray<PBlobChild*>& blobs = message.blobsChild();
+    if (!blobs.IsEmpty()) {
+      data->mClosure.mBlobImpls.SetCapacity(blobs.Length());
+
+      for (uint32_t i = 0, len = blobs.Length(); i < len; ++i) {
+        nsRefPtr<BlobImpl> impl =
+          static_cast<BlobChild*>(blobs[i])->GetBlobImpl();
+        data->mClosure.mBlobImpls.AppendElement(impl);
+      }
+    }
+
+    data->mClosure.mMessagePortIdentifiers.AppendElements(
+      message.transferredPorts());
+
+    if (!aData.AppendElement(data, mozilla::fallible)) {
+      return false;
+    }
+  }
+
+  return true;
+}
+
+/* static */ bool
+SharedMessagePortMessage::FromSharedToMessagesParent(
+                      MessagePortParent* aActor,
+                      const nsTArray<nsRefPtr<SharedMessagePortMessage>>& aData,
+                      FallibleTArray<MessagePortMessage>& aArray)
+{
+  MOZ_ASSERT(aArray.IsEmpty());
+
+  if (NS_WARN_IF(!aArray.SetCapacity(aData.Length(), mozilla::fallible))) {
+    return false;
+  }
+
+  PBackgroundParent* backgroundManager = aActor->Manager();
+  MOZ_ASSERT(backgroundManager);
+
+  for (auto& data : aData) {
+    MessagePortMessage* message = aArray.AppendElement(mozilla::fallible);
+    message->data().SwapElements(data->mData);
+
+    const nsTArray<nsRefPtr<BlobImpl>>& blobImpls = data->mClosure.mBlobImpls;
+    if (!blobImpls.IsEmpty()) {
+      message->blobsParent().SetCapacity(blobImpls.Length());
+
+      for (uint32_t i = 0, len = blobImpls.Length(); i < len; ++i) {
+        PBlobParent* blobParent =
+          BackgroundParent::GetOrCreateActorForBlobImpl(backgroundManager,
+                                                        blobImpls[i]);
+        message->blobsParent().AppendElement(blobParent);
+      }
+    }
+
+    message->transferredPorts().AppendElements(
+      data->mClosure.mMessagePortIdentifiers);
+  }
+
+  return true;
+}
+
+/* static */ bool
+SharedMessagePortMessage::FromMessagesToSharedParent(
+                      nsTArray<MessagePortMessage>& aArray,
+                      FallibleTArray<nsRefPtr<SharedMessagePortMessage>>& aData)
+{
+  MOZ_ASSERT(aData.IsEmpty());
+
+  if (NS_WARN_IF(!aData.SetCapacity(aArray.Length(), mozilla::fallible))) {
+    return false;
+  }
+
+  for (auto& message : aArray) {
+    nsRefPtr<SharedMessagePortMessage> data = new SharedMessagePortMessage();
+
+    data->mData.SwapElements(message.data());
+
+    const nsTArray<PBlobParent*>& blobs = message.blobsParent();
+    if (!blobs.IsEmpty()) {
+      data->mClosure.mBlobImpls.SetCapacity(blobs.Length());
+
+      for (uint32_t i = 0, len = blobs.Length(); i < len; ++i) {
+        nsRefPtr<BlobImpl> impl =
+          static_cast<BlobParent*>(blobs[i])->GetBlobImpl();
+        data->mClosure.mBlobImpls.AppendElement(impl);
+      }
+    }
+
+    data->mClosure.mMessagePortIdentifiers.AppendElements(
+      message.transferredPorts());
+
+    if (!aData.AppendElement(data, mozilla::fallible)) {
+      return false;
+    }
+  }
+
+  return true;
+}
+
+} // dom namespace
+} // mozilla namespace