The-Homebrew-Cloud/WUMSLoader
2
0
Fork 0
mirror of https://github.com/wiiu-env/WUMSLoader.git synced 2025-02-21 10:22:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

Only check heap in debug mode

Browse Source
This commit is contained in:
Maschell 2024-04-28 18:15:02 +02:00
parent 4774694061
commit 488fbc0c1f
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
wumsloader/src/utils/hooks.cpp
View File

@ -2,9 +2,11 @@
#include "globals.h" #include "globals.h"
#include "module/ModuleData.h" #include "module/ModuleData.h"
#include "utils/logger.h" #include "utils/logger.h"
#include <coreinit/memexpheap.h>
#include <memory> #include <memory>
#include <wums.h> #include <wums.h>
#ifdef DEBUG
static const char **hook_names = (const char *[]){ static const char **hook_names = (const char *[]){
"WUMS_HOOK_INIT_WUT_MALLOC", "WUMS_HOOK_INIT_WUT_MALLOC",
"WUMS_HOOK_FINI_WUT_MALLOC", "WUMS_HOOK_FINI_WUT_MALLOC",
@ -26,6 +28,7 @@ static const char **hook_names = (const char *[]){
"WUMS_HOOK_RELOCATIONS_DONE", "WUMS_HOOK_RELOCATIONS_DONE",
"WUMS_HOOK_APPLICATION_REQUESTS_EXIT", "WUMS_HOOK_APPLICATION_REQUESTS_EXIT",
"WUMS_HOOK_DEINIT"}; "WUMS_HOOK_DEINIT"};
#endif
void CallHook(const std::vector<std::shared_ptr<ModuleData>> &modules, wums_hook_type_t type, bool condition) { void CallHook(const std::vector<std::shared_ptr<ModuleData>> &modules, wums_hook_type_t type, bool condition) {
if (condition) { if (condition) {
@ -46,8 +49,6 @@ void CallHook(const std::shared_ptr<ModuleData> &module, wums_hook_type_t type,
} }
} }
extern "C" bool MEMCheckExpHeap(void *heap, bool logProblems);
void CallHook(const std::shared_ptr<ModuleData> &module, wums_hook_type_t type) { void CallHook(const std::shared_ptr<ModuleData> &module, wums_hook_type_t type) {
bool foundHook = false; bool foundHook = false;
for (auto &curHook : module->getHookDataList()) { for (auto &curHook : module->getHookDataList()) {
@ -92,11 +93,10 @@ void CallHook(const std::shared_ptr<ModuleData> &module, wums_hook_type_t type)
break; break;
} }
} }
if (foundHook && !MEMCheckExpHeap(MEMGetBaseHeapHandle(MEM_BASE_HEAP_MEM2), true)) {
DEBUG_FUNCTION_LINE_ERR("MEM2 default heap is corrupted while calling hook %s for module %s", hook_names[type], module->getExportName().c_str());
#ifdef DEBUG #ifdef DEBUG
if (foundHook && !MEMCheckExpHeap(MEMGetBaseHeapHandle(MEM_BASE_HEAP_MEM2), MEM_EXP_HEAP_CHECK_FLAGS_LOG_ERRORS)) {
DEBUG_FUNCTION_LINE_ERR("MEM2 default heap is corrupted while calling hook %s for module %s", hook_names[type], module->getExportName().c_str());
OSFatal("WUMSLoader: MEM2 default heap is corrupted. \n Please restart the console."); OSFatal("WUMSLoader: MEM2 default heap is corrupted. \n Please restart the console.");
#endif
} }
#endif
} }