From 53adf7a79398fa9fbde7e3236fe641753f21e032 Mon Sep 17 00:00:00 2001
From: orignal <i2porignal@yandex.ru>
Date: Wed, 11 Feb 2015 14:45:25 -0500
Subject: [PATCH] cleanup dead peers

---
 Transports.cpp | 30 ++++++++++++++++++++++++++----
 Transports.h   |  6 +++++-
 2 files changed, 31 insertions(+), 5 deletions(-)

diff --git a/Transports.cpp b/Transports.cpp
index 70c6b758..713f4ea1 100644
--- a/Transports.cpp
+++ b/Transports.cpp
@@ -95,7 +95,7 @@ namespace transport
 	Transports transports;	
 	
 	Transports::Transports (): 
-		m_IsRunning (false), m_Thread (nullptr), m_Work (m_Service),
+		m_IsRunning (false), m_Thread (nullptr), m_Work (m_Service), m_PeerCleanupTimer (m_Service),
 		m_NTCPServer (nullptr), m_SSUServer (nullptr), 
 		m_DHKeysPairSupplier (5) // 5 pre-generated keys
 	{		
@@ -134,10 +134,13 @@ namespace transport
 					LogPrint ("SSU server already exists");
 			}
 		}	
+		m_PeerCleanupTimer.expires_from_now (boost::posix_time::seconds(5*SESSION_CREATION_TIMEOUT));
+		m_PeerCleanupTimer.async_wait (std::bind (&Transports::HandlePeerCleanupTimer, this, std::placeholders::_1));
 	}
 		
 	void Transports::Stop ()
 	{	
+		m_PeerCleanupTimer.cancel ();	
 		m_Peers.clear ();
 		if (m_SSUServer)
 		{
@@ -202,7 +205,8 @@ namespace transport
 		if (it == m_Peers.end ())
 		{
 			auto r = netdb.FindRouter (ident);
-			it = m_Peers.insert (std::pair<i2p::data::IdentHash, Peer>(ident, { 0, r, nullptr})).first;
+			it = m_Peers.insert (std::pair<i2p::data::IdentHash, Peer>(ident, { 0, r, nullptr,
+				i2p::util::GetSecondsSinceEpoch () })).first;
 			if (!ConnectToPeer (ident, it->second))
 			{
 				DeleteI2NPMessage (msg);
@@ -228,7 +232,8 @@ namespace transport
 		if (it == m_Peers.end ())
 		{
 			auto r = netdb.FindRouter (ident);
-			it = m_Peers.insert (std::pair<i2p::data::IdentHash, Peer>(ident, { 0, r, nullptr})).first;
+			it = m_Peers.insert (std::pair<i2p::data::IdentHash, Peer>(ident, { 0, r, nullptr,
+				i2p::util::GetSecondsSinceEpoch () })).first;
 			if (!ConnectToPeer (ident, it->second))
 			{
 				for (auto it1: msgs)
@@ -419,7 +424,7 @@ namespace transport
 				}
 			}
 			else // incoming connection
-				m_Peers.insert (std::make_pair (ident, Peer{ 0, nullptr, session }));
+				m_Peers.insert (std::make_pair (ident, Peer{ 0, nullptr, session, i2p::util::GetSecondsSinceEpoch () }));
 		});			
 	}
 		
@@ -438,6 +443,23 @@ namespace transport
 			}
 		});	
 	}	
+
+	void Transports::HandlePeerCleanupTimer (const boost::system::error_code& ecode)
+	{
+		auto ts = i2p::util::GetSecondsSinceEpoch ();
+		for (auto it = m_Peers.begin (); it != m_Peers.end (); )
+		{
+			if (!it->second.session && ts > it->second.creationTime + SESSION_CREATION_TIMEOUT)
+			{
+				LogPrint (eLogError, "Session to peer ", it->first.ToBase64 (), " has not been created in ", SESSION_CREATION_TIMEOUT, " seconds");
+				it = m_Peers.erase (it);
+			}
+			else
+				it++;
+		}
+		m_PeerCleanupTimer.expires_from_now (boost::posix_time::seconds(5*SESSION_CREATION_TIMEOUT));
+		m_PeerCleanupTimer.async_wait (std::bind (&Transports::HandlePeerCleanupTimer, this, std::placeholders::_1));
+	}
 }
 }
 
diff --git a/Transports.h b/Transports.h
index c9b882c5..e1c85374 100644
--- a/Transports.h
+++ b/Transports.h
@@ -56,6 +56,7 @@ namespace transport
 		int numAttempts;
 		std::shared_ptr<const i2p::data::RouterInfo> router;
 		std::shared_ptr<TransportSession> session;
+		uint64_t creationTime;
 		std::vector<i2p::I2NPMessage *> delayedMessages;
 
 		~Peer ()
@@ -65,6 +66,7 @@ namespace transport
 		}	
 	};	
 	
+	const size_t SESSION_CREATION_TIMEOUT = 10; // in seconds
 	class Transports
 	{
 		public:
@@ -95,7 +97,8 @@ namespace transport
 			void PostMessages (i2p::data::IdentHash ident, std::vector<i2p::I2NPMessage *> msgs);
 			void PostCloseSession (std::shared_ptr<const i2p::data::RouterInfo> router);
 			bool ConnectToPeer (const i2p::data::IdentHash& ident, Peer& peer);
-			
+			void HandlePeerCleanupTimer (const boost::system::error_code& ecode);			
+
 			void NTCPResolve (const std::string& addr, const i2p::data::IdentHash& ident);
 			void HandleNTCPResolve (const boost::system::error_code& ecode, boost::asio::ip::tcp::resolver::iterator it,
  				const i2p::data::IdentHash& ident, std::shared_ptr<boost::asio::ip::tcp::resolver> resolver);
@@ -108,6 +111,7 @@ namespace transport
 			std::thread * m_Thread;	
 			boost::asio::io_service m_Service;
 			boost::asio::io_service::work m_Work;
+			boost::asio::deadline_timer m_PeerCleanupTimer;
 
 			NTCPServer * m_NTCPServer;
 			SSUServer * m_SSUServer;