mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2025-07-01 10:14:36 +02:00
96 lines
4.1 KiB
C++
96 lines
4.1 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.Add(obep);
|
|
LogPrint(eLogDebug, "Destination: Found OBEP matching IBGW");
|
|
} else
|
|
LogPrint(eLogWarning, "Destination: Could not find proper IBGW for matched outbound tunnel");
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
}
|