poll系统调用

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);//给客户端回复一个确认信息
						}
					}
				}
			}
		}

    }

}