磕磕绊绊还是差不多完成了,tcp多客户端在线词典
代码:
数据库导入:有点粗糙,不知道怎么搞成两列,一个单词中间还是空格卧槽难搞
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sqlite3.h>
#define ERR_MSG(msg) do{\fprintf(stderr,"__%d__",__LINE__);\perror(msg);\
}while(0)
int main(int argc, const char *argv[])
{//词典数据库初始化sqlite3 *ppDb;if(sqlite3_open("../ele_directory.db",&ppDb)!=0){printf("__%d__:sqlite3_open error\n",__LINE__);return -1;}char sql[128]="create table if not exists directory (word char);";char *errmsg="";int sql_res=sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg);if(sql_res!=0){printf("__%d__:sqlite3_exec :%s\n",__LINE__,errmsg);return -1;}//导入数据FILE* fp=fopen("../dict.txt","r");if(NULL==fp){ERR_MSG("fopen");return -1;}while(1){char line[64]="";bzero(line,sizeof(line));char* fs_res=fgets(line,sizeof(line),fp);if(NULL==fs_res){puts("文件读取完毕");break;}bzero(sql,sizeof(sql));sprintf(sql,"insert into directory values(\"%s\");",line);sql_res=sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg);printf("%s\n",sql);if(sql_res!=0){printf("__%d__:sqlite3_exec:%s\n",__LINE__,errmsg);return -1;}}puts("数据导入完成");if(sqlite3_close(ppDb)!=0){printf("__%d__:sqlite3_close error\n",__LINE__);return -1;}return 0;
}
服务器:
server.c
/** function: 在线词典服务器TCP多线程* @param [ in] * @param [out] * @return */
#include "head.h"int main(int argc, const char *argv[])
{//TCP服务器创建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;}printf("允许端口快速复用成功\n");struct sockaddr_in addr;addr.sin_family=AF_INET;addr.sin_port=htons(PORT);addr.sin_addr.s_addr=inet_addr(IP);if(bind(sfd,(struct sockaddr*)&addr,sizeof(addr))<0){ERR_MSG("bind");return -1;}printf("绑定成功\n");if(listen(sfd,128)<0){ERR_MSG("listen");return -1;}printf("监听状态\n");struct sockaddr_in cli_addr;socklen_t addrlen=sizeof(cli_addr);pthread_t pth;while(1){printf("客户端:%s:%d:连接前\n",\inet_ntoa(cli_addr.sin_addr),\ntohs(cli_addr.sin_addr.s_addr));int cfd=accept(sfd,(struct sockaddr*)&cli_addr,&addrlen);if(cfd<0){ERR_MSG("accept");return -1;}printf("客户端:%s:%d:连接成功\n",\inet_ntoa(cli_addr.sin_addr),\ntohs(cli_addr.sin_port));//调用线程cli_info client; client.cfd=cfd;client.cli_addr=cli_addr;if(pthread_create(&pth,NULL,fun,(void*)&client)<0){ERR_MSG("pthread_create");return -1;}printf("[%d]线程创建成功\n",cfd);pthread_detach(pth);//分离线程交由系统回收}close(sfd);return 0;
}
fun.c
#include "head.h"void * fun(void*arg){int cfd=((cli_info*)arg)->cfd;struct sockaddr_in cli_addr=((cli_info*)arg)->cli_addr;char buf[128]="";char user[16]="";//打开数据库sqlite3 *ppDb;if(sqlite3_open("../ele_directory.db",&ppDb)!=0){printf("__%d__:sqlite3_open error\n",__LINE__);return NULL;}//第一层//登录判断,(注册判断)1代表登录,2代表注册socklen_t addrlen=sizeof(cli_addr); int recv_res=recv(cfd,&buf,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return NULL;}if(recv_res==0){printf("客户端:%s:%d:断开连接\n",\inet_ntoa(cli_addr.sin_addr),\ntohs(cli_addr.sin_port));return NULL;}printf("11111=%s\n",buf);if(buf[0]=='1'){//登录判断printf("login...\n");int login_res;while(login_res=isLogin(ppDb,cfd,&user)){printf("login_res=%d\n",login_res);char*l_res=NULL;if(1==login_res)l_res="1";else if(2==login_res)l_res="2";else if(3==login_res) l_res="3";elsereturn NULL;printf("l_res=%s\n",l_res);int send_res=send(cfd,l_res,sizeof(recv_res),0);if(send<0){ERR_MSG("send");return NULL;}}int send_res=send(cfd,"0",1,0);if(send<0){ERR_MSG("send");return NULL;}printf("%s登陆成功\n",user);}else if(buf[0]=='2'){//注册判断printf("register...\n");int register_res;while(register_res=isRegister(ppDb,cfd,&user)){printf("register_res=%d",register_res);if(register_res!=1)return NULL;}printf("%s注册成功\n",user);return NULL;}else{printf("__%d__unknown error\n",__LINE__);return NULL;}//第二层//单词查询或者记录查询或退出while(1){char sql[128]="";char* errmsg=NULL;bzero(buf,sizeof(buf));recv_res=recv(cfd,&buf,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return NULL;}if(recv_res==0){printf("客户端:%s:%d:断开连接\n",\inet_ntoa(cli_addr.sin_addr),\ntohs(cli_addr.sin_port));//账号状态设为未登录bzero(sql,sizeof(sql));sprintf(sql,"update account set status=\"1\" where account=\"%s\";",user);if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);return NULL;}}if(buf[0]=='3'){//单词查询printf("word finding...\n");findWord(ppDb,cfd,user); }else if(buf[0]=='4'){//记录查询printf("record finding...\n");findRecord(ppDb,cfd,user);}else if(buf[0]=='5'){printf("客户端:%s:退出登录\n",user);//账号状态设为未登录bzero(sql,sizeof(sql));sprintf(sql,"update account set status=\"1\" where account=\"%s\";",user);if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);return NULL;}return NULL;}else{printf("__%d__unknown error\n",__LINE__);return NULL;}}//关闭数据库if(sqlite3_close(ppDb)!=0){printf("__%d__:sqlite3_close error\n",__LINE__);return NULL;}}
int isLogin(sqlite3 *ppDb,int cfd,char (*user)[]){char account[10]="";char password[10]="";char buf[128]="";//数据库查询,账号是否存在,若存在密码是否正确int recv_res=recv(cfd,&buf,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return -1;}else if(recv_res==0){printf("客户端:%s:断开连接\n",*user);return -1;}printf("buf=%s\n",buf);sscanf(buf,"%s %s",account,password);char sql[128]="";char** pres = NULL;int row, column;char* errmsg = NULL;sprintf(sql,"select * from account where account=\"%s\";",account);if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);return -1;}printf("sql=%s\n",sql);int exist_res=0;//是否存在for (int i=0; i<(row+1)*column; i++){printf("%s\t", pres[i]);if((i+1)%column == 0)putchar('\n');exist_res=1;//存在为1}printf("exist_res=%d\n",exist_res);if(0==exist_res){printf("账号不存在\n"); return 1;//账号不存在返回1}else{printf("账号存在,判断密码\n");bzero(sql,sizeof(sql));sprintf(sql,"select status from account where account=\"%s\" and password=\"%s\";",account,password);if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);return -1;}printf("sql=%s\n",sql);int login_res=0;//是否存在账号密码匹配for (int i=column; i<(row+1)*column; i++){printf("%s\t", pres[i]);if((i+1)%column == 0)putchar('\n');login_res=1;//存在为1}if(0==login_res){printf("密码错误\n");return 2;//账号密码错误返回2}else if(0==strcmp(pres[column],"0")){printf("账号重复登录\n");return 3;}//账号状态设为已登录bzero(sql,sizeof(sql));sprintf(sql,"update account set status=\"0\" where account=\"%s\";",account);if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);return -1;}printf("密码正确\n");strcpy(*user,account);return 0;//登陆成功返回0}
}
int isRegister(sqlite3 *ppDb,int cfd,char(*user)[]){char account[10]="";char password[10]="";char buf[128]="";//数据库查询,账号是否存在bzero(buf,sizeof(buf));int recv_res=recv(cfd,&buf,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return -1;}else if(recv_res==0){printf("客户端:%s:断开连接\n",*user);return -1;}printf("buf=%s\n",buf);sscanf(buf,"%s %s",account,password);char sql[128]="";char** pres = NULL;int row, column;char* errmsg = NULL;sprintf(sql,"select * from account where account=\"%s\";",account);if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);return -1;}printf("sql=%s\n",sql);int reg_res=0;//是否存在for (int i=0; i<(row+1)*column; i++){printf("%s\t", pres[i]);if((i+1)%column == 0)putchar('\n');reg_res=1;//存在为1}printf("reg_res=%d\n",reg_res);if(0==reg_res){//不存在printf("账号不存在,开始注册:\n"); bzero(sql,sizeof(sql));sprintf(sql,"insert into account values (\"%s\",\"%s\",\"1\");",account,password);printf("sql=%s\n",sql);if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);return -1;}printf("注册完成\n");strcpy(*user,account);char *r_res="0";//注册成功printf("r_res=%s\n",r_res); int send_res=send(cfd,r_res,sizeof(recv_res),0);if(send<0){ERR_MSG("send");return -1;}return 0;}else{printf("账号已存在\n");char *r_res="1";printf("r_res=%s\n",r_res); int send_res=send(cfd,r_res,sizeof(recv_res),0);if(send<0){ERR_MSG("send");return -1;}return 1;}}
int findWord(sqlite3 *ppDb,int cfd,char*user){printf("user=%s\n",user);char buf[128]="";char word[32]="";char sql[128]="";char*errmsg=NULL;int recv_res=recv(cfd,&word,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return -1;}if(recv_res==0){printf("客户端:%s:断开连接\n",user);//账号状态设为未登录sprintf(sql,"update account set status=\"1\" where account=\"%s\";",user);if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);return -1;}}printf("word=%s\n",word);char** pres = NULL;int row, column;bzero(sql,sizeof(sql));sprintf(sql,"select * from directory where word like \"%s%\";",word);if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);return -1;}printf("sql=%s\n",sql);for (int i=column; i<(row+1)*column; i++){bzero(buf,sizeof(buf));printf("%s\t", pres[i]);strcpy(buf,pres[i]);strcat(buf," ");if((i+1)%column == 0)strcat(buf,"\n");printf("buf=%s\n",buf);int send_res=send(cfd,buf,sizeof(buf),0);if(send<0){ERR_MSG("send");return -1;}if((i+1)%column == 0)putchar('\n');}int send_res=send(cfd,"find over",strlen("find over"),0);if(send<0){ERR_MSG("send");return -1;}printf("查找完毕\n");//记录(word,time,user)bzero(sql,sizeof(sql));time_t t=time(NULL);struct tm *timeinfo=localtime(&t);char r_time[64]="";sprintf(r_time,"%d-%d-%d:%d:%d:%d",\timeinfo->tm_year+1900,timeinfo->tm_mon+1,timeinfo->tm_mday,\timeinfo->tm_hour,timeinfo->tm_min,timeinfo->tm_sec);printf("word:%s;time:%s;user:%s\n",word,r_time,user);printf("sql=%s\n",sql);sprintf(sql,"insert into record values(\"%s\",\"%s\",\"%s\");",word,r_time,user);printf("sql=%s\n",sql);if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);return -1;}printf("记录完成\n");return 0;
}
int findRecord(sqlite3 *ppDb,int cfd,char*user){printf("user=%s\n",user);char sql[128]="";char** pres = NULL;int row, column;char* errmsg = NULL;char buf[128]="";sprintf(sql,"select * from record where user=\"%s\";",user);if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);return -1;}printf("sql=%s\n",sql);for (int i=column; i<(row+1)*column; i++){bzero(buf,sizeof(buf));printf("%s;%ld\t", pres[i],strlen(pres[i]));strcpy(buf,pres[i]);strcat(buf," ");if((i+1)%column == 0)strcat(buf,"\n");int send_res=send(cfd,buf,sizeof(buf),0);if(send<0){ERR_MSG("send");return -1;}if((i+1)%column == 0)putchar('\n');}printf("查找完毕\n");int send_res=send(cfd,"find record over",strlen("find record over"),0);if(send<0){ERR_MSG("send");return -1;}printf("记录查找完毕\n");return 0;
}
head.h
#ifndef __HEAD_H__
#define __HEAD_H__#define ERR_MSG(msg) do{\fprintf(stderr,"__%d__",__LINE__);\perror(msg);\
}while(0)
#define PORT 8888
#define IP "192.168.1.6"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <unistd.h>
#include <sqlite3.h>
#include <time.h>typedef struct clientInfo{int cfd;struct sockaddr_in cli_addr;
}cli_info;void * fun(void*arg);
int isLogin(sqlite3 *ppDb,int cfd,char(*user)[]);
int isRegister(sqlite3 *ppDb,int cfd,char(*user)[]);
int findWord(sqlite3 *ppDb,int cfd,char*user);
int findRecord(sqlite3 *ppDb,int cfd,char*user);
#endif
客户端:
client.c
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>//自定义报错提示
#define ERR_MSG(msg) do{\fprintf(stderr,"__%d__",__LINE__);\perror(msg);\
}while(0)
#define SER_PORT 8888
#define SER_IP "192.168.1.6"
/** function: TCP客户端* @param [ in] * @param [out] * @return */
int login(int cfd);
int registe(int cfd);
int findWord(int cfd);
int findRecord(int cfd);int main(int argc, const char *argv[])
{//1.创建socket套接字,int cfd=socket(AF_INET,SOCK_STREAM,0);if(cfd<0){ERR_MSG("socket");return -1;}puts("socket create");//允许端口快速复用int reuse = 1;if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0){ERR_MSG("setsockopt");return -1;}printf("允许端口快速复用成功\n");//2.连接服务端connectstruct sockaddr_in addr;addr.sin_family=AF_INET;addr.sin_port=htons(SER_PORT);addr.sin_addr.s_addr=inet_addr(SER_IP);if(connect(cfd,(struct sockaddr*)&addr,sizeof(addr))<0){ERR_MSG("connect");return -1;}puts("connect success");printf("-------------------------------------------------\n");printf("-------Wecome to eletronic directory system------\n");printf("-------------------------------------------------\n");printf("--------------1.登录-----------------------------\n");printf("--------------2.注册-----------------------------\n");printf("--------------3.退出-----------------------------\n");printf("-------------------------------------------------\n");printf("-------------------------------------------------\n");printf("-------------------------------------------------\n");char choice=0;while(1){printf("请选择1.登录;2注册;3.退出>>>");choice=getchar();while(getchar()!='\n');switch(choice){case '1'://登录if(5==login(cfd))return 0;break;case '2'://注册if(0==registe(cfd))return 0;case '3'://退出return 0;default :printf("错误输入,请重输\n");break; }}close(cfd);return 0;
}
int login(int cfd){char buf[32]="";char status=0;int send_res=send(cfd,"1",1,0);if(send_res<0){ERR_MSG("send");return -1;}while(1){printf("请输入账号 密码 (空格分开)>>>");fgets(buf,sizeof(buf),stdin);int send_res=send(cfd,buf,sizeof(buf),0);if(send_res<0){ERR_MSG("send");return -1;}int recv_res=recv(cfd,&status,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return -1;}else if(recv_res==0){printf("socket peer has shutdown\n");return -2;}printf("status=%c\n",status);if('1'==status){printf("账号不存在\n");}else if('2'==status){printf("密码错误\n");}else if('3'==status){printf("账号已登录\n");}else{printf("登陆成功\n");break;}}printf("-----------------------------------\n");printf("-------请输入你的操作--------------\n");printf("---------3.查找单词----------------\n");printf("---------4.记录查询----------------\n");printf("---------5.退出--------------------\n");printf("-----------------------------------\n");char choice=0;while(1){printf("请选择>>>3.查找单词4.记录查询5.退出");choice=getchar();while(getchar()!='\n');switch(choice){case '3'://查找单词findWord(cfd);break;case '4'://查询记录findRecord(cfd);break;case '5'://退出send_res=send(cfd,"5",1,0);if(send_res<0){ERR_MSG("send");return -1;}return 5;default :printf("错误输入,请重输\n");break; }}return 0;
}
int registe(int cfd){char buf[32]="";char status=0;int send_res=send(cfd,"2",1,0);if(send_res<0){ERR_MSG("send");return -1;}while(1){bzero(buf,sizeof(buf));printf("请输入账号 密码 (空格分开)>>>");fgets(buf,sizeof(buf),stdin);printf("%s",buf);int send_res=send(cfd,buf,sizeof(buf),0);if(send_res<0){ERR_MSG("send");return -1;}int recv_res=recv(cfd,&status,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return -1;}else if(recv_res==0){printf("socket peer has shutdown\n");return -2;}printf("status=%c\n",status);if(status=='0'){printf("注册成功,请重新进入系统\n");return 0;}else if(status=='1'){printf("账号已存在\n");}}
}
int findWord(int cfd){char buf[128]="";int send_res=send(cfd,"3",1,0);if(send_res<0){ERR_MSG("send");return -1;}printf("请输入你要查询的单词(支持模糊查询)>>>");fgets(buf,sizeof(buf),stdin);printf("%s",buf);buf[strlen(buf)-1] = '\0';send_res=send(cfd,buf,sizeof(buf),0);if(send_res<0){ERR_MSG("send");return -1;}while(1){bzero(buf,sizeof(buf));int recv_res=recv(cfd,buf,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return -1;}else if(recv_res==0){printf("socket peer has shutdown\n");return -2;}if(0==strcmp(buf,"find over"))break;printf("%s\n",buf);}putchar(10);printf("查找单词完毕\n");return 0;
}
int findRecord(int cfd){char buf[128]="";int send_res=send(cfd,"4",1,0);if(send_res<0){ERR_MSG("send");return -1;}while(1){bzero(buf,sizeof(buf));int recv_res=recv(cfd,buf,sizeof(buf),0);if(recv_res<0){ERR_MSG("recv");return -1;}else if(recv_res==0){printf("socket peer has shutdown\n");return -2;}if(0==strcmp(buf,"find record over"))break;printf("%s",buf);}putchar(10);printf("查找记录完毕\n");return 0;
}
又是依托屎山代码,好乱,不想去整理了