首页 > 代码库 > STL源码分析--空间配置器 第一级配置器

STL源码分析--空间配置器 第一级配置器

一、SGI STL配置器简介


SGI STL的配置器与众不同,它与标准规范不同。如果要在程序中明确使用SGI配置器,那么应该这样写:

[cpp] view plaincopyprint?
  1. vector<int,std::alloc> iv;  
他的名字是alloc,而且不接受任何参数。标准配置器的名字是allocator,而且可以接受参数。

SGI STL的每一个容器都已经指定了缺省配置器:alloc。我们很少需要自己去指定空间配置器。比如vector容器的声明:

[cpp] view plaincopyprint?
  1. template <class T, class Alloc = alloc>  // 预设使用alloc为配置器  
  2. class vector {  
  3. //...  
  4. }  


二、SGI标准的空间配置器


其实SGI也定义了一个符合部分标准,名为allocator的配置器,但是它自己不使用,也不建议我们使用,主要原因是效率不佳。它只是把C++的操作符::operator new和::operator delete做了一层简单的封装而已。下面仅仅贴出代码:

[cpp] view plaincopyprint?
  1. #ifndef DEFALLOC_H  
  2. #define DEFALLOC_H  
  3.   
  4. #include <new.h>  
  5. #include <stddef.h>  
  6. #include <stdlib.h>  
  7. #include <limits.h>  
  8. #include <iostream.h>  
  9. #include <algobase.h>  
  10.   
  11.   
  12. template <class T>  
  13. inline T* allocate(ptrdiff_t size, T*) {  
  14.     set_new_handler(0);  
  15.     T* tmp = (T*)(::operator new((size_t)(size * sizeof(T))));  
  16.     if (tmp == 0) {  
  17.     cerr << "out of memory" << endl;   
  18.     exit(1);  
  19.     }  
  20.     return tmp;  
  21. }  
  22.   
  23.   
  24. template <class T>  
  25. inline void deallocate(T* buffer) {  
  26.     ::operator delete(buffer);  
  27. }  
  28.   
  29. template <class T>  
  30. class allocator {  
  31. public:  
  32.     typedef T value_type;  
  33.     typedef T* pointer;  
  34.     typedef const T* const_pointer;  
  35.     typedef T& reference;  
  36.     typedef const T& const_reference;  
  37.     typedef size_t size_type;  
  38.     typedef ptrdiff_t difference_type;  
  39.   
  40.     pointer allocate(size_type n) {   
  41.     return ::allocate((difference_type)n, (pointer)0);  
  42.     }  
  43.     void deallocate(pointer p) { ::deallocate(p); }  
  44.     pointer address(reference x) { return (pointer)&x; }  
  45.     const_pointer const_address(const_reference x) {   
  46.     return (const_pointer)&x;   
  47.     }  
  48.     size_type init_page_size() {   
  49.     return max(size_type(1), size_type(4096/sizeof(T)));   
  50.     }  
  51.     size_type max_size() const {   
  52.     return max(size_type(1), size_type(UINT_MAX/sizeof(T)));   
  53.     }  
  54. };  
  55.   
  56. class allocator<void> {  
  57. public:  
  58.     typedef void* pointer;  
  59. };  
  60.   
  61. #endif  

三、SGI特殊的空间配置器alloc


通常,C++中用new操作符来分配内存都包括两个阶段:

(1)调用::operator new配置内存

(2)调用构造函数来构造对象内容


同理,delete操作也包括两个阶段:

(1)调用析构函数将对象析构

(2)调用::operator delete释放内存


为了精密分工,SGI allocator将两个阶段分开

内存配置操作由alloc:allocate负责,内存释放由alloc:deallocate负责;对象构造操作由::contructor()负责,对象析构由::destroy()负责。


配置器定义在头文件<memory>中,它里面又包括两个文件:

[cpp] view plaincopyprint?
  1. #include <stl_alloc.h>        // 负责内存空间的配置和器释放  
  2. #include <stl_construct.h>        // 负责对象的构造和析构  
下图显示了其结构:
图一 头文件 <memory>结构

1、对象的建构和结构函数construct()和destroy()

图二显示了这两个函数的结构和功能。他们被包含在头文件stl_construct.h中。


图二 函数construct()和destroy()示意图

函数construct()使用了定位new操作符,其源代码:

[cpp] view plaincopyprint?
  1. template <class T1, class T2>  
  2. inline void construct(T1* p, const T2& value) {  
  3.   new (p) T1(value);    // 定为new操作符placement new; 在指针p所指处构造对象  
  4. }  


函数destroy则有两个版本。

第一个版本较简单,接受一个指针作为参数,直接调用对象的析构函数即可,其源代码:

[cpp] view plaincopyprint?
  1. template <class T>  
  2. inline void destroy(T* pointer) {  
  3.     pointer->~T();   // 调用析构函数  
  4. }  
