Clean up
This commit is contained in:
nastys
@@ -0,0 +1,332 @@
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <Windows.h>
|
||||
#include <Dbt.h>
|
||||
#include <stdio.h>
|
||||
#include "Constants.h"
|
||||
#include "framework.h"
|
||||
#include "Input/Mouse/Mouse.h"
|
||||
#include "Input/Xinput/Xinput.h"
|
||||
#include "Input/Keyboard/Keyboard.h"
|
||||
#include "Input/DirectInput/DirectInput.h"
|
||||
#include "Input/DirectInput/Ds4/DualShock4.h"
|
||||
#include "Components/ComponentsManager.h"
|
||||
#include <tchar.h>
|
||||
#include <GL/freeglut.h>
|
||||
|
||||
#include <detours.h>
|
||||
#pragma comment(lib, "detours.lib")
|
||||
|
||||
void(__cdecl* divaEngineUpdate)() = (void(__cdecl*)())0x14018CC40;
|
||||
|
||||
LRESULT CALLBACK MessageWindowProcessCallback(HWND, UINT, WPARAM, LPARAM);
|
||||
DWORD WINAPI WindowMessageDispatcher(LPVOID);
|
||||
VOID RegisterMessageWindowClass();
|
||||
|
||||
struct
|
||||
{
|
||||
DWORD ID = NULL;
|
||||
HANDLE Handle = NULL;
|
||||
} MessageThread;
|
||||
|
||||
const wchar_t* MessageWindowClassName = TEXT("MessageWindowClass");
|
||||
const wchar_t* MessageWindowName = TEXT("MessageWindowTitle");
|
||||
|
||||
namespace TLAC
|
||||
{
|
||||
Components::ComponentsManager ComponentsManager;
|
||||
bool DeviceConnected = true;
|
||||
bool FirstUpdateTick = true;
|
||||
bool HasWindowFocus, HadWindowFocus;
|
||||
|
||||
void InitializeTick()
|
||||
{
|
||||
RegisterMessageWindowClass();
|
||||
if ((MessageThread.Handle = CreateThread(0, 0, WindowMessageDispatcher, 0, 0, 0)) == NULL)
|
||||
printf("[TLAC] InitializeTick(): CreateThread() Error: %d\n", GetLastError());
|
||||
|
||||
framework::DivaWindowHandle = FindWindow(0, framework::DivaWindowName);
|
||||
if (framework::DivaWindowHandle == NULL)
|
||||
framework::DivaWindowHandle = FindWindow(0, framework::GlutDefaultName);
|
||||
if (framework::DivaWindowHandle == NULL)
|
||||
framework::DivaWindowHandle = FindWindow(0, framework::freeGlutDefaultName);
|
||||
|
||||
HRESULT diInitResult = Input::InitializeDirectInput(framework::Module);
|
||||
if (FAILED(diInitResult))
|
||||
printf("[TLAC] InitializeTick(): Failed to initialize DirectInput. Error: 0x%08X\n", diInitResult);
|
||||
|
||||
ComponentsManager.Initialize();
|
||||
}
|
||||
|
||||
void UpdateTick()
|
||||
{
|
||||
if (FirstUpdateTick)
|
||||
{
|
||||
FirstUpdateTick = false;
|
||||
InitializeTick();
|
||||
}
|
||||
|
||||
if (DeviceConnected)
|
||||
{
|
||||
DeviceConnected = false;
|
||||
|
||||
if (!Input::DualShock4::InstanceInitialized())
|
||||
{
|
||||
if (Input::DualShock4::TryInitializeInstance())
|
||||
printf("[TLAC] UpdateTick(): DualShock4 connected and initialized\n");
|
||||
}
|
||||
}
|
||||
|
||||
ComponentsManager.Update();
|
||||
|
||||
HadWindowFocus = HasWindowFocus;
|
||||
HasWindowFocus = framework::DivaWindowHandle == NULL || GetForegroundWindow() == framework::DivaWindowHandle;
|
||||
|
||||
if ((HasWindowFocus) && (!framework::inputDisable))
|
||||
{
|
||||
Input::Keyboard::GetInstance()->PollInput();
|
||||
Input::Mouse::GetInstance()->PollInput();
|
||||
Input::Xinput::GetInstance()->PollInput();
|
||||
|
||||
if (Input::DualShock4::GetInstance() != nullptr)
|
||||
{
|
||||
if (!Input::DualShock4::GetInstance()->PollInput())
|
||||
{
|
||||
Input::DualShock4::DeleteInstance();
|
||||
printf("[TLAC] UpdateTick(): DualShock4 connection lost\n");
|
||||
}
|
||||
}
|
||||
|
||||
ComponentsManager.UpdateInput();
|
||||
}
|
||||
|
||||
if ((framework::inputDisable))
|
||||
{
|
||||
Input::Keyboard::GetInstance()->PollInput();
|
||||
Input::Mouse::GetInstance()->PollInput();
|
||||
Input::Xinput::GetInstance()->PollInput();
|
||||
|
||||
if (Input::DualShock4::GetInstance() != nullptr)
|
||||
{
|
||||
if (!Input::DualShock4::GetInstance()->PollInput())
|
||||
{
|
||||
Input::DualShock4::DeleteInstance();
|
||||
printf("[TLAC] UpdateTick(): DualShock4 connection lost\n");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (HasWindowFocus && !HadWindowFocus)
|
||||
ComponentsManager.OnFocusGain();
|
||||
|
||||
if (!HasWindowFocus && HadWindowFocus)
|
||||
ComponentsManager.OnFocusLost();
|
||||
}
|
||||
|
||||
void InitializeExtraSettings()
|
||||
{
|
||||
const LPCTSTR RESOLUTION_CONFIG_FILE_NAME = _T(".\\config.ini");
|
||||
auto nCustomRes = GetPrivateProfileIntW(L"resolution", L"r.enable", FALSE, RESOLUTION_CONFIG_FILE_NAME);
|
||||
auto nCustomWidth = GetPrivateProfileIntW(L"resolution", L"r.width", NULL, RESOLUTION_CONFIG_FILE_NAME);
|
||||
auto nCustomHeight = GetPrivateProfileIntW(L"resolution", L"r.height", NULL, RESOLUTION_CONFIG_FILE_NAME);
|
||||
auto nTAA = GetPrivateProfileIntW(L"graphics", L"taa", TRUE, RESOLUTION_CONFIG_FILE_NAME);
|
||||
auto nMLAA = GetPrivateProfileIntW(L"graphics", L"mlaa", TRUE, RESOLUTION_CONFIG_FILE_NAME);
|
||||
|
||||
if (nCustomRes)
|
||||
{
|
||||
printf("[TLAC] Custom internal resolution enabled\n");
|
||||
printf("[TLAC] X: %d Y: %d\n", nCustomWidth, nCustomHeight);
|
||||
// Requires -wqhd launch parameter or Render.dva plugin
|
||||
{
|
||||
DWORD oldProtect, bck;
|
||||
VirtualProtect((BYTE*)0x00000001409B8B68, 4, PAGE_EXECUTE_READWRITE, &oldProtect);
|
||||
*((int*)0x00000001409B8B68) = nCustomWidth;
|
||||
VirtualProtect((BYTE*)0x00000001409B8B68, 6, oldProtect, &bck);
|
||||
}
|
||||
{
|
||||
DWORD oldProtect, bck;
|
||||
VirtualProtect((BYTE*)0x00000001409B8B6C, 4, PAGE_EXECUTE_READWRITE, &oldProtect);
|
||||
*((int*)0x00000001409B8B6C) = nCustomHeight;
|
||||
VirtualProtect((BYTE*)0x00000001409B8B6C, 6, oldProtect, &bck);
|
||||
}
|
||||
|
||||
//*((int*)0x00000001409B8B6C) = maxHeight;
|
||||
//*((int*)0x00000001409B8B14) = maxWidth;
|
||||
//*((int*)0x00000001409B8B18) = maxHeight;
|
||||
|
||||
//*((int*)0x00000001409B8B1C) = maxWidth; // No parameters width?
|
||||
//*((int*)0x00000001409B8B20) = maxHeight; // No parameters height?
|
||||
}
|
||||
if (!nTAA)
|
||||
{
|
||||
DWORD oldProtect, bck;
|
||||
VirtualProtect((BYTE*)0x00000001411AB67C, 1, PAGE_EXECUTE_READWRITE, &oldProtect);
|
||||
*((byte*)0x00000001411AB67C + 0) = 0x00;
|
||||
VirtualProtect((BYTE*)0x00000001411AB67C, 1, oldProtect, &bck);
|
||||
|
||||
printf("[TLAC] TAA disabled\n");
|
||||
}
|
||||
if (!nMLAA)
|
||||
{
|
||||
DWORD oldProtect, bck;
|
||||
VirtualProtect((BYTE*)0x00000001411AB680, 1, PAGE_EXECUTE_READWRITE, &oldProtect);
|
||||
*((byte*)0x00000001411AB680 + 0) = 0x00;
|
||||
VirtualProtect((BYTE*)0x00000001411AB680, 1, oldProtect, &bck);
|
||||
|
||||
printf("[TLAC] MLAA disabled\n");
|
||||
}
|
||||
}
|
||||
|
||||
void Dispose()
|
||||
{
|
||||
ComponentsManager.Dispose();
|
||||
|
||||
delete Input::Keyboard::GetInstance();
|
||||
delete Input::Mouse::GetInstance();
|
||||
delete Input::DualShock4::GetInstance();
|
||||
|
||||
Input::DisposeDirectInput();
|
||||
|
||||
PostThreadMessage(MessageThread.ID, WM_QUIT, 0, 0);
|
||||
}
|
||||
}
|
||||
|
||||
DWORD WINAPI WindowMessageDispatcher(LPVOID lpParam)
|
||||
{
|
||||
HWND windowHandle = CreateWindowW(
|
||||
MessageWindowClassName,
|
||||
MessageWindowName,
|
||||
WS_OVERLAPPEDWINDOW,
|
||||
CW_USEDEFAULT, CW_USEDEFAULT,
|
||||
CW_USEDEFAULT, CW_USEDEFAULT,
|
||||
NULL, NULL,
|
||||
TLAC::framework::Module,
|
||||
NULL);
|
||||
|
||||
if (!windowHandle)
|
||||
{
|
||||
printf("[TLAC] WindowMessageDispatcher(): CreateWindowW() Error: %d\n", GetLastError());
|
||||
return 1;
|
||||
}
|
||||
|
||||
MessageThread.ID = GetCurrentThreadId();
|
||||
|
||||
MSG message;
|
||||
DWORD returnValue;
|
||||
|
||||
printf("[TLAC] WindowMessageDispatcher(): Entering message loop...\n");
|
||||
|
||||
while (1)
|
||||
{
|
||||
returnValue = GetMessage(&message, NULL, 0, 0);
|
||||
if (returnValue != -1)
|
||||
{
|
||||
TranslateMessage(&message);
|
||||
DispatchMessage(&message);
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("[TLAC] WindowMessageDispatcher(): GetMessage() Error: %d\n", returnValue);
|
||||
}
|
||||
}
|
||||
|
||||
DestroyWindow(windowHandle);
|
||||
return 0;
|
||||
}
|
||||
|
||||
BOOL RegisterDeviceInterface(HWND hWnd, HDEVNOTIFY* hDeviceNotify)
|
||||
{
|
||||
DEV_BROADCAST_DEVICEINTERFACE NotificationFilter = {};
|
||||
|
||||
NotificationFilter.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
|
||||
NotificationFilter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
|
||||
|
||||
*hDeviceNotify = RegisterDeviceNotification(hWnd, &NotificationFilter, DEVICE_NOTIFY_ALL_INTERFACE_CLASSES);
|
||||
|
||||
return *hDeviceNotify != NULL;
|
||||
}
|
||||
|
||||
LRESULT CALLBACK MessageWindowProcessCallback(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
|
||||
{
|
||||
switch (message)
|
||||
{
|
||||
case WM_CREATE:
|
||||
{
|
||||
HDEVNOTIFY hDevNotify = NULL;
|
||||
|
||||
if (!RegisterDeviceInterface(hWnd, &hDevNotify))
|
||||
printf("[TLAC] MessageWindowProcessCallback(): RegisterDeviceInterface() Error: %d\n", GetLastError());
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
case WM_DEVICECHANGE:
|
||||
{
|
||||
switch (wParam)
|
||||
{
|
||||
case DBT_DEVICEARRIVAL:
|
||||
TLAC::DeviceConnected = true;
|
||||
break;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
default:
|
||||
return DefWindowProc(hWnd, message, wParam, lParam);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
VOID RegisterMessageWindowClass()
|
||||
{
|
||||
WNDCLASS windowClass = { };
|
||||
|
||||
windowClass.lpfnWndProc = MessageWindowProcessCallback;
|
||||
windowClass.hInstance = TLAC::framework::Module;
|
||||
windowClass.lpszClassName = MessageWindowClassName;
|
||||
|
||||
RegisterClass(&windowClass);
|
||||
}
|
||||
|
||||
void hookedEngineUpdate()
|
||||
{
|
||||
TLAC::UpdateTick();
|
||||
//divaEngineUpdate();
|
||||
}
|
||||
|
||||
BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
|
||||
{
|
||||
HWND consoleHandle = GetConsoleWindow();
|
||||
|
||||
switch (ul_reason_for_call)
|
||||
{
|
||||
case DLL_PROCESS_ATTACH:
|
||||
|
||||
// Are you a dev and want to debug? Want to always be able to watch the console upon game initialization? Comment this.
|
||||
ShowWindow(consoleHandle, SW_HIDE);
|
||||
|
||||
printf("[TLAC] DllMain(): Installing hooks...\n");
|
||||
|
||||
|
||||
//InstallHook((void*)ENGINE_UPDATE_HOOK_TARGET_ADDRESS, (void*)UpdateTick, 0xE);
|
||||
DisableThreadLibraryCalls(hModule);
|
||||
DetourTransactionBegin();
|
||||
DetourUpdateThread(GetCurrentThread());
|
||||
DetourAttach(&(PVOID&)divaEngineUpdate, hookedEngineUpdate);
|
||||
DetourTransactionCommit();
|
||||
|
||||
TLAC::InitializeExtraSettings();
|
||||
|
||||
TLAC::framework::Module = hModule;
|
||||
break;
|
||||
|
||||
case DLL_PROCESS_DETACH:
|
||||
TLAC::Dispose();
|
||||
break;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
Reference in New Issue
Block a user