collect network bandwidth usage stats

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

diff --git a/src/net/ConnectionHandler.hpp b/src/net/ConnectionHandler.hpp
index 0f0afdaf1a6d24c3e2cb05396c9de7f31ea1ce4c..76103fecdf11a5db2f36aaf1088e42c00f5d158b 100644 (file)
--- a/src/net/ConnectionHandler.hpp
+++ b/src/net/ConnectionHandler.hpp
@@ -13,17 +13,32 @@ 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) { }
@@ -42,12 +57,24 @@ 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;
 
+       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;
+
 };
 
 }