Merge pull request #16 from orignal/master

Merge pull request from orignal/master
10 years ago · 1bad308878
parent 6e7c3677fa d938332db2
commit 1bad308878
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;
 	}