mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2025-04-28 03:37:49 +02:00
113 lines
3.1 KiB
C++
113 lines
3.1 KiB
C++
#include "MatchedDestination.h"
|
|
#include "Log.h"
|
|
#include "ClientContext.h"
|
|
|
|
|
|
namespace dotnet
|
|
{
|
|
namespace client
|
|
{
|
|
MatchedTunnelDestination::MatchedTunnelDestination(const dotnet::data::PrivateKeys & keys, const std::string & remoteName, const std::map<std::string, std::string> * params)
|
|
: ClientDestination(keys, false, params),
|
|
m_RemoteName(remoteName) {}
|
|
|
|
|
|
void MatchedTunnelDestination::ResolveCurrentLeaseSet()
|
|
{
|
|
auto addr = dotnet::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 dotnet::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();
|
|
});
|
|
}
|
|
}
|
|
|
|
|
|
bool MatchedTunnelDestination::Start()
|
|
{
|
|
if(ClientDestination::Start())
|
|
{
|
|
m_ResolveTimer = std::make_shared<boost::asio::deadline_timer>(GetService());
|
|
GetTunnelPool()->SetCustomPeerSelector(this);
|
|
ResolveCurrentLeaseSet();
|
|
return true;
|
|
}
|
|
else
|
|
return false;
|
|
}
|
|
|
|
bool MatchedTunnelDestination::Stop()
|
|
{
|
|
if(ClientDestination::Stop())
|
|
{
|
|
if(m_ResolveTimer)
|
|
m_ResolveTimer->cancel();
|
|
return true;
|
|
}
|
|
else
|
|
return false;
|
|
}
|
|
|
|
|
|
bool MatchedTunnelDestination::SelectPeers(dotnet::tunnel::Path & path, int hops, bool inbound)
|
|
{
|
|
auto pool = GetTunnelPool();
|
|
if(!dotnet::tunnel::StandardSelectPeers(path, hops, inbound, std::bind(&dotnet::tunnel::TunnelPool::SelectNextHop, pool, std::placeholders::_1)))
|
|
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<dotnet::data::RouterInfo> obep;
|
|
while(!obep && leases.size() > 0) {
|
|
auto idx = rand() % leases.size();
|
|
auto lease = leases[idx];
|
|
obep = dotnet::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;
|
|
}
|
|
|
|
bool MatchedTunnelDestination::OnBuildResult(const dotnet::tunnel::Path & path, bool inbound, dotnet::tunnel::TunnelBuildResult result)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
}
|