mirror of
https://github.com/wiiu-env/WiiUPluginLoaderBackend.git
synced 2024-12-26 21:01:52 +01:00
233 lines
8.9 KiB
C++
233 lines
8.9 KiB
C++
#include "hooks_patcher_static.h"
|
|
|
|
#include "globals.h"
|
|
#include "utils/config/ConfigUtils.h"
|
|
|
|
#include <coreinit/core.h>
|
|
#include <coreinit/messagequeue.h>
|
|
#include <hooks.h>
|
|
#include <padscore/wpad.h>
|
|
#include <vpad/input.h>
|
|
|
|
static uint8_t sVpadPressCooldown = 0xFF;
|
|
static bool sConfigMenuOpened = false;
|
|
static bool sWantsToOpenConfigMenu = false;
|
|
|
|
DECL_FUNCTION(void, GX2SwapScanBuffers, void) {
|
|
real_GX2SwapScanBuffers();
|
|
|
|
if (sWantsToOpenConfigMenu && !sConfigMenuOpened) {
|
|
sConfigMenuOpened = true;
|
|
ConfigUtils::openConfigMenu();
|
|
sConfigMenuOpened = false;
|
|
sWantsToOpenConfigMenu = false;
|
|
}
|
|
}
|
|
|
|
DECL_FUNCTION(void, GX2SetTVBuffer, void *buffer, uint32_t buffer_size, int32_t tv_render_mode, GX2SurfaceFormat format, GX2BufferingMode buffering_mode) {
|
|
gStoredTVBuffer.buffer = buffer;
|
|
gStoredTVBuffer.buffer_size = buffer_size;
|
|
gStoredTVBuffer.mode = tv_render_mode;
|
|
gStoredTVBuffer.surface_format = format;
|
|
gStoredTVBuffer.buffering_mode = buffering_mode;
|
|
|
|
return real_GX2SetTVBuffer(buffer, buffer_size, tv_render_mode, format, buffering_mode);
|
|
}
|
|
|
|
DECL_FUNCTION(void, GX2SetDRCBuffer, void *buffer, uint32_t buffer_size, uint32_t drc_mode, GX2SurfaceFormat surface_format, GX2BufferingMode buffering_mode) {
|
|
gStoredDRCBuffer.buffer = buffer;
|
|
gStoredDRCBuffer.buffer_size = buffer_size;
|
|
gStoredDRCBuffer.mode = drc_mode;
|
|
gStoredDRCBuffer.surface_format = surface_format;
|
|
gStoredDRCBuffer.buffering_mode = buffering_mode;
|
|
|
|
return real_GX2SetDRCBuffer(buffer, buffer_size, drc_mode, surface_format, buffering_mode);
|
|
}
|
|
|
|
static uint32_t lastData0 = 0;
|
|
|
|
|
|
DECL_FUNCTION(BOOL, OSSendMessage, OSMessageQueue *queue, OSMessage *message, OSMessageFlags flags) {
|
|
if (sConfigMenuOpened && queue == OSGetSystemMessageQueue()) {
|
|
if (message != nullptr) {
|
|
if (message->args[0] == 0xfacebacc) { // Release foreground
|
|
gConfigMenuShouldClose = true;
|
|
}
|
|
}
|
|
}
|
|
return real_OSSendMessage(queue, message, flags);
|
|
}
|
|
|
|
DECL_FUNCTION(uint32_t, OSReceiveMessage, OSMessageQueue *queue, OSMessage *message, uint32_t flags) {
|
|
const uint32_t res = real_OSReceiveMessage(queue, message, flags);
|
|
if (queue == OSGetSystemMessageQueue()) {
|
|
if (message != nullptr && res) {
|
|
if (lastData0 != message->args[0]) {
|
|
if (message->args[0] == 0xFACEF000) {
|
|
CallHook(gLoadedPlugins, WUPS_LOADER_HOOK_ACQUIRED_FOREGROUND);
|
|
} else if (message->args[0] == 0xD1E0D1E0) {
|
|
// Implemented via WUMS Hook
|
|
}
|
|
}
|
|
lastData0 = message->args[0];
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
DECL_FUNCTION(void, OSReleaseForeground) {
|
|
if (OSGetCoreId() == 1) {
|
|
CallHook(gLoadedPlugins, WUPS_LOADER_HOOK_RELEASE_FOREGROUND);
|
|
}
|
|
real_OSReleaseForeground();
|
|
}
|
|
|
|
DECL_FUNCTION(int32_t, VPADRead, int32_t chan, VPADStatus *buffer, uint32_t buffer_size, VPADReadError *error) {
|
|
if (sConfigMenuOpened) {
|
|
// Ignore reading vpad input only from other threads if the config menu is opened
|
|
if (OSGetCurrentThread() != gOnlyAcceptFromThread) {
|
|
return 0;
|
|
}
|
|
}
|
|
VPADReadError real_error = VPAD_READ_SUCCESS;
|
|
const int32_t result = real_VPADRead(chan, buffer, buffer_size, &real_error);
|
|
|
|
if (result > 0 && real_error == VPAD_READ_SUCCESS && buffer && ((buffer[0].hold & 0xFFFFF) == (VPAD_BUTTON_L | VPAD_BUTTON_DOWN | VPAD_BUTTON_MINUS)) && sVpadPressCooldown == 0 && !sConfigMenuOpened) {
|
|
|
|
sWantsToOpenConfigMenu = true;
|
|
sVpadPressCooldown = 0x3C;
|
|
return 0;
|
|
}
|
|
if (error) {
|
|
*error = real_error;
|
|
}
|
|
|
|
if (sVpadPressCooldown > 0) {
|
|
sVpadPressCooldown--;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
DECL_FUNCTION(void, WPADRead, WPADChan chan, WPADStatus *data) {
|
|
real_WPADRead(chan, data);
|
|
|
|
if (!sConfigMenuOpened && data && data[0].error == 0) {
|
|
if (const auto extensionType = data[0].extensionType; extensionType != 0xFF) {
|
|
if (extensionType == WPAD_EXT_CORE || extensionType == WPAD_EXT_NUNCHUK ||
|
|
extensionType == WPAD_EXT_MPLUS || extensionType == WPAD_EXT_MPLUS_NUNCHUK) {
|
|
if (data->buttons == (WPAD_BUTTON_B | WPAD_BUTTON_DOWN | WPAD_BUTTON_MINUS)) {
|
|
sWantsToOpenConfigMenu = true;
|
|
}
|
|
} else if (extensionType == WPAD_EXT_CLASSIC || extensionType == WPAD_EXT_MPLUS_CLASSIC) {
|
|
const auto *classic = reinterpret_cast<WPADStatusClassic *>(data);
|
|
if (classic->buttons == (WPAD_CLASSIC_BUTTON_L | WPAD_CLASSIC_BUTTON_DOWN | WPAD_CLASSIC_BUTTON_MINUS)) {
|
|
sWantsToOpenConfigMenu = true;
|
|
}
|
|
} else if (extensionType == WPAD_EXT_PRO_CONTROLLER) {
|
|
const auto *pro = reinterpret_cast<WPADStatusPro *>(data);
|
|
if (pro->buttons == (WPAD_PRO_TRIGGER_L | WPAD_PRO_BUTTON_DOWN | WPAD_PRO_BUTTON_MINUS)) {
|
|
sWantsToOpenConfigMenu = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// clang-format off
|
|
#define KiReport ((void(*)(const char *, ...)) 0xfff0ad0c)
|
|
// clang-format on
|
|
|
|
#pragma GCC push_options
|
|
#pragma GCC optimize("O0")
|
|
|
|
DECL_FUNCTION(uint32_t, SC17_FindClosestSymbol,
|
|
uint32_t addr,
|
|
uint32_t *outDistance,
|
|
char *symbolNameBuffer,
|
|
uint32_t symbolNameBufferLength,
|
|
char *moduleNameBuffer,
|
|
uint32_t moduleNameBufferLength) {
|
|
for (const auto &plugin : gLoadedPlugins) {
|
|
if (!plugin.isLinkedAndLoaded()) {
|
|
continue;
|
|
}
|
|
const auto sectionInfo = plugin.getPluginLinkInformation().getSectionInfo(".text");
|
|
if (!sectionInfo) {
|
|
continue;
|
|
}
|
|
|
|
if (!sectionInfo->isInSection(addr)) {
|
|
continue;
|
|
}
|
|
|
|
strncpy(moduleNameBuffer, plugin.getMetaInformation().getName().c_str(), moduleNameBufferLength - 1);
|
|
if (const auto functionSymbolData = plugin.getPluginLinkInformation().getNearestFunctionSymbolData(addr)) {
|
|
strncpy(symbolNameBuffer, functionSymbolData->getName().c_str(), moduleNameBufferLength - 1);
|
|
if (outDistance) {
|
|
*outDistance = addr - reinterpret_cast<uint32_t>(functionSymbolData->getAddress());
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
strncpy(symbolNameBuffer, ".text", symbolNameBufferLength);
|
|
|
|
if (outDistance) {
|
|
*outDistance = addr - sectionInfo->getAddress();
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
return real_SC17_FindClosestSymbol(addr, outDistance, symbolNameBuffer, symbolNameBufferLength, moduleNameBuffer, moduleNameBufferLength);
|
|
}
|
|
|
|
DECL_FUNCTION(uint32_t, KiGetAppSymbolName, uint32_t addr, char *buffer, int32_t bufSize) {
|
|
for (const auto &plugin : gLoadedPlugins) {
|
|
if (!plugin.isLinkedAndLoaded()) {
|
|
continue;
|
|
}
|
|
const auto sectionInfo = plugin.getPluginLinkInformation().getSectionInfo(".text");
|
|
if (!sectionInfo) {
|
|
continue;
|
|
}
|
|
|
|
if (!sectionInfo->isInSection(addr)) {
|
|
continue;
|
|
}
|
|
|
|
const auto pluginNameLen = strlen(plugin.getMetaInformation().getName().c_str());
|
|
int32_t spaceLeftInBuffer = bufSize - static_cast<int32_t>(pluginNameLen) - 1;
|
|
if (spaceLeftInBuffer < 0) {
|
|
spaceLeftInBuffer = 0;
|
|
}
|
|
strncpy(buffer, plugin.getMetaInformation().getName().c_str(), bufSize - 1);
|
|
|
|
if (const auto functionSymbolData = plugin.getPluginLinkInformation().getNearestFunctionSymbolData(addr)) {
|
|
buffer[pluginNameLen] = '|';
|
|
buffer[pluginNameLen + 1] = '\0';
|
|
strncpy(buffer + pluginNameLen + 1, functionSymbolData->getName().c_str(), spaceLeftInBuffer - 1);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
return real_KiGetAppSymbolName(addr, buffer, bufSize);
|
|
}
|
|
|
|
#pragma GCC pop_options
|
|
|
|
function_replacement_data_t method_hooks_static[] __attribute__((section(".data"))) = {
|
|
REPLACE_FUNCTION(GX2SwapScanBuffers, LIBRARY_GX2, GX2SwapScanBuffers),
|
|
REPLACE_FUNCTION(GX2SetTVBuffer, LIBRARY_GX2, GX2SetTVBuffer),
|
|
REPLACE_FUNCTION(GX2SetDRCBuffer, LIBRARY_GX2, GX2SetDRCBuffer),
|
|
REPLACE_FUNCTION(OSSendMessage, LIBRARY_COREINIT, OSSendMessage),
|
|
REPLACE_FUNCTION(OSReceiveMessage, LIBRARY_COREINIT, OSReceiveMessage),
|
|
REPLACE_FUNCTION(OSReleaseForeground, LIBRARY_COREINIT, OSReleaseForeground),
|
|
REPLACE_FUNCTION(VPADRead, LIBRARY_VPAD, VPADRead),
|
|
REPLACE_FUNCTION(WPADRead, LIBRARY_PADSCORE, WPADRead),
|
|
REPLACE_FUNCTION_VIA_ADDRESS(SC17_FindClosestSymbol, 0xfff10218, 0xfff10218),
|
|
REPLACE_FUNCTION_VIA_ADDRESS(KiGetAppSymbolName, 0xfff0e3a0, 0xfff0e3a0),
|
|
};
|
|
|
|
uint32_t method_hooks_static_size __attribute__((section(".data"))) = sizeof(method_hooks_static) / sizeof(function_replacement_data_t);
|