Added support for websocket

uBlock-LLC · Sep 26, 2018 · 4e75e4c · 4e75e4c
1 parent 46a249e
commit 4e75e4c
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 8 deletions.
diff --git a/platform/chromium/manifest.json b/platform/chromium/manifest.json
@@ -57,8 +57,7 @@
         "webNavigation",
         "webRequest",
         "webRequestBlocking",
-        "http://*/*",
-        "https://*/*"
+        "<all_urls>"
     ],
     "short_name": "uBlock"
 }
diff --git a/platform/chromium/vapi-background.js b/platform/chromium/vapi-background.js
@@ -710,6 +710,23 @@ vAPI.net.registerListeners = function() {
         normalizeRequestDetails(details);
         return onBeforeRequestClient(details);
     };
+
+    var urls, types;
+    urls = this.onBeforeRequest.urls || ['<all_urls>'];
+    types = this.onBeforeRequest.types ||  [];
+    if (
+        (chrome.webRequest.ResourceType.WEBSOCKET == 'websocket') &&
+        (urls.indexOf('<all_urls>') === -1)
+    ) {
+        types.push('websocket');
+        if ( urls.indexOf('ws://*/*') === -1 ) {
+            urls.push('ws://*/*');
+        }
+        if ( urls.indexOf('wss://*/*') === -1 ) {
+            urls.push('wss://*/*');
+        }
+    }
+
     chrome.webRequest.onBeforeRequest.addListener(
         onBeforeRequest,
         //function(details) {

diff --git a/platform/webext/manifest.json b/platform/webext/manifest.json
@@ -63,8 +63,7 @@
         "webNavigation",
         "webRequest",
         "webRequestBlocking",
-        "http://*/*",
-        "https://*/*"
+        "<all_urls>"
     ],
     "short_name": "uBlock"
 }
diff --git a/src/js/background.js b/src/js/background.js
@@ -91,8 +91,8 @@ return {
 
     // read-only
     systemSettings: {
-        compiledMagic: 'eopszukpnrcq',
-        selfieMagic: 'menhiasrxfed'
+        compiledMagic: 'eopszukpnrcj',
+        selfieMagic: 'menhiasrxfee'
     },
     restoreBackupSettings: {
         lastRestoreFile: '',

diff --git a/src/js/static-net-filtering.js b/src/js/static-net-filtering.js
@@ -62,7 +62,8 @@ var typeNameToTypeValue = {
             'script':  4 << 4,
     'xmlhttprequest':  5 << 4,
          'sub_frame':  6 << 4,
-             'other':  7 << 4,
+         'websocket':  9 << 4,
+             'other':  10 << 4,
 'cosmetic-filtering': 13 << 4,
      'inline-script': 14 << 4,
              'popup': 15 << 4,
@@ -1318,7 +1319,8 @@ FilterParser.prototype.toNormalizedType = {
           'elemhide': 'cosmetic-filtering',
      'inline-script': 'inline-script',
              'popup': 'popup',
-              'csp' : 'csp' 
+              'csp' : 'csp', 
+         'websocket': 'websocket'
 };
 
 /******************************************************************************/