实现功能:
- 用户注册,不能重复注册
- 保存用户信息
- 当账号在线时,其他用户不能使用该账号登录
- 单词查询功能
- 历史记录功能:存储单词、意思、以及查询时间
- 基于TCP,支持多客户端连接。
- 将dict.txt的数据导入到数据库中
- 输入“/q”退出客户端,与服务器断开连接
将字典导入sqlite3数据库:
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sqlite3.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
int putin_db(int fd, sqlite3 *db)
{
char *errmsg = NULL;
char cmd[128] = "";
char key[128] = "";
char translation[128] = "";
char buf[2] = "";
int flag = 1;
bzero(key,sizeof(key));
bzero(translation,sizeof(translation));
int i = 0;
int j = 0;
while(1)
{
if(read(fd,buf,1) == 0)
break;
if(flag == 1)
{
while(buf[0] != ' ')
{
key[i++] = buf[0];
read(fd,buf,1);
}
flag = 0;
}
while(buf[0] == ' ')
read(fd,buf,1);
if(flag == 0)
{
while(buf[0] != '\n')
{
translation[j++] = buf[0];
read(fd,buf,1);
}
flag = 1;
}
if(buf[0] == '\n')
{
sprintf(cmd,"insert into dict values (\"%s\", \"%s\");",key,translation);
if(0 != sqlite3_exec(db, cmd, NULL, NULL,&errmsg) )
{
printf("__%d__ %s\n",__LINE__,errmsg);
return -1;
}
bzero(key,sizeof(key));
bzero(translation,sizeof(translation));
bzero(cmd,sizeof(cmd));
i = 0;
j = 0;
}
}
}
int main(int argc, const char *argv[])
{
//打开数据库
sqlite3 *db = NULL;
if(0 != sqlite3_open("./dict.db",&db))
{
printf("__%d__ %s\n",__LINE__,sqlite3_errmsg(db));
return -1;
}
//创建表格
char *errmsg = NULL;
char sql[128] = "create table if not exists dict(word char, translation char);";
if(0 != sqlite3_exec(db, sql, NULL, NULL, &errmsg))
{
printf("__%d__ %s\n",__LINE__,errmsg);
return -1;
}
//打开dict文件
int fd = open("./dict.txt", O_RDONLY);
if(fd < 0)
{
perror("open false\n");
return -1;
}
putin_db(fd, db);
//关闭数据库
if(0 != sqlite3_close(db))
{
printf("__%d__ %s\n",__LINE__,sqlite3_errmsg(db));
return -1;
}
//关闭文件描述符
close(fd);
return 0;
}
电子词典服务器:
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sqlite3.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <time.h>
#define ERR_MSG(msg) do{\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.72.128" //服务器IP
struct cli_msg
{
int newfd;
struct sockaddr_in cin;
sqlite3 *db;
};
struct callback_data
{
int flag;
char buf[128];
char **text;
};
//打开数据库
sqlite3 *open_database(sqlite3 *db, char *path_db)
{
if(0 != sqlite3_open(path_db, &db))
{
printf("__%d__ %s\n",__LINE__,sqlite3_errmsg(db));
return NULL;
}
return db;
}
//删除表格
int delete_table(sqlite3 *db, char *sql)
{
char *errmsg = NULL;
if(0 != sqlite3_exec(db, sql, NULL, NULL, &errmsg))
{
printf("__%d__ %s\n",__LINE__,errmsg);
return -1;
}
return 0;
}
//命令执行语句
int create_table(sqlite3 *db, char *sql)
{
char *errmsg = NULL;
if(0 != sqlite3_exec(db, sql, NULL, NULL, &errmsg))
{
printf("__%d__ %s\n",__LINE__,errmsg);
return -1;
}
return 0;
}
//插入账号信息
int user_insert(char *buf,sqlite3 *dict_db)
{
char sql[128] = "";
sprintf(sql, "insert into users (user) values(\"%s\");",buf);
char *errmsg = NULL;
if(0 != sqlite3_exec(dict_db, sql, NULL, NULL, &errmsg))
return -1;
return 0;
}
//callback查询的回调函数
int select_callback(void *flag, int ncolumn, char **column_text, char **column_name)
{
struct callback_data *data = (struct callback_data *)flag;
data->text = column_text;
if(strcmp(column_text[2],"-1") == 0) //该账不号在线
{
data->flag = -1;
}
else //该账号在线
{
data->flag = 1;
}
return 0;
}
//发送协议
int send_ask(int newfd,const char * ask)
{
if(send(newfd, ask, strlen(ask), 0) < 0)
{
ERR_MSG("send_ask");
return -1;
}
return 0;
}
//接收客户端消息
int recv_ask(int newfd, char *buf)
{
int res = recv(newfd, buf, sizeof(buf), 0);
if(res < 0)
{
ERR_MSG("recv");
return -1;
}else if(0 == res)
{
printf("客户端下线\n");
return -1;
}
return 0;
}
//协议第一个字符
// 1——>>输入账号
// 2——>>输入密码
// 3——>>用户在线
// 4——>>密码错误
// 5——>>登录成功
//用户登录
int log_in (struct cli_msg info, char *user_log)
{
int newfd = info.newfd;
struct sockaddr_in cin = info.cin;
char user_num[128] = "1";
char user_pass[128] = "2";
char buf[128] = "";
char word[128] = "";
char *errmsg = NULL; //错误码
while(1)
{
bzero(buf, sizeof(buf));
bzero(word, sizeof(word));
//让客户端发送账号信息
if(send_ask(newfd,user_num) < 0)
return -1;
//接收账号信息
if(recv_ask(newfd,buf) < 0)
return -1;
//将账号插入用户表格
int state = user_insert(buf, info.db);
//将账号信息传参给回调函数
struct callback_data data;
data.flag = 0;
strcpy(data.buf,buf);
//查询账号的命令语句
char sql[128] = "";
sprintf(sql,"select * from users where user==\"%s\";",buf);
if(state < 0) //state < 0该账号已存在,检查该账号是否在线
{
if(sqlite3_exec(info.db, sql, select_callback, &data, &errmsg) != 0)
{
fprintf(stderr,"查询失败\n");
return -1;
}
if(data.flag > 0)
{
//给客户端发送在线信息
if(send_ask(newfd,"3") < 0)
return -1;
fprintf(stderr,"该账号在线,请重新登录\n");
continue;
}
while(1)
{
//发送输入密码的消息
if(send_ask(newfd, user_pass) < 0)
return -1;
//接收密码信息
if(recv_ask(newfd, word) < 0)
return -1;
if(strcmp(word, data.text[1]) == 0)
{
puts("登录成功");
if(send_ask(newfd, "5") < 0)
return -1;
strcpy(user_log,data.buf); //将账号返给上级函数
//修改状态
bzero(sql,sizeof(sql));
sprintf(sql, "update users set state=\"%d\" where user==\"%s\";",newfd,buf);
if(0 != sqlite3_exec(info.db, sql, NULL, NULL, &errmsg))
{
printf("__%d__ %s\n",__LINE__,errmsg);
return -1;
}
return 0;
}else
{
fprintf(stderr,"密码错误\n");
continue;
}
}
}
strcpy(user_log,data.buf); //将账号返给上级函数
//发送输入密码的消息
if(send_ask(newfd, user_pass) < 0)
return -1;
//接收密码信息
if(recv_ask(newfd, word) < 0)
return -1;
//登录成功
puts("登录成功");
if(send_ask(newfd, "5") < 0)
return -1;
//修改密码
bzero(sql,sizeof(sql));
sprintf(sql, "update users set password=\"%s\" where user==\"%s\";",word,buf);
if(0 != sqlite3_exec(info.db, sql, NULL, NULL, &errmsg))
{
printf("__%d__ %s\n",__LINE__,errmsg);
return -1;
}
//修改状态
bzero(sql,sizeof(sql));
sprintf(sql, "update users set state=\"%d\" where user==\"%s\";",newfd,buf);
if(0 != sqlite3_exec(info.db, sql, NULL, NULL, &errmsg))
{
printf("__%d__ %s\n",__LINE__,errmsg);
return -1;
}
return 0;
}
}
//查询单词意思的回调函数
int serch_callback(void *text, int ncolumn, char **column_text, char **column_name)
{
strcat((char*)text, column_text[1]); //将意思连接在texta
strcat((char*)text, "。"); //将意思连接在texta
return 0;
}
//获取当前时间
int get_time(char *tm)
{
time_t result;
time(&result);
strcpy(tm,asctime(localtime(&result)));
return 0;
}
//接收 ”/q“ ——>> 退出字典
//协议第一个字符
// 6——>>输入查询的单词
// 7——>>查询失败
// 8——>>查询成功
//查询单词 //结构体 用户表名 查询次数
int serch_word(struct cli_msg info, char *user, int num)
{
char buf[128] = ""; //接收单词
char means[128] = ""; //接收回调函数传参
char sign[128] = "6";
char *errmsg = NULL;
//发送查询单词的消息
if(send_ask(info.newfd, sign) < 0)
return -1;
//接收单词信息
if(recv_ask(info.newfd, buf) < 0)
return -1;
//判断用户是否退出 "/q"
if(strcmp(buf,"/q") == 0)
return -1;
//命令语句查询单词意思
char sql[256] = "";
sprintf(sql,"select * from dict where word == \"%s\";",buf);
if(0 != sqlite3_exec(info.db, sql, serch_callback, (void *)means, &errmsg))
{
fprintf(stderr,"查询失败\n");
if(send_ask(info.newfd, "7") < 0)
return -1;
return 0;
}
//查询失败
if(strlen(means) == 0)
{
if(send_ask(info.newfd, "7") < 0)
return -1;
return 0;
}
//获取当前时间
char time[128] = "";
get_time(time);
//将记录插入表格
bzero(sql,sizeof(sql));
sprintf(sql,"insert into \"%s\" values (%d, \"%s\", \"%s\", \"%s\");"\
,user ,num, buf, means, time);
if(0 != create_table(info.db, sql) )
{
fprintf(stderr,"__%d__ insert user's table false\n",__LINE__);
return -1;
}
char s_mean[128] = "8";
//将标识码添加在means中发送给客户端
strcat(s_mean, means);
strcpy(means,s_mean);
//发送查询到的单词意思
if(send_ask(info.newfd, means) < 0)
return -1;
//防止下个请求发送过快
usleep(50);
return 0;
}
//子线程与客户端交互
void *deal_cli_msg(void *arg)
{
//将结构体内信息取出另存
int newfd = (*(struct cli_msg*)arg).newfd;
struct sockaddr_in cin = (*(struct cli_msg*)arg).cin;
struct cli_msg info = *(struct cli_msg*)arg;
char buf[128] = "";
char text[128] = ""; //接收账号
ssize_t res = 0;
char *errmsg = NULL;
//用户登录
if(log_in(info,text) < 0) //text接收用户名
return NULL; //登录失败则退出
//用户登录成功
//创建用户记录表格
char sql[128] = "";
sprintf(sql, "create table if not exists \"%s\"(num int primary key, text char, mean char, time char);",text);
if(0 != create_table(info.db, sql) )
{
fprintf(stderr,"__%d__ create user's table false\n",__LINE__);
return NULL;
}
int i = 1; //记录单词查询个数
while(i++)
{
if(serch_word(info,text,i) < 0)
break;
}
//成功退出字典
//删除用户记录
bzero(sql,sizeof(sql));
sprintf(sql,"drop table \"%s\";", text);
if(0 != delete_table(info.db, sql) )
{
fprintf(stderr,"__%d__ delete user's table false\n",__LINE__);
return NULL;
}
//修改状态
bzero(sql,sizeof(sql));
sprintf(sql, "update users set state=\"%d\" where user==\"%s\";",-1,text);
if(0 != sqlite3_exec(info.db, sql, NULL, NULL, &errmsg))
{
printf("__%d__ %s\n",__LINE__,errmsg);
return NULL;
}
}
int main(int argc, const char *argv[])
{
//打开字典数据库
char path_db[128] ="./dict.db";
sqlite3 *dict_db = open_database(dict_db, path_db);
if(NULL == dict_db)
{
fprintf(stderr,"__%d__ open dict_db false\n",__LINE__);
return -1;
}
//创建用户信息表格
char sql[128] = "create table if not exists users(user char primary key, password char, state int);";
if(0 != create_table(dict_db, sql) )
{
fprintf(stderr,"__%d__ create user's table false\n",__LINE__);
return -1;
}
//创建流式套接字
int sfd = socket(AF_INET, SOCK_STREAM, 0);
if(sfd < 0)
{
ERR_MSG("socket");
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(8080); //端口号
sin.sin_addr.s_addr = inet_addr(IP); //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; //存储连接成功的客户端地址
int newfd = -1;
pthread_t tid;
struct cli_msg info;
int addrlen = sizeof(cin);
//循环从从已完成连接的队列头获取客户端信息
while(1)
{
newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
if(newfd < 0)
{
ERR_MSG("accept");
return -1;
}
info.newfd = newfd;
info.cin = cin;
info.db = dict_db;
//创建分支线程与客户端交互
if(pthread_create(&tid, NULL, deal_cli_msg, (void *)&info) != 0)
{
fprintf(stderr,"line:__%d__ pthread_create false\n",__LINE__);
return -1;
}
//分离线程
pthread_detach(tid);
}
//关闭字典数据库
if(0 != sqlite3_close(dict_db))
{
printf("__%d__ %s\n",__LINE__,sqlite3_errmsg(dict_db));
return -1;
}
//关闭文件描述符
close(sfd);
return 0;
}
电子词典客户端:
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#define ERR_MSG(msg) do{\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.72.128" //服务器IP
//接收服务器消息
int recv_ser(int cfd, char *buf, int len)
{
int res = recv(cfd, buf, len, 0);
if(res <= 0) //res == 0代表服务器下线
{
ERR_MSG("recv");
return -1;
}
return 0;
}
//发送消息
int send_ser(int cfd,const char * buf)
{
if(send(cfd, buf, sizeof(buf), 0) < 0)
{
ERR_MSG("send_ser");
return -1;
}
return 0;
}
//获取单词翻译
int putout(char *mean)
{
char *sp = mean+1;
printf("翻译:%s\n",sp);
return 0;
}
int main(int argc, const char *argv[])
{
//创建流式套接字
int cfd = socket(AF_INET, SOCK_STREAM, 0);
if(cfd < 0)
{
ERR_MSG("socket");
return -1;
}
//非必须绑定
//填充服务器地址信息
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(8080); //端口号
sin.sin_addr.s_addr = inet_addr(IP); //IP地址
//连接服务器
if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("connect");
return -1;
}
//接收协议
char ask[128] = "";
//发送消息
char buf[128] = "";
//协议第一个字符
// 1——>>输入账号
// 2——>>输入密码
// 3——>>用户在线
// 4——>>密码错误
// 5——>>登录成功
//接收 ”/q“ ——>> 退出字典
//协议第一个字符
// 6——>>输入查询的单词
// 7——>>查询失败
// 8——>>查询成功
//循环与服务器交互
while(1)
{
bzero(ask,sizeof(ask));
bzero(buf,sizeof(buf));
//接收服务器消息
if(recv_ser(cfd, ask, sizeof(ask)) < 0)
return -1;
switch (ask[0])
{
case '1':
printf("请输入账号:\n");
read(0,buf,sizeof(buf));
buf[strlen(buf)-1] = '\0';
if(send_ser(cfd, buf) < 0)
return -1;
break;
case '2':
printf("请输入密码:\n");
read(0,buf,sizeof(buf));
buf[strlen(buf)-1] = '\0';
if(send_ser(cfd, buf) < 0)
return -1;
break;
case '3':
printf("账号在线!\n");
break;
case '4':
printf("密码错误!\n");
break;
case '5':
printf("登录成功!\n");
break;
case '6':
printf("请输入查询的单词:\n");
read(0,buf,sizeof(buf));
buf[strlen(buf)-1] = '\0';
if(send_ser(cfd, buf) < 0)
return -1;
break;
case '7':
printf("未查询到该单词!\n");
break;
case '8':
putout(ask);
break;
}
//判断是否退出while循环
if(strcmp(buf,"/q") == 0)
break;
}
close(cfd);
return 0;
}