mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
120 lines
3.2 KiB
C++
120 lines
3.2 KiB
C++
#include <service/endpoint_state.hpp>
|
|
|
|
#include <exit/session.hpp>
|
|
#include <hook/shell.hpp>
|
|
#include <service/outbound_context.hpp>
|
|
#include <util/str.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace service
|
|
{
|
|
bool
|
|
EndpointState::SetOption(const std::string& k, const std::string& v,
|
|
const std::string& name)
|
|
{
|
|
if(k == "keyfile")
|
|
{
|
|
m_Keyfile = v;
|
|
}
|
|
if(k == "tag")
|
|
{
|
|
m_Tag = v;
|
|
LogInfo("Setting tag to ", v);
|
|
}
|
|
if(k == "prefetch-tag")
|
|
{
|
|
m_PrefetchTags.insert(v);
|
|
}
|
|
if(k == "prefetch-addr")
|
|
{
|
|
Address addr;
|
|
if(addr.FromString(v))
|
|
m_PrefetchAddrs.insert(addr);
|
|
}
|
|
if(k == "min-latency")
|
|
{
|
|
auto val = atoi(v.c_str());
|
|
if(val > 0)
|
|
m_MinPathLatency = val;
|
|
}
|
|
if(k == "bundle-rc")
|
|
{
|
|
m_BundleRC = IsTrueValue(v.c_str());
|
|
}
|
|
if(k == "blacklist-snode")
|
|
{
|
|
RouterID snode;
|
|
if(!snode.FromString(v))
|
|
{
|
|
LogError(name, " invalid snode value: ", v);
|
|
return false;
|
|
}
|
|
const auto result = m_SnodeBlacklist.insert(snode);
|
|
if(!result.second)
|
|
{
|
|
LogError(name, " duplicate blacklist-snode: ", snode.ToString());
|
|
return false;
|
|
}
|
|
LogInfo(name, " adding ", snode.ToString(), " to blacklist");
|
|
}
|
|
if(k == "on-up")
|
|
{
|
|
m_OnUp = hooks::ExecShellBackend(v);
|
|
if(m_OnUp)
|
|
LogInfo(name, " added on up script: ", v);
|
|
else
|
|
LogError(name, " failed to add on up script");
|
|
}
|
|
if(k == "on-down")
|
|
{
|
|
m_OnDown = hooks::ExecShellBackend(v);
|
|
if(m_OnDown)
|
|
LogInfo(name, " added on down script: ", v);
|
|
else
|
|
LogError(name, " failed to add on down script");
|
|
}
|
|
if(k == "on-ready")
|
|
{
|
|
m_OnReady = hooks::ExecShellBackend(v);
|
|
if(m_OnReady)
|
|
LogInfo(name, " added on ready script: ", v);
|
|
else
|
|
LogError(name, " failed to add on ready script");
|
|
}
|
|
return true;
|
|
}
|
|
|
|
util::StatusObject
|
|
EndpointState::ExtractStatus(util::StatusObject& obj) const
|
|
{
|
|
obj.Put("lastPublished", m_LastPublish);
|
|
obj.Put("lastPublishAttempt", m_LastPublishAttempt);
|
|
obj.Put("introset", m_IntroSet.ExtractStatus());
|
|
|
|
if(!m_Tag.IsZero())
|
|
obj.Put("tag", m_Tag.ToString());
|
|
static auto getSecond = [](const auto& item) -> const auto&
|
|
{
|
|
return item.second;
|
|
};
|
|
obj.PutContainer("deadSessions", m_DeadSessions, getSecond);
|
|
obj.PutContainer("remoteSessions", m_RemoteSessions, getSecond);
|
|
obj.PutContainer("lookups", m_PendingLookups, getSecond);
|
|
obj.PutContainer("snodeSessions", m_SNodeSessions,
|
|
[](const auto& item) { return item.second.first; });
|
|
|
|
util::StatusObject sessionObj{};
|
|
|
|
for(const auto& item : m_Sessions)
|
|
{
|
|
std::string k = item.first.ToHex();
|
|
sessionObj.Put(k, item.second.ExtractStatus());
|
|
}
|
|
|
|
obj.Put("converstations", sessionObj);
|
|
return obj;
|
|
}
|
|
} // namespace service
|
|
} // namespace llarp
|