| LEFT | RIGHT | 
|---|
| 1 #include <cstdio> | 1 #include <cstdio> | 
| 2 | 2 | 
| 3 #include "ActiveFilter.h" | 3 #include "ActiveFilter.h" | 
| 4 #include "StringScanner.h" | 4 #include "StringScanner.h" | 
| 5 | 5 | 
| 6 namespace | 6 namespace | 
| 7 { | 7 { | 
| 8   void ProcessDomain(String& docDomain, bool ignoreTrailingDot) | 8   const DependentString DEFAULT_DOMAIN(u""_str); | 
| 9   { |  | 
| 10     String::size_type len = docDomain.length(); |  | 
| 11     if (ignoreTrailingDot) |  | 
| 12     { |  | 
| 13       for (; len > 0; --len) |  | 
| 14       { |  | 
| 15         if (docDomain[len - 1] != u'.') |  | 
| 16           break; |  | 
| 17       } |  | 
| 18     } |  | 
| 19 | 9 | 
| 20     for (String::size_type i = 0; i < len; ++i) | 10   OwnedString to_string(unsigned int i) | 
| 21     { |  | 
| 22       String::value_type currChar = docDomain[i]; |  | 
| 23       // TODO: This needs to work for non-ASCII as well |  | 
| 24       if (currChar >= u'A' && currChar <= u'Z') |  | 
| 25         docDomain[i] = currChar + u'a' - u'A'; |  | 
| 26     } |  | 
| 27 |  | 
| 28     docDomain.reset(docDomain, 0, len); |  | 
| 29   } |  | 
| 30 |  | 
| 31   String to_string(unsigned int i) |  | 
| 32   { | 11   { | 
| 33     char buffer[11]; | 12     char buffer[11]; | 
| 34     int len = sprintf(buffer, "%u", i); | 13     int len = sprintf(buffer, "%u", i); | 
| 35     return std::move(String(buffer, len)); | 14 | 
|  | 15     OwnedString result(len); | 
|  | 16     for (String::size_type i = 0; i < len; i++) | 
|  | 17       result[i] = buffer[i]; | 
|  | 18     return result; | 
| 36   } | 19   } | 
| 37 } | 20 } | 
| 38 | 21 | 
| 39 ActiveFilter::ActiveFilter(const String& text, bool ignoreTrailingDot) | 22 ActiveFilter::ActiveFilter(Type type, const String& text, bool ignoreTrailingDot
     ) | 
