Browse Source

handle I2NP messages drops

pull/2013/head
orignal 5 months ago
parent
commit
f008478505
  1. 7
      libi2pd/I2NPProtocol.h
  2. 1
      libi2pd/NTCP2.cpp
  3. 6
      libi2pd/NetDb.cpp
  4. 6
      libi2pd/NetDbRequests.cpp
  5. 1
      libi2pd/SSU2Session.cpp
  6. 7
      libi2pd/Transports.h

7
libi2pd/I2NPProtocol.h

@ -13,6 +13,7 @@ @@ -13,6 +13,7 @@
#include <string.h>
#include <set>
#include <memory>
#include <functional>
#include "Crypto.h"
#include "I2PEndian.h"
#include "Identity.h"
@ -149,7 +150,8 @@ namespace tunnel @@ -149,7 +150,8 @@ namespace tunnel
uint8_t * buf;
size_t len, offset, maxLen;
std::shared_ptr<i2p::tunnel::InboundTunnel> from;
std::function<void ()> onDrop;
I2NPMessage (): buf (nullptr),len (I2NP_HEADER_SIZE + 2),
offset(2), maxLen (0), from (nullptr) {}; // reserve 2 bytes for NTCP header
@ -241,7 +243,6 @@ namespace tunnel @@ -241,7 +243,6 @@ namespace tunnel
SetSize (len - offset - I2NP_HEADER_SIZE);
SetChks (0);
}
void ToNTCP2 ()
{
uint8_t * ntcp2 = GetNTCP2Header ();
@ -253,6 +254,8 @@ namespace tunnel @@ -253,6 +254,8 @@ namespace tunnel
void RenewI2NPMessageHeader ();
bool IsExpired () const;
bool IsExpired (uint64_t ts) const; // in milliseconds
void Drop () { if (onDrop) { onDrop (); onDrop = nullptr; }; }
};
template<int sz>

1
libi2pd/NTCP2.cpp

@ -1140,6 +1140,7 @@ namespace transport @@ -1140,6 +1140,7 @@ namespace transport
if (!msg || msg->IsExpired (ts))
{
// drop null or expired message
if (msg) msg->Drop ();
m_SendQueue.pop_front ();
continue;
}

6
libi2pd/NetDb.cpp

@ -736,7 +736,11 @@ namespace data @@ -736,7 +736,11 @@ namespace data
!i2p::transport::transports.IsConnected (floodfill->GetIdentHash ()))
direct = false; // floodfill can't be reached directly
if (direct)
transports.SendMessage (floodfill->GetIdentHash (), dest->CreateRequestMessage (floodfill->GetIdentHash ()));
{
auto msg = dest->CreateRequestMessage (floodfill->GetIdentHash ());
msg->onDrop = [this, dest]() { this->m_Requests.SendNextRequest (dest); };
transports.SendMessage (floodfill->GetIdentHash (), msg);
}
else
{
auto pool = i2p::tunnel::tunnels.GetExploratoryPool ();

6
libi2pd/NetDbRequests.cpp

@ -188,7 +188,11 @@ namespace data @@ -188,7 +188,11 @@ namespace data
!i2p::transport::transports.IsConnected (nextFloodfill->GetIdentHash ()))
direct = false; // floodfill can't be reached directly
if (direct)
i2p::transport::transports.SendMessage (nextFloodfill->GetIdentHash (), dest->CreateRequestMessage (nextFloodfill->GetIdentHash ()));
{
auto msg = dest->CreateRequestMessage (nextFloodfill->GetIdentHash ());
msg->onDrop = [this, dest]() { this->SendNextRequest (dest); };
i2p::transport::transports.SendMessage (nextFloodfill->GetIdentHash (), msg);
}
else
{
auto pool = i2p::tunnel::tunnels.GetExploratoryPool ();

1
libi2pd/SSU2Session.cpp

@ -382,6 +382,7 @@ namespace transport @@ -382,6 +382,7 @@ namespace transport
if (!msg || msg->IsExpired (ts))
{
// drop null or expired message
if (msg) msg->Drop ();
m_SendQueue.pop_front ();
continue;
}

7
libi2pd/Transports.h

@ -86,6 +86,13 @@ namespace transport @@ -86,6 +86,13 @@ namespace transport
}
}
~Peer ()
{
// drop not sent delayed messages
for (auto& it: delayedMessages)
it->Drop ();
}
void Done ()
{
for (auto& it: sessions)

Loading…
Cancel
Save