poll系统调用的用途:与select类似,也是在指定时间内轮询一定数目的文件描述符,以测试其中是否有就绪的文件描述符。
poll系统调用函数的定义如下:
#include <poll.h>
int poll(struct pollfd* fds, nfds_t nfds, int timeout)
函数的作用是:此函数成功时,返回就绪文件描述符的总数,失败返回-1并设置errno。如果在超时时间内没有任何文件描述符就绪,将返回0;
参数nfds:指定被监听事件 fds 的大小,其类型 nfds_t 的定义如下:
typedef unsigned long int nfds_t;
参数timeout:指定poll的超时值,单位是毫秒。当timeout为-1时,poll调用将永远阻塞,直到某个事件发生;当timeout为0时,poll调用将立即返回。
参数fds: 是一个pollfd结构类型的数组,它指定所有我们感兴趣的文件描述符上发生的可读、可写和异常等事件。
pollfd结构体的定义如下:
struct pollfd
{
int fd; //文件描述符
short events; //注册的事件
short revents; //实际发生的事件,由内核填充
};
fd成员指定文件描述符;
events成员告诉poll监听fd上的那些事件;
revents成员由内核修改,用来通知应用程序fd上实际发生了那些事件。
初始化事件集中的数据:
void fds_init(struct pollfd fds[])
{
int i = 0;
for(;i < MAXFD;i++)
{
fds[i].fd = -1;
fds[i].events = 0;
fds[i].revents = 0;
}
}
添加一个事件集:
void fds_add(struct pollfd fds[],int fd)
{
int i = 0;
for(;i < MAXFD ;++i)
{
if(fds[i].fd == -1)
{
fds[i].fd = fd;
fds[i].events = POLLIN;
break;
}
}
}
删除一个事件集:
void fds_del(struct pollfd fds[],int fd)
{
int i = 0;
for(;i < MAXFD;++i)
{
if(fds[i].fd == fd)
{
fds[i].fd = -1;
fds[i].events = 0;
break;
}
}
}
完整代码如下:
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <assert.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/time.h>
#include <poll.h>
#define MAXFD 10
//初始化事件集中的数据
void fds_init(struct pollfd fds[])
{
int i = 0;
for(;i < MAXFD;i++)
{
fds[i].fd = -1;
fds[i].events = 0;
fds[i].revents = 0;
}
}
//添加一个事件集
void fds_add(struct pollfd fds[],int fd)
{
int i = 0;
for(;i < MAXFD ;++i)
{
if(fds[i].fd == -1)
{
fds[i].fd = fd;
fds[i].events = POLLIN;
break;
}
}
}
//删除一个事件集
void fds_del(struct pollfd fds[],int fd)
{
int i = 0;
for(;i < MAXFD;++i)
{
if(fds[i].fd == fd)
{
fds[i].fd = -1;
fds[i].events = 0;
break;
}
}
}
int main()
{
int sockfd = socket(AF_INET,SOCK_STREAM,0);//创建一个套接字
assert( sockfd != -1);
struct sockaddr_in saddr,caddr;//使用统一的TCP/IPV4协议族 socket地址
memset(&saddr,0,sizeof(saddr));
saddr.sin_family = AF_INET; //地址族
saddr.sin_port = htons(6000); //端口号
saddr.sin_addr.s_addr = inet_addr("127.0.0.1");
//将点分十进制字符串表示的IPv4地址转换成用网络字节序整数表示的ipv4地址。
int res = bind(sockfd,(struct sockaddr*)&saddr,sizeof(saddr));//命名套接字
assert(res != -1);
listen(sockfd,5);//创建监听队列
struct pollfd fds[MAXFD]; //创建一个事件集合
fds_init(fds); //初始化事件集
fds_add(fds,sockfd); //添加一个事件集
while( 1 )
{
int n = poll(fds,MAXFD,2000); // 调用poll,返回就绪的文件描述符总数
if(n == -1)//返回-1,poll函数调用失败
{
perror("poll error");
continue;
}
if(n == 0)//返回0,在超时时间内没有就绪的文件描述符
{
printf("time out\n");
continue;
}
else //有就绪的文件描述符
{
int i = 0;
for(;i < MAXFD;i++)//判断是那个文件描述符就绪
{
if(fds[i].fd == -1) //不是继续寻找下一个
{
continue;
}
if(fds[i].revents & POLLIN)//判断是否可读数据
{
if(sockfd == fds[i].fd) //如果是
{
int len = sizeof(caddr);
int c = accept(sockfd,(struct sockaddr*)&caddr,&len);
//接收一个连接,产生另外一个文件描述符,保持这个连接
if(c < 0)
{
continue;
}
fds_add(fds,c);
printf("accept c = %d\n",c);
}
else //如果是其他继续的文件描述符
{
char buff[128] = {0};
int num = recv(fds[i].fd,buff,127,0); //接收数据
if(num <= 0)
{
close(fds[i].fd); //关闭一个连接
fds_del(fds,fds[i].fd); //把这个文件描述符从事件集中删除
printf("one client over\n");
}
else
{
printf("recv(%d)=%s\n",fds[i].fd,buff);
send(fds[i].fd,"OK",2,0);//给客户端回复一个确认信息
}
}
}
}
}
}
}