From 370f16422df97a5c2a2d3a1f57bea0f48fb76ae2 Mon Sep 17 00:00:00 2001
From: acevest <zhaoyanbai@126.com>
Date: Thu, 4 Nov 2021 08:48:22 +0800
Subject: [PATCH] =?utf8?q?kmalloc=20=E5=A6=82=E6=9E=9C=E6=B2=A1=E6=9C=89?=
 =?utf8?q?=E6=89=BE=E5=88=B0=E5=90=88=E9=80=82=E7=9A=84kmem=5Fcache=5Ft?=
 =?utf8?q?=E5=B0=B1=E5=88=86=E9=85=8D=E5=A4=B1=E8=B4=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 mm/slub.c | 23 ++++++++++++++++++-----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/mm/slub.c b/mm/slub.c
index 2e48619..d182f2c 100644
--- a/mm/slub.c
+++ b/mm/slub.c
@@ -46,7 +46,9 @@ static bool calculate_params(kmem_cache_t *cache) {
 
     cache->objects = (PAGE_SIZE << cache->order) / cache->size;
 
-    if (0 == cache->objects) return false;
+    if (0 == cache->objects) {
+        return false;
+    }
 
     return true;
 }
@@ -61,7 +63,9 @@ static bool kmem_cache_init(kmem_cache_t *cache, const char *name, size_t size,
     cache->partial_cnt = 0;
     INIT_LIST_HEAD(&(cache->partial));
 
-    if (!calculate_params(cache)) goto err;
+    if (!calculate_params(cache)) {
+        goto err;
+    }
 
     return true;
 err:
@@ -124,7 +128,9 @@ static void *__slub_alloc(kmem_cache_t *cache, gfp_t gfpflags) {
         }
     }
 
-    if (cache->page == 0) return 0;
+    if (cache->page == 0) {
+        return 0;
+    }
 
     object = cache->page->freelist;
 
@@ -208,6 +214,7 @@ void kmem_cache_free(kmem_cache_t *cache, void *addr) {
 void *kmalloc(size_t size, gfp_t gfpflags) {
     unsigned int i;
     kmem_cache_t *cache = 0;
+    void *addr = 0;
 
     unsigned long flags;
     irq_save(flags);
@@ -220,8 +227,14 @@ void *kmalloc(size_t size, gfp_t gfpflags) {
         }
     }
 
-    void *addr = kmem_cache_alloc(cache, gfpflags);
+    // 如果没找到支持的cache则分配失败
+    if (0 == cache) {
+        goto err;
+    }
+
+    addr = kmem_cache_alloc(cache, gfpflags);
 
+err:
     irq_restore(flags);
 
     return addr;
@@ -297,5 +310,5 @@ void init_slub_system() {
         printk("kmalloc addr %08x\n", (unsigned long) addr);
         addrs[i] = addr;
     }
-#endif
+#endifq
 }
-- 
2.44.0