54 lines
2.0 KiB
C++
54 lines
2.0 KiB
C++
#include "applicationends_function_replacements.h"
|
|
#include "globals.h"
|
|
|
|
#include <coreinit/dynload.h>
|
|
#include <coreinit/messagequeue.h>
|
|
#include <wums.h>
|
|
|
|
static uint32_t lastData0 = 0;
|
|
|
|
void CallHook(wums_hook_type_t type) {
|
|
for (uint32_t i = 0; i < gModuleData->number_modules; i++) {
|
|
auto *curModule = &gModuleData->modules[i];
|
|
for (uint32_t j = 0; j < curModule->number_hook_entries; j++) {
|
|
auto hook_entry = &curModule->hook_entries[j];
|
|
auto hook_type = (wums_hook_type_t) hook_entry->type;
|
|
if (hook_type == type) {
|
|
if ((type == WUMS_HOOK_INIT_WRAPPER || type == WUMS_HOOK_FINI_WRAPPER) && curModule->skipInitFini) {
|
|
continue;
|
|
}
|
|
const void *target = (const void *) hook_entry->target;
|
|
((void (*)())((uint32_t *) target))();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
DECL_FUNCTION(void, _Exit, uint32_t status) {
|
|
CallHook(WUMS_HOOK_APPLICATION_ENDS);
|
|
CallHook(WUMS_HOOK_FINI_WUT_SOCKETS);
|
|
CallHook(WUMS_HOOK_FINI_WUT_DEVOPTAB);
|
|
real__Exit(status);
|
|
}
|
|
|
|
DECL_FUNCTION(uint32_t, OSReceiveMessage, OSMessageQueue *queue, OSMessage *message, uint32_t flags) {
|
|
uint32_t res = real_OSReceiveMessage(queue, message, flags);
|
|
if (queue == OSGetSystemMessageQueue()) {
|
|
if (message != nullptr && res) {
|
|
if (lastData0 != message->args[0]) {
|
|
if (message->args[0] == 0xD1E0D1E0) {
|
|
CallHook(WUMS_HOOK_APPLICATION_REQUESTS_EXIT);
|
|
}
|
|
}
|
|
lastData0 = message->args[0];
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
function_replacement_data_t applicationendshook_function_replacements[] = {
|
|
REPLACE_FUNCTION(OSReceiveMessage, LIBRARY_COREINIT, OSReceiveMessage),
|
|
REPLACE_FUNCTION(_Exit, LIBRARY_COREINIT, _Exit),
|
|
};
|
|
|
|
uint32_t applicationendshook_function_replacements_size = sizeof(applicationendshook_function_replacements) / sizeof(function_replacement_data_t); |