| // Copyright 2018 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef COMPONENTS_OPTIMIZATION_GUIDE_OPTIMIZATION_FILTER_H_ |
| #define COMPONENTS_OPTIMIZATION_GUIDE_OPTIMIZATION_FILTER_H_ |
| |
| #include <memory> |
| #include <vector> |
| |
| #include "base/macros.h" |
| #include "base/sequence_checker.h" |
| #include "components/optimization_guide/bloom_filter.h" |
| #include "third_party/re2/src/re2/re2.h" |
| #include "url/gurl.h" |
| |
| namespace optimization_guide { |
| |
| typedef std::vector<std::unique_ptr<re2::RE2>> RegexpList; |
| |
| // OptimizationFilter represents a filter with two underlying implementations: a |
| // Bloom filter and a set of regexps. This class has a 1:1 mapping with an |
| // OptimizationFilter protobuf message where this is the logical implementation |
| // of the proto data. |
| // |
| // TODO(dougarnett): consider moving this and BloomFilter under |
| // components/blacklist/. |
| class OptimizationFilter { |
| public: |
| explicit OptimizationFilter(std::unique_ptr<BloomFilter> bloom_filter, |
| std::unique_ptr<RegexpList> regexps); |
| |
| ~OptimizationFilter(); |
| |
| // Returns true if the given url is matched by this filter. |
| bool Matches(const GURL& url) const; |
| |
| private: |
| // Returns whether this filter contains a host suffix for the host part |
| // of |url|. It will check at most 5 host suffixes and it may ignore simple |
| // top level domain matches (such as "com" or "co.in"). |
| // |
| // A host suffix is comprised of domain name level elements (vs. characters). |
| // For example, "server1.www.company.co.in" has the following suffixes that |
| // would be checked for membership: |
| // "server1.www.company.co.in" |
| // "www.company.co.in" |
| // "company.co.in" |
| // This method will return true if any of those suffixes are present. |
| bool ContainsHostSuffix(const GURL& url) const; |
| |
| // Returns whether this filter contains a regexp that matches the given url. |
| bool MatchesRegexp(const GURL& url) const; |
| |
| std::unique_ptr<BloomFilter> bloom_filter_; |
| |
| std::unique_ptr<RegexpList> regexps_; |
| |
| SEQUENCE_CHECKER(sequence_checker_); |
| |
| DISALLOW_COPY_AND_ASSIGN(OptimizationFilter); |
| }; |
| |
| } // namespace optimization_guide |
| |
| #endif // COMPONENTS_OPTIMIZATION_GUIDE_OPTIMIZATION_FILTER_H_ |