about summary refs log tree commit diff
path: root/third_party/nix/src/libutil/monitor-fd.hh
blob: 54e9de848a13dbb71f3134464883142d30c58196 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#pragma once

#include <atomic>
#include <cstdlib>
#include <thread>

#include <poll.h>
#include <signal.h>
#include <sys/types.h>
#include <unistd.h>

namespace nix {

class MonitorFdHup {
 private:
  std::thread thread;

 public:
  MonitorFdHup(int fd) {
    thread = std::thread([fd]() {
      while (true) {
        /* Wait indefinitely until a POLLHUP occurs. */
        struct pollfd fds[1];
        fds[0].fd = fd;
        /* This shouldn't be necessary, but macOS doesn't seem to
           like a zeroed out events field.
           See rdar://37537852.
        */
        fds[0].events = POLLHUP;
        auto count = poll(fds, 1, -1);
        if (count == -1) abort();  // can't happen
        /* This shouldn't happen, but can on macOS due to a bug.
           See rdar://37550628.

           This may eventually need a delay or further
           coordination with the main thread if spinning proves
           too harmful.
         */
        if (count == 0) continue;
        assert(fds[0].revents & POLLHUP);
        triggerInterrupt();
        break;
      }
    });
  };

  ~MonitorFdHup() {
    pthread_cancel(thread.native_handle());
    thread.join();
  }
};

}  // namespace nix