Policy serialization using the contentScripts API.

1 files changed, 51 insertions, 52 deletions

diff --git a/src/content/media.js b/src/content/media.js
index ffa67e0..5da239f 100644
--- a/src/content/media.js
+++ b/src/content/media.js
@@ -1,60 +1,59 @@
-{
-  debug("Media Hook (blocked %s)", !!window.mediaBlocker, document.URL, document.documentElement && document.documentElement.innerHTML); // DEV_ONLY
-  (() => {
-    let unpatched = new Map();
-    function patch(obj, methodName, replacement) {
-       let methods = unpatched.get(obj) || {};
-       methods[methodName] = obj[methodName];
-       exportFunction(replacement, obj, {defineAs: methodName});
-       unpatched.set(obj, methods);
+ns.on("perms", ns => {
+  debug("Media Hook", document.URL, document.documentElement && document.documentElement.innerHTML, ns.perms.CURRENT); // DEV_ONLY
+  let mediaBlocker = !ns.allows("media");
+  let unpatched = new Map();
+  function patch(obj, methodName, replacement) {
+     let methods = unpatched.get(obj) || {};
+     methods[methodName] = obj[methodName];
+     exportFunction(replacement, obj, {defineAs: methodName});
+     unpatched.set(obj, methods);
+  }
+  let urlMap = new WeakMap();
+  patch(window.URL, "createObjectURL",  function(o, ...args) {
+    let url = unpatched.get(window.URL).createObjectURL.call(this, o, ...args);
+    if (o instanceof MediaSource) {
+      let urls = urlMap.get(o);
+      if (!urls) urlMap.set(o, urls = new Set());
+      urls.add(url);
     }
-    let urlMap = new WeakMap();
-    patch(window.URL, "createObjectURL",  function(o, ...args) {
-      let url = unpatched.get(window.URL).createObjectURL.call(this, o, ...args);
-      if (o instanceof MediaSource) {
-        let urls = urlMap.get(o);
-        if (!urls) urlMap.set(o, urls = new Set());
-        urls.add(url);
-      }
-      return url;
-    });
+    return url;
+  });
 
-    patch(window.MediaSource.prototype, "addSourceBuffer", function(mime, ...args) {
-      let ms = this;
-      let urls = urlMap.get(ms);
+  patch(window.MediaSource.prototype, "addSourceBuffer", function(mime, ...args) {
+    let ms = this;
+    let urls = urlMap.get(ms);
 
-      let request = {
-        id: "noscript-media",
-        type: "media",
-        url: document.URL,
-        documentUrl: document.URL,
-        embeddingDocument: true,
-      };
-      seen.record({policyType: "media", request, allowed: false});
-      notifyPage();
+    let request = {
+      id: "noscript-media",
+      type: "media",
+      url: document.URL,
+      documentUrl: document.URL,
+      embeddingDocument: true,
+    };
+    seen.record({policyType: "media", request, allowed: false});
+    notifyPage();
 
-      if (window.mediaBlocker) {
-        (async () => {
-          let me = Array.from(document.querySelectorAll("video,audio"))
-            .find(e => e.srcObject === ms || urls && urls.has(e.src));
+    if (mediaBlocker) {
+      (async () => {
+        let me = Array.from(document.querySelectorAll("video,audio"))
+          .find(e => e.srcObject === ms || urls && urls.has(e.src));
 
-          if (!me) return;
-          let exposedMime = `${mime} (MSE)`;
+        if (!me) return;
+        let exposedMime = `${mime} (MSE)`;
 
-          try {
-            let ph = PlaceHolder.create("media", request);
-            ph.replace(me);
-            PlaceHolder.listen();
-          } catch (e) {
-            error(e);
-          }
-        })();
-        throw new Error(`${exposedMime} blocked by NoScript`);
-      }
+        try {
+          let ph = PlaceHolder.create("media", request);
+          ph.replace(me);
+          PlaceHolder.listen();
+        } catch (e) {
+          error(e);
+        }
+      })();
+      throw new Error(`${exposedMime} blocked by NoScript`);
+    }
 
-      return unpatched.get(window.MediaSource.prototype).addSourceBuffer.call(ms, mime, ...args);
-    });
+    return unpatched.get(window.MediaSource.prototype).addSourceBuffer.call(ms, mime, ...args);
+  });
 
-  })();
-  document.URL;
-}
+})();
+document.URL;