lokinet/daemon/rcutil.cpp

#include <llarp.h>
#include <pthread.h>
#include <signal.h>
#include <stdio.h>
#include <string.h>
#include <experimental/filesystem>

namespace fs = std::experimental::filesystem;

static void
progress()
{
  printf(".");
  fflush(stdout);
}

struct llarp_main
{
  struct llarp_alloc mem;
  struct llarp_crypto crypto;
  struct llarp_router *router     = nullptr;
  struct llarp_threadpool *worker = nullptr;
  struct llarp_threadpool *thread = nullptr;
  struct llarp_logic *logic       = nullptr;
  struct llarp_config *config     = nullptr;
  struct llarp_nodedb *nodedb     = nullptr;
  struct llarp_ev_loop *mainloop  = nullptr;
  char nodedb_dir[256];
  int exitcode;

  int
  shutdown()
  {
    printf("Shutting down ");

    progress();
    if(mainloop)
      llarp_ev_loop_stop(mainloop);

    progress();
    if(worker)
      llarp_threadpool_stop(worker);

    progress();

    if(worker)
      llarp_threadpool_join(worker);

    progress();
    if(logic)
      llarp_logic_stop(logic);

    progress();

    if(router)
      llarp_stop_router(router);

    progress();
    llarp_free_router(&router);

    progress();
    llarp_free_config(&config);

    progress();
    llarp_ev_loop_free(&mainloop);

    progress();
    llarp_free_threadpool(&worker);

    progress();

    llarp_free_logic(&logic);
    progress();

    printf("\n");
    fflush(stdout);
    return exitcode;
  }
};

void
iter_main_config(struct llarp_config_iterator *itr, const char *section,
                 const char *key, const char *val)
{
  llarp_main *m = static_cast< llarp_main * >(itr->user);

  if(!strcmp(section, "router"))
  {
    if(!strcmp(key, "threads"))
    {
      int workers = atoi(val);
      if(workers > 0 && m->worker == nullptr)
      {
        m->worker = llarp_init_threadpool(workers, "llarp-worker");
      }
    }
  }
  if(!strcmp(section, "netdb"))
  {
    if(!strcmp(key, "dir"))
    {
      strncpy(m->nodedb_dir, val, sizeof(m->nodedb_dir));
    }
  }
}

llarp_main *sllarp = nullptr;

void
run_net(void *user)
{
  llarp_ev_loop_run(static_cast< llarp_ev_loop * >(user));
}

void
handle_signal(int sig)
{
  printf("\ninterrupted\n");
  llarp_ev_loop_stop(sllarp->mainloop);
  llarp_logic_stop(sllarp->logic);
}

#include <getopt.h>
#include <llarp/router_contact.h>
#include <llarp/time.h>
#include <fstream>

int
main(int argc, char *argv[])
{
  // --generate-blank /path/to/file.signed
  // --update-ifs /path/to/file.signed
  // --key /path/to/long_term_identity.key

  // --generate /path/to/file.signed
  // --update /path/to/file.signed
  // printf("has [%d]options\n", argc);
  if(argc < 3)
  {
    printf(
        "please specify --generate or --update with a path to a router contact "
        "file\n");
    return 0;
  }
  bool genMode;
  bool updMode;
  int c;
  char *rcfname;
  while(1)
  {
    static struct option long_options[] = {
        {"generate", required_argument, 0, 'g'},
        {"update", required_argument, 0, 'u'},
        {0, 0, 0, 0}};
    int option_index = 0;
    c = getopt_long(argc, argv, "gu", long_options, &option_index);
    if(c == -1)
      break;
    switch(c)
    {
      case 0:
        break;
      case 'g':
        // printf ("option -g with value `%s'\n", optarg);
        rcfname = optarg;
        genMode = true;
        break;
      case 'u':
        // printf ("option -u with value `%s'\n", optarg);
        rcfname = optarg;
        updMode = true;
        break;
      default:
        abort();
    }
  }
  printf("parsed options\n");
  if(!genMode && !updMode)
  {
    printf("I don't know what to do, no generate or update parameter\n");
    return 1;
  }

  sllarp = new llarp_main;
  llarp_mem_stdlib(&sllarp->mem);
  auto mem = &sllarp->mem;
  // llarp_new_config(&sllarp->config);
  // llarp_ev_loop_alloc(&sllarp->mainloop);
  llarp_crypto_libsodium_init(&sllarp->crypto);

  llarp_rc tmp;
  if(genMode)
  {
    printf("Creating [%s]\n", rcfname);
    // Jeff wanted tmp to be stack created
    // do we still need to zero it out?
    llarp_rc_clear(&tmp);
    // if we zero it out then
    // allocate fresh pointers that the bencoder can expect to be ready
    tmp.addrs = llarp_ai_list_new(mem);
    tmp.exits = llarp_xi_list_new(mem);
    // set updated timestamp
    tmp.last_updated = llarp_time_now_ms();
    // load longterm identity
    llarp_crypto crypt;
    fs::path ident_keyfile = "identity.key";
    llarp_seckey_t identity;
    llarp_findOrCreateIdentity(&crypt, ident_keyfile.c_str(), identity);
    // get identity public key
    uint8_t *pubkey = llarp_seckey_topublic(identity);
    llarp_rc_set_pubkey(&tmp, pubkey);
    // this causes a segfault
    // llarp_rc_sign(&crypt, &identity, &tmp);
    // set filename
    fs::path our_rc_file = rcfname;
    // write file
    llarp_rc_write(&tmp, our_rc_file.c_str());
    // release memory for tmp lists
    llarp_rc_free(&tmp);
  }
  if(updMode)
  {
    printf("Loading [%s]\n", rcfname);
    fs::path our_rc_file = rcfname;
    std::error_code ec;
    if(!fs::exists(our_rc_file, ec))
    {
      printf("File not found\n");
      return 0;
    }
    std::ifstream f(our_rc_file, std::ios::binary);
    if(!f.is_open())
    {
      printf("Can't open file\n");
      return 0;
    }
    byte_t tmpc[MAX_RC_SIZE];
    llarp_buffer_t buf;
    buf.base = tmpc;
    buf.cur  = buf.base;
    buf.sz   = sizeof(tmpc);
    f.read((char *)tmpc, sizeof(MAX_RC_SIZE));
    //printf("contents[%s]\n", tmpc);
    if(!llarp_rc_bdecode(mem, &tmp, &buf))
    {
      printf("Can't decode\n");
      return 0;
    }
    // set updated timestamp
    tmp.last_updated = llarp_time_now_ms();
    // load longterm identity
    llarp_crypto crypt;
    fs::path ident_keyfile = "identity.key";
    llarp_seckey_t identity;
    llarp_findOrCreateIdentity(&crypt, ident_keyfile.c_str(), identity);
    // get identity public key
    uint8_t *pubkey = llarp_seckey_topublic(identity);
    llarp_rc_set_pubkey(&tmp, pubkey);
    llarp_rc_sign(&crypt, identity, &tmp);
    // set filename
    fs::path our_rc_file_out = "update_debug.rc";
    // write file
    llarp_rc_write(&tmp, our_rc_file_out.c_str());
    // release memory for tmp lists
    llarp_rc_free(&tmp);
  }

  delete sllarp;
  return 1;
}