mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
edc52ad669
Originally, we store the search results in a CMemory instance which we then assign to the actual function pointer. CMemory is just a pointer class; we can assign the results directly to the actual function pointer. This commit reduces a lot of code verbosity, and also reduced roughly 2KiB worth of static pointers in the resulting executable. This commit also officially deprecates the support for any GameDLL's below S3 (Season 3), since it makes more sense to port the assets from earlier/later games back to the version this SDK supports.
121 lines
3.5 KiB
C++
121 lines
3.5 KiB
C++
#include "core/stdafx.h"
|
|
#include "windows/system.h"
|
|
#include "engine/host_state.h"
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
typedef BOOL(WINAPI* IGetVersionExA)(
|
|
_Inout_ LPOSVERSIONINFOA lpVersionInformation);
|
|
typedef BOOL(WINAPI* IPeekMessage)(
|
|
_Out_ LPMSG lpMsg,
|
|
_In_opt_ HWND hWnd,
|
|
_In_ UINT wMsgFilterMin,
|
|
_In_ UINT wMsgFilterMax,
|
|
_In_ UINT wRemoveMsg);
|
|
static IGetVersionExA VGetVersionExA = nullptr;
|
|
static IPeekMessage VPeekMessageA = nullptr;
|
|
static IPeekMessage VPeekMessageW = nullptr;
|
|
|
|
//#############################################################################
|
|
// SYSTEM HOOKS
|
|
//#############################################################################
|
|
|
|
BOOL
|
|
WINAPI
|
|
HGetVersionExA(
|
|
_Inout_ LPOSVERSIONINFOA lpVersionInformation)
|
|
{
|
|
#ifdef DEDICATED
|
|
// Return false for dedicated to skip 'SetProcessDpiAwareness' in 'CEngineAPI:OnStartup()'.
|
|
return NULL;
|
|
#else
|
|
return VGetVersionExA(lpVersionInformation);
|
|
#endif // DEDICATED
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
HPeekMessage(
|
|
_Out_ LPMSG lpMsg,
|
|
_In_opt_ HWND hWnd,
|
|
_In_ UINT wMsgFilterMin,
|
|
_In_ UINT wMsgFilterMax,
|
|
_In_ UINT wRemoveMsg)
|
|
{
|
|
#ifdef DEDICATED
|
|
// Return false for dedicated to reduce unnecessary overhead when calling 'PeekMessageA/W()' every frame.
|
|
return NULL;
|
|
#else
|
|
return VPeekMessageA(lpMsg, hWnd, wMsgFilterMin, wMsgFilterMax, wRemoveMsg);
|
|
#endif // DEDICATED
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
ConsoleHandlerRoutine(
|
|
DWORD eventCode)
|
|
{
|
|
switch (eventCode)
|
|
{
|
|
case CTRL_CLOSE_EVENT:
|
|
case CTRL_LOGOFF_EVENT:
|
|
case CTRL_SHUTDOWN_EVENT:
|
|
if (g_pHostState)
|
|
{
|
|
g_pHostState->m_iNextState = HostStates_t::HS_SHUTDOWN;
|
|
}
|
|
|
|
Sleep(10000);
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
//#############################################################################
|
|
// MANAGEMENT
|
|
//#############################################################################
|
|
|
|
void WinSys_Init()
|
|
{
|
|
#ifdef DEDICATED
|
|
VGetVersionExA = (IGetVersionExA)DetourFindFunction("KERNEL32.dll", "GetVersionExA");
|
|
VPeekMessageA = (IPeekMessage)DetourFindFunction("USER32.dll", "PeekMessageA");
|
|
VPeekMessageW = (IPeekMessage)DetourFindFunction("USER32.dll", "PeekMessageW");
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
DetourTransactionBegin();
|
|
DetourUpdateThread(GetCurrentThread());
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
DetourAttach(&(LPVOID&)VGetVersionExA, (PBYTE)HGetVersionExA);
|
|
DetourAttach(&(LPVOID&)VPeekMessageA, (PBYTE)HPeekMessage);
|
|
//DetourAttach(&(LPVOID&)VPeekMessageW, (PBYTE)HPeekMessage);
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
HRESULT hr = DetourTransactionCommit();
|
|
if (hr != NO_ERROR)
|
|
{
|
|
// Failed to hook into the process, terminate
|
|
Assert(0);
|
|
Error(eDLL_T::COMMON, 0xBAD0C0DE, "Failed to detour process: error code = %08x\n", hr);
|
|
}
|
|
#endif // DEDICATED
|
|
}
|
|
|
|
void WinSys_Shutdown()
|
|
{
|
|
#ifdef DEDICATED
|
|
///////////////////////////////////////////////////////////////////////////
|
|
DetourTransactionBegin();
|
|
DetourUpdateThread(GetCurrentThread());
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
DetourDetach(&(LPVOID&)VGetVersionExA, (PBYTE)HGetVersionExA);
|
|
DetourDetach(&(LPVOID&)VPeekMessageA, (PBYTE)HPeekMessage);
|
|
//DetourDetach(&(LPVOID&)VPeekMessageW, (PBYTE)HPeekMessage);
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
DetourTransactionCommit();
|
|
#endif // DEDICATED
|
|
}
|