WebBroswer2 事件处理

 这两天写一个辅助工具，用到了WebBrowser2控件，需要对DIID_DWebBrowserEvents2事件、DIID_HTMLDocumentEvents2进行处理
ATL提供了IDispEventSimpleImpl模板来简化事件处理，但是这个模板只能绑定一个事件，google了半天也没有找到好的解决方案，索性参考IDispEventSimpleImpl DIY了一个EasyDispEventImpl模板代码如下：

 

1. 1. struct EVENT_DIID_ENTRY
   2. {   
   3.     const IID* piid;
   4. };
   6. #define BEGIN_EVENT_IID_MAP(_class)/
   7.     typedef _class _GetEventIIDMapFinder;/
   8.     static const EVENT_DIID_ENTRY * _GetEventIIDMap()/
   9.     {/
   10.         typedef _class _atl_event_classtype;/
   11.         static const EVENT_DIID_ENTRY __map[] = {
   13. #define EVENT_IID_ENTRY(iid) {&iid},
   15. #define END_EVENT_IID_MAP() /
   16.     {NULL} }; return __map;/
   17. }
   19. class CWebBrowser2EventsBase
   20. {
   21. protected:
   22.     static _ATL_FUNC_INFO StatusTextChangeStruct;
   23.     static _ATL_FUNC_INFO TitleChangeStruct;
   24.     static _ATL_FUNC_INFO PropertyChangeStruct;
   25.     static _ATL_FUNC_INFO OnQuitStruct;
   26.     static _ATL_FUNC_INFO OnToolBarStruct;
   27.     static _ATL_FUNC_INFO OnMenuBarStruct;
   28.     static _ATL_FUNC_INFO OnStatusBarStruct;
   29.     static _ATL_FUNC_INFO OnFullScreenStruct;
   30.     static _ATL_FUNC_INFO OnTheaterModeStruct;
   31.     static _ATL_FUNC_INFO DownloadBeginStruct;
   32.     static _ATL_FUNC_INFO DownloadCompleteStruct;
   33.     static _ATL_FUNC_INFO NewWindow2Struct; 
   34.     static _ATL_FUNC_INFO CommandStateChangeStruct;
   35.     static _ATL_FUNC_INFO BeforeNavigate2Struct;
   36.     static _ATL_FUNC_INFO ProgressChangeStruct;
   37.     static _ATL_FUNC_INFO NavigateComplete2Struct;
   38.     static _ATL_FUNC_INFO DocumentComplete2Struct;
   39.     static _ATL_FUNC_INFO OnVisibleStruct;
   40.     static _ATL_FUNC_INFO SetSecureLockIconStruct;
   41.     static _ATL_FUNC_INFO NavigateErrorStruct;
   42.     static _ATL_FUNC_INFO PrivacyImpactedStateChangeStruct;
   43.     //static _ATL_FUNC_INFO FileDownloadStruct;   // Unsafe to use
   44.     //static _ATL_FUNC_INFO WindowClosingStruct;
   45.     static _ATL_FUNC_INFO KeyUpEventStruct; 
   46.     static _ATL_FUNC_INFO MouseOverEventStruct; 
   47.     static _ATL_FUNC_INFO ClickEventStruct;     
   48.     static _ATL_FUNC_INFO DbClickEventStruct;       
   49.     static _ATL_FUNC_INFO MouseOutEventStruct;      
   50.     
   51. };
   53. template <UINT nID, class T>
   54. class ATL_NO_VTABLE EasyDispEventImpl
   55. {
   56.     DWORD m_dwBrowserEventCookie;
   57.     DWORD m_dwDocumentEventCookie;
   58. public:
   59.     EasyDispEventImpl():m_dwBrowserEventCookie(0xFEFEFEFE),m_dwDocumentEventCookie(0xFEFEFEFE)
   60.     {}
   61.     STDMETHOD(_LocDEQueryInterface)(REFIID riid, void ** ppvObject)
   62.     {
   63.         ATLASSERT(ppvObject != NULL);
   64.         if (ppvObject == NULL)
   65.             return E_POINTER;
   66.         *ppvObject = NULL;
   68.         if (InlineIsEqualGUID(riid, IID_NULL))
   69.             return E_NOINTERFACE;
   71.         const EVENT_DIID_ENTRY * pMap = T::_GetEventIIDMap();
   72.         const EVENT_DIID_ENTRY * pFound = NULL;
   73.         while (pMap->piid != NULL)
   74.         {
   75.             if(InlineIsEqualGUID(riid,*pMap->piid))
   76.             {
   77.                 pFound = pMap;
   78.                 break;
   79.             }
   80.             pMap++;
   81.         }
   83.         if (pFound || InlineIsEqualUnknown(riid) || InlineIsEqualGUID(riid, __uuidof(IDispatch)))
   84.         {
   85.             *ppvObject = this;
   86.             AddRef();
   87. #ifdef _ATL_DEBUG_INTERFACES
   88.             _AtlDebugInterfacesModule.AddThunk((IUnknown**)ppvObject, _T("IDispEventImpl"), riid);
   89. #endif // _ATL_DEBUG_INTERFACES
   90.             return S_OK;
   91.         }
   92.         else
   93.             return E_NOINTERFACE;
   94.     }
   96.     // These are here only to support use in non-COM objects    
   97.     virtual ULONG STDMETHODCALLTYPE AddRef()
   98.     {
   99.         return 1;
   100.     }
   101.     virtual ULONG STDMETHODCALLTYPE Release()
   102.     {
   103.         return 1;
   104.     }
   106.     STDMETHOD(GetTypeInfoCount)(UINT* /*pctinfo*/)
   107.     {ATLTRACENOTIMPL(_T("IDispEventSimpleImpl::GetTypeInfoCount"));}
   109.     STDMETHOD(GetTypeInfo)(UINT /*itinfo*/, LCID /*lcid*/, ITypeInfo** /*pptinfo*/)
   110.     {ATLTRACENOTIMPL(_T("IDispEventSimpleImpl::GetTypeInfo"));}
   112.     STDMETHOD(GetIDsOfNames)(REFIID /*riid*/, LPOLESTR* /*rgszNames*/, UINT /*cNames*/,
   113.         LCID /*lcid*/, DISPID* /*rgdispid*/)
   114.     {ATLTRACENOTIMPL(_T("IDispEventSimpleImpl::GetIDsOfNames"));}
   116.     STDMETHOD(Invoke)(DISPID dispidMember, REFIID /*riid*/,
   117.         LCID lcid, WORD /*wFlags*/, DISPPARAMS* pdispparams, VARIANT* pvarResult,
   118.         EXCEPINFO* /*pexcepinfo*/, UINT* /*puArgErr*/)
   119.     {
   120.         const _ATL_EVENT_ENTRY<T>* pMap = T::_GetSinkMap();
   121.         const _ATL_EVENT_ENTRY<T>* pFound = NULL;       
   122.         while (pMap->piid != NULL)
   123.         {
   124.             const EVENT_DIID_ENTRY * pEvts = T::_GetEventIIDMap();
   125.             const EVENT_DIID_ENTRY * pFoundEvt = NULL;
   126.             while (pEvts->piid != NULL)
   127.             {
   128.                 if(InlineIsEqualGUID(*(pMap->piid),*(pEvts->piid)))
   129.                 {
   130.                     pFoundEvt = pEvts;
   131.                     break;
   132.                 }
   133.                 pEvts++;
   134.             }
   135.             if ((pMap->nControlID == nID) && (pMap->dispid == dispidMember) && pFoundEvt )
   136.             {               
   137.                 pFound = pMap;
   138.                 break;
   139.             }
   140.             pMap++;
   141.         }
   142.         if (pFound == NULL)
   143.             return S_OK;
   146.         _ATL_FUNC_INFO info;
   147.         _ATL_FUNC_INFO* pInfo;
   148.         if (pFound->pInfo != NULL)
   149.             pInfo = pFound->pInfo;
   150.         else
   151.         {
   152.             pInfo = &info;
   153.             HRESULT hr = GetFuncInfoFromId(*(pFound->piid), dispidMember, lcid, info);
   154.             if (FAILED(hr))
   155.                 return S_OK;
   156.         }
   157.         return InvokeFromFuncInfo(pFound->pfn, *pInfo, pdispparams, pvarResult);
   158.     }
   160.     //Helper for invoking the event
   161.     HRESULT InvokeFromFuncInfo(void (__stdcall T::*pEvent)(), _ATL_FUNC_INFO& info, DISPPARAMS* pdispparams, VARIANT* pvarResult)
   162.     {
   163.         ATLASSERT(pdispparams->cArgs == (UINT)info.nParams);
   165.         T* pT = static_cast<T*>(this);
   167.         // If this assert occurs, then add 
   168.         // #define _ATL_MAX_VARTYPES nnnn
   169.         // before including atlcom.h
   170.         ATLASSERT(info.nParams <= _ATL_MAX_VARTYPES);
   171.         if (info.nParams > _ATL_MAX_VARTYPES)
   172.         {
   173.             return E_FAIL;
   174.         }
   175.         VARIANTARG* rgVarArgs[_ATL_MAX_VARTYPES];
   176.         VARIANTARG** pVarArgs = info.nParams ? rgVarArgs : 0;
   178.         UINT nIndex = 0;
   180. #ifndef _ATL_IGNORE_NAMED_ARGS
   181.         for (nIndex; nIndex < pdispparams->cNamedArgs; nIndex++)
   182.             pVarArgs[pdispparams->rgdispidNamedArgs[nIndex]] = &pdispparams->rgvarg[nIndex];
   183. #endif
   185.         for (; nIndex < pdispparams->cArgs; nIndex++)
   186.             pVarArgs[info.nParams-nIndex-1] = &pdispparams->rgvarg[nIndex];
   188.         CComStdCallThunk<T> thunk;
   189.         thunk.Init(pEvent, pT);
   191.         CComVariant tmpResult;
   192.         if (pvarResult == NULL)
   193.             pvarResult = &tmpResult;
   195.         HRESULT hr = DispCallFunc(
   196.             &thunk,
   197.             0,
   198.             info.cc,
   199.             info.vtReturn,
   200.             info.nParams,
   201.             info.pVarTypes,
   202.             pVarArgs,
   203.             pvarResult);
   204.         ATLASSERT(SUCCEEDED(hr));
   205.         return hr;
   206.     }
   208.     //Helper for finding the function index for a DISPID
   209.     virtual HRESULT GetFuncInfoFromId(const IID& /*iid*/, DISPID /*dispidMember*/, LCID /*lcid*/, _ATL_FUNC_INFO& /*info*/)
   210.     {
   211.         ATLTRACE(_T("TODO: Classes using IDispEventSimpleImpl should override this method/n"));
   212.         ATLASSERT(0);
   213.         ATLTRACENOTIMPL(_T("IDispEventSimpleImpl::GetFuncInfoFromId"));
   214.     }
   215.     //Helpers for sinking events on random IUnknown*
   216.     HRESULT DispBrowserEventAdvise(IUnknown* pUnk)
   217.     {
   218.         ATLENSURE(m_dwBrowserEventCookie == 0xFEFEFEFE);        
   219.         return AtlAdvise(pUnk, (IUnknown*)this,DIID_DWebBrowserEvents2, &m_dwBrowserEventCookie);
   220.     }
   221.     HRESULT DispBrowserEventUnadvise(IUnknown* pUnk)
   222.     {
   223.         HRESULT hr = AtlUnadvise(pUnk,DIID_DWebBrowserEvents2, m_dwBrowserEventCookie);
   224.         m_dwBrowserEventCookie = 0xFEFEFEFE;
   225.         return hr;
   226.     }
   228.     HRESULT DispDocumentEventAdvise(IUnknown* pUnk)
   229.     {
   230.         ATLENSURE(m_dwDocumentEventCookie == 0xFEFEFEFE);       
   231.         return AtlAdvise(pUnk, (IUnknown*)this, DIID_HTMLDocumentEvents2, &m_dwDocumentEventCookie);
   232.     }
   233.     HRESULT DispDocumentEventUnadvise(IUnknown* pUnk)
   234.     {
   235.         HRESULT hr = AtlUnadvise(pUnk,DIID_HTMLDocumentEvents2, m_dwDocumentEventCookie);
   236.         m_dwDocumentEventCookie = 0xFEFEFEFE;
   237.         return hr;
   238.     }
   239.     HRESULT DispElementEventAdvise(IUnknown* pUnk, const IID* piid,DWORD * dwCookie)
   240.     {       
   241.         ///ATLASSERT(*dwCookie == 0xFEFEFEFE);
   242.         return AtlAdvise(pUnk, (IUnknown*)this, *piid,dwCookie);
   243.     }
   244.     HRESULT DispElementEventUnadvise(IUnknown* pUnk, const IID* piid,DWORD * dwCookie)
   245.     {
   246.         HRESULT hr = AtlUnadvise(pUnk, *piid,*dwCookie);
   247.         *dwCookie = 0xFEFEFEFE;
   248.         return hr;
   249.     }
   250. };
   252. #define EASY_SINK_ENTRY(id,iid, dispid,fn,info) {id, &iid,0, dispid, (void (__stdcall _atl_event_classtype::*)())fn, info},
   EASY_SINK_ENTRY 是SINK_ENTRY_INFO的替代物，因为我们自定了_ATL_EVENT_ENTRY结构的pInfo成员，因此nOffset可以为了0了。使用方法：
   

