From e0ae64871e317d356c0fbd48a0e4c911d1089708 Mon Sep 17 00:00:00 2001
From: hackademix
Date: Tue, 3 Jul 2018 17:02:58 +0200
Subject: Fixed regression: dynamic script injection breaking images loaded as
 frame content.

---
 src/bg/RequestUtil.js | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

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

diff --git a/src/bg/RequestUtil.js b/src/bg/RequestUtil.js
index 2020286..a62f139 100644
--- a/src/bg/RequestUtil.js
+++ b/src/bg/RequestUtil.js
@@ -11,6 +11,19 @@
   browser.webRequest.onCompleted.addListener(cleanup, filter);
   browser.webRequest.onErrorOccurred.addListener(cleanup, filter);
   var RequestUtil = {
+
+    getContentMetaData(request) {
+      if (request.content) return request.content;
+      let {responseHeaders} = request;
+      let content = request.content = {};
+      for (let h of responseHeaders) {
+        if (/^\s*Content-(Type|Disposition)\s*$/i.test(h.name)) {
+          content[h.name.split("-")[1].trim().toLowerCase()] = h.value;
+        }
+      }
+      return content;
+    },
+
     async executeOnStart(request, details) {
       let {requestId, tabId, frameId} = request;
       let scripts = pendingRequests.get(requestId);
@@ -25,8 +38,11 @@
 
       let filter = browser.webRequest.filterResponseData(requestId);
       let buffer = [];
+      let content = this.getContentMetaData(request);
       filter.onstart = async event => {
-        filter.write(new Uint8Array());
+        if (/ml$/i.test(content.type)) {
+          filter.write(new Uint8Array()); // work-around for https://bugzilla.mozilla.org/show_bug.cgi?id=1410755
+        }
         for (let details of scripts.values()) {
           details = Object.assign({
             runAt: "document_start",
-- 
cgit v1.2.3