第二个版本,其参数接受两个迭代器,将两个迭代器所指范围内的所有对象析构掉。而且,它采用了一种特别的技术:依据元素的型别,判断其是否有trivial destructor(无用的析构函数)进行不同的处理。这也是为了效率考虑。因为如果每个对象的析构函数都是trivial的,那么调用这些毫无作用的析构函数会对效率造成影响。

下面看其源代码:

[cpp] view plaincopyprint?
  1. // 如果元素的数值型別(value type)有 non-trivial destructor…  
  2. template <class ForwardIterator>  
  3. inline void  
  4. __destroy_aux(ForwardIterator first, ForwardIterator last, __false_type) {  
  5.   for ( ; first < last; ++first)  
  6.     destroy(&*first);//调用析构函数  
  7. }  
  8.   
  9. // 如果元素的数值型別(value type)有 trivial destructor…  
  10. template <class ForwardIterator>   
  11. inline void __destroy_aux(ForwardIterator, ForwardIterator, __true_type) {}//不调用析构函数  
  12.   
  13. // 判断元素的数值型別(value type)是否有 trivial destructor,分别调用上面的函数进行不同的处理  
  14. template <class ForwardIterator, class T>  
  15. inline void __destroy(ForwardIterator first, ForwardIterator last, T*) {  
  16.   typedef typename __type_traits<T>::has_trivial_destructor trivial_destructor;  
  17.   __destroy_aux(first, last, trivial_destructor());  
  18. }  
  19.   
  20. // 以下是 destroy() 第二版本,接受两个迭代器。它会设法找出元素的数值型別,  
  21. // 进而利用 __type_traits<> 求取最适当措施。  
  22. template <class ForwardIterator>  
  23. inline void destroy(ForwardIterator first, ForwardIterator last) {  
  24.   __destroy(first, last, value_type(first));  
  25. }  

第二版本还针对迭代器为char*和wchar_t*定义了特化版本:

[cpp] view plaincopyprint?
  1. inline void destroy(char*, char*) {}  
  2. inline void destroy(wchar_t*, wchar_t*) {}  


2、空间的配置和释放,std::alloc

对象构造前的空间分配和析构后的空间释放,定义在头文件<stl_alloc.h>中。其设计思想是:

  • 向system heap要求空间。
  • 考虑多线程状态。
  • 考虑内存不足时的应变措施。
  • 考虑过多“小额区块”可能造成的内存碎片问题。

考虑到小型区块可能造成的内存破碎问题,SGI设计了双层级配置器,第一级配置器直接使用malloc()和free(),第二级则视情况采用不同的策略。而且采用了复杂的内存池memory pool整理方式。整个设计究竟是只开放第一级配置器还是同事开放第二级配置器取决于宏__USE_MALLOC是否被定义。

[cpp] view plaincopyprint?
  1. # ifdef __USE_MALLOC   
  2. ...   
  3. typedef __malloc_alloc_template<0> malloc_alloc; <span style="font-family:KaiTi_GB2312;">//令 alloc为第一级配置器</span>  
  4. typedef malloc_alloc alloc;   
  5. # else   
  6. ...   
  7. //令 alloc 为第二级配置器   
  8. typedef __default_alloc_template<__NODE_ALLOCATOR_THREADS, 0> alloc;   
  9. #endif /* ! __USE_MALLOC */   
SGI并未定义它。

无论alloc被定义为第一级或者是第二级配置器,SGI还为它包装一个接口如下,使配置器的接口能够符合STL规格:

[cpp] view plaincopyprint?
  1. template<class T, class Alloc>  
  2. class simple_alloc {  
  3.   
  4. public:  
  5.     static T *allocate(size_t n)  
  6.                 { return 0 == n? 0 : (T*) Alloc::allocate(n * sizeof (T)); }  
  7.     static T *allocate(void)  
  8.                 { return (T*) Alloc::allocate(sizeof (T)); }  
  9.     static void deallocate(T *p, size_t n)  
  10.                 { if (0 != n) Alloc::deallocate(p, n * sizeof (T)); }  
  11.     static void deallocate(T *p)  
  12.                 { Alloc::deallocate(p, sizeof (T)); }  
  13. };  
SGI STL容器全部是使用这个simple_alloc接口。

第一级和第二级配置器之间的关系如图三所示。

图三 第一级配置器和第二级配置器

第一级和第二级配置器的包装接口和运用方式如下:



第一级配置器__malloc_alloc_template剖析

第一级配置器直接使用malloc(),free(),realloc()等C函数执行实际的内存配置、释放、重配置操作,并实现出类似C++ new handler机制。它有独特的out-of-memory内存处理机制:在抛出std::bad_alloc异常之前,调用内存不足处理例程尝试释放空间,如果用户没有定义相应的内存不足处理例程,那么还是会抛出异常。详细实现见函数oom_malloc(),oom_realloc()。

