APPS/unr_client/rel_client.c

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/sctp.h>
#include <arpa/inet.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <getopt.h>

#define SIZE_OF_MESSAGE (512 * 1024)
#define SIZE_OF_I_FRAME (51 * 1024)
#define SIZE_OF_P_FRAME (9 * 1024)
#define SIZE_OF_B_FRAME (3 * 1024)

#define DISCARD_PORT 30002
int fd;

int iframe_rel=0;
int pframe_rel=0;
int bframe_rel=0;

struct sockaddr_in addr;
char buffer[SIZE_OF_MESSAGE];
int icount=0, bcount=0, pcount=0;
int
send_iframe()
{
	int ret;
	buffer[0] = 'I';
	if ((ret = sctp_sendmsg(fd,
			 (const void *)buffer, SIZE_OF_I_FRAME,
			 (struct sockaddr *)&addr, sizeof(struct sockaddr_in),
			 htonl(73),       /* PPID */
			 0, /* flags */
			 1,                /* stream identifier */
			 0,                /* Max number of rtx */
			 0                 /* context */
		     )) < 0) {
		perror("sctp_sendmsg - I");
	} else {
		icount++;
	}
	printf("I:%d\r", icount);
	fflush(stdout);
	return(ret);
}

int
send_pframe()
{
	int ret;
	buffer[0] = 'P';
	if ((ret = sctp_sendmsg(fd,
			 (const void *)buffer, SIZE_OF_P_FRAME,
			 (struct sockaddr *)&addr, sizeof(struct sockaddr_in),
			 htonl(80),       /* PPID */
			 0, /* flags */
			 1,                /* stream identifier */
			 0,                /* Max number of rtx */
			 0                 /* context */
		     )) < 0) {
		perror("sctp_sendmsg - P");
	} else {
		pcount++;
	}
	printf("I:%d P:%d\r", icount, pcount);
	fflush(stdout);
	return(ret);
}

int
send_bframe()
{
	int ret;
	buffer[0] = 'B';
	if ((ret = sctp_sendmsg(fd,
			 (const void *)buffer, SIZE_OF_P_FRAME,
			 (struct sockaddr *)&addr, sizeof(struct sockaddr_in),
			 htonl(66),       /* PPID */
			 0, /* flags */
			 1,                /* stream identifier */
			 0,                /* Max number of rtx */
			 0                 /* context */
		     )) < 0) {
		perror("sctp_sendmsg - B");
	} else {
		bcount++;
	}
	printf("I:%d P:%d B:%d\r", icount, pcount, bcount);
	fflush(stdout);
	return(ret);
}



int main(int argc, char **argv) 
{
	unsigned int i, j, k;
	int num_messages=24;
	struct sockaddr_in bindaddr[10];
	char *baddr[10];
	int bndcnt=0;
	char *toaddr = NULL;
	int port = htons(DISCARD_PORT);

	while((i= getopt(argc,argv,"P:I:B:b:h:p:m:?")) != EOF)
	{
		switch(i) {
		case 'p':
			port = htons(strtol(optarg, NULL, 0));
			break;
		case 'P':
			pframe_rel = strtol(optarg, NULL, 0);
			break;
		case 'I':
			iframe_rel = strtol(optarg, NULL, 0);
			break;
		case 'B':
			bframe_rel = strtol(optarg, NULL, 0);
			break;


		case 'b':
			if (bndcnt < 10) {
				baddr[bndcnt] = optarg;
				bndcnt++;
			}
			break;
		case 'h':
			toaddr = optarg;
			break;
		case 'm':
			num_messages = strtol(optarg, NULL, 0);
			break;
		case '?':
		default:
			goto use;
			break;
		};
	}
	if (toaddr == NULL) {
	use:
		printf("use %s -h to-addr-in-dot-notation [-m message-cnt -b bindaddr]\n",
		       argv[0]);
		printf("                    [-P rel -I rel -B rel]\n");
		exit(0);
	}
	memset((void *)buffer, 'A', SIZE_OF_MESSAGE);
	if ((fd = socket(AF_INET, SOCK_SEQPACKET, IPPROTO_SCTP)) < 0)
		perror("socket");
	

	i = 1;
    	if(setsockopt(fd, IPPROTO_SCTP, SCTP_NODELAY, &i, sizeof(int))<0) {
        	perror("SCTP_NODELAY");
	}

	if (bndcnt) {
		/* prepare */
		memset(bindaddr, 0, sizeof(bindaddr));
		for(i=0; i<bndcnt; i++) {
			bindaddr[i].sin_len    = sizeof(struct sockaddr_in);
			bindaddr[i].sin_family = AF_INET;
			bindaddr[i].sin_port   = 0;
			inet_pton(AF_INET, baddr[i], &bindaddr[i].sin_addr);
		}
		/* bindx */
		if (sctp_bindx(fd, (struct sockaddr *)bindaddr,
			       bndcnt, SCTP_BINDX_ADD_ADDR) != 0) {
			perror("bind failed");
			exit(0);
		}
	}
	memset((void *)&addr, 0, sizeof(struct sockaddr_in));
	addr.sin_len    = sizeof(struct sockaddr_in);
	addr.sin_family = AF_INET;
	addr.sin_port   = port;
	inet_pton(AF_INET, toaddr, &addr.sin_addr);
	for (i = 0; i <  num_messages; i++) {
		if (send_iframe() < 0) {
			exit(-4);
		}
		if (send_pframe() < 0) {
			exit(-3);
		}
		for (j=0; j<5; j++) {
			if (send_pframe() < 0) {
				exit(-2);
			}
			for (k=0; k<6; k++) {
				if (send_bframe() < 0) {
					exit(-1);
				}
			}
		}
	}
	if (close(fd) < 0)
		perror("close");
	printf("\n");
	return(0);
}