Index: src/plugin/PluginWbPassThrough.h |
diff --git a/src/plugin/PluginWbPassThrough.h b/src/plugin/PluginWbPassThrough.h |
index c3ae80bfed2c16ddca68d26479d4df2d440c7d0c..c26a77d557f8f3a02e64954a2a7d5bf673c1fb47 100644 |
--- a/src/plugin/PluginWbPassThrough.h |
+++ b/src/plugin/PluginWbPassThrough.h |
@@ -15,85 +15,86 @@ |
* along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
*/ |
-#pragma once |
-#include <cstdint> |
-#include "ProtocolCF.h" |
-#include "ProtocolImpl.h" |
-#define IE_MAX_URL_LENGTH 2048 |
- |
-class WBPassthruSink : |
- public PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComMultiThreadModel>, |
- public IHttpNegotiate |
-{ |
- typedef PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComMultiThreadModel> BaseClass; |
- |
-public: |
- WBPassthruSink(); |
- |
- uint64_t m_currentPositionOfSentPage; |
- CComPtr<IInternetProtocol> m_pTargetProtocol; |
- int m_contentType; |
- std::wstring m_boundDomain; |
- bool m_isCustomResponse; |
- |
- int GetContentTypeFromMimeType(const CString& mimeType); |
- int GetContentTypeFromURL(const std::wstring& src); |
- int GetContentType(const CString& mimeType, const std::wstring& domain, const std::wstring& src); |
- bool IsFlashRequest(const wchar_t* const* additionalHeaders); |
-public: |
- BEGIN_COM_MAP(WBPassthruSink) |
- COM_INTERFACE_ENTRY(IHttpNegotiate) |
- COM_INTERFACE_ENTRY_CHAIN(BaseClass) |
- END_COM_MAP() |
- |
- BEGIN_SERVICE_MAP(WBPassthruSink) |
- SERVICE_ENTRY(IID_IHttpNegotiate) |
- END_SERVICE_MAP() |
- |
- STDMETHODIMP BeginningTransaction( |
- /* [in] */ LPCWSTR szURL, |
- /* [in] */ LPCWSTR szHeaders, |
- /* [in] */ DWORD dwReserved, |
- /* [out] */ LPWSTR *pszAdditionalHeaders); |
- |
- STDMETHODIMP OnResponse( |
- /* [in] */ DWORD dwResponseCode, |
- /* [in] */ LPCWSTR szResponseHeaders, |
- /* [in] */ LPCWSTR szRequestHeaders, |
- /* [out] */ LPWSTR *pszAdditionalRequestHeaders); |
- |
- HRESULT OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
- IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved, |
- IInternetProtocol* pTargetProtocol, bool& handled); |
- HRESULT OnRead(void *pv, ULONG cb, ULONG* pcbRead); |
- |
- STDMETHODIMP ReportProgress( |
- /* [in] */ ULONG ulStatusCode, |
- /* [in] */ LPCWSTR szStatusText); |
- |
- STDMETHODIMP ReportResult( |
- /* [in] */ HRESULT hrResult, |
- /* [in] */ DWORD dwError, |
- /* [in] */ LPCWSTR szResult); |
- |
- STDMETHODIMP Switch( |
- /* [in] */ PROTOCOLDATA *pProtocolData); |
-}; |
- |
-class WBPassthru; |
-typedef PassthroughAPP::CustomSinkStartPolicy<WBPassthru, WBPassthruSink> WBStartPolicy; |
- |
-class WBPassthru : public PassthroughAPP::CInternetProtocol<WBStartPolicy> |
-{ |
- typedef PassthroughAPP::CInternetProtocol<WBStartPolicy> BaseClass; |
-public: |
- WBPassthru(); |
- // IInternetProtocolRoot |
- STDMETHODIMP Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
- IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) override; |
- |
- //IInternetProtocol |
- STDMETHODIMP Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULONG *pcbRead) override; |
- |
- bool m_shouldSupplyCustomContent; |
-}; |
+#pragma once |
+#include <cstdint> |
+#include <AdblockPlus/FilterEngine.h> |
+#include "ProtocolCF.h" |
+#include "ProtocolImpl.h" |
+#define IE_MAX_URL_LENGTH 2048 |
+ |
+class WBPassthruSink : |
+ public PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComMultiThreadModel>, |
+ public IHttpNegotiate |
+{ |
+ typedef PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComMultiThreadModel> BaseClass; |
+ |
+public: |
+ WBPassthruSink(); |
+ |
+ uint64_t m_currentPositionOfSentPage; |
+ CComPtr<IInternetProtocol> m_pTargetProtocol; |
+ AdblockPlus::FilterEngine::ContentType m_contentType; |
+ std::wstring m_boundDomain; |
+ bool m_isCustomResponse; |
+ |
+ AdblockPlus::FilterEngine::ContentType GetContentTypeFromMimeType(const CString& mimeType); |
+ AdblockPlus::FilterEngine::ContentType GetContentTypeFromURL(const std::wstring& src); |
+ AdblockPlus::FilterEngine::ContentType GetContentType(const CString& mimeType, const std::wstring& domain, const std::wstring& src); |
+ bool IsFlashRequest(const wchar_t* const* additionalHeaders); |
+public: |
+ BEGIN_COM_MAP(WBPassthruSink) |
+ COM_INTERFACE_ENTRY(IHttpNegotiate) |
+ COM_INTERFACE_ENTRY_CHAIN(BaseClass) |
+ END_COM_MAP() |
+ |
+ BEGIN_SERVICE_MAP(WBPassthruSink) |
+ SERVICE_ENTRY(IID_IHttpNegotiate) |
+ END_SERVICE_MAP() |
+ |
+ STDMETHODIMP BeginningTransaction( |
+ /* [in] */ LPCWSTR szURL, |
+ /* [in] */ LPCWSTR szHeaders, |
+ /* [in] */ DWORD dwReserved, |
+ /* [out] */ LPWSTR *pszAdditionalHeaders); |
+ |
+ STDMETHODIMP OnResponse( |
+ /* [in] */ DWORD dwResponseCode, |
+ /* [in] */ LPCWSTR szResponseHeaders, |
+ /* [in] */ LPCWSTR szRequestHeaders, |
+ /* [out] */ LPWSTR *pszAdditionalRequestHeaders); |
+ |
+ HRESULT OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
+ IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved, |
+ IInternetProtocol* pTargetProtocol, bool& handled); |
+ HRESULT OnRead(void *pv, ULONG cb, ULONG* pcbRead); |
+ |
+ STDMETHODIMP ReportProgress( |
+ /* [in] */ ULONG ulStatusCode, |
+ /* [in] */ LPCWSTR szStatusText); |
+ |
+ STDMETHODIMP ReportResult( |
+ /* [in] */ HRESULT hrResult, |
+ /* [in] */ DWORD dwError, |
+ /* [in] */ LPCWSTR szResult); |
+ |
+ STDMETHODIMP Switch( |
+ /* [in] */ PROTOCOLDATA *pProtocolData); |
+}; |
+ |
+class WBPassthru; |
+typedef PassthroughAPP::CustomSinkStartPolicy<WBPassthru, WBPassthruSink> WBStartPolicy; |
+ |
+class WBPassthru : public PassthroughAPP::CInternetProtocol<WBStartPolicy> |
+{ |
+ typedef PassthroughAPP::CInternetProtocol<WBStartPolicy> BaseClass; |
+public: |
+ WBPassthru(); |
+ // IInternetProtocolRoot |
+ STDMETHODIMP Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
+ IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) override; |
+ |
+ //IInternetProtocol |
+ STDMETHODIMP Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULONG *pcbRead) override; |
+ |
+ bool m_shouldSupplyCustomContent; |
+}; |