i2pd/I2PTunnel.h

96 lines
2.3 KiB
C
Raw Normal View History

2014-08-13 05:14:19 +04:00
#ifndef I2PTUNNEL_H__
#define I2PTUNNEL_H__
#include <inttypes.h>
#include <string>
2014-08-13 23:25:52 +04:00
#include <set>
2014-08-13 05:14:19 +04:00
#include <boost/asio.hpp>
#include "Identity.h"
#include "Streaming.h"
namespace i2p
{
namespace stream
{
2014-08-13 23:25:52 +04:00
const size_t I2P_TUNNEL_CONNECTION_BUFFER_SIZE = 8192;
const int I2P_TUNNEL_CONNECTION_MAX_IDLE = 3600; // in seconds
2014-08-13 05:14:19 +04:00
class I2PTunnelConnection
{
public:
I2PTunnelConnection (boost::asio::ip::tcp::socket * socket,
const i2p::data::LeaseSet * leaseSet);
2014-08-20 23:03:10 +04:00
I2PTunnelConnection (Stream * stream, boost::asio::ip::tcp::socket * socket,
const boost::asio::ip::tcp::endpoint& target);
2014-08-13 05:14:19 +04:00
~I2PTunnelConnection ();
2014-08-13 23:25:52 +04:00
2014-08-13 05:14:19 +04:00
private:
2014-08-13 23:25:52 +04:00
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);
2014-08-20 23:03:10 +04:00
void HandleConnect (const boost::system::error_code& ecode);
2014-08-13 23:25:52 +04:00
private:
uint8_t m_Buffer[I2P_TUNNEL_CONNECTION_BUFFER_SIZE], m_StreamBuffer[I2P_TUNNEL_CONNECTION_BUFFER_SIZE];
2014-08-13 05:14:19 +04:00
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);
2014-08-13 23:25:52 +04:00
~I2PClientTunnel ();
void Start ();
void Stop ();
2014-08-13 05:14:19 +04:00
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;
2014-08-13 23:25:52 +04:00
const i2p::data::IdentHash * m_DestinationIdentHash;
2014-08-13 05:14:19 +04:00
const i2p::data::LeaseSet * m_RemoteLeaseSet;
2014-08-13 23:25:52 +04:00
std::set<I2PTunnelConnection *> m_Connections;
2014-08-13 05:14:19 +04:00
};
2014-08-20 23:03:10 +04:00
class I2PServerTunnel
{
public:
I2PServerTunnel (boost::asio::io_service& service, const std::string& address, int port,
const i2p::data::IdentHash& localDestination);
void Start ();
void Stop ();
private:
void Accept ();
void HandleAccept (Stream * stream);
private:
boost::asio::io_service& m_Service;
StreamingDestination * m_LocalDestination;
boost::asio::ip::tcp::endpoint m_Endpoint;
};
2014-08-13 05:14:19 +04:00
}
}
#endif