fde.h

/*
Copyright (c) 2012-2014 The SSDB Authors. All rights reserved.
Use of this source code is governed by a BSD-style license that can be
found in the LICENSE file.
*/
#ifndef UTIL_FDE_H
#define UTIL_FDE_H

#include <vector>
#include <sys/epoll.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>

#define FDEVENT_NONE (0)
#define FDEVENT_IN (1 << 0)
#define FDEVENT_PRI (1 << 1)
#define FDEVENT_OUT (1 << 2)
#define FDEVENT_HUP (1 << 3)
#define FDEVENT_ERR (1 << 4)

struct Fdevent {
    int fd;
    int s_flags; // subscribed events
    int events; // ready events
    struct {
        int num;
        void* ptr;
    } data;
};


class Fdevents {
public:
    typedef std::vector<struct Fdevent*> events_t;

private:
    static const int MAX_FDS = 8 * 1024;
    int ep_fd;
    struct epoll_event ep_events[MAX_FDS];
    events_t events;
    events_t ready_events;

    struct Fdevent* get_fde(int fd);

public:
    Fdevents();
    ~Fdevents();

    bool isset(int fd, int flag);
    int set(int fd, int flags, int data_num, void* data_ptr);
    int del(int fd);
    int clr(int fd, int flags);
    const events_t* wait(int timeout_ms = -1);
};

#endif