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

2019-07-13 09:24发布

原创作品,转载时请务必以超链接形式标明文章原始出处: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端:./server <端口号> <保存为文件名> 其中,server端先运行,client端与server端的端口号必须一致并且不能与已知端口冲突(例如8888即可) 下面将代码贴上:(使用UDP传输文件请参阅博文Linux网络编程之使用UDP传输文件 server端代码:server.c [cpp] view plain copy print?
  1. #include   
  2. #include   
  3. #include   
  4. #include   
  5. #include   
  6. #include   
  7. #include   
  8. #include   
  9. #include   
  10.   
  11. #define    MAXLINE        1024  
  12.   
  13. void usage(char *command)  
  14. {  
  15.     printf(”usage :%s portnum filename ”, 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;  
  29.   
  30.     if (argc != 3) {  
  31.         usage(argv[0]);  
  32.     }  
  33.     if ((fp = fopen(argv[2], “w”)) == NULL) {  
  34.         perror(”Open file failed ”);  
  35.         exit(0);  
  36.     }  
  37.     if ((sock_id = socket(AF_INET, SOCK_STREAM, 0)) < 0) {  
  38.         perror(”Create socket failed ”);  
  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);  
  46.   
  47.     if (bind(sock_id, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0 ) {  
  48.         perror(”Bind socket failed ”);  
  49.         exit(0);  
  50.     }  
  51.   
  52.     if (-1 == listen(sock_id, 10)) {  
  53.         perror(”Listen socket failed ”);  
  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 ”);  
  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! ”);  
  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 ”);  
  75.                 break;  
  76.             }  
  77.             bzero(buf,MAXLINE);  
  78.         }  
  79.         printf(” Finish Recieve ”);  
  80.         fclose(fp);  
  81.         close(link_id);  
  82.     }  
  83.     close(sock_id);   
  84.     return 0;  
  85. }  
#include #include #include #include #include #include #include #include #include #define MAXLINE 1024 void usage(char *command) { printf("usage :%s portnum filename ", command); exit(0); } int main(int argc,char **argv) { struct sockaddr_in serv_addr; struct sockaddr_in clie_addr; char buf[MAXLINE]; int sock_id; int link_id; int recv_len; int write_leng; int clie_addr_len; FILE *fp; if (argc != 3) { usage(argv[0]); } if ((fp = fopen(argv[2], "w")) == NULL) { perror("Open file failed "); exit(0); } if ((sock_id = socket(AF_INET, SOCK_STREAM, 0)) < 0) { perror("Create socket failed "); exit(0); } /*fill the server sockaddr_in struct commented by guoqingbo*/ memset(&serv_addr, 0, sizeof(serv_addr)); serv_addr.sin_family = AF_INET; serv_addr.sin_port = htons(atoi(argv[1])); serv_addr.sin_addr.s_addr = htonl(INADDR_ANY); if (bind(sock_id, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0 ) { perror("Bind socket failed "); exit(0); } if (-1 == listen(sock_id, 10)) { perror("Listen socket failed "); exit(0); } /* server part commented by guoqingbo*/ while (1) { clie_addr_len = sizeof(clie_addr); link_id = accept(sock_id, (struct sockaddr *)&clie_addr, &clie_addr_len); if (-1 == link_id) { perror("Accept socket failed "); exit(0); } bzero(buf, MAXLINE); while (recv_len = recv(link_id, buf, MAXLINE, 0)) { /* receiver data part commented by guoqingbo*/ if(recv_len < 0) { printf("Recieve Data From Server Failed! "); break; } printf("#"); write_leng = fwrite(buf, sizeof(char), recv_len, fp); if (write_leng < recv_len) { printf("Write file failed "); break; } bzero(buf,MAXLINE); } printf(" Finish Recieve "); fclose(fp); close(link_id); } close(sock_id); return 0; } client端代码:client.c [html] view plain copy print?
  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>  
  11.   
  12. #define    MAXLINE        1024  
  13.   
  14. void usage(char *command)  
  15. {  
  16.     printf(“usage :%s ipaddr portnum filename ”, 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;  
  28.      
  29.     if (argc != 4) {  
  30.         usage(argv[0]);  
  31.     }  
  32.       
  33.     /* open the file to be transported commented by guoqingbo*/  
  34.     if ((fp = fopen(argv[3],”r”)) == NULL) {  
  35.         perror(“Open file failed ”);  
  36.         exit(0);  
  37.     }  
  38.       
  39.     /* create the socket commented by guoqingbo*/  
  40.     if ((sock_id = socket(AF_INET,SOCK_STREAM,0)) < 0) {  
  41.         perror(“Create socket failed ”);  
  42.         exit(0);  
  43.     }  
  44.       
  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);  
  49.      
  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 ”);  
  54.        return -1;  
  55.     }  
  56.       
  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 ”);  
  63.             exit(0);  
  64.         }  
  65.         bzero(buf, MAXLINE);  
  66.     }  
  67.   
  68.     fclose(fp);  
  69.     close(sock_id);  
  70.     printf(“Send Finish ”);  
  71.     return 0;  
  72. }  
#include #include #include #include #include #include #include #include #include #include #define MAXLINE 1024 void usage(char *command) { printf("usage :%s ipaddr portnum filename ", command); exit(0); } int main(int argc,char **argv) { struct sockaddr_in serv_addr; char buf[MAXLINE]; int sock_id; int read_len; int send_len; FILE *fp; int i_ret; if (argc != 4) { usage(argv[0]); } /* open the file to be transported commented by guoqingbo*/ if ((fp = fopen(argv[3],"r")) == NULL) { perror("Open file failed "); exit(0); } /* create the socket commented by guoqingbo*/ if ((sock_id = socket(AF_INET,SOCK_STREAM,0)) < 0) { perror("Create socket failed "); exit(0); } memset(&serv_addr, 0, sizeof(serv_addr)); serv_addr.sin_family = AF_INET; serv_addr.sin_port = htons(atoi(argv[2])); inet_pton(AF_INET, argv[1], &serv_addr.sin_addr); /* connect the server commented by guoqingbo*/ i_ret = connect(sock_id, (struct sockaddr *)&serv_addr, sizeof(struct sockaddr)); if (-1 == i_ret) { printf("Connect socket failed "); return -1; } /* transported the file commented by guoqingbo*/ bzero(buf, MAXLINE); while ((read_len = fread(buf, sizeof(char), MAXLINE, fp)) >0 ) { send_len = send(sock_id, buf, read_len, 0); if ( send_len < 0 ) { perror("Send file failed "); exit(0); } bzero(buf, MAXLINE); } fclose(fp); close(sock_id); printf("Send Finish "); return 0; } 下转博文:Linux网络编程之使用UDP传输文件