任意大小分配的内存池实现

最近在写多媒体流方面的东西，发现要频繁的分配内存，如果采用new来分配速度会很慢，于是自己做了一个内存池。但是如果要开发任意大小分配的内存池是非常困难的，而且不能保证有new的速度快，在网上看到的内存池一般都是固定大小分配的。虽然任意大小分配的内存池很难实现，但是在某些特殊条件下，就变的很容易了。比如：如果内存释放的顺序是按分配的顺序释放的，这种情况在使用队列来存储大量数据的情况下非常有用。

 

实现这种内存池只需要一段较大的缓冲区和两个指针，一个标示已分配空间头的位置，一个表示已分配空间尾部位置。如果内存池没有足够的空间就退化为使用new来分配。

 

分配实现：

/*

 *  @func:  Alloc

 *  @brief: 分配nSize大小的空间

 *  @param: [in]  ULONG nSize 要分配的空间大小

 *  @ret:   PVOID     分配的空间的起始地址

 */

 

PVOID CRandomMemPool::Alloc(ULONG nSize)

{

       CAutoLock lock(&m_Lock);

 

       PBYTE pPtr = NULL;

       if (m_nTail > m_nHeader || (m_nTail == m_nHeader && m_nPoolLeft > 0))

       {

              if (m_nPoolSize - m_nTail >= nSize)

              {

                     pPtr = m_pPool + m_nTail;

                     m_nTail += nSize;

              }else if (m_nHeader >= nSize)

              {

                     pPtr = m_pPool;

                     m_nTail = nSize;

              }

       }else if (m_nTail < m_nHeader)

       {

              if (m_nHeader - m_nTail >= nSize)

              {

                     pPtr = m_pPool + m_nTail;

                     m_nTail += nSize;

              }

       }

 

       if (pPtr == NULL)

       {

              TRACE("the memory pool is empty, use new to alloc/r/n");

              pPtr = new BYTE[nSize];

       }else

       {

              m_nPoolLeft -= nSize;

       }

      

       return pPtr;

}

 

释放实现：

/*

 *  @func:  Free

 *  @brief: 释放空间，必须指定内存空间大小

 *  @param: [in]  PVOID pPtr    内存空间

 *                  [in]  ULONG uSize 内存空间大小

 *  @ret:   void

 */

void CRandomMemPool::Free(PVOID pPtr, ULONG nSize)

{

       CAutoLock lock(&m_Lock);

 

       if (pPtr >= m_pPool && pPtr < m_pPool + m_nPoolSize)

       {

              if (nSize > m_nPoolSize - m_nHeader)

                     m_nHeader = 0;

 

              //必须按顺序回收

              ASSERT(pPtr == m_pPool + m_nHeader);

             

              m_nHeader += nSize;

              m_nPoolLeft += nSize;

       }else

       {

              delete [] pPtr;

              pPtr = NULL;

       }

      

}