Added abstraction for frames, to fix domain-based rules, whitelisting and ad counter on Safari

safari/ext/content.js

 /*
  * This file is part of Adblock Plus <http://adblockplus.org/>,
  * Copyright (C) 2006-2013 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/>.
  */
 
 (function()
 {
   if (window == window.top)
     safari.self.tab.dispatchMessage("loading");
 
 
+  /* Events */
+
+  var ContentMessageEventTarget = function()
+  {
+    MessageEventTarget.call(this, safari.self);
+  };
+  ContentMessageEventTarget.prototype = {
+    __proto__: MessageEventTarget.prototype,
+    _getResponseDispatcher: function(event)
+    {
+      return event.target.tab;
+    },
+    _getSenderDetails: function(event)
+    {
+      return {};
+    }
+  };
+
+
   /* Background page proxy */
   var proxy = {
     objects: [],
     callbacks: [],
 
     send: function(message)
     {
       var evt = document.createEvent("Event");
       evt.initEvent("beforeload");
       return safari.self.tab.canLoad(evt, {type: "proxy", payload: message});
@@ skipping 249 matching lines @@
       case "link":
         if (/\bstylesheet\b/i.test(event.target.rel))
         {
           type = "stylesheet";
           break;
         }
       default:
         type = "other";
     }
 
-    if (!safari.self.tab.canLoad(event, {type: "webRequest", payload: {url: event.url, type: type}}))
+    if (!safari.self.tab.canLoad(
+      event, {
+        type: "webRequest",
+        payload: {
+          url: event.url,
+          type: type,
+          documentUrl: document.location.href,
+          isTopLevel: window == window.top
+        }
+      }
+    ))
     {
       event.preventDefault();
 
       // Safari doesn't dispatch an "error" or "load" event when preventing an
       // element from loading by cancelling the "beforeload" event. So we have
       // to dispatch it manually. Otherwise element collapsing wouldn't work.
       var evt = document.createEvent("Event");
       evt.initEvent(type == "sub_frame" ? "load" : "error");
       event.target.dispatchEvent(evt);
     }
   }, true);
 
 
   /* API */
 
   ext.backgroundPage = {
-    _eventTarget: safari.self,
-    _messageDispatcher: safari.self.tab,
-
-    sendMessage: sendMessage,
-    getWindow: function() { return proxy.getObject(0); }
+    sendMessage: function(message, responseCallback)
+    {
+      _sendMessage(
+        message, responseCallback,
+        safari.self.tab, safari.self,
+        {
+          documentUrl: document.location.href,
+          isTopLevel: window == window.top
+        }
+      );
+    },
+    getWindow: function()
+    {
+      return proxy.getObject(0);
+    }
   };
 
-  ext.onMessage = new MessageEventTarget(safari.self);
+  ext.onMessage = new ContentMessageEventTarget();
+
 
   // Safari does not pass the element which the context menu is refering to
   // so we need to add it to the event's user info.
   document.addEventListener("contextmenu", function(event)
   {
     var element = event.srcElement;
     safari.self.tab.setContextMenuEventUserInfo(event, {
       srcUrl: ("src" in element) ? element.src : null,
       tagName: element.tagName
     });
   }, false);
 })();