test: Add functions to find the amount of allocated memory

The malloc() implementations provides a way of finding out the approximate
amount of memory that is allocated. Add helper functions to make it easier
to access this and see changes over time. This is useful for tests that
want to check if memory has been allocated or freed.

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

diff --git a/include/test/ut.h b/include/test/ut.h
index fbfde10719f90f14dd7d714d5bfaa51c7204c251..f616c202f357a4f72d8ca75e4f5045174fb33bd0 100644 (file)
--- a/include/test/ut.h
+++ b/include/test/ut.h
@@ -149,4 +149,20 @@ void ut_failf(struct unit_test_state *uts, const char *fname, int line,
 /* Assert that an operation succeeds (returns 0) */
 #define ut_assertok(cond)      ut_asserteq(0, cond)
 
+/**
+ * ut_check_free() - Return the number of bytes free in the malloc() pool
+ *
+ * @return bytes free
+ */
+ulong ut_check_free(void);
+
+/**
+ * ut_check_delta() - Return the number of bytes allocated/freed
+ *
+ * @last: Last value from ut_check_free
+ * @return free memory delta from @last; positive means more memory has been
+ *     allocated, negative means less has been allocated (i.e. some is freed)
+ */
+long ut_check_delta(ulong last);
+
 #endif

diff --git a/test/ut.c b/test/ut.c
index 55798041bafe937370fc69bc9899000ccf766efe..265da4a0d89a8d05fa50e39751c32e6b52ed1119 100644 (file)
--- a/test/ut.c
+++ b/test/ut.c
@@ -6,6 +6,7 @@
  */
 
 #include <common.h>
+#include <malloc.h>
 #include <test/test.h>
 #include <test/ut.h>
 
@@ -32,3 +33,16 @@ void ut_failf(struct unit_test_state *uts, const char *fname, int line,
        putc('\n');
        uts->fail_count++;
 }
+
+ulong ut_check_free(void)
+{
+       struct mallinfo info = mallinfo();
+
+       return info.uordblks;
+}
+
+long ut_check_delta(ulong last)
+{
+       return ut_check_free() - last;
+}
+