初次使用C++標准庫實現共享內存的管理時,Vector每次分配內存個數不固定,回收也不固定,這樣的話,程序還需要繼續完善,下面就隨本文的講述來讓大家進一步的了解C++中的C++標准庫。
內存池管理程序源碼如下:
- #ifndef MY_ALLOCATOR_H_
- #define MY_ALLOCATOR_H_
- #include "stdafx.h"
- #include <limits>
- #include <iostream>
- namespace happyever
- {
- enum { NODENUMS = 2 };
- union _Obj
- {
- union _Obj* M_free_list_link;
- char M_client_data[1];
- } ;
- typedef union _Obj Obj;
- struct _Cookie
- {
- int iShmKey; /* 共享內存鍵值 */
- int iShmID; /* iShmKey對應的shmid */
- int iSemKey; /* 鎖信號鍵值 */
- int iSemID; /* 鎖信號標識 */
- int iTotalsize; /* 容器總容量 */
- void* pStartall; /* 共享內存自身地址 */
- char* pStartfree; /* 自由空間的開始地址*/
- char* pEndfree; /* 自由空間的結束地址*/
- int iUseNum[NODENUMS];
- /*用來存放free_list中節點的size*/
- short sFreelistIndex[NODENUMS];
- /*存放分配內存節點的鏈表*/
- Obj* uFreelist[NODENUMS];
- };
- typedef struct _Cookie Cookie;
- //Obj;
- //Cookie;
- static Cookie *pHead = NULL;
- template <class T>
- class MyAlloc
- {
- private:
- static const int ALIGN = sizeof(Obj);
- int round_up(int bytes);
- int freelist_index(int bytes);
- int freelist_getindex(int bytes);
- char* chunk_alloc(int size, int *nobjs);
- void* refill(int num,int n);
- public:
- // type definitions
- typedef T value_type;
- typedef T* pointer;
- typedef const T* const_pointer;
- typedef T& reference;
- typedef const T& const_reference;
- typedef std::size_t size_type;
- typedef std::ptrdiff_t difference_type;
- template <class U>
- struct rebind
- {
- typedef MyAlloc<U> other;
- };
以上程序只要稍微修改,就可以實現共享內存的管理,可以方便的使用C++標准庫提供的容器。加上信號量的鎖機制。以上為了學習而改寫的SGI的stl二級分配算法實現的。以上代碼存在一定的局限性。
我另外完整實現了共享內存管理的STL標准的alloctor程序,使用posix信號量加鎖。目前應用在aix的xlC編譯環境下。因為源碼涉及公司的商業秘密,所以不能公開。但基本上以上源碼已經體現了自己管理內存的完整思路,供這方面需求的朋友一起學習研究用。