diff --git a/arena.c b/arena.c index 50a5594..fb850c7 100644 --- a/arena.c +++ b/arena.c @@ -83,5 +83,7 @@ void *try_alloc_arena() if (!current_arena || !current_arena->free_count) { return NULL; } - return current_arena->free_blocks[--current_arena->free_count]; + void *ret = current_arena->free_blocks[--current_arena->free_count]; + track_block(ret); + return ret; }