C++中使用BHO来屏蔽特定网站

BHO(浏览器辅助对象)是一种简单的ATL COM对象,而Internet Explorer会在每次运行时都加载它;换句话来说,即每个Internet Explorer的实例都会加载它。BHO运行在Internet Explorer的地址空间内,能对可访问的对象(如窗口、模块等等)执行任何操作,且因为它依附于浏览器的主窗口,所以其生命期与浏览器实例的生命期一致。

  如果在系统中打开了活动桌面,BHO也能随同Windows Explorer一起启动。如果不想在Windows Explorer中运行BHO,可将如下代码添加到DLLMain中:

TCHAR strLoader[MAX_PATH];

::GetModuleFileName (NULL, strLoader, MAX_PATH);
if(stricmp("explorer.exe", strLoader) == 0)
return FALSE;

  BHO的COM Server必须实现IObjectWithSite,以便对象可以挂钩到浏览器事件,Internet Explorer会依靠IObjectWithSite将一个指针传递到它的IUnknown接口,所以,只需实现IObjectWithSite的SetSite方法就行了,如下所示:

STDMETHODIMP CBhoApp::SetSite(IUnknown *pUnkSite)
{
 //获取并存储IWebBrowser2指针
 m_spWebBrowser2 = pUnkSite;
 if (m_spWebBrowser2 == NULL)
  return E_INVALIDARG;

 //获取并存储IConnectionPointerContainer指针
 m_spCPC = m_spWebBrowser2;
 if (m_spCPC == NULL)
  return E_POINTER;

 //连接到宿主程序以接收事件通知
 return Connect();
}

  以下是Connect函数比较简单的实现:

HRESULT CBhoApp::Connect()
{
 HRESULT hr;
 CComPtr<IConnectionPoint> spCP;

 //获取访问WebBrowser事件的连接指针
 hr = m_spCPC->FindConnectionPoint(DIID_DWebBrowserEvents2, &spCP);
 if (FAILED(hr))
  return hr;

 //把事件处理程序传递给宿主程序Each time an event
 //每次有事件产生时,宿主程序都会调用我们实现的IDispatch接口的函数
 hr = spCP->Advise(reinterpret_cast<IDispatch*>(this),&m_dwCookie);
 return hr;
}

  通过调用Advise方法,告之浏览器BHO想要接受事件通知,这意味着BHO会把指向IDispatch的指针提供给浏览器(这是由于要进行组件事件处理),接下来,浏览器会调用IDispatch的Invoke方法,并传递给它一个事件的ID作为参数,因此,BHO必须实现Invoke方法以处理所发生的事件。

STDMETHODIMP CBhoApp::Invoke(DISPID dispidMember,REFIID riid,
LCID lcid,WORD wFlags,
DISPPARAMS *pDispParams,VARIANT
*pvarResult, EXCEPINFO *pExcepInfo,
UINT *puArgErr)
{
 //在使用ATL字符串转换宏(此处用的是OLE2T)防止编译错误时,必须先调用这个宏
 USES_CONVERSION;

 if(dispidMember == DISPID_BEFORENAVIGATE2)
 {
  BSTR bstrUrlName;
  HRESULT hr = m_spWebBrowser2->get_LocationURL(&bstrUrlName);
  if(FAILED(hr))
   return hr;

  LPTSTR psz = new TCHAR[SysStringLen(bstrUrlName)];
  lstrcpy(psz, OLE2T(bstrUrlName));

  //此处,直接比较www.xyz.com,也可从一个要屏蔽的网站列表中进行比较。
  if(stricmp("http://www.xyz.com/",psz) == 0)
  {
   VARIANT vFlags = ,vTargetFrameName = ;

   //如果不想显示"空白页",也可重定向至某个表明此网站已被屏蔽的页面。
   m_spWebBrowser2->Navigate(SysAllocString(L"about:blank"),&vFlags,&vTargetFrameName,
NULL,NULL);
   m_spWebBrowser2->put_Visible(VARIANT_TRUE);
   return S_FALSE;
  }
  return S_OK;
 }
 else if(dispidMember == DISPID_NAVIGATECOMPLETE2)
 //检查以防止页面的post导航加载(post-navigation loading)
 {
  BSTR bstrUrlName;
  HRESULT hr = m_spWebBrowser2->get_LocationURL(&bstrUrlName);
  if(FAILED(hr))
   return hr;

  //把文本从Unicode转换为ANSI
  LPTSTR psz = new TCHAR[SysStringLen(bstrUrlName)];
  lstrcpy(psz, OLE2T(bstrUrlName));
  ::OutputDebugString("In Navigate Complete");
  ::OutputDebugString(psz);
  if(stricmp("http://www.xyz.com/",psz) == 0)
  {
   VARIANT vFlags = ,vTargetFrameName = ;
   m_spWebBrowser2->Navigate(SysAllocString(L"about:blank"),
     &vFlags,&vTargetFrameName,
    NULL,NULL);
   m_spWebBrowser2->put_Visible(VARIANT_TRUE);
  }
  return S_OK;
 }
 return S_FALSE;
}

  还需修改工程的.rgs文件,依据所定格式添加以下注册表项:

HKLM SOFTWAREMicrosoftWindowsCurrentVersionExplorerBrowser Helper Objects
ForceRemove

  在编译完成后,使用regsvr32注册组件,如果想卸载,只需在regsvr32后带上/u。

posted @ 2009-05-04 17:48  jcss  阅读(477)  评论(0编辑  收藏  举报