1. 程式人生 > >kzalloc和kmalloc函式詳解

kzalloc和kmalloc函式詳解

用kzalloc申請記憶體的時候, 效果等同於先是用 kmalloc() 申請空間 , 然後用 memset() 來初始化 ,所有申請的元素都被初始化為 0.

  1. /** 
  2.  * kzalloc - allocate memory. The memory is set to zero. 
  3.  * @size: how many bytes of memory are required. 
  4.  * @flags: the type of memory to allocate (see kmalloc). 
  5.  */
  6. static inline void *kzalloc(size_t size, gfp_t flags)  
  7. {  
  8.     return kmalloc(size, flags | __GFP_ZERO);  
  9. }  
kzalloc 函式是帶引數呼叫kmalloc函式,新增的引數是或了標誌位__GFP_ZERO,
  1. void *__kmalloc(size_t size, gfp_t flags)  
  2. {  
  3.     struct kmem_cache *s;  
  4.     void *ret;  
  5.     if (unlikely(size > SLUB_MAX_SIZE))  
  6.         return kmalloc_large(size, flags);  
  7.     s = get_slab(size, flags);  
  8.     if (unlikely(ZERO_OR_NULL_PTR(s)))  
  9.         return s;  
  10.     ret = slab_alloc(s, flags, -1, _RET_IP_);  
  11.     trace_kmalloc(_RET_IP_, ret, size, s->size, flags);  
  12.     return ret;  
  13. }  
這個函式呼叫trace_kmalloc,flags引數不變,繼續往裡面可以看到
  1. static __always_inline void *slab_alloc(struct kmem_cache *s,  
  2.         gfp_t gfpflags, int node, unsigned long addr)  
  3. {  
  4.     void **object;  
  5.     struct kmem_cache_cpu *c;  
  6.     unsigned long flags;  
  7.     unsigned int objsize;  
  8.     gfpflags &= gfp_allowed_mask;  
  9.     lockdep_trace_alloc(gfpflags);  
  10.     might_sleep_if(gfpflags & __GFP_WAIT);  
  11.     if (should_failslab(s->objsize, gfpflags))  
  12.         return NULL;  
  13.     local_irq_save(flags);  
  14.     c = get_cpu_slab(s, smp_processor_id());  
  15.     objsize = c->objsize;  
  16.     if (unlikely(!c->freelist || !node_match(c, node)))  
  17.         object = __slab_alloc(s, gfpflags, node, addr, c);  
  18.     else {  
  19.         object = c->freelist;  
  20.         c->freelist = object[c->offset];  
  21.         stat(c, ALLOC_FASTPATH);  
  22.     }  
  23.     local_irq_restore(flags);  
  24.     if (unlikely((gfpflags & __GFP_ZERO) && object))  
  25.         memset(object, 0, objsize);  
  26.     kmemcheck_slab_alloc(s, gfpflags, object, c->objsize);  
  27.     kmemleak_alloc_recursive(object, objsize, 1, s->flags, gfpflags);  
  28.     returnobject;  
  29. }  
這裡主要判斷兩個標誌,WAIT和ZERO,和本文有關的關鍵程式碼就是

if (unlikely((gfpflags & __GFP_ZERO) && object))
memset(object, 0, objsize);


到此,這個函式區別於kmalloc的地方就清楚了

kmalloc 函式詳解

#include <linux/slab.h> void *kmalloc(size_t size, int flags);

