127 lines
4.1 KiB
C++
127 lines
4.1 KiB
C++
#include "pch.h"
|
||
#include "../../../Common/apihook.h"
|
||
#include "NetworkHandlerWrapper.h"
|
||
#include "../../../Events/SpoiledEvent.h"
|
||
#include "../../../Events/CreatureDiedEvent.h"
|
||
#include "../../../Events/EventDispatcher.h"
|
||
#include "ProcessManipulation.h"
|
||
|
||
namespace Interlude
|
||
{
|
||
void* NetworkHandlerWrapper::originalInitAddress = 0;
|
||
NetworkHandlerWrapper::NetworkHandler* NetworkHandlerWrapper::_target = 0;
|
||
|
||
void(__thiscall* NetworkHandlerWrapper::__Init)(NetworkHandler*, float) = 0;
|
||
Item* (__thiscall* NetworkHandlerWrapper::__GetNextItem)(NetworkHandler*, float, int) = 0;
|
||
User* (__thiscall* NetworkHandlerWrapper::__GetNextCreature)(NetworkHandler*, float, int) = 0;
|
||
float(__thiscall* NetworkHandlerWrapper::__GetMaxTickRate)(NetworkHandler*) = 0;
|
||
int(__thiscall* NetworkHandlerWrapper::__AddNetworkQueue)(NetworkHandler*, L2::NetworkPacket*) = 0;
|
||
int(__thiscall* NetworkHandlerWrapper::__OnDie)(NetworkHandler*, User*, L2ParamStack&) = 0;
|
||
|
||
//todo exception
|
||
Item* NetworkHandlerWrapper::GetNextItem(float_t radius, int prevId) const
|
||
{
|
||
if (__GetNextItem && _target) {
|
||
return (*__GetNextItem)(_target, radius, prevId);
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
//todo exception
|
||
User* NetworkHandlerWrapper::GetNextCreature(float_t radius, int prevId) const
|
||
{
|
||
if (__GetNextCreature && _target) {
|
||
return (*__GetNextCreature)(_target, radius, prevId);
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
User* NetworkHandlerWrapper::GetHero() const
|
||
{
|
||
const auto creatures = FindAllObjects<User*>(0.1f, [this](float_t radius, int32_t prevId) {
|
||
return GetNextCreature(radius, prevId);
|
||
});
|
||
|
||
for (const auto& kvp : creatures)
|
||
{
|
||
const auto& creature = static_cast<User*>(kvp.second);
|
||
if (creature->userType == L2::UserType::USER && creature->lvl > 0)
|
||
{
|
||
return creature;
|
||
}
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
void NetworkHandlerWrapper::Init(HMODULE hModule)
|
||
{
|
||
void* initAddress = GetProcAddress(hModule, "?Tick@UNetworkHandler@@UAEXM@Z");
|
||
originalInitAddress = splice(initAddress, __Init_hook);
|
||
(FARPROC&)__Init = (FARPROC)initAddress;
|
||
|
||
(FARPROC&)__GetNextItem = GetProcAddress(hModule, "?GetNextItem@UNetworkHandler@@UAEPAUItem@@MH@Z");
|
||
(FARPROC&)__GetNextCreature = GetProcAddress(hModule, "?GetNextCreature@UNetworkHandler@@UAEPAUUser@@MH@Z");
|
||
|
||
(FARPROC&)__GetMaxTickRate = (FARPROC)splice(
|
||
GetProcAddress(hModule, "?GetMaxTickRate@UGameEngine@@UAEMXZ"), __GetMaxTickRate_hook
|
||
);
|
||
(FARPROC&)__AddNetworkQueue = (FARPROC)splice(
|
||
GetProcAddress(hModule, "?AddNetworkQueue@UNetworkHandler@@UAEHPAUNetworkPacket@@@Z"), __AddNetworkQueue_hook
|
||
);
|
||
(FARPROC&)__OnDie = (FARPROC)splice(
|
||
GetProcAddress(hModule, "?OnDie@UGameEngine@@UAEHPAUUser@@AAVL2ParamStack@@@Z"), __OnDie_hook
|
||
);
|
||
}
|
||
|
||
void NetworkHandlerWrapper::Restore()
|
||
{
|
||
restore(originalInitAddress);
|
||
restore((void*&)__GetMaxTickRate);
|
||
restore((void*&)__AddNetworkQueue);
|
||
restore((void*&)__OnDie);
|
||
}
|
||
|
||
void __fastcall NetworkHandlerWrapper::__Init_hook(NetworkHandler* This, int /*edx*/, float unk)
|
||
{
|
||
if (_target == 0) {
|
||
_target = This;
|
||
|
||
InjectLibrary::StopCurrentProcess();
|
||
restore(originalInitAddress);
|
||
InjectLibrary::StartCurrentProcess();
|
||
|
||
(*__Init)(This, unk);
|
||
}
|
||
}
|
||
|
||
// TODO ini
|
||
// 0 - <20><><EFBFBD> <20><><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
float __fastcall NetworkHandlerWrapper::__GetMaxTickRate_hook(NetworkHandler* This, int)
|
||
{
|
||
float fps = (*__GetMaxTickRate)(This);
|
||
return 0.0f;
|
||
}
|
||
|
||
int __fastcall NetworkHandlerWrapper::__AddNetworkQueue_hook(NetworkHandler* This, int, L2::NetworkPacket* packet)
|
||
{
|
||
if (packet->id == static_cast<int>(L2::NetworkPacketId::SYSTEM_MESSAGE)) {
|
||
L2::SystemMessagePacket* p = static_cast<L2::SystemMessagePacket*>(packet);
|
||
if (
|
||
p->GetMessageId() == static_cast<int>(L2::SystemMessagePacket::Type::SPOIL_SUCCESS) ||
|
||
p->GetMessageId() == static_cast<int>(L2::SystemMessagePacket::Type::ALREADY_SPOILED)
|
||
) {
|
||
EventDispatcher::GetInstance().Dispatch(SpoiledEvent{});
|
||
}
|
||
}
|
||
|
||
return (*__AddNetworkQueue)(This, packet);
|
||
}
|
||
|
||
int __fastcall NetworkHandlerWrapper::__OnDie_hook(NetworkHandler* This, int, User* creature, L2ParamStack& stack)
|
||
{
|
||
EventDispatcher::GetInstance().Dispatch(CreatureDiedEvent{ creature->objectId, stack.GetBufferAsVector<int32_t>()});
|
||
|
||
return (*__OnDie)(This, creature, stack);
|
||
}
|
||
}
|