一.项目课题 ＜基于TCP的文件传输协议实现＞

在这里插入图片描述
客户端代码

需要cJSON.c文件和cJSON.h文件

在这里插入代码片#include "myheadth.h"
#include "myfun.h"#define TIME 10
int sockfd;
void heartbeat(int signum)
{cJSON* root = cJSON_CreateObject();cJSON_AddStringToObject(root,"request","heartbeat");        char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);cJSON_Delete(root);free(JSstr1);root = NULL;alarm(TIME);
}int main(int argc,char **argv)
{if(argc < 3){fprintf(stderr,"Usage <%s SERIP SERPOST>\n",argv[0]);return -1;}/* 1. 创建套接字*/sockfd = socket(AF_INET,SOCK_STREAM,0);if(sockfd == -1){perror("socket");return -1;}/* 2.连接服务器*/sin_t server = {AF_INET};server.sin_port = htons(atoi(argv[2]));server.sin_addr.s_addr = inet_addr(argv[1]);int len = sizeof(sin_t);connect(sockfd,(sad_t*)&server,len);//设置心跳定时器signal(SIGALRM,heartbeat);alarm(TIME);//函数声明
int findall_directory(int sockfd);
int upload_directory(int sockfd);
int download_directory(int sockfd);
void menu_2();while(1){menu();printf("请输入你的选择:");int choice;scanf("%d",&choice);getchar();if(choice == 0){quit(sockfd);alarm(TIME);break;}int flag = 0;switch(choice){case 1://用户登录user_login(sockfd);break;case 2://用户注册user_register(sockfd);break;case 3://用户注销user_logout(sockfd);break;  case 4:// 进入二级菜单menu_2();default:puts("输入有误请重新输入");flag = 1;break;}if(flag)continue;/* 4. 接收服务端消息*/char bufs[512] = {0};recv(sockfd,bufs,sizeof(bufs)-1,0);cJSON* servers = cJSON_Parse(bufs);if(servers == NULL){puts("malloc  node for parse failed");return -1;}alarm(TIME);cJSON* itemRever = cJSON_GetObjectItem(servers,"rever");printf("%s\n",itemRever -> valuestring);if(strcmp(itemRever -> valuestring,"登录成功!") == 0){while(1){menu_2();printf("请输入你的选择:");int choice2;scanf("%d",&choice2);getchar();if(choice2 == 0){alarm(TIME);break;}int flag2 = 0;switch(choice2){case 1://浏览所有目录信息findall_directory(sockfd);//需要修改break;case 2://上传所有信息upload_directory(sockfd);break;case 3://下载所有信息download_directory(sockfd);break;case 4://返回上一级return  0;break;default:puts("输入有误请重新输入");flag = 1;break;}alarm(TIME);if(flag2)continue;}}cJSON_Delete(servers);servers = NULL;}/* 5. 关闭套接字*/close(sockfd);return 0;
}

登录界面以及功能实现

在这里插入代码片#include "myfun.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include "cJSON.h"
void menu()
{printf("\n");printf("\033[1;36m");printf("================消息系统============\n");printf("1. 用户登录\n");printf("2. 用户注册\n");printf("3. 用户注销\n");printf("4. 进入功能\n");printf("0. 退出\n");printf("===================================\n");printf("\033[0m");printf("\n");
}void menu_2()
{printf("\n");printf("\033[1;36m");printf("================消息系统============\n");printf("1. 浏览所有目录信息\n");printf("2. 上传所有目录信息\n");printf("3. 下载所有目录信息\n");printf("0. 返回上一级\n");printf("===================================\n");printf("\033[0m");printf("\n");
}//用户登陆
int user_login(int sockfd)
{char user[20] = {0},passwd[8] = {0};printf("请输入用户名:");fgets(user,20,stdin);user[strlen(user)-1] = 0;printf("请输入密码:");fgets(passwd,8,stdin);passwd[strlen(passwd)-1] = 0;cJSON* root = cJSON_CreateObject();if(root == NULL){puts("malloc root node failed");return -1;}cJSON_AddStringToObject(root,"request","login");        cJSON_AddStringToObject(root,"user",user);       cJSON_AddStringToObject(root,"passwd",passwd);char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);cJSON_Delete(root);free(JSstr1);root = NULL;return 0;
}//用户注册
int user_register(int sockfd)
{char user[20] = {0},passwd[8] = {0};printf("请输入用户名:");fgets(user,20,stdin);user[strlen(user)-1] = 0;// getchar();printf("请输入密码:");fgets(passwd,8,stdin);passwd[strlen(passwd)-1] = 0;sleep(3);cJSON* root = cJSON_CreateObject();if(root == NULL){puts("malloc root node failed");return -1;}cJSON_AddStringToObject(root,"request","register");        cJSON_AddStringToObject(root,"user",user);       cJSON_AddStringToObject(root,"passwd",passwd);char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);cJSON_Delete(root);free(JSstr1);root = NULL;return 0;
}//用户注销
int user_logout(int sockfd)
{char user[20] = {0},passwd[8] = {0};printf("请输入用户名:");fgets(user,20,stdin);user[strlen(user)-1] = 0;printf("请输入密码:");fgets(passwd,8,stdin);passwd[strlen(passwd)-1] = 0;cJSON* root = cJSON_CreateObject();if(root == NULL){puts("malloc root node failed");return -1;}cJSON_AddStringToObject(root,"request","logout");        cJSON_AddStringToObject(root,"user",user);       cJSON_AddStringToObject(root,"passwd",passwd);char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);cJSON_Delete(root);free(JSstr1);root = NULL;return 0;
}// 浏览所有目录以及文件信息
int findall_directory(int sockfd) {cJSON* root = cJSON_CreateObject();if (root == NULL) {puts("malloc root node failed");return -1;}// 修改请求类型为 findall_directorycJSON_AddStringToObject(root, "request", "findall_directory");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);cJSON_Delete(root);free(JSstr1);root = NULL;char bufs[1024 * 1024] = {0};recv(sockfd, bufs, sizeof(bufs) - 1, 0);cJSON* directories = cJSON_Parse(bufs);if (directories == NULL) {puts("malloc node for parse failed");return -1;}cJSON* itemReverse = cJSON_GetObjectItem(directories, "reverse");if (strcmp(itemReverse->valuestring, "null") == 0) {cJSON_Delete(directories);directories = NULL;printf("未有目录信息\n");return 0;}// 解析目录信息cJSON* itemType = cJSON_GetObjectItem(directories, "type");cJSON* itemTitle = cJSON_GetObjectItem(directories, "title");cJSON* itemContent = cJSON_GetObjectItem(directories, "content");cJSON* itemData = cJSON_GetObjectItem(directories, "directory");// 打印目录信息printf("%-18s\t%-70s\t%s\n", itemType->valuestring, itemTitle->valuestring, itemContent->valuestring);if (cJSON_IsArray(itemData)) {int n = cJSON_GetArraySize(itemData);for (int i = 0; i < n; i++) {cJSON* subItem = cJSON_GetArrayItem(itemData, i);cJSON* itemData_name = cJSON_GetObjectItem(subItem, "name");cJSON* itemData_size = cJSON_GetObjectItem(subItem, "size");cJSON* itemData_created_date = cJSON_GetObjectItem(subItem, "created_date");// 打印目录的详细信息printf("%-30s\t%-10d\t%s\n", itemData_name->valuestring, itemData_size->valueint, itemData_created_date->valuestring);}}cJSON_Delete(directories);directories = NULL;return 0;
}// 假设的函数，用于收集目录和文件信息到 cJSON 对象
cJSON* collect_directory_and_file_info() {cJSON* root = cJSON_CreateObject();// 示例：添加一些目录和文件信息cJSON* directories = cJSON_AddArrayToObject(root, "directories");cJSON* files = cJSON_AddArrayToObject(root, "files");// 假设添加一个目录信息cJSON* dir1 = cJSON_CreateObject();cJSON_AddStringToObject(dir1, "name", "/path/to/directory");cJSON_AddNumberToObject(dir1, "size", 1024);cJSON_AddStringToObject(dir1, "created_date", "2025-01-12");cJSON_AddItemToArray(directories, dir1);// 假设添加一个文件信息cJSON* file1 = cJSON_CreateObject();cJSON_AddStringToObject(file1, "name", "/path/to/file.txt");cJSON_AddNumberToObject(file1, "size", 512);cJSON_AddStringToObject(file1, "created_date", "2025-01-11");cJSON_AddItemToArray(files, file1);return root;
}// 上传所有目录以及文件信息
int upload_directory(int sockfd) {cJSON* root = cJSON_CreateObject();if (root == NULL) {puts("malloc root node failed");return -1;}// 修改请求类型为 upload_directorycJSON_AddStringToObject(root, "request", "upload_directory");// 收集目录和文件信息cJSON* directory_and_file_info = collect_directory_and_file_info();cJSON_AddItemToObject(root, "data", directory_and_file_info);char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);cJSON_Delete(root);free(JSstr1);root = NULL;// 接收服务器的响应（可选）char bufs[1024 * 1024] = {0};recv(sockfd, bufs, sizeof(bufs) - 1, 0);cJSON* response = cJSON_Parse(bufs);if (response == NULL) {puts("malloc node for parse failed");return -1;}cJSON* itemReverse = cJSON_GetObjectItem(response, "reverse");if (strcmp(itemReverse->valuestring, "null") == 0) {cJSON_Delete(response);response = NULL;printf("上传失败\n");return -1;} else {printf("上传成功\n");}cJSON_Delete(response);response = NULL;return 0;
}// 下载所有目录及文件信息
int download_directory(int sockfd) {cJSON* root = cJSON_CreateObject();if (root == NULL) {puts("malloc root node failed");return -1;}// 修改请求类型为 download_directorycJSON_AddStringToObject(root, "request", "download_directory");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);cJSON_Delete(root);free(JSstr1);root = NULL;// 接收服务器的响应char bufs[1024 * 1024] = {0};recv(sockfd, bufs, sizeof(bufs) - 1, 0);cJSON* response = cJSON_Parse(bufs);if (response == NULL) {puts("malloc node for parse failed");return -1;}cJSON* itemReverse = cJSON_GetObjectItem(response, "reverse");if (strcmp(itemReverse->valuestring, "null") == 0) {cJSON_Delete(response);response = NULL;printf("没有可下载的目录或文件信息\n");return 0;} else {printf("开始下载目录和文件信息\n");}// 解析并处理下载的目录和文件信息cJSON* directories = cJSON_GetObjectItem(response, "directories");cJSON* files = cJSON_GetObjectItem(response, "files");if (cJSON_IsArray(directories)) {int n = cJSON_GetArraySize(directories);for (int i = 0; i < n; i++) {cJSON* subItem = cJSON_GetArrayItem(directories, i);cJSON* dir_name = cJSON_GetObjectItem(subItem, "name");cJSON* dir_size = cJSON_GetObjectItem(subItem, "size");cJSON* dir_created_date = cJSON_GetObjectItem(subItem, "created_date");// 打印目录信息，可添加下载逻辑printf("Directory: %s, Size: %d, Created Date: %s\n", dir_name->valuestring, dir_size->valueint, dir_created_date->valuestring);}}if (cJSON_IsArray(files)) {int m = cJSON_GetArraySize(files);for (int i = 0; i < m; i++) {cJSON* subItem = cJSON_GetArrayItem(files, i);cJSON* file_name = cJSON_GetObjectItem(subItem, "name");cJSON* file_size = cJSON_GetObjectItem(subItem, "size");cJSON* file_created_date = cJSON_GetObjectItem(subItem, "created_date");// 打印文件信息，可添加下载逻辑printf("File: %s, Size: %d, Created Date: %s\n", file_name->valuestring, file_size->valueint, file_created_date->valuestring);}}cJSON_Delete(response);response = NULL;return 0;
}int quit(int sockfd)
{cJSON* root = cJSON_CreateObject();if(root == NULL){puts("malloc root node failed");return -1;}cJSON_AddStringToObject(root,"request","quit");        char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);cJSON_Delete(root);free(JSstr1);root = NULL;return 0;
}

需要引用cJSON文件 json文件：

服务器端自己构建函数mufun.c

#include "myfun.h"
#include<dirent.h>
#include<sys/stat.h>typedef unsigned long long ull;//数据库连接
static MYSQL* mysql_connect()
{//数据库初始化MYSQL* mysql = mysql_init(NULL);if(mysql == NULL){fprintf(stderr,"mysql_init初始化失败\n");return NULL;}mysql_set_character_set(mysql,"utf8");//数据库连接if(NULL == mysql_real_connect(mysql,"localhost","root","123456","xiaohua",0,NULL,0)){fprintf(stderr,"conn:%s\n",mysql_error(mysql));mysql_close(mysql);return NULL; }return mysql;
}//登录
int login(const char* user,const char* passwd)
{MYSQL* mysql = mysql_connect();char sql[128] = {0};sprintf(sql,"select * from users");if(mysql_real_query(mysql,sql,strlen(sql))){fprintf(stderr,"exec:%s\n",mysql_error(mysql));mysql_close(mysql);return 0; }//获得字符集MYSQL_RES *result = mysql_store_result(mysql);if(result == NULL){fprintf(stderr,"store res:%s\n",mysql_error(mysql));mysql_close(mysql);return 0; }ull rnum = mysql_num_rows(result);//记录数量for(int i = 0; i  < rnum; i++){MYSQL_ROW row = mysql_fetch_row(result);if(strcmp(row[0],user) == 0 && strcmp(row[1],passwd) == 0){mysql_free_result(result);char sql2[128] = {0};sprintf(sql2,"update users set state = 1 where username = '%s'",user);mysql_real_query(mysql,sql2,strlen(sql2));mysql_close(mysql);return 1;}}mysql_free_result(result);mysql_close(mysql);return 0;
}//注册
int regist(const char* user,const char* passwd)
{MYSQL* mysql = mysql_connect();char sql1[128] = {0};sprintf(sql1,"create table if not exists users(username varchar(20) primary key,passwd varchar(8) not null,state int default 0)default charset=utf8");if(mysql_real_query(mysql,sql1,strlen(sql1))){fprintf(stderr,"exec:%s\n",mysql_error(mysql));mysql_close(mysql);return 0; }char sql2[128] = {0};sprintf(sql2,"insert into users(username,passwd) values('%s','%s')",user,passwd);if(mysql_real_query(mysql,sql2,strlen(sql2))){fprintf(stderr,"exec:%s\n",mysql_error(mysql));mysql_close(mysql);return 0; }mysql_close(mysql);return 1;
}//注销
int logout(const char* user,const char* passwd)
{MYSQL* mysql = mysql_connect();char sql[128] = {0};sprintf(sql,"select * from users");if(mysql_real_query(mysql,sql,strlen(sql))){fprintf(stderr,"exec:%s\n",mysql_error(mysql));mysql_close(mysql);return 0; }//获得字符集MYSQL_RES *result = mysql_store_result(mysql);if(result == NULL){mysql_close(mysql);return 0; }ull rnum = mysql_num_rows(result);//记录数量for(int i = 0; i  < rnum; i++){MYSQL_ROW row = mysql_fetch_row(result);if(strcmp(row[0],user) == 0 && strcmp(row[1],passwd) == 0){char sql2[128] = {0};sprintf(sql2,"delete from users where username = '%s'",user);if(mysql_real_query(mysql,sql2,strlen(sql2))){fprintf(stderr,"exec:%s\n",mysql_error(mysql));mysql_close(mysql);return 0; }mysql_free_result(result);mysql_close(mysql);return 1;}}mysql_free_result(result);mysql_close(mysql);return 0;   
}// 浏览所有目录文件信息
int findall_directory(int sockfd) {MYSQL* mysql = mysql_connect();if (mysql == NULL) {fprintf(stderr, "Failed to connect to MySQL database.\n");return -1;}char sql1[128] = {0};cJSON* root = cJSON_CreateObject();// 假设表名为 directories_files，包含目录和文件信息sprintf(sql1, "select * from xiaohua");if (mysql_real_query(mysql, sql1, strlen(sql1))) {fprintf(stderr, "Failed to execute SQL query: %s\n", mysql_error(mysql));mysql_close(mysql);cJSON_AddStringToObject(root, "reverse", "null");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);return 0;}MYSQL_RES *result = mysql_store_result(mysql);if (result == NULL) {fprintf(stderr, "Failed to store query result: %s\n", mysql_error(mysql));mysql_close(mysql);cJSON_AddStringToObject(root, "reverse", "null");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);return -1;}uint32_t fnum = mysql_num_fields(result);my_ulonglong rnum = mysql_num_rows(result);// 获得列名MYSQL_FIELD *fields = mysql_fetch_fields(result);cJSON_AddStringToObject(root, "reverse", "not null");// 假设表的列名为 dir_name, file_name, size, created_datecJSON_AddStringToObject(root, "dir_name", fields[0].name);cJSON_AddStringToObject(root, "file_name", fields[1].name);cJSON_AddStringToObject(root, "size", fields[2].name);cJSON_AddStringToObject(root, "created_date", fields[3].name);cJSON* array = cJSON_CreateArray();for (int i = 0; i < rnum; i++) {cJSON* obj = cJSON_CreateObject();MYSQL_ROW row = mysql_fetch_row(result);for (int j = 0; j < fnum; j++) {cJSON_AddStringToObject(obj, fields[j].name, row[j]);}cJSON_AddItemToArray(array, obj);}cJSON_AddItemToObject(root, "data", array);mysql_free_result(result);mysql_close(mysql);char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);cJSON_Delete(root);return 0;
}// 上传所有目录文件
int upload_directory(int sockfd) {cJSON* root = cJSON_CreateObject();cJSON* array = cJSON_CreateArray();// 打开当前目录DIR* dir = opendir(".");if (dir == NULL) {perror("opendir");cJSON_AddStringToObject(root, "reverse", "null");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);cJSON_Delete(root);return -1;}struct dirent* entry;while ((entry = readdir(dir))!= NULL) {if (entry->d_name[0] == '.') continue; // 跳过隐藏文件和目录struct stat st;if (stat(entry->d_name, &st) == -1) {perror("stat");continue;}cJSON* obj = cJSON_CreateObject();cJSON_AddStringToObject(obj, "name", entry->d_name);cJSON_AddNumberToObject(obj, "size", (int)st.st_size);cJSON_AddStringToObject(obj, "type", S_ISDIR(st.st_mode)? "directory" : "file");// 获取文件的创建时间char time_str[20];strftime(time_str, sizeof(time_str), "%Y-%m-%d %H:%M:%S", localtime(&st.st_ctime));cJSON_AddStringToObject(obj, "created_date", time_str);cJSON_AddItemToArray(array, obj);}closedir(dir);cJSON_AddItemToObject(root, "data", array);cJSON_AddStringToObject(root, "request", "upload_directory");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);cJSON_Delete(root);free(JSstr1);return 0;
}// 下载所有目录文件信息
int download_directory(int sockfd) {cJSON* root = cJSON_CreateObject();cJSON_AddStringToObject(root, "request", "download_directory");// 将请求信息发送到服务器char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd, JSstr1, strlen(JSstr1), 0);cJSON_Delete(root);free(JSstr1);// 接收服务器的响应char buffer[4096];int bytes_received = recv(sockfd, buffer, sizeof(buffer) - 1, 0);if (bytes_received <= 0) {perror("recv");return -1;}buffer[bytes_received] = '\0';// 解析服务器的响应cJSON* response = cJSON_Parse(buffer);if (response == NULL) {fprintf(stderr, "Failed to parse JSON response\n");return -1;}// 检查服务器是否返回错误信息cJSON* reverse = cJSON_GetObjectItem(response, "reverse");if (reverse!= NULL && strcmp(reverse->valuestring, "null") == 0) {fprintf(stderr, "Server returned an error\n");cJSON_Delete(response);return -1;}// 假设服务器返回的数据在 "data" 键下cJSON* data = cJSON_GetObjectItem(response, "data");if (data == NULL ||!cJSON_IsArray(data)) {fprintf(stderr, "Invalid data format from server\n");cJSON_Delete(response);return -1;}// 遍历服务器返回的目录文件信息int array_size = cJSON_GetArraySize(data);for (int i = 0; i < array_size; i++) {cJSON* item = cJSON_GetArrayItem(data, i);if (item == NULL) continue;cJSON* name = cJSON_GetObjectItem(item, "name");cJSON* size = cJSON_GetObjectItem(item, "size");cJSON* type = cJSON_GetObjectItem(item, "type");cJSON* created_date = cJSON_GetObjectItem(item, "created_date");if (name == NULL || size == NULL || type == NULL || created_date == NULL) {fprintf(stderr, "Invalid item format from server\n");continue;}// 这里可以添加逻辑将文件或目录信息保存到本地// 例如，打印文件或目录信息printf("Name: %s, Size: %d, Type: %s, Created Date: %s\n", name->valuestring, size->valueint, type->valuestring, created_date->valuestring);}cJSON_Delete(response);return 0;
}int check(int sockfd,const char* user)
{MYSQL* mysql = mysql_connect();char sql1[128] = {0};cJSON* root = cJSON_CreateObject();sprintf(sql1,"select * from unread where username = '%s'",user);if(mysql_real_query(mysql,sql1,strlen(sql1))){mysql_close(mysql);cJSON_AddStringToObject(root,"reverse","null");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);return 0; }MYSQL_RES *result = mysql_store_result(mysql);if(result == NULL){mysql_close(mysql);cJSON_AddStringToObject(root,"reverse","null");char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);return -1; }uint32_t      fnum = mysql_num_fields(result);my_ulonglong  rnum = mysql_num_rows(result);//获得列名MYSQL_FIELD *fields = mysql_fetch_fields(result);cJSON_AddStringToObject(root,"reverse","not null");cJSON_AddStringToObject(root,"type",fields[1].name);cJSON_AddStringToObject(root,"title",fields[2].name);cJSON_AddStringToObject(root,"content",fields[3].name);cJSON* array = cJSON_CreateArray();for(int i = 0; i  < rnum; i++){MYSQL_ROW  row = mysql_fetch_row(result);cJSON* obj = cJSON_CreateObject();for(int j = 1; j < fnum; j++){cJSON_AddStringToObject(obj,fields[j].name,row[j]);}cJSON_AddItemToArray(array,obj);}cJSON_AddItemToObject(root,"data",array);mysql_free_result(result);char sql2[128] = {0};sprintf(sql2,"delete from unread where username = '%s'",user);mysql_real_query(mysql,sql2,strlen(sql2));mysql_close(mysql);char* JSstr1 = cJSON_PrintUnformatted(root);send(sockfd,JSstr1,strlen(JSstr1),0);return 0;
}//退出
int quit(const char* user)
{MYSQL* mysql = mysql_connect();char sql[128] = {0};sprintf(sql,"update users set state = 0 where username = '%s'",user);if(mysql_real_query(mysql,sql,strlen(sql))){fprintf(stderr,"exec:%s\n",mysql_error(mysql));mysql_close(mysql);return -1; }mysql_close(mysql);return 0;
}

