#203 websockets

AdguardTeam · May 5, 2016 · f78be7e · f78be7e
1 parent 8b11814
commit f78be7e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Extension/lib/content-script/websocket.js b/Extension/lib/content-script/websocket.js
@@ -27,6 +27,9 @@
  */
 var overrideWebSocket = function () {
 
+    var postMessage = window.postMessage;
+    var addEventListener = window.addEventListener;
+
     'use strict';
     var Wrapped = window.WebSocket;
     var map = new WeakMap();
@@ -233,10 +236,10 @@ var overrideWebSocket = function () {
             initConnection(that, event.data.collapse);
         }
 
-        window.addEventListener("message", messageListener, false);
+        addEventListener.call(window, "message", messageListener, false);
 
         // Send a message to the background page to check if the request should be blocked
-        window.postMessage({
+        postMessage.call(window, {
             direction: 'from-page-script@adguard',
             elementUrl: url,
             documentUrl: document.URL