| 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 |
| (...skipping 19 matching lines...) Expand all Loading... |
| 30 CPluginDomTraverserCacheBase() : m_elements(0) {}; | 30 CPluginDomTraverserCacheBase() : m_elements(0) {}; |
| 31 void Init() { m_elements=0; } | 31 void Init() { m_elements=0; } |
| 32 }; | 32 }; |
| 33 | 33 |
| 34 template <class T> | 34 template <class T> |
| 35 class CPluginDomTraverserBase | 35 class CPluginDomTraverserBase |
| 36 { | 36 { |
| 37 | 37 |
| 38 public: | 38 public: |
| 39 | 39 |
| 40 CPluginDomTraverserBase(CPluginTab* tab); | 40 explicit CPluginDomTraverserBase(const PluginFilterPtr& pluginFilter); |
| 41 ~CPluginDomTraverserBase(); | 41 ~CPluginDomTraverserBase(); |
| 42 | 42 |
| 43 void TraverseHeader(bool isHeaderTraversed); | 43 void TraverseHeader(bool isHeaderTraversed); |
| 44 | 44 |
| 45 void TraverseDocument(IWebBrowser2* pBrowser, const std::wstring& domain, cons
t std::wstring& documentUrl); | 45 void TraverseDocument(IWebBrowser2* pBrowser, const std::wstring& domain, cons
t std::wstring& documentUrl); |
| 46 void TraverseSubdocument(IWebBrowser2* pBrowser, const std::wstring& domain, c
onst std::wstring& documentUrl); | 46 void TraverseSubdocument(IWebBrowser2* pBrowser, const std::wstring& domain, c
onst std::wstring& documentUrl); |
| 47 | 47 |
| 48 virtual void ClearCache(); | 48 virtual void ClearCache(); |
| 49 | 49 |
| 50 protected: | 50 protected: |
| 51 | 51 |
| 52 virtual bool OnIFrame(IHTMLElement* pEl, const std::wstring& url, const std::w
string& indent) { return true; } | 52 virtual bool OnIFrame(IHTMLElement* pEl, const std::wstring& url, const std::w
string& indent) { return true; } |
| 53 virtual bool OnElement(IHTMLElement* pEl, const std::wstring& tag, T* cache, b
ool isDebug, const std::wstring& indent) { return true; } | 53 virtual bool OnElement(IHTMLElement* pEl, const std::wstring& tag, T* cache, b
ool isDebug, const std::wstring& indent) { return true; } |
| 54 | 54 |
| 55 virtual bool IsEnabled(); | 55 virtual bool IsEnabled(); |
| 56 | 56 |
| 57 protected: | 57 protected: |
| 58 | 58 |
| 59 void TraverseDocument(IWebBrowser2* pBrowser, bool isMainDoc, const std::wstri
ng& indent); | 59 void TraverseDocument(IWebBrowser2* pBrowser, bool isMainDoc, const std::wstri
ng& indent); |
| 60 void TraverseChild(IHTMLElement* pEl, IWebBrowser2* pBrowser, const std::wstri
ng& indent, bool isCached=true); | 60 void TraverseChild(IHTMLElement* pEl, IWebBrowser2* pBrowser, const std::wstri
ng& indent, bool isCached=true); |
| 61 | 61 |
| 62 CComAutoCriticalSection m_criticalSection; | 62 CComAutoCriticalSection m_criticalSection; |
| 63 | 63 |
| 64 std::wstring m_domain; | 64 std::wstring m_domain; |
| 65 std::wstring m_documentUrl; | 65 std::wstring m_documentUrl; |
| 66 | 66 |
| 67 bool m_isHeaderTraversed; | 67 bool m_isHeaderTraversed; |
| 68 | 68 |
| 69 // Caching | |
| 70 long m_cacheDomElementCount; | |
| 71 | |
| 72 int m_cacheIndexLast; | 69 int m_cacheIndexLast; |
| 73 int m_cacheElementsMax; | 70 int m_cacheElementsMax; |
| 74 std::set<std::wstring> m_cacheDocumentHasFrames; | 71 std::set<std::wstring> m_cacheDocumentHasFrames; |
| 75 std::set<std::wstring> m_cacheDocumentHasIframes; | 72 std::set<std::wstring> m_cacheDocumentHasIframes; |
| 76 | 73 |
| 77 T* m_cacheElements; | 74 T* m_cacheElements; |
| 78 | 75 |
| 79 CPluginTab* m_tab; | 76 std::shared_ptr<const CPluginFilter> m_pluginFilter; |
| 80 CComPtr<IWebBrowser2> m_pBrowser; | |
| 81 }; | 77 }; |
| 82 | 78 |
| 83 template <class T> | 79 template <class T> |
| 84 CPluginDomTraverserBase<T>::CPluginDomTraverserBase(CPluginTab* tab) : | 80 CPluginDomTraverserBase<T>::CPluginDomTraverserBase(const PluginFilterPtr& plugi
nFilter) |
| 85 m_tab(tab), m_isHeaderTraversed(false), m_cacheDomElementCount(0), m_cacheInde
xLast(0), m_cacheElementsMax(5000) | 81 : m_pluginFilter(pluginFilter), m_isHeaderTraversed(false), m_cacheIndexLast(0
), m_cacheElementsMax(5000) |
| 86 { | 82 { |
| 87 m_cacheElements = new T[m_cacheElementsMax]; | 83 m_cacheElements = new T[m_cacheElementsMax]; |
| 88 } | 84 } |
| 89 | 85 |
| 90 | 86 |
| 91 template <class T> | 87 template <class T> |
| 92 CPluginDomTraverserBase<T>::~CPluginDomTraverserBase() | 88 CPluginDomTraverserBase<T>::~CPluginDomTraverserBase() |
| 93 { | 89 { |
| 94 delete [] m_cacheElements; | 90 delete [] m_cacheElements; |
| 95 } | 91 } |
| (...skipping 25 matching lines...) Expand all Loading... |
| 121 template <class T> | 117 template <class T> |
| 122 bool CPluginDomTraverserBase<T>::IsEnabled() | 118 bool CPluginDomTraverserBase<T>::IsEnabled() |
| 123 { | 119 { |
| 124 return CPluginSettings::GetInstance()->IsPluginEnabled(); | 120 return CPluginSettings::GetInstance()->IsPluginEnabled(); |
| 125 } | 121 } |
| 126 | 122 |
| 127 | 123 |
| 128 template <class T> | 124 template <class T> |
| 129 void CPluginDomTraverserBase<T>::TraverseDocument(IWebBrowser2* pBrowser, bool i
sMainDoc, const std::wstring& indent) | 125 void CPluginDomTraverserBase<T>::TraverseDocument(IWebBrowser2* pBrowser, bool i
sMainDoc, const std::wstring& indent) |
| 130 { | 126 { |
| 131 DWORD res = WaitForSingleObject(m_tab->m_filter.hideFiltersLoadedEvent, ENGINE
_STARTUP_TIMEOUT); | |
| 132 if (!IsEnabled()) return; | 127 if (!IsEnabled()) return; |
| 133 | 128 |
| 134 VARIANT_BOOL isBusy; | 129 VARIANT_BOOL isBusy; |
| 135 if (SUCCEEDED(pBrowser->get_Busy(&isBusy))) | 130 if (SUCCEEDED(pBrowser->get_Busy(&isBusy))) |
| 136 { | 131 { |
| 137 if (isBusy != VARIANT_FALSE) | 132 if (isBusy != VARIANT_FALSE) |
| 138 { | 133 { |
| 139 return; | 134 return; |
| 140 } | 135 } |
| 141 } | 136 } |
| (...skipping 314 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 456 { | 451 { |
| 457 m_cacheIndexLast = 0; | 452 m_cacheIndexLast = 0; |
| 458 m_cacheDocumentHasFrames.clear(); | 453 m_cacheDocumentHasFrames.clear(); |
| 459 m_cacheDocumentHasIframes.clear(); | 454 m_cacheDocumentHasIframes.clear(); |
| 460 } | 455 } |
| 461 m_criticalSection.Unlock(); | 456 m_criticalSection.Unlock(); |
| 462 } | 457 } |
| 463 | 458 |
| 464 | 459 |
| 465 #endif // _PLUGIN_DOM_TRAVERSER_BASE_H_ | 460 #endif // _PLUGIN_DOM_TRAVERSER_BASE_H_ |
| OLD | NEW |