diff --git a/OfAshAndSteel/dllmain.cpp b/OfAshAndSteel/dllmain.cpp new file mode 100644 index 0000000..0b20475 --- /dev/null +++ b/OfAshAndSteel/dllmain.cpp @@ -0,0 +1,454 @@ +#include "CommonHeaders.h" +#include "GameFixes.h" +#include "GameInformations.h" +#include "ObfuscateString.h" +#include "Memory.hpp" +#include "Maths.hpp" +#include "UEngine.hpp" +#include "SDK/Basic.hpp" +#include "SDK/Engine_classes.hpp" + +using namespace SDK; + +// Constants +const std::string PLUGIN_NAME = "OfAshAndSteel"; +const std::string PLUGIN_LOG = PLUGIN_NAME + ".log"; + +// Logger +std::shared_ptr logger; + +// Plugin states +static bool AOBScanDone = false; +static bool g_fix_enabled = false; +static bool g_fov_fix_enabled = false; +static bool g_ultrawide_fix_enabled = false; +static bool g_camera_fix_enabled = false; +static bool g_DOF_fix_enabled = false; +static bool g_CA_fix_enabled = false; +static bool g_Vignetting_fix_enabled = false; +static bool g_Fog_fix_enabled = false; +static bool g_VolumetricFog_fix_enabled = false; +static bool g_Console = false; +static int g_AdditionalFOVValue = 0; +static float g_CameraMultiplier = 1.f; + +// Shared values +static float g_FOV_In = 60.f; +static float g_FOV_Out = 60.f; +static float g_Camera_In = 300.f; +static float g_Camera_Out = 300.f; +static bool g_Console_Enabled = false; + +// AOB Unreal Engine offsets addresses +static uint8_t* GObjectsaddress = nullptr; +static uint8_t* AppendStringaddress = nullptr; +static uint8_t* ProcessEventaddress = nullptr; + +// AOB Scan pointers +static uint8_t* FOVaddress = nullptr; +static uint8_t* Cameraaddress = nullptr; +static uint8_t* DOFaddress = nullptr; +static uint8_t* CAaddress = nullptr; +static uint8_t* Vignettingaddress = nullptr; +static uint8_t* VolumetricFogaddress = nullptr; +static uint8_t* Fogaddress = nullptr; +static uint8_t* CameraComponentaddress = nullptr; +static uint8_t* ConstrainAspectRatioaddress = nullptr; +static uint8_t* AspectRatioAxisConstraintaddress = nullptr; + +// Hooking +static SafetyHookMid FOVHook{}; +static SafetyHookMid FogHook{}; +static SafetyHookMid CameraHook{}; + +// Prototypes +static void FOVFixEnabled(); +static void UltraWideFixEnabled(); +static void CameraFixEnabled(); +static void DOFFixEnabled(); +static void CAFixEnabled(); +static void VignettingFixEnabled(); +static void FogFixEnabled(); +static void VolumetricFogFixEnabled(); +static void EnableConsole(); + +extern "C" __declspec(dllexport) void SetFixEnabled(bool enabled, bool init) +{ + g_fix_enabled = enabled; + if (g_fix_enabled && !AOBScanDone) { + logger->info("--------------- AOB scan started ---------------"); + if (CameraComponentaddress == nullptr) { + constexpr auto FOVStringObfuscated = make_obfuscated<0x4A>("EB ?? F3 0F ?? ?? ?? ?? ?? ?? F3 0F ?? ?? ?? 8B 83"); + CameraComponentaddress = Memory::AOBScan("", FOVStringObfuscated.decrypt(), PAGE_EXECUTE_READ, logger); + + if (!CameraComponentaddress) + logger->warn("Camera component signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else { + logger->info("Camera component found at address: 0x{:X}.", reinterpret_cast(CameraComponentaddress)); + FOVaddress = CameraComponentaddress +0xa; + ConstrainAspectRatioaddress = CameraComponentaddress +0x18; + } + } + + if (!AspectRatioAxisConstraintaddress) { + constexpr auto AspectRatioAxisConstraintStringObfuscated = make_obfuscated<0x4A>("41 0F ?? ?? ?? ?? ?? ?? 48 8D ?? ?? ?? ?? ?? 4C ?? ?? 4D ?? ?? E8"); + AspectRatioAxisConstraintaddress = Memory::AOBScan("", AspectRatioAxisConstraintStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!AspectRatioAxisConstraintaddress) + logger->warn("HOR+ signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else + logger->info("HOR+ found at address: 0x{:X}.", reinterpret_cast(AspectRatioAxisConstraintaddress)); + } + + if (!Cameraaddress) { + constexpr auto CameraStringObfuscated = make_obfuscated<0x4A>("F3 0F ?? ?? ?? ?? ?? ?? F2 0F ?? ?? ?? ?? ?? ?? 89 8B"); + Cameraaddress = Memory::AOBScan("", CameraStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!Cameraaddress) + logger->warn("Camera not found. Maybe your game has been updated and is no more compatible with this plugin."); + else + logger->info("Camera found at address: 0x{:X}.", reinterpret_cast(Cameraaddress)); + } + + if (!DOFaddress) { + constexpr auto DOFStringObfuscated = make_obfuscated<0x4A>("8B ?? ?? E8 ?? ?? ?? ?? 8B ?? E8 ?? ?? ?? ?? 84 ?? 74 ?? 48"); + DOFaddress = Memory::AOBScan("", DOFStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!DOFaddress) + logger->warn("DOF signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else + logger->info("DOF signature found at address: 0x{:X}.", reinterpret_cast(DOFaddress)); + } + + if (!CAaddress) { + constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("7F ?? 89 B3 ?? ?? ?? ?? 8B ?? ?? 39 05 ?? ?? ?? ?? 0F"); + CAaddress = Memory::AOBScan("", CAStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!CAaddress) + logger->warn("Chromatic aberrations signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else + logger->info("Chromatic aberrations signature found at address: 0x{:X}.", reinterpret_cast(CAaddress)); + } + + if (!Vignettingaddress) { + constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("8B ?? 83 ?? ?? 7D ?? 89 B3 ?? ?? ?? ?? 83 F9"); + Vignettingaddress = Memory::AOBScan("", CAStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!Vignettingaddress) + logger->warn("Vignetting signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else + logger->info("Vignetting signature found at address: 0x{:X}.", reinterpret_cast(Vignettingaddress)); + } + + if (!Fogaddress) { + constexpr auto FogStringObfuscated = make_obfuscated<0x4A>("83 ?? ?? ?? 75 ?? B3 ?? EB ?? 32 ?? 48 8B ?? ?? ?? 48 ?? ?? 74 ?? E8 ?? ?? ?? ?? 84"); + Fogaddress = Memory::AOBScan("", FogStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!Fogaddress) + logger->warn("Fog signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else + logger->info("Fog signature found at address: 0x{:X}.", reinterpret_cast(Fogaddress)); + } + + if (!VolumetricFogaddress){ + constexpr auto VolumetricFogStringObfuscated = make_obfuscated<0x4A>("74 ?? F7 47 ?? ?? ?? ?? ?? 74 ?? 83 BB"); + VolumetricFogaddress = Memory::AOBScan("", VolumetricFogStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!VolumetricFogaddress) + logger->warn("Volumetric fog signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else + logger->info("Volumetric fog signature found at address: 0x{:X}.", reinterpret_cast(VolumetricFogaddress)); + } + + if (CameraComponentaddress && FOVaddress && Cameraaddress && DOFaddress && CAaddress && Vignettingaddress && + Fogaddress && VolumetricFogaddress && ConstrainAspectRatioaddress && AspectRatioAxisConstraintaddress) { + logger->info("All AOB signatures found. Ready to patch..."); + AOBScanDone = true; + } + + if (!GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) { + logger->info("------------ UEngine offsets search ------------"); + uint8_t* baseModule = reinterpret_cast(GetModuleHandleA(nullptr)); // Get game base address + + constexpr auto GObjetcsStringObfuscated = make_obfuscated<0x4A>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33"); + GObjectsaddress = Memory::AOBScan("", GObjetcsStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + constexpr auto GAppendStringStringObfuscated = make_obfuscated<0x4A>("48 89 ?? ?? ?? 48 89 ?? ?? ?? 57 48 ?? ?? ?? 8B ?? 48 ?? ?? 8B ?? 44 ?? ?? ?? C1 ?? ?? 48 ?? ?? 80 3D ?? ?? ?? ?? ?? 89 ?? ?? ?? 44 89 ?? ?? ?? 74 ?? 4C 8D ?? ?? ?? ?? ?? EB ?? 48 8D ?? ?? ?? ?? ?? E8 ?? ?? ?? ?? 4C ?? ?? C6 05 ?? ?? ?? ?? ?? 48 8B ?? ?? ?? 48 ?? ?? 48 ?? ?? ?? 8D ?? ?? 49 03 ?? ?? ?? E8 ?? ?? ?? ?? 83"); + AppendStringaddress = Memory::AOBScan("", GAppendStringStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + constexpr auto ProcessEventStringObfuscated = make_obfuscated<0x4A>("40 ?? 56 57 41 ?? 41 ?? 41 ?? 41 ?? 48 81 ?? ?? ?? ?? ?? 48 8D ?? ?? ?? 48 89 ?? ?? ?? ?? ?? 48 8B ?? ?? ?? ?? ?? 48 33 ?? 48 89 ?? ?? ?? ?? ?? 8B 41"); + ProcessEventaddress = Memory::AOBScan("", ProcessEventStringObfuscated.decrypt(), PAGE_EXECUTE_READ); + + if (!GObjectsaddress) + logger->warn("GObjects signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else { + uint32_t gObjectsOffset = static_cast(Memory::GetOffsetFromOpcode(GObjectsaddress + 0x3) - baseModule); + logger->info("GObjects offset is: 0x{:X}.", gObjectsOffset); + Offsets::GObjects = static_cast(gObjectsOffset); // Update GObjects offset + } + + if (!AppendStringaddress) + logger->warn("AppendString Event signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else { + std::optional gAppendStringOffsetOpt = UE::CalculateOffset("", AppendStringaddress); + uint32_t gAppendStringOffset = *gAppendStringOffsetOpt; + logger->info("AppendString offset is: 0x{:X}.", gAppendStringOffset); + Offsets::AppendString = static_cast(gAppendStringOffset);// Update AppendString offset + } + + if (!ProcessEventaddress) + logger->warn("Process Event signature not found. Maybe your game has been updated and is no more compatible with this plugin."); + else { + std::optional gProcessEventOffsetOpt = UE::CalculateOffset("", ProcessEventaddress); + uint32_t gProcessEventOffset = *gProcessEventOffsetOpt; + logger->info("Process Event offset is: 0x{:X}.", gProcessEventOffset); + Offsets::ProcessEvent = static_cast(gProcessEventOffset);// Update ProcessEvent offset + } + } + + logger->info("-------------- Fixes initialisation -------------"); + } + + if (!init && FOVaddress) FOVFixEnabled(); + if (!init && Cameraaddress) CameraFixEnabled(); + if (!init && DOFaddress) DOFFixEnabled(); + if (!init && CAaddress) CAFixEnabled(); + if (!init && Vignettingaddress) VignettingFixEnabled(); + if (!init && Fogaddress) FogFixEnabled(); + if (!init && VolumetricFogaddress) VolumetricFogFixEnabled(); + if (!init && AspectRatioAxisConstraintaddress && ConstrainAspectRatioaddress) UltraWideFixEnabled(); + if (!init) EnableConsole(); +} + +// Setters for Reshade addon call +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_ultrawide_fix_enabled = enabled; UltraWideFixEnabled(); } + if (fix == GameFixes::Camera) { g_camera_fix_enabled = enabled; CameraFixEnabled(); } + if (fix == GameFixes::DOF) { g_DOF_fix_enabled = enabled; DOFFixEnabled(); } + if (fix == GameFixes::ChromaticAberrations) { g_CA_fix_enabled = enabled; CAFixEnabled(); } + if (fix == GameFixes::Vignetting) { g_Vignetting_fix_enabled = enabled; VignettingFixEnabled(); } + if (fix == GameFixes::Fog) { g_Fog_fix_enabled = enabled; FogFixEnabled(); } + if (fix == GameFixes::VolumetricFog) { g_VolumetricFog_fix_enabled = enabled; VolumetricFogFixEnabled(); } +} + +extern "C" __declspec(dllexport) void SetFOV(int fov) +{ + g_AdditionalFOVValue = fov; +} + +extern "C" __declspec(dllexport) void SetCameraDistance(float cameraDistance) +{ + g_CameraMultiplier = cameraDistance; +} + +// Getters for Reshade addon call +extern "C" __declspec(dllexport) void GetGameInfos(GameInfos* infos) { + if (!infos) return; + + infos->FOVIn = g_FOV_In; + infos->FOVOut = g_FOV_Out; + infos->cameraIn = g_Camera_In; + infos->cameraOut = g_Camera_Out; + infos->consoleEnabled = g_Console_Enabled; +} + +// Code injection functions +static void FOVFixEnabled() { + if (g_fix_enabled && g_fov_fix_enabled && FOVaddress) { + if (!FOVHook) { // Hook only once + FOVHook = safetyhook::create_mid(FOVaddress, + [](SafetyHookContext& ctx) { + g_FOV_In = ctx.xmm0.f32[0]; + ctx.xmm0.f32[0] += (g_fix_enabled && g_fov_fix_enabled ? g_AdditionalFOVValue : g_FOV_In); + g_FOV_Out = ctx.xmm0.f32[0]; + }); + } + else FOVHook.enable(); + logger->info("FOV fix enabled"); + } + if (!(g_fix_enabled && g_fov_fix_enabled) && FOVaddress) { + if (FOVHook) FOVHook.disable(); + logger->info("FOV fix disabled"); + } +} + +static void CameraFixEnabled() { + if (g_fix_enabled && g_camera_fix_enabled && Cameraaddress) { + if (!CameraHook) { + CameraHook = safetyhook::create_mid(Cameraaddress + 0x8, + [](SafetyHookContext& ctx) { + g_Camera_In = ctx.xmm6.f32[0]; + ctx.xmm6.f32[0] *= g_CameraMultiplier; + g_Camera_Out = ctx.xmm6.f32[0]; + }); + } + else CameraHook.enable(); + logger->info("Camera distance fix enabled"); + } + if (!(g_fix_enabled && g_camera_fix_enabled) && CameraHook) { + if (CameraHook) CameraHook.disable(); + logger->info("Camera distance fix disabled"); + } +} + +static void FogFixEnabled() { + if (g_fix_enabled && g_Fog_fix_enabled && Fogaddress) { + if (!FogHook) { + FogHook = safetyhook::create_mid(Fogaddress, + [](SafetyHookContext& ctx) { + if (!ctx.rax) return; + uintptr_t* rax = reinterpret_cast(ctx.rax); + bool* bFog = reinterpret_cast(reinterpret_cast(rax) + 0x04); + *bFog = g_fix_enabled && g_Fog_fix_enabled ? false : true; + }); + } + logger->info("Fog fix enabled"); + } + if (!(g_fix_enabled && g_Fog_fix_enabled) && Fogaddress) + logger->info("Fog fix disabled"); +} + +// Memory patch fixes +static void UltraWideFixEnabled() { + if (g_fix_enabled && g_ultrawide_fix_enabled && AspectRatioAxisConstraintaddress && ConstrainAspectRatioaddress) { + Memory::PatchBytes(AspectRatioAxisConstraintaddress, "\x31\xD2\x90\x90\x90\x90\x90\x90", 8); // AspectRatio_MaintainYFOV = 0 + Memory::PatchBytes(ConstrainAspectRatioaddress, "\x31\xC0\x90\x90\x90\x90\x90", 7); // bConstrainAspectRatio = 0 + logger->info("Ultra wide fix enabled"); + } + if (!(g_fix_enabled && g_ultrawide_fix_enabled) && AspectRatioAxisConstraintaddress && ConstrainAspectRatioaddress) { + Memory::RestoreBytes(AspectRatioAxisConstraintaddress); + Memory::RestoreBytes(ConstrainAspectRatioaddress); + logger->info("Ultra wide fix disabled"); + } +} + +static void DOFFixEnabled() { + if (g_fix_enabled && g_DOF_fix_enabled && DOFaddress) { + Memory::PatchBytes(DOFaddress, "\x31\xFF\x90", 3); // xor edi,edi r.DepthOfFieldQuality = 0 + logger->info("Depth of field fix enabled"); + } + if (!(g_fix_enabled && g_DOF_fix_enabled) && DOFaddress) { + Memory::RestoreBytes(DOFaddress); + logger->info("Depth of field fix disabled"); + } +} + +static void CAFixEnabled() { + if (g_fix_enabled && g_CA_fix_enabled && CAaddress) { + Memory::PatchBytes(CAaddress, "\x90\x90", 2); // NOP x 2 r.SceneColorFringeQuality = 0 + logger->info("Chromatics aberrations fix enabled"); + } + if (!(g_fix_enabled && g_CA_fix_enabled) && CAaddress) { + Memory::RestoreBytes(CAaddress); + logger->info("Chromatics aberrations fix disabled"); + } +} + +static void VignettingFixEnabled() { + if (g_fix_enabled && g_Vignetting_fix_enabled && Vignettingaddress) { + Memory::PatchBytes(Vignettingaddress, "\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); + logger->info("Vignetting fix disabled"); + } +} + +static void VolumetricFogFixEnabled() { + if (g_fix_enabled && g_VolumetricFog_fix_enabled && VolumetricFogaddress) { + Memory::PatchBytes(VolumetricFogaddress, "\xEB", 1); // je-> jmp r.Fog=0 + logger->info("Volumetric fog fix enabled"); + } + if (!(g_fix_enabled && g_VolumetricFog_fix_enabled) && VolumetricFogaddress) { + Memory::RestoreBytes(VolumetricFogaddress); + logger->info("Volumetric fog fix disabled"); + } +} + +// UE Console creation +static void EnableConsole() +{ + if (g_Console_Enabled || !g_Console || !GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) { + if (!g_Console) + logger->info("------------------ User inputs ------------------"); + 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(NewObject); + auto end = std::chrono::high_resolution_clock::now(); + std::chrono::duration elapsed = end - start; + + for (int i = 0; i < UInputSettings::GetDefaultObj()->ConsoleKeys.Num(); i++) + { + UInputSettings::GetDefaultObj()->ConsoleKeys[i].KeyName = UKismetStringLibrary::Conv_StringToName(L"F2"); + } + logger->info("Console fully reactivated in {:.3f}s and bound to key F2", elapsed.count()); + logger->info("------------------ User inputs ------------------"); + + g_Console_Enabled = true; + } + else { + logger->error("Could not spawn console object"); + } + }).detach(); +} + +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("Of Ash & Steel", std::make_shared(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; +} \ No newline at end of file