/*
 * Copyright (C) 2016  Intel Corporation. All rights reserved.
 *
 * This file is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This file is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#include <errno.h>
#include <fcntl.h>
#include <inttypes.h>
#include <sys/epoll.h>
#include <sys/types.h>
#include <sys/uio.h>
#include <unistd.h>

#include "Poller.h"

extern const AP_HAL::HAL &hal;

namespace Linux {

bool Poller::register_pollable(Pollable *p, uint32_t events)
{
    /*
     * EPOLLWAKEUP prevents the system from hibernating or suspending when
     * inside epoll_wait() for this particular event.  It is silently
     * ignored if the process does not have the CAP_BLOCK_SUSPEND
     * capability.
     */
    events |= EPOLLWAKEUP;

    if (_epfd < 0) {
        return false;
    }

    struct epoll_event epev = { };
    epev.events = events;
    epev.data.ptr = static_cast<void *>(p);

    return epoll_ctl(_epfd, EPOLL_CTL_ADD, p->get_fd(), &epev) == 0;
}

void Poller::unregister_pollable(const Pollable *p)
{
    if (_epfd >= 0 && p->get_fd() >= 0) {
        epoll_ctl(_epfd, EPOLL_CTL_DEL, p->get_fd(), nullptr);
    }
}

int Poller::poll() const
{
    const int max_events = 16;
    epoll_event events[max_events];
    int r;

    do {
        r = epoll_wait(_epfd, events, max_events, -1);
    } while (r < 0 && errno == EINTR);

    if (r < 0) {
        return -errno;
    }

    for (int i = 0; i < r; i++) {
        Pollable *p = static_cast<Pollable *>(events[i].data.ptr);

        if (events[i].events & EPOLLIN) {
            p->on_can_read();
        }
        if (events[i].events & EPOLLOUT) {
            p->on_can_write();
        }
        if (events[i].events & EPOLLERR) {
            p->on_error();
        }
        if (events[i].events & EPOLLHUP) {
            p->on_hang_up();
        }
    }

    return r;
}

Pollable::~Pollable()
{
    /*
     * Make sure to remove the file descriptor from epoll since events could
     * continue to be reported if the file descriptor was dup()'ed.  However
     * we rely on user unregistering it rather than taking a reference to the
     * Poller.  Here we just close our file descriptor.
     */
    if (_fd >= 0) {
        close(_fd);
    }
}

}