| 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-2015 Eyeo GmbH | 3 * Copyright (C) 2006-2015 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 #pragma once | 18 #pragma once |
| 19 #include <cstdint> | 19 #include <cstdint> |
| 20 #include <AdblockPlus/FilterEngine.h> |
| 20 #include "ProtocolCF.h" | 21 #include "ProtocolCF.h" |
| 21 #include "ProtocolImpl.h" | 22 #include "ProtocolImpl.h" |
| 22 #define IE_MAX_URL_LENGTH 2048 | 23 #define IE_MAX_URL_LENGTH 2048 |
| 23 | 24 |
| 24 class WBPassthruSink : | 25 class WBPassthruSink : |
| 25 public PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComM
ultiThreadModel>, | 26 public PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CComM
ultiThreadModel>, |
| 26 public IHttpNegotiate | 27 public IHttpNegotiate |
| 27 { | 28 { |
| 28 typedef PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CCom
MultiThreadModel> BaseClass; | 29 typedef PassthroughAPP::CInternetProtocolSinkWithSP<WBPassthruSink, CCom
MultiThreadModel> BaseClass; |
| 29 | 30 |
| 30 public: | 31 public: |
| 31 WBPassthruSink(); | 32 WBPassthruSink(); |
| 32 | 33 |
| 33 uint64_t m_currentPositionOfSentPage; | 34 uint64_t m_currentPositionOfSentPage; |
| 34 CComPtr<IInternetProtocol> m_pTargetProtocol; | 35 CComPtr<IInternetProtocol> m_pTargetProtocol; |
| 35 » int m_contentType; | 36 » AdblockPlus::FilterEngine::ContentType m_contentType; |
| 36 std::wstring m_boundDomain; | 37 std::wstring m_boundDomain; |
| 37 bool m_isCustomResponse; | 38 bool m_isCustomResponse; |
| 38 | 39 |
| 39 » int GetContentTypeFromMimeType(const CString& mimeType); | 40 » AdblockPlus::FilterEngine::ContentType GetContentTypeFromMimeType(const
CString& mimeType); |
| 40 int GetContentTypeFromURL(const std::wstring& src); | 41 » AdblockPlus::FilterEngine::ContentType GetContentTypeFromURL(const std::
wstring& src); |
| 41 int GetContentType(const CString& mimeType, const std::wstring& domain, const
std::wstring& src); | 42 » AdblockPlus::FilterEngine::ContentType GetContentType(const CString& mim
eType, const std::wstring& domain, const std::wstring& src); |
| 42 bool IsFlashRequest(const wchar_t* const* additionalHeaders); | 43 bool IsFlashRequest(const wchar_t* const* additionalHeaders); |
| 43 public: | 44 public: |
| 44 BEGIN_COM_MAP(WBPassthruSink) | 45 BEGIN_COM_MAP(WBPassthruSink) |
| 45 COM_INTERFACE_ENTRY(IHttpNegotiate) | 46 COM_INTERFACE_ENTRY(IHttpNegotiate) |
| 46 COM_INTERFACE_ENTRY_CHAIN(BaseClass) | 47 COM_INTERFACE_ENTRY_CHAIN(BaseClass) |
| 47 END_COM_MAP() | 48 END_COM_MAP() |
| 48 | 49 |
| 49 BEGIN_SERVICE_MAP(WBPassthruSink) | 50 BEGIN_SERVICE_MAP(WBPassthruSink) |
| 50 SERVICE_ENTRY(IID_IHttpNegotiate) | 51 SERVICE_ENTRY(IID_IHttpNegotiate) |
| 51 END_SERVICE_MAP() | 52 END_SERVICE_MAP() |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 90 WBPassthru(); | 91 WBPassthru(); |
| 91 // IInternetProtocolRoot | 92 // IInternetProtocolRoot |
| 92 STDMETHODIMP Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, | 93 STDMETHODIMP Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
| 93 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) override
; | 94 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) override
; |
| 94 | 95 |
| 95 //IInternetProtocol | 96 //IInternetProtocol |
| 96 STDMETHODIMP Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULO
NG *pcbRead) override; | 97 STDMETHODIMP Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULO
NG *pcbRead) override; |
| 97 | 98 |
| 98 bool m_shouldSupplyCustomContent; | 99 bool m_shouldSupplyCustomContent; |
| 99 }; | 100 }; |
| OLD | NEW |