1 files changed, 59 insertions, 0 deletions
diff --git a/src/content/media.js b/src/content/media.js
new file mode 100644
index 0000000..22bf014
--- /dev/null
+++ b/src/content/media.js
@@ -0,0 +1,59 @@
+console.log("Media Hook", document.documentElement.innerHTML);
+try {
+  (() => {
+    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);
+    }
+    patch(window.console, "log", function(s, ...args) {
+      unpatched.get(window.console).log.call(`PATCHED ${s}`, ...args);
+    });
+    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;
+    });
+
+    patch(window.MediaSource.prototype, "addSourceBuffer", function(mime, ...args) {
+      let ms = this;
+      let urls = urlMap.get(ms);
+      let me = Array.from(document.querySelectorAll("video,audio"))
+        .find(e => e.srcObject === ms || urls && urls.has(e.src));
+      let exposedMime = `${mime} (MSE)`;
+
+      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) {
+        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);
+    });
+
+  })();
+} catch (e) {
+  error(e, "Cannot patch MediaSource");
+}