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

compiled/ActiveFilter.cpp

+#include <cstdio>
+
+#include "ActiveFilter.h"
+#include "StringScanner.h"
+
+namespace
+{
+  void ProcessDomain(String& docDomain, bool ignoreTrailingDot)
+  {
+    String::size_type len = docDomain.length();
+    if (ignoreTrailingDot)
+    {
+      for (; len > 0; --len)
+      {
+        if (docDomain[len - 1] != u'.')
+          break;
+      }
+    }
+
+    for (String::size_type i = 0; i < len; ++i)
+    {
+      String::value_type currChar = docDomain[i];
+      // TODO: This needs to work for non-ASCII as well
+      if (currChar >= u'A' && currChar <= u'Z')
+        docDomain[i] = currChar + u'a' - u'A';
+    }
+
+    docDomain.reset(docDomain, 0, len);
+  }
+
+  String to_string(unsigned int i)
+  {
+    char buffer[11];
+    int len = sprintf(buffer, "%u", i);
+    return std::move(String(buffer, len).ensure_own_buffer());
+  }
+}
+
+ActiveFilter::ActiveFilter(const String& text, bool ignoreTrailingDot)
+    : Filter(text), mDisabled(false), mHitCount(0), mLastHit(0),
+      mIgnoreTrailingDot(ignoreTrailingDot)
+{
+}
+
+void ActiveFilter::ParseDomains(String& str, String::size_type pos,
+    String::size_type len, String::value_type separator)
+{
+  String domains(str, pos, len);
+  DomainMap::size_type count = 2;
+  for (String::size_type i = 0; i < domains.length(); i++)
+    if (domains[i] == separator)
+      count++;
+
+  mDomains.reset(new DomainMap(count));
+  annotate_address(mDomains.get(), "DomainMap");
+
+  StringScanner scanner(domains, separator);
+  String::size_type start = 0;
+  bool reverse = false;
+  bool hasIncludes = false;
+  bool done = false;
+  while (!done)
+  {
+    done = scanner.done();
+    String::value_type currChar = scanner.next();
+    if (currChar == u'~' && scanner.position() == start)
+    {
+      start++;
+      reverse = true;
+    }
+    else if (currChar == separator)
+    {
+      if (scanner.position() > start)
+      {
+        String domain(domains, start, scanner.position() - start);
+        ProcessDomain(domain, mIgnoreTrailingDot);
+
+        enter_context("Adding to ActiveFilter.mDomains");
+        (*mDomains)[domain] = !reverse;
+        exit_context();
+
+        if (!reverse)
+          hasIncludes = true;
+      }
+      start = scanner.position() + 1;
+      reverse = false;
+    }
+  }
+  enter_context("Adding to ActiveFilter.mDomains");
+  (*mDomains)[u""_str] = !hasIncludes;
+  exit_context();
+}
+
+void ActiveFilter::AddSitekey(const String& sitekey)
+{
+  if (!mSitekeys)
+  {
+    mSitekeys.reset(new SitekeySet());
+    annotate_address(mSitekeys.get(), "SitekeySet");
+  }
+
+  enter_context("Adding to ActiveFilter.mSitekeys");
+  mSitekeys->insert(sitekey);
+  exit_context();
+}
+
+bool ActiveFilter::IsActiveOnDomain(String& docDomain, const String& sitekey) const
+{
+  if (mSitekeys && mSitekeys->find(sitekey) == mSitekeys->end())
+    return false;
+
+  // If no domains are set the rule matches everywhere
+  if (!mDomains)
+    return true;
+
+  // If the document has no host name, match only if the filter isn't restricted
+  // to specific domains
+  if (docDomain.empty())
+    return (*mDomains)[u""_str];
+
+  ProcessDomain(docDomain, mIgnoreTrailingDot);
+  while (true)
+  {
+    auto it = mDomains->find(docDomain);
+    if (it != mDomains->end())
+      return it->second;
+
+    String::size_type nextDot = docDomain.find(u'.');
+    if (nextDot == docDomain.npos)
+      break;
+    docDomain.reset(docDomain, nextDot + 1);
+  }
+  return (*mDomains)[u""_str];
+}
+
+bool ActiveFilter::IsActiveOnlyOnDomain(String& docDomain) const
+{
+  if (!mDomains || docDomain.empty() || (*mDomains)[u""_str])
+    return false;
+
+  ProcessDomain(docDomain, mIgnoreTrailingDot);
+  for (auto it = mDomains->begin(); it != mDomains->end(); ++it)
+  {
+    if (!it->second || it->first.equals(docDomain))
+      continue;
+
+    size_t len1 = it->first.length();
+    size_t len2 = docDomain.length();
+    if (len1 > len2 &&
+        String(it->first, len1 - len2).equals(docDomain) &&
+        it->first[len1 - len2 - 1] == u'.')
+    {
+      continue;
+    }
+
+    return false;
+  }
+  return true;
+}
+
+bool ActiveFilter::IsGeneric() const
+{
+  return !mSitekeys && (!mDomains || (*mDomains)[u""_str]);
+}
+
+String ActiveFilter::Serialize() const
+{
+  /* TODO this is very inefficient */
+  String result(Filter::Serialize());
+  if (mDisabled)
+    result.append(u"disabled=true\n"_str);
+  if (mHitCount)
+  {
+    result.append(u"hitCount="_str);
+    result.append(to_string(mHitCount));
+    result.append(u'\n');
+  }
+  if (mLastHit)
+  {
+    result.append(u"lastHit="_str);
+    result.append(to_string(mLastHit));
+    result.append(u'\n');
+  }
+  return std::move(result.ensure_own_buffer());
+}