net: Add setTotalBytes functionality for traffic data persistence

Add functions to set total bytes sent/received counts to enable restoring
traffic statistics between application sessions. This allows for continuous
traffic monitoring even after application restarts, improving the usefulness
of the traffic graph widget.

Author: rebroad

src/interfaces/node.h

@@ -157,6 +157,12 @@ class Node
     //! Get total bytes sent.
     virtual int64_t getTotalBytesSent() = 0;
 
+    //! Set total bytes recv
+    virtual void setTotalBytesRecv(uint64_t bytes) = 0;
+
+    //! Set total bytes sent
+    virtual void setTotalBytesSent(uint64_t bytes) = 0;
+
     //! Get mempool size.
     virtual size_t getMempoolSize() = 0;
 

src/net.cpp

@@ -3779,6 +3779,14 @@ uint64_t CConnman::GetTotalBytesSent() const
     return nTotalBytesSent;
 }
 
+void CConnman::SetTotalBytesRecv(uint64_t bytes) { nTotalBytesRecv.store(bytes); }
+
+void CConnman::SetTotalBytesSent(uint64_t bytes)
+{
+    WAIT_LOCK(m_total_bytes_sent_mutex, lock);
+    nTotalBytesSent = bytes;
+}
+
 ServiceFlags CConnman::GetLocalServices() const
 {
     return m_local_services;

src/net.h

@@ -1261,6 +1261,8 @@ class CConnman
 
     uint64_t GetTotalBytesRecv() const;
     uint64_t GetTotalBytesSent() const EXCLUSIVE_LOCKS_REQUIRED(!m_total_bytes_sent_mutex);
+    void SetTotalBytesRecv(uint64_t bytes);
+    void SetTotalBytesSent(uint64_t bytes);
 
     /** Get a unique deterministic randomizer. */
     CSipHasher GetDeterministicRandomizer(uint64_t id) const;

src/node/interfaces.cpp

@@ -285,6 +285,8 @@ class NodeImpl : public Node
     }
     int64_t getTotalBytesRecv() override { return m_context->connman ? m_context->connman->GetTotalBytesRecv() : 0; }
     int64_t getTotalBytesSent() override { return m_context->connman ? m_context->connman->GetTotalBytesSent() : 0; }
+    void setTotalBytesRecv(uint64_t bytes) override { if (m_context->connman) m_context->connman->SetTotalBytesRecv(bytes); }
+    void setTotalBytesSent(uint64_t bytes) override { if (m_context->connman) m_context->connman->SetTotalBytesSent(bytes); }
     size_t getMempoolSize() override { return m_context->mempool ? m_context->mempool->size() : 0; }
     size_t getMempoolDynamicUsage() override { return m_context->mempool ? m_context->mempool->DynamicMemoryUsage() : 0; }
     size_t getMempoolMaxUsage() override { return m_context->mempool ? m_context->mempool->m_opts.max_size_bytes : 0; }