From 37461da16af3814029d9ae6adcdb6483872a6eb4 Mon Sep 17 00:00:00 2001
From: PikalaxALT <pikalaxalt@gmail.com>
Date: Wed, 18 Aug 2021 20:51:11 -0400
Subject: NNS_FND_heapcommon

---
 arm9/lib/libnns/src/NNS_FND_heapcommon.c | 61 ++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 arm9/lib/libnns/src/NNS_FND_heapcommon.c

(limited to 'arm9/lib/libnns/src')

diff --git a/arm9/lib/libnns/src/NNS_FND_heapcommon.c b/arm9/lib/libnns/src/NNS_FND_heapcommon.c
new file mode 100644
index 00000000..2520aab2
--- /dev/null
+++ b/arm9/lib/libnns/src/NNS_FND_heapcommon.c
@@ -0,0 +1,61 @@
+#include "nitro.h"
+#include "NNS_FND_heapcommon.h"
+#include "NNS_FND_list.h"
+
+BOOL sRootListInitialized;
+NNSFndList sRootList;
+
+static inline void NNSi_FndHeapHeadSetOptionFlag(NNSiFndHeapHead *pHead, u8 option_flag)
+{
+    pHead->attribute &= ~0xFF;
+    pHead->attribute |= option_flag;
+}
+
+void *NNS_FndGetNextListObject(NNSFndList *, void *);
+
+NNSiFndHeapHead* FindContainHeap(NNSFndList * pList, const void * memBlock)
+{
+    NNSiFndHeapHead * pHead = NULL;
+
+    while ((pHead = NNS_FndGetNextListObject(pList, pHead)) != NULL)
+    {
+        if (pHead->heapStart <= memBlock && memBlock < pHead->heapEnd)
+        {
+            NNSiFndHeapHead * ret = FindContainHeap(&pHead->childList, memBlock);
+            if (ret == NULL)
+                ret = pHead;
+            return ret;
+        }
+    }
+    return NULL;
+}
+
+NNSFndList* FindListContainHeap(const void * memBlock)
+{
+    NNSFndList* ret = &sRootList;
+    NNSiFndHeapHead* pHead = FindContainHeap(&sRootList, memBlock);
+    if (pHead != NULL)
+        ret = &pHead->childList;
+    return ret;
+}
+
+void NNSi_FndInitHeapHead(NNSiFndHeapHead *pHead, u32 signature, void* heapStart, void* heapEnd, u16 optionFlag)
+{
+    pHead->signature = signature;
+    pHead->heapStart = heapStart;
+    pHead->heapEnd = heapEnd;
+    pHead->attribute = 0;
+    NNSi_FndHeapHeadSetOptionFlag(pHead, optionFlag);
+    NNS_FndInitList(&pHead->childList, 4);
+    if (!sRootListInitialized)
+    {
+        NNS_FndInitList(&sRootList, 4);
+        sRootListInitialized = TRUE;
+    }
+    NNS_FndAppendListObject(FindListContainHeap(pHead), pHead);
+}
+
+void NNSi_FndFinalizeHeap(NNSiFndHeapHead *pHead)
+{
+    NNS_FndRemoveListObject(FindListContainHeap(pHead), pHead);
+}
-- 
cgit v1.2.3