高效内存池的设计方案[c语言]

一、前言概述

本人在转发的博文《内存池的设计和实现》中，详细阐述了系统默认内存分配函数malloc/free的缺点，以及进行内存池设计的原因，在此不再赘述。通过对Nginx内存池以及《内存池的设计和实现》的分析后，现提出一种性能更优（申请/释放内存时间复杂度为O(1)）的内存池的设计方案。如有不妥之处，欢迎指正！如有其他的内存池的设计方案，欢迎共同分享和探讨。

二、结构设计

2.1 内存池结构

// 内存池结构体
typedef struct
{
int unitsize; // 内存单元大小，即unit的大小
int initnum; // 初始内存单元的数目
int grownum; // 每次新增内存单元的数目
int totalnum; // 内存单元总数
memblock_t *block; // memblock_t链表头
char *idleunit; // 空闲内存单元链表头
#if defined(__MEMPOOL_THREAD_SUPPORT__)
pthread_mutex_t lock; // 多线程加锁 —— 建议使用某种机制，消除多线程竞争加锁！
#endif /*__MEMPOOL_THREAD_SUPPORT__*/
}mempool_t;

// 内存块结构体
typdef struct
{
int unitnum; // 内存块总数
int idlenum; // 空闲内存块数
mempool_t *pool; // 宿主：所属mempool_t
char *lastunit; // 结束块地址(此变量可删除)
memblock_t *next; // 下一个memblock_t
}memblock_t;

// 内存单元信息
typedefstruct
{
memblock_t *block; // 属主：内存单元所属memblock_t
char *next; // 下一块内存块地址
}memunit_info_t;

2.2 总体结构

内存池的总体结构图为：

图1 总体结构图

2.3 运行机制

此内存池的运行机制如下：

1）将每一个内存单元的大小固定化，可提高内存分配效率。比如：内存单元分别为：{4, 8, 12, 16, 20, 24, 28, 32, 36, 40, 44, 48, 52, 56, 60, …}（单位：byte, 大小为4byte的整数倍）。Mempool_t之间是通过数组形式组织的，其大体结构如下：（注：为了明确Mempool_t之间的关系，未标出其他变量之间的关系）

图2 Mempool_t数组

注：为提高效率，通过数组存储Mempool_t，在申请内存空间时，可通过偏移量快速定位使用哪个大小的内存池。

2）内存池实际可供分配的内存单元是在Memblock_t中，当所有Memblock_t中的内存单元被使用完后，则需申请开辟一个新的Memblock_t，并加入到Memblock_t链表之中。Memblock_t的组织方式为：（注：为了明确Memblock_t之间的关系，未标出其他变量之间的关系）

图3 Memblock_t链表

3）使用链表组织空闲内存单元，可大大提高内存分配/释放时的效率（时间复杂度为O(1)）。Mempool_t中的idleunit是空闲内存单元的链表头。空闲内存单元的组织形式如下：（注：为明确空闲内存单元之间的关系，未标明其他变量之间的关系）

图4 空闲内存单元链表

说明：Memblock_t中的用红色数字标记的内存单元代表已被分配内存单元，用绿色数字标记的内存单元代表空闲内存单元。

4）当申请内存时，将idleunit指向的内存单元踢出空闲内存单元链表，并idleunit指向内存单元的后继，再返回该内存单元的地址。以图4为例，申请内存块后，空闲内存单元链表如图所示：（注：请对比与图4之间的变化）

图5 内存申请图

注：当申请的内存空间size比所有的内存单元都大时，则通过malloc（）向OS申请size+sizeof(memunit_info_t)的内存空间。

5）当释放内存单元unitn时，将unitn的后继改为idleunit的指向，同时将idleunit指向要释放的内存单元unitn。以图4为例，释红色数字标记的内存单元3后，空闲内存单元链表如图所示：（注：请对比与图4之间的变化）

图6 内存释放图

6）内存单元是通过链表形式进行组织管理的，因此，每个内存单元有额外的空间用来存放组织链表的信息。将图4进一步展开：（注：请结合图4一起看）

图7 内存单元内部结构

说明：

1. 每个内存单元的内部结构：memunit_info_t结构+unitsize大小的空间。每个内存单元的大小为：sizeof(memunit_info_t)+unitsize；

2. idleunit指向的是内存单元的data；空闲内存单元的next指向的是后继内存单元的data，无后继则为NULL；已分配的内存单元的next始终为NULL。

3. 内存单元的block指向宿主Memblock_t，这可快速的确定对当前内存单元属于哪个Memblock_t，再通过Memblock_t中的pool，可快速获知属于哪个Mempool_t。

4. 在分配内存时，返回给用户的是data的地址，而不是内存单元的地址。

7）在释放内存单元时，为使被释放内存单元加入空闲内存单元链表，可通过内存单元的block获知所属Memblock_t，再通过pool获知所属Mempool_t，因此，便可知空闲内存单元链表头idleunit，此时便可将被释放的内存单元加入空闲链表。

图8 所属Mempool_t

2.4 优缺点

通过对以上几点的分析，可知此内存池有以下优缺点：

优点：

1. 定位内存池的时间复杂度为O(1)

内存单元可申请使用的空间大小为4的整数倍，并依次递增。因此，定位内存池的算法:(n为内存池数组下标)

n = (size>>2) - 1;
if(0 != (size&(4-1))) n++;

2. 申请和释放内存的时间复杂度为O(1)

3. 内存碎片较小

4. 较小的互斥粒度：申请空间时，每次只锁住对应的mempool_t的内存池，依然可以申请其他size的内存池空间（注：如果再加入为每个线程分配一个内存池对象的机制，则可达到零互斥零竞争。这样的话，可不使用互斥机制，同时能够进一步提高性能）

缺点：

1. 内存单元的实际大小要比unitsize大sizeof(memunit_info_t)

2. 空闲内存单元链表中的内存单元是乱序串联的，因此会造成即使空闲内存单元个数超过单个Memblock_t内存单元总数时，可能依然无法释放任何一个Memblock_t。