1. 程式人生 > >Linux c實現一個tcp文件服務器和客戶端

Linux c實現一個tcp文件服務器和客戶端

repr snippets 功能 stderr strcpy fprintf inet_addr 編寫 create

總體需求:
編寫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. #define N 128
  13. #define PORT_NUM 8888
  14. typedef struct sockaddr SA;
  15. void ProcessList(int connfd)
  16. {
  17. char buf[N];
  18. DIR *mydir;
  19. struct dirent *myitem;
  20. mydir = opendir(".");
  21. while((myitem = readdir(mydir)) != NULL){
  22. if((strcmp(myitem->d_name, ".") == 0) || (strcmp(myitem->d_name, "..") == 0))
  23. continue;
  24. strcpy(buf, myitem->d_name);
  25. send( connfd, buf, N, 0);
  26. }
  27. close(connfd);
  28. return;
  29. }
  30. void ProcessGet(int connfd, char buf[])
  31. {
  32. int fd,nbyte;
  33. if(( fd = open(buf+1, O_RDONLY)) < 0){
  34. fprintf(stderr, "fail to open %s: %s\n",buf+1,strerror(errno));
  35. buf[0] = ‘N‘;
  36. send(connfd, buf, N, 0);
  37. return;
  38. }
  39. buf[0] = ‘Y‘;
  40. send(connfd, buf, N, 0);
  41. while(( nbyte = read( fd, buf, N)) > 0){
  42. send(connfd, buf, nbyte, 0);
  43. }
  44. close(connfd);
  45. return;
  46. }
  47. void ProcessPut(int connfd, char buf[])
  48. {
  49. int fd, nbyte;
  50. if(( fd = open(buf+1, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0){
  51. printf("fail to create %s on server\n",buf+1);
  52. return;
  53. }
  54. while(( nbyte = recv( connfd, buf, N, 0)) > 0){
  55. write(fd, buf, nbyte);
  56. }
  57. close(fd);
  58. return;
  59. }
  60. int main(int argc, char *argv[])
  61. {
  62. int listenfd, connfd;
  63. int optval = 1;
  64. char buf[N];
  65. struct sockaddr_in server_addr;
  66. if(( listenfd = socket( AF_INET, SOCK_STREAM, 0)) < 0 ){
  67. fprintf(stderr, "fail to socket: %s\n",strerror(errno));
  68. exit(-1);
  69. }
  70. #ifdef _DEBUG_
  71. printf("socket is %d\n", listenfd);
  72. #endif
  73. memset(&server_addr, 0, sizeof(server_addr));
  74. server_addr.sin_family = AF_INET;
  75. server_addr.sin_port = htons(PORT_NUM);
  76. server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
  77. setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
  78. if( bind( listenfd, (SA *)(&server_addr), sizeof(server_addr)) < 0){
  79. perror("fail to bind");
  80. exit(-1);
  81. }
  82. listen( listenfd, 5);
  83. while(1){
  84. if(( connfd = accept(listenfd, NULL, NULL)) < 0){
  85. perror("fail to accept");
  86. break;
  87. }
  88. recv( connfd, buf, N, 0);
  89. switch(buf[0]){
  90. case ‘L‘: ProcessList(connfd);
  91. break;
  92. case ‘G‘: ProcessGet(connfd, buf);
  93. break;
  94. case ‘P‘: ProcessPut(connfd, buf);
  95. break;
  96. default: printf("Input ");
  97. }
  98. close(connfd);
  99. }
  100. return 0;
  101. }

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

3.驗證結果(ubuntu14.04)

Linux c實現一個tcp文件服務器和客戶端