| 40     : Filter(text), mDisabled(false), mHitCount(0), mLastHit(0), | 23     : Filter(type, text), mDisabled(false), mHitCount(0), mLastHit(0), | 
| 41       mIgnoreTrailingDot(ignoreTrailingDot) | 24       mIgnoreTrailingDot(ignoreTrailingDot) | 
| 42 { | 25 { | 
| 43 } | 26 } | 
| 44 | 27 | 
| 45 void ActiveFilter::ParseDomains(const String& str, String::value_type separator) | 28 ActiveFilter::DomainMap* ActiveFilter::GetDomains() const | 
|  | 29 { | 
|  | 30   return mDomains.get(); | 
|  | 31 } | 
|  | 32 | 
|  | 33 ActiveFilter::SitekeySet* ActiveFilter::GetSitekeys() const | 
|  | 34 { | 
|  | 35   return mSitekeys.get(); | 
|  | 36 } | 
|  | 37 | 
|  | 38 void ActiveFilter::ParseDomains(const String& domains, | 
|  | 39     String::value_type separator) const | 
| 46 { | 40 { | 
| 47   DomainMap::size_type count = 2; | 41   DomainMap::size_type count = 2; | 
| 48   for (String::size_type i = 0; i < str.length(); i++) | 42   for (String::size_type i = 0; i < domains.length(); i++) | 
| 49     if (str[i] == separator) | 43     if (domains[i] == separator) | 
| 50       count++; | 44       count++; | 
| 51 | 45 | 
| 52   mDomains.reset(new DomainMap(count)); | 46   mDomains.reset(new DomainMap(count)); | 
| 53   annotate_address(mDomains.get(), "DomainMap"); | 47   annotate_address(mDomains.get(), "DomainMap"); | 
| 54 | 48 | 
| 55   StringScanner scanner(str, separator); | 49   StringScanner scanner(domains, 0, separator); | 
| 56   String::size_type start = 0; | 50   String::size_type start = 0; | 
| 57   bool reverse = false; | 51   bool reverse = false; | 
| 58   bool hasIncludes = false; | 52   bool hasIncludes = false; | 
| 59   bool done = false; | 53   bool done = scanner.done(); | 
| 60   while (!done) | 54   while (!done) | 
| 61   { | 55   { | 
| 62     done = scanner.done(); | 56     done = scanner.done(); | 
| 63     String::value_type currChar = scanner.next(); | 57     String::value_type currChar = scanner.next(); | 
| 64     if (currChar == u'~' && scanner.position() == start) | 58     if (currChar == u'~' && scanner.position() == start) | 
| 65     { | 59     { | 
| 66       start++; | 60       start++; | 
| 67       reverse = true; | 61       reverse = true; | 
| 68     } | 62     } | 
| 69     else if (currChar == separator) | 63     else if (currChar == separator) | 
| 70     { | 64     { | 
| 71       if (scanner.position() > start) | 65       String::size_type len = scanner.position() - start; | 
|  | 66       if (len > 0 && mIgnoreTrailingDot && domains[start + len - 1] == '.') | 
|  | 67         len--; | 
|  | 68       if (len > 0) | 
| 72       { | 69       { | 
| 73         String domain(str, start, scanner.position() - start); |  | 
| 74         ProcessDomain(domain, mIgnoreTrailingDot); |  | 
| 75 |  | 
| 76         enter_context("Adding to ActiveFilter.mDomains"); | 70         enter_context("Adding to ActiveFilter.mDomains"); | 
| 77         (*mDomains)[domain] = !reverse; | 71         (*mDomains)[DependentString(domains, start, len)] = !reverse; | 
| 78         exit_context(); | 72         exit_context(); | 
| 79 | 73 | 
| 80         if (!reverse) | 74         if (!reverse) | 
| 81           hasIncludes = true; | 75           hasIncludes = true; | 
| 82       } | 76       } | 
| 83       start = scanner.position() + 1; | 77       start = scanner.position() + 1; | 
| 84       reverse = false; | 78       reverse = false; | 
| 85     } | 79     } | 
| 86   } | 80   } | 
| 87   enter_context("Adding to ActiveFilter.mDomains"); | 81   enter_context("Adding to ActiveFilter.mDomains"); | 
| 88   (*mDomains)[u""_str] = !hasIncludes; | 82   (*mDomains)[DEFAULT_DOMAIN] = !hasIncludes; | 
| 89   exit_context(); | 83   exit_context(); | 
| 90 } | 84 } | 
| 91 | 85 | 
| 92 void ActiveFilter::AddSitekey(const String& sitekey) | 86 void ActiveFilter::AddSitekey(const String& sitekey) const | 
| 93 { | 87 { | 
| 94   if (!mSitekeys) | 88   if (!mSitekeys) | 
| 95   { | 89   { | 
| 96     mSitekeys.reset(new SitekeySet()); | 90     mSitekeys.reset(new SitekeySet()); | 
| 97     annotate_address(mSitekeys.get(), "SitekeySet"); | 91     annotate_address(mSitekeys.get(), "SitekeySet"); | 
| 98   } | 92   } | 
| 99 | 93 | 
| 100   enter_context("Adding to ActiveFilter.mSitekeys"); | 94   enter_context("Adding to ActiveFilter.mSitekeys"); | 
| 101   mSitekeys->insert(sitekey); | 95   mSitekeys->insert(sitekey); | 
| 102   exit_context(); | 96   exit_context(); | 
| 103 } | 97 } | 
| 104 | 98 | 
| 105 bool ActiveFilter::IsActiveOnDomain(String& docDomain, const String& sitekey) co
     nst | 99 bool ActiveFilter::IsActiveOnDomain(DependentString& docDomain, const String& si
     tekey) const | 
