_aligned_malloc自己實現

  1. void* aligned_malloc(size_t size, size_t align)  
  2. {  
  3.     void* raw_malloc_ptr;       //初始分配的地址  
  4.     void* aligned_ptr;          //最終我們獲得的alignment地址  
  5.       
  6.     if( align & (align - 1) )   //如果alignment不是2的n次方,返回  
  7.     {  
  8.         errno = EINVAL;  
  9.         return ( (void*)0 );  
  10.     }  
  11.       
  12.     if( 0 == size )  
  13.     {  
  14.         return ( (void*)0 );  
  15.     }   
  16.       
  17.     //將alignment置爲至少爲2*sizeof(void*),一種優化做法。  
  18.     if( align < 2*sizeof(void*) )  
  19.     {  
  20.         align = 2 * sizeof(void*);  
  21.     }  
  22.       
  23.     raw_malloc_ptr = malloc(size + align);  
  24.     if( !raw_malloc_ptr )  
  25.     {  
  26.         return ( (void*)0 );  
  27.     }  
  28.       
  29.     //Align  We have at least sizeof (void *) space below malloc'd ptr.   
  30.     aligned_ptr = (void*) ( ((size_t)raw_malloc_ptr + align) & ~((size_t)align - 1));  
  31.       
  32.     ( (void**)aligned_ptr )[-1] = raw_malloc_ptr;  
  33.       
  34.     return aligned_ptr;  
  35. }  

  1. void* aligned_free(void * aligned_ptr)  
  2. {  
  3.     if( aligned_ptr )  
  4.     {  
  5.         free( ((void**)aligned_ptr)[-1] );  
  6.     }  
  7. }  


再給出一套對比一下,可能講的更詳細:


  1. void* aligned_malloc(size_t size, int alignment)    
  2.     {    
  3.         // 分配足夠的內存, 這裏的算法很經典, 早期的STL中使用的就是這個算法    
  4.     
  5.         // 首先是維護FreeBlock指針佔用的內存大小    
  6.         const int pointerSize = sizeof (void*);    
  7.     
  8.         // alignment - 1 + pointerSize這個是FreeBlock內存對齊需要的內存大小    
  9.         // 前面的例子sizeof(T) = 20, __alignof(T) = 16,    
  10.         // g_MaxNumberOfObjectsInPool = 1000    
  11.         // 那麼調用本函數就是alignedMalloc(1000 * 20, 16)    
  12.         // 那麼alignment - 1 + pointSize = 19    
  13.         const int requestedSize = size + alignment - 1 + pointerSize;    
  14.     
  15.         // 分配的實際大小就是20000 + 19 = 20019    
  16.         void* raw = malloc(requestedSize);    
  17.     
  18.         // 這裏實Pool真正爲對象實例分配的內存地址    
  19.         uintptr_t start = (uintptr_t) raw + pointerSize;    
  20.         // 向上舍入操作    
  21.         // 解釋一下, __ALIGN - 1指明的是實際內存對齊的粒度    
  22.         // 例如__ALIGN = 8時, 我們只需要7就可以實際表示8個數(0~7)    
  23.         // 那麼~(__ALIGN - 1)就是進行舍入的粒度    
  24.         // 我們將(bytes) + __ALIGN-1)就是先進行進位, 然後截斷    
  25.         // 這就保證了我是向上舍入的    
  26.         // 例如byte = 100, __ALIGN = 8的情況    
  27.         // ~(__ALIGN - 1) = (1 000)B    
  28.         // ((bytes) + __ALIGN-1) = (1 101 011)B    
  29.         // (((bytes) + __ALIGN-1) & ~(__ALIGN - 1)) = (1 101 000 )B = (104)D    
  30.         // 104 / 8 = 13, 這就實現了向上舍入    
  31.         // 對於byte剛好滿足內存對齊的情況下, 結果保持byte大小不變    
  32.         // 記得《Hacker's Delight》上面有相關的計算    
  33.         // 這個表達式與下面給出的等價    
  34.         // ((((bytes) + _ALIGN - 1) * _ALIGN) / _ALIGN)    
  35.         // 但是SGI STL使用的方法效率非常高     
  36.         void* aligned = (void*) ((start + alignment - 1) & ~(alignment - 1));    
  37.     
  38.         // 這裏維護一個指向malloc()真正分配的內存    
  39.         *(void**) ((uintptr_t) aligned - pointerSize) = raw;    
  40.     
  41.         // 返回實例對象真正的地址    
  42.         return aligned;    
  43.     }    
  44.   
  45.   
  46.     // 這裏是內部維護的內存情況    
  47.     //                   這裏滿足內存對齊要求    
  48.     //                             |    
  49.     // ----------------------------------------------------------------------    
  50.     // | 內存對齊填充 | 維護的指針 | 對象1 | 對象2 | 對象3 | ...... | 對象n |    
  51.     // ----------------------------------------------------------------------    
  52.     // ^                     | 指向malloc()分配的地址起點    
  53.     // |                     |    
  54.     // -----------------------    
  55.     void aligned_free(void* aligned)    
  56.     {    
  57.         // 釋放操作很簡單了, 參見上圖    
  58.         void* raw = *(void**) ((uintptr_t) aligned - sizeof (void*));    
  59.         free(raw);    
  60.     }    
  61.     
  62.     bool isAligned(void* data, int alignment)    
  63.     {    
  64.         // 又是一個經典算法, 參見<Hacker's Delight>    
  65.         return ((uintptr_t) data & (alignment - 1)) == 0;    
  66.     }    

注:

<Hacker's Delight>這是第一本試圖揭開計算機運算神祕面紗的書,極其適合於編譯器編寫者、庫開發者、反工程愛好者研讀,將給大家展示一些鮮爲人知但非常有意義的二進制編碼技巧以及底層Algorithms技巧,其內容超級強悍。

發佈了41 篇原創文章 · 獲贊 5 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章