commit e4fdd1dd8483d25eebd7f2974320fc044b28ebaf
author Simon Glass <sjg@chromium.org> Tue Jul 30 08:39:35 2024 -0600
committer Tom Rini <trini@konsulko.com> Wed Aug 07 08:49:10 2024 -0600
tree 9dd6e83984ac9d1f633bda441591573d2ce1f5b7
parent e6cf54c87ebe0fad286b6112f84560de3bc34796

malloc: Support testing with realloc()

At present in tests it is possible to cause an out-of-memory condition
with malloc() but not realloc(). Add support to realloc() too, so code
which uses that function can be tested.

Signed-off-by: Simon Glass <sjg@chromium.org>

common/dlmalloc.c

diff --git a/common/dlmalloc.c b/common/dlmalloc.c
index 62e8557..1e1602a 100644
--- a/common/dlmalloc.c
+++ b/common/dlmalloc.c
@@ -1698,6 +1698,10 @@
 		panic("pre-reloc realloc() is not supported");
 	}
 #endif
+  if (CONFIG_IS_ENABLED(UNIT_TEST) && malloc_testing) {
+    if (--malloc_max_allocs < 0)
+      return NULL;
+  }
 
   newp    = oldp    = mem2chunk(oldmem);
   newsize = oldsize = chunksize(oldp);