112 lines
2.2 KiB
C++
112 lines
2.2 KiB
C++
|
// File: refcount.cpp
|
||
|
|
//
|
||
|
|
// RefCount class
|
||
|
|
|
||
|
|
#include "precomp.h"
|
||
|
|
#include "ConfUtil.h"
|
||
|
|
|
||
|
|
|
||
|
|
/* R E F C O U N T */
|
||
|
|
/*-------------------------------------------------------------------------
|
||
|
|
%%Function: Init
|
||
|
|
|
||
|
|
-------------------------------------------------------------------------*/
|
||
|
|
void RefCount::Init(OBJECTDESTROYEDPROC ObjectDestroyed)
|
||
|
|
{
|
||
|
|
ASSERT((!ObjectDestroyed) ||
|
||
|
|
IS_VALID_CODE_PTR(ObjectDestroyed, OBJECTDESTROYEDPROC));
|
||
|
|
|
||
|
|
m_ulcRef = 1;
|
||
|
|
m_ObjectDestroyed = ObjectDestroyed;
|
||
|
|
DbgMsgRefCount("Ref: %08X c=%d (created)", this, m_ulcRef);
|
||
|
|
|
||
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CRefCount));
|
||
|
|
#ifdef DEBUG
|
||
|
|
m_fTrack = FALSE;
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
/*-------------------------------------------------------------------------
|
||
|
|
%%Function: RefCount
|
||
|
|
|
||
|
|
-------------------------------------------------------------------------*/
|
||
|
|
RefCount::RefCount()
|
||
|
|
{
|
||
|
|
Init(NULL);
|
||
|
|
}
|
||
|
|
|
||
|
|
/*-------------------------------------------------------------------------
|
||
|
|
%%Function: RefCount
|
||
|
|
|
||
|
|
-------------------------------------------------------------------------*/
|
||
|
|
RefCount::RefCount(OBJECTDESTROYEDPROC ObjectDestroyed)
|
||
|
|
{
|
||
|
|
Init(ObjectDestroyed);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
RefCount::~RefCount(void)
|
||
|
|
{
|
||
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CRefCount));
|
||
|
|
|
||
|
|
// m_ulcRef may be any value.
|
||
|
|
DbgMsgRefCount("Ref: %08X c=%d (destroyed)", this, m_ulcRef);
|
||
|
|
|
||
|
|
if (m_ObjectDestroyed)
|
||
|
|
{
|
||
|
|
m_ObjectDestroyed();
|
||
|
|
m_ObjectDestroyed = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CRefCount));
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
ULONG STDMETHODCALLTYPE RefCount::AddRef(void)
|
||
|
|
{
|
||
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CRefCount));
|
||
|
|
|
||
|
|
ASSERT(m_ulcRef < ULONG_MAX);
|
||
|
|
m_ulcRef++;
|
||
|
|
DbgMsgRefCount("Ref: %08X c=%d (AddRef)", this, m_ulcRef);
|
||
|
|
|
||
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CRefCount));
|
||
|
|
|
||
|
|
#ifdef DEBUG
|
||
|
|
if (m_fTrack)
|
||
|
|
{
|
||
|
|
DbgMsg(iZONE_OBJECTS, "Obj: %08X c=%d (AddRef) *** Tracking", this, m_ulcRef);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
return m_ulcRef;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
ULONG STDMETHODCALLTYPE RefCount::Release(void)
|
||
|
|
{
|
||
|
|
ASSERT(IS_VALID_STRUCT_PTR(this, CRefCount));
|
||
|
|
|
||
|
|
if (m_ulcRef > 0)
|
||
|
|
{
|
||
|
|
m_ulcRef--;
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef DEBUG
|
||
|
|
if (m_fTrack)
|
||
|
|
{
|
||
|
|
DbgMsg(iZONE_OBJECTS, "Obj: %08X c=%d (Release) *** Tracking", this, m_ulcRef);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
ULONG ulcRef = m_ulcRef;
|
||
|
|
DbgMsgRefCount("Ref: %08X c=%d (Release)", this, m_ulcRef);
|
||
|
|
|
||
|
|
if (! ulcRef)
|
||
|
|
{
|
||
|
|
delete this;
|
||
|
|
}
|
||
|
|
|
||
|
|
return ulcRef;
|
||
|
|
}
|
||
|
|
|