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

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

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

[cpp] view
plaincopy

HRESULT __stdcall CA::QueryInterface(const IID& iid, void** ppv)  

{     

    if (iid == IID_IUnknown)  

    {  

        *ppv = static_cast<IX*>(this) ;  

    }  

    else if (iid == IID_IX)  

    {  

        *ppv = static_cast<IX*>(this) ;  

    }  

    else if (iid == IID_IY)  

    {  

        *ppv = static_cast<IY*>(this) ;  

    }  

    else  

    {        

        *ppv = NULL ;  

        return E_NOINTERFACE ;  

    }  

    reinterpret_cast<IUnknown*>(*ppv)->AddRef() ;  

    return S_OK ;  

}  

 

这里要注意的一点就是当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
plaincopy

class CA : public IX, public IY  

{  

//......  

private:  

    long m_cRef;  

} ;  

ULONG __stdcall CA::AddRef()  

{  

    return InterlockedIncrement(&m_cRef) ;  

}  

ULONG __stdcall CA::Release()   

{  

    if (InterlockedDecrement(&m_cRef) == 0)  

    {  

        delete this ;  

        return 0 ;  

    }  

    return m_cRef ;  

}  

 

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

这可不是一个简单的工作！

上面那个实现是为整个组件维护引用计数。也可以为每一个接口维护引用计数，

这样可以方便调试程序，也能支持资源的按需获取。

所以，客户必须在使用完某接口后，对该接口调用Release，而不是该组件的其它接口(这应该作为一个规则强制执行!)

无论如何,用户手动调用AddRef,Release都是一件非常麻烦的事情，另外QueryInterface中的void**参数也不是类型安全的。

所以可以使用智能接口指针简化编程(相当于添加中间层)。

在ATL中提供了两个COM接口指针类：CComPtr和CComQIPtr。

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

例如：

[cpp] view
plaincopy

CComPtr<IX> spIX ;  

//......  

//spIX->Release (); //不要这样  

spIX.Release (); //这样才对  

 

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

[cpp] view
plaincopy

template <class T>  

class _NoAddRefReleaseOnCComPtr : public T  

{  

private:  

    STDMETHOD_(ULONG, AddRef)()=0;  

    STDMETHOD_(ULONG, Release)()=0;  

};  

 

_NoAddRefReleaseOnCComPtr禁用了可能的AddRef，Release操作，

然后再这样实现->操作符重载

[cpp] view
plaincopy

_NoAddRefReleaseOnCComPtr<T>* operator->() const throw()  

{  

    ATLASSERT(p!=NULL);  

    return (_NoAddRefReleaseOnCComPtr<T>*)p;  

}  

 

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