| Index: src/plugin/PluginWbPassThrough.h |
| =================================================================== |
| --- a/src/plugin/PluginWbPassThrough.h |
| +++ b/src/plugin/PluginWbPassThrough.h |
| @@ -1,5 +1,5 @@ |
| #pragma once |
| - |
| +#include <cstdint> |
| #include "ProtocolCF.h" |
| #include "ProtocolImpl.h" |
| #define IE_MAX_URL_LENGTH 2048 |
| @@ -12,8 +12,7 @@ |
| public: |
| - bool m_shouldBlock; |
| - bool m_lastDataReported; |
| + uint64_t m_currentPositionOfSentPage; |
| CComPtr<IInternetProtocol> m_pTargetProtocol; |
| int GetContentTypeFromMimeType(CString mimeType); |
| @@ -43,8 +42,8 @@ |
| HRESULT OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
| IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved, |
| - IInternetProtocol* pTargetProtocol); |
| - HRESULT Read(void *pv, ULONG cb, ULONG* pcbRead); |
| + IInternetProtocol* pTargetProtocol, bool& handled); |
| + HRESULT OnRead(void *pv, ULONG cb, ULONG* pcbRead); |
| STDMETHODIMP ReportProgress( |
| /* [in] */ ULONG ulStatusCode, |
| @@ -60,10 +59,16 @@ |
| class WBPassthru : public PassthroughAPP::CInternetProtocol<WBStartPolicy> |
| { |
| public: |
| + WBPassthru(); |
| // IInternetProtocolRoot |
| STDMETHODIMP Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, |
| - IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved); |
| + IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) override; |
| //IInternetProtocol |
| - STDMETHODIMP Read( /* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULONG *pcbRead); |
| + STDMETHODIMP Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULONG *pcbRead) override; |
| + |
| + STDMETHODIMP LockRequest(/* [in] */ DWORD dwOptions) override; |
| + STDMETHODIMP UnlockRequest() override; |
| + |
| + bool m_shouldSupplyCustomContent; |
| }; |