Linux网络编程之使用TCP传输文件

原创作品，转载时请务必以超链接形式标明文章原始出处:http://blog.csdn.net/gqb666/article/details/8976263,作者:gqb666

最近在写Linux网络方面的demo,用TCP实现了一个简单的传输文件程序，适用于网卡设备及TCP/IP协议栈及网络环境测试时使用。用法容易，代码简洁，已经测试成功。其功能是使用TCP协议从client端向server端传输文件，用法如下:

     编译：

           client:gcc -o client client.c

           server:gcc -o server server.c

     运行：

           client端：./client <server IP> <端口号> <上传文件名>

           server端：./server <端口号> <保存为文件名>

其中，server端先运行，client端与server端的端口号必须一致并且不能与已知端口冲突（例如8888即可）

下面将代码贴上：（使用UDP传输文件请参阅博文Linux网络编程之使用UDP传输文件）

server端代码:server.c

[cpp] view plaincopy

1. #include <stdio.h>
2. #include <string.h>
3. #include <stdlib.h>
4. #include <string.h>
5. #include <unistd.h>
6. #include <sys/types.h>
7. #include <sys/socket.h>
8. #include <netinet/in.h>
9. #include <arpa/inet.h>
11. #define    MAXLINE        1024
13. void usage(char *command)  
14. {  
15.     printf("usage :%s portnum filename\n", command);  
16.     exit(0);  
17. }  
18. int main(int argc,char **argv)   
19. {  
20. struct sockaddr_in     serv_addr;  
21. struct sockaddr_in     clie_addr;  
22. char                   buf[MAXLINE];  
23. int                    sock_id;  
24. int                    link_id;  
25. int                    recv_len;  
26. int                    write_leng;  
27. int                    clie_addr_len;  
28. FILE                   *fp;  
30. if (argc != 3) {  
31.         usage(argv[0]);  
32.     }  
33. if ((fp = fopen(argv[2], "w")) == NULL) {  
34.         perror("Open file failed\n");  
35.         exit(0);  
36.     }  
37. if ((sock_id = socket(AF_INET, SOCK_STREAM, 0)) < 0) {  
38.         perror("Create socket failed\n");  
39.         exit(0);  
40.     }  
41. /*fill the server sockaddr_in struct commented by guoqingbo*/
42.     memset(&serv_addr, 0, sizeof(serv_addr));  
43.     serv_addr.sin_family = AF_INET;  
44.     serv_addr.sin_port = htons(atoi(argv[1]));  
45.     serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
47. if (bind(sock_id, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0 ) {  
48.         perror("Bind socket failed\n");  
49.         exit(0);  
50.     }  
52. if (-1 == listen(sock_id, 10)) {  
53.         perror("Listen socket failed\n");  
54.         exit(0);  
55.     }  
56. /* server part commented by guoqingbo*/
57. while (1) {  
58.         clie_addr_len = sizeof(clie_addr);  
59.         link_id = accept(sock_id, (struct sockaddr *)&clie_addr, &clie_addr_len);  
60. if (-1 == link_id) {  
61.             perror("Accept socket failed\n");  
62.             exit(0);  
63.         }  
64.         bzero(buf, MAXLINE);  
65. while (recv_len = recv(link_id, buf, MAXLINE, 0)) {  
66. /* receiver data part commented by guoqingbo*/
67. if(recv_len < 0) {  
68.                 printf("Recieve Data From Server Failed!\n");  
69. break;  
70.             }  
71.             printf("#");  
72.             write_leng = fwrite(buf, sizeof(char), recv_len, fp);  
73. if (write_leng < recv_len) {  
74.                 printf("Write file failed\n");  
75. break;  
76.             }  
77.             bzero(buf,MAXLINE);  
78.         }  
79.         printf("\nFinish Recieve\n");  
80.         fclose(fp);  
81.         close(link_id);  
82.     }  
83.     close(sock_id);   
84. return 0;  
85. }  

client端代码:client.c

[html] view plaincopy

1. #include <stdio.h>
2. #include <string.h>
3. #include <stdlib.h>
4. #include <string.h>
5. #include <unistd.h>
6. #include <fcntl.h>
7. #include <sys/types.h>
8. #include <sys/socket.h>
9. #include <netinet/in.h>
10. #include <arpa/inet.h>
12. #define    MAXLINE        1024  
14. void usage(char *command)  
15. {  
16.     printf("usage :%s ipaddr portnum filename\n", command);  
17.     exit(0);  
18. }  
19. int main(int argc,char **argv)  
20. {  
21.     struct sockaddr_in     serv_addr;  
22.     char                   buf[MAXLINE];  
23.     int                    sock_id;  
24.     int                    read_len;  
25.     int                    send_len;  
26.     FILE                   *fp;  
27.     int                    i_ret;  
29.     if (argc != 4) {  
30.         usage(argv[0]);  
31.     }  
33.     /* open the file to be transported commented by guoqingbo*/  
34.     if ((fp = fopen(argv[3],"r")) == NULL) {  
35.         perror("Open file failed\n");  
36.         exit(0);  
37.     }  
39.     /* create the socket commented by guoqingbo*/  
40.     if ((sock_id = socket(AF_INET,SOCK_STREAM,0)) < 0) {  
41.         perror("Create socket failed\n");  
42.         exit(0);  
43.     }  
45.     memset(&serv_addr, 0, sizeof(serv_addr));  
46. serv_addr.sin_family = AF_INET;  
47. serv_addr.sin_port = htons(atoi(argv[2]));  
48.     inet_pton(AF_INET, argv[1], &serv_addr.sin_addr);  
50.     /* connect the server commented by guoqingbo*/  
51. i_ret = connect(sock_id, (struct sockaddr *)&serv_addr, sizeof(struct sockaddr));  
52.     if (-1 == i_ret) {  
53.         printf("Connect socket failed\n");  
54.        return -1;  
55.     }  
57.     /* transported the file commented by guoqingbo*/  
58.     bzero(buf, MAXLINE);  
59.     while ((read_len = fread(buf, sizeof(char), MAXLINE, fp)) >0 ) {  
60. send_len = send(sock_id, buf, read_len, 0);  
61.         if ( send_len < 0 ) {  
62.             perror("Send file failed\n");  
63.             exit(0);  
64.         }  
65.         bzero(buf, MAXLINE);  
66.     }  
68.     fclose(fp);  
69.     close(sock_id);  
70.     printf("Send Finish\n");  
71.     return 0;  
72. }  

下转博文：Linux网络编程之使用UDP传输文件