Browse Source

send I2NP messages through NTCP2

pull/1221/head
orignal 6 years ago
parent
commit
b99f828583
  1. 1
      libi2pd/I2NPProtocol.h
  2. 61
      libi2pd/NTCP2.cpp
  3. 10
      libi2pd/NTCP2.h

1
libi2pd/I2NPProtocol.h

@ -199,6 +199,7 @@ namespace tunnel @@ -199,6 +199,7 @@ namespace tunnel
}
// for NTCP2 only
uint8_t * GetNTCP2Header () { return GetPayload () - I2NP_NTCP2_HEADER_SIZE; };
size_t GetNTCP2Length () const { return GetPayloadLength () + I2NP_NTCP2_HEADER_SIZE; };
void FromNTCP2 ()
{
const uint8_t * ntcp2 = GetNTCP2Header ();

61
libi2pd/NTCP2.cpp

@ -125,7 +125,7 @@ namespace transport @@ -125,7 +125,7 @@ namespace transport
m_IsEstablished (false), m_IsTerminated (false),
m_SessionRequestBuffer (nullptr), m_SessionCreatedBuffer (nullptr), m_SessionConfirmedBuffer (nullptr),
m_NextReceivedBuffer (nullptr), m_NextSendBuffer (nullptr),
m_ReceiveSequenceNumber (0), m_SendSequenceNumber (0)
m_ReceiveSequenceNumber (0), m_SendSequenceNumber (0), m_IsSending (false)
{
m_Establisher.reset (new NTCP2Establisher);
auto addr = in_RemoteRouter->GetNTCPAddress ();
@ -167,7 +167,7 @@ namespace transport @@ -167,7 +167,7 @@ namespace transport
{
m_IsEstablished = true;
m_Establisher.reset (nullptr);
// transports.PeerConnected (shared_from_this ());
transports.PeerConnected (shared_from_this ());
}
void NTCP2Session::CreateNonce (uint64_t seqn, uint8_t * nonce)
@ -455,12 +455,8 @@ namespace transport @@ -455,12 +455,8 @@ namespace transport
ReceiveLength ();
// TODO: remove
uint8_t pad[1024];
auto paddingLength = rand () % 1000;
RAND_bytes (pad + 3, paddingLength);
pad[0] = 254;
htobe16buf (pad + 1, paddingLength);
SendNextFrame (pad, paddingLength + 3);
//m_SendQueue.push_back (CreateDeliveryStatusMsg (1));
//SendQueue ();
}
void NTCP2Session::HandleSessionCreatedSent (const boost::system::error_code& ecode, std::size_t bytes_transferred)
@ -665,6 +661,7 @@ namespace transport @@ -665,6 +661,7 @@ namespace transport
LogPrint (eLogDebug, "NTCP2: sent length ", len + 16);
// send message
m_IsSending = true;
boost::asio::async_write (m_Socket, boost::asio::buffer (m_NextSendBuffer, len + 16 + 2), boost::asio::transfer_all (),
std::bind(&NTCP2Session::HandleNextFrameSent, shared_from_this (), std::placeholders::_1, std::placeholders::_2));
}
@ -673,6 +670,54 @@ namespace transport @@ -673,6 +670,54 @@ namespace transport
{
delete[] m_NextSendBuffer; m_NextSendBuffer = nullptr;
LogPrint (eLogDebug, "NTCP2: Next frame sent");
m_IsSending = false;
SendQueue ();
}
void NTCP2Session::SendQueue ()
{
if (!m_SendQueue.empty ())
{
uint8_t * payload = new uint8_t[NTCP2_UNENCRYPTED_FRAME_MAX_SIZE];
size_t s = 0;
// add I2NP blocks
while (!m_SendQueue.empty ())
{
auto msg = m_SendQueue.front ();
size_t len = msg->GetNTCP2Length ();
if (s + len + 3 <= NTCP2_UNENCRYPTED_FRAME_MAX_SIZE) // 3 bytes block header
{
payload[s] = eNTCP2BlkI2NPMessage; // blk
htobe16buf (payload + s + 1, len); // size
s += 3;
msg->ToNTCP2 ();
memcpy (payload + s, msg->GetNTCP2Header (), len);
s += len;
m_SendQueue.pop_front ();
}
else
break;
}
// add padding block
int paddingSize = (s*NTCP2_MAX_PADDING_RATIO)/100;
if (s + paddingSize + 3 > NTCP2_UNENCRYPTED_FRAME_MAX_SIZE) paddingSize = NTCP2_UNENCRYPTED_FRAME_MAX_SIZE - s -3;
if (paddingSize) paddingSize = rand () % paddingSize;
payload[s] = eNTCP2BlkPadding; // blk
htobe16buf (payload + s + 1, paddingSize); // size
s += 3;
RAND_bytes (payload + s, paddingSize);
s += paddingSize;
// send
SendNextFrame (payload, s);
delete[] payload;
}
}
void NTCP2Session::SendI2NPMessages (const std::vector<std::shared_ptr<I2NPMessage> >& msgs)
{
for (auto it: msgs)
m_SendQueue.push_back (it);
if (!m_IsSending) SendQueue ();
}
NTCP2Server::NTCP2Server ():

10
libi2pd/NTCP2.h

@ -4,6 +4,7 @@ @@ -4,6 +4,7 @@
#include <inttypes.h>
#include <memory>
#include <thread>
#include <list>
#include <openssl/bn.h>
#include <boost/asio.hpp>
#include "RouterInfo.h"
@ -13,6 +14,9 @@ namespace i2p @@ -13,6 +14,9 @@ namespace i2p
{
namespace transport
{
const size_t NTCP2_UNENCRYPTED_FRAME_MAX_SIZE = 65519;
const int NTCP2_MAX_PADDING_RATIO = 6; // in %
enum NTCP2BlockType
{
eNTCP2BlkDateTime = 0,
@ -68,7 +72,7 @@ namespace transport @@ -68,7 +72,7 @@ namespace transport
void ClientLogin (); // Alice
void ServerLogin (); // Bob
void SendI2NPMessages (const std::vector<std::shared_ptr<I2NPMessage> >& msgs) {}; // TODO
void SendI2NPMessages (const std::vector<std::shared_ptr<I2NPMessage> >& msgs);
private:
@ -100,6 +104,7 @@ namespace transport @@ -100,6 +104,7 @@ namespace transport
void SendNextFrame (const uint8_t * payload, size_t len);
void HandleNextFrameSent (const boost::system::error_code& ecode, std::size_t bytes_transferred);
void SendQueue ();
private:
@ -119,6 +124,9 @@ namespace transport @@ -119,6 +124,9 @@ namespace transport
uint64_t m_ReceiveSequenceNumber, m_SendSequenceNumber;
i2p::I2NPMessagesHandler m_Handler;
bool m_IsSending;
std::list<std::shared_ptr<I2NPMessage> > m_SendQueue;
};
class NTCP2Server

Loading…
Cancel
Save