2018-06-05 19:53:13 +03:00
|
|
|
#ifndef NTCP2_H__
|
|
|
|
#define NTCP2_H__
|
|
|
|
|
|
|
|
#include <inttypes.h>
|
|
|
|
#include <memory>
|
2018-06-06 22:38:18 +03:00
|
|
|
#include <boost/asio.hpp>
|
2018-06-05 19:53:13 +03:00
|
|
|
#include "RouterInfo.h"
|
|
|
|
#include "TransportSession.h"
|
|
|
|
|
|
|
|
namespace i2p
|
|
|
|
{
|
|
|
|
namespace transport
|
|
|
|
{
|
2018-06-06 22:38:18 +03:00
|
|
|
class NTCP2Server;
|
2018-06-05 19:53:13 +03:00
|
|
|
class NTCP2Session: public TransportSession, public std::enable_shared_from_this<NTCP2Session>
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
2018-06-06 22:38:18 +03:00
|
|
|
NTCP2Session (NTCP2Server& server, std::shared_ptr<const i2p::data::RouterInfo> in_RemoteRouter = nullptr); // TODO
|
2018-06-05 19:53:13 +03:00
|
|
|
~NTCP2Session ();
|
|
|
|
|
2018-06-06 22:38:18 +03:00
|
|
|
boost::asio::ip::tcp::socket& GetSocket () { return m_Socket; };
|
|
|
|
|
|
|
|
void ClientLogin (); // Alice
|
|
|
|
|
2018-06-05 19:53:13 +03:00
|
|
|
private:
|
|
|
|
|
2018-06-05 22:37:08 +03:00
|
|
|
bool KeyDerivationFunction (const uint8_t * rs, const uint8_t * pub, uint8_t * derived);
|
2018-06-05 19:53:13 +03:00
|
|
|
void CreateEphemeralKey (uint8_t * pub);
|
2018-06-06 22:38:18 +03:00
|
|
|
void SendSessionRequest ();
|
|
|
|
|
|
|
|
void HandleSessionRequestSent (const boost::system::error_code& ecode, std::size_t bytes_transferred);
|
2018-06-05 19:53:13 +03:00
|
|
|
|
|
|
|
private:
|
|
|
|
|
2018-06-06 22:38:18 +03:00
|
|
|
NTCP2Server& m_Server;
|
|
|
|
boost::asio::ip::tcp::socket m_Socket;
|
2018-06-05 19:53:13 +03:00
|
|
|
uint8_t m_ExpandedPrivateKey[64]; // x25519 ephemeral key
|
2018-06-06 22:38:18 +03:00
|
|
|
uint8_t m_RemoteStaticKey[32], m_RemoteIV[16];
|
|
|
|
uint8_t * m_SessionRequestBuffer;
|
|
|
|
};
|
|
|
|
|
|
|
|
class NTCP2Server
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
|
|
|
NTCP2Server () {};
|
|
|
|
~NTCP2Server () {} ;
|
|
|
|
boost::asio::io_service& GetService () { return m_Service; };
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
boost::asio::io_service m_Service;
|
2018-06-05 19:53:13 +03:00
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|