1 files changed, 125 insertions, 0 deletions
diff --git a/dom/base/ContentProcessMessageManager.cpp b/dom/base/ContentProcessMessageManager.cpp
new file mode 100644
index 0000000000..7fe9f4c2c7
--- /dev/null
+++ b/dom/base/ContentProcessMessageManager.cpp
@@ -0,0 +1,125 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* 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 "ContentProcessMessageManager.h"
+
+#include "nsContentCID.h"
+#include "mozilla/dom/ContentChild.h"
+#include "mozilla/dom/MessageManagerBinding.h"
+#include "mozilla/dom/ParentProcessMessageManager.h"
+#include "mozilla/dom/ScriptSettings.h"
+#include "mozilla/dom/ipc/SharedMap.h"
+#include "mozilla/HoldDropJSObjects.h"
+
+using namespace mozilla;
+using namespace mozilla::dom;
+
+bool ContentProcessMessageManager::sWasCreated = false;
+
+ContentProcessMessageManager::ContentProcessMessageManager(
+    nsFrameMessageManager* aMessageManager)
+    : MessageManagerGlobal(aMessageManager), mInitialized(false) {
+  mozilla::HoldJSObjects(this);
+}
+
+ContentProcessMessageManager::~ContentProcessMessageManager() {
+  mozilla::DropJSObjects(this);
+}
+
+ContentProcessMessageManager* ContentProcessMessageManager::Get() {
+  nsCOMPtr<nsIMessageSender> service =
+      do_GetService(NS_CHILDPROCESSMESSAGEMANAGER_CONTRACTID);
+  if (!service) {
+    return nullptr;
+  }
+  sWasCreated = true;
+  return static_cast<ContentProcessMessageManager*>(service.get());
+}
+
+already_AddRefed<mozilla::dom::ipc::SharedMap>
+ContentProcessMessageManager::GetSharedData() {
+  if (ContentChild* child = ContentChild::GetSingleton()) {
+    return do_AddRef(child->SharedData());
+  }
+  auto* ppmm = nsFrameMessageManager::sParentProcessManager;
+  return do_AddRef(ppmm->SharedData()->GetReadOnly());
+}
+
+bool ContentProcessMessageManager::WasCreated() { return sWasCreated; }
+
+void ContentProcessMessageManager::MarkForCC() {
+  MarkScopesForCC();
+  MessageManagerGlobal::MarkForCC();
+}
+
+NS_IMPL_CYCLE_COLLECTION_CLASS(ContentProcessMessageManager)
+
+NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(ContentProcessMessageManager)
+  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mMessageManager)
+NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END
+
+NS_IMPL_CYCLE_COLLECTION_TRACE_BEGIN(ContentProcessMessageManager)
+  NS_IMPL_CYCLE_COLLECTION_TRACE_PRESERVED_WRAPPER
+  tmp->nsMessageManagerScriptExecutor::Trace(aCallbacks, aClosure);
+NS_IMPL_CYCLE_COLLECTION_TRACE_END
+
+NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(ContentProcessMessageManager)
+  NS_IMPL_CYCLE_COLLECTION_UNLINK_PRESERVED_WRAPPER
+  NS_IMPL_CYCLE_COLLECTION_UNLINK(mMessageManager)
+  tmp->nsMessageManagerScriptExecutor::Unlink();
+  NS_IMPL_CYCLE_COLLECTION_UNLINK_WEAK_REFERENCE
+NS_IMPL_CYCLE_COLLECTION_UNLINK_END
+
+NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(ContentProcessMessageManager)
+  NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
+  NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIMessageSender)
+  NS_INTERFACE_MAP_ENTRY(nsIMessageSender)
+  NS_INTERFACE_MAP_ENTRY(nsISupportsWeakReference)
+NS_INTERFACE_MAP_END
+
+NS_IMPL_CYCLE_COLLECTING_ADDREF(ContentProcessMessageManager)
+NS_IMPL_CYCLE_COLLECTING_RELEASE(ContentProcessMessageManager)
+
+bool ContentProcessMessageManager::Init() {
+  if (mInitialized) {
+    return true;
+  }
+  mInitialized = true;
+
+  return nsMessageManagerScriptExecutor::Init();
+}
+
+JSObject* ContentProcessMessageManager::WrapObject(
+    JSContext* aCx, JS::Handle<JSObject*> aGivenProto) {
+  return ContentProcessMessageManager_Binding::Wrap(aCx, this, aGivenProto);
+}
+
+JSObject* ContentProcessMessageManager::GetOrCreateWrapper() {
+  JS::Rooted<JS::Value> val(RootingCx());
+  {
+    // Scope to run ~AutoJSAPI before working with a raw JSObject*.
+    AutoJSAPI jsapi;
+    jsapi.Init();
+
+    if (!GetOrCreateDOMReflectorNoWrap(jsapi.cx(), this, &val)) {
+      return nullptr;
+    }
+  }
+  MOZ_ASSERT(val.isObject());
+  return &val.toObject();
+}
+
+void ContentProcessMessageManager::LoadScript(const nsAString& aURL) {
+  Init();
+  JS::Rooted<JSObject*> messageManager(mozilla::dom::RootingCx(),
+                                       GetOrCreateWrapper());
+  LoadScriptInternal(messageManager, aURL, true);
+}
+
+void ContentProcessMessageManager::SetInitialProcessData(
+    JS::Handle<JS::Value> aInitialData) {
+  mMessageManager->SetInitialProcessData(aInitialData);
+}