Files
ReshadePluginsCore/TLOU2/dllmain.cpp

245 lines
8.2 KiB
C++

#include "Memory.hpp";
#include "Maths.hpp";
#include "ObfuscateString.h"
#include <Windows.h>
#include <string>
#include <spdlog/spdlog.h>
#include <spdlog/sinks/basic_file_sink.h>
#include <safetyhook.hpp>
// Constants
const std::string PLUGIN_NAME = "The Last Of Us : Part II";
const std::string PLUGIN_LOG = "TLOU2.log";
const std::string gameExecutable = "tlou-ii.exe";
// Logger
std::shared_ptr<spdlog::logger> logger;
// Screen informations
static int screenWidth = GetSystemMetrics(SM_CXSCREEN);
static int screenHeight = GetSystemMetrics(SM_CYSCREEN);
static float aspectRatio = (float)screenWidth / screenHeight;
// Plugin states
static bool AOBScanDone = false;
static bool g_fix_enabled = false;
static bool g_fov_fix_enabled = false;
static bool g_camera_fix_enabled = false;
static bool g_sharpness_fix_enabled = false;
static int g_AdditionalFOVValue = 0;
static float g_CameraValue = 0;
static float g_SharpnessValue = 0;
// Shared values
static float g_FOV_In = 0;
static float g_FOV_Out = 0;
// AOB Scan pointers
static uint8_t* FOVAddress = nullptr;
static uint8_t* CameraAddress = nullptr;
static uint8_t* SharpnessAddress = nullptr;
// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid CameraHook{};
static SafetyHookMid SharpnessHook{};
// Prototypes
static void FOVFixEnabled(bool fix_enabled);
static void CameraFixEnabled(bool fix_enabled);
static void SharpnessFixEnabled(bool fix_enabled);
extern "C" __declspec(dllexport) void SetFixEnabled(bool enabled)
{
g_fix_enabled = enabled;
if (g_fix_enabled && !AOBScanDone) {
logger->info("--------------- AOB scanning started ---------------");
// === AOB Scans ===
if (!FOVAddress) {
constexpr auto FOVPattern = make_obfuscated<0x4A>("E8 ?? ?? ?? ?? C5 FA ?? ?? ?? ?? ?? ?? C5 F8 ?? ?? E8 ?? ?? ?? ?? C5 7A");
FOVAddress = Memory::aob_scan(gameExecutable, FOVPattern.decrypt(), PAGE_EXECUTE_READ);
if (FOVAddress)
{
logger->info("FOV signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(FOVAddress));
FOVAddress += 0x5;
}
else logger->warn("FOV signature not found. Maybe the game was updated.");
}
if (!CameraAddress) {
constexpr auto CameraPattern = make_obfuscated<0x4A>("C5 FA ?? ?? ?? ?? ?? ?? C5 FA ?? ?? ?? C5 22 ?? ?? ?? C4 C1 ?? ?? ?? C5 FA ?? ?? ?? C5 7A");
CameraAddress = Memory::aob_scan(gameExecutable, CameraPattern.decrypt(), PAGE_EXECUTE_READ);
if (CameraAddress)
{
logger->info("Camera signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(CameraAddress));
CameraAddress += 0x8;
}
else logger->warn("Camera signature not found. Maybe the game was updated.");
}
if (!SharpnessAddress) {
constexpr auto SharpnessPattern = make_obfuscated<0x4A>("C5 FA ?? ?? ?? ?? ?? ?? 8B 86 ?? ?? ?? ?? 89 87 ?? ?? ?? ?? 0F B6 ?? ?? ?? ?? ?? 89");
SharpnessAddress = Memory::aob_scan(gameExecutable, SharpnessPattern.decrypt(), PAGE_EXECUTE_READ);
if (SharpnessAddress)
{
logger->info("Sharpness signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(SharpnessAddress));
}
else logger->warn("Sharpness signature not found. Maybe the game was updated.");
if (FOVAddress && CameraAddress && SharpnessAddress) {
logger->info("All AOB signatures found. Ready to patch...");
logger->info("--------------- AOB scanning finished ---------------");
AOBScanDone = true;
}
else
logger->warn("Some AOB signatures could not be found. Fixes may be partially unavailable.");
}
}
// === Activer ou désactiver les patchs ===
if (FOVAddress && CameraAddress && SharpnessAddress)
{
if (g_fix_enabled)
{
FOVFixEnabled(g_fov_fix_enabled);
CameraFixEnabled(g_camera_fix_enabled);
SharpnessFixEnabled(g_sharpness_fix_enabled);
}
else
{
FOVFixEnabled(false);
CameraFixEnabled(false);
SharpnessFixEnabled(false);
logger->info("All fixes disabled.");
}
}
}
// Setters for Reshade addon call
extern "C" __declspec(dllexport) void SetFOVFixEnabled(bool enabled, bool init)
{
g_fov_fix_enabled = enabled;
if (!init) FOVFixEnabled(g_fov_fix_enabled);
}
extern "C" __declspec(dllexport) void SetCameraFixEnabled(bool enabled, bool init)
{
g_camera_fix_enabled = enabled;
if (!init) CameraFixEnabled(g_camera_fix_enabled);
}
extern "C" __declspec(dllexport) void SetSharpnessFixEnabled(bool enabled, bool init)
{
g_sharpness_fix_enabled = enabled;
if (!init) SharpnessFixEnabled(g_sharpness_fix_enabled);
}
extern "C" __declspec(dllexport) void SetFOV(int fov)
{
g_AdditionalFOVValue = fov;
}
extern "C" __declspec(dllexport) void SetCamera(float cameraDistance)
{
g_CameraValue = cameraDistance;
}
extern "C" __declspec(dllexport) void SetSharpness(float sharpness)
{
g_SharpnessValue = sharpness;
}
// Getters for Reshade addon call
extern "C" __declspec(dllexport) float GetFOVIn() {
return g_FOV_In;
}
extern "C" __declspec(dllexport) float GetFOVOut() {
return g_FOV_Out;
}
static void FOVFixEnabled(bool fix_enabled) {
if (g_fix_enabled && fix_enabled && FOVAddress != nullptr) {
if (!FOVHook) { // Hook only once
FOVHook = safetyhook::create_mid(FOVAddress,
[](SafetyHookContext& ctx) {
if (ctx.xmm0.f32[0] == ctx.xmm15.f32[0]) {
g_FOV_In = ctx.xmm0.f32[0];
g_FOV_Out = ctx.xmm0.f32[0] += (g_fov_fix_enabled ? g_AdditionalFOVValue : 0); // World FOV
}
});
}
else FOVHook.enable();
logger->info("FOV fix enabled");
}
if (!fix_enabled && FOVHook) {
FOVHook.disable();
logger->info("FOV fix disabled");
}
}
static void CameraFixEnabled(bool fix_enabled) {
if (g_fix_enabled && fix_enabled && CameraAddress != nullptr) {
if (!CameraHook) {
CameraHook = safetyhook::create_mid(CameraAddress,
[](SafetyHookContext& ctx) {
ctx.xmm0.f32[0] = (g_camera_fix_enabled ? g_CameraValue : ctx.xmm0.f32[0]);
});
}
else CameraHook.enable();
logger->info("Camera fix enabled");
}
if (!fix_enabled && CameraHook) {
CameraHook.disable();
logger->info("Camera fix disabled");
}
}
static void SharpnessFixEnabled(bool fix_enabled) {
if (g_fix_enabled && fix_enabled && SharpnessAddress != nullptr) {
if (!SharpnessHook) {
SharpnessHook = safetyhook::create_mid(SharpnessAddress,
[](SafetyHookContext& ctx) {
ctx.xmm0.f32[0] = g_SharpnessValue;
});
}
else SharpnessHook.enable();
logger->info("Sharpness fix enabled");
}
if (!fix_enabled && SharpnessHook) {
SharpnessHook.disable();
logger->info("sharpness fix disabled");
}
}
// Initialisation de spdlog avec format personnalisé
static void InitializeLogger()
{
try
{
logger = spdlog::basic_logger_mt("Fixlib", PLUGIN_LOG, true);
spdlog::set_default_logger(logger);
// Format : [YYYY-MM-DD HH:MM:SS] [INFO] message
spdlog::set_pattern("[%Y-%m-%d %H:%M:%S] [%^%l%$] %v");
spdlog::set_level(spdlog::level::debug);
logger->flush_on(spdlog::level::debug); // Flush automatically
}
catch (const spdlog::spdlog_ex& ex)
{
std::string plugin_error_message = "Could not open " + PLUGIN_LOG;
MessageBoxA(nullptr, plugin_error_message.c_str(), "Logger Error", MB_ICONERROR | MB_OK);
}
}
HMODULE g_coreModule = nullptr;
// Entrée standard DLL
BOOL APIENTRY DllMain(HMODULE hModule, DWORD reason, LPVOID)
{
if (reason == DLL_PROCESS_ATTACH)
{
InitializeLogger();
logger->info("Plugin {} loaded.", PLUGIN_NAME);
}
else if (reason == DLL_PROCESS_DETACH)
{
logger->info("Plugin {} unloaded.", PLUGIN_NAME);
spdlog::drop_all();
}
return TRUE;
}