aboutsummaryrefslogtreecommitdiff
path: root/src/libutil/monitor-fd.hh
blob: 5ee0b88ef50f577804c58e44a005d91f94a2e5d6 (plain)
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
54
55
56
57
58
#pragma once

#include <thread>
#include <atomic>

#include <cstdlib>
#include <poll.h>
#include <sys/types.h>
#include <unistd.h>
#include <signal.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();
    }
};


}