PurpleI2P/i2pd
1
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2025-01-31 09:04:14 +00:00
Code Issues Releases Activity

receive incoming SSU2 packets

Browse Source
This commit is contained in:
orignal 2022-03-14 19:25:59 -04:00
parent b3c2e86436
commit 7b72d91549
2 changed files with 45 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
libi2pd/SSU2.cpp
View File

@ -225,7 +225,10 @@ namespace transport
} }
} }
if (port) if (port)
{
OpenSocket (port); OpenSocket (port);
Receive ();
}
else else
LogPrint (eLogError, "SSU2: Can't start server because port not specified "); LogPrint (eLogError, "SSU2: Can't start server because port not specified ");
break; break;
@ -238,7 +241,7 @@ namespace transport
{ {
StopIOService (); StopIOService ();
} }
void SSU2Server::OpenSocket (int port) void SSU2Server::OpenSocket (int port)
{ {
try try
@ -255,6 +258,36 @@ namespace transport
ThrowFatal ("Unable to start SSU2 transport at port ", port, ": ", ex.what ()); ThrowFatal ("Unable to start SSU2 transport at port ", port, ": ", ex.what ());
} }
} }
void SSU2Server::Receive ()
{
Packet * packet = m_PacketsPool.AcquireMt ();
m_Socket.async_receive_from (boost::asio::buffer (packet->buf, SSU2_MTU), packet->from,
std::bind (&SSU2Server::HandleReceivedFrom, this, std::placeholders::_1, std::placeholders::_2, packet));
}
void SSU2Server::HandleReceivedFrom (const boost::system::error_code& ecode, size_t bytes_transferred, Packet * packet)
{
if (!ecode)
{
packet->len = bytes_transferred;
ProcessNextPacket (packet->buf, packet->len, packet->from);
m_PacketsPool.ReleaseMt (packet);
Receive ();
}
else
{
m_PacketsPool.ReleaseMt (packet);
if (ecode != boost::asio::error::operation_aborted)
{
LogPrint (eLogError, "SSU2: Receive error: code ", ecode.value(), ": ", ecode.message ());
auto port = m_Socket.local_endpoint ().port ();
m_Socket.close ();
OpenSocket (port);
Receive ();
}
}
}
void SSU2Server::AddSession (uint64_t connID, std::shared_ptr<SSU2Session> session) void SSU2Server::AddSession (uint64_t connID, std::shared_ptr<SSU2Session> session)
{ {

11
libi2pd/SSU2.h
View File

@ -24,6 +24,7 @@ namespace transport
const int SSU2_TERMINATION_TIMEOUT = 330; // 5.5 minutes const int SSU2_TERMINATION_TIMEOUT = 330; // 5.5 minutes
const size_t SSU2_SOCKET_RECEIVE_BUFFER_SIZE = 0x1FFFF; // 128K const size_t SSU2_SOCKET_RECEIVE_BUFFER_SIZE = 0x1FFFF; // 128K
const size_t SSU2_SOCKET_SEND_BUFFER_SIZE = 0x1FFFF; // 128K const size_t SSU2_SOCKET_SEND_BUFFER_SIZE = 0x1FFFF; // 128K
const size_t SSU2_MTU = 1488;
enum SSU2MessageType enum SSU2MessageType
{ {
@ -78,6 +79,13 @@ namespace transport
class SSU2Server: private i2p::util::RunnableServiceWithWork class SSU2Server: private i2p::util::RunnableServiceWithWork
{ {
struct Packet
{
uint8_t buf[SSU2_MTU];
size_t len;
boost::asio::ip::udp::endpoint from;
};
public: public:
SSU2Server (); SSU2Server ();
@ -96,6 +104,8 @@ namespace transport
private: private:
void OpenSocket (int port); void OpenSocket (int port);
void Receive ();
void HandleReceivedFrom (const boost::system::error_code& ecode, size_t bytes_transferred, Packet * packet);
void ProcessNextPacket (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint); void ProcessNextPacket (uint8_t * buf, size_t len, const boost::asio::ip::udp::endpoint& senderEndpoint);
private: private:
@ -103,6 +113,7 @@ namespace transport
boost::asio::ip::udp::socket m_Socket; boost::asio::ip::udp::socket m_Socket;
std::unordered_map<uint64_t, std::shared_ptr<SSU2Session> > m_Sessions; std::unordered_map<uint64_t, std::shared_ptr<SSU2Session> > m_Sessions;
std::map<boost::asio::ip::udp::endpoint, std::shared_ptr<SSU2Session> > m_PendingOutgoingSessions; std::map<boost::asio::ip::udp::endpoint, std::shared_ptr<SSU2Session> > m_PendingOutgoingSessions;
i2p::util::MemoryPoolMt<Packet> m_PacketsPool;
}; };
} }
} }