I/O复用之epoll模型

 

epoll是Linux内核为处理大批量文件描述符而作了改进的poll，是Linux下多路复用IO接口select/poll的增强版本，它能显著提高程序在大量并发连接中只有少量活跃的情况下的系统CPU利用率。另一点原因就是获取事件的时候，它无须遍历整个被侦听的描述符集，只要遍历那些被内核IO事件异步唤醒而加入Ready队列的描述符集合就行了。epoll除了提供select/poll那种IO事件的水平触发（Level Triggered）外，还提供了边缘触发（Edge Triggered），这就使得用户空间程序有可能缓存IO状态，减少epoll_wait/epoll_pwait的调用，提高应用程序效率。

相比于select，epoll最大的好处在于它不会随着监听fd数目的增长而降低效率。因为在内核中的select实现中，它是采用轮询来处理的，轮询的fd数目越多，自然耗时越多 

epoll的接口非常简单，一共就三个函数：
1. int epoll_create(int size);
创建一个epoll的句柄，size用来告诉内核这个监听的数目一共有多大。这个参数不同于select()中的第一个参数，给出最大监听的fd+1的值。需要注意的是，当创建好epoll句柄后，它就是会占用一个fd值，在linux下如果查看/proc/进程id/fd/，是能够看到这个fd的，所以在使用完epoll后，必须调用close()关闭，否则可能导致fd被耗尽。


2. int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
epoll的事件注册函数，它不同与select()是在监听事件时告诉内核要监听什么类型的事件，而是在这里先注册要监听的事件类型。第一个参数是epoll_create()的返回值，第二个参数表示动作，用三个宏来表示：
EPOLL_CTL_ADD：注册新的fd到epfd中；
EPOLL_CTL_MOD：修改已经注册的fd的监听事件；
EPOLL_CTL_DEL：从epfd中删除一个fd； 

第三个参数是需要监听的fd，第四个参数是告诉内核需要监听什么事，struct epoll_event结构如下：

 1 typedef union epoll_data {
 2     void *ptr;
 3     int fd;
 4     __uint32_t u32;
 5     __uint64_t u64;
 6 } epoll_data_t;
 7 
 8 struct epoll_event {
 9     __uint32_t events; /* Epoll events */
10     epoll_data_t data; /* User data variable */
11 };

 events可以是以下几个宏的集合：
EPOLLIN ：表示对应的文件描述符可以读（包括对端SOCKET正常关闭）；
EPOLLOUT：表示对应的文件描述符可以写；
EPOLLPRI：表示对应的文件描述符有紧急的数据可读（这里应该表示有带外数据到来）；
EPOLLERR：表示对应的文件描述符发生错误；
EPOLLHUP：表示对应的文件描述符被挂断；
EPOLLET： 将EPOLL设为边缘触发(Edge Triggered)模式，这是相对于水平触发(Level Triggered)来说的。
EPOLLONESHOT：只监听一次事件，当监听完这次事件之后，如果还需要继续监听这个socket的话，需要再次把这个socket加入到EPOLL队列里


3. int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);
等待事件的产生，类似于select()调用。参数events用来从内核得到事件的集合，maxevents告之内核这个events有多大，这个 maxevents的值不能大于创建epoll_create()时的size，参数timeout是超时时间（毫秒，0会立即返回，-1将不确定，也有说法说是永久阻塞）。该函数返回需要处理的事件数目，如返回0表示已超时。

代码示例：

#include <stdio.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdlib.h>
#include <ctype.h>
#include <sys/select.h>
#include <sys/poll.h>
#include <sys/epoll.h>
#include <string.h>

#define	SERVERIP "127.0.0.1" 
#define	SERVERPORT	8000
#define	BUFSIZE	1500
#define	IPSIZE	16
#define	LISTEN	128
#define	EPOLLSIZE	60000


int main(void)
{
		struct sockaddr_in serveraddr,clientaddr;
		int ready=0;
		struct epoll_event eparr[EPOLLSIZE]; //就绪队列  传出
		struct epoll_event eptmp;
		char buf[BUFSIZE];
		int len;
		int clientfd;
		bzero(&serveraddr,sizeof(serveraddr));
		serveraddr.sin_family = AF_INET;
		serveraddr.sin_port = htons(SERVERPORT);
		inet_pton(AF_INET,SERVERIP,&serveraddr.sin_addr.s_addr);
		int serverfd = socket(AF_INET,SOCK_STREAM,0);
		bind(serverfd,(struct sockaddr*)&serveraddr,sizeof(serveraddr));
		listen(serverfd,LISTEN);
		int epfd = epoll_create(EPOLLSIZE); //创建epoll监听的红黑树
		eptmp.events=EPOLLIN;
		eptmp.data.fd = serverfd;
		epoll_ctl(epfd,EPOLL_CTL_ADD,serverfd,&eptmp);
		printf("EPoll Server Runing....\n");
		while(1)
		{
				ready = epoll_wait(epfd,eparr,EPOLLSIZE,-1);
				if(ready == -1)
						perror("EPoll Call Failed..\n");
				else{
						while(ready){
								ready--;
								if(eparr[ready].data.fd == serverfd)	
								{
										int clientsize = sizeof(clientaddr);
										//建立新的链接
										clientfd = accept(serverfd,(struct sockaddr*)&clientaddr,&clientsize);
										eptmp.events = EPOLLIN;
										eptmp.data.fd = clientfd;
										epoll_ctl(epfd,EPOLL_CTL_ADD,clientfd,&eptmp);
								}else{
										if((len = read(eparr[ready].data.fd,buf,sizeof(buf)))>0)
										{
												int j=0;
												while(len > j)
												{
														buf[j]=toupper(buf[j]);
														j++;
												}
												write(eparr[ready].data.fd,buf,len);
												bzero(buf,sizeof(buf));
										}else if(len == 0){
												epoll_ctl(epfd,EPOLL_CTL_DEL,eparr[ready].data.fd,NULL);
												close(eparr[ready].data.fd);
										}
								}

						}
				}
		}
}