byedpi/conev.h

#ifndef CONEV_H
#define CONEV_H

#include <stdint.h>
#include <stdbool.h>

#ifndef __linux__
    #define NOEPOLL
#endif

#ifdef _WIN32
    #include <ws2tcpip.h>
    #define poll(fds, cnt, to) WSAPoll(fds, cnt, to)
    #define close(fd) closesocket(fd)
#else
    #include <netinet/in.h>
    #include <sys/socket.h>
    #include <unistd.h>
    
    #ifndef NOEPOLL
        #include <sys/epoll.h>
        #define POLLIN EPOLLIN
        #define POLLOUT EPOLLOUT
        #define POLLERR EPOLLERR
        #define POLLHUP EPOLLHUP
        #define POLLRDHUP EPOLLRDHUP
    #else
        #include <sys/poll.h>
    #endif
#endif

#ifndef POLLRDHUP
    #define POLLRDHUP 0
#endif

struct poolhd;
struct eval;
typedef int (*evcb_t)(struct poolhd *, struct eval *, int);

union sockaddr_u {
    struct sockaddr sa;
    struct sockaddr_in in;
    struct sockaddr_in6 in6;
};

#define FLAG_S4 1
#define FLAG_S5 2
#define FLAG_CONN 4
#define FLAG_HTTP 8

struct buffer {
    size_t size;
    unsigned int offset;
    char *data;
    size_t lock;
    struct buffer *next;
};

struct eval {
    int fd;    
    int index;
    unsigned long long mod_iter;
    evcb_t cb;
    struct eval *pair;
    struct buffer *buff;
    int flag;
    union sockaddr_u addr;
    ssize_t recv_count;
    ssize_t round_sent;
    unsigned int round_count;
    int attempt;
    bool cache;
    bool mark; //
};

struct poolhd {
    int max;
    int count;
    int efd;
    struct eval **links;
    struct eval *items;
#ifndef NOEPOLL
    struct epoll_event *pevents;
#else
    struct pollfd *pevents;
#endif
    unsigned long long iters;
    bool brk;
    
    struct buffer *root_buff;
};

struct poolhd *init_pool(int count);

struct eval *add_event(struct poolhd *pool, evcb_t cb, int fd, int e);

struct eval *add_pair(struct poolhd *pool, struct eval *val, int sfd, int e);

void del_event(struct poolhd *pool, struct eval *val);

void destroy_pool(struct poolhd *pool);

struct eval *next_event(struct poolhd *pool, int *offs, int *type);

int mod_etype(struct poolhd *pool, struct eval *val, int type);

void loop_event(struct poolhd *pool);

struct buffer *buff_get(struct buffer *root, size_t size);

void buff_destroy(struct buffer *root);
#endif