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