poll.hpp

/**
 * @file poll.hpp
 * @brief Thin epoll wrapper. Adds and removes file descriptors in EPOLLET mode.
 *
 * CSCI 599: Network Systems for Cloud Computing
 * University of Southern California
 */

#ifndef __POLL_HPP__
#define __POLL_HPP__

#include <iostream>
#include <sys/epoll.h>

class __epoll {
private:
    int __epoll_fd;
    int __timeout;

public:
    const static int gnum = 128;

public:
    __epoll(int timeout)
        : __timeout(timeout) { }
    ~__epoll() { }

public:
    void create_poll() {
        __epoll_fd = epoll_create(gnum);
        if (__epoll_fd < 0)
            exit(5);
    }
    bool add_sock_to_poll(int sock, uint32_t events) {
        struct epoll_event ev;
        ev.events = events;
        ev.data.fd = sock;
        int n = epoll_ctl(__epoll_fd, EPOLL_CTL_ADD, sock, &ev);
        return n == 0;
    }
    bool delete_from_epoll(int sock) {
        return epoll_ctl(__epoll_fd, EPOLL_CTL_DEL, sock, nullptr) == 0;
    }
    int wait_poll(struct epoll_event revs[], int num) {
        return epoll_wait(__epoll_fd, revs, num, __timeout);
    }
    bool control_poll(int sock, uint32_t events) {
        events |= EPOLLET; // 无论是谁，一律设置为ET模式
        struct epoll_event ev;
        ev.events = events;
        ev.data.fd = sock;
        int n = epoll_ctl(__epoll_fd, EPOLL_CTL_MOD, sock, &ev);
        return n == 0;
    }

public:
};

#endif