时间戳

首页 > 解决方案 > 使用 `poll()` 检查阻塞

c - 使用 `poll()` 检查阻塞

问题描述

我想检查文件描述符是否会阻止某个事件，并且我想到了使用poll()0 超时：

int wouldblock(int fd, short event)
{
  struct pollfd pfd;
  pfd.fd = fd;
  pfd.events = event;
  return (poll(&pfd,1,0) == 0);
}

...
  if (wouldblock(0,POLLIN)) ...
...

流可用，并且 poll() 应该返回 1，或者它会阻塞并且超时将返回 0。（让我们暂时搁置错误检查）。

它有效（至少“它在我的机器上有效”）但我想知道我是否错过了什么？也许poll()是矫枉过正，我对系统压力太大了？

标签： cposix

解决方案

我想知道我是否错过了什么？

poll可以立即返回，操作仍然可以阻塞。poll()这可能会发生，因为从返回和开始操作之间有时间。当一个事件发生改变连接到文件描述符的事物的状态时，就会发生这种情况，并且该事件发生在您的进程返回poll()并即将开始操作之后。

最常见的情况是，两个进程同时从一个管道poll()ing 和read()ing - 可能会同时poll返回，但只有一个read获胜。

#include <unistd.h>
#include <stdio.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <poll.h>
#include <sys/wait.h>
int main(void)
{
    system("mkfifo /tmp/fifo ; echo > /tmp/fifo & sleep 0.1");
    fork();
    int fd = open("/tmp/fifo", O_RDWR);
    sleep(1);
    printf("poll=%d\n", (int)poll(&(struct pollfd){fd, POLLIN}, 1, 0));
    sleep(1);
    printf("read=%d\n", (int)read(fd, (char[1]){}, 1));
    printf("%d exiting!\n", (int)getppid());
    wait(0);
}

有足够的运气输出：

poll=1
poll=1
read=1
136769 exiting!

另一个fork()ed 进程正在等待read。

我记得在内核邮件列表上也看到了一个线程，该线程具有丢弃旧网络数据包以防止停止的功能。有了这样的功能，程序可以：poll()在网络套接字上，然后内核决定丢弃数据包，然后进程read()只会发现数据包不再存在。

poll()用作“通知”系统 -poll()返回文件描述符，如果它们有可读取的内容，您应该“检查”它们。然后，您应该通过调用来检查“真实”是否有要阅读的内容read()。

在实际应用程序中，我会将“wouldblock”函数向上移动一层，而不是检查文件描述符是否会阻塞，而是检查是否收到了我正在使用的任何协议中的完整逻辑数据包。调用它has_received_a_packet会用read()with接收数据O_NONBLOCK，在缓冲区中累积并检查是否接收到完整的数据包。

也许 poll() 太过分了，我对系统压力太大了？

我相信是这样，您可以直接打电话read并当场接收该数据。类似于如何ungetc实现：

struct buf {
   char data;
   bool hasit;
};
int readbuf_has_something(struct buf *readbuf, int fd) {
    if (!readbuf->hasit) {
        set_nonblock(fd);
        ssize_t r = read(fd, &readbuf->data, 1);
        if (r == 1) readbuf->hasit = 1;
        if (ret < 0) return -EIO;
    }
    return readbuf->hasit;
}
int readbuf_something(struct buf *readbuf, int fd) {
    if (readbuf->hasit) {
      readbuf->hasit = 0;
      return readbuf.data;
    }
    set_block(fd);
    ssize_t r = read(fd, &readbuf->data, 1);
    if (r == 1) return readbuf->data;
    return EOF;
}

这样的实现将导致更少的上下文更改，然后使用poll后跟read- 只需读取数据，它们就在那里。

推荐阅读