From 8301696801143ef07b2ede7da6d614e8d9992a1e Mon Sep 17 00:00:00 2001 From: Rusty Russell Date: Tue, 22 Mar 2011 13:14:23 +1030 Subject: [PATCH] talloc: fix leak in test/run-set_allocator.c --- ccan/talloc/test/run-set_allocator.c | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/ccan/talloc/test/run-set_allocator.c b/ccan/talloc/test/run-set_allocator.c index e485c623..e1c76d4c 100644 --- 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); -- 2.39.2