mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2025-02-02 11:04:00 +01:00
split stream to packets
This commit is contained in:
parent
a2707c19b8
commit
297afeb07b
113
Streaming.cpp
113
Streaming.cpp
|
@ -202,59 +202,68 @@ namespace stream
|
||||||
size_t Stream::Send (const uint8_t * buf, size_t len, int timeout)
|
size_t Stream::Send (const uint8_t * buf, size_t len, int timeout)
|
||||||
{
|
{
|
||||||
bool isNoAck = m_LastReceivedSequenceNumber < 0; // first packet
|
bool isNoAck = m_LastReceivedSequenceNumber < 0; // first packet
|
||||||
Packet * p = new Packet ();
|
while (len > 0)
|
||||||
uint8_t * packet = p->GetBuffer ();
|
|
||||||
// TODO: implement setters
|
|
||||||
size_t size = 0;
|
|
||||||
*(uint32_t *)(packet + size) = htobe32 (m_SendStreamID);
|
|
||||||
size += 4; // sendStreamID
|
|
||||||
*(uint32_t *)(packet + size) = htobe32 (m_RecvStreamID);
|
|
||||||
size += 4; // receiveStreamID
|
|
||||||
*(uint32_t *)(packet + size) = htobe32 (m_SequenceNumber++);
|
|
||||||
size += 4; // sequenceNum
|
|
||||||
if (isNoAck)
|
|
||||||
*(uint32_t *)(packet + size) = htobe32 (m_LastReceivedSequenceNumber);
|
|
||||||
else
|
|
||||||
*(uint32_t *)(packet + size) = 0;
|
|
||||||
size += 4; // ack Through
|
|
||||||
packet[size] = 0;
|
|
||||||
size++; // NACK count
|
|
||||||
size++; // resend delay
|
|
||||||
if (!m_IsOpen)
|
|
||||||
{
|
|
||||||
// initial packet
|
|
||||||
m_IsOpen = true;
|
|
||||||
uint16_t flags = PACKET_FLAG_SYNCHRONIZE | PACKET_FLAG_FROM_INCLUDED |
|
|
||||||
PACKET_FLAG_SIGNATURE_INCLUDED | PACKET_FLAG_MAX_PACKET_SIZE_INCLUDED;
|
|
||||||
if (isNoAck) flags |= PACKET_FLAG_NO_ACK;
|
|
||||||
*(uint16_t *)(packet + size) = htobe16 (flags);
|
|
||||||
size += 2; // flags
|
|
||||||
*(uint16_t *)(packet + size) = htobe16 (i2p::data::DEFAULT_IDENTITY_SIZE + 40 + 2); // identity + signature + packet size
|
|
||||||
size += 2; // options size
|
|
||||||
memcpy (packet + size, &m_LocalDestination->GetIdentity (), i2p::data::DEFAULT_IDENTITY_SIZE);
|
|
||||||
size += i2p::data::DEFAULT_IDENTITY_SIZE; // from
|
|
||||||
*(uint16_t *)(packet + size) = htobe16 (STREAMING_MTU);
|
|
||||||
size += 2; // max packet size
|
|
||||||
uint8_t * signature = packet + size; // set it later
|
|
||||||
memset (signature, 0, 40); // zeroes for now
|
|
||||||
size += 40; // signature
|
|
||||||
memcpy (packet + size, buf, len);
|
|
||||||
size += len; // payload
|
|
||||||
m_LocalDestination->Sign (packet, size, signature);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
{
|
||||||
// follow on packet
|
Packet * p = new Packet ();
|
||||||
*(uint16_t *)(packet + size) = 0;
|
uint8_t * packet = p->GetBuffer ();
|
||||||
size += 2; // flags
|
// TODO: implement setters
|
||||||
*(uint16_t *)(packet + size) = 0; // no options
|
size_t size = 0;
|
||||||
size += 2; // options size
|
*(uint32_t *)(packet + size) = htobe32 (m_SendStreamID);
|
||||||
memcpy (packet + size, buf, len);
|
size += 4; // sendStreamID
|
||||||
size += len; // payload
|
*(uint32_t *)(packet + size) = htobe32 (m_RecvStreamID);
|
||||||
}
|
size += 4; // receiveStreamID
|
||||||
p->len = size;
|
*(uint32_t *)(packet + size) = htobe32 (m_SequenceNumber++);
|
||||||
m_Service.post (boost::bind (&Stream::SendPacket, this, p));
|
size += 4; // sequenceNum
|
||||||
|
if (isNoAck)
|
||||||
|
*(uint32_t *)(packet + size) = htobe32 (m_LastReceivedSequenceNumber);
|
||||||
|
else
|
||||||
|
*(uint32_t *)(packet + size) = 0;
|
||||||
|
size += 4; // ack Through
|
||||||
|
packet[size] = 0;
|
||||||
|
size++; // NACK count
|
||||||
|
size++; // resend delay
|
||||||
|
if (!m_IsOpen)
|
||||||
|
{
|
||||||
|
// initial packet
|
||||||
|
m_IsOpen = true;
|
||||||
|
uint16_t flags = PACKET_FLAG_SYNCHRONIZE | PACKET_FLAG_FROM_INCLUDED |
|
||||||
|
PACKET_FLAG_SIGNATURE_INCLUDED | PACKET_FLAG_MAX_PACKET_SIZE_INCLUDED;
|
||||||
|
if (isNoAck) flags |= PACKET_FLAG_NO_ACK;
|
||||||
|
*(uint16_t *)(packet + size) = htobe16 (flags);
|
||||||
|
size += 2; // flags
|
||||||
|
*(uint16_t *)(packet + size) = htobe16 (i2p::data::DEFAULT_IDENTITY_SIZE + 40 + 2); // identity + signature + packet size
|
||||||
|
size += 2; // options size
|
||||||
|
memcpy (packet + size, &m_LocalDestination->GetIdentity (), i2p::data::DEFAULT_IDENTITY_SIZE);
|
||||||
|
size += i2p::data::DEFAULT_IDENTITY_SIZE; // from
|
||||||
|
*(uint16_t *)(packet + size) = htobe16 (STREAMING_MTU);
|
||||||
|
size += 2; // max packet size
|
||||||
|
uint8_t * signature = packet + size; // set it later
|
||||||
|
memset (signature, 0, 40); // zeroes for now
|
||||||
|
size += 40; // signature
|
||||||
|
size_t sentLen = STREAMING_MTU - size;
|
||||||
|
if (len < sentLen) sentLen = len;
|
||||||
|
memcpy (packet + size, buf, sentLen);
|
||||||
|
len -= sentLen;
|
||||||
|
size += sentLen; // payload
|
||||||
|
m_LocalDestination->Sign (packet, size, signature);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// follow on packet
|
||||||
|
*(uint16_t *)(packet + size) = 0;
|
||||||
|
size += 2; // flags
|
||||||
|
*(uint16_t *)(packet + size) = 0; // no options
|
||||||
|
size += 2; // options size
|
||||||
|
size_t sentLen = STREAMING_MTU - size;
|
||||||
|
if (len < sentLen) sentLen = len;
|
||||||
|
memcpy (packet + size, buf, sentLen);
|
||||||
|
len -= sentLen;
|
||||||
|
size += sentLen; // payload
|
||||||
|
}
|
||||||
|
p->len = size;
|
||||||
|
m_Service.post (boost::bind (&Stream::SendPacket, this, p));
|
||||||
|
}
|
||||||
|
|
||||||
return len;
|
return len;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue