原创：如何做linux下udp大文件传输

Question

近日小弟做了个linux下用户数据报协议大文件传输程式发上来与大家共勉。
代码在 redhat 9.0 下编译通过。
最大测试无差错传输文件：288M
最大测试传输速度：6.5M/S

可能这里并不需要这种类型的帖子，但希望各位能代小弟转贴，因为网络上很难搜索到这种可以解决丢包问题的代码级实现方法，希望每一个需要的人都能看的到。

服务器端：

2. #include<sys/socket.h>;
3. #include<string.h>;
4. #include<netinet/in.h>;
5. #include<stdio.h>;
6. #include<stdlib.h>;
7. #include<fcntl.h>;
8. #include<sys/stat.h>;
9. #include<unistd.h>;
10. #include<errno.h>;
11. #include<sys/select.h>;
12. #include<sys/time.h>;
13. #include<unistd.h>;
14. #include<sys/types.h>;
16. #define SERV_PORT 2500
17. #define MAX_SIZE 1024*40
19. void recvUDP(char name[20],int sockfd)
20. {
21.         int ret,fd;
22.         mode_t fdmode = (S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH);
23.         char mesg[MAX_SIZE];
24.         fd_set rdset;
25.         struct timeval tv;
26.         int rlen,wlen;
27.         int i;
29.         fd = open(name,O_RDWR|O_CREAT|O_APPEND,fdmode);
30.         if(fd == -1)
31.         {
32.                 printf("open file %s error:%n",name,strerror(errno));
33.                 exit(-1);
34.         }
36.         for(i=0;;i++)
37.         {
38.                 tv.tv_sec = 1;
39.                 tv.tv_usec = 0;
41.                 FD_ZERO(&rdset);
42.                 FD_SET(sockfd,&rdset);
43.                
44.                 ret = select(sockfd+1,&rdset,NULL,NULL,&tv);
45.                 if(ret == -1)
46.                 {
47.                         printf("select error %s\n",strerror(errno));
48.                         exit(-1);
49.                 }
50.                 else if(ret==0)
51.                 {
52.                         printf("select timeout,continue circle\n");
53.                         continue;
54.                 }
55.                
56.                 if(FD_ISSET(sockfd,&rdset))
57.                 {
58.                         memset(mesg,0,MAX_SIZE);
59.                         rlen = read(sockfd,mesg,MAX_SIZE);
60.                         if(rlen <=0 )
61.                         {
62.                                 printf("read error %s\n",strerror(errno));
63.                                 exit(-1);
64.                         }
65.                        
66.                         if(!strcmp(mesg,"end"))
67.                         {
68.                                 printf("recv end.\n");
69.                                 break;
70.                         }
72.                         wlen = write(fd,mesg,rlen);
73.                         if(wlen != rlen )
74.                         {
75.                                 printf("write error %s\n",strerror(errno));
76.                                 exit(-1);
77.                         }       
78.                 }
79.                
80.                 printf("The %d times write\n",i);
81.         }
83.         close(fd);
84. }
86. int main(int argc, char *argv[])
87. {
88.         int sockfd;
89.         int r;
90.         struct sockaddr_in servaddr;
92.         sockfd = socket(AF_INET,SOCK_DGRAM,0); /*create a socket*/
94.         /*init servaddr*/
95.         bzero(&servaddr,sizeof(servaddr));
96.         servaddr.sin_family = AF_INET;
97.         servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
98.         servaddr.sin_port = htons(SERV_PORT);
100.         /*bind address and port to socket*/
101.         if(bind(sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr)) == -1)
102.         {
103.                 perror("bind error");
104.                 exit(-1);
105.         }
107.         r = fcntl(sockfd, F_GETFL, 0);
108.         fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);
109.        
110.         recvUDP(argv[1],sockfd);
112.         return 0;
114. }

复制代码

客户端：

