用Visual Studio 2008开发IE BHO(浏览器帮助对象) 之二

上一篇文章开发的RayBHO只是BHO的一个框架,根本不具备任何功能.

在这篇文章里,我们将使继续扩展这个BHO,让它具备更强的功能.首先我们学习如何让BHO接收IE的事件通知,接者学习为ie添加一个按钮,并让BHO对按钮做出响应.

要让BHO能接收事件通知, 它必须让处理函数与浏览器事件建立连接点. 为响应这些事件，它必须实现IDispEventImpl, ATL提供了一个默认实现，可以帮助简化这个事件处理逻辑。
在RayBHO.h添加:

#include  "exdispid.h"
#include  "shlguid.h"

#include "exdispid.h"
#include "shlguid.h"

我们的CRayBHO必须派生自IDispEventImpl,修改后的代码如下:

class    ATL_NO_VTABLE CRayBHO :
     public    CComObjectRootEx<CComSingleThreadModel>,
     public    CComCoClass<CRayBHO, &CLSID_RayBHO>,
     public    IObjectWithSiteImpl<CRayBHO>,
     public    IDispatchImpl<IRayBHO, &IID_IRayBHO, &LIBID_MySolutionPluginLib, /*wMajor =*/   1  , /*wMinor =*/   0  >,
     public    IDispEventImpl<1  ,CRayBHO,&DIID_DWebBrowserEvents2,&LIBID_SHDocVw,1  ,1  >

class ATL_NO_VTABLE CRayBHO :
public CComObjectRootEx<CComSingleThreadModel>,
public CComCoClass<CRayBHO, &CLSID_RayBHO>,
public IObjectWithSiteImpl<CRayBHO>,
public IDispatchImpl<IRayBHO, &IID_IRayBHO, &LIBID_MySolutionPluginLib, /*wMajor =*/ 1, /*wMinor =*/ 0>,
public IDispEventImpl<1,CRayBHO,&DIID_DWebBrowserEvents2,&LIBID_SHDocVw,1,1>

DispEventImpl为处理事件提供了一种简单安全的方法。

IDispEventImpl与事件路由表配合工作，可以将事件路由到相应的处理程序函数。在例子中,我们将"DocumentComplete"的事件交由OnDocumentComplete函数进行处理.

在public段添加路由表:

BEGIN_SINK_MAP(CHelloWorldBHO)
SINK_ENTRY_EX( 1  , DIID_DWebBrowserEvents2, DISPID_DOCUMENTCOMPLETE, OnDocumentComplete)
END_SINK_MAP()

BEGIN_SINK_MAP(CHelloWorldBHO)
SINK_ENTRY_EX(1, DIID_DWebBrowserEvents2, DISPID_DOCUMENTCOMPLETE, OnDocumentComplete)
END_SINK_MAP()

上述声明中SINK_ENTRY_EX(1,...)中的"1"与接口声明中的IDispEventImpl<1,....>是对应的,在必要时可以用于区分来自不同接口的事件.

DocumentComplete将被路由到处理函数OnDocumentComplete:

void    STDMETHODCALLTYPE OnDocumentComplete(IDispatch* pDisp, VARIANT* URL);

   void STDMETHODCALLTYPE OnDocumentComplete(IDispatch* pDisp, VARIANT* URL);

它的参数和参数顺序与DocumentComplete事件所定义的相同,另请注意，不要试图从事件处理程序返回值,这是因为 Internet Explorer 会忽略任何从 Invoke 返回的值.
我们还声明了一个私有变量来跟踪事件映射的处理情况

BOOL m_fAdvised;

BOOL m_fAdvised;

SetSite函数中必须处理事件派遣:

