You've already forked mariadb-columnstore-engine
							
							
				mirror of
				https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
				synced 2025-11-03 17:13:17 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			3418 lines
		
	
	
		
			89 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			3418 lines
		
	
	
		
			89 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/* Copyright (C) 2014 InfiniDB, Inc.
 | 
						|
 | 
						|
   This program is free software; you can redistribute it and/or
 | 
						|
   modify it under the terms of the GNU General Public License
 | 
						|
   as published by the Free Software Foundation; version 2 of
 | 
						|
   the License.
 | 
						|
 | 
						|
   This program is distributed in the hope that it will be useful,
 | 
						|
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
   GNU General Public License for more details.
 | 
						|
 | 
						|
   You should have received a copy of the GNU General Public License
 | 
						|
   along with this program; if not, write to the Free Software
 | 
						|
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 | 
						|
   MA 02110-1301, USA. */
 | 
						|
 | 
						|
// $Id: impl.cpp 3495 2013-01-21 14:09:51Z rdempsey $
 | 
						|
 | 
						|
/* This code is based on udpcast-20090830. Most of the source code in that release
 | 
						|
   contains no copyright or licensing notices at all. The exception is fec.c, which
 | 
						|
   is not used here. The udpcast website, http://udpcast.linux.lu/, implies that
 | 
						|
   the source is covered under GPL. */
 | 
						|
 | 
						|
#include <net/if.h>
 | 
						|
#include <sys/socket.h>
 | 
						|
#include <netinet/in.h>
 | 
						|
#include <arpa/inet.h>
 | 
						|
#include <sys/ioctl.h>
 | 
						|
#include <cerrno>
 | 
						|
#include <pthread.h>
 | 
						|
using namespace std;
 | 
						|
 | 
						|
#include "bytestream.h"
 | 
						|
using namespace messageqcpp;
 | 
						|
 | 
						|
#include "udp-sender.h"
 | 
						|
#include "udpc-protoc.h"
 | 
						|
#include "util.h"
 | 
						|
#include "mc_fifo.h"
 | 
						|
 | 
						|
#include "impl.h"
 | 
						|
 | 
						|
struct  participantsDb {
 | 
						|
    int nrParticipants;
 | 
						|
    
 | 
						|
    struct clientDesc {
 | 
						|
	struct sockaddr_in addr;
 | 
						|
	int used;
 | 
						|
	int capabilities;
 | 
						|
	unsigned int rcvbuf;
 | 
						|
    } clientTable[MAX_CLIENTS];
 | 
						|
};
 | 
						|
struct produconsum {
 | 
						|
    unsigned int size;
 | 
						|
    volatile unsigned int produced;
 | 
						|
    unsigned int consumed;
 | 
						|
    volatile int atEnd;
 | 
						|
    pthread_mutex_t mutex;
 | 
						|
    volatile int consumerIsWaiting;
 | 
						|
    pthread_cond_t cond;
 | 
						|
    const char *name;
 | 
						|
};
 | 
						|
 | 
						|
struct stats {
 | 
						|
    int fd;
 | 
						|
    struct timeval lastPrinted;
 | 
						|
    long statPeriod;
 | 
						|
    int printUncompressedPos;
 | 
						|
};
 | 
						|
 | 
						|
struct sender_stats {
 | 
						|
    FILE *log;    
 | 
						|
    unsigned long long totalBytes;
 | 
						|
    unsigned long long retransmissions;
 | 
						|
    int clNo;
 | 
						|
    unsigned long periodBytes;
 | 
						|
    struct timeval periodStart;
 | 
						|
    long bwPeriod;
 | 
						|
    struct stats s;
 | 
						|
};
 | 
						|
 | 
						|
struct receiver_stats {
 | 
						|
    struct timeval tv_start;
 | 
						|
    int bytesOrig;
 | 
						|
    long long totalBytes;
 | 
						|
    int timerStarted;
 | 
						|
    struct stats s;
 | 
						|
};
 | 
						|
 | 
						|
#define SLICEMAGIC 0x41424344
 | 
						|
 | 
						|
typedef struct slice {
 | 
						|
    int base; /* base address of slice in buffer */
 | 
						|
    int sliceNo;
 | 
						|
    int bytes; /* bytes in slice */
 | 
						|
    int nextBlock; /* index of next buffer to be transmitted */
 | 
						|
    enum slice_state { 
 | 
						|
	SLICE_FREE, /* free slice, and in the queue of free slices */
 | 
						|
	SLICE_NEW, /* newly allocated. FEC calculation and first 
 | 
						|
		    * transmission */
 | 
						|
	SLICE_XMITTED, /* transmitted */
 | 
						|
	SLICE_ACKED, /* acknowledged (if applicable) */
 | 
						|
	SLICE_PRE_FREE, /* no longer used, but not returned to queue */
 | 
						|
	SLICE_RECEIVING,
 | 
						|
	SLICE_DONE,
 | 
						|
    };
 | 
						|
    volatile enum slice_state state;
 | 
						|
    char rxmitMap[MAX_SLICE_SIZE / BITS_PER_CHAR]; 
 | 
						|
    /* blocks to be retransmitted */
 | 
						|
 | 
						|
    char isXmittedMap[MAX_SLICE_SIZE / BITS_PER_CHAR]; 
 | 
						|
   /* blocks which have already been retransmitted during this round*/
 | 
						|
 | 
						|
    int rxmitId; /* used to distinguish among several retransmission 
 | 
						|
		  * requests, so that we can easily discard answers to "old"
 | 
						|
		  * requests */
 | 
						|
 | 
						|
    /* This structure is used to keep track of clients who answered, and
 | 
						|
     * to make the reqack message
 | 
						|
     */
 | 
						|
    struct reqackBm {
 | 
						|
	struct reqack ra;
 | 
						|
	char readySet[MAX_CLIENTS / BITS_PER_CHAR]; /* who is already ok? */
 | 
						|
    } sl_reqack;
 | 
						|
 | 
						|
    char answeredSet[MAX_CLIENTS / BITS_PER_CHAR]; /* who answered at all? */
 | 
						|
 | 
						|
    int nrReady; /* number of participants who are ready */
 | 
						|
    int nrAnswered; /* number of participants who answered; */
 | 
						|
    int needRxmit; /* does this need retransmission? */
 | 
						|
    int lastGoodBlock; /* last good block of slice (i.e. last block having not
 | 
						|
			* needed retransmission */
 | 
						|
 | 
						|
    int lastReqack; /* last req ack sent (debug) */
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    unsigned char *fec_data;
 | 
						|
#endif
 | 
						|
    int magic;
 | 
						|
    int blocksTransferred; /* blocks transferred during this slice */
 | 
						|
    int dataBlocksTransferred; /* data blocks transferred during this slice */
 | 
						|
    struct retransmit retransmit;
 | 
						|
    int freePos; /* where the next data part will be stored to */
 | 
						|
    int bytesKnown; /* is number of bytes known yet? */
 | 
						|
    short missing_data_blocks[MAX_FEC_INTERLEAVE]; 
 | 
						|
} *slice_t;
 | 
						|
 | 
						|
#define QUEUE_SIZE 256
 | 
						|
 | 
						|
struct returnChannel {
 | 
						|
    pthread_t thread; /* message receiving thread */
 | 
						|
    int rcvSock; /* socket on which we receive the messages */
 | 
						|
    produconsum_t incoming; /* where to enqueue incoming messages */
 | 
						|
    produconsum_t freeSpace; /* free space */
 | 
						|
    struct {
 | 
						|
	int clNo; /* client number */
 | 
						|
	union message msg; /* its message */
 | 
						|
    } q[QUEUE_SIZE];
 | 
						|
    struct net_config *config;
 | 
						|
    participantsDb_t participantsDb;
 | 
						|
};
 | 
						|
#define NR_SLICES 2
 | 
						|
 | 
						|
typedef struct senderState {
 | 
						|
    struct returnChannel rc;
 | 
						|
    struct fifo *fifo;
 | 
						|
 | 
						|
    struct net_config *config;
 | 
						|
    sender_stats_t stats;
 | 
						|
    int socket;
 | 
						|
    
 | 
						|
    struct slice slices[NR_SLICES];
 | 
						|
 | 
						|
    produconsum_t free_slices_pc;
 | 
						|
 | 
						|
    unsigned char *fec_data;
 | 
						|
    pthread_t fec_thread;
 | 
						|
    produconsum_t fec_data_pc;
 | 
						|
} *sender_state_t;
 | 
						|
 | 
						|
struct clientState {
 | 
						|
    struct fifo *fifo;
 | 
						|
    struct client_config *client_config;
 | 
						|
    struct net_config *net_config;
 | 
						|
    union serverDataMsg Msg;
 | 
						|
 | 
						|
    struct msghdr data_hdr;
 | 
						|
 | 
						|
    /* pre-prepared messages */
 | 
						|
    struct iovec data_iov[2];
 | 
						|
 | 
						|
    struct slice *currentSlice;
 | 
						|
    int currentSliceNo;
 | 
						|
    receiver_stats_t stats;
 | 
						|
    
 | 
						|
    produconsum_t free_slices_pc;
 | 
						|
    struct slice slices[NR_SLICES];
 | 
						|
 | 
						|
    /* Completely received slices */
 | 
						|
    int receivedPtr;
 | 
						|
    int receivedSliceNo;
 | 
						|
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    int use_fec; /* do we use forward error correction ? */
 | 
						|
#endif
 | 
						|
    produconsum_t fec_data_pc;
 | 
						|
    struct slice *fec_slices[NR_SLICES];
 | 
						|
    pthread_t fec_thread;
 | 
						|
 | 
						|
    /* A reservoir of free blocks for FEC */
 | 
						|
    produconsum_t freeBlocks_pc;
 | 
						|
    unsigned char **blockAddresses; /* adresses of blocks in local queue */
 | 
						|
 | 
						|
    unsigned char **localBlockAddresses;
 | 
						|
				/* local blocks: freed FEC blocks after we
 | 
						|
				 * have received the corresponding data */
 | 
						|
    int localPos;
 | 
						|
 | 
						|
    unsigned char *blockData;
 | 
						|
    unsigned char *nextBlock;
 | 
						|
 | 
						|
    int endReached; /* end of transmission reached:
 | 
						|
		       0: transmission in progress
 | 
						|
		       2: network transmission _and_ FEC 
 | 
						|
		          processing finished 
 | 
						|
		    */
 | 
						|
 | 
						|
    int netEndReached; /* In case of a FEC transmission; network
 | 
						|
			* transmission finished. This is needed to avoid
 | 
						|
			* a race condition, where the receiver thread would
 | 
						|
			* already prepare to wait for more data, at the same
 | 
						|
			* time that the FEC would set endReached. To avoid
 | 
						|
			* this, we do a select without timeout before
 | 
						|
			* receiving the last few packets, so that if the
 | 
						|
			* race condition strikes, we have a way to protect
 | 
						|
			* against
 | 
						|
			*/
 | 
						|
 | 
						|
    int selectedFd;
 | 
						|
 | 
						|
    int promptPrinted;  /* Has "Press any key..." prompt already been printed */
 | 
						|
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    fec_code_t fec_code;
 | 
						|
#endif
 | 
						|
};
 | 
						|
 | 
						|
#define S_UCAST socks[0]
 | 
						|
#define S_BCAST socks[1]
 | 
						|
#define S_MCAST_CTRL socks[2]
 | 
						|
#define S_MCAST_DATA socks[3]
 | 
						|
 | 
						|
#define SSEND(x) SEND(client_config->S_UCAST, x, client_config->serverAddr)
 | 
						|
 | 
						|
/**
 | 
						|
 * Receiver will passively listen to sender. Works best if sender runs
 | 
						|
 * in async mode
 | 
						|
 */
 | 
						|
#define FLAG_PASSIVE 0x0010
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * Do not write file synchronously
 | 
						|
 */
 | 
						|
#define FLAG_NOSYNC 0x0040
 | 
						|
 | 
						|
/*
 | 
						|
 * Don't ask for keyboard input on receiver end.
 | 
						|
 */
 | 
						|
#define FLAG_NOKBD 0x0080
 | 
						|
 | 
						|
/**
 | 
						|
 * Do write file synchronously
 | 
						|
 */
 | 
						|
#define FLAG_SYNC 0x0100
 | 
						|
 | 
						|
namespace
 | 
						|
{
 | 
						|
int udpc_isFullDuplex(int s, const char *ifname) {
 | 
						|
 | 
						|
#ifdef ETHTOOL_GLINK
 | 
						|
  struct ifreq ifr;
 | 
						|
  struct ethtool_cmd ecmd;
 | 
						|
 | 
						|
  ecmd.cmd = ETHTOOL_GSET;
 | 
						|
 | 
						|
  strncpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name)-1);
 | 
						|
  ifr.ifr_data = (char *) &ecmd;
 | 
						|
 | 
						|
  if(ioctl(s, SIOCETHTOOL, &ifr) == -1) {
 | 
						|
    /* Operation not supported */
 | 
						|
    return -1;
 | 
						|
  } else {
 | 
						|
    return ecmd.duplex;
 | 
						|
  }
 | 
						|
#else
 | 
						|
  return -1;
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
#define getSinAddr(addr) (((struct sockaddr_in *) addr)->sin_addr)
 | 
						|
 | 
						|
int udpc_ipIsZero(struct sockaddr_in *ip) {
 | 
						|
    return getSinAddr(ip).s_addr == 0;
 | 
						|
}
 | 
						|
 | 
						|
int hasLink(int s, const char *ifname) {
 | 
						|
 | 
						|
#ifdef ETHTOOL_GLINK
 | 
						|
  struct ifreq ifr;
 | 
						|
  struct ethtool_value edata;
 | 
						|
 | 
						|
  edata.cmd = ETHTOOL_GLINK;
 | 
						|
 | 
						|
  strncpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name)-1);
 | 
						|
  ifr.ifr_data = (char *) &edata;
 | 
						|
 | 
						|
  if(ioctl(s, SIOCETHTOOL, &ifr) == -1) {
 | 
						|
    /* Operation not supported */
 | 
						|
    return -1;
 | 
						|
  } else {
 | 
						|
    return edata.data;
 | 
						|
  }
 | 
						|
#else
 | 
						|
  return -1;
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
#define INET_ATON(a,i) inet_aton(a,i)
 | 
						|
 | 
						|
int udpc_doSend(int s, void *message, size_t len, struct sockaddr_in *to) {
 | 
						|
/*    flprintf("sent: %08x %d\n", *(int*) message, len);*/
 | 
						|
#ifdef LOSSTEST
 | 
						|
    loseSendPacket();
 | 
						|
#endif
 | 
						|
    return sendto(s, message, len, 0, (struct sockaddr*) to, sizeof(*to));
 | 
						|
}
 | 
						|
 | 
						|
void udpc_copyToMessage(unsigned char *dst, struct sockaddr_in *src) {    
 | 
						|
    memcpy(dst, (char *) &((struct sockaddr_in *)src)->sin_addr,
 | 
						|
	   sizeof(struct in_addr));
 | 
						|
}
 | 
						|
 | 
						|
void udpc_sendHello(struct net_config *net_config, int sock,
 | 
						|
	       int streaming) {
 | 
						|
//cerr << "sending hello..." << endl;
 | 
						|
    struct hello hello;
 | 
						|
    /* send hello message */
 | 
						|
    if(streaming)
 | 
						|
	hello.opCode = htons(CMD_HELLO_STREAMING);
 | 
						|
    else
 | 
						|
	hello.opCode = htons(CMD_HELLO);
 | 
						|
    hello.reserved = 0;
 | 
						|
    hello.capabilities = htonl(net_config->capabilities);
 | 
						|
    udpc_copyToMessage(hello.mcastAddr,&net_config->dataMcastAddr);
 | 
						|
    hello.blockSize = htons(net_config->blockSize);
 | 
						|
	//TODO: FIXME
 | 
						|
    //rgWaitAll(net_config, sock, net_config->controlMcastAddr.sin_addr.s_addr, sizeof(hello));
 | 
						|
    BCAST_CONTROL(sock, hello);
 | 
						|
}
 | 
						|
 | 
						|
char *udpc_getIpString(struct sockaddr_in *addr, char *buffer) {
 | 
						|
    long iaddr = htonl(getSinAddr(addr).s_addr);
 | 
						|
    sprintf(buffer,"%ld.%ld.%ld.%ld", 
 | 
						|
	    (iaddr >> 24) & 0xff,
 | 
						|
	    (iaddr >> 16) & 0xff,
 | 
						|
	    (iaddr >>  8) & 0xff,
 | 
						|
	    iaddr & 0xff);
 | 
						|
    return buffer;
 | 
						|
}
 | 
						|
 | 
						|
