(hryEngine)引擎基础组成——高效内存池(lookaside list)

转自 http://blogold.chinaunix.net/u3/102180/showart_2065839.html


//  工程名称:网络游戏服务器端引擎《hryEngine》 
//  参考项目:ReactOS、Apache、ACE
【路过的朋友请留下您宝贵的意见,谢谢!】

#ifndef _LOOKASIDELIST_
#define _LOOKASIDELIST_
/******************************************************************************
* LookasideList Functions *
******************************************************************************/

#include "..\Define.h"

#define LAL_MEMORY_BLOCK_SIZE    (8)
typedef struct _LAL_MEMORY_BLOCK
{    
    _LAL_MEMORY_BLOCK*    sNext;                    // 0

    DWORD                dwMemSize;                // 4

    CHAR                pMemory[1];                // 8

    
    _LAL_MEMORY_BLOCK()
    {
        memset(this,0,sizeof(_LAL_MEMORY_BLOCK));
    }
} LAL_MEMORY_BLOCK, *PLAL_MEMORY_BLOCK;

typedef struct _LAL_BLOCK_HEADER
{
    PLAL_MEMORY_BLOCK    pEntry;        // 0

    WORD                wLen;        // 4


    _LAL_BLOCK_HEADER()
    {
        memset(this,0,sizeof(_LAL_BLOCK_HEADER));
    }
} LAL_BLOCK_HEADER, *PLAL_BLOCK_HEADER;

#define LAL_MEMORY_OBJECT_SIZE    (14)
typedef struct _LAL_MEMORY_OBJECT
{
    _LAL_MEMORY_OBJECT*    pPrev;            // 0    for allocated list free

    _LAL_MEMORY_OBJECT*    pNext;            // 4

    PVOID                pParent;        // 8            

    USHORT                usIndex:15;        // 12

    USHORT                usStatus:1;        //            0: free 1: allocated

    CHAR                pMemory[1];        // 14


    _LAL_MEMORY_OBJECT()
    {
        memset(this,0,sizeof(_LAL_MEMORY_OBJECT));
    }    
} LAL_MEMORY_OBJECT, *PLAL_MEMORY_OBJECT;

typedef struct _LAL_OBJECT_HEADER
{
    PLAL_MEMORY_OBJECT    pEntry;        // 0

    DWORD                wLen;        // 4


    _LAL_OBJECT_HEADER()
    {
        memset(this,0,sizeof(_LAL_OBJECT_HEADER));
    }
} LAL_OBJECT_HEADER, *PLAL_OBJECT_HEADER;


class LOOKASIDE_LIST 
{
private:
    LAL_BLOCK_HEADER    m_sMemoryBlockList;
    DWORD                m_dwObjectSize;
    WORD                m_wMaxObjectNum;
    WORD                m_wDefaultObjectNum;
    WORD                m_wIncreaseObjectNum;

    BOOL
    AddBlock(
        IN OUT PLAL_BLOCK_HEADER pBlockHeader,
        IN OUT PLAL_OBJECT_HEADER pObjectHeader,
        IN CONST DWORD dwObjSize,
        IN CONST WORD wObjNum);
public:
    enum
    {
        PER_BLOCK_OBJ_MAX_NUM = 512,
        MEM_STATUS_FREE = 0,
        MEM_STATUS_ALLOCATED = 1,
    };
    LAL_OBJECT_HEADER    m_sFreeObjectList;
    LAL_OBJECT_HEADER    m_sAllocatedObjectList;
    DWORD                m_dwRealObjectSize;
public:
    LOOKASIDE_LIST();
    ~LOOKASIDE_LIST();
    BOOL 
    InitializeLookasideList(
        IN CONST DWORD dwObjectSize,
        IN CONST WORD wMaxObjNum,
        IN CONST WORD wDefaultObjNum,
        IN CONST WORD wIncreaseObjectNum);
    BOOL
    IncreaseBlock(VOID);
};

