| OLD | NEW |
| 1 #include "PluginStdAfx.h" | 1 #include "PluginStdAfx.h" |
| 2 | 2 |
| 3 #include "PluginFilter.h" | 3 #include "PluginFilter.h" |
| 4 | 4 |
| 5 #if (defined PRODUCT_ADBLOCKPLUS) | 5 #if (defined PRODUCT_ADBLOCKPLUS) |
| 6 #include "PluginSettings.h" | 6 #include "PluginSettings.h" |
| 7 #include "PluginClient.h" | 7 #include "PluginClient.h" |
| 8 #include "PluginClientFactory.h" | 8 #include "PluginClientFactory.h" |
| 9 #endif | 9 #endif |
| 10 | 10 |
| (...skipping 486 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 497 std::pair<CString, CFilterElementHide> pair = std::make_pair(filter->m
_tag, *filter); | 497 std::pair<CString, CFilterElementHide> pair = std::make_pair(filter->m
_tag, *filter); |
| 498 m_elementHideTags.insert(pair); | 498 m_elementHideTags.insert(pair); |
| 499 } | 499 } |
| 500 } | 500 } |
| 501 } while (separatorChar != '\0'); | 501 } while (separatorChar != '\0'); |
| 502 } | 502 } |
| 503 | 503 |
| 504 return true; | 504 return true; |
| 505 } | 505 } |
| 506 | 506 |
| 507 bool CPluginFilter::IsElementHidden(const CString& tag, IHTMLElement* pEl, const
CString& domain, const CString& indent) const | 507 bool CPluginFilter::IsElementHidden(const std::wstring& tag, IHTMLElement* pEl,
const std::wstring& domain, const std::wstring& indent) const |
| 508 { | 508 { |
| 509 CString tagCString = to_CString(tag); |
| 510 |
| 509 CString id; | 511 CString id; |
| 510 CComBSTR bstrId; | 512 CComBSTR bstrId; |
| 511 if (SUCCEEDED(pEl->get_id(&bstrId)) && bstrId) | 513 if (SUCCEEDED(pEl->get_id(&bstrId)) && bstrId) |
| 512 { | 514 { |
| 513 id = bstrId; | 515 id = bstrId; |
| 514 } | 516 } |
| 515 | 517 |
| 516 CString classNames; | 518 CString classNames; |
| 517 CComBSTR bstrClassNames; | 519 CComBSTR bstrClassNames; |
| 518 if (SUCCEEDED(pEl->get_className(&bstrClassNames)) && bstrClassNames) | 520 if (SUCCEEDED(pEl->get_className(&bstrClassNames)) && bstrClassNames) |
| 519 { | 521 { |
| 520 classNames = bstrClassNames; | 522 classNames = bstrClassNames; |
| 521 } | 523 } |
| 522 | 524 |
| 523 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); | 525 CriticalSection::Lock filterEngineLock(s_criticalSectionFilterMap); |
| 524 { | 526 { |
| 525 CString domainTest = domain; | 527 CString domainTest = to_CString(domain); |
| 526 | 528 |
| 527 // Search tag/id filters | 529 // Search tag/id filters |
| 528 if (!id.IsEmpty()) | 530 if (!id.IsEmpty()) |
| 529 { | 531 { |
| 530 std::pair<TFilterElementHideTagsNamed::const_iterator, TFilterElementHideT
agsNamed::const_iterator> idItEnum = | 532 std::pair<TFilterElementHideTagsNamed::const_iterator, TFilterElementHideT
agsNamed::const_iterator> idItEnum = |
| 531 m_elementHideTagsId.equal_range(std::make_pair(tag, id)); | 533 m_elementHideTagsId.equal_range(std::make_pair(tagCString, id)); |
| 532 for (TFilterElementHideTagsNamed::const_iterator idIt = idItEnum.first; id
It != idItEnum.second; idIt ++) | 534 for (TFilterElementHideTagsNamed::const_iterator idIt = idItEnum.first; id
It != idItEnum.second; idIt ++) |
| 533 { | 535 { |
| 534 if (idIt->second.IsMatchFilterElementHide(pEl)) | 536 if (idIt->second.IsMatchFilterElementHide(pEl)) |
| 535 { | 537 { |
| 536 #ifdef ENABLE_DEBUG_RESULT | 538 #ifdef ENABLE_DEBUG_RESULT |
| 537 DEBUG_HIDE_EL(indent + "HideEl::Found (tag/id) filter:" + idIt->second
.m_filterText) | 539 DEBUG_HIDE_EL(indent + "HideEl::Found (tag/id) filter:" + idIt->second
.m_filterText) |
| 538 CPluginDebug::DebugResultHiding(tag, "id:" + id, idIt->second.m_filt
erText); | 540 CPluginDebug::DebugResultHiding(tagCString, "id:" + id, idIt->second
.m_filterText); |
| 539 #endif | 541 #endif |
| 540 return true; | 542 return true; |
| 541 } | 543 } |
| 542 } | 544 } |
| 543 | 545 |
| 544 // Search general id | 546 // Search general id |
| 545 idItEnum = m_elementHideTagsId.equal_range(std::make_pair("", id)); | 547 idItEnum = m_elementHideTagsId.equal_range(std::make_pair("", id)); |
| 546 for (TFilterElementHideTagsNamed::const_iterator idIt = idItEnum.first; id
It != idItEnum.second; idIt ++) | 548 for (TFilterElementHideTagsNamed::const_iterator idIt = idItEnum.first; id
It != idItEnum.second; idIt ++) |
| 547 { | 549 { |
| 548 if (idIt->second.IsMatchFilterElementHide(pEl)) | 550 if (idIt->second.IsMatchFilterElementHide(pEl)) |
| 549 { | 551 { |
| 550 #ifdef ENABLE_DEBUG_RESULT | 552 #ifdef ENABLE_DEBUG_RESULT |
| 551 DEBUG_HIDE_EL(indent + "HideEl::Found (?/id) filter:" + idIt->second.m
_filterText) | 553 DEBUG_HIDE_EL(indent + "HideEl::Found (?/id) filter:" + idIt->second.m
_filterText) |
| 552 CPluginDebug::DebugResultHiding(tag, "id:" + id, idIt->second.m_filt
erText); | 554 CPluginDebug::DebugResultHiding(tagCString, "id:" + id, idIt->second
.m_filterText); |
| 553 #endif | 555 #endif |
| 554 return true; | 556 return true; |
| 555 } | 557 } |
| 556 } | 558 } |
| 557 } | 559 } |
| 558 | 560 |
| 559 // Search tag/className filters | 561 // Search tag/className filters |
| 560 if (!classNames.IsEmpty()) | 562 if (!classNames.IsEmpty()) |
| 561 { | 563 { |
| 562 int pos = 0; | 564 int pos = 0; |
| 563 CString className = classNames.Tokenize(L" \t\n\r", pos); | 565 CString className = classNames.Tokenize(L" \t\n\r", pos); |
| 564 while (pos >= 0) | 566 while (pos >= 0) |
| 565 { | 567 { |
| 566 std::pair<TFilterElementHideTagsNamed::const_iterator, TFilterElementHid
eTagsNamed::const_iterator> classItEnum = | 568 std::pair<TFilterElementHideTagsNamed::const_iterator, TFilterElementHid
eTagsNamed::const_iterator> classItEnum = |
| 567 m_elementHideTagsClass.equal_range(std::make_pair(tag, className)); | 569 m_elementHideTagsClass.equal_range(std::make_pair(tagCString, classNam
e)); |
| 568 | 570 |
| 569 for (TFilterElementHideTagsNamed::const_iterator classIt = classItEnum.f
irst; classIt != classItEnum.second; ++classIt) | 571 for (TFilterElementHideTagsNamed::const_iterator classIt = classItEnum.f
irst; classIt != classItEnum.second; ++classIt) |
| 570 { | 572 { |
| 571 if (classIt->second.IsMatchFilterElementHide(pEl)) | 573 if (classIt->second.IsMatchFilterElementHide(pEl)) |
| 572 { | 574 { |
| 573 #ifdef ENABLE_DEBUG_RESULT | 575 #ifdef ENABLE_DEBUG_RESULT |
| 574 DEBUG_HIDE_EL(indent + "HideEl::Found (tag/class) filter:" + classIt
->second.m_filterText) | 576 DEBUG_HIDE_EL(indent + "HideEl::Found (tag/class) filter:" + classIt
->second.m_filterText) |
| 575 CPluginDebug::DebugResultHiding(tag, "class:" + className, classIt
->second.m_filterText); | 577 CPluginDebug::DebugResultHiding(tagCString, "class:" + className,
classIt->second.m_filterText); |
| 576 #endif | 578 #endif |
| 577 return true; | 579 return true; |
| 578 } | 580 } |
| 579 } | 581 } |
| 580 | 582 |
| 581 // Search general class name | 583 // Search general class name |
| 582 classItEnum = m_elementHideTagsClass.equal_range(std::make_pair("", clas
sName)); | 584 classItEnum = m_elementHideTagsClass.equal_range(std::make_pair("", clas
sName)); |
| 583 for (TFilterElementHideTagsNamed::const_iterator classIt = classItEnum.f
irst; classIt != classItEnum.second; ++ classIt) | 585 for (TFilterElementHideTagsNamed::const_iterator classIt = classItEnum.f
irst; classIt != classItEnum.second; ++ classIt) |
| 584 { | 586 { |
| 585 if (classIt->second.IsMatchFilterElementHide(pEl)) | 587 if (classIt->second.IsMatchFilterElementHide(pEl)) |
| 586 { | 588 { |
| 587 #ifdef ENABLE_DEBUG_RESULT | 589 #ifdef ENABLE_DEBUG_RESULT |
| 588 DEBUG_HIDE_EL(indent + "HideEl::Found (?/class) filter:" + classIt->
second.m_filterText) | 590 DEBUG_HIDE_EL(indent + "HideEl::Found (?/class) filter:" + classIt->
second.m_filterText) |
| 589 CPluginDebug::DebugResultHiding(tag, "class:" + className, classIt
->second.m_filterText); | 591 CPluginDebug::DebugResultHiding(tagCString, "class:" + className,
classIt->second.m_filterText); |
| 590 #endif | 592 #endif |
| 591 return true; | 593 return true; |
| 592 } | 594 } |
| 593 } | 595 } |
| 594 | 596 |
| 595 // Next class name | 597 // Next class name |
| 596 className = classNames.Tokenize(L" \t\n\r", pos); | 598 className = classNames.Tokenize(L" \t\n\r", pos); |
| 597 } | 599 } |
| 598 } | 600 } |
| 599 | 601 |
| 600 // Search tag filters | 602 // Search tag filters |
| 601 std::pair<TFilterElementHideTags::const_iterator, TFilterElementHideTags::co
nst_iterator> tagItEnum | 603 std::pair<TFilterElementHideTags::const_iterator, TFilterElementHideTags::co
nst_iterator> tagItEnum |
| 602 = m_elementHideTags.equal_range(tag); | 604 = m_elementHideTags.equal_range(tagCString); |
| 603 for (TFilterElementHideTags::const_iterator tagIt = tagItEnum.first; tagIt !
= tagItEnum.second; ++ tagIt) | 605 for (TFilterElementHideTags::const_iterator tagIt = tagItEnum.first; tagIt !
= tagItEnum.second; ++ tagIt) |
| 604 { | 606 { |
| 605 if (tagIt->second.IsMatchFilterElementHide(pEl)) | 607 if (tagIt->second.IsMatchFilterElementHide(pEl)) |
| 606 { | 608 { |
| 607 #ifdef ENABLE_DEBUG_RESULT | 609 #ifdef ENABLE_DEBUG_RESULT |
| 608 DEBUG_HIDE_EL(indent + "HideEl::Found (tag) filter:" + tagIt->second.m_f
ilterText) | 610 DEBUG_HIDE_EL(indent + "HideEl::Found (tag) filter:" + tagIt->second.m_f
ilterText) |
| 609 CPluginDebug::DebugResultHiding(tag, "-", tagIt->second.m_filterText); | 611 CPluginDebug::DebugResultHiding(tagCString, "-", tagIt->second.m_filte
rText); |
| 610 #endif | 612 #endif |
| 611 return true; | 613 return true; |
| 612 } | 614 } |
| 613 } | 615 } |
| 614 } | 616 } |
| 615 | 617 |
| 616 return false; | 618 return false; |
| 617 } | 619 } |
| 618 | 620 |
| 619 bool CPluginFilter::LoadHideFilters(std::vector<std::wstring> filters) | 621 bool CPluginFilter::LoadHideFilters(std::vector<std::wstring> filters) |
| (...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 672 } | 674 } |
| 673 m_filterMapDefault[i].clear(); | 675 m_filterMapDefault[i].clear(); |
| 674 } | 676 } |
| 675 | 677 |
| 676 m_elementHideTags.clear(); | 678 m_elementHideTags.clear(); |
| 677 m_elementHideTagsId.clear(); | 679 m_elementHideTagsId.clear(); |
| 678 m_elementHideTagsClass.clear(); | 680 m_elementHideTagsClass.clear(); |
| 679 } | 681 } |
| 680 } | 682 } |
| 681 | 683 |
| 682 bool CPluginFilter::ShouldBlock(CString src, int contentType, const CString& dom
ain, bool addDebug) const | 684 bool CPluginFilter::ShouldBlock(const std::wstring& src, int contentType, const
std::wstring& domain, bool addDebug) const |
| 683 { | 685 { |
| 686 CString srcCString = to_CString(src); |
| 687 |
| 684 // We should not block the empty string, so all filtering does not make sense | 688 // We should not block the empty string, so all filtering does not make sense |
| 685 // Therefore we just return | 689 // Therefore we just return |
| 686 if (src.Trim().IsEmpty()) | 690 if (srcCString.Trim().IsEmpty()) |
| 687 { | 691 { |
| 688 return false; | 692 return false; |
| 689 } | 693 } |
| 690 | 694 |
| 691 CPluginSettings* settings = CPluginSettings::GetInstance(); | 695 CPluginSettings* settings = CPluginSettings::GetInstance(); |
| 692 | 696 |
| 693 CString type; | 697 CString type; |
| 694 if (addDebug) | 698 if (addDebug) |
| 695 { | 699 { |
| 696 type = "OTHER"; | 700 type = "OTHER"; |
| 697 | 701 |
| 698 std::map<int,CString>::const_iterator it = m_contentMapText.find(contentType
); | 702 std::map<int,CString>::const_iterator it = m_contentMapText.find(contentType
); |
| 699 if (it != m_contentMapText.end()) | 703 if (it != m_contentMapText.end()) |
| 700 { | 704 { |
| 701 type = it->second; | 705 type = it->second; |
| 702 } | 706 } |
| 703 } | 707 } |
| 704 | 708 |
| 705 CPluginClient* client = CPluginClient::GetInstance(); | 709 CPluginClient* client = CPluginClient::GetInstance(); |
| 706 if (client->Matches(std::wstring(src), std::wstring(type), std::wstring(domain
))) | 710 if (client->Matches(to_wstring(srcCString), to_wstring(type), domain)) |
| 707 { | 711 { |
| 708 if (addDebug) | 712 if (addDebug) |
| 709 { | 713 { |
| 710 DEBUG_FILTER("Filter::ShouldBlock " + type + " YES") | 714 DEBUG_FILTER("Filter::ShouldBlock " + type + " YES") |
| 711 | 715 |
| 712 #ifdef ENABLE_DEBUG_RESULT | 716 #ifdef ENABLE_DEBUG_RESULT |
| 713 CPluginDebug::DebugResultBlocking(type, src, domain); | 717 CPluginDebug::DebugResultBlocking(type, srcCString, to_CString(domain)); |
| 714 #endif | 718 #endif |
| 715 } | 719 } |
| 716 return true; | 720 return true; |
| 717 } | 721 } |
| 718 return false; | 722 return false; |
| 719 } | 723 } |
| OLD | NEW |