关于屏蔽CHtmlView脚本错误

当WebBrowser控件(CDHtmlDialog自动创建了WebBrowser控件)加载的网页中含有错误Javascript代码时默认情况下控件会弹出错误信息提示对话框，相对于用户体验来说这样的提示完全不是开发人员想要的，针对这个问题有两个解决方案，一是完全屏蔽掉错误提示，二是控制错误的提示并且记录错误信息同时也可以控制出现错误后Javascript是否继续执行。

1、屏蔽错误信息提示

m_pBrowserApp->put_Silent(VARIANT_TRUE);

在CDHtmlDialog::OnInitDialog()的代码中首先了创建WebBrowser控件，然后把控件的Browser对象赋值给m_pBrowserApp（这是CDHtmlDialog完成的不需要自己处理）。WebBrowser的put_Silent函数在官方给出的说明是禁用所有的对话框，但例外情况是它不会影响SSL安全认证需要的进示对话框。绝大多数情况下这就可以解决问题了，记得很久以前我遇到过一种情况就是虽然调用了put_Silent但是还是有极个别的js错误是无法屏蔽掉的依然会显示出来(在网页含有嵌套页面时会错误无法屏蔽，不知道是否还有其它情况)，现在找不到这样的网页了，如果谁遇到这种情况了建议给我发上个URL让我也重温一下当年阳光灿烂的时刻。

2、控制错误提示并进行记录

　　这要比第一种方法复杂上许多，简短的来说就是自定义COleControlSite类并实现IOleCommandTarget接口，IOleCommandTarget接口是错误控制的关健，错误发生时会触发此接口的Exec函数并为nCmdID参数赋值为OLECMDID_SHOWSCRIPTERROR,这样就可以得到错误信息了。

IOleCommandTarget : public IUnknown
   {
   public :
       virtual /* [input_sync] */ HRESULT STDMETHODCALLTYPE QueryStatus(
           /* [unique][in] */ __RPC__in_opt const GUID *pguidCmdGroup,
           /* [in] */ ULONG cCmds,
           /* [out][in][size_is] */ __RPC__inout_ecount_full(cCmds) OLECMD prgCmds[  ],
           /* [unique][out][in] */ __RPC__inout_opt OLECMDTEXT *pCmdText) = 0;

       virtual HRESULT STDMETHODCALLTYPE Exec(
           /* [unique][in] */ __RPC__in_opt const GUID *pguidCmdGroup,
           /* [in] */ DWORD nCmdID,
           /* [in] */ DWORD nCmdexecopt,
           /* [unique][in] */ __RPC__in_opt VARIANT *pvaIn,
           /* [unique][out][in] */ __RPC__inout_opt VARIANT *pvaOut) = 0;

   };

现在我们开始实现自定义的COleControlSite

class CMyControlSite : public COleControlSite
{

public :
   CMyControlSite(COleControlContainer *pCntr):COleControlSite(pCntr) {}

protected :

   DECLARE_INTERFACE_MAP()
   BEGIN_INTERFACE_PART(OleCommandTarget, IOleCommandTarget)
       STDMETHOD(QueryStatus)( const GUID *pguidCmdGroup, ULONG cCmds, OLECMD prgCmds[], OLECMDTEXT *pCmdText);
       STDMETHOD(Exec)( const GUID* pguidCmdGroup, DWORD nCmdID, DWORD nCmdexecopt, VARIANTARG* pvaIn, VARIANTARG* pvaOut);
   END_INTERFACE_PART(OleCommandTarget)

MFC提供了很多宏用于简化COM相关功能的开发，对COM接口的实现方式在MFC中具体体现方式是内嵌类，背后的设计思想是COM聚合，每个接口都产生一个内嵌类，所有的接口都聚合到外层的类。DECLARE_INTERFACE_MAP()  实际上就是定义一个数组以及查询操作，BEGIN_INTERFACE_PART定义一个命名为XOleCommandTarget的内嵌类(内嵌类的命名规则是XName)并定义IUnknown接口的三个方法AddRef、Release、QueryInterface。END_INTERFACE_PART定义一个m_xOleCommandTarget的成员类型为XOleCommandTarget(定义的成员命名规则就是m_xName)，并把XOleCommandTarget类声明为外层类的友元类在示例中外层类指CMyControlSite。

STDMETHOD宏定义为virtual __declspec(nothrow) HRESULT __stdcall，该宏定义函数为虚函数返回值为HRESULT，函数调用约定为__stdcall，并且在此函数上禁止异常。__declspec(nothrow)用定告诉编译器它修饰的函数以及此函数调用的函数不会产生C++异常调用从可以优化代码性能和代码尺寸（默认情况下C++编译器为了进行异常处理会在拥有throw调用的函数中自动生成相关的异常处理代码)。通常情况下HRESULT返回值就表达了错误信息，HRESULT是个32位值，不同的位域用于不同的目的，也可以使用自定义的位域，具体的信息可以参考http://en.wikipedia.org/wiki/HRESULT。由于COM本身的语言中立性所以不应该在COM组件对外公布的信息中掺杂特定语言相关的特性。如果需要提供更详尽的错误信息那么应该实现COM的IErrorInfo接口。言归正传以下是CMyControlSite的类实现代码

