udp传送文件

务器端：

[Copy to clipboard]  [ - ]

CODE:

#include<sys/socket.h>;
#include<string.h>;
#include<netinet/in.h>;
#include<stdio.h>;
#include<stdlib.h>;
#include<fcntl.h>;
#include<sys/stat.h>;
#include<unistd.h>;
#include<errno.h>;
#include<sys/select.h>;
#include<sys/time.h>;
#include<unistd.h>;
#include<sys/types.h>;

#define SERV_PORT 2500
#define MAX_SIZE 1024*40

void recvUDP(char name[20],int sockfd)
{
        int ret,fd;
        mode_t fdmode = (S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH);
        char mesg[MAX_SIZE];
        fd_set rdset;
        struct timeval tv;
        int rlen,wlen;
        int i;

        fd = open(name,O_RDWR|O_CREAT|O_APPEND,fdmode);
        if(fd == -1)
        {
                printf("open file %s error:%n",name,strerror(errno));
                exit(-1);
        }

        for(i=0;;i++)
        {
                tv.tv_sec = 1;
                tv.tv_usec = 0;

                FD_ZERO(&rdset);
                FD_SET(sockfd,&rdset);
                
                ret = select(sockfd+1,&rdset,NULL,NULL,&tv);
                if(ret == -1)
                {
                        printf("select error %s\n",strerror(errno));
                        exit(-1);
                }
                else if(ret==0)
                {
                        printf("select timeout,continue circle\n");
                        continue;
                }
                
                if(FD_ISSET(sockfd,&rdset))
                {
                        memset(mesg,0,MAX_SIZE);
                        rlen = read(sockfd,mesg,MAX_SIZE);
                        if(rlen <=0 )
                        {
                                printf("read error %s\n",strerror(errno));
                                exit(-1);
                        }
                        
                        if(!strcmp(mesg,"end"))
                        {
                                printf("recv end.\n");
                                break;
                        }

                        wlen = write(fd,mesg,rlen);
                        if(wlen != rlen )
                        {
                                printf("write error %s\n",strerror(errno));
                                exit(-1);
                        }        
                }
                
                printf("The %d times write\n",i);
        }

        close(fd);
}

int main(int argc, char *argv[])
{
        int sockfd;
        int r;
        struct sockaddr_in servaddr;

        sockfd = socket(AF_INET,SOCK_DGRAM,0); /*create a socket*/

        /*init servaddr*/
        bzero(&servaddr,sizeof(servaddr));
        servaddr.sin_family = AF_INET;
        servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
        servaddr.sin_port = htons(SERV_PORT);

        /*bind address and port to socket*/
        if(bind(sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr)) == -1)
        {
                perror("bind error");
                exit(-1);
        }

        r = fcntl(sockfd, F_GETFL, 0);
        fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);
        
        recvUDP(argv[1],sockfd);

        return 0;

}

客户端：

[Copy to clipboard]  [ - ]

CODE:

#include<sys/types.h>;
#include<sys/socket.h>;
#include<string.h>;
#include<netinet/in.h>;
#include<stdio.h>;
#include<stdlib.h>;
#include<arpa/inet.h>;
#include<fcntl.h>;
#include<sys/stat.h>;
#include<errno.h>;
#include<sys/sysinfo.h>;
#include<sys/select.h>;
#include<sys/time.h>;
#include<unistd.h>;

#define MAX_SIZE 1024*40
#define SERV_PORT 2500


void connectUDP(char name[20],int sockfd,struct sockaddr *pservaddr,socklen_t servlen)
{
                char buf[MAX_SIZE];
                fd_set wrset;
                struct timeval tv;
                int rlen,wlen;
                int fd;
                int ret;
                int i;

                if(connect(sockfd,(struct sockaddr *)pservaddr,servlen) == -1)
                {
                        perror("connet error");
                        exit(1);
                }
                else
                        printf("connect server ok!\n");

                fd = open(name,O_RDONLY);
                if(fd==-1)
                {
                        printf("fopen error %s\n",strerror(errno));
                        exit(-1);
                }
                
                i=0;
                
                while(1)
                {        
                        tv.tv_sec = 1;
                        tv.tv_usec = 0;
                        
                        FD_ZERO(&wrset);
                        FD_SET(sockfd,&wrset);
                        
                        ret = select(sockfd+1,NULL,&wrset,NULL,&tv);
                        if(ret == -1)
                        {
                                printf("select error %s\n",strerror(errno));
                                exit(-1);
                        }
                        else if(ret==0)
                        {
                                printf("select timeout,continue circle\n");
                                continue;
                        }
                        
                        memset(buf,0,MAX_SIZE);
                        if(FD_ISSET(sockfd,&wrset))
                        {                                        
                                rlen = read(fd,buf,MAX_SIZE);
                                if(rlen <0)
                                {
                                        printf("fread data error %s\n",strerror(errno));
                                        exit(-1);
                                }
                                else if(rlen==0)
                                {
                                        wlen = write(sockfd,"end",3);
                                        if(wlen !=3)
                                        {
                                                printf("write end error\n",strerror(errno));
                                                exit(-1);
                                        }

                                        printf("all complete\n");

                                        close(fd);
                                        close(sockfd);
                                        exit(0);
                                }

                                wlen = write(sockfd,buf,rlen);
                                if(wlen != rlen)
                                {
                                        printf("write data to sockfd error:%s\n",strerror(errno));
                                        exit(-1);
                                }
                                
                                i++;
                                
                                usleep(500);
                                printf("The %d times read\n",i);
                        }
                }
        }


int main(int argc ,char *argv[])
{        
        char *fh;
        struct sysinfo s_info;
        float time1,time2;
        int error1,error2;
        int sockfd;
        struct stat fsize;
        struct sockaddr_in servaddr;
        error1= sysinfo(&s_info);
        time1 = s_info.uptime;
        int r;

        if(argc != 3)
        {
                printf("useage:udpclient<IPaddress>;\n");
                exit(1);
        }
        bzero(&servaddr,sizeof(servaddr));
        servaddr.sin_family= AF_INET;
        servaddr.sin_port = htons(SERV_PORT);
        
        if(inet_pton(AF_INET,argv[1],&servaddr.sin_addr) <= 0)
        {
                printf("[%s]is not a valid IPaddress\n",argv[1]);
                exit(1);
        }
        fh = argv[2];

        sockfd =socket(AF_INET,SOCK_DGRAM,0);
        
        r = fcntl(sockfd, F_GETFL, 0);
        fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);
        
        connectUDP(argv[2],sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr));        
        fprintf(stderr,"ServerIP:\t%s\n",argv[1]);        
        if(stat(argv[2],&fsize) == -1)
                perror("failed to get fiel statusi\n");
        else        
                fprintf(stderr,"file name:\t%s\nfile size:\t%dK\n",argv[2],fsize.st_size/1024);
        error2=sysinfo(&s_info);
        time2 = s_info.uptime;
        printf("tranfice file time =%fs\n",(time2-time1));
}               

Makefile

[Copy to clipboard]  [ - ]

CODE:

all:send recv

send:send.c
        gcc -Wall send.c -o send
recv:recv.c
        gcc -Wall recv.c -o recv
        
clean:
        rm -rf send recv

http://blog.chinaunix.net/uid-7390305-id-2057060.html