电子词典项目需求
项目要求:
- 登录注册功能,不能重复登录,重复注册
- 单词查询功能
- 历史记录功能,存储单词,意思,以及查询时间
- 基于TCP,支持多客户端连接
- 采用数据库保存用户信息与历史记录
- 将dict.txt的数据导入到数据库中保存。
- 按下ctrl+c退出客户端后,注销该客户端的登录信息
格式要求:
- main函数只跑逻辑,不允许跑功能代码
- 功能代码封装成函数
- 先分析需求
- 将上述需求用流程图画出来
- 根据流程图搭建代码框架
- 一个功能一个功能实现
服务器
头文件
#ifndef __HEAD_H__
#define __HEAD_H__
#include <sqlite3.h>
#include <sys/socket.h>
#include <netinet/in.h>
//信号重命名
typedef void (*sighandler_t)(int);
//数据包结构体
struct data_pack{
char data_type; //数据包类型
char account_name[128]; //注册用户名
char account_password[128]; //注册密码
char word[128]; //单词
char translate[128]; //翻译
char htime[128]; //历史记录查询时间
};
//宏定义错误参数
#define ERR_MSG(msg) do{\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0)
//宏定义IP与端口
#define IP "192.168.8.129"
#define PORT 6666
//定义全局变量
struct data_pack data;
//词典数据库
int init_cidian_sjk();
//创建网络
int net_fwq(int sfd);
int deal_cli(int newfd,struct sockaddr_in cin);
//用户数据库
int init_user_sjk();
//注册函数
int regist_account(int newfd);
//登录函数
int login_account(int newfd);
//查询历史记录函数
int history_check(int newfd);
//查询单词函数
int word_search(int newfd);
#endif
功能函数代码
#include <stdio.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/wait.h>
#include <string.h>
#include <sqlite3.h>
#include <time.h>
#include "head.h"
//定义全局变量
sqlite3 *cidian_db = NULL;
sqlite3 *user_db = NULL;
//初始化词典数据库
int init_cidian_sjk(){
//创建并打开一个数据库
if(sqlite3_open("./cidian.db",&cidian_db) != SQLITE_OK){
fprintf(stderr,"sqlite3_open failed:%d:%s\n",sqlite3_errcode(cidian_db),sqlite3_errmsg(cidian_db));
return -1;
}
//创建一张表格
char sql[128] = "create table if not exists egdanci(danci char,fanyi char);";
char *errmsg = NULL;
if(sqlite3_exec(cidian_db,sql,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
//如果表格内有记录,清空表格记录,避免追加
char ssl[128] = "delete from egdanci;";
if(sqlite3_exec(cidian_db,ssl,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
printf("初始化电子词典数据库中\n");
//打开文件
FILE *file=fopen("./four_level.txt","r");
if(NULL==file){
perror("fopen");
return -1;
}
char buf[128]="";
char danci[128]="";
char fanyi_1[128]="";
char sdl[128]="";
while(fgets(buf,sizeof(buf),file) !=NULL){
buf[strlen(buf)-1]='\0';
for(int i=0; buf[i+2] != '\0'; i++){
if(buf[i]!=' ' && buf[i+1]==' ' &&buf[i+2]==' '){
strncpy(danci,buf,i+1);
}else if(buf[i]==' '&&buf[i+1]==' '&&buf[i+2]!=' '){
strcpy(fanyi_1,buf+i+2);
break;
}
}
bzero(sdl,sizeof(sdl));
sprintf(sdl,"insert into egdanci values(\"%s\",\"%s\")",danci,fanyi_1);
if(sqlite3_exec(cidian_db,sdl,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
bzero(danci,sizeof(danci));
bzero(fanyi_1,sizeof(fanyi_1));
bzero(buf,sizeof(buf));
}
fclose(file);
printf("电子词典初始化完毕\n");
return 0;
}
//创建用户数据库
int init_user_sjk(){
//创建并打开一个数据库
if(sqlite3_open("./user.db",&user_db) != SQLITE_OK){
fprintf(stderr,"sqlite3_open failed:%d:%s\n",sqlite3_errcode(user_db),sqlite3_errmsg(user_db));
return -1;
}
//创建注册信息表格
char srl[128] = "create table if not exists regist_user(user_name char PRIMARY KEY, user_password char);";
char *errmsg = NULL;
if(sqlite3_exec(user_db,srl,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
//创建历史记录表格
char shl[128] = "create table if not exists user_history(user_name char, danci char, fanyi char, time char);";
if(sqlite3_exec(user_db,shl,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
//创建登录信息表格
char sll[128] = "create table if not exists login_user(user_name char PRIMARY KEY, newfd int);";
if(sqlite3_exec(user_db,sll,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
printf("打开用户数据库成功\n");
}
//信号默认函数
void handler(int sig){
while(waitpid(-1,NULL,WNOHANG) > 0);
}
//创建网络
int net_fwq(int sfd){
//捕获17号信号
sighandler_t s = signal(SIGCHLD,handler);
if(SIG_ERR == s){
ERR_MSG("signal");
return -1;
}
//允许端口快速重用
int reuse = 1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0){
ERR_MSG("setsockopt");
return -1;
}
//绑定服务器
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
ERR_MSG("bind");
return -1;
}
//监听客户端
if(listen(sfd,128)<0){
ERR_MSG("listen");
return -1;
}
//定义真实信息地址结构体
struct sockaddr_in cin;
socklen_t addrlen = sizeof(cin);
//定义变量
int newfd=0;
pid_t pid = 0;
while(1){
//父进程负责连接客户端
newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
if(newfd < 0){
ERR_MSG("accept");
return -1;
}
pid = fork();
if(0 == pid){
printf("[%s:%d]newfd = %d 客户端连接成功\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
close(sfd);
deal_cli(newfd,cin);
close(newfd);
exit(0);
}else if(pid < 0){
ERR_MSG("fork");
return -1;
}
}
close(sfd);
return 0;
}
//子进程与客户端交互函数
int deal_cli(int newfd,struct sockaddr_in cin){
ssize_t res = 0;
char *errmsg = NULL;
while(1){
//接收客户端请求
bzero(data.account_name,sizeof(data.account_name));
bzero(data.account_password,sizeof(data.account_password));
bzero(data.word,sizeof(data.word));
bzero(data.translate,sizeof(data.translate));
res = recv(newfd,&data,sizeof(data),0);
if(res < 0){
ERR_MSG("recv");
return -1;
}else if(0 == res){
//异常退出
char quit_yichang[128]="";
bzero(quit_yichang,sizeof(quit_yichang));
sprintf(quit_yichang,"delete from login_user where newfd=%d",newfd);
if(sqlite3_exec(user_db,quit_yichang,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
}
fprintf(stderr,"[%s:%d]newfd = %d 客户端下线\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
break;
}
switch(data.data_type)
{
case 'R':
regist_account(newfd);
break;
case 'L':
login_account(newfd);
break;
case 'H':
history_check(newfd);
break;
case 'W':
word_search(newfd);
break;
default:
printf("数据包类型错误,请重新发送数据包\n");
break;
}
}
return 0;
}
//注册函数
int regist_account(int newfd){
//将注册信息加入到注册信息表中
char srl[520]="";
int flag = 0;
char *errmsg = NULL;
bzero(srl,sizeof(srl));
sprintf(srl,"insert into regist_user values(\"%s\",\"%s\")",data.account_name,data.account_password);
if(sqlite3_exec(user_db,srl,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
flag = 1;
}
//让客户端知道注册是否成功
if(send(newfd,&flag,sizeof(flag),0) < 0){
ERR_MSG("send");
return -1;
}
if(flag == 0){
printf("用户名[%s]注册成功\n",data.account_name);
}
return 0;
}
//注册信息表回调函数
int select_callback_regist(void *arg,int column,char **column_text,char **column_name){
*(int*)arg = 0;
if(strcmp(data.account_name,column_text[0]) == 0 && strcmp(data.account_password,column_text[1]) == 0){
*(int*)arg = 1;
}
return 0;
}
//登录函数
int login_account(int newfd){
//将登陆信息与注册信息比对,看用户是否注册成功
char buf[256] = "";
bzero(buf,sizeof(buf));
char *errmsg = NULL;
int flag = 0;
sprintf(buf,"select * from regist_user where user_name = \"%s\"",data.account_name);
if(sqlite3_exec(user_db,buf,select_callback_regist,&flag,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
if(flag == 0){
if(send(newfd,&flag,sizeof(flag),0) < 0){
ERR_MSG("send");
return -1;
}
}else if(flag == 1){
//将登录信息插入登录信息表中
char sll[256]="";
bzero(sll,sizeof(sll));
sprintf(sll,"insert into login_user values(\"%s\",%d)",data.account_name,newfd);
if(sqlite3_exec(user_db,sll,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
flag = 2;
}
if(send(newfd,&flag,sizeof(flag),0) < 0){
ERR_MSG("send");
return -1;
}
}
if(flag == 1){
printf("用户名[%s]登录成功\n",data.account_name);
}
return 0;
}
//查询历史记录回调函数
int select_callback_history(void *arg,int column,char **column_text,char **column_name){
bzero(data.word,sizeof(data.word));
bzero(data.translate,sizeof(data.translate));
bzero(data.htime,sizeof(data.htime));
strcpy(data.word,column_text[1]);
strcpy(data.translate,column_text[2]);
strcpy(data.htime,column_text[3]);
//发送
if(send(*(int*)arg,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
return 0;
}
//查询历史记录函数
int history_check(int newfd){
//查询历史记录
char buf[256] = "";
char *errmsg = NULL;
sprintf(buf,"select * from user_history where user_name = \"%s\"",data.account_name);
if(sqlite3_exec(user_db,buf,select_callback_history,&newfd,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
//历史记录发送结束标志
data.data_type = 'E';
//发送
if(send(newfd,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
}
//查询单词回调函数
int select_callback_word(void *arg,int column,char **column_text,char **column_name){
char *errmsg = NULL;
//查询成功,进入回调函数,将数据包类型改成'Y'
data.data_type = 'Y';
//发送单词翻译
bzero(data.translate,sizeof(data.translate));
strcpy(data.translate,column_text[1]);
if(send(*(int*)arg,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
//设置查询时间
time_t t1,t2;
struct tm* info =NULL;
char timing[128]="";
t2=time(&t1);
info = localtime(&t1);
bzero(timing,sizeof(timing));
sprintf(timing,"%d-%02d-%02d %02d:%02d:%02d",info->tm_year+1900,info->tm_mon+1,info->tm_mday,info->tm_hour,info->tm_min,info->tm_sec);
//将查询结果放入历史记录表中
char insert_history[512]="";
sprintf(insert_history,"insert into user_history values(\"%s\",\"%s\",\"%s\",\"%s\")",data.account_name,data.word,data.translate,timing);
if(sqlite3_exec(user_db,insert_history,NULL,NULL,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
return 0;
}
//查询单词函数
int word_search(int newfd){
//将输入的单词放入词典数据库中比对
char buf[256] = "";
char *errmsg = NULL;
sprintf(buf,"select * from egdanci where danci = \"%s\"",data.word);
if(sqlite3_exec(cidian_db,buf,select_callback_word,&newfd,&errmsg) != SQLITE_OK){
fprintf(stderr,"line:%d sqlite3_exec:%s\n",__LINE__,errmsg);
return -1;
}
//表示查询结束
if(data.data_type == 'Y'){
data.data_type = 'C';
if(send(newfd,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
}else{
//查询失败,将数据包类型改成'S'
data.data_type = 'S';
if(send(newfd,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
}
return 0;
}
主函数代码
#include <stdio.h>
#include "head.h"
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
int main(int argc, const char *argv[])
{
//初始化词典数据库
init_cidian_sjk();
//创建用户数据库
init_user_sjk();
//创建网络-----使用TCP多进程并发服务器
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd < 0){
ERR_MSG("socket");
return -1;
}
net_fwq(sfd);
close(sfd);
return 0;
}
客户端
整体代码
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
//定义数据包结构体
struct data_pack{
char data_type; //数据包类型
char account_name[128]; //注册用户名
char account_password[128]; //注册密码
char word[128]; //单词
char translate[128]; //翻译
char htime[128]; //历史记录查询时间
};
//宏定义错误参数
#define ERR_MSG(msg) do{\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0)
//宏定义IP与端口
#define IP "192.168.8.129"
#define PORT 6666
//定义全局变量
struct data_pack data;
//客户端退出函数
int exit_account();
//函数声明
int net_khd();
int regist_account(int sfd);
int login_account(int sfd);
int history_check(int sfd);
int word_search(int sfd);
//主函数
int main(int argc, const char *argv[])
{
//创建网络
int res;
int sfd = net_khd();
while(1){
system("clear");
printf("------------------\n");
printf("------1.注册------\n");
printf("------2.登录------\n");
printf("------3.退出------\n");
printf("------------------\n");
printf("请输入>>>");
int choose = 0;
scanf("%d",&choose);
while(getchar()!=10);
switch(choose)
{
case 1:
regist_account(sfd);
break;
case 2:
res = login_account(sfd);
if(res == 1){
break;
}else if(res == 0){
while(1){
system("clear");
printf("----------------------\n");
printf("------1.查询单词------\n");
printf("------2.查询历史------\n");
printf("------3.返回上级------\n");
printf("----------------------\n");
printf("请输入>>>");
int chose = 0;
scanf("%d",&chose);
while(getchar()!=10);
switch(chose)
{
case 1:
word_search(sfd);
break;
case 2:
history_check(sfd);
break;
case 3:
goto END;
default:
printf("输入错误,请重新输入\n");
break;
}
printf("输入任意字符清零>>>");
while(getchar()!=10);
}
END:
break;
}
case 3:
goto LAST;
default:
printf("输入错误,请重新输入\n");
break;
}
printf("输入任意字符清零>>>");
while(getchar()!=10);
}
LAST:
close(sfd);
return 0;
}
//创建客户端网络
int net_khd(){
//创建网络
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd < 0){
ERR_MSG("socket");
return -1;
}
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
ERR_MSG("connect");
return -1;
}
return sfd;
}
//注册
int regist_account(int sfd){
//初始化flag
int flag = 2;
//修改数据包类型
data.data_type = 'R';
//注册用户名
bzero(data.account_name,sizeof(data.account_name));
printf("请输入用户名>>>");
fgets(data.account_name,sizeof(data.account_name),stdin);
data.account_name[strlen(data.account_name)-1] = '\0';
//注册密码
bzero(data.account_password,sizeof(data.account_password));
printf("请输入密码>>>");
fgets(data.account_password,sizeof(data.account_password),stdin);
data.account_password[strlen(data.account_password)-1] = '\0';
//发送
if(send(sfd,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
//接收flag
recv(sfd,&flag,sizeof(flag),0);
if(0==flag){
printf("注册成功\n");
}else if(1==flag){
printf("用户已存在,注册失败\n");
}
return 0;
}
//登录
int login_account(int sfd){
//初始化flag
int flag = 0;
//修改数据包类型
data.data_type='L';
//登录用户名
bzero(data.account_name,sizeof(data.account_name));
printf("请输入用户名>>>");
fgets(data.account_name,sizeof(data.account_name),stdin);
data.account_name[strlen(data.account_name)-1] = '\0';
//登录密码
bzero(data.account_password,sizeof(data.account_password));
printf("请输入密码>>>");
fgets(data.account_password,sizeof(data.account_password),stdin);
data.account_password[strlen(data.account_password)-1] = '\0';
//发送
if(send(sfd,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
//接收flag
if(recv(sfd,&flag,sizeof(flag),0) < 0 ){
ERR_MSG("recv");
return -1;
}
if(1==flag){
printf("登录成功\n");
return 0;
}else if(2==flag){
printf("用户重复登录\n");
return 1;
}else if(0==flag){
printf("用户不存在或账户密码输入错误\n");
return 1;
}
}
//历史记录查询
int history_check(int sfd){
//修改数据包类型
data.data_type='H';
bzero(data.word,sizeof(data.word));
bzero(data.translate,sizeof(data.translate));
bzero(data.htime,sizeof(data.htime));
//发送
if(send(sfd,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
//接收历史记录数据包
while(1){
if(recv(sfd,&data,sizeof(data),0) < 0){
ERR_MSG("recv");
return -1;
}
if(data.data_type == 'E'){
return 0;
}
printf("%s %-10s %-10s %-10s\n",data.account_name,data.word,data.translate,data.htime);
}
}
//单词查询
int word_search(int sfd){
//初始化标志位
int flag = 0;
//修改数据包类型
data.data_type='W';
//输入单词
bzero(data.word,sizeof(data.word));
printf("请输入要查询的单词>>>");
fgets(data.word,sizeof(data.word),stdin);
data.word[strlen(data.word)-1] = '\0';
//发送
if(send(sfd,&data,sizeof(data),0) < 0){
ERR_MSG("send");
return -1;
}
while(1){
if(recv(sfd,&data,sizeof(data),0) < 0){
ERR_MSG("recv");
return -1;
}
if(data.data_type == 'S'){
printf("无法查询,单词输入错误,请重新输入\n");
return 0;
}else if(data.data_type == 'C'){
printf("查询单词成功\n");
return 0;
}else if(data.data_type == 'Y'){
printf("%s %s\n",data.word,data.translate);
}
}
}
Makefile
fwq khd:main_fwq_dianzicidian.o dm_fwq_dianzicidian.o khd_dianzicidian.o
gcc main_fwq_dianzicidian.o dm_fwq_dianzicidian.o -o fwq -lsqlite3
gcc khd_dianzicidian.o -o khd
main_fwq_dianzicidian.o:main_fwq_dianzicidian.c
gcc -c main_fwq_dianzicidian.c -o main_fwq_dianzicidian.o
dm_fwq_dianzicidian.o:dm_fwq_dianzicidian.c
gcc -c dm_fwq_dianzicidian.c -o dm_fwq_dianzicidian.o -lsqlite3
khd_dianzicidian.o:khd_dianzicidian.c
gcc -c khd_dianzicidian.c -o khd_dianzicidian.o
.PHONY:clean
clean:
rm *.o fwq khd