IO多路复用之poll

一、回顾

我们在上一章当中已经熟悉了IO多路复用以及select的使用，当然我们也知道了select有着在文件描述符过多时系统开销大以及监视的文件描述符有着最大限制的缺点。

而本章将会讲poll多路复用，它相比select有了一定的进步，但是也有着一部分select中存在的问题，现在让我们进入本章去深入了解poll多路复用吧。

二、poll函数

poll函数的机制与select类似，管理多个描述符也是轮询，与select相比它没有最大文件描述符数量的限制，但是它与select包含大量文件描述符的数组被整体复制于用户态和内核的地址空间之间，而不论这些文件描述符是否就绪，它的开销随着文件描述符数量的增加而线性增大，fd数量过大后就会导致性能下降。

man手册poll解释部分如下：

#include <poll.h>

int poll(struct pollfd *fds, nfds_t nfds, int timeout);

struct pollfd {
               int   fd;         /* file descriptor */
               short events;     /* requested events */
               short revents;    /* returned events */
           };

参数解释：fds用来指向一个struct pollfd类型的数组，每一个pollfd结构体指定了一个被监视的文件描述符，指示poll()监视多个文件描述符。每个结构体的events域是监视该文件描述符的事件掩码，由用户来设置这个域。revents域是文件描述符的操作结果事件掩码，内核在调用返回时设置这个域，events域中请求的任何事件都可能在revents域中返回；nfds用于指定数组中监听的元素个数；timeout为超时时间，如果设置为NULL则永不超时。

指定events标志以及测试revents标志的一些常值：

注意：POLLIN | POLLPRI等价于select()的读事件，POLLOUT |POLLWRBAND等价于select()的写事件。POLLIN等价于POLLRDNORM |POLLRDBAND，而POLLOUT则等价于POLLWRNORM。例如，要同时监视一个文件描述符是否可读和可写，我们可以设置 events为POLLIN |POLLOUT。在poll返回时，我们可以检查revents中的标志，对应于文件描述符请求的events结构体。如果POLLIN事件被设置，则文件描述符可以被读取而不阻塞。如果POLLOUT被设置，则文件描述符可以写入而不导致阻塞。这些标志并不是互斥的：它们可能被同时设置，表示这个文件描述符的读取和写入操作都会正常返回而不阻塞。

该函数成功调用时，poll()返回结构体中revents域不为0的文件描述符个数；如果在超时前没有任何事件发生，poll()返回0；
失败时，poll()返回-1，并设置errno为下列值之一：

• EBADF　　 一个或多个结构体中指定的文件描述符无效。
• EFAULTfds　　 指针指向的地址超出进程的地址空间。
• EINTR　　　　 请求的事件之前产生一个信号，调用可以重新发起。
• EINVALnfds　　参数超出PLIMIT_NOFILE值。
• ENOMEM　　 可用内存不足，无法完成请求。

三、poll实现socket服务器

（1）流程图：

（2）代码讲解：

fds_array[0].fd = listenfd;
fds_array[0].events = POLLIN;

这里需要在struct pollfd类型的数组fds_array中指定的event的标志，我们需要等待读操作，所以指定POLLIN标志。

rv = poll(fds_array, max+1, -1);

这里进行poll操作，参数分别为struct pollfd类型的数组fds_array，监听元素最大数量max+1，超时时间-1，表示永不超时，最后把返回值赋值给rv。

if (fds_array[0].revents & POLLIN)
{
	if( (connfd=accept(listenfd, (struct sockaddr *)NULL, NULL)) < 0)
	{
		printf("accept new client failure: %s\n", strerror(errno));
		continue;
	}
	found = 0;

	for(i=1; i<ARRAY_SIZE(fds_array) ; i++)
	{
		if( fds_array[i].fd < 0 )
		{
			printf("accept new client[%d] and add it into array\n", connfd );
			fds_array[i].fd = connfd;
			fds_array[i].events = POLLIN;
			found = 1;
			break;
		}
	}

	if( !found )
	{
		printf("accept new client[%d] but full, so refuse it\n", connfd);
		close(connfd);
		continue;
	}

	max = i>max ? i : max;
	if (--rv <= 0)
		continue;
}

这个地方通过测试revent标志判断是否为新的客户端到来。

有新客户端则进行accept操作，并用for循环更新fds_array[i]，用if( !found )来判断客户端是否已满。

最后要判断更新数组中监听元素个数。

其余代码与select实现服务器中的代码几乎相同，详见：IO多路复用之select

四、 总结

本文讲解了poll多路复用，详细的讲解了poll函数，对它的参数、功能和优缺点等都做了分析，希望这篇博客可以让大家有所收获。