初识nginx——内存池篇

为了自身使用的方便，Nginx封装了很多有用的数据结构，比如ngx_str_t ,ngx_array_t, ngx_pool_t 等等，对于内存池，nginx设计的十分精炼，值得我们学习，本文介绍内存池基本知识，nginx内存池的结构和关键代码，并用一个实际的代码例子作了进一步的讲解

一、内存池概述

内存池是在真正使用内存之前，预先申请分配一定数量的、大小相等(一般情况下)的内存块留作备用。当有新的内存需求时，就从内存池中分出一部分内存块，若内存块不够用时，再继续申请新的内存。

内存池的好处有减少向系统申请和释放内存的时间开销，解决内存频繁分配产生的碎片，提示程序性能，减少程序员在编写代码中对内存的关注等

目前一些常见的内存池实现方案有STL中的内存分配区，boost中的object_pool,nginx中的ngx_pool_t,google的开源项目TCMalloc等

二、nginx内存池综述

nginx为每一个层级都会创建一个内存池，进行内存的管理，比如一个模板，tcp连接，http请求等，在对应的生命周期结束的时候会摧毁整个内存池，把分配的内存一次性归还给操作系统。

在分配的内存上，nginx有小块内存和大块内存的概念，小块内存 nginx在分配的时候会尝试在当前的内存池节点中分配，而大块内存会调用系统函数malloc向操作系统申请

在释放内存的时候，nginx没有专门提供针对释放小块内存的函数，小块内存会在ngx_destory_pool 和 ngx_reset_pool的时候一并释放

区分小块内存和大块内存的原因有2个，

1、针对大块内存如果它的生命周期远远短于所属的内存池，那么提供一个单独的释放函数是十分有意义的，但不区分大块内存和小块内存，针对大的内存块便会无法提前释放了

2、大块内存与小块内存的界限是一页内存（p->max = (size < NGX_MAX_ALLOC_FROM_POOL) ? size : NGX_MAX_ALLOC_FROM_POOL，NGX_MAX_ALLOC_FROM_POOL的值通过调用getpagesize()获得），大于一页的内存在物理上不一定是连续的，所以如果分配的内存大于一页的话，从内存池中使用，和向操作系统重新申请效率差不多是等价的

nginx内存池提供的函数主要有以下几个

三、nginx内存池详解

nginx使用了ngx_pool_s用于表示整个内存池对象,ngx_pool_data_t表示单个内存池节点的分配信息，ngx_pool_large_s表示大块内存

它们的结构和含义如下

struct ngx_pool_large_s {
ngx_pool_large_t *next;
void *alloc;
};

next: 指向下一个大块内存

alloc：指向分配的大块内存

struct ngx_pool_s {
ngx_pool_data_t d;
size_t max;
ngx_pool_t *current;
ngx_chain_t *chain;
ngx_pool_large_t *large;
ngx_pool_cleanup_t *cleanup;
ngx_log_t *log;
};

d:内存池的节点的数据分配情况

max: 单个内存池节点容量的最大值

current: 指向当前的内存池节点

chain：指向一个ngx_chain_t结构

large：指向大块内存链表

cleanup:释放内存池的callback

log：用于输出日志

typedef struct {
u_char *last;
u_char *end;
ngx_pool_t *next;
ngx_uint_t failed;
} ngx_pool_data_t;

last: 内存池节点已分配的末位地址，下一次分配会尝试从此开始

end：内存池节点的结束位置

next：next指向下一个内存池节点

failed: 当前内存池节点分配失败次数

nginx 内存池示意图1

在分配内存的时候，nginx会判断当前要分配的内存是小块内存还是大块内存，大块内存调用ngx_palloc_large进行分配，小块内存nginx先会尝试从内存池的当前节点(p->current)中分配，如果内存池当前节点的剩余空间不足，nginx会调用ngx_palloc_block新创建一个内存池节点并从中分配，

如果内存池当前节点的分配失败次数已经大于等于6次（p->d.failed++ > 4）,则将当前内存池节点前移一个

(这里有个需要注意的地方，当当前内存节点的剩余空间不够分配时，nginx会重新创建一个ngx_pool_t对象，并且将pool.d->next指向新的ngx_pool_t,新分配的ngx_pool_t对象只用到了ngx_pool_data_t区域，并没有头部信息，头部信息部分已经被当做内存分配区域了)

nginx 内存池示意图2(新建了一个内存池节点和分配了2个大块内存，其中一个已经释放)

关键代码

创建内存池代码

ngx_pool_t *

ngx_create_pool(size_t size, ngx_log_t *log)

{

ngx_pool_t *p;

p = ngx_memalign(NGX_POOL_ALIGNMENT, size, log);//间接调用了posix_memalign分配内存

if (p == NULL) {

return NULL;

}

p->d.last = (u_char *) p + sizeof(ngx_pool_t);//初始的前面几个字节用于储存内存池头部信息,所以下一次分配的开始应该前移头部的大小

p->d.end = (u_char *) p + size;//内存池节点的结尾

p->d.next = NULL;//由于当前内存池只有一个节点所以next为NULL

p->d.failed = 0;

size = size - sizeof(ngx_pool_t);

p->max = (size < NGX_MAX_ALLOC_FROM_POOL) ? size : NGX_MAX_ALLOC_FROM_POOL;//设置小块内存和大块内存的判断标准

p->current = p;

p->chain = NULL;

p->large = NULL;

p->cleanup = NULL;

p->log = log;

return p;

}

