Mirrors/i2pd
2
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2025-04-24 01:46:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

delete messages sent to disconnected session

Browse source
This commit is contained in:
orignal 2015-02-09 16:33:52 -05:00
parent 89883e2078
commit 923b64eb48
2 changed files with 28 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
NTCPSession.cpp
View file

@ -678,6 +678,11 @@ namespace transport
{ {
if (msg) if (msg)
{ {
if (m_IsTerminated)
{
DeleteI2NPMessage (msg);
return;
}
if (m_IsSending) if (m_IsSending)
m_SendQueue.push_back (msg); m_SendQueue.push_back (msg);
else else
@ -692,6 +697,12 @@ namespace transport
void NTCPSession::PostI2NPMessages (std::vector<I2NPMessage *> msgs) void NTCPSession::PostI2NPMessages (std::vector<I2NPMessage *> msgs)
{ {
if (m_IsTerminated)
{
for (auto it: msgs)
DeleteI2NPMessage (it);
return;
}
if (m_IsSending) if (m_IsSending)
{ {
for (auto it: msgs) for (auto it: msgs)

13
SSUSession.cpp
View file

@ -836,7 +836,12 @@ namespace transport
void SSUSession::PostI2NPMessage (I2NPMessage * msg) void SSUSession::PostI2NPMessage (I2NPMessage * msg)
{ {
if (msg) if (msg)
{
if (m_State == eSessionStateEstablished)
m_Data.Send (msg); m_Data.Send (msg);
else
DeleteI2NPMessage (msg);
}
} }
void SSUSession::SendI2NPMessages (const std::vector<I2NPMessage *>& msgs) void SSUSession::SendI2NPMessages (const std::vector<I2NPMessage *>& msgs)
@ -845,10 +850,18 @@ namespace transport
} }
void SSUSession::PostI2NPMessages (std::vector<I2NPMessage *> msgs) void SSUSession::PostI2NPMessages (std::vector<I2NPMessage *> msgs)
{
if (m_State == eSessionStateEstablished)
{ {
for (auto it: msgs) for (auto it: msgs)
if (it) m_Data.Send (it); if (it) m_Data.Send (it);
} }
else
{
for (auto it: msgs)
DeleteI2NPMessage (it);
}
}
void SSUSession::ProcessData (uint8_t * buf, size_t len) void SSUSession::ProcessData (uint8_t * buf, size_t len)
{ {