STL-内存的配置与释放

STL有两级空间配置器，默认是使用第二级。第二级空间配置器会在某些情况下去调用第一级空间配置器。空间配置器都是在allocate函数内分配内存，在deallocate函数内释放内存。

第一级空间配置器

第一级配置器只是对malloc函数和free函数的简单封装，在allocate内调用malloc，在deallocate内调用free。同时第一级配置器的oom_malloc函数，用来处理malloc失败的情况。如下所示：

allocate对malloc函数简单封装 :

static void *allocate(size_t n)
{void *result = malloc(n);if (NULL == result)result = oom_malloc(n);return result;
}

deallocate对free函数简单封装 :

static void deallocate(void *p, size_t) { free(p); }

oom_malloc调用外部提供的malloc失败处理函数，然后重新试着再次调用malloc。重复执行此过程，直到malloc成功为止 :

template <int inst>
void* __malloc_alloc<inst>::oom_malloc(size_t n)
{void (*my_malloc_handler)();void *result;for (;;){my_malloc_handler = malloc_alloc_oom_handler;if (NULL == my_malloc_handler)__THROW_BAD_ALLOC;(*my_malloc_handler)();result = malloc(n);if (result)return result;}
}

第二级空间配置器

第一级配置器直接调用malloc和free带来了几个问题：

1. 内存分配/释放的效率低。

2. 当配置大量的小内存块时，会导致内存碎片比较严重。

3. 配置内存时，需要额外的部分空间存储内存块信息，所以配置大量的小内存块时，还会导致额外内存负担。

第二级配置器维护了一个自由链表数组，每次需要分配内存时，直接从相应的链表上取出一个内存节点就完成工作，效率很高。

自由链表数组

自由链表数组其实就是个指针数组，数组中的每个指针元素指向一个链表的起始节点。数组大小为16，即维护了16个链表，链表的每个节点就是实际的内存块，相同链表上的内存块大小都相同，不同链表的内存块大小不同，从8一直到128。如下所示，obj为链表上的节点，free_list就是链表数组。

//自由链表
union obj
{union obj *free_list_link;char data[1];
};
//自由链表数组
static obj *volatile free_list[__NFREELISTS];

内存分配

allocate函数内先判断要分配的内存大小，若大于128字节，直接调用第一级配置器，否则根据要分配的内存大小从16个链表中选出一个链表，取出该链表的第一个节点。若相应的链表为空，则调用refill函数填充该链表。如下：

template <bool threads>
void *__default_alloc<threads>::allocate(size_t n)
{obj *volatile *my_free_list;obj *result;if (n > (size_t)__MAX_BYTES) //调用第一级配置器return malloc_alloc::allocate(n);my_free_list = free_list + FREELIST_INDEX(n);result = *my_free_list;if (result == NULL){//第n号链表无内存块，则准备重新填充该链表void *r = refill(ROUND_UP(n));return r;}*my_free_list = result->free_list_link;return result;
}

填充链表

若allocate函数内要取出节点的链表为空，则会调用refill函数填充该链表。

refill函数内会先调用chunk_alloc函数从内存池分配一大块内存，该内存大小默认为20个链表节点大小，当内存池的内存也不足时，返回的内存块节点数目会不足20个。接着refill的工作就是将这一大块内存分成20份相同大小的内存块，并将各内存块连接起来形成一个链表。如下：

template <bool threads>
void *__default_alloc<threads>::refill(size_t n)
{int nobjs = __NOBJS;char *chunk = chunk_alloc(n, nobjs);  //从内存池获取内存if (nobjs == 1)  //只能分配一块，则直接返回给调用者return chunk;obj *volatile *my_free_list;obj *result, *next_obj, *current_obj;result = (obj *)chunk;my_free_list = free_list + FREELIST_INDEX(n);*my_free_list = next_obj = (obj *)(chunk + n);for (int i = 1; i < nobjs - 1; i++)  //将剩下的区块添加进链表{current_obj = next_obj;next_obj = (obj *)(char *)(next_obj + n);current_obj->free_list_link = next_obj;}//最后一块current_obj = next_obj;current_obj->free_list_link = NULL;return result;
}

内存池

