More reliable attempt to run onResponseHeader listener the last of installed extension.

2 files changed, 20 insertions, 10 deletions

diff --git a/src/bg/RequestGuard.js b/src/bg/RequestGuard.js
index 703623f..38801de 100644
--- a/src/bg/RequestGuard.js
+++ b/src/bg/RequestGuard.js
@@ -462,19 +462,33 @@ var RequestGuard = (() => {
 
     onResponseStarted(request) {
       debug("onResponseStarted", request);
-      if (request.type === "main_frame") {
-        TabStatus.initTab(request.tabId);
+      let {url, tabId, frameId, type} = request;
+      if (type === "main_frame") {
+        TabStatus.initTab(tabId);
       }
       let scriptBlocked = request.responseHeaders.some(
         h => CSP.isMine(h) && CSP.blocks(h.value, "script")
       );
-      debug("%s scriptBlocked=%s setting noscriptFrame on ", request.url, scriptBlocked, request.tabId, request.frameId);
+      debug("%s scriptBlocked=%s setting noscriptFrame on ", url, scriptBlocked, tabId, frameId);
       TabStatus.record(request, "noscriptFrame", scriptBlocked);
       let pending = pendingRequests.get(request.requestId);
       if (pending) {
         pending.scriptBlocked = scriptBlocked;
-        if (!pending.headersProcessed) {
-          debug("[WARNING] onHeadersReceived could not process", request);
+        if (!(pending.headersProcessed && 
+            (scriptBlocked || ns.policy.can(url, "script", request.documentURL))
+          )) {
+          debug("[WARNING] onHeadersReceived %s %o", frameId, tabId,
+            pending.headersProcessed ? "has been overridden on": "could not process", 
+            request);
+
+          if (tabId !== -1) {
+            debug("[WARNING] Reloading %s frame %s of tab %s.", url, frameId, tabId);
+            browser.tabs.executeScript(tabId, {
+              runAt: "document_start",
+              code: "window.location.reload(false)",
+              frameId
+            });
+          }
         }
       }
     },
diff --git a/src/bg/deferWebTraffic.js b/src/bg/deferWebTraffic.js
index 15d177b..3f27dd5 100644
--- a/src/bg/deferWebTraffic.js
+++ b/src/bg/deferWebTraffic.js
@@ -34,11 +34,7 @@ function deferWebTraffic(promiseToWaitFor, next) {
         if (frameId !== 0) {
           documentUrl = request.frameAncestors.pop().url;
         }
-        if (tabId !== -1) {
-          reloadTab(tabId);
-        } else {
-          debug("No tab to reload for %s %s from %s", type, url, documentUrl);
-        }
+        reloadTab(tabId);
       }
     }
     debug("Deferring %s %s from %s", type, url, documentUrl);