2. #include<sys/types.h>;
3. #include<sys/socket.h>;
4. #include<string.h>;
5. #include<netinet/in.h>;
6. #include<stdio.h>;
7. #include<stdlib.h>;
8. #include<arpa/inet.h>;
9. #include<fcntl.h>;
10. #include<sys/stat.h>;
11. #include<errno.h>;
12. #include<sys/sysinfo.h>;
13. #include<sys/select.h>;
14. #include<sys/time.h>;
15. #include<unistd.h>;
17. #define MAX_SIZE 1024*40
18. #define SERV_PORT 2500
21. void connectUDP(char name[20],int sockfd,struct sockaddr *pservaddr,socklen_t servlen)
22. {
23.                 char buf[MAX_SIZE];
24.                 fd_set wrset;
25.                 struct timeval tv;
26.                 int rlen,wlen;
27.                 int fd;
28.                 int ret;
29.                 int i;
31.                 if(connect(sockfd,(struct sockaddr *)pservaddr,servlen) == -1)
32.                 {
33.                         perror("connet error");
34.                         exit(1);
35.                 }
36.                 else
37.                         printf("connect server ok!\n");
39.                 fd = open(name,O_RDONLY);
40.                 if(fd==-1)
41.                 {
42.                         printf("fopen error %s\n",strerror(errno));
43.                         exit(-1);
44.                 }
45.                
46.                 i=0;
47.                
48.                 while(1)
49.                 {       
50.                         tv.tv_sec = 1;
51.                         tv.tv_usec = 0;
52.                        
53.                         FD_ZERO(&wrset);
54.                         FD_SET(sockfd,&wrset);
55.                        
56.                         ret = select(sockfd+1,NULL,&wrset,NULL,&tv);
57.                         if(ret == -1)
58.                         {
59.                                 printf("select error %s\n",strerror(errno));
60.                                 exit(-1);
61.                         }
62.                         else if(ret==0)
63.                         {
64.                                 printf("select timeout,continue circle\n");
65.                                 continue;
66.                         }
67.                        
68.                         memset(buf,0,MAX_SIZE);
69.                         if(FD_ISSET(sockfd,&wrset))
70.                         {                                       
71.                                 rlen = read(fd,buf,MAX_SIZE);
72.                                 if(rlen <0)
73.                                 {
74.                                         printf("fread data error %s\n",strerror(errno));
75.                                         exit(-1);
76.                                 }
77.                                 else if(rlen==0)
78.                                 {
79.                                         wlen = write(sockfd,"end",3);
80.                                         if(wlen !=3)
81.                                         {
82.                                                 printf("write end error\n",strerror(errno));
83.                                                 exit(-1);
84.                                         }
86.                                         printf("all complete\n");
88.                                         close(fd);
89.                                         close(sockfd);
90.                                         exit(0);
91.                                 }
93.                                 wlen = write(sockfd,buf,rlen);
94.                                 if(wlen != rlen)
95.                                 {
96.                                         printf("write data to sockfd error:%s\n",strerror(errno));
97.                                         exit(-1);
98.                                 }
99.                                
100.                                 i++;
101.                                
102.                                 usleep(500);
103.                                 printf("The %d times read\n",i);
104.                         }
105.                 }
106.         }
109. int main(int argc ,char *argv[])
110. {       
111.         char *fh;
112.         struct sysinfo s_info;
113.         float time1,time2;
114.         int error1,error2;
115.         int sockfd;
116.         struct stat fsize;
117.         struct sockaddr_in servaddr;
118.         error1= sysinfo(&s_info);
119.         time1 = s_info.uptime;
120.         int r;
122.         if(argc != 3)
123.         {
124.                 printf("useage:udpclient<IPaddress>;\n");
125.                 exit(1);
126.         }
127.         bzero(&servaddr,sizeof(servaddr));
128.         servaddr.sin_family= AF_INET;
129.         servaddr.sin_port = htons(SERV_PORT);
130.        
131.         if(inet_pton(AF_INET,argv[1],&servaddr.sin_addr) <= 0)
132.         {
133.                 printf("[%s]is not a valid IPaddress\n",argv[1]);
134.                 exit(1);
135.         }
136.         fh = argv[2];
138.         sockfd =socket(AF_INET,SOCK_DGRAM,0);
139.        
140.         r = fcntl(sockfd, F_GETFL, 0);
141.         fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);
142.        
143.         connectUDP(argv[2],sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr));       
144.         fprintf(stderr,"ServerIP:\t%s\n",argv[1]);       
145.         if(stat(argv[2],&fsize) == -1)
146.                 perror("failed to get fiel statusi\n");
147.         else       
148.                 fprintf(stderr,"file name:\t%s\nfile size:\t%dK\n",argv[2],fsize.st_size/1024);
149.         error2=sysinfo(&s_info);
150.         time2 = s_info.uptime;
151.         printf("tranfice file time =%fs\n",(time2-time1));
152. }               

复制代码

Makefile

2. all:send recv
4. send:send.c
5.         gcc -Wall send.c -o send
6. recv:recv.c
7.         gcc -Wall recv.c -o recv
8.        
9. clean:
10.         rm -rf send recv

复制代码

[ 本帖最后由 platinum 于 2006-5-12 12:02 编辑 ]

Answer 1

希望大家可以转到任何可以转的linux论坛。谢谢！

Answer 2

大哥不会是让我们都当做范例学习吧.

Answer 3

赞一下LZ

Answer 4

请楼主以后发代码时，用 [ code ] 括起来，或者禁止显示表情，否则贴出的代码没法看

Answer 5

memset是多此一举，并且大大耗费时间。

Answer 6

很好啊，值得学习！

谢谢楼主。

Answer 7

为什么说这是 "可以解决丢包问题的代码级实现方法"??

哪里是亮点,lz讲明白点不好吗??

否则我读代码的时候会遗漏过去体会不到.

还有如果网络状况很差 或者 发送比接收的机器速度快 等情况下是否会能保证不丢报呢?

Answer 8

原帖由 思一克 于 2006-5-12 13:11 发表
memset是多此一举，并且大大耗费时间。

agree.

Answer 9

无知者无畏亚