lokinet/llarp/router/router.cpp

1190 lines
31 KiB
C++
Raw Normal View History

#include <memory>
#include <router/router.hpp>
#include <config/config.hpp>
2019-09-01 12:10:49 +00:00
#include <constants/proto.hpp>
#include <constants/files.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 <stdexcept>
#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>
#include <util/str.hpp>
2019-09-05 17:39:09 +00:00
#include <ev/ev.hpp>
2017-11-28 14:05:31 +00:00
#include "tooling/router_event.hpp"
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
2020-02-25 22:32:57 +00:00
#if defined(WITH_SYSTEMD)
#include <systemd/sd-daemon.h>
#endif
static constexpr std::chrono::milliseconds ROUTER_TICK_INTERVAL = 1s;
2018-05-20 17:45:47 +00:00
namespace llarp
{
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)
, 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)
#ifdef LOKINET_HIVE
, _randomStartDelay(std::chrono::milliseconds((llarp::randint() % 1250) + 2000))
#else
2020-02-05 16:16:46 +00:00
, _randomStartDelay(std::chrono::seconds((llarp::randint() % 30) + 10))
#endif
2018-06-20 17:45:44 +00:00
{
m_keyManager = std::make_shared<KeyManager>();
_stopping.store(false);
_running.store(false);
_lastTick = llarp::time_now_ms();
2020-01-18 20:47:36 +00:00
m_NextExploreAt = Clock_t::now();
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{{"running", true},
{"numNodesKnown", _nodedb->num_loaded()},
{"dht", _dht->impl->ExtractStatus()},
{"services", _hiddenServiceContext.ExtractStatus()},
{"exit", _exitContext.ExtractStatus()},
{"links", _linkManager.ExtractStatus()},
{"outboundMessages", _outboundMessageHandler.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
}
2020-01-30 17:23:16 +00:00
void
Router::GossipRCIfNeeded(const RouterContact rc)
{
/// if we are not a service node forget about gossip
if (not IsServiceNode())
2020-01-30 17:23:16 +00:00
return;
/// wait for random uptime
if (std::chrono::milliseconds{Uptime()} < _randomStartDelay)
2020-01-30 17:23:16 +00:00
return;
_rcGossiper.GossipRC(rc);
}
2018-12-13 00:03:19 +00:00
bool
Router::GetRandomGoodRouter(RouterID& router)
2018-12-13 00:03:19 +00:00
{
if (whitelistRouters)
{
return _rcLookupHandler.GetRandomWhitelistRouter(router);
}
auto pick_router = [&](auto& collection) -> bool {
2019-05-14 14:36:18 +00:00
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
};
std::shared_lock 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 auto PumpInterval = 25ms;
const auto now = Now();
if (_stopping.load())
return;
if (_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
{
return _outboundMessageHandler.QueueMessage(remote, msg, handler);
}
void
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, bool isRouter, llarp_nodedb* nodedb)
{
if (nodedb == nullptr)
{
throw std::invalid_argument("nodedb cannot be null");
}
_nodedb = nodedb;
if (not m_keyManager->initialize(*conf, true, isRouter))
throw std::runtime_error("KeyManager failed to initialize");
if (!FromConfig(conf))
throw std::runtime_error("FromConfig() failed");
if (!InitOutboundLinks())
throw std::runtime_error("InitOutboundLinks() failed");
if (not EnsureIdentity())
throw std::runtime_error("EnsureIdentity() failed");
return true;
}
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()))
{
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();
2020-02-28 01:23:36 +00:00
_running.store(false);
}
void
Router::handle_router_ticker()
2018-11-28 15:18:18 +00:00
{
ticker_job_id = 0;
2020-02-24 19:40:45 +00:00
Tick();
ScheduleTicker(ROUTER_TICK_INTERVAL);
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
if (nextOnionKey.SaveToFile(f.c_str()))
2019-01-29 13:20:27 +00:00
{
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);
if (rotateKeys)
2020-01-21 17:31:48 +00:00
{
// propagate RC by renegotiating sessions
ForEachPeer([](ILinkSession* s) {
if (s->RenegotiateSession())
2020-01-21 17:31:48 +00:00
LogInfo("renegotiated session");
else
LogWarn("failed to renegotiate session");
});
}
if (IsServiceNode())
return SaveRC();
else
return true;
2019-03-25 15:41:37 +00:00
}
2019-07-12 17:21:29 +00:00
bool
Router::FromConfig(Config* conf)
{
2019-07-07 11:29:44 +00:00
// Set netid before anything else
if (!conf->router.m_netId.empty() && strcmp(conf->router.m_netId.c_str(), llarp::DEFAULT_NETID))
{
const auto& netid = conf->router.m_netId;
llarp::LogWarn(
"!!!! you have manually set netid to be '",
netid,
"' which does not equal '",
llarp::DEFAULT_NETID,
"' 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()));
2019-07-07 11:29:44 +00:00
// reset netid in our rc
_rc.netID = llarp::NetID();
}
2019-07-07 11:29:44 +00:00
// IWP config
m_OutboundPort = conf->links.m_OutboundLink.port;
2019-07-12 17:21:29 +00:00
// Router config
_rc.SetNick(conf->router.m_nickname);
_outboundSessionMaker.maxConnectedRouters = conf->router.m_maxConnectedRouters;
_outboundSessionMaker.minConnectedRouters = conf->router.m_minConnectedRouters;
2020-06-04 18:38:35 +00:00
encryption_keyfile = m_keyManager->m_encKeyPath;
our_rc_file = m_keyManager->m_rcPath;
transport_keyfile = m_keyManager->m_transportKeyPath;
ident_keyfile = m_keyManager->m_idKeyPath;
2020-05-06 20:38:44 +00:00
_ourAddress = conf->router.m_publicAddress;
2020-03-31 16:04:47 +00:00
RouterContact::BlockBogons = conf->router.m_blockBogons;
2019-08-26 23:29:17 +00:00
2019-07-12 17:21:29 +00:00
// Lokid Config
usingSNSeed = conf->lokid.usingSNSeed;
2019-07-12 17:21:29 +00:00
whitelistRouters = conf->lokid.whitelistRouters;
2020-05-06 20:38:44 +00:00
lokidRPCAddr = IpAddress(conf->lokid.lokidRPCAddr); // TODO: make config's option an IpAddress
lokidRPCUser = conf->lokid.lokidRPCUser;
2019-07-12 17:21:29 +00:00
lokidRPCPassword = conf->lokid.lokidRPCPassword;
2020-06-04 18:38:35 +00:00
if (usingSNSeed)
ident_keyfile = conf->lokid.ident_keyfile;
// TODO: add config flag for "is service node"
if (conf->links.m_InboundLinks.size())
{
m_isServiceNode = true;
}
2020-05-04 16:33:44 +00:00
networkConfig = conf->network;
2020-04-27 15:24:05 +00:00
/// build a set of strictConnectPubkeys (
/// TODO: make this consistent with config -- do we support multiple strict connections
// or not?
std::set<RouterID> strictConnectPubkeys;
2020-04-27 15:24:05 +00:00
if (not networkConfig.m_strictConnect.empty())
{
2020-04-27 15:24:05 +00:00
const auto& val = networkConfig.m_strictConnect;
if (IsServiceNode())
throw std::runtime_error("cannot use strict-connect option as service node");
2020-04-27 15:24:05 +00:00
// try as a RouterID and as a PubKey, convert to RouterID if needed
llarp::RouterID snode;
llarp::PubKey pk;
if (pk.FromString(val))
2020-04-27 15:24:05 +00:00
strictConnectPubkeys.emplace(pk);
else if (snode.FromString(val))
2020-04-27 15:24:05 +00:00
strictConnectPubkeys.insert(snode);
else
2020-04-27 15:24:05 +00:00
throw std::invalid_argument(stringify("invalid key for strict-connect: ", val));
}
std::vector<fs::path> configRouters = conf->connect.routers;
configRouters.insert(
configRouters.end(), conf->bootstrap.routers.begin(), conf->bootstrap.routers.end());
// if our conf had no bootstrap files specified, try the default location of
// <DATA_DIR>/bootstrap.signed. If this isn't present, leave a useful error message
if (configRouters.size() == 0 and not m_isServiceNode)
{
// TODO: use constant
fs::path defaultBootstrapFile = conf->router.m_dataDir / "bootstrap.signed";
if (fs::exists(defaultBootstrapFile))
configRouters.push_back(defaultBootstrapFile);
else
{
LogError("No bootstrap files specified in config file, and the default");
LogError("bootstrap file ", defaultBootstrapFile, " does not exist.");
2020-04-29 20:19:48 +00:00
LogError("Please provide a bootstrap file (e.g. run 'lokinet-bootstrap)'");
throw std::runtime_error("No bootstrap files available.");
}
}
BootstrapList b_list;
for (const auto& router : configRouters)
{
bool isListFile = false;
{
std::ifstream inf(router.c_str(), std::ios::binary);
if (inf.is_open())
{
const char ch = inf.get();
isListFile = ch == 'l';
}
}
if (isListFile)
{
if (not BDecodeReadFile(router, b_list))
{
throw std::runtime_error(stringify("failed to read bootstrap list file '", router, "'"));
}
}
else
{
RouterContact rc;
if (not rc.Read(router))
{
throw std::runtime_error(
stringify("failed to decode bootstrap RC, file='", router, "' rc=", rc));
}
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);
// create inbound links, if we are a service node
for (const LinksConfig::LinkInfo& serverConfig : conf->links.m_InboundLinks)
{
auto server = iwp::NewInboundLink(
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));
2020-03-19 22:20:45 +00:00
const std::string& key = serverConfig.interface;
int af = serverConfig.addressFamily;
uint16_t port = serverConfig.port;
if (!server->Configure(netloop(), key, af, port))
{
throw std::runtime_error(stringify("failed to bind inbound link on ", key, " port ", port));
}
_linkManager.AddLink(std::move(server), true);
}
// Network config
2020-05-20 22:48:13 +00:00
if (conf->network.m_enableProfiling.has_value() and not*conf->network.m_enableProfiling)
2018-12-19 17:48:29 +00:00
{
routerProfiling().Disable();
LogWarn("router profiling explicitly disabled");
2018-12-19 17:48:29 +00:00
}
if (!conf->network.m_routerProfilesFile.empty())
{
routerProfilesFile = conf->network.m_routerProfilesFile;
routerProfiling().Load(routerProfilesFile.c_str());
llarp::LogInfo("setting profiles to ", routerProfilesFile);
}
// API config
enableRPCServer = conf->api.m_enableRPCServer;
2020-05-06 20:38:44 +00:00
rpcBindAddr = IpAddress(conf->api.m_rpcBindAddr); // TODO: make this an IpAddress in config
if (not IsServiceNode())
{
hiddenServiceContext().AddEndpoint(*conf);
}
// Logging config
LogContext::Instance().Initialize(
conf->logging.m_logLevel,
conf->logging.m_logType,
conf->logging.m_logFile,
conf->router.m_nickname,
diskworker());
2018-12-19 17:48:29 +00:00
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; })
2019-05-09 15:36:39 +00:00
> 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
{
2020-02-24 19:40:45 +00:00
static constexpr auto ReportStatsInterval = 1h;
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())
2019-07-15 16:56:09 +00:00
{
LogInfo(NumberOfConnectedClients(), " client connections");
LogInfo(_rc.Age(now), " since we last updated our RC");
LogInfo(_rc.TimeUntilExpires(now), " until our RC expires");
2019-07-15 16:56:09 +00:00
}
2020-05-15 12:38:04 +00:00
LogInfo(now - m_LastStatsReport, " last reported stats");
2019-07-15 16:56:09 +00:00
m_LastStatsReport = now;
}
void
Router::Tick()
2018-06-19 17:11:24 +00:00
{
if (_stopping)
2019-04-23 18:29:42 +00:00
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)
2020-02-25 22:32:57 +00:00
{
std::stringstream ss;
ss << "WATCHDOG=1\nSTATUS=v" << llarp::VERSION_STR;
if (IsServiceNode())
2020-02-25 22:32:57 +00:00
{
ss << " snode | known/svc/clients: " << nodedb()->num_loaded() << "/"
<< NumberOfConnectedRouters() << "/" << NumberOfConnectedClients() << " | "
<< pathContext().CurrentTransitPaths() << " active paths";
2020-02-25 22:32:57 +00:00
}
else
{
ss << " client | known/connected: " << nodedb()->num_loaded() << "/"
<< NumberOfConnectedRouters() << " | path success: ";
hiddenServiceContext().ForEachService([&ss](const auto& name, const auto& ep) {
ss << " [" << name << " " << std::setprecision(4)
<< (100.0 * ep->CurrentBuildStats().SuccessRatio()) << "%]";
return true;
});
2020-02-25 22:32:57 +00:00
}
const auto status = ss.str();
::sd_notify(0, status.c_str());
}
#endif
2019-03-04 17:03:18 +00:00
routerProfiling().Tick();
if (ShouldReportStats(now))
2019-07-15 16:56:09 +00:00
{
ReportStats();
}
2020-02-24 19:40:45 +00:00
_rcGossiper.Decay(now);
2020-01-30 17:23:16 +00:00
_rcLookupHandler.PeriodicUpdate(now);
2019-07-15 16:56:09 +00:00
const bool isSvcNode = IsServiceNode();
if (_rc.ExpiresSoon(now, std::chrono::milliseconds(randint() % 10000))
|| (now - _rc.last_updated) > rcRegenInterval)
2018-12-19 17:48:29 +00:00
{
LogInfo("regenerating RC");
if (!UpdateOurRC(false))
LogError("Failed to update our RC");
}
else
{
GossipRCIfNeeded(_rc);
2020-01-30 17:23:16 +00:00
}
const bool gotWhitelist = _rcLookupHandler.HaveReceivedWhitelist();
2020-03-08 12:09:48 +00:00
// remove RCs for nodes that are no longer allowed by network policy
nodedb()->RemoveIf([&](const RouterContact& rc) -> bool {
2020-03-09 15:08:56 +00:00
// don't purge bootstrap nodes from nodedb
if (IsBootstrapNode(rc.pubkey))
2020-03-08 12:09:48 +00:00
return false;
// if for some reason we stored an RC that isn't a valid router
// purge this entry
if (not rc.IsPublicRouter())
2020-03-08 12:09:48 +00:00
return true;
2020-03-08 12:12:23 +00:00
// clients have a notion of a whilelist
// we short circuit logic here so we dont remove
// routers that are not whitelisted for first hops
if (not isSvcNode)
2020-03-08 12:09:48 +00:00
return false;
// if we have a whitelist enabled and we don't
// have the whitelist yet don't remove the entry
if (whitelistRouters and not gotWhitelist)
return false;
// if we have no whitelist enabled or we have
// the whitelist enabled and we got the whitelist
// check against the whitelist and remove if it's not
2020-03-09 15:05:40 +00:00
// in the whitelist OR if there is no whitelist don't remove
2020-03-08 12:09:48 +00:00
return not _rcLookupHandler.RemoteIsAllowed(rc.pubkey);
});
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)
2019-12-03 17:03:19 +00:00
{
connected += _linkManager.NumberOfPendingConnections();
}
const int interval = isSvcNode ? 5 : 2;
2020-01-18 20:46:22 +00:00
const auto timepoint_now = Clock_t::now();
if (timepoint_now >= m_NextExploreAt)
2020-01-18 20:46:22 +00:00
{
2020-01-18 20:55:50 +00:00
_rcLookupHandler.ExploreNetwork();
2020-01-18 20:46:22 +00:00
m_NextExploreAt = timepoint_now + std::chrono::seconds(interval);
}
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);
2020-01-14 17:01:41 +00:00
// save profiles
if (routerProfiling().ShouldSave(now))
2019-03-25 15:41:37 +00:00
{
diskworker()->addJob([&]() { routerProfiling().Save(routerProfilesFile.c_str()); });
2019-03-25 15:41:37 +00:00
}
2020-01-14 17:01:41 +00:00
// save nodedb
if (nodedb()->ShouldSaveToDisk(now))
2020-01-14 17:01:41 +00:00
{
nodedb()->AsyncFlushToDisk();
}
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())
2019-09-10 14:16:32 +00:00
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
{
return CryptoManager::instance()->sign(sig, identity(), buf);
}
void
2020-02-24 19:40:45 +00:00
Router::ScheduleTicker(llarp_time_t interval)
2018-06-06 12:46:26 +00:00
{
ticker_job_id = _logic->call_later(interval, 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(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);
}
bool
Router::StartJsonRpc()
{
if (_running || _stopping)
return false;
if (enableRPCServer)
2018-10-09 12:06:30 +00:00
{
2020-05-06 20:38:44 +00:00
if (rpcBindAddr.isEmpty())
{
rpcBindAddr = DefaultRPCBindAddr;
}
rpcServer = std::make_unique<rpc::Server>(this);
2020-05-04 18:39:14 +00:00
if (not rpcServer->Start(rpcBindAddr))
{
LogError("failed to bind jsonrpc to ", rpcBindAddr);
2020-05-04 18:39:14 +00:00
return false;
}
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);
2020-05-04 18:39:14 +00:00
if (not rpcCaller->Start(lokidRPCAddr))
{
2020-05-04 18:39:14 +00:00
LogError("RPC Caller to ", lokidRPCAddr, " failed to start");
return false;
}
LogInfo("RPC Caller to ", lokidRPCAddr, " started");
}
2018-10-09 12:06:30 +00:00
if (!cryptoworker->start())
2019-07-15 17:19:31 +00:00
{
LogError("crypto worker failed to start");
return false;
}
if (!disk->start())
2019-07-15 17:19:31 +00:00
{
LogError("disk worker failed to start");
return false;
}
2018-11-19 16:40:49 +00:00
routerProfiling().Load(routerProfilesFile.c_str());
// set public signing key
_rc.pubkey = seckey_topublic(identity());
// set router version if service node
if (IsServiceNode())
2020-01-25 16:28:07 +00:00
{
2020-01-25 17:21:28 +00:00
_rc.routerVersion = RouterVersion(llarp::VERSION, LLARP_PROTO_VERSION);
}
2019-04-08 18:21:01 +00:00
_linkManager.ForEachInboundLink([&](LinkLayer_ptr link) {
AddressInfo ai;
if (link->GetOurAddressInfo(ai))
{
// override ip and port
if (not _ourAddress.isEmpty())
2019-04-08 18:21:01 +00:00
{
ai.fromIpAddress(_ourAddress);
2019-04-08 18:21:01 +00:00
}
if (RouterContact::BlockBogons && IsBogon(ai.ip))
return;
LogInfo("adding address: ", ai);
_rc.addrs.push_back(ai);
if (ExitEnabled())
{
const IpAddress address = ai.toIpAddress();
_rc.exits.emplace_back(_rc.pubkey, address);
LogInfo("Exit relay started, advertised as exiting at: ", address);
}
}
});
// 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 (IsServiceNode())
{
if (!SaveRC())
{
LogError("failed to save RC");
return false;
}
}
_outboundSessionMaker.SetOurRouter(pubkey());
if (!_linkManager.StartLinks(_logic, cryptoworker))
{
LogWarn("One or more links failed to start.");
return false;
}
if (IsServiceNode())
{
// initialize as service node
if (!InitServiceNode())
{
LogError("Failed to initialize service node");
return false;
}
const RouterID us = pubkey();
LogInfo("initalized service node: ", us);
// init gossiper here
2020-03-04 00:50:20 +00:00
_rcGossiper.Init(&_linkManager, us, this);
// 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;
}
}
LogInfo("starting hidden service context...");
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));
2020-02-24 19:40:45 +00:00
ScheduleTicker(ROUTER_TICK_INTERVAL);
_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 > 0s && _now > _startedAt)
return _now - _startedAt;
2020-02-24 19:40:45 +00:00
return 0s;
}
void
Router::AfterStopLinks()
{
Close();
}
void
Router::AfterStopIssued()
{
StopLinks();
nodedb()->AsyncFlushToDisk();
2020-02-24 19:40:45 +00:00
_logic->call_later(200ms, 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");
2020-02-25 22:32:57 +00:00
#if defined(WITH_SYSTEMD)
sd_notify(0, "STOPPING=1\nSTATUS=Shutting down");
#endif
2019-02-22 16:21:05 +00:00
hiddenServiceContext().StopAll();
_exitContext.Stop();
if (rpcServer)
rpcServer->Stop();
paths.PumpUpstream();
_linkManager.PumpLinks();
2020-02-24 19:40:45 +00:00
_logic->call_later(200ms, 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);
}
std::string
Router::ShortName() const
{
return RouterID(pubkey()).ToString().substr(0, 8);
}
uint32_t
Router::NextPathBuildNumber()
{
return path_build_count++;
}
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())
2019-12-03 17:03:19 +00:00
{
connected += _linkManager.NumberOfPendingConnections();
}
if (connected >= want)
2019-12-03 17:03:19 +00:00
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());
_exitContext.AddExitEndpoint("default-connectivity", networkConfig, dnsConfig);
return true;
2018-11-26 13:29:45 +00:00
}
2018-06-14 17:35:12 +00:00
bool
Router::ValidateConfig(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
{
auto link = iwp::NewOutboundLink(
m_keyManager,
util::memFn(&AbstractRouter::rc, this),
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),
util::memFn(&AbstractRouter::SessionClosed, this),
util::memFn(&AbstractRouter::PumpLL, this));
if (!link)
throw std::runtime_error("NewOutboundLink() failed to provide a link");
2019-08-07 16:33:29 +00:00
const auto afs = {AF_INET, AF_INET6};
for (const auto af : afs)
{
2020-04-29 20:19:48 +00:00
if (not link->Configure(netloop(), "*", af, m_OutboundPort))
2019-01-05 13:48:00 +00:00
continue;
2020-04-29 20:19:48 +00:00
2019-08-07 16:33:29 +00:00
_linkManager.AddLink(std::move(link), false);
return true;
}
throw std::runtime_error(
stringify("Failed to init AF_INET and AF_INET6 on port ", m_OutboundPort));
}
2018-06-21 13:33:42 +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