SOCKET网络编程五：IO多路复用epoll

一、关于select、poll、epoll这三者使用、对比介绍这篇：

网络编程--IO多路复用之select,poll和epoll详解

二、poll的使用和上篇的select大同小异：

int main(void)
{
	//当客户端结束时，服务器还维持一个僵尸进程
	signal(SIGCHLD, SIG_IGN);//使用signal忽略SIGCHLD信号(第一种方法)
	/*signal(SIGCHLD, handle_sigchld);//(第二种方法)*/

	int listenfd;//建立套接字
	if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
/*	if((listenfd=socket(PF_INET,SOCK_STREAM, 0))<0)*/
	ERR_EXIT("socket");
	//建立服务器地址
	struct sockaddr_in servaddr;
	memset(&servaddr,0,sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(5188);
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
	/*servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");*/
	/*inet_aton("127.0.0.1",&servaddr.sin_addr);*/

	 int on = 1;
	 //为了使服务器处于TIME_WATTING状态，能再次绑定地址
	 if(setsockopt(listenfd,SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))<0)
		ERR_EXIT("setsockopt");
	 //绑定地址
	if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
	ERR_EXIT("bind");
	//监听套接字
	if(listen(listenfd,SOMAXCONN)<0)//套接字变为被动套接字
		ERR_EXIT("listen");
	//存放客户端地址
	struct sockaddr_in peeraddr;
  	socklen_t peerlen;
  	
	int conn;//接受accept的套接字

	int i;
	int maxi = 0;
	struct pollfd client[2048];//假设一个进程可以打开最大的文件描述符数为2048
	for(i=0; i<2048; i++)
	{
		client[i].fd = -1;//空闲可以容纳新的客户端
	} 

	int nready;
	client[0].fd = listenfd;
	client[0].events = POLLIN;//读可读事件感兴趣
	//目前，加入到数组中的文件描述符个数等于1
	while(1)
	{
		nready = poll(client, maxi+1, -1);
		//nready = select(maxfd+1, &rset, NULL,NULL, NULL);
		if(nready==-1)
		{
			if(errno == EINTR)
				continue;
			ERR_EXIT("poll");
		}
		if(nready==0)
			continue;

		if(client[0].revents & POLLIN)//返回发生的事件与POLLIN相同
		//if(FD_ISSET(listenfd, &rset))
		{
			peerlen= sizeof(peeraddr);
			conn=accept(listenfd, (struct sockaddr*)&peeraddr, &peerlen);
			if(conn==-1)
				ERR_EXIT("accept");
			for(i=0; i<2048;i++)
			{
				if(client[i].fd<0)
				{
					client[i].fd=conn;//将新的套接字加入
					if(i>maxi)
						maxi=i;
					break;
				}

			}
			if(i==2048)
			{
				fprintf(stderr,"too many clients");
				exit(EXIT_FAILURE);
			}
			printf("ip=%s port=%d\n", inet_ntoa(peeraddr.sin_addr), ntohs(peeraddr.sin_port));
			//FD_SET(conn, &allset);
			client[i].events = POLLIN;//添加事件

			
			if(--nready<=0)
				continue;
		}
		//监听套接口总为client[0]
		//已连接套接口从1到maxi
		for(i=1; i<=maxi; i++)
		{
			conn=client[i].fd;
			if(conn == -1)
				continue;

			if(client[i].events & POLLIN)//事件为POLLIN则处理
			//if(FD_ISSET(conn, &rset))
			{
				char recvbuf[1024] = {0};
				int ret = readline(conn, recvbuf,1024);//先接受包头
				if(ret==-1)//读取失败则退出
					ERR_EXIT("readline");
				else if(ret==0)//客户端关闭
				{
					printf("client_close\n");
					//FD_CLR(conn, &allset);
					client[i].fd = -1;
					close(conn);
				}
				
				
				fputs(recvbuf,stdout);//打印标准输出
				writen(conn,recvbuf, strlen(recvbuf));
				if(--nready<=0)
					break;
			}
		}
	}
	return 0;
}

三、我们主要关注epoll

//仅修改服务器端
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>//使用signal函数
#include <sys/wait.h>//使用wait函数
#include <sys/epoll.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>

#include <vector>
#include <algorithm>

typedef std::vector<struct epoll_event> EventList;

#define ERR_EXIT(m) \
	do \
	{ \
		perror(m);	\
		exit(EXIT_FAILURE);\
	}while(0)

