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。