blob: ae994db85dcc89dc5fc8f367f11474a65cf0528a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// Copyright 2013 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.
#include "components/url_matcher/regex_set_matcher.h"
#include "base/logging.h"
#include "base/stl_util.h"
#include "base/strings/string_util.h"
#include "components/url_matcher/substring_set_matcher.h"
#include "third_party/re2/re2/filtered_re2.h"
#include "third_party/re2/re2/re2.h"
namespace url_matcher {
RegexSetMatcher::RegexSetMatcher() {}
RegexSetMatcher::~RegexSetMatcher() {
DeleteSubstringPatterns();
}
void RegexSetMatcher::AddPatterns(
const std::vector<const StringPattern*>& regex_list) {
if (regex_list.empty())
return;
for (size_t i = 0; i < regex_list.size(); ++i) {
regexes_[regex_list[i]->id()] = regex_list[i];
}
RebuildMatcher();
}
void RegexSetMatcher::ClearPatterns() {
regexes_.clear();
RebuildMatcher();
}
bool RegexSetMatcher::Match(const std::string& text,
std::set<StringPattern::ID>* matches) const {
size_t old_number_of_matches = matches->size();
if (regexes_.empty())
return false;
if (!filtered_re2_.get()) {
LOG(ERROR) << "RegexSetMatcher was not initialized";
return false;
}
// FilteredRE2 expects lowercase for prefiltering, but we still
// match case-sensitively.
std::vector<RE2ID> atoms(FindSubstringMatches(base::ToLowerASCII(text)));
std::vector<RE2ID> re2_ids;
filtered_re2_->AllMatches(text, atoms, &re2_ids);
for (size_t i = 0; i < re2_ids.size(); ++i) {
StringPattern::ID id = re2_id_map_[re2_ids[i]];
matches->insert(id);
}
return old_number_of_matches != matches->size();
}
bool RegexSetMatcher::IsEmpty() const {
return regexes_.empty();
}
std::vector<RegexSetMatcher::RE2ID> RegexSetMatcher::FindSubstringMatches(
const std::string& text) const {
std::set<int> atoms_set;
substring_matcher_->Match(text, &atoms_set);
return std::vector<RE2ID>(atoms_set.begin(), atoms_set.end());
}
void RegexSetMatcher::RebuildMatcher() {
re2_id_map_.clear();
filtered_re2_.reset(new re2::FilteredRE2());
if (regexes_.empty())
return;
for (RegexMap::iterator it = regexes_.begin(); it != regexes_.end(); ++it) {
RE2ID re2_id;
RE2::ErrorCode error = filtered_re2_->Add(
it->second->pattern(), RE2::DefaultOptions, &re2_id);
if (error == RE2::NoError) {
DCHECK_EQ(static_cast<RE2ID>(re2_id_map_.size()), re2_id);
re2_id_map_.push_back(it->first);
} else {
// Unparseable regexes should have been rejected already in
// URLMatcherFactory::CreateURLMatchesCondition.
LOG(ERROR) << "Could not parse regex (id=" << it->first << ", "
<< it->second->pattern() << ")";
}
}
std::vector<std::string> strings_to_match;
filtered_re2_->Compile(&strings_to_match);
substring_matcher_.reset(new SubstringSetMatcher);
DeleteSubstringPatterns();
// Build SubstringSetMatcher from |strings_to_match|.
// SubstringSetMatcher doesn't own its strings.
for (size_t i = 0; i < strings_to_match.size(); ++i) {
substring_patterns_.push_back(
new StringPattern(strings_to_match[i], i));
}
substring_matcher_->RegisterPatterns(substring_patterns_);
}
void RegexSetMatcher::DeleteSubstringPatterns() {
STLDeleteElements(&substring_patterns_);
}
} // namespace url_matcher
|