static 
inline 
PLAL_MEMORY_OBJECT
BodyToHeader ( 
             IN PVOID addr );

PVOID
LalAlloc(
         IN LOOKASIDE_LIST* pLookaside);

VOID
LalFree(
         IN LOOKASIDE_LIST* pLookaside,
         IN PVOID pMemory);


#endif


/*
* PROJECT: hryEngine.LookasideList
* PURPOSE: memory pool by lookaside list
* PROGRAMMERS: Liao Huaifu(279670696@qq.com,deuglhf@hotmail.com)
                 Shanghai
                 09/9-26~09/10-8
*/


#include <windows.h>
#include <malloc.h>
#include "LookasideList.h"

LOOKASIDE_LIST::LOOKASIDE_LIST()
{
    memset(this,0,sizeof(LOOKASIDE_LIST));
}

LOOKASIDE_LIST::~LOOKASIDE_LIST()
{
    if (m_sMemoryBlockList.pEntry && m_sMemoryBlockList.wLen)
    {
        PLAL_MEMORY_BLOCK pBlock = m_sMemoryBlockList.pEntry;
        while(pBlock)
        {
            PLAL_MEMORY_BLOCK tmpBlock = pBlock->sNext;
            delete[] (char*)pBlock;
            pBlock = NULL;
            pBlock = tmpBlock;
        }
    }
}

BOOL 
LOOKASIDE_LIST::InitializeLookasideList(
                        IN CONST DWORD dwObjectSize,
                        IN CONST WORD wMaxObjNum,
                        IN CONST WORD wDefaultObjNum,
                        IN CONST WORD wIncreaseObjectNum)
{
    m_dwObjectSize = dwObjectSize;
    m_wMaxObjectNum = wMaxObjNum;
    m_wDefaultObjectNum = wDefaultObjNum;
    m_wIncreaseObjectNum = wIncreaseObjectNum;

    WORD num = 0;
    (wDefaultObjNum%PER_BLOCK_OBJ_MAX_NUM == 0)? 
        num = wDefaultObjNum/PER_BLOCK_OBJ_MAX_NUM : 
        num = wDefaultObjNum/PER_BLOCK_OBJ_MAX_NUM + 1;
    for(int i=0;i<num;i++)
    {
        AddBlock(&m_sMemoryBlockList,&m_sFreeObjectList,m_dwObjectSize,PER_BLOCK_OBJ_MAX_NUM);
    }
    return TRUE;
}

BOOL
LOOKASIDE_LIST::AddBlock(
         IN OUT PLAL_BLOCK_HEADER pBlockHeader,
         IN OUT PLAL_OBJECT_HEADER pObjectHeader,
         IN CONST DWORD dwObjSize,
         IN CONST WORD wObjNum)
{
    if (!pBlockHeader || !pObjectHeader || !dwObjSize || !dwObjSize)
    {
        return FALSE;
    }

    m_dwRealObjectSize = dwObjSize + LAL_MEMORY_OBJECT_SIZE;
    DWORD dwBlockSize = wObjNum * m_dwRealObjectSize + LAL_MEMORY_BLOCK_SIZE;
    char *pMemory = new char[dwBlockSize];
    if (!pMemory)
    {
        return FALSE;
    }

    PLAL_MEMORY_BLOCK pBlock = (PLAL_MEMORY_BLOCK)pMemory;
    pBlock->sNext = pBlockHeader->pEntry;
    pBlock->dwMemSize = dwBlockSize;
    //pBlock->pMemory = pMemory + LAL_MEMORY_BLOCK_SIZE;


    pBlockHeader->pEntry = pBlock;
    pBlockHeader->wLen++;

    // init object memory chain table

    DWORD dwOffset = 0;     
    for(int i=0;i<wObjNum;i++)
    {
        dwOffset = LAL_MEMORY_BLOCK_SIZE + (i*m_dwRealObjectSize);
        PLAL_MEMORY_OBJECT pObject = (PLAL_MEMORY_OBJECT)(pMemory + dwOffset);
        pObject->pNext = pObjectHeader->pEntry;
        pObject->pPrev = NULL;
        pObjectHeader->pEntry = pObject;
        pObjectHeader->wLen++;
        pObject->pParent = pMemory;
        pObject->usIndex = i+1;
        pObject->usStatus = MEM_STATUS_FREE;
        //pObject->pMemory = pMemory + dwOffset + LAL_MEMORY_OBJECT_SIZE;

    }
    return TRUE;
}

