main.cpp

#include "kadsim.h"

using namespace std;

KadConf::KadConf(int n_bits, int k, int alpha, int n_nodes)
{
  this->n_bits = n_bits;
  this->k = k;
  this->alpha = alpha;
  this->n_nodes = n_nodes;
}

void
KadConf::save(std::ostream& fout)
{
  fout << "n_bits " << n_bits << "\n";
  fout << "k " << k << "\n";
  fout << "alpha " << alpha << "\n";
  fout << "n_nodes " << n_nodes << "\n";
}

void
usage()
{
  std::cerr << "usage: kadsim\n";
  std::cerr << "\t-b\tn_bits\n";
  std::cerr << "\t-k\tKademlia K parameter\n";
  std::cerr << "\t-a\tKademlia alpha parameter\n";
  std::cerr << "\t-n\tnumber of nodes\n";
  std::cerr << "\t-c\tinitial number of connections per node\n";
  std::cerr << "\t-N\tnumber of files\n";
  std::cerr << "\t-S\trandom seed\n";
  exit(1);
}

void
parse_error()
{
  std::cerr << "parse error\n";
  exit(1);
}

int main(int argc, char **argv)
{
  int c;
  int n_bits = 64;
  int k = 20;
  int alpha = 3;
  int n_nodes = 1500;
  int n_init_conn = 100;
  int n_files = 5000;
  int rand_seed = 0;
  char *fname = NULL;
     
  opterr = 0;
     
  while ((c = getopt (argc, argv, "b:k:a:n:c:S:f:N:")) != -1)
    {
      switch (c)
	{
	case 'b':
	  n_bits = atoi(optarg);
	  break;
	case 'k':
	  k = atoi(optarg);
	  break;
	case 'a':
	  alpha = atoi(optarg);
	  break;
	case 'n':
	  n_nodes = atoi(optarg);
	  break;
	case 'c':
	  n_init_conn = atoi(optarg);
	  break;
	case 'S':
	  rand_seed = atoi(optarg);
	  break ;
	case 'f':
	  fname = strdup(optarg);
	  break ;
	case 'N':
	  n_files = atoi(optarg);
	  break ;
	case '?':
	default:
	  usage();
	}
    }

  if (fname)
    {
      std::ifstream fin(fname);
      if (!fin.is_open())
	{
	  std::cerr << "unable to open " << fname << "\n";
	  exit(1);
	}

      char buf[256], *p;
#define GETLINE() \
      fin.getline(buf, sizeof (buf));	  \
      if (NULL == (p = rindex(buf, ' '))) \
	parse_error();			  \
      p++;
      GETLINE(); n_bits = atoi(p);
      GETLINE(); k = atoi(p);
      GETLINE(); alpha = atoi(p);
      GETLINE(); n_nodes = atoi(p);
    }

  KadConf conf(n_bits, k, alpha, n_nodes);
  //conf.save(std::cout);
  KadNetwork network(&conf);
  Shell shell;

  srand(rand_seed);

  network.initialize_nodes(n_init_conn);
  network.initialize_files(n_files);
  network.check_files();

  shell.set_cmds(cmd_defs);
  shell.set_handle(&network);
  shell.set_prompt("kadsim> ");
  shell.loop();

  return EXIT_SUCCESS;
}