mirror of https://github.com/PurpleI2P/i2pd.git
orignal
11 years ago
5 changed files with 110 additions and 3 deletions
@ -0,0 +1,53 @@
@@ -0,0 +1,53 @@
|
||||
#include <boost/bind.hpp> |
||||
#include "I2PTunnel.h" |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace stream |
||||
{ |
||||
I2PTunnelConnection::I2PTunnelConnection (boost::asio::ip::tcp::socket * socket, |
||||
const i2p::data::LeaseSet * leaseSet): m_Socket (socket) |
||||
{ |
||||
m_Stream = i2p::stream::CreateStream (*leaseSet); |
||||
} |
||||
|
||||
I2PTunnelConnection::~I2PTunnelConnection () |
||||
{ |
||||
if (m_Stream) |
||||
{ |
||||
m_Stream->Close (); |
||||
DeleteStream (m_Stream); |
||||
} |
||||
|
||||
delete m_Socket; |
||||
} |
||||
|
||||
I2PClientTunnel::I2PClientTunnel (boost::asio::io_service& service, const std::string& destination, int port): |
||||
m_Service (service), m_Acceptor (m_Service, boost::asio::ip::tcp::endpoint (boost::asio::ip::tcp::v4(), port)), |
||||
m_Destination (destination), m_RemoteLeaseSet (nullptr) |
||||
{ |
||||
} |
||||
|
||||
|
||||
void I2PClientTunnel::Accept () |
||||
{ |
||||
auto newSocket = new boost::asio::ip::tcp::socket (m_Service); |
||||
m_Acceptor.async_accept (*newSocket, boost::bind (&I2PClientTunnel::HandleAccept, this, |
||||
boost::asio::placeholders::error, newSocket)); |
||||
} |
||||
|
||||
void I2PClientTunnel::HandleAccept (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket) |
||||
{ |
||||
if (!ecode) |
||||
{ |
||||
if (m_RemoteLeaseSet) |
||||
new I2PTunnelConnection (socket, m_RemoteLeaseSet); |
||||
else |
||||
delete socket; |
||||
Accept (); |
||||
} |
||||
else |
||||
delete socket; |
||||
} |
||||
} |
||||
} |
@ -0,0 +1,50 @@
@@ -0,0 +1,50 @@
|
||||
#ifndef I2PTUNNEL_H__ |
||||
#define I2PTUNNEL_H__ |
||||
|
||||
#include <inttypes.h> |
||||
#include <string> |
||||
#include <boost/asio.hpp> |
||||
#include "Identity.h" |
||||
#include "Streaming.h" |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace stream |
||||
{ |
||||
class I2PTunnelConnection |
||||
{ |
||||
public: |
||||
|
||||
I2PTunnelConnection (boost::asio::ip::tcp::socket * socket, |
||||
const i2p::data::LeaseSet * leaseSet); |
||||
~I2PTunnelConnection (); |
||||
|
||||
private: |
||||
|
||||
boost::asio::ip::tcp::socket * m_Socket; |
||||
Stream * m_Stream; |
||||
}; |
||||
|
||||
class I2PClientTunnel |
||||
{ |
||||
public: |
||||
|
||||
I2PClientTunnel (boost::asio::io_service& service, const std::string& destination, int port); |
||||
|
||||
private: |
||||
|
||||
void Accept (); |
||||
void HandleAccept (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket); |
||||
|
||||
private: |
||||
|
||||
boost::asio::io_service& m_Service; |
||||
boost::asio::ip::tcp::acceptor m_Acceptor; |
||||
std::string m_Destination; |
||||
i2p::data::IdentHash m_DestinationIdentHash; |
||||
const i2p::data::LeaseSet * m_RemoteLeaseSet; |
||||
}; |
||||
} |
||||
} |
||||
|
||||
#endif |
Loading…
Reference in new issue