I/O复用——poll函数

poll() 与 select() 类似，与 select() 在本质上没有多大差别，管理多个文件描述符也是进行时间轮询，根据描述符的状态进行处理，但是 poll()没有最大文件描述符数量的限制（但是数量过大后性能也是会下降）。poll() 和 select() 同样存在一个缺点就是，包含大量文件描述符的数组被整体复制于用户态和内核的地址空间之间，而不论这些文件描述符是否就绪，它的开销随着文件描述符数量的增加而线性增大。

1、函数原型：

头文件：

#include <poll.h>

int poll(struct pollfd *fds, nfds_t nfds, int timeout);

2、函数参数

fds：指向一个结构体数组的第0个元素的指针，每个数组元素都是一个struct pollfd结构，用于指定测试某个给定的fd的条件。

struct pollfd{

int fd; //文件描述符

short events; //等待的事件

short revents; //实际发生的事件

};

fd：每一个 pollfd 结构体指定了一个被监视的文件描述符，可以传递多个结构体，指示 poll() 监视多个文件描述符。

events/revents:要测试的条件由 events 成员指定，函数在相应的 revents 成员中返回该描述符的状态(每个描述符都有两个变量，一个为调用值，另一个为返回结果，从而避免值-结果参数)这两个成员中的每一个都由指定某个特定条件的一位或多位组合而成。下标列出指定 events 标志以及测试 revents 标志的一些常值。

nfds:用来指定第一个参数数组元素个数

timeout: 指定等待的毫秒数

3、函数的返回值

成功时，poll() 返回结构体中 revents 域不为 0 的文件描述符个数；如果在超时前没有任何事件发生，poll()返回 0；失败时，poll() 返回 -1。

poll.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <assert.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
 
#define MAXFD 10

int fds_init(struct pollfd *fds)
{
	int i = 0;
	for(;i < MAXFD;i++)
	{
		fds[i].fd = -1;
		fds[i].events = 0;
	}
}
 
int 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;//注册读事件
			fds[i].revents = 0;
			break;
		}
	}
}
 
int 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;
	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");
 
	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 a = poll(fds,MAXFD,5000);
		if(a < 0)
		{
			perror("poll error");
			continue;
		}
		if(a > 0)
		{
			int i = 0;
			for(;i < MAXFD;i++)
			{
				if(fds[i].revents & POLLIN)
				{
					if(fds[i].fd == sockfd)
					{
						struct sockaddr caddr;
						int len = sizeof(caddr);
						int c = accept(sockfd,&caddr,&len);
						printf("accept(%d)\n",c);
						fds_add(fds,c);			
					}
					else
					{
						char buff[128] = {0};
						int n = recv(fds[i].fd,buff,127,0);
						if(n <= 0)
						{
							printf("one client over\n");
							close(fds[i].fd);
							fds_del(fds,fds[i].fd);	
						}
						else
						{
							printf("buff(%d) = %s\n",fds[i].fd,buff);
							send(fds[i].fd,"OK",2,0);
						}
					}
	
				}
			}	
		}
		
	}
	
	return 0;
}

运行结果：

I/O复用——poll函数

猜你喜欢