The-Homebrew-Cloud/MemoryMappingModule
2
0
Fork 0
mirror of https://github.com/wiiu-env/MemoryMappingModule.git synced 2024-11-22 09:49:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Round up the size of the allocated memory to avoid heap corruption

Browse Source
This commit is contained in:
Maschell 2022-02-11 20:36:37 +01:00
parent cd56a105e3
commit f513e1a374
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
source/memory_mapping.cpp
View File

@ -395,6 +395,9 @@ void MemoryMapping_setupMemoryMapping() {
OSInitMutex(&allocMutex); OSInitMutex(&allocMutex);
} }
#define ROUNDDOWN(val, align) ((val) & ~(align - 1))
#define ROUNDUP(val, align) ROUNDDOWN(((val) + (align - 1)), align)
void *MemoryMapping_allocEx(uint32_t size, int32_t align, bool videoOnly) { void *MemoryMapping_allocEx(uint32_t size, int32_t align, bool videoOnly) {
OSLockMutex(&allocMutex); OSLockMutex(&allocMutex);
void *res = nullptr; void *res = nullptr;
@ -411,7 +414,10 @@ void *MemoryMapping_allocEx(uint32_t size, int32_t align, bool videoOnly) {
continue; continue;
} }
res = MEMAllocFromExpHeapEx(heapHandle, size, align); // We round up the size to avoid heap corruption.
// FSReadFile expects the buffer size to be a multiple of 0x40
// This can remove once all modules/plugins have been updated :)
res = MEMAllocFromExpHeapEx(heapHandle, ROUNDUP(size, 0x40), align);
if (res != nullptr) { if (res != nullptr) {
break; break;
} }