Files
ReshadePluginsCore/Cronos The New Dawn/dllmain.cpp

356 lines
15 KiB
C++

#include "CommonHeaders.h"
#include "UEngine.hpp";
#include "UEvars.hpp"
#include "SDK/Engine_classes.hpp"
#include "SDK/GameBase_classes.hpp"
using namespace SDK;
// Constants
const std::string PLUGIN_NAME = "CronosTND";
const std::string PLUGIN_LOG = PLUGIN_NAME + ".log";
const float baseAspect = 1.7777778;
// 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_Console = false;
static bool g_fix_enabled = false;
static bool g_fov_fix_enabled = false;
static bool g_aspect_fix_enabled = false;
static bool g_DOF_fix_enabled = false;
static bool g_Vignetting_fix_enabled = false;
static bool g_Fog_fix_enabled = false;
static bool g_cam_distance_fix_enabled = false;
static bool g_SkipIntro_fix_enabled = false;
static int g_AdditionalFOVValue = 0;
static float g_cameraDistanceMultiplier = 1.f;
static bool user_inputs_logged = false;
// Shared values
static float g_FOV_In = 0;
static float g_Compensated_FOV = 0;
static float g_FOV_Out = 0;
static float g_Camera_In = 180;
static float g_Camera_Out = 180;
static bool g_Console_Enabled = false;
// AOB Scan pointers
static uint8_t* FOVaddress = nullptr;
static uint8_t* Aspectaddress = nullptr;
static uint8_t* DOFaddress = nullptr;
static uint8_t* Vignettingaddress = nullptr;
static uint8_t* Fogaddress = nullptr;
static uint8_t* CameraDistanceaddress = nullptr;
// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid AspectHook{};
static SafetyHookMid FogHook{};
static SafetyHookMid CameraDistanceHook{};
static SafetyHookMid PEHook{};
// Prototypes
static void FOVFixEnabled();
static void AspectFixEnabled();
static void DOFFixEnabled();
static void VignettingFixEnabled();
static void FogFixEnabled();
static void CameraDistanceFixEnabled();
static void EnableConsole();
static void ProcessEvent();
extern "C" __declspec(dllexport) void SetFixEnabled(bool enabled, bool init)
{
g_fix_enabled = enabled;
if (!AOBScanDone) { // Unreal Engine 5.5.4
logger->info("--------------- AOB scan started ---------------");
using AOBScan::Make;
using OffsetScan::Make;
constexpr auto FOVStringObfuscated = make_obfuscated<0x8B>("EB ?? F3 0F ?? ?? ?? ?? ?? ?? F3 0F ?? ?? ?? 8B 83 ?? ?? ?? ?? 89");
constexpr auto AspectStringObfuscated = make_obfuscated<0x39>("C7 43 ?? 39 8E E3 3F C7 87 ?? ?? ?? ?? 39 8E E3 3F 48 83 ?? ?? ?? ?? ?? ?? 75");
constexpr auto DOFStringObfuscated = make_obfuscated<0x0F>("74 ?? 48 ?? ?? 8B ?? ?? 48 ?? ?? E8 ?? ?? ?? ?? 0F");
constexpr auto VignettingStringObfuscated = make_obfuscated<0x7D>("8B ?? 83 ?? ?? 7D ?? 44 89");
constexpr auto FogStringObfuscated = make_obfuscated<0xB3>("F6 ?? ?? ?? ?? 74 ?? 48 8B ?? ?? ?? ?? ?? 83 78 ?? ?? 75 ?? B3");
constexpr auto CamDistanceStringObfuscated = make_obfuscated<0xF2>("E8 ?? ?? ?? ?? F2 0F ?? ?? ?? ?? 48 8D ?? ?? F3 0F ?? ?? ?? ?? ?? ?? 48 8D ?? ?? ?? ?? ?? 0F");
// Prepare all data for scanning
std::vector<AOBScanEntry> signatures = {
Make(&FOVaddress, FOVStringObfuscated, "FOV"),
Make(&Aspectaddress, AspectStringObfuscated, "Ultrawide"),
Make(&CameraDistanceaddress, CamDistanceStringObfuscated, "Camera distance"),
Make(&DOFaddress, DOFStringObfuscated, "DOF"),
Make(&Vignettingaddress, VignettingStringObfuscated, "Vignetting"),
Make(&Fogaddress, FogStringObfuscated, "Fog")
};
// Scan all signature in a batch
Memory::AOBScanBatch(signatures, logger);
if (FOVaddress && Aspectaddress && DOFaddress && Vignettingaddress && Fogaddress && CameraDistanceaddress)
logger->info("All AOB signatures found. Ready to patch...");
// Unreal Engine Offsets updates (needed when the game reveives an update and to prevent the fix from crashing the game)
if (!GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) {
logger->info("------------ UEngine offsets search ------------");
uint8_t* baseModule = reinterpret_cast<uint8_t*>(GetModuleHandleA(nullptr)); // Get game base address
constexpr auto GObjetcsStringObfuscated = make_obfuscated<0x33>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
constexpr auto AppendStringStringObfuscated = make_obfuscated<0x80>("48 89 ?? ?? ?? 48 89 ?? ?? ?? 57 48 83 ?? ?? 80 3D ?? ?? ?? ?? ?? 48 8B F2 8B");
constexpr auto ProcessEventStringObfuscated = make_obfuscated<0x56>("40 ?? 56 57 41 ?? 41 ?? 41 ?? 41 ?? 48 81 ?? ?? ?? ?? ?? 48 8D ?? ?? ?? 48 89 ?? ?? ?? ?? ?? 48 8B ?? ?? ?? ?? ?? 48 33 ?? 48 89 ?? ?? ?? ?? ?? 4D ?? ?? 48");
// Prepare all data for scanning
std::vector<OffsetScanEntry> UEoffsetsScans = {
Make(&GObjectsaddress, GObjetcsStringObfuscated, "GObjects", OffsetCalcType::GetOffsetFromOpcode, &Offsets::GObjects, 0x3),
Make(&AppendStringaddress, AppendStringStringObfuscated, "AppendString", OffsetCalcType::UE_CalculateOffset, &Offsets::AppendString),
Make(&ProcessEventaddress, ProcessEventStringObfuscated, "ProcessEvent", OffsetCalcType::UE_CalculateOffset, &Offsets::ProcessEvent)
};
// Retrieve all Unreal Engine offsets in a batch
Memory::OffsetScanBatch(UEoffsetsScans, baseModule, logger, "");
}
AOBScanDone = true;
logger->info("-------------- Fixes initialisation -------------");
}
if (!init && FOVaddress) FOVFixEnabled();
if (!init && Aspectaddress) AspectFixEnabled();
if (!init && DOFaddress) DOFFixEnabled();
if (!init && Vignettingaddress) VignettingFixEnabled();
if (!init && Fogaddress) FogFixEnabled();
if (!init && CameraDistanceaddress) CameraDistanceFixEnabled();
ProcessEvent();
}
// Setters for Reshade addon calls
extern "C" __declspec(dllexport) void SetFixesEnabled(GameFixes fix, bool enabled) { // Set each fix individually
if (fix == GameFixes::DevConsole) { g_Console = enabled; EnableConsole(); }
if (fix == GameFixes::FOV) { g_fov_fix_enabled = enabled; FOVFixEnabled(); }
if (fix == GameFixes::UltraWide) { g_aspect_fix_enabled = enabled; AspectFixEnabled(); }
if (fix == GameFixes::Camera) { g_cam_distance_fix_enabled = enabled; CameraDistanceFixEnabled(); }
if (fix == GameFixes::DOF) { g_DOF_fix_enabled = enabled; DOFFixEnabled(); }
if (fix == GameFixes::Vignetting) { g_Vignetting_fix_enabled = enabled; VignettingFixEnabled(); }
if (fix == GameFixes::Fog) { g_Fog_fix_enabled = enabled; FogFixEnabled(); }
if (fix == GameFixes::SkipIntro) { g_SkipIntro_fix_enabled = enabled; }
}
extern "C" __declspec(dllexport) void SetFOV(int fov) {
g_AdditionalFOVValue = fov;
}
extern "C" __declspec(dllexport) void SetCameraDistance(float cameraDistance) {
g_cameraDistanceMultiplier = cameraDistance;
}
// Getters for Reshade addon calls
extern "C" __declspec(dllexport) void GetGameInfos(GameInfos* infos) {
if (!infos) return;
infos->FOVIn = g_FOV_In;
infos->FOVOut = g_FOV_Out;
infos->CompensatedFOV = g_Compensated_FOV;
infos->cameraIn = g_Camera_In;
infos->cameraOut = g_Camera_Out;
infos->consoleEnabled = g_Console_Enabled;
}
static bool inContinue = false;
static void ProcessEvent() {
if (!PEHook && ProcessEventaddress) {
PEHook = safetyhook::create_mid(ProcessEventaddress + 0xc,
[](SafetyHookContext& ctx) {
UObject* object = (UObject*)ctx.rcx;
UFunction* func = (UFunction*)ctx.rdx;
if (object && func && g_DOF_fix_enabled) {
std::string funcName = func->GetName();
std::string objectName = object->GetName();
std::string objectFullName = object->GetFullName();
// Skip disclaimers
if (objectName.contains("StartGameMenuWidget")) {
auto* widget = static_cast<UStartGameWidget*>(object);
if (widget && widget->Class && widget->IsA(UStartGameWidget::StaticClass()) && funcName == "OnNext" && !inContinue) {
inContinue = true;
if (g_SkipIntro_fix_enabled) widget->Continue();
inContinue = false;
}
}
}
});
}
}
// Code injection functions
static void FOVFixEnabled() {
if (g_fix_enabled && (g_fov_fix_enabled || g_aspect_fix_enabled) && FOVaddress) {
if (!FOVHook) { // Hook only once
FOVHook = safetyhook::create_mid(FOVaddress + 0xa,
[](SafetyHookContext& ctx) {
g_FOV_In = ctx.xmm0.f32[0];
if (g_fix_enabled && g_aspect_fix_enabled)
ctx.xmm0.f32[0] = Maths::CompensateHorizontalFOV(g_FOV_In, baseAspect, aspectRatio);
g_Compensated_FOV = ctx.xmm0.f32[0];
g_FOV_Out = ctx.xmm0.f32[0] += (g_fov_fix_enabled ? g_AdditionalFOVValue : 0);
});
}
else FOVHook.enable();
logger->info("FOV fix enabled");
}
if (!(g_fix_enabled && (g_fov_fix_enabled || g_aspect_fix_enabled)) && FOVaddress) {
if (FOVHook) FOVHook.disable();
logger->info("FOV fix disabled");
}
}
// Memory patch fixes
static void AspectFixEnabled() {
if (g_fix_enabled && g_aspect_fix_enabled && Aspectaddress) {
const char* newAspectRatio = Memory::Float32ToHexBytes(aspectRatio); // Converts new apsect ratio in 4 char bytes
Memory::PatchBytes(Aspectaddress + 0x3, newAspectRatio, 4); // mov [rdi+00000254],3FE38E39 <= patched
logger->info("Aspect fix enabled");
}
if (!(g_fix_enabled && g_aspect_fix_enabled) && Aspectaddress) {
Memory::RestoreBytes(Aspectaddress + 0x3);
logger->info("Aspect ratio fix disabled");
}
}
static void DOFFixEnabled() {
if (g_fix_enabled && g_DOF_fix_enabled && DOFaddress) {
Memory::PatchBytes(DOFaddress + 0x5, "\x31\xF6\x90", 3); // xor esi,esi r.DepthOfFieldQuality = 0
logger->info("Depth of field fix enabled");
}
if (!(g_fix_enabled && g_DOF_fix_enabled) && DOFaddress) {
Memory::RestoreBytes(DOFaddress + 0x5);
logger->info("Depth of field fix disabled");
}
}
static void VignettingFixEnabled() {
if (g_fix_enabled && g_Vignetting_fix_enabled && Vignettingaddress) {
Memory::PatchBytes(Vignettingaddress + 0x5, "\x31\xC9", 2); // xor ecx,ecx r.Tonemapper.Quality=0
logger->info("Vignetting fix enabled");
}
if (!(g_fix_enabled && g_Vignetting_fix_enabled) && Vignettingaddress) {
Memory::RestoreBytes(Vignettingaddress + 0x5);
logger->info("Vignetting fix disabled");
}
}
static void FogFixEnabled() {
if (g_fix_enabled && g_Fog_fix_enabled && Fogaddress) {
if (!FogHook) { // Hook only once
FogHook = safetyhook::create_mid(Fogaddress + 0x12,
[](SafetyHookContext& ctx) {
ctx.rflags &= ~0x40; // ZF=0
});
}
logger->info("Fog fix enabled");
}
}
static void CameraDistanceFixEnabled() {
if (g_fix_enabled && g_cam_distance_fix_enabled && CameraDistanceaddress) {
if (!CameraDistanceHook) { // Hook only once
CameraDistanceHook = safetyhook::create_mid(CameraDistanceaddress + 0x17,
[](SafetyHookContext& ctx) {
g_Camera_In = ctx.xmm1.f32[0];
ctx.xmm1.f32[0] *= g_cameraDistanceMultiplier;
g_Camera_Out = ctx.xmm1.f32[0];
});
}
else CameraDistanceHook.enable();
logger->info("Camera distance fix enabled");
}
if (!(g_fix_enabled && g_cam_distance_fix_enabled) && CameraDistanceaddress) {
if (CameraDistanceHook) CameraDistanceHook.disable();
logger->info("Camera distance fix disabled");
}
}
// UE Console creation
static void EnableConsole() {
if (g_Console_Enabled || !g_Console || !GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) {
if (!g_Console && !user_inputs_logged) {
logger->info("------------------ User inputs ------------------");
user_inputs_logged = true;
}
return;
}
logger->info("-------------- Console re-enabling --------------");
std::thread([&]() {
auto start = std::chrono::high_resolution_clock::now(); // Measure the time to renable console
UEngine* Engine = nullptr;
for (int i = 0; i < 100; ++i) { // gives 10 seconds to find UE Engine
std::this_thread::sleep_for(std::chrono::milliseconds(100));
Engine = UEngine::GetEngine();
if (Engine && Engine->ConsoleClass && Engine->GameViewport)
break;
}
if (!Engine || !Engine->ConsoleClass || !Engine->GameViewport) {
logger->error("Console could not be found in engine.");
return;
}
logger->info("Console found in engine");
/* Creates a new UObject of class-type specified by Engine->ConsoleClass */
UObject* NewObject = UGameplayStatics::SpawnObject(Engine->ConsoleClass, Engine->GameViewport);
if (NewObject) {
logger->info("Successfully spawned console object");
// Set the console viewport so that it will be displayed
Engine->GameViewport->ViewportConsole = static_cast<UConsole*>(NewObject);
auto end = std::chrono::high_resolution_clock::now();
std::chrono::duration<double> elapsed = end - start;
logger->info("Console fully reactivated in {:.3f}s and bound to key Tilde", elapsed.count());
logger->info("------------------ User inputs ------------------");
g_Console_Enabled = true;
}
else {
logger->error("Could not spawn console object");
}
}).detach();
}
// Logging
static void InitializeLogger() {
try {
std::filesystem::path log_path = std::filesystem::absolute(PLUGIN_LOG);
if (std::filesystem::exists(log_path))
std::filesystem::remove(log_path);
logger = std::make_shared<spdlog::logger>("Cronos : The New Dawn", std::make_shared<spdlog::sinks::rotating_file_sink_st>(PLUGIN_LOG, 10 * 1024 * 1024, 1));
logger->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);
}
}
// Standard dll entry
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;
}