lokinet/daemon/main.cpp

312 lines
8.8 KiB
C++
Raw Normal View History

#include <config/config.hpp> // for ensure_config
#include <constants/version.hpp>
#include <llarp.hpp>
#include <util/lokinet_init.h>
2019-01-11 01:59:44 +00:00
#include <util/fs.hpp>
2019-09-01 12:10:49 +00:00
#include <util/logging/logger.hpp>
2019-09-16 19:40:31 +00:00
#include <util/logging/ostream_logger.hpp>
#include <util/str.hpp>
2020-06-29 20:09:59 +00:00
#include <util/thread/logic.hpp>
2019-01-11 01:59:44 +00:00
#include <csignal>
#include <cxxopts.hpp>
2018-07-20 04:50:28 +00:00
#include <string>
#include <iostream>
#include <future>
2018-05-20 16:15:16 +00:00
2020-05-21 14:18:23 +00:00
#ifdef USE_JEMALLOC
#include <new>
#include <jemalloc/jemalloc.h>
void*
operator new(std::size_t sz)
{
void* ptr = malloc(sz);
if (ptr)
return ptr;
else
throw std::bad_alloc{};
}
void
operator delete(void* ptr) noexcept
{
free(ptr);
}
void
operator delete(void* ptr, size_t) noexcept
{
free(ptr);
}
#endif
#ifdef _WIN32
#define wmin(x, y) (((x) < (y)) ? (x) : (y))
#define MIN wmin
2019-08-02 03:25:48 +00:00
extern "C" LONG FAR PASCAL
win32_signal_handler(EXCEPTION_POINTERS*);
#endif
2020-06-29 19:55:59 +00:00
std::shared_ptr<llarp::Context> ctx;
std::promise<int> exit_code;
2018-04-30 16:14:20 +00:00
void
handle_signal(int sig)
2018-05-18 17:50:21 +00:00
{
2020-06-29 20:09:59 +00:00
if (ctx)
LogicCall(ctx->logic, std::bind(&llarp::Context::HandleSignal, ctx.get(), sig));
else
std::cerr << "Received signal " << sig << ", but have no context yet. Ignoring!" << std::endl;
2018-05-18 17:50:21 +00:00
}
#ifdef _WIN32
int
startWinsock()
{
WSADATA wsockd;
int err;
err = ::WSAStartup(MAKEWORD(2, 2), &wsockd);
if (err)
{
perror("Failed to start Windows Sockets");
return err;
}
::CreateMutex(nullptr, FALSE, "lokinet_win32_daemon");
return 0;
}
2018-12-14 12:50:45 +00:00
extern "C" BOOL FAR PASCAL
handle_signal_win32(DWORD fdwCtrlType)
{
UNREFERENCED_PARAMETER(fdwCtrlType);
handle_signal(SIGINT);
2018-12-23 13:29:11 +00:00
return TRUE; // probably unreachable
2018-12-14 12:50:45 +00:00
}
#endif
/// this sets up, configures and runs the main context
static void
run_main_context(const fs::path confFile, const llarp::RuntimeOptions opts)
{
2020-06-29 19:55:59 +00:00
try
{
2020-06-29 19:55:59 +00:00
// this is important, can downgrade from Info though
llarp::LogDebug("Running from: ", fs::current_path().string());
llarp::LogInfo("Using config file: ", confFile);
llarp::Config conf;
conf.Load(confFile, opts.isRouter, confFile.parent_path());
ctx = std::shared_ptr<llarp::Context>();
ctx->Configure(opts, {});
2020-06-29 19:55:59 +00:00
signal(SIGINT, handle_signal);
signal(SIGTERM, handle_signal);
#ifndef _WIN32
signal(SIGHUP, handle_signal);
#endif
2020-06-29 19:55:59 +00:00
ctx->Setup(opts);
2020-06-29 19:55:59 +00:00
llarp::util::SetThreadName("llarp-mainloop");
2020-06-29 19:55:59 +00:00
auto result = ctx->Run(opts);
exit_code.set_value(result);
}
catch (std::exception& e)
{
llarp::LogError("Fatal: caught exception while running: ", e.what());
exit_code.set_exception(std::current_exception());
2020-06-29 19:55:59 +00:00
}
catch (...)
{
llarp::LogError("Fatal: caught non-standard exception while running");
exit_code.set_exception(std::current_exception());
}
}
int
main(int argc, char* argv[])
{
auto result = Lokinet_INIT();
if (result)
{
return result;
}
2020-06-29 19:55:59 +00:00
llarp::RuntimeOptions opts;
#ifdef _WIN32
if (startWinsock())
return -1;
2018-12-14 12:50:45 +00:00
SetConsoleCtrlHandler(handle_signal_win32, TRUE);
2019-08-02 03:25:48 +00:00
// SetUnhandledExceptionFilter(win32_signal_handler);
#endif
cxxopts::Options options(
"lokinet",
"LokiNET is a free, open source, private, "
"decentralized, \"market based sybil resistant\" "
"and IP based onion routing network");
options.add_options()("v,verbose", "Verbose", cxxopts::value<bool>())(
"h,help", "help", cxxopts::value<bool>())("version", "version", cxxopts::value<bool>())(
"g,generate", "generate client config", cxxopts::value<bool>())(
"r,relay", "run as relay instead of client", cxxopts::value<bool>())(
"f,force", "overwrite", cxxopts::value<bool>())(
"c,colour", "colour output", cxxopts::value<bool>()->default_value("true"))(
"b,background",
"background mode (start, but do not connect to the network)",
cxxopts::value<bool>())(
"config", "path to configuration file", cxxopts::value<std::string>());
2019-04-21 21:48:28 +00:00
options.parse_positional("config");
2019-10-08 14:52:01 +00:00
bool genconfigOnly = false;
bool overwrite = false;
2020-06-29 19:55:59 +00:00
fs::path configFile;
try
{
auto result = options.parse(argc, argv);
if (result.count("verbose") > 0)
{
SetLogLevel(llarp::eLogDebug);
llarp::LogDebug("debug logging activated");
}
if (!result["colour"].as<bool>())
2019-09-16 19:40:31 +00:00
{
llarp::LogContext::Instance().logStream =
std::make_unique<llarp::OStreamLogStream>(false, std::cerr);
2019-09-16 19:40:31 +00:00
}
if (result.count("help"))
{
2019-04-23 20:52:13 +00:00
std::cout << options.help() << std::endl;
return 0;
}
if (result.count("version"))
{
2020-06-29 19:55:59 +00:00
std::cout << llarp::VERSION_FULL << std::endl;
return 0;
}
if (result.count("generate") > 0)
{
genconfigOnly = true;
}
if (result.count("background") > 0)
2019-05-28 00:19:25 +00:00
{
opts.background = true;
2019-05-28 00:19:25 +00:00
}
2020-06-29 19:55:59 +00:00
if (result.count("router") > 0)
{
2020-06-29 19:55:59 +00:00
opts.isRouter = true;
}
if (result.count("force") > 0)
{
overwrite = true;
}
if (result.count("config") > 0)
{
auto arg = result["config"].as<std::string>();
if (!arg.empty())
{
2020-06-29 19:55:59 +00:00
configFile = arg;
}
}
}
catch (const cxxopts::option_not_exists_exception& ex)
{
std::cerr << ex.what();
2019-04-23 20:52:13 +00:00
std::cout << options.help() << std::endl;
return 1;
}
2020-06-29 19:55:59 +00:00
if (!configFile.empty())
{
// when we have an explicit filepath
2020-06-29 19:55:59 +00:00
fs::path basedir = configFile.parent_path();
2019-01-29 11:23:40 +00:00
if (genconfigOnly)
{
2020-06-29 19:55:59 +00:00
llarp::ensureConfig(basedir, configFile, overwrite, opts.isRouter);
}
else
{
std::error_code ec;
2020-06-29 19:55:59 +00:00
if (!fs::exists(configFile, ec))
2019-01-29 11:23:40 +00:00
{
2020-06-29 19:55:59 +00:00
llarp::LogError("Config file not found ", configFile);
return 1;
2019-01-29 11:23:40 +00:00
}
if (ec)
throw std::runtime_error(llarp::stringify("filesystem error: ", ec));
}
}
else
{
llarp::ensureConfig(
2020-06-29 19:55:59 +00:00
llarp::GetDefaultDataDir(), llarp::GetDefaultConfigPath(), overwrite, opts.isRouter);
configFile = llarp::GetDefaultConfigPath();
}
if (genconfigOnly)
{
return 0;
}
2018-07-13 13:36:51 +00:00
2020-06-29 19:55:59 +00:00
std::thread main_thread{std::bind(&run_main_context, configFile, opts)};
auto ftr = exit_code.get_future();
do
2018-05-27 18:03:10 +00:00
{
// do periodic non lokinet related tasks here
2020-06-29 19:55:59 +00:00
if (ctx and ctx->IsUp() and not ctx->LooksAlive())
{
2020-06-29 19:55:59 +00:00
for (const auto& wtf : {"you have been visited by the mascott of the "
"deadlocked router.",
"⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⣀⣴⣾⣿⣿⣿⣿⣿⣿⣿⣿⣿⣷⣄⠄⠄⠄⠄",
"⠄⠄⠄⠄⠄⢀⣀⣀⡀⠄⠄⠄⡠⢲⣾⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣷⡀⠄⠄",
"⠄⠄⠄⠔⣈⣀⠄⢔⡒⠳⡴⠊⠄⠸⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⡿⠿⣿⣿⣧⠄⠄",
"⠄⢜⡴⢑⠖⠊⢐⣤⠞⣩⡇⠄⠄⠄⠙⢿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣿⣆⠄⠝⠛⠋⠐",
"⢸⠏⣷⠈⠄⣱⠃⠄⢠⠃⠐⡀⠄⠄⠄⠄⠙⠻⢿⣿⣿⣿⣿⣿⣿⣿⡿⠛⠸⠄⠄⠄⠄",
"⠈⣅⠞⢁⣿⢸⠘⡄⡆⠄⠄⠈⠢⡀⠄⠄⠄⠄⠄⠄⠉⠙⠛⠛⠛⠉⠉⡀⠄⠡⢀⠄⣀",
"⠄⠙⡎⣹⢸⠄⠆⢘⠁⠄⠄⠄⢸⠈⠢⢄⡀⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⠃⠄⠄⠄⠄⠄",
"⠄⠄⠑⢿⠈⢆⠘⢼⠄⠄⠄⠄⠸⢐⢾⠄⡘⡏⠲⠆⠠⣤⢤⢤⡤⠄⣖⡇⠄⠄⠄⠄⠄",
"⣴⣶⣿⣿⣣⣈⣢⣸⠄⠄⠄⠄⡾⣷⣾⣮⣤⡏⠁⠘⠊⢠⣷⣾⡛⡟⠈⠄⠄⠄⠄⠄⠄",
"⣿⣿⣿⣿⣿⠉⠒⢽⠄⠄⠄⠄⡇⣿⣟⣿⡇⠄⠄⠄⠄⢸⣻⡿⡇⡇⠄⠄⠄⠄⠄⠄⠄",
"⠻⣿⣿⣿⣿⣄⠰⢼⠄⠄⠄⡄⠁⢻⣍⣯⠃⠄⠄⠄⠄⠈⢿⣻⠃⠈⡆⡄⠄⠄⠄⠄⠄",
"⠄⠙⠿⠿⠛⣿⣶⣤⡇⠄⠄⢣⠄⠄⠈⠄⢠⠂⠄⠁⠄⡀⠄⠄⣀⠔⢁⠃⠄⠄⠄⠄⠄",
"⠄⠄⠄⠄⠄⣿⣿⣿⣿⣾⠢⣖⣶⣦⣤⣤⣬⣤⣤⣤⣴⣶⣶⡏⠠⢃⠌⠄⠄⠄⠄⠄⠄",
"⠄⠄⠄⠄⠄⠿⠿⠟⠛⡹⠉⠛⠛⠿⠿⣿⣿⣿⣿⣿⡿⠂⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄⠄",
"⠠⠤⠤⠄⠄⣀⠄⠄⠄⠑⠠⣤⣀⣀⣀⡘⣿⠿⠙⠻⡍⢀⡈⠂⠄⠄⠄⠄⠄⠄⠄⠄⠄",
"⠄⠄⠄⠄⠄⠄⠑⠠⣠⣴⣾⣿⣿⣿⣿⣿⣿⣇⠉⠄⠻⣿⣷⣄⡀⠄⠄⠄⠄⠄⠄⠄⠄",
"file a bug report now or be cursed with this "
"annoying image in your syslog for all time."})
{
2020-06-29 19:55:59 +00:00
LogError(wtf);
llarp::LogContext::Instance().ImmediateFlush();
}
2020-06-29 19:55:59 +00:00
std::abort();
}
} while (ftr.wait_for(std::chrono::seconds(1)) != std::future_status::ready);
main_thread.join();
const auto code = ftr.get();
llarp::LogContext::Instance().ImmediateFlush();
#ifdef _WIN32
::WSACleanup();
#endif
if (ctx)
{
2020-06-29 19:55:59 +00:00
ctx.reset();
}
2018-05-27 19:13:25 +00:00
return code;
2017-09-28 17:02:05 +00:00
}