Browse Source

select high bandwidth routers for client tunnels

pull/102/head
orignal 10 years ago
parent
commit
79d2f69837
  1. 20
      NetDb.cpp
  2. 4
      NetDb.h
  3. 13
      TunnelPool.cpp
  4. 2
      TunnelPool.h

20
NetDb.cpp

@ -785,19 +785,31 @@ namespace data @@ -785,19 +785,31 @@ namespace data
}
}
const RouterInfo * NetDb::GetRandomRouter () const
{
return GetRandomRouter (
[](const RouterInfo * router)->bool
{
return !router->IsHidden ();
});
}
const RouterInfo * NetDb::GetRandomRouter (const RouterInfo * compatibleWith) const
{
if (compatibleWith)
return GetRandomRouter (
[compatibleWith](const RouterInfo * router)->bool
{
return !router->IsHidden () && router->IsCompatible (*compatibleWith);
});
else
}
const RouterInfo * NetDb::GetHighBandwidthRandomRouter (const RouterInfo * compatibleWith) const
{
return GetRandomRouter (
[](const RouterInfo * router)->bool
[compatibleWith](const RouterInfo * router)->bool
{
return !router->IsHidden ();
return !router->IsHidden () && router->IsCompatible (*compatibleWith) &&
(router->GetCaps () & RouterInfo::eHighBandwidth);
});
}

4
NetDb.h

@ -79,7 +79,9 @@ namespace data @@ -79,7 +79,9 @@ namespace data
void HandleDatabaseSearchReplyMsg (I2NPMessage * msg);
void HandleDatabaseLookupMsg (I2NPMessage * msg);
const RouterInfo * GetRandomRouter (const RouterInfo * compatibleWith = nullptr) const;
const RouterInfo * GetRandomRouter () const;
const RouterInfo * GetRandomRouter (const RouterInfo * compatibleWith) const;
const RouterInfo * GetHighBandwidthRandomRouter (const RouterInfo * compatibleWith) const;
void PostI2NPMsg (I2NPMessage * msg);

13
TunnelPool.cpp

@ -197,6 +197,15 @@ namespace tunnel @@ -197,6 +197,15 @@ namespace tunnel
i2p::garlic::routing.PostI2NPMsg (msg);
}
const i2p::data::RouterInfo * TunnelPool::SelectNextHop (const i2p::data::RouterInfo * prevHop) const
{
auto hop = m_NumHops >= 3 ? i2p::data::netdb.GetHighBandwidthRandomRouter (prevHop) :
i2p::data::netdb.GetRandomRouter (prevHop);
if (!hop)
hop = i2p::data::netdb.GetRandomRouter ();
return hop;
}
void TunnelPool::CreateInboundTunnel ()
{
OutboundTunnel * outboundTunnel = GetNextOutboundTunnel ();
@ -219,7 +228,7 @@ namespace tunnel @@ -219,7 +228,7 @@ namespace tunnel
}
for (int i = 0; i < numHops; i++)
{
auto hop = i2p::data::netdb.GetRandomRouter (prevHop);
auto hop = SelectNextHop (prevHop);
prevHop = hop;
hops.push_back (hop);
}
@ -251,7 +260,7 @@ namespace tunnel @@ -251,7 +260,7 @@ namespace tunnel
std::vector<const i2p::data::RouterInfo *> hops;
for (int i = 0; i < m_NumHops; i++)
{
auto hop = i2p::data::netdb.GetRandomRouter (prevHop);
auto hop = SelectNextHop (prevHop);
prevHop = hop;
hops.push_back (hop);
}

2
TunnelPool.h

@ -7,6 +7,7 @@ @@ -7,6 +7,7 @@
#include <utility>
#include "Identity.h"
#include "LeaseSet.h"
#include "RouterInfo.h"
#include "I2NPProtocol.h"
#include "TunnelBase.h"
#include "RouterContext.h"
@ -53,6 +54,7 @@ namespace tunnel @@ -53,6 +54,7 @@ namespace tunnel
template<class TTunnels>
typename TTunnels::value_type GetNextTunnel (TTunnels& tunnels,
typename TTunnels::value_type suggested = nullptr);
const i2p::data::RouterInfo * SelectNextHop (const i2p::data::RouterInfo * prevHop) const;
private:

Loading…
Cancel
Save