服务器:

#include<myhead.h>#define PORT 9999
#define IP "192.168.125.130"struct Msg
{char type;char name[20];char text[100];
};
struct sock
{int fd;struct sockaddr_in cin;
};int main(int argc, const char *argv[])
{if(argc != 3){printf("输入有误请重新输入\n");printf("./a.out IP PORT\n");return -1;}int sfd = -1;if((sfd = socket(AF_INET,SOCK_STREAM,0)) == -1){perror("socket error");return -1;}//设置快速重用int reuse = 1;if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))==-1){perror("setsockopt error");return -1;}//创建绑定结构体struct sockaddr_in sin;sin.sin_family = AF_INET;//sin.sin_port = htons(PORT);sin.sin_port = htons(atoi(argv[2]));//sin.sin_addr.s_addr = inet_addr(IP);sin.sin_addr.s_addr = inet_addr(argv[1]);//绑定地址信息结构体if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){perror("bind error");return -1;}printf("bind success\n");//设置为监听状态if(listen(sfd,128) == -1){perror("listen error");return -1;}printf("listen success\n");struct sockaddr_in cin;socklen_t socklen = sizeof(cin);fd_set readfds;//读文件描述符集合//清空集合FD_ZERO(&readfds);//文件描述符加入集合FD_SET(0,&readfds);FD_SET(sfd,&readfds);int maxfd = sfd;int newfd = -1;int res = -1;char buf[128] ="";fd_set tempfds;struct sock arr_cin[1024];//成员信息结构体数组struct Msg peo;//接受客户端发来的结构体while(1){tempfds = readfds;//判断产生的事件res = select(maxfd+1,&tempfds,NULL,NULL,NULL);if(res == -1){perror("select error");return -1;}else if(res == 0){printf("time out\n");return -1;}//判断事件for(int i=0;i<=maxfd;i++){if(!FD_ISSET(i,&tempfds)){continue;}//发生事件的文件描述符if(i == sfd){//设置为接收状态if((newfd = accept(sfd,(struct sockaddr*)&cin,&socklen)) == -1){perror("accept error");return -1;}arr_cin[newfd].cin = cin;arr_cin[newfd].fd = newfd;printf("[%s:%d] 连接成功, newfd = %d\n", inet_ntoa(arr_cin[newfd].cin.sin_addr),\ntohs(arr_cin[newfd].cin.sin_port), newfd); //将新加入的客户端加入数组FD_SET(newfd,&readfds);if(newfd > maxfd){maxfd = newfd;}}else if(i == 0){peo.type = 'c';bzero(peo.text,sizeof(peo.text));fgets(peo.text,sizeof(peo.text),stdin);peo.text[strlen(peo.text)-1]=0;strcpy(peo.name,"系统");//发送给全部客服端for(int index=4;index<=maxfd&&FD_ISSET(index,&readfds);index++){send(index,&peo,sizeof(peo),0);}}else{//接受客户端发来的消息，并发送给其他人bzero(&peo,sizeof(peo));res = recv(i,&peo,sizeof(peo),0);if(res == 0){printf("客户端下线\n");close(i);FD_CLR(i,&readfds);for(int i=maxfd;i>0;i--){if(FD_ISSET(i,&readfds)){maxfd = i;break;}}continue;}if(peo.type == 'L'){sprintf(peo.text,"-----%s上线-------",peo.name);for(int index = 4;index<=maxfd&&FD_ISSET(index,&readfds);index++){if( i== arr_cin[index].fd )continue;send(index,&peo,sizeof(peo),0);}}else if(peo.type == 'c'){for(int index = 4;index<=maxfd&&FD_ISSET(index,&readfds);index++){if( i != arr_cin[index].fd ){printf("%s:%s\n",peo.name,peo.text);//测试send(arr_cin[index].fd,&peo,sizeof(peo),0);}}}else if(peo.type == 'Q'){sprintf(peo.text,"-----%s下线-------",peo.name);for(int index = 4;index<=maxfd&&FD_ISSET(index,&readfds);index++){if( i== arr_cin[index].fd )continue;send(index,&peo,sizeof(peo),0);}close(i);FD_CLR(i,&readfds);for(int i=maxfd;i>0;i--){if(FD_ISSET(i,&readfds)){maxfd = i;break;}}continue;}}}}close(sfd);return 0;
}

客户端:

#include<myhead.h>#define PORT 9999
#define IP "192.168.125.130"struct Msg
{char type;char name[20];char text[100];
};int main(int argc, const char *argv[])
{if(argc != 3){printf("输入有误请重新输入\n");printf("./a.out IP PORT\n");return -1;}int sfd = -1;if((sfd = socket(AF_INET,SOCK_STREAM,0)) == -1){perror("socket error");return -1;}//设置快速重用int reuse = 1;if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))==-1){perror("setsockopt error");return -1;}//链接struct sockaddr_in sin;sin.sin_family = AF_INET;//sin.sin_port = htons(PORT);//sin.sin_addr.s_addr = inet_addr(IP);sin.sin_port = htons(atoi(argv[2]));sin.sin_addr.s_addr = inet_addr(argv[1]);if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){perror("connect error");return -1;}printf("connect success\n");//将成员信息发送给服务器struct Msg information;printf("请输入姓名>>>");scanf("%s",information.name);getchar();information.type = 'L';strcpy(information.text, " ");printf("-----%s登陆成功-----\n",information.name);send(sfd,&information,sizeof(information),0);fd_set readfds;FD_SET(0,&readfds);FD_SET(sfd,&readfds);int maxfd = sfd;int res = -1;char buf[128] ="";fd_set tempfds;while(1){tempfds = readfds;res = select(maxfd+1,&tempfds,NULL,NULL,NULL);if(res == -1){perror("select error");return -1;}else if(res == 0){printf("time out\n");return -1;}//到此发生事件if(FD_ISSET(0,&tempfds)){struct Msg mem;mem.type = 'c';//设置类型strcpy(mem.name,information.name);//复制名字//客户端消息发送个服务器，服务器发送给其他客户端bzero(mem.text,sizeof(mem.text));fgets(mem.text,sizeof(mem.text),stdin);mem.text[strlen(mem.text)-1]=0;if(strcmp(mem.text,"quit") == 0){mem.type = 'Q';send(sfd,&mem,sizeof(mem),0);break;}send(sfd,&mem,sizeof(mem),0);}else{struct Msg mem;//接收服务器发来的消息bzero(&mem,sizeof(mem));res = recv(sfd,&mem,sizeof(mem),0);if(res == 0){printf("客户端下线\n");break;}else if(res == -1){perror("recv");return -1;}if(mem.type == 'L'||mem.type == 'Q'){	printf("%s\n",mem.text);}elseprintf("%s:%s\n",mem.name,mem.text);}}return 0;
}