Issue 3670 - Make rules case sensitive where possible

lib/contentBlockerList.js

Index: lib/contentBlockerList.js
diff --git a/lib/contentBlockerList.js b/lib/contentBlockerList.js
index 792b54e624c219634ef6cc21d77c3aa4690507fa..3810cc36c02c1c7c78a151f45789c3ff1038ef2b 100644
--- a/lib/contentBlockerList.js
+++ b/lib/contentBlockerList.js
@@ -50,7 +50,7 @@ function escapeRegExp(s)
 
 function matchDomain(domain)
 {
-  return "^https?://([^/:]*\\.)?" + escapeRegExp(domain) + "[/:]";
+  return "^https?://([^/:]*\\.)?" + escapeRegExp(domain).toLowerCase() + "[/:]";
 }
 
 function convertElemHideFilter(filter, elemhideSelectorExceptions)
@@ -65,10 +65,15 @@ function convertElemHideFilter(filter, elemhideSelectorExceptions)
     return {matchDomains: included.map(matchDomain), selector: filter.selector};
 }
 
+// Convert the "regexpSource" part of a filter's text to a regular expression,
+// also deciding if the expression can safely be converted to and matched as
+// lowercase.
 function toRegExp(text)
 {
   let result = [];
   let lastIndex = text.length - 1;
+  let containsHostname = false;
+  let caseSensitive = false;
 
   for (let i = 0; i < text.length; i++)
   {
@@ -98,11 +103,24 @@ function toRegExp(text)
         if (i == 1 && text[0] == "|")
         {
           result.push("https?://");
+          containsHostname = caseSensitive = true;
           break;
         }
-      case ".": case "+": case "?": case "$":
-      case "{": case "}": case "(": case ")":
-      case "[": case "]": case "\\":
+        result.push("\\", c);
+        break;
+      case "/": case "?": case "*": case "^":
+        if (containsHostname && i < lastIndex)
+          caseSensitive = false;
+        else if (i >= 2 && text[i-2] == ":" && text[i-1] == "/" && c == "/")
+          containsHostname = caseSensitive = true;
+        if (c != "?")
+        {
+          result.push(c);
+          break;
+        }
+      case ".": case "+": case "$": case "{":
+      case "}": case "(": case ")": case "[":
+      case "]": case "\\":
         result.push("\\", c);
         break;
       default:
@@ -110,12 +128,12 @@ function toRegExp(text)
     }
   }
 
-  return result.join("");
+  return {regexp: result.join(""), caseSensitive: caseSensitive};
 }
 
-function getRegExpSource(filter)
+function getRegExpTrigger(filter)
 {
-  let source = toRegExp(filter.regexpSource.replace(
+  let result = toRegExp(filter.regexpSource.replace(
     // Safari expects punycode, filter lists use unicode
     /^(\|\||\|?https?:\/\/)([\w\-.*\u0080-\uFFFF]+)/i,
     function (match, prefix, domain)
@@ -124,11 +142,21 @@ function getRegExpSource(filter)
     }
   ));
 
+  let trigger = {"url-filter": result.regexp};
+
   // Limit rules to to HTTP(S) URLs
-  if (!/^(\^|http)/i.test(source))
-    source = "^https?://.*" + source;
+  if (!/^(\^|http)/i.test(trigger["url-filter"]))
+    trigger["url-filter"] = "^https?://.*" + trigger["url-filter"];
 
-  return source;
+  // For rules containing only a hostname we know that we're matching against
+  // a lowercase string and can therefore enable case sensitive matching.
+  if (result.caseSensitive)
+  {
+    trigger["url-filter"] = trigger["url-filter"].toLowerCase();
+    trigger["url-filter-is-case-sensitive"] = true;
+  }
+
+  return trigger;
 }
 
 function getResourceTypes(filter)
@@ -175,11 +203,14 @@ function addDomainPrefix(domains)
 
 function convertFilter(filter, action, withResourceTypes)
 {
-  let trigger = {"url-filter": getRegExpSource(filter)};
+  let trigger = getRegExpTrigger(filter);
   let included = [];
   let excluded = [];
 
-  parseDomains(filter.domains, included, excluded);
+  if (filter.matchCase)
+    trigger["url-filter-is-case-sensitive"] = true;
+
+ parseDomains(filter.domains, included, excluded);
 
   if (withResourceTypes)
     trigger["resource-type"] = getResourceTypes(filter);
@@ -377,7 +408,8 @@ ContentBlockerList.prototype.generateRules = function(filter)
       selector = convertIDSelectorsToAttributeSelectors(selector);
 
       addRule({
-        trigger: {"url-filter": matchDomain},
+        trigger: {"url-filter": matchDomain,
+                  "url-filter-is-case-sensitive": true},
         action: {type: "css-display-none",
                  selector: selector}
       });