1 files changed, 55 insertions, 0 deletions
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..73646030
--- /dev/null
+++ b/arm9/lib/libnns/src/NNS_FND_heapcommon.c
@@ -0,0 +1,55 @@
+#include "nitro.h"
+#include "NNS_FND_heapcommon.h"
+#include "NNS_FND_list.h"
+
+BOOL sRootListInitialized;
+NNSFndList sRootList;
+
+ARM_FUNC void *NNS_FndGetNextListObject(NNSFndList *, void *);
+
+ARM_FUNC static 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;
+}
+
+ARM_FUNC static NNSFndList* FindListContainHeap(const void * memBlock)
+{
+    NNSFndList* ret = &sRootList;
+    NNSiFndHeapHead* pHead = FindContainHeap(&sRootList, memBlock);
+    if (pHead != NULL)
+        ret = &pHead->childList;
+    return ret;
+}
+
+ARM_FUNC 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;
+    SetOptForHeap(pHead, optionFlag);
+    NNS_FndInitList(&pHead->childList, 4);
+    if (!sRootListInitialized)
+    {
+        NNS_FndInitList(&sRootList, 4);
+        sRootListInitialized = TRUE;
+    }
+    NNS_FndAppendListObject(FindListContainHeap(pHead), pHead);
+}
+
+ARM_FUNC void NNSi_FndFinalizeHeap(NNSiFndHeapHead *pHead)
+{
+    NNS_FndRemoveListObject(FindListContainHeap(pHead), pHead);
+}