STDMETHODIMP CRayBHO::SetSite(IUnknown*pUnkSite)
{
    if   (pUnkSite!=NULL)
    {
        HRESULT hr;
        CComPtr<IServiceProvider> sp;

        hr = pUnkSite->QueryInterface(&sp);
        if   (SUCCEEDED(hr) && sp)
        {
            //缓存指向IWebBrowser2的指针
            hr = sp->QueryService(IID_IWebBrowserApp, IID_IWebBrowser2, ( void   **)&m_spWebBrowser);

            if   (SUCCEEDED(hr)&&m_spWebBrowser!=0  )
            {
                //注册DWebBrowserEvents2事件。
                hr=DispEventAdvise(m_spWebBrowser2);

                 if   (SUCCEEDED(hr))
                {
                    m_fAdvised=TRUE;
                }
            }
        }
        m_spUnkSite = pUnkSite;
         this   ->m_bIsIe7=this   ->IsIE7();

        //hr = sp->QueryInterface(IID_IOleCommandTarget,(void**)&m_spTarget);
         //this->GetInternetExplorerVersion();

    }
     else
    {
        //取消注册事件。
         if   (m_fAdvised)
        {
            DispEventUnadvise(m_spWebBrowser);
            m_fAdvised=FALSE;
        }
         //在此释放缓存的指针和其他资源。
        m_spWebBrowser.Release();
        //m_spTarget.Release();
    }
    //调用基类实现。
     return    IObjectWithSiteImpl<CRayBHO>::SetSite(pUnkSite);
}

STDMETHODIMP CRayBHO::SetSite(IUnknown*pUnkSite)
{
if(pUnkSite!=NULL)
{
HRESULT hr;
CComPtr<IServiceProvider> sp;
hr = pUnkSite->QueryInterface(&sp);
if(SUCCEEDED(hr) && sp)
{
//缓存指向IWebBrowser2的指针
hr = sp->QueryService(IID_IWebBrowserApp, IID_IWebBrowser2, (void**)&m_spWebBrowser);
if(SUCCEEDED(hr)&&m_spWebBrowser!=0)
{
//注册DWebBrowserEvents2事件。
hr=DispEventAdvise(m_spWebBrowser2);
if(SUCCEEDED(hr))
{
m_fAdvised=TRUE;
}
}
}
m_spUnkSite = pUnkSite;
this->m_bIsIe7=this->IsIE7();
//hr = sp->QueryInterface(IID_IOleCommandTarget,(void**)&m_spTarget);
//this->GetInternetExplorerVersion();
}
else
{
//取消注册事件。
if(m_fAdvised)
{
DispEventUnadvise(m_spWebBrowser);
m_fAdvised=FALSE;
}
//在此释放缓存的指针和其他资源。
m_spWebBrowser.Release();
//m_spTarget.Release();
}
//调用基类实现。
return IObjectWithSiteImpl<CRayBHO>::SetSite(pUnkSite);
}

我从网上找了一个OnDocumentComplete函数的例子并将之修改成范型,它对HTML Dom进行操作,将图像的属性设置为Display:None, 具体操作与javascript类似,不再赘述.

void       STDMETHODCALLTYPE   CRayBHO::OnDocumentComplete(IDispatch*pDisp,VARIANT*pvarURL)
{
    HRESULT hr = S_OK;
     // 查询 IWebBrowser2 接口。
    CComQIPtr<IWebBrowser2> spTempWebBrowser = pDisp;
    // 此事件是否与顶级浏览器相关联?
     if    (spTempWebBrowser && m_spWebBrowser && m_spWebBrowser.IsEqualObject(spTempWebBrowser))
    {
        // 从浏览器中获取当前文档对象……
        CComPtr<IDispatch>  spDispDoc;
        hr = m_spWebBrowser->get_Document(&spDispDoc);
         if    (SUCCEEDED(hr))
        {
            // ……并查询 HTML 文档。
            CComQIPtr<IHTMLDocument2> spHTMLDoc = spDispDoc;
            if    (spHTMLDoc != NULL) {
                 // 最后，删除这些图像。
                RemoveImages(spHTMLDoc);
            }
        }
    }
}

