Issue 521- Inject our stylesheet on per-site basis rather than globally

lib/contentPolicy.js

 /*
  * This file is part of Adblock Plus <http://adblockplus.org/>,
  * Copyright (C) 2006-2014 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
@@ skipping 12 matching lines @@
 Cu.import("resource://gre/modules/Services.jsm");
 
 let {TimeLine} = require("timeline");
 let {Utils} = require("utils");
 let {Prefs} = require("prefs");
 let {FilterStorage} = require("filterStorage");
 let {BlockingFilter, WhitelistFilter} = require("filterClasses");
 let {defaultMatcher} = require("matcher");
 let {objectMouseEventHander} = require("objectTabs");
 let {RequestNotifier} = require("requestNotifier");
+let {ElemHide} = require("elemHide");
 
 /**
  * List of explicitly supported content types
  * @type Array of String
  */
 let contentTypes = ["OTHER", "SCRIPT", "IMAGE", "STYLESHEET", "OBJECT", "SUBDOCUMENT", "DOCUMENT", "XMLHTTPREQUEST", "OBJECT_SUBREQUEST", "FONT", "MEDIA"];
 
 /**
  * List of content types that aren't associated with a visual document area
  * @type Array of String
@@ skipping 92 matching lines @@
     Utils.styleService.loadAndRegisterSheet(collapseStyle, Ci.nsIStyleSheetService.USER_SHEET);
     onShutdown.add(function()
     {
       Utils.styleService.unregisterSheet(collapseStyle, Ci.nsIStyleSheetService.USER_SHEET);
     })
     TimeLine.log("done registering stylesheet");
 
     TimeLine.leave("Done initializing content policy");
   },
 
-  processWindow: function(wnd, contentType)
+  /**
+   * Checks whether a window or its parent is whitelisted or has a valid site-key.
+   * @param wnd {nsIDOMWindow}
+   */
+  shouldNeverBlockWindow: function(wnd)
   {
     let uri = wnd.document.documentURIObject;
     if (!uri)
       return true;
 
     if (!Policy.isBlockableScheme(uri.scheme))
       return true;
 
     let topWnd = wnd.top;
     if (!topWnd || !topWnd.location || !topWnd.location.href)
       return true;
 
     let testWnd = wnd;
     let parentWndLocation = getWindowLocation(testWnd);
     while (true)
     {
       let testWndLocation = parentWndLocation;
       parentWndLocation = (testWnd == testWnd.parent ? testWndLocation : getWindowLocation(testWnd.parent));
 
       let match = Policy.isWhitelisted(testWndLocation, parentWndLocation);
 
-      if (!(match instanceof WhitelistFilter))

[tschuster, 2014/06/04 18:07:40]

I think we should make this match === null, because isWhitelisted can only
return an instance of WhitelistFilter  or null

+      if (match == null)
       {
         let keydata = (testWnd.document && testWnd.document.documentElement ? testWnd.document.documentElement.getAttribute("data-adblockkey") : null);
         if (keydata && keydata.indexOf("_") >= 0)
         {
           let [key, signature] = keydata.split("_", 2);
           let keyMatch = defaultMatcher.matchesByKey(testWndLocation, key.replace(/=/g, ""), docDomain);
           if (keyMatch && Utils.crypto)
           {
             // Website specifies a key that we know but is the signature valid?
             let uri = Services.io.newURI(testWndLocation, null, null);
             let params = [
               uri.path.replace(/#.*/, ""),  // REQUEST_URI
               uri.asciiHost,                // HTTP_HOST
               Utils.httpProtocol.userAgent  // HTTP_USER_AGENT
             ];
             if (Utils.verifySignature(key, signature, params.join("\0")))
-              match = keyMatch; // XXX what is this?

[tschuster, 2014/06/04 18:07:40]

Okay I am not sure about this case, is that |keyMatch| a whitelistfilter?
Otherwise we probably have to return it.

+              match = keyMatch;
           }

[tschuster, 2014/06/05 19:00:34]

I have an other concern: This is never going to work when we are in
"content-document-global-created" because the DOM is not ready yet. Is this code
relevant to elemhide filters?

         }
       }
 
       if (match instanceof WhitelistFilter)
       {
         FilterStorage.increaseHitCount(match, wnd);
         RequestNotifier.addNodeData(testWnd.document, topWnd, Policy.type.DOCUMENT, getHostname(parentWndLocation), false, testWndLocation, match);
         return true;
-      }
-
-      if (contentType == Policy.type.ELEMHIDE)
-      {
-        let parentDocDomain = getHostname(parentWndLocation);
-        match = defaultMatcher.matchesAny(testWndLocation, "ELEMHIDE", parentDocDomain, false);
-        if (match instanceof WhitelistFilter)
-        {
-          FilterStorage.increaseHitCount(match, wnd);
-          RequestNotifier.addNodeData(testWnd.document, topWnd, contentType, parentDocDomain, false, testWndLocation, match);
-          return true;
-        }
       }
 
       if (testWnd.parent == testWnd)
         break;
       else
         testWnd = testWnd.parent;
     }
 
     return false;
   },
 
   /**
    * Checks whether a node should be blocked, hides it if necessary
    * @param wnd {nsIDOMWindow}
    * @param node {nsIDOMElement}
    * @param contentType {String}
    * @param location {nsIURI}
    * @param collapse {Boolean} true to force hiding of the node
    * @return {Boolean} false if the node should be blocked
    */
   processNode: function(wnd, node, contentType, location, collapse)
   {
-    if (contentType == Policy.type.ELEMHIDE)
-      throw new Error("unexpected!");
+    if (Policy.shouldNeverBlockWindow(wnd))
+      return true;
+
+    let topWnd = wnd.top;
+    let originWindow = Utils.getOriginWindow(wnd);
+    let wndLocation = originWindow.location.href;
+    let docDomain = getHostname(wndLocation);
 
     // Data loaded by plugins should be attached to the document
     if (contentType == Policy.type.OBJECT_SUBREQUEST && node instanceof Ci.nsIDOMElement)
       node = node.ownerDocument;
 
     // Fix type for objects misrepresented as frames or images
     if (contentType != Policy.type.OBJECT && (node instanceof Ci.nsIDOMHTMLObjectElement || node instanceof Ci.nsIDOMHTMLEmbedElement))
       contentType = Policy.type.OBJECT;
 
-    if (Policy.processWindow(wnd, contentType))
-      return true;
-
-    let originWindow = Utils.getOriginWindow(wnd);
-    let wndLocation = originWindow.location.href;
-    let docDomain = getHostname(wndLocation);
-    let topWnd = wnd.top;
-
     let locationText = location.spec;
-    let thirdParty = isThirdParty(location, docDomain);
-
-    let match = defaultMatcher.matchesAny(locationText, Policy.typeDescr[contentType] || "", docDomain, thirdParty);
-    if (match instanceof BlockingFilter && node.ownerDocument && !(contentType in Policy.nonVisual))
-    {
-      let prefCollapse = (match.collapse != null ? match.collapse : !Prefs.fastcollapse);
-      if (collapse || prefCollapse)
-        schedulePostProcess(node);
-    }
-
-    // Track mouse events for objects
-    if (!match && contentType == Policy.type.OBJECT && node.nodeType == Ci.nsIDOMNode.ELEMENT_NODE)
-    {
-      node.addEventListener("mouseover", objectMouseEventHander, true);
-      node.addEventListener("mouseout", objectMouseEventHander, true);
+    let match = null;
+    if (contentType == Policy.type.ELEMHIDE)
+    {
+      let testWnd = wnd;
+      let parentWndLocation = getWindowLocation(testWnd);
+      while (true)
+      {
+        let testWndLocation = parentWndLocation;
+        parentWndLocation = (testWnd == testWnd.parent ? testWndLocation : getWindowLocation(testWnd.parent));
+        let parentDocDomain = getHostname(parentWndLocation);
+        match = defaultMatcher.matchesAny(testWndLocation, "ELEMHIDE", parentDocDomain, false);
+        if (match instanceof WhitelistFilter)
+        {
+          FilterStorage.increaseHitCount(match, wnd);
+          RequestNotifier.addNodeData(testWnd.document, topWnd, contentType, parentDocDomain, false, testWndLocation, match);
+          return true;
+        }
+
+        if (testWnd.parent == testWnd)
+          break;
+        else
+          testWnd = testWnd.parent;
+      }
+
+      match = location;
+      locationText = match.text.replace(/^.*?#/, '#');
+      location = locationText;
+
+      if (!match.isActiveOnDomain(docDomain))
+        return true;
+
+      let exception = ElemHide.getException(match, docDomain);
+      if (exception)
+      {
+        FilterStorage.increaseHitCount(exception, wnd);
+        RequestNotifier.addNodeData(node, topWnd, contentType, docDomain, thirdParty, locationText, exception);
+        return true;
+      }
+    }
+
+    let thirdParty = (contentType == Policy.type.ELEMHIDE ? false : isThirdParty(location, docDomain));
+
+    if (!match && Prefs.enabled)
+    {
+      match = defaultMatcher.matchesAny(locationText, Policy.typeDescr[contentType] || "", docDomain, thirdParty);
+      if (match instanceof BlockingFilter && node.ownerDocument && !(contentType in Policy.nonVisual))
+      {
+        let prefCollapse = (match.collapse != null ? match.collapse : !Prefs.fastcollapse);
+        if (collapse || prefCollapse)
+          schedulePostProcess(node);
+      }
+
+      // Track mouse events for objects
+      if (!match && contentType == Policy.type.OBJECT && node.nodeType == Ci.nsIDOMNode.ELEMENT_NODE)
+      {
+        node.addEventListener("mouseover", objectMouseEventHander, true);
+        node.addEventListener("mouseout", objectMouseEventHander, true);
+      }
     }
 
     // Store node data
     RequestNotifier.addNodeData(node, topWnd, contentType, docDomain, thirdParty, locationText, match);
     if (match)
       FilterStorage.increaseHitCount(match, wnd);
 
     return !match || match instanceof WhitelistFilter;
   },
 
@@ skipping 478 matching lines @@
   if (!wnd || wnd.closed)
     return;
 
   if (entry.type == Policy.type.OBJECT)
   {
     node.removeEventListener("mouseover", objectMouseEventHander, true);
     node.removeEventListener("mouseout", objectMouseEventHander, true);
   }
   Policy.processNode(wnd, node, entry.type, Utils.makeURI(entry.location), true);
 }