NetworkWrapper.h

#include <iostream>
#include <sstream>
#include <tuple>

#include "MTAStuff/sdk/SharedUtil.h"
#include "MTAStuff/core/CDynamicLibrary.h"
#include "MTAStuff/sdk/SharedUtil.h"
#include "MTAStuff/sdk/net/Packets.h"
#include "MTAStuff/sdk/net/bitstream.h"
#include "MTAStuff/sdk/net/CNetServer.h"

#include <Python.h>

#include "Packets/PlayerJoinDataPacket.h"

#include "Utils.h"

#ifndef WIN32
#define __stdcall
#endif

struct Packet
{
	NetServerPlayerID player;
	unsigned char ucPacketID;
	NetBitStreamInterface* pBitStream;
	unsigned char ucPriority;
	unsigned char ucReliability;

	Packet(NetServerPlayerID socket, unsigned char packetId, NetBitStreamInterface* bitStream, unsigned char priority, unsigned char reliability)
		: player(socket), ucPacketID(packetId), pBitStream(bitStream), ucPriority(priority), ucReliability(reliability) {}
};

struct SerialExtraAndVersion
{
	std::string serial;
	std::string extra;
	std::string version;

	SerialExtraAndVersion(std::string serial, std::string extra, std::string version)
		: serial(serial), extra(extra), version(version) {}
};

struct PlayerAddress
{
	std::string strIP = "";
	unsigned short usPort = 0;
};

class MTANetworkWrapper
{
private:
	unsigned short usID;
	static uint16_t nextId;
	static std::map<uint16_t, MTANetworkWrapper*> netWrappers;
	static std::map<NetServerPlayerID, MTANetworkWrapper*> netWrappersPerSocket;

	bool m_QueuedPacket = false;
	CDynamicLibrary m_NetworkLibLoader;

	unsigned int  m_uiPacketIndex;
	unsigned int  m_uiPacket = 0;
	unsigned long m_ulPlayerListAddress = 0;
	PyObject* m_pPacketData = nullptr;

	CNetServer* m_pNetwork;
	std::map<ulong, NetServerPlayerID> m_Players;
	bool m_bRunning;
	std::thread m_MainThread;
	std::thread m_PulseThread;
	std::thread m_PacketHandlerThread;
	std::queue<Packet> m_Packets;
	std::mutex mutex;

	void PulseLoop();
	void MainLoop();
public:
	MTANetworkWrapper();

	bool Setup(const char* szServerIdPath, const char* szNetLibPath, const char* szIP, unsigned short usPort, unsigned int uiPlayerCount,
		const char* szServerName, unsigned long* pulMtaVersionType, PyObject* pFunction);

	void Start();

	PyObject* GetLastPackets();

	void Send(unsigned long ulAddress, unsigned char ucPacketId, unsigned short usBitStreamVersion, const char* szData, unsigned long ulDataSize, unsigned char ucPriority, unsigned char ucReliability);

	bool IsValidSocket(unsigned long ulAddress);

	PlayerAddress GetPlayerAddress(unsigned long ulAddress);

	void SetClientBitStreamVersion(unsigned long ulAddress, unsigned short usVersion);

	void SetAntiCheatChecks(const char* szDisableComboACMap, const char* szDisableACMap, const char* szEnableSDMap, int iEnableClientChecks, bool bHideAC, const char* szImgMods);

	void GetAntiCheatInfo(unsigned long ulAddress);

	SerialExtraAndVersion GetClientData(unsigned long ulAddress);

	void  GetModPackets(unsigned long ulAddress);

	const char* GetNetRoute();

	const char* GetPingStatus();

	SBandwidthStatistics GetBandwidthStatistics();
	SPacketStat GetPacketStats();

	void Stop();

	void Destroy();

	ushort GetId();
	bool StaticPacketHandler(unsigned char ucPacketID, const NetServerPlayerID& player, NetBitStreamInterface* pBitStream, SNetExtraInfo* pNetExtraInfo);
	bool IsValidSocket(NetServerPlayerID id);


	static MTANetworkWrapper* GetNetWrapper(int iId);
	static MTANetworkWrapper* GetNetWrapper(NetServerPlayerID id);
};