From: zhang jiao <zhangjiao2@cmss.chinamobile.com>
Add a return value check of malloc() and calloc() in lib/slab.c
Signed-off-by: zhang jiao <zhangjiao2@cmss.chinamobile.com>
---
tools/lib/slab.c | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/tools/lib/slab.c b/tools/lib/slab.c
index 981a21404f32..07c98926c0e4 100644
--- a/tools/lib/slab.c
+++ b/tools/lib/slab.c
@@ -19,6 +19,8 @@ void *kmalloc(size_t size, gfp_t gfp)
return NULL;
ret = malloc(size);
+ if (!ret)
+ return NULL;
uatomic_inc(&kmalloc_nr_allocated);
if (kmalloc_verbose)
printf("Allocating %p from malloc\n", ret);
@@ -45,6 +47,8 @@ void *kmalloc_array(size_t n, size_t size, gfp_t gfp)
return NULL;
ret = calloc(n, size);
+ if (!ret)
+ return NULL;
uatomic_inc(&kmalloc_nr_allocated);
if (kmalloc_verbose)
printf("Allocating %p from calloc\n", ret);
--
2.33.0