lokinet/llarp/router.cpp

1362 lines
34 KiB
C++
Raw Normal View History

2018-11-19 23:40:38 +00:00
#include <llarp/proto.hpp>
2018-08-10 21:34:11 +00:00
#include <llarp/iwp.hpp>
#include <llarp/link_message.hpp>
2018-09-04 12:55:20 +00:00
#include <llarp/link/utp.hpp>
2018-10-19 11:41:36 +00:00
#include <llarp/rpc.hpp>
2018-05-21 13:17:07 +00:00
#include "buffer.hpp"
#include "encode.hpp"
#include "llarp/net.hpp"
#include "logger.hpp"
#include "router.hpp"
#include "str.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>
2018-11-08 12:31:50 +00:00
#if defined(RPI) || defined(ANDROID)
#include <unistd.h>
#endif
2018-05-20 17:45:47 +00:00
namespace llarp
{
void
router_iter_config(llarp_config_iterator *iter, const char *section,
const char *key, const char *val);
2018-05-30 20:56:47 +00:00
struct async_verify_context
{
Router *router;
2018-09-06 11:46:19 +00:00
TryConnectJob *establish_job;
2018-05-30 20:56:47 +00:00
};
2018-02-01 13:21:00 +00:00
} // namespace llarp
2017-11-28 14:05:31 +00:00
2018-09-06 11:46:19 +00:00
struct TryConnectJob
{
2018-09-06 11:46:19 +00:00
llarp::RouterContact rc;
llarp::ILinkLayer *link;
llarp::Router *router;
uint16_t triesLeft;
TryConnectJob(const llarp::RouterContact &remote, llarp::ILinkLayer *l,
uint16_t tries, llarp::Router *r)
2018-09-06 11:46:19 +00:00
: rc(remote), link(l), router(r), triesLeft(tries)
{
}
void
Failed()
{
llarp::LogInfo("session to ", llarp::RouterID(rc.pubkey.data()), " closed");
link->CloseSessionTo(rc.pubkey);
}
void
Success()
{
2018-11-20 15:46:26 +00:00
router->FlushOutboundFor(rc.pubkey, link);
}
void
AttemptTimedout()
{
router->routerProfiling.MarkTimeout(rc.pubkey);
if(ShouldRetry())
{
Attempt();
return;
}
2018-11-21 14:10:02 +00:00
if(!router->IsServiceNode())
{
if(router->routerProfiling.IsBad(rc.pubkey))
llarp_nodedb_del_rc(router->nodedb, rc.pubkey);
}
// delete this
2018-09-06 20:31:58 +00:00
router->pendingEstablishJobs.erase(rc.pubkey);
}
void
Attempt()
{
--triesLeft;
if(!link->TryEstablishTo(rc))
llarp::LogError("did not attempt connection to ", rc.pubkey,
" and it has ", rc.addrs.size(), " advertised addresses");
}
bool
ShouldRetry() const
{
return triesLeft > 0;
}
};
static void
on_try_connecting(void *u)
{
2018-09-06 11:46:19 +00:00
TryConnectJob *j = static_cast< TryConnectJob * >(u);
j->Attempt();
}
2018-08-30 18:48:43 +00:00
bool
llarp_router_try_connect(llarp::Router *router,
2018-08-30 18:48:43 +00:00
const llarp::RouterContact &remote,
uint16_t numretries)
{
// do we already have a pending job for this remote?
if(router->HasPendingConnectJob(remote.pubkey))
{
llarp::LogDebug("We have pending connect jobs to ", remote.pubkey);
return false;
}
2018-09-06 11:46:19 +00:00
auto link = router->outboundLink.get();
auto itr = router->pendingEstablishJobs.insert(std::make_pair(
remote.pubkey.data(),
2018-09-06 11:46:19 +00:00
std::make_unique< TryConnectJob >(remote, link, numretries, router)));
TryConnectJob *job = itr.first->second.get();
// try establishing async
2018-12-10 14:14:55 +00:00
router->logic->queue_job({job, &on_try_connecting});
return true;
2018-08-30 18:48:43 +00:00
}
bool
llarp_findOrCreateIdentity(llarp_crypto *crypto, const char *fpath,
byte_t *secretkey)
{
llarp::LogDebug("find or create ", fpath);
fs::path path(fpath);
std::error_code ec;
if(!fs::exists(path, ec))
{
llarp::LogInfo("generating new identity key");
crypto->identity_keygen(secretkey);
std::ofstream f(path.string(), std::ios::binary);
if(f.is_open())
{
f.write((char *)secretkey, SECKEYSIZE);
}
}
std::ifstream f(path.string(), std::ios::binary);
if(f.is_open())
{
f.read((char *)secretkey, SECKEYSIZE);
return true;
}
llarp::LogInfo("failed to get identity key");
return false;
2018-08-14 21:17:18 +00:00
}
// C++ ...
bool
llarp_findOrCreateEncryption(llarp_crypto *crypto, const char *fpath,
llarp::SecretKey &encryption)
{
llarp::LogDebug("find or create ", fpath);
fs::path path(fpath);
std::error_code ec;
if(!fs::exists(path, ec))
{
llarp::LogInfo("generating new encryption key");
crypto->encryption_keygen(encryption);
std::ofstream f(path.string(), std::ios::binary);
if(f.is_open())
{
f.write((char *)encryption.data(), SECKEYSIZE);
}
}
std::ifstream f(path.string(), std::ios::binary);
if(f.is_open())
2018-06-20 12:34:48 +00:00
{
f.read((char *)encryption.data(), SECKEYSIZE);
return true;
2018-06-20 12:34:48 +00:00
}
llarp::LogInfo("failed to get encryption key");
return false;
}
namespace llarp
{
void
Router::HandleLinkSessionEstablished(llarp::RouterContact rc,
llarp::ILinkLayer *link)
{
async_verify_RC(rc, link);
}
Router::Router(struct llarp_threadpool *_tp, struct llarp_ev_loop *_netloop,
llarp::Logic *_logic)
: ready(false)
, netloop(_netloop)
, tp(_tp)
, logic(_logic)
, paths(this)
, exitContext(this)
, dht(llarp_dht_context_new(this))
, inbound_link_msg_parser(this)
, hiddenServiceContext(this)
2018-06-20 17:45:44 +00:00
{
// set rational defaults
this->ip4addr.sin_family = AF_INET;
this->ip4addr.sin_port = htons(1090);
#ifdef TESTNET
disk = tp;
#else
disk = llarp_init_threadpool(1, "llarp-diskio");
#endif
llarp_crypto_init(&crypto);
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
}
bool
Router::HandleRecvLinkMessageBuffer(llarp::ILinkSession *session,
llarp_buffer_t buf)
{
if(!session)
{
llarp::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 llarp::RouterID &remote, llarp_time_t until)
2018-10-07 15:29:36 +00:00
{
llarp::LogDebug("persist session to ", remote, " until ", until);
m_PersistingSessions[remote] =
std::max(until, m_PersistingSessions[remote]);
2018-06-01 14:08:54 +00:00
}
constexpr size_t MaxPendingSendQueueSize = 8;
bool
Router::SendToOrQueue(const llarp::RouterID &remote,
const llarp::ILinkMessage *msg)
2018-06-14 15:10:31 +00:00
{
for(const auto &link : inboundLinks)
2018-11-21 17:46:33 +00:00
{
if(link->HasSessionTo(remote.data()))
{
SendTo(remote, msg, link.get());
return true;
}
2018-11-21 17:46:33 +00:00
}
if(outboundLink && outboundLink->HasSessionTo(remote.data()))
2018-11-21 17:46:33 +00:00
{
SendTo(remote, msg, outboundLink.get());
return true;
2018-11-21 17:46:33 +00:00
}
// no link available
// this will create an entry in the obmq if it's not already there
auto itr = outboundMessageQueue.find(remote);
if(itr == outboundMessageQueue.end())
{
outboundMessageQueue.insert(std::make_pair(remote, MessageQueue()));
}
// encode
llarp_buffer_t buf =
llarp::StackBuffer< decltype(linkmsg_buffer) >(linkmsg_buffer);
if(!msg->BEncode(&buf))
return false;
// queue buffer
auto &q = outboundMessageQueue[remote];
if(q.size() < MaxPendingSendQueueSize)
{
buf.sz = buf.cur - buf.base;
q.emplace(buf.sz);
memcpy(q.back().data(), buf.base, buf.sz);
}
else
{
llarp::LogWarn("tried to queue a message to ", remote,
" but the queue is full so we drop it like it's hawt");
}
llarp::RouterContact remoteRC;
// we don't have an open session to that router right now
if(llarp_nodedb_get_rc(nodedb, remote, remoteRC))
{
// try connecting directly as the rc is loaded from disk
llarp_router_try_connect(this, remoteRC, 10);
return true;
}
// we don't have the RC locally so do a dht lookup
dht->impl.LookupRouter(remote,
std::bind(&Router::HandleDHTLookupForSendTo, this,
remote, std::placeholders::_1));
return true;
2018-06-14 15:10:31 +00:00
}
2018-06-13 12:58:51 +00:00
void
Router::HandleDHTLookupForSendTo(
llarp::RouterID remote,
const std::vector< llarp::RouterContact > &results)
{
if(results.size())
{
if(whitelistRouters
&& lokinetRouters.find(remote) == lokinetRouters.end())
{
return;
}
if(results[0].Verify(&crypto))
{
llarp_nodedb_put_rc(nodedb, results[0]);
llarp_router_try_connect(this, results[0], 10);
}
}
else
{
DiscardOutboundFor(remote);
}
}
void
Router::ForEachPeer(
std::function< void(const llarp::ILinkSession *, bool) > visit) const
{
outboundLink->ForEachSession(
[visit](const llarp::ILinkSession *peer) { visit(peer, true); });
for(const auto &link : inboundLinks)
{
link->ForEachSession(
[visit](const llarp::ILinkSession *peer) { visit(peer, false); });
}
}
void
Router::try_connect(fs::path rcfile)
2018-05-16 18:13:18 +00:00
{
llarp::RouterContact remote;
if(!remote.Read(rcfile.string().c_str()))
{
llarp::LogError("failure to decode or verify of remote RC");
return;
}
if(remote.Verify(&crypto))
{
llarp::LogDebug("verified signature");
// store into filesystem
if(!llarp_nodedb_put_rc(nodedb, remote))
{
llarp::LogWarn("failed to store");
}
if(!llarp_router_try_connect(this, remote, 10))
{
// or error?
llarp::LogWarn("session already made");
}
}
else
llarp::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(!EnsureEncryptionKey())
return false;
return llarp_findOrCreateIdentity(&crypto, ident_keyfile.string().c_str(),
identity);
}
2018-04-05 14:23:14 +00:00
bool
Router::EnsureEncryptionKey()
{
return llarp_findOrCreateEncryption(
&crypto, encryption_keyfile.string().c_str(), encryption);
}
2018-01-29 14:27:24 +00:00
void
Router::AddInboundLink(std::unique_ptr< llarp::ILinkLayer > &link)
2018-05-20 17:45:47 +00:00
{
inboundLinks.push_back(std::move(link));
2018-05-20 17:45:47 +00:00
}
bool
Router::Configure(struct llarp_config *conf)
{
llarp_config_iterator iter;
iter.user = this;
iter.visit = llarp::router_iter_config;
llarp_config_iter(conf, &iter);
if(!InitOutboundLink())
return false;
if(!Ready())
{
return false;
}
return EnsureIdentity();
}
2018-11-21 14:10:02 +00:00
bool
Router::Ready()
{
return outboundLink != nullptr;
}
bool
Router::SaveRC()
2018-08-18 14:01:21 +00:00
{
llarp::LogDebug("verify RC signature");
if(!rc().Verify(&crypto))
{
rc().Dump< MAX_RC_SIZE >();
llarp::LogError("RC is invalid, not saving");
return false;
}
return rc().Write(our_rc_file.string().c_str());
2018-06-20 12:34:48 +00:00
}
2018-05-29 13:40:26 +00:00
bool
Router::IsServiceNode() const
{
return inboundLinks.size() > 0;
}
2018-05-30 20:56:47 +00:00
void
Router::Close()
{
for(const auto &link : inboundLinks)
2018-05-30 20:56:47 +00:00
{
link->Stop();
2018-05-30 20:56:47 +00:00
}
inboundLinks.clear();
if(outboundLink)
{
outboundLink->Stop();
outboundLink.reset(nullptr);
}
}
void
Router::on_verify_client_rc(llarp_async_verify_rc *job)
{
llarp::async_verify_context *ctx =
static_cast< llarp::async_verify_context * >(job->user);
ctx->router->pendingEstablishJobs.erase(job->rc.pubkey);
auto router = ctx->router;
llarp::PubKey pk(job->rc.pubkey);
router->FlushOutboundFor(pk, router->GetLinkWithSessionByPubkey(pk));
2018-11-20 15:46:26 +00:00
delete ctx;
2018-11-22 15:02:51 +00:00
router->pendingVerifyRC.erase(pk);
2018-05-30 20:56:47 +00:00
}
void
Router::on_verify_server_rc(llarp_async_verify_rc *job)
2018-09-06 11:46:19 +00:00
{
llarp::async_verify_context *ctx =
static_cast< llarp::async_verify_context * >(job->user);
auto router = ctx->router;
llarp::PubKey pk(job->rc.pubkey);
if(!job->valid)
{
if(ctx->establish_job)
{
// was an outbound attempt
ctx->establish_job->Failed();
}
delete ctx;
router->DiscardOutboundFor(pk);
router->pendingVerifyRC.erase(pk);
2018-09-06 13:16:24 +00:00
return;
}
// we're valid, which means it's already been committed to the nodedb
2018-09-06 13:16:24 +00:00
llarp::LogDebug("rc verified and saved to nodedb");
2018-06-10 14:05:48 +00:00
if(router->validRouters.count(pk))
{
router->validRouters.erase(pk);
}
2018-05-30 20:56:47 +00:00
llarp::RouterContact rc = job->rc;
router->validRouters.insert(std::make_pair(pk.data(), rc));
2018-05-30 20:56:47 +00:00
// track valid router in dht
router->dht->impl.nodes->PutNode(rc);
2018-05-30 20:56:47 +00:00
// mark success in profile
router->routerProfiling.MarkSuccess(pk);
// this was an outbound establish job
if(ctx->establish_job)
{
ctx->establish_job->Success();
}
2018-11-28 15:18:18 +00:00
else
router->FlushOutboundFor(pk, router->GetLinkWithSessionByPubkey(pk));
delete ctx;
router->pendingVerifyRC.erase(pk);
2018-11-28 15:18:18 +00:00
}
void
Router::handle_router_ticker(void *user, uint64_t orig, uint64_t left)
2018-11-28 15:18:18 +00:00
{
if(left)
return;
Router *self = static_cast< Router * >(user);
self->ticker_job_id = 0;
self->Tick();
self->ScheduleTicker(orig);
2018-11-28 15:18:18 +00:00
}
bool
Router::ConnectionToRouterAllowed(const llarp::RouterID &router) const
{
if(strictConnectPubkeys.size() && strictConnectPubkeys.count(router) == 0)
return false;
else if(IsServiceNode() && whitelistRouters)
return lokinetRouters.count(router) != 0;
else
return true;
}
void
Router::HandleDHTLookupForExplore(
llarp::RouterID remote,
const std::vector< llarp::RouterContact > &results)
2018-08-14 21:17:18 +00:00
{
if(results.size() == 0)
return;
for(const auto &rc : results)
{
if(rc.Verify(&crypto))
llarp_nodedb_put_rc(nodedb, rc);
else
return;
}
if(ConnectionToRouterAllowed(remote))
{
TryEstablishTo(remote);
}
2018-08-14 21:17:18 +00:00
}
void
Router::TryEstablishTo(const llarp::RouterID &remote)
2018-08-14 21:17:18 +00:00
{
if(!ConnectionToRouterAllowed(remote))
{
llarp::LogWarn("not connecting to ", remote,
" as it's not permitted by config");
2018-10-07 15:29:36 +00:00
return;
}
llarp::RouterContact rc;
if(llarp_nodedb_get_rc(nodedb, remote, rc))
{
// try connecting async
llarp_router_try_connect(this, rc, 5);
}
else if(IsServiceNode() || !routerProfiling.IsBad(remote))
{
if(dht->impl.HasRouterLookup(remote))
return;
llarp::LogInfo("looking up router ", remote);
// dht lookup as we don't know it
dht->impl.LookupRouter(
remote,
std::bind(&Router::HandleDHTLookupForTryEstablishTo, this, remote,
std::placeholders::_1));
}
else
{
llarp::LogWarn("not connecting to ", remote, " as it's unreliable");
}
2018-08-14 21:17:18 +00:00
}
void
Router::OnConnectTimeout(const llarp::RouterID &remote)
{
auto itr = pendingEstablishJobs.find(remote);
if(itr != pendingEstablishJobs.end())
{
itr->second->AttemptTimedout();
}
}
2018-08-14 21:17:18 +00:00
void
Router::HandleDHTLookupForTryEstablishTo(
llarp::RouterID remote,
const std::vector< llarp::RouterContact > &results)
{
if(results.size() == 0)
{
if(!IsServiceNode())
routerProfiling.MarkTimeout(remote);
}
for(const auto &result : results)
{
if(whitelistRouters
&& lokinetRouters.find(result.pubkey) == lokinetRouters.end())
continue;
llarp_nodedb_put_rc(nodedb, result);
llarp_router_try_connect(this, result, 10);
}
}
size_t
Router::NumberOfConnectedRouters() const
{
return validRouters.size();
}
2018-06-19 17:11:24 +00:00
void
Router::Tick()
2018-06-19 17:11:24 +00:00
{
// llarp::LogDebug("tick router");
auto now = llarp_ev_loop_time_now_ms(netloop);
paths.TickPaths(now);
paths.ExpirePaths(now);
2018-08-14 21:17:18 +00:00
{
auto itr = m_PersistingSessions.begin();
while(itr != m_PersistingSessions.end())
2018-08-14 21:17:18 +00:00
{
auto link = GetLinkWithSessionByPubkey(itr->first);
if(now < itr->second)
2018-08-14 21:17:18 +00:00
{
if(link)
{
llarp::LogDebug("keepalive to ", itr->first);
link->KeepAliveSessionTo(itr->first);
}
else
{
llarp::LogDebug("establish to ", itr->first);
TryEstablishTo(itr->first);
}
++itr;
2018-08-14 21:17:18 +00:00
}
else
{
llarp::LogInfo("commit to ", itr->first, " expired");
itr = m_PersistingSessions.erase(itr);
2018-08-14 21:17:18 +00:00
}
}
}
if(inboundLinks.size() == 0)
{
size_t N = llarp_nodedb_num_loaded(nodedb);
if(N < minRequiredRouters)
{
llarp::LogInfo("We need at least ", minRequiredRouters,
" service nodes to build paths but we have ", N);
// TODO: only connect to random subset
if(bootstrapRCList.size())
{
for(const auto &rc : bootstrapRCList)
{
llarp_router_try_connect(this, rc, 4);
dht->impl.ExploreNetworkVia(rc.pubkey.data());
}
}
else
llarp::LogError("we have no bootstrap nodes specified");
2018-08-14 21:17:18 +00:00
}
paths.BuildPaths(now);
hiddenServiceContext.Tick(now);
2018-08-14 21:17:18 +00:00
}
if(NumberOfConnectedRouters() < minConnectedRouters)
{
ConnectToRandomRouters(minConnectedRouters);
}
exitContext.Tick(now);
if(rpcCaller)
rpcCaller->Tick(now);
2018-08-22 16:19:51 +00:00
}
void
Router::SendTo(llarp::RouterID remote, const llarp::ILinkMessage *msg,
llarp::ILinkLayer *selected)
2018-08-22 16:19:51 +00:00
{
llarp_buffer_t buf =
llarp::StackBuffer< decltype(linkmsg_buffer) >(linkmsg_buffer);
if(!msg->BEncode(&buf))
{
llarp::LogWarn("failed to encode outbound message, buffer size left: ",
llarp_buffer_size_left(buf));
return;
}
// set size of message
buf.sz = buf.cur - buf.base;
buf.cur = buf.base;
llarp::LogDebug("send ", buf.sz, " bytes to ", remote);
if(selected)
{
if(selected->SendTo(remote, buf))
return;
}
bool sent = outboundLink->SendTo(remote, buf);
if(!sent)
2018-06-19 17:11:24 +00:00
{
for(const auto &link : inboundLinks)
{
if(!sent)
{
sent = link->SendTo(remote, buf);
}
}
2018-06-19 17:11:24 +00:00
}
if(!sent)
llarp::LogWarn("message to ", remote, " was dropped");
2018-06-19 17:11:24 +00:00
}
2018-05-30 20:56:47 +00:00
void
Router::ScheduleTicker(uint64_t ms)
2018-06-06 12:46:26 +00:00
{
ticker_job_id = logic->call_later({ms, this, &handle_router_ticker});
2018-06-06 12:46:26 +00:00
}
void
Router::SessionClosed(const llarp::RouterID &remote)
2018-06-20 12:34:48 +00:00
{
__llarp_dht_remove_peer(dht, remote);
// remove from valid routers if it's a valid router
validRouters.erase(remote);
llarp::LogInfo("Session to ", remote, " fully closed");
2018-06-20 12:34:48 +00:00
}
llarp::ILinkLayer *
Router::GetLinkWithSessionByPubkey(const llarp::RouterID &pubkey)
2018-06-06 12:46:26 +00:00
{
if(outboundLink->HasSessionTo(pubkey))
return outboundLink.get();
for(const auto &link : inboundLinks)
2018-06-06 12:46:26 +00:00
{
if(link->HasSessionTo(pubkey))
return link.get();
2018-06-06 12:46:26 +00:00
}
return nullptr;
2018-06-06 12:46:26 +00:00
}
2018-05-30 20:56:47 +00:00
void
Router::FlushOutboundFor(llarp::RouterID remote, llarp::ILinkLayer *chosen)
2018-07-03 13:33:37 +00:00
{
llarp::LogDebug("Flush outbound for ", remote);
2018-07-03 13:33:37 +00:00
auto itr = outboundMessageQueue.find(remote);
if(itr == outboundMessageQueue.end())
{
pendingEstablishJobs.erase(remote);
return;
}
if(!chosen)
{
DiscardOutboundFor(remote);
pendingEstablishJobs.erase(remote);
return;
}
while(itr->second.size())
{
auto buf = llarp::ConstBuffer(itr->second.front());
if(!chosen->SendTo(remote, buf))
llarp::LogWarn("failed to send outboud message to ", remote, " via ",
chosen->Name());
2018-11-22 15:53:11 +00:00
itr->second.pop();
}
2018-11-22 15:02:51 +00:00
pendingEstablishJobs.erase(remote);
2018-06-01 14:08:54 +00:00
}
2018-06-13 12:58:51 +00:00
void
Router::DiscardOutboundFor(const llarp::RouterID &remote)
{
outboundMessageQueue.erase(remote);
}
bool
Router::GetRandomConnectedRouter(llarp::RouterContact &result) const
2018-11-22 15:02:51 +00:00
{
auto sz = validRouters.size();
if(sz)
2018-11-22 15:02:51 +00:00
{
auto itr = validRouters.begin();
if(sz > 1)
std::advance(itr, llarp_randint() % sz);
result = itr->second;
return true;
2018-11-22 15:02:51 +00:00
}
return false;
2018-11-22 15:02:51 +00:00
}
void
Router::async_verify_RC(const llarp::RouterContact &rc,
llarp::ILinkLayer *link)
2018-10-09 12:06:30 +00:00
{
if(pendingVerifyRC.count(rc.pubkey))
return;
if(rc.IsPublicRouter() && whitelistRouters)
2018-10-09 12:06:30 +00:00
{
if(lokinetRouters.find(rc.pubkey) == lokinetRouters.end())
{
llarp::LogInfo(rc.pubkey, " is NOT a valid service node, rejecting");
link->CloseSessionTo(rc.pubkey);
return;
}
2018-10-09 12:06:30 +00:00
}
llarp_async_verify_rc *job = &pendingVerifyRC[rc.pubkey];
llarp::async_verify_context *ctx = new llarp::async_verify_context();
ctx->router = this;
ctx->establish_job = nullptr;
auto itr = pendingEstablishJobs.find(rc.pubkey);
if(itr != pendingEstablishJobs.end())
ctx->establish_job = itr->second.get();
job->user = ctx;
job->rc = rc;
job->valid = false;
job->hook = nullptr;
job->nodedb = nodedb;
job->logic = logic;
// job->crypto = &crypto; // we already have this
job->cryptoworker = tp;
job->diskworker = disk;
if(rc.IsPublicRouter())
job->hook = &Router::on_verify_server_rc;
else
job->hook = &Router::on_verify_client_rc;
llarp_nodedb_async_verify(job);
}
bool
Router::Run(struct llarp_nodedb *nodedb)
{
this->nodedb = nodedb;
if(enableRPCServer)
2018-10-09 12:06:30 +00:00
{
if(rpcBindAddr.empty())
{
rpcBindAddr = DefaultRPCBindAddr;
}
rpcServer = std::make_unique< llarp::rpc::Server >(this);
while(!rpcServer->Start(rpcBindAddr))
{
llarp::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
}
llarp::LogInfo("Bound RPC server to ", rpcBindAddr);
2018-10-09 12:06:30 +00:00
}
llarp_threadpool_start(tp);
llarp_threadpool_start(disk);
2018-11-19 16:40:49 +00:00
routerProfiling.Load(routerProfilesFile.c_str());
llarp::Addr publicAddr(this->addrInfo);
if(this->publicOverride)
{
llarp::LogDebug("public address:port ", publicAddr);
}
llarp::LogInfo("You have ", inboundLinks.size(), " inbound links");
for(const auto &link : inboundLinks)
{
llarp::AddressInfo addr;
if(!link->GetOurAddressInfo(addr))
continue;
llarp::Addr a(addr);
if(this->publicOverride && a.sameAddr(publicAddr))
{
llarp::LogInfo("Found adapter for public address");
}
if(!llarp::IsBogon(*a.addr6()))
{
llarp::LogInfo("Loading Addr: ", a, " into our RC");
_rc.addrs.push_back(addr);
}
};
if(this->publicOverride)
{
llarp::ILinkLayer *link = nullptr;
// llarp::LogWarn("Need to load our public IP into RC!");
if(inboundLinks.size() == 1)
{
link = inboundLinks[0].get();
}
else
{
if(inboundLinks.size())
{
link = inboundLinks[0].get();
}
else
{
llarp::LogWarn(
"No need to set public ipv4 and port if no external interface "
"binds, turning off public override");
this->publicOverride = false;
link = nullptr;
}
}
if(link && link->GetOurAddressInfo(this->addrInfo))
{
// override ip and port
this->addrInfo.ip = *publicAddr.addr6();
this->addrInfo.port = publicAddr.port();
llarp::LogInfo("Loaded our public ", publicAddr, " override into RC!");
_rc.addrs.push_back(this->addrInfo);
}
}
// set public encryption key
_rc.enckey = llarp::seckey_topublic(encryption);
// set public signing key
_rc.pubkey = llarp::seckey_topublic(identity);
if(ExitEnabled())
{
llarp::nuint32_t a = publicAddr.xtonl();
// TODO: enable this once the network can serialize xi
//_rc.exits.emplace_back(_rc.pubkey, a);
llarp::LogInfo(
"Neato tehl33toh, You are a freaking exit relay. w00t!!!!! your exit "
"is advertised as exiting at ",
a);
}
llarp::LogInfo("Signing rc...");
if(!_rc.Sign(&crypto, identity))
{
llarp::LogError("failed to sign rc");
return false;
}
if(!SaveRC())
{
llarp::LogError("failed to save RC");
return false;
}
llarp::LogInfo("have ", llarp_nodedb_num_loaded(nodedb), " routers");
2018-09-10 13:43:36 +00:00
llarp::LogDebug("starting outbound link");
if(!outboundLink->Start(logic))
{
llarp::LogWarn("outbound link failed to start");
return false;
}
int IBLinksStarted = 0;
2018-06-10 14:05:48 +00:00
// start links
for(const auto &link : inboundLinks)
2018-06-10 14:05:48 +00:00
{
if(link->Start(logic))
{
llarp::LogDebug("Link ", link->Name(), " started");
IBLinksStarted++;
}
else
llarp::LogWarn("Link ", link->Name(), " failed to start");
2018-06-10 14:05:48 +00:00
}
if(IBLinksStarted > 0)
{
// initialize as service node
if(!InitServiceNode())
{
llarp::LogError("Failed to initialize service node");
return false;
}
llarp::RouterID us = pubkey();
llarp::LogInfo("initalized service node: ", us);
}
else
{
// we are a client
// regenerate keys and resign rc before everything else
crypto.identity_keygen(identity);
crypto.encryption_keygen(encryption);
_rc.pubkey = llarp::seckey_topublic(identity);
_rc.enckey = llarp::seckey_topublic(encryption);
if(!_rc.Sign(&crypto, identity))
{
llarp::LogError("failed to regenerate keys and sign RC");
return false;
}
// generate default hidden service
llarp::LogInfo("setting up default network endpoint");
if(!CreateDefaultHiddenService())
{
llarp::LogError("failed to set up default network endpoint");
return false;
}
}
llarp::LogInfo("starting hidden service context...");
if(!hiddenServiceContext.StartAll())
{
llarp::LogError("Failed to start hidden service context");
return false;
}
llarp_dht_context_start(dht, pubkey());
ScheduleTicker(1000);
return true;
2018-06-10 14:05:48 +00:00
}
void
Router::Stop()
2018-11-26 13:29:45 +00:00
{
this->Close();
this->routerProfiling.Save(this->routerProfilesFile.c_str());
2018-11-26 13:29:45 +00:00
}
2018-08-02 23:30:34 +00:00
bool
Router::HasSessionTo(const llarp::RouterID &remote) const
{
return validRouters.find(remote) != validRouters.end();
}
void
Router::ConnectToRandomRouters(int want)
{
int wanted = want;
Router *self = this;
llarp_nodedb_visit_loaded(
self->nodedb, [self, &want](const llarp::RouterContact &other) -> bool {
// check if we really want to
if(!self->ConnectionToRouterAllowed(other.pubkey))
return want > 0;
if(llarp_randint() % 2 == 0
&& !(self->HasSessionTo(other.pubkey)
|| self->HasPendingConnectJob(other.pubkey)))
{
llarp_router_try_connect(self, other, 5);
--want;
}
return want > 0;
});
if(wanted != want)
llarp::LogInfo("connecting to ", abs(want - wanted), " out of ", wanted,
" random routers");
}
bool
Router::InitServiceNode()
2018-11-26 13:29:45 +00:00
{
llarp::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::ReloadConfig(__attribute__((unused)) const llarp_config *conf)
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::InitOutboundLink()
2018-06-18 22:03:50 +00:00
{
if(outboundLink)
return true;
auto link = llarp::utp::NewServer(this);
if(!link->EnsureKeys(transport_keyfile.string().c_str()))
{
llarp::LogError("failed to load ", transport_keyfile);
return false;
}
2018-06-20 17:45:44 +00:00
auto afs = {AF_INET, AF_INET6};
2018-06-20 17:45:44 +00:00
for(auto af : afs)
{
if(link->Configure(netloop, "*", af, 0))
{
outboundLink = std::move(link);
llarp::LogInfo("outbound link ready");
return true;
}
}
return false;
}
2018-06-21 13:33:42 +00:00
bool
Router::CreateDefaultHiddenService()
{
// fallback defaults
static const std::unordered_map< std::string,
std::function< std::string(void) > >
netConfigDefaults = {
{"ifname", llarp::findFreeLokiTunIfName},
{"ifaddr", llarp::findFreePrivateRange},
{"local-dns", []() -> std::string { return "127.0.0.1:53"; }},
{"upstream-dns", []() -> std::string { return "1.1.1.1:53"; }}};
// populate with fallback defaults if values not present
auto itr = netConfigDefaults.begin();
while(itr != netConfigDefaults.end())
{
if(netConfig.count(itr->first) == 0)
{
netConfig.emplace(std::make_pair(itr->first, itr->second()));
}
++itr;
}
// add endpoint
return hiddenServiceContext.AddDefaultEndpoint(netConfig);
}
2018-08-02 23:30:34 +00:00
bool
Router::HasPendingConnectJob(const llarp::RouterID &remote)
{
return pendingEstablishJobs.find(remote) != pendingEstablishJobs.end();
}
2018-01-19 16:51:27 +00:00
bool
Router::LoadHiddenServiceConfig(const char *fname)
2018-07-09 17:32:11 +00:00
{
llarp::LogDebug("opening hidden service config ", fname);
llarp::service::Config conf;
if(!conf.Load(fname))
2018-07-09 17:32:11 +00:00
return false;
for(const auto &config : conf.services)
{
llarp::service::Config::section_t filteredConfig;
mergeHiddenServiceConfig(config.second, filteredConfig.second);
filteredConfig.first = config.first;
if(!hiddenServiceContext.AddEndpoint(filteredConfig))
return false;
}
return true;
2018-07-09 17:32:11 +00:00
}
void
router_iter_config(llarp_config_iterator *iter, const char *section,
const char *key, const char *val)
2018-05-20 17:45:47 +00:00
{
2018-12-04 16:16:43 +00:00
llarp::LogDebug(section, " ", key, "=", val);
Router *self = static_cast< Router * >(iter->user);
2018-08-10 21:34:11 +00:00
int af;
uint16_t proto;
if(StrEq(val, "eth"))
{
#ifdef AF_LINK
2018-05-29 13:40:26 +00:00
af = AF_LINK;
#endif
#ifdef AF_PACKET
2018-05-29 13:40:26 +00:00
af = AF_PACKET;
#endif
proto = LLARP_ETH_PROTO;
}
2018-05-20 17:45:47 +00:00
else
{
// try IPv4 first
2018-05-29 12:15:48 +00:00
af = AF_INET;
proto = std::atoi(val);
2018-05-20 17:45:47 +00:00
}
if(StrEq(section, "bind"))
{
if(!StrEq(key, "*"))
{
2018-09-04 12:55:20 +00:00
auto server = llarp::utp::NewServer(self);
if(!server->EnsureKeys(self->transport_keyfile.string().c_str()))
{
llarp::LogError("failed to ensure keyfile ", self->transport_keyfile);
return;
}
if(server->Configure(self->netloop, key, af, proto))
{
self->AddInboundLink(server);
return;
}
if(af == AF_INET6)
{
// we failed to configure IPv6
// try IPv4
llarp::LogInfo("link ", key,
" failed to configure IPv6, trying IPv4");
af = AF_INET;
if(server->Configure(self->netloop, key, af, proto))
{
self->AddInboundLink(server);
return;
}
2018-06-23 14:55:25 +00:00
}
llarp::LogError("Failed to set up curvecp link");
}
}
else if(StrEq(section, "network"))
{
2018-11-14 18:02:27 +00:00
if(StrEq(key, "profiles"))
{
self->routerProfilesFile = val;
self->routerProfiling.Load(val);
llarp::LogInfo("setting profiles to ", self->routerProfilesFile);
}
else if(StrEq(key, "strict-connect"))
{
if(self->IsServiceNode())
{
llarp::LogError("cannot use strict-connect option as service node");
return;
}
llarp::RouterID snode;
llarp::PubKey pk;
if(pk.FromString(val))
{
if(self->strictConnectPubkeys.insert(pk.data()).second)
llarp::LogInfo("added ", pk, " to strict connect list");
else
llarp::LogWarn("duplicate key for strict connect: ", pk);
}
else if(snode.FromString(val))
{
if(self->strictConnectPubkeys.insert(snode).second)
llarp::LogInfo("added ", snode, " to strict connect list");
else
llarp::LogWarn("duplicate key for strict connect: ", snode);
}
else
llarp::LogError("invalid key for strict-connect: ", val);
}
2018-11-14 18:02:27 +00:00
else
{
2018-11-26 13:29:45 +00:00
self->netConfig.insert(std::make_pair(key, val));
}
}
2018-10-09 12:06:30 +00:00
else if(StrEq(section, "api"))
{
if(StrEq(key, "enabled"))
{
self->enableRPCServer = IsTrueValue(val);
}
if(StrEq(key, "bind"))
{
self->rpcBindAddr = val;
}
if(StrEq(key, "authkey"))
{
// TODO: add pubkey to whitelist
}
}
2018-07-09 17:32:11 +00:00
else if(StrEq(section, "services"))
{
if(self->LoadHiddenServiceConfig(val))
{
llarp::LogInfo("loaded hidden service config for ", key);
}
else
{
llarp::LogWarn("failed to load hidden service config for ", key);
}
}
2018-11-21 17:46:33 +00:00
else if(StrEq(section, "lokid"))
{
if(StrEq(key, "enabled"))
{
self->whitelistRouters = IsTrueValue(val);
}
if(StrEq(key, "jsonrpc"))
{
self->lokidRPCAddr = val;
}
}
2018-11-11 13:14:19 +00:00
else if(StrEq(section, "dns"))
{
if(StrEq(key, "upstream"))
{
llarp::LogInfo("add upstream resolver ", val);
2018-11-26 13:29:45 +00:00
self->netConfig.emplace(std::make_pair("upstream-dns", val));
2018-11-11 13:14:19 +00:00
}
if(StrEq(key, "bind"))
{
llarp::LogInfo("set local dns to ", val);
2018-11-26 13:29:45 +00:00
self->netConfig.emplace(std::make_pair("local-dns", val));
2018-11-11 13:14:19 +00:00
}
}
else if(StrEq(section, "connect")
|| (StrEq(section, "bootstrap") && StrEq(key, "add-node")))
{
self->bootstrapRCList.emplace_back();
auto &rc = self->bootstrapRCList.back();
if(rc.Read(val) && rc.Verify(&self->crypto))
{
llarp::LogInfo("Added bootstrap node ", RouterID(rc.pubkey.data()));
}
else
{
llarp::LogError("malformed rc file: ", val);
self->bootstrapRCList.pop_back();
}
2018-04-05 14:43:16 +00:00
}
2018-05-23 20:37:43 +00:00
else if(StrEq(section, "router"))
{
if(StrEq(key, "nickname"))
{
self->_rc.SetNick(val);
2018-08-30 18:48:43 +00:00
// set logger name here
_glog.nodeName = self->rc().Nick();
}
2018-06-10 14:05:48 +00:00
if(StrEq(key, "encryption-privkey"))
{
self->encryption_keyfile = val;
}
2018-05-23 20:37:43 +00:00
if(StrEq(key, "contact-file"))
{
self->our_rc_file = val;
}
if(StrEq(key, "transport-privkey"))
{
self->transport_keyfile = val;
}
if(StrEq(key, "ident-privkey"))
{
self->ident_keyfile = val;
}
if(StrEq(key, "public-address"))
{
llarp::LogInfo("public ip ", val, " size ", strlen(val));
2018-06-29 12:15:15 +00:00
if(strlen(val) < 17)
{
// assume IPv4
// inet_pton(AF_INET, val, &self->ip4addr.sin_addr);
2018-06-29 12:15:15 +00:00
// struct sockaddr dest;
// sockaddr *dest = (sockaddr *)&self->ip4addr;
llarp::Addr a(val);
llarp::LogInfo("setting public ipv4 ", a);
2018-06-29 12:15:15 +00:00
self->addrInfo.ip = *a.addr6();
2018-06-28 11:59:50 +00:00
self->publicOverride = true;
}
2018-06-29 12:15:15 +00:00
// llarp::Addr a(val);
}
if(StrEq(key, "public-port"))
{
llarp::LogInfo("Setting public port ", val);
2018-10-09 01:38:25 +00:00
int p = atoi(val);
// Not needed to flip upside-down - this is done in llarp::Addr(const
// AddressInfo&)
2018-10-09 03:33:56 +00:00
self->ip4addr.sin_port = p;
self->addrInfo.port = p;
2018-06-29 12:15:15 +00:00
self->publicOverride = true;
}
2018-05-23 20:37:43 +00:00
}
2018-11-14 18:02:27 +00:00
} // namespace llarp
2018-02-01 13:21:00 +00:00
} // namespace llarp