Add simple memory leak check to all tests.
authorU. Artie Eoff <ullysses.a.eoff@intel.com>
Fri, 23 Mar 2012 17:01:23 +0000 (10:01 -0700)
committerKristian Høgsberg <krh@bitplanet.net>
Sat, 24 Mar 2012 18:37:42 +0000 (14:37 -0400)
Wrap all tests with a memory balance check to detect potential
memory leaks.
Fixed a few tests that had memory leaks contained in the tests
themselves.

Signed-off-by: U. Artie Eoff <ullysses.a.eoff@intel.com>
tests/Makefile.am
tests/array-test.c
tests/map-test.c
tests/test-runner.c
tests/test-runner.h

index cf81c0e..d7d12e0 100644 (file)
@@ -12,5 +12,5 @@ event_loop_test_SOURCES = event-loop-test.c test-runner.c
 AM_CFLAGS = $(GCC_CFLAGS)
 LDADD = $(top_builddir)/src/libwayland-util.la \
        $(top_builddir)/src/libwayland-server.la \
-       -lrt $(FFI_LIBS)
+       -lrt -ldl $(FFI_LIBS)
 
index cb62713..2be369f 100644 (file)
@@ -133,4 +133,6 @@ TEST(array_for_each)
        wl_array_for_each(p, &array)
                assert(*p == elements[i++]);
        assert(i == 5);
+
+       wl_array_release(&array);
 }
index 5ef940f..00de337 100644 (file)
@@ -90,4 +90,6 @@ TEST(map_remove)
        l = wl_map_insert_new(&map, WL_MAP_SERVER_SIDE, &d);
        assert(l == WL_SERVER_ID_START + 1);
        assert(wl_map_lookup(&map, l) == &d);
+
+       wl_map_release(&map);
 }
index 301c736..321d5c0 100644 (file)
@@ -20,6 +20,8 @@
  * OF THIS SOFTWARE.
  */
 
+#define _GNU_SOURCE
+
 #include <unistd.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <sys/wait.h>
 #include <string.h>
 #include <assert.h>
+#include <dlfcn.h>
 #include "test-runner.h"
 
+static int num_alloc;
+static void* (*sys_malloc)(size_t);
+static void (*sys_free)(void*);
+
 extern const struct test __start_test_section, __stop_test_section;
 
+void* malloc(size_t size)
+{
+       num_alloc++;
+       return sys_malloc(size);
+}
+
+void free(void* mem)
+{
+       if (mem != NULL)
+               num_alloc--;
+       sys_free(mem);
+}
+
 static const struct test *
 find_test(const char *name)
 {
@@ -46,7 +66,10 @@ find_test(const char *name)
 static void
 run_test(const struct test *t)
 {
+       int cur_alloc = num_alloc;
+
        t->run();
+       assert(("memory leak detected in test.", cur_alloc == num_alloc));
        exit(EXIT_SUCCESS);
 }
 
@@ -57,6 +80,10 @@ int main(int argc, char *argv[])
        int total, pass;
        siginfo_t info;
 
+       /* Load system malloc and free */
+       sys_malloc = dlsym(RTLD_NEXT, "malloc");
+       sys_free = dlsym(RTLD_NEXT, "free");
+
        if (argc == 2) {
                t = find_test(argv[1]);
                if (t == NULL) {
index c1fdff9..238e613 100644 (file)
@@ -1,3 +1,6 @@
+#ifndef _TEST_RUNNER_H_
+#define _TEST_RUNNER_H_
+
 struct test {
        const char *name;
        void (*run)(void);
@@ -12,3 +15,5 @@ struct test {
        };                                                      \
                                                                \
        static void name(void)
+
+#endif