Issue 1264 - Generate custom response for subdocument on subdomain request

src/plugin/PluginWbPassThrough.h

 #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:
 
-        bool m_shouldBlock;
-        bool m_lastDataReported;
+        uint64_t m_currentPositionOfSentPage;
         CComPtr<IInternetProtocol> m_pTargetProtocol;
 
   int GetContentTypeFromMimeType(CString mimeType);
   int GetContentTypeFromURL(CString src);
   int GetContentType(CString mimeType, CString domain, CString src);
 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);
-        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,
                 /* [in] */ LPCWSTR szStatusText);
 
         STDMETHODIMP Switch(
                 /* [in] */ PROTOCOLDATA *pProtocolData);
 };
 
 class WBPassthru;
 typedef PassthroughAPP::CustomSinkStartPolicy<WBPassthru, WBPassthruSink> WBStartPolicy;
 
 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;
 };