PeerStreamer

/*

 *  Copyright (c) 2010 Luca Abeni

 *  Copyright (c) 2010 Csaba Kiraly

 *

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

 */

#include <sys/select.h>

#include <sys/time.h>

#include <time.h>

#include <stdint.h>

#include <stdlib.h>

#include <stdio.h>

#include <stdbool.h>

#include <string.h>

#include <net_helper.h>

#include <grapes_msg_types.h>

#include <peerset.h>

#include <peer.h>

#include "chunk_signaling.h"

#include "streaming.h"

#include "topology.h"

#include "loop.h"

#include "dbg.h"

#define BUFFSIZE 512 * 1024

#define FDSSIZE 16

static struct timeval period = {0, 500000};

static struct timeval tnext;

#ifdef HTTPIO_MHD

extern pthread_mutex_t cb_mutex;

#endif

void tout_init(struct timeval *tv)

{

  struct timeval tnow;

  if (tnext.tv_sec == 0) {

    gettimeofday(&tnext, NULL);

  }

  gettimeofday(&tnow, NULL);

  if(timercmp(&tnow, &tnext, <)) {

    timersub(&tnext, &tnow, tv);

  } else {

    *tv = (struct timeval){0, 0};

  }

}

void loop(struct nodeID *s, int csize, int buff_size)

{

  int done = 0;

  static uint8_t buff[BUFFSIZE];

  int cnt = 0;

  period.tv_sec = csize / 1000000;

  period.tv_usec = csize % 1000000;

  peers_init();

  stream_init(buff_size, s);

  update_peers(NULL, NULL, 0);

  while (!done) {

    int len, res;

    struct timeval tv;

    tout_init(&tv);

    res = wait4data(s, &tv, NULL);

    if (res > 0) {

      struct nodeID *remote;

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

      if (len < 0) {

        fprintf(stderr,"Error receiving message. Maybe larger than %d bytes\n", BUFFSIZE);

        nodeid_free(remote);

        continue;

      }

      switch (buff[0] /* Message Type */) {

                case MSG_TYPE_TMAN:

        case MSG_TYPE_TOPOLOGY:

          update_peers(remote, buff, len);

          break;

        case MSG_TYPE_CHUNK:

          dprintf("Chunk message received:\n");

          received_chunk(remote, buff, len);

          break;

        case MSG_TYPE_SIGNALLING:

          sigParseData(remote, buff, len);

          break;

        default:

          fprintf(stderr, "Unknown Message Type %x\n", buff[0]);

      }

      nodeid_free(remote);

    } else {

      struct timeval tmp;

      //send_chunk();

      send_offer();

      if (cnt++ % 10 == 0) {

        update_peers(NULL, NULL, 0);

      }

      timeradd(&tnext, &period, &tmp);

      tnext = tmp;

    }

  }

}

void source_loop(const char *fname, struct nodeID *s, int csize, int chunks, bool loop)

{

  int done = 0;

  static uint8_t buff[BUFFSIZE];

  int cnt = 0;

  int fds[FDSSIZE];

  fds[0] = -1;

  period.tv_sec = csize  / 1000000;

  period.tv_usec = csize % 1000000;

  peers_init();

  if (source_init(fname, s, loop, fds, FDSSIZE) < 0) {

    fprintf(stderr,"Cannot initialize source, exiting");

    return;

  }

  while (!done) {

    int len, res;

    struct timeval tv, *ptv;

    int wait4fds[FDSSIZE], *pfds;

#ifdef HTTPIO

    memcpy(wait4fds, fds, sizeof(fds));

    res = wait4data(s, NULL, wait4fds);

#else

    if (fds[0] == -1) {

      tout_init(&tv);

      ptv = &tv;

      pfds = NULL;

    } else {

      memcpy(wait4fds, fds, sizeof(fds));

      pfds = wait4fds;

      ptv = NULL;

    }

    res = wait4data(s, ptv, pfds);

#endif

    if (res == 1) {

      struct nodeID *remote;

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

      if (len < 0) {

        fprintf(stderr,"Error receiving message. Maybe larger than %d bytes\n", BUFFSIZE);

        nodeid_free(remote);

        continue;

      }

      dprintf("Received message (%c) from %s\n", buff[0], node_addr(remote));

      switch (buff[0] /* Message Type */) {

                case MSG_TYPE_TMAN:

        case MSG_TYPE_TOPOLOGY:

          fprintf(stderr, "Top Parse\n");

#ifdef HTTPIO_MHD

          pthread_mutex_lock(&cb_mutex);

#endif

          update_peers(remote, buff, len);

#ifdef HTTPIO_MHD

          pthread_mutex_unlock(&cb_mutex);

#endif

          break;

        case MSG_TYPE_CHUNK:

          fprintf(stderr, "Some dumb peer pushed a chunk to me! peer:%s\n",node_addr(remote));

          break;

        case MSG_TYPE_SIGNALLING:

#ifdef HTTPIO_MHD

          pthread_mutex_lock(&cb_mutex);

#endif

          sigParseData(remote, buff, len);

#ifdef HTTPIO_MHD

          pthread_mutex_unlock(&cb_mutex);

#endif

          break;

        default:

          fprintf(stderr, "Bad Message Type %x\n", buff[0]);

      }

      nodeid_free(remote);

    } else if (res == 0 || res == 2) {        //timeout or data arrived from source

      int i;

      struct timeval tmp, d;

      struct chunk *c;

      d.tv_sec = 0;

      c = generated_chunk(&d.tv_usec);

      if (c) {

        add_chunk(c);

        for (i = 0; i < chunks; i++) {        // @TODO: why this cycle?

          send_chunk();

        }

        if (cnt++ % 10 == 0) {

            update_peers(NULL, NULL, 0);

        }

      }

      timeradd(&tnext, &d, &tmp);

      tnext = tmp;

    }

  }

}