From 67763248cc088b179ca51ae728890e10c24bc0ec Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Tue, 24 Sep 2024 22:06:44 -0400
Subject: [PATCH] add peer test session endpoint to connected recently after
 msg 6 and 7

---
 libi2pd/SSU2Session.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/libi2pd/SSU2Session.cpp b/libi2pd/SSU2Session.cpp
index ce3f1637..bc47690d 100644
--- a/libi2pd/SSU2Session.cpp
+++ b/libi2pd/SSU2Session.cpp
@@ -3064,6 +3064,7 @@ namespace transport
 		SetSourceConnID (sourceConnID);
 		SetDestConnID (destConnID);	
 		SetState (eSSU2SessionStatePeerTest);	
+		SetTerminationTimeout (SSU2_PEER_TEST_EXPIRATION_TIMEOUT);
 	}	
 
 	bool SSU2PeerTestSession::ProcessPeerTest (uint8_t * buf, size_t len)
@@ -3140,6 +3141,7 @@ namespace transport
 					SendPeerTest (7, buf + offset, len - offset);
 				else
 					LogPrint (eLogWarning, "SSU2: Unknown address for peer test 6");
+				GetServer ().AddConnectedRecently (GetRemoteEndpoint (), i2p::util::GetSecondsSinceEpoch ());
 				GetServer ().RequestRemoveSession (GetConnID ());
 				break;
 			}			
@@ -3148,6 +3150,7 @@ namespace transport
 				auto addr = GetAddress ();
 				if (addr && addr->IsV6 ())
 					i2p::context.SetStatusV6 (eRouterStatusOK); // set status OK for ipv6 even if from SSU2
+				GetServer ().AddConnectedRecently (GetRemoteEndpoint (), i2p::util::GetSecondsSinceEpoch ());
 				GetServer ().RequestRemoveSession (GetConnID ());	
 				break;
 			}