From d83ae9c923b2b271ffd2a86279dad91c7713146d Mon Sep 17 00:00:00 2001 From: Andrey Smirnov Date: Mon, 15 Oct 2018 10:00:17 -0700 Subject: tlsf_malloc: dummy_malloc: Share code for calloc() Calloc() implementation for TLSF does not correctly check for malloc() failure which can result in a NULL pointer exception when trying to calloc() extra large buffers. Since both TLSF and dummy malloc implementations of calloc() are exactly the same, pick implementation for the latter (which does aforementioned check) and share it between the two. Signed-off-by: Andrey Smirnov Signed-off-by: Sascha Hauer --- common/Makefile | 4 ++-- common/calloc.c | 19 +++++++++++++++++++ common/dummy_malloc.c | 13 ------------- common/tlsf_malloc.c | 16 ---------------- 4 files changed, 21 insertions(+), 31 deletions(-) create mode 100644 common/calloc.c (limited to 'common') diff --git a/common/Makefile b/common/Makefile index 13920cc5a6..861365bd55 100644 --- a/common/Makefile +++ b/common/Makefile @@ -34,8 +34,8 @@ obj-$(CONFIG_GLOBALVAR) += globalvar.o obj-$(CONFIG_GREGORIAN_CALENDER) += date.o obj-$(CONFIG_KALLSYMS) += kallsyms.o obj-$(CONFIG_MALLOC_DLMALLOC) += dlmalloc.o -obj-$(CONFIG_MALLOC_TLSF) += tlsf_malloc.o tlsf.o -obj-$(CONFIG_MALLOC_DUMMY) += dummy_malloc.o +obj-$(CONFIG_MALLOC_TLSF) += tlsf_malloc.o tlsf.o calloc.o +obj-$(CONFIG_MALLOC_DUMMY) += dummy_malloc.o calloc.o obj-$(CONFIG_MEMINFO) += meminfo.o obj-$(CONFIG_MENU) += menu.o obj-$(CONFIG_MODULES) += module.o diff --git a/common/calloc.c b/common/calloc.c new file mode 100644 index 0000000000..2b933ec272 --- /dev/null +++ b/common/calloc.c @@ -0,0 +1,19 @@ +#include +#include + +/* + * calloc calls malloc, then zeroes out the allocated chunk. + */ +void *calloc(size_t n, size_t elem_size) +{ + size_t size = elem_size * n; + void *r = malloc(size); + + if (!r) + return r; + + memset(r, 0x0, size); + + return r; +} +EXPORT_SYMBOL(calloc); diff --git a/common/dummy_malloc.c b/common/dummy_malloc.c index 641baa125a..fa4f5d126c 100644 --- a/common/dummy_malloc.c +++ b/common/dummy_malloc.c @@ -50,16 +50,3 @@ void *realloc(void *ptr, size_t size) { BUG(); } - -void *calloc(size_t n, size_t elem_size) -{ - size_t size = elem_size * n; - void *r = malloc(size); - - if (!r) - return r; - - memset(r, 0x0, size); - - return r; -} diff --git a/common/tlsf_malloc.c b/common/tlsf_malloc.c index a3541d8256..aa3ab23975 100644 --- a/common/tlsf_malloc.c +++ b/common/tlsf_malloc.c @@ -39,22 +39,6 @@ void *malloc(size_t bytes) } EXPORT_SYMBOL(malloc); -/* - * calloc calls malloc, then zeroes out the allocated chunk. - */ -void *calloc(size_t n, size_t elem_size) -{ - void *mem; - size_t sz; - - sz = n * elem_size; - mem = malloc(sz); - memset(mem, 0, sz); - - return mem; -} -EXPORT_SYMBOL(calloc); - void free(void *mem) { tlsf_free(tlsf_mem_pool, mem); -- cgit v1.2.3