src/util.cpp

   1 /*
   2  * Copyright (C) 2017 Mentor Graphics Development (Deutschland) GmbH
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include "util.hpp"
  18
  19 #include <cerrno>
  20 #include <cstdarg>
  21 #include <cstdio>
  22 #include <cstdlib>
  23 #include <ctime>
  24
  25 #include <unistd.h>
  26
  27 void Poller::add_fd(int fd, std::function<int(int)> handler) {
  28    pfds.emplace_back(pollfd{.fd = fd, .events = POLLIN, .revents = 0});
  29    handlers.emplace_back(std::move(handler));
  30 }
  31
  32 int Poller::check_events() {
  33    int ret = 0;
  34    if ((ret = poll(this->pfds.data(), this->pfds.size(), -1)) != -1 &&
  35        errno != EINTR) {
  36       for (unsigned i = 0; i < pfds.size(); i++) {
  37          if ((pfds[i].revents & POLLIN) != 0) {
  38             if (handlers[i](pfds[i].fd) == -1) {
  39                return -1;
  40             }
  41             pfds[i].revents = 0;
  42             pfds[i].events = POLLIN;
  43          }
  44       }
  45    }
  46    return ret;
  47 }
  48
  49 unique_fd::~unique_fd() {
  50    if (this->fd != -1) {
  51       close(this->fd);
  52    }
  53 }