linux下epoll编程

#include <unistd.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <fcntl.h>
#include <sys/wait.h>
#include <sys/epoll.h>

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>

#include <vector>
#include <algorithm>
#include <iostream>

typedef std::vector<struct epoll_event> EventList;

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

int main(void)
{
	signal(SIGPIPE, SIG_IGN);
	//客户端关闭套接字，服务器端就会收到SIGPIPE信号（这个信号默认用于退出当前进程),SIG_IGN用于屏蔽这个信号
	signal(SIGCHLD, SIG_IGN);
	//signal（SIGCHLD,SIG_IGN） 通知内核，自己对子进程的结束不感兴趣，那么子进程结束后，内核会回收， 并不再给父进程发送信号

	int idlefd = open("/dev/null", O_RDONLY | O_CLOEXEC);
	int listenfd;
	//if ((listenfd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
	if ((listenfd = socket(PF_INET, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, IPPROTO_TCP)) < 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);

	int on = 1;
	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)//SOMAXCONN表示最大连接数
		ERR_EXIT("listen");

	std::vector<int> clients;
	int epollfd;
	epollfd = epoll_create1(EPOLL_CLOEXEC);
	//这个函数用来创建一个文件描述符来唯一标示内核中的一个事件表
	//)参数EPOLL_CLOEXEC是这个epoll专用描述符epfd所关联的socketfd的最大个数,这个参数实际上是不起什么作用的
	//EPOLL_CLOEXEC = 02000000,//在新建的epfd上设置FD_CLOEXEC。

	struct epoll_event event; //事件结构体
	event.data.fd = listenfd;//指定事件所从属的目标文件描述符
	event.events = EPOLLIN/* | EPOLLET*/;//监听套接字加入 关注他的读的触发事件（电平触发）
	epoll_ctl(epollfd, EPOLL_CTL_ADD, listenfd, &event);//添加进去关注
	//向epollfd这个事件表中注册( EPOLL_CTL_ADD)listenfd这个套接字上的事件(event);
	
	EventList events(16);//向量数组类的实例化，16个事件
	struct sockaddr_in peeraddr;
	socklen_t peerlen;
	int connfd;

	int nready;
	while (1)
	{
	   
		nready = epoll_wait(epollfd, &*events.begin(), static_cast<int>(events.size()), -1);
		//epoll_wait在一段超时时间内等待epollfd上的事件，如果检测到事件，
		//就将所有就绪的事件从内核事件表epollfd中复制到&*events.begin()指向的数组中，这个数组只用于输出epoll_wait检测到的就绪事件
		//nready是返回的&*events.begin()这个数组中装啦多少个就绪的文件描述符
        //这里就是poll和epoll在使用上的差异
		//poll必须遍历所有已注册文件描述符并找到其中的就绪着
		//而epoll只用遍历就绪的nready个(在&*events.begin()中存着)在文件描述符
		if (nready == -1)
		{
			if (errno == EINTR) //接收到中断信号
				continue;
			
			ERR_EXIT("epoll_wait");
		}
		if (nready == 0)	// nothing happended //超时
			continue;

		if ((size_t)nready == events.size())//如果events中已经存满了事件（上面定义啦16个),就扩容1倍
			events.resize(events.size()*2);

		for (int i = 0; i < nready; ++i) //仅遍历就绪的的nready个文件描述符（这些都是要处理的）
		{
			if (events[i].data.fd == listenfd)
			{
				peerlen = sizeof(peeraddr);
				connfd = ::accept4(listenfd, (struct sockaddr*)&peeraddr,
						&peerlen, SOCK_NONBLOCK | SOCK_CLOEXEC);

				if (connfd == -1)//函数发生错误
				{
					if (errno == EMFILE)//表示每个进程允许打开的文件描述符数量最大值已经到达
					{//遇到这种情况的解决方法是
					/*
					准备一个空闲的文件描述符。遇到这种情况，先关闭这个空闲文件，
					获得一个文件描述符名额;再accept(2)拿到socket连接的文件描述符；随后立刻close(2)，
					这样就优雅地断开了与客户端的连接；
					最后重新打开空闲文件，把“坑”填上，以备再次出现这种情况时使用
					*/
					
						close(idlefd);
						idlefd = accept(listenfd, NULL, NULL);
						close(idlefd);
						idlefd = open("/dev/null", O_RDONLY | O_CLOEXEC);
						continue;
					}
					else
						ERR_EXIT("accept4");
				}


				std::cout<<"ip="<<inet_ntoa(peeraddr.sin_addr)<<
					" port="<<ntohs(peeraddr.sin_port)<<std::endl;

				clients.push_back(connfd);
				
				event.data.fd = connfd;
				event.events = EPOLLIN/* | EPOLLET*/;
				epoll_ctl(epollfd, EPOLL_CTL_ADD, connfd, &event);
			}
			else if (events[i].events & EPOLLIN)//有数据可读事件发生
			{
				connfd = events[i].data.fd;
				if (connfd < 0)
					continue;

				char buf[1024] = {0};
				int ret = read(connfd, buf, 1024);
				if (ret == -1)
					ERR_EXIT("read");
				if (ret == 0)
				{
					std::cout<<"client close"<<std::endl;
					close(connfd);
					event = events[i];
					epoll_ctl(epollfd, EPOLL_CTL_DEL, connfd, &event);
					//向epollfd这个事件表中删除( EPOLL_CTL_ADD)connfd这个套接字上的事件(event);
					clients.erase(std::remove(clients.begin(), clients.end(), connfd), clients.end());//删除这个元素
					continue;//进入下次循环
				}

				std::cout<<buf;
				write(connfd, buf, strlen(buf));//向这个数据套接字中写入数据
			}

		}
	}

	return 0;
}