1 files changed, 96 insertions, 0 deletions
diff --git a/dom/base/messageWakeupService.js b/dom/base/messageWakeupService.js
new file mode 100644
index 000000000..73d7c2431
--- /dev/null
+++ b/dom/base/messageWakeupService.js
@@ -0,0 +1,96 @@
+/* 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/. */
+
+Components.utils.import("resource://gre/modules/XPCOMUtils.jsm");
+
+const Cc = Components.classes;
+const Ci = Components.interfaces;
+
+const CATEGORY_WAKEUP_REQUEST = "wakeup-request";
+
+function MessageWakeupService() { };
+
+MessageWakeupService.prototype =
+{
+  classID:          Components.ID("{f9798742-4f7b-4188-86ba-48b116412b29}"),
+  QueryInterface:   XPCOMUtils.generateQI([Ci.nsIObserver]),
+
+  messagesData: [],
+
+  get messageManager() {
+    if (!this._messageManager)
+      this._messageManager = Cc["@mozilla.org/parentprocessmessagemanager;1"].
+                             getService(Ci.nsIMessageListenerManager);
+    return this._messageManager;
+  },
+
+  requestWakeup: function(aMessageName, aCid, aIid, aMethod) {
+    this.messagesData[aMessageName] = {
+      cid: aCid,
+      iid: aIid,
+      method: aMethod,
+    };
+
+    this.messageManager.addMessageListener(aMessageName, this);
+  },
+
+  receiveMessage: function(aMessage) {
+    let data = this.messagesData[aMessage.name];
+    // TODO: When bug 593407 is ready, stop doing the wrappedJSObject hack
+    //       and use this line instead:
+    //                  QueryInterface(Ci.nsIMessageListener);
+    let service = Cc[data.cid][data.method](Ci[data.iid]).
+                  wrappedJSObject;
+
+    // The receiveMessage() call itself may spin an event loop, and we
+    // do not want to swap listeners in that - it would cause the current
+    // message to be answered by two listeners. So, we call that first,
+    // then queue the swap for the next event loop
+    let ret = service.receiveMessage(aMessage);
+
+    if (data.timer) {
+      // Handle the case of two such messages happening in quick succession
+      data.timer.cancel();
+      data.timer = null;
+    }
+
+    data.timer = Cc["@mozilla.org/timer;1"].createInstance(Ci.nsITimer);
+    let self = this;
+    data.timer.initWithCallback(function() {
+      self.messageManager.addMessageListener(aMessage.name, service);
+      self.messageManager.removeMessageListener(aMessage.name, self);
+      delete self.messagesData[aMessage.name];
+    }, 0, Ci.nsITimer.TYPE_ONE_SHOT);
+
+    return ret;
+  },
+
+  observe: function TM_observe(aSubject, aTopic, aData) {
+    switch (aTopic) {
+      case "profile-after-change":
+        {
+          var catMan = Cc["@mozilla.org/categorymanager;1"].
+                           getService(Ci.nsICategoryManager);
+          var entries = catMan.enumerateCategory(CATEGORY_WAKEUP_REQUEST);
+          while (entries.hasMoreElements()) {
+            var entry = entries.getNext().QueryInterface(Ci.nsISupportsCString).data;
+            var value = catMan.getCategoryEntry(CATEGORY_WAKEUP_REQUEST, entry);
+            var parts = value.split(",");
+            var cid = parts[0];
+            var iid = parts[1];
+            var method = parts[2];
+            var messages = parts.slice(3);
+            messages.forEach(function(messageName) {
+              this.requestWakeup(messageName, cid, iid, method);
+            }, this);
+          }
+        }
+        break;
+    }
+  },
+};
+
+var components = [MessageWakeupService];
+this.NSGetFactory = XPCOMUtils.generateNSGetFactory(components);
+