chunk_alloc函数内管理了一块内存池，当refill函数要填充链表时，就会调用chunk_alloc函数，从内存池取出相应的内存。

在chunk_alloc函数内首先判断内存池大小是否足够填充一个有20个节点的链表，若内存池足够大，则直接返回20个内存节点大小的内存块给refill。如下：

        if (size_left >= total_size)  //内存池剩余空间满足需求{result = start_free;start_free += total_size;return result;}

若内存池大小无法满足20个内存节点的大小，但至少满足1个内存节点，则直接返回相应的内存节点大小的内存块给refill；

        else if (size_left >= size)  //剩余空间不能全部满足，但至少满足一块{nobjs = size_left / size;result = start_free;start_free += nobjs * size;return result;}

若内存池连1个内存节点大小的内存块都无法提供，则chunk_alloc函数会将内存池中那一点点的内存大小分配给其他合适的链表，然后去调用malloc函数分配的内存大小为所需的两倍。若malloc成功，则返回相应的内存大小给refill；若malloc失败，会先搜寻其他链表的可用的内存块，添加到内存池，然后递归调用chunk_alloc函数来分配内存，若其他链表也无内存块可用，则只能调用第一级空间配置器，因为第一级空间配置器有malloc失败的出错处理函数，最终的希望只能寄托在那里了。

如下是整个chunk_alloc函数：

template <bool threads>
char *__default_alloc<threads>::chunk_alloc(size_t size, int& nobjs)
{size_t total_size = size * nobjs;char *result;size_t size_left = end_free - start_free;if (size_left >= total_size)  //内存池剩余空间满足需求{result = start_free;start_free += total_size;return result;}else if (size_left >= size)  //剩余空间不能全部满足，但至少满足一块{nobjs = size_left / size;result = start_free;start_free += nobjs * size;return result;}else  //连一个区块都无法满足{if (size_left > 0)  //将残余内存分配给其他合适的链表{obj *volatile *my_free_list = free_list + FREELIST_INDEX(size_left);((obj *)start_free)->free_list_link = *my_free_list;  //在头部插入*my_free_list = (obj *)start_free;}size_t bytes_to_get = 2 * total_size + ROUND_UP(heap_size >> 4);start_free = (char *)malloc(bytes_to_get);if (start_free == NULL)  //堆空间不足{int i;obj *volatile *my_free_list;obj *p;for (i = size; i < __MAX_BYTES; i++){my_free_list = free_list + FREELIST_INDEX(i);p = *my_free_list;if (p != NULL){*my_free_list = p->free_list_link;start_free = (char *)p;end_free = start_free + i;return chunk_alloc(size, nobjs);}}end_free = NULL;//调用第一级配置器start_free = (char *)malloc_alloc::allocate(bytes_to_get);}heap_size += bytes_to_get;end_free = start_free + heap_size;return chunk_alloc(size, nobjs);}
}

内存释放

第二级配置器的deallocate函数并不会直接释放内存块。当内存块大小大于128字节时才会直接释放，否则会将内存块回收到相应的链表当中。如下：

void __default_alloc<threads>::deallocate(void *p, size_t n)
{//大于__MAX_BYTES，则释放该内存if (n > (size_t)__MAX_BYTES)malloc_alloc::deallocate(p, n);obj *q = (obj *)p;obj *volatile *my_free_list;my_free_list = free_list + FREELIST_INDEX(n);//小于__MAX_BYTES，则回收区块,并未释放q->free_list_link = *my_free_list;*my_free_list = q;
}

内存对外接口

STL对外提供了一个simple_alloc类，该类提供统一的接口：allocate函数、deallocate函数，使得外部无需关心使用的是几级内存配置器。另外simple_alloc类将外部所需的对象个数转换为字节。如下。

template <typename T, typename Alloc>
class simple_alloc
{
public:static T *allocate(size_t n) // 个数{return n == 0 ? 0 : (T*)Alloc::allocate(n * sizeof(T)); // 将个数转换为字节}static T *allocate(void){return (T*)Alloc::allocate(sizeof(T));}static void deallocate(T *p, size_t n) // 个数{if (n != 0)Alloc::deallocate(p, n * sizeof(T));}static void deallocate(T *p){Alloc::deallocate(p, sizeof(T));}
};