lokinet/daemon/main.c
2018-01-29 09:27:24 -05:00

72 lines
2.0 KiB
C

#include <llarp.h>
#include <stdio.h>
#include <string.h>
struct llarp_main {
struct llarp_router *router;
struct llarp_threadpool *tp;
struct llarp_config *config;
struct llarp_ev_loop *mainloop;
};
void iter_main_config(struct llarp_config_iterator *itr, const char *section,
const char *key, const char *val) {
struct llarp_main *m = (struct llarp_main *)itr->user;
if (!strcmp(section, "threadpool")) {
if (!strcmp(key, "workers")) {
int workers = atoi(val);
if (!m->tp && workers > 0) {
m->tp = llarp_init_threadpool(workers);
}
}
}
}
int shutdown_llarp(struct llarp_main *m) {
printf("Shutting down .");
llarp_stop_router(m->router);
printf(".");
llarp_ev_loop_stop(m->mainloop);
printf(".");
llarp_threadpool_join(m->tp);
printf(".");
llarp_free_router(&m->router);
printf(".");
llarp_free_config(&m->config);
printf(".");
llarp_ev_loop_free(&m->mainloop);
printf(".");
llarp_free_threadpool(&m->tp);
printf(".\n");
}
int main(int argc, char *argv[]) {
struct llarp_main llarp = {NULL, NULL, NULL, NULL};
const char *conffname = "daemon.ini";
if (argc > 1)
conffname = argv[1];
llarp_mem_jemalloc();
llarp_new_config(&llarp.config);
llarp_ev_loop_alloc(&llarp.mainloop);
printf("%s loaded\n", LLARP_VERSION);
if (!llarp_load_config(llarp.config, conffname)) {
printf("Loaded config %s\n", conffname);
struct llarp_config_iterator iter;
iter.user = &llarp;
iter.visit = iter_main_config;
llarp_config_iter(llarp.config, &iter);
if (!llarp.tp)
llarp.tp = llarp_init_threadpool(2);
llarp.router = llarp_init_router(llarp.tp);
if (!llarp_configure_router(llarp.router, llarp.config)) {
printf("Running\n");
llarp_run_router(llarp.router, llarp.mainloop);
llarp_ev_loop_run(llarp.mainloop);
} else
printf("Failed to configure router\n");
} else
printf("Failed to load config %s\n", conffname);
return shutdown_llarp(&llarp);
}