Linux c实现一个tcp文件服务器和客户端

总体需求：
编写tcp文件服务器和客户端。客户端可以上传和下载文件。



================================================

分解需求



客户端功能描述：

1）要求支持以下命令：
help: 显示客户端所有命令和说明, 在本地实现即可，help的内容不需要从服务器传回。
list: 显示服务器端可下载文件列表，列表内容需要从服务器端传回。
get <filename>: 根据<filename>下载指定文件，<filename>只包含文件名，如果出现"/"等字符任务是路径，不予支持；下载后的文件存放在当前工作路径下即可。
put <pathname>: 上传文件 <pathname> 必须为客户端本机的合法路径，客户端搜索到文件后推送给服务器
quit: 退出客户端
2）客户端启动后可以反复输入命令，除非用户输入quit才会退出。
3) 每次命令（list/get/put）建立一个连接，命令执行完毕后关闭该连接。


服务器端功能：

1）文件服务器不要求支持并发，同时只支持一个连接，即一个客户端发起的一次命令。一次命令执行完毕后关闭连接并继续等待下一次连接请求。
2）文件服务器启动后一直执行，除非被人为强制关闭。
3）文件服务器端需要设定一个目录用于存放所有的文件，该目录路径不支持可配置，且该目录不要求再包含子目录。称其为"文件存放目录"。
4）对list服务，服务器端从"文件存放目录"下列举出所有的文件名称并发送给客户端。
5）对get服务，服务器根据用户指定的文件名缺省从"文件存放目录"搜索该文件并推送文件到客户端，推送不会删除原服务器上的文件。
6）对put服务，服务器将用户推送的文件存储到"文件存放目录"，如果存在同名的文件则拒绝。
7）若执行命令和传输过程中发生错误，关闭当前连接并进入等待下一个连接。


=================================================
提示：

请在编码前仔细设计一个简单的应用层协议规定客户端和服务器端之间命令传输的请求和应答流程和格式。

二。参考代码：

1.服务器端参考代码

[cpp] view plain copy
 
  1. #include<stdio.h>  
  2. #include<stdlib.h>  
  3. #include<string.h>  
  4. #include<unistd.h>  
  5. #include<sys/types.h>  
  6. #include<sys/socket.h>  
  7. #include<netinet/in.h>  
  8. #include<arpa/inet.h>  
  9. #include<fcntl.h>  
  10. #include<dirent.h>  
  11. #include<errno.h>  
  12.   
  13. #define N 128   
  14. #define PORT_NUM 8888  
  15.   
  16. typedef struct sockaddr SA;  
  17.   
  18. void ProcessList(int connfd)  
  19. {  
  20.     char buf[N];  
  21.     DIR *mydir;  
  22.     struct dirent *myitem;  
  23.   
  24.     mydir = opendir(".");  
  25.     while((myitem = readdir(mydir)) != NULL){  
  26.         if((strcmp(myitem->d_name, ".") == 0) || (strcmp(myitem->d_name, "..") == 0))  
  27.             continue;  
  28.         strcpy(buf, myitem->d_name);  
  29.         send( connfd, buf, N, 0);  
  30.     }  
  31.     close(connfd);  
  32.   
  33.     return;  
  34. }  
  35.   
  36. void ProcessGet(int connfd, char buf[])  
  37. {  
  38.     int fd,nbyte;  
  39.   
  40.     if(( fd = open(buf+1, O_RDONLY)) < 0){  
  41.         fprintf(stderr, "fail to open %s: %s\n",buf+1,strerror(errno));  
  42.         buf[0] = 'N';  
  43.         send(connfd, buf, N, 0);  
  44.         return;  
  45.     }  
  46.     buf[0] = 'Y';  
  47.     send(connfd, buf, N, 0);  
  48.     while(( nbyte = read( fd, buf, N)) > 0){  
  49.         send(connfd, buf, nbyte, 0);      
  50.     }  
  51.   
  52.     close(connfd);  
  53.   
  54.     return;  
  55. }  
  56.   
  57. void ProcessPut(int connfd, char buf[])  
  58. {  
  59.     int fd, nbyte;  
  60.   
  61.     if(( fd = open(buf+1, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0){  
  62.         printf("fail to create %s on server\n",buf+1);  
  63.         return;  
  64.     }  
  65.   
  66.     while(( nbyte = recv( connfd, buf, N, 0)) > 0){  
  67.         write(fd, buf, nbyte);    
  68.     }  
  69.     close(fd);  
  70.   
  71.     return;  
  72. }  
  73.   
  74. int main(int argc, char *argv[])  
  75. {  
  76.     int listenfd, connfd;  
  77.     int optval = 1;  
  78.     char buf[N];  
  79.     struct sockaddr_in server_addr;  
  80.   
  81.     if(( listenfd = socket( AF_INET, SOCK_STREAM, 0)) < 0 ){  
  82.         fprintf(stderr, "fail to socket: %s\n",strerror(errno));  
  83.         exit(-1);  
  84.     }  
  85.   
  86. #ifdef _DEBUG_  
  87.     printf("socket is %d\n", listenfd);  
  88. #endif  
  89.   
  90.     memset(&server_addr, 0, sizeof(server_addr));  
  91.     server_addr.sin_family = AF_INET;  
  92.     server_addr.sin_port = htons(PORT_NUM);  
  93.     server_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
  94.       
  95.     setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));  
  96.   
  97.     if( bind( listenfd, (SA *)(&server_addr), sizeof(server_addr)) < 0){  
  98.         perror("fail to bind");  
  99.         exit(-1);  
  100.     }  
  101.     listen( listenfd, 5);  
  102.   
  103.     while(1){  
  104.         if(( connfd = accept(listenfd, NULL, NULL)) < 0){  
  105.             perror("fail to accept");  
  106.             break;  
  107.         }     
  108.         recv( connfd, buf, N, 0);  
  109.   
  110.         switch(buf[0]){  
  111.             case 'L': ProcessList(connfd);  
  112.                 break;  
  113.             case 'G': ProcessGet(connfd, buf);  
  114.                 break;  
  115.             case 'P': ProcessPut(connfd, buf);  
  116.                 break;  
  117.             default: printf("Input ");  
  118.         }  
  119.         close(connfd);  
  120.     }  
  121.   
  122.     return 0;  
  123. }  