/*
activate_nonblock —— 设置I/O为非阻塞模式
*/
void activate_nonblock(int fd)
{
	int ret;
	int flags = fcntl(fd, F_GETFL);//获取文件标志
	if(flags == -1)
	{
		ERR_EXIT("fcntl");
	}

	flags |= O_NONBLOCK;//设置非阻塞模式
	ret = fcntl(fd, F_SETFL, flags);
	if(ret == -1)
		ERR_EXIT("fcntl");
}
//读确定大小的包
//ssize_t有符号数，size_t为无符号数
ssize_t readn(int fd, void *buf, size_t count)
{
	size_t nleft=count;//剩余字节数
	ssize_t nread;//已经接受的字节数
	char *bufp = (char*)buf;//

	while(nleft>0)
	{
		if((nread=read(fd, bufp, nleft))<0)
		{
			if(errno==EINTR)//信号中断
			{
				continue;
			}
			return -1;//否则出错
		}
		else if(nread==0)//表示对方关闭传送
		{
			return count-nleft;//返回已读字节数
		}
		bufp+=nread;//进行指针偏移
		nleft -= nread;
	} 
	return count;
}
ssize_t writen(int fd, const void *buf, size_t count)
{
	size_t nleft=count;//剩余字节数
	ssize_t nwritten;//已经写入的字节数
	char *bufp = (char*)buf;//

	while(nleft>0)
	{
		if((nwritten=write(fd, bufp, nleft))<0)
		{
			if(errno==EINTR)//信号中断
			{
				continue;
			}
			return -1;//否则出错
		}
		else if(nwritten==0)
		{
			continue;
		}
		bufp+=nwritten;//进行指针偏移
		nleft -= nwritten;
	} 
	return count;
}
ssize_t recv_peek(int sockfd, void *buf,size_t len)
{
	while(1)
	{
		//recv函数只用于套接口
		//recv函数读取后，不将数据在缓冲区清除
		int ret= recv(sockfd, buf, len, MSG_PEEK);
		if(ret == -1 && errno == EINTR)
			continue;
		return ret;
	}
}

ssize_t readline(int sockfd, void *buf, size_t maxline)
{
	int ret;//设置窥探返回值
	int nread;//设置已窥探字符数
	char *bufp = (char*)buf;//缓存buf
	int nleft = maxline;//设置maxline为包最大长度，nleft为剩余需读取字符数
	while(1)
	{
		ret=recv_peek(sockfd, bufp, nleft);
		if(ret<0)
		{
			return ret;
		}
		else if(ret=0)
		{
			return ret;//对方终止了传送
		}

		nread=ret;
		int i;//检测有没有‘\n’字符，有则读取
		for(i=0; i<nread;i++)
		{
			if(bufp[i]=='\n')
			{
				ret= readn(sockfd,bufp, i+1);
				if(ret != i+1)//已经窥探到有i+1字符，如果没有则错误
					exit(EXIT_FAILURE);
				return ret;
			}
		}
		//如果没有读到‘\n’，则将消息读入，直到最大包
		nleft -= nread;
		ret= readn(sockfd, bufp, nread);
		if(ret != nread)//已经窥探到有nread个字符，如果readn函数不能读取这么多，则错误
		{
			exit(EXIT_FAILURE);
		}
		bufp += nread;
	}
	return -1;
}

