|
|
|
@ -133,6 +133,7 @@ namespace ssu
@@ -133,6 +133,7 @@ namespace ssu
|
|
|
|
|
uint32_t relayTag = be32toh (*(uint32_t *)(buf + sizeof (SSUHeader) + 263)); |
|
|
|
|
SendSessionConfirmed (buf + sizeof (SSUHeader), ourAddress, relayTag); |
|
|
|
|
m_State = eSessionStateEstablished; |
|
|
|
|
Established (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -149,6 +150,7 @@ namespace ssu
@@ -149,6 +150,7 @@ namespace ssu
|
|
|
|
|
LogPrint ("Session confirmed received"); |
|
|
|
|
// TODO:
|
|
|
|
|
m_State = eSessionStateEstablished; |
|
|
|
|
Established (); |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
LogPrint ("Unexpected payload type ", (int)(header->flag >> 4)); |
|
|
|
@ -362,13 +364,35 @@ namespace ssu
@@ -362,13 +364,35 @@ namespace ssu
|
|
|
|
|
void SSUSession::Close () |
|
|
|
|
{ |
|
|
|
|
SendSesionDestroyed (); |
|
|
|
|
if (!m_DelayedMessages.empty ()) |
|
|
|
|
{ |
|
|
|
|
for (auto it :m_DelayedMessages) |
|
|
|
|
delete it; |
|
|
|
|
m_DelayedMessages.clear (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SSUSession::Established () |
|
|
|
|
{ |
|
|
|
|
if (!m_DelayedMessages.empty ()) |
|
|
|
|
{ |
|
|
|
|
for (auto it :m_DelayedMessages) |
|
|
|
|
Send (it); |
|
|
|
|
m_DelayedMessages.clear (); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SSUSession::SendI2NPMessage (I2NPMessage * msg) |
|
|
|
|
{ |
|
|
|
|
// TODO:
|
|
|
|
|
if (msg) |
|
|
|
|
{ |
|
|
|
|
if (m_State == eSessionStateEstablished) |
|
|
|
|
Send (msg); |
|
|
|
|
else |
|
|
|
|
m_DelayedMessages.push_back (msg); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void SSUSession::ProcessData (uint8_t * buf, size_t len) |
|
|
|
|
{ |
|
|
|
|
//uint8_t * start = buf;
|
|
|
|
@ -479,6 +503,52 @@ namespace ssu
@@ -479,6 +503,52 @@ namespace ssu
|
|
|
|
|
FillHeaderAndEncrypt (PAYLOAD_TYPE_SESSION_DESTROYED, buf, 48, m_SessionKey, iv, m_MacKey); |
|
|
|
|
m_Server->Send (buf, 48, m_RemoteEndpoint); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SSUSession::Send (i2p::I2NPMessage * msg) |
|
|
|
|
{ |
|
|
|
|
uint32_t msgID = htobe32 (msg->ToSSU ()); |
|
|
|
|
size_t payloadSize = SSU_MTU - sizeof (SSUHeader) - 9; // 9 = flag + #frg(1) + messageID(4) + frag info (3)
|
|
|
|
|
size_t len = msg->GetLength (); |
|
|
|
|
uint8_t * msgBuf = msg->GetSSUHeader (); |
|
|
|
|
|
|
|
|
|
uint32_t fragmentNum = 0; |
|
|
|
|
while (len > 0) |
|
|
|
|
{ |
|
|
|
|
uint8_t buf[SSU_MTU + 18], iv[16]; |
|
|
|
|
buf[0] = DATA_FLAG_WANT_REPLY; // for compatibility
|
|
|
|
|
buf[1] = 1; // always 1 message fragment per message
|
|
|
|
|
*(uint32_t *)(buf + 2) = msgID; |
|
|
|
|
bool isLast = (len <= payloadSize); |
|
|
|
|
size_t size = isLast ? len : payloadSize; |
|
|
|
|
uint32_t fragmentInfo = (fragmentNum << 17); |
|
|
|
|
if (isLast) |
|
|
|
|
fragmentInfo |= 0x010000; |
|
|
|
|
|
|
|
|
|
fragmentInfo |= size; |
|
|
|
|
fragmentInfo = htobe32 (fragmentInfo); |
|
|
|
|
memcpy (buf + 6, (uint8_t *)(&fragmentInfo) + 1, 3); |
|
|
|
|
memcpy (buf + 9, msgBuf, size); |
|
|
|
|
|
|
|
|
|
size += sizeof (SSUHeader) + 9; |
|
|
|
|
if (size % 16) // make sure 16 bytes boundary
|
|
|
|
|
size = (size/16 + 1)*16; |
|
|
|
|
|
|
|
|
|
CryptoPP::RandomNumberGenerator& rnd = i2p::context.GetRandomNumberGenerator (); |
|
|
|
|
rnd.GenerateBlock (iv, 16); // random iv
|
|
|
|
|
// encrypt message with session key
|
|
|
|
|
FillHeaderAndEncrypt (PAYLOAD_TYPE_DATA, buf, size, m_SessionKey, iv, m_MacKey); |
|
|
|
|
m_Server->Send (buf, size, m_RemoteEndpoint); |
|
|
|
|
|
|
|
|
|
if (!isLast) |
|
|
|
|
{ |
|
|
|
|
len -= payloadSize; |
|
|
|
|
msgBuf += payloadSize; |
|
|
|
|
} |
|
|
|
|
else |
|
|
|
|
len = 0; |
|
|
|
|
fragmentNum++; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
SSUServer::SSUServer (boost::asio::io_service& service, int port): |
|
|
|
|
m_Endpoint (boost::asio::ip::udp::v4 (), port), m_Socket (service, m_Endpoint) |
|
|
|
@ -506,6 +576,7 @@ namespace ssu
@@ -506,6 +576,7 @@ namespace ssu
|
|
|
|
|
void SSUServer::Send (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& to) |
|
|
|
|
{ |
|
|
|
|
m_Socket.send_to (boost::asio::buffer (buf, len), to); |
|
|
|
|
LogPrint ("SSU sent ", len, " bytes"); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void SSUServer::Receive () |
|
|
|
|