mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2024-11-04 06:00:37 +00:00
129 lines
3.6 KiB
C++
129 lines
3.6 KiB
C++
#ifndef I2PTUNNEL_H__
|
|
#define I2PTUNNEL_H__
|
|
|
|
#include <inttypes.h>
|
|
#include <string>
|
|
#include <set>
|
|
#include <memory>
|
|
#include <boost/asio.hpp>
|
|
#include "Identity.h"
|
|
#include "Destination.h"
|
|
#include "Streaming.h"
|
|
|
|
namespace i2p
|
|
{
|
|
namespace client
|
|
{
|
|
const size_t I2P_TUNNEL_CONNECTION_BUFFER_SIZE = 8192;
|
|
const int I2P_TUNNEL_CONNECTION_MAX_IDLE = 3600; // in seconds
|
|
const int I2P_TUNNEL_DESTINATION_REQUEST_TIMEOUT = 10; // in seconds
|
|
|
|
class I2PTunnel;
|
|
class I2PTunnelConnection: public std::enable_shared_from_this<I2PTunnelConnection>
|
|
{
|
|
public:
|
|
|
|
I2PTunnelConnection (I2PTunnel * owner, boost::asio::ip::tcp::socket * socket,
|
|
const i2p::data::LeaseSet * leaseSet);
|
|
I2PTunnelConnection (I2PTunnel * owner, std::shared_ptr<i2p::stream::Stream> stream, boost::asio::ip::tcp::socket * socket,
|
|
const boost::asio::ip::tcp::endpoint& target);
|
|
~I2PTunnelConnection ();
|
|
|
|
void I2PConnect ();
|
|
void Connect ();
|
|
|
|
private:
|
|
|
|
void Terminate ();
|
|
|
|
void Receive ();
|
|
void HandleReceived (const boost::system::error_code& ecode, std::size_t bytes_transferred);
|
|
void HandleWrite (const boost::system::error_code& ecode);
|
|
|
|
void StreamReceive ();
|
|
void HandleStreamReceive (const boost::system::error_code& ecode, std::size_t bytes_transferred);
|
|
void HandleConnect (const boost::system::error_code& ecode);
|
|
|
|
private:
|
|
|
|
uint8_t m_Buffer[I2P_TUNNEL_CONNECTION_BUFFER_SIZE], m_StreamBuffer[I2P_TUNNEL_CONNECTION_BUFFER_SIZE];
|
|
boost::asio::ip::tcp::socket * m_Socket;
|
|
std::shared_ptr<i2p::stream::Stream> m_Stream;
|
|
I2PTunnel * m_Owner;
|
|
boost::asio::ip::tcp::endpoint m_RemoteEndpoint;
|
|
};
|
|
|
|
class I2PTunnel
|
|
{
|
|
public:
|
|
|
|
I2PTunnel (boost::asio::io_service& service, ClientDestination * localDestination):
|
|
m_Service (service), m_LocalDestination (localDestination) {};
|
|
virtual ~I2PTunnel () { ClearConnections (); };
|
|
|
|
void AddConnection (std::shared_ptr<I2PTunnelConnection> conn);
|
|
void RemoveConnection (std::shared_ptr<I2PTunnelConnection> conn);
|
|
void ClearConnections ();
|
|
ClientDestination * GetLocalDestination () { return m_LocalDestination; };
|
|
void SetLocalDestination (ClientDestination * dest) { m_LocalDestination = dest; };
|
|
|
|
boost::asio::io_service& GetService () { return m_Service; };
|
|
|
|
private:
|
|
|
|
boost::asio::io_service& m_Service;
|
|
ClientDestination * m_LocalDestination;
|
|
std::set<std::shared_ptr<I2PTunnelConnection> > m_Connections;
|
|
};
|
|
|
|
class I2PClientTunnel: public I2PTunnel
|
|
{
|
|
public:
|
|
|
|
I2PClientTunnel (boost::asio::io_service& service, const std::string& destination, int port,
|
|
ClientDestination * localDestination = nullptr);
|
|
~I2PClientTunnel ();
|
|
|
|
void Start ();
|
|
void Stop ();
|
|
|
|
private:
|
|
|
|
void Accept ();
|
|
void HandleAccept (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket);
|
|
void HandleDestinationRequestTimer (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket);
|
|
void CreateConnection (boost::asio::ip::tcp::socket * socket);
|
|
|
|
private:
|
|
|
|
boost::asio::ip::tcp::acceptor m_Acceptor;
|
|
boost::asio::deadline_timer m_Timer;
|
|
std::string m_Destination;
|
|
const i2p::data::IdentHash * m_DestinationIdentHash;
|
|
const i2p::data::LeaseSet * m_RemoteLeaseSet;
|
|
};
|
|
|
|
class I2PServerTunnel: public I2PTunnel
|
|
{
|
|
public:
|
|
|
|
I2PServerTunnel (boost::asio::io_service& service, const std::string& address, int port,
|
|
ClientDestination * localDestination);
|
|
|
|
void Start ();
|
|
void Stop ();
|
|
|
|
private:
|
|
|
|
void Accept ();
|
|
void HandleAccept (std::shared_ptr<i2p::stream::Stream> stream);
|
|
|
|
private:
|
|
|
|
boost::asio::ip::tcp::endpoint m_Endpoint;
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif
|