BEGIN_INTERFACE_MAP(CMyControlSite, COleControlSite)
   INTERFACE_PART(CMyControlSite, IID_IOleCommandTarget, OleCommandTarget)
END_INTERFACE_MAP()


HRESULT CMyControlSite::XOleCommandTarget::Exec
( const GUID* pguidCmdGroup, DWORD nCmdID,
DWORD nCmdexecopt, VARIANTARG* pvaIn, VARIANTARG* pvaOut )
{
   HRESULT hr = OLECMDERR_E_NOTSUPPORTED;
   //return S_OK;
   if (pguidCmdGroup && IsEqualGUID(*pguidCmdGroup, CGID_DocHostCommandHandler))
   {

       switch (nCmdID)
       {

       case OLECMDID_SHOWSCRIPTERROR:
           {
               IHTMLDocument2*             pDoc = NULL;
               IHTMLWindow2*               pWindow = NULL;
               IHTMLEventObj*              pEventObj = NULL;
               BSTR                        rgwszNames[5] =
               {
                   SysAllocString(L "errLine" ),
                   SysAllocString(L "errCharacter" ),
                   SysAllocString(L "errCode" ),
                   SysAllocString(L "errMsg" ),
                   SysAllocString(L "errUrl" )
               };
               DISPID                      rgDispIDs[5];
               VARIANT                     rgvaEventInfo[5];
               DISPPARAMS                  params;
               BOOL                        fContinueRunningScripts = true ;

               params.cArgs = 0;
               params.cNamedArgs = 0;

               hr = pvaIn->punkVal->QueryInterface(IID_IHTMLDocument2, ( void **) &pDoc);

               hr = pDoc->get_parentWindow(&pWindow);
               pDoc->Release();

               hr = pWindow->get_event(&pEventObj);

               for ( int i = 0; i < 5; i++)
               {

                   hr = pEventObj->GetIDsOfNames(IID_NULL, &rgwszNames[i], 1,
                       LOCALE_SYSTEM_DEFAULT, &rgDispIDs[i]);

                   hr = pEventObj->Invoke(rgDispIDs[i], IID_NULL,
                       LOCALE_SYSTEM_DEFAULT,
                       DISPATCH_PROPERTYGET, ¶ms, &rgvaEventInfo[i],
                       NULL, NULL);
                   //可以在此记录错误信息

                   //必须使用SysFreeString来释放SysAllocString分配的内存,SysAllocString在分配的内存中记录了字符的长度
                   SysFreeString(rgwszNames[i]);
               }

               // At this point, you would normally alert the user with
               // the information about the error, which is now contained
               // in rgvaEventInfo[]. Or, you could just exit silently.

               (*pvaOut).vt = VT_BOOL;
               if (fContinueRunningScripts)
               {
                   // 在页面中继续执行脚本
                   (*pvaOut).boolVal = VARIANT_TRUE;
               }
               else
               {
                   // 停止在页面中执行脚本
                   (*pvaOut).boolVal = VARIANT_FALSE;
               }
               break ;
           }
       default :
           hr =OLECMDERR_E_NOTSUPPORTED;
           break ;
       }
   }
   else
   {
       hr = OLECMDERR_E_UNKNOWNGROUP;
   }
   return (hr);
}


ULONG FAR EXPORT CMyControlSite::XOleCommandTarget::AddRef()
{
   METHOD_PROLOGUE(CMyControlSite, OleCommandTarget)
       return pThis->ExternalAddRef();
}


