Issue 6791 - Implement abort-current-inline-script snippet

lib/content/snippets.js

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-present eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
  */
 
 /* eslint-env webextensions */
 /* eslint no-console: "off" */
 
 "use strict";
 
+function regexEscape(s)
+{
+  return s.replace(/[-/\\^$*+?.()|[\]{}]/g, "\\$&");
+}
+
+function findProperty(path, root = window)
+{
+  let owner = root;
+
+  let chain = path.split(".");
+  let property = chain.pop();
+
+  for (let name of chain)
+  {
+    if (!owner)
+      break;
+
+    owner = owner[name];
+  }
+
+  if (!owner)
+    return null;
+
+  return {owner, property};
+}
+
+function injectPropertyAccessValidator(object, property, validate)
+{
+  let value = object[property];
+
+  Object.defineProperty(object, property, {
+    get()
+    {
+      validate();
+      return value;
+    },
+    set(newValue)
+    {
+      validate();
+      value = newValue;
+    }
+  });
+}
+
 function injectCode(code)
 {
   let script = document.createElement("script");
 
   script.type = "application/javascript";
   script.async = false;
 
   // Firefox 58 only bypasses site CSPs when assigning to 'src',
   // while Chrome 67 only bypasses site CSPs when using 'textContent'.
   if (browser.runtime.getURL("").startsWith("chrome-extension://"))
@@ skipping 15 matching lines @@
 function stringifyFunctionCall(func, ...params)
 {
   return `"use strict";(${func})(${params.map(JSON.stringify).join(",")});`;
 }
 
 function makeInjector(injectable)
 {
   return (...args) => injectCode(stringifyFunctionCall(injectable, ...args));
 }
 
+function getMagic()
+{
+  return String.fromCharCode(Date.now() % 26 + 97) +
+         Math.floor(Math.random() * 982451653 + 982451653).toString(36);
+}
+
+function suppressMagicError(magic)
+{
+  let {onerror} = window;
+  window.onerror = function(message, ...rest)
+  {
+    if (typeof message == "string" && message.includes(magic))
+      return true;
+
+    if (typeof onerror instanceof Function)
+      return onerror.call(this, message, ...rest);
+  };
+}
+
 function log(...args)
 {
   console.log(...args);
 }
 
 exports.log = log;
 
 function uabinjectDefuser()
 {
   window.trckd = true;
   window.uabpdl = true;
   window.uabInject = true;
   window.uabDetect = true;
 }
 
 exports["uabinject-defuser"] = makeInjector(uabinjectDefuser);
+
+// This is an implementation of the abort-current-inline-script technique used
+// by uBlock Origin
+// https://github.com/uBlockOrigin/uAssets/blob/68cf8a364fb55deb96264c4e546b58f4c851d782/filters/resources.txt#L1943
+function abortCurrentInlineScript(target, needle)
+{
+  if (!target)
+    return;
+
+  let re = null;
+
+  if (needle)
+  {
+    re = new RegExp(/^\/.+\/$/.test(needle) ?
+                      needle.slice(1, -1) :
+                      regexEscape(needle));
+  }
+
+  let {owner, property} = findProperty(target) || {};
+
+  if (!owner)
+    return;
+
+  let descriptor = Object.getOwnPropertyDescriptor(owner, property);
+  if (descriptor && descriptor.get)
+    return;
+
+  let magic = getMagic();
+
+  injectPropertyAccessValidator(owner, property, () =>
+  {
+    let element = document.currentScript;
+    if (element instanceof HTMLScriptElement &&
+        element.src == "" && (!re || re.test(element.textContent)))
+    {
+      throw new ReferenceError(magic);
+    }
+  });
+
+  suppressMagicError(magic);
+}
+
+exports["abort-current-inline-script"] = makeInjector(abortCurrentInlineScript);