lokinet/llarp/router/router.cpp

1241 lines
32 KiB
C++
Raw Normal View History

#include <memory>
#include <router/router.hpp>
#include <config/config.hpp>
2019-08-12 11:27:46 +00:00
#include <constants/limits.hpp>
2019-09-01 12:10:49 +00:00
#include <constants/proto.hpp>
#include <crypto/crypto_libsodium.hpp>
2019-09-01 12:10:49 +00:00
#include <crypto/crypto.hpp>
#include <dht/context.hpp>
2019-01-19 18:16:40 +00:00
#include <dht/node.hpp>
2019-03-29 15:17:49 +00:00
#include <iwp/iwp.hpp>
2018-12-12 01:32:10 +00:00
#include <link/server.hpp>
#include <messages/link_message.hpp>
#include <net/net.hpp>
#include <rpc/rpc.hpp>
#include <util/buffer.hpp>
#include <util/encode.hpp>
2019-09-01 12:10:49 +00:00
#include <util/logging/file_logger.hpp>
#include <util/logging/json_logger.hpp>
#include <util/logging/logger_syslog.hpp>
#include <util/logging/logger.hpp>
2019-09-01 12:38:03 +00:00
#include <util/meta/memfn.hpp>
2019-09-01 12:58:27 +00:00
#include <util/metrics/metrics.hpp>
#include <util/str.hpp>
2019-09-05 17:39:09 +00:00
#include <ev/ev.hpp>
2017-11-28 14:05:31 +00:00
2018-05-20 17:45:47 +00:00
#include <fstream>
2018-10-09 01:38:25 +00:00
#include <cstdlib>
#include <iterator>
#include <unordered_map>
2019-09-01 12:10:49 +00:00
#include <utility>
#if defined(ANDROID) || defined(IOS)
2018-11-08 12:31:50 +00:00
#include <unistd.h>
#endif
constexpr uint64_t ROUTER_TICK_INTERVAL_MS = 1000;
2018-05-20 17:45:47 +00:00
namespace llarp
{
2019-07-09 13:47:24 +00:00
Router::Router(std::shared_ptr< llarp::thread::ThreadPool > _tp,
llarp_ev_loop_ptr __netloop, std::shared_ptr< Logic > l)
: ready(false)
2019-07-30 23:42:13 +00:00
, _netloop(std::move(__netloop))
, cryptoworker(std::move(_tp))
, _logic(std::move(l))
, paths(this)
, _exitContext(this)
2019-07-09 13:58:16 +00:00
, disk(std::make_shared< llarp::thread::ThreadPool >(1, 1000,
"diskworker"))
, _dht(llarp_dht_context_new(this))
, inbound_link_msg_parser(this)
2019-02-22 16:21:05 +00:00
, _hiddenServiceContext(this)
2018-06-20 17:45:44 +00:00
{
m_keyManager = std::make_shared< KeyManager >();
// set rational defaults
this->ip4addr.sin_family = AF_INET;
this->ip4addr.sin_port = htons(1090);
_stopping.store(false);
_running.store(false);
_lastTick = llarp::time_now_ms();
2018-06-20 17:45:44 +00:00
}
Router::~Router()
2018-09-19 13:27:15 +00:00
{
llarp_dht_context_free(_dht);
2018-09-19 13:27:15 +00:00
}
2019-02-11 17:14:43 +00:00
util::StatusObject
Router::ExtractStatus() const
2019-02-08 19:43:25 +00:00
{
if(_running)
{
return util::StatusObject{
2019-10-04 19:43:30 +00:00
{"running", true},
{"numNodesKnown", _nodedb->num_loaded()},
{"dht", _dht->impl->ExtractStatus()},
{"services", _hiddenServiceContext.ExtractStatus()},
{"exit", _exitContext.ExtractStatus()},
{"links", _linkManager.ExtractStatus()}};
}
else
{
2019-10-04 19:43:30 +00:00
return util::StatusObject{{"running", false}};
}
2019-02-08 19:43:25 +00:00
}
bool
Router::HandleRecvLinkMessageBuffer(ILinkSession *session,
const llarp_buffer_t &buf)
{
if(_stopping)
return true;
if(!session)
{
LogWarn("no link session");
return false;
}
return inbound_link_msg_parser.ProcessFrom(session, buf);
2018-10-07 15:29:36 +00:00
}
void
Router::PersistSessionUntil(const RouterID &remote, llarp_time_t until)
2018-10-07 15:29:36 +00:00
{
_linkManager.PersistSessionUntil(remote, until);
2018-06-01 14:08:54 +00:00
}
2018-12-13 00:03:19 +00:00
bool
Router::GetRandomGoodRouter(RouterID &router)
{
if(whitelistRouters)
{
return _rcLookupHandler.GetRandomWhitelistRouter(router);
}
2019-05-14 14:36:18 +00:00
auto pick_router = [&](auto &collection) -> bool {
const auto sz = collection.size();
auto itr = collection.begin();
if(sz == 0)
return false;
if(sz > 1)
std::advance(itr, randint() % sz);
router = itr->first;
return true;
2019-05-14 14:36:18 +00:00
};
absl::ReaderMutexLock l(&nodedb()->access);
2019-05-14 14:36:18 +00:00
return pick_router(nodedb()->entries);
2018-12-13 00:03:19 +00:00
}
2019-04-30 16:07:17 +00:00
void
Router::PumpLL()
{
static constexpr size_t PumpJobThreshhold = 50;
static constexpr llarp_time_t PumpInterval = 25;
const auto now = Now();
if(_stopping.load())
return;
if(_logic->numPendingJobs() >= PumpJobThreshhold
&& _lastPump + PumpInterval >= now)
{
return;
}
_lastPump = now;
paths.PumpDownstream();
2019-11-04 18:25:05 +00:00
paths.PumpUpstream();
_outboundMessageHandler.Tick();
_linkManager.PumpLinks();
2019-04-30 16:07:17 +00:00
}
bool
Router::SendToOrQueue(const RouterID &remote, const ILinkMessage *msg,
SendStatusHandler handler)
2018-06-14 15:10:31 +00:00
{
if(handler == nullptr)
{
using std::placeholders::_1;
handler = std::bind(&Router::MessageSent, this, remote, _1);
}
return _outboundMessageHandler.QueueMessage(remote, msg, handler);
}
void
2019-04-05 14:58:22 +00:00
Router::ForEachPeer(std::function< void(const ILinkSession *, bool) > visit,
bool randomize) const
{
_linkManager.ForEachPeer(visit, randomize);
}
void
Router::ForEachPeer(std::function< void(ILinkSession *) > visit)
{
_linkManager.ForEachPeer(visit);
}
void
Router::try_connect(fs::path rcfile)
2018-05-16 18:13:18 +00:00
{
RouterContact remote;
if(!remote.Read(rcfile.string().c_str()))
{
LogError("failure to decode or verify of remote RC");
return;
}
if(remote.Verify(Now()))
{
LogDebug("verified signature");
_outboundSessionMaker.CreateSessionTo(remote, nullptr);
}
else
LogError(rcfile, " contains invalid RC");
2018-05-16 18:13:18 +00:00
}
2018-05-20 17:45:47 +00:00
bool
Router::EnsureIdentity()
{
if(whitelistRouters)
{
#if defined(ANDROID) || defined(IOS)
LogError("running a service node on mobile device is not possible.");
return false;
#else
#if defined(_WIN32)
LogError("running a service node on windows is not possible.");
return false;
#endif
#endif
}
2018-04-05 14:23:14 +00:00
_identity = m_keyManager->identityKey;
_encryption = m_keyManager->encryptionKey;
if(_identity.IsZero())
return false;
if(_encryption.IsZero())
return false;
return true;
}
2018-01-29 14:27:24 +00:00
bool
Router::Configure(Config *conf, llarp_nodedb *nodedb)
{
if(nodedb == nullptr)
{
LogError(
"Attempting to Router::Configure but passed null nodedb pointer");
return false;
}
_nodedb = nodedb;
if(not m_keyManager->initialize(*conf, true))
2019-07-12 17:23:38 +00:00
return false;
if(!FromConfig(conf))
return false;
if(!InitOutboundLinks())
return false;
return EnsureIdentity();
}
2018-11-21 14:10:02 +00:00
2019-03-25 15:41:37 +00:00
/// called in disk worker thread
void
Router::HandleSaveRC() const
2019-03-25 15:41:37 +00:00
{
std::string fname = our_rc_file.string();
_rc.Write(fname.c_str());
2019-03-25 15:41:37 +00:00
}
bool
Router::SaveRC()
2018-08-18 14:01:21 +00:00
{
LogDebug("verify RC signature");
if(!_rc.Verify(Now()))
{
2019-05-24 02:01:36 +00:00
Dump< MAX_RC_SIZE >(rc());
LogError("RC is invalid, not saving");
return false;
}
diskworker()->addJob(std::bind(&Router::HandleSaveRC, this));
2019-03-25 15:41:37 +00:00
return true;
2018-06-20 12:34:48 +00:00
}
2018-05-29 13:40:26 +00:00
bool
Router::IsServiceNode() const
{
return m_isServiceNode;
}
2018-05-30 20:56:47 +00:00
void
Router::Close()
{
LogInfo("closing router");
2019-06-02 21:17:05 +00:00
llarp_ev_loop_stop(_netloop);
2019-07-09 13:47:24 +00:00
disk->stop();
disk->shutdown();
}
void
Router::handle_router_ticker()
2018-11-28 15:18:18 +00:00
{
ticker_job_id = 0;
LogicCall(logic(), std::bind(&Router::Tick, this));
ScheduleTicker(ROUTER_TICK_INTERVAL_MS);
2018-11-28 15:18:18 +00:00
}
bool
Router::ParseRoutingMessageBuffer(const llarp_buffer_t &buf,
routing::IMessageHandler *h,
const PathID_t &rxid)
{
return inbound_routing_msg_parser.ParseMessageBuffer(buf, h, rxid, this);
}
bool
Router::ConnectionToRouterAllowed(const RouterID &router) const
{
return _rcLookupHandler.RemoteIsAllowed(router);
2019-05-09 15:36:39 +00:00
}
size_t
Router::NumberOfConnectedRouters() const
{
return _linkManager.NumberOfConnectedRouters();
2019-05-09 15:36:39 +00:00
}
size_t
Router::NumberOfConnectedClients() const
{
return _linkManager.NumberOfConnectedClients();
}
2018-06-19 17:11:24 +00:00
bool
Router::UpdateOurRC(bool rotateKeys)
{
SecretKey nextOnionKey;
RouterContact nextRC = _rc;
if(rotateKeys)
{
CryptoManager::instance()->encryption_keygen(nextOnionKey);
2019-01-29 13:20:27 +00:00
std::string f = encryption_keyfile.string();
2019-03-25 15:41:37 +00:00
// TODO: use disk worker
2019-01-29 13:20:27 +00:00
if(nextOnionKey.SaveToFile(f.c_str()))
{
nextRC.enckey = seckey_topublic(nextOnionKey);
_encryption = nextOnionKey;
2019-01-29 13:20:27 +00:00
}
}
if(!nextRC.Sign(identity()))
return false;
if(!nextRC.Verify(time_now_ms(), false))
return false;
_rc = std::move(nextRC);
// propagate RC by renegotiating sessions
ForEachPeer([](ILinkSession *s) {
if(s->RenegotiateSession())
LogInfo("renegotiated session");
else
LogWarn("failed to renegotiate session");
});
2019-02-04 23:18:43 +00:00
return SaveRC();
2019-03-25 15:41:37 +00:00
}
2019-07-12 17:21:29 +00:00
bool
2019-07-12 17:23:38 +00:00
Router::FromConfig(Config *conf)
{
2019-07-07 11:29:44 +00:00
// Set netid before anything else
2019-09-09 11:36:21 +00:00
if(!conf->router.netId().empty()
&& strcmp(conf->router.netId().c_str(), llarp::DEFAULT_NETID))
{
2019-07-01 23:55:02 +00:00
const auto &netid = conf->router.netId();
2019-07-07 11:29:44 +00:00
llarp::LogWarn("!!!! you have manually set netid to be '", netid,
"' which does not equal '", llarp::DEFAULT_NETID,
2019-07-07 11:29:44 +00:00
"' you will run as a different network, good luck "
"and don't forget: something something MUH traffic "
"shape correlation !!!!");
NetID::DefaultValue() =
NetID(reinterpret_cast< const byte_t * >(netid.c_str()));
// reset netid in our rc
_rc.netID = llarp::NetID();
}
2019-08-07 16:33:29 +00:00
const auto linktypename = conf->router.defaultLinkProto();
_defaultLinkType = LinkFactory::TypeFromName(linktypename);
if(_defaultLinkType == LinkFactory::LinkType::eLinkUnknown)
{
LogError("failed to set link type to '", linktypename,
"' as that is invalid");
return false;
}
2019-07-07 11:29:44 +00:00
// IWP config
2019-08-07 16:33:29 +00:00
m_OutboundPort = std::get< LinksConfig::Port >(conf->links.outboundLink());
2019-07-12 17:21:29 +00:00
// Router config
_rc.SetNick(conf->router.nickname());
_outboundSessionMaker.maxConnectedRouters =
conf->router.maxConnectedRouters();
_outboundSessionMaker.minConnectedRouters =
conf->router.minConnectedRouters();
encryption_keyfile = conf->router.encryptionKeyfile();
our_rc_file = conf->router.ourRcFile();
transport_keyfile = conf->router.transportKeyfile();
addrInfo = conf->router.addrInfo();
publicOverride = conf->router.publicOverride();
ip4addr = conf->router.ip4addr();
2019-07-12 17:21:29 +00:00
2019-08-26 23:29:17 +00:00
if(!conf->router.blockBogons().value_or(true))
{
RouterContact::BlockBogons = false;
}
2019-07-12 17:21:29 +00:00
// Lokid Config
usingSNSeed = conf->lokid.usingSNSeed;
ident_keyfile = conf->lokid.ident_keyfile;
whitelistRouters = conf->lokid.whitelistRouters;
lokidRPCAddr = conf->lokid.lokidRPCAddr;
lokidRPCUser = conf->lokid.lokidRPCUser;
lokidRPCPassword = conf->lokid.lokidRPCPassword;
// TODO: add config flag for "is service node"
2019-08-07 16:33:29 +00:00
if(conf->links.inboundLinks().size())
{
m_isServiceNode = true;
}
std::set< RouterID > strictConnectPubkeys;
2019-07-02 00:20:58 +00:00
if(!conf->network.strictConnect().empty())
{
2019-07-02 00:20:58 +00:00
const auto &val = conf->network.strictConnect();
if(IsServiceNode())
{
llarp::LogError("cannot use strict-connect option as service node");
2019-07-12 17:21:29 +00:00
return false;
}
llarp::RouterID snode;
llarp::PubKey pk;
if(pk.FromString(val))
{
if(strictConnectPubkeys.emplace(pk).second)
llarp::LogInfo("added ", pk, " to strict connect list");
else
llarp::LogWarn("duplicate key for strict connect: ", pk);
}
else if(snode.FromString(val))
{
if(strictConnectPubkeys.insert(snode).second)
{
llarp::LogInfo("added ", snode, " to strict connect list");
netConfig.emplace("strict-connect", val);
}
else
llarp::LogWarn("duplicate key for strict connect: ", snode);
}
else
llarp::LogError("invalid key for strict-connect: ", val);
}
std::vector< std::string > configRouters = conf->connect.routers;
configRouters.insert(configRouters.end(), conf->bootstrap.routers.begin(),
conf->bootstrap.routers.end());
BootstrapList b_list;
for(const auto &router : configRouters)
{
bool isListFile = false;
{
std::ifstream inf(router, std::ios::binary);
if(inf.is_open())
{
const char ch = inf.get();
isListFile = ch == 'l';
}
}
if(isListFile)
{
if(not BDecodeReadFile(router.c_str(), b_list))
{
LogWarn("failed to read bootstrap list file '", router, "'");
return false;
}
}
else
{
RouterContact rc;
if(not rc.Read(router.c_str()))
{
llarp::LogWarn("failed to decode bootstrap RC, file='", router,
"' rc=", rc);
return false;
}
b_list.insert(rc);
}
}
for(auto &rc : b_list)
{
if(not rc.Verify(Now()))
{
LogWarn("ignoring invalid RC: ", RouterID(rc.pubkey));
continue;
}
bootstrapRCList.emplace(std::move(rc));
}
LogInfo("Loaded ", bootstrapRCList.size(), " bootstrap routers");
// Init components after relevant config settings loaded
_outboundMessageHandler.Init(&_linkManager, _logic);
_outboundSessionMaker.Init(&_linkManager, &_rcLookupHandler,
&_routerProfiling, _logic, _nodedb,
threadpool());
_linkManager.Init(&_outboundSessionMaker);
_rcLookupHandler.Init(_dht, _nodedb, threadpool(), &_linkManager,
&_hiddenServiceContext, strictConnectPubkeys,
bootstrapRCList, whitelistRouters, m_isServiceNode);
if(!usingSNSeed)
{
ident_keyfile = conf->router.identKeyfile();
}
// create inbound links, if we are a service node
2019-08-07 16:33:29 +00:00
for(const auto &serverConfig : conf->links.inboundLinks())
{
2019-08-07 16:33:29 +00:00
// get default factory
auto inboundLinkFactory = LinkFactory::Obtain(_defaultLinkType, true);
// for each option if provided ...
for(const auto &opt : std::get< LinksConfig::Options >(serverConfig))
{
// try interpreting it as a link type
const auto linktype = LinkFactory::TypeFromName(opt);
if(linktype != LinkFactory::LinkType::eLinkUnknown)
{
// override link factory if it's a valid link type
auto factory = LinkFactory::Obtain(linktype, true);
if(factory)
{
inboundLinkFactory = std::move(factory);
break;
}
}
}
auto server = inboundLinkFactory(
m_keyManager, util::memFn(&AbstractRouter::rc, this),
util::memFn(&AbstractRouter::HandleRecvLinkMessageBuffer, this),
2019-08-07 16:33:29 +00:00
util::memFn(&AbstractRouter::Sign, this),
util::memFn(&IOutboundSessionMaker::OnSessionEstablished,
&_outboundSessionMaker),
util::memFn(&AbstractRouter::CheckRenegotiateValid, this),
util::memFn(&IOutboundSessionMaker::OnConnectTimeout,
&_outboundSessionMaker),
util::memFn(&AbstractRouter::SessionClosed, this),
util::memFn(&AbstractRouter::PumpLL, this));
2019-08-07 16:33:29 +00:00
const auto &key = std::get< LinksConfig::Interface >(serverConfig);
int af = std::get< LinksConfig::AddressFamily >(serverConfig);
uint16_t port = std::get< LinksConfig::Port >(serverConfig);
if(!server->Configure(netloop(), key, af, port))
{
LogError("failed to bind inbound link on ", key, " port ", port);
return false;
}
_linkManager.AddLink(std::move(server), true);
}
// set network config
netConfig = conf->network.netConfig();
// Network config
if(conf->network.enableProfiling().has_value())
2018-12-19 17:48:29 +00:00
{
if(not conf->network.enableProfiling().value())
{
routerProfiling().Disable();
LogWarn("router profiling explicitly disabled");
}
2018-12-19 17:48:29 +00:00
}
if(!conf->network.routerProfilesFile().empty())
{
routerProfilesFile = conf->network.routerProfilesFile();
routerProfiling().Load(routerProfilesFile.c_str());
llarp::LogInfo("setting profiles to ", routerProfilesFile);
}
// API config
enableRPCServer = conf->api.enableRPCServer();
rpcBindAddr = conf->api.rpcBindAddr();
2019-06-04 13:19:45 +00:00
// Services config
for(const auto &service : conf->services.services)
2019-06-04 13:19:45 +00:00
{
if(LoadHiddenServiceConfig(service.second))
{
llarp::LogInfo("loaded hidden service config for ", service.first);
}
else
2019-06-04 13:19:45 +00:00
{
llarp::LogWarn("failed to load hidden service config for ",
service.first);
2019-06-04 13:19:45 +00:00
}
}
// Logging config
auto logfile = conf->logging.m_LogFile;
if(conf->logging.m_LogJSON)
2019-06-04 13:29:16 +00:00
{
LogContext::Instance().logStream = std::make_unique< JSONLogStream >(
diskworker(), logfile, 100, logfile != stdout);
2019-06-04 13:29:16 +00:00
}
else if(logfile != stdout)
2019-06-04 13:29:16 +00:00
{
LogContext::Instance().logStream =
std::make_unique< FileLogStream >(diskworker(), logfile, 100, true);
2019-06-04 13:29:16 +00:00
}
2018-12-19 17:48:29 +00:00
netConfig.insert(conf->dns.netConfig.begin(), conf->dns.netConfig.end());
return true;
}
bool
Router::CheckRenegotiateValid(RouterContact newrc, RouterContact oldrc)
2019-03-31 15:09:59 +00:00
{
return _rcLookupHandler.CheckRenegotiateValid(newrc, oldrc);
2019-03-31 15:09:59 +00:00
}
bool
2019-05-09 15:36:39 +00:00
Router::IsBootstrapNode(const RouterID r) const
{
2019-05-09 15:36:39 +00:00
return std::count_if(
bootstrapRCList.begin(), bootstrapRCList.end(),
[r](const RouterContact &rc) -> bool { return rc.pubkey == r; })
> 0;
}
2019-04-05 14:58:22 +00:00
2019-07-15 16:56:09 +00:00
bool
Router::ShouldReportStats(llarp_time_t now) const
{
2019-07-17 12:25:51 +00:00
static constexpr llarp_time_t ReportStatsInterval = 60 * 60 * 1000;
2019-07-15 16:56:09 +00:00
return now - m_LastStatsReport > ReportStatsInterval;
}
void
Router::ReportStats()
{
const auto now = Now();
LogInfo(nodedb()->num_loaded(), " RCs loaded");
LogInfo(bootstrapRCList.size(), " bootstrap peers");
LogInfo(NumberOfConnectedRouters(), " router connections");
if(IsServiceNode())
{
LogInfo(NumberOfConnectedClients(), " client connections");
LogInfo(_rc.Age(now), " ms since we last updated our RC");
LogInfo(_rc.TimeUntilExpires(now), " ms until our RC expires");
}
LogInfo(now, " system time");
LogInfo(m_LastStatsReport, " last reported stats");
m_LastStatsReport = now;
}
void
Router::Tick()
2018-06-19 17:11:24 +00:00
{
2019-04-23 18:29:42 +00:00
if(_stopping)
return;
// LogDebug("tick router");
2019-11-05 16:58:53 +00:00
const auto now = Now();
2018-12-19 17:48:29 +00:00
#if defined(WITH_SYSTEMD)
::sd_notify(0, "WATCHDOG=1");
#endif
2019-03-04 17:03:18 +00:00
routerProfiling().Tick();
2019-07-15 16:56:09 +00:00
if(ShouldReportStats(now))
{
ReportStats();
}
_rcLookupHandler.PeriodicUpdate(now);
2019-07-15 16:56:09 +00:00
const bool isSvcNode = IsServiceNode();
2019-08-05 13:39:27 +00:00
if(_rc.ExpiresSoon(now, randint() % 10000)
|| (now - _rc.last_updated) > rcRegenInterval)
2018-12-19 17:48:29 +00:00
{
2019-08-05 13:39:27 +00:00
LogInfo("regenerating RC");
if(!UpdateOurRC(false))
LogError("Failed to update our RC");
}
2019-03-31 15:09:59 +00:00
2019-08-05 13:39:27 +00:00
if(isSvcNode)
{
// remove RCs for nodes that are no longer allowed by network policy
2019-05-09 15:36:39 +00:00
nodedb()->RemoveIf([&](const RouterContact &rc) -> bool {
2019-05-11 20:53:08 +00:00
if(IsBootstrapNode(rc.pubkey))
return false;
return !_rcLookupHandler.RemoteIsAllowed(rc.pubkey);
2019-05-09 15:36:39 +00:00
});
2018-12-19 17:48:29 +00:00
}
2019-02-25 12:46:40 +00:00
_linkManager.CheckPersistingSessions(now);
2019-12-03 17:03:19 +00:00
size_t connected = NumberOfConnectedRouters();
if(not isSvcNode)
{
connected += _linkManager.NumberOfPendingConnections();
}
_rcLookupHandler.ExploreNetwork();
size_t connectToNum = _outboundSessionMaker.minConnectedRouters;
const auto strictConnect = _rcLookupHandler.NumberOfStrictConnectRouters();
if(strictConnect > 0 && connectToNum > strictConnect)
{
connectToNum = strictConnect;
}
if(connected < connectToNum)
{
size_t dlt = connectToNum - connected;
LogInfo("connecting to ", dlt, " random routers to keep alive");
_outboundSessionMaker.ConnectToRandomRouters(dlt);
2018-08-14 21:17:18 +00:00
}
2019-11-05 16:58:53 +00:00
_hiddenServiceContext.Tick(now);
_exitContext.Tick(now);
2019-11-05 16:58:53 +00:00
if(rpcCaller)
rpcCaller->Tick(now);
2019-03-25 15:41:37 +00:00
// save profiles async
if(routerProfiling().ShouldSave(now))
{
diskworker()->addJob(
[&]() { routerProfiling().Save(routerProfilesFile.c_str()); });
2019-03-25 15:41:37 +00:00
}
2019-09-10 14:16:32 +00:00
// get connected peers
std::set< dht::Key_t > peersWeHave;
_linkManager.ForEachPeer([&peersWeHave](ILinkSession *s) {
if(!s->IsEstablished())
return;
peersWeHave.emplace(s->GetPubKey());
});
// remove any nodes we don't have connections to
_dht->impl->Nodes()->RemoveIf([&peersWeHave](const dht::Key_t &k) -> bool {
return peersWeHave.count(k) == 0;
});
2019-11-05 16:58:53 +00:00
// expire paths
paths.ExpirePaths(now);
// update tick timestamp
_lastTick = llarp::time_now_ms();
}
2018-08-22 16:19:51 +00:00
bool
Router::Sign(Signature &sig, const llarp_buffer_t &buf) const
{
2019-06-13 21:58:17 +00:00
metrics::TimerGuard t("Router", "Sign");
return CryptoManager::instance()->sign(sig, identity(), buf);
}
void
Router::ScheduleTicker(uint64_t ms)
2018-06-06 12:46:26 +00:00
{
ticker_job_id =
_logic->call_later(ms, std::bind(&Router::handle_router_ticker, this));
2018-06-06 12:46:26 +00:00
}
void
Router::SessionClosed(RouterID remote)
2018-06-20 12:34:48 +00:00
{
2019-02-25 12:46:40 +00:00
dht::Key_t k(remote);
dht()->impl->Nodes()->DelNode(k);
LogInfo("Session to ", remote, " fully closed");
2018-06-20 12:34:48 +00:00
}
bool
Router::GetRandomConnectedRouter(RouterContact &result) const
2018-06-06 12:46:26 +00:00
{
return _linkManager.GetRandomConnectedRouter(result);
2018-06-06 12:46:26 +00:00
}
2018-05-30 20:56:47 +00:00
void
Router::HandleDHTLookupForExplore(ABSL_ATTRIBUTE_UNUSED RouterID remote,
const std::vector< RouterContact > &results)
2018-07-03 13:33:37 +00:00
{
for(const auto &rc : results)
{
_rcLookupHandler.CheckRC(rc);
}
2018-06-01 14:08:54 +00:00
}
2018-06-13 12:58:51 +00:00
// TODO: refactor callers and remove this function
void
Router::LookupRouter(RouterID remote, RouterLookupHandler resultHandler)
2018-10-09 12:06:30 +00:00
{
_rcLookupHandler.GetRC(
remote,
[=](const RouterID &id, const RouterContact *const rc,
const RCRequestResult result) {
(void)id;
if(resultHandler)
{
std::vector< RouterContact > routers;
if(result == RCRequestResult::Success && rc != nullptr)
{
routers.push_back(*rc);
}
resultHandler(routers);
}
});
}
void
Router::SetRouterWhitelist(const std::vector< RouterID > &routers)
{
_rcLookupHandler.SetRouterWhitelist(routers);
}
/// this function ensure there are sane defualts in a net config
static void
EnsureNetConfigDefaultsSane(
std::unordered_multimap< std::string, std::string > &netConfig)
{
static const std::unordered_map< std::string,
std::function< std::string(void) > >
netConfigDefaults = {
{"ifname", llarp::FindFreeTun},
{"ifaddr", llarp::FindFreeRange},
{"local-dns", []() -> std::string { return "127.0.0.1:53"; }}};
// populate with fallback defaults if values not present
auto itr = netConfigDefaults.begin();
while(itr != netConfigDefaults.end())
{
auto found = netConfig.find(itr->first);
if(found == netConfig.end() || found->second.empty())
{
2019-09-09 12:10:26 +00:00
auto val = itr->second();
if(!val.empty())
netConfig.emplace(itr->first, std::move(val));
}
++itr;
}
}
bool
Router::StartJsonRpc()
{
if(_running || _stopping)
return false;
if(enableRPCServer)
2018-10-09 12:06:30 +00:00
{
if(rpcBindAddr.empty())
{
rpcBindAddr = DefaultRPCBindAddr;
}
rpcServer = std::make_unique< rpc::Server >(this);
while(!rpcServer->Start(rpcBindAddr))
{
LogError("failed to bind jsonrpc to ", rpcBindAddr);
2018-11-08 12:31:50 +00:00
#if defined(ANDROID) || defined(RPI)
sleep(1);
2018-11-08 12:31:50 +00:00
#else
std::this_thread::sleep_for(std::chrono::seconds(1));
2018-11-08 12:31:50 +00:00
#endif
}
LogInfo("Bound RPC server to ", rpcBindAddr);
2018-10-09 12:06:30 +00:00
}
return true;
}
bool
Router::Run()
{
if(_running || _stopping)
return false;
if(whitelistRouters)
{
rpcCaller = std::make_unique< rpc::Caller >(this);
rpcCaller->SetAuth(lokidRPCUser, lokidRPCPassword);
while(!rpcCaller->Start(lokidRPCAddr))
{
LogError("failed to start jsonrpc caller to ", lokidRPCAddr);
#if defined(ANDROID) || defined(RPI)
sleep(1);
#else
std::this_thread::sleep_for(std::chrono::seconds(1));
#endif
}
LogInfo("RPC Caller to ", lokidRPCAddr, " started");
}
2018-10-09 12:06:30 +00:00
2019-07-15 17:19:31 +00:00
if(!cryptoworker->start())
{
LogError("crypto worker failed to start");
return false;
}
if(!disk->start())
{
LogError("disk worker failed to start");
return false;
}
2018-11-19 16:40:49 +00:00
routerProfiling().Load(routerProfilesFile.c_str());
Addr publicAddr(this->addrInfo);
if(this->publicOverride)
{
LogDebug("public address:port ", publicAddr);
}
2019-04-08 18:21:01 +00:00
// set public signing key
_rc.pubkey = seckey_topublic(identity());
AddressInfo ai;
_linkManager.ForEachInboundLink([&](LinkLayer_ptr link) {
2019-01-05 13:45:05 +00:00
if(link->GetOurAddressInfo(ai))
{
2019-01-05 13:45:05 +00:00
// override ip and port
if(this->publicOverride)
{
2019-01-05 13:45:05 +00:00
ai.ip = *publicAddr.addr6();
ai.port = publicAddr.port();
}
2019-08-26 23:29:17 +00:00
if(RouterContact::BlockBogons && IsBogon(ai.ip))
return;
2019-01-05 13:45:05 +00:00
_rc.addrs.push_back(ai);
2019-04-08 18:21:01 +00:00
if(ExitEnabled())
{
const llarp::Addr addr(ai);
2019-04-08 18:25:06 +00:00
const nuint32_t a{addr.addr4()->s_addr};
2019-04-08 18:21:01 +00:00
_rc.exits.emplace_back(_rc.pubkey, a);
LogInfo("Exit relay started, advertised as exiting at: ", a);
2019-04-08 18:21:01 +00:00
}
}
});
// set public encryption key
_rc.enckey = seckey_topublic(encryption());
2019-04-08 18:21:01 +00:00
LogInfo("Signing rc...");
if(!_rc.Sign(identity()))
{
LogError("failed to sign rc");
return false;
}
if(!SaveRC())
{
LogError("failed to save RC");
return false;
}
2019-08-23 11:32:52 +00:00
_outboundSessionMaker.SetOurRouter(pubkey());
2019-09-05 17:39:09 +00:00
if(!_linkManager.StartLinks(_logic, cryptoworker))
2018-06-10 14:05:48 +00:00
{
LogWarn("One or more links failed to start.");
return false;
2018-06-10 14:05:48 +00:00
}
EnsureNetConfigDefaultsSane(netConfig);
2019-08-12 11:20:57 +00:00
const auto limits =
IsServiceNode() ? llarp::limits::snode : llarp::limits::client;
_outboundSessionMaker.minConnectedRouters = std::max(
_outboundSessionMaker.minConnectedRouters, limits.DefaultMinRouters);
_outboundSessionMaker.maxConnectedRouters = std::max(
_outboundSessionMaker.maxConnectedRouters, limits.DefaultMaxRouters);
2019-08-12 11:20:57 +00:00
if(IsServiceNode())
{
// initialize as service node
if(!InitServiceNode())
{
LogError("Failed to initialize service node");
return false;
}
RouterID us = pubkey();
LogInfo("initalized service node: ", us);
2019-08-12 11:20:57 +00:00
// relays do not use profiling
routerProfiling().Disable();
}
else
{
// we are a client
// regenerate keys and resign rc before everything else
CryptoManager::instance()->identity_keygen(_identity);
CryptoManager::instance()->encryption_keygen(_encryption);
_rc.pubkey = seckey_topublic(identity());
_rc.enckey = seckey_topublic(encryption());
if(!_rc.Sign(identity()))
{
LogError("failed to regenerate keys and sign RC");
return false;
}
if(!CreateDefaultHiddenService())
{
LogError("failed to set up default network endpoint");
return false;
}
}
LogInfo("starting hidden service context...");
2019-02-22 16:21:05 +00:00
if(!hiddenServiceContext().StartAll())
{
LogError("Failed to start hidden service context");
return false;
}
{
ssize_t loaded = _nodedb->LoadAll();
llarp::LogInfo("loaded ", loaded, " RCs");
if(loaded < 0)
{
// shouldn't be possible
return false;
}
}
llarp_dht_context_start(dht(), pubkey());
for(const auto &rc : bootstrapRCList)
{
if(this->nodedb()->Insert(rc))
{
LogInfo("added bootstrap node ", RouterID(rc.pubkey));
}
else
{
LogError("Failed to add bootstrap node ", RouterID(rc.pubkey));
}
_dht->impl->Nodes()->PutNode(rc);
}
LogInfo("have ", _nodedb->num_loaded(), " routers");
_netloop->add_ticker(std::bind(&Router::PumpLL, this));
ScheduleTicker(ROUTER_TICK_INTERVAL_MS);
_running.store(true);
_startedAt = Now();
#if defined(WITH_SYSTEMD)
::sd_notify(0, "READY=1");
#endif
2019-12-09 13:08:30 +00:00
LogContext::Instance().DropToRuntimeLevel();
return _running;
}
bool
Router::IsRunning() const
{
return _running;
}
llarp_time_t
Router::Uptime() const
{
const llarp_time_t _now = Now();
if(_startedAt && _now > _startedAt)
return _now - _startedAt;
return 0;
}
void
Router::AfterStopLinks()
{
Close();
}
void
Router::AfterStopIssued()
{
StopLinks();
nodedb()->AsyncFlushToDisk();
_logic->call_later(200, std::bind(&Router::AfterStopLinks, this));
}
void
Router::StopLinks()
{
_linkManager.Stop();
2018-06-10 14:05:48 +00:00
}
void
Router::Stop()
2018-11-26 13:29:45 +00:00
{
if(!_running)
return;
if(_stopping)
return;
_stopping.store(true);
2019-12-09 13:08:30 +00:00
LogContext::Instance().RevertRuntimeLevel();
LogInfo("stopping router");
2019-02-22 16:21:05 +00:00
hiddenServiceContext().StopAll();
_exitContext.Stop();
if(rpcServer)
rpcServer->Stop();
paths.PumpUpstream();
_linkManager.PumpLinks();
_logic->call_later(200, std::bind(&Router::AfterStopIssued, this));
2018-11-26 13:29:45 +00:00
}
2018-08-02 23:30:34 +00:00
bool
Router::HasSessionTo(const RouterID &remote) const
{
return _linkManager.HasSessionTo(remote);
}
void
2019-12-03 17:03:19 +00:00
Router::ConnectToRandomRouters(int _want)
{
2019-12-03 17:03:19 +00:00
const size_t want = _want;
auto connected = NumberOfConnectedRouters();
if(not IsServiceNode())
{
connected += _linkManager.NumberOfPendingConnections();
}
if(connected >= want)
return;
_outboundSessionMaker.ConnectToRandomRouters(want);
}
bool
Router::InitServiceNode()
2018-11-26 13:29:45 +00:00
{
LogInfo("accepting transit traffic");
paths.AllowTransit();
llarp_dht_allow_transit(dht());
return _exitContext.AddExitEndpoint("default-connectivity", netConfig);
2018-11-26 13:29:45 +00:00
}
2018-06-14 17:35:12 +00:00
bool
Router::ValidateConfig(ABSL_ATTRIBUTE_UNUSED Config *conf) const
{
return true;
}
bool
Router::Reconfigure(Config *)
2018-06-18 22:05:02 +00:00
{
// TODO: implement me
return true;
2018-05-20 17:45:47 +00:00
}
2018-01-08 13:49:05 +00:00
bool
Router::TryConnectAsync(RouterContact rc, uint16_t tries)
{
(void)tries;
if(rc.pubkey == pubkey())
{
return false;
}
if(!_rcLookupHandler.RemoteIsAllowed(rc.pubkey))
{
return false;
}
_outboundSessionMaker.CreateSessionTo(rc, nullptr);
return true;
}
bool
2019-01-05 13:45:05 +00:00
Router::InitOutboundLinks()
2018-06-18 22:03:50 +00:00
{
2019-08-07 16:33:29 +00:00
const auto linkTypeName = LinkFactory::NameFromType(_defaultLinkType);
LogInfo("initialize outbound link: ", linkTypeName);
auto factory = LinkFactory::Obtain(_defaultLinkType, false);
if(factory == nullptr)
{
LogError("cannot initialize outbound link of type '", linkTypeName,
"' as it has no implementation");
return false;
}
auto link =
factory(m_keyManager, util::memFn(&AbstractRouter::rc, this),
2019-08-07 16:33:29 +00:00
util::memFn(&AbstractRouter::HandleRecvLinkMessageBuffer, this),
util::memFn(&AbstractRouter::Sign, this),
util::memFn(&IOutboundSessionMaker::OnSessionEstablished,
&_outboundSessionMaker),
util::memFn(&AbstractRouter::CheckRenegotiateValid, this),
util::memFn(&IOutboundSessionMaker::OnConnectTimeout,
&_outboundSessionMaker),
2019-11-04 18:53:53 +00:00
util::memFn(&AbstractRouter::SessionClosed, this),
util::memFn(&AbstractRouter::PumpLL, this));
2019-08-07 16:33:29 +00:00
if(!link)
return false;
2019-08-07 16:33:29 +00:00
const auto afs = {AF_INET, AF_INET6};
2019-08-07 16:33:29 +00:00
for(const auto af : afs)
{
2019-08-07 16:33:29 +00:00
if(!link->Configure(netloop(), "*", af, m_OutboundPort))
2019-01-05 13:48:00 +00:00
continue;
2019-08-07 16:33:29 +00:00
_linkManager.AddLink(std::move(link), false);
return true;
}
2019-08-07 16:33:29 +00:00
return false;
}
2018-06-21 13:33:42 +00:00
bool
Router::CreateDefaultHiddenService()
{
// add endpoint
2019-02-22 16:21:05 +00:00
return hiddenServiceContext().AddDefaultEndpoint(netConfig);
}
2018-08-02 23:30:34 +00:00
bool
Router::LoadHiddenServiceConfig(string_view fname)
2018-07-09 17:32:11 +00:00
{
LogDebug("opening hidden service config ", fname);
service::Config conf;
if(!conf.Load(fname))
2018-07-09 17:32:11 +00:00
return false;
for(const auto &config : conf.services)
{
service::Config::section_t filteredConfig;
mergeHiddenServiceConfig(config.second, filteredConfig.second);
filteredConfig.first = config.first;
2019-02-22 16:21:05 +00:00
if(!hiddenServiceContext().AddEndpoint(filteredConfig))
return false;
}
return true;
2018-07-09 17:32:11 +00:00
}
void
Router::MessageSent(const RouterID &remote, SendStatus status)
{
if(status == SendStatus::Success)
{
LogDebug("Message successfully sent to ", remote);
}
else
{
2019-09-09 11:36:21 +00:00
LogDebug("Message failed sending to ", remote);
}
}
2018-02-01 13:21:00 +00:00
} // namespace llarp