你應(yīng)該曾經(jīng)糾結(jié)過是用kmalloc()，還是vmalloc()？現(xiàn)在你不用那么糾結(jié)了，因為內(nèi)核里面現(xiàn)在有個API叫kvmalloc()，可以認(rèn)為是kmalloc()和vmalloc()的雙劍合一。屠龍刀和倚天劍的合體。

內(nèi)核里面有大量的代碼現(xiàn)在都使用了kvmalloc()，譬如：

source/ipc/msg.c

static int newque(struct ipc_namespace *ns, struct ipc_params *params) {struct msg_queue *msq;int retval;key_t key = params->key;int msgflg = params->flg;msq = kvmalloc(sizeof(*msq), GFP_KERNEL);if (unlikely(!msq))return -ENOMEM;... }

這個代碼在早期的內(nèi)核里面是（比如v4.0-rc7/source/ipc/msg.c）：

static int newque(struct ipc_namespace *ns, struct ipc_params *params) {struct msg_queue *msq;int id, retval;key_t key = params->key;int msgflg = params->flg;msq = ipc_rcu_alloc(sizeof(*msq));if (!msq)return -ENOMEM;...}

看起來是用的這個函數(shù)申請內(nèi)存：

ipc_rcu_alloc(sizeof(*msq))

那么這個ipc_rc_alloc()是怎么回事呢？

void *ipc_alloc(int size) {void *out;if (size > PAGE_SIZE)out = vmalloc(size);elseout = kmalloc(size, GFP_KERNEL);return out; }

邏輯上是，大于一頁的時候用vmalloc()，小于等于1頁用kmalloc()。

而kvmalloc()的實現(xiàn)代碼里面則對類似邏輯進(jìn)行了非常智能地處理：

void *kvmalloc_node(size_t size, gfp_t flags, int node) {gfp_t kmalloc_flags = flags;void *ret;/** vmalloc uses GFP_KERNEL for some internal allocations (e.g page tables)* so the given set of flags has to be compatible.*/if ((flags & GFP_KERNEL) != GFP_KERNEL)return kmalloc_node(size, flags, node);/** We want to attempt a large physically contiguous block first because* it is less likely to fragment multiple larger blocks and therefore* contribute to a long term fragmentation less than vmalloc fallback.* However make sure that larger requests are not too disruptive - no* OOM killer and no allocation failure warnings as we have a fallback.*/if (size > PAGE_SIZE) {kmalloc_flags |= __GFP_NOWARN;if (!(kmalloc_flags & __GFP_RETRY_MAYFAIL))kmalloc_flags |= __GFP_NORETRY;}ret = kmalloc_node(size, kmalloc_flags, node);/** It doesn't really make sense to fallback to vmalloc for sub page* requests*/if (ret || size <= PAGE_SIZE)return ret;return __vmalloc_node_flags_caller(size, node, flags,__builtin_return_address(0)); } EXPORT_SYMBOL(kvmalloc_node);static inline void *kvmalloc(size_t size, gfp_t flags) {return kvmalloc_node(size, flags, NUMA_NO_NODE); }

大于一個page的時候，會先用kmalloc()進(jìn)行__GFP_NORETRY的嘗試，如果嘗試失敗就fallback到vmalloc（NORETRY標(biāo)記避免了kmalloc在申請內(nèi)存失敗地情況下，反復(fù)嘗試甚至做OOM來獲得內(nèi)存）。

當(dāng)然，kvmalloc()的size如果小于1個page，則沿用老的kmalloc()邏輯，而且也不會設(shè)置__GFP_NORETRY，如果反復(fù)嘗試失敗的話，也不會fallback到vmalloc()，因為vmalloc()申請小于1個page的內(nèi)存是不合適的。

可觀看我今天的技術(shù)分享小視頻：

凡事都沒有絕對的，當(dāng)咱們還在糾結(jié)是kmalloc()還是vmalloc()的時候，人家已經(jīng)造出了kvmalloc()。咱的糾結(jié)，相對于人家的創(chuàng)造，是不是有一種要鉆進(jìn)去地洞的感覺？思考是最重要的，腦洞要開地大一點，被動地學(xué)習(xí)永遠(yuǎn)只是追著別人的腦子跑。?

(END)

更多精彩，盡在"Linux閱碼場"，掃描下方二維碼關(guān)注

您的鼓勵是我們前行的動力

總結(jié)

以上是生活随笔為你收集整理的宋宝华： kvmalloc ——倚天剑屠龙刀两大神器合体？的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。