高性能内存池实现

发表于 3年以前 | 总阅读数：336 次

对于从事c/c++开发的人来说，malloc/new再也熟悉不过了，对于堆上的内存分配，都是使用标准库提供的函数来进行内存分配，而这些函数最终也会进入到系统调用(brk等)，每次的内存申请和释放，都可能会涉及到底层内存数据的调整，所以效率会非常低。如果我们一次申请一块很大的内存块，后续所有的内存申请和分配，都是基于这一块内存来进行，这样效率就会提升很多，本文主要就是实现一个高效的固定大小的内存池。

结构

图一

实现

内存块定义：

typedef struct memory_block {
        unsigned int size;
        unsigned int free_size;
        unsigned int first_free;

        struct memory_block *next;
        char a_data[1];
} s_memory_block;

内存池Header:

typedef struct memory_pool {
        unsigned int unit_size;
        unsigned int init_size;
        unsigned int grow_size;
        s_memory_block *block;
} s_memory_pool;

初始化内存池：

//此函数初始化内存池Header
s_memory_pool *memory_pool_create(unsigned int size) {
        s_memory_pool *mp;
        mp = (s_memory_pool*)malloc(sizeof(s_memory_pool));
        mp->first_block = NULL;
        mp->init_size = 10000;
        mp->grow_size = 10000;
        if(size < sizeof(unsigned int))
                mp->obj_size = sizeof(unsigned int);
        //对齐
        mp->unit_size = (size + (MEMPOOL_ALIGNMENT-1)) & ~(MEMPOOL_ALIGNMENT-1);
        return mp;
}

当初始化之后，内存池结构变成：

图二

从函数实现内容来看，是初始化了内存池的头。

内存分配函数：

1、从mp的first_block开始，如果其为空，则表明该内存池为首次创建，需要分配内存块，并在该内存块内进行链式初始化，返回该块的第一小块地址。

2、从first_block开始，查找一个有内存可分配的block，如果有，则分配，并将first_free指向该块的下一个地址。

3、重新建一个block，进行分配，并将该新块插入到mp的头部

void *memory_alloc(s_memory_pool *mp)  {

        register unsigned int i;
        register char *data;
        //unsigned int length;

        if(mp->first_block == NULL)//memory_pool is NULL
        {

                s_memory_block *mb;
                mb = (s_memory_block *)malloc((mp->init_size)*(mp->obj_size) + sizeof(s_memory_block));//create first memory_block
                if(mb == NULL)
                {
                        perror("memory allocate failed!\n");
                        return NULL;
                }

                /* init the first block */
                mb->next = NULL;
                mb->free_size = mp->init_size - 1;
                mb->first_free = 1;
                mb->size = mp->init_size*mp->obj_size;

                mp->first_block = mb;

                data = mb->a_data;

                /* set the mark */
                for(i=1; i<mp->init_size; ++i)
                {
                       //初始化块，链接方式类似于链表
                }

                return (void *)mb->a_data;
        } //如图三

        s_memory_block *pm_block = mp->first_block;

        while((pm_block!=NULL) && (pm_block->free_size==0))
        {
                pm_block = pm_block->next;
        }

        if(pm_block != NULL)
        {
                char *pfree = pm_block->a_data + pm_block->first_free * mp->obj_size;
                //查找一个可用块返回
                // ......

                return (void *)pfree;
        }
        else
        {
                if(mp->grow_size == 0) 
                    return NULL;
                s_memory_block *new_block = (s_memory_block *)malloc((mp->grow_size)*(mp->obj_size) + sizeof(s_memory_block));

                if(new_block == NULL)
                        return NULL;

                data = new_block->a_data;

                for(i=1; i<mp->grow_size; ++i)
                {
                        //链式
                }

               //将新块插入到链表头首部

                return (void *)(new_block->a_data);
        }
}

释放函数：

1、遍历该内存池，查找所要释放的内存块pfree所在的block

2、将该block的first指向该pfree的偏移

3、该pfree的偏移指向之前block的first

注：2、3处相当于链表的插入

inline void* memory_free(s_memory_pool *mp, void *pfree)
{
  if(mp->first_block == NULL)
    return;

  s_memory_block *pm_block = mp->first_block;
  s_memory_block *pm_pre_block = mp->first_block;

  /* research the memory_block which the pfree in */
  while(pm_block && ((unsigned int)pfree<(unsigned int)pm_block->a_data ||
    (unsigned int)pfree>((unsigned int)pm_block->a_data+pm_block->size)))
  {
    //pm_pre_block = pm_block;
    pm_block = pm_block->next;
    if(pm_block == NULL)
      return NULL;
  }

  //获取偏移地址
  unsigned int offset = pfree -(void*) pm_block->a_data;

  if((offset&(mp->obj_size -1)) > 0)
    return pfree;
  //将释放的内存块返回给内存池对应的Block块
  pm_block->free_size++;
  *((unsigned int *)pfree) = pm_block->first_free;
  pm_block->first_free=(unsigned int)(offset/mp->obj_size);

  return NULL;

}