COM学习笔记3_IUnknown的三个接口函数

对于COM组件，正确处理IUnknown的三个接口函数非常重要。

对于QueryInterface，一个参考处理如下：

[cpp]  view plain copy

1. HRESULT __stdcall CA::QueryInterface(const IID& iid, void** ppv)  
2. {     
3.     if (iid == IID_IUnknown)  
4.     {  
5.         *ppv = static_cast<IX*>(this) ;  
6.     }  
7.     else if (iid == IID_IX)  
8.     {  
9.         *ppv = static_cast<IX*>(this) ;  
10.     }  
11.     else if (iid == IID_IY)  
12.     {  
13.         *ppv = static_cast<IY*>(this) ;  
14.     }  
15.     else  
16.     {        
17.         *ppv = NULL ;  
18.         return E_NOINTERFACE ;  
19.     }  
20.     reinterpret_cast<IUnknown*>(*ppv)->AddRef() ;  
21.     return S_OK ;  
22. }  

 

这里要注意的一点就是当iid == IID_IUnknown时，返回的并不是
static_cast<IUnknown*>(this) ;
而是
static_cast<IX*>(this) ;
这是因为把this转为IUnknown*是不明确的(IX和IY都从IUnknown派生)。
另外IX和IY并不按虚拟方式从IUnknown派生，是因为虚拟派生可能产生与COM不兼容的vtbl。

QueryInterface可以有不同的实现，只需要遵循一定的规则。这些规则可见《COM技术内幕》3.2节。

AddRef和Release的一个简单实现如下：

[cpp]  view plain copy

1. class CA : public IX, public IY  
2. {  
3. //......  
4. private:  
5.     long m_cRef;  
6. } ;  
7. ULONG __stdcall CA::AddRef()  
8. {  
9.     return InterlockedIncrement(&m_cRef) ;  
10. }  
11. ULONG __stdcall CA::Release()   
12. {  
13.     if (InterlockedDecrement(&m_cRef) == 0)  
14.     {  
15.         delete this ;  
16.         return 0 ;  
17.     }  
18.     return m_cRef ;  
19. }  

 

组件的实现很简单，但客户端使用起来可就不那么容易，客户必须记得用完每一个接口指针后调用Release。
这可不是一个简单的工作！

上面那个实现是为整个组件维护引用计数。也可以为每一个接口维护引用计数，
这样可以方便调试程序，也能支持资源的按需获取。
所以，客户必须在使用完某接口后，对该接口调用Release，而不是该组件的其它接口(这应该作为一个规则强制执行!)

无论如何,用户手动调用AddRef,Release都是一件非常麻烦的事情，另外QueryInterface中的void**参数也不是类型安全的。
所以可以使用智能接口指针简化编程(相当于添加中间层)。
在ATL中提供了两个COM接口指针类：CComPtr和CComQIPtr。

注：若需要手动释放智能接口指针所指向的接口，不能通过简单通过->转发Release操作。
例如：

[cpp]  view plain copy

1. CComPtr<IX> spIX ;  
2. //......  
3. //spIX->Release (); //不要这样  
4. spIX.Release (); //这样才对  

 

为了避免用户像这样spIX->Release ()调用Release，CComPtr在重载->操作符时，不是简单把内部接口指针返回，而是定义了一个类：

[cpp]  view plain copy

1. template <class T>  
2. class _NoAddRefReleaseOnCComPtr : public T  
3. {  
4. private:  
5.     STDMETHOD_(ULONG, AddRef)()=0;  
6.     STDMETHOD_(ULONG, Release)()=0;  
7. };  

 

_NoAddRefReleaseOnCComPtr禁用了可能的AddRef，Release操作，
然后再这样实现->操作符重载

[cpp]  view plain copy

1. _NoAddRefReleaseOnCComPtr<T>* operator->() const throw()  
2. {  
3.     ATLASSERT(p!=NULL);  
4.     return (_NoAddRefReleaseOnCComPtr<T>*)p;  
5. }  

 

这样如果用户像这样操作spIX->Release ()会遭遇一个编译错误。