mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-17 15:25:35 +00:00
550 lines
11 KiB
C++
550 lines
11 KiB
C++
#include <llarp.hpp>
|
|
#include <constants/version.hpp>
|
|
|
|
#include <config/config.hpp>
|
|
#include <crypto/crypto_libsodium.hpp>
|
|
#include <crypto/crypto_noop.hpp>
|
|
#include <dht/context.hpp>
|
|
#include <ev/ev.hpp>
|
|
#include <ev/vpnio.hpp>
|
|
#include <nodedb.hpp>
|
|
#include <router/router.hpp>
|
|
#include <service/context.hpp>
|
|
#include <util/logging/logger.h>
|
|
|
|
#include <cxxopts.hpp>
|
|
#include <csignal>
|
|
|
|
#if(__FreeBSD__) || (__OpenBSD__) || (__NetBSD__)
|
|
#include <pthread_np.h>
|
|
#endif
|
|
|
|
namespace llarp
|
|
{
|
|
bool
|
|
Context::CallSafe(std::function< void(void) > f)
|
|
{
|
|
return logic && LogicCall(logic, f);
|
|
}
|
|
|
|
bool
|
|
Context::Configure()
|
|
{
|
|
// llarp::LogInfo("loading config at ", configfile);
|
|
if(configfile.size())
|
|
{
|
|
if(!config->Load(configfile.c_str()))
|
|
{
|
|
config.release();
|
|
llarp::LogError("failed to load config file ", configfile);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// System config
|
|
if(!config->system.pidfile.empty())
|
|
{
|
|
SetPIDFile(config->system.pidfile);
|
|
}
|
|
auto threads = config->router.workerThreads();
|
|
if(threads <= 0)
|
|
threads = 1;
|
|
worker = std::make_shared< llarp::thread::ThreadPool >(threads, 1024,
|
|
"llarp-worker");
|
|
auto jobQueueSize = config->router.jobQueueSize();
|
|
if(jobQueueSize < 1024)
|
|
jobQueueSize = 1024;
|
|
logic = std::make_shared< Logic >(jobQueueSize);
|
|
|
|
nodedb_dir = config->netdb.nodedbDir();
|
|
|
|
return true;
|
|
}
|
|
|
|
void
|
|
Context::SetPIDFile(const std::string &fname)
|
|
{
|
|
pidfile = fname;
|
|
}
|
|
|
|
bool
|
|
Context::IsUp() const
|
|
{
|
|
return router && router->IsRunning();
|
|
}
|
|
|
|
bool
|
|
Context::LooksAlive() const
|
|
{
|
|
return router && router->LooksAlive();
|
|
}
|
|
|
|
int
|
|
Context::LoadDatabase()
|
|
{
|
|
if(!llarp_nodedb::ensure_dir(nodedb_dir.c_str()))
|
|
{
|
|
llarp::LogError("nodedb_dir is incorrect");
|
|
return 0;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
int
|
|
Context::Setup()
|
|
{
|
|
llarp::LogInfo(llarp::VERSION_FULL, " ", llarp::RELEASE_MOTTO);
|
|
llarp::LogInfo("starting up");
|
|
mainloop = llarp_make_ev_loop();
|
|
logic->set_event_loop(mainloop.get());
|
|
|
|
mainloop->set_logic(logic);
|
|
|
|
crypto = std::make_unique< sodium::CryptoLibSodium >();
|
|
cryptoManager = std::make_unique< CryptoManager >(crypto.get());
|
|
|
|
router = std::make_unique< Router >(worker, mainloop, logic);
|
|
|
|
nodedb = std::make_unique< llarp_nodedb >(router->diskworker(), nodedb_dir);
|
|
|
|
if(!router->Configure(config.get(), nodedb.get()))
|
|
{
|
|
llarp::LogError("Failed to configure router");
|
|
return 1;
|
|
}
|
|
|
|
// must be done after router is made so we can use its disk io worker
|
|
// must also be done after configure so that netid is properly set if it
|
|
// is provided by config
|
|
if(!this->LoadDatabase())
|
|
return 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
Context::Run(llarp_main_runtime_opts opts)
|
|
{
|
|
if(router == nullptr)
|
|
{
|
|
// we are not set up so we should die
|
|
llarp::LogError("cannot run non configured context");
|
|
return 1;
|
|
}
|
|
if(!WritePIDFile())
|
|
return 1;
|
|
// run
|
|
if(!router->StartJsonRpc())
|
|
return 1;
|
|
|
|
if(!opts.background)
|
|
{
|
|
if(!router->Run())
|
|
return 2;
|
|
}
|
|
|
|
// run net io thread
|
|
llarp::LogInfo("running mainloop");
|
|
llarp_ev_loop_run_single_process(mainloop, logic);
|
|
if(closeWaiter)
|
|
{
|
|
// inform promise if called by CloseAsync
|
|
closeWaiter->set_value();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
Context::CloseAsync()
|
|
{
|
|
/// already closing
|
|
if(closeWaiter)
|
|
return;
|
|
if(CallSafe(std::bind(&Context::HandleSignal, this, SIGTERM)))
|
|
closeWaiter = std::make_unique< std::promise< void > >();
|
|
}
|
|
|
|
void
|
|
Context::Wait()
|
|
{
|
|
if(closeWaiter)
|
|
{
|
|
closeWaiter->get_future().wait();
|
|
closeWaiter.reset();
|
|
}
|
|
}
|
|
|
|
bool
|
|
Context::WritePIDFile() const
|
|
{
|
|
if(pidfile.size())
|
|
{
|
|
std::ofstream f(pidfile);
|
|
f << std::to_string(getpid());
|
|
return f.good();
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void
|
|
Context::RemovePIDFile() const
|
|
{
|
|
if(pidfile.size())
|
|
{
|
|
fs::path f = pidfile;
|
|
std::error_code ex;
|
|
if(fs::exists(f, ex))
|
|
{
|
|
if(!ex)
|
|
fs::remove(f);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
Context::HandleSignal(int sig)
|
|
{
|
|
if(sig == SIGINT || sig == SIGTERM)
|
|
{
|
|
SigINT();
|
|
}
|
|
// TODO(despair): implement hot-reloading config on NT
|
|
#ifndef _WIN32
|
|
if(sig == SIGHUP)
|
|
{
|
|
llarp::LogInfo("SIGHUP");
|
|
if(router)
|
|
{
|
|
router->hiddenServiceContext().ForEachService(
|
|
[](const std::string &name,
|
|
const llarp::service::Endpoint_ptr &ep) -> bool {
|
|
ep->ResetInternalState();
|
|
llarp::LogInfo("Reset internal state for ", name);
|
|
return true;
|
|
});
|
|
router->PumpLL();
|
|
Config newconfig;
|
|
if(!newconfig.Load(configfile.c_str()))
|
|
{
|
|
llarp::LogError("failed to load config file ", configfile);
|
|
return;
|
|
}
|
|
// validate config
|
|
if(!router->ValidateConfig(&newconfig))
|
|
{
|
|
llarp::LogWarn("new configuration is invalid");
|
|
return;
|
|
}
|
|
// reconfigure
|
|
if(!router->Reconfigure(&newconfig))
|
|
{
|
|
llarp::LogError("Failed to reconfigure so we will stop.");
|
|
router->Stop();
|
|
return;
|
|
}
|
|
llarp::LogInfo("router reconfigured");
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void
|
|
Context::SigINT()
|
|
{
|
|
if(router)
|
|
{
|
|
/// async stop router on sigint
|
|
router->Stop();
|
|
}
|
|
else
|
|
{
|
|
if(logic)
|
|
logic->stop();
|
|
llarp_ev_loop_stop(mainloop);
|
|
Close();
|
|
}
|
|
}
|
|
|
|
void
|
|
Context::Close()
|
|
{
|
|
llarp::LogDebug("stop workers");
|
|
if(worker)
|
|
worker->stop();
|
|
|
|
llarp::LogDebug("free config");
|
|
config.release();
|
|
|
|
llarp::LogDebug("free workers");
|
|
worker.reset();
|
|
|
|
llarp::LogDebug("free nodedb");
|
|
nodedb.release();
|
|
|
|
llarp::LogDebug("free router");
|
|
router.release();
|
|
|
|
llarp::LogDebug("free logic");
|
|
logic.reset();
|
|
|
|
RemovePIDFile();
|
|
}
|
|
|
|
bool
|
|
Context::LoadConfig(const std::string &fname)
|
|
{
|
|
config = std::make_unique< Config >();
|
|
configfile = fname;
|
|
return Configure();
|
|
}
|
|
} // namespace llarp
|
|
|
|
struct llarp_main
|
|
{
|
|
llarp_main(llarp_config *conf);
|
|
~llarp_main() = default;
|
|
std::unique_ptr< llarp::Context > ctx;
|
|
};
|
|
|
|
struct llarp_config
|
|
{
|
|
llarp::Config impl;
|
|
llarp_config() = default;
|
|
|
|
llarp_config(const llarp_config *other) : impl(other->impl)
|
|
{
|
|
}
|
|
};
|
|
|
|
extern "C"
|
|
{
|
|
size_t
|
|
llarp_main_size()
|
|
{
|
|
return sizeof(llarp_main);
|
|
}
|
|
|
|
size_t
|
|
llarp_config_size()
|
|
{
|
|
return sizeof(llarp_config);
|
|
}
|
|
|
|
struct llarp_config *
|
|
llarp_default_config()
|
|
{
|
|
llarp_config *conf = new llarp_config();
|
|
#ifdef ANDROID
|
|
// put andrid config overrides here
|
|
#endif
|
|
#ifdef IOS
|
|
// put IOS config overrides here
|
|
#endif
|
|
return conf;
|
|
}
|
|
|
|
void
|
|
llarp_config_free(struct llarp_config *conf)
|
|
{
|
|
if(conf)
|
|
delete conf;
|
|
}
|
|
|
|
struct llarp_main *
|
|
llarp_main_init_from_config(struct llarp_config *conf)
|
|
{
|
|
if(conf == nullptr)
|
|
return nullptr;
|
|
llarp_main *m = new llarp_main(conf);
|
|
if(m->ctx->Configure())
|
|
return m;
|
|
delete m;
|
|
return nullptr;
|
|
}
|
|
|
|
bool
|
|
llarp_config_read_file(struct llarp_config *conf, const char *fname)
|
|
{
|
|
if(conf == nullptr)
|
|
return false;
|
|
return conf->impl.Load(fname);
|
|
}
|
|
|
|
bool
|
|
llarp_config_load_file(const char *fname, struct llarp_config **conf)
|
|
{
|
|
llarp_config *c = new llarp_config();
|
|
if(c->impl.Load(fname))
|
|
{
|
|
*conf = c;
|
|
return true;
|
|
}
|
|
delete c;
|
|
*conf = nullptr;
|
|
return false;
|
|
}
|
|
|
|
void
|
|
llarp_main_signal(struct llarp_main *ptr, int sig)
|
|
{
|
|
LogicCall(ptr->ctx->logic,
|
|
std::bind(&llarp::Context::HandleSignal, ptr->ctx.get(), sig));
|
|
}
|
|
|
|
int
|
|
llarp_main_setup(struct llarp_main *ptr)
|
|
{
|
|
return ptr->ctx->Setup();
|
|
}
|
|
|
|
int
|
|
llarp_main_run(struct llarp_main *ptr, struct llarp_main_runtime_opts opts)
|
|
{
|
|
return ptr->ctx->Run(opts);
|
|
}
|
|
|
|
const char *
|
|
llarp_version()
|
|
{
|
|
return llarp::VERSION_FULL;
|
|
}
|
|
|
|
ssize_t
|
|
llarp_vpn_io_readpkt(struct llarp_vpn_pkt_reader *r, unsigned char *dst,
|
|
size_t dstlen)
|
|
{
|
|
if(r == nullptr)
|
|
return -1;
|
|
if(not r->queue.enabled())
|
|
return -1;
|
|
auto pkt = r->queue.popFront();
|
|
ManagedBuffer mbuf = pkt.ConstBuffer();
|
|
const llarp_buffer_t &buf = mbuf;
|
|
if(buf.sz > dstlen || buf.sz == 0)
|
|
return -1;
|
|
std::copy_n(buf.base, buf.sz, dst);
|
|
return buf.sz;
|
|
}
|
|
|
|
bool
|
|
llarp_vpn_io_writepkt(struct llarp_vpn_pkt_writer *w, unsigned char *pktbuf,
|
|
size_t pktlen)
|
|
{
|
|
if(pktlen == 0 || pktbuf == nullptr)
|
|
return false;
|
|
if(w == nullptr)
|
|
return false;
|
|
llarp_vpn_pkt_queue::Packet_t pkt;
|
|
llarp_buffer_t buf(pktbuf, pktlen);
|
|
if(not pkt.Load(buf))
|
|
return false;
|
|
return w->queue.pushBack(std::move(pkt))
|
|
== llarp::thread::QueueReturn::Success;
|
|
}
|
|
|
|
bool
|
|
llarp_main_inject_vpn_by_name(struct llarp_main *ptr, const char *name,
|
|
struct llarp_vpn_io *io,
|
|
struct llarp_vpn_ifaddr_info info)
|
|
{
|
|
if(name == nullptr || io == nullptr)
|
|
return false;
|
|
if(ptr == nullptr || ptr->ctx == nullptr || ptr->ctx->router == nullptr)
|
|
return false;
|
|
auto ep = ptr->ctx->router->hiddenServiceContext().GetEndpointByName(name);
|
|
return ep && ep->InjectVPN(io, info);
|
|
}
|
|
|
|
void
|
|
llarp_vpn_io_close_async(struct llarp_vpn_io *io)
|
|
{
|
|
if(io == nullptr || io->impl == nullptr)
|
|
return;
|
|
static_cast< llarp_vpn_io_impl * >(io->impl)->AsyncClose();
|
|
}
|
|
|
|
bool
|
|
llarp_vpn_io_init(struct llarp_main *ptr, struct llarp_vpn_io *io)
|
|
{
|
|
if(io == nullptr || ptr == nullptr)
|
|
return false;
|
|
llarp_vpn_io_impl *impl = new llarp_vpn_io_impl(ptr, io);
|
|
io->impl = impl;
|
|
return true;
|
|
}
|
|
|
|
struct llarp_vpn_pkt_writer *
|
|
llarp_vpn_io_packet_writer(struct llarp_vpn_io *io)
|
|
{
|
|
if(io == nullptr || io->impl == nullptr)
|
|
return nullptr;
|
|
llarp_vpn_io_impl *vpn = static_cast< llarp_vpn_io_impl * >(io->impl);
|
|
return &vpn->writer;
|
|
}
|
|
|
|
struct llarp_vpn_pkt_reader *
|
|
llarp_vpn_io_packet_reader(struct llarp_vpn_io *io)
|
|
{
|
|
if(io == nullptr || io->impl == nullptr)
|
|
return nullptr;
|
|
llarp_vpn_io_impl *vpn = static_cast< llarp_vpn_io_impl * >(io->impl);
|
|
return &vpn->reader;
|
|
}
|
|
|
|
void
|
|
llarp_main_free(struct llarp_main *ptr)
|
|
{
|
|
delete ptr;
|
|
}
|
|
|
|
const char *
|
|
llarp_main_get_default_endpoint_name(struct llarp_main *)
|
|
{
|
|
return "default";
|
|
}
|
|
|
|
void
|
|
llarp_main_stop(struct llarp_main *ptr)
|
|
{
|
|
if(ptr == nullptr)
|
|
return;
|
|
ptr->ctx->CloseAsync();
|
|
ptr->ctx->Wait();
|
|
}
|
|
|
|
bool
|
|
llarp_main_configure(struct llarp_main *ptr, struct llarp_config *conf)
|
|
{
|
|
if(ptr == nullptr || conf == nullptr)
|
|
return false;
|
|
// give new config
|
|
ptr->ctx->config.reset(new llarp::Config(conf->impl));
|
|
return ptr->ctx->Configure();
|
|
}
|
|
|
|
bool
|
|
llarp_main_is_running(struct llarp_main *ptr)
|
|
{
|
|
return ptr && ptr->ctx->router && ptr->ctx->router->IsRunning();
|
|
}
|
|
}
|
|
|
|
llarp_main::llarp_main(llarp_config *conf)
|
|
|
|
: ctx(new llarp::Context())
|
|
{
|
|
ctx->config.reset(new llarp::Config(conf->impl));
|
|
}
|
|
|
|
namespace llarp
|
|
{
|
|
Context *
|
|
Context::Get(llarp_main *m)
|
|
{
|
|
if(m == nullptr || m->ctx == nullptr)
|
|
return nullptr;
|
|
return m->ctx.get();
|
|
}
|
|
} // namespace llarp
|