I/O多路复用之poll

对于poll的理论讲解，可参考文章：http://www.cnblogs.com/Anker/p/3261006.html

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　　     可用内存不足，无法完成请求。

实战：写一个echo程序，服务器原样返回客户端发送过来的内容。用poll写。

client：

#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <stdlib.h>
#include <poll.h>

#define err_exit(m)\
        {\
                perror(m);\
                exit(EXIT_FAILURE);\
        }

#define SERV_PORT 9877

#define BUFSIZE 4096

int max(int a, int b)
{
        if (a > b)
                return a;
        else
                return b;
}

//客户端具体操作函数
void str_cli(FILE *fp, int sockfd);

int main(int argc, char **argv)
{
        if (argc != 2)
        {
                printf("argument error\n");
                exit(0);
        }
        int sockfd;
        struct sockaddr_in servaddr;
        int status;

        bzero(&servaddr, sizeof(servaddr));
        servaddr.sin_family = AF_INET;
        servaddr.sin_port = htons(SERV_PORT);
        inet_pton(AF_INET, argv[1], &servaddr.sin_addr);//将点分十进制IP地址转化为网络字节序的二进制地址

        sockfd = socket(AF_INET, SOCK_STREAM, 0);
        if (sockfd == -1)
                err_exit("socket");
        status = connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));//连接服务器
        if (status == -1)
                err_exit("connect");
        str_cli(stdin, sockfd);
        exit(0);
}

void str_cli(FILE *fp, int sockfd)
{
        int n;
        char buf[BUFSIZE];
        struct pollfd fds[2];
        fds[0].fd = sockfd;
        fds[0].events = POLLIN;
        fds[1].fd = fileno(fp);
        fds[1].events = POLLIN;

        for (;;)
        {
                poll(fds, 2, -1);//等待套接字或标准输入就绪
                if (fds[0].revents & POLLIN)//套接字可读，说明服务器发送数据过来
                {
                        if ((n = read(sockfd, buf, BUFSIZE)) == 0)
                        {
                                printf("server terminated prematurely");
                                exit(1);
                        }
                        write(fileno(stdout), buf, n);
                }
                if (fds[1].revents & POLLIN)//标准输入可读，说明客户端有数据要写往服务器
                {
                        if ((n = read(fileno(fp), buf, BUFSIZE)) == 0)
                        {
                                shutdown(sockfd, SHUT_WR);
                                continue;
                        }
                        write(sockfd, buf, n);
                }
        }

}

server:

#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <stdlib.h>
#include <signal.h>
#include <errno.h>
#include <poll.h>

//出错函数
#define err_exit(m)\
        {\
                perror(m);\
                exit(EXIT_FAILURE);\
        }

//最大连接数
#define LISTENQ 1024

//服务器端口号
#define SERV_PORT 9877

//接收和发送的缓冲区大小
#define BUFSIZE 4096

#define OPEN_MAX 1000

//处理客户端请求函数
void str_echo(int confd);

int main(int argc, char **argv)
{
        int confd, listenfd;
        struct sockaddr_in cliaddr, servaddr;
        socklen_t clilen;
        int status;
        char buff[BUFSIZE];
        struct pollfd fds[OPEN_MAX];
        int nready;
        int i;
        ssize_t n;

        //设置协议地址结构内容
        bzero(&servaddr, sizeof(servaddr));
        servaddr.sin_family = AF_INET;
        servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
        servaddr.sin_port = htons(SERV_PORT);

        listenfd = socket(AF_INET, SOCK_STREAM, 0);//创建套接字
        if (listenfd == -1)
                err_exit("socket");
        status = bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));//将协议、IP地址、端口绑定到套接字
        if (status == -1)
                err_exit("bind");
        status = listen(listenfd, LISTENQ);//使套接字变为监听套接字
        if (status == -1)
                err_exit("listen");

        fds[0].fd = listenfd;
        fds[0].events = POLLIN;
        for (i = 1; i < OPEN_MAX; i++)
                fds[i].fd = -1;
        int max = 0;

        while (1)
        {
                nready = poll(fds, max + 1, -1);//等待描述符集中有描述符就绪
                if (fds[0].revents & POLLIN)//监听套接字可读
                {
                        clilen = sizeof(cliaddr);
                        confd = accept(listenfd, (struct sockaddr *)&cliaddr, &clilen);//等待连接完成
                        if (confd == -1)
                        {
                                if (errno = EINTR)
                                        continue ;
                                else
                                        err_exit("accept");
                        }
                        printf("connection from %s, port %d\n",
                                inet_ntop(AF_INET, &cliaddr.sin_addr, buff, sizeof(buff)),
                                ntohs(cliaddr.sin_port));
                        for (i = 1; i < OPEN_MAX; i++)
                        {
                                if (fds[i].fd < 0)
                                {
                                        fds[i].fd = confd;//将已连接套接字加入描述符集
                                        break;
                                }
                        }
                        if (i == OPEN_MAX)
                        {
                                printf("too many clients\n");
                                exit(1);
                        }
                        fds[i].events = POLLIN;
                        if (i > max)
                                max = i;
                        if (--nready <= 0)
                                continue;//只有监听套接字就绪
                }
                for (i = 1; i <= max; i++) //有已连接套接字可读,轮询处理
                {
                        if (fds[i].fd < 0)//对描述如进行测试是否就绪
                                continue;
                        n = read(fds[i].fd, buff, BUFSIZE);
                        if (n == 0)
                        {
                                close(fds[i].fd);
                                fds[i].fd = -1;
                                continue;
                        }
                        write(fds[i].fd, buff, n);
                }
        }
}

测试结果：