malloc_usable_size() is a standard libc function. Implement it for the
barebox allocators, dlmalloc and tlsf_malloc.

Signed-off-by: Sascha Hauer <[email protected]>
---
 common/dlmalloc.c    | 12 ++++++++++++
 common/tlsf_malloc.c | 11 +++++++++++
 include/malloc.h     |  1 +
 3 files changed, 24 insertions(+)

diff --git a/common/dlmalloc.c b/common/dlmalloc.c
index c41487d54b..99f666aada 100644
--- a/common/dlmalloc.c
+++ b/common/dlmalloc.c
@@ -1431,6 +1431,18 @@ void free(void *mem)
                frontlink(p, sz, idx, bck, fwd);
 }
 
+size_t malloc_usable_size(void *mem)
+{
+       mchunkptr p;
+       size_t size;
+
+       if (!mem)
+               return 0;
+
+       p = mem2chunk(mem);
+       return chunksize(p);
+}
+
 /*
   Realloc algorithm:
 
diff --git a/common/tlsf_malloc.c b/common/tlsf_malloc.c
index 981f09de41..f07ef5645b 100644
--- a/common/tlsf_malloc.c
+++ b/common/tlsf_malloc.c
@@ -38,6 +38,17 @@ void free(void *mem)
 }
 EXPORT_SYMBOL(free);
 
+size_t malloc_usable_size(void *mem)
+{
+       size_t size;
+
+       if (!mem)
+               return 0;
+
+       return tlsf_block_size(mem);
+}
+EXPORT_SYMBOL(malloc_usable_size);
+
 void *realloc(void *oldmem, size_t bytes)
 {
        void *mem = tlsf_realloc(tlsf_mem_pool, oldmem, bytes);
diff --git a/include/malloc.h b/include/malloc.h
index d63853b91e..5cdff0a4f9 100644
--- a/include/malloc.h
+++ b/include/malloc.h
@@ -6,6 +6,7 @@
 #include <types.h>
 
 void *malloc(size_t) __alloc_size(1);
+size_t malloc_usable_size(void *);
 void free(void *);
 void *realloc(void *, size_t) __realloc_size(2);
 void *memalign(size_t, size_t) __alloc_size(2);
-- 
2.39.2


Reply via email to