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函数的执行结果是一样的。。。