#include <std_include.hpp>
#include "loader/component_loader.hpp"
#include "game/game.hpp"

#include "console.hpp"

#include <utils/hook.hpp>

namespace security
{
	namespace
	{
		utils::hook::detour ui_replace_directive_hook;

		void set_cached_playerdata_stub(const int localclient, const int index1, const int index2)
		{
			if (index1 >= 0 && index1 < 18 && index2 >= 0 && index2 < 42)
			{
				reinterpret_cast<void(*)(int, int, int)>(0x140536A60)(localclient, index1, index2);
			}
		}

		void remap_cached_entities(game::mp::cachedSnapshot_t& snapshot)
		{
			static bool printed = false;
			if(snapshot.num_clients > 1200 && !printed)
			{
				printed = true;
				console::info("Too many entities (%d)... remapping!\n", snapshot.num_clients);
			}
			
			snapshot.num_clients = std::min(snapshot.num_clients, 1200);
		}

		void remap_cached_entities_stub(utils::hook::assembler& a)
		{
			a.pushad64();

			a.mov(rcx, rbx);
			a.call_aligned(remap_cached_entities);

			a.popad64();
			a.jmp(0x14044DE51);
		}

		void sv_execute_client_message_stub(game::mp::client_t* client, game::msg_t* msg)
		{
			if ((client->reliableSequence - client->reliableAcknowledge) < 0)
			{
				client->reliableAcknowledge = client->reliableSequence;
				console::info("Negative reliableAcknowledge from %s - cl->reliableSequence is %i, reliableAcknowledge is %i\n",
				              client->name, client->reliableSequence, client->reliableAcknowledge);
				return;
			}

			utils::hook::invoke<void>(0x14043AA90, client, msg);
		}

		void ui_replace_directive_stub(const int local_client_num, const char* src_string, char* dst_string, const int dst_buffer_size)
		{
			assert(src_string);
			if (!src_string)
			{
				return;
			}

			assert(dst_string);
			if (!dst_string)
			{
				return;
			}

			assert(dst_buffer_size > 0);
			if (dst_buffer_size <= 0)
			{
				return;
			}

			constexpr std::size_t MAX_HUDELEM_TEXT_LEN = 0x100;
			if (std::strlen(src_string) > MAX_HUDELEM_TEXT_LEN)
			{
				return;
			}

			ui_replace_directive_hook.invoke<void>(local_client_num, src_string, dst_string, dst_buffer_size);
		}

		int hud_elem_set_enum_string_stub(char* string, const char* format, ...)
		{
			va_list ap;
			va_start(ap, format);
			const auto len = vsnprintf(string, 0x800, format, ap);
			va_end(ap);

			return len;
		}

		int sv_add_bot_stub(char* string, const char* format, ...)
		{
			va_list ap;
			va_start(ap, format);
			const auto len = vsnprintf(string, 0x400, format, ap);
			va_end(ap);

			return len;
		}

		int sv_add_test_client_stub(char* string, const char* format, ...)
		{
			va_list ap;
			va_start(ap, format);
			const auto len = vsnprintf(string, 0x400, format, ap);
			va_end(ap);

			return len;
		}
	}

	class component final : public component_interface
	{
	public:
		void post_unpack() override
		{
			// sprinf
			utils::hook::call(SELECT_VALUE(0x1402203DF, 0x1402F0BAF), hud_elem_set_enum_string_stub);

			if (game::environment::is_sp()) return;

			// Patch vulnerability in PlayerCards_SetCachedPlayerData
			utils::hook::call(0x1401BB909, set_cached_playerdata_stub);

			// sprinf
			utils::hook::call(0x140439075, sv_add_bot_stub);
			utils::hook::call(0x14043932A, sv_add_test_client_stub);

			// Patch entity overflow
			utils::hook::jump(0x14044DE3A, assemble(remap_cached_entities_stub), true);

			// It is possible to make the server hang if left unchecked
			utils::hook::call(0x140443051, sv_execute_client_message_stub);

			ui_replace_directive_hook.create(0x1404A9640, ui_replace_directive_stub);
		}
	};
}

REGISTER_COMPONENT(security::component)