void do_service(int conn)//执行连接建立后，内容读取，和返回内容
{
	char recvbuf[1024];
	int n;
		while(1)
		{
			memset(recvbuf, 0, sizeof(recvbuf));
			int ret = readline(conn, recvbuf,1024);//先接受包头
			if(ret==-1)//读取失败则退出
			ERR_EXIT("readline");
			else if(ret==0)//客户端关闭
			{
				printf("client_close\n");
				break;
			}
			
			
			fputs(recvbuf,stdout);//打印标准输出
			writen(conn,recvbuf, strlen(recvbuf));
		}
}
void ehco_srv(int conn)//执行连接建立后，内容读取，和返回内容
{
	char recvbuf[1024];
	int n;
		while(1)
		{
			memset(recvbuf, 0, sizeof(recvbuf));
			int ret = readline(conn, recvbuf,1024);//先接受包头
			if(ret==-1)//读取失败则退出
			ERR_EXIT("readline");
			else if(ret==0)//客户端关闭
			{
				printf("client_close\n");
				break;
			}
			
			
			fputs(recvbuf,stdout);//打印标准输出
			writen(conn,recvbuf, strlen(recvbuf));
		}
}
void handle_sigchld(int sig)
{
	//只等待一个子进程结束，就退出消除一个子进程产生的僵尸进程，无法处理并发的多个子进程，造成的僵尸进程
	//wait(NULL);
	//等待选项为不挂起，用来处理客户端多个并发子进程连接后关闭时造成的多个僵尸进程
	while(waitpid(-1, NULL, WNOHANG)>0);
}
void handle_sigpipe(int sig)
{
	printf("recv a sig=%d\n", sig);
}
int main(void)
{
	int count = 0;
	//处理A调用了close,A不能再接收和发送数据段，但对等方B可以向A发送数据包，
	//如果B这样做了，A返回RST段，B再向A发送数据，A会产生SIGPIPE信号，该信号直接忽略就好
	signal(SIGPIPE, handle_sigpipe);
	//当客户端结束时，服务器还维持一个僵尸进程
	/*signal(SIGCHLD, SIG_IGH)；//使用signal忽略SIGCHLD信号(第一种方法)*/
	signal(SIGCHLD, handle_sigchld);//(第二种方法)

	int listenfd;//建立套接字
	if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
	/*	if((listenfd=socket(PF_INET,SOCK_STREAM, 0))<0)*/
		ERR_EXIT("socket");
	//建立服务器地址
	struct sockaddr_in servaddr;
	memset(&servaddr,0,sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(5188);
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
	/*servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");*/
	/*inet_aton("127.0.0.1",&servaddr.sin_addr);*/

	 int on = 1;
	 //为了使服务器处于TIME_WATTING状态，能再次绑定地址
	 if(setsockopt(listenfd,SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))<0)
		ERR_EXIT("setsockopt");
	 //绑定地址
	if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
		ERR_EXIT("bind");
	//监听套接字
	if(listen(listenfd,SOMAXCONN)<0)//套接字变为被动套接字
		ERR_EXIT("listen");

	//epoll函数修改
	std::vector<int> clients;//存放已连接套接字
	int epollfd;//存放epoll实例
	epollfd = epoll_create1(EPOLL_CLOEXEC);//创建一个EPOLL实例

	//将感兴趣的事件加入到epoll队列
	struct epoll_event event;
	event.data.fd = listenfd;
	event.events = EPOLLIN | EPOLLET;//边沿触发
	epoll_ctl(epollfd, EPOLL_CTL_ADD, listenfd, &event);

	EventList events(16);//存放并发事件

	//存放客户端地址
	struct sockaddr_in peeraddr;
	socklen_t peerlen;

	int conn;//接受accept的套接字
	
	int i;

	int nready;
	while(1)
	{//&*events.begin()动态数组首地址
		nready = epoll_wait(epollfd, &*events.begin(), static_cast<int>(events.size()), -1);

		//nready = poll(client，maxi+1, -1);
		//nready = select(maxfd+1, &rset, NULL,NULL, NULL);
		if(nready==-1)
		{
			if(errno == EINTR)
				continue;
			ERR_EXIT("epoll_wait");
		}
		if(nready==0)
			continue;
		//数组已满，等待的数已经满了，扩充数组
		if((size_t)nready == events.size())//等待的事件的个数
			events.resize(events.size()*2);

		//开始处理已检测到的事件
		for(i=0; i<nready; i++)
		{
			//处理监听套接口
			if(events[i].data.fd == listenfd)
			{
				peerlen = sizeof(peeraddr);
				conn = accept(listenfd, (struct sockaddr*)&peeraddr, &peerlen);
				if(conn==-1)
				{
					ERR_EXIT("accept");
				}
				printf("ip=%s port=%d\n", inet_ntoa(peeraddr.sin_addr), ntohs(peeraddr.sin_port));
				clients.push_back(conn);

				activate_nonblock(conn);

				//监听套接口变为已连接套接口，放入队列，进行管理，等待数据到来
				event.data.fd = conn;
				event.events = EPOLLIN | EPOLLET;
				epoll_ctl(epollfd, EPOLL_CTL_ADD, conn, &event);
			}
			else if(events[i].events & EPOLLIN)//已连接套接字
			{
				conn = events[i].data.fd;
				if(conn<0)
					continue;
				char recvbuf[1024] = {0};
				int ret = readline(conn, recvbuf, 1024);
				if(ret ==-1)
					ERR_EXIT("readline");
				if(ret ==0)
				{
					printf("client close\n");
					close(conn);

					event= events[i];
					epoll_ctl(epollfd, EPOLL_CTL_DEL, conn, &event);
					clients.erase(std::remove(clients.begin(), clients.end(), conn), clients.end());
				}
				fputs(recvbuf,stdout);//打印标准输出
				writen(conn,recvbuf, strlen(recvbuf));
			}
		}
	}
	return 0;
}