mirror of https://github.com/PurpleI2P/i2pd.git
chertov
11 years ago
8 changed files with 269 additions and 174 deletions
@ -0,0 +1,138 @@
@@ -0,0 +1,138 @@
|
||||
#include "Log.h" |
||||
#include "SSU.h" |
||||
#include "SSUData.h" |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace ssu |
||||
{ |
||||
SSUData::SSUData (SSUSession& session): |
||||
m_Session (session) |
||||
{ |
||||
} |
||||
|
||||
SSUData::~SSUData () |
||||
{ |
||||
for (auto it: m_IncomleteMessages) |
||||
if (it.second) |
||||
{ |
||||
DeleteI2NPMessage (it.second->msg); |
||||
delete it.second; |
||||
} |
||||
} |
||||
|
||||
void SSUData::ProcessMessage (uint8_t * buf, size_t len) |
||||
{ |
||||
//uint8_t * start = buf;
|
||||
uint8_t flag = *buf; |
||||
buf++; |
||||
LogPrint ("Process SSU data flags=", (int)flag); |
||||
if (flag & DATA_FLAG_EXPLICIT_ACKS_INCLUDED) |
||||
{ |
||||
// explicit ACKs
|
||||
uint8_t numAcks =*buf; |
||||
buf++; |
||||
// TODO: process ACKs
|
||||
buf += numAcks*4; |
||||
} |
||||
if (flag & DATA_FLAG_ACK_BITFIELDS_INCLUDED) |
||||
{ |
||||
// explicit ACK bitfields
|
||||
uint8_t numBitfields =*buf; |
||||
buf++; |
||||
for (int i = 0; i < numBitfields; i++) |
||||
{ |
||||
buf += 4; // msgID
|
||||
// TODO: process ACH bitfields
|
||||
while (*buf & 0x80) // not last
|
||||
buf++; |
||||
buf++; // last byte
|
||||
} |
||||
} |
||||
uint8_t numFragments = *buf; // number of fragments
|
||||
buf++; |
||||
for (int i = 0; i < numFragments; i++) |
||||
{ |
||||
uint32_t msgID = be32toh (*(uint32_t *)buf); // message ID
|
||||
buf += 4; |
||||
uint8_t frag[4]; |
||||
frag[0] = 0; |
||||
memcpy (frag + 1, buf, 3); |
||||
buf += 3; |
||||
uint32_t fragmentInfo = be32toh (*(uint32_t *)frag); // fragment info
|
||||
uint16_t fragmentSize = fragmentInfo & 0x1FFF; // bits 0 - 13
|
||||
bool isLast = fragmentInfo & 0x010000; // bit 16
|
||||
uint8_t fragmentNum = fragmentInfo >> 17; // bits 23 - 17
|
||||
LogPrint ("SSU data fragment ", (int)fragmentNum, " of message ", msgID, " size=", (int)fragmentSize, isLast ? " last" : " non-last"); |
||||
I2NPMessage * msg = nullptr; |
||||
if (fragmentNum > 0) // follow-up fragment
|
||||
{ |
||||
auto it = m_IncomleteMessages.find (msgID); |
||||
if (it != m_IncomleteMessages.end ()) |
||||
{ |
||||
if (fragmentNum == it->second->nextFragmentNum) |
||||
{ |
||||
// expected fragment
|
||||
msg = it->second->msg; |
||||
memcpy (msg->buf + msg->len, buf, fragmentSize); |
||||
msg->len += fragmentSize; |
||||
it->second->nextFragmentNum++; |
||||
} |
||||
else if (fragmentNum < it->second->nextFragmentNum) |
||||
// duplicate fragment
|
||||
LogPrint ("Duplicate fragment ", fragmentNum, " of message ", msgID, ". Ignored"); |
||||
else |
||||
{ |
||||
// missing fragment
|
||||
LogPrint ("Missing fragments from ", it->second->nextFragmentNum, " to ", fragmentNum - 1, " of message ", msgID); |
||||
//TODO
|
||||
} |
||||
|
||||
if (isLast) |
||||
{ |
||||
delete it->second; |
||||
m_IncomleteMessages.erase (it); |
||||
} |
||||
} |
||||
else |
||||
// TODO:
|
||||
LogPrint ("Unexpected follow-on fragment ", fragmentNum, " of message ", msgID); |
||||
} |
||||
else // first fragment
|
||||
{ |
||||
msg = NewI2NPMessage (); |
||||
memcpy (msg->GetSSUHeader (), buf, fragmentSize); |
||||
msg->len += fragmentSize - sizeof (I2NPHeaderShort); |
||||
} |
||||
|
||||
if (msg) |
||||
{ |
||||
if (!fragmentNum && !isLast) |
||||
m_IncomleteMessages[msgID] = new IncompleteMessage (msg); |
||||
if (isLast) |
||||
{ |
||||
m_Session.SendMsgAck (msgID); |
||||
msg->FromSSU (msgID); |
||||
if (m_Session.GetState () == eSessionStateEstablished) |
||||
i2p::HandleI2NPMessage (msg); |
||||
else |
||||
{ |
||||
// we expect DeliveryStatus
|
||||
if (msg->GetHeader ()->typeID == eI2NPDeliveryStatus) |
||||
{ |
||||
LogPrint ("SSU session established"); |
||||
m_Session.Established (); |
||||
} |
||||
else |
||||
LogPrint ("SSU unexpected message ", (int)msg->GetHeader ()->typeID); |
||||
DeleteI2NPMessage (msg); |
||||
} |
||||
} |
||||
} |
||||
buf += fragmentSize; |
||||
} |
||||
} |
||||
|
||||
} |
||||
} |
||||
|
||||
@ -0,0 +1,48 @@
@@ -0,0 +1,48 @@
|
||||
#ifndef SSU_DATA_H__ |
||||
#define SSU_DATA_H__ |
||||
|
||||
#include <inttypes.h> |
||||
#include <map> |
||||
#include "I2NPProtocol.h" |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace ssu |
||||
{ |
||||
|
||||
// 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; |
||||
|
||||
class SSUSession; |
||||
class SSUData |
||||
{ |
||||
public: |
||||
|
||||
SSUData (SSUSession& session); |
||||
~SSUData (); |
||||
|
||||
void ProcessMessage (uint8_t * buf, size_t len); |
||||
|
||||
private: |
||||
|
||||
struct IncompleteMessage |
||||
{ |
||||
I2NPMessage * msg; |
||||
uint8_t nextFragmentNum; |
||||
|
||||
IncompleteMessage (I2NPMessage * m): msg (m), nextFragmentNum (1) {}; |
||||
}; |
||||
|
||||
SSUSession& m_Session; |
||||
std::map<uint32_t, IncompleteMessage *> m_IncomleteMessages; |
||||
}; |
||||
} |
||||
} |
||||
|
||||
#endif |
||||
|
||||
Loading…
Reference in new issue