非阻塞获取字符

Question

• 平台：Linux 3.2.0 x86 (Debian 7)
• 编译器：GCC 4.7.2 (Debian 4.7.2-5)

如果字符已经存在于标准输入中，我正在编写一个从标准输入读取单个字符的函数。如果 stdin 为空，则该函数假定什么都不做并返回 -1。我搜索了非阻塞输入，并被指向poll() 或select()。首先我尝试使用 select() 但我无法让它工作，所以我尝试了 poll() 并得出了相同的结论。我不确定这些函数究竟做了什么，但根据我对 poll() 文档的理解，如果我这样称呼它：

struct pollfd pollfds;
pollfds = STDIN_FILENO;
pollfds.events = POLLIN;
poll(pollfds, 1, 0);

if(pollfds.revents & POLLIN) 如果“可以不阻塞地读取高优先级数据以外的数据”，则为真。但是 poll() 在我的测试情况下总是超时。我如何测试功能可能是问题，但我想要的功能正是我正在测试的。这是目前的功能和测试情况。

#include <poll.h>
#include <stdio.h>
#include <unistd.h>

int ngetc(char *c)
{       
    struct pollfd pollfds;
    pollfds.fd = STDIN_FILENO;
    pollfds.events = POLLIN;

    poll(&pollfds, 1, 0);

    if(pollfds.revents & POLLIN)
    {
            //Bonus points to the persons that can tell me if
            //read() will change the value of '*c' if an error
            //occurs during the read
        read(STDIN_FILENO, c, 1);
            return 0;
    }
    else return -1;
}

//Test Situation:
//Try to read a character left in stdin by an fgets() call
int main()
{
    int ret = 0;
    char c = 0;
    char str[256];

    //Make sure to enter more than 2 characters so that the excess
    //is left in stdin by fgets()
    fgets(str, 2, stdin);

    ret = ngetc(&c);

    printf("ret = %i\nc = %c\n", ret, c);

    return 0;
}

Answer 1

您的 IO 操作不正确，POSIX 手册和所有其他相关文档都明确表示永远不要混合在 FILE *s 和文件描述符上完成的 IO。你已经非常公然地违反了这条规则。这条规则是存在的，因为FILE *s 使用 buffering 这意味着在调用fgets 之后，read 将没有任何东西可以获取，因为fgets 已经读取了所有待处理的数据放入保存在FILE * 结构中的缓冲区中。

因此，由于无法检查 ISO C IO 方法是否会阻塞，我们只能使用文件描述符。

既然我们知道STDIN_FILENO只是数字0，我们可以使用

fcntl (0, F_SETFL, O_NONBLOCK);

这会将文件描述符 0 上的所有 reads 转换为非阻塞模式，如果您想使用不同的文件描述符以便可以不理会 0，那么只需使用 dup 复制它。

这样你就可以完全远离poll，实现ngetc为

ssize_t 
ngetc (char *c)
{
  return read (0, c, 1);
}

或者更好的是，一个宏

#define ngetc(c) (read (0, (c), 1))

这样您就可以得到一个简单的实现来满足您的需求。

编辑：如果您仍然担心终端缓冲输入，您可以随时更改终端的设置，请参阅How to disable line buffering of input in xterm from program? 了解有关如何执行此操作的更多信息。

编辑：不能使用fgetc 代替read 的原因与使用fgets 不起作用的原因相同。当FILE * IO 函数之一运行时，它会从关联的文件描述符中读取所有数据。但是一旦发生这种情况，poll 将永远不会返回，因为它正在等待一个始终为空的文件描述符，read 也会发生同样的事情。因此，我建议您遵循文档的建议，永远不要混合 streams（IO 使用 fgets、fgetc 等）和 文件描述符（IO使用read、write等）

Answer 2

你的代码有两个问题。

1. 根据poll的manual，给超时赋值0会立即返回

   如果 timeout 的值为 0，poll() 将立即返回。如果 timeout 的值为 -1，poll() 将阻塞直到请求的事件发生或直到调用被中断。

2. fgets 不符合您的预期，它来自 stdio 库并将缓冲读取。假设你输入了 3 个字母并回车，fgets 之后的第三个字母将无法提供给poll。

所以注释掉 fgets 行并将 -1 分配给 poll 中的超时，然后再次运行它以查看是否是您想要的。

Answer 3

我没有得到上述答案的预期行为，实际上我还必须考虑this answer 它将 TTY 设置为非规范模式。

#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <termios.h>

int main(int argc, char *argv[]) 
{
    struct termios t;
    tcgetattr(0, &t);
    t.c_lflag &= ~ICANON;
    tcsetattr(0, TCSANOW, &t);

    fcntl(0, F_SETFL, fcntl(0, F_GETFL) | O_NONBLOCK);

    printf("Starting loop (press i or q)...\n");

    for (int i = 0; ; i++) {
        char c = 0;
        read (0, &c, 1);

        switch (c) {
        case 'i':
            printf("\niteration: %d\n", i);
            break;

        case 'q':
            printf("\n");
            exit(0);
        }
    }

    return 0; 
}