mirrors/i2pd
2
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd synced 2024-11-10 00:00:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

merge incomplete DATAGRAM SEND message

Browse Source
This commit is contained in:
orignal 2015-03-27 14:36:54 -04:00
parent 2a4d1c978e
commit 8c174cd548
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
SAM.cpp
View File

@ -220,7 +220,8 @@ namespace client
if (processed < bytes_transferred)
{
m_BufferOffset = bytes_transferred - processed;
memmove (m_Buffer, m_Buffer + processed, m_BufferOffset);
if (processed > 0)
memmove (m_Buffer, m_Buffer + processed, m_BufferOffset);
}
// since it's SAM v1 reply is not expected
Receive ();
@ -420,7 +421,10 @@ namespace client
LogPrint (eLogError, "SAM session is not created from DATAGRAM SEND");
}
else
LogPrint (eLogError, "SAM sent datagram size ", size, " exceeds buffer");
{
LogPrint (eLogWarning, "SAM sent datagram size ", size, " exceeds buffer");
return 0; // try to receive more
}
return offset + size;
}
@ -529,6 +533,12 @@ namespace client
void SAMSocket::Receive ()
{
if (m_BufferOffset >= SAM_SOCKET_BUFFER_SIZE)
{
LogPrint (eLogError, "Buffer is full. Terminate");
Terminate ();
return;
}
m_Socket.async_read_some (boost::asio::buffer(m_Buffer + m_BufferOffset, SAM_SOCKET_BUFFER_SIZE - m_BufferOffset),
std::bind((m_SocketType == eSAMSocketTypeStream) ? &SAMSocket::HandleReceived : &SAMSocket::HandleMessage,
shared_from_this (), std::placeholders::_1, std::placeholders::_2));