給 kmalloc 的第一個引數是要分配的塊的大小. 第 2 個引數, 分配標誌, 非常有趣, 因為它以幾個方式控制 kmalloc 的行為.
最一般使用的標誌, GFP_KERNEL, 意思是這個分配((內部最終通過呼叫 __get_free_pages 來進行, 它是 GFP_ 字首的來源) 代表執行在核心空間的程序而進行的. 換句話說, 這意味著呼叫函式是代表一個程序在執行一個系統呼叫. 使用 GFP_KENRL 意味著 kmalloc 能夠使當前程序在少記憶體的情況下睡眠來等待一頁. 一個使用 GFP_KERNEL 來分配記憶體的函式必須, 因此, 是可重入的並且不能在原子上下文中執行. 噹噹前程序睡眠, 核心採取正確的動作來定位一些空閒記憶體, 或者通過重新整理快取到磁碟或者交換出去一個使用者程序的記憶體.
GFP_KERNEL 不一直是使用的正確分配標誌; 有時 kmalloc 從一個程序的上下文的外部呼叫. 例如, 這類的呼叫可能發生在中斷處理, tasklet, 和核心定時器中. 在這個情況下, 當前程序不應當被置為睡眠, 並且驅動應當使用一個 GFP_ATOMIC 標誌來代替. 核心正常地試圖保持一些空閒頁以便來滿足原子的分配. 當使用 GFP_ATOMIC 時, kmalloc 能夠使用甚至最後一個空閒頁. 如果這最後一個空閒頁不存在, 但是, 分配失敗.
其他用來代替或者增添 GFP_KERNEL 和 GFP_ATOMIC 的標誌, 儘管它們 2 個涵蓋大部分裝置驅動的需要. 所有的標誌定義在 <linux/gfp.h>, 並且每個標誌用一個雙下劃線做字首, 例如 __GFP_DMA. 另外, 有符號代表常常使用的標誌組合; 這些缺乏字首並且有時被稱為分配優先順序. 後者包括:
GFP_ATOMIC
用來從中斷處理和程序上下文之外的其他程式碼中分配記憶體. 從不睡眠.
GFP_KERNEL
核心記憶體的正常分配. 可能睡眠.
GFP_USER
用來為使用者空間頁來分配記憶體; 它可能睡眠.
GFP_HIGHUSER
如同 GFP_USER, 但是從高階記憶體分配, 如果有. 高階記憶體在下一個子節描述.
GFP_NOIO
GFP_NOFS
這個標誌功能如同 GFP_KERNEL, 但是它們增加限制到核心能做的來滿足請求. 一個 GFP_NOFS 分配不允許進行任何檔案系統呼叫, 而 GFP_NOIO 根本不允許任何 I/O 初始化. 它們主要地用在檔案系統和虛擬記憶體程式碼, 那裡允許一個分配睡眠, 但是遞迴的檔案系統呼叫會是一個壞注意.
上面列出的這些分配標誌可以是下列標誌的相或來作為引數, 這些標誌改變這些分配如何進行:
__GFP_DMA
這個標誌要求分配在能夠 DMA 的記憶體區. 確切的含義是平臺依賴的並且在下面章節來解釋.
__GFP_HIGHMEM
這個標誌指示分配的記憶體可以位於高階記憶體.
__GFP_COLD
正常地, 記憶體分配器盡力返回"緩衝熱"的頁 -- 可能在處理器緩衝中找到的頁. 相反, 這個標誌請求一個"冷"頁, 它在一段時間沒被使用. 它對分配頁作 DMA 讀是有用的, 此時在處理器緩衝中出現是無用的.
__GFP_NOWARN
這個很少用到的標誌阻止核心來發出警告(使用 printk ), 當一個分配無法滿足.
__GFP_HIGH
這個標誌標識了一個高優先順序請求, 它被允許來消耗甚至被核心保留給緊急狀況的最後的記憶體頁.
__GFP_REPEAT
__GFP_NOFAIL
__GFP_NORETRY
這些標誌修改分配器如何動作, 當它有困難滿足一個分配. __GFP_REPEAT 意思是" 更盡力些嘗試" 通過重複嘗試 -- 但是分配可能仍然失敗. __GFP_NOFAIL 標誌告訴分配器不要失敗; 它盡最大努力來滿足要求. 使用 __GFP_NOFAIL 是強烈不推薦的; 可能從不會有有效的理由在一個裝置驅動中使用它. 最後, __GFP_NORETRY 告知分配器立即放棄如果得不到請求的記憶體.
kmalloc 能夠分配的記憶體塊的大小有一個上限. 這個限制隨著體系和核心配置選項而變化. 如果你的程式碼是要完全可移植, 它不能指望可以分配任何大於 128 KB. 如果你需要多於幾個 KB
這方面的原因:
kmalloc並不直接從分頁機制中獲得空閒頁面而是從slab頁面分配器那兒獲得需要的頁面,slab的實現程式碼限制了最大分配的大小為128k,即131072bytes,理論上你可以通過更改slab.c中的 cache_sizes陣列中的最大值使得kmalloc可以獲得更大的頁面數,不知道有沒有甚麼副效應或者沒有必要這樣做,因為獲取較大記憶體的方法有很多,想必128k是經驗總結後的合適值。
alloc_page( )可以分配的最大連續頁面是4K
 static inline struct page * alloc_pages(unsigned int gfp_mask, unsigned int order) 

/* 
 * Gets optimized away by the compiler. 
 */ 
 if (order >= MAX_ORDER) 
 return NULL; 
 return _alloc_pages(gfp_mask, order); 
 } 

alloc_pages最大分配頁面數為512個,則可用記憶體數最大為2^9*4K=2M
  

轉載地址:http://blog.csdn.net/xujianqun/article/details/6715243#