Browse Source

handle single packet

pull/1752/head
orignal 3 years ago
parent
commit
05946125b5
  1. 23
      libi2pd/SSU2.cpp
  2. 3
      libi2pd/SSU2.h

23
libi2pd/SSU2.cpp

@ -1330,13 +1330,13 @@ namespace transport @@ -1330,13 +1330,13 @@ namespace transport
{
i2p::transport::transports.UpdateReceivedBytes (bytes_transferred);
packet->len = bytes_transferred;
std::vector<Packet *> packets;
packets.push_back (packet);
boost::system::error_code ec;
size_t moreBytes = socket.available (ec);
if (!ec)
if (!ec && moreBytes)
{
std::vector<Packet *> packets;
packets.push_back (packet);
while (moreBytes && packets.size () < 32)
{
packet = m_PacketsPool.AcquireMt ();
@ -1355,9 +1355,10 @@ namespace transport @@ -1355,9 +1355,10 @@ namespace transport
break;
}
}
GetService ().post (std::bind (&SSU2Server::HandleReceivedPackets, this, packets));
}
GetService ().post (std::bind (&SSU2Server::HandleReceivedPacket, this, packets));
else
GetService ().post (std::bind (&SSU2Server::HandleReceivedPacket, this, packet));
Receive (socket);
}
else
@ -1374,7 +1375,17 @@ namespace transport @@ -1374,7 +1375,17 @@ namespace transport
}
}
void SSU2Server::HandleReceivedPacket (std::vector<Packet *> packets)
void SSU2Server::HandleReceivedPacket (Packet * packet)
{
if (packet)
{
ProcessNextPacket (packet->buf, packet->len, packet->from);
m_PacketsPool.ReleaseMt (packet);
if (m_LastSession) m_LastSession->FlushData ();
}
}
void SSU2Server::HandleReceivedPackets (std::vector<Packet *> packets)
{
for (auto& packet: packets)
ProcessNextPacket (packet->buf, packet->len, packet->from);

3
libi2pd/SSU2.h

@ -257,7 +257,8 @@ namespace transport @@ -257,7 +257,8 @@ namespace transport
void Receive (boost::asio::ip::udp::socket& socket);
void HandleReceivedFrom (const boost::system::error_code& ecode, size_t bytes_transferred,
Packet * packet, boost::asio::ip::udp::socket& socket);
void HandleReceivedPacket (std::vector<Packet *> packets);
void HandleReceivedPacket (Packet * packet);
void HandleReceivedPackets (std::vector<Packet *> packets);
void ProcessNextPacket (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
void ScheduleTermination ();

Loading…
Cancel
Save