|
|
|
@ -1,38 +1,30 @@
|
|
|
|
|
#ifndef EV_EPOLL_HPP
|
|
|
|
|
#define EV_EPOLL_HPP
|
|
|
|
|
#include "ev.hpp"
|
|
|
|
|
#include <unistd.h>
|
|
|
|
|
#include <sys/epoll.h>
|
|
|
|
|
#include <netinet/in.h>
|
|
|
|
|
#include <sys/epoll.h>
|
|
|
|
|
#include <unistd.h>
|
|
|
|
|
#include "ev.hpp"
|
|
|
|
|
|
|
|
|
|
namespace llarp {
|
|
|
|
|
struct udp_listener : public ev_io {
|
|
|
|
|
llarp_udp_io* udp;
|
|
|
|
|
|
|
|
|
|
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(int fd, llarp_udp_io* u) : ev_io(fd), udp(u){};
|
|
|
|
|
|
|
|
|
|
~udp_listener() {}
|
|
|
|
|
|
|
|
|
|
virtual int read()
|
|
|
|
|
{
|
|
|
|
|
sockaddr src;
|
|
|
|
|
socklen_t slen;
|
|
|
|
|
int ret = ::recvfrom(fd, buff, sizeof(buff), 0, &src, &slen);
|
|
|
|
|
if (ret == -1) return -1;
|
|
|
|
|
udp->recvfrom(udp, &src, buff, ret);
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
~udp_listener() {}
|
|
|
|
|
|
|
|
|
|
virtual int sendto(const sockaddr * to, const void * data, size_t sz)
|
|
|
|
|
{
|
|
|
|
|
socklen_t slen;
|
|
|
|
|
switch(to->sa_family)
|
|
|
|
|
{
|
|
|
|
|
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;
|
|
|
|
@ -41,106 +33,84 @@ namespace llarp
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
return ::sendto(fd, data, sz, 0, to, slen);
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
return ::sendto(fd, data, sz, 0, to, slen);
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
}; // namespace llarp
|
|
|
|
|
|
|
|
|
|
struct llarp_epoll_loop : public llarp_ev_loop
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
struct llarp_epoll_loop : public llarp_ev_loop {
|
|
|
|
|
int epollfd;
|
|
|
|
|
|
|
|
|
|
llarp_epoll_loop() : epollfd(-1)
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
~llarp_epoll_loop()
|
|
|
|
|
{
|
|
|
|
|
}
|
|
|
|
|
llarp_epoll_loop() : epollfd(-1) {}
|
|
|
|
|
|
|
|
|
|
~llarp_epoll_loop() {}
|
|
|
|
|
|
|
|
|
|
bool init()
|
|
|
|
|
{
|
|
|
|
|
if(epollfd == -1)
|
|
|
|
|
epollfd = epoll_create1(EPOLL_CLOEXEC);
|
|
|
|
|
bool init() {
|
|
|
|
|
if (epollfd == -1) epollfd = epoll_create1(EPOLL_CLOEXEC);
|
|
|
|
|
return epollfd != -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int run()
|
|
|
|
|
{
|
|
|
|
|
int run() {
|
|
|
|
|
epoll_event events[1024];
|
|
|
|
|
int result;
|
|
|
|
|
do
|
|
|
|
|
{
|
|
|
|
|
char readbuf[2048];
|
|
|
|
|
do {
|
|
|
|
|
result = epoll_wait(epollfd, events, 1024, 0);
|
|
|
|
|
if(result > 0)
|
|
|
|
|
{
|
|
|
|
|
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() == -1)
|
|
|
|
|
{
|
|
|
|
|
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) {
|
|
|
|
|
epoll_ctl(epollfd, EPOLL_CTL_DEL, ev->fd, nullptr);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
++idx;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
while(result != -1);
|
|
|
|
|
} while (result != -1);
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int udp_bind(const sockaddr * addr)
|
|
|
|
|
{
|
|
|
|
|
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;
|
|
|
|
|
default:
|
|
|
|
|
return -1;
|
|
|
|
|
switch (addr->sa_family) {
|
|
|
|
|
case AF_INET:
|
|
|
|
|
slen = sizeof(struct sockaddr_in);
|
|
|
|
|
break;
|
|
|
|
|
case AF_INET6:
|
|
|
|
|
slen = sizeof(struct sockaddr_in6);
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
int fd = socket(addr->sa_family, SOCK_DGRAM, 0);
|
|
|
|
|
if (fd == -1) return -1;
|
|
|
|
|
if(bind(fd, addr, slen) == -1)
|
|
|
|
|
{
|
|
|
|
|
if (bind(fd, addr, slen) == -1) {
|
|
|
|
|
close(fd);
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
return fd;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool udp_listen(llarp_udp_io * l)
|
|
|
|
|
{
|
|
|
|
|
bool udp_listen(llarp_udp_io* l) {
|
|
|
|
|
int fd = udp_bind((sockaddr*)l->addr);
|
|
|
|
|
if (fd == -1) return false;
|
|
|
|
|
llarp::udp_listener * listener = new llarp::udp_listener(fd, l);
|
|
|
|
|
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)
|
|
|
|
|
{
|
|
|
|
|
if (epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &ev) == -1) {
|
|
|
|
|
delete listener;
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void stop()
|
|
|
|
|
{
|
|
|
|
|
if(epollfd != -1)
|
|
|
|
|
::close(epollfd);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void stop() {
|
|
|
|
|
if (epollfd != -1) ::close(epollfd);
|
|
|
|
|
|
|
|
|
|
epollfd = -1;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|