lokinet/llarp/service/endpoint.cpp

1354 lines
37 KiB
C++
Raw Normal View History

#include <chrono>
#include <memory>
#include <service/endpoint.hpp>
#include <dht/context.hpp>
#include <dht/key.hpp>
2018-12-12 00:48:54 +00:00
#include <dht/messages/findintro.hpp>
2019-01-16 00:24:16 +00:00
#include <dht/messages/findrouter.hpp>
#include <dht/messages/gotintro.hpp>
#include <dht/messages/gotrouter.hpp>
#include <dht/messages/pubintro.hpp>
#include <nodedb.hpp>
#include <profiling.hpp>
#include <router/abstractrouter.hpp>
#include <routing/dht_message.hpp>
2019-06-19 22:30:07 +00:00
#include <routing/path_transfer_message.hpp>
#include <service/endpoint_state.hpp>
#include <service/endpoint_util.hpp>
#include <service/hidden_service_address_lookup.hpp>
#include <service/outbound_context.hpp>
2018-12-12 02:15:08 +00:00
#include <service/protocol.hpp>
2019-09-01 13:26:16 +00:00
#include <util/thread/logic.hpp>
#include <util/str.hpp>
#include <util/buffer.hpp>
2019-09-01 12:38:03 +00:00
#include <util/meta/memfn.hpp>
#include <hook/shell.hpp>
#include <link/link_manager.hpp>
#include <tooling/dht_event.hpp>
2019-09-01 13:26:16 +00:00
2019-07-30 23:42:13 +00:00
#include <utility>
namespace llarp
{
namespace service
{
Endpoint::Endpoint(AbstractRouter* r, Context* parent)
: path::Builder(r, 3, path::default_len), context(parent), m_RecvQueue(128)
{
m_state = std::make_unique<EndpointState>();
m_state->m_Router = r;
m_state->m_Name = "endpoint";
2019-11-28 23:08:02 +00:00
m_RecvQueue.enable();
}
bool
2020-05-17 17:44:00 +00:00
Endpoint::Configure(const NetworkConfig& conf, [[maybe_unused]] const DnsConfig& dnsConf)
{
2020-06-08 12:42:10 +00:00
if (conf.m_Paths.has_value())
numPaths = *conf.m_Paths;
2020-06-08 12:42:10 +00:00
if (conf.m_Hops.has_value())
numHops = *conf.m_Hops;
2020-05-14 17:51:27 +00:00
return m_state->Configure(conf);
2018-08-09 19:02:17 +00:00
}
2019-04-08 12:01:52 +00:00
llarp_ev_loop_ptr
Endpoint::EndpointNetLoop()
{
if (m_state->m_IsolatedNetLoop)
return m_state->m_IsolatedNetLoop;
2019-07-06 17:03:40 +00:00
return Router()->netloop();
}
2018-08-16 14:34:15 +00:00
bool
Endpoint::NetworkIsIsolated() const
{
return m_state->m_IsolatedLogic.get() != nullptr && m_state->m_IsolatedNetLoop != nullptr;
2018-08-09 19:02:17 +00:00
}
2018-08-10 03:51:38 +00:00
bool
Endpoint::HasPendingPathToService(const Address& addr) const
{
return m_state->m_PendingServiceLookups.find(addr) != m_state->m_PendingServiceLookups.end();
2018-08-10 03:51:38 +00:00
}
void
2019-11-05 16:58:53 +00:00
Endpoint::RegenAndPublishIntroSet(bool forceRebuild)
{
2019-11-05 16:58:53 +00:00
const auto now = llarp::time_now_ms();
std::set<Introduction> introset;
if (!GetCurrentIntroductionsWithFilter(
introset, [now](const service::Introduction& intro) -> bool {
return not intro.ExpiresSoon(now, path::min_intro_lifetime);
}))
{
LogWarn(
"could not publish descriptors for endpoint ",
Name(),
" because we couldn't get enough valid introductions");
if (ShouldBuildMore(now) || forceRebuild)
2018-11-22 15:52:04 +00:00
ManualRebuild(1);
return;
}
introSet().I.clear();
for (auto& intro : introset)
{
introSet().I.emplace_back(std::move(intro));
}
if (introSet().I.size() == 0)
{
2019-04-21 15:40:32 +00:00
LogWarn("not enough intros to publish introset for ", Name());
if (ShouldBuildMore(now) || forceRebuild)
ManualRebuild(1);
return;
}
auto maybe = m_Identity.EncryptAndSignIntroSet(introSet(), now);
if (not maybe)
{
2020-01-27 21:30:41 +00:00
LogWarn("failed to generate introset for endpoint ", Name());
return;
}
if (PublishIntroSet(*maybe, Router()))
{
2019-04-21 15:40:32 +00:00
LogInfo("(re)publishing introset for endpoint ", Name());
}
else
{
2019-04-21 15:40:32 +00:00
LogWarn("failed to publish intro set for endpoint ", Name());
}
}
bool
Endpoint::IsReady() const
{
const auto now = Now();
if (introSet().I.size() == 0)
return false;
if (introSet().IsExpired(now))
return false;
return true;
}
bool
Endpoint::HasPendingRouterLookup(const RouterID remote) const
{
const auto& routers = m_state->m_PendingRouters;
return routers.find(remote) != routers.end();
}
bool
Endpoint::GetEndpointWithConvoTag(
const ConvoTag tag, llarp::AlignedBuffer<32>& addr, bool& snode) const
{
auto itr = Sessions().find(tag);
if (itr != Sessions().end())
{
snode = false;
addr = itr->second.remote.Addr();
return true;
}
2019-07-30 23:42:13 +00:00
for (const auto& item : m_state->m_SNodeSessions)
{
if (item.second.second == tag)
{
2019-07-30 23:42:13 +00:00
snode = true;
addr = item.first;
2019-07-30 23:42:13 +00:00
return true;
}
}
2019-07-30 23:42:13 +00:00
return false;
}
bool
Endpoint::IntrosetIsStale() const
{
return introSet().HasExpiredIntros(Now());
}
2019-02-11 17:14:43 +00:00
util::StatusObject
Endpoint::ExtractStatus() const
2019-02-08 19:43:25 +00:00
{
auto obj = path::Builder::ExtractStatus();
obj["identity"] = m_Identity.pub.Addr().ToString();
return m_state->ExtractStatus(obj);
2019-02-08 19:43:25 +00:00
}
2019-11-05 17:01:34 +00:00
void Endpoint::Tick(llarp_time_t)
{
2019-11-05 16:58:53 +00:00
const auto now = llarp::time_now_ms();
2019-04-23 16:13:22 +00:00
path::Builder::Tick(now);
2018-07-19 04:58:39 +00:00
// publish descriptors
if (ShouldPublishDescriptors(now))
{
2019-11-05 16:58:53 +00:00
RegenAndPublishIntroSet();
}
2019-02-21 19:26:59 +00:00
m_state->m_RemoteLookupFilter.Decay(now);
2018-12-13 12:27:14 +00:00
// expire snode sessions
EndpointUtil::ExpireSNodeSessions(now, m_state->m_SNodeSessions);
// expire pending tx
EndpointUtil::ExpirePendingTx(now, m_state->m_PendingLookups);
2018-08-14 21:17:18 +00:00
// expire pending router lookups
EndpointUtil::ExpirePendingRouterLookups(now, m_state->m_PendingRouters);
2018-08-14 21:17:18 +00:00
2019-02-05 14:50:33 +00:00
// deregister dead sessions
EndpointUtil::DeregisterDeadSessions(now, m_state->m_DeadSessions);
// tick remote sessions
EndpointUtil::TickRemoteSessions(
now, m_state->m_RemoteSessions, m_state->m_DeadSessions, Sessions());
2019-02-09 14:37:24 +00:00
// expire convotags
EndpointUtil::ExpireConvoSessions(now, Sessions());
2018-09-24 15:52:25 +00:00
}
bool
Endpoint::Stop()
{
// stop remote sessions
EndpointUtil::StopRemoteSessions(m_state->m_RemoteSessions);
// stop snode sessions
EndpointUtil::StopSnodeSessions(m_state->m_SNodeSessions);
if (m_OnDown)
m_OnDown->NotifyAsync(NotifyParams());
2019-04-21 15:40:32 +00:00
return path::Builder::Stop();
}
2018-07-18 03:10:21 +00:00
uint64_t
Endpoint::GenTXID()
{
uint64_t txid = randint();
const auto& lookups = m_state->m_PendingLookups;
while (lookups.find(txid) != lookups.end())
2018-07-18 03:10:21 +00:00
++txid;
return txid;
}
2018-07-16 03:32:13 +00:00
std::string
Endpoint::Name() const
{
return m_state->m_Name + ":" + m_Identity.pub.Name();
2018-07-16 03:32:13 +00:00
}
2018-08-04 02:59:32 +00:00
void
Endpoint::PutLookup(IServiceLookup* lookup, uint64_t txid)
{
m_state->m_PendingLookups.emplace(txid, std::unique_ptr<IServiceLookup>(lookup));
2018-08-04 02:59:32 +00:00
}
bool
2019-05-03 13:15:03 +00:00
Endpoint::HandleGotIntroMessage(dht::GotIntroMessage_constptr msg)
{
std::set<EncryptedIntroSet> remote;
for (const auto& introset : msg->found)
{
if (not introset.Verify(Now()))
{
2020-01-27 21:30:41 +00:00
LogError(Name(), " got invalid introset");
return false;
2018-07-18 22:50:05 +00:00
}
2019-07-06 17:03:40 +00:00
remote.insert(introset);
}
auto& lookups = m_state->m_PendingLookups;
auto itr = lookups.find(msg->txid);
if (itr == lookups.end())
2018-07-18 03:10:21 +00:00
{
LogWarn(
"invalid lookup response for hidden service endpoint ", Name(), " txid=", msg->txid);
2018-07-20 04:50:28 +00:00
return true;
2018-07-18 03:10:21 +00:00
}
std::unique_ptr<IServiceLookup> lookup = std::move(itr->second);
lookups.erase(itr);
if (not lookup->HandleResponse(remote))
lookups.emplace(msg->txid, std::move(lookup));
2018-08-14 21:17:18 +00:00
return true;
}
bool
Endpoint::HasInboundConvo(const Address& addr) const
{
for (const auto& item : Sessions())
{
if (item.second.remote.Addr() == addr && item.second.inbound)
return true;
}
return false;
}
2018-08-09 19:02:17 +00:00
void
Endpoint::PutSenderFor(const ConvoTag& tag, const ServiceInfo& info, bool inbound)
2018-08-09 19:02:17 +00:00
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2018-08-09 19:02:17 +00:00
{
itr = Sessions().emplace(tag, Session{}).first;
2019-07-01 13:44:25 +00:00
itr->second.inbound = inbound;
itr->second.remote = info;
2018-08-09 19:02:17 +00:00
}
2018-10-29 16:48:36 +00:00
itr->second.lastUsed = Now();
2018-08-09 19:02:17 +00:00
}
bool
Endpoint::GetSenderFor(const ConvoTag& tag, ServiceInfo& si) const
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2018-08-09 19:02:17 +00:00
return false;
si = itr->second.remote;
return true;
}
void
Endpoint::PutIntroFor(const ConvoTag& tag, const Introduction& intro)
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2018-08-09 19:02:17 +00:00
{
return;
2018-08-09 19:02:17 +00:00
}
itr->second.intro = intro;
2018-10-29 16:48:36 +00:00
itr->second.lastUsed = Now();
2018-08-09 19:02:17 +00:00
}
bool
Endpoint::GetIntroFor(const ConvoTag& tag, Introduction& intro) const
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2018-08-09 19:02:17 +00:00
return false;
intro = itr->second.intro;
return true;
}
2019-02-21 16:45:33 +00:00
void
Endpoint::PutReplyIntroFor(const ConvoTag& tag, const Introduction& intro)
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2019-02-21 16:45:33 +00:00
{
return;
2019-02-21 16:45:33 +00:00
}
itr->second.replyIntro = intro;
itr->second.lastUsed = Now();
2019-02-21 16:45:33 +00:00
}
bool
Endpoint::GetReplyIntroFor(const ConvoTag& tag, Introduction& intro) const
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2019-02-21 16:45:33 +00:00
return false;
intro = itr->second.replyIntro;
return true;
}
2018-08-09 19:02:17 +00:00
bool
Endpoint::GetConvoTagsForService(const Address& addr, std::set<ConvoTag>& tags) const
2018-08-09 19:02:17 +00:00
{
return EndpointUtil::GetConvoTagsForService(Sessions(), addr, tags);
2018-08-09 19:02:17 +00:00
}
bool
Endpoint::GetCachedSessionKeyFor(const ConvoTag& tag, SharedSecret& secret) const
2018-08-09 19:02:17 +00:00
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2018-08-09 19:02:17 +00:00
return false;
secret = itr->second.sharedKey;
2018-08-09 19:02:17 +00:00
return true;
}
void
Endpoint::PutCachedSessionKeyFor(const ConvoTag& tag, const SharedSecret& k)
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2018-08-09 19:02:17 +00:00
{
itr = Sessions().emplace(tag, Session{}).first;
2018-08-09 19:02:17 +00:00
}
itr->second.sharedKey = k;
itr->second.lastUsed = Now();
2018-08-09 19:02:17 +00:00
}
2019-09-19 20:28:12 +00:00
void
Endpoint::MarkConvoTagActive(const ConvoTag& tag)
{
auto itr = Sessions().find(tag);
if (itr != Sessions().end())
2019-09-19 20:28:12 +00:00
{
itr->second.lastUsed = Now();
}
}
bool
Endpoint::LoadKeyFile()
{
const auto& keyfile = m_state->m_Keyfile;
if (!keyfile.empty())
{
if (!m_Identity.EnsureKeys(keyfile, Router()->keyManager()->needBackup()))
{
LogError("Can't ensure keyfile [", keyfile, "]");
return false;
}
}
else
{
m_Identity.RegenerateKeys();
}
return true;
}
bool
Endpoint::Start()
{
m_state->m_RemoteLookupFilter.DecayInterval(500ms);
// how can I tell if a m_Identity isn't loaded?
if (!m_DataHandler)
2018-08-09 19:02:17 +00:00
{
m_DataHandler = this;
}
2018-08-16 14:34:15 +00:00
// this does network isolation
while (m_state->m_OnInit.size())
2018-08-09 19:02:17 +00:00
{
if (m_state->m_OnInit.front()())
m_state->m_OnInit.pop_front();
2018-08-09 19:02:17 +00:00
else
{
2019-04-21 15:40:32 +00:00
LogWarn("Can't call init of network isolation");
2018-08-09 19:02:17 +00:00
return false;
}
2018-08-09 19:02:17 +00:00
}
return true;
}
Endpoint::~Endpoint()
{
if (m_OnUp)
m_OnUp->Stop();
if (m_OnDown)
m_OnDown->Stop();
if (m_OnReady)
m_OnReady->Stop();
}
2018-07-18 03:10:21 +00:00
2020-02-10 17:52:24 +00:00
bool
Endpoint::PublishIntroSet(const EncryptedIntroSet& introset, AbstractRouter* r)
2020-02-10 17:52:24 +00:00
{
const auto paths = GetManyPathsWithUniqueEndpoints(this, llarp::dht::IntroSetRelayRedundancy);
if (paths.size() != llarp::dht::IntroSetRelayRedundancy)
{
LogWarn(
"Cannot publish intro set because we only have ",
paths.size(),
" paths, but need ",
llarp::dht::IntroSetRelayRedundancy);
return false;
}
2020-02-10 17:34:14 +00:00
// do publishing for each path selected
size_t published = 0;
for (const auto& path : paths)
2020-02-10 17:34:14 +00:00
{
for (size_t i = 0; i < llarp::dht::IntroSetRequestsPerRelay; ++i)
2020-02-10 17:34:14 +00:00
{
r->NotifyRouterEvent<tooling::PubIntroSentEvent>(
2020-03-07 01:20:11 +00:00
r->pubkey(),
llarp::dht::Key_t{introset.derivedSigningKey.as_array()},
RouterID(path->hops[path->hops.size() - 1].rc.pubkey),
published);
if (PublishIntroSetVia(introset, r, path, published))
published++;
2020-02-10 17:34:14 +00:00
}
}
if (published != llarp::dht::IntroSetStorageRedundancy)
LogWarn(
"Publish introset failed: could only publish ",
published,
" copies but wanted ",
llarp::dht::IntroSetStorageRedundancy);
return published == llarp::dht::IntroSetStorageRedundancy;
2018-07-18 03:10:21 +00:00
}
2018-09-18 14:48:06 +00:00
struct PublishIntroSetJob : public IServiceLookup
{
2020-01-27 21:30:41 +00:00
EncryptedIntroSet m_IntroSet;
2018-09-18 14:48:06 +00:00
Endpoint* m_Endpoint;
uint64_t m_relayOrder;
PublishIntroSetJob(
Endpoint* parent, uint64_t id, EncryptedIntroSet introset, uint64_t relayOrder)
2018-09-18 14:48:06 +00:00
: IServiceLookup(parent, id, "PublishIntroSet")
2019-07-30 23:42:13 +00:00
, m_IntroSet(std::move(introset))
2018-09-18 14:48:06 +00:00
, m_Endpoint(parent)
, m_relayOrder(relayOrder)
2018-09-18 14:48:06 +00:00
{
}
std::shared_ptr<routing::IMessage>
2019-07-30 23:42:13 +00:00
BuildRequestMessage() override
2018-09-18 14:48:06 +00:00
{
auto msg = std::make_shared<routing::DHTMessage>();
msg->M.emplace_back(
std::make_unique<dht::PublishIntroMessage>(m_IntroSet, txid, true, m_relayOrder));
2018-09-18 14:48:06 +00:00
return msg;
}
bool
HandleResponse(const std::set<EncryptedIntroSet>& response) override
2018-09-18 14:48:06 +00:00
{
if (not response.empty())
2018-09-18 14:48:06 +00:00
m_Endpoint->IntroSetPublished();
else
m_Endpoint->IntroSetPublishFail();
return true;
}
};
2018-07-18 03:10:21 +00:00
void
Endpoint::IntroSetPublishFail()
{
auto now = Now();
if (ShouldPublishDescriptors(now))
{
2019-11-05 16:58:53 +00:00
RegenAndPublishIntroSet();
}
else if (NumInStatus(path::ePathEstablished) < 3)
{
if (introSet().HasExpiredIntros(now))
ManualRebuild(1);
}
2018-09-18 14:48:06 +00:00
}
bool
Endpoint::PublishIntroSetVia(
const EncryptedIntroSet& introset,
AbstractRouter* r,
path::Path_ptr path,
uint64_t relayOrder)
2018-09-18 14:48:06 +00:00
{
2020-02-24 19:52:49 +00:00
auto job = new PublishIntroSetJob(this, GenTXID(), introset, relayOrder);
if (job->SendRequestViaPath(path, r))
2018-09-18 14:48:06 +00:00
{
m_state->m_LastPublishAttempt = Now();
2018-09-18 14:48:06 +00:00
return true;
}
return false;
2018-07-18 03:10:21 +00:00
}
2019-05-07 17:46:38 +00:00
void
Endpoint::ResetInternalState()
{
path::Builder::ResetInternalState();
static auto resetState = [](auto& container, auto getter) {
std::for_each(container.begin(), container.end(), [getter](auto& item) {
getter(item)->ResetInternalState();
});
2019-05-07 17:46:38 +00:00
};
resetState(m_state->m_RemoteSessions, [](const auto& item) { return item.second; });
resetState(m_state->m_SNodeSessions, [](const auto& item) { return item.second.first; });
2019-05-07 17:46:38 +00:00
}
2018-07-18 03:10:21 +00:00
bool
2018-07-18 22:50:05 +00:00
Endpoint::ShouldPublishDescriptors(llarp_time_t now) const
2018-07-18 03:10:21 +00:00
{
if (not m_PublishIntroSet)
return false;
auto next_pub = m_state->m_LastPublishAttempt
+ (m_state->m_IntroSet.HasExpiredIntros(now) ? INTROSET_PUBLISH_RETRY_INTERVAL
: INTROSET_PUBLISH_INTERVAL);
return now >= next_pub;
2018-07-18 03:10:21 +00:00
}
void
Endpoint::IntroSetPublished()
{
const auto now = Now();
// We usually get 4 confirmations back (one for each DHT location), which
// is noisy: suppress this log message if we already had a confirmation in
// the last second.
if (m_state->m_LastPublish < now - 1s)
LogInfo(Name(), " IntroSet publish confirmed");
else
LogDebug(Name(), " Additional IntroSet publish confirmed");
m_state->m_LastPublish = now;
if (m_OnReady)
m_OnReady->NotifyAsync(NotifyParams());
m_OnReady = nullptr;
2018-07-18 03:10:21 +00:00
}
void
Endpoint::IsolatedNetworkMainLoop()
{
m_state->m_IsolatedNetLoop = llarp_make_ev_loop();
m_state->m_IsolatedLogic = std::make_shared<llarp::Logic>();
if (SetupNetworking())
llarp_ev_loop_run_single_process(m_state->m_IsolatedNetLoop, m_state->m_IsolatedLogic);
else
{
m_state->m_IsolatedNetLoop.reset();
m_state->m_IsolatedLogic.reset();
}
2018-08-09 19:02:17 +00:00
}
2019-05-10 16:19:33 +00:00
bool
Endpoint::SelectHop(
llarp_nodedb* db,
const std::set<RouterID>& prev,
RouterContact& cur,
size_t hop,
path::PathRole roles)
2019-05-10 16:19:33 +00:00
{
std::set<RouterID> exclude = prev;
for (const auto& snode : SnodeBlacklist())
2019-05-10 16:19:33 +00:00
exclude.insert(snode);
if (hop == numHops - 1 and numHops > 1)
2020-01-06 15:28:37 +00:00
{
// diversify endpoints
ForEachPath([&exclude](const path::Path_ptr& path) { exclude.insert(path->Endpoint()); });
2020-01-06 15:28:37 +00:00
}
2019-05-10 16:19:33 +00:00
return path::Builder::SelectHop(db, exclude, cur, hop, roles);
}
void
Endpoint::PathBuildStarted(path::Path_ptr path)
{
path::Builder::PathBuildStarted(path);
}
2018-07-22 23:14:29 +00:00
void
2019-04-21 15:40:32 +00:00
Endpoint::PutNewOutboundContext(const service::IntroSet& introset)
2018-07-22 23:14:29 +00:00
{
Address addr;
introset.A.CalculateAddress(addr.as_array());
2018-07-22 23:14:29 +00:00
auto& remoteSessions = m_state->m_RemoteSessions;
auto& serviceLookups = m_state->m_PendingServiceLookups;
if (remoteSessions.count(addr) >= MAX_OUTBOUND_CONTEXT_COUNT)
2018-07-22 23:14:29 +00:00
{
auto itr = remoteSessions.find(addr);
auto range = serviceLookups.equal_range(addr);
auto i = range.first;
while (i != range.second)
{
i->second(addr, itr->second.get());
++i;
}
serviceLookups.erase(addr);
return;
2018-07-22 23:14:29 +00:00
}
auto it = remoteSessions.emplace(addr, std::make_shared<OutboundContext>(introset, this));
2019-04-21 15:40:32 +00:00
LogInfo("Created New outbound context for ", addr.ToString());
2018-07-22 23:14:29 +00:00
// inform pending
auto range = serviceLookups.equal_range(addr);
auto itr = range.first;
if (itr != range.second)
2018-07-22 23:14:29 +00:00
{
itr->second(addr, it->second.get());
++itr;
2018-07-22 23:14:29 +00:00
}
serviceLookups.erase(addr);
2018-07-22 23:14:29 +00:00
}
void
Endpoint::HandleVerifyGotRouter(dht::GotRouterMessage_constptr msg, llarp_async_verify_rc* j)
{
auto& pendingRouters = m_state->m_PendingRouters;
auto itr = pendingRouters.find(j->rc.pubkey);
if (itr != pendingRouters.end())
{
if (j->valid)
itr->second.InformResult(msg->foundRCs);
else
itr->second.InformResult({});
pendingRouters.erase(itr);
}
delete j;
}
2018-08-10 21:34:11 +00:00
bool
2019-05-03 13:15:03 +00:00
Endpoint::HandleGotRouterMessage(dht::GotRouterMessage_constptr msg)
2018-08-10 21:34:11 +00:00
{
if (not msg->foundRCs.empty())
{
for (const auto& rc : msg->foundRCs)
{
llarp_async_verify_rc* job = new llarp_async_verify_rc();
job->nodedb = Router()->nodedb();
job->cryptoworker = Router()->threadpool();
job->diskworker = Router()->diskworker();
job->logic = Router()->logic();
job->hook = std::bind(&Endpoint::HandleVerifyGotRouter, this, msg, std::placeholders::_1);
job->rc = rc;
llarp_nodedb_async_verify(job);
}
2019-05-03 13:15:03 +00:00
}
else
{
auto& routers = m_state->m_PendingRouters;
auto itr = routers.begin();
while (itr != routers.end())
{
if (itr->second.txid == msg->txid)
{
itr->second.InformResult({});
itr = routers.erase(itr);
}
else
++itr;
}
2018-08-10 21:34:11 +00:00
}
return true;
2018-08-10 21:34:11 +00:00
}
void
Endpoint::EnsureRouterIsKnown(const RouterID& router)
{
if (router.IsZero())
2018-08-14 22:07:58 +00:00
return;
if (!Router()->nodedb()->Has(router))
2018-08-10 21:34:11 +00:00
{
2019-05-03 13:15:03 +00:00
LookupRouterAnon(router, nullptr);
2018-12-19 17:48:29 +00:00
}
}
2018-08-10 21:34:11 +00:00
2018-12-19 17:48:29 +00:00
bool
2019-05-03 13:15:03 +00:00
Endpoint::LookupRouterAnon(RouterID router, RouterLookupHandler handler)
2018-12-19 17:48:29 +00:00
{
using llarp::dht::FindRouterMessage;
auto& routers = m_state->m_PendingRouters;
if (routers.find(router) == routers.end())
2018-12-19 17:48:29 +00:00
{
auto path = GetEstablishedPathClosestTo(router);
routing::DHTMessage msg;
auto txid = GenTXID();
msg.M.emplace_back(std::make_unique<FindRouterMessage>(txid, router));
2018-12-19 17:48:29 +00:00
if (path && path->SendRoutingMessage(msg, Router()))
2018-12-19 17:48:29 +00:00
{
RouterLookupJob job(this, handler);
assert(msg.M.size() == 1);
auto dhtMsg = dynamic_cast<FindRouterMessage*>(msg.M[0].get());
assert(dhtMsg != nullptr);
m_router->NotifyRouterEvent<tooling::FindRouterSentEvent>(m_router->pubkey(), *dhtMsg);
routers.emplace(router, RouterLookupJob(this, handler));
2018-12-19 17:48:29 +00:00
return true;
2018-08-10 21:34:11 +00:00
}
}
2018-12-19 17:48:29 +00:00
return false;
2018-08-10 21:34:11 +00:00
}
void
Endpoint::HandlePathBuilt(path::Path_ptr p)
{
2019-06-02 21:19:10 +00:00
p->SetDataHandler(util::memFn(&Endpoint::HandleHiddenServiceFrame, this));
p->SetDropHandler(util::memFn(&Endpoint::HandleDataDrop, this));
p->SetDeadChecker(util::memFn(&Endpoint::CheckPathIsDead, this));
path::Builder::HandlePathBuilt(p);
}
bool
Endpoint::HandleDataDrop(path::Path_ptr p, const PathID_t& dst, uint64_t seq)
{
LogWarn(Name(), " message ", seq, " dropped by endpoint ", p->Endpoint(), " via ", dst);
return true;
}
std::unordered_map<std::string, std::string>
Endpoint::NotifyParams() const
{
return {{"LOKINET_ADDR", m_Identity.pub.Addr().ToString()}};
}
2019-11-28 23:08:02 +00:00
void
Endpoint::FlushRecvData()
{
do
{
auto maybe = m_RecvQueue.tryPopFront();
if (not maybe)
2019-11-28 23:08:02 +00:00
return;
auto ev = std::move(*maybe);
2019-11-28 23:08:02 +00:00
ProtocolMessage::ProcessAsync(ev.fromPath, ev.pathid, ev.msg);
} while (true);
2019-11-28 23:08:02 +00:00
}
void
Endpoint::QueueRecvData(RecvDataEvent ev)
{
if (m_RecvQueue.full() || m_RecvQueue.empty())
2019-11-28 23:08:02 +00:00
{
auto self = this;
LogicCall(m_router->logic(), [self]() { self->FlushRecvData(); });
}
m_RecvQueue.pushBack(std::move(ev));
}
bool
Endpoint::HandleDataMessage(
path::Path_ptr path, const PathID_t from, std::shared_ptr<ProtocolMessage> msg)
{
2019-06-06 10:52:27 +00:00
msg->sender.UpdateAddr();
PutSenderFor(msg->tag, msg->sender, true);
2019-06-28 14:12:20 +00:00
PutReplyIntroFor(msg->tag, path->intro);
Introduction intro;
intro.pathID = from;
intro.router = PubKey(path->Endpoint());
2019-06-28 14:48:00 +00:00
intro.expiresAt = std::min(path->ExpireTime(), msg->introReply.expiresAt);
2019-06-28 14:12:20 +00:00
PutIntroFor(msg->tag, intro);
2018-09-18 17:48:26 +00:00
return ProcessDataMessage(msg);
}
2018-11-29 14:01:13 +00:00
bool
2019-07-01 13:44:25 +00:00
Endpoint::HasPathToSNode(const RouterID ident) const
2018-11-29 14:01:13 +00:00
{
auto range = m_state->m_SNodeSessions.equal_range(ident);
auto itr = range.first;
while (itr != range.second)
{
if (itr->second.first->IsReady())
2018-11-29 14:01:13 +00:00
{
return true;
}
++itr;
}
return false;
}
2018-11-29 13:12:35 +00:00
bool
Endpoint::ProcessDataMessage(std::shared_ptr<ProtocolMessage> msg)
2018-11-29 13:12:35 +00:00
{
2020-05-21 14:18:23 +00:00
if ((msg->proto == eProtocolExit
&& (m_state->m_ExitEnabled || msg->sender.Addr() == m_state->m_ExitNode))
|| msg->proto == eProtocolTrafficV4 || msg->proto == eProtocolTrafficV6)
2018-11-29 13:12:35 +00:00
{
De-abseil, part 2: mutex, locks, (most) time - util::Mutex is now a std::shared_timed_mutex, which is capable of exclusive and shared locks. - util::Lock is still present as a std::lock_guard<util::Mutex>. - the locking annotations are preserved, but updated to the latest supported by clang rather than using abseil's older/deprecated ones. - ACQUIRE_LOCK macro is gone since we don't pass mutexes by pointer into locks anymore (WTF abseil). - ReleasableLock is gone. Instead there are now some llarp::util helper methods to obtain unique and/or shared locks: - `auto lock = util::unique_lock(mutex);` gets an RAII-but-also unlockable object (std::unique_lock<T>, with T inferred from `mutex`). - `auto lock = util::shared_lock(mutex);` gets an RAII shared (i.e. "reader") lock of the mutex. - `auto lock = util::unique_locks(mutex1, mutex2, mutex3);` can be used to atomically lock multiple mutexes at once (returning a tuple of the locks). This are templated on the mutex which makes them a bit more flexible than using a concrete type: they can be used for any type of lockable mutex, not only util::Mutex. (Some of the code here uses them for getting locks around a std::mutex). Until C++17, using the RAII types is painfully verbose: ```C++ // pre-C++17 - needing to figure out the mutex type here is annoying: std::unique_lock<util::Mutex> lock(mutex); // pre-C++17 and even more verbose (but at least the type isn't needed): std::unique_lock<decltype(mutex)> lock(mutex); // our compromise: auto lock = util::unique_lock(mutex); // C++17: std::unique_lock lock(mutex); ``` All of these functions will also warn (under gcc or clang) if you discard the return value. You can also do fancy things like `auto l = util::unique_lock(mutex, std::adopt_lock)` (which lets a lock take over an already-locked mutex). - metrics code is gone, which also removes a big pile of code that was only used by metrics: - llarp::util::Scheduler - llarp::thread::TimerQueue - llarp::util::Stopwatch
2020-02-21 17:21:11 +00:00
util::Lock l(m_state->m_InboundTrafficQueueMutex);
m_state->m_InboundTrafficQueue.emplace(msg);
2019-05-22 17:47:33 +00:00
return true;
2018-11-29 13:12:35 +00:00
}
if (msg->proto == eProtocolControl)
2018-11-29 13:12:35 +00:00
{
// TODO: implement me (?)
// right now it's just random noise
2018-11-29 13:12:35 +00:00
return true;
}
return false;
}
2019-03-08 16:00:45 +00:00
void
Endpoint::RemoveConvoTag(const ConvoTag& t)
{
Sessions().erase(t);
2019-03-08 16:00:45 +00:00
}
bool
Endpoint::HandleHiddenServiceFrame(path::Path_ptr p, const ProtocolFrame& frame)
{
if (frame.R)
2019-03-08 16:00:45 +00:00
{
// handle discard
ServiceInfo si;
if (!GetSenderFor(frame.T, si))
2019-03-08 16:00:45 +00:00
return false;
// verify source
if (!frame.Verify(si))
2019-03-08 16:00:45 +00:00
return false;
// remove convotag it doesn't exist
LogWarn("remove convotag T=", frame.T);
RemoveConvoTag(frame.T);
2019-03-08 16:00:45 +00:00
return true;
}
if (!frame.AsyncDecryptAndVerify(
EndpointLogic(), p, CryptoWorker(), m_Identity, m_DataHandler))
2019-03-08 16:00:45 +00:00
{
// send discard
ProtocolFrame f;
f.R = 1;
f.T = frame.T;
2019-03-08 16:00:45 +00:00
f.F = p->intro.pathID;
if (!f.Sign(m_Identity))
2019-03-08 16:00:45 +00:00
return false;
2019-05-02 16:23:31 +00:00
{
2020-02-17 22:33:45 +00:00
LogWarn("invalidating convotag T=", frame.T);
De-abseil, part 2: mutex, locks, (most) time - util::Mutex is now a std::shared_timed_mutex, which is capable of exclusive and shared locks. - util::Lock is still present as a std::lock_guard<util::Mutex>. - the locking annotations are preserved, but updated to the latest supported by clang rather than using abseil's older/deprecated ones. - ACQUIRE_LOCK macro is gone since we don't pass mutexes by pointer into locks anymore (WTF abseil). - ReleasableLock is gone. Instead there are now some llarp::util helper methods to obtain unique and/or shared locks: - `auto lock = util::unique_lock(mutex);` gets an RAII-but-also unlockable object (std::unique_lock<T>, with T inferred from `mutex`). - `auto lock = util::shared_lock(mutex);` gets an RAII shared (i.e. "reader") lock of the mutex. - `auto lock = util::unique_locks(mutex1, mutex2, mutex3);` can be used to atomically lock multiple mutexes at once (returning a tuple of the locks). This are templated on the mutex which makes them a bit more flexible than using a concrete type: they can be used for any type of lockable mutex, not only util::Mutex. (Some of the code here uses them for getting locks around a std::mutex). Until C++17, using the RAII types is painfully verbose: ```C++ // pre-C++17 - needing to figure out the mutex type here is annoying: std::unique_lock<util::Mutex> lock(mutex); // pre-C++17 and even more verbose (but at least the type isn't needed): std::unique_lock<decltype(mutex)> lock(mutex); // our compromise: auto lock = util::unique_lock(mutex); // C++17: std::unique_lock lock(mutex); ``` All of these functions will also warn (under gcc or clang) if you discard the return value. You can also do fancy things like `auto l = util::unique_lock(mutex, std::adopt_lock)` (which lets a lock take over an already-locked mutex). - metrics code is gone, which also removes a big pile of code that was only used by metrics: - llarp::util::Scheduler - llarp::thread::TimerQueue - llarp::util::Stopwatch
2020-02-21 17:21:11 +00:00
util::Lock lock(m_state->m_SendQueueMutex);
m_state->m_SendQueue.emplace_back(
std::make_shared<const routing::PathTransferMessage>(f, frame.F), p);
2019-05-02 16:23:31 +00:00
}
return true;
2019-03-08 16:00:45 +00:00
}
return true;
}
2019-04-23 16:13:22 +00:00
void Endpoint::HandlePathDied(path::Path_ptr)
2018-09-17 15:32:37 +00:00
{
2019-11-05 16:58:53 +00:00
RegenAndPublishIntroSet(true);
2019-03-30 13:02:10 +00:00
}
bool
Endpoint::CheckPathIsDead(path::Path_ptr, llarp_time_t dlt)
{
2019-04-05 14:58:22 +00:00
return dlt > path::alive_timeout;
}
2018-08-10 21:34:11 +00:00
bool
Endpoint::OnLookup(
const Address& addr, std::optional<IntroSet> introset, const RouterID& endpoint)
2018-08-10 21:34:11 +00:00
{
2019-08-02 09:27:27 +00:00
const auto now = Router()->Now();
auto& fails = m_state->m_ServiceLookupFails;
auto& lookups = m_state->m_PendingServiceLookups;
if (not introset or introset->IsExpired(now))
{
LogError(Name(), " failed to lookup ", addr.ToString(), " from ", endpoint);
fails[endpoint] = fails[endpoint] + 1;
// inform one
2019-07-29 15:10:20 +00:00
auto range = lookups.equal_range(addr);
auto itr = range.first;
if (itr != range.second)
{
itr->second(addr, nullptr);
2019-07-29 15:10:20 +00:00
itr = lookups.erase(itr);
}
2018-08-10 21:34:11 +00:00
return false;
}
2020-03-02 16:56:47 +00:00
// check for established outbound context
if (m_state->m_RemoteSessions.count(addr) > 0)
2020-03-02 16:56:47 +00:00
return true;
PutNewOutboundContext(*introset);
2018-08-10 21:34:11 +00:00
return true;
}
2020-02-18 16:00:45 +00:00
void
Endpoint::MarkAddressOutbound(const Address& addr)
{
m_state->m_OutboundSessions.insert(addr);
}
bool
Endpoint::WantsOutboundSession(const Address& addr) const
{
return m_state->m_OutboundSessions.count(addr) > 0;
}
2018-07-19 04:58:39 +00:00
bool
Endpoint::EnsurePathToService(
const Address remote, PathEnsureHook hook, llarp_time_t /*timeoutMS*/)
2018-07-19 04:58:39 +00:00
{
/// how many routers to use for lookups
2020-03-02 16:17:50 +00:00
static constexpr size_t NumParallelLookups = 2;
/// how many requests per router
static constexpr size_t RequestsPerLookup = 2;
2020-02-18 16:00:45 +00:00
MarkAddressOutbound(remote);
auto& sessions = m_state->m_RemoteSessions;
2018-07-22 23:14:29 +00:00
{
auto itr = sessions.find(remote);
if (itr != sessions.end())
2018-07-22 23:14:29 +00:00
{
2018-08-22 15:52:10 +00:00
hook(itr->first, itr->second.get());
2018-07-22 23:14:29 +00:00
return true;
}
}
// filter check for address
if (not m_state->m_RemoteLookupFilter.Insert(remote))
return false;
auto& lookups = m_state->m_PendingServiceLookups;
const auto paths = GetManyPathsWithUniqueEndpoints(this, NumParallelLookups);
using namespace std::placeholders;
size_t lookedUp = 0;
2020-02-10 17:52:24 +00:00
const dht::Key_t location = remote.ToKey();
uint64_t order = 0;
for (const auto& path : paths)
{
for (size_t count = 0; count < RequestsPerLookup; ++count)
2020-02-10 17:52:24 +00:00
{
2020-03-02 16:12:29 +00:00
HiddenServiceAddressLookup* job = new HiddenServiceAddressLookup(
this,
util::memFn(&Endpoint::OnLookup, this),
location,
PubKey{remote.as_array()},
order,
GenTXID());
LogInfo(
"doing lookup for ",
remote,
" via ",
path->Endpoint(),
" at ",
location,
" order=",
order);
2020-03-02 16:12:29 +00:00
order++;
if (job->SendRequestViaPath(path, Router()))
2020-03-02 16:12:29 +00:00
{
lookups.emplace(remote, hook);
lookedUp++;
}
else
LogError(Name(), " send via path failed for lookup");
2020-02-10 17:52:24 +00:00
}
}
2020-03-02 16:18:26 +00:00
return lookedUp == (NumParallelLookups * RequestsPerLookup);
2018-07-19 04:58:39 +00:00
}
bool
2019-07-01 13:44:25 +00:00
Endpoint::EnsurePathToSNode(const RouterID snode, SNodeEnsureHook h)
2018-11-29 13:12:35 +00:00
{
static constexpr size_t MaxConcurrentSNodeSessions = 16;
auto& nodeSessions = m_state->m_SNodeSessions;
if (nodeSessions.size() >= MaxConcurrentSNodeSessions)
{
// a quick client side work arround before we do proper limiting
LogError(Name(), " has too many snode sessions");
return false;
}
using namespace std::placeholders;
if (nodeSessions.count(snode) == 0)
2018-11-29 13:12:35 +00:00
{
ConvoTag tag;
// TODO: check for collision lol no we don't but maybe we will...
// some day :DDDDD
tag.Randomize();
auto session = std::make_shared<exit::SNodeSession>(
snode,
2019-07-01 13:44:25 +00:00
[=](const llarp_buffer_t& pkt) -> bool {
/// TODO: V6
return HandleInboundPacket(tag, pkt, eProtocolTrafficV4);
2019-07-01 13:44:25 +00:00
},
Router(),
numPaths,
numHops,
false,
ShouldBundleRC());
m_state->m_SNodeSessions.emplace(snode, std::make_pair(session, tag));
2018-11-29 13:12:35 +00:00
}
2019-04-30 21:36:27 +00:00
EnsureRouterIsKnown(snode);
auto range = nodeSessions.equal_range(snode);
auto itr = range.first;
while (itr != range.second)
{
if (itr->second.first->IsReady())
h(snode, itr->second.first);
2019-03-07 15:17:29 +00:00
else
2019-04-30 21:36:27 +00:00
{
itr->second.first->AddReadyHook(std::bind(h, snode, _1));
itr->second.first->BuildOne();
2019-04-30 21:36:27 +00:00
}
++itr;
}
return true;
2018-11-29 13:12:35 +00:00
}
bool
Endpoint::SendToSNodeOrQueue(const RouterID& addr, const llarp_buffer_t& buf)
2018-11-29 13:12:35 +00:00
{
auto pkt = std::make_shared<net::IPPacket>();
if (!pkt->Load(buf))
2018-11-29 13:12:35 +00:00
return false;
2019-04-30 21:36:27 +00:00
EnsurePathToSNode(addr, [pkt](RouterID, exit::BaseSession_ptr s) {
if (s)
2019-04-30 21:36:27 +00:00
s->QueueUpstreamTraffic(*pkt, routing::ExitPadSize);
});
return true;
2018-11-29 13:12:35 +00:00
}
2019-04-30 16:07:17 +00:00
void Endpoint::Pump(llarp_time_t)
2019-04-25 17:15:56 +00:00
{
const auto& sessions = m_state->m_SNodeSessions;
auto& queue = m_state->m_InboundTrafficQueue;
auto epPump = [&]() {
2019-11-28 23:08:02 +00:00
FlushRecvData();
2019-05-22 16:20:50 +00:00
// send downstream packets to user for snode
for (const auto& item : sessions)
item.second.first->FlushDownstream();
// send downstream traffic to user for hidden service
De-abseil, part 2: mutex, locks, (most) time - util::Mutex is now a std::shared_timed_mutex, which is capable of exclusive and shared locks. - util::Lock is still present as a std::lock_guard<util::Mutex>. - the locking annotations are preserved, but updated to the latest supported by clang rather than using abseil's older/deprecated ones. - ACQUIRE_LOCK macro is gone since we don't pass mutexes by pointer into locks anymore (WTF abseil). - ReleasableLock is gone. Instead there are now some llarp::util helper methods to obtain unique and/or shared locks: - `auto lock = util::unique_lock(mutex);` gets an RAII-but-also unlockable object (std::unique_lock<T>, with T inferred from `mutex`). - `auto lock = util::shared_lock(mutex);` gets an RAII shared (i.e. "reader") lock of the mutex. - `auto lock = util::unique_locks(mutex1, mutex2, mutex3);` can be used to atomically lock multiple mutexes at once (returning a tuple of the locks). This are templated on the mutex which makes them a bit more flexible than using a concrete type: they can be used for any type of lockable mutex, not only util::Mutex. (Some of the code here uses them for getting locks around a std::mutex). Until C++17, using the RAII types is painfully verbose: ```C++ // pre-C++17 - needing to figure out the mutex type here is annoying: std::unique_lock<util::Mutex> lock(mutex); // pre-C++17 and even more verbose (but at least the type isn't needed): std::unique_lock<decltype(mutex)> lock(mutex); // our compromise: auto lock = util::unique_lock(mutex); // C++17: std::unique_lock lock(mutex); ``` All of these functions will also warn (under gcc or clang) if you discard the return value. You can also do fancy things like `auto l = util::unique_lock(mutex, std::adopt_lock)` (which lets a lock take over an already-locked mutex). - metrics code is gone, which also removes a big pile of code that was only used by metrics: - llarp::util::Scheduler - llarp::thread::TimerQueue - llarp::util::Stopwatch
2020-02-21 17:21:11 +00:00
util::Lock lock(m_state->m_InboundTrafficQueueMutex);
while (not queue.empty())
{
const auto& msg = queue.top();
2019-07-18 16:28:17 +00:00
const llarp_buffer_t buf(msg->payload);
HandleInboundPacket(msg->tag, buf, msg->proto);
queue.pop();
}
};
if (NetworkIsIsolated())
{
LogicCall(EndpointLogic(), epPump);
}
else
{
epPump();
}
2019-04-30 16:07:17 +00:00
auto router = Router();
// TODO: locking on this container
for (const auto& item : m_state->m_RemoteSessions)
2019-04-30 16:07:17 +00:00
item.second->FlushUpstream();
// TODO: locking on this container
for (const auto& item : sessions)
item.second.first->FlushUpstream();
2019-09-19 20:28:12 +00:00
{
De-abseil, part 2: mutex, locks, (most) time - util::Mutex is now a std::shared_timed_mutex, which is capable of exclusive and shared locks. - util::Lock is still present as a std::lock_guard<util::Mutex>. - the locking annotations are preserved, but updated to the latest supported by clang rather than using abseil's older/deprecated ones. - ACQUIRE_LOCK macro is gone since we don't pass mutexes by pointer into locks anymore (WTF abseil). - ReleasableLock is gone. Instead there are now some llarp::util helper methods to obtain unique and/or shared locks: - `auto lock = util::unique_lock(mutex);` gets an RAII-but-also unlockable object (std::unique_lock<T>, with T inferred from `mutex`). - `auto lock = util::shared_lock(mutex);` gets an RAII shared (i.e. "reader") lock of the mutex. - `auto lock = util::unique_locks(mutex1, mutex2, mutex3);` can be used to atomically lock multiple mutexes at once (returning a tuple of the locks). This are templated on the mutex which makes them a bit more flexible than using a concrete type: they can be used for any type of lockable mutex, not only util::Mutex. (Some of the code here uses them for getting locks around a std::mutex). Until C++17, using the RAII types is painfully verbose: ```C++ // pre-C++17 - needing to figure out the mutex type here is annoying: std::unique_lock<util::Mutex> lock(mutex); // pre-C++17 and even more verbose (but at least the type isn't needed): std::unique_lock<decltype(mutex)> lock(mutex); // our compromise: auto lock = util::unique_lock(mutex); // C++17: std::unique_lock lock(mutex); ``` All of these functions will also warn (under gcc or clang) if you discard the return value. You can also do fancy things like `auto l = util::unique_lock(mutex, std::adopt_lock)` (which lets a lock take over an already-locked mutex). - metrics code is gone, which also removes a big pile of code that was only used by metrics: - llarp::util::Scheduler - llarp::thread::TimerQueue - llarp::util::Stopwatch
2020-02-21 17:21:11 +00:00
util::Lock lock(m_state->m_SendQueueMutex);
// send outbound traffic
for (const auto& item : m_state->m_SendQueue)
{
item.second->SendRoutingMessage(*item.first, router);
MarkConvoTagActive(item.first->T.T);
}
m_state->m_SendQueue.clear();
2019-09-19 20:28:12 +00:00
}
2019-12-30 13:20:50 +00:00
UpstreamFlush(router);
router->linkManager().PumpLinks();
2019-04-25 17:15:56 +00:00
}
bool
Endpoint::EnsureConvo(
const AlignedBuffer<32> /*addr*/, bool snode, ConvoEventListener_ptr /*ev*/)
{
if (snode)
{
}
// TODO: something meaningful
return false;
}
2018-08-22 15:52:10 +00:00
bool
Endpoint::SendToServiceOrQueue(
const service::Address& remote, const llarp_buffer_t& data, ProtocolType t)
2018-08-22 15:52:10 +00:00
{
if (data.sz == 0)
2019-11-29 00:37:58 +00:00
return false;
// inbound converstation
2019-07-18 16:28:17 +00:00
const auto now = Now();
2018-11-14 12:23:08 +00:00
if (HasInboundConvo(remote))
{
auto transfer = std::make_shared<routing::PathTransferMessage>();
ProtocolFrame& f = transfer->T;
std::shared_ptr<path::Path> p;
std::set<ConvoTag> tags;
if (GetConvoTagsForService(remote, tags))
{
2019-06-28 14:12:20 +00:00
// the remote guy's intro
Introduction remoteIntro;
2019-06-28 14:12:20 +00:00
Introduction replyPath;
SharedSecret K;
// pick tag
for (const auto& tag : tags)
{
if (tag.IsZero())
continue;
if (!GetCachedSessionKeyFor(tag, K))
continue;
if (!GetReplyIntroFor(tag, replyPath))
2019-06-28 14:12:20 +00:00
continue;
if (!GetIntroFor(tag, remoteIntro))
2019-06-28 14:12:20 +00:00
continue;
// get path for intro
ForEachPath([&](path::Path_ptr path) {
if (path->intro == replyPath)
{
p = path;
2019-07-01 20:45:00 +00:00
return;
}
if (p && p->ExpiresSoon(now) && path->IsReady()
&& path->intro.router == replyPath.router)
{
2019-06-28 14:12:20 +00:00
p = path;
}
2019-06-28 14:12:20 +00:00
});
if (p)
{
2019-06-28 14:12:20 +00:00
f.T = tag;
}
}
if (p)
{
// TODO: check expiration of our end
auto m = std::make_shared<ProtocolMessage>(f.T);
m->PutBuffer(data);
f.N.Randomize();
f.C.Zero();
transfer->Y.Randomize();
m->proto = t;
m->introReply = p->intro;
PutReplyIntroFor(f.T, m->introReply);
m->sender = m_Identity.pub;
m->seqno = GetSeqNoForConvo(f.T);
f.S = 1;
f.F = m->introReply.pathID;
transfer->P = remoteIntro.pathID;
auto self = this;
return CryptoWorker()->addJob([transfer, p, m, K, self]() {
if (not transfer->T.EncryptAndSign(*m, K, self->m_Identity))
{
LogError("failed to encrypt and sign");
return;
}
De-abseil, part 2: mutex, locks, (most) time - util::Mutex is now a std::shared_timed_mutex, which is capable of exclusive and shared locks. - util::Lock is still present as a std::lock_guard<util::Mutex>. - the locking annotations are preserved, but updated to the latest supported by clang rather than using abseil's older/deprecated ones. - ACQUIRE_LOCK macro is gone since we don't pass mutexes by pointer into locks anymore (WTF abseil). - ReleasableLock is gone. Instead there are now some llarp::util helper methods to obtain unique and/or shared locks: - `auto lock = util::unique_lock(mutex);` gets an RAII-but-also unlockable object (std::unique_lock<T>, with T inferred from `mutex`). - `auto lock = util::shared_lock(mutex);` gets an RAII shared (i.e. "reader") lock of the mutex. - `auto lock = util::unique_locks(mutex1, mutex2, mutex3);` can be used to atomically lock multiple mutexes at once (returning a tuple of the locks). This are templated on the mutex which makes them a bit more flexible than using a concrete type: they can be used for any type of lockable mutex, not only util::Mutex. (Some of the code here uses them for getting locks around a std::mutex). Until C++17, using the RAII types is painfully verbose: ```C++ // pre-C++17 - needing to figure out the mutex type here is annoying: std::unique_lock<util::Mutex> lock(mutex); // pre-C++17 and even more verbose (but at least the type isn't needed): std::unique_lock<decltype(mutex)> lock(mutex); // our compromise: auto lock = util::unique_lock(mutex); // C++17: std::unique_lock lock(mutex); ``` All of these functions will also warn (under gcc or clang) if you discard the return value. You can also do fancy things like `auto l = util::unique_lock(mutex, std::adopt_lock)` (which lets a lock take over an already-locked mutex). - metrics code is gone, which also removes a big pile of code that was only used by metrics: - llarp::util::Scheduler - llarp::thread::TimerQueue - llarp::util::Stopwatch
2020-02-21 17:21:11 +00:00
util::Lock lock(self->m_state->m_SendQueueMutex);
self->m_state->m_SendQueue.emplace_back(transfer, p);
});
}
}
}
2020-02-18 16:00:45 +00:00
else
2018-08-22 15:52:10 +00:00
{
auto& sessions = m_state->m_RemoteSessions;
auto range = sessions.equal_range(remote);
auto itr = range.first;
while (itr != range.second)
{
if (itr->second->ReadyToSend())
{
itr->second->AsyncEncryptAndSendTo(data, t);
return true;
}
++itr;
}
2020-02-18 16:00:45 +00:00
// if we want to make an outbound session
if (WantsOutboundSession(remote))
{
// add pending traffic
auto& traffic = m_state->m_PendingTraffic;
traffic[remote].emplace_back(data, t);
return EnsurePathToService(
remote,
[self = this](Address addr, OutboundContext* ctx) {
if (ctx)
{
ctx->UpdateIntroSet();
for (auto& pending : self->m_state->m_PendingTraffic[addr])
{
ctx->AsyncEncryptAndSendTo(pending.Buffer(), pending.protocol);
}
}
self->m_state->m_PendingTraffic.erase(addr);
},
2020-02-24 19:40:45 +00:00
1500ms);
}
2018-08-22 15:52:10 +00:00
}
return false;
}
2018-08-22 15:52:10 +00:00
2019-03-08 17:00:13 +00:00
bool
Endpoint::HasConvoTag(const ConvoTag& t) const
{
return Sessions().find(t) != Sessions().end();
2019-03-08 17:00:13 +00:00
}
2018-08-09 19:02:17 +00:00
uint64_t
Endpoint::GetSeqNoForConvo(const ConvoTag& tag)
{
auto itr = Sessions().find(tag);
if (itr == Sessions().end())
2018-08-09 19:02:17 +00:00
return 0;
return ++(itr->second.seqno);
}
2019-03-08 14:36:24 +00:00
bool
Endpoint::ShouldBuildMore(llarp_time_t now) const
{
if (path::Builder::BuildCooldownHit(now))
2019-11-05 16:58:53 +00:00
return false;
size_t numBuilding = NumInStatus(path::ePathBuilding);
if (numBuilding > 0)
return false;
return ((now - lastBuild) > path::intro_path_spread)
|| NumInStatus(path::ePathEstablished) < path::min_intro_paths;
}
std::shared_ptr<Logic>
2018-08-09 19:02:17 +00:00
Endpoint::RouterLogic()
2018-07-19 04:58:39 +00:00
{
return Router()->logic();
2018-07-19 04:58:39 +00:00
}
std::shared_ptr<Logic>
2018-08-09 19:02:17 +00:00
Endpoint::EndpointLogic()
{
return m_state->m_IsolatedLogic ? m_state->m_IsolatedLogic : Router()->logic();
2018-08-09 19:02:17 +00:00
}
std::shared_ptr<llarp::thread::ThreadPool>
Endpoint::CryptoWorker()
2018-07-19 04:58:39 +00:00
{
return Router()->threadpool();
2018-07-19 04:58:39 +00:00
}
AbstractRouter*
Endpoint::Router()
{
return m_state->m_Router;
}
2020-05-21 14:18:23 +00:00
void
Endpoint::BlacklistSNode(const RouterID snode)
{
m_state->m_SnodeBlacklist.insert(snode);
}
const std::set<RouterID>&
Endpoint::SnodeBlacklist() const
{
return m_state->m_SnodeBlacklist;
}
const IntroSet&
Endpoint::introSet() const
{
return m_state->m_IntroSet;
}
IntroSet&
Endpoint::introSet()
{
return m_state->m_IntroSet;
}
const ConvoMap&
Endpoint::Sessions() const
{
return m_state->m_Sessions;
}
ConvoMap&
Endpoint::Sessions()
{
return m_state->m_Sessions;
}
2018-07-12 18:21:44 +00:00
} // namespace service
2018-07-16 03:32:13 +00:00
} // namespace llarp