summaryrefslogtreecommitdiff
path: root/include/nitro/os_alloc.c
diff options
context:
space:
mode:
Diffstat (limited to 'include/nitro/os_alloc.c')
-rw-r--r--include/nitro/os_alloc.c517
1 files changed, 517 insertions, 0 deletions
diff --git a/include/nitro/os_alloc.c b/include/nitro/os_alloc.c
new file mode 100644
index 00000000..022932ec
--- /dev/null
+++ b/include/nitro/os_alloc.c
@@ -0,0 +1,517 @@
+//
+// Created by mart on 4/23/20.
+//
+
+#include "os_alloc.h"
+#include "consts.h"
+#include "os_system.h"
+
+extern BOOL OSi_MainExArenaEnabled; // TODO: located at 0x021d36f4
+extern BOOL OSi_Initialized; // TODO: located at 0x021d36f0
+extern u32 OS_GetConsoleType();
+extern u32 OS_CONSOLE_SIZE_MASK;
+extern u32 OS_CONSOLE_SIZE_4MB;
+extern Cell* DLExtract(Cell* list, Cell* cell);
+extern Cell* DLInsert(Cell* list, Cell* cell);
+extern Cell* DLAddFront(Cell* list, Cell* cell);
+
+void* OS_AllocFromArenaHi(OSArenaId id, u32 size, u32 align) {
+#ifdef MATCH_ASM
+ __asm {
+ stmdb sp!,{ r4 r5 r6 lr }
+ mov r4,r0
+ mov r6,r1
+ mov r5,r2
+ bl OS_GetArenaHi
+ cmp r0,#0x0
+ moveq r0,#0x0
+ ldmeqia sp!,{ r4 r5 r6 lr }
+ bxeq lr
+ sub r1,r5,#0x1
+ mvn r2,r1
+ and r0,r0,r2
+ sub r1,r0,r6
+ mov r0,r4
+ and r5,r1,r2
+ bl OS_GetArenaLo
+ cmp r5,r0
+ movcc r0,#0x0
+ ldmccia sp!,{ r4 r5 r6 lr }
+ bxcc lr
+ mov r0,r4
+ mov r1,r5
+ bl OS_SetArenaHi
+ mov r0,r5
+ ldmia sp!,{ r4 r5 r6 lr }
+ bx lr
+ };
+#else
+ void* ptr;
+ u8* arenaHi;
+
+ arenaHi = OS_GetArenaHi(id);
+ if (!arenaHi) {
+ return NULL;
+ }
+
+ arenaHi = (u8 *)OSi_TRUNC(arenaHi, align);
+ arenaHi -= size;
+ arenaHi = ptr = (void *)OSi_TRUNC(arenaHi, align);
+
+ if (arenaHi < (u8*)OS_GetArenaLo(id)) {
+ return NULL;
+ }
+
+ OS_SetArenaHi(id, arenaHi);
+
+ return ptr;
+#endif
+}
+
+void* OS_AllocFromArenaLo(OSArenaId id, u32 size, u32 align) {
+#ifdef MATCH_ASM
+ __asm {
+ stmdb sp!,{ r4 r5 r6 r7 lr }
+ sub sp,sp,#0x4
+ mov r7,r0
+ mov r6,r1
+ mov r5,r2
+ bl OS_GetArenaLo
+ cmp r0,#0x0
+ addeq sp,sp,#0x4
+ moveq r0,#0x0
+ ldmeqia sp!,{ r4 r5 r6 r7 lr }
+ bxeq lr
+ add r0,r0,r5
+ sub r1,r5,#0x1
+ mvn r2,r1
+ sub r0,r0,#0x1
+ and r4,r2,r0
+ add r0,r4,r6
+ add r0,r0,r5
+ sub r1,r0,#0x1
+ mov r0,r7
+ and r5,r2,r1
+ bl OS_GetArenaHi
+ cmp r5,r0
+ addhi sp,sp,#0x4
+ movhi r0,#0x0
+ ldmhiia sp!,{ r4 r5 r6 r7 lr }
+ bxhi lr
+ mov r0,r7
+ mov r1,r5
+ bl OS_SetArenaLo
+ mov r0,r4
+ add sp,sp,#0x4
+ ldmia sp!,{ r4 r5 r6 r7 lr }
+ bx lr
+ }
+#else
+ void* ptr;
+ u8* arenaLo;
+ ptr = OS_GetArenaLo(id);
+ if (!ptr) {
+ return NULL;
+ }
+ arenaLo = ptr = (void *)OSi_ROUND(ptr, align);
+ arenaLo += size;
+ arenaLo = (u8 *)OSi_ROUND(arenaLo, align);
+ if (arenaLo > (u8*)OS_GetArenaHi(id)) {
+ return NULL;
+ }
+ OS_SetArenaLo(id, arenaLo);
+
+ return ptr;
+#endif
+}
+
+void OS_SetArenaLo(OSArenaId id, void* newLo) {
+#ifdef MATCH_ASM
+ __asm {
+ mov r0,r0, lsl #0x2
+ add r0,r0,#0x2700000
+ add r0,r0,#0xff000
+ str r1,[r0,#0xda0]
+ bx lr
+ }
+#else
+ OSi_GetArenaInfo().lo[id] = newLo;
+#endif
+}
+
+void OS_SetArenaHi(OSArenaId id, void* newHi) {
+#ifdef MATCH_ASM
+ __asm {
+ mov r0,r0, lsl #0x2
+ add r0,r0,#0x2700000
+ add r0,r0,#0xff000
+ str r1,[r0,#0xdc4]
+ bx lr
+ }
+#else
+ OSi_GetArenaInfo().lo[id] = newHi;
+#endif
+}
+
+void* OS_GetInitArenaLo(OSArenaId id) {
+#ifdef MATCH_ASM
+ __asm {
+ // TODO: idk how to do switch case stuff properly in asm
+ }
+#else
+ switch (id) {
+ case OS_ARENA_MAIN:
+ return (void *)0x0225ffa0;
+ case OS_ARENA_MAINEX:
+ if (!OSi_MainExArenaEnabled || (OS_GetConsoleType() & OS_CONSOLE_SIZE_MASK) == OS_CONSOLE_SIZE_4MB) {
+ return NULL;
+ } else {
+ return (void *)0x023e0000;
+ }
+ case OS_ARENA_ITCM:
+ return (void *)0x01ff8720;
+ case OS_ARENA_DTCM:
+ return (void *)0x027e0080;
+ case OS_ARENA_SHARED:
+ return (void *)0x027ff000;
+ case OS_ARENA_WRAM_MAIN:
+ return (void *)0x037f8000;
+ default:
+ return NULL;
+ }
+#endif
+}
+
+void* OS_GetInitArenaHi(OSArenaId id) {
+#ifdef MATCH_ASM
+ __asm {
+ // TODO: idk how to do switch case stuff properly in asm
+ }
+#else
+ switch (id) {
+ case OS_ARENA_MAIN:
+ return (void *)0x023e0000;
+ case OS_ARENA_MAINEX:
+ if (!OSi_MainExArenaEnabled || (OS_GetConsoleType() & OS_CONSOLE_SIZE_MASK) == OS_CONSOLE_SIZE_4MB) {
+ return (void *)0;
+ } else {
+ return (void *)0x02700000;
+ }
+ case OS_ARENA_ITCM:
+ return (void *)0x02000000;
+ case OS_ARENA_DTCM:
+ return (void *)0x027e0080;
+ case OS_ARENA_SHARED:
+ return (void *)0x027ff680;
+ case OS_ARENA_WRAM_MAIN:
+ return (void *)0x037f8000;
+ default:
+ return NULL;
+ }
+#endif
+}
+
+void* OS_GetArenaLo(OSArenaId id) {
+#ifdef MATCH_ASM
+ __asm {
+ mov r0,r0, lsl #0x2
+ add r0,r0,#0x2700000
+ add r0,r0,#0xff000
+ ldr r0,[r0,#0xda0]
+ bx lr
+ }
+#else
+ return OSi_GetArenaInfo().lo[id];
+#endif
+}
+
+void* OS_GetArenaHi(OSArenaId id) {
+#ifdef MATCH_ASM
+ __asm {
+ mov r0,r0, lsl #0x2
+ add r0,r0,#0x2700000
+ add r0,r0,#0xff000
+ ldr r0,[r0,#0xdc4]
+ bx lr
+ }
+#else
+ return OSi_GetArenaInfo().hi[id];
+#endif
+}
+
+void OS_InitArenaEx() {
+#ifdef MATCH_ASM
+ __asm {
+ stmdb sp!,{ lr }
+ sub sp,sp,#0x4
+ mov r0,#0x2
+ bl OS_GetInitArenaHi
+ mov r1,r0
+ mov r0,#0x2
+ bl OS_SetArenaHi
+ mov r0,#0x2
+ bl OS_GetInitArenaLo
+ mov r1,r0
+ mov r0,#0x2
+ bl OS_SetArenaLo
+ ldr r0,OSi_MainExArenaEnabled
+ ldr r0,[r0,#0x0]
+ cmp r0,#0x0
+ beq LAB_020cc5b8
+ bl OS_GetConsoleType
+ and r0,r0,#0x3
+ cmp r0,#0x1
+ addne sp,sp,#0x4
+ ldmneia sp!,{ lr }
+ bxne lr
+ LAB_020cc5b8:
+ ldr r0,#0200002b
+ bl OS_SetProtectionRegion1
+ ldr r0,#0x023e0021
+ bl OS_SetProtectionRegion2
+ add sp,sp,#0x4
+ ldmia sp!,{ lr }
+ bx lr
+ }
+#else
+ void* uVar1;
+
+ uVar1 = OS_GetInitArenaHi(OS_ARENA_MAINEX);
+ OS_SetArenaHi(2,uVar1);
+ uVar1 = OS_GetInitArenaLo(OS_ARENA_MAINEX);
+ OS_SetArenaLo(2,uVar1);
+ if (!OSi_MainExArenaEnabled || (OS_GetConsoleType() & OS_CONSOLE_SIZE_MASK) == OS_CONSOLE_SIZE_4MB) {
+ return;
+ }
+ // TODO:
+ // OS_SetProtectionRegion1(&UNK_0200002b);
+ // OS_SetProtectionRegion2(0x023e0021);
+#endif
+}
+
+void OS_InitArena() {
+#ifdef MATCH_ASM
+ __asm {
+ stmdb sp!,{ lr }
+ sub sp,sp,#0x4
+ ldr r1,OSi_Initialized
+ ldr r0,[r1,#0x0]
+ cmp r0,#0x0
+ addne sp,sp,#0x4
+ ldmneia sp!,{ lr }
+ bxne lr
+ mov r2,#0x1
+ mov r0,#0x0
+ str r2,[r1,#0x0]
+ bl OS_GetInitArenaHi
+ mov r1,r0
+ mov r0,#0x0
+ bl OS_SetArenaHi
+ mov r0,#0x0
+ bl OS_GetInitArenaLo
+ mov r1,r0
+ mov r0,#0x0
+ bl OS_SetArenaLo
+ mov r0,#0x2
+ mov r1,#0x0
+ bl OS_SetArenaLo
+ mov r0,#0x2
+ mov r1,#0x0
+ bl OS_SetArenaHi
+ mov r0,#0x3
+ bl OS_GetInitArenaHi
+ mov r1,r0
+ mov r0,#0x3
+ bl OS_SetArenaHi
+ mov r0,#0x3
+ bl OS_GetInitArenaLo
+ mov r1,r0
+ mov r0,#0x3
+ bl OS_SetArenaLo
+ mov r0,#0x4
+ bl OS_GetInitArenaHi
+ mov r1,r0
+ mov r0,#0x4
+ bl OS_SetArenaHi
+ mov r0,#0x4
+ bl OS_GetInitArenaLo
+ mov r1,r0
+ mov r0,#0x4
+ bl OS_SetArenaLo
+ mov r0,#0x5
+ bl OS_GetInitArenaHi
+ mov r1,r0
+ mov r0,#0x5
+ bl OS_SetArenaHi
+ mov r0,#0x5
+ bl OS_GetInitArenaLo
+ mov r1,r0
+ mov r0,#0x5
+ bl OS_SetArenaLo
+ mov r0,#0x6
+ bl OS_GetInitArenaHi
+ mov r1,r0
+ mov r0,#0x6
+ bl OS_SetArenaHi
+ mov r0,#0x6
+ bl OS_GetInitArenaLo
+ mov r1,r0
+ mov r0,#0x6
+ bl OS_SetArenaLo
+ add sp,sp,#0x4
+ ldmia sp!,{ lr }
+ bx lr
+ }
+#else
+ if (OSi_Initialized) {
+ return;
+ }
+ OSi_Initialized = TRUE;
+
+ OS_SetArenaHi(OS_ARENA_MAIN, OS_GetInitArenaHi(OS_ARENA_MAIN));
+ OS_SetArenaLo(OS_ARENA_MAIN, OS_GetInitArenaLo(OS_ARENA_MAIN));
+
+ OS_SetArenaLo(OS_ARENA_MAINEX, (void *)0);
+ OS_SetArenaHi(OS_ARENA_MAINEX, (void *)0);
+
+ OS_SetArenaHi(OS_ARENA_ITCM, OS_GetInitArenaHi(OS_ARENA_ITCM));
+ OS_SetArenaLo(OS_ARENA_ITCM, OS_GetInitArenaLo(OS_ARENA_ITCM));
+
+ OS_SetArenaHi(OS_ARENA_DTCM, OS_GetInitArenaHi(OS_ARENA_DTCM));
+ OS_SetArenaLo(OS_ARENA_DTCM, OS_GetInitArenaLo(OS_ARENA_DTCM));
+
+ OS_SetArenaHi(OS_ARENA_SHARED, OS_GetInitArenaHi(OS_ARENA_SHARED));
+ OS_SetArenaLo(OS_ARENA_SHARED, OS_GetInitArenaLo(OS_ARENA_SHARED));
+
+ OS_SetArenaHi(OS_ARENA_WRAM_MAIN, OS_GetInitArenaHi(OS_ARENA_WRAM_MAIN));
+ OS_SetArenaLo(OS_ARENA_WRAM_MAIN, OS_GetInitArenaLo(OS_ARENA_WRAM_MAIN));
+#endif
+}
+
+#define HEADERSIZE OSi_ROUND(sizeof(Cell), 32)
+#define MINOBJSIZE (HEADERSIZE+32)
+
+void OS_FreeToHeap(OSArenaId id, OSHeapHandle heap, void* ptr) {
+#ifdef MATCH_ASM
+ __asm {
+ stmdb sp!,{ r4 r5 r6 r7 lr }
+ sub sp,sp,#0x4
+ mov r7,param_1
+ mov r5,param_2
+ mov r4,param_3
+ bl OS_DisableInterrupts
+ ldr param_2,OSiHeapInfo
+ mov r6,param_1
+ ldr param_1,[param_2,r7,lsl #0x2]
+ cmp r5,#0x0
+ ldrlt r5,[param_1,#0x0]
+ ldr param_2,[param_1,#0x10]
+ mov param_1,#0xc
+ mla r7,r5,param_1,param_2
+ sub r4,r4,#0x20
+ ldr param_1,[r7,#0x8]
+ mov param_2,r4
+ bl DLExtract
+ str param_1,[r7,#0x8]
+ ldr param_1,[r7,#0x4]
+ mov param_2,r4
+ bl DLInsert
+ str param_1,[r7,#0x4]
+ mov param_1,r6
+ bl OS_RestoreInterrupts
+ add sp,sp,#0x4
+ ldmia sp!,{ r4 r5 r6 r7 lr }
+ bx lr
+ }
+#else
+ OSHeapInfo *heapInfo;
+ HeapDesc *hd;
+ Cell *cell;
+
+ OSIntrMode enabled = OS_DisableInterrupts();
+ heapInfo = OSiHeapInfo[id];
+
+ if (heap < 0) {
+ heap = heapInfo->currentHeap;
+ }
+
+ cell = (Cell *) ((char *)ptr - HEADERSIZE);
+ hd = &heapInfo->heapArray[heap];
+
+ hd->allocated = DLExtract(hd->allocated, cell);
+ hd->free = DLInsert(hd->free, cell);
+
+ OS_RestoreInterrupts(enabled);
+
+#endif
+}
+
+void* OS_AllocFromHeap(OSArenaId id, OSHeapHandle heap, u32 size) {
+#ifdef MATCH_ASM
+ __asm {
+
+ }
+#else
+ OSHeapInfo* heapInfo;
+ HeapDesc* hd;
+ Cell* cell;
+ Cell* newCell;
+ long leftoverSize;
+
+ OSIntrMode enabled = OS_DisableInterrupts();
+ heapInfo = OSiHeapInfo[id];
+ if (!heapInfo) {
+ OS_RestoreInterrupts(enabled);
+ return NULL;
+ }
+
+ if (heap < 0) {
+ heap = heapInfo->currentHeap;
+ }
+
+ hd = &heapInfo->heapArray[heap];
+
+ size += HEADERSIZE;
+ size = OSi_ROUND(size, 32);
+
+ for (cell = hd->free; cell != NULL; cell = cell->next) {
+ if ((long)size <= cell->size) {
+ break;
+ }
+ }
+
+ if (cell == NULL) {
+ OS_RestoreInterrupts(enabled);
+ return NULL;
+ }
+
+ leftoverSize = cell->size - (long)size;
+ if (leftoverSize < MINOBJSIZE) {
+ hd->free = DLExtract(hd->free, cell);
+ } else {
+ cell->size = (long)size;
+
+ newCell = (Cell *) ((char *)cell + size);
+ newCell->size = leftoverSize;
+
+ newCell->prev = cell->prev;
+ newCell->next = cell->next;
+
+ if (newCell->next != NULL) {
+ newCell->next->prev = newCell;
+ }
+
+ if (newCell->prev != NULL) {
+ newCell->prev->next = newCell;
+ } else {
+ hd->free = newCell;
+ }
+ }
+
+ hd->allocated = DLAddFront(hd->allocated, cell);
+
+ OS_RestoreInterrupts(enabled);
+ return (void *)((char *)cell + HEADERSIZE);
+
+#endif
+} \ No newline at end of file
generated by cgit v1.2.3 (git 2.25.1) at 2025-05-17 01:52:45 +0000