Fix: also count after realloc

tizilogic · Mar 24, 2023 · 7c15086 · 7c15086
1 parent 4a6fdfe
commit 7c15086
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/Sources/krink/memory.c b/Sources/krink/memory.c
@@ -52,7 +52,9 @@ static void kr_alloctrack_malloc(void *ptr, size_t size) {
 			kr_allocs[i].size = size;
 			++kr_pool_num_allocs;
 			kr_pool_allocated += size;
-			kr_pool_allocated_max = (kr_pool_allocated > kr_pool_allocated_max) ? kr_pool_allocated : kr_pool_allocated_max;
+			kr_pool_allocated_max = (kr_pool_allocated > kr_pool_allocated_max)
+			                            ? kr_pool_allocated
+			                            : kr_pool_allocated_max;
 			return;
 		}
 	}
@@ -64,12 +66,13 @@ static void kr_alloctrack_malloc(void *ptr, size_t size) {
 	}
 	kr_allocs[kr_allocs_count].ptr = ptr;
 	kr_allocs[kr_allocs_count].size = size;
+	++kr_pool_num_allocs;
 	kr_allocs_count *= 2;
 }
 
 static void kr_alloctrack_free(void *ptr) {
 	if (ptr == NULL) return;
-	
+
 	for (int i = 0; i < kr_allocs_count; ++i) {
 		if (kr_allocs[i].ptr == ptr) {
 			kr_allocs[i].ptr = NULL;
@@ -87,7 +90,9 @@ static void kr_alloctrack_realloc(void *old, void *ptr, size_t size) {
 		if (kr_allocs[i].ptr == old) {
 			kr_allocs[i].ptr = ptr;
 			kr_pool_allocated += size - kr_allocs[i].size;
-			kr_pool_allocated_max = (kr_pool_allocated > kr_pool_allocated_max) ? kr_pool_allocated : kr_pool_allocated_max;
+			kr_pool_allocated_max = (kr_pool_allocated > kr_pool_allocated_max)
+			                            ? kr_pool_allocated
+			                            : kr_pool_allocated_max;
 			kr_allocs[i].size = size;
 			return;
 		}