一种巧妙的内存池算法(HeapBlock)

在一个简单的GUI库中看到的算法，非常巧妙，适用于需要频繁分配和释放相同大小数据块的情况，如GUI库中的窗口结构，SOCKET结构等，算法额外开支极小。

[cpp] view plain copy print ?

1. typedef DWORD HBLOCKHEAP; 
2. typedef unsigned char BYTE, *PBYTE; 
3.  
4. //使用之前需要先分配一整块内存，并调用此函数进行初始化，初始化后将形成一个链，只是这个链没有前后指针， 
5. //指针数据保存在数据块实体数据前4字节中 
6. HBLOCKHEAP BlockHeap_Initialize(void* buffer, int buffersize, int blocksize) 
7. { 
8.     PBYTE pBlock = (PBYTE)buffer; 
9.  
10.     for(buffersize -= blocksize; pBlock - (PBYTE)buffer < buffersize ; pBlock += blocksize) 
11.     { 
12.              //此处将后一块的位置保存在前一块的前4字节中，(PBYTE*)pBlock的转换只是为了将pBlock暂时解释成BYTE*,才能赋值4字节内存地址 
13.         *(PBYTE*)pBlock = pBlock + blocksize; 
14.          } 
15.          *(PBYTE*)pBlock = NULL;//最后一块的下一块为空块 
16.     return (HBLOCKHEAP)buffer;//返回此链的首块，此值需要调用者保存 
17. } 
18.  
19.  
20. //从内存池中分配数据时，直接将链首返回给调用者，同时将链首指向的下一块指定为新的链首 
21. void* BlockHeap_Alloc(HBLOCKHEAP* pHeap) 
22. { 
23.     void* pBlock = (void*)(*pHeap); 
24.  
25.     if(pBlock) 
26.         *pHeap =(HBLOCKHEAP)( *(PBYTE*)(*pHeap) ); 
27.  
28.     return pBlock; 
29. } 
30.  
31.  
32. //释放时将要释放的块指定为新的链首，并指向原链首即可 
33. void BlockHeap_Free(HBLOCKHEAP* pHeap, void* pBlock) 
34. { 
35.     *(PBYTE*)pBlock = (PBYTE)(*pHeap); 
36.     *pHeap = (HBLOCKHEAP)pBlock; 
37. } 
38.  
39.  
40.  
41. //基本的调用示例 
42. typedef struct{ 
43.     int a; 
44.     int b; 
45.     int c; 
46.     ... 
47. }SOCKET, *PSOCKET; 
48.  
49.  
50. static SOCKET s_sockets[100];    //先分配的内存 
51. static HBLOCKHEAP s_hSockets;    //保存链首的变量 
52.  
53. //初始化并得到链首 
54. s_hSockets = BlockHeap_Initialize(s_sockets, sizeof(s_sockets), sizeof(s_sockets[0])); 
55. //分配内存 
56. SOCKET *sock = BlockHeap_Alloc(&s_hSockets); 
57.  
58. //其它操作... 
59.  
60.  
61. //释放内存 
62. BlockHeap_Free(&s_hSockets, sock);