Issue #1467, #3397 - Rewrite the map from threads to tabs

src/plugin/PluginWbPassThrough.cpp

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-2016 Eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ skipping 124 matching lines @@
   }
 }
 
 WBPassthruSink::WBPassthruSink()
   : m_currentPositionOfSentPage(0)
   , m_contentType(ContentType::CONTENT_TYPE_OTHER)
   , m_isCustomResponse(false)
 {
 }
 
-ContentType WBPassthruSink::GetContentTypeFromMimeType(const std::wstring& mimeType)
-{
-  if ((mimeType.find(L"text/html") != std::wstring::npos) ||
-      (mimeType.find(L"application/xhtml+xml") != std::wstring::npos))
-  {
-    return ContentType::CONTENT_TYPE_SUBDOCUMENT;
-  }
-  if (mimeType.find(L"image/") != std::wstring::npos)
-  {
-    return ContentType::CONTENT_TYPE_IMAGE;
-  }
-  if (mimeType.find(L"text/css") != std::wstring::npos)
-  {
-    return ContentType::CONTENT_TYPE_STYLESHEET;
-  }
-  if ((mimeType.find(L"application/javascript") != std::wstring::npos) || (mimeType.find(L"application/json") != std::wstring::npos))
-  {
-    return ContentType::CONTENT_TYPE_SCRIPT;
-  }
-  if (mimeType.find(L"application/x-shockwave-flash") != std::wstring::npos)
-  {
-    return ContentType::CONTENT_TYPE_OBJECT;
-  }
-  // It is important to have this check last, since it is rather generic, and might overlay text/html, for example
-  if (mimeType.find(L"xml") != std::wstring::npos)
-  {
-    return ContentType::CONTENT_TYPE_XMLHTTPREQUEST;
-  }
-
-  return ContentType::CONTENT_TYPE_OTHER;
-}
-
-ContentType WBPassthruSink::GetContentTypeFromURL(const std::wstring& src)
-{
-  std::wstring schemeAndHierarchicalPart = GetSchemeAndHierarchicalPart(src);
-  auto contentType = GetContentTypeFromString(schemeAndHierarchicalPart);
-  if (contentType == ContentType::CONTENT_TYPE_OTHER &&
-    AdblockPlus::IE::InstalledMajorVersion() == 8)
-  {
-    std::wstring queryString = GetQueryString(src);
-    wchar_t* nextToken = nullptr;
-    const wchar_t* token = wcstok_s(&queryString[0], L"&=", &nextToken);
-    while (token != nullptr)
-    {
-      contentType = GetContentTypeFromString(token);
-      if (contentType != ContentType::CONTENT_TYPE_OTHER)
+namespace
+{
+  /**
+   * Heuristic to infer an ABP content type from the media type list of an Accept: header field.
+   *
+   * Known IE Accept: strings for documents and images:
+   *     text/html, application/xhtml+xml, image/jxr, *\/*
+   *     image/png, image/svg+xml, image/jxr, image/*;q=0.8, *\/*;q = 0.5
+   * In IE9 the image/jxr part is missing.
+   */
+  ContentType InferContentTypeFromMediaTypeList(const std::wstring& mediaTypeList)
+  {
+    if ((mediaTypeList.find(L"text/html") != std::wstring::npos) ||
+      (mediaTypeList.find(L"application/xhtml+xml") != std::wstring::npos))
+    {
+      return ContentType::CONTENT_TYPE_SUBDOCUMENT;
+    }
+    if (mediaTypeList.find(L"image/") != std::wstring::npos)
+    {
+      return ContentType::CONTENT_TYPE_IMAGE;
+    }
+    if (mediaTypeList.find(L"text/css") != std::wstring::npos)
+    {
+      return ContentType::CONTENT_TYPE_STYLESHEET;
+    }
+    if ((mediaTypeList.find(L"application/javascript") != std::wstring::npos) || (mediaTypeList.find(L"application/json") != std::wstring::npos))
+    {
+      return ContentType::CONTENT_TYPE_SCRIPT;
+    }
+    if (mediaTypeList.find(L"application/x-shockwave-flash") != std::wstring::npos)
+    {
+      return ContentType::CONTENT_TYPE_OBJECT;
+    }
+    // It is important to have this check last, since it is rather generic, and might overlay text/html, for example
+    if (mediaTypeList.find(L"xml") != std::wstring::npos)
+    {
+      return ContentType::CONTENT_TYPE_XMLHTTPREQUEST;
+    }
+    return ContentType::CONTENT_TYPE_OTHER;
+  }
+
+  ContentType InferContentTypeFromUrl(const std::wstring& src)
+  {
+    std::wstring schemeAndHierarchicalPart = GetSchemeAndHierarchicalPart(src);
+    auto contentType = GetContentTypeFromString(schemeAndHierarchicalPart);
+    if (contentType == ContentType::CONTENT_TYPE_OTHER &&
+      AdblockPlus::IE::InstalledMajorVersion() == 8)
+    {
+      std::wstring queryString = GetQueryString(src);
+      wchar_t* nextToken = nullptr;
+      const wchar_t* token = wcstok_s(&queryString[0], L"&=", &nextToken);
+      while (token != nullptr)
       {
-         return contentType;
+        contentType = GetContentTypeFromString(token);
+        if (contentType != ContentType::CONTENT_TYPE_OTHER)
+        {
+           return contentType;
+        }
+        token = wcstok_s(nullptr, L"&=", &nextToken);
       }
-      token = wcstok_s(nullptr, L"&=", &nextToken);
-    }
-  }
-  return contentType;
-}
-
-ContentType WBPassthruSink::GetContentType(const std::wstring& mimeType, const std::wstring& domain, const std::wstring& src)
-{
-  // No referer or mime type
-  // BINDSTRING_XDR_ORIGIN works only for IE v8+
-  if (mimeType.empty() && domain.empty() && AdblockPlus::IE::InstalledMajorVersion() >= 8)
-  {
-    return ContentType::CONTENT_TYPE_XMLHTTPREQUEST;
-  }
-  ContentType contentType = GetContentTypeFromMimeType(mimeType);
-  if (contentType == ContentType::CONTENT_TYPE_OTHER)
-  {
-    contentType = GetContentTypeFromURL(src);
-  }
-  return contentType;
+    }
+    return contentType;
+  }
+
+  ContentType InferContentType(const std::wstring& mediaTypeList, const std::wstring& domain, const std::wstring& src)
+  {
+    // No referer or mime type
+    // BINDSTRING_XDR_ORIGIN works only for IE v8+
+    if (mediaTypeList.empty() && domain.empty() && AdblockPlus::IE::InstalledMajorVersion() >= 8)
+    {
+      return ContentType::CONTENT_TYPE_XMLHTTPREQUEST;
+    }
+    ContentType contentType = InferContentTypeFromMediaTypeList(mediaTypeList);
+    if (contentType == ContentType::CONTENT_TYPE_OTHER)
+    {
+      contentType = InferContentTypeFromUrl(src);
+    }
+    return contentType;
+  }
 }
 
 ////////////////////////////////////////////////////////////////////////////////////////
 //WBPassthruSink
 //Monitor and/or cancel every request and responde
 //WB makes, including images, sounds, scripts, etc
 ////////////////////////////////////////////////////////////////////////////////////////
 HRESULT WBPassthruSink::OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink,
                                 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved,
                                 IInternetProtocol* pTargetProtocol)
