i2pd/SSUData.h

101 lines
2.3 KiB
C
Raw Normal View History

2014-04-22 15:39:26 +00:00
#ifndef SSU_DATA_H__
#define SSU_DATA_H__
#include <inttypes.h>
2014-07-15 02:06:58 +00:00
#include <string.h>
2014-04-22 15:39:26 +00:00
#include <map>
2014-06-12 15:14:22 +00:00
#include <vector>
2014-07-15 02:06:58 +00:00
#include <set>
2014-07-20 14:38:39 +00:00
#include <boost/asio.hpp>
2014-04-22 15:39:26 +00:00
#include "I2NPProtocol.h"
namespace i2p
{
namespace ssu
{
2014-07-15 02:06:58 +00:00
const size_t SSU_MTU = 1484;
2014-07-20 14:38:39 +00:00
const int RESEND_INTERVAL = 3; // in seconds
const int MAX_NUM_RESENDS = 5;
2014-04-22 15:39:26 +00:00
// 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;
2014-07-15 02:06:58 +00:00
struct Fragment
{
2014-07-20 14:38:39 +00:00
int fragmentNum;
size_t len;
2014-07-15 02:06:58 +00:00
bool isLast;
2014-07-20 14:38:39 +00:00
uint8_t buf[SSU_MTU + 18];
2014-07-15 02:06:58 +00:00
2014-07-20 14:38:39 +00:00
Fragment () = default;
2014-07-15 02:06:58 +00:00
Fragment (int n, const uint8_t * b, int l, bool last):
fragmentNum (n), len (l), isLast (last) { memcpy (buf, b, len); };
};
struct FragmentCmp
{
bool operator() (const Fragment * f1, const Fragment * f2) const
{
return f1->fragmentNum < f2->fragmentNum;
};
};
struct IncompleteMessage
{
I2NPMessage * msg;
int nextFragmentNum;
std::set<Fragment *, FragmentCmp> savedFragments;
IncompleteMessage (I2NPMessage * m): msg (m), nextFragmentNum (0) {};
~IncompleteMessage () { for (auto it: savedFragments) { delete it; }; };
};
2014-07-19 22:53:02 +00:00
struct SentMessage
{
2014-07-20 14:38:39 +00:00
std::vector<Fragment *> fragments;
2014-07-19 22:53:02 +00:00
uint32_t nextResendTime; // in seconds
int numResends;
2014-07-20 14:38:39 +00:00
~SentMessage () { for (auto it: fragments) { delete it; }; };
2014-07-19 22:53:02 +00:00
};
2014-07-15 02:06:58 +00:00
2014-04-22 15:39:26 +00:00
class SSUSession;
class SSUData
{
public:
SSUData (SSUSession& session);
~SSUData ();
void ProcessMessage (uint8_t * buf, size_t len);
2014-06-10 14:39:29 +00:00
void Send (i2p::I2NPMessage * msg);
2014-04-22 15:39:26 +00:00
2014-06-10 15:19:31 +00:00
private:
void SendMsgAck (uint32_t msgID);
void SendFragmentAck (uint32_t msgID, int fragmentNum);
2014-07-17 19:22:32 +00:00
void ProcessAcks (uint8_t *& buf, uint8_t flag);
void ProcessFragments (uint8_t * buf);
void ProcessSentMessageAck (uint32_t msgID);
2014-06-10 15:19:31 +00:00
2014-07-20 14:38:39 +00:00
void ScheduleResend ();
void HandleResendTimer (const boost::system::error_code& ecode);
2014-07-15 02:06:58 +00:00
2014-07-20 14:38:39 +00:00
private:
2014-04-22 15:39:26 +00:00
SSUSession& m_Session;
std::map<uint32_t, IncompleteMessage *> m_IncomleteMessages;
2014-07-20 14:38:39 +00:00
std::map<uint32_t, SentMessage *> m_SentMessages;
boost::asio::deadline_timer m_ResendTimer;
2014-04-22 15:39:26 +00:00
};
}
}
#endif