如何为我班级的每个界面单独定义一些功能?

时间:2015-05-20 09:05:51

标签: c++ com

我正在学习COM,我也是C ++的初学者。我定义了三个接口:

#include <objbase.h>

// pure abstract base class
interface IX : IUnknown{  
  virtual void __stdcall Fx() = 0; // pure virtual function
};

// pure abstract base class
interface IY : IUnknown{
  virtual void __stdcall Fy() = 0; // pure virtual function
};

// pure abstract base class
interface IZ : IUnknown{
  virtual void __stdcall Fz() = 0; // pure virtual function
};

extern "C"{
  extern const IID IID_IX;
  extern const IID IID_IY;
  extern const IID IID_IZ;
}

然后我创建了一个实现这些接口的类:

#include <iostream>
#include "IFace.h"

// COM component
class CA :
  public IX,
  public IY,
  public IZ
{
public:
  CA();
  ~CA();
  virtual void __stdcall Fx(); // IX
  virtual void __stdcall Fy(); // IY
  virtual void __stdcall Fz(); // IZ
  // IUnknown (this interface implemented by IX, IY and IZ)
  virtual HRESULT __stdcall QueryInterface(const IID& iid, void** ppv);
  virtual ULONG __stdcall AddRef();
  virtual ULONG __stdcall Release();
private:
  // Each interface has own counter (for convenience of debugging)
  ULONG ix_counter;
  ULONG iy_counter;
  ULONG iz_counter;
};
...

我想为每个接口使用单独的实例计数器(为了方便调试)。在这种情况下,我必须如何定义AddRef()Release()函数?

我可以这样做:

ULONG __stdcall CA::AddRef(){
  // increase interface counter here
}

但是此代码是我IX类的IYIZCA个实例的 common 。如果实例是IX,那么我只需要增加ix_counter计数器。如果实例是IY,那么我只需要增加iy_counter计数器等等。

我试过这样的方法:

virtual HRESULT __stdcall QueryInterface(const IID& iid, void** ppv); // common implementation
virtual ULONG __stdcall IX::AddRef(); // for IX instances
virtual ULONG __stdcall IX::Release(); // for IX instances

virtual ULONG __stdcall IY::AddRef(); // for IY instances
virtual ULONG __stdcall IY::Release(); // for IY instances

virtual ULONG __stdcall IZ::AddRef(); // for IZ instances
virtual ULONG __stdcall IZ::Release(); // for IZ instances

而不是

virtual ULONG __stdcall AddRef();
virtual ULONG __stdcall Release();

我也试过了:

// This is wrong definitions:
ULONG __stdcall CA::IX::AddRef(){
  return InterlockedIncrement(&ix_counter);
}

ULONG __stdcall CA::IX::Release(){
  InterlockedDecrement(&ix_counter);
  ULONG count = ix_counter + iy_counter + iz_counter;
  if (0 == count) delete this;
  return count;
}

而不是

ULONG __stdcall CA::AddRef(){
      // increase interface counter here
    }

但我的最后一次替换是错误的。

如何为每个界面单独定义AddRef()Release()个函数?

3 个答案:

答案 0 :(得分:1)

  

我想为每个接口使用单独的实例计数器(为了方便调试)。在这种情况下我如何定义AddRef()和Release()函数?

这完全违反了COM规则。其中一个要求是,每当您通过指向特定对象的任何类型的任何指针检索IUnknown*时,您将获得相同的指针值。这就是为什么你不能以任何有意义的方式拥有一个单独的计数器。

答案 1 :(得分:1)

您可以拥有一个特定于接口的计数器。您的代码可能如下所示:

// (interfaces defined)

template <typename T, typename Base>
class CUnknownWithCounterT :
    public Base
{
public:
    ULONG m_nCounter;

public:
    CUnknownWithCounterT() :
        m_nCounter(0)
    {
    }
    ULONG __stdcall AddRef()
    {
        m_nCounter++;
        return static_cast<T*>(this)->InternalAddRef();
    }
    ULONG __stdcall Release()
    {
        m_nCounter--;
        return static_cast<T*>(this)->InternalRelease();
    }
};

class CA :
    public CUnknownWithCounterT<CA, IX>,
    public CUnknownWithCounterT<CA, IY>,
    public CUnknownWithCounterT<CA, IZ>
{
public:
    ULONG m_nMainCounter;

public:
    // QueryInterface as stuff...
    ULONG __stdcall InternalAddRef()
    {
        return ++m_nMainCounter;
    }
    ULONG __stdcall InternalRelease()
    {
        return --m_nMainCounter;
    }
};

准备好了:

CA A;
IX* pX = &A;
IY* pY = &A;
IZ* pZ = &A;
pY->AddRef();
pZ->AddRef();
pZ->AddRef();
_tprintf(_T("%d %d %d\n"), 
    A.CUnknownWithCounterT<CA, IX>::m_nCounter, 
    A.CUnknownWithCounterT<CA, IY>::m_nCounter, 
    A.CUnknownWithCounterT<CA, IZ>::m_nCounter, 
    0);
// 0 1 2

但是,您还需要分别覆盖QueryInterface,因为它还以相应计数器需要增量的方式返回指针。

实现将符合COM(m_nMainCounter具有COM的计数器),问题是各个计数器是否准确无法帮助您找到引用泄漏。如果它们在大多数情况下都很有帮助,那么你肯定会对故障排除有用。

答案 2 :(得分:0)

我认为你应该从实现IUnknown的类继承公共实现。这与COM定义的内容相似。