| OLD | NEW |
| 1 #pragma once | 1 #pragma once |
| 2 #include <cstdint> | 2 #include <cstdint> |
| 3 #include <AdblockPlus/FilterEngine.h> |
| 3 #include "ProtocolCF.h" | 4 #include "ProtocolCF.h" |
| 4 #include "ProtocolImpl.h" | 5 #include "ProtocolImpl.h" |
| 5 #define IE_MAX_URL_LENGTH 2048 | 6 #define IE_MAX_URL_LENGTH 2048 |
| 6 | 7 |
| 7 class WBPassthruSink : | 8 class WBPassthruSink : |
| 8 public PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComM
ultiThreadModel>, | 9 public PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComM
ultiThreadModel>, |
| 9 public IHttpNegotiate | 10 public IHttpNegotiate |
| 10 { | 11 { |
| 11 typedef PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CCom
MultiThreadModel> BaseClass; | 12 typedef PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CCom
MultiThreadModel> BaseClass; |
| 12 | 13 |
| 13 public: | 14 public: |
| 14 WBPassthruSink(); | 15 WBPassthruSink(); |
| 15 | 16 |
| 16 uint64_t m_currentPositionOfSentPage; | 17 uint64_t m_currentPositionOfSentPage; |
| 17 CComPtr<IInternetProtocol> m_pTargetProtocol; | 18 CComPtr<IInternetProtocol> m_pTargetProtocol; |
| 18 » int m_contentType; | 19 » AdblockPlus::FilterEngine::ContentType m_contentType; |
| 19 std::wstring m_boundDomain; | 20 std::wstring m_boundDomain; |
| 20 bool m_isCustomResponse; | 21 bool m_isCustomResponse; |
| 21 | 22 |
| 22 » int GetContentTypeFromMimeType(const CString& mimeType); | 23 » AdblockPlus::FilterEngine::ContentType GetContentTypeFromMimeType(const
CString& mimeType); |
| 23 » int GetContentTypeFromURL(const CString& src); | 24 » AdblockPlus::FilterEngine::ContentType GetContentTypeFromURL(const CStri
ng& src); |
| 24 » int GetContentType(const CString& mimeType, const std::wstring& domain,
const CString& src); | 25 » AdblockPlus::FilterEngine::ContentType GetContentType(const CString& mim
eType, const std::wstring& domain, const CString& src); |
| 25 bool IsFlashRequest(const wchar_t* const* additionalHeaders); | 26 bool IsFlashRequest(const wchar_t* const* additionalHeaders); |
| 26 public: | 27 public: |
| 27 BEGIN_COM_MAP(WBPassthruSink) | 28 BEGIN_COM_MAP(WBPassthruSink) |
| 28 COM_INTERFACE_ENTRY(IHttpNegotiate) | 29 COM_INTERFACE_ENTRY(IHttpNegotiate) |
| 29 COM_INTERFACE_ENTRY_CHAIN(BaseClass) | 30 COM_INTERFACE_ENTRY_CHAIN(BaseClass) |
| 30 END_COM_MAP() | 31 END_COM_MAP() |
| 31 | 32 |
| 32 BEGIN_SERVICE_MAP(WBPassthruSink) | 33 BEGIN_SERVICE_MAP(WBPassthruSink) |
| 33 SERVICE_ENTRY(IID_IHttpNegotiate) | 34 SERVICE_ENTRY(IID_IHttpNegotiate) |
| 34 END_SERVICE_MAP() | 35 END_SERVICE_MAP() |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 73 WBPassthru(); | 74 WBPassthru(); |
| 74 // IInternetProtocolRoot | 75 // IInternetProtocolRoot |
| 75 STDMETHODIMP Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, | 76 STDMETHODIMP Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
| 76 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) override
; | 77 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) override
; |
| 77 | 78 |
| 78 //IInternetProtocol | 79 //IInternetProtocol |
| 79 STDMETHODIMP Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULO
NG *pcbRead) override; | 80 STDMETHODIMP Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULO
NG *pcbRead) override; |
| 80 | 81 |
| 81 bool m_shouldSupplyCustomContent; | 82 bool m_shouldSupplyCustomContent; |
| 82 }; | 83 }; |
| OLD | NEW |