网络编程,IO多路复用
1.使用IO多路复用完成TCP并发服务器
#include<myhead.h>
#define SER_PORT 8888 //服务器端口号
#define SER_IP "192.168.124.10" //服务器IP地址int main(int argc, const char *argv[])
{//1、创建用于连接的套接字int sfd = socket(AF_INET, SOCK_STREAM, 0); //参数1:通信域,表明使用的是ipv4协议//参数2:通信方式,使用TCP通信//参数3:0表示之前已经指定协议 IPPROTO_TCPif(sfd == -1){perror("socket error");return -1;}printf("sfd = %d\n", sfd); //3//将端口号快速重用函数int reuse = 1;if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1){perror("setsockopt error");return -1;}printf("端口号快速重用成功\n");//2、给当前套接字绑定IP地址和端口号//2.1填充要绑定的地址信息结构体struct sockaddr_in sin;sin.sin_family = AF_INET; //通信域sin.sin_port = htons(SER_PORT); //端口号sin.sin_addr.s_addr = inet_addr(SER_IP); //ip地址//2.2 绑定if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) == -1){perror("bind error");return -1;}printf("bind success %s %s %d\n", __FILE__, __func__, __LINE__);//3、将套接字设置成监听状态if(listen(sfd, 128) == -1){perror("listen error");return -1;}printf("listen success %s %s %d\n", __FILE__, __func__, __LINE__);//4、阻塞等待客户端的链接请求//4.1定义容器接收客户端的地址信息struct sockaddr_in cin; //用于接收地址信息socklen_t socklen = sizeof(cin); //用于接收地址信息的大小int newfd = -1;//11、准备一个文件描述符容器fd_set readfds, tempfds; //22、清空容器FD_ZERO(&readfds);//33、将要检测的文件描述符放入集合FD_SET(0, &readfds);FD_SET(sfd, &readfds);//定义一个变量存储容器中最大的文件描述符int maxfd = sfd;//定义一个地址信息结构体数组,每一个元素对应一个客户端文件描述符struct sockaddr_in cin_arr[1024];while(1){tempfds = readfds; //存储一份数据//使用select函数对容器中的文件描述符进行检测int res = select(maxfd+1,&tempfds,NULL,NULL,NULL);if(res == -1){perror("select error");return -1;}else if(res == 0){printf("timeout\n");return -1;}//判断是否是文件描述符触发事件for(int cli=0;cli<=maxfd;cli++){if(!FD_ISSET(cli,&tempfds)){continue;}if(cli == sfd){//4.2 接收客户端的链接newfd = accept(sfd, (struct sockaddr*)&cin, &socklen); if(newfd == -1){perror("accept error");return -1;}printf("[%s:%d]发来链接请求 %s %s %d\n", \inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),__FILE__, __func__, __LINE__);//更新地址信息结构体数组cin_arr[newfd] = cin;//将newfd放入readfds中FD_SET(newfd, &readfds);//更新maxfdif(newfd > maxfd){maxfd = newfd;}}else//判断0号文件描述符是否在集合中if(cli == 0){//终端输入char wbuf[128] = "";scanf("%s", wbuf);printf("触发了终端输入事件。。。\n");if(strcmp(wbuf, "quit") == 0){goto END;}//将消息发送给所有客户端for(int i=4; i<=maxfd; i++ ){sendto(i, wbuf, sizeof(wbuf), 0,(struct sockaddr*) &cin_arr[i], sizeof(cin_arr[i]));}}else{//5、跟客户端进行消息通信char buf[128] = "";//将数组清空bzero(buf, sizeof(buf));//读取客户端发来的消息//int res = read(newfd, buf, sizeof(buf));int res = recv(cli, buf, sizeof(buf), 0);if(res == 0){printf("客户端已经下线\n");//关闭当前通信的套接字close(cli);//将当前文件描述符从文件描述符集合中移除FD_CLR(cli, &readfds);//更新maxfdfor(int i=maxfd; i>=sfd; i--){if(FD_ISSET(i, &readfds)){maxfd = i;break;}}continue;}printf("[%s:%d] : %s\n", inet_ntoa(cin_arr[cli].sin_addr), ntohs(cin_arr[cli].sin_port), buf);//给客户端发消息strcat(buf, "*_*");//write(newfd, buf, sizeof(buf));send(cli, buf, sizeof(buf), 0);printf("发送成功\n");}}}END://6、关闭套接字close(sfd);return 0;}