udpserver.c

UDP server/client 实现

#include <stdlib.h> 
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <fcntl.h>
#include <ctype.h>
#include <unistd.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

int main()
{
	int fdSock;
	char chBuf[1024];
	struct sockaddr_in myAddr,peerAddr;
	int iRetSock,iRetFile;
	FILE *pFile;
	size_t nSize=sizeof(peerAddr);
	int i=0;
	//
	fdSock=socket(AF_INET,SOCK_DGRAM,0);
	myAddr.sin_family=AF_INET;
	myAddr.sin_port=htons(5555);
	myAddr.sin_addr.s_addr=htonl(INADDR_ANY);
	if(bind(fdSock,(struct sockaddr *)&myAddr,sizeof(myAddr))!=0){
		perror("bind error:");
		close(fdSock);
		return -1;
	}
	//接收文件名	
	iRetSock=recvfrom(fdSock,chBuf,sizeof(chBuf)-1,0,(struct sockaddr *)&peerAddr,&nSize);
	if(iRetSock<1){
		perror("recv file name fail.");
		close(fdSock);
		return -2;
	}		   			  
	chBuf[iRetSock]=0;			   			  
	pFile=fopen(chBuf,"rb");
	if(pFile==NULL){
		printf("open file %s fail.\n",chBuf);
		close(fdSock);
		return -3;
	}
	while(1){
		iRetFile=fread(chBuf,1,sizeof(chBuf),pFile);
		iRetSock=sendto(fdSock,chBuf,iRetFile,0,(struct sockaddr *)&peerAddr,nSize);
		if(iRetSock!=iRetFile){
			perror("send data fail!");
			break;	
		}
		if(iRetFile<1024){	//表示读到文件末尾,读文件失败的情况当作读到文件末尾的情况处理
			break;
		}
		printf("send :%d,iRet:%d\n",i++,iRetSock);			
		sleep(1);
	}
	fclose(pFile);
	close(fdSock);
	return 0;
}