mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-11 07:10:36 +00:00
273270916e
This commit reflects changes to clang-format rules. Unfortunately, these rule changes create a massive change to the codebase, which causes an apparent rewrite of git history. Git blame's --ignore-rev flag can be used to ignore this commit when attempting to `git blame` some code.
540 lines
16 KiB
C++
540 lines
16 KiB
C++
#include <service/outbound_context.hpp>
|
|
|
|
#include <router/abstractrouter.hpp>
|
|
#include <service/async_key_exchange.hpp>
|
|
#include <service/hidden_service_address_lookup.hpp>
|
|
#include <service/endpoint.hpp>
|
|
#include <nodedb.hpp>
|
|
#include <profiling.hpp>
|
|
#include <util/meta/memfn.hpp>
|
|
|
|
#include <service/endpoint_util.hpp>
|
|
|
|
#include <random>
|
|
#include <algorithm>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace service
|
|
{
|
|
bool
|
|
OutboundContext::Stop()
|
|
{
|
|
markedBad = true;
|
|
return path::Builder::Stop();
|
|
}
|
|
|
|
bool
|
|
OutboundContext::IsDone(llarp_time_t now) const
|
|
{
|
|
(void)now;
|
|
return AvailablePaths(path::ePathRoleAny) == 0 && ShouldRemove();
|
|
}
|
|
|
|
bool
|
|
OutboundContext::ShouldBundleRC() const
|
|
{
|
|
return m_Endpoint->ShouldBundleRC();
|
|
}
|
|
|
|
bool
|
|
OutboundContext::HandleDataDrop(path::Path_ptr p, const PathID_t& dst, uint64_t seq)
|
|
{
|
|
// pick another intro
|
|
if (dst == remoteIntro.pathID && remoteIntro.router == p->Endpoint())
|
|
{
|
|
LogWarn(Name(), " message ", seq, " dropped by endpoint ", p->Endpoint(), " via ", dst);
|
|
if (MarkCurrentIntroBad(Now()))
|
|
{
|
|
SwapIntros();
|
|
}
|
|
UpdateIntroSet();
|
|
}
|
|
return true;
|
|
}
|
|
|
|
OutboundContext::OutboundContext(const IntroSet& introset, Endpoint* parent)
|
|
: path::Builder(parent->Router(), 4, path::default_len)
|
|
, SendContext(introset.A, {}, this, parent)
|
|
, location(introset.A.Addr().ToKey())
|
|
, currentIntroSet(introset)
|
|
|
|
{
|
|
updatingIntroSet = false;
|
|
for (const auto& intro : introset.I)
|
|
{
|
|
if (intro.expiresAt > m_NextIntro.expiresAt)
|
|
m_NextIntro = intro;
|
|
}
|
|
}
|
|
|
|
OutboundContext::~OutboundContext() = default;
|
|
|
|
/// actually swap intros
|
|
void
|
|
OutboundContext::SwapIntros()
|
|
{
|
|
if (remoteIntro != m_NextIntro)
|
|
{
|
|
LogInfo(Name(), " swap intro to use ", RouterID(m_NextIntro.router));
|
|
remoteIntro = m_NextIntro;
|
|
m_DataHandler->PutIntroFor(currentConvoTag, remoteIntro);
|
|
ShiftIntroduction(false);
|
|
}
|
|
}
|
|
|
|
bool
|
|
OutboundContext::OnIntroSetUpdate(
|
|
const Address&, nonstd::optional<IntroSet> foundIntro, const RouterID& endpoint)
|
|
{
|
|
if (markedBad)
|
|
return true;
|
|
updatingIntroSet = false;
|
|
if (foundIntro.has_value())
|
|
{
|
|
if (foundIntro->T == 0s)
|
|
{
|
|
LogWarn(Name(), " got introset with zero timestamp: ", foundIntro.value());
|
|
return true;
|
|
}
|
|
if (currentIntroSet.T > foundIntro->T)
|
|
{
|
|
LogInfo("introset is old, dropping");
|
|
return true;
|
|
}
|
|
|
|
const llarp_time_t now = Now();
|
|
if (foundIntro->IsExpired(now))
|
|
{
|
|
LogError("got expired introset from lookup from ", endpoint);
|
|
return true;
|
|
}
|
|
currentIntroSet = foundIntro.value();
|
|
ShiftIntroduction(false);
|
|
}
|
|
else
|
|
{
|
|
++m_LookupFails;
|
|
LogWarn(Name(), " failed to look up introset, fails=", m_LookupFails);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
OutboundContext::ReadyToSend() const
|
|
{
|
|
return (!remoteIntro.router.IsZero()) && GetPathByRouter(remoteIntro.router) != nullptr;
|
|
}
|
|
|
|
void
|
|
OutboundContext::ShiftIntroRouter(const RouterID r)
|
|
{
|
|
const auto now = Now();
|
|
Introduction selectedIntro;
|
|
for (const auto& intro : currentIntroSet.I)
|
|
{
|
|
if (intro.expiresAt > selectedIntro.expiresAt && intro.router != r)
|
|
{
|
|
selectedIntro = intro;
|
|
}
|
|
}
|
|
if (selectedIntro.router.IsZero() || selectedIntro.ExpiresSoon(now))
|
|
return;
|
|
LogWarn(Name(), " shfiting intro off of ", r, " to ", RouterID(selectedIntro.router));
|
|
m_NextIntro = selectedIntro;
|
|
}
|
|
|
|
void
|
|
OutboundContext::HandlePathBuildTimeout(path::Path_ptr p)
|
|
{
|
|
ShiftIntroRouter(p->Endpoint());
|
|
path::Builder::HandlePathBuildTimeout(p);
|
|
}
|
|
|
|
void
|
|
OutboundContext::HandlePathBuilt(path::Path_ptr p)
|
|
{
|
|
path::Builder::HandlePathBuilt(p);
|
|
/// don't use it if we are marked bad
|
|
if (markedBad)
|
|
return;
|
|
p->SetDataHandler(util::memFn(&OutboundContext::HandleHiddenServiceFrame, this));
|
|
p->SetDropHandler(util::memFn(&OutboundContext::HandleDataDrop, this));
|
|
// we now have a path to the next intro, swap intros
|
|
if (p->Endpoint() == m_NextIntro.router)
|
|
SwapIntros();
|
|
else
|
|
{
|
|
LogInfo(Name(), " built to non aligned router: ", p->Endpoint());
|
|
}
|
|
}
|
|
|
|
void
|
|
OutboundContext::AsyncGenIntro(const llarp_buffer_t& payload, ProtocolType t)
|
|
{
|
|
if (not currentConvoTag.IsZero())
|
|
return;
|
|
if (remoteIntro.router.IsZero())
|
|
SwapIntros();
|
|
|
|
auto path = m_PathSet->GetNewestPathByRouter(remoteIntro.router);
|
|
if (path == nullptr)
|
|
{
|
|
// try parent as fallback
|
|
path = m_Endpoint->GetPathByRouter(remoteIntro.router);
|
|
if (path == nullptr)
|
|
{
|
|
if (!BuildCooldownHit(Now()))
|
|
BuildOneAlignedTo(remoteIntro.router);
|
|
LogWarn(Name(), " dropping intro frame, no path to ", remoteIntro.router);
|
|
return;
|
|
}
|
|
}
|
|
currentConvoTag.Randomize();
|
|
auto frame = std::make_shared<ProtocolFrame>();
|
|
auto ex = std::make_shared<AsyncKeyExchange>(
|
|
m_Endpoint->RouterLogic(),
|
|
remoteIdent,
|
|
m_Endpoint->GetIdentity(),
|
|
currentIntroSet.K,
|
|
remoteIntro,
|
|
m_DataHandler,
|
|
currentConvoTag,
|
|
t);
|
|
|
|
ex->hook = std::bind(&OutboundContext::Send, shared_from_this(), std::placeholders::_1, path);
|
|
|
|
ex->msg.PutBuffer(payload);
|
|
ex->msg.introReply = path->intro;
|
|
frame->F = ex->msg.introReply.pathID;
|
|
m_Endpoint->CryptoWorker()->addJob(std::bind(&AsyncKeyExchange::Encrypt, ex, frame));
|
|
}
|
|
|
|
std::string
|
|
OutboundContext::Name() const
|
|
{
|
|
return "OBContext:" + m_Endpoint->Name() + "-" + currentIntroSet.A.Addr().ToString();
|
|
}
|
|
|
|
void
|
|
OutboundContext::UpdateIntroSet()
|
|
{
|
|
if (updatingIntroSet || markedBad)
|
|
return;
|
|
const auto addr = currentIntroSet.A.Addr();
|
|
// we want to use the parent endpoint's paths because outbound context
|
|
// does not implement path::PathSet::HandleGotIntroMessage
|
|
const auto paths = GetManyPathsWithUniqueEndpoints(m_Endpoint, 2);
|
|
uint64_t relayOrder = 0;
|
|
for (const auto& path : paths)
|
|
{
|
|
HiddenServiceAddressLookup* job = new HiddenServiceAddressLookup(
|
|
m_Endpoint,
|
|
util::memFn(&OutboundContext::OnIntroSetUpdate, shared_from_this()),
|
|
location,
|
|
PubKey{addr.as_array()},
|
|
relayOrder,
|
|
m_Endpoint->GenTXID());
|
|
relayOrder++;
|
|
if (job->SendRequestViaPath(path, m_Endpoint->Router()))
|
|
updatingIntroSet = true;
|
|
}
|
|
}
|
|
|
|
util::StatusObject
|
|
OutboundContext::ExtractStatus() const
|
|
{
|
|
auto obj = path::Builder::ExtractStatus();
|
|
obj["currentConvoTag"] = currentConvoTag.ToHex();
|
|
obj["remoteIntro"] = remoteIntro.ExtractStatus();
|
|
obj["sessionCreatedAt"] = to_json(createdAt);
|
|
obj["lastGoodSend"] = to_json(lastGoodSend);
|
|
obj["seqno"] = sequenceNo;
|
|
obj["markedBad"] = markedBad;
|
|
obj["lastShift"] = to_json(lastShift);
|
|
obj["remoteIdentity"] = remoteIdent.Addr().ToString();
|
|
obj["currentRemoteIntroset"] = currentIntroSet.ExtractStatus();
|
|
obj["nextIntro"] = m_NextIntro.ExtractStatus();
|
|
|
|
std::transform(
|
|
m_BadIntros.begin(),
|
|
m_BadIntros.end(),
|
|
std::back_inserter(obj["badIntros"]),
|
|
[](const auto& item) -> util::StatusObject { return item.first.ExtractStatus(); });
|
|
return obj;
|
|
}
|
|
|
|
bool
|
|
OutboundContext::Pump(llarp_time_t now)
|
|
{
|
|
// we are probably dead af
|
|
if (m_LookupFails > 16 || m_BuildFails > 10)
|
|
return true;
|
|
|
|
// check for expiration
|
|
if (remoteIntro.ExpiresSoon(now))
|
|
{
|
|
UpdateIntroSet();
|
|
// shift intro if it expires "soon"
|
|
if (ShiftIntroduction())
|
|
SwapIntros(); // swap intros if we shifted
|
|
}
|
|
// lookup router in intro if set and unknown
|
|
m_Endpoint->EnsureRouterIsKnown(remoteIntro.router);
|
|
// expire bad intros
|
|
auto itr = m_BadIntros.begin();
|
|
while (itr != m_BadIntros.end())
|
|
{
|
|
if (now > itr->second && now - itr->second > path::default_lifetime)
|
|
itr = m_BadIntros.erase(itr);
|
|
else
|
|
++itr;
|
|
}
|
|
// send control message if we look too quiet
|
|
if (lastGoodSend > 0s)
|
|
{
|
|
if (now - lastGoodSend > (sendTimeout / 2))
|
|
{
|
|
if (!GetNewestPathByRouter(remoteIntro.router))
|
|
{
|
|
if (!BuildCooldownHit(now))
|
|
BuildOneAlignedTo(remoteIntro.router);
|
|
}
|
|
else
|
|
{
|
|
Encrypted<64> tmp;
|
|
tmp.Randomize();
|
|
llarp_buffer_t buf(tmp.data(), tmp.size());
|
|
AsyncEncryptAndSendTo(buf, eProtocolControl);
|
|
}
|
|
}
|
|
}
|
|
// if we are dead return true so we are removed
|
|
return lastGoodSend > 0s ? (now >= lastGoodSend && now - lastGoodSend > sendTimeout)
|
|
: (now >= createdAt && now - createdAt > connectTimeout);
|
|
}
|
|
|
|
bool
|
|
OutboundContext::SelectHop(
|
|
llarp_nodedb* db,
|
|
const std::set<RouterID>& prev,
|
|
RouterContact& cur,
|
|
size_t hop,
|
|
path::PathRole roles)
|
|
{
|
|
if (m_NextIntro.router.IsZero() || prev.count(m_NextIntro.router))
|
|
{
|
|
ShiftIntroduction(false);
|
|
}
|
|
if (m_NextIntro.router.IsZero())
|
|
return false;
|
|
std::set<RouterID> exclude = prev;
|
|
exclude.insert(m_NextIntro.router);
|
|
for (const auto& snode : m_Endpoint->SnodeBlacklist())
|
|
exclude.insert(snode);
|
|
if (hop == 0)
|
|
{
|
|
// exclude any exits as our first hop
|
|
const auto exits = m_Endpoint->GetExitRouters();
|
|
exclude.insert(exits.begin(), exits.end());
|
|
}
|
|
if (hop == numHops - 1)
|
|
{
|
|
m_Endpoint->EnsureRouterIsKnown(m_NextIntro.router);
|
|
if (db->Get(m_NextIntro.router, cur))
|
|
return true;
|
|
++m_BuildFails;
|
|
return false;
|
|
}
|
|
return path::Builder::SelectHop(db, exclude, cur, hop, roles);
|
|
}
|
|
|
|
bool
|
|
OutboundContext::ShouldBuildMore(llarp_time_t now) const
|
|
{
|
|
if (markedBad || path::Builder::BuildCooldownHit(now))
|
|
return false;
|
|
const bool canBuild =
|
|
NumInStatus(path::ePathBuilding) == 0 and path::Builder::ShouldBuildMore(now);
|
|
if (not canBuild)
|
|
return false;
|
|
llarp_time_t t = 0s;
|
|
ForEachPath([&t](path::Path_ptr path) {
|
|
if (path->IsReady())
|
|
t = std::max(path->ExpireTime(), t);
|
|
});
|
|
return t >= now + path::default_lifetime / 4;
|
|
}
|
|
|
|
bool
|
|
OutboundContext::MarkCurrentIntroBad(llarp_time_t now)
|
|
{
|
|
return MarkIntroBad(remoteIntro, now);
|
|
}
|
|
|
|
bool
|
|
OutboundContext::MarkIntroBad(const Introduction& intro, llarp_time_t now)
|
|
{
|
|
// insert bad intro
|
|
m_BadIntros[intro] = now;
|
|
// try shifting intro without rebuild
|
|
if (ShiftIntroduction(false))
|
|
{
|
|
// we shifted
|
|
// check if we have a path to the next intro router
|
|
if (GetNewestPathByRouter(m_NextIntro.router))
|
|
return true;
|
|
// we don't have a path build one if we aren't building too fast
|
|
if (!BuildCooldownHit(now))
|
|
BuildOneAlignedTo(m_NextIntro.router);
|
|
return true;
|
|
}
|
|
|
|
// we didn't shift check if we should update introset
|
|
if (now - lastShift >= MIN_SHIFT_INTERVAL || currentIntroSet.HasExpiredIntros(now)
|
|
|| currentIntroSet.IsExpired(now))
|
|
{
|
|
// update introset
|
|
LogInfo(Name(), " updating introset");
|
|
UpdateIntroSet();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
OutboundContext::ShiftIntroduction(bool rebuild)
|
|
{
|
|
bool success = false;
|
|
auto now = Now();
|
|
if (now - lastShift < MIN_SHIFT_INTERVAL)
|
|
return false;
|
|
bool shifted = false;
|
|
std::vector<Introduction> intros = currentIntroSet.I;
|
|
if (intros.size() > 1)
|
|
{
|
|
std::random_device rd;
|
|
std::mt19937 g(rd());
|
|
std::shuffle(intros.begin(), intros.end(), g);
|
|
}
|
|
|
|
// to find a intro on the same router as before that is newer
|
|
for (const auto& intro : intros)
|
|
{
|
|
if (intro.ExpiresSoon(now))
|
|
continue;
|
|
if (m_Endpoint->SnodeBlacklist().count(intro.router))
|
|
continue;
|
|
if (m_BadIntros.find(intro) == m_BadIntros.end() && remoteIntro.router == intro.router)
|
|
{
|
|
if (intro.expiresAt > m_NextIntro.expiresAt)
|
|
{
|
|
success = true;
|
|
m_NextIntro = intro;
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
if (!success)
|
|
{
|
|
/// pick newer intro not on same router
|
|
for (const auto& intro : intros)
|
|
{
|
|
if (m_Endpoint->SnodeBlacklist().count(intro.router))
|
|
continue;
|
|
m_Endpoint->EnsureRouterIsKnown(intro.router);
|
|
if (intro.ExpiresSoon(now))
|
|
continue;
|
|
if (m_BadIntros.find(intro) == m_BadIntros.end() && m_NextIntro != intro)
|
|
{
|
|
if (intro.expiresAt > m_NextIntro.expiresAt)
|
|
{
|
|
shifted = intro.router != m_NextIntro.router;
|
|
m_NextIntro = intro;
|
|
success = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (m_NextIntro.router.IsZero())
|
|
return false;
|
|
if (shifted)
|
|
lastShift = now;
|
|
if (rebuild && !BuildCooldownHit(Now()))
|
|
BuildOneAlignedTo(m_NextIntro.router);
|
|
return success;
|
|
}
|
|
|
|
void
|
|
OutboundContext::HandlePathDied(path::Path_ptr path)
|
|
{
|
|
// unconditionally update introset
|
|
UpdateIntroSet();
|
|
const RouterID endpoint(path->Endpoint());
|
|
// if a path to our current intro died...
|
|
if (endpoint == remoteIntro.router)
|
|
{
|
|
// figure out how many paths to this router we have
|
|
size_t num = 0;
|
|
ForEachPath([&](const path::Path_ptr& p) {
|
|
if (p->Endpoint() == endpoint && p->IsReady())
|
|
++num;
|
|
});
|
|
// if we have more than two then we are probably fine
|
|
if (num > 2)
|
|
return;
|
|
// if we have one working one ...
|
|
if (num == 1)
|
|
{
|
|
num = 0;
|
|
ForEachPath([&](const path::Path_ptr& p) {
|
|
if (p->Endpoint() == endpoint)
|
|
++num;
|
|
});
|
|
// if we have 2 or more established or pending don't do anything
|
|
if (num > 2)
|
|
return;
|
|
BuildOneAlignedTo(endpoint);
|
|
}
|
|
else if (num == 0)
|
|
{
|
|
// we have no paths to this router right now
|
|
// hop off it
|
|
Introduction picked;
|
|
// get the latest intro that isn't on that endpoint
|
|
for (const auto& intro : currentIntroSet.I)
|
|
{
|
|
if (intro.router == endpoint)
|
|
continue;
|
|
if (intro.expiresAt > picked.expiresAt)
|
|
picked = intro;
|
|
}
|
|
// we got nothing
|
|
if (picked.router.IsZero())
|
|
{
|
|
return;
|
|
}
|
|
m_NextIntro = picked;
|
|
// check if we have a path to this router
|
|
num = 0;
|
|
ForEachPath([&](const path::Path_ptr& p) {
|
|
// don't count timed out paths
|
|
if (p->Status() != path::ePathTimeout && p->Endpoint() == m_NextIntro.router)
|
|
++num;
|
|
});
|
|
// build a path if one isn't already pending build or established
|
|
BuildOneAlignedTo(m_NextIntro.router);
|
|
}
|
|
}
|
|
}
|
|
|
|
bool
|
|
OutboundContext::HandleHiddenServiceFrame(path::Path_ptr p, const ProtocolFrame& frame)
|
|
{
|
|
return m_Endpoint->HandleHiddenServiceFrame(p, frame);
|
|
}
|
|
|
|
} // namespace service
|
|
|
|
} // namespace llarp
|