BOOL
LOOKASIDE_LIST::IncreaseBlock(VOID)
{
    if (m_wMaxObjectNum && m_sAllocatedObjectList.wLen>=m_wMaxObjectNum)
    {
        return FALSE;
    }
    return AddBlock(&m_sMemoryBlockList,&m_sFreeObjectList,m_dwObjectSize,m_wIncreaseObjectNum);
}

PVOID
LalAlloc(
         IN LOOKASIDE_LIST* pLookaside)
{
    if (!pLookaside)    return    NULL;
    if (!pLookaside->m_sFreeObjectList.pEntry)
    {
        if (!pLookaside->IncreaseBlock())
            return NULL;
    }

    PLAL_MEMORY_OBJECT pVel = pLookaside->m_sFreeObjectList.pEntry;
    pLookaside->m_sFreeObjectList.wLen--;
    pLookaside->m_sFreeObjectList.pEntry = pVel->pNext;
    pVel->usStatus = LOOKASIDE_LIST::MEM_STATUS_ALLOCATED;
    pVel->pNext = pVel->pPrev = NULL;
#ifdef DEBUG_VER
    if (pLookaside->m_sAllocatedObjectList.pEntry)
    {
        pVel->pNext = pLookaside->m_sAllocatedObjectList.pEntry;
        pLookaside->m_sAllocatedObjectList.pEntry->pPrev = pVel;    
        pLookaside->m_sAllocatedObjectList.pEntry = pVel;
    }
    pLookaside->m_sAllocatedObjectList.pEntry = pVel;
    pLookaside->m_sAllocatedObjectList.wLen++;
#endif
    return pVel->pMemory;
}

VOID
LalFree(
         IN LOOKASIDE_LIST* pLookaside,
         IN PVOID pMemory)
{
    if (!pLookaside || !pMemory) return;
    
    PLAL_MEMORY_OBJECT pVel = BodyToHeader(pMemory);
    PCHAR pUnit = (PCHAR)pVel;
    PCHAR pL = ((PCHAR)pVel->pParent) + (pVel->usIndex-1)*(pLookaside->m_dwRealObjectSize);
    PCHAR pH = ((PCHAR)pVel->pParent) + (pVel->usIndex+1)*(pLookaside->m_dwRealObjectSize);
    if (pUnit > pL && pUnit < pH)
    {
        if (pVel->usStatus)
        {
#ifdef DEBUG_VER
            PLAL_MEMORY_OBJECT pPrev = pVel->pPrev;
            PLAL_MEMORY_OBJECT pNext = pVel->pNext;
            if (pPrev)    
                pPrev->pNext = pNext;
            if (pNext)    
                pNext->pPrev = pPrev;
            else // pPrev = pNext = null

                pLookaside->m_sAllocatedObjectList.pEntry = NULL;    
            pLookaside->m_sAllocatedObjectList.wLen--;
#endif
            pVel->usStatus = LOOKASIDE_LIST::MEM_STATUS_FREE;    
            pVel->pPrev = NULL;
            pVel->pNext = pLookaside->m_sFreeObjectList.pEntry;     
            pLookaside->m_sFreeObjectList.pEntry = pVel;
            pLookaside->m_sFreeObjectList.wLen++;
        }
        else
        {
            // memory had been destroy

        }
    }
}

static 
inline 
PLAL_MEMORY_OBJECT
BodyToHeader ( 
             IN PVOID addr )
{
    return (PLAL_MEMORY_OBJECT)(((char*)addr) - LAL_MEMORY_OBJECT_SIZE);
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值