Linux聊天室 -- select && 多线程实现

/******************************************************************************* * 服务器处理用户聊天操作实现文件 * 2015-12-16 yrr实现 * ********************************************************************************/ #include "config.h" extern ListNode *userList; /************************************************** 函数名：groupChat 功能：群聊函数实现 参数：msg--用户发送的群聊消息 sockfd -- 发送者套接字 返回值：成功登陆返回SUCCESS 否则返回异常类型 ****************************************************/ int groupChat(Message *msg , int sockfd) { ListNode *p; int ret; /*声明数据库变量*/ sqlite3 *db; sqlite3_stmt *stmt; const char *tail; /*声明sql语句存储变量*/ char sql[128]; /*消息发送缓冲区*/ char buf[MAX_LINE]; /*消息内容*/ Message message; memset(&message , 0 , sizeof(message)); strcpy(message.sendName , (*msg).sendName); strcpy(message.recvName , (*msg).recvName); message.msgType = (*msg).msgType; /*查看在线用户*/ p = userList; /*除了自己无人在线*/ if(p->next == NULL) { /*改变消息类型为RESULT*/ message.msgType = RESULT; strcpy(message.content, stateMsg(ALL_NOT_ONLINE)); memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send(sockfd , buf , sizeof(buf) , 0); return ALL_NOT_ONLINE; }//if /*向所有在线用户发送消息*/ else { strcpy(message.recvName , ""); strcpy(message.content , (*msg).content); strcpy(message.msgTime , (*msg).msgTime); while(p!=NULL) { if(strcmp((p->user).userName , message.sendName) != 0) { memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send((p->user).sockfd , buf , sizeof(buf) , 0); }//else p = p->next; }//while /*（1）打开数据库*/ ret = sqlite3_open(DB_NAME, &db); if(ret != SQLITE_OK) { printf("unable open database!\n"); return FAILED; }//if /*（2）执行插入操作*/ memset(sql , 0 , sizeof(sql)); sprintf(sql , "insert into Message(msgType , sendName , recvName , content , msgTime)\ values(%d,'%s','%s','%s', '%s');",message.msgType , message.sendName , message.recvName,message.content , message.msgTime); ret = sqlite3_prepare(db , sql , strlen(sql) , &stmt , &tail); if(ret != SQLITE_OK) { ret = sqlite3_step(stmt); sqlite3_finalize(stmt); sqlite3_close(db); return FAILED; }//if /*（3）顺利插入*/ ret = sqlite3_step(stmt); sqlite3_finalize(stmt); sqlite3_close(db); /*群聊处理成功*/ return SUCCESS; }//else } /************************************************** 函数名：personalChat 功能：私聊函数实现 参数：msg--用户发送的群聊消息 sockfd -- 发送者套接字 返回值：成功登陆返回SUCCESS 否则返回异常类型 ****************************************************/ int personalChat(Message *msg , int sockfd) { ListNode *p; int ret; /*声明数据库变量*/ sqlite3 *db; sqlite3_stmt *stmt; const char *tail; /*声明sql语句存储变量*/ char sql[128]; /*消息发送缓冲区*/ char buf[MAX_LINE]; /*消息内容*/ Message message; memset(&message , 0 , sizeof(message)); strcpy(message.sendName , (*msg).sendName); strcpy(message.recvName , (*msg).recvName); message.msgType = (*msg).msgType; /*消息发送对象和接收对象相同*/ if(strcmp((*msg).sendName , (*msg).recvName) == 0) { printf("消息不能发送到自己！\n"); /*改变消息类型为RESULT*/ message.msgType = RESULT; strcpy(message.content, stateMsg(MESSAGE_SELF)); memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send(sockfd , buf , sizeof(buf) , 0); return MESSAGE_SELF; }//if /*查找接收信息用户*/ p = userList; while(p != NULL && strcmp((p->user).userName , (*msg).recvName) != 0) { p = p->next; }//while if(p == NULL) { printf("该用户不在线！\n"); /*改变消息类型为RESULT*/ message.msgType = RESULT; strcpy(message.content, stateMsg(ID_NOT_ONLINE)); memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send(sockfd , buf , sizeof(buf) , 0); return ID_NOT_ONLINE; }//if else{ strcpy(message.content , (*msg).content); strcpy(message.msgTime , (*msg).msgTime); memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send((p->user).sockfd , buf , sizeof(buf) , 0); /*写到数据库*/ /*（1）打开数据库*/ ret = sqlite3_open(DB_NAME, &db); if(ret != SQLITE_OK) { printf("unable open database!\n"); return FAILED; }//if /*（2）执行插入操作*/ memset(sql , 0 , sizeof(sql)); sprintf(sql , "insert into Message(msgType , sendName , recvName , content , msgTime)\ values(%d,'%s','%s','%s', '%s');",message.msgType , message.sendName , message.recvName,message.content , message.msgTime); printf("%s\n" , sql); ret = sqlite3_prepare(db , sql , strlen(sql) , &stmt , &tail); if(ret != SQLITE_OK) { ret = sqlite3_step(stmt); sqlite3_finalize(stmt); sqlite3_close(db); return FAILED; }//if /*（3）顺利插入*/ ret = sqlite3_step(stmt); sqlite3_finalize(stmt); sqlite3_close(db); /*私聊处理成功*/ return SUCCESS; }//else } /************************************************** 函数名：viewUserList 功能：查看在线用户列表函数实现 参数：msg--用户发送的群聊消息 sockfd -- 发送者套接字 返回值：成功登陆返回SUCCESS 否则返回异常类型 ****************************************************/ int viewUserList(Message *msg , int sockfd) { ListNode *p; int ret; /*消息发送缓冲区*/ char buf[MAX_LINE]; /*消息内容*/ Message message; memset(&message , 0 , sizeof(message)); strcpy(message.sendName , (*msg).sendName); strcpy(message.recvName , (*msg).recvName); message.msgType = (*msg).msgType; /*查看在线用户*/ p = userList; if(p == NULL) { /*改变消息类型为RESULT*/ message.msgType = RESULT; strcpy(message.content, stateMsg(ALL_NOT_ONLINE)); memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send(sockfd , buf , sizeof(buf) , 0); return ALL_NOT_ONLINE; }//if else{ /*否则消息类型不变*/ strcpy(message.content , ""); while(p!=NULL) { strcat(message.content , "\t"); strcat(message.content , (p->user).userName); p = p->next; }//while memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send(sockfd , buf , sizeof(buf) , 0); printf("查看在线列表结果：%s\n", message.content); } return SUCCESS; } /************************************************** 函数名：viewUserList 功能：查看聊天记录 参数：msg--用户发送的群聊消息 sockfd -- 发送者套接字 返回值：成功登陆返回SUCCESS 否则返回异常类型 ****************************************************/ int viewRecords(Message *msg , int sockfd) { int ret; char buf[MAX_LINE] , record[MAX_LINE]; /*声明数据库变量*/ sqlite3 *db; char *errmsg = NULL; char **dbRet; int nRow , nCol , i , j , idx; /*声明sql语句存储变量*/ char sql[128]; /*存储操作结果消息*/ Message message; memset(&message , 0 , sizeof(message)); strcpy(message.sendName , (*msg).sendName); /*判断是否接收群消息*/ if(strcmp( (*msg).recvName , "all") == 0) strcpy(message.recvName , ""); else strcpy(message.recvName , (*msg).recvName); message.msgType = (*msg).msgType; /*（1）打开数据库*/ ret = sqlite3_open(DB_NAME, &db); if(ret != SQLITE_OK) { printf("unable open database.\n"); /*改变消息类型为RESULT*/ message.msgType = RESULT; strcpy(message.content, stateMsg(FAILED)); memset(buf , 0 , MAX_LINE); memcpy(buf , &message , sizeof(message)); send(sockfd , buf , sizeof(buf) , 0); return FAILED; }//if /*（2）读出两者的聊天记录，以二进制方式*/ memset(sql , 0 , sizeof(sql)); if(st