1. _ATL_FUNC_INFO DocumentComplete2Struct={CC_STDCALL, VT_EMPTY, 2, {VT_DISPATCH, VT_BYREF|VT_VARIANT}};
2. //....
3. _ATL_FUNC_INFO ClickEventStruct = {CC_STDCALL, VT_EMPTY, 1, {VT_DISPATCH}};
4. 
   
6. class CMyClass : public EasyDispEventImpl<1, CMyClass > 
7. {
8. public:
9.     BEGIN_SINK_MAP(CMyClass)    
10.         EASY_SINK_ENTRY(1, DIID_DWebBrowserEvents2, DISPID_DOCUMENTCOMPLETE, __DocumentComplete, &DocumentComplete2Struct)
11.         EASY_SINK_ENTRY(1, DIID_HTMLDocumentEvents2, DISPID_HTMLELEMENTEVENTS2_ONCLICK, __Click, &ClickEventStruct)
12.     END_SINK_MAP()
14. void __stdcall __DocumentComplete(IDispatch* pDisp, VARIANT* pvURL)
15.     {
16.         //TODO...
17.     }
19.   void __stdcall __Click(IDispatch* pDisp)
20.     {
21.         T* pT = static_cast<T*>(this);
22.         CComQIPtr<IHTMLEventObj2> evt(pDisp);
23.                 //TODO .... 
24.     }
26. };