• 项目复习:基于TCP的文件服务器

     ser.c(服务器):

    1. #include "./fun.h"
    2.  
    3. int main(int argc,const char * argv[])
    4. {
    5.     //1.判断入参
    6.     if(argc!=3){
    7.         fprintf(stderr,"入参为空，请检查\n");
    8.         return -1;
    9.     }
    10.     //端口号转整型
    11.     int port=atoi(argv[2]);
    12.     //变量声明
    13.     struct sockaddr_in sin;
    14.     int sinLen=sizeof(sin);
    15.     struct sockaddr_in cin;
    16.     int cinLen=sizeof(cin);
    17.     Info_t info;
    18.     int newfd;
    19.     pthread_t tid;
    20.     //2.创建套接字
    21.     int sfd=socket(AF_INET,SOCK_STREAM,0);
    22.     if(sfd==-1){
    23.         PRINT_ERR("socket error");
    24.     }
    25.     //3.快速重用端口号
    26.     int optval=1;
    27.     if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&optval,sizeof(optval))==-1){
    28.         PRINT_ERR("setsockopt error");
    29.     }
    30.     //3.填充服务器结构体信息
    31.     sin.sin_family=AF_INET;
    32.     sin.sin_addr.s_addr=inet_addr(argv[1]);
    33.     sin.sin_port=htons(port);
    34.     //4.绑定
    35.     if(bind(sfd,(struct sockaddr *)&sin,sinLen)==-1){
    36.         PRINT_ERR("bind error");
    37.     }
    38.     //5.监听
    39.     if(listen(sfd,10)==-1){
    40.         PRINT_ERR("listen error");
    41.     }
    42.     //6.多线程并发
    43.     while(1){
    44.         newfd=accept(sfd,(struct sockaddr *)&cin,&sinLen);
    45.         if(newfd==-1){
    46.             PRINT_ERR("accept error");
    47.         }
    48.         info.newfd=newfd;
    49.         //创建线程，将信息结构体传入线程
    50.         if(pthread_create(&tid,NULL,callBack,&info)==-1){
    51.             PRINT_ERR("pthread_create error");
    52.         }
    53.     }
    54.     //7.关闭套接字
    55.     close(sfd);
    56.     return 0;
    57. }

    cli.c(客户端):

    1. #include "./fun.h"
    2.  
    3. int main(int argc, const char *argv[])
    4. {
    5.     // 1.判断入参
    6.     if (argc != 3) {
    7.         fprintf(stderr, "入参为空，请检查\n");
    8.         return -1;
    9.     }
    10.     //变量声明
    11.     struct sockaddr_in sin;
    12.     int sinLen = sizeof(sin);
    13.     buf_t buf;
    14.     char path[128] = "";
    15.     // 2.端口号转整型
    16.     int port = atoi(argv[2]);
    17.     // 3.创建套接字
    18.     int sfd = socket(AF_INET, SOCK_STREAM, 0);
    19.     if (sfd == -1) {
    20.         PRINT_ERR("socket error");
    21.     }
    22.     // 4.填充结构体信息
    23.     sin.sin_family = AF_INET;
    24.     sin.sin_addr.s_addr = inet_addr(argv[1]);
    25.     sin.sin_port = htons(port);
    26.     // 5.连接服务器
    27.     if (connect(sfd, (struct sockaddr *)&sin, sinLen) == -1) {
    28.         PRINT_ERR("connect error");
    29.     }
    30.     while (1) {
    31.         // 6.打印选项菜单
    32.         puts("*****************************");
    33.         puts("*********1.查看文件***********");
    34.         puts("*********2.下载文件***********");
    35.         puts("*********3.上传文件***********");
    36.         puts("*****************************");
    37.         printf("请输入选项:>>>");
    38.         memset(&buf, 0, sizeof(buf));
    39.         memset(path, 0, sizeof(path));
    40.  
    41.         scanf("%d", &buf.type);
    42.         getchar();
    43.         switch (buf.type) {
    44.         case 1:
    45.             break;
    46.         case 2:
    47.             if (getcwd(path, sizeof(path)) == NULL) {
    48.                 PRINT_ERR("getcwd error");
    49.             }
    50.             strcpy(buf.data1, path);
    51.             strcat(buf.data1, "/download/");
    52.             printf("请输入要下载的文件名:>>>");
    53.             if (fgets(buf.data2, sizeof(buf.data2), stdin) == NULL) {
    54.                 PRINT_ERR("fgets error");
    55.             }
    56.             buf.data2[strlen(buf.data2) - 1] = '\0';
    57.             strcat(buf.data1, buf.data2);
    58.             break;
    59.         case 3:
    60.             if (getcwd(path, sizeof(path)) == NULL) {
    61.                 PRINT_ERR("getcwd error");
    62.             }
    63.             strcpy(buf.data1, path);
    64.             strcat(buf.data1, "/download/");
    65.             printf("请输入要上传的文件名:>>>");
    66.             if (fgets(buf.data2, sizeof(buf.data2), stdin) == NULL) {
    67.                 PRINT_ERR("fgets error");
    68.             }
    69.             buf.data2[strlen(buf.data2) - 1] = '\0';
    70.             strcat(buf.data1, buf.data2);
    71.             break;
    72.         default:
    73.             printf("error");
    74.         }
    75.  
    76.         // 7.发送消息
    77.         if (send(sfd, &buf, sizeof(buf), 0) == -1) {
    78.             PRINT_ERR("send error");
    79.         }
    80.         // printf("1\n");
    81.         // 8.接收消息
    82.         while (1) {
    83.             memset(&buf, 0, sizeof(buf));
    84.             if (recv(sfd, &buf, sizeof(buf), 0) == -1) {
    85.                 PRINT_ERR("recv error");
    86.             }
    87.             // buf.data2是quit说明服务器读取文件名已经读完，退出循环
    88.             if (strcmp(buf.data2, "quit") == 0)
    89.                 break;
    90.             puts(buf.data1);
    91.             if (buf.type == 2 || buf.type == 3)
    92.                 break;
    93.         }
    94.         //按任意键继续
    95.         printf("<<请按回车键继续>>");
    96.         getchar();
    97.     }
    98.     // 8.关闭套接字
    99.     close(sfd);
    100.     return 0;
    101. }

    fun.h(头文件):

    1. #ifndef __FUN_H__
    2. #define __FUN_H__
    3. #include 
    4. /********共用***********/
    5. //结构体声明
    6. //线程信息结构体
    7. typedef struct{
    8.     int newfd;
    9. }Info_t;
    10. //消息收发结构体
    11. typedef struct{
    12.     int type;
    13.     char data1[256];
    14.     char data2[256];
    15. }buf_t;
    16. /********服务器端***********/
    17. //函数声明
    18. void *callBack(void *arg);
    19. //显示文件
    20. void showFile(int newfd);
    21. //下载文件
    22. void downloadFile(int newfd,buf_t buf);\
    23. //上传文件
    24. void uploadFile(int newfd,buf_t buf);
    25.  
    26. /********客户端***********/
    27.  
    28.  
    29. #endif

    fun.c(功能文件):

    1. #include "fun.h"
    2.  
    3. void *callBack(void *arg)
    4. {
    5.     //转分离态
    6.     pthread_detach(pthread_self());
    7.     Info_t info = *(Info_t *)arg;
    8.     int newfd = info.newfd;
    9.  
    10.     //变量声明
    11.     buf_t buf;
    12.     //循环收发数据
    13.     while (1) {
    14.         if (recv(newfd, &buf, sizeof(buf), 0) == -1) {
    15.             VPRINT_ERR("recv error");
    16.         }
    17.         switch (buf.type) {
    18.         case 1:
    19.             showFile(newfd);
    20.             break;
    21.         case 2:
    22.             downloadFile(newfd, buf);
    23.             break;
    24.         case 3:
    25.             uploadFile(newfd, buf);
    26.             break;
    27.         default:
    28.             printf("error");
    29.         }
    30.     }
    31.  
    32.     //关闭文件描述符
    33.     close(newfd);
    34.     //退出线程
    35.     pthread_exit(0);
    36. }
    37. //显示文件
    38. void showFile(int newfd)
    39. {
    40.     //变量声明
    41.     DIR *dir = NULL;
    42.     buf_t buf;
    43.     struct dirent *dp = NULL;
    44.     //打开目录
    45.     dir = opendir("./");
    46.     if (NULL == dir) {
    47.         VPRINT_ERR("opendir error");
    48.     }
    49.     while (1) {
    50.         //循环读取目录，直到读完
    51.         dp = readdir(dir);
    52.         if (dp == NULL) {
    53.             if (errno == 0) {
    54.                 strcpy(buf.data2, "quit");
    55.                 if (send(newfd, &buf, sizeof(buf), 0) == -1) {
    56.                     VPRINT_ERR("send error");
    57.                 }
    58.                 break;
    59.             } else {
    60.                 VPRINT_ERR("readdir error");
    61.             }
    62.         }
    63.         memset(&buf, 0, sizeof(buf));
    64.         buf.type = 1;
    65.         strcpy(buf.data1, dp->d_name);
    66.         //循环发送
    67.         if (send(newfd, &buf, sizeof(buf), 0) == -1) {
    68.             VPRINT_ERR("send error");
    69.         }
    70.     }
    71.     //关闭目录，dir指向NULL
    72.     closedir(dir);
    73.     dir = NULL;
    74. }
    75. //下载文件
    76. void downloadFile(int newfd, buf_t buf)
    77. {
    78.     //变量声明
    79.     char server_path[256] = "./";
    80.     char client_path[256] = "";
    81.     char str[128] = "";
    82.     int ret;
    83.  
    84.     strcpy(client_path, buf.data1);
    85.     strcat(server_path, buf.data2);
    86.     // 1.打开2个文件
    87.     int fdr = open(server_path, O_RDONLY);
    88.     int fdw = open(client_path, O_WRONLY | O_CREAT | O_TRUNC, 0664);
    89.     if (fdr == -1 || fdw == -1) {
    90.         VPRINT_ERR("open error");
    91.     }
    92.     while (1) {
    93.         // 2.循环读取
    94.         ret = read(fdr, str, sizeof(str));
    95.         if (ret == -1) {
    96.             VPRINT_ERR("read error");
    97.         } else if (ret == 0) {
    98.             break;
    99.         }
    100.         // 3.循环写入
    101.         write(fdw, str, ret);
    102.     }
    103.     // 4.发送消息
    104.     memset(&buf, 0, sizeof(buf));
    105.     buf.type = 2;
    106.     strcpy(buf.data1, "下载成功");
    107.     if (send(newfd, &buf, sizeof(buf), 0) == -1) {
    108.         VPRINT_ERR("send error");
    109.     }
    110.     // 5.关闭2个文件
    111.     close(fdr);
    112.     close(fdw);
    113. }
    114. //上传文件
    115. void uploadFile(int newfd, buf_t buf)
    116. {
    117.     //变量声明
    118.     char server_path[256] = "./";
    119.     char client_path[256] = "";
    120.     char str[128] = "";
    121.     int ret;
    122.  
    123.     strcpy(client_path, buf.data1);
    124.     strcat(server_path, buf.data2);
    125.     // 1.打开2个文件
    126.     int fdr = open(client_path, O_RDONLY);
    127.     int fdw = open(server_path, O_WRONLY | O_CREAT | O_TRUNC, 0664);
    128.     if (fdr == -1 || fdw == -1) {
    129.         VPRINT_ERR("open error");
    130.     }
    131.     while (1) {
    132.         // 2.循环读取
    133.         ret = read(fdr, str, sizeof(str));
    134.         if (ret == -1) {
    135.             VPRINT_ERR("read error");
    136.         } else if (ret == 0) {
    137.             break;
    138.         }
    139.         // 3.循环写入
    140.         write(fdw, str, ret);
    141.     }
    142.     // 4.发送消息
    143.     memset(&buf, 0, sizeof(buf));
    144.     buf.type = 3;
    145.     strcpy(buf.data1, "上传成功");
    146.     if (send(newfd, &buf, sizeof(buf), 0) == -1) {
    147.         VPRINT_ERR("send error");
    148.     }
    149.     // 5.关闭2个文件
    150.     close(fdr);
    151.     close(fdw);
    152. }

    工程管理文件(makefile):

    1. -include config.mk
    2. #版本1
    3. # a.out:$(NAME).o $(FUN).o
    4. # 	$(CC) $^ -o $@
    5. # %.o:%.c
    6. # 	$(CC) -c $^ -o $@
    7.  
    8. # .PHONY:clean
    9. # clean:
    10. # 	rm -rf $(NAME).o $(FUN).o a.out
    11.  
    12. #版本2（生成动态库版本）
    13. # a.out:$(NAME).o lib$(FUN).so
    14. # 	$(CC) $< -L./lib -l$(FUN) -o $@
    15. # lib$(FUN).so:$(FUN).o
    16. # 	$(CC) -shared $^ -o $@
    17. # 	mv $@ ./lib
    18. # $(NAME).o:$(NAME).c
    19. # 	$(CC) -c $^ -o $@
    20. # $(FUN).o:$(FUN).c
    21. # 	$(CC) -Wall -fPIC -c $^ -o $@
    22. # .PHONY:clean
    23. # clean:
    24. # 	rm -rf *.o lib/*.so a.out
    26. #版本3（生成动态库版本+链接线程库版）
    27. # a.out:$(NAME).o lib$(FUN).so
    28. # 	$(CC) $< -L./lib -l$(FUN) -o $@ -lpthread
    29. # lib$(FUN).so:$(FUN).o
    30. # 	$(CC) -shared $^ -o $@
    31. # 	mv $@ ./lib
    32. # $(NAME).o:$(NAME).c
    33. # 	$(CC) -c $^ -o $@
    34. # $(FUN).o:$(FUN).c
    35. # 	$(CC) -Wall -fPIC -c $^ -o $@
    36. # .PHONY:clean
    37. # clean:
    38. # 	rm -rf *.o lib/*.so a.out
    40. #版本4（生成动态库版本+非父子进程通信版本）
    41. # .PHONY:all
    42. # all:F A B
    43. # A:$(NAMEA).o lib$(FUN).so
    44. # 	$(CC) $< -L./lib -l$(FUN) -o $@
    45. # B:$(NAMEB).o lib$(FUN).so
    46. # 	$(CC) $< -L./lib -l$(FUN) -o $@
    47. # F:$(NAMEF).o lib$(FUN).so
    48. # 	$(CC) $< -L./lib -l$(FUN) -o $@
    49. # lib$(FUN).so:$(FUN).o
    50. # 	$(CC) -shared $^ -o $@
    51. # 	mv $@ ./lib
    52. # $(NAMEF).o:$(NAMEF).c
    53. # 	$(CC) -c $^ -o $@
    54. # $(NAMEA).o:$(NAMEA).c
    55. # 	$(CC) -c $^ -o $@
    56. # $(NAMEB).o:$(NAMEB).c
    57. # 	$(CC) -c $^ -o $@
    58. # $(FUN).o:$(FUN).c
    59. # 	$(CC) -Wall -fPIC -c $^ -o $@
    60. # .PHONY:clean
    61. # clean:
    62. # 	rm -rf *.o lib/*.so A B F
    64. #版本5（生成动态库版本+非父子进程通信版本+链接线程库版本）
    65. # .PHONY:all
    66. # all:F A B
    67. # A:$(NAMEA).o lib$(FUN).so
    68. # 	$(CC) $< -L./lib -l$(FUN) -o $@ -lpthread
    69. # B:$(NAMEB).o lib$(FUN).so
    70. # 	$(CC) $< -L./lib -l$(FUN) -o $@ -lpthread
    71. # F:$(NAMEF).o lib$(FUN).so
    72. # 	$(CC) $< -L./lib -l$(FUN) -o $@
    73. # lib$(FUN).so:$(FUN).o
    74. # 	$(CC) -shared $^ -o $@
    75. # 	mv $@ ./lib
    76. # $(NAMEF).o:$(NAMEF).c
    77. # 	$(CC) -c $^ -o $@
    78. # $(NAMEA).o:$(NAMEA).c
    79. # 	$(CC) -c $^ -o $@
    80. # $(NAMEB).o:$(NAMEB).c
    81. # 	$(CC) -c $^ -o $@
    82. # $(FUN).o:$(FUN).c
    83. # 	$(CC) -Wall -fPIC -c $^ -o $@
    84. # .PHONY:clean
    85. # clean:
    86. # 	rm -rf *.o lib/*.so A B F
    88. #版本6（生成动态库版本+客户端服务器通信版本）
    89. # .PHONY:all
    90. # all:cli ser
    91. # ser:$(NAMESER).o lib$(FUN).so
    92. # 	$(CC) $< -L./lib -l$(FUN) -o $@
    93. # cli:$(NAMECLI).o lib$(FUN).so
    94. # 	$(CC) $< -L./lib -l$(FUN) -o $@
    95. # lib$(FUN).so:$(FUN).o
    96. # 	$(CC) -shared $^ -o $@
    97. # 	mv $@ ./lib
    98. # $(NAMESER).o:$(NAMESER).c
    99. # 	$(CC) -c $^ -o $@
    100. # $(NAMECLI).o:$(NAMECLI).c
    101. # 	$(CC) -c $^ -o $@
    102. # $(FUN).o:$(FUN).c
    103. # 	$(CC) -Wall -fPIC -c $^ -o $@
    104. # .PHONY:clean
    105. # clean:
    106. # 	rm -rf *.o lib/*.so ser cli
    108. #版本6（生成动态库版本+客户端服务器通信版本+线程库版本）
    109. .PHONY:all
    110. all:cli ser
    111. ser:$(NAMESER).o lib$(FUN).so
    112. 	$(CC) $< -L./lib -l$(FUN) -o $@ -lpthread
    113. cli:$(NAMECLI).o lib$(FUN).so
    114. 	$(CC) $< -L./lib -l$(FUN) -o $@ -lpthread
    115. lib$(FUN).so:$(FUN).o
    116. 	$(CC) -shared $^ -o $@
    117. 	mv $@ ./lib
    118. $(NAMESER).o:$(NAMESER).c
    119. 	$(CC) -c $^ -o $@
    120. $(NAMECLI).o:$(NAMECLI).c
    121. 	$(CC) -c $^ -o $@
    122. $(FUN).o:$(FUN).c
    123. 	$(CC) -Wall -fPIC -c $^ -o $@
    124. .PHONY:clean
    125. clean:
    126. 	rm -rf *.o lib/*.so ser cli

    工程管理配置文件(config.mk):

    1. #对应版本1，2，3的makefile
    2. # CC=gcc
    3. # NAME=test
    4. # FUN=fun
    5.  
    6. #对应版本4,5的makefile
    7. # CC=gcc
    8. # NAMEA=Atest
    9. # NAMEB=Btest
    10. # NAMEF=Ftest
    11. # FUN=fun
    12.  
    13. #对应版本6的makefile
    14. CC=gcc
    15. NAMESER=ser
    16. NAMECLI=cli
    17. FUN=fun

    测试结果:

    编译makefile：

     打开服务器和客户端：

     上传和下载文件的位置：

     选择1查看文件：

     选择2下载文件：

     另起一个终端客户端，选择3上传文件：

     关闭服务器和客户端，diff测试下载以及上传的文件是否和原来的一致：

     

  • 相关阅读:
    TS 常考知识点记录
    8/23 网络流、最大流+hungary算法
    人工智能学习01--errors
    【Arduino+ESP32专题】外部中断的使用
    【Github】git本地仓库建立与远程连接
    前端面试题 - 元素的innerText outerText innerHTML的区别？
    【Python】-- Turtle绘图（使用代码画喜欢的图形！）
    云原生Kubernetes：K8S集群使用带凭证的harbor仓库
    figma插件都有哪些好用的，分享11个提效插件
    10_18Qt
  • 原文地址：https://blog.csdn.net/qq_53183630/article/details/128156488