ULONG FAR EXPORT CMyControlSite::XOleCommandTarget::Release()
{
   METHOD_PROLOGUE(CMyControlSite, OleCommandTarget)
       return pThis->ExternalRelease();
}

HRESULT FAR EXPORT CMyControlSite::XOleCommandTarget::QueryInterface(REFIID riid, void **ppvObj)
{
   METHOD_PROLOGUE(CMyControlSite, OleCommandTarget)
       HRESULT hr = ( HRESULT )pThis->ExternalQueryInterface(&riid, ppvObj);
   return hr;
}

STDMETHODIMP CMyControlSite::XOleCommandTarget::QueryStatus(
   /* [unique][in] */ const GUID __RPC_FAR *pguidCmdGroup,
   /* [in] */ ULONG cCmds,
   /* [out][in][size_is] */ OLECMD __RPC_FAR prgCmds[ ],
   /* [unique][out][in] */ OLECMDTEXT __RPC_FAR *pCmdText
   )
{
   METHOD_PROLOGUE(CMyControlSite, OleCommandTarget)
       return OLECMDERR_E_NOTSUPPORTED;
}

实现CMyControlSite后需要应用到CDHtmlDialog上，重写CreateControlSite虚函数既可

virtual BOOL CreateControlSite(COleControlContainer* pContainer,
       COleControlSite** ppSite, UINT  nID , REFCLSID  clsid )
{
       if (ppSite == NULL)
   {
       ASSERT(FALSE);
       return FALSE;
   }

   CMyControlSite *pBrowserSite =
       new CMyControlSite (pContainer, this );
   if (!pBrowserSite)
       return FALSE;

   *ppSite = pBrowserSite;
   return TRUE;
}

现在就可以去编译测试了。到目前还有一个问题没有考虑，如果这段代码整合到现有的CDHtmlDialog应用中而现有的代码使用了其它默认的设定比如说自定义WebBrowser的右健菜单或使用了GetIDispatch函数等情况下原有的代码就不能正常工作了。这部分功能是在MFC的CBrowserControlSite类中处理的，所以CMyControlSite应该把CBrowserControlSite的功能也实现一遍以使CDHtmlDialog的原有封装性不被破坏。在CMyControlSite中实现IDocHostUIHandler接口既可完成此功能。代码声明如下

public :

CMyControlSite(COleControlContainer *pCnt, CDHtmlDialog *pHandler):COleControlSite(pCnt),m_pHandler(pHandler) {}

protected :

CDHtmlDialog *m_pHandler;<br>

BEGIN_INTERFACE_PART(DocHostUIHandler, IDocHostUIHandler)
       STDMETHOD(ShowContextMenu)( DWORD , LPPOINT, LPUNKNOWN, LPDISPATCH);
       STDMETHOD(GetHostInfo)(DOCHOSTUIINFO*);
       STDMETHOD(ShowUI)( DWORD , LPOLEINPLACEACTIVEOBJECT,
           LPOLECOMMANDTARGET, LPOLEINPLACEFRAME, LPOLEINPLACEUIWINDOW);
       STDMETHOD(HideUI)( void );
       STDMETHOD(UpdateUI)( void );
       STDMETHOD(EnableModeless)( BOOL );
       STDMETHOD(OnDocWindowActivate)( BOOL );
       STDMETHOD(OnFrameWindowActivate)( BOOL );
       STDMETHOD(ResizeBorder)(LPCRECT, LPOLEINPLACEUIWINDOW, BOOL );
       STDMETHOD(TranslateAccelerator)(LPMSG, const GUID*, DWORD );
       STDMETHOD(GetOptionKeyPath)(OLECHAR **, DWORD );
       STDMETHOD(GetDropTarget)(LPDROPTARGET, LPDROPTARGET*);
       STDMETHOD(GetExternal)(LPDISPATCH*);
       STDMETHOD(TranslateUrl)( DWORD , OLECHAR*, OLECHAR **);
       STDMETHOD(FilterDataObject)(LPDATAOBJECT , LPDATAOBJECT*);
   END_INTERFACE_PART(DocHostUIHandler)

以下是实现代码

