Add Vampire The Masquerade Bloodlines 2
This commit is contained in:
415
VampireTMBL2/dllmain.cpp
Normal file
415
VampireTMBL2/dllmain.cpp
Normal file
@@ -0,0 +1,415 @@
|
|||||||
|
#include <unordered_set>
|
||||||
|
#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"
|
||||||
|
#include "SDK/UMG_classes.hpp"
|
||||||
|
|
||||||
|
using namespace SDK;
|
||||||
|
|
||||||
|
// Constants
|
||||||
|
const std::string PLUGIN_NAME = "VampireTMBL2";
|
||||||
|
const std::string PLUGIN_LOG = PLUGIN_NAME + ".log";
|
||||||
|
|
||||||
|
// Logger
|
||||||
|
std::shared_ptr<spdlog::logger> 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 int g_AdditionalFOVValue = 0;
|
||||||
|
|
||||||
|
// Shared values
|
||||||
|
static float g_FOV_In = 60.f;
|
||||||
|
static float g_FOV_Out = 60.f;
|
||||||
|
static bool g_Console_Enabled = false;
|
||||||
|
|
||||||
|
// AOB Unreal Engine offsets addresses
|
||||||
|
static uint8_t* GObjectsaddress = nullptr;
|
||||||
|
static uint8_t* GNamesaddress = 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 CameraHook{};
|
||||||
|
static SafetyHookMid FogHook{};
|
||||||
|
static SafetyHookMid ProcessEventHook{};
|
||||||
|
|
||||||
|
// Prototypes
|
||||||
|
static void FOVFixEnabled();
|
||||||
|
static void UltraWideFixEnabled();
|
||||||
|
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) { // Unreal Engine 5.3.2
|
||||||
|
constexpr auto FOVStringObfuscated = make_obfuscated<0x4A>("EB ?? F3 0F ?? ?? ?? ?? ?? ?? F3 0F ?? ?? ?? 8B 83 ?? ?? ?? ?? 89");
|
||||||
|
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<uintptr_t>(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<uintptr_t>(AspectRatioAxisConstraintaddress));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!DOFaddress) {
|
||||||
|
constexpr auto DOFStringObfuscated = make_obfuscated<0x4A>("8B ?? ?? 48 ?? ?? E8 ?? ?? ?? ?? 0F ?? ?? 48 6B ?? ?? 48 8D");
|
||||||
|
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<uintptr_t>(DOFaddress));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!CAaddress) {
|
||||||
|
constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("7F ?? 89 B3 ?? ?? ?? ?? 8B ?? ?? 39 05 ?? ?? ?? ?? 0F 8F");
|
||||||
|
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<uintptr_t>(CAaddress));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!Vignettingaddress) {
|
||||||
|
constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("8B ?? 83 ?? ?? 7D ?? 89 B3 ?? ?? ?? ?? EB");
|
||||||
|
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<uintptr_t>(Vignettingaddress));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!Fogaddress) {
|
||||||
|
constexpr auto FogStringObfuscated = make_obfuscated<0x4A>("74 ?? 48 8B ?? ?? ?? ?? ?? 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<uintptr_t>(Fogaddress));
|
||||||
|
Fogaddress += 0x9;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!VolumetricFogaddress){
|
||||||
|
constexpr auto VolumetricFogStringObfuscated = make_obfuscated<0x4A>("84 ?? 0F 84 ?? ?? ?? ?? 48 ?? ?? 0F 84 ?? ?? ?? ?? 83 3D ?? ?? ?? ?? ?? 0F 84");
|
||||||
|
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<uintptr_t>(VolumetricFogaddress));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (CameraComponentaddress && FOVaddress && DOFaddress && CAaddress && Vignettingaddress &&
|
||||||
|
Fogaddress && VolumetricFogaddress && ConstrainAspectRatioaddress && AspectRatioAxisConstraintaddress) {
|
||||||
|
logger->info("All AOB signatures found. Ready to patch...");
|
||||||
|
AOBScanDone = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!GObjectsaddress || !GNamesaddress || !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<0x4A>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
|
||||||
|
GObjectsaddress = Memory::AOBScan("", GObjetcsStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
|
||||||
|
constexpr auto GNamesStringObfuscated = make_obfuscated<0x4A>("48 8D ?? ?? ?? ?? ?? EB ?? 48 8D ?? ?? ?? ?? ?? E8 ?? ?? ?? ?? C6 05 ?? ?? ?? ?? ?? 0F ?? ?? 4C");
|
||||||
|
GNamesaddress = Memory::AOBScan("", GNamesStringObfuscated.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 ?? ?? 48 89 ?? ?? ?? ?? ?? 4D 8B");
|
||||||
|
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<uint32_t>(Memory::GetOffsetFromOpcode(GObjectsaddress + 0x3) - baseModule);
|
||||||
|
logger->info("GObjects offset is: 0x{:X}.", gObjectsOffset);
|
||||||
|
Offsets::GObjects = static_cast<UC::uint32>(gObjectsOffset); // Update GObjects offset
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!GNamesaddress)
|
||||||
|
logger->warn("GNames signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
|
||||||
|
else {
|
||||||
|
uint32_t gNamesOffset = static_cast<uint32_t>(Memory::GetOffsetFromOpcode(GNamesaddress + 0x3) - baseModule);
|
||||||
|
logger->info("GNames offset is: 0x{:X}.", gNamesOffset);
|
||||||
|
Offsets::GNames = static_cast<UC::uint32>(gNamesOffset); // Update GNames 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<uint32_t> gProcessEventOffsetOpt = UE::CalculateOffset("", ProcessEventaddress);
|
||||||
|
uint32_t gProcessEventOffset = *gProcessEventOffsetOpt;
|
||||||
|
logger->info("Process Event offset is: 0x{:X}.", gProcessEventOffset);
|
||||||
|
Offsets::ProcessEvent = static_cast<UC::uint32>(gProcessEventOffset);// Update ProcessEvent offset
|
||||||
|
}
|
||||||
|
}
|
||||||
|
logger->info("-------------- Fixes initialisation -------------");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!init && FOVaddress) FOVFixEnabled();
|
||||||
|
if (!init && AspectRatioAxisConstraintaddress && ConstrainAspectRatioaddress) UltraWideFixEnabled();
|
||||||
|
if (!init && DOFaddress) DOFFixEnabled();
|
||||||
|
if (!init && CAaddress) CAFixEnabled();
|
||||||
|
if (!init && Vignettingaddress) VignettingFixEnabled();
|
||||||
|
if (!init && Fogaddress) FogFixEnabled();
|
||||||
|
if (!init && VolumetricFogaddress) VolumetricFogFixEnabled();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Setters for Reshade addon call
|
||||||
|
extern "C" __declspec(dllexport) void SetFixesEnabled(GameFixes fix, bool enabled)
|
||||||
|
{ // Set each fix individually
|
||||||
|
if (fix == GameFixes::DevConsole) EnableConsole();
|
||||||
|
if (fix == GameFixes::FOV) { g_fov_fix_enabled = enabled; FOVFixEnabled(); }
|
||||||
|
if (fix == GameFixes::UltraWide) { g_ultrawide_fix_enabled = enabled; UltraWideFixEnabled(); }
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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->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 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<uintptr_t*>(ctx.rax);
|
||||||
|
bool* bFog = reinterpret_cast<bool*>(reinterpret_cast<uint8_t*>(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\xC9\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, "\x30\xC0", 2); // 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()
|
||||||
|
{
|
||||||
|
logger->info("-------------- Console re-enabling --------------");
|
||||||
|
if (!GObjectsaddress || !GNamesaddress || !ProcessEventaddress) {
|
||||||
|
logger->warn("Could not re-enable console");
|
||||||
|
logger->info("------------------ User inputs ------------------");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
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;
|
||||||
|
|
||||||
|
// Set the all the console shortkey to F2
|
||||||
|
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<spdlog::logger>("Vampire TMBL2", 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;
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user