| 106 { | 100 { | 
| 107   if (mSitekeys && mSitekeys->find(sitekey) == mSitekeys->end()) | 101   auto sitekeys = GetSitekeys(); | 
|  | 102   if (sitekeys && !sitekeys->find(sitekey)) | 
| 108     return false; | 103     return false; | 
| 109 | 104 | 
| 110   // If no domains are set the rule matches everywhere | 105   // If no domains are set the rule matches everywhere | 
| 111   if (!mDomains) | 106   auto domains = GetDomains(); | 
|  | 107   if (!domains) | 
| 112     return true; | 108     return true; | 
| 113 | 109 | 
| 114   // If the document has no host name, match only if the filter isn't restricted | 110   // If the document has no host name, match only if the filter isn't restricted | 
| 115   // to specific domains | 111   // to specific domains | 
| 116   if (docDomain.empty()) | 112   if (docDomain.empty()) | 
| 117     return (*mDomains)[u""_str]; | 113     return (*domains)[DEFAULT_DOMAIN]; | 
| 118 | 114 | 
| 119   ProcessDomain(docDomain, mIgnoreTrailingDot); | 115   docDomain.toLower(); | 
|  | 116 | 
|  | 117   String::size_type len = docDomain.length(); | 
|  | 118   if (len > 0 && mIgnoreTrailingDot && docDomain[len - 1] == '.') | 
|  | 119     docDomain.reset(docDomain, 0, len - 1); | 
| 120   while (true) | 120   while (true) | 
| 121   { | 121   { | 
| 122     auto it = mDomains->find(docDomain); | 122     auto it = domains->find(docDomain); | 
| 123     if (it != mDomains->end()) | 123     if (it) | 
| 124       return it->second; | 124       return it->second; | 
| 125 | 125 | 
| 126     String::size_type nextDot = docDomain.find(u'.'); | 126     String::size_type nextDot = docDomain.find(u'.'); | 
| 127     if (nextDot == docDomain.npos) | 127     if (nextDot == docDomain.npos) | 
| 128       break; | 128       break; | 
| 129     docDomain.reset(docDomain, nextDot + 1); | 129     docDomain.reset(docDomain, nextDot + 1); | 
| 130   } | 130   } | 
| 131   return (*mDomains)[u""_str]; | 131   return (*domains)[DEFAULT_DOMAIN]; | 
| 132 } | 132 } | 
| 133 | 133 | 
| 134 bool ActiveFilter::IsActiveOnlyOnDomain(String& docDomain) const | 134 bool ActiveFilter::IsActiveOnlyOnDomain(DependentString& docDomain) const | 
| 135 { | 135 { | 
| 136   if (!mDomains || docDomain.empty() || (*mDomains)[u""_str]) | 136   auto domains = GetDomains(); | 
|  | 137   if (!domains || docDomain.empty() || (*domains)[DEFAULT_DOMAIN]) | 
| 137     return false; | 138     return false; | 
| 138 | 139 | 
| 139   ProcessDomain(docDomain, mIgnoreTrailingDot); | 140   docDomain.toLower(); | 
| 140   for (auto it = mDomains->begin(); it != mDomains->end(); ++it) | 141 | 
|  | 142   String::size_type len = docDomain.length(); | 
|  | 143   if (len > 0 && mIgnoreTrailingDot && docDomain[len - 1] == '.') | 
|  | 144     docDomain.reset(docDomain, 0, len - 1); | 
|  | 145   for (auto it = domains->begin(); it != domains->end(); ++it) | 
| 141   { | 146   { | 
| 142     if (!it->second || it->first.equals(docDomain)) | 147     if (!it->second || it->first.equals(docDomain)) | 
| 143       continue; | 148       continue; | 
| 144 | 149 | 
| 145     size_t len1 = it->first.length(); | 150     size_t len1 = it->first.length(); | 
| 146     size_t len2 = docDomain.length(); | 151     size_t len2 = docDomain.length(); | 
| 147     if (len1 > len2 && | 152     if (len1 > len2 && | 
| 148         String(it->first, len1 - len2).equals(docDomain) && | 153         DependentString(it->first, len1 - len2).equals(docDomain) && | 
| 149         it->first[len1 - len2 - 1] == u'.') | 154         it->first[len1 - len2 - 1] == u'.') | 
| 150     { | 155     { | 
| 151       continue; | 156       continue; | 
| 152     } | 157     } | 
| 153 | 158 | 
| 154     return false; | 159     return false; | 
| 155   } | 160   } | 
| 156   return true; | 161   return true; | 
| 157 } | 162 } | 
| 158 | 163 | 
| 159 bool ActiveFilter::IsGeneric() const | 164 bool ActiveFilter::IsGeneric() const | 
| 160 { | 165 { | 
| 161   return !mSitekeys && (!mDomains || (*mDomains)[u""_str]); | 166   auto sitekeys = GetSitekeys(); | 
|  | 167   auto domains = GetDomains(); | 
|  | 168   return !sitekeys && (!domains || (*domains)[DEFAULT_DOMAIN]); | 
| 162 } | 169 } | 
| 163 | 170 | 
| 164 String ActiveFilter::Serialize() const | 171 OwnedString ActiveFilter::Serialize() const | 
| 165 { | 172 { | 
| 166   /* TODO this is very inefficient */ | 173   /* TODO this is very inefficient */ | 
| 167   String result(Filter::Serialize()); | 174   OwnedString result(Filter::Serialize()); | 
| 168   if (mDisabled) | 175   if (mDisabled) | 
| 169     result.append(u"disabled=true\n"_str); | 176     result.append(u"disabled=true\n"_str); | 
| 170   if (mHitCount) | 177   if (mHitCount) | 
| 171   { | 178   { | 
| 172     result.append(u"hitCount="_str); | 179     result.append(u"hitCount="_str); | 
| 173     result.append(to_string(mHitCount)); | 180     result.append(to_string(mHitCount)); | 
| 174     result.append(u'\n'); | 181     result.append(u'\n'); | 
| 175   } | 182   } | 
| 176   if (mLastHit) | 183   if (mLastHit) | 
| 177   { | 184   { | 
| 178     result.append(u"lastHit="_str); | 185     result.append(u"lastHit="_str); | 
| 179     result.append(to_string(mLastHit)); | 186     result.append(to_string(mLastHit)); | 
| 180     result.append(u'\n'); | 187     result.append(u'\n'); | 
| 181   } | 188   } | 
| 182   return std::move(result); | 189   return result; | 
| 183 } | 190 } | 
| LEFT | RIGHT | 
|---|