linux 下用c语言编写的聊天室程序,服务器和客户端

/** server.c** Created on: 2012-6-15* Author: root*/#include <stdio.h>#include <stdlib.h>#include <pthread.h>#include <string.h>#include <unistd.h>#include <sys/socket.h>#include <sys/types.h>#include <error.h>#include<netinet/in.h>#define PORT 7999#define MAX_NUM 3 //client连接最大个数#define MAX_CLIENT 15#define MAX_SIZE 1024pthread_rwlock_t idx_lock, wait_lock;//client 信息typedef struct _client {int sockfd;char name[20];pthread_t pid;int flg;} c_client;c_client client[MAX_CLIENT];//定义client;//等待的clientstruct _client_ {int sockfd;char name[20];pthread_t pid;struct _client_ *next;};typedef struct _client_ c_client_c;c_client_c *head = NULL;c_client_c *temp_c1 = NULL, *temp_c2 = NULL;//等待的//初始化client信息void init_client() {int i = 0;for (i = 0; i < MAX_CLIENT; i++) {client[i].sockfd = -1;memset(client[i].name, 0, 20);client[i].pid = -1;client[i].flg = -1;}}//查找结构体数组中sockfd为-1的下标值int find_fd(c_client *client) {int i = 0;while (i < MAX_NUM) {// printf("====%d\n",client[i].sockfd);if (client[i].sockfd == -1)return i;i++;}return -1;}//判断登录格式int logform(char *buf) {char *p = strstr(buf, "LOGIN\r\n");int n = strlen(buf);char *q = p + n - 4;if (p != NULL && p + 7 != q && strcmp(q, "\r\n\r\n") == 0)return 1;elsereturn 0;}int cmpname(char *buf, c_client *p_client) {int i = 0;char *p = strtok(buf + 7, "\r\n\r\n");while (client[i].sockfd != -1 && client[i].sockfd != p_client->sockfd && i < MAX_NUM) {if (strcmp(client[i].name, p) == 0)return 0;i++;}return 1;}//SHOWvoid showuser(c_client *p_client) {int i = 0;char buf[1024] = { 0 };strcpy(buf, "200\r\n");for (i = 0; i < MAX_NUM; i++) {if (client[i].sockfd != -1) {sprintf(buf + strlen(buf), "%s\r\n", client[i].name);}}sprintf(buf + strlen(buf), "\r\n");send(p_client->sockfd, buf, strlen(buf), 0);}//ALLvoid sendto_all(c_client *p_client, char *buf) {int i = 0;char sendbuf[1024] = { 0 };sprintf(sendbuf, "AFROM\r\n%s\r\n%s", p_client->name, buf + 5);for (i = 0; i < MAX_NUM; i++) {if (client[i].sockfd != -1 && client[i].flg != -1)if(send(client[i].sockfd, sendbuf, strlen(sendbuf), 0) <= 0){printf("send errrrrr\n");exit(1);}}}int findname(char *name) {int i = 0;for (i = 0; i < MAX_NUM; i++) {if (client[i].sockfd != -1 && strcmp(client[i].name, name) == 0) return client[i].sockfd;}return 0;}//TOvoid sendto_one(c_client *p_client, char *buf) {int i = 0;char sendbuf[1024] = { 0 };char name[20] = { 0 };char *p = strtok(buf + 4, "\r\n");//TO\r\n：4个字符后取出\r\n前的名字strcpy(name, p);int sock = findname(name);if (!sock) {sprintf(sendbuf, "ERROR2\r\n%s用户不存在\r\n\r\n", name);send(p_client->sockfd, sendbuf, strlen(sendbuf), 0);} else {sprintf(sendbuf, "FROM\r\n%s\r\n%s", p_client->name, buf + 4 + strlen( name) + 2);if(send(sock, sendbuf, strlen(sendbuf), 0)<=0){printf("send errrrrr\n");exit(1);}}}void pthread_fun(void* cclient);//quitvoid quit(c_client *p_client){int i=0;int idx;char buf[1024] = {0};c_client_c *temp;printf("--%s退出聊天室\n",p_client->name);close(p_client->sockfd);p_client->sockfd = -1;p_client->pid = -1;p_client->flg = -1;sprintf(buf,"NOTICE1\r\n%s退出聊天室\r\n\r\n",p_client->name);memset(p_client->name,0,20);for(i=0;i<MAX_NUM;i++){if(client[i].sockfd != -1 && client[i].flg != -1)send(client[i].sockfd,buf,strlen(buf),0);}if(head != NULL && head->next != NULL){memset(buf,0,1024);pthread_rwlock_rdlock(&idx_lock);idx = find_fd(client);pthread_rwlock_unlock(&idx_lock);client[idx].sockfd = head->next->sockfd;pthread_rwlock_wrlock(&wait_lock);temp = head->next;head->next = head->next->next;free(temp);pthread_rwlock_unlock(&wait_lock);sprintf(buf,"NOTICE\r\n您已被唤醒,请继续操作\r\n\r\n");send(client[idx].sockfd,buf,strlen(buf),0);if (pthread_create(&client[idx].pid, NULL, (void *)pthread_fun,(void *) &client[idx]) != 0) {perror("pthread_create");exit(1);}pthread_detach(client[idx].pid);}}void pthread_fun(void* cclient) {c_client *p_client = (c_client *) cclient;char buf[MAX_SIZE] = { 0 };char sendbuf[1024] = { 0 };int i, n;char *p;sprintf(sendbuf, "%s", "NOTICE\r\n通讯通道开启\r\n\r\n");if (send(p_client->sockfd, sendbuf, strlen(sendbuf), 0) <= 0) {printf("send err\n");}memset(sendbuf, 0, 1024);while (1) {memset(buf, 0, MAX_SIZE);n = recv(p_client->sockfd, buf, sizeof(buf) - 1, MSG_NOSIGNAL);if (n <= 0) {close(p_client->sockfd);p_client->sockfd = -1;break;}if (logform(buf)) {if (cmpname(buf, p_client) == 0) {send(p_client->sockfd, "ERROR\r\n用户名重复\r\n\r\n", 26, 0);continue;} else {p_client->flg = 1;p = strtok(buf + 7, "\r\n\r\n");strcpy(p_client->name, p);sprintf(sendbuf, "100\r\n%s\r\n\r\n", p_client->name);send(p_client->sockfd, sendbuf, sizeof(sendbuf), 0);printf("%s进入聊天室\n", p_client->name);for (i = 0; i < MAX_NUM; i++) {if (client[i].sockfd != -1 && client[i].sockfd!= p_client->sockfd && client[i].flg != -1)send(client[i].sockfd, sendbuf, sizeof(sendbuf), 0);}memset(sendbuf, 0, 1024);while (1) {memset(buf, 0, MAX_SIZE);if ((n = recv(p_client->sockfd, buf, MAX_SIZE, 0)) <= 0) {perror("recv err");break;}// printf("recv=%s\n",buf);if ((p = strstr(buf, "\r\n\r\n")) != NULL && *(p + 4)== '\0') {if (!strncmp(buf, "SHOW\r\n\r\n", 8)) {showuser(p_client);//客户端执行show后，发送给客户端已连接的用户continue;}if (!strncmp(buf, "ALL\r\n", 5)) {sendto_all(p_client, buf);continue;}if (!strncmp(buf, "TO\r\n",4)) {sendto_one(p_client, buf);continue;}if (!strncmp(buf, "QUIT\r\n\r\n", 8))quit(p_client);// break;pthread_exit(NULL);} else {send(p_client->sockfd, "ERROR\r\n信息不符合协议要求\r\n\r\n",38, 0);}}}} else {send(p_client->sockfd, "ERROR\r\n未登录,请您登录再进行其他操作\r\n\r\n", 56, 0);}}pthread_exit(NULL);}int main() {int ser_sockfd, clt_sockfd;struct sockaddr_in addr;int idx;char buf[1024] = { 0 };// pthread_rwlock_t idx_lock,wait_lock;pthread_rwlock_init(&idx_lock, NULL);pthread_rwlock_init(&wait_lock, NULL);init_client();//创建服务器sockfdif ((ser_sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {perror("socket");exit(1);}//设置服务器网络地址bzero(&addr, sizeof(addr));addr.sin_family = AF_INET;addr.sin_port = htons(PORT);addr.sin_addr.s_addr = htonl(INADDR_ANY);//设置端口可重用int opt = 1;setsockopt(ser_sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));//将套接字绑定到服务器的网络地址上if (bind(ser_sockfd, (struct sockaddr*)&addr, sizeof(addr)) == -1) { perror("bind");exit(1);}printf("bind success\n");//监听连接请求--监听队列长度为10if (listen(ser_sockfd, 10) == -1) {perror("listen");exit(1);}printf("listen success\n");while (1) {if ((clt_sockfd = accept(ser_sockfd, NULL, NULL)) == -1) {perror("accept");exit(1);}pthread_rwlock_rdlock(&idx_lock);idx = find_fd(client);// printf("idx=%d\n",idx);pthread_rwlock_unlock(&idx_lock);if (idx != -1) { //连接末满client[idx].sockfd = clt_sockfd;if (pthread_create(&client[idx].pid, NULL,(void *) pthread_fun,(void *)&client[idx]) != 0) {perror("pthread_create");exit(1);}pthread_detach(client[idx].pid);} else { //连接已满,等待temp_c1 = (c_client_c *) malloc(sizeof(c_client_c));temp_c1->sockfd = clt_sockfd;temp_c1->next = NULL;pthread_rwlock_wrlock(&wait_lock);if (head == NULL) {head = (c_client_c *) malloc(sizeof(c_client_c));head->next = temp_c1;} else {for (temp_c2 = head; temp_c2->next != NULL; temp_c2= temp_c2->next);temp_c2->next = temp_c1;}pthread_rwlock_unlock(&wait_lock);memset(buf, 0, 1024);sprintf(buf, "NOTICE\r\n服务器已满,请等候\r\n\r\n");//客户端接受则等待if (send(temp_c1->sockfd, buf, strlen(buf), 0) <= 0) {printf("sendr err\n");}}}}/** client.c** Created on: 2012-6-18* Author: root*/#include <stdio.h>#include <stdlib.h>#include <string.h>#include <sys/socket.h>#include <unistd.h>#include <sys/types.h>#include <pthread.h>#include <netinet/in.h>#include <error.h>#include <arpa/inet.h>#include <termios.h>#define MAX_SIZE 1024#define PORT 7999static int FLAGE = -1;char name[20] = {0};void fun_show(int sockfd){char sendbuf[256] = {0};sprintf(sendbuf,"SHOW\r\n\r\n");if(send(sockfd,sendbuf,strlen(sendbuf),0)<=0){ printf("send err\n");close(sockfd);exit(1);}}void fun_all(int sockfd){char sendbuf[MAX_SIZE] = {0};sprintf(sendbuf,"ALL\r\n",5);printf("输入发送的内容:\n");scanf("%s",sendbuf+5);sprintf(sendbuf+strlen(sendbuf),"\r\n\r\n");if(send(sockfd,sendbuf,strlen(sendbuf),0) <= 0){ printf("send err\n");close(sockfd);exit(1);}}void fun_one(int sockfd){char sendbuf[MAX_SIZE] = {0};char name3[20] = {0};printf("输入聊天对象:");scanf("%s",name3);sprintf(sendbuf,"TO\r\n%s\r\n",name3);printf("输入聊天内容:\n");scanf("%s",sendbuf+strlen(sendbuf));sprintf(sendbuf+strlen(sendbuf),"\r\n\r\n");if(send(sockfd,sendbuf,strlen(sendbuf),0) <= 0){printf("send err\n");close(sockfd);exit(1);}}void fun_quit(int sockfd){char sendbuf[256] = "QUIT\r\n\r\n";if(send(sockfd,sendbuf,strlen(sendbuf),0) <= 0){printf("send err\n");close(sockfd);exit(1);}}void *pthread_fun(int *sock){int sockfd = *sock;char recvbuf[1024] = {0};int n = 0;char *p,*q;char name2[20] = {0};while(1){memset(recvbuf,0,1024);n = recv(sockfd,recvbuf,sizeof(recvbuf),0);if(n<=0){printf("recv failed\n");exit(1);}if(!strncmp(recvbuf,"NOTICE\r\n通讯通道开启\r\n\r\n",30)){ printf("通讯通道开启\n");}if (!strncmp(recvbuf, "100\r\n", 5)) {char *p = strtok(recvbuf + 5, "\r\n\r\n");//100\r\n：3个字符后取出\r\n\r\n前的名字strcpy(name2, p);printf("[NOTICE]%s进入聊天室\n", name2);FLAGE = 4;}if (!strncmp(recvbuf, "ERROR\r\n用户名重复\r\n\r\n", 26)) {printf("用户名重复\n");FLAGE = 3;}if (!strncmp(recvbuf, "200\r\n", 5)) {p = strtok(recvbuf+5,"\r\n\r\n");while(p != NULL){printf("%s\n",p);p = strtok(NULL,"\r\n\r\n");}printf("please input con:\n");}if(!strncmp(recvbuf,"AFROM\r\n",7)){// printf("recvbuf=%s\n",recvbuf);p = strtok(recvbuf+6,"\r\n");q = strtok(NULL,"\r\n\r\n");printf("(%s)[群聊]:%s\n",p, q);printf("please input con:\n");}if(!strncmp(recvbuf,"FROM\r\n",6)){p = strtok(recvbuf+6,"\r\n");q = strtok(NULL,"\r\n\r\n");printf("(%s)[私聊](%s):%s\n",p,name,q);printf("please input con:\n");}if(!strncmp(recvbuf,"ERROR2\r\n",8)){p = strtok(recvbuf + 8, "\r\n");printf("%s\n", p);}if(!strncmp(recvbuf,"NOTICE1\r\n",9)){p = strtok(recvbuf + 9, "\r\n");printf("用户%s\n", p);}if(!strncmp(recvbuf,"NOTICE\r\n服务器已满,请等候\r\n\r\n",37)){printf("服务器已满,请等候\n");}if(!strncmp(recvbuf,"NOTICE\r\n您已被唤醒,请继续操作\r\n\r\n",45)){ printf("你已经被唤醒,请继续操作\n");FLAGE = 2;}}}int main(int argc,char **argv){if(argc != 2){printf("input server ip:\n");exit(1);}pthread_t pid;int sockfd;struct sockaddr_in addr;char recvbuf[1024] = {0};char sendbuf[1024] = {0};int k=0;int n;char str[6] = {0};char *p,*q;if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1){perror("socket");exit(1);}bzero(&addr,sizeof(addr));addr.sin_family = AF_INET;addr.sin_port = htons(PORT);if(inet_pton(AF_INET,argv[1],(void *)&addr.sin_addr) <= 0){perror("inet_pton");exit(1);}// inet_aton(argv[1], &addr.sin_addr);if(connect(sockfd,(struct sockaddr *)&addr,sizeof(addr)) == -1){ perror("connect");exit(1);}printf("connect success\n");// pid = fork();pthread_create(&pid,NULL,(void *)pthread_fun,(void*)&sockfd);pthread_detach(pid);// if(pid > 0){usleep(100);//让子进程先运行while(FLAGE==0){printf("服务器已满,wait...\n");sleep(1);}tcflush(0,TCIFLUSH);while(FLAGE==2){ //当FLAGE＝2时正好父进程运行，则等待,FLAGE=1 usleep(100);}if(FLAGE == 1){while(k<3){//fflush(stdin);memset(name,0,20);printf("login:");scanf("%s",name);//fflush(stdin);sprintf(sendbuf,"LOGIN\r\n%s\r\n\r\n",name);send(sockfd,sendbuf,strlen(sendbuf),0);memset(sendbuf,0,1024);for(;;){if(FLAGE==3 || FLAGE==4)break;}if(FLAGE==3){k++;if(k==3){printf("输入过多,退出\n");}else{printf("还有%d次机会登录\n", 3 - k);}FLAGE=1;continue;}if(FLAGE==4){while(1){memset(str,0,6);scanf("%s",str);if(!strcmp(str,"show")){printf("显示在线用户:\n");fun_show(sockfd);continue;}else if(!strcmp(str,"all")){printf("群聊模式");fun_all(sockfd);continue;}else if(!strcmp(str,"to")){printf("私聊模式");fun_one(sockfd);continue;}else if(!strcmp(str,"quit")){printf("退出聊天室\n");fun_quit(sockfd);close(sockfd);return;}else if(!strcmp(str,"help")){printf("all[聊天内容] 群聊\n");printf("to[name][聊天内容] 私聊\n");printf("quit 退出程序\n");printf("help 显示帮助信息\n");printf("please input con:\n");continue;}else{printf("请输入help查看指令:\n");continue;}}}}}// }wait(NULL);}。