|
|
|
#ifndef SSU_H__
|
|
|
|
#define SSU_H__
|
|
|
|
|
|
|
|
#include <inttypes.h>
|
|
|
|
#include <map>
|
|
|
|
#include <list>
|
|
|
|
#include <boost/asio.hpp>
|
|
|
|
#include <cryptopp/modes.h>
|
|
|
|
#include <cryptopp/aes.h>
|
|
|
|
#include "I2PEndian.h"
|
|
|
|
#include "RouterInfo.h"
|
|
|
|
#include "I2NPProtocol.h"
|
|
|
|
|
|
|
|
namespace i2p
|
|
|
|
{
|
|
|
|
namespace ssu
|
|
|
|
{
|
|
|
|
#pragma pack(1)
|
|
|
|
struct SSUHeader
|
|
|
|
{
|
|
|
|
uint8_t mac[16];
|
|
|
|
uint8_t iv[16];
|
|
|
|
uint8_t flag;
|
|
|
|
uint32_t time;
|
|
|
|
};
|
|
|
|
#pragma pack()
|
|
|
|
|
|
|
|
const int SSU_MTU = 1484;
|
|
|
|
|
|
|
|
// payload types (4 bits)
|
|
|
|
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;
|
|
|
|
const uint8_t PAYLOAD_TYPE_TEST = 7;
|
|
|
|
const uint8_t PAYLOAD_TYPE_SESSION_DESTROYED = 8;
|
|
|
|
|
|
|
|
// 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;
|
|
|
|
|
|
|
|
enum SessionState
|
|
|
|
{
|
|
|
|
eSessionStateUnknown,
|
|
|
|
eSessionStateRequestSent,
|
|
|
|
eSessionStateRequestReceived,
|
|
|
|
eSessionStateCreatedSent,
|
|
|
|
eSessionStateCreatedReceived,
|
|
|
|
eSessionStateConfirmedSent,
|
|
|
|
eSessionStateConfirmedReceived,
|
|
|
|
eSessionRelayRequestSent,
|
|
|
|
eSessionRelayRequestReceived,
|
|
|
|
eSessionRelayResponseReceived,
|
|
|
|
eSessionStateEstablished
|
|
|
|
};
|
|
|
|
|
|
|
|
class SSUServer;
|
|
|
|
class SSUSession
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
|
|
|
SSUSession (SSUServer * server, boost::asio::ip::udp::endpoint& remoteEndpoint,
|
|
|
|
const i2p::data::RouterInfo * router = nullptr);
|
|
|
|
void ProcessNextMessage (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
|
|
|
|
|
|
|
|
void Connect ();
|
|
|
|
void ConnectThroughIntroducer (const i2p::data::RouterInfo::Introducer& introducer);
|
|
|
|
void Close ();
|
|
|
|
boost::asio::ip::udp::endpoint& GetRemoteEndpoint () { return m_RemoteEndpoint; };
|
|
|
|
void SendI2NPMessage (I2NPMessage * msg);
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
void CreateAESandMacKey (uint8_t * pubKey, uint8_t * aesKey, uint8_t * macKey);
|
|
|
|
|
|
|
|
void ProcessMessage (uint8_t * buf, size_t len); // call for established session
|
|
|
|
void ProcessSessionRequest (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
|
|
|
|
void SendSessionRequest ();
|
|
|
|
void SendRelayRequest (const i2p::data::RouterInfo::Introducer& introducer);
|
|
|
|
void ProcessSessionCreated (uint8_t * buf, size_t len);
|
|
|
|
void SendSessionCreated (const uint8_t * x);
|
|
|
|
void ProcessSessionConfirmed (uint8_t * buf, size_t len);
|
|
|
|
void SendSessionConfirmed (const uint8_t * y, const uint8_t * ourAddress, uint32_t relayTag);
|
|
|
|
void ProcessRelayResponse (uint8_t * buf, size_t len);
|
|
|
|
void Established ();
|
|
|
|
void ProcessData (uint8_t * buf, size_t len);
|
|
|
|
void SendMsgAck (uint32_t msgID);
|
|
|
|
void SendSesionDestroyed ();
|
|
|
|
void Send (i2p::I2NPMessage * msg);
|
|
|
|
|
|
|
|
bool ProcessIntroKeyEncryptedMessage (uint8_t expectedPayloadType, uint8_t * buf, size_t len);
|
|
|
|
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);
|
|
|
|
const uint8_t * GetIntroKey () const;
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
SSUServer * m_Server;
|
|
|
|
boost::asio::ip::udp::endpoint m_RemoteEndpoint;
|
|
|
|
const i2p::data::RouterInfo * m_RemoteRouter;
|
|
|
|
SessionState m_State;
|
|
|
|
CryptoPP::CBC_Mode<CryptoPP::AES>::Encryption m_Encryption;
|
|
|
|
CryptoPP::CBC_Mode<CryptoPP::AES>::Decryption m_Decryption;
|
|
|
|
uint8_t m_SessionKey[32], m_MacKey[32];
|
|
|
|
std::map<uint32_t, I2NPMessage *> m_IncomleteMessages;
|
|
|
|
std::list<i2p::I2NPMessage *> m_DelayedMessages;
|
|
|
|
};
|
|
|
|
|
|
|
|
class SSUServer
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
|
|
|
SSUServer (boost::asio::io_service& service, int port);
|
|
|
|
~SSUServer ();
|
|
|
|
void Start ();
|
|
|
|
void Stop ();
|
|
|
|
SSUSession * GetSession (const i2p::data::RouterInfo * router);
|
|
|
|
void DeleteSession (SSUSession * session);
|
|
|
|
void DeleteAllSessions ();
|
|
|
|
|
|
|
|
const boost::asio::ip::udp::endpoint& GetEndpoint () const { return m_Endpoint; };
|
|
|
|
void Send (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& to);
|
|
|
|
void ReassignSession (const boost::asio::ip::udp::endpoint& oldEndpoint, const boost::asio::ip::udp::endpoint& newEndpoint);
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
void Receive ();
|
|
|
|
void HandleReceivedFrom (const boost::system::error_code& ecode, std::size_t bytes_transferred);
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
boost::asio::ip::udp::endpoint m_Endpoint;
|
|
|
|
boost::asio::ip::udp::socket m_Socket;
|
|
|
|
boost::asio::ip::udp::endpoint m_SenderEndpoint;
|
|
|
|
uint8_t m_ReceiveBuffer[2*SSU_MTU];
|
|
|
|
std::map<boost::asio::ip::udp::endpoint, SSUSession *> m_Sessions;
|
|
|
|
|
|
|
|
public:
|
|
|
|
// for HTTP only
|
|
|
|
const decltype(m_Sessions)& GetSessions () const { return m_Sessions; };
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|