fixed race condition

NTCPSession.cpp

@@ -751,17 +751,24 @@ namespace transport
 	void NTCPServer::AddNTCPSession (std::shared_ptr<NTCPSession> session)
 	{
 		if (session)
+		{
+			std::unique_lock<std::mutex> l(m_NTCPSessionsMutex);	
 			m_NTCPSessions[session->GetRemoteIdentity ().GetIdentHash ()] = session;
 		}
+	}	
 
 	void NTCPServer::RemoveNTCPSession (std::shared_ptr<NTCPSession> session)
 	{
 		if (session)
+		{
+			std::unique_lock<std::mutex> l(m_NTCPSessionsMutex);	
 			m_NTCPSessions.erase (session->GetRemoteIdentity ().GetIdentHash ());
 		}
+	}	
 
 	std::shared_ptr<NTCPSession> NTCPServer::FindNTCPSession (const i2p::data::IdentHash& ident)
 	{
+		std::unique_lock<std::mutex> l(m_NTCPSessionsMutex);	
 		auto it = m_NTCPSessions.find (ident);
 		if (it != m_NTCPSessions.end ())
 			return it->second;

NTCPSession.h

@@ -6,6 +6,7 @@
 #include <map>
 #include <memory>
 #include <thread>
+#include <mutex>
 #include <boost/asio.hpp>
 #include <cryptopp/modes.h>
 #include <cryptopp/aes.h>
@@ -167,6 +168,7 @@ namespace transport
 			boost::asio::io_service m_Service;
 			boost::asio::io_service::work m_Work;
 			boost::asio::ip::tcp::acceptor * m_NTCPAcceptor, * m_NTCPV6Acceptor;
+			std::mutex m_NTCPSessionsMutex;
 			std::map<i2p::data::IdentHash, std::shared_ptr<NTCPSession> > m_NTCPSessions;
 
 		public: