网站风格分析,网站建设分金手指专业二,wordpress 5.1后台打开慢,旧宫做网站的公司练习#xff1a; 将 epoll 服务器 客户端拿来用 客户端#xff1a;写一个界面#xff0c;里面有注册登录 服务器#xff1a;处理注册和登录逻辑#xff0c;注册的话将注册的账号密码写入数据库#xff0c;登录的话查询数据库中是否存在账号#xff0c;并验证密码是否正确…
练习 将 epoll 服务器 客户端拿来用 客户端写一个界面里面有注册登录 服务器处理注册和登录逻辑注册的话将注册的账号密码写入数据库登录的话查询数据库中是否存在账号并验证密码是否正确 额外功能客户端登录的时候服务器向客户端发送一个验证码只有验证码也正确的时候才能登录成功· server.c
#include stdio.h
#include string.h
#include unistd.h
#include stdlib.h
#include sys/types.h
#include sys/socket.h
#include arpa/inet.h
#include fcntl.h
#include sys/epoll.h
#include sqlite3.h
#include errno.h
#include time.h#define MAX_EVENTS 100typedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;enum Type {TYPE_REGIST,TYPE_LOGIN,TYPE_CHAT
};typedef struct Pack {enum Type type;char name[20];char pswd[20];char tarname[20];char text[1024];
} pack_t;sqlite3 *db; // 数据库指针// 生成验证码
void generate_code(char *code, int length) {const char charset[] 0123456789;srand(time(NULL));for (int i 0; i length; i) {code[i] charset[rand() % 10];}code[length] \0;
}// 设置文件描述符非阻塞模式
void set_nonblocking(int fd) {int flags fcntl(fd, F_GETFL, 0);fcntl(fd, F_SETFL, flags | O_NONBLOCK);
}// 处理客户端请求
void handle_client(int client) {pack_t pack {0};int res read(client, pack, sizeof(pack));if (res 0) {printf(客户端断开连接\n);close(client);return;}char sql[256];char *errmsg NULL;switch (pack.type) {case TYPE_REGIST:snprintf(sql, sizeof(sql), INSERT INTO users (name, password) VALUES (%s, %s);, pack.name, pack.pswd);if (sqlite3_exec(db, sql, 0, 0, errmsg) ! SQLITE_OK) {printf(注册失败: %s\n, errmsg);strcpy(pack.text, 注册失败该账号可能已存在);sqlite3_free(errmsg);} else {strcpy(pack.text, 注册成功);}write(client, pack, sizeof(pack));break;case TYPE_LOGIN: {snprintf(sql, sizeof(sql), SELECT password FROM users WHERE name %s;, pack.name);sqlite3_stmt *stmt;if (sqlite3_prepare_v2(db, sql, -1, stmt, 0) SQLITE_OK) {if (sqlite3_step(stmt) SQLITE_ROW) {const char *db_pswd (const char*)sqlite3_column_text(stmt, 0);if (strcmp(db_pswd, pack.pswd) 0) {char code[6];generate_code(code, 5);snprintf(pack.text, sizeof(pack.text), 验证码%s\n, code);write(client, pack, sizeof(pack));// 等待客户端输入验证码read(client, pack, sizeof(pack));if (strcmp(pack.text, code) 0) {strcpy(pack.text, 登录成功);} else {strcpy(pack.text, 验证码错误登录失败);}} else {strcpy(pack.text, 密码错误);}} else {strcpy(pack.text, 账号不存在);}sqlite3_finalize(stmt);}write(client, pack, sizeof(pack));break;}case TYPE_CHAT:printf(收到消息: %s\n, pack.text);break;}
}int main(int argc, const char *argv[]) {if (argc ! 2) {printf(请输入端口号\n);return 1;}int port atoi(argv[1]);// 初始化数据库if (sqlite3_open(users.db, db) ! SQLITE_OK) {printf(无法打开数据库\n);return 1;}const char *create_table_sql CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT UNIQUE, password TEXT);;char *errmsg NULL;if (sqlite3_exec(db, create_table_sql, 0, 0, errmsg) ! SQLITE_OK) {printf(创建表失败: %s\n, errmsg);sqlite3_free(errmsg);return 1;}// 创建服务器 socketint server socket(AF_INET, SOCK_STREAM, 0);addr_in_t addr {0};addr.sin_family AF_INET;addr.sin_port htons(port);addr.sin_addr.s_addr INADDR_ANY;bind(server, (addr_t*)addr, sizeof(addr));listen(server, 10);// 设置非阻塞模式set_nonblocking(server);// 创建 epollint epoll_fd epoll_create1(0);struct epoll_event ev, events[MAX_EVENTS];ev.events EPOLLIN;ev.data.fd server;epoll_ctl(epoll_fd, EPOLL_CTL_ADD, server, ev);printf(服务器启动监听端口 %d...\n, port);while (1) {int n epoll_wait(epoll_fd, events, MAX_EVENTS, -1);for (int i 0; i n; i) {if (events[i].data.fd server) {// 处理新连接int client accept(server, NULL, NULL);if (client 0) {perror(accept);continue;}set_nonblocking(client);ev.events EPOLLIN | EPOLLET;ev.data.fd client;epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client, ev);printf(新客户端连接: %d\n, client);} else {// 处理客户端请求handle_client(events[i].data.fd);}}}sqlite3_close(db);close(server);return 0;
}client.c
#include stdio.h
#include string.h
#include unistd.h
#include stdlib.h
#include sys/types.h
#include sys/socket.h
#include arpa/inet.htypedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;enum Type {TYPE_REGIST,TYPE_LOGIN,TYPE_CHAT
};typedef struct Pack {enum Type type;char name[20];char pswd[20];char tarname[20];char text[1024];
} pack_t;void handle_response(int client) {pack_t pack;int res read(client, pack, sizeof(pack));if (res 0) {printf(服务器: %s\n, pack.text);}
}int main(int argc, const char *argv[]) {if (argc ! 2) {printf(输入端口号\n);return 1;}int port atoi(argv[1]);int client socket(AF_INET, SOCK_STREAM, 0);addr_in_t addr {0};addr.sin_family AF_INET;addr.sin_port htons(port);addr.sin_addr.s_addr inet_addr(192.168.126.245);if (connect(client, (addr_t*)addr, sizeof(addr)) -1) {perror(连接失败);return 1;}printf(连接服务器成功\n);while (1) {int ch;printf(1: 注册\n2: 登录\n3: 聊天\n0: 退出\n请选择: );scanf(%d, ch);while (getchar() ! \n);pack_t pack {0};switch (ch) {case 1:printf(输入账号: );scanf(%s, pack.name);while (getchar() ! \n);printf(输入密码: );scanf(%s, pack.pswd);while (getchar() ! \n);pack.type TYPE_REGIST;write(client, pack, sizeof(pack));handle_response(client);break;case 2:printf(输入账号: );scanf(%s, pack.name);while (getchar() ! \n);printf(输入密码: );scanf(%s, pack.pswd);while (getchar() ! \n);pack.type TYPE_LOGIN;write(client, pack, sizeof(pack));handle_response(client); // 服务器发送验证码printf(输入验证码: );scanf(%s, pack.text);while (getchar() ! \n);write(client, pack, sizeof(pack));handle_response(client); // 登录成功 or 失败break;case 3:printf(输入聊天对象: );scanf(%s, pack.tarname);while (getchar() ! \n);printf(输入消息: );fgets(pack.text, sizeof(pack.text), stdin);pack.text[strcspn(pack.text, \n)] \0; // 去除换行符pack.type TYPE_CHAT;write(client, pack, sizeof(pack));handle_response(client);break;case 0:close(client);return 0;default:printf(无效选项\n);}}return 0;
}