src/util.cpp


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

#include "util.hpp"

#include <cerrno>
#include <cstdarg>
#include <cstdio>
#include <cstdlib>
#include <ctime>

#include <unistd.h>

void Poller::add_fd(int fd, std::function<int(int)> handler) {
   pfds.emplace_back(pollfd{.fd = fd, .events = POLLIN, .revents = 0});
   handlers.emplace_back(std::move(handler));
}

int Poller::check_events() {
   int ret = 0;
   if ((ret = poll(this->pfds.data(), this->pfds.size(), -1)) != -1 &&
       errno != EINTR) {
      for (unsigned i = 0; i < pfds.size(); i++) {
         if ((pfds[i].revents & POLLIN) != 0) {
            if (handlers[i](pfds[i].fd) == -1) {
               return -1;
            }
            pfds[i].revents = 0;
            pfds[i].events = POLLIN;
         }
      }
   }
   return ret;
}

unique_fd::~unique_fd() {
   if (this->fd != -1) {
      close(this->fd);
   }
}