i2pd/TunnelPool.cpp

206 lines
5.4 KiB
C++
Raw Normal View History

2014-03-17 20:50:03 +00:00
#include "I2PEndian.h"
#include "CryptoConst.h"
2014-03-14 16:35:02 +00:00
#include "Tunnel.h"
#include "NetDb.h"
2014-03-15 00:24:12 +00:00
#include "Timestamp.h"
2014-03-17 20:50:03 +00:00
#include "Garlic.h"
2014-03-14 16:35:02 +00:00
#include "TunnelPool.h"
namespace i2p
{
namespace tunnel
{
2014-04-01 17:55:09 +00:00
TunnelPool::TunnelPool (i2p::data::LocalDestination& localDestination, int numTunnels):
2014-03-16 20:03:20 +00:00
m_LocalDestination (localDestination), m_NumTunnels (numTunnels), m_LastOutboundTunnel (nullptr)
2014-03-14 16:35:02 +00:00
{
}
TunnelPool::~TunnelPool ()
{
for (auto it: m_InboundTunnels)
2014-03-15 00:51:51 +00:00
it->SetTunnelPool (nullptr);
2014-03-16 20:03:20 +00:00
for (auto it: m_OutboundTunnels)
it->SetTunnelPool (nullptr);
}
void TunnelPool::TunnelCreated (InboundTunnel * createdTunnel)
{
m_InboundTunnels.insert (createdTunnel);
}
2014-03-15 01:22:59 +00:00
void TunnelPool::TunnelExpired (InboundTunnel * expiredTunnel)
{
2014-03-18 12:15:43 +00:00
if (expiredTunnel)
{
expiredTunnel->SetTunnelPool (nullptr);
m_InboundTunnels.erase (expiredTunnel);
}
2014-04-01 17:55:09 +00:00
m_LocalDestination.UpdateLeaseSet ();
2014-03-15 01:22:59 +00:00
}
2014-03-16 20:03:20 +00:00
void TunnelPool::TunnelCreated (OutboundTunnel * createdTunnel)
{
m_OutboundTunnels.insert (createdTunnel);
}
void TunnelPool::TunnelExpired (OutboundTunnel * expiredTunnel)
{
2014-03-18 12:15:43 +00:00
if (expiredTunnel)
2014-03-21 19:54:55 +00:00
{
2014-03-18 12:15:43 +00:00
expiredTunnel->SetTunnelPool (nullptr);
m_OutboundTunnels.erase (expiredTunnel);
2014-03-21 19:54:55 +00:00
}
2014-03-17 20:50:03 +00:00
if (expiredTunnel == m_LastOutboundTunnel)
m_LastOutboundTunnel = nullptr;
2014-03-16 20:03:20 +00:00
}
2014-03-15 01:22:59 +00:00
std::vector<InboundTunnel *> TunnelPool::GetInboundTunnels (int num) const
{
std::vector<InboundTunnel *> v;
int i = 0;
for (auto it : m_InboundTunnels)
{
if (i >= num) break;
2014-03-21 19:54:55 +00:00
if (!it->IsFailed ())
{
v.push_back (it);
i++;
}
}
return v;
}
2014-03-16 20:03:20 +00:00
OutboundTunnel * TunnelPool::GetNextOutboundTunnel ()
{
if (m_OutboundTunnels.empty ()) return nullptr;
auto tunnel = *m_OutboundTunnels.begin ();
if (m_LastOutboundTunnel && tunnel == m_LastOutboundTunnel)
{
for (auto it: m_OutboundTunnels)
2014-03-21 19:54:55 +00:00
if (it != m_LastOutboundTunnel && !it->IsFailed ())
2014-03-16 20:03:20 +00:00
{
tunnel = it;
break;
}
}
m_LastOutboundTunnel = tunnel;
return tunnel;
}
InboundTunnel * TunnelPool::GetNextInboundTunnel ()
{
return GetNextTunnel (m_InboundTunnels);
}
template<class TTunnels>
typename TTunnels::value_type TunnelPool::GetNextTunnel (TTunnels& tunnels)
{
if (tunnels.empty ()) return nullptr;
for (auto it: tunnels)
if (!it->IsFailed ())
return it;
return nullptr;
}
void TunnelPool::CreateTunnels ()
{
2014-03-15 00:24:12 +00:00
int num = m_InboundTunnels.size ();
for (int i = num; i < m_NumTunnels; i++)
CreateInboundTunnel ();
2014-03-16 20:03:20 +00:00
num = m_OutboundTunnels.size ();
for (int i = num; i < m_NumTunnels; i++)
CreateOutboundTunnel ();
}
2014-03-17 20:50:03 +00:00
void TunnelPool::TestTunnels ()
{
auto& rnd = i2p::context.GetRandomNumberGenerator ();
for (auto it: m_Tests)
{
LogPrint ("Tunnel test ", (int)it.first, " failed");
// both outbound and inbound tunnels considered as invalid
2014-03-21 19:54:55 +00:00
it.second.first->SetFailed (true);
it.second.second->SetFailed (true);
2014-03-17 20:50:03 +00:00
}
m_Tests.clear ();
auto it1 = m_OutboundTunnels.begin ();
auto it2 = m_InboundTunnels.begin ();
while (it1 != m_OutboundTunnels.end () && it2 != m_InboundTunnels.end ())
{
2014-03-21 22:26:11 +00:00
bool failed = false;
if ((*it1)->IsFailed ())
{
failed = true;
it1++;
}
if ((*it2)->IsFailed ())
{
failed = true;
it2++;
}
if (!failed)
{
uint32_t msgID = rnd.GenerateWord32 ();
m_Tests[msgID] = std::make_pair (*it1, *it2);
(*it1)->SendTunnelDataMsg ((*it2)->GetNextIdentHash (), (*it2)->GetNextTunnelID (),
CreateDeliveryStatusMsg (msgID));
it1++; it2++;
}
2014-03-17 20:50:03 +00:00
}
}
void TunnelPool::ProcessDeliveryStatus (I2NPMessage * msg)
{
I2NPDeliveryStatusMsg * deliveryStatus = (I2NPDeliveryStatusMsg *)msg->GetPayload ();
auto it = m_Tests.find (be32toh (deliveryStatus->msgID));
if (it != m_Tests.end ())
{
LogPrint ("Tunnel test ", it->first, " successive. ", i2p::util::GetMillisecondsSinceEpoch () - be64toh (deliveryStatus->timestamp), " milliseconds");
m_Tests.erase (it);
}
else
i2p::garlic::routing.HandleDeliveryStatusMessage (msg->GetPayload (), msg->GetLength ()); // TODO:
DeleteI2NPMessage (msg);
}
void TunnelPool::CreateInboundTunnel ()
{
2014-03-16 20:03:20 +00:00
OutboundTunnel * outboundTunnel = m_OutboundTunnels.size () > 0 ?
*m_OutboundTunnels.begin () : tunnels.GetNextOutboundTunnel ();
LogPrint ("Creating destination inbound tunnel...");
auto firstHop = i2p::data::netdb.GetRandomRouter (outboundTunnel ? outboundTunnel->GetEndpointRouter () : nullptr);
2014-03-15 01:22:59 +00:00
auto secondHop = i2p::data::netdb.GetRandomRouter (firstHop);
auto * tunnel = tunnels.CreateTunnel<InboundTunnel> (
new TunnelConfig (std::vector<const i2p::data::RouterInfo *>
{
firstHop,
2014-03-15 13:16:55 +00:00
secondHop
// TODO: switch to 3-hops later
2014-03-15 13:16:55 +00:00
/*i2p::data::netdb.GetRandomRouter (secondHop) */
}),
outboundTunnel);
2014-03-15 00:51:51 +00:00
tunnel->SetTunnelPool (this);
2014-03-14 16:35:02 +00:00
}
2014-03-16 20:03:20 +00:00
void TunnelPool::CreateOutboundTunnel ()
{
InboundTunnel * inboundTunnel = m_InboundTunnels.size () > 0 ?
*m_InboundTunnels.begin () : tunnels.GetNextInboundTunnel ();
if (inboundTunnel)
{
LogPrint ("Creating destination outbound tunnel...");
auto firstHop = i2p::data::netdb.GetRandomRouter (&i2p::context.GetRouterInfo ());
auto secondHop = i2p::data::netdb.GetRandomRouter (firstHop);
auto * tunnel = tunnels.CreateTunnel<OutboundTunnel> (
new TunnelConfig (std::vector<const i2p::data::RouterInfo *>
{
firstHop,
secondHop
},
inboundTunnel->GetTunnelConfig ()));
tunnel->SetTunnelPool (this);
}
}
2014-03-14 16:35:02 +00:00
}
}