Issues #696,1231,1264,1265 - Improve handling in PassthroughApp

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:
+        WBPassthruSink();
 
-        bool m_shouldBlock;
-        bool m_lastDataReported;
+        uint64_t m_currentPositionOfSentPage;
         CComPtr<IInternetProtocol> m_pTargetProtocol;
+        int m_contentType;
+        std::wstring m_boundDomain;
+        bool m_blockedInTransaction;
 
-  int GetContentTypeFromMimeType(CString mimeType);
-  int GetContentTypeFromURL(CString src);
-  int GetContentType(CString mimeType, std::wstring domain, CString src);
+        int GetContentTypeFromMimeType(const CString& mimeType);
+        int GetContentTypeFromURL(const CString& src);
+        int GetContentType(const CString& mimeType, const std::wstring& domain, const 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 ReportResult(
+                /* [in] */ HRESULT hrResult,
+                /* [in] */ DWORD dwError,
+                /* [in] */ LPCWSTR szResult);
+
         STDMETHODIMP Switch(
                 /* [in] */ PROTOCOLDATA *pProtocolData);
 };
 
 class WBPassthru;
 typedef PassthroughAPP::CustomSinkStartPolicy<WBPassthru, WBPassthruSink> WBStartPolicy;
 
 class WBPassthru : public PassthroughAPP::CInternetProtocol<WBStartPolicy>
 {
+  typedef PassthroughAPP::CInternetProtocol<WBStartPolicy> BaseClass;
 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;
+  bool m_hasOriginalStartCalled;
 };