Added custom allocator support

test/test_memory.c

@@ -3,32 +3,74 @@
 #include "console_colors.h"
 
 #include <stdio.h>
+#include <stdlib.h>
 
-int main() {
+//the test allocator
+static int callCount = 0;
+
+void* allocator(void* pointer, size_t oldSize, size_t newSize) {
+	callCount++;
+
+	if (newSize == 0 && oldSize == 0) {
+		return NULL;
+	}
+
+	if (newSize == 0) {
+		free(pointer);
+		return NULL;
+	}
+
+	void* mem = realloc(pointer, newSize);
+
+	if (mem == NULL) {
+		exit(-1);
+	}
+
+	return mem;
+}
+
+void testMemoryAllocation() {
 	{
 		//test single pointer
 		int* integer = ALLOCATE(int, 1);
-
 		FREE(int, integer);
 	}
 
 	{
-		//test single pointer
+		//test single pointer array
 		int* array = ALLOCATE(int, 10);
 
+		array[1] = 42; //access the given memory
+
 		FREE_ARRAY(int, array, 10);
 	}
 
 	{
-		//test single pointer
+		//test multiple pointer arrays
 		int* array1 = ALLOCATE(int, 10);
 		int* array2 = ALLOCATE(int, 10);
 
+		array1[1] = 42; //access the given memory
+		array2[1] = 42; //access the given memory
+
 		FREE_ARRAY(int, array1, 10);
 		FREE_ARRAY(int, array2, 10);
 	}
+}
+
+int main() {
+	//test the default allocator
+	testMemoryAllocation();
+
+	//test the custom allocator
+	setAllocator(allocator);
+	testMemoryAllocation();
+
+	if (callCount != 8) {
+		fprintf(stderr, ERROR "Unexpected call count for custom allocator; was called %d times" RESET, callCount);
+		return -1;
+	}
 
 	printf(NOTICE "All good\n" RESET);
 	return 0;
 }
-