e7562bcdff
* feat(protections): better logging * feat(protections): improve protections * fix(globals): save all spoofing settings
80 lines
2.7 KiB
C++
80 lines
2.7 KiB
C++
#include "gta/net_game_event.hpp"
|
|
#include "hooking.hpp"
|
|
#include "pointers.hpp"
|
|
|
|
#include <rage/sysMemAllocator.hpp>
|
|
|
|
namespace big
|
|
{
|
|
bool is_reliable_message(rage::netQueuedMessage* msg)
|
|
{
|
|
rage::datBitBuffer buffer(msg->m_data_buffer, INT_MAX);
|
|
buffer.m_flagBits |= 1; // read
|
|
buffer.Seek(10); // size
|
|
return buffer.ReadBits(1); // flags
|
|
}
|
|
|
|
void free_message(rage::netQueuedMessage* msg, rage::sysMemAllocator* allocator)
|
|
{
|
|
if (msg->m_data_buffer)
|
|
allocator->TryFree(msg->m_data_buffer);
|
|
|
|
allocator->TryFree(msg);
|
|
}
|
|
|
|
void* hooks::allocate_memory_reliable(rage::netConnection* cxn, int required_memory)
|
|
{
|
|
if (!cxn || !required_memory)
|
|
return nullptr;
|
|
|
|
auto memory = reinterpret_cast<rage::sysMemAllocator*>(cxn->m_allocator)->Allocate(required_memory, 0, 0);
|
|
|
|
if (memory)
|
|
return memory;
|
|
|
|
LOG(WARNING) << "Failed to allocate " << required_memory << " bytes for reliable message, free space: "
|
|
<< reinterpret_cast<rage::sysMemAllocator*>(cxn->m_allocator)->GetMemoryAvailable() << ". Trying to free some memory";
|
|
|
|
g_pointers->m_gta.m_connection_manager_try_free_memory(cxn->m_net_connection_mgr);
|
|
|
|
memory = reinterpret_cast<rage::sysMemAllocator*>(cxn->m_allocator)->Allocate(required_memory, 0, 0);
|
|
|
|
if (memory)
|
|
return memory;
|
|
|
|
LOG(WARNING) << "Failed to allocate " << required_memory << " bytes for reliable message, free space: "
|
|
<< reinterpret_cast<rage::sysMemAllocator*>(cxn->m_allocator)->GetMemoryAvailable() << ". Failed to free some memory, clearing all messages (including reliables) for connection";
|
|
|
|
while (cxn->m_normal_message_queue.m_count)
|
|
{
|
|
auto msg = cxn->m_normal_message_queue.m_first;
|
|
|
|
g_pointers->m_gta.m_remove_message_from_queue(&cxn->m_normal_message_queue, msg);
|
|
|
|
if (is_reliable_message(msg))
|
|
g_pointers->m_gta.m_remove_message_from_unacked_reliables(&cxn->m_unacked_reliable_message_list, &msg->word4C);
|
|
|
|
free_message(msg, reinterpret_cast<rage::sysMemAllocator*>(cxn->m_allocator));
|
|
}
|
|
|
|
while (cxn->m_reliables_resend_queue.m_count)
|
|
{
|
|
auto msg = cxn->m_reliables_resend_queue.m_first;
|
|
|
|
g_pointers->m_gta.m_remove_message_from_queue(&cxn->m_reliables_resend_queue, msg);
|
|
g_pointers->m_gta.m_remove_message_from_unacked_reliables(&cxn->m_unacked_reliable_message_list,
|
|
&msg->word4C); // messages in this queue are always reliables
|
|
free_message(msg, reinterpret_cast<rage::sysMemAllocator*>(cxn->m_allocator));
|
|
}
|
|
|
|
|
|
memory = reinterpret_cast<rage::sysMemAllocator*>(cxn->m_allocator)->Allocate(required_memory, 0, 0);
|
|
|
|
if (memory)
|
|
return memory;
|
|
|
|
g_notification_service->push_error("Protections", "The network message allocator is out of memory"); // this never reaches here but why not
|
|
|
|
return nullptr;
|
|
}
|
|
} |