iw4x-client/src/Components/Modules/Network.hpp

#define NETWORK_MAX_PACKETS_PER_SECOND 100'000

namespace Components
{
	class Network : public Component
	{
	public:
		class Address
		{
		public:
			Address() { this->SetType(Game::netadrtype_t::NA_BAD); };
			Address(std::string addrString);
			Address(Game::netadr_t addr) : address(addr) {}
			Address(Game::netadr_t* addr) : Address(*addr) {}
			Address(const Address& obj) : address(obj.address) {};
			Address(const Proto::Network::Address& addr) { this->Deserialize(addr); };
			bool operator!=(const Address &obj) { return !(*this == obj); };
			bool operator==(const Address &obj);

			void SetPort(unsigned short port);
			unsigned short GetPort();

			void SetIP(DWORD ip);
			void SetIP(Game::netIP_t ip);
			Game::netIP_t GetIP();

			void SetType(Game::netadrtype_t type);
			Game::netadrtype_t GetType();

			Game::netadr_t* Get();
			const char* GetString();

			bool IsLocal();
			bool IsSelf();
			bool IsValid();

			void Serialize(Proto::Network::Address* protoAddress);
			void Deserialize(const Proto::Network::Address& protoAddress);

		private:
			Game::netadr_t address;
		};

		typedef void(Callback)(Address address, std::string data);
		typedef void(CallbackRaw)();

		Network();
		~Network();
		const char* GetName() { return "Network"; };

		static void Handle(std::string packet, Callback* callback);
		static void OnStart(CallbackRaw* callback);

		// Send quake-styled binary data
		static void Send(Address target, std::string data);
		static void Send(Game::netsrc_t type, Address target, std::string data);

		// Allows sending raw data without quake header
		static void SendRaw(Address target, std::string data);
		static void SendRaw(Game::netsrc_t type, Address target, std::string data);

		// Send quake-style command using binary data
		static void SendCommand(Address target, std::string command, std::string data = "");
		static void SendCommand(Game::netsrc_t type, Address target, std::string command, std::string data = "");

		static void Broadcast(unsigned short port, std::string data);
		static void BroadcastRange(unsigned int min, unsigned int max, std::string data);
		static void BroadcastAll(std::string data);

	private:
		static SOCKET TcpSocket;
		static std::string SelectedPacket;
		static wink::signal<wink::slot<CallbackRaw>> StartupSignal;
		static std::map<std::string, wink::slot<Callback>> PacketHandlers;

		static int PacketInterceptionHandler(const char* packet);
		static void DeployPacket(Game::netadr_t* from, Game::msg_t* msg);
		static void DeployPacketStub();

		static void NetworkStart();
		static void NetworkStartStub();
	};
}