talloc: fix leak in test/run-set_allocator.c

[ccan] / ccan / talloc / test / run-set_allocator.c

diff --git a/ccan/talloc/test/run-set_allocator.c b/ccan/talloc/test/run-set_allocator.c
index e485c62337c64a6487e5018f03a5bc0f4025659d..e1c76d4c045995956b9e1b68961a77dd6da49aa5 100644 (file)
--- a/ccan/talloc/test/run-set_allocator.c
+++ b/ccan/talloc/test/run-set_allocator.c
@@ -34,16 +34,27 @@ int main(int argc, char *argv[])
        failtest_init(argc, argv);
        talloc_set_allocator(my_malloc, my_free, my_realloc);
        p1 = talloc_array(NULL, int, 10);
+       if (!p1)
+               failtest_exit(exit_status());
        ok1(my_malloc_count == 1);
        ok1(my_free_count == 0);
        ok1(my_realloc_count == 0);
 
-       p1 = talloc_realloc(NULL, p1, int, 10000);
+       p2 = talloc_realloc(NULL, p1, int, 10000);
+       if (!p2) {
+               talloc_free(p1);
+               failtest_exit(exit_status());
+       }
+       p1 = p2;
        ok1(my_malloc_count == 1);
        ok1(my_free_count == 0);
        ok1(my_realloc_count == 1);
 
        p2 = talloc(p1, int);
+       if (!p2) {
+               talloc_free(p1);
+               failtest_exit(exit_status());
+       }
        ok1(my_malloc_count == 2);
        ok1(my_free_count == 0);
        ok1(my_realloc_count == 1);