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

src/plugin/AdblockPlusDomTraverser.cpp

 #include "PluginStdAfx.h"
 
 #include "PluginClient.h"
 #include "PluginFilter.h"
 #include "PluginSettings.h"
 
 #include "AdblockPlusDomTraverser.h"
 
 
 CPluginDomTraverser::CPluginDomTraverser(CPluginTab* tab) : CPluginDomTraverserBase(tab)
 {
 }
 
 
 bool CPluginDomTraverser::OnIFrame(IHTMLElement* pEl, const CString& url, CString& indent)
 {
   CPluginClient* client = CPluginClient::GetInstance();
 
   // If src should be blocked, set style display:none on iframe
-  bool isBlocked = client->ShouldBlock(url, CFilter::contentTypeSubdocument, m_domain);
+  bool isBlocked = client->ShouldBlock(to_wstring(url), CFilter::contentTypeSubdocument, m_domain);
   if (isBlocked)
   {
     HideElement(pEl, "iframe", url, true, indent);
   }
 
   return !isBlocked;
 }
 
 
 bool CPluginDomTraverser::OnElement(IHTMLElement* pEl, const CString& tag, CPluginDomTraverserCache* cache, bool isDebug, CString& indent)
 {
   if (cache->m_isHidden)
   {
     return false;
   }
 
   // Check if element is hidden
   CPluginClient* client = CPluginClient::GetInstance();
 
-  cache->m_isHidden = client->IsElementHidden(tag, pEl, m_domain, indent, m_tab->m_filter.get());
+  cache->m_isHidden = client->IsElementHidden(to_wstring(tag), pEl, m_domain, to_wstring(indent), m_tab->m_filter.get());
   if (cache->m_isHidden)
   {
     HideElement(pEl, tag, "", false, indent);
     return false;
   }
 
   // Images
   if (tag == "img")
   {
     CComVariant vAttr;
 
     if (SUCCEEDED(pEl->getAttribute(L"src", 0, &vAttr)) && vAttr.vt == VT_BSTR && ::SysStringLen(vAttr.bstrVal) > 0)
     {
       CString src = vAttr.bstrVal;
       CPluginClient::UnescapeUrl(src);
 
       // If src should be blocked, set style display:none on image
-      cache->m_isHidden = client->ShouldBlock(src, CFilter::contentTypeImage, m_domain);
+      cache->m_isHidden = client->ShouldBlock(to_wstring(src), CFilter::contentTypeImage, m_domain);
       if (cache->m_isHidden)
       {
         HideElement(pEl, "image", src, true, indent);
         return false;
       }
     }
   }
   // Objects
   else if (tag == "object")
   {
@@ skipping 14 matching lines @@
         src = sObjectHtml.Mid(posBegin, posEnd - posBegin);
 
         // eg. http://w3schools.com/html/html_examples.asp
         if (src.Left(2) == "//")
         {
           src = "http:" + src;
         }
 
         if (!src.IsEmpty())
         {
-          //                            cache->m_isHidden = client->ShouldBlock(src, CFilter::contentTypeObject, m_domain);
           if (cache->m_isHidden)
           {
             HideElement(pEl, "object", src, true, indent);
             return false;
           }
         }
 
         posBegin = sObjectHtml.Find(L"VALUE=\"", posBegin);
         posEnd = posBegin >= 0 ? sObjectHtml.Find(L"\"", posBegin + 7) : -1;
       }
     }
   }
 
   return true;
 }
 
 
 bool CPluginDomTraverser::IsEnabled()
 {
   CPluginClient* client = CPluginClient::GetInstance();
-  return client && CPluginSettings::GetInstance()->IsPluginEnabled() && !client->IsWhitelistedUrl(std::wstring(m_domain));
+  return client && CPluginSettings::GetInstance()->IsPluginEnabled() && !client->IsWhitelistedUrl(m_domain);
 }
 
 
 void CPluginDomTraverser::HideElement(IHTMLElement* pEl, const CString& type, const CString& url, bool isDebug, CString& indent)
 {
   CComPtr<IHTMLStyle> pStyle;
 
   if (SUCCEEDED(pEl->get_style(&pStyle)) && pStyle)
   {
     CComBSTR bstrDisplay;
@@ skipping 11 matching lines @@
 
 #ifdef ENABLE_DEBUG_RESULT
         if (isDebug)
         {
           CPluginDebug::DebugResultHiding(type, url, "-");
         }
 #endif // ENABLE_DEBUG_RESULT
     }
   }
 }