void    CRayBHO::RemoveImages(IHTMLDocument2* pDocument)
{
    CComPtr<IHTMLElementCollection> spImages;
     // 从 DOM 中获取图像集。
    HRESULT hr = pDocument->get_images(&spImages);
    if    (hr == S_OK && spImages != NULL) {
         // 获取集合中的图像数。
         long    cImages = 0  ;
        hr = spImages->get_length(&cImages);
        if    (hr == S_OK && cImages > 0  )
        {
            for    (int    i = 0  ; i < cImages; i++)
            {
                CComVariant svarItemIndex(i);
                CComVariant svarEmpty;
                CComPtr<IDispatch> spdispImage;
                 // 按索引从集合中获取图像。
                hr = spImages->item(svarItemIndex, svarEmpty, &spdispImage);
                if    (hr == S_OK && spdispImage != NULL)
                {
                    // 首先，查询通用 HTML 元素接口……
                    CComQIPtr<IHTMLElement> spElement = spdispImage;
                    if    (spElement)
                    {
                        // ……然后请求样式接口。
                        CComPtr<IHTMLStyle> spStyle;
                        hr = spElement->get_style(&spStyle);
                         // 设置 display="none" 以隐藏图像。
                         if    (hr == S_OK && spStyle != NULL)
                        {
                            static    const    CComBSTR sbstrNone(L"none"  );
                            spStyle->put_display(sbstrNone);
                        }
                    }
                }
            }
        }
    }
}

void	STDMETHODCALLTYPE	CRayBHO::OnDocumentComplete(IDispatch*pDisp,VARIANT*pvarURL)
{
HRESULT hr = S_OK;
// 查询 IWebBrowser2 接口。
CComQIPtr<IWebBrowser2> spTempWebBrowser = pDisp;
// 此事件是否与顶级浏览器相关联?
if (spTempWebBrowser && m_spWebBrowser && m_spWebBrowser.IsEqualObject(spTempWebBrowser))
{
// 从浏览器中获取当前文档对象……
CComPtr<IDispatch>  spDispDoc;
hr = m_spWebBrowser->get_Document(&spDispDoc);
if (SUCCEEDED(hr))
{
// ……并查询 HTML 文档。
CComQIPtr<IHTMLDocument2> spHTMLDoc = spDispDoc;
if (spHTMLDoc != NULL) {
// 最后，删除这些图像。
RemoveImages(spHTMLDoc);
}
}
}
}
void CRayBHO::RemoveImages(IHTMLDocument2* pDocument)
{
CComPtr<IHTMLElementCollection> spImages;
// 从 DOM 中获取图像集。
HRESULT hr = pDocument->get_images(&spImages);
if (hr == S_OK && spImages != NULL) {
// 获取集合中的图像数。
long cImages = 0;
hr = spImages->get_length(&cImages);
if (hr == S_OK && cImages > 0)
{
for (int i = 0; i < cImages; i++)
{
CComVariant svarItemIndex(i);
CComVariant svarEmpty;
CComPtr<IDispatch> spdispImage;
// 按索引从集合中获取图像。
hr = spImages->item(svarItemIndex, svarEmpty, &spdispImage);
if (hr == S_OK && spdispImage != NULL)
{
// 首先，查询通用 HTML 元素接口……
CComQIPtr<IHTMLElement> spElement = spdispImage;
if (spElement)
{
// ……然后请求样式接口。
CComPtr<IHTMLStyle> spStyle;
hr = spElement->get_style(&spStyle);
// 设置 display="none" 以隐藏图像。
if (hr == S_OK && spStyle != NULL)
{
static const CComBSTR sbstrNone(L"none");
spStyle->put_display(sbstrNone);
}
}
}
}
}
}
}

利用VC++操作HTML并没有想象中的繁琐, 你可以开发出更有趣的东西,比如从数据库自动填表单的BHO等等.

 

 

转载出处：http://www.cnblogs.com/jcss2008/archive/2009/05/05/1449654.html