#include<th.h>#define ERR_MSG(msg) do{\fprintf(stderr,"__%d__",__LINE__);\perror(msg);\
}while(0)#define PORT 6666
#define IP "192.168.2.3"//键盘输入事件
int serverkeyboard(fd_set readfds)
{char buf[128] = "";int sndfd = -1;bzero(buf,sizeof(buf));int res = scanf("%d %s",&sndfd,buf);while(getchar() != 10);if(res != 2) //终端输入的数据格式错误{printf("输入数据的格式错误，: fd string\n");return -1;}if(sndfd <= 2 || FD_ISSET(sndfd,&readfds) == 0){printf("非法的文件描述符: sndfd = %d\n",sndfd);return -1;}if(send(sndfd,buf,sizeof(buf),0) < 0){ERR_MSG("send error");return -1;}printf("send success\n");return 0;
}//客户端连接事件
int clientconnection(int sfd,struct sockaddr_in saveCin[],fd_set *preadfds,int *pmaxfd)
{int newfd = -1;struct sockaddr_in cin;socklen_t addrlen = sizeof(cin);newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);if(newfd < 0){ERR_MSG("accept error");return -1;}printf("[%s:%d]客户端连接成功 newfd = %d\n",\inet_ntoa(cin.sin_addr),\ntohs(cin.sin_port),\newfd);saveCin[newfd] = cin; //将ｃｉｎ另存到ｎｅｗｆｄ对应的下标位置去FD_SET(newfd,preadfds); //将ｎｅｗｆｄ添加到集合中*pmaxfd = *pmaxfd > newfd ? *pmaxfd : newfd; //更新maxfdreturn 0;
}//客户端交互事件
int clientsr(int fd,struct sockaddr_in * saveCin, fd_set *preadfds,int * pmaxfd)
{char buf[128] = "";bzero(buf,sizeof(buf));//接收ssize_t res = recv(fd,buf,sizeof(buf),0);if(res < 0){ERR_MSG("recv error");return -1;}else if(0 == res){printf("[%s:%d]客户端下线 newfd = %d\n",\inet_ntoa(saveCin[fd].sin_addr),\ntohs(saveCin[fd].sin_port),fd);close(fd);FD_CLR(fd,preadfds); //将下线的文件描述符踢出集合//由于踢出的文件描述符可能是最大的文件描述符//所以需要更新maxfd/** for(; *pmaxfd >= 0; *pmaxfd--)* {* if(FD_ISSET(*pmaxfd,preadfds))* {* break;* }*/while(FD_ISSET(*pmaxfd,preadfds) == 0 && (*pmaxfd)-- >= 0);return 0;}printf("[%s:%d] newfd = %d : %s\n",\inet_ntoa(saveCin[fd].sin_addr),\ntohs(saveCin[fd].sin_port),fd,buf);//发送strcat(buf,"*_*");if(send(fd,buf,sizeof(buf),0) < 0){ERR_MSG("send error");return -1;}printf("send success\n");return 0;
}int main(int argc, const char *argv[])
{int sfd;sfd = socket(AF_INET,SOCK_STREAM,0);if(sfd < 0){ERR_MSG("socket error");return -1;}//允许端口被快速复用int reuse = 1;if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0){ERR_MSG("setsockopt error");return -1;}printf("允许端口被快速复用成功\n");//填充服务器信息结构体struct sockaddr_in sin; //创建服务器地址信息结构体sin.sin_family = AF_INET;sin.sin_port = htons(PORT);sin.sin_addr.s_addr = inet_addr(IP);socklen_t serveraddrlen = sizeof(sin);if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0){ERR_MSG("bind error");return -1;}printf("服务器信息结构体绑定成功\n");if(listen(sfd,128) < 0){ERR_MSG("listen error");return -1;}printf("服务器监听成功\n");//创建一个读集合fd_set readfds,tempfds;FD_ZERO(&readfds);//将需要监测的文件描述符添加到集合中FD_SET(0,&readfds);FD_SET(sfd,&readfds);int maxfd = sfd; //存储最大的文件描述符int s_res = -1;ssize_t res = -1;char buf[128] = "";//储存连接成功的客户端的地址信息，用下标来对应文件描述符struct sockaddr_in saveCin[1024];while(1){tempfds = readfds;s_res = select(maxfd + 1, &tempfds,NULL,NULL,NULL);if(s_res < 0){ERR_MSG("select error");return -1;}else if(0 == s_res){printf("time out,,\n");break;}printf("__%d__\n",__LINE__);for(int i = 0; i < maxfd; i++){if(FD_ISSET(i,&tempfds) == 0){continue;}if(0 == i) //键盘输入事件{printf("触发键盘输入事件\n");serverkeyboard(readfds);}else if(sfd == i){printf("触发客户端连接事件\n");clientconnection(sfd,saveCin,&readfds,&maxfd);}else{printf("触发客户端交互事件\n");clientsr(i,saveCin,&readfds,&maxfd);}}}if(close(sfd) < 0){ERR_MSG("close error");return -1;}return 0;
}
