lokinet/llarp/ev_epoll.hpp

#ifndef EV_EPOLL_HPP
#define EV_EPOLL_HPP
#include <llarp/net.h>
#include <sys/epoll.h>
#include <unistd.h>
#include <cstdio>
#include "ev.hpp"

namespace llarp {
struct udp_listener : public ev_io {
  llarp_udp_io* udp;

  udp_listener(int fd, llarp_udp_io* u) : ev_io(fd), udp(u){};

  ~udp_listener() {}

  virtual int read(void* buf, size_t sz) {
    sockaddr src;
    socklen_t slen;
    int ret = ::recvfrom(fd, buf, sz, 0, &src, &slen);
    if (ret == -1) return -1;
    udp->recvfrom(udp, &src, buf, ret);
    return 0;
  }

  virtual int sendto(const sockaddr* to, const void* data, size_t sz) {
    socklen_t slen;
    switch (to->sa_family) {
      case AF_INET:
        slen = sizeof(struct sockaddr_in);
        break;
      case AF_INET6:
        slen = sizeof(struct sockaddr_in6);
        break;
      default:
        return -1;
    }
    return ::sendto(fd, data, sz, 0, to, slen);
  }
};
};  // namespace llarp

struct llarp_epoll_loop : public llarp_ev_loop {
  int epollfd;

  llarp_epoll_loop() : epollfd(-1) {}

  ~llarp_epoll_loop() {}

  bool init() {
    if (epollfd == -1) epollfd = epoll_create1(EPOLL_CLOEXEC);
    return epollfd != -1;
  }

  int run() {
    epoll_event events[1024];
    int result;
    char readbuf[2048];
    do {
      result = epoll_wait(epollfd, events, 1024, -1);
      if (result > 0) {
        int idx = 0;
        while (idx < result) {
          llarp::ev_io* ev = static_cast<llarp::ev_io*>(events[idx].data.ptr);
          if (events[idx].events & EPOLLIN) {
            if (ev->read(readbuf, sizeof(readbuf)) == -1) {
              close_ev(ev);
              delete ev;
            }
          }
          ++idx;
        }
      }
    } while (result != -1);
    return result;
  }

  int udp_bind(const sockaddr* addr) {
    socklen_t slen;
    switch (addr->sa_family) {
      case AF_INET:
        slen = sizeof(struct sockaddr_in);
        break;
      case AF_INET6:
        slen = sizeof(struct sockaddr_in6);
        break;
      case AF_PACKET:
        slen = sizeof(struct sockaddr_ll);
        break;
      default:
        return -1;
    }
    int fd = socket(addr->sa_family, SOCK_DGRAM, 0);
    if (fd == -1) return -1;
    if (bind(fd, addr, slen) == -1) {
      close(fd);
      return -1;
    }
    return fd;
  }

  bool close_ev(llarp::ev_io * ev)
  {
    return epoll_ctl(epollfd, EPOLL_CTL_DEL, ev->fd, nullptr) != -1;
  }

  bool udp_listen(llarp_udp_io* l) {
    int fd = udp_bind(&l->addr);
    if (fd == -1) return false;
    llarp::udp_listener* listener = new llarp::udp_listener(fd, l);
    epoll_event ev;
    ev.data.ptr = listener;
    ev.events = EPOLLIN;
    if (epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &ev) == -1) {
      delete listener;
      return false;
    }
    l->impl = listener;
    return true;
  }

  bool udp_close(llarp_udp_io* l)
  {
    auto listener = static_cast<llarp::udp_listener*>(l->impl);
    if(!listener) return false;
    return close_ev(listener);
  }
  

  void stop() {
    if (epollfd != -1) ::close(epollfd);

    epollfd = -1;
  }
};

#endif