More edge cases covered in dynamic script injection.

1 files changed, 3 insertions, 4 deletions

diff --git a/src/content/media.js b/src/content/media.js
index b975bef..ead6e05 100644
--- a/src/content/media.js
+++ b/src/content/media.js
@@ -1,5 +1,5 @@
-debug("Media Hook (blocked %s)", !!window.mediaBlocker, document.URL, document.documentElement && document.documentElement.innerHTML);
-try {
+if (correctFrame()) {
+  debug("Media Hook (blocked %s)", !!window.mediaBlocker, document.URL, document.documentElement && document.documentElement.innerHTML);
   (() => {
     let unpatched = new Map();
     function patch(obj, methodName, replacement) {
@@ -56,6 +56,5 @@ try {
     });
 
   })();
-} catch (e) {
-  error(e, "Cannot patch MediaSource");
+  document.URL;
 }