i2pd/NTCPSession.h

178 lines
4.4 KiB
C
Raw Normal View History

2013-09-10 01:35:46 +00:00
#ifndef NTCP_SESSION_H__
#define NTCP_SESSION_H__
#include <inttypes.h>
2013-11-29 12:52:09 +00:00
#include <list>
2013-09-10 01:35:46 +00:00
#include <boost/asio.hpp>
#include <cryptopp/modes.h>
#include <cryptopp/aes.h>
#include <cryptopp/adler32.h>
#include "RouterInfo.h"
2013-10-23 02:43:29 +00:00
#include "I2NPProtocol.h"
2013-09-10 01:35:46 +00:00
namespace i2p
{
namespace ntcp
{
#pragma pack(1)
struct NTCPPhase1
{
uint8_t pubKey[256];
uint8_t HXxorHI[32];
};
struct NTCPPhase2
{
uint8_t pubKey[256];
struct
{
uint8_t hxy[32];
uint32_t timestamp;
uint8_t filler[12];
} encrypted;
};
struct NTCPPhase3
{
uint16_t size;
i2p::data::Identity ident;
2013-09-10 01:35:46 +00:00
uint32_t timestamp;
uint8_t padding[15];
uint8_t signature[40];
};
struct NTCPPhase4
{
uint8_t signature[40];
uint8_t padding[8];
};
struct SignedData // used for signature in Phase3 and Phase4
{
uint8_t x[256];
uint8_t y[256];
uint8_t ident[32];
uint32_t tsA;
uint32_t tsB;
};
#pragma pack()
2013-11-29 12:52:09 +00:00
const int TERMINATION_TIMEOUT = 60; // 1 minute
2013-09-10 01:35:46 +00:00
class NTCPSession
{
public:
2013-12-22 16:29:57 +00:00
NTCPSession (boost::asio::io_service& service, i2p::data::RouterInfo& in_RemoteRouterInfo);
2013-09-10 01:35:46 +00:00
virtual ~NTCPSession () {};
2013-11-29 12:52:09 +00:00
boost::asio::ip::tcp::socket& GetSocket () { return m_Socket; };
2013-09-10 01:35:46 +00:00
bool IsEstablished () const { return m_IsEstablished; };
2013-12-12 03:06:04 +00:00
i2p::data::RouterInfo& GetRemoteRouterInfo () { return m_RemoteRouterInfo; };
2013-09-10 01:35:46 +00:00
void ClientLogin ();
void ServerLogin ();
2013-10-23 02:43:29 +00:00
void SendI2NPMessage (I2NPMessage * msg);
protected:
2013-09-10 01:35:46 +00:00
void Terminate ();
2013-11-29 12:52:09 +00:00
virtual void Connected ();
void SendTimeSyncMessage ();
void SetIsEstablished (bool isEstablished) { m_IsEstablished = isEstablished; }
2013-09-10 01:35:46 +00:00
private:
void CreateAESKey (uint8_t * pubKey, uint8_t * aesKey);
// client
void SendPhase3 ();
void HandlePhase1Sent (const boost::system::error_code& ecode, std::size_t bytes_transferred);
void HandlePhase2Received (const boost::system::error_code& ecode, std::size_t bytes_transferred);
void HandlePhase3Sent (const boost::system::error_code& ecode, std::size_t bytes_transferred, uint32_t tsA);
void HandlePhase4Received (const boost::system::error_code& ecode, std::size_t bytes_transferred, uint32_t tsA);
//server
void SendPhase2 ();
void SendPhase4 (uint32_t tsB);
void HandlePhase1Received (const boost::system::error_code& ecode, std::size_t bytes_transferred);
void HandlePhase2Sent (const boost::system::error_code& ecode, std::size_t bytes_transferred, uint32_t tsB);
void HandlePhase3Received (const boost::system::error_code& ecode, std::size_t bytes_transferred, uint32_t tsB);
void HandlePhase4Sent (const boost::system::error_code& ecode, std::size_t bytes_transferred);
// common
void Receive ();
void HandleReceived (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2013-10-27 15:23:15 +00:00
void DecryptNextBlock (const uint8_t * encrypted);
void Send (i2p::I2NPMessage * msg);
void HandleSent (const boost::system::error_code& ecode, std::size_t bytes_transferred, i2p::I2NPMessage * msg);
2013-09-10 01:35:46 +00:00
2013-11-29 12:52:09 +00:00
// timer
void ScheduleTermination ();
void HandleTerminationTimer (const boost::system::error_code& ecode);
2013-09-10 01:35:46 +00:00
private:
2013-11-29 12:52:09 +00:00
boost::asio::ip::tcp::socket m_Socket;
boost::asio::deadline_timer m_TerminationTimer;
2013-09-10 01:35:46 +00:00
bool m_IsEstablished;
CryptoPP::CBC_Mode<CryptoPP::AES>::Decryption m_Decryption;
CryptoPP::CBC_Mode<CryptoPP::AES>::Encryption m_Encryption;
CryptoPP::Adler32 m_Adler;
2013-12-22 16:29:57 +00:00
i2p::data::RouterInfo& m_RemoteRouterInfo;
2013-09-10 01:35:46 +00:00
NTCPPhase1 m_Phase1;
NTCPPhase2 m_Phase2;
NTCPPhase3 m_Phase3;
NTCPPhase4 m_Phase4;
uint8_t m_ReceiveBuffer[i2p::NTCP_MAX_MESSAGE_SIZE*2], m_TimeSyncBuffer[16];
2013-09-10 01:35:46 +00:00
int m_ReceiveBufferOffset;
2013-11-29 12:52:09 +00:00
i2p::I2NPMessage * m_NextMessage;
std::list<i2p::I2NPMessage *> m_DelayedMessages;
2013-10-27 15:23:15 +00:00
size_t m_NextMessageOffset;
2013-09-10 01:35:46 +00:00
};
class NTCPClient: public NTCPSession
{
public:
2014-01-21 23:01:11 +00:00
NTCPClient (boost::asio::io_service& service, const boost::asio::ip::address& address, int port, i2p::data::RouterInfo& in_RouterInfo);
2013-09-10 01:35:46 +00:00
private:
void Connect ();
void HandleConnect (const boost::system::error_code& ecode);
private:
boost::asio::ip::tcp::endpoint m_Endpoint;
};
class NTCPServerConnection: public NTCPSession
{
public:
2013-12-22 16:29:57 +00:00
NTCPServerConnection (boost::asio::io_service& service):
NTCPSession (service, m_DummyRemoteRouterInfo) {};
2013-09-10 01:35:46 +00:00
2013-11-29 12:52:09 +00:00
protected:
2013-09-10 01:35:46 +00:00
2013-11-29 12:52:09 +00:00
virtual void Connected ();
2013-12-22 16:29:57 +00:00
private:
i2p::data::RouterInfo m_DummyRemoteRouterInfo;
2013-09-10 01:35:46 +00:00
};
}
}
#endif