Merge remote-tracking branch 'origin/master' into pikalax_work

1 files changed, 32 insertions, 0 deletions

diff --git a/arm9/lib/NitroSDK/src/custom_allocator.c b/arm9/lib/NitroSDK/src/custom_allocator.c
new file mode 100644
index 00000000..283c3500
--- /dev/null
+++ b/arm9/lib/NitroSDK/src/custom_allocator.c
@@ -0,0 +1,32 @@
+#include "custom_allocator.h"
+
+#include "global.h"
+#include "OS_alloc.h"
+
+static FreeFunc sDestructor;
+static AllocFunc sAllocator;
+
+// Custom allocator
+ARM_FUNC void* CallCustomAllocator(u32 size)
+{
+    if (sAllocator != NULL)
+        return sAllocator(size);
+    else
+        return OS_AllocFromHeap(OS_ARENA_MAIN, -1, size);
+}
+
+// Custom destructor
+ARM_FUNC void CallCustomDestructor(void * ptr)
+{
+    if (sDestructor != NULL)
+        sDestructor(ptr);
+    else
+        OS_FreeToHeap(OS_ARENA_MAIN, -1, ptr);
+}
+
+// Custom alloc/free setter
+ARM_FUNC void SetCustomAllocatorAndDestructor(AllocFunc allocator, FreeFunc destructor)
+{
+    sAllocator = allocator;
+    sDestructor = destructor;
+}