collect network bandwidth usage stats

[blank.git] / src / net / ConnectionHandler.hpp

diff --git a/src/net/ConnectionHandler.hpp b/src/net/ConnectionHandler.hpp
index 56538c6178505d0ee35c764bcfac4b684e1723b4..76103fecdf11a5db2f36aaf1088e42c00f5d158b 100644 (file)
--- a/src/net/ConnectionHandler.hpp
+++ b/src/net/ConnectionHandler.hpp
@@ -11,16 +11,40 @@ namespace blank {
 class ConnectionHandler {
 
 public:
+       ConnectionHandler();
+
+       /// packet loss as factor
+       float PacketLoss() const noexcept { return packet_loss; }
+       /// smooth average round trip time in milliseconds
+       float RoundTripTime() const noexcept { return rtt; }
+       /// estimated kilobytes transferred per second
+       float Upstream() const noexcept { return tx_kbps; }
+       /// estimated kilobytes received per second
+       float Downstream() const noexcept { return rx_kbps; }
+
+       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;
+       void UpdateRTT(std::uint16_t) noexcept;
+       bool SamplePacket(std::uint16_t) const noexcept;
+       int HeadDiff(std::uint16_t) const noexcept;
+       void UpdateStats() 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
        virtual void OnPacketLost(std::uint16_t) { }
 
-       virtual void OnTimeout() { }
-
-private:
        virtual void On(const Packet::Ping &) { }
        virtual void On(const Packet::Login &) { }
        virtual void On(const Packet::Join &) { }
@@ -29,6 +53,27 @@ private:
        virtual void On(const Packet::SpawnEntity &) { }
        virtual void On(const Packet::DespawnEntity &) { }
        virtual void On(const Packet::EntityUpdate &) { }
+       virtual void On(const Packet::PlayerCorrection &) { }
+       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;
+
+       Uint32 stamps[16];
+       std::size_t stamp_cursor;
+       std::uint16_t stamp_last;
+       float rtt;
+
+       Uint32 next_sample;
+       std::size_t tx_bytes;
+       std::size_t rx_bytes;
+       float tx_kbps;
+       float rx_kbps;
 
 };