mrecv.c

自己编写的一个组播发送

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/ioctl.h>
#include <netinet/in.h>
#include <arpa/inet.h>

int main(int argc,char **argv)	{
	int recv_sockfd;
	int recv_buf_size=220*1024;
	struct sockaddr_in mcast_addr,from_addr;
	struct ip_mreq recv_mcast_req;
	int mcast_addr_len,from_addr_len;
	FILE *fp;
	char *recvFile="/home/jackie/mcast/RecvMtv.mpg";
	int bufSize=3.5*1024;
	char buf[bufSize];
	int Msize=1024*1024; //1 Mbytes
	double sum=0.0;
	int counter=0;
	int num_bytes;
	int i;

	recv_sockfd=socket(AF_INET,SOCK_DGRAM,0);
	bzero(&mcast_addr,sizeof(struct sockaddr_in));
	mcast_addr.sin_family=AF_INET;
//	mcast_addr.sin_addr.s_addr=inet_addr("234.1.1.13");
//	inet_pton(AF_INET, "192.168.1.13",&mcast_addr.sin_addr);
	mcast_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	mcast_addr.sin_port=htons(1234);
	mcast_addr_len=sizeof(mcast_addr);

	//bzero(&from_addr,sizeof(struct sockaddr_in));
	//from_addr.sin_family=AF_INET;
	//inet_pton(AF_INET,"234.1.1.13",&from_addr.sin_addr);
	//from_addr.sin_port=htons(1234);
	from_addr_len=sizeof(from_addr);

	bind(recv_sockfd,(struct sockaddr *)&mcast_addr,sizeof(struct sockaddr_in));

	int reuseaddr=1;
	setsockopt(recv_sockfd, SOL_SOCKET, SO_REUSEADDR, &reuseaddr, sizeof(int));
	setsockopt(recv_sockfd,SOL_SOCKET,SO_RCVBUF,&recv_buf_size,sizeof(int));
	
	inet_pton(AF_INET,"234.1.1.13",&(recv_mcast_req.imr_multiaddr));
	recv_mcast_req.imr_interface.s_addr =htonl(INADDR_ANY);
	setsockopt(recv_sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const char *)&recv_mcast_req, sizeof(struct ip_mreq));
	setsockopt(recv_sockfd, IPPROTO_IP, IP_MULTICAST_TTL, (u_char *)0, sizeof(u_char));
	setsockopt(recv_sockfd, IPPROTO_IP, IP_MULTICAST_LOOP, (u_char*)0, sizeof(u_char));
	

	if((fp=fopen(recvFile,"wb"))==NULL)	{
			printf("\ncannot open file\n");
			exit(0);
	}
	printf("\nopen file\n");
	int writeSize;
	while(1)	{
		//printf("recv data now\n");	
		for(i=0;i<bufSize;i++)
			buf[i]=0;
		//printf("test\n");
		if ((num_bytes=recvfrom(recv_sockfd, buf, bufSize, 0, (struct sockaddr *)&from_addr,&from_addr_len)) == -1) {
			printf("receive data - error\n");
			exit(0);
		}
		//printf("receive %d data\n", num_bytes);
		
		if(strcmp(buf,"the file send completely")==0){
			fclose(fp);
			printf("\nfile write completely\n");
			break;
		}	
		writeSize=fwrite(buf,sizeof(char),bufSize,fp);
		//printf("write %d data\n", writeSize);
		
	
		
		if( writeSize<0){
			printf("\nfile write error\n");
			exit(0);
		}
		sum += num_bytes;
		if(sum/Msize >= counter) {
			counter++;
			printf("++receive %.2fMbytes data stream\n", sum/Msize);
		}
		usleep(1);	
	}
		
}