服务器端fock.c

在这里插入代码片#include "mysock.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include "cJSON.h"
#include <dirent.h>
#include <sys/stat.h>
#include <arpa/inet.h>// 假设以下函数在其他地方实现，在此声明它们
int findall_directory(int sockc);
int download_directory(int sockc);
int upload_directory(int sokc);
int login(const char *user, const char *passwd);
int regist(const char *user, const char *passwd);
int logout(const char *user, const char *passwd);
int  quit(const char *user);/*监听函数,返回监听套接字*/
int mysock_init(const char *Ip, const char *Port)
{/* 1. 创建套接字*/int sockl = socket(AF_INET, SOCK_STREAM, 0);if (sockl == -1)return -1;/* 2. 绑定自己地址信息*/struct sockaddr_in server = {0};server.sin_family = AF_INET;server.sin_port = htons(atoi(Port));inet_pton(AF_INET, Ip, &server.sin_addr); //将字符串变成网络字节序socklen_t len = sizeof(server);if (-1 == bind(sockl, (struct sockaddr *)&server, len)){close(sockl);return -1;}/* 3. 监听客户端连接*/if (-1 == listen(sockl, 5)){close(sockl);return -1;}//返回监听套接字return sockl;
}char user_name[20] = {0};/*连接函数,接收客户端连接*/
int mysock_accept(int sockl)
{struct sockaddr_in peer = {0}; //对方socklen_t len = sizeof(peer);int sockc = accept(sockl, (struct sockaddr *)&peer, &len);if (sockc == -1)return -1;char IP[INET_ADDRSTRLEN + 1];inet_ntop(AF_INET, &peer.sin_addr, IP, INET_ADDRSTRLEN);printf("[%s:%d]已连接\n", IP, ntohs(peer.sin_port));//生成连接套接字return sockc;
}/* 接收客户端信息的函数*/
void *from_client(void *argp)
{int sockc = *(int *)argp;struct sockaddr_in peer = {0};socklen_t len = sizeof(peer);getpeername(sockc, (struct sockaddr *)&peer, &len);// 获取客户端地址信息while (1){char str[512] = {0};int n = recv(sockc, str, sizeof(str), 0);if (n == 0){printf("客户端[%s:%d]已断开连接\n", inet_ntoa(peer.sin_addr), ntohs(peer.sin_port));quit(user_name);break;}/* 处理客户端请求*/to_client(str, sockc);}return NULL;
}void to_client(const char *jsStr, int sockc)
{/* 给客户端发消息*/cJSON *root = cJSON_Parse(jsStr);if (root == NULL)return;cJSON *itemRequest = cJSON_GetObjectItem(root, "request");cJSON *itemUser = cJSON_GetObjectItem(root, "user");cJSON *itemPasswd = cJSON_GetObjectItem(root, "passwd");//登录if (strcmp(itemRequest->valuestring, "login") == 0){puts("用户登录");char *res[] = {"登录失败!", "登录成功!"};int i = login(itemUser->valuestring, itemPasswd->valuestring);cJSON *rever = cJSON_CreateObject();if (rever == NULL)return;if (i == 1){memset(user_name, 0, 20);strcpy(user_name, itemUser->valuestring);}cJSON_AddStringToObject(rever, "rever", res[i]);char *JSstr1 = cJSON_PrintUnformatted(rever);send(sockc, JSstr1, strlen(JSstr1), 0);cJSON_Delete(rever);}//注册else if (strcmp(itemRequest->valuestring, "register") == 0){puts("用户注册");char *res[] = {"注册失败!", "注册成功,请登录..."};int i = regist(itemUser->valuestring, itemPasswd->valuestring);cJSON *rever = cJSON_CreateObject();if (rever == NULL)return;cJSON_AddStringToObject(rever, "rever", res[i]);char *JSstr1 = cJSON_PrintUnformatted(rever);send(sockc, JSstr1, strlen(JSstr1), 0);cJSON_Delete(rever);}//注销else if (strcmp(itemRequest->valuestring, "logout") == 0){puts("用户注销");char *res[] = {"注销失败!", "注销成功!"};int i = logout(itemUser->valuestring, itemPasswd->valuestring);cJSON *rever = cJSON_CreateObject();if (rever == NULL)return;cJSON_AddStringToObject(rever, "rever", res[i]);char *JSstr1 = cJSON_PrintUnformatted(rever);send(sockc, JSstr1, strlen(JSstr1), 0);cJSON_Delete(rever);}// 浏览所有目录文件信息else if (strcmp(itemRequest->valuestring, "findall_directory") == 0){puts("浏览所有目录文件信息");findall_directory(sockc);}// 上传目录文件信息else if (strcmp(itemRequest->valuestring, "upload_directory") == 0){puts("上传目录文件信息");int upload_directory(int sockc) {cJSON *root = cJSON_CreateObject();cJSON *array = cJSON_CreateArray();DIR *dir = opendir(".");if (dir == NULL) {perror("opendir");cJSON_AddStringToObject(root, "reverse", "null");char *JSstr = cJSON_PrintUnformatted(root);send(sockc, JSstr, strlen(JSstr), 0);cJSON_Delete(root);free(JSstr);return -1;}struct dirent *entry;while ((entry = readdir(dir))!= NULL) {if (entry->d_name[0] == '.') continue;struct stat st;if (stat(entry->d_name, &st) == -1) {perror("stat");continue;}cJSON *obj = cJSON_CreateObject();cJSON_AddStringToObject(obj, "name", entry->d_name);cJSON_AddNumberToObject(obj, "size", (int)st.st_size);cJSON_AddStringToObject(obj, "type", S_ISDIR(st.st_mode)? "directory" : "file");cJSON_AddItemToArray(array, obj);}closedir(dir);cJSON_AddItemToObject(root, "data", array);cJSON_AddStringToObject(root, "request", "upload_directory");char *JSstr = cJSON_PrintUnformatted(root);send(sockc, JSstr, strlen(JSstr), 0);cJSON_Delete(root);free(JSstr);return 0;}upload_directory(sockc);}// 下载所有目录信息else if (strcmp(itemRequest->valuestring, "download_directory") == 0){puts("下载所有目录信息");// 假设 download_directory 函数的参数为 sockcdownload_directory(sockc);}//检测是否退出else if (strcmp(itemRequest->valuestring, "quit") == 0){quit(user_name);}//心跳检测机制else if (strcmp(itemRequest->valuestring, "heartbeat") == 0){struct sockaddr_in peer = {0}; //对方socklen_t len = sizeof(peer);getpeername(sockc, (struct sockaddr *)&peer, &len);char IP[INET_ADDRSTRLEN + 1];inet_ntop(AF_INET, &peer.sin_addr, IP, INET_ADDRSTRLEN);printf("[%s:%d]还在连接中\n", IP, ntohs(peer.sin_port));}cJSON_Delete(root);
}

