From 0ad5f95eb479f8450058333541631640fc8ee4bb Mon Sep 17 00:00:00 2001
From: hackademix
Date: Sat, 7 Jul 2018 01:28:20 +0200
Subject: Fixed dynamic script injection breaking images shown as frame
 content.

---
 src/bg/RequestGuard.js | 23 +++++++++++------------
 1 file changed, 11 insertions(+), 12 deletions(-)

(limited to 'src/bg/RequestGuard.js')

diff --git a/src/bg/RequestGuard.js b/src/bg/RequestGuard.js
index b23853f..dcb79a7 100644
--- a/src/bg/RequestGuard.js
+++ b/src/bg/RequestGuard.js
@@ -90,9 +90,6 @@ var RequestGuard = (() => {
 
       if (what === "noscriptFrame") {
         let nsf = records.noscriptFrames;
-        if (frameId in nsf) {
-          return null;
-        }
         nsf[frameId] = optValue;
         what = optValue ? "blocked" : "allowed";
         if (frameId === 0) {
@@ -278,18 +275,20 @@ var RequestGuard = (() => {
       };
       if (tabId < 0) return;
       if (pending) request.initialUrl = pending.initialUrl;
-      try {
-        browser.tabs.sendMessage(
-          tabId,
-          {type: "seen", request, allowed, policyType, ownFrame: true},
-          {frameId}
-        );
-      } catch (e) {
-        debug(`Couldn't deliver "seen" message for ${type}@${url} ${allowed ? "A" : "F" } to document ${documentUrl} (${frameId}/${tabId}`, e);
+      if (type !== "sub_frame") { // we couldn't deliver it to frameId, since it's generally not loaded yet
+        try {
+          await browser.tabs.sendMessage(
+            tabId,
+            {type: "seen", request, allowed, policyType, ownFrame: true},
+            {frameId}
+          );
+        } catch (e) {
+          debug(`Couldn't deliver "seen" message for ${type}@${url} ${allowed ? "A" : "F" } to document ${documentUrl} (${frameId}/${tabId})`, e);
+        }
       }
       if (frameId === 0) return;
       try {
-        browser.tabs.sendMessage(
+        await browser.tabs.sendMessage(
           tabId,
           {type: "seen", request, allowed, policyType},
           {frameId: 0}
-- 
cgit v1.2.3