162 lines
4 KiB
C++
162 lines
4 KiB
C++
/*==========================================================================
|
|
*
|
|
* Copyright (C) 1998 Microsoft Corporation. All Rights Reserved.
|
|
*
|
|
* File: RCBuffer.cpp
|
|
* Content: RefCount Buffers
|
|
*
|
|
*
|
|
* History:
|
|
* Date By Reason
|
|
* ==== == ======
|
|
* 01/12/00 mjn Created
|
|
* 01/31/00 mjn Allow user defined Alloc and Free
|
|
***************************************************************************/
|
|
|
|
#include "dncmni.h"
|
|
#include "RCBuffer.h"
|
|
|
|
|
|
#undef DPF_SUBCOMP
|
|
#define DPF_SUBCOMP DN_SUBCOMP_COMMON
|
|
|
|
|
|
//**********************************************************************
|
|
// Constant definitions
|
|
//**********************************************************************
|
|
|
|
//**********************************************************************
|
|
// Macro definitions
|
|
//**********************************************************************
|
|
|
|
//**********************************************************************
|
|
// Structure definitions
|
|
//**********************************************************************
|
|
|
|
//**********************************************************************
|
|
// Variable definitions
|
|
//**********************************************************************
|
|
|
|
//**********************************************************************
|
|
// Function prototypes
|
|
//**********************************************************************
|
|
|
|
//**********************************************************************
|
|
// Function definitions
|
|
//**********************************************************************
|
|
|
|
|
|
//**********************************************************************
|
|
// ------------------------------
|
|
// CRefCountBuffer::Initialize
|
|
//
|
|
// Entry: CFixedPool <CRefCountBuffer> *pFPRefCountBuffer
|
|
// const DWORD dwBufferSize
|
|
//
|
|
// Exit: Error Code: DN_OK
|
|
// DNERR_OUTOFMEMORY
|
|
// ------------------------------
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "CRefCountBuffer::Initialize"
|
|
|
|
HRESULT CRefCountBuffer::Initialize(CLockedContextClassFixedPool <CRefCountBuffer> *pFPRefCountBuffer,
|
|
PFNALLOC_REFCOUNT_BUFFER pfnAlloc,
|
|
PFNFREE_REFCOUNT_BUFFER pfnFree,
|
|
void *const pvContext,
|
|
const DWORD dwBufferSize)
|
|
{
|
|
DPFX(DPFPREP, 3,"Entered");
|
|
|
|
DNASSERT(pFPRefCountBuffer != NULL);
|
|
DNASSERT((pfnAlloc == NULL && pfnFree == NULL) || (pfnAlloc != NULL && pfnFree != NULL));
|
|
|
|
m_pFPOOLRefCountBuffer = pFPRefCountBuffer;
|
|
|
|
if (dwBufferSize)
|
|
{
|
|
if (pfnAlloc)
|
|
{
|
|
m_pfnAlloc = pfnAlloc;
|
|
m_pfnFree = pfnFree;
|
|
m_pvContext = pvContext;
|
|
}
|
|
else
|
|
{
|
|
m_pfnAlloc = RefCountBufferDefaultAlloc;
|
|
m_pfnFree = RefCountBufferDefaultFree;
|
|
}
|
|
m_dnBufferDesc.pBufferData = static_cast<BYTE*>((pfnAlloc)(m_pvContext,dwBufferSize));
|
|
if (m_dnBufferDesc.pBufferData == NULL)
|
|
{
|
|
return(DPNERR_OUTOFMEMORY);
|
|
}
|
|
m_dnBufferDesc.dwBufferSize = dwBufferSize;
|
|
}
|
|
|
|
return(DPN_OK);
|
|
}
|
|
|
|
/* REMOVE
|
|
//**********************************************************************
|
|
// ------------------------------
|
|
// CRefCountBuffer::ReturnSelfToPool
|
|
//
|
|
// Entry: Nothing
|
|
//
|
|
// Exit: Nothing
|
|
// ------------------------------
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "CRefCountBuffer::ReturnSelfToPool"
|
|
|
|
void CRefCountBuffer::ReturnSelfToPool( void )
|
|
{
|
|
DPFX(DPFPREP, 3,"Entered");
|
|
|
|
DNASSERT(m_lRefCount == 0);
|
|
m_pFPOOLRefCountBuffer->Release( this );
|
|
}
|
|
|
|
*/
|
|
//**********************************************************************
|
|
// ------------------------------
|
|
// CRefCountBuffer::Release
|
|
//
|
|
// Entry: Nothing
|
|
//
|
|
// Exit: Nothing
|
|
// ------------------------------
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "CRefCountBuffer::Release"
|
|
|
|
void CRefCountBuffer::Release( void )
|
|
{
|
|
DNASSERT(m_lRefCount > 0);
|
|
if ( InterlockedDecrement( &m_lRefCount ) == 0 )
|
|
{
|
|
if (m_pfnFree && m_dnBufferDesc.pBufferData)
|
|
{
|
|
(*m_pfnFree)(m_pvContext,m_dnBufferDesc.pBufferData);
|
|
m_dnBufferDesc.pBufferData = NULL;
|
|
m_dnBufferDesc.dwBufferSize = 0;
|
|
}
|
|
ReturnSelfToPool();
|
|
}
|
|
}
|
|
|
|
|
|
PVOID RefCountBufferDefaultAlloc(void *const pv,const DWORD dwSize)
|
|
{
|
|
return(DNMalloc(dwSize));
|
|
}
|
|
|
|
|
|
void RefCountBufferDefaultFree(PVOID pv,PVOID pvBuffer)
|
|
{
|
|
DNFree(pvBuffer);
|
|
}
|
|
|
|
|