From 60d000a27717e52716a725d6fe785344bbb984f4 Mon Sep 17 00:00:00 2001
From: Maschell <Maschell@gmx.de>
Date: Wed, 20 Oct 2021 23:52:09 +0200
Subject: [PATCH] Replace the Spinlock with a Mutex, flush the heap header

---
 source/memory_mapping.cpp | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/source/memory_mapping.cpp b/source/memory_mapping.cpp
index 52ac7e2..68b7cca 100644
--- a/source/memory_mapping.cpp
+++ b/source/memory_mapping.cpp
@@ -11,9 +11,11 @@
 #include "CThread.h"
 #include <cstring>
 #include <coreinit/debug.h>
+#include <coreinit/mutex.h>
 #include <cstdio>
 
-OSSpinLock allocFreeSpinlock;
+OSMutex allocFreeMutex;
+
 
 // #define DEBUG_FUNCTION_LINE(x,...)
 
@@ -382,10 +384,11 @@ void MemoryMapping_setupMemoryMapping() {
     //readTestValuesFromMemory();
 
     //runOnAllCores(writeSegmentRegister,&srTableCpy);
+    OSInitMutex(&allocFreeMutex);
 }
 
 void *MemoryMapping_allocEx(uint32_t size, int32_t align, bool videoOnly) {
-    OSUninterruptibleSpinLock_Acquire(&allocFreeSpinlock);
+    OSLockMutex(&allocFreeMutex);
     void *res = nullptr;
     for (int32_t i = 0; /* waiting for a break */; i++) {
         if (mem_mapping[i].physical_addresses == nullptr) {
@@ -400,6 +403,8 @@ void *MemoryMapping_allocEx(uint32_t size, int32_t align, bool videoOnly) {
             continue;
         }
 
+        DCFlushRange(heap, sizeof(MEMExpHeap));
+
         res = MEMAllocFromExpHeapEx(heapHandle, size, align);
         auto cur = heap->usedList.head;
         while (cur != nullptr) {
@@ -415,7 +420,7 @@ void *MemoryMapping_allocEx(uint32_t size, int32_t align, bool videoOnly) {
             break;
         }
     }
-    OSUninterruptibleSpinLock_Release(&allocFreeSpinlock);
+    OSUnlockMutex(&allocFreeMutex);
     return res;
 }
 
@@ -431,7 +436,7 @@ void MemoryMapping_free(void *ptr) {
     if (ptr == nullptr) {
         return;
     }
-    OSUninterruptibleSpinLock_Acquire(&allocFreeSpinlock);
+    OSLockMutex(&allocFreeMutex);
     auto ptr_val = (uint32_t) ptr;
     for (int32_t i = 0; /* waiting for a break */; i++) {
         if (mem_mapping[i].physical_addresses == nullptr) {
@@ -442,6 +447,7 @@ void MemoryMapping_free(void *ptr) {
             auto *heap = (MEMExpHeap *) heapHandle;
 
             MEMFreeToExpHeap((MEMHeapHandle) mem_mapping[i].effective_start_address, ptr);
+            DCFlushRange(heap, sizeof(MEMExpHeap));
             auto cur = heap->usedList.head;
             while (cur != nullptr) {
                 DCFlushRange(cur, sizeof(MEMExpHeapBlock));
@@ -455,7 +461,7 @@ void MemoryMapping_free(void *ptr) {
             break;
         }
     }
-    OSUninterruptibleSpinLock_Release(&allocFreeSpinlock);
+    OSUnlockMutex(&allocFreeMutex);
 }
 
 uint32_t MemoryMapping_MEMGetAllocatableSize() {