即时通讯软件

客户端：

#include <stdio.h>

服务端：

#include <stdio.h>#include <errno.h>#include <stdlib.h>#include <unistd.h>#include <string.h>#include <netdb.h>#include <sys/types.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#include <ctype.h>#include <pthread.h>#include <sqlite3.h>//定义端口#define PORT 9999//登录成功标志变量地址int *adress;//客户机结构体                            struct client_msg{int cmd_c; //功能选项 1.注册 2.登录。3.退出int id_c;int cmd2_c; //功能选项 1.客户机查看在线人数。2.私聊。3.群聊。4.注销。5.超级用户踢人。6.禁言。7.解禁。8.文件传输。99.客户机强退。char name_c[20];char psw[10];char tak_c[100];};//服务器结构体                            struct server_msg{int cmd_s;  //0表示注册成功 1表示登录成功 2表示登录失败 3表示私聊信息发送成功 4表示私聊信息发送失败int id_s;                  char fname[20];char tak_s[100];};//链表结构体typedef struct Node{int id_l;int fd_l; struct Node *next;}linkedlist;linkedlist *list;//链表函数linkedlist *createlinkedlist();int getcount(linkedlist *list);//void insert1(linkedlist *list, int id, int fd);linkedlist *insert2(linkedlist *list, int id, int fd);int queryfd(linkedlist *list, int id);int queryid(linkedlist *list, int fd);void print(linkedlist *list);linkedlist *dellinkedlist(linkedlist *list, int fd);//线程处理函数static void * handle_client(void * arg);//回调函数int select_callback(void *date, int column_count,char* columnValue[],char* columnName[]);int main(){//用来创建socketint socket_fd;socket_fd = socket(AF_INET, SOCK_STREAM, 0);if(-1 == socket_fd){perror("socket");exit(EXIT_FAILURE);}printf("socket success/n");//用来绑定struct sockaddr_in myaddr;memset(&myaddr, 0, sizeof(struct sockaddr));myaddr.sin_family = AF_INET;//设置协议myaddr.sin_port = htons(PORT);//端口号myaddr.sin_addr.s_addr = INADDR_ANY;//任意ip都可以通信int ret;ret = bind(socket_fd, (struct sockaddr*)&myaddr, sizeof(struct sockaddr));if(-1 == ret){perror("bind");exit(EXIT_FAILURE);}printf("bind success/n");//监听ret = listen(socket_fd, 6);if(-1 == ret){perror("listen");exit(EXIT_FAILURE);}printf("listen success/n");struct sockaddr_in clientaddr;pthread_t cli_thread;//建立数据库int sqlret;sqlite3 *db;char *ErrMsg;sqlret = sqlite3_open("./chatroom.db", &db);//打开数据库if(sqlret != SQLITE_OK){printf("error : open sqlite/n");}printf("建立数据库成功/n");char *create_str = "create table people (id INTEGER primary key, name TEXT, psw TEXT);";sqlret = sqlite3_exec(db, create_str, 0, 0, &ErrMsg);if(sqlret != SQLITE_OK){   printf("error : create   %s/n", sqlite3_errmsg(db));}printf("建立表成功/n");//关闭数据库ret = sqlite3_close(db);if(ret != SQLITE_OK){   printf("error : close/n");}    char snd_buf[1024];//创建链表list = createlinkedlist();while(1){int size = sizeof(struct sockaddr_in);//此处发生第一次阻塞，满足条件：有新的客户端接入   int new_fd;   new_fd = accept(socket_fd, (struct sockaddr*)&clientaddr, &size);   if(-1 == new_fd)   {    perror("accept");   }   printf("accept success/n");//线程处理新接入的客户机ret = pthread_create(&cli_thread, NULL, handle_client, (void *)new_fd);if(-1 == ret){perror("pthread_create");   exit(EXIT_FAILURE);}}return 0;}static void * handle_client(void *arg){int new_fd = (int)arg;char buffer[1024];char snd_buf[1024];int ret;static int num = 520;//用作idwhile(1){//接收信息printf("reading.../n");ret = read(new_fd, buffer, 1024);if(-1 == ret){printf("read fail/n");perror("read");exit(EXIT_FAILURE);}printf("read success/n");//解析信息   struct client_msg fclient; //定义结构体变量memset(&fclient, 0, sizeof(fclient));//清空结构体memcpy(&fclient, buffer, sizeof(fclient));//把接收到的信息转换成结构体//printf("%d/t%s/t%s/t%s/n", fclient.cmd_c,  fclient.name_c, fclient.psw, fclient.tak_c);struct server_msg server1;server1.cmd_s = -1;switch(fclient.cmd_c){case 1:{printf("有新的客户请求注册/n");//*********插入注册信息****************int sqlret;sqlite3 *db;sqlret = sqlite3_open("./chatroom.db", &db);if(sqlret != SQLITE_OK){printf("error : open sqlite/n");}char *insert_str = "insert into people (id, name, psw) values (%d, '%s', '%s');";char *ErrMsg;char str[1024];//插入注册信息到people表sprintf(str, insert_str, num, fclient.name_c, fclient.psw);ret = sqlite3_exec(db, str, 0, 0, &ErrMsg);if(ret != SQLITE_OK){printf("error : insert   %s/n", sqlite3_errmsg(db));exit(EXIT_FAILURE);}ret = sqlite3_close(db);if(ret != SQLITE_OK){printf("error : close/n");}server1.cmd_s = 0;server1.id_s = num;num++;//***********发送服务器结构体*********memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("发送成功，等待下一指令....../n");break;}case 2:{printf("有新的客户端请求登录/n");//printf("%d/t%s/t%s/t%d/n", fclient.cmd_c,  fclient.name_c, fclient.psw, fclient.id_c);//回调函数查找int ret;sqlite3 *db;char *ErrMsg;char str[1024];//打开数据库ret = sqlite3_open("./chatroom.db", &db);if(ret != SQLITE_OK){printf("error : open/n");}//查找表内数据int check = -1;adress = &check;//登录成功标志变量printf("*********查询信息********/n");char *select_str = "select * from people where id = %d and psw = '%s';";sprintf(str, select_str, fclient.id_c, fclient.psw);printf("开始查找/n");ret = sqlite3_exec(db, str, select_callback, 0, &ErrMsg);if(ret != SQLITE_OK)           {               printf("error : select  %s/n", sqlite3_errmsg(db));break;           }printf("check = %d/n", check);//关闭数据库ret = sqlite3_close(db);if(ret != SQLITE_OK){printf("error : close/n");}if(check == 1)//登录成功{//server1.cmd_s = 1;if(520 == fclient.id_c){server1.cmd_s = 10;}else{server1.cmd_s = 1;}memset(snd_buf, 0, 1024);//清空发送缓存   memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("登录成功，发送反馈成功，等待下一指令....../n");//插入信息到链表list = insert2(list, fclient.id_c, new_fd);printf("现在打印链表/n");print(list);printf("在线人数为：%d/n", getcount(list));//reading...while(1){printf("现在等待客户端命令/n");//**************读取信息*******************************************memset(buffer, 0, 1024);ret = read(new_fd, buffer, 1024);if(-1 == ret){printf("read fail/n");perror("read");exit(EXIT_FAILURE);}printf("read success/n");//解析信息struct client_msg fclient; //定义结构体变量memset(&fclient, 0, sizeof(fclient));//清空结构体memcpy(&fclient, buffer, sizeof(fclient));//把接收到的信息转换成结构体//******************读取完成****************************************printf("客户端命令为 ：%d/n", fclient.cmd2_c);switch(fclient.cmd2_c){case 1://客户机请求查看在线人数{printf("客户机请求查看在线人数/n");server1.cmd_s = 9;print(list);server1.id_s = getcount(list);memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("发送反馈成功，等待下一指令....../n");break;}case 2://私聊{printf("私聊ID：%d/t私聊内容：%s/n", fclient.id_c, fclient.tak_c);int send_fd = queryfd(list, fclient.id_c);printf("1send_fd is %d/n", send_fd);if(0 != send_fd){printf("2send_fd is %d/n", send_fd);strcpy(server1.tak_s, fclient.tak_c);//私聊信息存到服务器结构体printf("本身id ：%d/n", queryid(list, new_fd));server1.id_s = queryid(list, new_fd);server1.cmd_s = 5;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(send_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（私聊信息）发送成功/n");server1.cmd_s = 3;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（私聊反馈）发送成功/n");}else{printf("（私聊信息）发送失败/n");server1.cmd_s = 4;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（私聊反馈）发送成功/n");}break;}case 3://群聊{printf("客户机请求群聊/n");server1.id_s = queryid(list, new_fd);printf("发起人%d/t群聊内容：%s/n", server1.id_s, fclient.tak_c);int i;linkedlist *h = list;for(i = 0; i<getcount(list); i++){if(h -> fd_l != new_fd){strcpy(server1.tak_s, fclient.tak_c);//私聊信息存到服务器结构体//printf("本身id ：%d/n", queryid(list, new_fd));server1.cmd_s = 6;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(h -> fd_l, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf(" （私聊信息）发送成功/n");}h = h -> next;}server1.cmd_s = 7;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（私聊反馈）发送成功/n");break;}case 4://注销{printf("用户请求注销下线/n");list = dellinkedlist(list, new_fd);print(list);server1.cmd_s = 8;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（注销反馈）发送成功/n");goto label1;}case 5://踢人{printf("超级用户请求踢人，被踢者ID为%d/n", fclient.id_c);int ti_fd = queryfd(list, fclient.id_c);//list = dellinkedlist(list, ti_fd);//print(list);server1.cmd_s = 11;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(ti_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（踢人指令）发送成功/n");server1.cmd_s = 12;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（踢人反馈）发送成功/n");break;}case 6://禁言{printf("超级用户请求禁言，被禁言者ID为%d/n");int jin_fd = queryfd(list, fclient.id_c);server1.cmd_s = 13;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(jin_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（禁言指令）发送成功/n");server1.cmd_s = 14;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（禁言反馈）发送成功/n");break;}case 7://解禁{printf("超级用户请求解除禁言，被解除者ID为%d/n");int jiejin_fd = queryfd(list, fclient.id_c);server1.cmd_s = 15;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(jiejin_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（解禁指令）发送成功/n");server1.cmd_s = 16;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（解禁反馈）发送成功/n");break;}case 8://文件传输{printf("客户机请求发送文件******************/n");printf("发送给%d/t文件名：%s/n", fclient.id_c, fclient.name_c);int id_sdf = queryid(list, new_fd);int fd_rsvf = queryfd(list, fclient.id_c);server1.cmd_s = 17;server1.id_s = id_sdf;strcpy(server1.fname, fclient.name_c);memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(fd_rsvf, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（文件传输指令）发送成功/n");////转发//while(1)printf("文件发送ing/n");char buffer_file[1024] = {0};while(read(new_fd, buffer_file, 1024) != -1)  //失败返回-1{printf("server : buffer_file is %s/n", buffer_file);if(write(fd_rsvf, buffer_file, 1024) == -1)  //失败返回-1{perror("send");exit(-1);}if(0 == strcmp(buffer_file, "END")){printf("文件主体传输完成/n");break;}memset(buffer_file, 0, 1024);//清空缓存}perror("recv");//exit(-1);server1.cmd_s = 18;memset(snd_buf, 0, 1024);//清空发送缓存memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("（文件传输反馈）发送成功/n");break;}case 99://客户机强行退出{printf("客户机强行退出/n");list = dellinkedlist(list, new_fd);print(list);goto label1;}}}}else//登录失败{server1.cmd_s = 2;memset(snd_buf, 0, 1024);//清空发送缓存   memcpy(snd_buf, &server1, sizeof(server1)); //结构体转换成字符串ret = write(new_fd, snd_buf, 1024);//发送信息if(-1 == ret){perror("write");exit(EXIT_FAILURE);}printf("登录失败，发送反馈成功，等待下一指令....../n");break;}break;}case 3:{printf("客户机退出程序/n");goto clientout;}case 99:{printf("客户机强行退出/n");goto clientout;}}label1:printf("**************/n");}clientout:close(new_fd);}linkedlist *createlinkedlist(){linkedlist *h;h = NULL;return h;}linkedlist *dellinkedlist(linkedlist *list, int fd){linkedlist *p = list -> next;linkedlist *h = list;if(list -> fd_l == fd){free(h);return list -> next;}else{while(p -> fd_l != fd){p = p -> next;h = h -> next;}h -> next = p -> next;free(p);return list;}}linkedlist *insert2(linkedlist *list, int id, int fd){linkedlist *head, *p;head = list;p = (linkedlist *)malloc(sizeof(linkedlist));if(head == NULL){head = p;p -> next = NULL;}else{p -> next = head;head = p;}p -> id_l = id;p -> fd_l = fd;return head;}int getcount(linkedlist *list){linkedlist *h = list;int count = 0;while(h != NULL){count++;h = h -> next;}return count;}int queryid(linkedlist *list, int fd){linkedlist *p, *h;h = list;while(h != NULL){if(fd == h -> fd_l){return h -> id_l;}h = h -> next;}}int queryfd(linkedlist *list, int id){linkedlist *h = list;while(h != NULL){if(id == h -> id_l){return h -> fd_l;}h = h -> next;}return 0;}void print(linkedlist *list){linkedlist *h = list;while(h != NULL){printf("list id is %d/tfd is %d/n",h ->id_l, h -> fd_l);        h = h -> next;}printf("/n");}int select_callback(void *date, int column_count,char* columnValue[],char* columnName[])//回调函数查找{if(column_count != 0){*adress = 1;}else{*adress = 2;}    return 0; }//回调函数查找