mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
145 lines
2.3 KiB
C++
145 lines
2.3 KiB
C++
#ifndef EV_EPOLL_HPP
|
|
#define EV_EPOLL_HPP
|
|
|
|
#include <ev/ev.hpp>
|
|
#include <net/net.h>
|
|
#include <net/net.hpp>
|
|
#include <util/buffer.hpp>
|
|
#include <util/buffer.hpp>
|
|
#include <util/logger.hpp>
|
|
#include <util/mem.hpp>
|
|
|
|
#include <cassert>
|
|
#include <cstdio>
|
|
#include <fcntl.h>
|
|
#include <signal.h>
|
|
#include <sys/epoll.h>
|
|
#include <sys/un.h>
|
|
#include <tuntap.h>
|
|
#include <unistd.h>
|
|
|
|
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()
|
|
{
|
|
}
|
|
|
|
bool
|
|
tick() override;
|
|
|
|
int
|
|
read(byte_t* buf, size_t sz) override;
|
|
|
|
int
|
|
sendto(const sockaddr* to, const void* data, size_t sz) override;
|
|
};
|
|
|
|
struct tun : public ev_io
|
|
{
|
|
llarp_tun_io* t;
|
|
int writefd;
|
|
device* tunif;
|
|
tun(llarp_tun_io* tio, llarp_ev_loop_ptr l)
|
|
: ev_io(-1, new LossyWriteQueue_t("tun_write_queue", l, l))
|
|
, t(tio)
|
|
, writefd(-1)
|
|
, tunif(tuntap_init())
|
|
|
|
{
|
|
|
|
};
|
|
|
|
int
|
|
sendto(const sockaddr* to, const void* data, size_t sz) override;
|
|
|
|
bool
|
|
tick() override;
|
|
|
|
void
|
|
flush_write() override;
|
|
|
|
ssize_t
|
|
do_write(void* buf, size_t sz) override;
|
|
|
|
int
|
|
read(byte_t* buf, size_t sz) override;
|
|
|
|
static int
|
|
wait_for_fd_promise(struct device* dev);
|
|
|
|
bool
|
|
setup();
|
|
|
|
~tun()
|
|
{
|
|
if(tunif)
|
|
tuntap_destroy(tunif);
|
|
}
|
|
};
|
|
}; // namespace llarp
|
|
|
|
struct llarp_epoll_loop
|
|
: public llarp_ev_loop,
|
|
public std::enable_shared_from_this< llarp_epoll_loop >
|
|
{
|
|
int epollfd;
|
|
|
|
llarp_epoll_loop() : epollfd(-1)
|
|
{
|
|
}
|
|
|
|
~llarp_epoll_loop()
|
|
{
|
|
}
|
|
|
|
bool
|
|
tcp_connect(struct llarp_tcp_connecter* tcp, const sockaddr* remoteaddr);
|
|
|
|
llarp::ev_io*
|
|
bind_tcp(llarp_tcp_acceptor* tcp, const sockaddr* bindaddr);
|
|
|
|
virtual bool
|
|
udp_listen(llarp_udp_io* l, const sockaddr* src);
|
|
|
|
bool
|
|
running() const;
|
|
|
|
bool
|
|
init();
|
|
|
|
int
|
|
tick(int ms);
|
|
|
|
int
|
|
run();
|
|
|
|
int
|
|
udp_bind(const sockaddr* addr);
|
|
|
|
bool
|
|
close_ev(llarp::ev_io* ev);
|
|
|
|
llarp::ev_io*
|
|
create_tun(llarp_tun_io* tun);
|
|
|
|
llarp::ev_io*
|
|
create_udp(llarp_udp_io* l, const sockaddr* src);
|
|
|
|
bool
|
|
add_ev(llarp::ev_io* e, bool write);
|
|
|
|
bool
|
|
udp_close(llarp_udp_io* l);
|
|
|
|
void
|
|
stop();
|
|
};
|
|
|
|
#endif
|