libfunctionpatcher/source/utils.cpp

181 lines
8.7 KiB
C++

#include "logger.h"
#include <coreinit/debug.h>
#include <coreinit/dynload.h>
#include <function_patcher/function_patching.h>
static OSDynLoad_Module sModuleHandle = nullptr;
// Introduced in API version 2.
static FunctionPatcherStatus (*sFPGetVersion)(FunctionPatcherAPIVersion *) = nullptr;
static FunctionPatcherStatus (*sFPAddFunctionPatch)(function_replacement_data_t *function_data, PatchedFunctionHandle *outHandle, bool *outHasBeenPatched) = nullptr;
static FunctionPatcherStatus (*sFPRemoveFunctionPatch)(PatchedFunctionHandle handle) = nullptr;
static FunctionPatcherStatus (*sFPIsFunctionPatched)(PatchedFunctionHandle handle, bool *outIsPatched) = nullptr;
// Deprecated functions
static bool (*sFunctionPatcherPatchFunction)(function_replacement_data_t *function_data, PatchedFunctionHandle *outHandle) = nullptr;
static bool (*sFunctionPatcherRestoreFunction)(PatchedFunctionHandle handle) = nullptr;
static FunctionPatcherAPIVersion sFunctionPatcherVersion = FUNCTION_PATCHER_MODULE_API_VERSION_ERROR;
const char *FunctionPatcher_GetStatusStr(FunctionPatcherStatus status) {
switch (status) {
case FUNCTION_PATCHER_RESULT_SUCCESS:
return "FUNCTION_PATCHER_RESULT_SUCCESS";
case FUNCTION_PATCHER_RESULT_MODULE_NOT_FOUND:
return "FUNCTION_PATCHER_RESULT_MODULE_NOT_FOUND";
case FUNCTION_PATCHER_RESULT_MODULE_MISSING_EXPORT:
return "FUNCTION_PATCHER_RESULT_MODULE_MISSING_EXPORT";
case FUNCTION_PATCHER_RESULT_UNSUPPORTED_VERSION:
return "FUNCTION_PATCHER_RESULT_UNSUPPORTED_VERSION";
case FUNCTION_PATCHER_RESULT_INVALID_ARGUMENT:
return "FUNCTION_PATCHER_RESULT_INVALID_ARGUMENT";
case FUNCTION_PATCHER_RESULT_LIB_UNINITIALIZED:
return "FUNCTION_PATCHER_RESULT_LIB_UNINITIALIZED";
case FUNCTION_PATCHER_RESULT_UNSUPPORTED_COMMAND:
return "FUNCTION_PATCHER_RESULT_UNSUPPORTED_COMMAND";
case FUNCTION_PATCHER_RESULT_UNKNOWN_ERROR:
return "FUNCTION_PATCHER_RESULT_UNKNOWN_ERROR";
case FUNCTION_PATCHER_RESULT_PATCH_NOT_FOUND:
return "FUNCTION_PATCHER_RESULT_PATCH_NOT_FOUND";
case FUNCTION_PATCHER_RESULT_UNSUPPORTED_STRUCT_VERSION:
return "FUNCTION_PATCHER_RESULT_UNSUPPORTED_STRUCT_VERSION";
}
return "FUNCTION_PATCHER_RESULT_UNKNOWN_ERROR";
}
FunctionPatcherStatus FunctionPatcher_InitLibrary() {
if (OSDynLoad_Acquire("homebrew_functionpatcher", &sModuleHandle) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_ERR("OSDynLoad_Acquire failed.");
return FUNCTION_PATCHER_RESULT_MODULE_NOT_FOUND;
}
if (OSDynLoad_FindExport(sModuleHandle, FALSE, "FPGetVersion", (void **) &sFPGetVersion) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_ERR("FindExport FPGetVersion failed.");
return FUNCTION_PATCHER_RESULT_MODULE_MISSING_EXPORT;
}
auto res = FunctionPatcher_GetVersion(&sFunctionPatcherVersion);
if (res != FUNCTION_PATCHER_RESULT_SUCCESS) {
sFunctionPatcherVersion = FUNCTION_PATCHER_MODULE_API_VERSION_ERROR;
return FUNCTION_PATCHER_RESULT_UNSUPPORTED_VERSION;
}
if (OSDynLoad_FindExport(sModuleHandle, FALSE, "FPAddFunctionPatch", (void **) &sFPAddFunctionPatch) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_ERR("FindExport FPAddFunctionPatch failed.");
sFPAddFunctionPatch = nullptr;
}
if (OSDynLoad_FindExport(sModuleHandle, FALSE, "FPRemoveFunctionPatch", (void **) &sFPRemoveFunctionPatch) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_ERR("FindExport FPRemoveFunctionPatch failed.");
sFPRemoveFunctionPatch = nullptr;
}
if (OSDynLoad_FindExport(sModuleHandle, FALSE, "FPIsFunctionPatched", (void **) &sFPIsFunctionPatched) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_ERR("FindExport FPIsFunctionPatched failed.");
sFPIsFunctionPatched = nullptr;
}
return FUNCTION_PATCHER_RESULT_SUCCESS;
}
FunctionPatcherStatus FunctionPatcher_DeInitLibrary() {
sFPAddFunctionPatch = nullptr;
sFPRemoveFunctionPatch = nullptr;
sFPIsFunctionPatched = nullptr;
sFunctionPatcherVersion = FUNCTION_PATCHER_MODULE_API_VERSION_ERROR;
OSDynLoad_Release(sModuleHandle);
sModuleHandle = nullptr;
return FUNCTION_PATCHER_RESULT_SUCCESS;
}
bool FunctionPatcherPatchFunctionDecl(function_replacement_data_t *, PatchedFunctionHandle *);
bool FunctionPatcherPatchFunction(function_replacement_data_t *function_data, PatchedFunctionHandle *outHandle) {
if (sFunctionPatcherPatchFunction == nullptr) {
if (OSDynLoad_Acquire("homebrew_functionpatcher", &sModuleHandle) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_WARN("OSDynLoad_Acquire failed.");
return false;
}
if (OSDynLoad_FindExport(sModuleHandle, FALSE, "FunctionPatcherPatchFunction", (void **) &sFunctionPatcherPatchFunction) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_WARN("FindExport FunctionPatcherPatchFunction failed.");
return false;
}
}
return reinterpret_cast<decltype(&FunctionPatcherPatchFunctionDecl)>(sFunctionPatcherPatchFunction)(function_data, outHandle);
}
bool FunctionPatcherRestoreFunctionDecl(PatchedFunctionHandle);
bool FunctionPatcherRestoreFunction(PatchedFunctionHandle handle) {
if (sFunctionPatcherRestoreFunction == nullptr) {
if (OSDynLoad_Acquire("homebrew_functionpatcher", &sModuleHandle) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_WARN("OSDynLoad_Acquire failed.");
return false;
}
if (OSDynLoad_FindExport(sModuleHandle, FALSE, "FunctionPatcherRestoreFunction", (void **) &sFunctionPatcherRestoreFunction) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_WARN("FindExport FunctionPatcherRestoreFunction failed.");
return false;
}
}
return reinterpret_cast<decltype(&FunctionPatcherRestoreFunctionDecl)>(sFunctionPatcherRestoreFunction)(handle);
}
FunctionPatcherStatus FunctionPatcher_GetVersion(FunctionPatcherAPIVersion *outVersion) {
if (sFPGetVersion == nullptr) {
if (OSDynLoad_Acquire("homebrew_functionpatcher", &sModuleHandle) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_WARN("OSDynLoad_Acquire failed.");
return FUNCTION_PATCHER_RESULT_MODULE_NOT_FOUND;
}
if (OSDynLoad_FindExport(sModuleHandle, FALSE, "FPGetVersion", (void **) &sFPGetVersion) != OS_DYNLOAD_OK) {
DEBUG_FUNCTION_LINE_WARN("FindExport FPGetVersion failed.");
return FUNCTION_PATCHER_RESULT_MODULE_MISSING_EXPORT;
}
}
return reinterpret_cast<decltype(&FunctionPatcher_GetVersion)>(sFPGetVersion)(outVersion);
}
FunctionPatcherStatus FunctionPatcher_AddFunctionPatch(function_replacement_data_t *function_data, PatchedFunctionHandle *outHandle, bool *outHasBeenPatched) {
if (sFunctionPatcherVersion == FUNCTION_PATCHER_MODULE_API_VERSION_ERROR) {
return FUNCTION_PATCHER_RESULT_LIB_UNINITIALIZED;
}
if (sFPAddFunctionPatch == nullptr || sFunctionPatcherVersion < 2) {
return FUNCTION_PATCHER_RESULT_UNSUPPORTED_COMMAND;
}
if (function_data == nullptr) {
return FUNCTION_PATCHER_RESULT_INVALID_ARGUMENT;
}
return reinterpret_cast<decltype(&FunctionPatcher_AddFunctionPatch)>(sFPAddFunctionPatch)(function_data, outHandle, outHasBeenPatched);
}
FunctionPatcherStatus FunctionPatcher_RemoveFunctionPatch(PatchedFunctionHandle handle) {
if (sFunctionPatcherVersion == FUNCTION_PATCHER_MODULE_API_VERSION_ERROR) {
return FUNCTION_PATCHER_RESULT_LIB_UNINITIALIZED;
}
if (sFPRemoveFunctionPatch == nullptr || sFunctionPatcherVersion < 2) {
return FUNCTION_PATCHER_RESULT_UNSUPPORTED_COMMAND;
}
if (handle == 0) {
return FUNCTION_PATCHER_RESULT_INVALID_ARGUMENT;
}
return reinterpret_cast<decltype(&FunctionPatcher_RemoveFunctionPatch)>(sFPRemoveFunctionPatch)(handle);
}
FunctionPatcherStatus FunctionPatcher_IsFunctionPatched(PatchedFunctionHandle handle, bool *outIsFunctionPatched) {
if (sFunctionPatcherVersion == FUNCTION_PATCHER_MODULE_API_VERSION_ERROR) {
return FUNCTION_PATCHER_RESULT_LIB_UNINITIALIZED;
}
if (sFPIsFunctionPatched == nullptr || sFunctionPatcherVersion < 2) {
return FUNCTION_PATCHER_RESULT_UNSUPPORTED_COMMAND;
}
if (handle == 0 || outIsFunctionPatched == nullptr) {
return FUNCTION_PATCHER_RESULT_INVALID_ARGUMENT;
}
return reinterpret_cast<decltype(&FunctionPatcher_IsFunctionPatched)>(sFPIsFunctionPatched)(handle, outIsFunctionPatched);
}