[apue] 使用 poll 检测管道断开

海海

发布于 2022-08-19 10:15:06

7590

发布于 2022-08-19 10:15:06

文章被收录于专栏：goodcitizen

一般使用 poll 检测 socket 或标准输入时，只要指定 POLLIN 标志位，就可以检测是否有数据到达，或者连接断开：

 1 struct pollfd fds[3];
 2 fds[0].fd = STDIN_FILENO;
 3 fds[0].events = POLLIN; 
 4 fds[1].fd = sock_fd; 
 5 fds[1].events = POLLIN; 
 6 fds[2].fd = pipe_fd; 
 7 fds[2].events = POLLIN; 
 8 ret = poll(fds, 3, -1);
 9 if (ret > 0) { 
10     if (fds[0].revents & POLLIN) {
11         // handle stdin
12         ...
13     }
14     if (fds[1].revents & POLLIN) { 
15         // handle socket input
16         ...
17     }
18     if (fds[2].revents & POLLIN) { 
19         // handle pipe input
20         ...
21     }        
22 }

当 read 结果返回 0 时表示相应连接断开。

而对于 pipe，只检测POLLIN是感知不到管道断开的，当管道断开时，会在revents设置POLLHUP，必需额外检测此标志位：

1 if (pfd[2].revents & POLLHUP) {
2       // handle pipe break
3       ...
4 }

而当 poll 一个已经关闭的句柄时（句柄号 >=0 有效），poll 本身并不返回错误，而是给对应的句柄事件中设置 POLLNVAL 标志位：

1 if (pfd[2].revents & POLLNVAL) {
2       // handle pipe close
3       ...
4 }

若 poll 一个无效句柄时（句柄号为-1），poll 本身仍不返回错误，但该句柄一定没有任何事件可供检测与返回。因此可用于占位处理，

例如固定从数组某个下标中取出某个句柄时可以在不相关位置设置-1句柄，这样就不用再去判断当前有事件的句柄的源句柄是哪一个了：

1 struct pollfd fds[3];
2 fds[0].fd = STDIN_FILENO;
3 fds[0].events = POLLIN; 
4 fds[1].fd = -1; 
5 fds[1].events = POLLIN; 
6 fds[2].fd = pipe_fd; 
7 fds[2].events = POLLIN; 
8 ret = poll(fds, 3, -1);
9 ……

例如当没有 socket 句柄时，该位置保持-1，这样可以不用将管道句柄上移，从而可以固定从fds[2]中取出管道句柄。

当然如果传入 poll 的句柄数组中所有句柄都为无效句柄时，poll仍不返回错误，此时若提供超时，可当成sleep使用；

若不提供超时，则会进入无限期等待……

测试代码

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019-06-12，如有侵权请联系 cloudcommunity@tencent.com 删除

socket编程