X-Git-Url: http://git.localhorst.tv/?a=blobdiff_plain;f=src%2Fnet%2FConnectionHandler.hpp;h=ff5e0e42204c8ab9998b83ac9630aae71e19d210;hb=b4e5dad2ed5c7e77573de413f3bf5be88577856d;hp=0f0afdaf1a6d24c3e2cb05396c9de7f31ea1ce4c;hpb=933ca0fe6c660e482edd45742d981f2de59a32df;p=blank.git

diff --git a/src/net/ConnectionHandler.hpp b/src/net/ConnectionHandler.hpp
index 0f0afda..ff5e0e4 100644
--- a/src/net/ConnectionHandler.hpp
+++ b/src/net/ConnectionHandler.hpp
@@ -1,6 +1,7 @@
 #ifndef BLANK_NET_CONNECTIONHANDLER_HPP_
 #define BLANK_NET_CONNECTIONHANDLER_HPP_
 
+#include "CongestionControl.hpp"
 #include "Packet.hpp"
 
 #include <SDL_net.h>
@@ -13,18 +14,20 @@ class ConnectionHandler {
 public:
 	ConnectionHandler();
 
-	float PacketLoss() const noexcept { return packet_loss; }
+	const CongestionControl &NetStat() const noexcept { return cc; }
 
+	void PacketSent(std::uint16_t) noexcept;
 	void PacketLost(std::uint16_t);
 	void PacketReceived(std::uint16_t);
 
+	void PacketIn(const UDPpacket &) noexcept;
+	void PacketOut(const UDPpacket &) noexcept;
+
 	void Handle(const UDPpacket &);
 
 	virtual void OnTimeout() { }
 
 private:
-	void UpdatePacketLoss() noexcept;
-
 	// called as soon as the remote end ack'd given packet
 	virtual void OnPacketReceived(std::uint16_t) { }
 	// called if the remote end probably didn't get given packet
@@ -42,11 +45,10 @@ private:
 	virtual void On(const Packet::ChunkBegin &) { }
 	virtual void On(const Packet::ChunkData &) { }
 	virtual void On(const Packet::BlockUpdate &) { }
+	virtual void On(const Packet::Message &) { }
 
 private:
-	unsigned int packets_lost;
-	unsigned int packets_received;
-	float packet_loss;
+	CongestionControl cc;
 
 };