PeerStreamer

/*

 *  Copyright (c) 2009 Luca Abeni

 *

 *  This is free software; see gpl-3.0.txt

 *

 *  This is a small test program for the gossip based TopologyManager

 *  To try the simple test: run it with

 *    ./topology_test -I <network interface> -P <port> [-i <remote IP> -p <remote port>]

 *    the "-i" and "-p" parameters can be used to add an initial neighbour

 *    (otherwise, the peer risks to stay out of the overlay).

 *    For example, run

 *      ./topology_test -I eth0 -P 6666

 *    on a computer, and then

 *      ./topology_test -I eth0 -P 2222 -i <ip_address> -p 6666

 *  on another one ... Of course, one of the two peers has to use -i... -p...

 *  (in general, to be part of the overlay a peer must either use

 *  "-i<known peer IP> -p<known peer port>" or be referenced by another peer).

 */

#include <stdlib.h>

#include <stdint.h>

#include <stdio.h>

#include <string.h>

#include <getopt.h>

#include "net_helper.h"

#include "peersampler.h"

#include "net_helpers.h"

static const char *psample_config;

static struct psample_context *context;

static const char *my_addr = "127.0.0.1";

static int port = 6666;

static int srv_port;

static const char *srv_ip;

static char *fprefix;

static void cmdline_parse(int argc, char *argv[])

{

  int o;

  while ((o = getopt(argc, argv, "s:p:i:P:I:c")) != -1) {

    switch(o) {

      case 'p':

        srv_port = atoi(optarg);

        break;

      case 'i':

        srv_ip = strdup(optarg);

        break;

      case 'P':

        port =  atoi(optarg);

        break;

      case 'I':

        my_addr = iface_addr(optarg);

        break;

      case 's':

        fprefix = strdup(optarg);

        break;

      case 'c':

        psample_config = "protocol=cyclon";

        break;

      default:

        fprintf(stderr, "Error: unknown option %c\n", o);

        exit(-1);

    }

  }

}

static struct nodeID *init(void)

{

  struct nodeID *myID;

  myID = net_helper_init(my_addr, port, "");

  if (myID == NULL) {

    fprintf(stderr, "Error creating my socket (%s:%d)!\n", my_addr, port);

    return NULL;

  }

  context = psample_init(myID, NULL, 0, psample_config);

  return myID;

}

static void loop(struct nodeID *s)

{

  int done = 0;

#define BUFFSIZE 1024

  static uint8_t buff[BUFFSIZE];

  int cnt = 0;

  psample_parse_data(context, NULL, 0);

  while (!done) {

    int len;

    int news;

    const struct timeval tout = {1, 0};

    struct timeval t1;

    t1 = tout;

    news = wait4data(s, &t1, NULL);

    if (news > 0) {

      struct nodeID *remote;

      len = recv_from_peer(s, &remote, buff, BUFFSIZE);

      psample_parse_data(context, buff, len);

      nodeid_free(remote);

    } else {

      psample_parse_data(context, NULL, 0);

      if (cnt % 10 == 0) {

        const struct nodeID **neighbourhoods;

        int n, i;

        neighbourhoods = psample_get_cache(context, &n);

        printf("I have %d neighbours:\n", n);

        for (i = 0; i < n; i++) {

          printf("\t%d: %s\n", i, node_addr(neighbourhoods[i]));

        }

        fflush(stdout);

        if (fprefix) {

          FILE *f;

          char fname[64];

          sprintf(fname, "%s-%d.txt", fprefix, port);

          f = fopen(fname, "w");

          if (f) fprintf(f, "#Cache size: %d\n", n);

          for (i = 0; i < n; i++) {

            if (f) fprintf(f, "%d\t\t%d\t%s\n", port, i, node_addr(neighbourhoods[i]));

          }

          fclose(f);

        }

      }

      cnt++;

    }

  }

}

int main(int argc, char *argv[])

{

  struct nodeID *my_sock;

  cmdline_parse(argc, argv);

  my_sock = init();

  if (my_sock == NULL) {

    return -1;

  }

  if (srv_port != 0) {

    struct nodeID *knownHost;

    knownHost = create_node(srv_ip, srv_port);

    if (knownHost == NULL) {

      fprintf(stderr, "Error creating knownHost socket (%s:%d)!\n", srv_ip, srv_port);

      return -1;

    }

    psample_add_peer(context, knownHost, NULL, 0);

  }

  loop(my_sock);

  return 0;

}