Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

147 lines
3.1 KiB

/*****************************************************************************
*
* (C) COPYRIGHT MICROSOFT CORPORATION, 2000
*
* TITLE: cntutils.inl
*
* VERSION: 1.0
*
* AUTHOR: LazarI
*
* DATE: 23-Dec-2000
*
* DESCRIPTION: Containers and algorithms utility templates (Impl.)
*
*****************************************************************************/
////////////////////////////////////////////////
//
// class CFastHeap<T>
//
// fast cached heap for fixed chunks
// of memory (MT safe)
//
template <class T>
CFastHeap<T>::CFastHeap<T>(int iCacheSize):
#if DBG
m_iPhysicalAllocs(0),
m_iLogicalAllocs(0),
#endif
m_pFreeList(NULL),
m_iCacheSize(iCacheSize),
m_iCached(0)
{
// nothing
}
template <class T>
CFastHeap<T>::~CFastHeap<T>()
{
// delete the cache
while( m_pFreeList )
{
HeapItem *pItem = m_pFreeList;
m_pFreeList = m_pFreeList->pNext;
delete pItem;
#if DBG
m_iPhysicalAllocs--;
#endif
}
#if DBG
ASSERT(0 == m_iPhysicalAllocs);
ASSERT(0 == m_iLogicalAllocs);
#endif
}
template <class T>
HRESULT CFastHeap<T>::Alloc(const T &data, HANDLE *ph)
{
CCSLock::Locker lock(m_csLock);
HRESULT hr = E_INVALIDARG;
if( ph )
{
*ph = NULL;
HeapItem *pi = NULL;
if( m_pFreeList )
{
// we have an item in the cache, just use it
pi = m_pFreeList;
m_pFreeList = m_pFreeList->pNext;
m_iCached--;
ASSERT(m_iCached >= 0);
}
else
{
// no items in the cache, allocate new one
pi = new HeapItem;
#if DBG
if( pi )
{
m_iPhysicalAllocs++;
}
#endif
}
if( pi )
{
pi->data = data;
*ph = reinterpret_cast<HANDLE>(pi);
#if DBG
m_iLogicalAllocs++;
#endif
}
hr = (*ph) ? S_OK : E_OUTOFMEMORY;
}
return hr;
}
template <class T>
HRESULT CFastHeap<T>::Free(HANDLE h)
{
CCSLock::Locker lock(m_csLock);
HRESULT hr = E_INVALIDARG;
if( h )
{
#if DBG
m_iLogicalAllocs--;
#endif
ASSERT(m_iCached >= 0);
HeapItem *pi = reinterpret_cast<HeapItem*>(h);
if( m_iCached < m_iCacheSize )
{
// the number of cached items is less than the
// cache size, so we put this item in the cache
pi->pNext = m_pFreeList;
m_pFreeList = pi;
m_iCached++;
}
else
{
// enough items cached, delete this one
delete pi;
#if DBG
m_iPhysicalAllocs--;
#endif
}
hr = S_OK;
}
return hr;
}
template <class T>
HRESULT CFastHeap<T>::GetItem(HANDLE h, T **ppData)
{
// just return the item ptr. no need to aquire the CS as this is
// readonly function and Alloc/Free cannot invalidate the item.
*ppData = &(reinterpret_cast<HeapItem*>(h)->data);
return S_OK;
}