ActiveProxy: reset the internal state of ActiveProxy object when begi…

…nning a new connection (#206)
elastos · Sep 7, 2023 · aa9e386 · aa9e386
1 parent 70a49de
commit aa9e386
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/addons/activeproxy/activeproxy.cc b/src/addons/activeproxy/activeproxy.cc
@@ -185,7 +185,7 @@ bool ActiveProxy::needsNewConnection() noexcept
 
     if (connections.empty()) {
         if (serverPk.has_value())
-            serverPk.reset();
+            reset();
         return true;
     }
 

diff --git a/src/addons/activeproxy/activeproxy.h b/src/addons/activeproxy/activeproxy.h
@@ -165,6 +165,23 @@ class ActiveProxy : public Addon {
 
     void announcePeer() noexcept;
 
+    void reset() noexcept {
+        sessionKey = CryptoBox::KeyPair();
+        serverPk.reset();
+
+        lastConnectTimestamp = 0;
+        serverFails = 0;
+        reconnectDelay = 0;
+
+        idleTimestamp = UINT64_MAX;
+
+        lastIdleCheckTimestamp = 0;
+        lastHealthCheckTimestamp = 0;
+        lastAnnouncePeerTimestamp = 0;
+
+        inFlights = 0;
+    }
+
 private:
     Sp<Node> node;