1 files changed, 15 insertions, 1 deletions
diff --git a/contrib/Socket.h b/contrib/Socket.h
index ab2a14a..29b618a 100644
--- a/contrib/Socket.h
+++ b/contrib/Socket.h
@@ -31,9 +31,11 @@
 #include "ThreadsafeQueue.h"
 #include <cstdlib>
 #include <atomic>
-#include <string>
+#include <chrono>
 #include <list>
 #include <memory>
+#include <optional>
+#include <string>
 #include <thread>
 #include <vector>
 
@@ -211,6 +213,8 @@ class TCPSocket {
 
         SOCKET get_sockfd() const { return m_sock; }
 
+        InetAddress get_remote_address() const { return m_remote_address; }
+
     private:
         explicit TCPSocket(int sockfd);
         explicit TCPSocket(int sockfd, InetAddress remote_address);
@@ -236,6 +240,8 @@ class TCPClient {
         TCPSocket m_sock;
         std::string m_hostname;
         int m_port;
+
+        std::optional<std::chrono::steady_clock::time_point> m_last_received_packet_ts;
 };
 
 /* Helper class for TCPDataDispatcher, contains a queue of pending data and
@@ -250,6 +256,12 @@ class TCPConnection
 
         ThreadsafeQueue<std::vector<uint8_t> > queue;
 
+        struct stats_t {
+            size_t buffer_fullness = 0;
+            InetAddress remote_address;
+        };
+        stats_t get_stats() const;
+
     private:
         std::atomic<bool> m_running;
         std::thread m_sender_thread;
@@ -272,6 +284,8 @@ class TCPDataDispatcher
         void start(int port, const std::string& address);
         void write(const std::vector<uint8_t>& data);
 
+        std::vector<TCPConnection::stats_t> get_stats() const;
+
     private:
         void process();