???小队 2018-09-19 13:18 采纳率: 0%
浏览 960
已结题

求助:C++内存池Mempool问题

先附代码:
MemoryBlock.h
#pragma once
#include
#include
using namespace std;

struct MemoryBlock
{
USHORT nSize;//可分配内存总大小
USHORT nFree;//可分配内存单元数目
USHORT nFirst;//第一个可用的内存单元位置
MemoryBlock* pNext;//指向下一个内存块
char aData[1];

static void* operator new(size_t, USHORT nTypes, USHORT nUnitSize)
{
   return ::operator new(sizeof(MemoryBlock) + nTypes * nUnitSize);//申请一个内存块空间
}
static void  operator delete(void *p, size_t)
{
    ::operator delete (p);//删除内存块空间
}

MemoryBlock(USHORT nTypes = 1, USHORT nUnitSize = 0);
~MemoryBlock() {}

};

MemoryBlock::MemoryBlock(USHORT nTypes, USHORT nUnitSize)
: nSize(nTypes * nUnitSize),
nFree(nTypes - 1),
nFirst(1),
pNext(0)
{
char * pData = aData;
for (USHORT i = 1; i < nTypes; i++)//初始化1到nTypes-1的指向
{
reinterpret_cast<USHORT>(pData) = i;
pData += nUnitSize;
}
}

MemoryPool.h
#pragma once
#include
#include
#include"MemoryBlock.h"
using namespace std;

class MemoryPool
{
private:
MemoryBlock * pBlock;//指向第一个内存块
USHORT nUnitSize;//一个可分配单元的大小
USHORT nInitSize;//第一个内存块可分配空间数目
USHORT nGrowSize;//后续申请的内存块的可分配空间数目

public:
MemoryPool(USHORT nUnitSize,
USHORT nInitSize = 1024,
USHORT nGrowSize = 256);//单元大小,第一个内存块的可分配空间数目,第二个内存块之后的可分配空间数目 (可设置)
~MemoryPool();

void*           Alloc();//分配内存 
void            Free(void* pfree);//回收内存 
void FreeMemoryBlock(MemoryBlock *pblock);//销毁 

};

MemoryPool.cpp
#include"MemoryPool.h"

const USHORT MEMPOOL_ALIGNMENT = 2;

MemoryPool::MemoryPool(USHORT _nUnitSize,
USHORT _nInitSize, USHORT _nGrowSize)
{
pBlock = NULL;

nInitSize = _nInitSize;

nGrowSize = _nGrowSize;

    if (_nUnitSize > 4)
        nUnitSize = (_nUnitSize + (MEMPOOL_ALIGNMENT - 1)) & ~(MEMPOOL_ALIGNMENT - 1); //将大于4个字节的大小_nUnitSize往上"取整到"大于_nUnitSize的最小的MEMPOOL_ ALIGNMENT的倍数
    else if (_nUnitSize <= 2)
        nUnitSize = 2;              
    else
        nUnitSize = 4;

}

void* MemoryPool::Alloc()
{
if (!pBlock)//如果是第一次申请

{
MemoryBlock* pmb_first = new (nInitSize, nUnitSize)MemoryBlock(nInitSize, nUnitSize);
pBlock = pmb_first;
return (void*)pmb_first->aData;
}

MemoryBlock* pMyBlock = pBlock;
while (pMyBlock && !pMyBlock->nFree)//pMyBlock没走到最后并且当前内存块没有可分配结点
{
    pMyBlock = pMyBlock->pNext;//往之后的内存块遍历
}

if (pMyBlock)//如果找到可分配结点的内存块     
{
    char* pFree = pMyBlock->aData + (pMyBlock->nFirst*nUnitSize);
    pMyBlock->nFirst = *((USHORT*)pFree);
    pMyBlock->nFree--;//可分配节点自减 
    return (void*)pFree;
}
else//如果找不到,此时pMyBlock值为0 
{
    if (!nGrowSize)
        return NULL;

    pMyBlock = new(nGrowSize, nUnitSize) MemoryBlock(nGrowSize, nUnitSize);
    if (!pMyBlock)//创建不成功
        return NULL;

    pMyBlock->pNext = pBlock;//把新建的block放到最前
    pBlock = pMyBlock;

    return (void*)(pMyBlock->aData);
}

}

void MemoryPool::Free(void* pfree)
{
if (pBlock == NULL)
return;
MemoryBlock* pmb_block = pBlock;
MemoryBlock* pmb_preblock = pBlock;
while ((ULONG)pfree<(ULONG)pmb_block->aData ||
(ULONG)pfree>(ULONG)(pmb_block->aData + pmb_block->nSize))//pfree不在当前block中
{
pmb_preblock = pmb_block;//前一个block块
pmb_block = pmb_block->pNext;
if (!pmb_block)
return;
}
pmb_block->nFree++;//可分配数目+1
((USHORT)pfree) = pmb_block->nFirst;
pmb_block->nFirst = (USHORT)((ULONG)pfree - (ULONG)pmb_block->aData) / nUnitSize;
if (pmb_block->nFree*nUnitSize == pmb_block->nSize)//如何该链块为空
{
pmb_preblock->pNext = pmb_block->pNext;
if ((ULONG)pmb_preblock == (ULONG)pBlock)
pBlock = NULL;
delete pmb_block;
}
}

MemoryPool::~MemoryPool(void)
{
if (pBlock)
FreeMemoryBlock(pBlock);
}

void MemoryPool::FreeMemoryBlock(MemoryBlock *pblock)
{
if (pblock->pNext)
FreeMemoryBlock(pblock->pNext);
delete pblock;
pblock = NULL;
}

请问:如何将信息存入内存池,如何从内存池中取出已存入的信息,信息可以是string类或是自己定义的类吗?

  • 写回答

3条回答 默认 最新

  • devmiao 2018-09-19 15:51
    关注
    评论

报告相同问题?

悬赏问题

  • ¥15 YoloV5 第三方库的版本对照问题
  • ¥15 请完成下列相关问题!
  • ¥15 drone 推送镜像时候 purge: true 推送完毕后没有删除对应的镜像,手动拷贝到服务器执行结果正确在样才能让指令自动执行成功删除对应镜像,如何解决?
  • ¥15 求daily translation(DT)偏差订正方法的代码
  • ¥15 js调用html页面需要隐藏某个按钮
  • ¥15 ads仿真结果在圆图上是怎么读数的
  • ¥20 Cotex M3的调试和程序执行方式是什么样的?
  • ¥20 java项目连接sqlserver时报ssl相关错误
  • ¥15 一道python难题3
  • ¥15 牛顿斯科特系数表表示