如何检查标准输入是否为空且没有挂起？

Question

我希望我的程序从命令行选项或标准输入（重定向）读取数据。那么如何检查stdin是否为空呢？

如果stdin 为空（等待输入），则执行fgetc 会挂起我的程序。调用 feof 返回 false。

也试过了

ioctl(0, I_NREAD, &n) == 0 && n > 0

但我没有定义 I_NREAD（在 Raspi 上，但我想要便携）。

如何实现？

Answer 1

select 函数可以在 POSIX 系统上完成。在 Windows 或其他系统上，您需要其他选项。

请注意，如果 shell 设置了管道并且操作系统安排子进程首先运行(first | child)，它可能会执行选择并且找不到任何准备读取的输入。

请注意，我将超时从 0 更改为 100,000 微秒（0.1 秒），这有助于避免随机丢失输入。但是，如果系统负载过重，它仍然可能无法检测到标准输入已准备好读取。

见：

#include <stdio.h>
#include <sys/select.h>

int main() {
  fd_set read_fds;
  struct timeval tv = {0, 100000};
  int fd = fileno(stdin);
  int r;

  FD_ZERO(&read_fds);
  FD_SET(fd, &read_fds);
  r = select(fd + 1, &read_fds, NULL, NULL, &tv);
  if (r > 0) {
    char buffer[256];
    if (fgets(buffer, 256, stdin)) {
      puts(buffer);
    }
  }
  return 0;
}