Issue 7250 - Optimize third-party request check

lib/domain.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/>.
  */
 
 "use strict";
 
 const publicSuffixes = require("../data/publicSuffixList.json");
 
 /**
+ * Map of public suffixes to their offsets.
+ * @type {Map.<string,number>}
+ */
+let publicSuffixMap = buildPublicSuffixMap();
+
+/**
+ * Builds a map of public suffixes to their offsets.
+ * @returns {Map.<string,number>}
+ */
+function buildPublicSuffixMap()
+{
+  let map = new Map();
+
+  for (let key in publicSuffixes)
+    map.set(key, publicSuffixes[key]);
+
+  return map;
+}
+
+/**
+ * Yields all suffixes for a domain. For example, given the domain
+ * <code>www.example.com</code>, this function yields
+ * <code>www.example.com</code>, <code>example.com</code>, and
+ * <code>com</code>, in that order.
+ *
+ * @param {string} domain The domain.
+ * @yields {string} The next suffix for the domain.
+ */
+function* suffixes(domain)
+{
+  while (domain != "")
+  {
+    yield domain;
+
+    let dotIndex = domain.indexOf(".");
+    domain = dotIndex == -1 ? "" : domain.substr(dotIndex + 1);
+  }
+}
+
+/**
  * Checks whether the given hostname is a domain.
  *
  * @param {string} hostname
  * @returns {boolean}
  */
 function isDomain(hostname)
 {
   // No hostname or IPv4 address, also considering hexadecimal octets.
   if (/^((0x[\da-f]+|\d+)(\.|$))*$/i.test(hostname))
     return false;
 
   // IPv6 address. Since there can't be colons in domains, we can
   // just check whether there are any colons to exclude IPv6 addresses.
   return hostname.indexOf(":") == -1;
 }
 
 /**
  * Gets the base domain for the given hostname.
  *
  * @param {string} hostname
  * @returns {string}
  */
 function getDomain(hostname)
 {
-  let bits = hostname.split(".");
-  let cutoff = bits.length - 2;
+  let slices = [];
+  let cutoff = NaN;
 
-  for (let i = 0; i < bits.length; i++)
+  for (let suffix of suffixes(hostname))
   {
-    let offset = publicSuffixes[bits.slice(i).join(".")];
+    slices.push(suffix);
+
+    let offset = publicSuffixMap.get(suffix);
 
     if (typeof offset != "undefined")
     {
-      cutoff = i - offset;
+      cutoff = slices.length - 1 - offset;
       break;
     }
   }
 
+  if (isNaN(cutoff))
+    return slices.length > 2 ? slices[slices.length - 2] : hostname;
+
   if (cutoff <= 0)
     return hostname;
 
-  return bits.slice(cutoff).join(".");
+  return slices[cutoff];
 }
 
+exports.getDomain = getDomain;
+
 /**
  * Checks whether a request's origin is different from its document's origin.
  *
  * @param {URL} url The request URL.
  * @param {string} documentHostname The IDNA-encoded hostname of the document.
  *
  * @returns {boolean}
  */
 function isThirdParty(url, documentHostname)
 {
-  let requestHostname = url.hostname.replace(/\.+$/, "");
-  documentHostname = documentHostname.replace(/\.+$/, "");
+  let requestHostname = url.hostname;
+
+  if (requestHostname[requestHostname.length - 1] == ".")
+    requestHostname = requestHostname.replace(/\.+$/, "");
+
+  if (documentHostname[documentHostname.length - 1] == ".")
+    documentHostname = documentHostname.replace(/\.+$/, "");
 
   if (requestHostname == documentHostname)
     return false;
 
   if (!isDomain(requestHostname) || !isDomain(documentHostname))
     return true;
 
   return getDomain(requestHostname) != getDomain(documentHostname);
 }
 
 exports.isThirdParty = isThirdParty;