bind streaming destination to port for server tunnel

chadf · Mar 3, 2015 · c66ba37 · c66ba37
1 parent 6fc0b2e
commit c66ba37
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/I2PTunnel.cpp b/I2PTunnel.cpp
@@ -246,6 +246,7 @@ namespace client
 	I2PServerTunnel::I2PServerTunnel (const std::string& address, int port, std::shared_ptr<ClientDestination> localDestination): 
 		I2PService (localDestination), m_Endpoint (boost::asio::ip::address::from_string (address), port)
 	{
+		m_PortDestination = localDestination->CreateStreamingDestination (port);
 	}
 
 	void I2PServerTunnel::Start ()
@@ -260,9 +261,15 @@ namespace client
 
 	void I2PServerTunnel::Accept ()
 	{
+		if (m_PortDestination)
+			m_PortDestination->SetAcceptor (std::bind (&I2PServerTunnel::HandleAccept, this, std::placeholders::_1));
+
 		auto localDestination = GetLocalDestination ();	
 		if (localDestination)
-			localDestination->AcceptStreams (std::bind (&I2PServerTunnel::HandleAccept, this, std::placeholders::_1));
+		{
+			if (!localDestination->IsAcceptingStreams ()) // set it as default if not set yet
+				localDestination->AcceptStreams (std::bind (&I2PServerTunnel::HandleAccept, this, std::placeholders::_1));
+		}
 		else
 			LogPrint ("Local destination not set for server tunnel");
 	}

diff --git a/I2PTunnel.h b/I2PTunnel.h
@@ -95,7 +95,8 @@ namespace client
 
 		private:
 
-			boost::asio::ip::tcp::endpoint m_Endpoint;		
+			boost::asio::ip::tcp::endpoint m_Endpoint;	
+			std::shared_ptr<i2p::stream::StreamingDestination> m_PortDestination;	
 	};
 }
 }