i2pd/SSU.h

120 lines
3.2 KiB
C
Raw Normal View History

2014-01-23 21:10:33 +00:00
#ifndef SSU_H__
#define SSU_H__
#include <inttypes.h>
2014-01-24 21:30:07 +00:00
#include <map>
2014-01-23 21:10:33 +00:00
#include <boost/asio.hpp>
2014-01-27 21:52:17 +00:00
#include <cryptopp/modes.h>
#include <cryptopp/aes.h>
#include "I2PEndian.h"
2014-01-28 21:49:54 +00:00
#include "RouterInfo.h"
2014-01-29 21:49:53 +00:00
#include "I2NPProtocol.h"
2014-01-23 21:10:33 +00:00
namespace i2p
{
namespace ssu
{
2014-01-27 21:52:17 +00:00
#pragma pack(1)
struct SSUHeader
{
uint8_t mac[16];
uint8_t iv[16];
uint8_t flag;
uint32_t time;
};
#pragma pack()
2014-01-23 21:10:33 +00:00
const int SSU_MTU = 1484;
2014-01-30 13:28:11 +00:00
// payload types (4 bits)
2014-01-24 21:30:07 +00: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;
2014-01-30 13:28:11 +00:00
const uint8_t PAYLOAD_TYPE_SESSION_DESTROY = 8;
2014-01-24 21:30:07 +00:00
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;
enum SessionState
{
eSessionStateUnknown,
eSessionStateRequestSent,
eSessionStateRequestReceived,
eSessionStateCreatedSent,
eSessionStateCreatedReceived,
eSessionStateConfirmedSent,
eSessionStateConfirmedReceived,
eSessionStateEstablised
};
2014-01-28 21:49:54 +00:00
class SSUServer;
2014-01-24 21:30:07 +00:00
class SSUSession
{
public:
2014-01-28 21:49:54 +00:00
SSUSession (SSUServer * server, const boost::asio::ip::udp::endpoint& remoteEndpoint,
i2p::data::RouterInfo * router = nullptr);
2014-01-29 21:49:53 +00:00
void ProcessNextMessage (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
void Connect ();
void SendI2NPMessage (I2NPMessage * msg);
2014-01-28 21:49:54 +00:00
2014-01-27 21:52:17 +00:00
private:
2014-01-29 21:49:53 +00:00
void CreateAESKey (uint8_t * pubKey, uint8_t * aesKey); // TODO: shouldn't be here
void ProcessSessionRequest (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
void SendSessionRequest ();
void ProcessSessionCreated (uint8_t * buf, size_t len);
2014-01-30 19:03:11 +00:00
void SendSessionCreated (const uint8_t * x);
2014-01-28 21:49:54 +00:00
2014-01-29 21:49:53 +00:00
bool ProcessIntroKeyEncryptedMessage (uint8_t expectedPayloadType, uint8_t * buf, size_t len);
void FillHeaderAndEncrypt (uint8_t payloadType, uint8_t * buf, size_t len, uint8_t * aesKey, uint8_t * iv, uint8_t * macKey);
2014-01-28 21:49:54 +00:00
void Decrypt (uint8_t * buf, size_t len, uint8_t * aesKey);
bool Validate (uint8_t * buf, size_t len, uint8_t * macKey);
2014-01-24 21:30:07 +00:00
private:
2014-01-28 21:49:54 +00:00
SSUServer * m_Server;
boost::asio::ip::udp::endpoint m_RemoteEndpoint;
2014-01-29 21:49:53 +00:00
i2p::data::RouterInfo * m_RemoteRouter;
2014-01-24 21:30:07 +00:00
SessionState m_State;
2014-01-28 21:49:54 +00:00
CryptoPP::CBC_Mode<CryptoPP::AES>::Encryption m_Encryption;
2014-01-29 21:49:53 +00:00
CryptoPP::CBC_Mode<CryptoPP::AES>::Decryption m_Decryption;
uint8_t m_SessionKey[32];
2014-01-24 21:30:07 +00:00
};
2014-01-23 21:10:33 +00:00
class SSUServer
{
public:
SSUServer (boost::asio::io_service& service, int port);
2014-01-24 21:30:07 +00:00
~SSUServer ();
2014-01-23 21:10:33 +00:00
void Start ();
void Stop ();
2014-01-28 21:49:54 +00:00
SSUSession * GetSession (i2p::data::RouterInfo * router);
2014-01-30 19:03:11 +00:00
const boost::asio::ip::udp::endpoint& GetEndpoint () const { return m_Endpoint; };
2014-01-28 21:49:54 +00:00
void Send (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& to);
2014-01-23 21:10:33 +00:00
private:
void Receive ();
void HandleReceivedFrom (const boost::system::error_code& ecode, std::size_t bytes_transferred);
private:
2014-01-30 19:03:11 +00:00
boost::asio::ip::udp::endpoint m_Endpoint;
2014-01-23 21:10:33 +00:00
boost::asio::ip::udp::socket m_Socket;
boost::asio::ip::udp::endpoint m_SenderEndpoint;
2014-01-28 21:49:54 +00:00
uint8_t m_ReceiveBuffer[2*SSU_MTU];
2014-01-24 21:30:07 +00:00
std::map<boost::asio::ip::udp::endpoint, SSUSession *> m_Sessions;
2014-01-23 21:10:33 +00:00
};
}
}
#endif