[PATCH] lib/test_meminit: handle kmem_cache_alloc() failure

[PATCH] lib/test_meminit: handle kmem_cache_alloc() failure

[Tuo Li]

In do_kmem_cache_size(), kmem_cache_alloc() may fail and return NULL.
The returned pointer is subsequently passed to check_buf() and
fill_with_garbage_skip() without being checked, which may result in a
NULL pointer dereference.

Handle the allocation failure by marking the test as failed and
continuing with the next iteration.

Fixes: 5015a300a522 ("lib: introduce test_meminit module")
Signed-off-by: Tuo Li <islituo@gmail.com>
---
 lib/test_meminit.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/lib/test_meminit.c b/lib/test_meminit.c
index 68c3b9da090e..db4b4398e1c1 100644
--- a/lib/test_meminit.c
+++ b/lib/test_meminit.c
@@ -241,6 +241,10 @@ static int __init do_kmem_cache_size(size_t size, bool want_ctor,
 		}
 
 		buf = kmem_cache_alloc(c, alloc_mask);
+		if (!buf) {
+			fail = true;
+			continue;
+		}
 		/* Check that buf is zeroed, if it must be. */
 		fail |= check_buf(buf, size, want_ctor, want_rcu, want_zero);
 		fill_with_garbage_skip(buf, size, want_ctor ? CTOR_BYTES : 0);
-- 
2.43.0