exclude failed tunnel from pool

10 years ago · d938332db2
parent b34629b9c0
commit d938332db2
1 changed files with 10 additions and 2 deletions
--- a/TunnelPool.cpp
+++ b/TunnelPool.cpp
@ -37,7 +37,11 @@ namespace tunnel

 	void TunnelPool::TunnelExpired (InboundTunnel * expiredTunnel)
 	{
-		m_InboundTunnels.erase (expiredTunnel);
+		if (expiredTunnel)
+		{	
+			expiredTunnel->SetTunnelPool (nullptr);
+			m_InboundTunnels.erase (expiredTunnel);
+		}	
 		if (m_LocalDestination)
 			m_LocalDestination->UpdateLeaseSet ();
 	}	
@ -49,7 +53,11 @@ namespace tunnel

 	void TunnelPool::TunnelExpired (OutboundTunnel * expiredTunnel)
 	{
-		m_OutboundTunnels.erase (expiredTunnel);
+		if (expiredTunnel)
+		{	
+			expiredTunnel->SetTunnelPool (nullptr);
+			m_OutboundTunnels.erase (expiredTunnel);
+		}	
 		if (expiredTunnel == m_LastOutboundTunnel)
 			m_LastOutboundTunnel = nullptr;
 	}