exclude failed tunnel from pool

gema-arta · Mar 18, 2014 · d938332 · d938332
1 parent b34629b
commit d938332
Showing 1 changed file with 10 additions and 2 deletions.
diff --git 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;
 	}