Issue 4125 - [emscripten] Convert filter classes to C++

compiled/Filter.cpp

-#include <emscripten.h>
-#include <unordered_map>
-
 #include "Filter.h"
 #include "CommentFilter.h"
+#include "InvalidFilter.h"
 #include "RegExpFilter.h"
+#include "BlockingFilter.h"
+#include "WhitelistFilter.h"
+#include "ElemHideBase.h"
 #include "ElemHideFilter.h"
 #include "ElemHideException.h"
-#include "tools.h"
-
-Filter::Filter(const std::u16string& text)
-{
-  this->text = text;
-}
-
-const std::u16string Filter::Serialize()
-{
-  return (
-    u"[Filter]\n"
-    u"text=" + text + u"\n"
-  );
-}
+#include "CSSPropertyFilter.h"
+#include "StringMap.h"
 
 namespace
 {
-  std::unordered_map<std::u16string,Filter*> knownFilters;
+  StringMap<Filter*> knownFilters(8192);
 
-  void trim_spaces(std::u16string& str)
+  void NormalizeWhitespace(DependentString& text)
   {
-    size_t pos;
+    String::size_type start = 0;
+    String::size_type end = text.length();
 
-    // Remove leading whitespace
-    pos = str.find_first_not_of(u' ');
-    if (pos > 0)
-      str.erase(0, pos);
+    // Remove leading spaces and special characters like line breaks
+    for (; start < end; start++)
+      if (text[start] > ' ')
+        break;
 
-    // Remove trailing whitespace
-    pos = str.find_last_not_of(u' ');
-    if (pos < str.length() - 1)
-      str.erase(pos + 1);
-  }
+    // Now look for invalid characters inside the string
+    String::size_type pos;
+    for (pos = start; pos < end; pos++)
+      if (text[pos] < ' ')
+        break;
 
-  void remove_spaces(std::u16string& str)
-  {
-    for (size_t i = 0, l = str.length(); i < l; ++i)
+    if (pos < end)
     {
-      if (str[i] == u' ')
+      // Found invalid characters, copy all the valid characters while skipping
+      // the invalid ones.
+      String::size_type delta = 1;
+      for (pos = pos + 1; pos < end; pos++)
       {
-        str.erase(i, 1);
-        --i;
-        --l;
+        if (text[pos] < ' ')
+          delta++;
+        else
+          text[pos - delta] = text[pos];
       }
+      end -= delta;
     }
+
+    // Remove trailing spaces
+    for (; end > 0; end--)
+      if (text[end - 1] != ' ')
+        break;
+
+    // Set new string boundaries
+    text.reset(text, start, end - start);
   }
 }
 
-extern "C"
+Filter::Filter(Type type, const String& text)
+    : mType(type), mText(text)
 {
-  Filter* EMSCRIPTEN_KEEPALIVE Filter_FromText(char16_t* textPtr, int textLen)
+  annotate_address(this, "Filter");
+}
+
+Filter::~Filter()
+{
+  knownFilters.erase(mText);
+}
+
+OwnedString Filter::Serialize() const
+{
+  OwnedString result(u"[Filter]\ntext="_str);
+  result.append(mText);
+  result.append(u'\n');
+  return result;
+}
+
+Filter* Filter::FromText(DependentString& text)
+{
+  NormalizeWhitespace(text);
+  if (text.empty())
+    return nullptr;
+
+  // Parsing also normalizes the filter text, so it has to be done before the
+  // lookup in knownFilters.
+  union
   {
-    std::u16string text(textPtr, textLen);
-    auto it = knownFilters.find(text);
-    if (it != knownFilters.end())
-      return it->second;
+    RegExpFilterData regexp;
+    ElemHideData elemhide;
+  } data;
+  DependentString error;
 
-    Filter* filter = CommentFilter::Create(text);
-    if (!filter)
-      filter = ElemHideBase::Create(text);
-    if (!filter)
-      filter = RegExpFilter::Create(text);
-    return knownFilters[text] = filter;
-  }
-  char16_t* EMSCRIPTEN_KEEPALIVE Filter_Normalize(char16_t* textPtr, size_t textLen, size_t* resultLen)
+  Filter::Type type = CommentFilter::Parse(text);
+  if (type == Filter::Type::UNKNOWN)
+    type = ElemHideBase::Parse(text, data.elemhide);
+  if (type == Filter::Type::UNKNOWN)
+    type = RegExpFilter::Parse(text, error, data.regexp);
+
+  auto knownFilter = knownFilters.find(text);
+  if (knownFilter)
   {
-    std::u16string text(textPtr, textLen);
-
-    // Remove special characters like line breaks
-    for (size_t i = 0, l = text.length(); i < l; ++i)
-    {
-      if (text[i] < u' ')
-      {
-        text.erase(i, 1);
-        --i;
-        --l;
-      }
-    }
-
-    trim_spaces(text);
-
-    {
-      size_t domainsEnd;
-      size_t selectorStart;
-      Filter::Type type = ElemHideBase::Parse(text, &domainsEnd, &selectorStart);
-      if (type != Filter::Type::UNKNOWN)
-      {
-        std::u16string domains = text.substr(0, domainsEnd);
-        std::u16string selector = text.substr(selectorStart);
-        remove_spaces(domains);
-        trim_spaces(selector);
-        return stringToBuffer(domains + (type == Filter::Type::ELEMHIDEEXCEPTION ? u"#@#" : u"##") + selector, resultLen);
-      }
-    }
-
-    if (CommentFilter::Parse(text) == Filter::Type::UNKNOWN)
-      remove_spaces(text);
-    return stringToBuffer(text, resultLen);
+    knownFilter->second->AddRef();
+    return knownFilter->second;
   }
 
-  Filter::Type EMSCRIPTEN_KEEPALIVE Filter_get_type(Filter* filter)
+  FilterPtr filter;
+  switch (type)
   {
-    return filter->GetType();
+    case Filter::Type::COMMENT:
+      filter = new CommentFilter(text);
+      break;
+    case Filter::Type::INVALID:
+      filter = new InvalidFilter(text, error);
+      break;
+    case Filter::Type::BLOCKING:
+      filter = new BlockingFilter(text, data.regexp);
+      break;
+    case Filter::Type::WHITELIST:
+      filter = new WhitelistFilter(text, data.regexp);
+      break;
+    case Filter::Type::ELEMHIDE:
+      filter = new ElemHideFilter(text, data.elemhide);
+      break;
+    case Filter::Type::ELEMHIDEEXCEPTION:
+      filter = new ElemHideException(text, data.elemhide);
+      break;
+    case Filter::Type::CSSPROPERTY:
+      filter = new CSSPropertyFilter(text, data.elemhide);
+      if (static_cast<CSSPropertyFilter*>(filter.get())->IsGeneric())
+        filter = new InvalidFilter(text, u"filter_cssproperty_nodomain"_str);
+      break;
+    default:
+      // This should never happen but just in case
+      return nullptr;
   }
 
-  char16_t* EMSCRIPTEN_KEEPALIVE Filter_Serialize(Filter* filter, size_t* resultLen)
-  {
-    return stringToBuffer(filter->Serialize(), resultLen);
-  }
+  // This is a hack: we looked up the entry using text but create it using
+  // filter->mText. This works because both are equal at this point. However,
+  // text refers to a temporary buffer which will go away.
+  enter_context("Adding to known filters");
+  knownFilter.assign(filter->mText, filter.get());
+  exit_context();
+
+  return filter.release();
 }