Issue 3108 - Inject our about: module into all processes

lib/elemHide.js

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-2015 Eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
  */
 
 /**
  * @fileOverview Element hiding implementation.
  */
 
 Cu.import("resource://gre/modules/Services.jsm");
 
 let {Utils} = require("utils");
 let {IO} = require("io");
 let {Prefs} = require("prefs");
 let {ElemHideException} = require("filterClasses");
 let {FilterNotifier} = require("filterNotifier");
-let {AboutHandler} = require("elemHideHitRegistration");
 
 /**
  * Lookup table, filters by their associated key
  * @type Object
  */
 let filterByKey = Object.create(null);
 
 /**
  * Lookup table, keys of the filters by filter text
  * @type Object
@@ skipping 34 matching lines @@
    * Inidicates whether the element hiding stylesheet is currently applied.
    * @type Boolean
    */
   applied: false,
 
   /**
    * Called on module startup.
    */
   init: function()
   {
+    let messageManager = Cc["@mozilla.org/parentprocessmessagemanager;1"]
+                           .getService(Ci.nsIMessageListenerManager);
+    let hitHandler = (message => {
+      let result = ElemHide.shouldHide(message.data);
+      let target = message.target.QueryInterface(Ci.nsIMessageSender);
+      target.sendAsyncMessage(message.data.responseMessage, result);
+    });
+    messageManager.addMessageListener("AdblockPlus:ElemHideHit", hitHandler);
+    onShutdown.add(() => messageManager.removeMessageListener("AdblockPlus:ElemHideHit", hitHandler));
+
     Prefs.addListener(function(name)
     {
       if (name == "enabled")
         ElemHide.apply();
     });
-    onShutdown.add(function()
-    {
-      ElemHide.unapply();
-    });
+    onShutdown.add(() => ElemHide.unapply());
 
     let styleFile = IO.resolveFilePath(Prefs.data_directory);
     styleFile.append("elemhide.css");
     styleURL = Services.io.newFileURI(styleFile).QueryInterface(Ci.nsIFileURL);
   },
 
   /**
    * Removes all known filters
    */
   clear: function()
@@ skipping 79 matching lines @@
 
     let list = exceptions[filter.selector];
     for (let i = list.length - 1; i >= 0; i--)
       if (list[i].isActiveOnDomain(docDomain))
         return list[i];
 
     return null;
   },
 
   /**
+   * Processes an element hiding hit
+   * @param {String} message.key
+   *    key of the matching element hiding rule
+   * @param {Object[]} message.frames
+   *    information required to reconstruct frame
+   *    data for the hit
+   * @return {Boolean}
+   */
+  shouldHide: function(message)
+  {
+    let filter = ElemHide.getFilterByKey(message.key);
+    if (!filter || !message.frames.length)
+      return false;
+
+    let fakeFrame = null;
+    for (let i = message.frames.length - 1; i >= 0; i--)
+    {
+      fakeFrame = {
+        parent: fakeFrame,
+        location: {
+          href: message.frames[i].location
+        },
+        document: {
+          documentElement: {}
+        },
+        QueryInterface: function() {return this;},
+        getInterface: function() {return this;},
+        usePrivateBrowsing: (fakeFrame ? fakeFrame.usePrivateBrowsing : message.frames[i].privateBrowsing)
+      };
+      fakeFrame.top = fakeFrame.parent || fakeFrame;
+      if (!fakeFrame.parent)
+        fakeFrame.parent = fakeFrame;
+
+      let sitekey = message.frames[i].sitekey || null;
+      fakeFrame.document.documentElement.getAttribute = function(attr)
+      {
+        if (attr == "data-adblockkey")
+          return sitekey;
+        else
+          return null;
+      };
+    }
+
+    let {Policy} = require("contentPolicy");
+    return !Policy.processNode(fakeFrame, fakeFrame.document, Policy.type.ELEMHIDE, filter);

[Wladimir Palant, 2015/10/16 12:15:10]

Side-effect here: the hit info is attached to a fake window, meaning that it
won't appear in the blockable items list. Not really feasible to fix until we
start injecting the content policy into all processes as well (at this point we
will also be able to stop faking DOM structures here).

+  },
+
+  /**
    * Will be set to true if apply() is running (reentrance protection).
    * @type Boolean
    */
   _applying: false,
 
   /**
    * Will be set to true if an apply() call arrives while apply() is already
    * running (delayed execution).
    * @type Boolean
    */
@@ skipping 104 matching lines @@
 
     if (!hasFilters)
       throw Cr.NS_ERROR_NOT_AVAILABLE;
 
     function escapeChar(match)
     {
       return "\\" + match.charCodeAt(0).toString(16) + " ";
     }
 
     // Return CSS data
-    let cssTemplate = "-moz-binding: url(about:" + AboutHandler.aboutPrefix + "?%ID%#dummy) !important;";
+    let cssTemplate = "-moz-binding: url(about:abp-elemhidehit?%ID%#dummy) !important;";
     for (let domain in domains)
     {
       let rules = [];
       let list = domains[domain];
 
       if (domain)
         yield ('@-moz-document domain("' + domain.split(",").join('"),domain("') + '"){').replace(/[^\x01-\x7F]/g, escapeChar);
       else
       {
         // Only allow unqualified rules on a few protocols to prevent them from blocking chrome
@@ skipping 57 matching lines @@
       let filter = filterByKey[key];
       if (specificOnly && (!filter.domains || filter.domains[""]))
         continue;
 
       if (filter.isActiveOnDomain(domain) && !this.getException(filter, domain))
         result.push(filter.selector);
     }
     return result;
   }
 };