i2pd/I2PTunnel.h

339 lines
11 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>
#include <tuple>
2014-11-23 19:33:58 +03:00
#include <memory>
#include <sstream>
2014-08-13 05:14:19 +04:00
#include <boost/asio.hpp>
#include "Identity.h"
#include "Destination.h"
2016-08-21 22:02:17 +03:00
#include "Datagram.h"
2014-08-13 05:14:19 +04:00
#include "Streaming.h"
#include "I2PService.h"
2014-08-13 05:14:19 +04:00
namespace i2p
{
namespace client
2014-08-13 05:14:19 +04:00
{
2016-12-23 15:27:34 +03:00
const size_t I2P_TUNNEL_CONNECTION_BUFFER_SIZE = 65536;
2014-08-13 23:25:52 +04:00
const int I2P_TUNNEL_CONNECTION_MAX_IDLE = 3600; // in seconds
2014-10-15 20:07:06 +04:00
const int I2P_TUNNEL_DESTINATION_REQUEST_TIMEOUT = 10; // in seconds
2016-01-11 21:48:18 +03:00
// for HTTP tunnels
const char X_I2P_DEST_HASH[] = "X-I2P-DestHash"; // hash in base64
2016-01-19 17:36:56 +03:00
const char X_I2P_DEST_B64[] = "X-I2P-DestB64"; // full address in base64
const char X_I2P_DEST_B32[] = "X-I2P-DestB32"; // .b32.i2p address
class I2PTunnelConnection: public I2PServiceHandler, public std::enable_shared_from_this<I2PTunnelConnection>
2014-08-13 05:14:19 +04:00
{
public:
I2PTunnelConnection (I2PService * owner, std::shared_ptr<boost::asio::ip::tcp::socket> socket,
2015-03-13 20:29:27 +03:00
std::shared_ptr<const i2p::data::LeaseSet> leaseSet, int port = 0); // to I2P
I2PTunnelConnection (I2PService * owner, std::shared_ptr<boost::asio::ip::tcp::socket> socket,
2015-06-02 20:03:22 +03:00
std::shared_ptr<i2p::stream::Stream> stream); // to I2P using simplified API
I2PTunnelConnection (I2PService * owner, std::shared_ptr<i2p::stream::Stream> stream, std::shared_ptr<boost::asio::ip::tcp::socket> socket,
const boost::asio::ip::tcp::endpoint& target, bool quiet = true); // from I2P
2014-08-13 05:14:19 +04:00
~I2PTunnelConnection ();
void I2PConnect (const uint8_t * msg = nullptr, size_t len = 0);
void Connect (bool isUniqueLocal = true);
2014-11-24 06:23:17 +03:00
2015-06-02 20:03:22 +03:00
protected:
2014-08-13 05:14:19 +04:00
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);
2015-06-02 20:03:22 +03:00
virtual void Write (const uint8_t * buf, size_t len); // can be overloaded
2014-08-13 23:25:52 +04:00
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];
std::shared_ptr<boost::asio::ip::tcp::socket> m_Socket;
2014-11-23 19:33:58 +03:00
std::shared_ptr<i2p::stream::Stream> m_Stream;
2014-11-24 06:23:17 +03:00
boost::asio::ip::tcp::endpoint m_RemoteEndpoint;
bool m_IsQuiet; // don't send destination
};
2015-06-02 20:03:22 +03:00
class I2PTunnelConnectionHTTP: public I2PTunnelConnection
{
public:
I2PTunnelConnectionHTTP (I2PService * owner, std::shared_ptr<i2p::stream::Stream> stream,
std::shared_ptr<boost::asio::ip::tcp::socket> socket,
const boost::asio::ip::tcp::endpoint& target, const std::string& host);
protected:
void Write (const uint8_t * buf, size_t len);
private:
2015-06-03 19:30:15 +03:00
std::string m_Host;
std::stringstream m_InHeader, m_OutHeader;
bool m_HeaderSent;
2016-01-11 21:48:18 +03:00
std::shared_ptr<const i2p::data::IdentityEx> m_From;
2015-06-02 20:03:22 +03:00
};
2016-02-22 22:33:21 +03:00
class I2PTunnelConnectionIRC: public I2PTunnelConnection
{
public:
I2PTunnelConnectionIRC (I2PService * owner, std::shared_ptr<i2p::stream::Stream> stream,
std::shared_ptr<boost::asio::ip::tcp::socket> socket,
2016-03-04 09:37:38 +03:00
const boost::asio::ip::tcp::endpoint& target, const std::string& m_WebircPass);
2016-02-22 22:33:21 +03:00
protected:
void Write (const uint8_t * buf, size_t len);
private:
2016-03-05 04:35:53 +03:00
2016-02-22 22:33:21 +03:00
std::shared_ptr<const i2p::data::IdentityEx> m_From;
std::stringstream m_OutPacket, m_InPacket;
2016-03-05 04:35:53 +03:00
bool m_NeedsWebIrc;
std::string m_WebircPass;
2016-02-22 22:33:21 +03:00
};
2015-01-08 05:49:35 +03:00
class I2PClientTunnel: public TCPIPAcceptor
2014-08-13 05:14:19 +04:00
{
2015-01-08 05:49:35 +03:00
protected:
// Implements TCPIPAcceptor
std::shared_ptr<I2PServiceHandler> CreateHandler(std::shared_ptr<boost::asio::ip::tcp::socket> socket);
2015-01-08 05:49:35 +03:00
2014-08-13 05:14:19 +04:00
public:
2016-01-14 04:21:53 +03:00
I2PClientTunnel (const std::string& name, const std::string& destination,
const std::string& address, int port, std::shared_ptr<ClientDestination> localDestination, int destinationPort = 0);
2015-01-08 05:49:35 +03:00
~I2PClientTunnel () {}
2014-08-13 23:25:52 +04:00
void Start ();
void Stop ();
2016-01-14 04:21:53 +03:00
const char* GetName() { return m_Name.c_str (); }
2014-08-13 05:14:19 +04:00
private:
const i2p::data::IdentHash * GetIdentHash ();
2014-08-13 05:14:19 +04:00
2016-01-14 04:21:53 +03:00
private:
std::string m_Name, m_Destination;
2014-08-13 23:25:52 +04:00
const i2p::data::IdentHash * m_DestinationIdentHash;
2015-03-13 20:29:27 +03:00
int m_DestinationPort;
2016-08-21 22:02:17 +03:00
};
2014-08-20 23:03:10 +04:00
2016-09-03 16:38:53 +03:00
/** 2 minute timeout for udp sessions */
const uint64_t I2P_UDP_SESSION_TIMEOUT = 1000 * 60 * 2;
/** max size for i2p udp */
const size_t I2P_UDP_MAX_MTU = i2p::datagram::MAX_DATAGRAM_SIZE;
2016-09-03 20:58:34 +03:00
2016-09-03 16:38:53 +03:00
struct UDPSession
{
i2p::datagram::DatagramDestination * m_Destination;
boost::asio::ip::udp::socket IPSocket;
i2p::data::IdentHash Identity;
boost::asio::ip::udp::endpoint FromEndpoint;
boost::asio::ip::udp::endpoint SendEndpoint;
uint64_t LastActivity;
uint16_t LocalPort;
uint16_t RemotePort;
uint8_t m_Buffer[I2P_UDP_MAX_MTU];
UDPSession(boost::asio::ip::udp::endpoint localEndpoint,
const std::shared_ptr<i2p::client::ClientDestination> & localDestination,
2016-10-09 17:55:55 +03:00
boost::asio::ip::udp::endpoint remote, const i2p::data::IdentHash * ident,
2016-09-03 16:38:53 +03:00
uint16_t ourPort, uint16_t theirPort);
void HandleReceived(const boost::system::error_code & ecode, std::size_t len);
void Receive();
};
2016-09-04 00:53:46 +03:00
/** read only info about a datagram session */
struct DatagramSessionInfo
{
/** the name of this forward */
std::string Name;
/** ident hash of local destination */
std::shared_ptr<const i2p::data::IdentHash> LocalIdent;
/** ident hash of remote destination */
std::shared_ptr<const i2p::data::IdentHash> RemoteIdent;
/** ident hash of IBGW in use currently in this session or nullptr if none is set */
std::shared_ptr<const i2p::data::IdentHash> CurrentIBGW;
/** ident hash of OBEP in use for this session or nullptr if none is set */
std::shared_ptr<const i2p::data::IdentHash> CurrentOBEP;
/** i2p router's udp endpoint */
boost::asio::ip::udp::endpoint LocalEndpoint;
/** client's udp endpoint */
boost::asio::ip::udp::endpoint RemoteEndpoint;
/** how long has this converstation been idle in ms */
uint64_t idle;
};
2016-12-23 15:27:34 +03:00
typedef std::shared_ptr<UDPSession> UDPSessionPtr;
2016-09-04 00:53:46 +03:00
2016-09-03 16:38:53 +03:00
/** server side udp tunnel, many i2p inbound to 1 ip outbound */
class I2PUDPServerTunnel
2016-09-04 00:53:46 +03:00
{
2016-09-03 16:38:53 +03:00
public:
I2PUDPServerTunnel(const std::string & name,
std::shared_ptr<i2p::client::ClientDestination> localDestination,
boost::asio::ip::address localAddress,
2016-09-03 16:38:53 +03:00
boost::asio::ip::udp::endpoint forwardTo, uint16_t port);
~I2PUDPServerTunnel();
/** expire stale udp conversations */
void ExpireStale(const uint64_t delta=I2P_UDP_SESSION_TIMEOUT);
void Start();
const char * GetName() const { return m_Name.c_str(); }
2016-09-04 00:53:46 +03:00
std::vector<std::shared_ptr<DatagramSessionInfo> > GetSessions();
2016-09-12 19:05:01 +03:00
std::shared_ptr<ClientDestination> GetLocalDestination () const { return m_LocalDest; }
void SetUniqueLocal(bool isUniqueLocal = true) { m_IsUniqueLocal = isUniqueLocal; }
2016-09-03 16:38:53 +03:00
private:
2016-09-12 19:05:01 +03:00
2016-09-03 16:38:53 +03:00
void HandleRecvFromI2P(const i2p::data::IdentityEx& from, uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
2016-12-23 15:27:34 +03:00
UDPSessionPtr ObtainUDPSession(const i2p::data::IdentityEx& from, uint16_t localPort, uint16_t remotePort);
2016-09-12 19:05:01 +03:00
2016-09-03 16:38:53 +03:00
private:
bool m_IsUniqueLocal;
2016-09-03 16:38:53 +03:00
const std::string m_Name;
boost::asio::ip::address m_LocalAddress;
boost::asio::ip::udp::endpoint m_RemoteEndpoint;
std::mutex m_SessionsMutex;
2016-12-23 15:27:34 +03:00
std::vector<UDPSessionPtr> m_Sessions;
2016-09-03 16:38:53 +03:00
std::shared_ptr<i2p::client::ClientDestination> m_LocalDest;
};
2016-09-04 00:53:46 +03:00
class I2PUDPClientTunnel
2016-09-03 16:38:53 +03:00
{
public:
I2PUDPClientTunnel(const std::string & name, const std::string &remoteDest,
boost::asio::ip::udp::endpoint localEndpoint, std::shared_ptr<i2p::client::ClientDestination> localDestination,
uint16_t remotePort);
~I2PUDPClientTunnel();
void Start();
const char * GetName() const { return m_Name.c_str(); }
2016-09-04 00:53:46 +03:00
std::vector<std::shared_ptr<DatagramSessionInfo> > GetSessions();
2016-09-12 19:05:01 +03:00
2016-09-03 20:58:34 +03:00
bool IsLocalDestination(const i2p::data::IdentHash & destination) const { return destination == m_LocalDest->GetIdentHash(); }
2016-09-12 19:05:01 +03:00
std::shared_ptr<ClientDestination> GetLocalDestination () const { return m_LocalDest; }
void ExpireStale(const uint64_t delta=I2P_UDP_SESSION_TIMEOUT);
2016-09-12 19:05:01 +03:00
2016-09-03 16:38:53 +03:00
private:
2016-11-17 18:43:27 +03:00
typedef std::pair<boost::asio::ip::udp::endpoint, uint64_t> UDPConvo;
void RecvFromLocal();
void HandleRecvFromLocal(const boost::system::error_code & e, std::size_t transferred);
2016-09-03 16:38:53 +03:00
void HandleRecvFromI2P(const i2p::data::IdentityEx& from, uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
void TryResolving();
const std::string m_Name;
std::mutex m_SessionsMutex;
std::map<uint16_t, UDPConvo > m_Sessions; // maps i2p port -> local udp convo
2016-09-03 16:38:53 +03:00
const std::string m_RemoteDest;
std::shared_ptr<i2p::client::ClientDestination> m_LocalDest;
const boost::asio::ip::udp::endpoint m_LocalEndpoint;
i2p::data::IdentHash * m_RemoteIdent;
std::thread * m_ResolveThread;
boost::asio::ip::udp::socket m_LocalSocket;
boost::asio::ip::udp::endpoint m_RecvEndpoint;
uint8_t m_RecvBuff[I2P_UDP_MAX_MTU];
2016-09-03 16:38:53 +03:00
uint16_t RemotePort;
bool m_cancel_resolve;
};
class I2PServerTunnel: public I2PService
2014-08-20 23:03:10 +04:00
{
public:
2016-01-14 04:21:53 +03:00
I2PServerTunnel (const std::string& name, const std::string& address, int port,
std::shared_ptr<ClientDestination> localDestination, int inport = 0, bool gzip = true);
2014-08-20 23:03:10 +04:00
void Start ();
void Stop ();
2015-03-16 21:52:42 +03:00
void SetAccessList (const std::set<i2p::data::IdentHash>& accessList);
void SetUniqueLocal (bool isUniqueLocal) { m_IsUniqueLocal = isUniqueLocal; }
2015-06-02 20:03:22 +03:00
const std::string& GetAddress() const { return m_Address; }
int GetPort () const { return m_Port; };
uint16_t GetLocalPort () const { return m_PortDestination->GetLocalPort (); };
2015-06-02 20:03:22 +03:00
const boost::asio::ip::tcp::endpoint& GetEndpoint () const { return m_Endpoint; }
2016-01-14 04:21:53 +03:00
const char* GetName() { return m_Name.c_str (); }
2016-07-28 18:16:29 +03:00
void SetMaxConnsPerMinute(const uint32_t conns) { m_PortDestination->SetMaxConnsPerMinute(conns); }
2016-08-21 22:02:17 +03:00
2016-07-28 18:16:29 +03:00
private:
2014-08-20 23:03:10 +04:00
void HandleResolve (const boost::system::error_code& ecode, boost::asio::ip::tcp::resolver::iterator it,
std::shared_ptr<boost::asio::ip::tcp::resolver> resolver);
2014-08-20 23:03:10 +04:00
void Accept ();
2014-11-23 19:33:58 +03:00
void HandleAccept (std::shared_ptr<i2p::stream::Stream> stream);
2015-06-02 20:03:22 +03:00
virtual void CreateI2PConnection (std::shared_ptr<i2p::stream::Stream> stream);
2014-08-20 23:03:10 +04:00
private:
bool m_IsUniqueLocal;
2016-01-14 04:21:53 +03:00
std::string m_Name, m_Address;
2015-06-02 20:03:22 +03:00
int m_Port;
boost::asio::ip::tcp::endpoint m_Endpoint;
2015-03-16 21:52:42 +03:00
std::shared_ptr<i2p::stream::StreamingDestination> m_PortDestination;
std::set<i2p::data::IdentHash> m_AccessList;
2016-08-21 22:02:17 +03:00
bool m_IsAccessList;
2014-08-20 23:03:10 +04:00
};
2015-05-20 23:00:09 +03:00
class I2PServerTunnelHTTP: public I2PServerTunnel
{
public:
2016-01-14 04:21:53 +03:00
I2PServerTunnelHTTP (const std::string& name, const std::string& address, int port,
2016-02-26 04:32:05 +03:00
std::shared_ptr<ClientDestination> localDestination, const std::string& host,
int inport = 0, bool gzip = true);
2015-06-02 20:03:22 +03:00
private:
2016-02-26 04:32:05 +03:00
void CreateI2PConnection (std::shared_ptr<i2p::stream::Stream> stream);
private:
std::string m_Host;
2015-05-20 23:00:09 +03:00
};
2016-02-22 22:33:21 +03:00
class I2PServerTunnelIRC: public I2PServerTunnel
{
public:
I2PServerTunnelIRC (const std::string& name, const std::string& address, int port,
2016-03-04 09:37:38 +03:00
std::shared_ptr<ClientDestination> localDestination, const std::string& webircpass,
int inport = 0, bool gzip = true);
private:
void CreateI2PConnection (std::shared_ptr<i2p::stream::Stream> stream);
2016-02-22 22:33:21 +03:00
private:
2016-03-04 09:37:38 +03:00
std::string m_WebircPass;
2016-02-22 22:33:21 +03:00
};
}
2014-08-13 05:14:19 +04:00
}
#endif