More edge cases covered in dynamic script injection.

1 files changed, 3 insertions, 6 deletions

diff --git a/src/content/webglHook.js b/src/content/webglHook.js
index c3b7aaa..39637e2 100644
--- a/src/content/webglHook.js
+++ b/src/content/webglHook.js
@@ -1,5 +1,5 @@
-debug("WebGL Hook", document.URL, document.documentElement && document.documentElement.innerHTML);
-try {
+if (correctFrame()) {
+  debug("WebGL Hook", document.URL, document.documentElement && document.documentElement.innerHTML);
   let proto = HTMLCanvasElement.prototype;
   let getContext = proto.getContext;
   exportFunction(function(type, ...rest) {
@@ -24,8 +24,5 @@ try {
     }
     return getContext.call(this, type, ...rest);
   }, proto, {defineAs: "getContext"});
-} catch (e) {
-  console.error(e);
+  document.URL;
 }
-
-null;