net_if_t *udpc_getNetIf(const char *wanted) {
 | 
						|
#ifndef __MINGW32__
 | 
						|
	struct ifreq ibuf[100];
 | 
						|
	struct ifreq *ifrp, *ifend, *chosen;
 | 
						|
	struct ifconf ifc;
 | 
						|
	int s;
 | 
						|
#else /* __MINGW32__ */
 | 
						|
	int i;
 | 
						|
 | 
						|
	int etherNo=-1;
 | 
						|
	int wantedEtherNo=-2; /* Wanted ethernet interface */
 | 
						|
 | 
						|
	MIB_IPADDRTABLE *iptab=NULL;
 | 
						|
	MIB_IFTABLE *iftab=NULL;
 | 
						|
 | 
						|
	MIB_IPADDRROW *iprow, *chosen=NULL;
 | 
						|
	MIB_IFROW *chosenIf=NULL;
 | 
						|
	WORD wVersionRequested; /* Version of Winsock to load */
 | 
						|
	WSADATA wsaData; /* Winsock implementation details */
 | 
						|
	ULONG a;
 | 
						|
 | 
						|
	int r;
 | 
						|
#endif /* __MINGW32__ */
 | 
						|
 | 
						|
	int lastGoodness=0;
 | 
						|
	struct in_addr wantedAddress;
 | 
						|
	int isAddress=0;
 | 
						|
	int wantedLen=0;
 | 
						|
	net_if_t *net_if;
 | 
						|
 | 
						|
	if(wanted == NULL) {
 | 
						|
	    wanted = getenv("IFNAME");
 | 
						|
	}
 | 
						|
 | 
						|
	if(wanted && INET_ATON(wanted, &wantedAddress))
 | 
						|
	    isAddress=1;
 | 
						|
	else
 | 
						|
	    wantedAddress.s_addr=0;
 | 
						|
 | 
						|
	if(wanted)
 | 
						|
	    wantedLen=strlen(wanted);
 | 
						|
 | 
						|
	net_if = MALLOC(net_if_t);
 | 
						|
			//TODO: FIXME
 | 
						|
	//if(net_if == NULL)
 | 
						|
	//    udpc_fatal(1, "Out of memory error");
 | 
						|
 | 
						|
#ifndef __MINGW32__
 | 
						|
 | 
						|
	s = socket(PF_INET, SOCK_DGRAM, 0);
 | 
						|
	if (s < 0) {
 | 
						|
	    perror("make socket");
 | 
						|
	    exit(1);
 | 
						|
	}	
 | 
						|
 | 
						|
	ifc.ifc_len = sizeof(ibuf);
 | 
						|
	ifc.ifc_buf = (caddr_t) ibuf;
 | 
						|
 | 
						|
	if (ioctl(s, SIOCGIFCONF, (char *)&ifc) < 0 ||
 | 
						|
            ifc.ifc_len < (signed int)sizeof(struct ifreq)) {
 | 
						|
                perror("udpcast: SIOCGIFCONF: ");
 | 
						|
                exit(1);
 | 
						|
        }
 | 
						|
 | 
						|
	ifend = (struct ifreq *)((char *)ibuf + ifc.ifc_len);
 | 
						|
	chosen=NULL;
 | 
						|
 | 
						|
	for (ifrp = ibuf ; ifrp < ifend;
 | 
						|
#ifdef IFREQ_SIZE
 | 
						|
	     ifrp = IFREQ_SIZE(*ifrp) + (char *)ifrp
 | 
						|
#else
 | 
						|
	     ifrp++
 | 
						|
#endif       
 | 
						|
	     ) {
 | 
						|
	    unsigned long iaddr = getSinAddr(&ifrp->ifr_addr).s_addr;
 | 
						|
	    int goodness;
 | 
						|
 | 
						|
	    if(ifrp->ifr_addr.sa_family != PF_INET)
 | 
						|
		continue;
 | 
						|
	    
 | 
						|
	    if(wanted) {
 | 
						|
		if(isAddress && iaddr == wantedAddress.s_addr) {
 | 
						|
		    goodness=8;
 | 
						|
		} else if(strcmp(wanted, ifrp->ifr_name) ==0) {
 | 
						|
		    /* perfect match on interface name */
 | 
						|
		    goodness=12;
 | 
						|
		} else if(wanted != NULL && 
 | 
						|
			  strncmp(wanted, ifrp->ifr_name, wantedLen) ==0) {
 | 
						|
		    /* prefix match on interface name */
 | 
						|
		    goodness=7;
 | 
						|
		} else {
 | 
						|
		    /* no match, try next */
 | 
						|
		    continue;
 | 
						|
		}
 | 
						|
	    } else {
 | 
						|
		if(iaddr == 0) {
 | 
						|
		    /* disregard interfaces whose address is zero */
 | 
						|
		    goodness = 1;
 | 
						|
		} else if(iaddr == htonl(0x7f000001)) {
 | 
						|
		    /* disregard localhost type devices */
 | 
						|
		    goodness = 2;
 | 
						|
		} else if(strcmp("eth0", ifrp->ifr_name) == 0 ||
 | 
						|
			  strcmp("en0",  ifrp->ifr_name) == 0) {
 | 
						|
		    /* prefer first ethernet interface */
 | 
						|
		    goodness = 6;
 | 
						|
		} else if(strncmp("eth0:", ifrp->ifr_name, 5) == 0) {
 | 
						|
		    /* second choice: any secondary addresses of first ethernet */
 | 
						|
		    goodness = 5;
 | 
						|
		} else if(strncmp("eth", ifrp->ifr_name, 3) == 0 ||
 | 
						|
			  strncmp("en",  ifrp->ifr_name, 2) == 0) {
 | 
						|
		    /* and, if not available, any other ethernet device */
 | 
						|
		    goodness = 4;
 | 
						|
		} else {
 | 
						|
		    goodness = 3;
 | 
						|
		}
 | 
						|
	    }
 | 
						|
 | 
						|
	    if(hasLink(s, ifrp->ifr_name))
 | 
						|
	      /* Good or unknown link status privileged over known 
 | 
						|
	       * disconnected */
 | 
						|
	      goodness += 3;
 | 
						|
 | 
						|
	    /* If all else is the same, prefer interfaces that 
 | 
						|
	     * have broadcast */
 | 
						|
	    goodness = goodness * 2;
 | 
						|
	    if(goodness >= lastGoodness) {
 | 
						|
		/* Privilege broadcast-enabled interfaces */
 | 
						|
		if(ioctl(s,  SIOCGIFBRDADDR, ifrp) < 0)
 | 
						|
		{
 | 
						|
			//TODO: FIXME
 | 
						|
		    //udpc_fatal(-1, "Error getting broadcast address for %s: %s", ifrp->ifr_name, strerror(errno));
 | 
						|
		}
 | 
						|
		if(getSinAddr(&ifrp->ifr_ifru.ifru_broadaddr).s_addr)
 | 
						|
		    goodness++;
 | 
						|
	    }
 | 
						|
 | 
						|
	    if(goodness > lastGoodness) {
 | 
						|
		chosen = ifrp;
 | 
						|
		lastGoodness = goodness;
 | 
						|
		net_if->addr.s_addr = iaddr;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
 | 
						|
 | 
						|
	if(!chosen) {
 | 
						|
	    fprintf(stderr, "No suitable network interface found\n");
 | 
						|
	    fprintf(stderr, "The following interfaces are available:\n");
 | 
						|
 | 
						|
	    for (ifrp = ibuf ; ifrp < ifend;
 | 
						|
#ifdef IFREQ_SIZE
 | 
						|
		 ifrp = IFREQ_SIZE(*ifrp) + (char *)ifrp
 | 
						|
#else
 | 
						|
		 ifrp++
 | 
						|
#endif
 | 
						|
		 ) {
 | 
						|
		char buffer[16];
 | 
						|
 | 
						|
		if(ifrp->ifr_addr.sa_family != PF_INET)
 | 
						|
		    continue;
 | 
						|
 | 
						|
		fprintf(stderr, "\t%s\t%s\n",
 | 
						|
			ifrp->ifr_name, 
 | 
						|
			udpc_getIpString((struct sockaddr_in *)&ifrp->ifr_addr, buffer));
 | 
						|
	    }
 | 
						|
	    exit(1);
 | 
						|
	}
 | 
						|
 | 
						|
	net_if->name = strdup(chosen->ifr_name);
 | 
						|
 | 
						|
#ifdef HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
 | 
						|
	/* Index for multicast subscriptions */
 | 
						|
	if(ioctl(s,  SIOCGIFINDEX, chosen) < 0)
 | 
						|
	{
 | 
						|
			//TODO: FIXME
 | 
						|
	    //udpc_fatal(-1, "Error getting index for %s: %s", net_if->name, strerror(errno));
 | 
						|
	}
 | 
						|
	net_if->index = chosen->ifr_ifindex;
 | 
						|
#endif
 | 
						|
 | 
						|
	/* Broadcast */
 | 
						|
	if(ioctl(s,  SIOCGIFBRDADDR, chosen) < 0)
 | 
						|
	{
 | 
						|
			//TODO: FIXME
 | 
						|
	    //udpc_fatal(-1, "Error getting broadcast address for %s: %s", net_if->name, strerror(errno));
 | 
						|
	}
 | 
						|
	net_if->bcast = getSinAddr(&chosen->ifr_ifru.ifru_broadaddr);
 | 
						|
 | 
						|
	close(s);
 | 
						|
 | 
						|
#else /* __MINGW32__ */
 | 
						|
 | 
						|
	/* WINSOCK initialization */	
 | 
						|
	wVersionRequested = MAKEWORD(2, 0); /* Request Winsock v2.0 */
 | 
						|
	if (WSAStartup(wVersionRequested, &wsaData) != 0) /* Load Winsock DLL */ {
 | 
						|
	    fprintf(stderr,"WSAStartup() failed");
 | 
						|
	    exit(1);
 | 
						|
	}
 | 
						|
	/* End WINSOCK initialization */
 | 
						|
	
 | 
						|
 | 
						|
	a=0;
 | 
						|
	r=GetIpAddrTable(iptab, &a, TRUE);
 | 
						|
	iptab=malloc(a);
 | 
						|
	r=GetIpAddrTable(iptab, &a, TRUE);
 | 
						|
 | 
						|
	a=0;
 | 
						|
	r=GetIfTable(iftab, &a, TRUE);
 | 
						|
	iftab=malloc(a);
 | 
						|
	r=GetIfTable(iftab, &a, TRUE);
 | 
						|
 | 
						|
	if(wanted && !strncmp(wanted, "eth", 3) && wanted[3]) {
 | 
						|
	    char *ptr;
 | 
						|
	    int n = strtoul(wanted+3, &ptr, 10);
 | 
						|
	    if(!*ptr)
 | 
						|
		wantedEtherNo=n;
 | 
						|
	}
 | 
						|
 | 
						|
	for(i=0; i<iptab->dwNumEntries; i++) {
 | 
						|
	    int goodness=-1;
 | 
						|
	    unsigned long iaddr;
 | 
						|
	    int isEther=0;
 | 
						|
	    MIB_IFROW *ifrow;
 | 
						|
 | 
						|
	    iprow = &iptab->table[i];
 | 
						|
	    iaddr = iprow->dwAddr;
 | 
						|
 | 
						|
	    ifrow = getIfRow(iftab, iprow->dwIndex);
 | 
						|
 | 
						|
	    if(ifrow && ifrow->dwPhysAddrLen == 6 && iprow->dwBCastAddr) {
 | 
						|
		isEther=1;
 | 
						|
		etherNo++;
 | 
						|
	    }
 | 
						|
 | 
						|
	    if(wanted) {
 | 
						|
		if(isAddress && iaddr == wantedAddress.s_addr) {
 | 
						|
		    goodness=8;
 | 
						|
		} else if(isEther && wantedEtherNo == etherNo) {
 | 
						|
			goodness=9;
 | 
						|
		} else if(ifrow->dwPhysAddrLen) {
 | 
						|
		    int j;
 | 
						|
		    const char *ptr=wanted;
 | 
						|
		    for(j=0; *ptr && j<ifrow->dwPhysAddrLen; j++) {
 | 
						|
			int digit = strtoul(ptr, (char**)&ptr, 16);
 | 
						|
			if(digit != ifrow->bPhysAddr[j])
 | 
						|
			    break; /* Digit mismatch */
 | 
						|
			if(*ptr == '-' || *ptr == ':') {
 | 
						|
			    ptr++;
 | 
						|
			}
 | 
						|
		    }
 | 
						|
		    if(!*ptr && j == ifrow->dwPhysAddrLen) {
 | 
						|
			goodness=9;
 | 
						|
		    }
 | 
						|
		}
 | 
						|
	    } else {
 | 
						|
		if(iaddr == 0) {
 | 
						|
		    /* disregard interfaces whose address is zero */
 | 
						|
		    goodness = 1;
 | 
						|
		} else if(iaddr == htonl(0x7f000001)) {
 | 
						|
		    /* disregard localhost type devices */
 | 
						|
		    goodness = 2;
 | 
						|
		} else if(isEther) {
 | 
						|
		    /* prefer ethernet */
 | 
						|
		    goodness = 6;
 | 
						|
		} else if(ifrow->dwPhysAddrLen) {
 | 
						|
		    /* then prefer interfaces which have a physical address */
 | 
						|
		    goodness = 4;
 | 
						|
		} else {
 | 
						|
		    goodness = 3;
 | 
						|
		}
 | 
						|
	    }
 | 
						|
 | 
						|
	    goodness = goodness * 2;
 | 
						|
	    /* If all else is the same, prefer interfaces that 
 | 
						|
	     * have broadcast */
 | 
						|
	    if(goodness >= lastGoodness) {
 | 
						|
		/* Privilege broadcast-enabled interfaces */
 | 
						|
		if(iprow->dwBCastAddr)
 | 
						|
		    goodness++;
 | 
						|
	    }
 | 
						|
 | 
						|
	    if(goodness > lastGoodness) {
 | 
						|
		chosen = iprow;
 | 
						|
		chosenIf = ifrow;
 | 
						|
		lastGoodness = goodness;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
	
 | 
						|
	if(!chosen) {
 | 
						|
	    fprintf(stderr, "No suitable network interface found%s%s\n", 
 | 
						|
		    wanted ? " for " : "", wanted ? wanted : "");
 | 
						|
	    fprintf(stderr, "The following interfaces are available:\n");
 | 
						|
 | 
						|
	    for(i=0; i<iptab->dwNumEntries; i++) {
 | 
						|
		char buffer[16];
 | 
						|
		struct sockaddr_in addr;
 | 
						|
		MIB_IFROW *ifrow;
 | 
						|
		char *name=NULL;
 | 
						|
		iprow = &iptab->table[i];
 | 
						|
		addr.sin_addr.s_addr = iprow->dwAddr;
 | 
						|
		ifrow = getIfRow(iftab, iprow->dwIndex);
 | 
						|
		name = fmtName(ifrow);
 | 
						|
		fprintf(stderr, " %15s  %s\n",
 | 
						|
			udpc_getIpString(&addr, buffer),
 | 
						|
			name ? name : "");
 | 
						|
		if(name)
 | 
						|
		    free(name);
 | 
						|
	    }
 | 
						|
	    exit(1);
 | 
						|
	}
 | 
						|
 | 
						|
	net_if->bcast.s_addr = net_if->addr.s_addr = chosen->dwAddr;
 | 
						|
	if(chosen->dwBCastAddr)
 | 
						|
	    net_if->bcast.s_addr |= ~chosen->dwMask;
 | 
						|
	if(chosenIf) {
 | 
						|
	    net_if->name = fmtName(chosenIf);
 | 
						|
	} else {
 | 
						|
	    net_if->name = "*";
 | 
						|
	}
 | 
						|
	free(iftab);
 | 
						|
	free(iptab);
 | 
						|
#endif /* __MINGW32__ */
 | 
						|
 | 
						|
	return net_if;
 | 
						|
}
 | 
						|
 | 
						|
#define IP_MREQN ip_mreqn
 | 
						|
 | 
						|
int fillMreq(net_if_t *net_if, struct in_addr addr,
 | 
						|
		    struct IP_MREQN *mreq) {
 | 
						|
#ifdef HAVE_STRUCT_IP_MREQN_IMR_IFINDEX
 | 
						|
    mreq->imr_ifindex = net_if->index;
 | 
						|
    mreq->imr_address.s_addr = 0;
 | 
						|
#else
 | 
						|
    mreq->imr_interface = net_if->addr;
 | 
						|
#endif
 | 
						|
    mreq->imr_multiaddr = addr;
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int mcastOp(int sock, net_if_t *net_if, struct in_addr addr,
 | 
						|
		   int code, const char *message) {
 | 
						|
    struct IP_MREQN mreq;
 | 
						|
    int r;
 | 
						|
    
 | 
						|
    fillMreq(net_if, addr, &mreq);
 | 
						|
    r = setsockopt(sock, SOL_IP, code, (char*)&mreq, sizeof(mreq));
 | 
						|
    if(r < 0) {
 | 
						|
	perror(message);
 | 
						|
	exit(1);
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_setMcastDestination(int sock, net_if_t *net_if, struct sockaddr_in *addr) {
 | 
						|
#ifdef WINDOWS
 | 
						|
    int r;
 | 
						|
    struct sockaddr_in interface_addr;
 | 
						|
    struct in_addr if_addr;
 | 
						|
    getMyAddress(net_if, &interface_addr);
 | 
						|
    if_addr = getSinAddr(&interface_addr);
 | 
						|
    r = setsockopt (sock, IPPROTO_IP, IP_MULTICAST_IF, 
 | 
						|
		    (char *) &if_addr, sizeof(if_addr));
 | 
						|
    if(r < 0)
 | 
						|
	fatal(1, "Set multicast send interface");
 | 
						|
    return 0;
 | 
						|
#else
 | 
						|
    /* IP_MULTICAST_IF not correctly supported on Cygwin */
 | 
						|
    return mcastOp(sock, net_if, getSinAddr(addr), IP_MULTICAST_IF,
 | 
						|
		   "Set multicast send interface");
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
int initSockAddress(addr_type_t addr_type,
 | 
						|
			   net_if_t *net_if, 
 | 
						|
			   in_addr_t ip,
 | 
						|
			   unsigned short port, 
 | 
						|
			   struct sockaddr_in *addr) {
 | 
						|
    memset ((char *) addr, 0, sizeof(struct sockaddr_in));
 | 
						|
    addr->sin_family = AF_INET;
 | 
						|
    addr->sin_port = htons(port);
 | 
						|
 | 
						|
	//TODO: FIXME
 | 
						|
    //if(!net_if && addr_type != ADDR_TYPE_MCAST)
 | 
						|
	//udpc_fatal(1, "initSockAddr without ifname\n");
 | 
						|
 | 
						|
    switch(addr_type) {
 | 
						|
    case ADDR_TYPE_UCAST:
 | 
						|
	addr->sin_addr = net_if->addr;
 | 
						|
	break;
 | 
						|
    case ADDR_TYPE_BCAST:
 | 
						|
	addr->sin_addr = net_if->bcast;
 | 
						|
	break;
 | 
						|
    case ADDR_TYPE_MCAST:
 | 
						|
	addr->sin_addr.s_addr = ip;
 | 
						|
	break;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int mcastListen(int sock, net_if_t *net_if, struct sockaddr_in *addr) {
 | 
						|
    return mcastOp(sock, net_if, getSinAddr(addr), IP_ADD_MEMBERSHIP,
 | 
						|
		   "Subscribe to multicast group");
 | 
						|
}
 | 
						|
 | 
						|
int udpc_makeSocket(addr_type_t addr_type, 
 | 
						|
	       net_if_t *net_if, 
 | 
						|
	       struct sockaddr_in *tmpl,
 | 
						|
	       int port) {
 | 
						|
    int ret, s;
 | 
						|
    struct sockaddr_in myaddr;
 | 
						|
    in_addr_t ip=0;
 | 
						|
 | 
						|
#ifdef WINDOWS
 | 
						|
    static int lastSocket=-1;
 | 
						|
    /* Very ugly hack, but hey!, this is for Windows */
 | 
						|
 | 
						|
    if(addr_type == ADDR_TYPE_MCAST) {
 | 
						|
	mcastListen(lastSocket, net_if, tmpl);
 | 
						|
	return -1;
 | 
						|
    } else if(addr_type != ADDR_TYPE_UCAST)
 | 
						|
	return -1;
 | 
						|
#endif
 | 
						|
 | 
						|
    s = socket(PF_INET, SOCK_DGRAM, 0);
 | 
						|
    if (s < 0) {
 | 
						|
	perror("make socket");
 | 
						|
	exit(1);
 | 
						|
    }
 | 
						|
 | 
						|
    if(addr_type == ADDR_TYPE_MCAST && tmpl != NULL) {
 | 
						|
	ip = tmpl->sin_addr.s_addr;
 | 
						|
    }
 | 
						|
 | 
						|
    ret = initSockAddress(addr_type, net_if, ip, port, &myaddr);
 | 
						|
	//TODO: FIXME
 | 
						|
    //if(ret < 0)
 | 
						|
//	udpc_fatal(1, "Could not get socket address fot %d/%s", 
 | 
						|
//		   addr_type, net_if->name);
 | 
						|
if (addr_type == ADDR_TYPE_BCAST)
 | 
						|
{
 | 
						|
//cerr << "for addr_type == ADDR_TYPE_BCAST, myaddr.sin_addr.s_addr = 0x" << hex << myaddr.sin_addr.s_addr << dec << endl;
 | 
						|
}
 | 
						|
    if(addr_type == ADDR_TYPE_BCAST && myaddr.sin_addr.s_addr == 0) {
 | 
						|
      /* Attempting to bind to broadcast address on not-broadcast media ... */
 | 
						|
      closesocket(s);
 | 
						|
      return -1;
 | 
						|
    }
 | 
						|
    ret = bind(s, (struct sockaddr *) &myaddr, sizeof(myaddr));
 | 
						|
	//TODO: FIXME
 | 
						|
//    if (ret < 0) {
 | 
						|
//	char buffer[16];
 | 
						|
//	udpc_fatal(1, "bind socket to %s:%d (%s)\n",
 | 
						|
//		   udpc_getIpString(&myaddr, buffer), 
 | 
						|
//		   udpc_getPort(&myaddr),
 | 
						|
//		   strerror(errno));
 | 
						|
//    }
 | 
						|
 | 
						|
    if(addr_type == ADDR_TYPE_MCAST)
 | 
						|
	mcastListen(s, net_if, &myaddr);
 | 
						|
#ifdef WINDOWS
 | 
						|
    lastSocket=s;
 | 
						|
#endif
 | 
						|
    return s;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_setSocketToBroadcast(int sock) {
 | 
						|
    /* set the socket to broadcast */
 | 
						|
    int p = 1;
 | 
						|
    return setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (char *)&p, sizeof(int));
 | 
						|
}
 | 
						|
 | 
						|
int udpc_getBroadCastAddress(net_if_t *net_if, struct sockaddr_in *addr, 
 | 
						|
			short port){
 | 
						|
    int r= initSockAddress(ADDR_TYPE_BCAST, net_if, INADDR_ANY, port, addr);
 | 
						|
    if(addr->sin_addr.s_addr == 0) {
 | 
						|
      /* Quick hack to make it work for loopback */
 | 
						|
      struct sockaddr_in ucast;
 | 
						|
      initSockAddress(ADDR_TYPE_UCAST, net_if, INADDR_ANY, port, &ucast);
 | 
						|
 | 
						|
      if((ntohl(ucast.sin_addr.s_addr) & 0xff000000) == 0x7f000000)
 | 
						|
	addr->sin_addr.s_addr = ucast.sin_addr.s_addr;
 | 
						|
    }
 | 
						|
    return r;
 | 
						|
}
 | 
						|
 | 
						|
int safe_inet_aton(const char *address, struct in_addr *ip) {
 | 
						|
    if(!INET_ATON(address, ip))
 | 
						|
	{
 | 
						|
	//TODO: FIXME
 | 
						|
	//udpc_fatal(-1, "Bad address %s", address);
 | 
						|
	}
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_getMcastAllAddress(struct sockaddr_in *addr, const char *address,
 | 
						|
		       short port){
 | 
						|
    struct in_addr ip;
 | 
						|
    int ret;
 | 
						|
 | 
						|
    if(address == NULL || address[0] == '\0')
 | 
						|
	safe_inet_aton("224.0.0.1", &ip);
 | 
						|
    else {
 | 
						|
	if((ret=safe_inet_aton(address, &ip))<0)
 | 
						|
	   return ret;
 | 
						|
    }
 | 
						|
    return initSockAddress(ADDR_TYPE_MCAST, NULL, ip.s_addr, port, addr);
 | 
						|
}
 | 
						|
 | 
						|
void setPort(struct sockaddr_in *addr, unsigned short port) {
 | 
						|
    ((struct sockaddr_in *) addr)->sin_port = htons(port);
 | 
						|
}
 | 
						|
 | 
						|
int isMcastAddress(struct sockaddr_in *addr) {
 | 
						|
    int ip = ntohl(addr->sin_addr.s_addr) >> 24;
 | 
						|
    return ip >= 0xe0 && ip < 0xf0;
 | 
						|
}
 | 
						|
 | 
						|
void udpc_clearIp(struct sockaddr_in *addr) {
 | 
						|
    addr->sin_addr.s_addr = 0;
 | 
						|
    addr->sin_family = AF_INET;
 | 
						|
}
 | 
						|
 | 
						|
void udpc_setSendBuf(int sock, unsigned int bufsize) {
 | 
						|
    if(setsockopt(sock, SOL_SOCKET, SO_SNDBUF, (char*)&bufsize, sizeof(bufsize))< 0)
 | 
						|
	perror("Set send buffer");
 | 
						|
}
 | 
						|
 | 
						|
int udpc_setTtl(int sock, int ttl) {
 | 
						|
    /* set the socket to broadcast */
 | 
						|
    return setsockopt(sock, SOL_IP, IP_MULTICAST_TTL, (char*)&ttl, sizeof(int));
 | 
						|
}
 | 
						|
 | 
						|
int udpc_getMyAddress(net_if_t *net_if, struct sockaddr_in *addr) {
 | 
						|
    return initSockAddress(ADDR_TYPE_UCAST, net_if, INADDR_ANY, 0, addr);
 | 
						|
}
 | 
						|
 | 
						|
void udpc_getDefaultMcastAddress(net_if_t *net_if, struct sockaddr_in *mcast) {
 | 
						|
    udpc_getMyAddress(net_if, mcast);
 | 
						|
    mcast->sin_addr.s_addr &= htonl(0x07ffffff);
 | 
						|
    mcast->sin_addr.s_addr |= htonl(0xe8000000);
 | 
						|
}
 | 
						|
 | 
						|
void udpc_copyIpFrom(struct sockaddr_in *dst, struct sockaddr_in *src) {
 | 
						|
    dst->sin_addr = src->sin_addr;
 | 
						|
    dst->sin_family = src->sin_family;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_getSelectedSock(int *socks, int nr, fd_set *read_set) {
 | 
						|
    int i;
 | 
						|
    int maxFd;
 | 
						|
    maxFd=-1;
 | 
						|
    for(i=0; i<nr; i++) {
 | 
						|
	if(socks[i] == -1)
 | 
						|
	    continue;
 | 
						|
	if(FD_ISSET(socks[i], read_set))
 | 
						|
	    return socks[i];
 | 
						|
    }
 | 
						|
    return -1;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_ipIsEqual(struct sockaddr_in *left, struct sockaddr_in *right) {
 | 
						|
    return getSinAddr(left).s_addr == getSinAddr(right).s_addr;
 | 
						|
}
 | 
						|
 | 
						|
void udpc_closeSock(int *socks, int nr, int target) {
 | 
						|
    int i;
 | 
						|
    int sock = socks[target];
 | 
						|
 | 
						|
    socks[target]=-1;
 | 
						|
    for(i=0; i<nr; i++)
 | 
						|
	if(socks[i] == sock)
 | 
						|
	    return;
 | 
						|
    closesocket(sock);
 | 
						|
}
 | 
						|
int prepareForSelect(int *socks, int nr, fd_set *read_set) {
 | 
						|
    int i;
 | 
						|
    int maxFd;
 | 
						|
    FD_ZERO(read_set);
 | 
						|
    maxFd=-1;
 | 
						|
    for(i=0; i<nr; i++) {
 | 
						|
	if(socks[i] == -1)
 | 
						|
	    continue;
 | 
						|
	FD_SET(socks[i], read_set);
 | 
						|
	if(socks[i] > maxFd)
 | 
						|
	    maxFd = socks[i];
 | 
						|
    }
 | 
						|
    return maxFd;
 | 
						|
}
 | 
						|
 | 
						|
int doReceive(int s, void *message, size_t len, 
 | 
						|
	      struct sockaddr_in *from, int portBase) {
 | 
						|
    socklen_t slen;
 | 
						|
    int r;
 | 
						|
    unsigned short port;
 | 
						|
 | 
						|
    slen = sizeof(*from);
 | 
						|
#ifdef LOSSTEST
 | 
						|
    loseRecvPacket(s);
 | 
						|
#endif
 | 
						|
    r = recvfrom(s, message, len, 0, (struct sockaddr *)from, &slen);
 | 
						|
    if (r < 0)
 | 
						|
	return r;
 | 
						|
    port = ntohs(from->sin_port);
 | 
						|
    if(port != RECEIVER_PORT(portBase) && port != SENDER_PORT(portBase)) {
 | 
						|
	return -1;
 | 
						|
    }
 | 
						|
/*    flprintf("recv: %08x %d\n", *(int*) message, r);*/
 | 
						|
    return r;
 | 
						|
}
 | 
						|
 | 
						|
participantsDb_t udpc_makeParticipantsDb(void)
 | 
						|
{
 | 
						|
    return MALLOC(struct participantsDb);
 | 
						|
}
 | 
						|
 | 
						|
int udpc_removeParticipant(struct participantsDb *db, int i) {
 | 
						|
    if(db->clientTable[i].used) {
 | 
						|
	db->clientTable[i].used = 0;
 | 
						|
	db->nrParticipants--;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_lookupParticipant(struct participantsDb *db, struct sockaddr_in *addr) {
 | 
						|
    int i;
 | 
						|
    for (i=0; i < MAX_CLIENTS; i++) {
 | 
						|
	if (db->clientTable[i].used && 
 | 
						|
	    udpc_ipIsEqual(&db->clientTable[i].addr, addr)) {
 | 
						|
	    return i;
 | 
						|
	}
 | 
						|
    }
 | 
						|
    return -1;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_nrParticipants(participantsDb_t db) {
 | 
						|
    return db->nrParticipants;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_addParticipant(participantsDb_t db,
 | 
						|
		   struct sockaddr_in *addr, 
 | 
						|
		   int capabilities,
 | 
						|
		   unsigned int rcvbuf,
 | 
						|
		   int pointopoint) {
 | 
						|
//cerr << "adding a participant..." << endl;
 | 
						|
    int i;
 | 
						|
 | 
						|
    if((i = udpc_lookupParticipant(db, addr)) >= 0)
 | 
						|
	return i;
 | 
						|
 | 
						|
    for (i=0; i < MAX_CLIENTS; i++) {
 | 
						|
	if (!db->clientTable[i].used) {
 | 
						|
	    db->clientTable[i].addr = *addr;
 | 
						|
	    db->clientTable[i].used = 1;
 | 
						|
	    db->clientTable[i].capabilities = capabilities;
 | 
						|
	    db->clientTable[i].rcvbuf = rcvbuf;
 | 
						|
	    db->nrParticipants++;
 | 
						|
 | 
						|
	    return i;
 | 
						|
	} else if(pointopoint)
 | 
						|
	    return -1;
 | 
						|
    }
 | 
						|
 | 
						|
    return -1; /* no space left in participant's table */
 | 
						|
}
 | 
						|
 | 
						|
int selectWithoutConsole(int maxFd, 
 | 
						|
		      fd_set *read_set, struct timeval *tv) {
 | 
						|
    int ret;
 | 
						|
 | 
						|
    ret = select(maxFd, read_set, NULL, NULL, tv);
 | 
						|
    if(ret < 0)
 | 
						|
	return -1;
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
#if 0
 | 
						|
void sendHello(struct net_config *net_config, int sock,
 | 
						|
	       int streaming) {
 | 
						|
    struct hello hello;
 | 
						|
    /* send hello message */
 | 
						|
    if(streaming)
 | 
						|
	hello.opCode = htons(CMD_HELLO_STREAMING);
 | 
						|
    else
 | 
						|
	hello.opCode = htons(CMD_HELLO);
 | 
						|
    hello.reserved = 0;
 | 
						|
    hello.capabilities = htonl(net_config->capabilities);
 | 
						|
    udpc_copyToMessage(hello.mcastAddr,&net_config->dataMcastAddr);
 | 
						|
    hello.blockSize = htons(net_config->blockSize);
 | 
						|
    //rgWaitAll(net_config, sock, net_config->controlMcastAddr.sin_addr.s_addr, sizeof(hello));
 | 
						|
    BCAST_CONTROL(sock, hello);
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
int checkClientWait(participantsDb_t db, 
 | 
						|
			   struct net_config *net_config,
 | 
						|
			   time_t *firstConnected)
 | 
						|
{
 | 
						|
    time_t now;
 | 
						|
    if (!udpc_nrParticipants(db) || !firstConnected || !*firstConnected)
 | 
						|
	return 0; /* do not start: no receivers */
 | 
						|
 | 
						|
    now = time(0);
 | 
						|
    /*
 | 
						|
     * If we have a max_client_wait, start the transfer after first client
 | 
						|
     * connected + maxSendWait
 | 
						|
     */
 | 
						|
    if(net_config->max_receivers_wait &&
 | 
						|
       (now >= *firstConnected + net_config->max_receivers_wait)) {
 | 
						|
#ifdef USE_SYSLOG
 | 
						|
	    syslog(LOG_INFO, "max wait[%d] passed: starting", 
 | 
						|
			    net_config->max_receivers_wait );
 | 
						|
#endif
 | 
						|
	return 1; /* send-wait passed: start */
 | 
						|
    }
 | 
						|
 | 
						|
    /*
 | 
						|
     * Otherwise check to see if the minimum of clients
 | 
						|
     *  have checked in.
 | 
						|
     */
 | 
						|
    else if (udpc_nrParticipants(db) >= net_config->min_receivers &&
 | 
						|
	/*
 | 
						|
	 *  If there are enough clients and there's a min wait time, we'll
 | 
						|
	 *  wait around anyway until then.
 | 
						|
	 *  Otherwise, we always transfer
 | 
						|
	 */
 | 
						|
	(!net_config->min_receivers_wait || 
 | 
						|
	 now >= *firstConnected + net_config->min_receivers_wait)) {
 | 
						|
#ifdef USE_SYSLOG
 | 
						|
	    syslog(LOG_INFO, "min receivers[%d] reached: starting", 
 | 
						|
			    net_config->min_receivers );
 | 
						|
#endif
 | 
						|
	    return 1;
 | 
						|
    } else
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
int sendConnectionReply(participantsDb_t db,
 | 
						|
			       int sock,
 | 
						|
			       struct net_config *config,
 | 
						|
			       struct sockaddr_in *client, 
 | 
						|
			       int capabilities,
 | 
						|
			       unsigned int rcvbuf) {
 | 
						|
    struct connectReply reply;
 | 
						|
 | 
						|
    if(rcvbuf == 0)
 | 
						|
	rcvbuf = 65536;
 | 
						|
 | 
						|
    if(capabilities & CAP_BIG_ENDIAN) {
 | 
						|
	reply.opCode = htons(CMD_CONNECT_REPLY);
 | 
						|
	reply.clNr = 
 | 
						|
	    htonl(udpc_addParticipant(db,
 | 
						|
				      client, 
 | 
						|
				      capabilities,
 | 
						|
				      rcvbuf,
 | 
						|
				      config->flags & FLAG_POINTOPOINT));
 | 
						|
	reply.blockSize = htonl(config->blockSize);
 | 
						|
    } else {
 | 
						|
	//TODO: FIXME
 | 
						|
	//udpc_fatal(1, "Little endian protocol no longer supported");
 | 
						|
    }
 | 
						|
    reply.reserved = 0;
 | 
						|
 | 
						|
    if(config->flags & FLAG_POINTOPOINT) {
 | 
						|
	udpc_copyIpFrom(&config->dataMcastAddr, client);
 | 
						|
    }
 | 
						|
 | 
						|
    /* new parameters: always big endian */
 | 
						|
    reply.capabilities = ntohl(config->capabilities);
 | 
						|
    udpc_copyToMessage(reply.mcastAddr,&config->dataMcastAddr);
 | 
						|
    /*reply.mcastAddress = mcastAddress;*/
 | 
						|
    //rgWaitAll(config, sock, client->sin_addr.s_addr, sizeof(reply));
 | 
						|
    if(SEND(sock, reply, *client) < 0) {
 | 
						|
	perror("reply add new client");
 | 
						|
	return -1;
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int mainDispatcher(int *fd, int nr,
 | 
						|
			  participantsDb_t db,
 | 
						|
			  struct net_config *net_config,
 | 
						|
			  int *tries,
 | 
						|
			  time_t *firstConnected)
 | 
						|
{
 | 
						|
    struct sockaddr_in client;
 | 
						|
    union message fromClient;
 | 
						|
    fd_set read_set;
 | 
						|
    int ret;
 | 
						|
    int msgLength;
 | 
						|
    int startNow=0;
 | 
						|
    int selected;
 | 
						|
 | 
						|
    if (firstConnected && !*firstConnected && udpc_nrParticipants(db)) {
 | 
						|
	*firstConnected = time(0);
 | 
						|
#ifdef USE_SYSLOG
 | 
						|
        syslog(LOG_INFO,
 | 
						|
	 "first connection: min wait[%d] secs - max wait[%d] - min clients[%d]",
 | 
						|
	  net_config->min_receivers_wait, net_config->max_receivers_wait, 
 | 
						|
	  net_config->min_receivers );
 | 
						|
#endif
 | 
						|
    }
 | 
						|
 | 
						|
    while(!startNow) {
 | 
						|
	struct timeval tv;
 | 
						|
	struct timeval *tvp;
 | 
						|
	int nr_desc;
 | 
						|
 | 
						|
	int maxFd = prepareForSelect(fd, nr, &read_set);
 | 
						|
 | 
						|
	if(net_config->rexmit_hello_interval) {
 | 
						|
	    tv.tv_usec = (net_config->rexmit_hello_interval % 1000)*1000;
 | 
						|
	    tv.tv_sec = net_config->rexmit_hello_interval / 1000;
 | 
						|
	    tvp = &tv;
 | 
						|
	} else if(firstConnected && udpc_nrParticipants(db)) {
 | 
						|
	    tv.tv_usec = 0;
 | 
						|
	    tv.tv_sec = 2;
 | 
						|
	    tvp = &tv;
 | 
						|
	} else
 | 
						|
	    tvp = 0;
 | 
						|
	nr_desc = selectWithoutConsole(maxFd+1, &read_set, tvp);
 | 
						|
	if(nr_desc < 0) {
 | 
						|
	    perror("select");
 | 
						|
	    return -1;
 | 
						|
	}
 | 
						|
	if(nr_desc > 0)
 | 
						|
	    /* key pressed, or receiver activity */
 | 
						|
	    break;
 | 
						|
 | 
						|
	if(net_config->rexmit_hello_interval) {
 | 
						|
	    /* retransmit hello message */
 | 
						|
	    udpc_sendHello(net_config, fd[0], 0);
 | 
						|
	    (*tries)++;
 | 
						|
	    if(net_config->autostart != 0 && *tries > net_config->autostart)
 | 
						|
		startNow=1;
 | 
						|
	}
 | 
						|
 | 
						|
	if(firstConnected)
 | 
						|
	    startNow = 
 | 
						|
		startNow || checkClientWait(db, net_config, firstConnected);
 | 
						|
    }
 | 
						|
 | 
						|
    selected = udpc_getSelectedSock(fd, nr, &read_set);
 | 
						|
    if(selected == -1)
 | 
						|
	return startNow;
 | 
						|
 | 
						|
    BZERO(fromClient); /* Zero it out in order to cope with short messages
 | 
						|
			* from older versions */
 | 
						|
 | 
						|
    msgLength = RECV(selected, fromClient, client, net_config->portBase);
 | 
						|
    if(msgLength < 0) {
 | 
						|
	perror("problem getting data from client");
 | 
						|
	return 0; /* don't panic if we get weird messages */
 | 
						|
    }
 | 
						|
 | 
						|
    if(net_config->flags & FLAG_ASYNC)
 | 
						|
	return 0;
 | 
						|
 | 
						|
    switch(ntohs(fromClient.opCode)) {
 | 
						|
	case CMD_CONNECT_REQ:
 | 
						|
	    sendConnectionReply(db, fd[0],
 | 
						|
				net_config,
 | 
						|
				&client, 
 | 
						|
				CAP_BIG_ENDIAN |
 | 
						|
				ntohl(fromClient.connectReq.capabilities),
 | 
						|
				ntohl(fromClient.connectReq.rcvbuf));
 | 
						|
	    return startNow;
 | 
						|
	case CMD_GO:
 | 
						|
	    return 1;
 | 
						|
	case CMD_DISCONNECT:
 | 
						|
	    ret = udpc_lookupParticipant(db, &client);
 | 
						|
	    if (ret >= 0)
 | 
						|
		udpc_removeParticipant(db, ret);
 | 
						|
	    return startNow;
 | 
						|
	default:
 | 
						|
	    break;
 | 
						|
    }
 | 
						|
 | 
						|
    return startNow;
 | 
						|
}
 | 
						|
 | 
						|
static int isPointToPoint(participantsDb_t db, int flags) {
 | 
						|
    if(flags & FLAG_POINTOPOINT)
 | 
						|
	return 1;
 | 
						|
    if(flags & (FLAG_NOPOINTOPOINT | FLAG_ASYNC))
 | 
						|
	return 0;
 | 
						|
    return udpc_nrParticipants(db) == 1;
 | 
						|
}
 | 
						|
 | 
						|
int getProducedAmount(produconsum_t pc) {
 | 
						|
    unsigned int produced = pc->produced;
 | 
						|
    unsigned int consumed = pc->consumed;
 | 
						|
    if(produced < consumed)
 | 
						|
	return produced + 2 * pc->size - consumed;
 | 
						|
    else
 | 
						|
	return produced - consumed;
 | 
						|
}
 | 
						|
 | 
						|
int _consumeAny(produconsum_t pc, unsigned int minAmount,
 | 
						|
		       struct timespec *ts) {
 | 
						|
    unsigned int amount;
 | 
						|
#if DEBUG
 | 
						|
    flprintf("%s: Waiting for %d bytes (%d:%d)\n", 
 | 
						|
	    pc->name, minAmount, pc->consumed, pc->produced);
 | 
						|
#endif
 | 
						|
    pc->consumerIsWaiting=1;
 | 
						|
    amount = getProducedAmount(pc);
 | 
						|
    if(amount >= minAmount || pc->atEnd) {	
 | 
						|
	pc->consumerIsWaiting=0;
 | 
						|
#if DEBUG
 | 
						|
	flprintf("%s: got %d bytes\n",pc->name, amount);
 | 
						|
#endif
 | 
						|
	return amount;
 | 
						|
    }
 | 
						|
    pthread_mutex_lock(&pc->mutex);
 | 
						|
    while((amount=getProducedAmount(pc)) < minAmount && !pc->atEnd) {
 | 
						|
#if DEBUG
 | 
						|
	flprintf("%s: ..Waiting for %d bytes (%d:%d)\n", 
 | 
						|
		pc->name, minAmount, pc->consumed, pc->produced);
 | 
						|
#endif
 | 
						|
	if(ts == 0)
 | 
						|
	    pthread_cond_wait(&pc->cond, &pc->mutex);
 | 
						|
	else {
 | 
						|
	    int r;
 | 
						|
#if DEBUG
 | 
						|
	    flprintf("Before timed wait\n");
 | 
						|
#endif
 | 
						|
	    r=pthread_cond_timedwait(&pc->cond, &pc->mutex, ts);
 | 
						|
#if DEBUG
 | 
						|
	    flprintf("After timed wait %d\n", r);
 | 
						|
#endif
 | 
						|
	    if(r == ETIMEDOUT) {
 | 
						|
		amount=getProducedAmount(pc);
 | 
						|
		break;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
    }
 | 
						|
    pthread_mutex_unlock(&pc->mutex);
 | 
						|
#if DEBUG
 | 
						|
    flprintf("%s: Got them %d (for %d) %d\n", pc->name, 
 | 
						|
	    amount, minAmount, pc->atEnd);
 | 
						|
#endif
 | 
						|
    pc->consumerIsWaiting=0;
 | 
						|
    return amount;
 | 
						|
}
 | 
						|
 | 
						|
produconsum_t pc_makeProduconsum(int size, const char *name)
 | 
						|
{
 | 
						|
    produconsum_t pc = MALLOC(struct produconsum);
 | 
						|
    pc->size = size;
 | 
						|
    pc->produced = 0;
 | 
						|
    pc->consumed = 0;
 | 
						|
    pc->atEnd = 0;
 | 
						|
    pthread_mutex_init(&pc->mutex, NULL);
 | 
						|
    pc->consumerIsWaiting = 0;
 | 
						|
    pthread_cond_init(&pc->cond, NULL);
 | 
						|
    pc->name = name;
 | 
						|
    return pc;
 | 
						|
}
 | 
						|
 | 
						|
int pc_consumeAnyWithTimeout(produconsum_t pc, struct timespec *ts)
 | 
						|
{
 | 
						|
    return _consumeAny(pc, 1, ts);
 | 
						|
}
 | 
						|
 | 
						|
unsigned int pc_getProducerPosition(produconsum_t pc)
 | 
						|
{
 | 
						|
    return pc->produced % pc->size;
 | 
						|
}
 | 
						|
 | 
						|
unsigned int pc_getWaiting(produconsum_t pc)
 | 
						|
{
 | 
						|
    return getProducedAmount(pc);
 | 
						|
}
 | 
						|
 | 
						|
int pc_consumeAny(produconsum_t pc)
 | 
						|
{
 | 
						|
    return _consumeAny(pc, 1, 0);
 | 
						|
}
 | 
						|
 | 
						|
int pc_consume(produconsum_t pc, int amount)
 | 
						|
{
 | 
						|
    return _consumeAny(pc, amount, 0);
 | 
						|
}
 | 
						|
 | 
						|
void wakeConsumer(produconsum_t pc)
 | 
						|
{
 | 
						|
    if(pc->consumerIsWaiting) {
 | 
						|
	pthread_mutex_lock(&pc->mutex);
 | 
						|
	pthread_cond_signal(&pc->cond);
 | 
						|
	pthread_mutex_unlock(&pc->mutex);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void pc_produceEnd(produconsum_t pc)
 | 
						|
{
 | 
						|
    pc->atEnd = 1;
 | 
						|
    wakeConsumer(pc);
 | 
						|
}
 | 
						|
 | 
						|
int pc_consumed(produconsum_t pc, int amount)
 | 
						|
{
 | 
						|
    unsigned int consumed = pc->consumed;
 | 
						|
    if(consumed >= 2*pc->size - amount) {
 | 
						|
	consumed += amount - 2 *pc->size;
 | 
						|
    } else {
 | 
						|
	consumed += amount;
 | 
						|
    }
 | 
						|
    pc->consumed = consumed;
 | 
						|
    return amount;
 | 
						|
}
 | 
						|
 | 
						|
unsigned int pc_getConsumerPosition(produconsum_t pc)
 | 
						|
{
 | 
						|
    return pc->consumed % pc->size;
 | 
						|
}
 | 
						|
 | 
						|
void pc_produce(produconsum_t pc, unsigned int amount)
 | 
						|
{
 | 
						|
    unsigned int produced = pc->produced;
 | 
						|
    unsigned int consumed = pc->consumed;
 | 
						|
 | 
						|
    /* sanity checks:
 | 
						|
     * 1. should not produce more than size
 | 
						|
     * 2. do not pass consumed+size
 | 
						|
     */
 | 
						|
    if(amount > pc->size) {
 | 
						|
	//TODO: FIXME
 | 
						|
	//udpc_fatal(1, "Buffer overflow in produce %s: %d > %d \n", pc->name, amount, pc->size);
 | 
						|
    }
 | 
						|
 | 
						|
    produced += amount;
 | 
						|
    if(produced >= 2*pc->size)
 | 
						|
	produced -= 2*pc->size;
 | 
						|
 | 
						|
    if(produced > consumed + pc->size ||
 | 
						|
       (produced < consumed && produced > consumed - pc->size)) {
 | 
						|
	//TODO: FIXME
 | 
						|
	//udpc_fatal(1, "Buffer overflow in produce %s: %d > %d [%d] \n", pc->name, produced, consumed, pc->size);
 | 
						|
    }
 | 
						|
 | 
						|
    pc->produced = produced;
 | 
						|
    wakeConsumer(pc);
 | 
						|
}
 | 
						|
 | 
						|
void udpc_initFifo(struct fifo *fifo, int blockSize)
 | 
						|
{
 | 
						|
    fifo->dataBufSize = blockSize * 4096;
 | 
						|
    fifo->dataBuffer = (unsigned char*)malloc(fifo->dataBufSize+4096);
 | 
						|
    fifo->dataBuffer += 4096 - (((unsigned long)fifo->dataBuffer) % 4096);
 | 
						|
 | 
						|
    /* Free memory queue is initially full */
 | 
						|
    fifo->freeMemQueue = pc_makeProduconsum(fifo->dataBufSize, "free mem");
 | 
						|
    pc_produce(fifo->freeMemQueue, fifo->dataBufSize);
 | 
						|
 | 
						|
    fifo->data = pc_makeProduconsum(fifo->dataBufSize, "receive");
 | 
						|
}
 | 
						|
 | 
						|
THREAD_RETURN returnChannelMain(void *args) {
 | 
						|
    struct returnChannel *returnChannel = (struct returnChannel *) args;
 | 
						|
 | 
						|
    while(1) {
 | 
						|
	struct sockaddr_in from;
 | 
						|
	int clNo;
 | 
						|
	int pos = pc_getConsumerPosition(returnChannel->freeSpace);
 | 
						|
	pc_consumeAny(returnChannel->freeSpace);
 | 
						|
 | 
						|
	RECV(returnChannel->rcvSock, 
 | 
						|
	     returnChannel->q[pos].msg, from,
 | 
						|
	     returnChannel->config->portBase);
 | 
						|
	clNo = udpc_lookupParticipant(returnChannel->participantsDb, &from);
 | 
						|
	if (clNo < 0) { 
 | 
						|
	    /* packet from unknown provenance */
 | 
						|
	    continue;
 | 
						|
	}
 | 
						|
	returnChannel->q[pos].clNo = clNo;
 | 
						|
	pc_consumed(returnChannel->freeSpace, 1);
 | 
						|
	pc_produce(returnChannel->incoming, 1);
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
void initReturnChannel(struct returnChannel *returnChannel,
 | 
						|
			      struct net_config *config,
 | 
						|
			      int sock) {
 | 
						|
    returnChannel->config = config;
 | 
						|
    returnChannel->rcvSock = sock;
 | 
						|
    returnChannel->freeSpace = pc_makeProduconsum(QUEUE_SIZE,"msg:free-queue");
 | 
						|
    pc_produce(returnChannel->freeSpace, QUEUE_SIZE);
 | 
						|
    returnChannel->incoming = pc_makeProduconsum(QUEUE_SIZE,"msg:incoming");
 | 
						|
 | 
						|
    pthread_create(&returnChannel->thread, NULL,
 | 
						|
		   returnChannelMain, returnChannel);
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
void senderStatsAddBytes(sender_stats_t ss, long bytes) {
 | 
						|
    if(ss != NULL) {
 | 
						|
	ss->totalBytes += bytes;
 | 
						|
 | 
						|
	if(ss->bwPeriod) {
 | 
						|
	    double tdiff, bw;
 | 
						|
	    struct timeval tv;
 | 
						|
	    gettimeofday(&tv, 0);
 | 
						|
	    ss->periodBytes += bytes;
 | 
						|
	    if(tv.tv_sec - ss->periodStart.tv_sec < ss->bwPeriod-1)
 | 
						|
		return;
 | 
						|
	    tdiff = (tv.tv_sec-ss->periodStart.tv_sec) * 1000000.0 +
 | 
						|
		tv.tv_usec - ss->periodStart.tv_usec;
 | 
						|
	    if(tdiff < ss->bwPeriod * 1000000.0)
 | 
						|
		return;
 | 
						|
	    bw = ss->periodBytes * 8.0 / tdiff;
 | 
						|
	    ss->periodBytes=0;
 | 
						|
	    ss->periodStart = tv;
 | 
						|
	}
 | 
						|
    }
 | 
						|
}
 | 
						|
int ackSlice(struct slice *slice, struct net_config *net_config,
 | 
						|
		    struct fifo *fifo, sender_stats_t stats)
 | 
						|
{
 | 
						|
    if(slice->state == slice::SLICE_ACKED)
 | 
						|
	/* already acked */
 | 
						|
	return 0;
 | 
						|
    if(!(net_config->flags & FLAG_SN)) {
 | 
						|
	if(net_config->discovery == net_config::DSC_DOUBLING) {
 | 
						|
		net_config->sliceSize += net_config->sliceSize / 4;
 | 
						|
	   if(net_config->sliceSize >= net_config->max_slice_size) {
 | 
						|
	       net_config->sliceSize = net_config->max_slice_size;
 | 
						|
	       net_config->discovery = net_config::DSC_REDUCING;
 | 
						|
	   }
 | 
						|
       }
 | 
						|
    }
 | 
						|
    slice->state = slice::SLICE_ACKED;
 | 
						|
    pc_produce(fifo->freeMemQueue, slice->bytes);
 | 
						|
 | 
						|
    /* Statistics */
 | 
						|
    senderStatsAddBytes(stats, slice->bytes);
 | 
						|
 | 
						|
    /* End Statistics */
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int isSliceAcked(struct slice *slice)
 | 
						|
{
 | 
						|
    if(slice->state == slice::SLICE_ACKED) {
 | 
						|
	return 1;
 | 
						|
    } else {
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int freeSlice(sender_state_t sendst, struct slice *slice) {
 | 
						|
    int i;
 | 
						|
    i = slice - sendst->slices;
 | 
						|
    slice->state = slice::SLICE_PRE_FREE;
 | 
						|
    while(1) {
 | 
						|
	int pos = pc_getProducerPosition(sendst->free_slices_pc);
 | 
						|
	if(sendst->slices[pos].state == slice::SLICE_PRE_FREE)
 | 
						|
	    sendst->slices[pos].state = slice::SLICE_FREE;
 | 
						|
	else
 | 
						|
	    break;
 | 
						|
	pc_produce(sendst->free_slices_pc, 1);
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int isSliceXmitted(struct slice *slice) {
 | 
						|
    if(slice->state == slice::SLICE_XMITTED) {
 | 
						|
	return 1;
 | 
						|
    } else {
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int getSliceBlocks(struct slice *slice, struct net_config *net_config)
 | 
						|
{
 | 
						|
    return (slice->bytes + net_config->blockSize - 1) / net_config->blockSize;
 | 
						|
}
 | 
						|
 | 
						|
int sendReqack(struct slice *slice, struct net_config *net_config,
 | 
						|
		      struct fifo *fifo, sender_stats_t stats,
 | 
						|
		      int sock)
 | 
						|
{
 | 
						|
    /* in async mode, just confirm slice... */
 | 
						|
    if((net_config->flags & FLAG_ASYNC) && slice->bytes != 0) {
 | 
						|
	ackSlice(slice, net_config, fifo, stats);
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    if((net_config->flags & FLAG_ASYNC)
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
       && 
 | 
						|
       (net_config->flags & FLAG_FEC)
 | 
						|
#endif
 | 
						|
       ) {
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    if(!(net_config->flags & FLAG_SN) && slice->rxmitId != 0) {
 | 
						|
	int nrBlocks;
 | 
						|
	nrBlocks = getSliceBlocks(slice, net_config);
 | 
						|
	if(slice->lastGoodBlock != 0 && slice->lastGoodBlock < nrBlocks) {
 | 
						|
	    net_config->discovery = net_config::DSC_REDUCING;
 | 
						|
	    if (slice->lastGoodBlock < net_config->sliceSize / 2) {
 | 
						|
		net_config->sliceSize = net_config->sliceSize / 2;
 | 
						|
	    } else {
 | 
						|
		net_config->sliceSize = slice->lastGoodBlock;
 | 
						|
	    }
 | 
						|
	    if(net_config->sliceSize < 32) {
 | 
						|
		/* a minimum of 32 */
 | 
						|
		net_config->sliceSize = 32;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
    }
 | 
						|
 | 
						|
    slice->lastGoodBlock = 0;
 | 
						|
    slice->sl_reqack.ra.opCode = htons(CMD_REQACK);
 | 
						|
    slice->sl_reqack.ra.sliceNo = htonl(slice->sliceNo);
 | 
						|
    slice->sl_reqack.ra.bytes = htonl(slice->bytes);
 | 
						|
 | 
						|
    slice->sl_reqack.ra.reserved = 0;
 | 
						|
    memcpy((void*)&slice->answeredSet,(void*)&slice->sl_reqack.readySet,
 | 
						|
	   sizeof(slice->answeredSet));
 | 
						|
    slice->nrAnswered = slice->nrReady;
 | 
						|
 | 
						|
    /* not everybody is ready yet */
 | 
						|
    slice->needRxmit = 0;
 | 
						|
    memset(slice->rxmitMap, 0, sizeof(slice->rxmitMap));
 | 
						|
    memset(slice->isXmittedMap, 0, sizeof(slice->isXmittedMap));
 | 
						|
    slice->sl_reqack.ra.rxmit = htonl(slice->rxmitId);
 | 
						|
    
 | 
						|
    //rgWaitAll(net_config, sock, net_config->dataMcastAddr.sin_addr.s_addr, sizeof(slice->sl_reqack));
 | 
						|
    BCAST_DATA(sock, slice->sl_reqack);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
struct slice *findSlice(struct slice *slice1,
 | 
						|
			       struct slice *slice2,
 | 
						|
			       int sliceNo)
 | 
						|
{
 | 
						|
    if(slice1 != NULL && slice1->sliceNo == sliceNo)
 | 
						|
	return slice1;
 | 
						|
    if(slice2 != NULL && slice2->sliceNo == sliceNo)
 | 
						|
	return slice2;
 | 
						|
    return NULL;
 | 
						|
}
 | 
						|
 | 
						|
void markParticipantAnswered(slice_t slice, int clNo)
 | 
						|
{
 | 
						|
    if(BIT_ISSET(clNo, slice->answeredSet))
 | 
						|
	/* client already has answered */
 | 
						|
	return;
 | 
						|
    slice->nrAnswered++;
 | 
						|
    SET_BIT(clNo, slice->answeredSet);
 | 
						|
}
 | 
						|
 | 
						|
int udpc_isParticipantValid(struct participantsDb *db, int i) {
 | 
						|
    return db->clientTable[i].used;
 | 
						|
}
 | 
						|
 | 
						|
void senderSetAnswered(sender_stats_t ss, int clNo) {
 | 
						|
    if(ss != NULL)
 | 
						|
	ss->clNo = clNo;
 | 
						|
}
 | 
						|
 | 
						|
int handleOk(sender_state_t sendst,
 | 
						|
		    struct slice *slice,
 | 
						|
		    int clNo)
 | 
						|
{
 | 
						|
    if(slice == NULL)
 | 
						|
	return 0;
 | 
						|
    if(!udpc_isParticipantValid(sendst->rc.participantsDb, clNo)) {
 | 
						|
	//udpc_flprintf("Invalid participant %d\n", clNo);
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
    if (BIT_ISSET(clNo, slice->sl_reqack.readySet)) {
 | 
						|
        /* client is already marked ready */
 | 
						|
    } else {
 | 
						|
	SET_BIT(clNo, slice->sl_reqack.readySet);
 | 
						|
	slice->nrReady++;
 | 
						|
	senderSetAnswered(sendst->stats, clNo);
 | 
						|
	markParticipantAnswered(slice, clNo);
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int handleDisconnect1(struct slice *slice, int clNo)
 | 
						|
{    
 | 
						|
    if(slice != NULL) {
 | 
						|
	if (BIT_ISSET(clNo, slice->sl_reqack.readySet)) {
 | 
						|
	    /* avoid counting client both as left and ready */
 | 
						|
	    CLR_BIT(clNo, slice->sl_reqack.readySet);
 | 
						|
	    slice->nrReady--;
 | 
						|
	}
 | 
						|
	if (BIT_ISSET(clNo, slice->answeredSet)) {
 | 
						|
	    slice->nrAnswered--;
 | 
						|
	    CLR_BIT(clNo, slice->answeredSet);
 | 
						|
	}
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int handleDisconnect(participantsDb_t db,
 | 
						|
			    struct slice *slice1,
 | 
						|
			    struct slice *slice2,
 | 
						|
			    int clNo)
 | 
						|
{
 | 
						|
    handleDisconnect1(slice1, clNo);
 | 
						|
    handleDisconnect1(slice2, clNo);
 | 
						|
    udpc_removeParticipant(db, clNo);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int handleRetransmit(sender_state_t sendst,
 | 
						|
			    struct slice *slice,
 | 
						|
			    int clNo, unsigned char *map, int rxmit)
 | 
						|
{
 | 
						|
    unsigned int i;
 | 
						|
 | 
						|
    if(!udpc_isParticipantValid(sendst->rc.participantsDb, clNo)) {
 | 
						|
	//udpc_flprintf("Invalid participant %d\n", clNo);
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
    if(slice == NULL)
 | 
						|
	return 0;    
 | 
						|
    if (rxmit < slice->rxmitId) {
 | 
						|
	/* late answer to previous Req Ack */
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
    for(i=0; i <sizeof(slice->rxmitMap) / sizeof(char); i++) {
 | 
						|
	slice->rxmitMap[i] |= ~map[i];
 | 
						|
    }
 | 
						|
    slice->needRxmit = 1;
 | 
						|
    markParticipantAnswered(slice, clNo);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
int handleNextMessage(sender_state_t sendst,
 | 
						|
			     struct slice *xmitSlice,
 | 
						|
			     struct slice *rexmitSlice)
 | 
						|
{
 | 
						|
    int pos = pc_getConsumerPosition(sendst->rc.incoming);
 | 
						|
    union message *msg = &sendst->rc.q[pos].msg;
 | 
						|
    int clNo = sendst->rc.q[pos].clNo;
 | 
						|
 | 
						|
    pc_consumeAny(sendst->rc.incoming);
 | 
						|
    switch(ntohs(msg->opCode)) {
 | 
						|
	case CMD_OK:
 | 
						|
	    handleOk(sendst, 
 | 
						|
		     findSlice(xmitSlice, rexmitSlice, ntohl(msg->ok.sliceNo)), clNo);
 | 
						|
	    break;
 | 
						|
	case CMD_DISCONNECT:
 | 
						|
	    handleDisconnect(sendst->rc.participantsDb, 
 | 
						|
			     xmitSlice, rexmitSlice, clNo);
 | 
						|
	    break;	    
 | 
						|
	case CMD_RETRANSMIT:
 | 
						|
	    handleRetransmit(sendst,
 | 
						|
			     findSlice(xmitSlice, rexmitSlice,
 | 
						|
				       ntohl(msg->retransmit.sliceNo)),
 | 
						|
			     clNo,
 | 
						|
			     msg->retransmit.map,
 | 
						|
			     msg->retransmit.rxmit);
 | 
						|
	    break;
 | 
						|
	default:
 | 
						|
		//TODO: FIXME
 | 
						|
	    //udpc_flprintf("Bad command %04x\n", (unsigned short) msg->opCode);
 | 
						|
	    break;
 | 
						|
    }
 | 
						|
    pc_consumed(sendst->rc.incoming, 1);
 | 
						|
    pc_produce(sendst->rc.freeSpace, 1);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int sendRawData(int sock,
 | 
						|
		       struct net_config *config, 
 | 
						|
		       char *header, int headerSize, 
 | 
						|
		       unsigned char *data, int dataSize)
 | 
						|
{
 | 
						|
    struct iovec iov[2];
 | 
						|
    struct msghdr hdr;
 | 
						|
    int packetSize;
 | 
						|
    int ret;
 | 
						|
    
 | 
						|
    iov[0].iov_base = header;
 | 
						|
    iov[0].iov_len = headerSize;
 | 
						|
 | 
						|
    iov[1].iov_base = data;
 | 
						|
    iov[1].iov_len = dataSize;
 | 
						|
 | 
						|
    hdr.msg_name = &config->dataMcastAddr;
 | 
						|
    hdr.msg_namelen = sizeof(struct sockaddr_in);
 | 
						|
    hdr.msg_iov = iov;
 | 
						|
    hdr.msg_iovlen  = 2;
 | 
						|
    initMsgHdr(&hdr);
 | 
						|
 | 
						|
    packetSize = dataSize + headerSize;
 | 
						|
    //rgWaitAll(config, sock, config->dataMcastAddr.sin_addr.s_addr, packetSize);
 | 
						|
    ret = sendmsg(sock, &hdr, 0);
 | 
						|
    if (ret < 0) {
 | 
						|
	//TODO: FIXME
 | 
						|
    }
 | 
						|
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
int transmitDataBlock(sender_state_t sendst, struct slice *slice, int i)
 | 
						|
{
 | 
						|
    struct fifo *fifo = sendst->fifo;
 | 
						|
    struct net_config *config = sendst->config;
 | 
						|
    struct dataBlock msg;
 | 
						|
 | 
						|
    idbassert(i < MAX_SLICE_SIZE);
 | 
						|
    
 | 
						|
    msg.opCode  = htons(CMD_DATA);
 | 
						|
    msg.sliceNo = htonl(slice->sliceNo);
 | 
						|
    msg.blockNo = htons(i);
 | 
						|
 | 
						|
    msg.reserved = 0;
 | 
						|
    msg.reserved2 = 0;
 | 
						|
    msg.bytes = htonl(slice->bytes);
 | 
						|
    
 | 
						|
    sendRawData(sendst->socket, config, 
 | 
						|
		(char *) &msg, sizeof(msg),
 | 
						|
		fifo->dataBuffer + 
 | 
						|
		(slice->base + i * config->blockSize) % fifo->dataBufSize,
 | 
						|
		config->blockSize);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
void senderStatsAddRetransmissions(sender_stats_t ss, int retransmissions) {
 | 
						|
    if(ss != NULL) {
 | 
						|
	ss->retransmissions += retransmissions;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int sendSlice(sender_state_t sendst, struct slice *slice,
 | 
						|
		     int retransmitting)
 | 
						|
{    
 | 
						|
    struct net_config *config = sendst->config;
 | 
						|
 | 
						|
    int nrBlocks, i, rehello;
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    int fecBlocks;
 | 
						|
#endif
 | 
						|
    int retransmissions=0;
 | 
						|
 | 
						|
    if(retransmitting) {
 | 
						|
	slice->nextBlock = 0;
 | 
						|
	if(slice->state != slice::SLICE_XMITTED)
 | 
						|
	    return 0;
 | 
						|
    } else {
 | 
						|
	if(slice->state != slice::SLICE_NEW)
 | 
						|
	    return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    nrBlocks = getSliceBlocks(slice, config);
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    if((config->flags & FLAG_FEC) && !retransmitting) {
 | 
						|
	fecBlocks = config->fec_redundancy * config->fec_stripes;
 | 
						|
    } else {
 | 
						|
	fecBlocks = 0;
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    if((sendst->config->flags & FLAG_STREAMING)) {
 | 
						|
      rehello = nrBlocks - sendst->config->rehelloOffset;
 | 
						|
      if(rehello < 0)
 | 
						|
	rehello = 0;
 | 
						|
    } else {
 | 
						|
      rehello = -1;
 | 
						|
    }
 | 
						|
 | 
						|
    /* transmit the data */
 | 
						|
    for(i = slice->nextBlock; i < nrBlocks
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	  + fecBlocks
 | 
						|
#endif
 | 
						|
	  ; i++) {
 | 
						|
	if(retransmitting) {
 | 
						|
	    if(!BIT_ISSET(i, slice->rxmitMap) ||
 | 
						|
	       BIT_ISSET(i, slice->isXmittedMap)) {
 | 
						|
		/* if slice is not in retransmit list, or has _already_
 | 
						|
		 * been retransmitted, skip it */
 | 
						|
		if(i > slice->lastGoodBlock)
 | 
						|
		    slice->lastGoodBlock = i;
 | 
						|
		continue;
 | 
						|
	    }
 | 
						|
	    SET_BIT(i, slice->isXmittedMap);
 | 
						|
	    retransmissions++;
 | 
						|
	}
 | 
						|
 | 
						|
	if(i == rehello) {
 | 
						|
	    udpc_sendHello(sendst->config, sendst->socket, 1);
 | 
						|
	}
 | 
						|
 | 
						|
	if(i < nrBlocks)
 | 
						|
	    transmitDataBlock(sendst, slice, i);
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	else
 | 
						|
	    transmitFecBlock(sendst, slice, i - nrBlocks);
 | 
						|
#endif
 | 
						|
	if(!retransmitting && pc_getWaiting(sendst->rc.incoming)) {
 | 
						|
	    i++;
 | 
						|
	    break;
 | 
						|
	}
 | 
						|
    }
 | 
						|
 | 
						|
    if(retransmissions)
 | 
						|
	senderStatsAddRetransmissions(sendst->stats, retransmissions);
 | 
						|
    slice->nextBlock = i;
 | 
						|
    if(i == nrBlocks
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
       + fecBlocks
 | 
						|
#endif
 | 
						|
       ) {
 | 
						|
	slice->needRxmit = 0;
 | 
						|
	if(!retransmitting)
 | 
						|
	    slice->state = slice::SLICE_XMITTED;
 | 
						|
	return 2;
 | 
						|
    }
 | 
						|
    return 1;
 | 
						|
}
 | 
						|
 | 
						|
int doRetransmissions(sender_state_t sendst,
 | 
						|
			     struct slice *slice)
 | 
						|
{
 | 
						|
    if(slice->state == slice::SLICE_ACKED)
 | 
						|
	return 0; /* nothing to do */
 | 
						|
 | 
						|
    /* FIXME: reduce slice size if needed */
 | 
						|
    if(slice->needRxmit) {
 | 
						|
	/* do some retransmissions */
 | 
						|
	sendSlice(sendst, slice, 1);
 | 
						|
    }
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
struct slice *makeSlice(sender_state_t sendst, int sliceNo) {
 | 
						|
    struct net_config *config = sendst->config;
 | 
						|
    struct fifo *fifo = sendst->fifo;
 | 
						|
    int i;
 | 
						|
    struct slice *slice=NULL;
 | 
						|
 | 
						|
    pc_consume(sendst->free_slices_pc, 1);
 | 
						|
    i = pc_getConsumerPosition(sendst->free_slices_pc);
 | 
						|
    slice = &sendst->slices[i];
 | 
						|
    idbassert(slice->state == slice::SLICE_FREE);
 | 
						|
    BZERO(*slice);
 | 
						|
    pc_consumed(sendst->free_slices_pc, 1);
 | 
						|
 | 
						|
    slice->base = pc_getConsumerPosition(sendst->fifo->data);
 | 
						|
    slice->sliceNo = sliceNo;
 | 
						|
    slice->bytes = pc_consume(fifo->data, 10*config->blockSize);
 | 
						|
 | 
						|
    /* fixme: use current slice size here */
 | 
						|
    if(slice->bytes > config->blockSize * config->sliceSize)
 | 
						|
	slice->bytes = config->blockSize * config->sliceSize;
 | 
						|
    pc_consumed(fifo->data, slice->bytes);
 | 
						|
    slice->nextBlock = 0;
 | 
						|
    slice->state = slice::SLICE_NEW;
 | 
						|
    BZERO(slice->sl_reqack.readySet);
 | 
						|
    slice->nrReady = 0;
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    slice->fec_data = sendst->fec_data + (i * config->fec_stripes * 
 | 
						|
					  config->fec_redundancy *
 | 
						|
					  config->blockSize);
 | 
						|
#endif
 | 
						|
    return slice;
 | 
						|
}
 | 
						|
 | 
						|
void cancelReturnChannel(struct returnChannel *returnChannel) {
 | 
						|
    /* No need to worry about the pthread_cond_wait in produconsum, because
 | 
						|
     * at the point where we enter here (to cancel the thread), we are sure
 | 
						|
     * that nobody else uses that produconsum any more
 | 
						|
     */
 | 
						|
    pthread_cancel(returnChannel->thread);
 | 
						|
    pthread_join(returnChannel->thread, NULL);
 | 
						|
}
 | 
						|
 | 
						|
THREAD_RETURN netSenderMain(void	*args0)
 | 
						|
{
 | 
						|
    sender_state_t sendst = (sender_state_t) args0;
 | 
						|
    struct net_config *config = sendst->config;
 | 
						|
    struct timeval tv;
 | 
						|
    struct timespec ts;
 | 
						|
    int atEnd = 0;
 | 
						|
    int nrWaited=0;
 | 
						|
    unsigned long waitAverage=10000; /* Exponential average of last wait times */
 | 
						|
 | 
						|
    struct slice *xmitSlice=NULL; /* slice being transmitted a first time */
 | 
						|
    struct slice *rexmitSlice=NULL; /* slice being re-transmitted */
 | 
						|
    int sliceNo = 0;
 | 
						|
 | 
						|
    /* transmit the data */
 | 
						|
    if(config->default_slice_size == 0) {
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	if(config->flags & FLAG_FEC) {
 | 
						|
	    config->sliceSize = 
 | 
						|
		config->fec_stripesize * config->fec_stripes;
 | 
						|
	} else
 | 
						|
#endif
 | 
						|
	  if(config->flags & FLAG_ASYNC)
 | 
						|
	    config->sliceSize = 1024;
 | 
						|
	else if (sendst->config->flags & FLAG_SN) {
 | 
						|
	    sendst->config->sliceSize = 112;
 | 
						|
	} else
 | 
						|
	    sendst->config->sliceSize = 130;
 | 
						|
	sendst->config->discovery = net_config::DSC_DOUBLING;
 | 
						|
    } else {
 | 
						|
	config->sliceSize = config->default_slice_size;
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	if((config->flags & FLAG_FEC) &&
 | 
						|
	   (config->sliceSize > 128 * config->fec_stripes))
 | 
						|
	    config->sliceSize = 128 * config->fec_stripes;
 | 
						|
#endif
 | 
						|
    }
 | 
						|
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    if( (sendst->config->flags & FLAG_FEC) &&
 | 
						|
	config->max_slice_size > config->fec_stripes * 128)
 | 
						|
      config->max_slice_size = config->fec_stripes * 128;
 | 
						|
#endif
 | 
						|
 | 
						|
    if(config->sliceSize > config->max_slice_size)
 | 
						|
	config->sliceSize = config->max_slice_size;
 | 
						|
 | 
						|
    idbassert(config->sliceSize <= MAX_SLICE_SIZE);
 | 
						|
 | 
						|
    do {
 | 
						|
	/* first, cleanup rexmit Slice if needed */
 | 
						|
 | 
						|
	if(rexmitSlice != NULL) {
 | 
						|
	    if(rexmitSlice->nrReady == 
 | 
						|
	       udpc_nrParticipants(sendst->rc.participantsDb)){
 | 
						|
#if DEBUG
 | 
						|
		flprintf("slice is ready\n");
 | 
						|
#endif
 | 
						|
		ackSlice(rexmitSlice, sendst->config, sendst->fifo, 
 | 
						|
			 sendst->stats);
 | 
						|
	    }
 | 
						|
	    if(isSliceAcked(rexmitSlice)) {
 | 
						|
		freeSlice(sendst, rexmitSlice);
 | 
						|
		rexmitSlice = NULL;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
 | 
						|
	/* then shift xmit slice to rexmit slot, if possible */
 | 
						|
	if(rexmitSlice == NULL &&  xmitSlice != NULL && 
 | 
						|
	   isSliceXmitted(xmitSlice)) {
 | 
						|
	    rexmitSlice = xmitSlice;
 | 
						|
	    xmitSlice = NULL;
 | 
						|
	    sendReqack(rexmitSlice, sendst->config, sendst->fifo, sendst->stats,
 | 
						|
		       sendst->socket);
 | 
						|
	}
 | 
						|
 | 
						|
	/* handle any messages */
 | 
						|
	if(pc_getWaiting(sendst->rc.incoming)) {
 | 
						|
#if DEBUG
 | 
						|
	    flprintf("Before message %d\n",
 | 
						|
		    pc_getWaiting(sendst->rc.incoming));
 | 
						|
#endif
 | 
						|
	    handleNextMessage(sendst, xmitSlice, rexmitSlice);
 | 
						|
 | 
						|
	    /* restart at beginning of loop: we may have acked the rxmit
 | 
						|
	     * slice, makeing it possible to shift the pipe */
 | 
						|
	    continue;
 | 
						|
	}
 | 
						|
 | 
						|
	/* do any needed retransmissions */
 | 
						|
	if(rexmitSlice != NULL && rexmitSlice->needRxmit) {
 | 
						|
	    doRetransmissions(sendst, rexmitSlice);
 | 
						|
	    /* restart at beginning: new messages may have arrived during
 | 
						|
	     * retransmission  */
 | 
						|
	    continue;
 | 
						|
	}
 | 
						|
 | 
						|
	/* if all participants answered, send req ack */
 | 
						|
	if(rexmitSlice != NULL && 
 | 
						|
	   rexmitSlice->nrAnswered == 
 | 
						|
	   udpc_nrParticipants(sendst->rc.participantsDb)) {
 | 
						|
	    rexmitSlice->rxmitId++;
 | 
						|
	    sendReqack(rexmitSlice, sendst->config, sendst->fifo, sendst->stats,
 | 
						|
		       sendst->socket);
 | 
						|
	}
 | 
						|
 | 
						|
	if(xmitSlice == NULL && !atEnd) {
 | 
						|
#if DEBUG
 | 
						|
	    flprintf("SN=%d\n", sendst->config->flags & FLAG_SN);
 | 
						|
#endif
 | 
						|
	    if((sendst->config->flags & FLAG_SN) ||
 | 
						|
	       rexmitSlice == NULL) {
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
		if(sendst->config->flags & FLAG_FEC) {
 | 
						|
		    int i;
 | 
						|
		    pc_consume(sendst->fec_data_pc, 1);
 | 
						|
		    i = pc_getConsumerPosition(sendst->fec_data_pc);
 | 
						|
		    xmitSlice = &sendst->slices[i];
 | 
						|
		    pc_consumed(sendst->fec_data_pc, 1);
 | 
						|
		} else
 | 
						|
#endif
 | 
						|
		  {
 | 
						|
		    xmitSlice = makeSlice(sendst, sliceNo++);
 | 
						|
		}
 | 
						|
		if(xmitSlice->bytes == 0)
 | 
						|
		    atEnd = 1;
 | 
						|
	    }
 | 
						|
	}
 | 
						|
	 
 | 
						|
	if(xmitSlice != NULL && xmitSlice->state == slice::SLICE_NEW) {
 | 
						|
	    sendSlice(sendst, xmitSlice, 0);
 | 
						|
#if DEBUG
 | 
						|
	    flprintf("%d Interrupted at %d/%d\n", xmitSlice->sliceNo, 
 | 
						|
		     xmitSlice->nextBlock, 
 | 
						|
		     getSliceBlocks(xmitSlice, sendst->config));
 | 
						|
#endif
 | 
						|
	    continue;
 | 
						|
	}
 | 
						|
	if(atEnd && rexmitSlice == NULL && xmitSlice == NULL)
 | 
						|
	    break;
 | 
						|
 | 
						|
	if(sendst->config->flags & FLAG_ASYNC)
 | 
						|
	    break;
 | 
						|
 | 
						|
#if DEBUG
 | 
						|
	flprintf("Waiting for timeout...\n");
 | 
						|
#endif
 | 
						|
	gettimeofday(&tv, 0);
 | 
						|
	ts.tv_sec = tv.tv_sec;
 | 
						|
	ts.tv_nsec = (long int)((tv.tv_usec + 1.1*waitAverage) * 1000);
 | 
						|
 | 
						|
#ifdef WINDOWS
 | 
						|
	/* Windows has a granularity of 1 millisecond in its timer. Take this
 | 
						|
	 * into account here */
 | 
						|
	#define GRANULARITY 1000000
 | 
						|
	ts.tv_nsec += 3*GRANULARITY/2;
 | 
						|
	ts.tv_nsec -= ts.tv_nsec % GRANULARITY;
 | 
						|
#endif
 | 
						|
 | 
						|
#define BILLION 1000000000
 | 
						|
 | 
						|
	while(ts.tv_nsec >= BILLION) {
 | 
						|
	    ts.tv_nsec -= BILLION;
 | 
						|
	    ts.tv_sec++;
 | 
						|
	}
 | 
						|
 | 
						|
	if(rexmitSlice->rxmitId > 10)
 | 
						|
	    /* after tenth retransmission, wait minimum one second */
 | 
						|
	    ts.tv_sec++;
 | 
						|
 | 
						|
	if(pc_consumeAnyWithTimeout(sendst->rc.incoming, &ts) != 0) {
 | 
						|
#if DEBUG
 | 
						|
	    flprintf("Have data\n");
 | 
						|
#endif
 | 
						|
	    {
 | 
						|
		struct timeval tv2;
 | 
						|
		unsigned long timeout;
 | 
						|
		gettimeofday(&tv2, 0);
 | 
						|
		timeout = 
 | 
						|
		    (tv2.tv_sec - tv.tv_sec) * 1000000+
 | 
						|
		    tv2.tv_usec - tv.tv_usec;
 | 
						|
		if(nrWaited)
 | 
						|
		    timeout += waitAverage;
 | 
						|
		waitAverage += 9; /* compensate against rounding errors */
 | 
						|
		waitAverage = (long unsigned int)((0.9 * waitAverage + 0.1 * timeout));
 | 
						|
	    }
 | 
						|
	    nrWaited = 0;
 | 
						|
	    continue;
 | 
						|
	}
 | 
						|
	if(rexmitSlice == NULL) {
 | 
						|
		//TODO: FIXME
 | 
						|
	    //udpc_flprintf("Weird. Timeout and no rxmit slice");
 | 
						|
	    break;
 | 
						|
	}
 | 
						|
	if(nrWaited > 5){
 | 
						|
#ifndef WINDOWS
 | 
						|
	    /* on Cygwin, we would get too many of those messages... */
 | 
						|
	    nrWaited=0;
 | 
						|
#endif
 | 
						|
	}
 | 
						|
	nrWaited++;
 | 
						|
	if(rexmitSlice->rxmitId > config->retriesUntilDrop) {
 | 
						|
	    int i;
 | 
						|
            for(i=0; i < MAX_CLIENTS; i++) {
 | 
						|
                if(udpc_isParticipantValid(sendst->rc.participantsDb, i) && 
 | 
						|
		   !BIT_ISSET(i, rexmitSlice->sl_reqack.readySet)) {
 | 
						|
                    udpc_removeParticipant(sendst->rc.participantsDb, i);
 | 
						|
                    if(udpc_nrParticipants(sendst->rc.participantsDb) == 0)
 | 
						|
			goto exit_main_loop;
 | 
						|
                }
 | 
						|
            }
 | 
						|
	    continue;
 | 
						|
	}
 | 
						|
	rexmitSlice->rxmitId++;
 | 
						|
	sendReqack(rexmitSlice, sendst->config, sendst->fifo, sendst->stats,
 | 
						|
		   sendst->socket);
 | 
						|
    } while(udpc_nrParticipants(sendst->rc.participantsDb)||
 | 
						|
	    (config->flags & FLAG_ASYNC));
 | 
						|
 exit_main_loop:
 | 
						|
    cancelReturnChannel(&sendst->rc);
 | 
						|
    pc_produceEnd(sendst->fifo->freeMemQueue);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int spawnNetSender(struct fifo *fifo,
 | 
						|
		   int sock,
 | 
						|
		   struct net_config *config,
 | 
						|
		   participantsDb_t db)
 | 
						|
{
 | 
						|
    int i;
 | 
						|
 | 
						|
    sender_state_t sendst = MALLOC(struct senderState);
 | 
						|
    sendst->fifo = fifo;
 | 
						|
    sendst->socket = sock;
 | 
						|
    sendst->config = config;
 | 
						|
    //sendst->stats = stats;
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    if(sendst->config->flags & FLAG_FEC)
 | 
						|
      sendst->fec_data =  xmalloc(NR_SLICES *
 | 
						|
				  config->fec_stripes * 
 | 
						|
				  config->fec_redundancy *
 | 
						|
				  config->blockSize);
 | 
						|
#endif
 | 
						|
    sendst->rc.participantsDb = db;
 | 
						|
    initReturnChannel(&sendst->rc, sendst->config, sendst->socket);
 | 
						|
 | 
						|
    sendst->free_slices_pc = pc_makeProduconsum(NR_SLICES, "free slices");
 | 
						|
    pc_produce(sendst->free_slices_pc, NR_SLICES);
 | 
						|
    for(i = 0; i <NR_SLICES; i++)
 | 
						|
	sendst->slices[i].state = slice::SLICE_FREE;
 | 
						|
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    if(sendst->config->flags & FLAG_FEC) {
 | 
						|
	/* Free memory queue is initially full */
 | 
						|
	fec_init();
 | 
						|
	sendst->fec_data_pc = pc_makeProduconsum(NR_SLICES, "fec data");
 | 
						|
 | 
						|
	pthread_create(&sendst->fec_thread, NULL, fecMain, sendst);
 | 
						|
    }
 | 
						|
#endif
 | 
						|
 | 
						|
    pthread_create(&fifo->thread, NULL, netSenderMain, sendst);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
struct sockaddr_in *udpc_getParticipantIp(participantsDb_t db, int i)
 | 
						|
{
 | 
						|
    return &db->clientTable[i].addr;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_getParticipantCapabilities(participantsDb_t db, int i)
 | 
						|
{
 | 
						|
    return db->clientTable[i].capabilities;
 | 
						|
}
 | 
						|
 | 
						|
unsigned int udpc_getParticipantRcvBuf(participantsDb_t db, int i)
 | 
						|
{
 | 
						|
    return db->clientTable[i].rcvbuf;
 | 
						|
}
 | 
						|
 | 
						|
int pc_consumeContiguousMinAmount(produconsum_t pc, int amount)
 | 
						|
{
 | 
						|
    int n = _consumeAny(pc, amount, 0);
 | 
						|
    int l = pc->size - (pc->consumed % pc->size);
 | 
						|
    if(n > l)
 | 
						|
	n = l;
 | 
						|
    return n;
 | 
						|
    
 | 
						|
}
 | 
						|
 | 
						|
#define BLOCKSIZE 4096
 | 
						|
 | 
						|
void localReader(struct fifo* fifo, const uint8_t* buf, uint32_t len)
 | 
						|
{
 | 
						|
//cerr << "starting to send " << len << " bytes" << endl;
 | 
						|
    uint32_t offset = 0;
 | 
						|
    while(1) {
 | 
						|
	int pos = pc_getConsumerPosition(fifo->freeMemQueue);
 | 
						|
	int bytes = 
 | 
						|
	    pc_consumeContiguousMinAmount(fifo->freeMemQueue, BLOCKSIZE);
 | 
						|
	if(bytes > (pos + bytes) % BLOCKSIZE)
 | 
						|
	    bytes -= (pos + bytes) % BLOCKSIZE;
 | 
						|
 | 
						|
        if (offset + bytes > len) bytes = len - offset;
 | 
						|
 | 
						|
//cerr << "sending " << bytes << " bytes from bs..." << endl;
 | 
						|
	//bytes = read(in, fifo->dataBuffer + pos, bytes);
 | 
						|
	memcpy(fifo->dataBuffer + pos, buf + offset, bytes);
 | 
						|
        offset += bytes;
 | 
						|
 | 
						|
	if (bytes == 0) {
 | 
						|
	    /* the end */
 | 
						|
	    pc_produceEnd(fifo->data);
 | 
						|
	    break;
 | 
						|
	} else {
 | 
						|
	    pc_consumed(fifo->freeMemQueue, bytes);
 | 
						|
	    pc_produce(fifo->data, bytes);
 | 
						|
	}
 | 
						|
    }
 | 
						|
//cerr << "done sending" << endl;
 | 
						|
}
 | 
						|
 | 
						|
unsigned int udpc_getRcvBuf(int sock) {
 | 
						|
    unsigned int bufsize;
 | 
						|
    socklen_t len=sizeof(int);
 | 
						|
    if(getsockopt(sock, SOL_SOCKET, SO_RCVBUF, (char *)&bufsize, &len) < 0)
 | 
						|
	return -1;
 | 
						|
    return bufsize;
 | 
						|
}
 | 
						|
 | 
						|
int sendConnectReq(struct client_config *client_config,
 | 
						|
			  struct net_config *net_config,
 | 
						|
			  int haveServerAddress) {
 | 
						|
//cerr << "sending a connect request" << endl;
 | 
						|
    struct connectReq connectReq;
 | 
						|
 | 
						|
    if(net_config->flags & FLAG_PASSIVE)
 | 
						|
	return 0;
 | 
						|
 | 
						|
    connectReq.opCode = htons(CMD_CONNECT_REQ);
 | 
						|
    connectReq.reserved = 0;
 | 
						|
    connectReq.capabilities = htonl(RECEIVER_CAPABILITIES);
 | 
						|
    connectReq.rcvbuf = htonl(udpc_getRcvBuf(client_config->S_UCAST));
 | 
						|
    if(haveServerAddress)
 | 
						|
      return SSEND(connectReq);
 | 
						|
    else
 | 
						|
      return BCAST_CONTROL(client_config->S_UCAST, connectReq);
 | 
						|
}
 | 
						|
 | 
						|
void udpc_setRcvBuf(int sock, unsigned int bufsize) {
 | 
						|
    if(setsockopt(sock, SOL_SOCKET, SO_RCVBUF, 
 | 
						|
		  (char*) &bufsize, sizeof(bufsize))< 0)
 | 
						|
	{
 | 
						|
	//TODO: FIXME
 | 
						|
	//perror("Set receiver buffer");
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void udpc_copyFromMessage(struct sockaddr_in *dst, unsigned char *src) {
 | 
						|
    memcpy((char *) &dst->sin_addr, src, sizeof(struct in_addr));
 | 
						|
}
 | 
						|
 | 
						|
unsigned short udpc_getPort(struct sockaddr_in *addr) {
 | 
						|
    return ntohs(((struct sockaddr_in *) addr)->sin_port);
 | 
						|
}
 | 
						|
 | 
						|
int udpc_selectSock(int *socks, int nr, int startTimeout) {
 | 
						|
    fd_set read_set;
 | 
						|
    int r;
 | 
						|
    int maxFd;
 | 
						|
    struct timeval tv, *tvp;
 | 
						|
    if(startTimeout) {
 | 
						|
	tv.tv_sec = startTimeout;
 | 
						|
	tv.tv_usec = 0;
 | 
						|
	tvp = &tv;
 | 
						|
    } else {
 | 
						|
	tvp = NULL;
 | 
						|
    }
 | 
						|
    maxFd = prepareForSelect(socks, nr, &read_set);
 | 
						|
    r = select(maxFd+1, &read_set, NULL, NULL, tvp);
 | 
						|
    if(r < 0)
 | 
						|
	return r;
 | 
						|
    return udpc_getSelectedSock(socks, nr, &read_set);
 | 
						|
}
 | 
						|
 | 
						|
void udpc_zeroSockArray(int *socks, int nr) {
 | 
						|
    int i;
 | 
						|
 | 
						|
    for(i=0; i<nr; i++)
 | 
						|
	socks[i] = -1;
 | 
						|
}
 | 
						|
 | 
						|
unsigned char *getBlockSpace(struct clientState *clst)
 | 
						|
{
 | 
						|
    int pos;
 | 
						|
 | 
						|
    if(clst->localPos) {
 | 
						|
	clst->localPos--;
 | 
						|
	return clst->localBlockAddresses[clst->localPos];
 | 
						|
    }
 | 
						|
 | 
						|
    pc_consume(clst->freeBlocks_pc, 1);
 | 
						|
    pos = pc_getConsumerPosition(clst->freeBlocks_pc);
 | 
						|
    pc_consumed(clst->freeBlocks_pc, 1);
 | 
						|
    return clst->blockAddresses[pos];
 | 
						|
}
 | 
						|
 | 
						|
void setNextBlock(struct clientState *clst)
 | 
						|
{
 | 
						|
    clst->nextBlock = getBlockSpace(clst);
 | 
						|
}
 | 
						|
 | 
						|
int setupMessages(struct clientState *clst) {
 | 
						|
    /* the messages received from the server */
 | 
						|
    clst->data_iov[0].iov_base = (void *)&clst->Msg;
 | 
						|
    clst->data_iov[0].iov_len = sizeof(clst->Msg);
 | 
						|
    
 | 
						|
    /* namelen set just before reception */
 | 
						|
    clst->data_hdr.msg_iov = clst->data_iov;
 | 
						|
    /* iovlen set just before reception */
 | 
						|
 | 
						|
    initMsgHdr(&clst->data_hdr);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
struct slice *initSlice(struct clientState *clst, 
 | 
						|
			       struct slice *slice,
 | 
						|
			       int sliceNo)
 | 
						|
{
 | 
						|
    idbassert(slice->state == slice::SLICE_FREE || slice->state == slice::SLICE_RECEIVING);
 | 
						|
 | 
						|
    slice->magic = SLICEMAGIC;
 | 
						|
    slice->state = slice::SLICE_RECEIVING;
 | 
						|
    slice->blocksTransferred = 0;
 | 
						|
    slice->dataBlocksTransferred = 0;
 | 
						|
    BZERO(slice->retransmit);
 | 
						|
    slice->freePos = 0;
 | 
						|
    slice->bytes = 0;
 | 
						|
    if(clst->currentSlice != NULL && !clst->currentSlice->bytesKnown) {
 | 
						|
	//udpc_fatal(1, "Previous slice size not known\n");
 | 
						|
    }
 | 
						|
    if(clst->currentSliceNo != sliceNo-1) {
 | 
						|
	//udpc_fatal(1, "Slice no mismatch %d <-> %d\n", sliceNo, clst->currentSliceNo);
 | 
						|
    }
 | 
						|
    slice->bytesKnown = 0;
 | 
						|
    slice->sliceNo = sliceNo;
 | 
						|
 | 
						|
    BZERO(slice->missing_data_blocks);
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    BZERO(slice->fec_stripes);
 | 
						|
    BZERO(slice->fec_blocks);
 | 
						|
    BZERO(slice->fec_descs);
 | 
						|
#endif
 | 
						|
    clst->currentSlice = slice;
 | 
						|
    clst->currentSliceNo = sliceNo;
 | 
						|
    return slice;
 | 
						|
}
 | 
						|
struct slice *newSlice(struct clientState *clst, int sliceNo)
 | 
						|
{
 | 
						|
    struct slice *slice=NULL;
 | 
						|
    int i;
 | 
						|
 | 
						|
    pc_consume(clst->free_slices_pc, 1);
 | 
						|
    i = pc_getConsumerPosition(clst->free_slices_pc);
 | 
						|
    pc_consumed(clst->free_slices_pc, 1);
 | 
						|
    slice = &clst->slices[i];
 | 
						|
    idbassert(slice->state == slice::SLICE_FREE);
 | 
						|
 | 
						|
    /* wait for free data memory */
 | 
						|
    slice->base = pc_getConsumerPosition(clst->fifo->freeMemQueue);
 | 
						|
    pc_consume(clst->fifo->freeMemQueue, 
 | 
						|
	       clst->net_config->blockSize * MAX_SLICE_SIZE);
 | 
						|
    initSlice(clst, slice, sliceNo);
 | 
						|
    return slice;
 | 
						|
}
 | 
						|
 | 
						|
#define ADR(x, bs) (fifo->dataBuffer + \
 | 
						|
	(slice->base+(x)*bs) % fifo->dataBufSize)
 | 
						|
 | 
						|
void closeAllExcept(struct clientState *clst, int fd) {
 | 
						|
    int i;
 | 
						|
    int *socks = clst->client_config->socks;
 | 
						|
    
 | 
						|
    if(clst->selectedFd >= 0)
 | 
						|
      return;
 | 
						|
 | 
						|
    clst->selectedFd = fd;
 | 
						|
    for(i=1; i<NR_CLIENT_SOCKS; i++)
 | 
						|
	if(socks[i] != -1 && socks[i] != fd)
 | 
						|
	    udpc_closeSock(socks, NR_CLIENT_SOCKS, i);
 | 
						|
}
 | 
						|
 | 
						|
struct slice *rcvFindSlice(struct clientState *clst, int sliceNo)
 | 
						|
{
 | 
						|
    if(! clst->currentSlice) {
 | 
						|
	/* Streaming mode? */
 | 
						|
	clst->currentSliceNo = sliceNo-1;
 | 
						|
	return newSlice(clst, sliceNo);
 | 
						|
    }
 | 
						|
    if(sliceNo <= clst->currentSliceNo) {
 | 
						|
	struct slice *slice = clst->currentSlice;
 | 
						|
	int pos = slice - clst->slices;
 | 
						|
	idbassert(slice == NULL || slice->magic == SLICEMAGIC);
 | 
						|
	while(slice->sliceNo != sliceNo) {
 | 
						|
	    if(slice->state == slice::SLICE_FREE)
 | 
						|
		return NULL;
 | 
						|
	    idbassert(slice->magic == SLICEMAGIC);
 | 
						|
	    pos--;
 | 
						|
	    if(pos < 0)
 | 
						|
		pos += NR_SLICES;
 | 
						|
	    slice = &clst->slices[pos];
 | 
						|
	}
 | 
						|
	return slice;
 | 
						|
    }
 | 
						|
    if(sliceNo != clst->currentSliceNo + 1) {
 | 
						|
	//TODO: FIXME
 | 
						|
	//udpc_flprintf("Slice %d skipped\n", sliceNo-1);
 | 
						|
	//exit(1);
 | 
						|
    }
 | 
						|
 | 
						|
    if((clst->net_config->flags & FLAG_STREAMING) &&
 | 
						|
       sliceNo != clst->currentSliceNo) {
 | 
						|
	return initSlice(clst, clst->currentSlice, sliceNo);	    
 | 
						|
    }
 | 
						|
 | 
						|
    if(sliceNo > clst->receivedSliceNo + 2) {
 | 
						|
 | 
						|
	//TODO: FIXME
 | 
						|
#if 0
 | 
						|
	slice_t slice = rcvFindSlice(clst, clst->receivedSliceNo+1);
 | 
						|
	udpc_flprintf("Dropped by server now=%d last=%d\n", sliceNo, clst->receivedSliceNo);
 | 
						|
	if(slice != NULL)
 | 
						|
	    printMissedBlockMap(clst, slice);
 | 
						|
	exit(1);
 | 
						|
#endif
 | 
						|
    }
 | 
						|
    return newSlice(clst, sliceNo);
 | 
						|
}
 | 
						|
 | 
						|
void setSliceBytes(struct slice *slice, 
 | 
						|
			  struct clientState *clst,
 | 
						|
			  int bytes) {
 | 
						|
    idbassert(slice->magic == SLICEMAGIC);
 | 
						|
    if(slice->bytesKnown) {
 | 
						|
	if(slice->bytes != bytes) {
 | 
						|
		//TODO: FIXME
 | 
						|
	    //udpc_fatal(1, "Byte number mismatch %d <-> %d\n", bytes, slice->bytes);
 | 
						|
	}
 | 
						|
    } else {
 | 
						|
	slice->bytesKnown = 1;
 | 
						|
	slice->bytes = bytes;
 | 
						|
	if(bytes == 0)
 | 
						|
	    clst->netEndReached=1;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void advanceReceivedPointer(struct clientState *clst) {
 | 
						|
    int pos = clst->receivedPtr;
 | 
						|
    while(1) {	
 | 
						|
	slice_t slice = &clst->slices[pos];
 | 
						|
	if(
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	   slice->state != SLICE_FEC &&
 | 
						|
	   slice->state != SLICE_FEC_DONE &&
 | 
						|
#endif
 | 
						|
	   slice->state != slice::SLICE_DONE)
 | 
						|
	    break;
 | 
						|
	pos++;
 | 
						|
	clst->receivedSliceNo = slice->sliceNo;
 | 
						|
	if(pos >= NR_SLICES)
 | 
						|
	    pos -= NR_SLICES;
 | 
						|
    }
 | 
						|
    clst->receivedPtr = pos;
 | 
						|
}
 | 
						|
 | 
						|
void receiverStatsAddBytes(receiver_stats_t rs, long bytes) {
 | 
						|
    if(rs != NULL)
 | 
						|
	rs->totalBytes += bytes;
 | 
						|
}
 | 
						|
 | 
						|
void cleanupSlices(struct clientState *clst, unsigned int doneState)
 | 
						|
{
 | 
						|
    while(1) {	
 | 
						|
	int pos = pc_getProducerPosition(clst->free_slices_pc);
 | 
						|
	int bytes;
 | 
						|
	slice_t slice = &clst->slices[pos];
 | 
						|
	if(slice->state != (signed)doneState)
 | 
						|
	    break;
 | 
						|
	receiverStatsAddBytes(clst->stats, slice->bytes);
 | 
						|
	bytes = slice->bytes;
 | 
						|
 | 
						|
	/* signal data received */
 | 
						|
	if(bytes == 0) {
 | 
						|
	    pc_produceEnd(clst->fifo->data);
 | 
						|
	} else
 | 
						|
	    pc_produce(clst->fifo->data, slice->bytes);
 | 
						|
 | 
						|
	/* free up slice structure */
 | 
						|
	clst->slices[pos].state = slice::SLICE_FREE;
 | 
						|
	pc_produce(clst->free_slices_pc, 1);
 | 
						|
	
 | 
						|
	/* if at end, exit this thread */
 | 
						|
	if(!bytes) {
 | 
						|
	    clst->endReached = 2;
 | 
						|
	}
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void checkSliceComplete(struct clientState *clst,
 | 
						|
			       struct slice *slice)
 | 
						|
{
 | 
						|
    int blocksInSlice;
 | 
						|
 | 
						|
    idbassert(slice->magic == SLICEMAGIC);
 | 
						|
    if(slice->state != slice::SLICE_RECEIVING) 
 | 
						|
	/* bad starting state */
 | 
						|
	return; 
 | 
						|
 | 
						|
    /* is this slice ready ? */
 | 
						|
    idbassert(clst->net_config->blockSize != 0);
 | 
						|
    blocksInSlice = (slice->bytes + clst->net_config->blockSize - 1) / 
 | 
						|
	clst->net_config->blockSize;
 | 
						|
    if(blocksInSlice == slice->blocksTransferred) {
 | 
						|
	pc_consumed(clst->fifo->freeMemQueue, slice->bytes);
 | 
						|
	clst->net_config->flags &= ~FLAG_STREAMING;
 | 
						|
	if(blocksInSlice == slice->dataBlocksTransferred)
 | 
						|
	    slice->state = slice::SLICE_DONE;
 | 
						|
	else {
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	    idbassert(clst->use_fec == 1);
 | 
						|
	    slice->state = SLICE_FEC;
 | 
						|
#else
 | 
						|
	    idbassert(0);
 | 
						|
#endif
 | 
						|
	}
 | 
						|
	advanceReceivedPointer(clst);
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	if(clst->use_fec) {
 | 
						|
	    int n = pc_getProducerPosition(clst->fec_data_pc);
 | 
						|
	    idbassert(slice->state == SLICE_DONE || slice->state == SLICE_FEC);
 | 
						|
	    clst->fec_slices[n] = slice;
 | 
						|
	    pc_produce(clst->fec_data_pc, 1);
 | 
						|
	} else
 | 
						|
#endif
 | 
						|
	    cleanupSlices(clst, slice::SLICE_DONE);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int processDataBlock(struct clientState *clst,
 | 
						|
			    int sliceNo,
 | 
						|
			    int blockNo,
 | 
						|
			    int bytes)
 | 
						|
{
 | 
						|
//cerr << "processDataBlock(): " << sliceNo << ", " << blockNo << ", " << bytes << endl;
 | 
						|
    struct fifo *fifo = clst->fifo;
 | 
						|
    struct slice *slice = rcvFindSlice(clst, sliceNo);
 | 
						|
    unsigned char *shouldAddress, *isAddress;
 | 
						|
 | 
						|
    idbassert(slice == NULL || slice->magic == SLICEMAGIC);
 | 
						|
 | 
						|
    if(slice == NULL || 
 | 
						|
       slice->state == slice::SLICE_FREE ||
 | 
						|
       slice->state == slice::SLICE_DONE
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
       ||
 | 
						|
       slice->state == SLICE_FEC ||
 | 
						|
       slice->state == SLICE_FEC_DONE
 | 
						|
#endif
 | 
						|
       ) {
 | 
						|
	/* an old slice. Ignore */
 | 
						|
//cerr << "ignore" << endl;
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    if(sliceNo > clst->currentSliceNo+2)
 | 
						|
	{
 | 
						|
	//TODO: FIXME
 | 
						|
	//udpc_fatal(1, "We have been dropped by sender\n");
 | 
						|
	}
 | 
						|
 | 
						|
    if(BIT_ISSET(blockNo, slice->retransmit.map)) {
 | 
						|
	/* we already have this packet, ignore */
 | 
						|
#if 0
 | 
						|
	flprintf("Packet %d:%d not for us\n", sliceNo, blockNo);
 | 
						|
#endif
 | 
						|
//cerr << "dup" << endl;
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    if(slice->base % clst->net_config->blockSize) {
 | 
						|
	//TODO: FIXME
 | 
						|
	//udpc_fatal(1, "Bad base %d, not multiple of block size %d\n", slice->base, clst->net_config->blockSize);
 | 
						|
//cerr << "bad base" << endl;
 | 
						|
    }
 | 
						|
//cerr << "good slice" << endl;
 | 
						|
 | 
						|
    shouldAddress = ADR(blockNo, clst->net_config->blockSize);
 | 
						|
    isAddress = (unsigned char*)clst->data_hdr.msg_iov[1].iov_base;
 | 
						|
    if(shouldAddress != isAddress) {
 | 
						|
	/* copy message to the correct place */
 | 
						|
	memcpy(shouldAddress, isAddress,  clst->net_config->blockSize);
 | 
						|
    }
 | 
						|
 | 
						|
    if(clst->client_config->sender_is_newgen && bytes != 0)
 | 
						|
	setSliceBytes(slice, clst, bytes);
 | 
						|
    if(clst->client_config->sender_is_newgen && bytes == 0)
 | 
						|
	clst->netEndReached = 0;
 | 
						|
 | 
						|
    SET_BIT(blockNo, slice->retransmit.map);
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    if(slice->fec_stripes) {
 | 
						|
	int stripe = blockNo % slice->fec_stripes;
 | 
						|
	slice->missing_data_blocks[stripe]--;
 | 
						|
	idbassert(slice->missing_data_blocks[stripe] >= 0);
 | 
						|
	if(slice->missing_data_blocks[stripe] <
 | 
						|
	   slice->fec_blocks[stripe]) {
 | 
						|
	    int blockIdx;
 | 
						|
	    /* FIXME: FEC block should be enqueued in local queue here...*/
 | 
						|
	    slice->fec_blocks[stripe]--;
 | 
						|
	    blockIdx = stripe+slice->fec_blocks[stripe]*slice->fec_stripes;
 | 
						|
	    idbassert(slice->fec_descs[blockIdx].adr != 0);
 | 
						|
	    clst->localBlockAddresses[clst->localPos++] = 
 | 
						|
		slice->fec_descs[blockIdx].adr;
 | 
						|
	    slice->fec_descs[blockIdx].adr=0;
 | 
						|
	    slice->blocksTransferred--;
 | 
						|
	}
 | 
						|
    }
 | 
						|
#endif
 | 
						|
    slice->dataBlocksTransferred++;
 | 
						|
    slice->blocksTransferred++;
 | 
						|
    while(slice->freePos < MAX_SLICE_SIZE && 
 | 
						|
	  BIT_ISSET(slice->freePos, slice->retransmit.map))
 | 
						|
	slice->freePos++;
 | 
						|
    checkSliceComplete(clst, slice);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int sendOk(struct client_config *client_config, unsigned int sliceNo)
 | 
						|
{
 | 
						|
    struct ok ok;
 | 
						|
    ok.opCode = htons(CMD_OK);
 | 
						|
    ok.reserved = 0;
 | 
						|
    ok.sliceNo = htonl(sliceNo);
 | 
						|
    return SSEND(ok);
 | 
						|
}
 | 
						|
 | 
						|
int sendRetransmit(struct clientState *clst,
 | 
						|
			  struct slice *slice,
 | 
						|
			  int rxmit) {
 | 
						|
    struct client_config *client_config = clst->client_config;
 | 
						|
 | 
						|
    idbassert(slice->magic == SLICEMAGIC);
 | 
						|
    slice->retransmit.opCode = htons(CMD_RETRANSMIT);
 | 
						|
    slice->retransmit.reserved = 0;
 | 
						|
    slice->retransmit.sliceNo = htonl(slice->sliceNo);
 | 
						|
    slice->retransmit.rxmit = htonl(rxmit);
 | 
						|
    return SSEND(slice->retransmit);
 | 
						|
}
 | 
						|
 | 
						|
int processReqAck(struct clientState *clst,
 | 
						|
			 int sliceNo, int bytes, int rxmit)
 | 
						|
{   
 | 
						|
    struct slice *slice = rcvFindSlice(clst, sliceNo);
 | 
						|
    int blocksInSlice;
 | 
						|
    char *readySet = (char *) clst->data_hdr.msg_iov[1].iov_base;
 | 
						|
 | 
						|
    idbassert(slice == NULL || slice->magic == SLICEMAGIC);
 | 
						|
 | 
						|
    {
 | 
						|
	struct timeval tv;
 | 
						|
	gettimeofday(&tv, 0);
 | 
						|
	/* usleep(1); DEBUG: FIXME */
 | 
						|
    }
 | 
						|
    if(BIT_ISSET(clst->client_config->clientNumber, readySet)) {
 | 
						|
	/* not for us */
 | 
						|
	return 0;
 | 
						|
    }
 | 
						|
 | 
						|
    if(slice == NULL) {
 | 
						|
	/* an old slice => send ok */
 | 
						|
	return sendOk(clst->client_config, sliceNo);
 | 
						|
    }
 | 
						|
 | 
						|
    setSliceBytes(slice, clst, bytes);
 | 
						|
    idbassert(clst->net_config->blockSize != 0);
 | 
						|
    blocksInSlice = (slice->bytes + clst->net_config->blockSize - 1) / 
 | 
						|
	clst->net_config->blockSize;
 | 
						|
    if (blocksInSlice == slice->blocksTransferred) {
 | 
						|
	/* send ok */
 | 
						|
	sendOk(clst->client_config, slice->sliceNo);
 | 
						|
    } else {
 | 
						|
	sendRetransmit(clst, slice, rxmit);
 | 
						|
    }
 | 
						|
    checkSliceComplete(clst, slice); /* needed for the final 0 sized slice */
 | 
						|
    advanceReceivedPointer(clst);
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    if(!clst->use_fec)
 | 
						|
	cleanupSlices(clst, SLICE_DONE);
 | 
						|
#endif
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int udpc_isAddressEqual(struct sockaddr_in *a, struct sockaddr_in *b) {
 | 
						|
    return !memcmp((char *) a, (char *)b, 8);
 | 
						|
}
 | 
						|
int dispatchMessage(struct clientState *clst)
 | 
						|
{
 | 
						|
    int ret;
 | 
						|
    struct sockaddr_in lserver;
 | 
						|
    struct fifo *fifo = clst->fifo;
 | 
						|
    int fd = -1;
 | 
						|
    struct client_config *client_config = clst->client_config;
 | 
						|
 | 
						|
    /* set up message header */
 | 
						|
    if (clst->currentSlice != NULL &&
 | 
						|
	clst->currentSlice->freePos < MAX_SLICE_SIZE) {
 | 
						|
	struct slice *slice = clst->currentSlice;
 | 
						|
	idbassert(slice == NULL || slice->magic == SLICEMAGIC);
 | 
						|
	clst->data_iov[1].iov_base = 
 | 
						|
	    ADR(slice->freePos, clst->net_config->blockSize);
 | 
						|
    } else {
 | 
						|
	clst->data_iov[1].iov_base = clst->nextBlock;
 | 
						|
    }
 | 
						|
 | 
						|
    clst->data_iov[1].iov_len = clst->net_config->blockSize;
 | 
						|
    clst->data_hdr.msg_iovlen = 2;
 | 
						|
 | 
						|
    clst->data_hdr.msg_name = &lserver;
 | 
						|
    clst->data_hdr.msg_namelen = sizeof(struct sockaddr_in);
 | 
						|
 | 
						|
    while(clst->endReached || clst->netEndReached) {
 | 
						|
	int oldEndReached = clst->endReached;
 | 
						|
	int nr_desc;
 | 
						|
	struct timeval tv;
 | 
						|
	fd_set read_set;
 | 
						|
 | 
						|
	int maxFd = prepareForSelect(client_config->socks,
 | 
						|
				     NR_CLIENT_SOCKS, &read_set);
 | 
						|
 | 
						|
	tv.tv_sec = clst->net_config->exitWait / 1000;
 | 
						|
	tv.tv_usec = (clst->net_config->exitWait % 1000)*1000;
 | 
						|
	nr_desc = select(maxFd,  &read_set, 0, 0,  &tv);
 | 
						|
	if(nr_desc < 0) {
 | 
						|
	    break;
 | 
						|
	}
 | 
						|
	fd = udpc_getSelectedSock(client_config->socks, NR_CLIENT_SOCKS, &read_set);
 | 
						|
	if(fd >= 0)
 | 
						|
	  break;
 | 
						|
 | 
						|
	/* Timeout expired */
 | 
						|
	if(oldEndReached >= 2) {
 | 
						|
	    clst->endReached = 3;
 | 
						|
	    return 0;
 | 
						|
	}
 | 
						|
    }
 | 
						|
 | 
						|
    if(fd < 0)
 | 
						|
	fd = clst->selectedFd;
 | 
						|
 | 
						|
    if(fd < 0) {
 | 
						|
	struct timeval tv, *tvp;
 | 
						|
	fd_set read_set;
 | 
						|
	int maxFd = prepareForSelect(client_config->socks,
 | 
						|
				     NR_CLIENT_SOCKS, &read_set);
 | 
						|
	clst->promptPrinted=1;
 | 
						|
 | 
						|
	if(clst->net_config->startTimeout == 0) {
 | 
						|
	    tvp=NULL;
 | 
						|
	} else {
 | 
						|
	    tv.tv_sec = clst->net_config->startTimeout;
 | 
						|
	    tv.tv_usec = 0;
 | 
						|
	    tvp = &tv;
 | 
						|
	}
 | 
						|
 | 
						|
//cerr << "waiting for data..." << endl;
 | 
						|
	ret = selectWithoutConsole(maxFd+1, &read_set, tvp);
 | 
						|
	if(ret < 0) {
 | 
						|
	  perror("Select");
 | 
						|
	  return 0;
 | 
						|
	}
 | 
						|
	if(ret == 0) {
 | 
						|
		clst->endReached=3;
 | 
						|
		clst->netEndReached=3;
 | 
						|
		pc_produceEnd(clst->fifo->data);
 | 
						|
		return 1;
 | 
						|
	}
 | 
						|
	fd = udpc_getSelectedSock(clst->client_config->socks,
 | 
						|
			     NR_CLIENT_SOCKS, &read_set);
 | 
						|
    }
 | 
						|
 | 
						|
#ifdef LOSSTEST
 | 
						|
    loseRecvPacket(fd);
 | 
						|
    ret=RecvMsg(fd, &clst->data_hdr, 0);
 | 
						|
#else
 | 
						|
    ret=recvmsg(fd, &clst->data_hdr, 0);
 | 
						|
//cerr << "got some data on fd " << fd << endl;
 | 
						|
#endif
 | 
						|
    if (ret < 0) {
 | 
						|
#if DEBUG
 | 
						|
	flprintf("data recvfrom %d: %s\n", fd, strerror(errno));
 | 
						|
#endif
 | 
						|
	return -1;
 | 
						|
    }
 | 
						|
 | 
						|
#if 0
 | 
						|
    fprintf(stderr, "received packet for slice %d, block %d\n", 
 | 
						|
	     ntohl(Msg.sliceNo), ntohs(db.blockNo));
 | 
						|
#endif
 | 
						|
 | 
						|
    if(!udpc_isAddressEqual(&lserver, 
 | 
						|
			    &clst->client_config->serverAddr)) {
 | 
						|
	return -1;
 | 
						|
    }
 | 
						|
 | 
						|
    switch(ntohs(clst->Msg.opCode)) {
 | 
						|
	case CMD_DATA:
 | 
						|
//cerr << "got CMD_DATA" << endl;
 | 
						|
	    closeAllExcept(clst, fd);
 | 
						|
	    //udpc_receiverStatsStartTimer(clst->stats);
 | 
						|
	    clst->client_config->isStarted = 1;
 | 
						|
	    return processDataBlock(clst,
 | 
						|
				    ntohl(clst->Msg.dataBlock.sliceNo),
 | 
						|
				    ntohs(clst->Msg.dataBlock.blockNo),
 | 
						|
				    ntohl(clst->Msg.dataBlock.bytes));
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
	case CMD_FEC:
 | 
						|
//cerr << "got CMD_FEC" << endl;
 | 
						|
	    closeAllExcept(clst, fd);
 | 
						|
	    //receiverStatsStartTimer(clst->stats);
 | 
						|
	    clst->client_config->isStarted = 1;
 | 
						|
	    return processFecBlock(clst,
 | 
						|
				   ntohs(clst->Msg.fecBlock.stripes),
 | 
						|
				   ntohl(clst->Msg.fecBlock.sliceNo),
 | 
						|
				   ntohs(clst->Msg.fecBlock.blockNo),
 | 
						|
				   ntohl(clst->Msg.fecBlock.bytes));
 | 
						|
#endif
 | 
						|
	case CMD_REQACK:
 | 
						|
//cerr << "got CMD_REQACK" << endl;
 | 
						|
	    closeAllExcept(clst, fd);
 | 
						|
	    //receiverStatsStartTimer(clst->stats);
 | 
						|
	    clst->client_config->isStarted = 1;
 | 
						|
	    return processReqAck(clst,
 | 
						|
				 ntohl(clst->Msg.reqack.sliceNo),
 | 
						|
				 ntohl(clst->Msg.reqack.bytes),
 | 
						|
				 ntohl(clst->Msg.reqack.rxmit));
 | 
						|
	case CMD_HELLO_STREAMING:
 | 
						|
	case CMD_HELLO_NEW:
 | 
						|
	case CMD_HELLO:
 | 
						|
//cerr << "got CMD_HELLO" << endl;
 | 
						|
	    /* retransmission of hello to find other participants ==> ignore */
 | 
						|
	    return 0;
 | 
						|
	default:
 | 
						|
	    break;
 | 
						|
    }
 | 
						|
 | 
						|
    return -1;
 | 
						|
}
 | 
						|
 | 
						|
THREAD_RETURN netReceiverMain(void *args0)
 | 
						|
{    
 | 
						|
    struct clientState *clst = (struct clientState *) args0;
 | 
						|
 | 
						|
    clst->currentSliceNo = 0;
 | 
						|
    setupMessages(clst);
 | 
						|
    
 | 
						|
    clst->currentSliceNo = -1;
 | 
						|
    clst->currentSlice = NULL;
 | 
						|
    clst->promptPrinted = 0;
 | 
						|
    if(! (clst->net_config->flags & FLAG_STREAMING))
 | 
						|
	newSlice(clst, 0);
 | 
						|
    else {
 | 
						|
	clst->currentSlice = NULL;
 | 
						|
	clst->currentSliceNo = 0;
 | 
						|
    }
 | 
						|
 | 
						|
    while(clst->endReached < 3) {
 | 
						|
	dispatchMessage(clst);
 | 
						|
    }
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    if(clst->use_fec)
 | 
						|
	pthread_join(clst->fec_thread, NULL);
 | 
						|
#endif
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int spawnNetReceiver(struct fifo *fifo,
 | 
						|
		     struct client_config *client_config,
 | 
						|
		     struct net_config *net_config)
 | 
						|
{
 | 
						|
    int i;
 | 
						|
    struct clientState  *clst = MALLOC(struct clientState);
 | 
						|
    clst->fifo = fifo;
 | 
						|
    clst->client_config = client_config;
 | 
						|
    clst->net_config = net_config;
 | 
						|
    //clst->stats = stats;
 | 
						|
    clst->endReached = 0;
 | 
						|
    clst->netEndReached = 0;
 | 
						|
    clst->selectedFd = -1;
 | 
						|
 | 
						|
    clst->free_slices_pc = pc_makeProduconsum(NR_SLICES, "free slices");
 | 
						|
    pc_produce(clst->free_slices_pc, NR_SLICES);
 | 
						|
    for(i = 0; i <NR_SLICES; i++)
 | 
						|
	clst->slices[i].state = slice::SLICE_FREE;
 | 
						|
    clst->receivedPtr = 0;
 | 
						|
    clst->receivedSliceNo = 0;
 | 
						|
 | 
						|
#ifdef BB_FEATURE_UDPCAST_FEC
 | 
						|
    fec_init(); /* fec new involves memory
 | 
						|
		 * allocation. Better do it here */
 | 
						|
    clst->use_fec = 0;
 | 
						|
    clst->fec_data_pc = pc_makeProduconsum(NR_SLICES, "fec data");
 | 
						|
#endif
 | 
						|
 | 
						|
#define NR_BLOCKS 4096
 | 
						|
    clst->freeBlocks_pc = pc_makeProduconsum(NR_BLOCKS, "free blocks");
 | 
						|
    pc_produce(clst->freeBlocks_pc, NR_BLOCKS);
 | 
						|
    clst->blockAddresses = (unsigned char**)calloc(NR_BLOCKS, sizeof(char *));
 | 
						|
    clst->localBlockAddresses = (unsigned char**)calloc(NR_BLOCKS, sizeof(char *));
 | 
						|
    clst->blockData = (unsigned char*)malloc(NR_BLOCKS * net_config->blockSize);
 | 
						|
    for(i = 0; i < NR_BLOCKS; i++)
 | 
						|
	clst->blockAddresses[i] = clst->blockData + i * net_config->blockSize;
 | 
						|
    clst->localPos=0;
 | 
						|
 | 
						|
    setNextBlock(clst);
 | 
						|
    return pthread_create(&client_config->thread, NULL, netReceiverMain, clst);
 | 
						|
}
 | 
						|
 | 
						|
unsigned int pc_getSize(produconsum_t pc) {
 | 
						|
    return pc->size;
 | 
						|
}
 | 
						|
 | 
						|
int writer(struct fifo *fifo, SBS outbs) {
 | 
						|
//cerr << "start appending to outbs" << endl;
 | 
						|
	outbs->restart();
 | 
						|
    int fifoSize = pc_getSize(fifo->data);
 | 
						|
    while(1) {
 | 
						|
	int pos=pc_getConsumerPosition(fifo->data);
 | 
						|
	int bytes = pc_consumeContiguousMinAmount(fifo->data, BLOCKSIZE);
 | 
						|
	if (bytes == 0) {
 | 
						|
//cerr << "done appending to outbs: " << outbs->length() << endl;
 | 
						|
	    return 0;
 | 
						|
	}
 | 
						|
 | 
						|
	/*
 | 
						|
	 * If we have more than blocksize, round down to nearest blocksize
 | 
						|
	 * multiple
 | 
						|
	 */
 | 
						|
	if(pos + bytes != fifoSize && 
 | 
						|
	   bytes > (pos + bytes) % BLOCKSIZE)
 | 
						|
	    bytes -= (pos + bytes) % BLOCKSIZE;
 | 
						|
 | 
						|
	/* make sure we don't write to big a chunk... Better to
 | 
						|
	 * liberate small chunks one by one rather than attempt to
 | 
						|
	 * write out a bigger chunk and block reception for too
 | 
						|
	 * long */
 | 
						|
	if (bytes > 128 * 1024)
 | 
						|
	    bytes = 64 * 1024;
 | 
						|
 | 
						|
	//bytes = write(outFile, fifo->dataBuffer + pos, bytes);
 | 
						|
//cerr << "appending " << bytes << " bytes to outbs..." << endl;
 | 
						|
	outbs->append(fifo->dataBuffer + pos, bytes);
 | 
						|
	pc_consumed(fifo->data, bytes);
 | 
						|
	pc_produce(fifo->freeMemQueue, bytes);
 | 
						|
    }
 | 
						|
}
 | 
						|
}
 | 
						|
 | 
						|
namespace multicast
 | 
						|
{
 | 
						|
 | 
						|
MulticastImpl::MulticastImpl(int min_receivers, const string& ifName, int portBase, int bufSize) :
 | 
						|
	fIfName(ifName),
 | 
						|
	fDb(0)
 | 
						|
{
 | 
						|
	udpc_clearIp(&fNet_config.dataMcastAddr);
 | 
						|
	fNet_config.mcastRdv = 0;
 | 
						|
	fNet_config.blockSize = 1024; //1456;
 | 
						|
	fNet_config.sliceSize = 16;
 | 
						|
	fNet_config.portBase = portBase;
 | 
						|
	fNet_config.nrGovernors = 0;
 | 
						|
	fNet_config.flags = FLAG_NOKBD;
 | 
						|
	fNet_config.capabilities = 0;
 | 
						|
	fNet_config.min_slice_size = 16;
 | 
						|
	fNet_config.max_slice_size = 1024;
 | 
						|
	fNet_config.default_slice_size = 0;
 | 
						|
	fNet_config.ttl = 1;
 | 
						|
	fNet_config.rexmit_hello_interval = 2000;
 | 
						|
	fNet_config.autostart = 0;
 | 
						|
	fNet_config.requestedBufSize = bufSize;
 | 
						|
 | 
						|
	fNet_config.min_receivers = min_receivers;
 | 
						|
	fNet_config.max_receivers_wait=0;
 | 
						|
	fNet_config.min_receivers_wait=0;
 | 
						|
 | 
						|
	fNet_config.retriesUntilDrop = 200;
 | 
						|
 | 
						|
	fNet_config.rehelloOffset = 50;
 | 
						|
 | 
						|
	fStat_config.log = 0;
 | 
						|
	fStat_config.bwPeriod = 0;
 | 
						|
	fStat_config.printUncompressedPos = -1;
 | 
						|
	fStat_config.statPeriod = DEFLT_STAT_PERIOD;
 | 
						|
 | 
						|
	fNet_config.net_if = 0;
 | 
						|
 | 
						|
	//full-duplex
 | 
						|
	fNet_config.flags |= FLAG_SN;
 | 
						|
 | 
						|
	if (fIfName.empty())
 | 
						|
		fIfName = "eth0";
 | 
						|
 | 
						|
	fSock[0] = -1;
 | 
						|
	fSock[1] = -1;
 | 
						|
	fSock[2] = -1;
 | 
						|
}
 | 
						|
 | 
						|
MulticastImpl::~MulticastImpl()
 | 
						|
{
 | 
						|
	delete fDb;
 | 
						|
 | 
						|
	for (int i=0; i<3; i++)
 | 
						|
	{
 | 
						|
		if (fSock[i] >= 0)
 | 
						|
		{
 | 
						|
			shutdown(fSock[i], SHUT_RDWR);
 | 
						|
			close(fSock[i]);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void MulticastImpl::startSender()
 | 
						|
{
 | 
						|
    int tries;
 | 
						|
    time_t firstConnected = 0;
 | 
						|
    time_t *firstConnectedP;
 | 
						|
 | 
						|
    //participantsDb_t db;
 | 
						|
 | 
						|
    /* make the socket and print banner */
 | 
						|
    //int fSock[3];
 | 
						|
    int nr=0;
 | 
						|
    int fd;
 | 
						|
    int r;
 | 
						|
    int j;
 | 
						|
 | 
						|
    fNet_config.net_if = udpc_getNetIf(fIfName.c_str());
 | 
						|
 | 
						|
    fSock[nr++] = udpc_makeSocket(ADDR_TYPE_UCAST,
 | 
						|
			    fNet_config.net_if,
 | 
						|
			    NULL,
 | 
						|
			    SENDER_PORT(fNet_config.portBase));
 | 
						|
//cerr << "sock[" << (nr-1) << "] = " << fSock[(nr-1)] << endl;
 | 
						|
 | 
						|
    if(! (fNet_config.flags & (FLAG_SN | FLAG_NOTSN)) ) {
 | 
						|
      if(udpc_isFullDuplex(fSock[0], fNet_config.net_if->name) == 1) {
 | 
						|
	fNet_config.flags |= FLAG_SN;
 | 
						|
      }
 | 
						|
    }
 | 
						|
    
 | 
						|
    fd = udpc_makeSocket(ADDR_TYPE_BCAST,
 | 
						|
		    fNet_config.net_if,
 | 
						|
		    NULL,
 | 
						|
		    SENDER_PORT(fNet_config.portBase));
 | 
						|
    if(fd >= 0)
 | 
						|
	fSock[nr++] = fd;
 | 
						|
//cerr << "sock[" << (nr-1) << "] = " << fSock[(nr-1)] << endl;
 | 
						|
 | 
						|
    if(fNet_config.requestedBufSize)
 | 
						|
	udpc_setSendBuf(fSock[0], fNet_config.requestedBufSize);
 | 
						|
 | 
						|
    fNet_config.controlMcastAddr.sin_addr.s_addr =0;
 | 
						|
    if(fNet_config.ttl == 1 && fNet_config.mcastRdv == NULL) {
 | 
						|
	udpc_getBroadCastAddress(fNet_config.net_if,
 | 
						|
			    &fNet_config.controlMcastAddr,
 | 
						|
			    RECEIVER_PORT(fNet_config.portBase));
 | 
						|
	udpc_setSocketToBroadcast(fSock[0]);
 | 
						|
    } 
 | 
						|
 | 
						|
    if(fNet_config.controlMcastAddr.sin_addr.s_addr == 0) {
 | 
						|
	udpc_getMcastAllAddress(&fNet_config.controlMcastAddr,
 | 
						|
			   fNet_config.mcastRdv,
 | 
						|
			   RECEIVER_PORT(fNet_config.portBase));
 | 
						|
	/* Only do the following if controlMcastAddr is indeed an
 | 
						|
	   mcast address ... */
 | 
						|
	if(isMcastAddress(&fNet_config.controlMcastAddr)) {
 | 
						|
	    udpc_setMcastDestination(fSock[0], fNet_config.net_if,
 | 
						|
				&fNet_config.controlMcastAddr);
 | 
						|
	    udpc_setTtl(fSock[0], fNet_config.ttl);
 | 
						|
	    fSock[nr++] = udpc_makeSocket(ADDR_TYPE_MCAST,
 | 
						|
				    fNet_config.net_if,
 | 
						|
				    &fNet_config.controlMcastAddr,
 | 
						|
				    SENDER_PORT(fNet_config.portBase));
 | 
						|
//cerr << "sock[" << (nr-1) << "] = " << fSock[(nr-1)] << endl;
 | 
						|
	}
 | 
						|
    }
 | 
						|
 | 
						|
    if(!(fNet_config.flags & FLAG_POINTOPOINT) &&
 | 
						|
       udpc_ipIsZero(&fNet_config.dataMcastAddr)) {
 | 
						|
	udpc_getDefaultMcastAddress(fNet_config.net_if, 
 | 
						|
			       &fNet_config.dataMcastAddr);
 | 
						|
    }
 | 
						|
 | 
						|
    if(fNet_config.flags & FLAG_POINTOPOINT) {
 | 
						|
	udpc_clearIp(&fNet_config.dataMcastAddr);
 | 
						|
    }
 | 
						|
 | 
						|
    setPort(&fNet_config.dataMcastAddr, RECEIVER_PORT(fNet_config.portBase));
 | 
						|
 | 
						|
    fNet_config.capabilities = SENDER_CAPABILITIES;
 | 
						|
    if(fNet_config.flags & FLAG_ASYNC)
 | 
						|
	fNet_config.capabilities |= CAP_ASYNC;
 | 
						|
 | 
						|
    udpc_sendHello(&fNet_config, fSock[0], 0);
 | 
						|
 | 
						|
    fDb = udpc_makeParticipantsDb();
 | 
						|
    tries = 0;
 | 
						|
 | 
						|
    if(fNet_config.min_receivers || fNet_config.min_receivers_wait ||
 | 
						|
       fNet_config.max_receivers_wait)
 | 
						|
	firstConnectedP = &firstConnected;
 | 
						|
    else
 | 
						|
	firstConnectedP = NULL;	
 | 
						|
 | 
						|
    while (!(r = mainDispatcher(fSock, nr, fDb, &fNet_config, &tries, firstConnectedP)))
 | 
						|
        ;
 | 
						|
 | 
						|
    for(j=1; j<nr; j++)
 | 
						|
      if(fSock[j] != fSock[0])
 | 
						|
	closesocket(fSock[j]);
 | 
						|
 | 
						|
    if(r == 1) {
 | 
						|
	int i;
 | 
						|
	for(i=1; i<nr; i++)
 | 
						|
	    udpc_closeSock(fSock, nr, i);
 | 
						|
    }
 | 
						|
	//doTransfer(fSock[0], fDb, &fNet_config, &fStat_config);
 | 
						|
}
 | 
						|
 | 
						|
void MulticastImpl::doTransfer(const uint8_t* buf, uint32_t len)
 | 
						|
{
 | 
						|
    int i;
 | 
						|
    int ret;
 | 
						|
    struct fifo fifo;
 | 
						|
    int isPtP = isPointToPoint(fDb, fNet_config.flags);
 | 
						|
 | 
						|
    fNet_config.rcvbuf=0;
 | 
						|
 | 
						|
    for(i=0; i<MAX_CLIENTS; i++)
 | 
						|
	if(udpc_isParticipantValid(fDb, i)) {
 | 
						|
	    unsigned int pRcvBuf = udpc_getParticipantRcvBuf(fDb, i);
 | 
						|
	    if(isPtP)
 | 
						|
		udpc_copyIpFrom(&fNet_config.dataMcastAddr, 
 | 
						|
			   udpc_getParticipantIp(fDb,i));
 | 
						|
	    fNet_config.capabilities &= 
 | 
						|
		udpc_getParticipantCapabilities(fDb, i);
 | 
						|
	    if(pRcvBuf != 0 && 
 | 
						|
	       (fNet_config.rcvbuf == 0 || fNet_config.rcvbuf > pRcvBuf))
 | 
						|
		fNet_config.rcvbuf = pRcvBuf;
 | 
						|
	}
 | 
						|
 | 
						|
    if(isMcastAddress(&fNet_config.dataMcastAddr))
 | 
						|
	udpc_setMcastDestination(fSock[0], fNet_config.net_if, 
 | 
						|
			    &fNet_config.dataMcastAddr);
 | 
						|
 | 
						|
    if(! (fNet_config.capabilities & CAP_BIG_ENDIAN))
 | 
						|
	{
 | 
						|
		//TODO: FIXME
 | 
						|
	//udpc_fatal(1, "Peer with incompatible endianness");
 | 
						|
	}
 | 
						|
 | 
						|
    if(! (fNet_config.capabilities & CAP_NEW_GEN)) {
 | 
						|
       fNet_config.dataMcastAddr = fNet_config.controlMcastAddr;
 | 
						|
       fNet_config.flags &= ~(FLAG_SN | FLAG_ASYNC);
 | 
						|
    }
 | 
						|
 | 
						|
    if(fNet_config.flags & FLAG_BCAST)
 | 
						|
       fNet_config.dataMcastAddr = fNet_config.controlMcastAddr;
 | 
						|
 | 
						|
    udpc_initFifo(&fifo, fNet_config.blockSize);
 | 
						|
    ret = spawnNetSender(&fifo, fSock[0], &fNet_config, fDb);
 | 
						|
    localReader(&fifo, buf, len);
 | 
						|
 | 
						|
    pthread_join(fifo.thread, NULL);    
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
void MulticastImpl::startReceiver()
 | 
						|
{
 | 
						|
    union serverControlMsg Msg;
 | 
						|
    int connectReqSent=0;
 | 
						|
    struct sockaddr_in myIp;
 | 
						|
    int haveServerAddress;
 | 
						|
 | 
						|
    fClient_config.sender_is_newgen = 0;
 | 
						|
 | 
						|
    fNet_config.net_if = udpc_getNetIf(fIfName.c_str());
 | 
						|
{
 | 
						|
fprintf(stderr, "net_if:\n\taddr = 0x%x\n\tbcast = 0x%x\n\tname = %s\n\tindex = %d\n", fNet_config.net_if->addr.s_addr, fNet_config.net_if->bcast.s_addr, fNet_config.net_if->name, fNet_config.net_if->index);
 | 
						|
}
 | 
						|
 | 
						|
    udpc_zeroSockArray(fClient_config.socks, NR_CLIENT_SOCKS);
 | 
						|
 | 
						|
    fClient_config.S_UCAST = udpc_makeSocket(ADDR_TYPE_UCAST,
 | 
						|
				       fNet_config.net_if,
 | 
						|
				       0, RECEIVER_PORT(fNet_config.portBase));
 | 
						|
//cerr << "S_UCAST = " << fClient_config.S_UCAST << endl;
 | 
						|
    fClient_config.S_BCAST = udpc_makeSocket(ADDR_TYPE_BCAST,
 | 
						|
				       fNet_config.net_if,
 | 
						|
				       0, RECEIVER_PORT(fNet_config.portBase));
 | 
						|
//cerr << "S_BCAST = " << fClient_config.S_BCAST << endl;
 | 
						|
 | 
						|
    if(fNet_config.ttl == 1 && fNet_config.mcastRdv == NULL) {
 | 
						|
	udpc_getBroadCastAddress(fNet_config.net_if,
 | 
						|
			    &fNet_config.controlMcastAddr,
 | 
						|
			    SENDER_PORT(fNet_config.portBase));
 | 
						|
	udpc_setSocketToBroadcast(fClient_config.S_UCAST);
 | 
						|
    } else {
 | 
						|
	udpc_getMcastAllAddress(&fNet_config.controlMcastAddr,
 | 
						|
			   fNet_config.mcastRdv,
 | 
						|
			   SENDER_PORT(fNet_config.portBase));
 | 
						|
	if(isMcastAddress(&fNet_config.controlMcastAddr)) {
 | 
						|
	    udpc_setMcastDestination(fClient_config.S_UCAST, fNet_config.net_if,
 | 
						|
				&fNet_config.controlMcastAddr);
 | 
						|
	    udpc_setTtl(fClient_config.S_UCAST, fNet_config.ttl);
 | 
						|
	    
 | 
						|
	    fClient_config.S_MCAST_CTRL =
 | 
						|
		udpc_makeSocket(ADDR_TYPE_MCAST,
 | 
						|
			   fNet_config.net_if,
 | 
						|
			   &fNet_config.controlMcastAddr,
 | 
						|
			   RECEIVER_PORT(fNet_config.portBase));
 | 
						|
//cerr << "S_MCAST_CTRL = " << fClient_config.S_MCAST_CTRL << endl;
 | 
						|
	    // TODO: subscribe address as receiver to!
 | 
						|
	}
 | 
						|
    }
 | 
						|
    udpc_clearIp(&fNet_config.dataMcastAddr);
 | 
						|
 | 
						|
    connectReqSent = 0;
 | 
						|
    haveServerAddress = 0;
 | 
						|
 | 
						|
    fClient_config.clientNumber= 0; /*default number for asynchronous transfer*/
 | 
						|
    while(1) {
 | 
						|
	// int len;
 | 
						|
	int msglen;
 | 
						|
	int sock;
 | 
						|
 | 
						|
	if (!connectReqSent) {
 | 
						|
	    if (sendConnectReq(&fClient_config, &fNet_config,
 | 
						|
			       haveServerAddress) < 0) {
 | 
						|
		//TODO: FIXME
 | 
						|
		//perror("sendto to locate server");
 | 
						|
	    }
 | 
						|
	    connectReqSent = 1;
 | 
						|
	}
 | 
						|
 | 
						|
	haveServerAddress=0;
 | 
						|
 | 
						|
//cerr << "waiting for msg..." << flush << endl;
 | 
						|
	sock = udpc_selectSock(fClient_config.socks, NR_CLIENT_SOCKS,
 | 
						|
			       fNet_config.startTimeout);
 | 
						|
//cerr << "got something" << endl;
 | 
						|
	if(sock < 0) {
 | 
						|
		//TODO: FIXME
 | 
						|
	}
 | 
						|
 | 
						|
	// len = sizeof(server);
 | 
						|
	msglen=RECV(sock, 
 | 
						|
		    Msg, fClient_config.serverAddr, fNet_config.portBase);
 | 
						|
	if (msglen < 0) {
 | 
						|
		//TODO: FIXME
 | 
						|
	    //perror("recvfrom to locate server");
 | 
						|
	    //exit(1);
 | 
						|
	}
 | 
						|
	
 | 
						|
	if(udpc_getPort(&fClient_config.serverAddr) != 
 | 
						|
	   SENDER_PORT(fNet_config.portBase))
 | 
						|
	    /* not from the right port */
 | 
						|
	    continue;
 | 
						|
 | 
						|
	switch(ntohs(Msg.opCode)) {
 | 
						|
	    case CMD_CONNECT_REPLY:
 | 
						|
//cerr << "got conrep" << endl;
 | 
						|
		fClient_config.clientNumber = ntohl(Msg.connectReply.clNr);
 | 
						|
		fNet_config.blockSize = ntohl(Msg.connectReply.blockSize);
 | 
						|
 | 
						|
		if(ntohl(Msg.connectReply.capabilities) & CAP_NEW_GEN) {
 | 
						|
		    fClient_config.sender_is_newgen = 1;
 | 
						|
		    udpc_copyFromMessage(&fNet_config.dataMcastAddr,
 | 
						|
				    Msg.connectReply.mcastAddr);
 | 
						|
		}
 | 
						|
		if (fClient_config.clientNumber == -1) {
 | 
						|
		//TODO: FIXME
 | 
						|
		    //udpc_fatal(1, "Too many clients already connected\n");
 | 
						|
		}
 | 
						|
		goto break_loop;
 | 
						|
 | 
						|
	    case CMD_HELLO_STREAMING:
 | 
						|
	    case CMD_HELLO_NEW:
 | 
						|
	    case CMD_HELLO:
 | 
						|
//cerr << "got hello" << endl;
 | 
						|
		connectReqSent = 0;
 | 
						|
		if(ntohs(Msg.opCode) == CMD_HELLO_STREAMING)
 | 
						|
			fNet_config.flags |= FLAG_STREAMING;
 | 
						|
		if(ntohl(Msg.hello.capabilities) & CAP_NEW_GEN) {
 | 
						|
		    fClient_config.sender_is_newgen = 1;
 | 
						|
		    udpc_copyFromMessage(&fNet_config.dataMcastAddr,
 | 
						|
				    Msg.hello.mcastAddr);
 | 
						|
		    fNet_config.blockSize = ntohs(Msg.hello.blockSize);
 | 
						|
		    if(ntohl(Msg.hello.capabilities) & CAP_ASYNC)
 | 
						|
			fNet_config.flags |= FLAG_PASSIVE;
 | 
						|
		    if(fNet_config.flags & FLAG_PASSIVE)
 | 
						|
			goto break_loop;
 | 
						|
		}
 | 
						|
		haveServerAddress=1;
 | 
						|
		continue;
 | 
						|
	    case CMD_CONNECT_REQ:
 | 
						|
	    case CMD_DATA:
 | 
						|
	    case CMD_FEC:
 | 
						|
		continue;
 | 
						|
	    default:
 | 
						|
		break;
 | 
						|
	}
 | 
						|
 | 
						|
		//TODO: FIXME
 | 
						|
	//udpc_fatal(1, "Bad server reply %04x. Other transfer in progress?\n", (unsigned short) ntohs(Msg.opCode));
 | 
						|
    }
 | 
						|
 | 
						|
 break_loop:
 | 
						|
 | 
						|
    udpc_getMyAddress(fNet_config.net_if, &myIp);
 | 
						|
 | 
						|
    if(!udpc_ipIsZero(&fNet_config.dataMcastAddr)  &&
 | 
						|
       !udpc_ipIsEqual(&fNet_config.dataMcastAddr, &myIp) &&
 | 
						|
       (udpc_ipIsZero(&fNet_config.controlMcastAddr) ||
 | 
						|
       !udpc_ipIsEqual(&fNet_config.dataMcastAddr, &fNet_config.controlMcastAddr)
 | 
						|
	)) {
 | 
						|
	fClient_config.S_MCAST_DATA = 
 | 
						|
	  udpc_makeSocket(ADDR_TYPE_MCAST, fNet_config.net_if, 
 | 
						|
		     &fNet_config.dataMcastAddr, 
 | 
						|
		     RECEIVER_PORT(fNet_config.portBase));
 | 
						|
//cerr << "S_MCAST_DATA = " << fClient_config.S_MCAST_DATA << endl;
 | 
						|
    }
 | 
						|
 | 
						|
 | 
						|
    if(fNet_config.requestedBufSize) {
 | 
						|
      int i;
 | 
						|
      for(i=0; i<NR_CLIENT_SOCKS; i++)
 | 
						|
	if(fClient_config.socks[i] != -1)
 | 
						|
	  udpc_setRcvBuf(fClient_config.socks[i],fNet_config.requestedBufSize);
 | 
						|
    }
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
void MulticastImpl::receive(SBS outbs)
 | 
						|
{
 | 
						|
	struct fifo fifo;
 | 
						|
 | 
						|
	udpc_initFifo(&fifo, fNet_config.blockSize);
 | 
						|
	fifo.data = pc_makeProduconsum(fifo.dataBufSize, "receive");
 | 
						|
	fClient_config.isStarted = 0;
 | 
						|
 | 
						|
	spawnNetReceiver(&fifo, &fClient_config, &fNet_config);
 | 
						|
	writer(&fifo, outbs);
 | 
						|
 | 
						|
	pthread_join(fClient_config.thread, NULL);
 | 
						|
}
 | 
						|
 | 
						|
}
 | 
						|
 |