i2pd/SSU.h

189 lines
6.2 KiB
C
Raw Normal View History

2014-01-24 01:10:33 +04:00
#ifndef SSU_H__
#define SSU_H__
#include <inttypes.h>
2014-01-25 01:30:07 +04:00
#include <map>
2014-02-10 03:28:34 +04:00
#include <list>
2014-04-09 22:58:30 +04:00
#include <set>
2014-01-24 01:10:33 +04:00
#include <boost/asio.hpp>
2014-01-28 01:52:17 +04:00
#include <cryptopp/modes.h>
#include <cryptopp/aes.h>
#include "I2PEndian.h"
2014-04-04 22:56:46 +04:00
#include "Identity.h"
2014-01-29 01:49:54 +04:00
#include "RouterInfo.h"
2014-01-30 01:49:53 +04:00
#include "I2NPProtocol.h"
2014-01-24 01:10:33 +04:00
namespace i2p
{
namespace ssu
{
2014-01-28 01:52:17 +04:00
#pragma pack(1)
struct SSUHeader
{
uint8_t mac[16];
uint8_t iv[16];
uint8_t flag;
uint32_t time;
2014-04-08 00:41:29 +04:00
uint8_t GetPayloadType () const { return flag >> 4; };
2014-01-28 01:52:17 +04:00
};
#pragma pack()
2014-04-07 23:31:38 +04:00
const size_t SSU_MTU = 1484;
2014-03-31 06:55:03 +04:00
const int SSU_CONNECT_TIMEOUT = 5; // 5 seconds
const int SSU_TERMINATION_TIMEOUT = 330; // 5.5 minutes
2014-01-24 01:10:33 +04:00
2014-01-30 17:28:11 +04:00
// payload types (4 bits)
2014-01-25 01:30:07 +04:00
const uint8_t PAYLOAD_TYPE_SESSION_REQUEST = 0;
const uint8_t PAYLOAD_TYPE_SESSION_CREATED = 1;
const uint8_t PAYLOAD_TYPE_SESSION_CONFIRMED = 2;
const uint8_t PAYLOAD_TYPE_RELAY_REQUEST = 3;
const uint8_t PAYLOAD_TYPE_RELAY_RESPONSE = 4;
const uint8_t PAYLOAD_TYPE_RELAY_INTRO = 5;
const uint8_t PAYLOAD_TYPE_DATA = 6;
2014-04-07 23:31:38 +04:00
const uint8_t PAYLOAD_TYPE_PEER_TEST = 7;
2014-02-09 06:06:40 +04:00
const uint8_t PAYLOAD_TYPE_SESSION_DESTROYED = 8;
2014-01-25 01:30:07 +04:00
2014-02-08 00:47:10 +04:00
// data flags
const uint8_t DATA_FLAG_EXTENDED_DATA_INCLUDED = 0x02;
const uint8_t DATA_FLAG_WANT_REPLY = 0x04;
const uint8_t DATA_FLAG_REQUEST_PREVIOUS_ACKS = 0x08;
const uint8_t DATA_FLAG_EXPLICIT_CONGESTION_NOTIFICATION = 0x10;
const uint8_t DATA_FLAG_ACK_BITFIELDS_INCLUDED = 0x40;
const uint8_t DATA_FLAG_EXPLICIT_ACKS_INCLUDED = 0x80;
2014-01-25 01:30:07 +04:00
enum SessionState
{
eSessionStateUnknown,
eSessionStateRequestSent,
eSessionStateRequestReceived,
eSessionStateCreatedSent,
eSessionStateCreatedReceived,
eSessionStateConfirmedSent,
eSessionStateConfirmedReceived,
eSessionStateRelayRequestSent,
eSessionStateRelayRequestReceived,
eSessionStateIntroduced,
eSessionStateEstablished,
eSessionStateFailed
2014-04-14 00:59:54 +04:00
};
2014-01-25 01:30:07 +04:00
2014-01-29 01:49:54 +04:00
class SSUServer;
2014-01-25 01:30:07 +04:00
class SSUSession
{
public:
2014-04-02 18:49:16 +04:00
SSUSession (SSUServer& server, boost::asio::ip::udp::endpoint& remoteEndpoint,
2014-04-08 03:28:06 +04:00
const i2p::data::RouterInfo * router = nullptr, bool peerTest = false);
2014-01-30 01:49:53 +04:00
void ProcessNextMessage (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
2014-03-31 06:55:03 +04:00
~SSUSession ();
2014-01-30 01:49:53 +04:00
void Connect ();
2014-04-09 19:58:57 +04:00
void Introduce (uint32_t iTag, const uint8_t * iKey);
void WaitForIntroduction ();
2014-02-09 06:06:40 +04:00
void Close ();
boost::asio::ip::udp::endpoint& GetRemoteEndpoint () { return m_RemoteEndpoint; };
2014-03-12 15:37:43 +04:00
const i2p::data::RouterInfo * GetRemoteRouter () const { return m_RemoteRouter; };
2014-01-30 01:49:53 +04:00
void SendI2NPMessage (I2NPMessage * msg);
2014-04-08 00:19:33 +04:00
void SendPeerTest (); // Alice
2014-01-28 01:52:17 +04:00
private:
2014-04-14 00:59:54 +04:00
void CreateAESandMacKey (const uint8_t * pubKey, uint8_t * aesKey, uint8_t * macKey);
2014-01-30 01:49:53 +04:00
2014-04-07 23:31:38 +04:00
void ProcessMessage (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint); // call for established session
2014-01-30 01:49:53 +04:00
void ProcessSessionRequest (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
void SendSessionRequest ();
2014-04-09 19:58:57 +04:00
void SendRelayRequest (uint32_t iTag, const uint8_t * iKey);
2014-01-30 01:49:53 +04:00
void ProcessSessionCreated (uint8_t * buf, size_t len);
2014-01-30 23:03:11 +04:00
void SendSessionCreated (const uint8_t * x);
2014-02-04 23:20:58 +04:00
void ProcessSessionConfirmed (uint8_t * buf, size_t len);
2014-04-08 23:35:08 +04:00
void SendSessionConfirmed (const uint8_t * y, const uint8_t * ourAddress);
2014-02-22 01:13:36 +04:00
void ProcessRelayResponse (uint8_t * buf, size_t len);
2014-04-08 22:03:15 +04:00
void ProcessRelayIntro (uint8_t * buf, size_t len);
2014-02-10 03:28:34 +04:00
void Established ();
void Failed ();
2014-03-31 06:55:03 +04:00
void HandleConnectTimer (const boost::system::error_code& ecode);
2014-04-07 23:31:38 +04:00
void ProcessPeerTest (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
void SendPeerTest (uint32_t nonce, uint32_t address, uint16_t port, uint8_t * introKey); // Charlie to Alice
void ProcessData (uint8_t * buf, size_t len);
2014-02-08 00:47:10 +04:00
void SendMsgAck (uint32_t msgID);
2014-02-09 06:06:40 +04:00
void SendSesionDestroyed ();
2014-02-10 03:28:34 +04:00
void Send (i2p::I2NPMessage * msg);
2014-04-07 23:31:38 +04:00
void Send (uint8_t type, const uint8_t * payload, size_t len); // with session key
2014-02-09 06:06:40 +04:00
2014-02-09 17:52:56 +04:00
void FillHeaderAndEncrypt (uint8_t payloadType, uint8_t * buf, size_t len, const uint8_t * aesKey, const uint8_t * iv, const uint8_t * macKey);
void Decrypt (uint8_t * buf, size_t len, const uint8_t * aesKey);
bool Validate (uint8_t * buf, size_t len, const uint8_t * macKey);
2014-02-13 01:36:13 +04:00
const uint8_t * GetIntroKey () const;
2014-01-25 01:30:07 +04:00
2014-04-08 05:40:28 +04:00
void ScheduleTermination ();
void HandleTerminationTimer (const boost::system::error_code& ecode);
2014-01-25 01:30:07 +04:00
private:
struct IncompleteMessage
{
I2NPMessage * msg;
uint8_t nextFragmentNum;
IncompleteMessage (I2NPMessage * m): msg (m), nextFragmentNum (1) {};
};
2014-01-25 01:30:07 +04:00
2014-04-02 18:49:16 +04:00
SSUServer& m_Server;
2014-01-29 01:49:54 +04:00
boost::asio::ip::udp::endpoint m_RemoteEndpoint;
2014-02-09 17:52:56 +04:00
const i2p::data::RouterInfo * m_RemoteRouter;
2014-04-02 18:49:16 +04:00
boost::asio::deadline_timer m_Timer;
2014-04-04 22:56:46 +04:00
i2p::data::DHKeysPair * m_DHKeysPair; // X - for client and Y - for server
2014-04-08 03:28:06 +04:00
bool m_PeerTest;
2014-04-08 23:35:08 +04:00
SessionState m_State;
2014-04-14 00:59:54 +04:00
bool m_IsSessionKey;
2014-04-08 23:35:08 +04:00
uint32_t m_RelayTag;
2014-04-09 22:58:30 +04:00
std::set<uint32_t> m_PeerTestNonces;
2014-01-29 01:49:54 +04:00
CryptoPP::CBC_Mode<CryptoPP::AES>::Encryption m_Encryption;
2014-01-30 01:49:53 +04:00
CryptoPP::CBC_Mode<CryptoPP::AES>::Decryption m_Decryption;
2014-02-03 23:40:38 +04:00
uint8_t m_SessionKey[32], m_MacKey[32];
std::map<uint32_t, IncompleteMessage *> m_IncomleteMessages;
2014-02-10 03:28:34 +04:00
std::list<i2p::I2NPMessage *> m_DelayedMessages;
2014-01-25 01:30:07 +04:00
};
2014-01-24 01:10:33 +04:00
class SSUServer
{
public:
SSUServer (boost::asio::io_service& service, int port);
2014-01-25 01:30:07 +04:00
~SSUServer ();
2014-01-24 01:10:33 +04:00
void Start ();
void Stop ();
2014-04-08 03:28:06 +04:00
SSUSession * GetSession (const i2p::data::RouterInfo * router, bool peerTest = false);
2014-03-26 05:17:03 +04:00
SSUSession * FindSession (const i2p::data::RouterInfo * router);
2014-02-09 06:06:40 +04:00
void DeleteSession (SSUSession * session);
2014-02-22 01:13:36 +04:00
void DeleteAllSessions ();
2014-03-31 06:55:03 +04:00
boost::asio::io_service& GetService () { return m_Socket.get_io_service(); };
2014-01-30 23:03:11 +04:00
const boost::asio::ip::udp::endpoint& GetEndpoint () const { return m_Endpoint; };
2014-01-29 01:49:54 +04:00
void Send (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& to);
2014-01-24 01:10:33 +04:00
private:
void Receive ();
void HandleReceivedFrom (const boost::system::error_code& ecode, std::size_t bytes_transferred);
private:
2014-01-30 23:03:11 +04:00
boost::asio::ip::udp::endpoint m_Endpoint;
2014-01-24 01:10:33 +04:00
boost::asio::ip::udp::socket m_Socket;
boost::asio::ip::udp::endpoint m_SenderEndpoint;
2014-01-29 01:49:54 +04:00
uint8_t m_ReceiveBuffer[2*SSU_MTU];
2014-01-25 01:30:07 +04:00
std::map<boost::asio::ip::udp::endpoint, SSUSession *> m_Sessions;
2014-02-25 07:28:28 +04:00
public:
// for HTTP only
const decltype(m_Sessions)& GetSessions () const { return m_Sessions; };
2014-01-24 01:10:33 +04:00
};
}
}
#endif