首页 > 代码库 > 多路复用I/O poll()

多路复用I/O poll()

1、基本知识

  poll的机制与select类似,与select在本质上没有多大差别,管理多个描述符也是进行轮询,根据描述符的状态进行处理,但是poll没有最大文件描述符数量的限制。poll和select同样存在一个缺点就是,包含大量文件描述符的数组被整体复制于用户态和内核的地址空间之间,而不论这些文件描述符是否就绪,它的开销随着文件描述符数量的增加而线性增大。

2、poll函数

  函数格式如下所示:

# include <poll.h>
int poll ( struct pollfd * fds, unsigned int nfds, int timeout);

pollfd结构体定义如下:

struct pollfd {

int fd;         /* 文件描述符 */
short events;         /* 等待的事件 */
short revents;       /* 实际发生了的事件 */
} ; 

  每一个pollfd结构体指定了一个被监视的文件描述符,可以传递多个结构体,指示poll()监视多个文件描述符。每个结构体的events域是监视该文件描述符的事件掩码,由用户来设置这个域。revents域是文件描述符的操作结果事件掩码,内核在调用返回时设置这个域。events域中请求的任何事件都可能在revents域中返回。合法的事件如下:

  POLLIN         有数据可读。

  POLLRDNORM       有普通数据可读。

  POLLRDBAND      有优先数据可读。

  POLLPRI         有紧迫数据可读。

  POLLOUT            写数据不会导致阻塞。

  POLLWRNORM       写普通数据不会导致阻塞。

  POLLWRBAND        写优先数据不会导致阻塞。

  POLLMSGSIGPOLL     消息可用。

  此外,revents域中还可能返回下列事件:
  POLLER     指定的文件描述符发生错误。

  POLLHUP   指定的文件描述符挂起事件。

  POLLNVAL  指定的文件描述符非法。

这些事件在events域中无意义,因为它们在合适的时候总是会从revents中返回。

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

  timeout参数指定等待的毫秒数,无论I/O是否准备好,poll都会返回。timeout指定为负数值表示无限超时,使poll()一直挂起直到一个指定事件发生;timeout为0指示poll调用立即返回并列出准备好I/O的文件描述符,但并不等待其它的事件。这种情况下,poll()就像它的名字那样,一旦选举出来,立即返回。


  返回值和错误代码
  成功时,poll()返回结构体中revents域不为0的文件描述符个数;如果在超时前没有任何事件发生,poll()返回0;失败时,poll()返回-1,并设置errno为下列值之一:
  EBADF         一个或多个结构体中指定的文件描述符无效。

  EFAULTfds   指针指向的地址超出进程的地址空间。

  EINTR      请求的事件之前产生一个信号,调用可以重新发起。

  EINVALnfds  参数超出PLIMIT_NOFILE值。

  ENOMEM       可用内存不足,无法完成请求。

3、测出程序

  编写一个echo server程序,功能是客户端向服务器发送信息,服务器接收输出并原样发送回给客户端,客户端接收到输出到终端。

  服务器端程序如下:

#include <stdio.h>
#include <stdio.h>
#include "debug.h"
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>

int main()
{
    int listenfd = socket(AF_INET, SOCK_STREAM, 0);
    if(-1 == listenfd)
        errsys("socket");

    struct sockaddr_in myaddr = {0};
    struct sockaddr_in clientaddr = {0};
    myaddr.sin_family = AF_INET;
    myaddr.sin_port = htons(8888);
    myaddr.sin_addr.s_addr = inet_addr("0.0.0.0");//INADDR_ANY
    int len = sizeof myaddr;

    if(-1 == bind(listenfd, (struct sockaddr*)&myaddr, len))
        errsys("bind");

    if(-1 == listen(listenfd, 10))
        errsys("listen");


#define BUFSIZE 100
#define MAXNFD  1024 
    struct pollfd myfds[MAXNFD] = {0};                            //定义一个大小为 MAXFD 的就绪状态询问表
    myfds[0].fd = listenfd;
    myfds[0].events = POLLIN;
    int maxnum = 1;
    
    int nready;
    char buf[MAXNFD][BUFSIZE] = {0};
    while(1)
    {
        if(-1 == (nready = poll(myfds, maxnum, -1)) )            //询问列表中描述符是否准备就绪
            errsys("poll");
        
        
        int i = 0;
        for(;i<maxnum; i++)                                            //遍历额列表中的套接字描述符
        {
            if(myfds[i].revents & POLLIN)                            //第 i 个描述符读请求准备就绪
            {
    
                if(myfds[i].fd == listenfd)                                //监听套接字描述符准备就绪
                {
                    int sockfd = accept(listenfd, (struct sockaddr*)&clientaddr, &len);                //获取 client 套接字
                    if(-1 == sockfd)
                        errsys("accept");
                    debug("incoming: %s\n", inet_ntoa( clientaddr.sin_addr) );

                    myfds[maxnum].fd = sockfd;                        //要读取client数据,将client 套接字描述符添加到 询问表中
                    myfds[maxnum].events = POLLIN;                    //要问的事情:是否准备好读取    POLLIN server读(client发送)   POLLOUT server写(clinet接收)
                    maxnum++;                //要询问的描述符数量

                    if(--nready==0)
                        continue;
                }
                else
                {
                    int ret = read(myfds[i].fd, buf[myfds[i].fd], sizeof buf[0]);
                    if(0 == ret)
                    {
                        close(myfds[i].fd);                    //client 关闭,释放套接字
                        myfds[i].fd = -1;
                        continue;
                    }
                    myfds[i].events = POLLOUT;            // 接收完毕 ,接下来判读写请求时否就绪
                }
            }
            else if(myfds[i].revents & POLLOUT)                //第 i 个描述符 写请求准备就绪
            {
                int ret = write(myfds[i].fd, buf[myfds[i].fd], sizeof buf[0]);
                printf("ret %d: %d\n", myfds[i].fd, ret);
                myfds[i].events = POLLIN;            // 再次设为 读请求。
            }
        }
    }

    close(listenfd);
}

 

多路复用I/O poll()