WiiUPluginLoaderBackend/source/plugin/FunctionData.h

115 lines
3.8 KiB
C
Raw Normal View History

2018-09-22 19:52:52 +02:00
/****************************************************************************
* Copyright (C) 2018-2020 Maschell
2018-09-22 19:52:52 +02:00
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
****************************************************************************/
#pragma once
2018-09-22 19:52:52 +02:00
#include "utils/logger.h"
#include <function_patcher/fpatching_defines.h>
#include <function_patcher/function_patching.h>
2022-02-04 16:25:44 +01:00
#include <string>
2018-09-22 19:52:52 +02:00
class FunctionData {
public:
FunctionData(void *paddress, void *vaddress, std::string name, function_replacement_library_type_t library, void *replaceAddr, void *replaceCall,
2021-09-25 14:26:18 +02:00
FunctionPatcherTargetProcess targetProcess) {
2022-02-04 16:25:44 +01:00
this->paddress = paddress;
this->vaddress = vaddress;
this->name = std::move(name);
2022-02-04 16:25:44 +01:00
this->library = library;
this->targetProcess = targetProcess;
2022-02-04 16:25:44 +01:00
this->replaceAddr = replaceAddr;
this->replaceCall = replaceCall;
2018-09-22 19:52:52 +02:00
}
2020-12-26 14:17:50 +01:00
~FunctionData() = default;
2018-09-22 19:52:52 +02:00
2020-12-26 14:17:50 +01:00
[[nodiscard]] const std::string &getName() const {
2018-09-22 19:52:52 +02:00
return this->name;
}
2020-12-26 14:17:50 +01:00
[[nodiscard]] function_replacement_library_type_t getLibrary() const {
2018-09-22 19:52:52 +02:00
return this->library;
}
2020-12-26 14:17:50 +01:00
[[nodiscard]] const void *getPhysicalAddress() const {
return paddress;
}
2020-05-03 12:30:15 +02:00
2020-12-26 14:17:50 +01:00
[[nodiscard]] const void *getVirtualAddress() const {
return vaddress;
}
2020-12-26 14:17:50 +01:00
[[nodiscard]] const void *getReplaceAddress() const {
2018-09-22 19:52:52 +02:00
return replaceAddr;
}
2020-12-26 14:17:50 +01:00
[[nodiscard]] const void *getReplaceCall() const {
2018-09-22 19:52:52 +02:00
return replaceCall;
}
2020-12-26 14:17:50 +01:00
[[nodiscard]] FunctionPatcherTargetProcess getTargetProcess() const {
return targetProcess;
}
bool patch() {
if (handle == 0) {
function_replacement_data_t functionData = {
.VERSION = FUNCTION_REPLACEMENT_DATA_STRUCT_VERSION,
.physicalAddr = reinterpret_cast<uint32_t>(this->paddress),
.virtualAddr = reinterpret_cast<uint32_t>(this->vaddress),
.replaceAddr = reinterpret_cast<uint32_t>(this->replaceAddr),
.replaceCall = static_cast<uint32_t *>(this->replaceCall),
.library = this->library,
.function_name = this->name.c_str(),
.targetProcess = this->targetProcess};
if (!FunctionPatcherPatchFunction(&functionData, &handle)) {
DEBUG_FUNCTION_LINE_ERR("Failed to patch function");
return false;
}
} else {
DEBUG_FUNCTION_LINE("Function is already patched");
}
return true;
}
bool restore() {
if (handle != 0) {
if (!FunctionPatcherRestoreFunction(handle)) {
DEBUG_FUNCTION_LINE_ERR("Failed to restore function patch");
return false;
}
handle = 0;
} else {
DEBUG_FUNCTION_LINE("Was not patched.");
}
return true;
}
2018-09-22 19:52:52 +02:00
private:
2020-12-26 14:17:50 +01:00
void *paddress = nullptr;
void *vaddress = nullptr;
2018-09-22 19:52:52 +02:00
std::string name;
function_replacement_library_type_t library;
FunctionPatcherTargetProcess targetProcess;
2020-12-26 14:17:50 +01:00
void *replaceAddr = nullptr;
void *replaceCall = nullptr;
PatchedFunctionHandle handle = 0;
2018-09-22 19:52:52 +02:00
};