Issue 4599 - Update the page structure more eagerly

chrome/ext/background.js

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-2016 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 105 matching lines @@
     if (!frames)
       frames = framesOfTabs[tabId] = Object.create(null);
 
     var frame = frames[frameId];
     if (!frame)
       frame = frames[frameId] = {};
 
     return frame;
   }
 
-  chrome.webNavigation.onBeforeNavigate.addListener(function(details)
-  {
-    // Capture parent frame here because onCommitted doesn't get this info.
-    var frame = createFrame(details.tabId, details.frameId);
-    frame.parent = framesOfTabs[details.tabId][details.parentFrameId] || null;
-  });
-
   var eagerlyUpdatedPages = new ext.PageMap();
 
   ext._updatePageFrameStructure = function(frameId, tabId, url, eager)
   {
     if (frameId == 0)
     {
       let page = new Page({id: tabId, url: url});
 
       if (eagerlyUpdatedPages.get(page) != url)
       {
@@ skipping 18 matching lines @@
             ext.pages.onLoading._dispatch(page);
         });
       }
     }
 
     // Update frame URL in frame structure
     var frame = createFrame(tabId, frameId);
     frame.url = new URL(url);
   };
 
+  chrome.webNavigation.onBeforeNavigate.addListener(function(details)
+  {
+    // Capture parent frame here because onCommitted doesn't get this info.
+    var frame = createFrame(details.tabId, details.frameId);
+    frame.parent = framesOfTabs[details.tabId][details.parentFrameId] || null;
+
+    // Chrome will sometimes perform requests for a pre-rendered page before
+    // navigation was confirmed and the onComitted event was fired. Those
+    // requests will often wrongly be considered as third party, since their
+    // page structure was not recorded. As a work around we update the page
+    // structure eagerly for new tabs (which includes pre-rendered pages).
+    if (!(details.tabId in framesOfTabs))
+      ext._updatePageFrameStructure(details.frameId, details.tabId, details.url);
+  });
+
   chrome.webNavigation.onCommitted.addListener(function(details)
   {
     ext._updatePageFrameStructure(details.frameId, details.tabId, details.url);
   });
 
   function forgetTab(tabId)
   {
     ext.pages.onRemoved._dispatch(tabId);
 
     ext._removeFromAllPageMaps(tabId);
@@ skipping 458 matching lines @@
   ext.windows = {
     create: function(createData, callback)
     {
       chrome.windows.create(createData, function(createdWindow)
       {
         afterTabLoaded(callback)(createdWindow.tabs[0]);
       });
     }
   };
 })();