Issue 3259 - Store blockable items selection in the content process

lib/child/requestNotifier.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 Stores Adblock Plus data to be attached to a window.
  */
 let {Services} = Cu.import("resource://gre/modules/Services.jsm", {});
 
 let {Utils} = require("utils");
 let {Flasher} = require("child/flasher");
 
 let nodeData = new WeakMap();
 let windowStats = new WeakMap();
+let windowData = new WeakMap();
 let requestEntryMaxId = 0;
 
 /**
  * Active RequestNotifier instances by their ID
  * @type Map.<number,RequestNotifier>
  */
 let notifiers = new Map();
 
 addMessageListener("AdblockPlus:StartWindowScan", onStartScan);
 addMessageListener("AdblockPlus:ShutdownNotifier", onNotifierShutdown);
 addMessageListener("AdblockPlus:FlashNodes", onFlashNodes);
 addMessageListener("AdblockPlus:RetrieveNodeSize", onRetrieveNodeSize);
 addMessageListener("AdblockPlus:StoreNodesForEntries", onStoreNodes);
 addMessageListener("AdblockPlus:RetrieveWindowStats", onRetrieveWindowStats);
+addMessageListener("AdblockPlus:StoreWindowData", onStoreWindowData);
+addMessageListener("AdblockPlus:RetrieveWindowData", onRetrieveWindowData);
 
 onShutdown.add(() => {
   removeMessageListener("AdblockPlus:StartWindowScan", onStartScan);
   removeMessageListener("AdblockPlus:ShutdownNotifier", onNotifierShutdown);
   removeMessageListener("AdblockPlus:FlashNodes", onFlashNodes);
   removeMessageListener("AdblockPlus:RetrieveNodeSize", onRetrieveNodeSize);
   removeMessageListener("AdblockPlus:StoreNodesForEntries", onStoreNodes);
   removeMessageListener("AdblockPlus:RetrieveWindowStats", onRetrieveWindowStats);
+  removeMessageListener("AdblockPlus:StoreWindowData", onStoreWindowData);
+  removeMessageListener("AdblockPlus:RetrieveWindowData", onRetrieveWindowData);
 });
 
 function onStartScan(message)
 {
   let {notifierID, outerWindowID} = message.data;
   let window = Services.wm.getOuterWindowWithId(outerWindowID);
   if (window)
     new RequestNotifier(window, notifierID);
 }
 
@@ skipping 35 matching lines @@
   if (window)
   {
     let stats = RequestNotifier.getWindowStatistics(window);
     sendAsyncMessage("AdblockPlus:RetrieveWindowStatsResponse", {
       responseID,
       stats
     });
   }
 }
 
+function onStoreWindowData(message)
+{
+  let {outerWindowID, data} = message.data;
+  let window = Services.wm.getOuterWindowWithId(outerWindowID);
+  if (window)
+    windowData.set(window.document, data);
+};
+
+function onRetrieveWindowData(message)
+{
+  let {responseID, outerWindowID} = message.data;
+  let window = Services.wm.getOuterWindowWithId(outerWindowID);
+  if (window)
+  {
+    let data = windowData.get(window.document) || null;
+    sendAsyncMessage("AdblockPlus:RetrieveWindowDataResponse", {
+      responseID,
+      data
+    });
+  }
+};
+
 /**
  * Creates a notifier object for a particular window. After creation the window
  * will first be scanned for previously saved requests. Once that scan is
  * complete only new requests for this window will be reported.
  * @param {Window} window  window to attach the notifier to
  * @param {Integer} notifierID  Parent notifier ID to be messaged
  */
 function RequestNotifier(window, notifierID)
 {
   this.window = window;
@@ skipping 332 matching lines @@
       node = node.parentNode;
     }
     else
     {
       node = null;
     }
   }
 
   return null;
 };