Mirrors/i2pd
2
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2025-02-02 11:04:00 +01:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

check if session is terminated before send

Browse source
This commit is contained in:
orignal 2020-10-24 21:22:48 -04:00
parent 1b63c9f6ad
commit e9f11e204e
1 changed files with 17 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
libi2pd_client/I2CP.cpp
View file

@ -806,17 +806,23 @@ namespace client
void I2CPSession::SendMessagePayloadMessage (const uint8_t * payload, size_t len) void I2CPSession::SendMessagePayloadMessage (const uint8_t * payload, size_t len)
{ {
// we don't use SendI2CPMessage to eliminate additional copy // we don't use SendI2CPMessage to eliminate additional copy
auto l = len + 10 + I2CP_HEADER_SIZE; auto socket = m_Socket;
uint8_t * buf = new uint8_t[l]; if (socket)
htobe32buf (buf + I2CP_HEADER_LENGTH_OFFSET, len + 10); {
buf[I2CP_HEADER_TYPE_OFFSET] = I2CP_MESSAGE_PAYLOAD_MESSAGE; auto l = len + 10 + I2CP_HEADER_SIZE;
htobe16buf (buf + I2CP_HEADER_SIZE, m_SessionID); uint8_t * buf = new uint8_t[l];
htobe32buf (buf + I2CP_HEADER_SIZE + 2, m_MessageID++); htobe32buf (buf + I2CP_HEADER_LENGTH_OFFSET, len + 10);
htobe32buf (buf + I2CP_HEADER_SIZE + 6, len); buf[I2CP_HEADER_TYPE_OFFSET] = I2CP_MESSAGE_PAYLOAD_MESSAGE;
memcpy (buf + I2CP_HEADER_SIZE + 10, payload, len); htobe16buf (buf + I2CP_HEADER_SIZE, m_SessionID);
boost::asio::async_write (*m_Socket, boost::asio::buffer (buf, l), boost::asio::transfer_all (), htobe32buf (buf + I2CP_HEADER_SIZE + 2, m_MessageID++);
std::bind(&I2CPSession::HandleI2CPMessageSent, shared_from_this (), htobe32buf (buf + I2CP_HEADER_SIZE + 6, len);
std::placeholders::_1, std::placeholders::_2, buf)); memcpy (buf + I2CP_HEADER_SIZE + 10, payload, len);
boost::asio::async_write (*socket, boost::asio::buffer (buf, l), boost::asio::transfer_all (),
std::bind(&I2CPSession::HandleI2CPMessageSent, shared_from_this (),
std::placeholders::_1, std::placeholders::_2, buf));
}
else
LogPrint (eLogError, "I2CP: Can't write to the socket");
} }
I2CPServer::I2CPServer (const std::string& interface, int port, bool isSingleThread): I2CPServer::I2CPServer (const std::string& interface, int port, bool isSingleThread):