Archives/i2pd
2
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2024-11-16 00:12:43 +00:00
Code Issues Projects Releases Wiki Activity

memory pool for SSU packets

Browse Source
This commit is contained in:
orignal 2017-01-16 22:22:51 -05:00
parent 9cb8e194b0
commit 4317694c64
2 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
SSU.cpp
View File

@ -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
{
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
else
{
LogPrint (eLogError, "SSU: receive_from error: ", ec.message ());
delete packet;
m_Packets.ReleaseMt (packet);
break;
}
}
@ -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
else
{
LogPrint (eLogError, "SSU: v6 receive_from error: ", ec.message ());
delete packet;
m_Packets.ReleaseMt (packet);
break;
}
}
@ -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
if (session) session->FlushData ();
session = nullptr;
}
delete packet;
}
m_Packets.ReleaseMt (packets);
if (session) session->FlushData ();
}

4
SSU.h
View File

@ -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
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; };