BEGIN_INTERFACE_MAP(CMyControlSite, COleControlSite)
   INTERFACE_PART(CMyControlSite, IID_IDocHostUIHandler, DocHostUIHandler)
   INTERFACE_PART(CMyControlSite, IID_IOleCommandTarget, OleCommandTarget)
END_INTERFACE_MAP()

STDMETHODIMP CMyControlSite::XDocHostUIHandler::GetExternal(LPDISPATCH *lppDispatch)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->GetExternal(lppDispatch);
}



STDMETHODIMP CMyControlSite::XDocHostUIHandler::ShowContextMenu(
   DWORD dwID, LPPOINT ppt, LPUNKNOWN pcmdTarget, LPDISPATCH pdispReserved)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->ShowContextMenu(dwID, ppt, pcmdTarget, pdispReserved);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::GetHostInfo(
   DOCHOSTUIINFO *pInfo)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->GetHostInfo(pInfo);
}


STDMETHODIMP CMyControlSite::XDocHostUIHandler::ShowUI(
   DWORD dwID, LPOLEINPLACEACTIVEOBJECT pActiveObject,
   LPOLECOMMANDTARGET pCommandTarget, LPOLEINPLACEFRAME pFrame,
   LPOLEINPLACEUIWINDOW pDoc)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->ShowUI(dwID, pActiveObject, pCommandTarget, pFrame, pDoc);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::HideUI( void )
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->HideUI();
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::UpdateUI( void )
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->UpdateUI();
}


STDMETHODIMP CMyControlSite::XDocHostUIHandler::EnableModeless( BOOL fEnable)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->EnableModeless(fEnable);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::OnDocWindowActivate( BOOL fActivate)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->OnDocWindowActivate(fActivate);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::OnFrameWindowActivate(
   BOOL fActivate)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->OnFrameWindowActivate(fActivate);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::ResizeBorder(
   LPCRECT prcBorder, LPOLEINPLACEUIWINDOW pUIWindow, BOOL fFrameWindow)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->ResizeBorder(prcBorder, pUIWindow, fFrameWindow);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::TranslateAccelerator(
   LPMSG lpMsg, const GUID* pguidCmdGroup, DWORD nCmdID)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->TranslateAccelerator(lpMsg, pguidCmdGroup, nCmdID);
}


STDMETHODIMP CMyControlSite::XDocHostUIHandler::GetOptionKeyPath(
   LPOLESTR* pchKey, DWORD dwReserved)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->GetOptionKeyPath(pchKey, dwReserved);
}


STDMETHODIMP CMyControlSite::XDocHostUIHandler::GetDropTarget(
   LPDROPTARGET pDropTarget, LPDROPTARGET* ppDropTarget)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->GetDropTarget(pDropTarget, ppDropTarget);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::TranslateUrl(
   DWORD dwTranslate, OLECHAR* pchURLIn, OLECHAR** ppchURLOut)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->TranslateUrl(dwTranslate, pchURLIn, ppchURLOut);
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::FilterDataObject(
   LPDATAOBJECT pDataObject, LPDATAOBJECT* ppDataObject)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
   return pThis->m_pHandler->FilterDataObject(pDataObject, ppDataObject);
}


STDMETHODIMP_( ULONG ) CMyControlSite::XDocHostUIHandler::AddRef()
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
       return pThis->ExternalAddRef();
}

STDMETHODIMP_( ULONG ) CMyControlSite::XDocHostUIHandler::Release()
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
       return pThis->ExternalRelease();
}

STDMETHODIMP CMyControlSite::XDocHostUIHandler::QueryInterface(
   REFIID iid, LPVOID far* ppvObj)
{
   METHOD_PROLOGUE_EX_(CMyControlSite, DocHostUIHandler)
       return pThis->ExternalQueryInterface(&iid, ppvObj);
}

STDMETHODIMP宏的定义是HRESULT __stdcall，STDMETHODIMP_宏指定了返回值，这两个宏用在cpp实现文件中，对应用于声明时使用的STDMETHOD和STDMETHOD_。

METHOD_PROLOGUE_EX_宏定义了pThis指针来指向外层类。

以上代码基于VS2008，由于不同版本的VS所带的MFC库版本不尽一致所以需要根本具体的版本来处理，目前已知的不同部分主要集中在CreateControlSite上。

 

转载地址：http://www.cnblogs.com/thinkingfor/archive/2010/11/10/1873689.html