ngx_palloc分配函数代码

void *

ngx_palloc(ngx_pool_t *pool, size_t size)

{

u_char *m;

ngx_pool_t *p;

if (size <= pool->max) //判断是小块内存还是大块内存

{

p = pool->current;

do {

m = ngx_align_ptr(p->d.last, NGX_ALIGNMENT);

if ((size_t) (p->d.end - m) >= size) {//尝试在已有的内存池节点中分配内存

p->d.last = m + size;

return m;

}

p = p->d.next;

} while (p);

return ngx_palloc_block(pool, size);//当前已有节点都分配失败，创建一个新的内存池节点

}

return ngx_palloc_large(pool, size);//分配大块内存

}

消耗内存池

void

ngx_destroy_pool(ngx_pool_t *pool)

{

ngx_pool_t *p, *n;

ngx_pool_large_t *l;

ngx_pool_cleanup_t *c;

for (c = pool->cleanup; c; c = c->next) {

if (c->handler) {

ngx_log_debug1(NGX_LOG_DEBUG_ALLOC, pool->log, 0,"run cleanup: %p", c);

c->handler(c->data);//调用需要在内存池释放时同步调用的方法

}

for (l = pool->large; l; l = l->next) {//释放大块内存

ngx_log_debug1(NGX_LOG_DEBUG_ALLOC, pool->log, 0, "free: %p", l->alloc);

if (l->alloc) {

ngx_free(l->alloc);

}

#if (NGX_DEBUG)

/*

* we could allocate the pool->log from this pool

* so we cannot use this log while free()ing the pool

*/

for (p = pool, n = pool->d.next; /* void */; p = n, n = n->d.next) {

ngx_log_debug2(NGX_LOG_DEBUG_ALLOC, pool->log, 0,"free: %p, unused: %uz", p, p->d.end - p->d.last);

if (n == NULL) {

break;

}

#endif

for (p = pool, n = pool->d.next; /* void */; p = n, n = n->d.next) {

ngx_free(p);//间接调用free释放内存

if (n == NULL) {

break;

}

四、示例代码

这里是直接替换了原有nginx代码的main函数 (src/core/nginx.c)

void print_pool(ngx_pool_t *pool)

{

if (pool->large != NULL)

{

printf("has large memory\n");

for(ngx_pool_large_t* i = pool->large; i!=NULL; i = i->next)

{

printf("\t\tlarge next=0x%x\n", i->next);

printf("\t\tlarge alloc=0x%x\n", i->alloc);

}

int i=1;

while(pool)

{

printf("pool=0x%x,index:%d\n", pool, i++);

printf("\t\tlast=0x%x\n", (pool->d).last);

printf("\t\tend=0x%x\n",(pool->d).end);

printf("\t\tnext=0x%x\n",(pool->d).next);

printf("\t\tfailed=%d\n",pool->d.failed);

printf("\t\tmax=%d\n",pool->max);

printf("\t\tcurrent=0x%x\n",pool->current);

printf("\t\tchain=0x%x\n",pool->chain);

printf("\t\tlarge=0x%x\n",pool->large);

printf("\t\tcleanup=0x%x\n",pool->cleanup);

printf("\t\tlog=0x%x\n",pool->log);

printf("\t\tavailable pool memory=%d\n", pool->d.end-pool->d.last);

printf("\n");

pool=pool->d.next;

}

void print_array(int *a,int size)

{

for(int i=0; i<size; i++)

{

printf("%d,",a[i]);

}

printf("\n");

}

int main()

{

ngx_pool_t *pool;

int array_size = 128;

int array_size_large = 1024;

int page_size = getpagesize();//获得一页的大小

printf("page_size:%d\n", page_size);

printf("----------------------------\n");

printf("create a new pool");

pool = ngx_create_pool(1024, NULL);//创建一个大小为1024的内存池

print_pool(pool);

printf("----------------------------\n");

printf("alloc block 1 from the pool:\n");

int *a1 = ngx_palloc(pool, sizeof(int) * array_size);//分配第一块内存用于创建数组

for (int i=0; i< array_size; i++)

{

a1[i] = i+1;

}

print_pool(pool);

printf("----------------------------\n");

printf("alloc block 2 from the pool:\n");

int *a2 = ngx_palloc(pool, sizeof(int) * array_size);//分配第二块内存用于创建数组，这个时候会创建第二个内存池节点

for (int i=0; i< array_size; i++)

{

a2[i] = 12345678;

}

print_pool(pool);

printf("----------------------------\n");

printf("alloc large memory:\n");

printf("\t\tlarge next before=0x%x\n", pool->current->d.last);

int * a3 = ngx_palloc(pool, sizeof(int) * array_size_large);//由于大小超过了max的值 ngx_palloc中会调用ngx_palloc_large分配大块内存

printf("\t\tlarge next after=0x%x\n", pool->large);

for (int i=0; i< array_size_large; i++)

{

a3[i] = i+1;

}

print_pool(pool);

print_array(a1,array_size);

print_array(a2,array_size);

print_array(a3,array_size_large);

ngx_destroy_pool(pool);

return 0;

}

运行结果:

通过红框可以看到ngx_pool_t中只有第一个内存池节点的头部信息是有意义的，后续调用ngx_palloc_block创建的节点的头部信息都已经被数据覆盖。

五、总结

nginx的代码设计的十分灵活，既方便我们开发，也方便我们复用其中的结构，其中内存池的使用对我们学习nginx，了解nginx如何管理内存有着十分重要的意义。