mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2024-11-16 00:12:43 +00:00
108 lines
3.0 KiB
C++
108 lines
3.0 KiB
C++
/*
|
|
* Copyright (c) 2013-2021, The PurpleI2P Project
|
|
*
|
|
* This file is part of Purple i2pd project and licensed under BSD3
|
|
*
|
|
* See full license text in LICENSE file at top of project tree
|
|
*/
|
|
|
|
#include "MatchedDestination.h"
|
|
#include "Log.h"
|
|
#include "ClientContext.h"
|
|
|
|
|
|
namespace i2p
|
|
{
|
|
namespace client
|
|
{
|
|
MatchedTunnelDestination::MatchedTunnelDestination(const i2p::data::PrivateKeys & keys, const std::string & remoteName, const std::map<std::string, std::string> * params)
|
|
: RunnableClientDestination(keys, false, params),
|
|
m_RemoteName(remoteName) {}
|
|
|
|
|
|
void MatchedTunnelDestination::ResolveCurrentLeaseSet()
|
|
{
|
|
auto addr = i2p::client::context.GetAddressBook().GetAddress (m_RemoteName);
|
|
if(addr && addr->IsIdentHash ())
|
|
{
|
|
m_RemoteIdent = addr->identHash;
|
|
auto ls = FindLeaseSet(m_RemoteIdent);
|
|
if(ls)
|
|
HandleFoundCurrentLeaseSet(ls);
|
|
else
|
|
RequestDestination(m_RemoteIdent, std::bind(&MatchedTunnelDestination::HandleFoundCurrentLeaseSet, this, std::placeholders::_1));
|
|
}
|
|
else
|
|
LogPrint(eLogWarning, "Destination: failed to resolve ", m_RemoteName);
|
|
}
|
|
|
|
void MatchedTunnelDestination::HandleFoundCurrentLeaseSet(std::shared_ptr<const i2p::data::LeaseSet> ls)
|
|
{
|
|
if(ls)
|
|
{
|
|
LogPrint(eLogDebug, "Destination: resolved remote lease set for ", m_RemoteName);
|
|
m_RemoteLeaseSet = ls;
|
|
}
|
|
else
|
|
{
|
|
m_ResolveTimer->expires_from_now(boost::posix_time::seconds(1));
|
|
m_ResolveTimer->async_wait([&](const boost::system::error_code & ec) {
|
|
if(!ec) ResolveCurrentLeaseSet();
|
|
});
|
|
}
|
|
}
|
|
|
|
|
|
void MatchedTunnelDestination::Start()
|
|
{
|
|
ClientDestination::Start();
|
|
m_ResolveTimer = std::make_shared<boost::asio::deadline_timer>(GetService());
|
|
GetTunnelPool()->SetCustomPeerSelector(this);
|
|
ResolveCurrentLeaseSet();
|
|
}
|
|
|
|
void MatchedTunnelDestination::Stop()
|
|
{
|
|
ClientDestination::Stop();
|
|
if(m_ResolveTimer)
|
|
m_ResolveTimer->cancel();
|
|
}
|
|
|
|
|
|
bool MatchedTunnelDestination::SelectPeers(i2p::tunnel::Path & path, int hops, bool inbound)
|
|
{
|
|
auto pool = GetTunnelPool();
|
|
if(!i2p::tunnel::StandardSelectPeers(path, hops, inbound,
|
|
std::bind(&i2p::tunnel::TunnelPool::SelectNextHop, pool, std::placeholders::_1, std::placeholders::_2)))
|
|
return false;
|
|
// more here for outbound tunnels
|
|
if(!inbound && m_RemoteLeaseSet)
|
|
{
|
|
if(m_RemoteLeaseSet->IsExpired())
|
|
{
|
|
ResolveCurrentLeaseSet();
|
|
}
|
|
if(m_RemoteLeaseSet && !m_RemoteLeaseSet->IsExpired())
|
|
{
|
|
// remote lease set is good
|
|
auto leases = m_RemoteLeaseSet->GetNonExpiredLeases();
|
|
// pick lease
|
|
std::shared_ptr<i2p::data::RouterInfo> obep;
|
|
while(!obep && leases.size() > 0) {
|
|
auto idx = rand() % leases.size();
|
|
auto lease = leases[idx];
|
|
obep = i2p::data::netdb.FindRouter(lease->tunnelGateway);
|
|
leases.erase(leases.begin()+idx);
|
|
}
|
|
if(obep) {
|
|
path.push_back(obep->GetRouterIdentity());
|
|
LogPrint(eLogDebug, "Destination: found OBEP matching IBGW");
|
|
} else
|
|
LogPrint(eLogWarning, "Destination: could not find proper IBGW for matched outbound tunnel");
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
}
|