Linuxc聊天室，有日志功能可群聊可私聊

#include<stdio.h> #include<stdlib.h> #include<string.h> #include<sys/socket.h> #include<netinet/in.h> #include<unistd.h> #include<ctype.h> #include<arpa/inet.h> #include<errno.h> #include<time.h> #define LISTENQ 1000 //最大连接数 #define MAXLINE 512 //发送消息的最长字节 #define MAXMEM 1000 #define NAMELEN 20 //名字长度 struct socketcfd { char name[20]; int fd; }; int listenfd; //分别记录服务器端的套接字与连接的多个客户端的套接字 struct socketcfd connfd[MAXMEM]; void rcv_snd(void *arg) //服务器接受并转发消息函数 { char* ask="请输入你的名字："; char* ask1="使用说明：\n1、群聊可以直接发送消息，\n2、私聊请先输入@+人名+空格+内容，\n3、退出直接输入“bye”\n-------------------------------------------"; char* ch="此用户不存在"; char* ch1="群聊"; char* ch2="私聊"; char buff[MAXLINE]; //用户姓名 char buff1[MAXLINE]; //聊天内容（包括私聊和群聊，然后对它进行分析） char buff2[MAXLINE]; //发送消息的时间 char buff3[MAXLINE]; //私聊时接受用户的姓名 char buff4[MAXLINE]; //私聊时发送的内容 time_t ticks; int i=0; int retval; int len; int k=0,j=0,m=0; int p; int n=(int *)arg; //获取此进程对应的套接字用户的名字 write(connfd[n].fd,ask,strlen(ask)); len=read(connfd[n].fd,connfd[n].name,NAMELEN); if(len>0){ connfd[n].name[len]=0; } printf("用户姓名：%s\n",connfd[n].name); //把当前用户的加入告知所有用户 strcpy(buff,connfd[n].name); strcat(buff,"\t加入聊天"); for(i=0;i<MAXMEM;i++){ if(connfd[i].fd!=-1) write(connfd[i].fd,buff,strlen(buff)); } //告诉用户使用规则 write(connfd[n].fd,ask1,strlen(ask1)); //接受当前用户的信息并将其转发给所有用户或者指定用户 while(1){ if((len=read(connfd[n].fd,buff1,MAXLINE))>0) { buff1[len]=0; //当前用户的输入信息为“bye”时，当前用户退出 if(strcmp("bye",buff1)==0){ printf("%s用户已退出\n",connfd[n].name); close(connfd[n].fd); connfd[n].fd=-1; pthread_exit(&retval); } //通过分析发送的信息判断是群聊还是私聊 if(buff1[0]=='@'){ j=0;m=0; for(k=1;k<sizeof(buff1);k++){ if(buff1[k]==' ') p=k; } for(i=1;i<=MAXLINE;i++){ if(i>p){ buff4[m]=buff1[i]; m++; } else if(i<p){ buff3[j]=buff1[i]; j++; } } buff3[j]='\0'; buff4[m]='\0'; ticks=time(NULL); sprintf(buff2,"%.24s\r\n",ctime(&ticks)); strcpy(buff,buff2); strcat(buff,ch2); strcat(buff,"\t"); strcat(buff,connfd[n].name); strcat(buff,"\n"); strcat(buff,buff4); for(i=0;i<20;i++){ if(strcmp(connfd[i].name,buff3)==0){ write(connfd[i].fd,buff,strlen(buff)); } else if(strcmp(connfd[i].name,buff3)!=0&&i>MAXMEM){ write(connfd[n].fd,ch,strlen(ch)); } } }else{ ticks=time(NULL); sprintf(buff2,"%.24s\r\n",ctime(&ticks)); strcpy(buff,buff2); strcat(buff,ch1); strcat(buff,"\t"); strcat(buff,connfd[n].name); strcat(buff,"\n"); strcat(buff,buff1); for(i=0;i<MAXMEM;i++){ if(connfd[i].fd!=-1){ write(connfd[i].fd,buff,strlen(buff)); } } } } } } void quit() //服务器关闭函数 { char msg[10]; while(1){ scanf("%s",msg); if(strcmp("quit",msg)==0){ printf("......服务器退出.....\n"); close(listenfd); exit(0); } } } int main() { pthread_t thread1,thread2; struct sockaddr_in servaddr,cliaddr; socklen_t len; time_t ticks; char buff[MAXLINE]; char addr_p[INET_ADDRSTRLEN]; int n=0; //调用socket函数创建服务器端的套接字 printf("Socket...\n"); listenfd=socket(AF_INET,SOCK_STREAM,0); if(listenfd<0){ printf("Socket created failed.\n"); return -1; } //调用bind函数使得服务器端的套接字与地址实现绑定 printf("Bind...\n"); servaddr.sin_family=AF_INET; servaddr.sin_port=htons(8080); servaddr.sin_addr.s_addr=htonl(INADDR_ANY); if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0){ printf("Bind failed.\n"); return -1; } //调用listen函数，将一个主动连接套接字变为被动的倾听套接字 //在此过程中完成tcp的三次握手连接 printf("listening...\n"); listen(listenfd,LISTENQ); //创建一个线程，对服务器程序进行管理（关闭） pthread_create(&thread1,NULL,(void*)(&quit),NULL); //记录空闲的客户端的套接字描述符（-1为空闲） int i=0; for(i=0;i<MAXMEM;i++){ connfd[i].fd=-1; } while(1){ len=sizeof(cliaddr); for(i=0;i<MAXMEM;i++){ if(connfd[i].fd==-1) break; } //调用accept函数从listen接受的连接队列中取得一个连接 connfd[i].fd=accept(listenfd,(struct sockaddr*)&cliaddr,&len); ticks=time(NULL); sprintf(buff,"%Y-%m-%d%X \r \n",ctime(&ticks)); inet_ntop(AF_INET,&cliaddr,addr_p,sizeof(addr_p)); printf("第%d个客户端连接成功\n",i+1); printf("%s Connect from: %s,port %d\n",buff,addr_p,ntohs(cliaddr.sin_port)); //打印客户端连接的时间，地址和端口号 //针对当前套接字创建一个线程，对当前套接字的消息进行处理 pthread_create(&thread2,NULL,(void*)(&rcv_snd),(void*)i); } pthread_join(thread2,NULL); pthread_join(thread1,NULL); return 0; }