服务器端 server.c

#include "myfun.h"
#include "mysock.h"
#include "pthreadpool.h"int main(int argc,char **argv)
{if(argc < 3){fprintf(stderr,"Usage <%s SERIP SERPOST>\n",argv[0]);return -1;}//调用函数threadpool_t pool;threadpool_init(&pool,8,10);/* 1. 调用监听函数*/int sockl = mysock_init(argv[1],argv[2]);if(sockl == -1){perror("mysock_init");return -1;}while(1){/* 2.调用连接函数,接收客户端连接*/int sockc = mysock_accept(sockl);if(sockc == -1)continue;threadpool_addtask(&pool,from_client,(void*)(uint64_t)sockc);}return 0;
}

服务器pthreadPool.c


#include "pthreadpool.h"
#include <stdlib.h>void* thread_fun(void *argp)
{threadpool_t* pool = (threadpool_t*)argp;task_t    *p = NULL;while(1){//互斥锁pthread_mutex_lock(&pool->mutex);//判断队列情况//任务队列为空等待while(pool -> queue_num == 0)pthread_cond_wait(&pool->queue_empty,&pool->mutex);//任务队列不为空,被线程池中线程竞争p = pool -> head;pool -> queue_num--;//表示任务队列只有一个节点if(pool -> queue_num == 0)pool -> head = pool -> tail = NULL;elsepool -> head = p -> next;//判断任务队列是否为满通知其他线程if(pool->queue_num == pool -> queue_max_size -1)pthread_cond_signal(&pool->queue_full);pthread_mutex_unlock(&pool->mutex);//调用函数(*(p -> taskfun))(p -> argp);//调用完毕回收资源free(p);p = NULL;}
}

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.mzph.cn/diannao/66936.shtml

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

Type-C双屏显示器方案

Type-C双屏显示器方案

在数字化时代&#xff0c;高效的信息处理和视觉体验已成为我们日常生活和工作的关键需求。随着科技的进步&#xff0c;一款结合了便携性和高效视觉输出的设备——双屏便携屏&#xff0c;逐渐崭露头角&#xff0c;成为追求高效工作和娱乐体验人群的新宠。本文将深入探讨双屏便携…
阅读更多...
AI开发 - 算法基础 递归 的概念和入门（三）递归的进阶学习

AI开发 - 算法基础 递归 的概念和入门（三）递归的进阶学习

前面我们通过2篇文章&#xff0c;一起了解了 递归&#xff0c;以及使用递归来解决汉诺塔问题。今天我们在这个基础上&#xff0c;进一步地熟悉和学习递归。 这篇学习笔记将涵盖递归的基本概念、应用、优化技巧、陷阱及与迭代的对比&#xff0c;并通过具体的 Python 代码示例和…
阅读更多...
【C++多线程编程：六种锁】

【C++多线程编程：六种锁】

目录 普通互斥锁&#xff1a; 轻量级锁 独占锁&#xff1a; std::lock_guard&#xff1a; std::unique_lock: 共享锁&#xff1a; 超时的互斥锁 递归锁 普通互斥锁&#xff1a; std::mutex确保任意时刻只有一个线程可以访问共享资源&#xff0c;在多线程中常用于保…
阅读更多...
计算机视觉算法实战——车道线检测

计算机视觉算法实战——车道线检测

✨个人主页欢迎您的访问 ✨期待您的三连 ✨ ✨个人主页欢迎您的访问 ✨期待您的三连 ✨ ✨个人主页欢迎您的访问 ✨期待您的三连✨ ​​​​​​ ​​​​​​​​​​​​ ​​​​​ 车道线检测是计算机视觉领域的一个重要研究方向&#xff0c;尤其在自动驾驶和高级驾驶辅助…
阅读更多...
【微服务】面试 3、 服务监控 SkyWalking

【微服务】面试 3、 服务监控 SkyWalking

微服务监控的原因 问题定位&#xff1a;在微服务架构中&#xff0c;客户端&#xff08;如 PC 端、APP 端、小程序等&#xff09;请求后台服务需经过网关再路由到各个微服务&#xff0c;服务间可能存在多链路调用。当某一微服务挂掉时&#xff0c;在复杂的调用链路中难以迅速确定…
阅读更多...
EasyExcel的应用

EasyExcel的应用

一、简单使用 引入依赖&#xff1a; 这里我们可以使用最新的4.0.2版本&#xff0c;也可以选择之前的稳定版本&#xff0c;3.1.x以后的版本API大致相同&#xff0c;新的版本也会向前兼容&#xff08;3.1.x之前的版本&#xff0c;部分API可能在高版本被废弃&#xff09;&…
阅读更多...
【MySQL数据库】基础总结

【MySQL数据库】基础总结

目录 前言 一、概述 二、 SQL 1. SQL通用语法 2. SQL分类 3. DDL 3.1 数据库操作 3.2 表操作 4. DML 5. DQL 5.1 基础查询 5.2 条件查询 5.3 聚合函数 5.4 分组查询 5.5 排序查询 5.6 分页查询 6. DCL 6.1 管理用户 6.2 权限控制 三、数据类型 1. 数值类…
阅读更多...
aws(学习笔记第二十三课) step functions进行开发（lambda函数调用）

aws(学习笔记第二十三课) step functions进行开发（lambda函数调用）

aws(学习笔记第二十三课) 开发step functions状态机的应用程序 学习内容&#xff1a; step functions状态机的概念开发简单的step functions状态机 1. step functions状态机概念 官方说明文档和实例程序 AWS的官方给出了学习的链接和实例程序。使用SAM创建step functions 借…
阅读更多...
安卓开发动画

安卓开发动画

1.gif图片动画 边缘会有锯齿 2.json动画 用lottie json文件动画 实现 Android Studio使用lottie&#xff0c;加载json文件&#xff0c;实现动画效果_android 加载json动画-CSDN博客 遇到的坑 1.不播放&#xff0c;可能因为设置了图片&#xff08;跟动画一样的图片&#xf…
阅读更多...
【Docker】入门教程

【Docker】入门教程

目录 一、Docker的安装 二、Docker的命令 Docker命令实验 1.下载镜像 2.启动容器 3.修改页面 4.保存镜像 5.分享社区 三、Docker存储 1.目录挂载 2.卷映射 四、Docker网络 1.容器间相互访问 2.Redis主从同步集群 3.启动MySQL 五、Docker Compose 1.命令式安装 …
阅读更多...
算法练习7——拦截导弹的系统数量求解

算法练习7——拦截导弹的系统数量求解

题目描述 某国为了防御敌国的导弹袭击&#xff0c;发展出一种导弹拦截系统。但是这种导弹拦截系统有一个缺陷&#xff1a;虽然它的第一发炮弹能够到达任意的高度&#xff0c;但是以后每一发炮弹都不能高于前一发的高度。 假设某天雷达捕捉到敌国的导弹来袭。由于该系统还在试用…
阅读更多...
如何使用高性能内存数据库Redis

如何使用高性能内存数据库Redis

一、详细介绍 1.1、Redis概述 Redis&#xff08;Remote Dictionary Server&#xff09;是一个开源的、内存中的数据结构存储系统&#xff0c;它可以用作数据库、缓存和消息中间件。Redis支持多种类型的数据结构&#xff0c;如字符串&#xff08;strings&#xff09;、哈希&am…
阅读更多...
【Linux系列】`find / -name cacert.pem` 文件搜索

【Linux系列】`find / -name cacert.pem` 文件搜索

&#x1f49d;&#x1f49d;&#x1f49d;欢迎来到我的博客&#xff0c;很高兴能够在这里和您见面&#xff01;希望您在这里可以感受到一份轻松愉快的氛围&#xff0c;不仅可以获得有趣的内容和知识&#xff0c;也可以畅所欲言、分享您的想法和见解。 推荐:kwan 的首页,持续学…
阅读更多...
最大拿牌的得分

最大拿牌的得分

假设有个游戏&#xff0c;一列牌有不同分数&#xff0c;但是只能从两头拿 &#xff0c;拿到最后分数最高的人获胜&#xff0c;假设两个人都是聪明人&#xff0c;求最后的最高分是多少&#xff1f; 思路&#xff1a;递归算法&#xff0c;一个人拿左边牌&#xff0c;另一个人的得…
阅读更多...
UE材质Fab Megascans

UE材质Fab Megascans

2025年Bridge里已经不能直接导入资产了&#xff0c;显示GET IT ON FAB 只能在Fab中导入资产&#xff0c; 纹理打包技术从RMA改成了ORM O&#xff1a;AO 环境光遮蔽 R&#xff1a;Roughness 粗糙度 M&#xff1a;Metallic 金属度 在Fab中找到材质&#xff0c;点击Add to P…
阅读更多...
前后端本地启动

前后端本地启动

一、后端启动 1. 项目导入 目标&#xff1a;将后端代码从远程仓库&#xff08;GitHub&#xff09;导入到开发工具中&#xff08;例如 IntelliJ IDEA&#xff09;&#xff0c;方便我们对项目进行编辑和运行。 步骤&#xff1a; 打开 IntelliJ IDEA&#xff08;下文简称 IDEA…
阅读更多...
ansible 检查目录大小

ansible 检查目录大小

检查目录大小 worker_du.yml# ansible-playbook -i hosts worker_du.yml --limit w10 --- - name: 检查目录大小hosts:- w10 # 可以根据需要修改目标主机# 可以添加更多主机tasks:- name: 获取每台主机 /root/worker01 目录大小shell: du -sh /root/worker01/ | awk {print …
阅读更多...
【NP-hard问题】NP与NP-hard问题通俗解释

【NP-hard问题】NP与NP-hard问题通俗解释

最近在研究NP-hard问题&#xff0c;讲一下自己的对于NP与NP-hard问题的通俗解释 一、NP-Hard 问题是什么意思&#xff1f; 什么是 NP&#xff1f; NP 问题可以理解为「检查答案很容易&#xff0c;但找到答案很难」。 举个例子&#xff1a; 假设你在一个迷宫里&#xff0c;…
阅读更多...
【黑灰产】假钱包推广套路

【黑灰产】假钱包推广套路

假钱包推广产业链研究 市面上钱包的主要推广方式&#xff1a; 1&#xff0c;竞价&#xff08;搜索引擎&#xff09;&#xff0c;误导客户为真正官方钱包从而完成下载使用 优点&#xff1a;精准&#xff0c;客户大 缺点&#xff1a;竞价户容易挂&#xff0c;投资大 2&#xff0…
阅读更多...
C#范围表达式,模式匹配,逆变和协变--11

C#范围表达式,模式匹配,逆变和协变--11

目录 一.范围表达式 1.概述 2.语法 3.代码示例 4.实现原理 5.应用场景 二.模式匹配 1.概述 2.核心概念 3.常用模式类型 4.Switch表达式 5.使用示例 6.优势 三.逆变和协变 1.概述 2.泛型类型参数的变性 3.协变示例 4.逆变示例 5.注意事项 6.应用场景 总结 一…
阅读更多...
最新文章

Copyright @ 2022~2023