首页 > 代码库 > ATL实现一个组件多个dual接口,multidisp
ATL实现一个组件多个dual接口,multidisp
最近想自己写个按键精灵的插件,于是接触到这个问题: 怎么在一个组件里实现两个自动化接口。
主要针对的ATL,MFC貌似没这个问题,具体MFC是怎么实现的自己没有深究。
按键精灵的插件会在一个组件里实现两个dispinterface,具体请看oleview工具截图:
刚开始对这个问题不理解,以为不是问题,自己用ATL尝试了几次,才发现不是那么回事,于是google之。
MSDN上是这么说的,看这里
ATL不提供任何为将多个双重接口支持。IDispatch的单个实现。 但是,有几个已知的方法来手动合并接口,如创建包含创建一个新的对象,执行 QueryInterface 函数或使用嵌套的对象一个基于typeinfo的实现的单独 IDispatch 接口来创建 IDispatch 接口的模板选件类。
这些方法都有潜在的命名空间冲突问题,以及代码复杂性和可维护性。 建议不要创建多个双绑定接口。
虽然ATL不支持,但是上面也说了,还是有方法的,于是再google之,终于找到一篇相关问题的文章,里面说的很细,还提供了几种不同的方案:网址:https://www.sellsbrothers.com/posts/details/12657
自己比较喜欢第2和第3种方案,对比来说,第3种方案比较容易理解和实现。
当然我是用的第3种方案的简单实现,没有从typeinfo接口再继承,还是自己实现了一个类,代理其实接口的IDispatch调用,废话不说了,上代码:
#ifndef _XMULTIDISPIMPL_H_ #define _XMULTIDISPIMPL_H_ #include <atlcom.h> #define INTERFACE_MASK 0xFFFF0000UL #define DISPID_MASK 0x0000FFFFUL template<class tihclass = CComTypeInfoHolder> struct _TIH_ENTRY { tihclass *ptih; // 类型库指针,实现IDispatch调用 DWORD dispEncode; // 函数调用id编码,在GetIdsOfNames函数中对返回的dispid进行编码,尝试解决dispid重复的问题 DWORD offset; // 接口虚函数表偏移,IDispatchImpl<...> }; template <class T, class tihclass = CComTypeInfoHolder> class ATL_NO_VTABLE XMultiDispImpl: public IDispatch { public: typedef _TIH_ENTRY<tihclass> TIH_ENTRY; public: STDMETHOD(GetTypeInfoCount)(UINT* pctinfo) { //TODO: 考虑是否按多个类型库处理 *pctinfo = 1; return S_OK; } STDMETHOD(GetTypeInfo)(UINT itinfo, LCID lcid, ITypeInfo** pptinfo) { //TODO: 考虑是否按多个类型库处理 T* pT = static_cast<T*> (this); TIH_ENTRY* pEntry = pT->GetTypeInfoHolder(); if (pEntry->ptih) { // 默认返回第一个接口的类型库 return pEntry->ptih->GetTypeInfo(itinfo, lcid, pptinfo); } return E_FAIL; } STDMETHOD(GetIDsOfNames)(REFIID riid, LPOLESTR* rgszNames, UINT cNames, LCID lcid, DISPID* rgdispid) { // NOTE: 函数名字不能冲突, // 名字相同时按顺序查找接口映射表中的接口, // 返回第一个匹配的接口函数对应的dispid T* pT = static_cast<T*> (this); TIH_ENTRY* pEntry = pT->GetTypeInfoHolder(); HRESULT hr = DISP_E_UNKNOWNNAME; while (pEntry->ptih != NULL) { hr = pEntry->ptih->GetIDsOfNames(riid, rgszNames, cNames, lcid, rgdispid); if (SUCCEEDED(hr)) { for (UINT i = 0; i < cNames; i++) { rgdispid[i] |= pEntry->dispEncode; } return hr; } else if (hr != DISP_E_UNKNOWNNAME) { return hr; } pEntry++; } return DISP_E_UNKNOWNNAME; } STDMETHOD(Invoke)(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS* pdispparams, VARIANT* pvarResult, EXCEPINFO* pexcepinfo, UINT* puArgErr) { T* pT = static_cast<T*> (this); TIH_ENTRY* pEntry = pT->GetTypeInfoHolder(); HRESULT hr = DISP_E_MEMBERNOTFOUND; if (dispidMember & INTERFACE_MASK) { // 函数id是编码过的,查找对应的接口进行调用,一般是脚本一类的动态调用 while (pEntry->ptih != NULL) { if (pEntry->dispEncode == (dispidMember & INTERFACE_MASK)) { // 找到接口,调用并退出 hr = pEntry->ptih->Invoke((IDispatch*)(((DWORD)pT)+pEntry->offset), (dispidMember & DISPID_MASK), riid, lcid, wFlags, pdispparams, pvarResult, pexcepinfo, puArgErr); return hr; } pEntry++; } } else { // 函数id未编码,逐个接口进行尝试,一般是VC生成的接口类进行的静态调用 // NOTE: 不同的接口,如果存在dispid相同的函数, // 请保证其函数参数个数或者参数类型或者返回值类型不要相同, // 否则可能会调用到错误的接口函数 while (pEntry->ptih != NULL) { hr = pEntry->ptih->Invoke((IDispatch*)(((DWORD)pT)+pEntry->offset), dispidMember, riid, lcid, wFlags, pdispparams, pvarResult, pexcepinfo, puArgErr); if (SUCCEEDED(hr)) { // 调用成功退出 return hr; } pEntry++; } } return DISP_E_MEMBERNOTFOUND; } }; // 映射表宏定义,需要在组件的头文件中引用 #define BEGIN_MULTI_DISPATCH_MAP(CLS) typedef CLS theDerived; static theDerived::TIH_ENTRY* GetTypeInfoHolder() { const DWORD _dwCnt = __COUNTER__; static theDerived::TIH_ENTRY pDispEntries[] = { // 函数id编码,占用id的高16位bit #define MULTI_DISPATCH_ENCODE() (((DWORD)(__COUNTER__) - _dwCnt) << 16) #define MULTI_DISPATCH_ENTRY(theBase) { &theBase::_tih, MULTI_DISPATCH_ENCODE(), offsetofclass(theBase, theDerived) }, #define END_MULTI_DISPATCH_MAP() { NULL, 0UL, 0UL } }; return(pDispEntries); } #endif // sentry
使用方法,在组件类的头文件中,让我们的组件继承我们的类:
class ATL_NO_VTABLE CQMPlugin : public CComObjectRootEx<CComSingleThreadModel>, public CComCoClass<CQMPlugin, &CLSID_QMPlugin>, public ISupportErrorInfo, <span style="color:#3366ff;">public XMultiDispImpl<CQMPlugin>,</span> public IDispatchImpl<IQMPlugin, &IID_IQMPlugin, &LIBID_zdLib, /*wMajor =*/ 1, /*wMinor =*/ 0>, public IDispatchImpl<IQMPluginStandard, &IID_IQMPluginStandard, &LIBID_zdLib, /*wMajor =*/ 1, /*wMinor =*/ 0>
蓝色是要手动添加的代码
在BEGIN_COM_MAP和END_COM_MAP中添加如下代码:
BEGIN_COM_MAP(CQMPlugin) <span style="color:#3366ff;">COM_INTERFACE_ENTRY2(IDispatch, XMultiDispImpl<CQMPlugin>)</span> COM_INTERFACE_ENTRY(IQMPlugin) COM_INTERFACE_ENTRY(IQMPluginStandard) COM_INTERFACE_ENTRY(ISupportErrorInfo) END_COM_MAP()意思是说当外部程序查询IDispatch接口,返回我们实现的类的虚函数表
下面就是要添加接口映射表了,目前感觉这里还是看着不是很爽,暂时没有解决办法:
<span style="color:#3366ff;">typedef IDispatchImpl<IQMPlugin, &IID_IQMPlugin, &LIBID_zdLib, /*wMajor =*/ 1, /*wMinor =*/ 0> TQMPlugin; typedef IDispatchImpl<IQMPluginStandard, &IID_IQMPluginStandard, &LIBID_zdLib, /*wMajor =*/ 1, /*wMinor =*/ 0> TQMPluginStandard; BEGIN_MULTI_DISPATCH_MAP(CQMPlugin) MULTI_DISPATCH_ENTRY(TQMPlugin) MULTI_DISPATCH_ENTRY(TQMPluginStandard) END_MULTI_DISPATCH_MAP()</span>
记住要先typedef 再用MULTI_DISPATCH_ENTRY,不然会编译失败,这也是让人不爽的地方。
其他的可以按正常的ATLCOM接口开发步骤进行开发了。
下面就是注意事项了:
1. 如果要在一个组件里实现多个disp接口,对于每个接口的方法或者属性,不要出现重名的情况,代码中有说明;
2. 函数的dispid可以相同,但是如果dispid相同,请一定让两个函数的参数个数,参数类型或者返加值类型不要全部相同,不然可能调用到错误的接口函数;
3. 理论上这个类实现的多接口是支持静态调用和动态调用的
4. 对于dispid相同的情况,代码是通过在dispid的高16bit设置标志还区别的,对于VBS一类的动态脚本调用是没有问题的,在脚本里可以把组件当成只实现了一个接口
5. 因为使用的dispid的高16bit,所以这个类最多支持65536个接口,同时每个接口的方法和属性不能超过65536个,有需要的可以自行在代码里调整。
最后希望代码能帮助到大家,没有什么比自己的代码被别人认可更让人。。。,找不到形容词了,欢迎大定留言哈。
ATL实现一个组件多个dual接口,multidisp