OLD | NEW |
1 /* | 1 /* |
2 * This file is part of Adblock Plus <https://adblockplus.org/>, | 2 * This file is part of Adblock Plus <https://adblockplus.org/>, |
3 * Copyright (C) 2006-2015 Eyeo GmbH | 3 * Copyright (C) 2006-2015 Eyeo GmbH |
4 * | 4 * |
5 * Adblock Plus is free software: you can redistribute it and/or modify | 5 * Adblock Plus is free software: you can redistribute it and/or modify |
6 * it under the terms of the GNU General Public License version 3 as | 6 * it under the terms of the GNU General Public License version 3 as |
7 * published by the Free Software Foundation. | 7 * published by the Free Software Foundation. |
8 * | 8 * |
9 * Adblock Plus is distributed in the hope that it will be useful, | 9 * Adblock Plus is distributed in the hope that it will be useful, |
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of | 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 * GNU General Public License for more details. | 12 * GNU General Public License for more details. |
13 * | 13 * |
14 * You should have received a copy of the GNU General Public License | 14 * You should have received a copy of the GNU General Public License |
15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
16 */ | 16 */ |
17 | 17 |
18 #include "PluginStdAfx.h" | 18 #include "PluginStdAfx.h" |
19 #include "AdblockPlusClient.h" | 19 #include "AdblockPlusClient.h" |
20 #include "PluginFilter.h" | 20 #include "PluginFilter.h" |
21 #include "PluginSettings.h" | 21 #include "PluginSettings.h" |
22 #include "PluginMutex.h" | 22 #include "PluginMutex.h" |
23 #include "PluginSettings.h" | 23 #include "PluginSettings.h" |
24 #include "PluginSystem.h" | 24 #include "PluginSystem.h" |
25 #include "PluginClass.h" | 25 #include "PluginClass.h" |
26 #include "mlang.h" | 26 #include "mlang.h" |
| 27 #include "TokenSequence.h" |
27 #include "..\shared\CriticalSection.h" | 28 #include "..\shared\CriticalSection.h" |
28 #include "..\shared\Utils.h" | 29 #include "..\shared\Utils.h" |
29 #include "..\shared\MsHTMLUtils.h" | 30 #include "..\shared\MsHTMLUtils.h" |
30 | 31 |
31 // The filters are described at http://adblockplus.org/en/filters | 32 // The filters are described at http://adblockplus.org/en/filters |
32 | 33 |
33 static CriticalSection s_criticalSectionFilterMap; | 34 static CriticalSection s_criticalSectionFilterMap; |
34 | 35 |
35 // ============================================================================ | 36 namespace |
36 // CFilterElementHideAttrSelector | |
37 // ============================================================================ | |
38 | |
39 CFilterElementHideAttrSelector::CFilterElementHideAttrSelector() : m_type(TYPE_N
ONE), m_pos(POS_NONE), m_bstrAttr(NULL) | |
40 { | 37 { |
| 38 std::wstring ToLowerString(const std::wstring& s) |
| 39 { |
| 40 return ToWstring(ToCString(s).MakeLower()); |
| 41 } |
41 } | 42 } |
42 | 43 |
43 CFilterElementHideAttrSelector::CFilterElementHideAttrSelector(const CFilterElem
entHideAttrSelector& filter) | |
44 { | |
45 m_type = filter.m_type; | |
46 m_pos = filter.m_pos; | |
47 m_bstrAttr = filter.m_bstrAttr; | |
48 | |
49 m_value = filter.m_value; | |
50 } | |
51 | |
52 CFilterElementHideAttrSelector::~CFilterElementHideAttrSelector() | |
53 { | |
54 } | |
55 | |
56 | |
57 // ============================================================================ | 44 // ============================================================================ |
58 // CFilterElementHide | 45 // CFilterElementHide |
59 // ============================================================================ | 46 // ============================================================================ |
| 47 namespace |
| 48 { |
| 49 class ElementHideParseException |
| 50 : public std::runtime_error |
| 51 { |
| 52 std::string message(const std::wstring& filterText, const std::string& reaso
n) |
| 53 { |
| 54 std::string s( |
| 55 "CFilterElementHide::CFilterElementHide, error parsing selector \"" |
| 56 + ToUtf8String(filterText) + "\" (" + reason + ")" |
| 57 ); |
| 58 DEBUG_FILTER(ToUtf16String(s)); |
| 59 return s; |
| 60 } |
| 61 public: |
| 62 ElementHideParseException(const std::wstring& filterText, const std::string&
reason) |
| 63 : std::runtime_error(message(filterText, reason)) |
| 64 {} |
| 65 }; |
| 66 } |
60 | 67 |
61 CFilterElementHide::CFilterElementHide(const CString& filterText) : m_filterText
(filterText), m_type(ETraverserComplexType::TRAVERSER_TYPE_ERROR) | 68 CFilterElementHide::CFilterElementHide(const std::wstring& filterText) |
| 69 : m_filterText(filterText), m_type(ETraverserComplexType::TRAVERSER_TYPE_ERROR
) |
62 { | 70 { |
| 71 std::wstring filterSuffix(filterText); // The unparsed remainder of the input
filter text |
| 72 |
63 // Find tag name, class or any (*) | 73 // Find tag name, class or any (*) |
64 CString filterString = filterText; | 74 wchar_t firstTag = filterText[0]; |
65 | |
66 wchar_t firstTag = filterString.GetAt(0); | |
67 // Any tag | |
68 if (firstTag == '*') | 75 if (firstTag == '*') |
69 { | 76 { |
70 filterString = filterString.Mid(1); | 77 // Any tag |
| 78 filterSuffix = filterSuffix.substr(1); |
71 } | 79 } |
72 // Any tag (implicitely) | |
73 else if (firstTag == '[' || firstTag == '.' || firstTag == '#') | 80 else if (firstTag == '[' || firstTag == '.' || firstTag == '#') |
74 { | 81 { |
| 82 // Any tag (implicitly) |
75 } | 83 } |
76 // Real tag | |
77 else if (isalnum(firstTag)) | 84 else if (isalnum(firstTag)) |
78 { | 85 { |
79 //TODO: Add support for descendant selectors | 86 // Real tag |
80 int pos = filterString.FindOneOf(L".#[("); | 87 //TODO: Add support for descendant selectors |
81 | 88 auto pos = filterSuffix.find_first_of(L".#[("); |
82 if (pos < 0) | 89 if (pos == std::wstring::npos) |
83 pos = filterString.GetLength(); | 90 { |
84 m_tag = filterString.Left(pos).MakeLower(); | 91 pos = filterSuffix.length(); |
85 | 92 } |
86 filterString = filterString.Mid(pos); | 93 m_tag = ToLowerString(filterSuffix.substr(0, pos)); |
| 94 filterSuffix = filterSuffix.substr(pos); |
87 } | 95 } |
88 // Error | |
89 else | 96 else |
90 { | 97 { |
91 DEBUG_FILTER("Filter::Error parsing selector:" + filterText + " (invalid tag
)"); | 98 // Error |
92 throw std::runtime_error(CStringA("Filter::Error parsing selector:" + filter
Text + " (invalid tag)").GetString()); | 99 throw ElementHideParseException(filterText, "invalid tag"); |
93 } | 100 } |
94 | 101 |
95 // Find Id and class name | 102 // Find Id and class name |
96 | 103 if (!filterSuffix.empty()) |
97 if (!filterString.IsEmpty()) | |
98 { | 104 { |
99 wchar_t firstId = filterString.GetAt(0); | 105 wchar_t firstId = filterSuffix[0]; |
100 | 106 |
101 // Id | 107 // Id |
102 if (firstId == '#') | 108 if (firstId == '#') |
103 { | 109 { |
104 int pos = filterString.Find('['); | 110 auto pos = filterSuffix.find('['); |
105 if (pos < 0) | 111 if (pos == std::wstring::npos) |
106 { | 112 { |
107 pos = filterString.GetLength(); | 113 pos = filterSuffix.length(); |
108 } | 114 } |
109 m_tagId = filterString.Mid(1, pos - 1); | 115 m_tagId = filterSuffix.substr(1, pos - 1); |
110 filterString = filterString.Mid(pos); | 116 filterSuffix = filterSuffix.substr(pos); |
111 pos = m_tagId.Find(L"."); | 117 pos = m_tagId.find(L"."); |
112 if (pos > 0) | 118 if (pos != std::wstring::npos) |
113 { | 119 { |
114 m_tagClassName = m_tagId.Mid(pos + 1); | 120 if (pos == 0) |
115 m_tagId = m_tagId.Left(pos); | 121 { |
| 122 throw ElementHideParseException(filterText, "empty tag id"); |
| 123 } |
| 124 m_tagClassName = m_tagId.substr(pos + 1); |
| 125 m_tagId = m_tagId.substr(0, pos); |
116 } | 126 } |
117 } | 127 } |
118 // Class name | 128 // Class name |
119 else if (firstId == '.') | 129 else if (firstId == '.') |
120 { | 130 { |
121 int pos = filterString.Find('['); | 131 auto pos = filterSuffix.find('['); |
122 if (pos < 0) | 132 if (pos == std::wstring::npos) |
123 { | 133 { |
124 pos = filterString.GetLength(); | 134 pos = filterSuffix.length(); |
125 } | 135 } |
126 m_tagClassName = filterString.Mid(1, pos - 1); | 136 m_tagClassName = filterSuffix.substr(1, pos - 1); |
127 filterString = filterString.Mid(pos); | 137 filterSuffix = filterSuffix.substr(pos); |
128 } | 138 } |
129 } | 139 } |
130 | 140 |
131 char chAttrStart = '['; | 141 while (!filterSuffix.empty()) |
132 char chAttrEnd = ']'; | 142 { |
| 143 if (filterSuffix[0] != '[') |
| 144 { |
| 145 throw ElementHideParseException(filterText, "expected '['"); |
| 146 } |
| 147 auto endPos = filterSuffix.find(']') ; |
| 148 if (endPos == std::wstring::npos) |
| 149 { |
| 150 throw ElementHideParseException(filterText, "expected ']'"); |
| 151 } |
| 152 std::wstring arg = filterSuffix.substr(1, endPos - 1); |
| 153 filterSuffix = filterSuffix.substr(endPos + 1); |
133 | 154 |
134 while (!filterString.IsEmpty()) | 155 CFilterElementHideAttrSelector attrSelector; |
135 { | 156 auto posEquals = arg.find('='); |
136 if (filterString.GetAt(0) != chAttrStart) | 157 if (posEquals != std::wstring::npos) |
137 { | 158 { |
138 DEBUG_FILTER("Filter::Error parsing selector:" + filterText + " (more data
)"); | 159 if (posEquals == 0) |
139 throw std::runtime_error(CStringA("Filter::Error parsing selector:" + filt
erText + " (more data)").GetString()); | |
140 } | |
141 int endPos = filterString.Find(']') ; | |
142 if (endPos < 0) | |
143 { | |
144 DEBUG_FILTER("Filter::Error parsing selector:" + filterText + " (more data
)"); | |
145 throw std::runtime_error(CStringA("Filter::Error parsing selector:" + filt
erText + " (more data)").GetString()); | |
146 } | |
147 | |
148 CFilterElementHideAttrSelector attrSelector; | |
149 | |
150 CString arg = filterString.Mid(1, endPos - 1); | |
151 filterString = filterString.Mid(endPos + 1); | |
152 | |
153 int delimiterPos = arg.Find('='); | |
154 if (delimiterPos > 0) | |
155 { | |
156 attrSelector.m_value = arg.Mid(delimiterPos + 1); | |
157 if (attrSelector.m_value.GetLength() >= 2 && attrSelector.m_value.GetAt(0)
== '\"' && attrSelector.m_value.GetAt(attrSelector.m_value.GetLength() - 1) ==
'\"') | |
158 { | 160 { |
159 attrSelector.m_value = attrSelector.m_value.Mid(1, attrSelector.m_value.
GetLength() - 2); | 161 throw ElementHideParseException(filterText, "empty attribute name before
'='"); |
| 162 } |
| 163 attrSelector.m_value = arg.substr(posEquals + 1); |
| 164 if (attrSelector.m_value.length() >= 2 && attrSelector.m_value[0] == '\"'
&& attrSelector.m_value[attrSelector.m_value.length() - 1] == '\"') |
| 165 { |
| 166 attrSelector.m_value = attrSelector.m_value.substr(1, attrSelector.m_val
ue.length() - 2); |
160 } | 167 } |
161 | 168 |
162 if (arg.GetAt(delimiterPos - 1) == '^') | 169 if (arg[posEquals - 1] == '^') |
163 { | 170 { |
164 attrSelector.m_bstrAttr = arg.Left(delimiterPos - 1); | |
165 attrSelector.m_pos = CFilterElementHideAttrPos::STARTING; | 171 attrSelector.m_pos = CFilterElementHideAttrPos::STARTING; |
166 } | 172 } |
167 else if (arg.GetAt(delimiterPos - 1) == '*') | 173 else if (arg[posEquals - 1] == '*') |
168 { | 174 { |
169 attrSelector.m_bstrAttr = arg.Left(delimiterPos - 1); | |
170 attrSelector.m_pos = CFilterElementHideAttrPos::ANYWHERE; | 175 attrSelector.m_pos = CFilterElementHideAttrPos::ANYWHERE; |
171 } | 176 } |
172 else if (arg.GetAt(delimiterPos - 1) == '$') | 177 else if (arg[posEquals - 1] == '$') |
173 { | 178 { |
174 attrSelector.m_bstrAttr = arg.Left(delimiterPos - 1); | |
175 attrSelector.m_pos = CFilterElementHideAttrPos::ENDING; | 179 attrSelector.m_pos = CFilterElementHideAttrPos::ENDING; |
176 } | 180 } |
| 181 if (attrSelector.m_pos != CFilterElementHideAttrPos::POS_NONE) |
| 182 { |
| 183 if (posEquals == 1) |
| 184 { |
| 185 throw ElementHideParseException(filterText, "empty attribute name befo
re " + std::string(1, static_cast<char>(arg[0])) + "'='"); |
| 186 } |
| 187 attrSelector.m_attr = arg.substr(0, posEquals - 1); |
| 188 } |
177 else | 189 else |
178 { | 190 { |
179 attrSelector.m_bstrAttr = arg.Left(delimiterPos); | |
180 attrSelector.m_pos = CFilterElementHideAttrPos::EXACT; | 191 attrSelector.m_pos = CFilterElementHideAttrPos::EXACT; |
| 192 attrSelector.m_attr = arg.substr(0, posEquals); |
181 } | 193 } |
182 } | 194 } |
183 CString tag = attrSelector.m_bstrAttr; | 195 |
184 if (tag == "style") | 196 if (attrSelector.m_attr == L"style") |
185 { | 197 { |
186 attrSelector.m_type = CFilterElementHideAttrType::STYLE; | 198 attrSelector.m_type = CFilterElementHideAttrType::STYLE; |
187 attrSelector.m_value.MakeLower(); | 199 attrSelector.m_value = ToLowerString(attrSelector.m_value); |
188 } | 200 } |
189 else if (tag == "id") | 201 else if (attrSelector.m_attr == L"id") |
190 { | 202 { |
191 attrSelector.m_type = CFilterElementHideAttrType::ID; | 203 attrSelector.m_type = CFilterElementHideAttrType::ID; |
192 } | 204 } |
193 else if (tag == "class") | 205 else if (attrSelector.m_attr == L"class") |
194 { | 206 { |
195 attrSelector.m_type = CFilterElementHideAttrType::CLASS; | 207 attrSelector.m_type = CFilterElementHideAttrType::CLASS; |
196 } | 208 } |
197 m_attributeSelectors.push_back(attrSelector); | 209 m_attributeSelectors.push_back(attrSelector); |
198 | |
199 } | 210 } |
200 | 211 |
201 // End check | 212 // End check |
202 if (!filterString.IsEmpty()) | 213 if (!filterSuffix.empty()) |
203 { | 214 { |
204 DEBUG_FILTER("Filter::Error parsing selector:" + filterFile + "/" + filterTe
xt + " (more data)") | 215 throw ElementHideParseException(filterText, "extra characters at end"); |
205 throw new std::runtime_error(CStringA("Filter::Error parsing selector:" + f
ilterText + " (more data)").GetString()); | |
206 } | 216 } |
207 } | 217 } |
208 | 218 |
209 CFilterElementHide::CFilterElementHide(const CFilterElementHide& filter) | 219 CFilterElementHide::CFilterElementHide(const CFilterElementHide& filter) |
210 { | 220 { |
211 m_filterText = filter.m_filterText; | 221 m_filterText = filter.m_filterText; |
212 | 222 |
213 m_tagId = filter.m_tagId; | 223 m_tagId = filter.m_tagId; |
214 m_tagClassName = filter.m_tagClassName; | 224 m_tagClassName = filter.m_tagClassName; |
215 | 225 |
(...skipping 27 matching lines...) Expand all Loading... |
243 CFilter::CFilter() : m_isMatchCase(false), m_isFirstParty(false), | 253 CFilter::CFilter() : m_isMatchCase(false), m_isFirstParty(false), |
244 m_isThirdParty(false), | 254 m_isThirdParty(false), |
245 m_isFromStart(false), m_isFromEnd(false), m_hitCount(0) | 255 m_isFromStart(false), m_isFromEnd(false), m_hitCount(0) |
246 { | 256 { |
247 } | 257 } |
248 | 258 |
249 | 259 |
250 bool CFilterElementHide::IsMatchFilterElementHide(IHTMLElement* pEl) const | 260 bool CFilterElementHide::IsMatchFilterElementHide(IHTMLElement* pEl) const |
251 { | 261 { |
252 HRESULT hr; | 262 HRESULT hr; |
253 | 263 /* |
254 if (!m_tagId.IsEmpty()) | 264 * If a tag id is specified, it must match |
| 265 */ |
| 266 if (!m_tagId.empty()) |
255 { | 267 { |
256 CComBSTR id; | 268 CComBSTR idBstr; |
257 hr = pEl->get_id(&id); | 269 if (FAILED(pEl->get_id(&idBstr)) || !idBstr || m_tagId != ToWstring(idBstr)) |
258 if ((hr != S_OK) || (id != CComBSTR(m_tagId))) | |
259 { | 270 { |
260 return false; | 271 return false; |
261 } | 272 } |
262 } | 273 } |
263 if (!m_tagClassName.IsEmpty()) | 274 /* |
| 275 * If a class name is specified, it must match |
| 276 */ |
| 277 if (!m_tagClassName.empty()) |
264 { | 278 { |
265 CComBSTR classNameBSTR; | 279 CComBSTR classNameListBstr; |
266 hr = pEl->get_className(&classNameBSTR); | 280 hr = pEl->get_className(&classNameListBstr); |
267 if (hr == S_OK) | 281 if (FAILED(hr) || !classNameListBstr) |
268 { | 282 { |
269 CString className = classNameBSTR; | 283 return false; // We can't match a class name if there's no class name |
270 int start = 0; | 284 } |
271 CString specificClass; | 285 std::wstring classNameList(ToWstring(classNameListBstr)); |
272 bool foundMatch = false; | 286 if (classNameList.empty()) |
273 while ((specificClass = className.Tokenize(L" ", start)) != L"") | 287 { |
| 288 return false; |
| 289 } |
| 290 // TODO: Consider case of multiple classes. (m_tagClassName can be something
like "foo.bar") |
| 291 /* |
| 292 * Match when 'm_tagClassName' appears as a token within classNameList |
| 293 */ |
| 294 bool foundMatch = false; |
| 295 TokenSequence<std::wstring> ts(classNameList, L" "); |
| 296 for (auto iterClassName = ts.cbegin(); iterClassName != ts.cend(); ++iterCla
ssName) |
| 297 { |
| 298 if (*iterClassName == m_tagClassName) |
274 { | 299 { |
275 // TODO: Consider case of multiple classes. (m_tagClassName can be somet
hing like "foo.bar") | 300 foundMatch = true; |
276 if (specificClass == m_tagClassName) | 301 break; |
277 { | |
278 foundMatch = true; | |
279 } | |
280 } | |
281 if (!foundMatch) | |
282 { | |
283 return false; | |
284 } | 302 } |
285 } | 303 } |
286 } | 304 if (!foundMatch) |
287 if (!m_tag.IsEmpty()) | |
288 { | |
289 CComBSTR tagName; | |
290 hr = pEl->get_tagName(&tagName); | |
291 tagName.ToLower(); | |
292 if ((hr != S_OK) || (tagName != CComBSTR(m_tag))) | |
293 { | 305 { |
294 return false; | 306 return false; |
295 } | 307 } |
296 } | 308 } |
297 | 309 /* |
298 // Check attributes | 310 * If a tag name is specified, it must match |
299 for (std::vector<CFilterElementHideAttrSelector>::const_iterator attrIt = m_at
tributeSelectors.begin(); | 311 */ |
300 attrIt != m_attributeSelectors.end(); ++ attrIt) | 312 if (!m_tag.empty()) |
301 { | 313 { |
302 ATL::CString value; | 314 CComBSTR tagNameBstr; |
| 315 if (FAILED(pEl->get_tagName(&tagNameBstr)) || !tagNameBstr) |
| 316 { |
| 317 return false; |
| 318 } |
| 319 if (m_tag != ToLowerString(ToWstring(tagNameBstr))) |
| 320 { |
| 321 return false; |
| 322 } |
| 323 } |
| 324 /* |
| 325 * Match each attribute |
| 326 */ |
| 327 for (auto attrIt = m_attributeSelectors.begin(); attrIt != m_attributeSelector
s.end(); ++attrIt) |
| 328 { |
| 329 std::wstring value; |
303 bool attrFound = false; | 330 bool attrFound = false; |
304 if (attrIt->m_type == CFilterElementHideAttrType::STYLE) | 331 if (attrIt->m_type == CFilterElementHideAttrType::STYLE) |
305 { | 332 { |
306 CComPtr<IHTMLStyle> pStyle; | 333 CComPtr<IHTMLStyle> pStyle; |
307 if (SUCCEEDED(pEl->get_style(&pStyle)) && pStyle) | 334 if (SUCCEEDED(pEl->get_style(&pStyle)) && pStyle) |
308 { | 335 { |
309 CComBSTR bstrStyle; | 336 CComBSTR styleBstr; |
310 | 337 if (SUCCEEDED(pStyle->get_cssText(&styleBstr)) && styleBstr) |
311 if (SUCCEEDED(pStyle->get_cssText(&bstrStyle)) && bstrStyle) | |
312 { | 338 { |
313 value = bstrStyle; | 339 value = ToLowerString(ToWstring(styleBstr)); |
314 value.MakeLower(); | |
315 attrFound = true; | 340 attrFound = true; |
316 } | 341 } |
317 } | 342 } |
318 } | 343 } |
319 else if (attrIt->m_type == CFilterElementHideAttrType::CLASS) | 344 else if (attrIt->m_type == CFilterElementHideAttrType::CLASS) |
320 { | 345 { |
321 CComBSTR bstrClassNames; | 346 CComBSTR classNamesBstr; |
322 if (SUCCEEDED(pEl->get_className(&bstrClassNames)) && bstrClassNames) | 347 if (SUCCEEDED(pEl->get_className(&classNamesBstr)) && classNamesBstr) |
323 { | 348 { |
324 value = bstrClassNames; | 349 value = ToWstring(classNamesBstr); |
325 attrFound = true; | 350 attrFound = true; |
326 } | 351 } |
327 } | 352 } |
328 else if (attrIt->m_type == CFilterElementHideAttrType::ID) | 353 else if (attrIt->m_type == CFilterElementHideAttrType::ID) |
329 { | 354 { |
330 CComBSTR bstrId; | 355 CComBSTR idBstr; |
331 if (SUCCEEDED(pEl->get_id(&bstrId)) && bstrId) | 356 if (SUCCEEDED(pEl->get_id(&idBstr)) && idBstr) |
332 { | 357 { |
333 value = bstrId; | 358 value = ToWstring(idBstr); |
334 attrFound = true; | 359 attrFound = true; |
335 } | 360 } |
336 } | 361 } |
337 else | 362 else |
338 { | 363 { |
339 auto attributeValue = GetHtmlElementAttribute(*pEl, attrIt->m_bstrAttr); | 364 CComBSTR attrArgument(attrIt->m_attr.length(), attrIt->m_attr.c_str()); |
340 if (attrFound = attributeValue.isAttributeFound) | 365 auto x = GetHtmlElementAttribute(*pEl, attrArgument); |
| 366 attrFound = x.isAttributeFound; |
| 367 if (attrFound) |
341 { | 368 { |
342 value = ToCString(attributeValue.attributeValue); | 369 value = x.attributeValue; |
343 } | 370 } |
344 } | 371 } |
345 | 372 |
346 if (attrFound) | 373 if (attrFound) |
347 { | 374 { |
348 if (attrIt->m_pos == CFilterElementHideAttrPos::EXACT) | 375 if (attrIt->m_pos == CFilterElementHideAttrPos::EXACT) |
349 { | 376 { |
350 // TODO: IE rearranges the style attribute completely. Figure out if any
thing can be done about it. | 377 // TODO: IE rearranges the style attribute completely. Figure out if any
thing can be done about it. |
351 if (value != attrIt->m_value) | 378 if (value != attrIt->m_value) |
352 return false; | 379 return false; |
353 } | 380 } |
354 else if (attrIt->m_pos == CFilterElementHideAttrPos::STARTING) | 381 else if (attrIt->m_pos == CFilterElementHideAttrPos::STARTING) |
355 { | 382 { |
356 if (value.Left(attrIt->m_value.GetLength()) != attrIt->m_value) | 383 if (value.compare(0, attrIt->m_value.length(), attrIt->m_value) != 0) |
357 return false; | 384 return false; |
358 } | 385 } |
359 else if (attrIt->m_pos == CFilterElementHideAttrPos::ENDING) | 386 else if (attrIt->m_pos == CFilterElementHideAttrPos::ENDING) |
360 { | 387 { |
361 if (value.Right(attrIt->m_value.GetLength()) != attrIt->m_value) | 388 size_t valueLength = value.length(); |
| 389 size_t attrLength = attrIt->m_value.length(); |
| 390 if (valueLength < attrLength) |
| 391 return false; |
| 392 if (value.compare(valueLength - attrLength, attrLength, attrIt->m_value)
!= 0) |
362 return false; | 393 return false; |
363 } | 394 } |
364 else if (attrIt->m_pos == CFilterElementHideAttrPos::ANYWHERE) | 395 else if (attrIt->m_pos == CFilterElementHideAttrPos::ANYWHERE) |
365 { | 396 { |
366 if (value.Find(attrIt->m_value) < 0) | 397 if (value.find(attrIt->m_value) == std::wstring::npos) |
367 return false; | 398 return false; |
368 } | 399 } |
369 else if (attrIt->m_value.IsEmpty()) | 400 else if (attrIt->m_value.empty()) |
370 { | 401 { |
371 return true; | 402 return true; |
372 } | 403 } |
373 } | 404 } |
374 else | 405 else |
375 { | 406 { |
376 return false; | 407 return false; |
377 } | 408 } |
378 } | 409 } |
379 | 410 |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
416 | 447 |
417 return true; | 448 return true; |
418 } | 449 } |
419 | 450 |
420 | 451 |
421 | 452 |
422 // ============================================================================ | 453 // ============================================================================ |
423 // CPluginFilter | 454 // CPluginFilter |
424 // ============================================================================ | 455 // ============================================================================ |
425 | 456 |
426 CPluginFilter::CPluginFilter(const CString& dataPath) : m_dataPath(dataPath) | 457 CPluginFilter::CPluginFilter(const std::wstring& dataPath) |
| 458 : m_dataPath(dataPath) |
427 { | 459 { |
428 ClearFilters(); | 460 ClearFilters(); |
429 } | 461 } |
430 | 462 |
431 | 463 |
432 bool CPluginFilter::AddFilterElementHide(CString filterText) | 464 bool CPluginFilter::AddFilterElementHide(std::wstring filterText) |
433 { | 465 { |
434 DEBUG_FILTER("Input: " + filterText + " filterFile" + filterFile); | 466 DEBUG_FILTER(L"Input: " + filterText + L" filterFile" + filterFile); |
435 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); | 467 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); |
436 { | 468 { |
437 CString filterString = filterText; | |
438 // Create filter descriptor | 469 // Create filter descriptor |
439 std::auto_ptr<CFilterElementHide> filter; | 470 std::auto_ptr<CFilterElementHide> filter; |
440 | |
441 CString wholeFilterString = filterString; | |
442 wchar_t separatorChar; | 471 wchar_t separatorChar; |
443 do | 472 do |
444 { | 473 { |
445 int chunkEnd = filterText.FindOneOf(L"+>"); | 474 auto chunkEnd = filterText.find_first_of(L"+>"); |
446 if (chunkEnd > 0) | 475 if (chunkEnd != std::wstring::npos && chunkEnd > 0) |
447 { | 476 { |
448 separatorChar = filterText.GetAt(chunkEnd); | 477 separatorChar = filterText[chunkEnd]; |
449 } | 478 } |
450 else | 479 else |
451 { | 480 { |
452 chunkEnd = filterText.GetLength(); | 481 chunkEnd = filterText.length(); |
453 separatorChar = L'\0'; | 482 separatorChar = L'\0'; |
454 } | 483 } |
455 | 484 |
456 CString filterChunk = filterText.Left(chunkEnd).TrimRight(); | |
457 std::auto_ptr<CFilterElementHide> filterParent(filter); | 485 std::auto_ptr<CFilterElementHide> filterParent(filter); |
458 | 486 filter.reset(new CFilterElementHide(TrimStringRight(filterText.substr(0, c
hunkEnd)))); |
459 filter.reset(new CFilterElementHide(filterChunk)); | |
460 | |
461 if (filterParent.get() != 0) | 487 if (filterParent.get() != 0) |
462 { | 488 { |
463 filter->m_predecessor.reset(filterParent.release()); | 489 filter->m_predecessor.reset(filterParent.release()); |
464 } | 490 } |
465 | 491 |
466 if (separatorChar != L'\0') // complex selector | 492 if (separatorChar != L'\0') // complex selector |
467 { | 493 { |
468 filterText = filterText.Mid(chunkEnd + 1).TrimLeft(); | 494 filterText = TrimStringLeft(filterText.substr(chunkEnd + 1)); |
469 if (separatorChar == '+') | 495 if (separatorChar == '+') |
470 filter->m_type = CFilterElementHide::TRAVERSER_TYPE_IMMEDIATE; | 496 filter->m_type = CFilterElementHide::TRAVERSER_TYPE_IMMEDIATE; |
471 else if (separatorChar == '>') | 497 else if (separatorChar == '>') |
472 filter->m_type = CFilterElementHide::TRAVERSER_TYPE_PARENT; | 498 filter->m_type = CFilterElementHide::TRAVERSER_TYPE_PARENT; |
473 } | 499 } |
474 else // Terminating element (simple selector) | 500 else // Terminating element (simple selector) |
475 { | 501 { |
476 if (!filter->m_tagId.IsEmpty()) | 502 if (!filter->m_tagId.empty()) |
477 { | 503 { |
478 m_elementHideTagsId.insert(std::make_pair(std::make_pair(filter->m_tag
, filter->m_tagId), *filter)); | 504 m_elementHideTagsId.insert(std::make_pair(std::make_pair(filter->m_tag
, filter->m_tagId), *filter)); |
479 } | 505 } |
480 else if (!filter->m_tagClassName.IsEmpty()) | 506 else if (!filter->m_tagClassName.empty()) |
481 { | 507 { |
482 m_elementHideTagsClass.insert(std::make_pair(std::make_pair(filter->m_
tag, filter->m_tagClassName), *filter)); | 508 m_elementHideTagsClass.insert(std::make_pair(std::make_pair(filter->m_
tag, filter->m_tagClassName), *filter)); |
483 } | 509 } |
484 else | 510 else |
485 { | 511 { |
486 std::pair<CString, CFilterElementHide> pair = std::make_pair(filter->m
_tag, *filter); | 512 m_elementHideTags.insert(std::make_pair(filter->m_tag, *filter)); |
487 m_elementHideTags.insert(pair); | |
488 } | 513 } |
489 } | 514 } |
490 } while (separatorChar != '\0'); | 515 } while (separatorChar != '\0'); |
491 } | 516 } |
492 | 517 |
493 return true; | 518 return true; |
494 } | 519 } |
495 | 520 |
496 bool CPluginFilter::IsElementHidden(const std::wstring& tag, IHTMLElement* pEl,
const std::wstring& domain, const std::wstring& indent) const | 521 bool CPluginFilter::IsElementHidden(const std::wstring& tag, IHTMLElement* pEl,
const std::wstring& domain, const std::wstring& indent) const |
497 { | 522 { |
498 CString tagCString = ToCString(tag); | 523 std::wstring id; |
499 | 524 CComBSTR idBstr; |
500 CString id; | 525 if (SUCCEEDED(pEl->get_id(&idBstr)) && idBstr) |
501 CComBSTR bstrId; | |
502 if (SUCCEEDED(pEl->get_id(&bstrId)) && bstrId) | |
503 { | 526 { |
504 id = bstrId; | 527 id = ToWstring(idBstr); |
505 } | 528 } |
506 | 529 std::wstring classNames; |
507 CString classNames; | 530 CComBSTR classNamesBstr; |
508 CComBSTR bstrClassNames; | 531 if (SUCCEEDED(pEl->get_className(&classNamesBstr)) && classNamesBstr) |
509 if (SUCCEEDED(pEl->get_className(&bstrClassNames)) && bstrClassNames) | |
510 { | 532 { |
511 classNames = bstrClassNames; | 533 classNames = ToWstring(classNamesBstr); |
512 } | 534 } |
513 | 535 |
514 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); | 536 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); |
515 { | 537 { |
516 // Search tag/id filters | 538 // Search tag/id filters |
517 if (!id.IsEmpty()) | 539 if (!id.empty()) |
518 { | 540 { |
519 std::pair<TFilterElementHideTagsNamed::const_iterator, TFilterElementHideT
agsNamed::const_iterator> idItEnum = | 541 auto idItEnum = m_elementHideTagsId.equal_range(std::make_pair(tag, id)); |
520 m_elementHideTagsId.equal_range(std::make_pair(tagCString, id)); | 542 for (auto idIt = idItEnum.first; idIt != idItEnum.second; ++idIt) |
521 for (TFilterElementHideTagsNamed::const_iterator idIt = idItEnum.first; id
It != idItEnum.second; idIt ++) | |
522 { | 543 { |
523 if (idIt->second.IsMatchFilterElementHide(pEl)) | 544 if (idIt->second.IsMatchFilterElementHide(pEl)) |
524 { | 545 { |
525 #ifdef ENABLE_DEBUG_RESULT | 546 #ifdef ENABLE_DEBUG_RESULT |
526 DEBUG_HIDE_EL(indent + "HideEl::Found (tag/id) filter:" + idIt->second
.m_filterText) | 547 DEBUG_HIDE_EL(indent + L"HideEl::Found (tag/id) filter:" + idIt->secon
d.m_filterText); |
527 CPluginDebug::DebugResultHiding(tag, L"id:" + ToWstring(id), ToWstri
ng(idIt->second.m_filterText)); | 548 CPluginDebug::DebugResultHiding(tag, L"id:" + id, idIt->second.m_filte
rText); |
528 #endif | 549 #endif |
529 return true; | 550 return true; |
530 } | 551 } |
531 } | 552 } |
532 | 553 |
533 // Search general id | 554 // Search general id |
534 idItEnum = m_elementHideTagsId.equal_range(std::make_pair("", id)); | 555 idItEnum = m_elementHideTagsId.equal_range(std::make_pair(L"", id)); |
535 for (TFilterElementHideTagsNamed::const_iterator idIt = idItEnum.first; id
It != idItEnum.second; idIt ++) | 556 for (auto idIt = idItEnum.first; idIt != idItEnum.second; ++idIt) |
536 { | 557 { |
537 if (idIt->second.IsMatchFilterElementHide(pEl)) | 558 if (idIt->second.IsMatchFilterElementHide(pEl)) |
538 { | 559 { |
539 #ifdef ENABLE_DEBUG_RESULT | 560 #ifdef ENABLE_DEBUG_RESULT |
540 DEBUG_HIDE_EL(indent + "HideEl::Found (?/id) filter:" + idIt->second.m
_filterText) | 561 DEBUG_HIDE_EL(indent + L"HideEl::Found (?/id) filter:" + idIt->second.
m_filterText); |
541 CPluginDebug::DebugResultHiding(tag, L"id:" + ToWstring(id), ToWstri
ng(idIt->second.m_filterText)); | 562 CPluginDebug::DebugResultHiding(tag, L"id:" + id, idIt->second.m_filte
rText); |
542 #endif | 563 #endif |
543 return true; | 564 return true; |
544 } | 565 } |
545 } | 566 } |
546 } | 567 } |
547 | 568 |
548 // Search tag/className filters | 569 // Search tag/className filters |
549 if (!classNames.IsEmpty()) | 570 if (!classNames.empty()) |
550 { | 571 { |
551 int pos = 0; | 572 TokenSequence<std::wstring> ts(classNames, L" \t\n\r"); |
552 CString className = classNames.Tokenize(L" \t\n\r", pos); | 573 for (auto iterClassName = ts.cbegin(); iterClassName != ts.cend(); ++iterC
lassName) |
553 while (pos >= 0) | |
554 { | 574 { |
555 std::pair<TFilterElementHideTagsNamed::const_iterator, TFilterElementHid
eTagsNamed::const_iterator> classItEnum = | 575 std::wstring className = *iterClassName; |
556 m_elementHideTagsClass.equal_range(std::make_pair(tagCString, classNam
e)); | 576 auto classItEnum = m_elementHideTagsClass.equal_range(std::make_pair(tag
, className)); |
557 | 577 for (auto classIt = classItEnum.first; classIt != classItEnum.second; ++
classIt) |
558 for (TFilterElementHideTagsNamed::const_iterator classIt = classItEnum.f
irst; classIt != classItEnum.second; ++classIt) | |
559 { | 578 { |
560 if (classIt->second.IsMatchFilterElementHide(pEl)) | 579 if (classIt->second.IsMatchFilterElementHide(pEl)) |
561 { | 580 { |
562 #ifdef ENABLE_DEBUG_RESULT | 581 #ifdef ENABLE_DEBUG_RESULT |
563 DEBUG_HIDE_EL(indent + "HideEl::Found (tag/class) filter:" + classIt
->second.m_filterText) | 582 DEBUG_HIDE_EL(indent + L"HideEl::Found (tag/class) filter:" + classI
t->second.m_filterText); |
564 CPluginDebug::DebugResultHiding(tag, L"class:" + ToWstring(classNa
me), ToWstring(classIt->second.m_filterText)); | 583 CPluginDebug::DebugResultHiding(tag, L"class:" + className, classIt-
>second.m_filterText); |
565 #endif | 584 #endif |
566 return true; | 585 return true; |
567 } | 586 } |
568 } | 587 } |
569 | 588 |
570 // Search general class name | 589 // Search general class name |
571 classItEnum = m_elementHideTagsClass.equal_range(std::make_pair("", clas
sName)); | 590 classItEnum = m_elementHideTagsClass.equal_range(std::make_pair(L"", cla
ssName)); |
572 for (TFilterElementHideTagsNamed::const_iterator classIt = classItEnum.f
irst; classIt != classItEnum.second; ++ classIt) | 591 for (auto classIt = classItEnum.first; classIt != classItEnum.second; ++
classIt) |
573 { | 592 { |
574 if (classIt->second.IsMatchFilterElementHide(pEl)) | 593 if (classIt->second.IsMatchFilterElementHide(pEl)) |
575 { | 594 { |
576 #ifdef ENABLE_DEBUG_RESULT | 595 #ifdef ENABLE_DEBUG_RESULT |
577 DEBUG_HIDE_EL(indent + L"HideEl::Found (?/class) filter:" + ToWStrin
g(classIt->second.m_filterText)); | 596 DEBUG_HIDE_EL(indent + L"HideEl::Found (?/class) filter:" + classIt-
>second.m_filterText); |
578 CPluginDebug::DebugResultHiding(tag, L"class:" + ToWstring(className
), ToWstring(classIt->second.m_filterText)); | 597 CPluginDebug::DebugResultHiding(tag, L"class:" + className, classIt-
>second.m_filterText); |
579 #endif | 598 #endif |
580 return true; | 599 return true; |
581 } | 600 } |
582 } | 601 } |
583 | |
584 // Next class name | |
585 className = classNames.Tokenize(L" \t\n\r", pos); | |
586 } | 602 } |
587 } | 603 } |
588 | 604 |
589 // Search tag filters | 605 // Search tag filters |
590 std::pair<TFilterElementHideTags::const_iterator, TFilterElementHideTags::co
nst_iterator> tagItEnum | 606 auto tagItEnum = m_elementHideTags.equal_range(tag); |
591 = m_elementHideTags.equal_range(tagCString); | 607 for (auto tagIt = tagItEnum.first; tagIt != tagItEnum.second; ++tagIt) |
592 for (TFilterElementHideTags::const_iterator tagIt = tagItEnum.first; tagIt !
= tagItEnum.second; ++ tagIt) | |
593 { | 608 { |
594 if (tagIt->second.IsMatchFilterElementHide(pEl)) | 609 if (tagIt->second.IsMatchFilterElementHide(pEl)) |
595 { | 610 { |
596 #ifdef ENABLE_DEBUG_RESULT | 611 #ifdef ENABLE_DEBUG_RESULT |
597 DEBUG_HIDE_EL(indent + "HideEl::Found (tag) filter:" + tagIt->second.m_f
ilterText) | 612 DEBUG_HIDE_EL(indent + L"HideEl::Found (tag) filter:" + tagIt->second.m_
filterText); |
598 CPluginDebug::DebugResultHiding(tag, L"-", ToWstring(tagIt->second.m_f
ilterText)); | 613 CPluginDebug::DebugResultHiding(tag, L"-", tagIt->second.m_filterText); |
599 #endif | 614 #endif |
600 return true; | 615 return true; |
601 } | 616 } |
602 } | 617 } |
603 } | 618 } |
604 | 619 |
605 return false; | 620 return false; |
606 } | 621 } |
607 | 622 |
608 bool CPluginFilter::LoadHideFilters(std::vector<std::wstring> filters) | 623 bool CPluginFilter::LoadHideFilters(std::vector<std::wstring> filters) |
609 { | 624 { |
610 ClearFilters(); | 625 ClearFilters(); |
611 bool isRead = false; | 626 bool isRead = false; |
612 CPluginClient* client = CPluginClient::GetInstance(); | 627 CPluginClient* client = CPluginClient::GetInstance(); |
613 | 628 |
614 // Parse hide string | 629 // Parse hide string |
615 int pos = 0; | 630 int pos = 0; |
616 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); | 631 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); |
617 { | 632 { |
618 for (std::vector<std::wstring>::iterator it = filters.begin(); it < filters.
end(); ++it) | 633 for (auto it = filters.begin(); it < filters.end(); ++it) |
619 { | 634 { |
620 CString filter((*it).c_str()); | 635 std::wstring filter(TrimString(*it)); |
621 // If the line is not commented out | 636 // If the line is not commented out |
622 if (!filter.Trim().IsEmpty() && filter.GetAt(0) != '!' && filter.GetAt(0)
!= '[') | 637 if (!filter.empty() && filter[0] != '!' && filter[0] != '[') |
623 { | 638 { |
624 int filterType = 0; | 639 int filterType = 0; |
625 | 640 |
626 // See http://adblockplus.org/en/filters for further documentation | 641 // See http://adblockplus.org/en/filters for further documentation |
627 | 642 |
628 try | 643 try |
629 { | 644 { |
630 AddFilterElementHide(filter); | 645 AddFilterElementHide(filter); |
631 } | 646 } |
632 catch(...) | 647 catch(...) |
633 { | 648 { |
634 #ifdef ENABLE_DEBUG_RESULT | 649 #ifdef ENABLE_DEBUG_RESULT |
635 CPluginDebug::DebugResult(L"Error loading hide filter: " + ToWstring(f
ilter)); | 650 CPluginDebug::DebugResult(L"Error loading hide filter: " + filter); |
636 #endif | 651 #endif |
637 } | 652 } |
638 } | 653 } |
639 } | 654 } |
640 } | 655 } |
641 | 656 |
642 return isRead; | 657 return isRead; |
643 } | 658 } |
644 | 659 |
645 void CPluginFilter::ClearFilters() | 660 void CPluginFilter::ClearFilters() |
(...skipping 30 matching lines...) Expand all Loading... |
676 CPluginDebug::DebugResultBlocking(type, srcTrimmed, domain); | 691 CPluginDebug::DebugResultBlocking(type, srcTrimmed, domain); |
677 } | 692 } |
678 else | 693 else |
679 { | 694 { |
680 CPluginDebug::DebugResultIgnoring(type, srcTrimmed, domain); | 695 CPluginDebug::DebugResultIgnoring(type, srcTrimmed, domain); |
681 } | 696 } |
682 } | 697 } |
683 #endif | 698 #endif |
684 return result; | 699 return result; |
685 } | 700 } |
OLD | NEW |