内存不足处理例程保存在函数指针__malloc_alloc_oom_handler里面。

下面列出代码:

[cpp] view plaincopyprint?
  1. #if 0   
  2. #   include <new>   
  3. #   define  __THROW_BAD_ALLOC throw bad_alloc   
  4. #elif !defined(__THROW_BAD_ALLOC)   
  5. #   include <iostream.h>   
  6. #   define  __THROW_BAD_ALLOC cerr << "out of memory" << endl; exit(1)   
  7. #endif   
  8.    
  9. // malloc-based allocator. 通常比稍后介绍的 default alloc 速度慢,   
  10. //一般而言是 thread-safe,并且对于空间的运用比较高效(efficient)。   
  11. //以下是第一级配置器。   
  12. //注意,无「template 型别参数」。至于「非型别参数」inst,完全没派上用场。  
  13. template <int inst>     
  14. class __malloc_alloc_template {   
  15.    
  16. private:   
  17. //以下都是函式指标,所代表的函式将用来处理内存不足的情况。   
  18. // oom : out of memory.   
  19. static void *oom_malloc(size_t);   
  20. static void *oom_realloc(void *, size_t);   
  21. static void (* __malloc_alloc_oom_handler)();   
  22.    
  23. public:   
  24.    
  25. static void * allocate(size_t n)   
  26. {   
  27.     void  *result =malloc(n);//第一级配置器直接使用 malloc()   
  28.     // 以下,无法满足需求时,改用 oom_malloc()   
  29.     if (0 == result) result = oom_malloc(n);   
  30.     return  result;   
  31. }   
  32. static void deallocate(void *p, size_t /* n */)   
  33. {   
  34. free(p); //第一级配置器直接使用 free()   
  35. }   
  36.    
  37. static void * reallocate(void *p, size_t /* old_sz */size_t new_sz)   
  38. {   
  39.     void  *  result  =realloc(p, new_sz);//第一级配置器直接使用 rea  
  40.     // 以下,无法满足需求时,改用 oom_realloc()   
  41.     if (0 == result) result = oom_realloc(p, new_sz);   
  42.     return  result;   
  43. }   
  44.    
  45. //以下模拟 C++的 set_new_handler(). 换句话说,你可以透过它,   
  46. //指定你自己的 out-of-memory handler   
  47. static void (* set_malloc_handler(void (*f)()))()   
  48. {   
  49.     void  (*  old)()  =  __malloc_alloc_oom_handler;   
  50. __malloc_alloc_oom_handler = f;   
  51.     return(old);   
  52. }   
  53. };   
  54.    
  55. // malloc_alloc out-of-memory handling   
  56. //初值为 0。有待客端设定。   
  57. template <int inst>   
  58. void (* __malloc_alloc_template<inst>::__malloc_alloc_oom_handler)() = 0;   
  59.    
  60. template <int inst>   
  61. void * __malloc_alloc_template<inst>::oom_malloc(size_t n)   
  62. {   
  63.     void  (* my_malloc_handler)();   
  64.     void  *result;   
  65.    
  66.     for (;;)  {   
  67.    
  68. //不断尝试释放、配置、再释放、再配置…   
  69. my_malloc_handler = __malloc_alloc_oom_handler;   
  70.         if  (0  ==  my_malloc_handler)  {  __THROW_BAD_ALLOC; }   
  71.         (*my_malloc_handler)();//呼叫处理例程,企图释放内存。   
  72.         result = malloc(n);  //再次尝试配置内存。   
  73.         if  (result)  return(result);   
  74.     }   
  75. }   
  76.    
  77. template <int inst>   
  78. void * __malloc_alloc_template<inst>::oom_realloc(void *p, size_t n)   
  79. {   
  80.     void  (* my_malloc_handler)();   
  81.     void  *result;   
  82.        for (;;)  {  //不断尝试释放、配置、再释放、再配置…   
  83. my_malloc_handler = __malloc_alloc_oom_handler;   
  84.         if  (0  ==  my_malloc_handler)  {  __THROW_BAD_ALLOC; }   
  85.         (*my_malloc_handler)();//呼叫处理例程,企图释放内存。   
  86.         result = realloc(p, n);//再次尝试配置内存。   
  87.         if  (result)  return(result);   
  88.     }   
  89. }   
  90.    
  91. //注意,以下直接将参数 inst指定为 0。   
  92. typedef __malloc_alloc_template<0> malloc_alloc;   
  93.    

需要注意的是:设计“内存不足处理例程”是客端的责任。关于设计“内存不足处理例程”的具体做法可参考《Effective C++(第二版)》条款7。

STL源码分析--空间配置器 第一级配置器