写在前面
1. 本文内容对应《 UNIX 环境高级编程》 ( 第 2 版 ) 》第 14 章。
2. 总结了 IO 多路转接的概念,以及 selec t 、pselect 和poll 函数的用法 。
3. 希望本文对您有所帮助,也欢迎您给我提意见和建议。
IO
多路转接
当从一个文件描述符读,然后写到另一个文件描述符,可以在下列形式的循环中使用阻塞 IO 。
while ((n = read(STDIN_FILENO, buf, BUFSIZ)) > 0) if (write(STDOUT_FILENO, buf, n) != n) err_sys("write error"); |
但是,如果必须从两个描述符读,那么就可能长时间阻塞在一个描述符上,而另一个描述符虽然有很多数据却不能得到及时处理。
IO 多路转接( I/O Multiplexing ),先构造一张有关描述符的列表,然后调用一个函数,直到这些描述符中的一个已准备好进行 I/O 时,该函数才返回。在返回时,它告诉进程哪些描述符已准备好可以进行 I/O 。 POSIX.1 标准定义了 select 和 pselect 函数,而 poll 则是对该基本部分的 XSI 扩展。
select
#include <sys/select.h> int select(int maxfdp1, fd_set *restrict readfds, fd_set *restrict writefds, fd_set *restrict exceptfds, struct timeval *restrict tvptr); Returns: count of ready descriptors, 0 on timeout, -1 on error |
最后一个参数 tvptr ,指定愿意等待的时间:
l tvptr==NULL ,永久等待。当所指定的描述符中的一个已经准备好或捕捉到一个信号则返回。如果捕捉到一个信号,则 select 返回 -1 , errno 设置为 EINTR 。
l tvptr->tv_sec==0 && tvptr->tv_usec==0 ,完全不等待。这是得到多个描述符状态而不阻塞 select 函数的轮询方法。
l tvptr->tv_sec!=0 || tvptr->tv_usec!=0 ,等待指定的秒数和微秒数。当指定的描述符之一已准备好,或当指定的时间值已经超时,或捕捉到信号时,函数返回。在 linux 下,若在该时间值尚未超过时 select 就返回,那么将用余留时间更新 tvptr 指向的结构。
中间三个参数 readfd s 、writefds 和exceptfds 是指向描述符集的指针。这三个描述符集说明了我们关心的可读、可写或出于异常条件的各个描述符。每个描述符集存放在一个fd_set 数据类型中。这种数据类型为每一可能的描述符保持一位。描述符集的函数接口(可能实现为宏)包括:调用FD_ZERO 将一个指定的fd_set 变量的所有位设置为0 ;调用FD_SET 设置一个fd_set 变量的指定位;调用FD_CLR 将一指定位清楚;调用FD_ISSET 测试一指定位是否设置。
#include <sys/select.h> int FD_ISSET(int fd, fd_set *fdset); Returns: nonzero if fd is in set, 0 otherwise void FD_CLR(int fd, fd_set *fdset); void FD_SET(int fd, fd_set *fdset); void FD_ZERO(fd_set *fdset); |
select 的第一个参数 maxfdp1 的意思是最大描述符加 1 ,即在三个描述符集中找出最大描述符编号值,然后加 1 。
select 有三个可能的返回值:
l 返回值 -1 表示出错。如捕捉到一个信号。在这种情况下,将不修改其中任何描述符集。
l 返回值 0 表示没有描述符准备好。
l 正返回值表示已经准备好的描述符数,该值是三个描述符集中已经准备好的描述符数之和。
对于准备好的意思是,对读写集中的一个描述符的 read 或 write 操作不会阻塞,对异常状态集中的一个描述符有一个未决异常状态(包括 1 )在网络连接上到达的带外数据 2 )在处于数据包模式的伪终端上发生了某些状态)。如果在一个描述符上碰到了文件结尾处,则 select 认为该描述符是可读的,因为调用 read 将返回 0 。
pselect
#include <sys/select.h> int pselect(int maxfdp1, fd_set *restrict readfds, fd_set *restrict writefds, fd_set *restrict exceptfds, const struct timespec *restrict tsptr, const sigset_t *restrict sigmask); Returns: count of ready descriptors, 0 on timeout, -1 on error |
它与 select 的区别在于:
l pselect 使用 timespec 结构指定超时值。 timespec 结构以秒和纳秒表示时间,而非秒和微秒。
l pselect 的超时值被声明为 const ,这保证了调用 pselect 不会改变 timespec 结构。
l pselect 可使用一个可选择的信号屏蔽字。在调用 pselect 时,以原子操作的方式安装该信号屏蔽字,在返回时恢复以前的信号屏蔽字。
poll
#include <poll.h> int poll(struct pollfd fdarray[], nfds_t nfds, int timeout); Returns: count of ready descriptors, 0 on timeout, 1 on error struct pollfd { int fd; /* file descriptor to check, or <0 to ignore */ short events; /* events of interest on fd */ short revents; /* events that occurred on fd */ }; |
与 select 不同, poll 不是为每个状态构造一个描述符集,而是构造一个 pollfd 结构数组,每个数组元素指定一个描述符编号以及对其所关心的状态。 pollfd 结构中的 events 告诉内核我们队该描述符关心的是什么。 poll 函数返回时,内核设置 revents 成员,以说明对该描述符已经发生了什么事件。 fdarray 数组中的元素数由 nfds 说明。 timeout 参数说明愿意等待的时间: 1 ) timeout==-1 ,永久等待; 2 ) timeout==0 ,不等待; 3 ) timeout>0 ,等待 timeout 毫秒。
实验程序如下:
#include <stdio.h> #include <stdlib.h> #include <fcntl.h> #include <sys/select.h> #include <poll.h> #include "apue.h"
int main() { fd_set rfds, wfds; int fd, result; char buf[10]; struct pollfd fds[2];
if((fd = open("tempselect", O_CREAT | O_WRONLY, S_IRUSR | S_IWUSR)) < 0) err_sys("open tempselect error"); FD_ZERO(&rfds); FD_ZERO(&wfds); FD_SET(fd, &rfds); FD_SET(STDIN_FILENO, &wfds); if((result = select(fd+1, &rfds, &wfds, NULL, NULL)) == -1) perror("select error"); else if(result == 0) printf("no fd ready/n"); else { printf("%d fd(s) ready/n", result); if(FD_ISSET(fd, &rfds)) printf("fd is ready for read/n"); if(read(fd, buf, 10) < 0) perror("read fd error"); if(FD_ISSET(STDIN_FILENO, &wfds)) printf("STDIN is ready for write/n"); }
fds[0].fd = fd; fds[0].events = POLLIN; fds[1].fd = STDIN_FILENO; fds[1].events = POLLOUT; if((result = poll(fds, 2, -1)) == -1) perror("poll error"); else { printf("%d fd(s) ready/n", result); if(fds[0].revents == POLLIN) printf("fd is ready for read/n"); if(read(fd, buf, 10) < 0) perror("read fd error"); if(fds[1].revents == POLLOUT) printf("STDIN is ready for write/n"); } exit(0); } |
运行结果如下。可以看到 select 和 poll 函数只负责确定 read 和 write 操作是否会被阻塞,而不进行文件访问权限检查。
pydeng@pydeng-laptop:~/apue.2e/mytest$ ./a.out 2 fd(s) ready fd is ready for read read fd error: Bad file descriptor STDIN is ready for write 2 fd(s) ready fd is ready for read read fd error: Bad file descriptor STDIN is ready for write |