Index: compiled/ActiveFilter.cpp |
=================================================================== |
new file mode 100644 |
--- /dev/null |
+++ b/compiled/ActiveFilter.cpp |
@@ -0,0 +1,183 @@ |
+#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)); |
+ } |
+} |
+ |
+ActiveFilter::ActiveFilter(const String& text, bool ignoreTrailingDot) |
+ : Filter(text), mDisabled(false), mHitCount(0), mLastHit(0), |
+ mIgnoreTrailingDot(ignoreTrailingDot) |
+{ |
+} |
+ |
+void ActiveFilter::ParseDomains(const String& str, String::value_type separator) |
+{ |
+ DomainMap::size_type count = 2; |
+ for (String::size_type i = 0; i < str.length(); i++) |
+ if (str[i] == separator) |
+ count++; |
+ |
+ mDomains.reset(new DomainMap(count)); |
+ annotate_address(mDomains.get(), "DomainMap"); |
+ |
+ StringScanner scanner(str, 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(str, 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); |
+} |