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; |
}; |