IO复用：poll函数

   poll提供的功能与select函数类似，不过在处理流设备时，它能够提供额外的信息 。

   #include<poll.h>

   int  poll(struct pollfd *fdarray，unsigned long nfds，int  timeout)；

       返回：若有就绪的描述符则为其数目，若超时则为0，若出错则为-1；

  第一个参数是指向一个结构第一个元素的指针。每个数组元素都是一个pollfd结构，用于指定测试某个给定的描述符fd的条件。

   struct pollfd {

       int      fd;

       short  events;

       short  revents;

  };

  要测试的条件由events成员指定，函数在相应的revents成员中返回该描述符的状态。在下表中给出一些常值：




    我们将该图分为三个部分：第一个部分是处理输入的四个常值，第二个部分是处理输出的三个常值，第三个部分是处理错误的三个常值。其中，第三部分的三个常值不能在events中设置，但是当相应条件存在时就在revents中返回。

    timeout参数指定poll函数返回前等待多长时间。它是一个指定应等待毫秒数的正值。





  INFTIM常值定义为一个负值，其定义在头文件#include<poll.h>，不过很多系统把它定义在头文件#include<sys/stropts.h>中。

   当发生错误时，poll函数的返回值为-1，若定时器到时之前没有任何描述符就绪，则返回0，否则返回就绪描述符的个数，即revents成员值非0的描述符个数。

   下面给出一个用poll实现的例子，其功能和之前的博客对于select函数实现的功能一样，客户端给服务器发送消息，服务器将消息回射到客户端。。。

服务器：
#include"../util.h"

int sock_bind(const char *ip,short port)
{
int sockserfd;
//创建套接字
sockserfd = socket(AF_INET,SOCK_STREAM,0);
if(sockserfd == -1)
perror("socket");
struct sockaddr_in addrSer;
addrSer.sin_family = AF_INET;
addrSer.sin_port = htons(port);
addrSer.sin_addr.s_addr = inet_addr(ip);

int res;
//绑定
res = bind(sockserfd,(struct sockaddr *)&addrSer,sizeof(struct sockaddr));

return sockserfd;
}

void handle_connection(struct pollfd *clientfds,int maxi)
{
int n;
char buf[256];
for(int i = 1; i <= maxi; ++i)
{
if(clientfds[i].fd == -1)
continue;
//当客户端的描述符状态为可读时
if(clientfds[i].revents & POLLIN)
{
//接收客户端的数据
n = recv(clientfds[i].fd,buf,256,0);
//当n小于0时说明客户端关闭
if(n <= 0)
{
close(clientfds[i].fd);
clientfds[i].fd = -1;
continue;
}
printf("recv msg:>%s\n",buf);
//将接收到的数据又进行发送
send(clientfds[i].fd,buf,n,0);

}

}

}
void do_poll(int sockfd)
{
pollfd clientfds[OPEN_SIZE];
//0下标为服务器
//将服务器的测试条件设置为可读
clientfds[0].fd = sockfd;
clientfds[0].events = POLLIN;

//初始化描述符
for(int i = 1; i < OPEN_SIZE; ++i)
clientfds[i].fd = -1;

int maxi = 0;
int nready;
int connfd;
int i = 0;
struct sockaddr_in addrCli;
for(;;)
{
//nready表示就绪的描述符
nready = poll(clientfds,maxi + 1,-1);
if(nready == -1)
{
perror("poll");
exit(1);
}
//当服务器的返回事件为可读
if(clientfds[0].revents & POLLIN)
{
//新客户端建立连接
socklen_t addrlen = sizeof(struct sockaddr);
connfd = accept(sockfd,(struct sockaddr *)&addrCli,&addrlen);

if(connfd == -1)
{
perror("accept");
continue;
}
printf("accept a new client :%s:%d\n",inet_ntoa(addrCli.sin_addr),addrCli.sin_port);
for(i = 1 ; i < OPEN_SIZE; ++i)
{
if(clientfds[i].fd < 0)
{
clientfds[i].fd = connfd;
break;
}
}
if(i == OPEN_SIZE)
{
printf("too many clients");
exit(1);
}
//将客户端的测试条件设置为可读
clientfds[i].events = POLLIN;
if(i > maxi)
maxi = i;
if(--nready <= 0)
continue;
}
handle_connection(clientfds,maxi);

}
}
int main(int argc,char*argv[])
{
int sockSer;
sockSer = sock_bind(IPADDR,PORT);
listen(sockSer,LISTENQ);
do_poll(sockSer);
return 0;
}

客户端：

#include"../util.h"

void handle_connection(int sockfd)
{
pollfd fds[2];
fds[0].fd = sockfd;
fds[0].events = POLLIN;
fds[1].fd = STDIN_FILENO;
fds[1].events = POLLIN;

int n;
char buf[256];
for(;;)
{
poll(fds,2,-1);
//当服务器可读时
if(fds[0].revents & POLLIN)
{
//接收服务器的消息
n = recv(sockfd,buf,256,0);
//如果n等于0，说明服务器关闭
if(n <= 0)
{
printf("server is closed.\n");
close(sockfd);
}
//将buf的内容显示在屏幕上
write(STDIN_FILENO,buf,n);
}
//当标准输入输出可读时
if(fds[1].revents & POLLIN)
{
//将屏幕的内容读到buf中
n = read(STDIN_FILENO,buf,256);
//当n等于0时，继续循环
if(n == 0)
continue;
//向服务器发送数据
write(sockfd,buf,n);
}
}
}

int main(int argc,char*argv[])
{
int sockCli;
//创建套接字
sockCli = socket(AF_INET,SOCK_STREAM,0);
if(sockCli == -1)
perror("sockCli");

struct sockaddr_in addrSer;
socklen_t addrlen;
addrSer.sin_family = AF_INET;
addrSer.sin_port = htons(PORT);
addrSer.sin_addr.s_addr = inet_addr(IPADDR);

addrlen = sizeof(struct sockaddr);
//客户端进行连接
connect(sockCli,(struct sockaddr *)&addrSer,addrlen);

handle_connection(sockCli);
return 0;
}

公用头文件：

#pragma once
#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<poll.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#include<string.h>

#define IPADDR "127.0.0.1"
#define PORT 8787
#define MAXLINE 1024
#define LISTENQ 5

#define SIZE 10
#define OPEN_SIZE 5

程序的执行结果和利用select函数的执行结果是一样的。。。