don't look for session if a packet is from same endpoint as previous

This commit is contained in:
orignal 2015-02-09 14:48:01 -05:00
parent c6c414c382
commit 23907e6cb1
3 changed files with 15 additions and 14 deletions

25
SSU.cpp
View file

@ -211,22 +211,25 @@ namespace transport
void SSUServer::HandleReceivedPackets (std::vector<SSUPacket *> packets)
{
std::shared_ptr<SSUSession> session;
for (auto it1: packets)
{
auto packet = it1;
std::shared_ptr<SSUSession> session;
auto it = m_Sessions.find (packet->from);
if (it != m_Sessions.end ())
session = it->second;
if (!session)
if (session && session->GetRemoteEndpoint () != packet->from) // we received packet for other session than previous
{
session = std::make_shared<SSUSession> (*this, packet->from);
session->WaitForConnect ();
auto it = m_Sessions.find (packet->from);
if (it != m_Sessions.end ())
session = it->second;
if (!session)
{
std::unique_lock<std::mutex> l(m_SessionsMutex);
m_Sessions[packet->from] = session;
}
LogPrint ("New SSU session from ", packet->from.address ().to_string (), ":", packet->from.port (), " created");
session = std::make_shared<SSUSession> (*this, packet->from);
session->WaitForConnect ();
{
std::unique_lock<std::mutex> l(m_SessionsMutex);
m_Sessions[packet->from] = session;
}
LogPrint ("New SSU session from ", packet->from.address ().to_string (), ":", packet->from.port (), " created");
}
}
session->ProcessNextMessage (packet->buf, packet->len, packet->from);
delete packet;