i2pd/I2PTunnel.h

128 lines
3.8 KiB
C
Raw Normal View History

2014-08-13 01:14:19 +00:00
#ifndef I2PTUNNEL_H__
#define I2PTUNNEL_H__
#include <inttypes.h>
#include <string>
2014-08-13 19:25:52 +00:00
#include <set>
2014-11-23 16:33:58 +00:00
#include <memory>
2014-08-13 01:14:19 +00:00
#include <boost/asio.hpp>
#include "Identity.h"
#include "Destination.h"
2014-08-13 01:14:19 +00:00
#include "Streaming.h"
namespace i2p
{
namespace client
2014-08-13 01:14:19 +00:00
{
2014-08-13 19:25:52 +00:00
const size_t I2P_TUNNEL_CONNECTION_BUFFER_SIZE = 8192;
const int I2P_TUNNEL_CONNECTION_MAX_IDLE = 3600; // in seconds
2014-10-15 16:07:06 +00:00
const int I2P_TUNNEL_DESTINATION_REQUEST_TIMEOUT = 10; // in seconds
const i2p::data::SigningKeyType I2P_TUNNEL_DEFAULT_KEY_TYPE = i2p::data::SIGNING_KEY_TYPE_ECDSA_SHA256_P256;
class I2PTunnel;
2014-11-24 03:23:17 +00:00
class I2PTunnelConnection: public std::enable_shared_from_this<I2PTunnelConnection>
2014-08-13 01:14:19 +00:00
{
public:
I2PTunnelConnection (I2PTunnel * owner, boost::asio::ip::tcp::socket * socket,
const i2p::data::LeaseSet * leaseSet); // to I2P
I2PTunnelConnection (I2PTunnel * owner, boost::asio::ip::tcp::socket * socket,
std::shared_ptr<i2p::stream::Stream> stream); // to I2P using simplified API :)
2014-11-23 16:33:58 +00:00
I2PTunnelConnection (I2PTunnel * owner, std::shared_ptr<i2p::stream::Stream> stream, boost::asio::ip::tcp::socket * socket,
const boost::asio::ip::tcp::endpoint& target, bool quiet = true); // from I2P
2014-08-13 01:14:19 +00:00
~I2PTunnelConnection ();
2014-08-13 19:25:52 +00:00
void I2PConnect (const uint8_t * msg = nullptr, size_t len = 0);
2014-11-24 03:23:17 +00:00
void Connect ();
2014-08-13 01:14:19 +00:00
private:
2014-08-13 19:25:52 +00: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 19:03:10 +00:00
void HandleConnect (const boost::system::error_code& ecode);
2014-08-13 19:25:52 +00:00
private:
uint8_t m_Buffer[I2P_TUNNEL_CONNECTION_BUFFER_SIZE], m_StreamBuffer[I2P_TUNNEL_CONNECTION_BUFFER_SIZE];
2014-08-13 01:14:19 +00:00
boost::asio::ip::tcp::socket * m_Socket;
2014-11-23 16:33:58 +00:00
std::shared_ptr<i2p::stream::Stream> m_Stream;
I2PTunnel * m_Owner;
2014-11-24 03:23:17 +00:00
boost::asio::ip::tcp::endpoint m_RemoteEndpoint;
bool m_IsQuiet; // don't send destination
};
class I2PTunnel
{
public:
I2PTunnel (ClientDestination * localDestination = nullptr);
virtual ~I2PTunnel () { ClearConnections (); };
2014-11-24 03:23:17 +00:00
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_LocalDestination->GetService (); };
private:
ClientDestination * m_LocalDestination;
2014-11-24 03:23:17 +00:00
std::set<std::shared_ptr<I2PTunnelConnection> > m_Connections;
2014-08-13 01:14:19 +00:00
};
class I2PClientTunnel: public I2PTunnel
2014-08-13 01:14:19 +00:00
{
public:
I2PClientTunnel (const std::string& destination, int port, ClientDestination * localDestination = nullptr);
2014-08-13 19:25:52 +00:00
~I2PClientTunnel ();
void Start ();
void Stop ();
2014-08-13 01:14:19 +00:00
private:
const i2p::data::IdentHash * GetIdentHash ();
2014-08-13 01:14:19 +00:00
void Accept ();
void HandleAccept (const boost::system::error_code& ecode, boost::asio::ip::tcp::socket * socket);
void HandleStreamRequestComplete (std::shared_ptr<i2p::stream::Stream> stream, boost::asio::ip::tcp::socket * socket);
2014-10-15 16:07:06 +00:00
2014-08-13 01:14:19 +00:00
private:
boost::asio::ip::tcp::acceptor m_Acceptor;
2014-10-15 16:07:06 +00:00
boost::asio::deadline_timer m_Timer;
2014-08-13 01:14:19 +00:00
std::string m_Destination;
2014-08-13 19:25:52 +00:00
const i2p::data::IdentHash * m_DestinationIdentHash;
2014-08-13 01:14:19 +00:00
};
2014-08-20 19:03:10 +00:00
class I2PServerTunnel: public I2PTunnel
2014-08-20 19:03:10 +00:00
{
public:
I2PServerTunnel (const std::string& address, int port, ClientDestination * localDestination);
2014-08-20 19:03:10 +00:00
void Start ();
void Stop ();
private:
void Accept ();
2014-11-23 16:33:58 +00:00
void HandleAccept (std::shared_ptr<i2p::stream::Stream> stream);
2014-08-20 19:03:10 +00:00
private:
boost::asio::ip::tcp::endpoint m_Endpoint;
};
2014-08-13 01:14:19 +00:00
}
}
#endif