i2pd/NTCPSession.h

184 lines
5.7 KiB
C
Raw Normal View History

2013-09-10 05:35:46 +04:00
#ifndef NTCP_SESSION_H__
#define NTCP_SESSION_H__
#include <inttypes.h>
2015-01-12 01:41:56 +03:00
#include <map>
2014-11-26 00:30:15 +03:00
#include <memory>
2015-01-12 01:41:56 +03:00
#include <thread>
2015-01-12 20:15:54 +03:00
#include <mutex>
2015-01-12 01:41:56 +03:00
#include <boost/asio.hpp>
2013-09-10 05:35:46 +04:00
#include <cryptopp/modes.h>
#include <cryptopp/aes.h>
#include <cryptopp/adler32.h>
2014-05-07 06:30:09 +04:00
#include "aes.h"
2014-04-04 21:30:13 +04:00
#include "Identity.h"
2013-09-10 05:35:46 +04:00
#include "RouterInfo.h"
2013-10-23 06:43:29 +04:00
#include "I2NPProtocol.h"
2014-10-21 00:09:59 +04:00
#include "TransportSession.h"
2013-09-10 05:35:46 +04:00
namespace i2p
{
namespace transport
2013-09-10 05:35:46 +04:00
{
#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;
};
#pragma pack()
const size_t NTCP_MAX_MESSAGE_SIZE = 16384;
2015-01-23 06:00:41 +03:00
const size_t NTCP_BUFFER_SIZE = 4160; // fits 4 tunnel messages (4*1028)
2014-04-08 05:40:28 +04:00
const int NTCP_TERMINATION_TIMEOUT = 120; // 2 minutes
2014-11-25 22:29:06 +03:00
const size_t NTCP_DEFAULT_PHASE3_SIZE = 2/*size*/ + i2p::data::DEFAULT_IDENTITY_SIZE/*387*/ + 4/*ts*/ + 15/*padding*/ + 40/*signature*/; // 448
2014-10-20 23:19:56 +04:00
2015-01-12 01:41:56 +03:00
class NTCPServer;
2014-11-26 00:30:15 +03:00
class NTCPSession: public TransportSession, public std::enable_shared_from_this<NTCPSession>
2013-09-10 05:35:46 +04:00
{
public:
2015-01-12 01:41:56 +03:00
NTCPSession (NTCPServer& server, std::shared_ptr<const i2p::data::RouterInfo> in_RemoteRouter = nullptr);
~NTCPSession ();
2014-11-25 23:16:03 +03:00
void Terminate ();
2013-09-10 05:35:46 +04:00
2013-11-29 16:52:09 +04:00
boost::asio::ip::tcp::socket& GetSocket () { return m_Socket; };
2013-09-10 05:35:46 +04:00
bool IsEstablished () const { return m_IsEstablished; };
void ClientLogin ();
void ServerLogin ();
2013-10-23 06:43:29 +04:00
void SendI2NPMessage (I2NPMessage * msg);
2015-01-21 05:05:57 +03:00
void SendI2NPMessages (const std::vector<I2NPMessage *>& msgs);
size_t GetNumSentBytes () const { return m_NumSentBytes; };
size_t GetNumReceivedBytes () const { return m_NumReceivedBytes; };
2013-10-23 06:43:29 +04:00
2015-01-21 20:08:15 +03:00
private:
2013-10-23 06:43:29 +04:00
2015-01-12 05:00:38 +03:00
void PostI2NPMessage (I2NPMessage * msg);
2015-01-21 05:05:57 +03:00
void PostI2NPMessages (std::vector<I2NPMessage *> msgs);
2014-11-25 23:16:03 +03:00
void Connected ();
2013-11-29 16:52:09 +04:00
void SendTimeSyncMessage ();
void SetIsEstablished (bool isEstablished) { m_IsEstablished = isEstablished; }
2013-09-10 05:35:46 +04:00
2014-11-02 04:53:45 +03:00
void CreateAESKey (uint8_t * pubKey, i2p::crypto::AESKey& key);
2013-09-10 05:35:46 +04:00
// 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 ();
2014-11-25 18:35:35 +03:00
void SendPhase4 (uint32_t tsA, uint32_t tsB);
2013-09-10 05:35:46 +04:00
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);
2014-11-25 20:33:51 +03:00
void HandlePhase3ExtraReceived (const boost::system::error_code& ecode, std::size_t bytes_transferred, uint32_t tsB, size_t paddingLen);
void HandlePhase3 (uint32_t tsB, size_t paddingLen);
2013-09-10 05:35:46 +04:00
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);
2014-09-18 19:11:51 +04:00
bool DecryptNextBlock (const uint8_t * encrypted);
2013-10-27 19:23:15 +04:00
void Send (i2p::I2NPMessage * msg);
boost::asio::const_buffers_1 CreateMsgBuffer (I2NPMessage * msg);
void Send (const std::vector<I2NPMessage *>& msgs);
2015-01-21 20:08:15 +03:00
void HandleSent (const boost::system::error_code& ecode, std::size_t bytes_transferred, std::vector<I2NPMessage *> msgs);
2013-11-29 16:52:09 +04:00
// timer
void ScheduleTermination ();
void HandleTerminationTimer (const boost::system::error_code& ecode);
2013-09-10 05:35:46 +04:00
private:
2015-01-12 01:41:56 +03:00
NTCPServer& m_Server;
2013-11-29 16:52:09 +04:00
boost::asio::ip::tcp::socket m_Socket;
boost::asio::deadline_timer m_TerminationTimer;
2013-09-10 05:35:46 +04:00
bool m_IsEstablished;
2014-05-07 06:30:09 +04:00
i2p::crypto::CBCDecryption m_Decryption;
i2p::crypto::CBCEncryption m_Encryption;
2013-09-10 05:35:46 +04:00
CryptoPP::Adler32 m_Adler;
2014-09-12 06:15:20 +04:00
struct Establisher
{
NTCPPhase1 phase1;
NTCPPhase2 phase2;
} * m_Establisher;
2013-09-10 05:35:46 +04:00
2014-11-25 18:14:18 +03:00
i2p::crypto::AESAlignedBuffer<NTCP_BUFFER_SIZE + 16> m_ReceiveBuffer;
i2p::crypto::AESAlignedBuffer<16> m_TimeSyncBuffer;
2013-09-10 05:35:46 +04:00
int m_ReceiveBufferOffset;
2013-11-29 16:52:09 +04:00
i2p::I2NPMessage * m_NextMessage;
2013-10-27 19:23:15 +04:00
size_t m_NextMessageOffset;
2015-01-23 06:00:41 +03:00
i2p::I2NPMessagesHandler m_Handler;
size_t m_NumSentBytes, m_NumReceivedBytes;
2013-09-10 05:35:46 +04:00
};
2015-01-12 01:41:56 +03:00
// TODO: move to NTCP.h/.cpp
class NTCPServer
{
public:
NTCPServer (int port);
~NTCPServer ();
void Start ();
void Stop ();
void AddNTCPSession (std::shared_ptr<NTCPSession> session);
void RemoveNTCPSession (std::shared_ptr<NTCPSession> session);
std::shared_ptr<NTCPSession> FindNTCPSession (const i2p::data::IdentHash& ident);
void Connect (const boost::asio::ip::address& address, int port, std::shared_ptr<NTCPSession> conn);
boost::asio::io_service& GetService () { return m_Service; };
private:
void Run ();
void HandleAccept (std::shared_ptr<NTCPSession> conn, const boost::system::error_code& error);
void HandleAcceptV6 (std::shared_ptr<NTCPSession> conn, const boost::system::error_code& error);
void HandleConnect (const boost::system::error_code& ecode, std::shared_ptr<NTCPSession> conn);
private:
bool m_IsRunning;
std::thread * m_Thread;
boost::asio::io_service m_Service;
boost::asio::io_service::work m_Work;
boost::asio::ip::tcp::acceptor * m_NTCPAcceptor, * m_NTCPV6Acceptor;
2015-01-12 20:15:54 +03:00
std::mutex m_NTCPSessionsMutex;
2015-01-12 01:41:56 +03:00
std::map<i2p::data::IdentHash, std::shared_ptr<NTCPSession> > m_NTCPSessions;
public:
// for HTTP/I2PControl
const decltype(m_NTCPSessions)& GetNTCPSessions () const { return m_NTCPSessions; };
};
2013-09-10 05:35:46 +04:00
}
}
#endif