1 files changed, 88 insertions, 0 deletions
diff --git a/project-base/src/memory.h b/project-base/src/memory.h
new file mode 100644
index 0000000..ee9b5a7
--- /dev/null
+++ b/project-base/src/memory.h
@@ -0,0 +1,88 @@
+/* 
+*  BSD 2-Clause “Simplified” License
+*  Copyright (c) 2019, Aldrik Ramaekers, aldrik.ramaekers@protonmail.com
+*  All rights reserved.
+*/
+
+#ifndef INCLUDE_MEMORY
+#define INCLUDE_MEMORY
+
+#ifndef MODE_DEBUG
+#define mem_alloc(size) malloc(size);
+#define mem_free(p) free(p)
+#define mem_realloc(p, size) realloc(p, size)
+#define memory_print_leaks() {}
+#else
+#include <stdio.h>
+typedef struct t_mem_entry
+{
+	void* ptr;
+	u64 len;
+} mem_entry;
+array _mem_registry = {0,0,0,0,0};
+mutex _mem_mut;
+
+void* _registered_alloc(u64 size)
+{
+	if (_mem_registry.data == 0) {
+		_mem_mut = mutex_create();
+		array new_array;
+		new_array.length = 0;
+		new_array.reserved_length = 0;
+		new_array.entry_size = sizeof(mem_entry);
+		new_array.data = malloc(new_array.entry_size);
+		new_array.reserve_jump = 1;
+		new_array.mutex = mutex_create_recursive();
+
+		_mem_registry = new_array; 
+		array_reserve(&_mem_registry, 1000);
+	}
+	mutex_lock(&_mem_mut);
+	void* addr = malloc(size);
+	mem_entry new_entry;
+	new_entry.ptr = addr;
+	new_entry.len = size;
+	array_push(&_mem_registry, (uint8_t*)&new_entry);
+	mutex_unlock(&_mem_mut);
+	return addr;
+}
+
+void _registered_free(void* addr)
+{
+	mutex_lock(&_mem_mut);
+	for (int i = 0; i < _mem_registry.length; i++)
+	{
+		mem_entry *entry = array_at(&_mem_registry, i);
+		if (entry->ptr == addr) {
+			array_remove_at(&_mem_registry, i);
+			free(addr);
+			mutex_unlock(&_mem_mut);
+			return;
+		}
+	}
+	mutex_unlock(&_mem_mut);
+}
+
+void _registered_print_leaks()
+{
+	mutex_lock(&_mem_mut);
+	printf("Leaks:\n");
+	for (s32 i = 0; i < _mem_registry.length; i++)
+	{
+		mem_entry* entry = array_at(&_mem_registry, i);
+		printf("#%d: %I64d\n", i+1, entry->len);
+	}
+	mutex_unlock(&_mem_mut);
+}
+
+#define mem_alloc(size) _registered_alloc(size);
+#define mem_free(p) _registered_free(p)
+#define mem_realloc(p, size) realloc(p, size)
+#define memory_print_leaks() _registered_print_leaks();
+#endif
+
+#define STBI_MALLOC(sz) mem_alloc(sz)
+#define STBI_REALLOC(p, newsz) mem_realloc(p, newsz)
+#define STBI_FREE(p) mem_free(p)
+
+#endif
+\ No newline at end of file