2.客户端参考代码：

[cpp] view plain copy
 
  1. #include<stdio.h>  
  2. #include<stdlib.h>  
  3. #include<unistd.h>  
  4. #include<string.h>  
  5. #include<fcntl.h>  
  6. #include<errno.h>  
  7. #include<sys/types.h>  
  8. #include<sys/socket.h>  
  9. #include<netinet/in.h>  
  10. #include<arpa/inet.h>  
  11.   
  12. #define N 128  
  13. #define PORT_NUM 8888  
  14.   
  15. typedef struct sockaddr SA;   
  16.   
  17. void PrintHelp()  
  18. {  
  19.     printf("help: display help info\n");  
  20.     printf("list: get file list of server\n");  
  21.     printf("get : get <file>\n");  
  22.     printf("put : put <file>\n");  
  23.     printf("quit: quit the client\n");  
  24.   
  25.     return;  
  26. }  
  27.   
  28. void ProcessList(struct sockaddr_in server_addr)  
  29. {  
  30.     int sockfd, nbyte;  
  31.     char buf[N];  
  32.   
  33.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
  34.         printf("fail to list\n");  
  35.         return;  
  36.     }  
  37.   
  38.     if( connect(sockfd, (SA *)(&server_addr), sizeof(server_addr)) < 0){  
  39.         printf("fail to connect server\n");  
  40.         goto ERROR_1;  
  41.     }  
  42.   
  43.     strcpy(buf, "L");  
  44.     send(sockfd, buf, N, 0);  
  45.   
  46.     while(( nbyte = recv( sockfd, buf, N, 0)) != 0){  
  47.         printf("%s\n",buf);   
  48.     }  
  49.   
  50. ERROR_1:  
  51.     close(sockfd);  
  52.   
  53.     return;  
  54. }  
  55.   
  56. void ProcessGet(struct sockaddr_in server_addr, char command[])  
  57. {  
  58.     int sockfd, nbyte, fd;  
  59.     char buf[N];  
  60.   
  61.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
  62.         printf("fail to get\n");  
  63.         return;  
  64.     }  
  65.   
  66.     if( connect( sockfd, (SA *)(&server_addr), sizeof(server_addr)) < 0){  
  67.         printf("fail to connect server\n");  
  68.         goto ERROR_2;  
  69.     }  
  70.   
  71.     sprintf(buf, "G%s", command+4);  
  72.     send(sockfd, buf, N, 0);  
  73.     recv(sockfd, buf, N, 0);  
  74.   
  75.     if(buf[0] == 'N'){  
  76.         printf("No such file on server\n");  
  77.         goto ERROR_2;  
  78.     }  
  79.   
  80.     if(( fd = open(command+4, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0){  
  81.         printf("fail to create local file %s\n",command+4);  
  82.         goto ERROR_2;  
  83.     }  
  84.   
  85.     while(( nbyte = recv(sockfd, buf, N, 0)) > 0){  
  86.         write(fd, buf, nbyte);    
  87.     }  
  88.     close(fd);  
  89.   
  90. ERROR_2:  
  91.     close(sockfd);  
  92.   
  93.     return;  
  94. }  
  95.   
  96. void ProcessPut(struct sockaddr_in server_addr, char command[])  
  97. {  
  98.     int sockfd, fd, nbyte;  
  99.     char buf[N];  
  100.   
  101.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
  102.         printf("fail to get\n");  
  103.         return;  
  104.     }  
  105.       
  106.     if( connect(sockfd,(SA *)(&server_addr),sizeof(server_addr)) < 0){  
  107.         printf("fail to connect server\n");  
  108.         goto ERROR_3;  
  109.     }  
  110.   
  111.     if((fd = open(command+4, O_RDONLY)) < 0){  
  112.         printf("fail to open %s\n",command+4);  
  113.         goto ERROR_3;  
  114.     }  
  115.     sprintf(buf, "P%s", command+4);  
  116.     send(sockfd, buf, N, 0);  
  117.   
  118.     while(( nbyte = read(fd, buf, N)) > 0){  
  119.         send(sockfd, buf, nbyte, 0);      
  120.     }  
  121.     close(fd);  
  122.   
  123. ERROR_3:  
  124.     close(sockfd);  
  125.   
  126.     return;  
  127. }  
  128.   
  129. int main(int argc, char *argv[])  
  130. {  
  131.     int sockfd, fd, nbyte;  
  132.     char command[32];  
  133.     struct sockaddr_in server_addr;  
  134.   
  135.     if(argc < 3){  
  136.         printf("Usage: %s <server_ip> <port>\n",argv[0]);  
  137.         exit(-1);  
  138.     }  
  139.   
  140.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
  141.         fprintf(stderr, "fail to socket: %s\n", strerror(errno));  
  142.         exit(-1);  
  143.     }  
  144.   
  145. #ifdef _DEBUG_  
  146.     printf("socket is %d\n",sockfd);  
  147. #endif  
  148.       
  149.     memset(&server_addr, 0, sizeof(server_addr));  
  150.     server_addr.sin_family = AF_INET;  
  151.     server_addr.sin_port = htons(atoi(argv[2]));  
  152.     server_addr.sin_addr.s_addr = inet_addr(argv[1]);  
  153.     //server_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
  154.   
  155.     while(1){  
  156.         printf("<client>");  
  157.         fgets(command, 32, stdin);  
  158.         command[strlen(command)-1] = '\0';  
  159.         if(strcmp( command, "help") == 0){  
  160.             PrintHelp();      
  161.         }  
  162.         else if(strcmp( command, "list") == 0){  
  163.             ProcessList(server_addr);     
  164.         }  
  165.         else if(strncmp( command, "get", 3) == 0){  
  166.             ProcessGet(server_addr, command);     
  167.         }  
  168.         else if(strncmp( command, "put", 3) == 0){  
  169.             ProcessPut(server_addr, command);  
  170.         }  
  171.         else if(strcmp( command, "quit") == 0){  
  172.             printf("Bye\n");  
  173.             break;  
  174.         }  
  175.         else{  
  176.             printf("Wrong command, 'help' for command list.\n");      
  177.         }  
  178.     }  
  179.   
  180.     return 0;  
  181. }  

3.验证结果（ubuntu14.04）

posted on 2017-05-20 11:43  baraka  阅读(791)  评论(0)  编辑  收藏  举报