@@ skipping 113 matching lines @@
   QueryServiceFromClient(&httpNegotiate);
   // This fills the pszAdditionalHeaders with more headers. One of which is the Referer header, which we need.
   // There doesn't seem to be any other way to get this header before the request has been made.
   HRESULT nativeHr = httpNegotiate ? httpNegotiate->BeginningTransaction(szURL, szHeaders, dwReserved, pszAdditionalHeaders) : S_OK;
 
   if (pszAdditionalHeaders && *pszAdditionalHeaders)
   {
     m_boundDomain = ExtractHttpHeader<std::wstring>(*pszAdditionalHeaders, L"Referer:", L"\n");
   }
   m_boundDomain = TrimString(m_boundDomain);
-  m_contentType = GetContentType(ToUtf16String(ExtractHttpAcceptHeader(m_spTargetProtocol)), m_boundDomain, src);
-
-  CPluginTab* tab = CPluginClass::GetTabCurrentThread();
+  m_contentType = InferContentType(ToUtf16String(ExtractHttpAcceptHeader(m_spTargetProtocol)), m_boundDomain, src);
+
+  CPluginTab* tab = CPluginClass::GetTabForCurrentThread();
   CPluginClient* client = CPluginClient::GetInstance();
 
   if (tab && client)
   {
     std::wstring documentUrl = tab->GetDocumentUrl();
     // Page is identical to document => don't block
     if (documentUrl == src)
     {
       return nativeHr;
     }
@@ skipping 85 matching lines @@
   }
 
   return OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, m_spInternetProtocol);
 }
 
 STDMETHODIMP WbPassthroughProtocol::Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULONG *pcbRead)
 {
   WBPassthruSink* pSink = GetSink();
   return pSink->OnRead(pv, cb, pcbRead);
 }