Browse Source

memory pool for SSU packets

pull/781/head
orignal 8 years ago
parent
commit
4317694c64
  1. 16
      SSU.cpp
  2. 4
      SSU.h

16
SSU.cpp

@ -187,14 +187,14 @@ namespace transport @@ -187,14 +187,14 @@ namespace transport
void SSUServer::Receive ()
{
SSUPacket * packet = new SSUPacket ();
SSUPacket * packet = m_Packets.AcquireMt ();
m_Socket.async_receive_from (boost::asio::buffer (packet->buf, SSU_MTU_V4), packet->from,
std::bind (&SSUServer::HandleReceivedFrom, this, std::placeholders::_1, std::placeholders::_2, packet));
}
void SSUServer::ReceiveV6 ()
{
SSUPacket * packet = new SSUPacket ();
SSUPacket * packet = m_Packets.AcquireMt ();
m_SocketV6.async_receive_from (boost::asio::buffer (packet->buf, SSU_MTU_V6), packet->from,
std::bind (&SSUServer::HandleReceivedFromV6, this, std::placeholders::_1, std::placeholders::_2, packet));
}
@ -213,7 +213,7 @@ namespace transport @@ -213,7 +213,7 @@ namespace transport
{
while (moreBytes && packets.size () < 25)
{
packet = new SSUPacket ();
packet = m_Packets.AcquireMt ();
packet->len = m_Socket.receive_from (boost::asio::buffer (packet->buf, SSU_MTU_V4), packet->from, 0, ec);
if (!ec)
{
@ -224,7 +224,7 @@ namespace transport @@ -224,7 +224,7 @@ namespace transport
else
{
LogPrint (eLogError, "SSU: receive_from error: ", ec.message ());
delete packet;
m_Packets.ReleaseMt (packet);
break;
}
}
@ -260,7 +260,7 @@ namespace transport @@ -260,7 +260,7 @@ namespace transport
{
while (moreBytes && packets.size () < 25)
{
packet = new SSUPacket ();
packet = m_Packets.AcquireMt ();
packet->len = m_SocketV6.receive_from (boost::asio::buffer (packet->buf, SSU_MTU_V6), packet->from, 0, ec);
if (!ec)
{
@ -271,7 +271,7 @@ namespace transport @@ -271,7 +271,7 @@ namespace transport
else
{
LogPrint (eLogError, "SSU: v6 receive_from error: ", ec.message ());
delete packet;
m_Packets.ReleaseMt (packet);
break;
}
}
@ -282,7 +282,7 @@ namespace transport @@ -282,7 +282,7 @@ namespace transport
}
else
{
delete packet;
m_Packets.ReleaseMt (packet);
if (ecode != boost::asio::error::operation_aborted)
{
LogPrint (eLogError, "SSU: v6 receive error: ", ecode.message ());
@ -323,8 +323,8 @@ namespace transport @@ -323,8 +323,8 @@ namespace transport
if (session) session->FlushData ();
session = nullptr;
}
delete packet;
}
m_Packets.ReleaseMt (packets);
if (session) session->FlushData ();
}

4
SSU.h

@ -15,6 +15,7 @@ @@ -15,6 +15,7 @@
#include "RouterInfo.h"
#include "I2NPProtocol.h"
#include "SSUSession.h"
#include "util.h" // for MemoryPool
namespace i2p
{
@ -122,7 +123,8 @@ namespace transport @@ -122,7 +123,8 @@ namespace transport
std::map<boost::asio::ip::udp::endpoint, std::shared_ptr<SSUSession> > m_Sessions, m_SessionsV6;
std::map<uint32_t, std::shared_ptr<SSUSession> > m_Relays; // we are introducer
std::map<uint32_t, PeerTest> m_PeerTests; // nonce -> creation time in milliseconds
i2p::util::MemoryPoolMt<SSUPacket> m_Packets;
public:
// for HTTP only
const decltype(m_Sessions)& GetSessions () const { return m_Sessions; };

Loading…
Cancel
Save