1. 问题提出
接下来再回到IO 多路复用文中提出的问题(请允许我稍作修改):
// fd0 表示标准输入,即 0 号描述符, fd1, fd2 分别是以只读的方式打开的两个不同有名管道的描述符(a.fifo, b.fifo)
int n;
char buf[64];
int fd0 = STDIN_FILENO;
int fd1 = open("a.fifo", O_RDONLY);
int fd2 = open("b.fifo", O_RDONLY);
while(1) {
n = read(fd0, buf, 64);
write(STDOUT_FILENO, buf, n);
n = read(fd1, buf, 64);
write(STDOUT_FILENO, buf, n);
n = read(fd2, buf, 64);
write(STDOUT_FILENO, buf, n);
}
这段程序存在的问题之前我也讨论过了,只要有一个 read 发生阻塞,即使其它描述符上有数据可读,也没办法执行。
而我们之前的解决方案就是通过 I/O Multiplexing 技术,我们设想了一个 select 函数,来帮我们管理这 3 个描述符。
2. select 函数
实际上,select 函数是真实存在的,它的原型比我们之前提到的稍稍复杂:
int select(int nfds, fd_set *readfds, fd_set *writefds,
fd_set *exceptfds, struct timeval *timeout);
我知道,就算你看到这个函数你也不会觉得它有多么复杂,这多亏了在上一文中我们非常细致的描述了 fd_set 是什么玩意儿。
2.1 参数
这里的 select 函数提供的参数比较多,其中有三个参数都是 fd_set 类型,先讲这个:
- readfds,你想监听这个集合中的描述符中是否有数据可读。
- writefds,你想监听这个集合中的描述符是否可写,写 IO 当然也会发生阻塞,比如缓冲区满了。
- exceptfds,你想监听这个集合中的描述符是否发生异常,有两种情况,一种常见的是在网络编程中出现,另一种不太常见的是在伪终端中出现,这两种我们都没学过,所以暂时不去管它。
对于我们第 1 节中的问题来说,我们只关心三个描述符是否有数据可读,因此对于 writefds 和 exceptfds 都可以设置成空。
参数 nfds 表示,在传入参数的那三个集合中,最大的描述符的值 + 1. 比如 readfds 中保存了 1、3、5 号描述符,writefds 中保存了 2、3、6,exceptfds 为空,那么 nfds 就应该设置成最大的描述符的值(6) + 1,也就是 nfds = 7.
最后一个参数是等待时间,如果传空,表示永远等待,直到指定的三个集合中的描述符有事件(有数据可读、有数据可写、有异常事件发生)到来。
如果不传空值,则表示最长愿意等待多久。
它的结构如下:
struct timeval {
long tv_sec; /* 秒 */
long tv_usec; /* 微秒 */
};
如果给三个集合都设置成空,nfds = 0,只给时间参数传值,那么 select 函数就相当于一个加强版本的 sleep 函数,因为它提供的时间精度是微秒。
2.2 返回值
为什么返回值要作当独一节?主要还是有点多……
select 函数的返回值体现在两方面:
- 函数返回值
- 修改三个集合参数
2.2.1 函数返回值
对于函数返回值来说,主要有 3 种情况:
(1) 返回值 < 0,这种一般表示函数执行出错,比如使用了不可用的描述符。有时候,select 会被信号打断,也会返回 < 0 的值。因为 select 函数是不支持自动重启动的,所以被信号打断会立即返回,然后把 errno 的值设置成 EINTR.
(2) 返回值 = 0,这种只对设置了超时时间的方式才会出现,如果返回 0,表示超时时间到了,还没有事件发生。
(3) 返回值 > 0,表示监听的描述符中,有几个事件发生。也就是读集合中的事件数 + 写集合中的事件数 + 异常集合中的事件数。就算不同集合间有重复的描述符,也会累计。
2.2.2 修改参数
对于 select 函数,如何才能知道哪些描述符上发生了事件,其实在IO 多路复用一文中已经简单提到过,它主要修改三个传入的描述符集合。
如果某个集合中的描述符上有事件到来,在 select 返回时,会保留该描述符,所有其它未发生事件的描述符全部清除。
这是一种很重要的方式,因为我们必须要知道哪些描述符发生了事件,比如对于 readfds 中的描述符,我们只能去 read 发生了事件的描述符,如果不这样,就会导致阻塞。
对于超时参数来说,如果在超时时间到达前发生异常或有事件到来,则该参数会被被更新为剩余时间。
3. 实验
程序 select.c 从标准输入、a.fifo 和 b.fifo 中读数据并打印到屏幕。
程序 writepipe.c 主要向管道文件写数据。
3.1 代码
// select.c
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/select.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#define PERR(msg) do { perror(msg); exit(1); } while(0);
int process(char* prompt, int fd) {
int n;
char buf[64];
char line[64];
n = read(fd, buf, 64);
if (n < 0) {
// read 执行出错
PERR("read");
}
else if (n == 0) {
// 如果对端关闭,read 返回 0
sprintf(line, "%s closed\n", prompt);
puts(line);
return 0;
}
else if (n > 0) {
buf[n] = 0;
sprintf(line, "%s say: %s", prompt, buf);
puts(line);
}
return n;
}
int main () {
int n, res;
char buf[64];
fd_set st;
FD_ZERO(&st);
int fd0 = STDIN_FILENO;
int fd1 = open("a.fifo", O_RDONLY);
printf("open pipe: fd = %d\n", fd1);
int fd2 = open("b.fifo", O_RDONLY);
printf("open pipe: fd = %d\n", fd2);
FD_SET(fd0, &st);
FD_SET(fd1, &st);
FD_SET(fd2, &st);
// 最后一个 open 的描述符值是最大的
int maxfd = fd2 + 1;
while(1) {
// 因为 tmpset 参数会被 select 修改,所以要重新赋值。
fd_set tmpset = st;
res = select(maxfd, &tmpset, NULL, NULL, NULL);
if (res < 0) {
// select 执行出错,对于被信号中断的,需要单独处理,这里暂时不考虑,后面的文章会讲
PERR("select");
}
else if (res == 0) {
// 超时,先不用管
continue;
}
// 判断返回的集合中是否包含对应的描述符,如果包含,说明的事件(可读)到来。
if (FD_ISSET(fd0, &tmpset)) {
n = process("fd0", fd0);
// 如果返回值为 0,表示对端关闭,后面的也一样。
if (n == 0) FD_CLR(fd0, &st);
}
if (FD_ISSET(fd1, &tmpset)) {
n = process("fd1", fd1);
if (n == 0) FD_CLR(fd1, &st);
}
if (FD_ISSET(fd2, &tmpset)) {
n = process("fd2", fd2);
if (n == 0) FD_CLR(fd2, &st);
}
}
}
3.2 writepipe 程序
writepipe 程序主要向管道文件写数据。它从命令行接收管道文件的名字。
// writepipe.c
#include <unistd.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
int main(int argc, char* argv[]) {
if (argc < 2) {
printf("Usage: %s <fifoname>\n", argv[0]);
return 1;
}
char buf[64];
int n;
int fd = open(argv[1], O_WRONLY);
if (fd < 0) {
perror("open pipe");
return 1;
}
while(1) {
n = read(STDIN_FILENO, buf, 64);
write(fd, buf, n);
}
return 0;
}
3.3 编译和运行
- 编译
$ gcc select.c -o select
$ gcc writepipe.c -o writepipe
- 运行
先创建两个管道文件:
$ mkfifo a.fifo
$ mkfifo b.fifo
打开三个终端,分别运行:
$ ./select
$ ./writepipe a.fifo
$ ./writepipe b.fifo
图1 运行结果
4. 总结
- 掌握 select 的用法
实际上,在网络编程中,select 用来同时处理多个连接也是一种常用的手段。
练习:完成本文中的实验。