2019-01-11 02:19:36 +01:00
|
|
|
#include <ev/ev.h>
|
2019-01-10 20:41:51 +01:00
|
|
|
#include <util/mem.hpp>
|
|
|
|
#include <util/string_view.hpp>
|
2019-09-01 15:26:16 +02:00
|
|
|
#include <util/thread/logic.hpp>
|
2019-06-18 01:19:39 +02:00
|
|
|
#include <net/net_addr.hpp>
|
2018-11-06 23:48:17 +01:00
|
|
|
|
2019-06-18 01:19:39 +02:00
|
|
|
#include <cstddef>
|
|
|
|
#include <cstring>
|
2018-11-06 23:48:17 +01:00
|
|
|
|
2019-06-14 03:36:31 +02:00
|
|
|
// We libuv now
|
|
|
|
#include <ev/ev_libuv.hpp>
|
2019-12-11 22:05:40 +01:00
|
|
|
#if defined(_WIN32) || defined(_WIN64) || defined(__NT__)
|
2019-03-25 19:37:11 +01:00
|
|
|
#define SHUT_RDWR SD_BOTH
|
2019-01-11 02:19:36 +01:00
|
|
|
#include <ev/ev_win32.hpp>
|
2018-04-30 15:18:57 +02:00
|
|
|
#endif
|
|
|
|
|
2019-04-08 14:01:52 +02:00
|
|
|
llarp_ev_loop_ptr
|
2019-02-11 15:43:48 +01:00
|
|
|
llarp_make_ev_loop()
|
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_loop_ptr r = std::make_shared<libuv::Loop>();
|
2019-02-11 15:43:48 +01:00
|
|
|
r->init();
|
2019-03-15 15:31:54 +01:00
|
|
|
r->update_time();
|
2019-02-11 15:43:48 +01:00
|
|
|
return r;
|
|
|
|
}
|
|
|
|
|
2018-06-06 14:46:26 +02:00
|
|
|
void
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_loop_run_single_process(llarp_ev_loop_ptr ev, std::shared_ptr<llarp::Logic> logic)
|
2018-06-06 14:46:26 +02:00
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
while (ev->running())
|
2018-06-06 14:46:26 +02:00
|
|
|
{
|
2019-03-15 15:31:54 +01:00
|
|
|
ev->update_time();
|
2018-08-10 05:51:38 +02:00
|
|
|
ev->tick(EV_TICK_INTERVAL);
|
2019-05-15 17:54:26 +02:00
|
|
|
llarp::LogContext::Instance().logStream->Tick(ev->time_now());
|
2018-06-06 14:46:26 +02:00
|
|
|
}
|
2019-11-23 05:47:08 +01:00
|
|
|
logic->clear_event_loop();
|
2019-06-02 23:17:05 +02:00
|
|
|
ev->stopped();
|
2018-06-06 14:46:26 +02:00
|
|
|
}
|
|
|
|
|
2018-05-22 17:54:19 +02:00
|
|
|
int
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_add_udp(struct llarp_ev_loop* ev, struct llarp_udp_io* udp, const struct sockaddr* src)
|
2018-05-22 17:54:19 +02:00
|
|
|
{
|
2018-05-16 18:41:20 +02:00
|
|
|
udp->parent = ev;
|
2020-04-07 20:38:56 +02:00
|
|
|
if (ev->udp_listen(udp, src))
|
2018-05-22 17:54:19 +02:00
|
|
|
return 0;
|
2018-05-16 18:41:20 +02:00
|
|
|
return -1;
|
2018-01-29 15:27:24 +01:00
|
|
|
}
|
2018-01-19 17:51:27 +01:00
|
|
|
|
2018-05-22 17:54:19 +02:00
|
|
|
int
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_close_udp(struct llarp_udp_io* udp)
|
2018-05-22 17:54:19 +02:00
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
if (udp->parent->udp_close(udp))
|
2018-05-22 17:54:19 +02:00
|
|
|
return 0;
|
2018-05-16 18:41:20 +02:00
|
|
|
return -1;
|
2018-01-29 15:27:24 +01:00
|
|
|
}
|
2018-01-29 15:19:00 +01:00
|
|
|
|
2018-10-29 17:48:36 +01:00
|
|
|
llarp_time_t
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_loop_time_now_ms(const llarp_ev_loop_ptr& loop)
|
2018-10-29 17:48:36 +01:00
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
if (loop)
|
2019-03-15 15:31:54 +01:00
|
|
|
return loop->time_now();
|
2018-11-20 13:44:18 +01:00
|
|
|
return llarp::time_now_ms();
|
2018-10-29 17:48:36 +01:00
|
|
|
}
|
|
|
|
|
2018-05-22 17:54:19 +02:00
|
|
|
void
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_loop_stop(const llarp_ev_loop_ptr& loop)
|
2018-05-18 19:10:48 +02:00
|
|
|
{
|
2018-05-22 17:54:19 +02:00
|
|
|
loop->stop();
|
2018-05-18 19:10:48 +02:00
|
|
|
}
|
|
|
|
|
2018-05-22 17:54:19 +02:00
|
|
|
int
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_udp_sendto(struct llarp_udp_io* udp, const sockaddr* to, const llarp_buffer_t& buf)
|
2018-05-22 17:54:19 +02:00
|
|
|
{
|
2019-06-02 23:17:05 +02:00
|
|
|
return udp->sendto(udp, to, buf.base, buf.sz);
|
2018-05-22 17:54:19 +02:00
|
|
|
}
|
2018-08-15 17:36:34 +02:00
|
|
|
|
|
|
|
bool
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_add_tun(struct llarp_ev_loop* loop, struct llarp_tun_io* tun)
|
2018-08-15 17:36:34 +02:00
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
if (tun->ifaddr[0] == 0 || strcmp(tun->ifaddr, "auto") == 0)
|
2019-01-16 22:03:45 +01:00
|
|
|
{
|
2019-07-08 17:26:06 +02:00
|
|
|
LogError("invalid ifaddr on tun: ", tun->ifaddr);
|
|
|
|
return false;
|
2019-01-16 22:03:45 +01:00
|
|
|
}
|
2020-04-07 20:38:56 +02:00
|
|
|
if (tun->ifname[0] == 0 || strcmp(tun->ifname, "auto") == 0)
|
2019-01-16 22:03:45 +01:00
|
|
|
{
|
2019-07-08 17:26:06 +02:00
|
|
|
LogError("invalid ifname on tun: ", tun->ifname);
|
|
|
|
return false;
|
2019-01-16 22:03:45 +01:00
|
|
|
}
|
2019-09-05 03:25:50 +02:00
|
|
|
#if !defined(_WIN32)
|
2019-06-02 23:17:05 +02:00
|
|
|
return loop->tun_listen(tun);
|
2018-12-14 13:04:14 +01:00
|
|
|
#else
|
|
|
|
UNREFERENCED_PARAMETER(loop);
|
2020-04-07 20:38:56 +02:00
|
|
|
auto dev = new win32_tun_io(tun);
|
2018-12-14 13:04:14 +01:00
|
|
|
tun->impl = dev;
|
|
|
|
// We're not even going to add this to the socket event loop
|
2020-04-07 20:38:56 +02:00
|
|
|
if (dev)
|
2018-12-14 13:04:14 +01:00
|
|
|
{
|
|
|
|
dev->setup();
|
2019-12-11 22:18:47 +01:00
|
|
|
return dev->add_ev(loop); // start up tun and add to event queue
|
2018-12-14 13:04:14 +01:00
|
|
|
}
|
2019-05-06 14:17:34 +02:00
|
|
|
llarp::LogWarn("Loop could not create tun");
|
2018-12-14 13:04:14 +01:00
|
|
|
return false;
|
2019-06-11 18:44:05 +02:00
|
|
|
#endif
|
2018-12-14 13:04:14 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_ev_tun_async_write(struct llarp_tun_io* tun, const llarp_buffer_t& buf)
|
2018-12-14 13:04:14 +01:00
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
if (buf.sz > EV_WRITE_BUF_SZ)
|
2018-12-14 13:04:14 +01:00
|
|
|
{
|
|
|
|
llarp::LogWarn("packet too big, ", buf.sz, " > ", EV_WRITE_BUF_SZ);
|
|
|
|
return false;
|
|
|
|
}
|
2019-05-20 14:19:33 +02:00
|
|
|
#ifndef _WIN32
|
2019-06-03 14:02:54 +02:00
|
|
|
return tun->writepkt(tun, buf.base, buf.sz);
|
2018-12-14 13:04:14 +01:00
|
|
|
#else
|
2020-04-07 20:38:56 +02:00
|
|
|
return static_cast<win32_tun_io*>(tun->impl)->queue_write(buf.base, buf.sz);
|
2018-12-14 13:04:14 +01:00
|
|
|
#endif
|
2019-05-06 14:17:34 +02:00
|
|
|
}
|
2018-08-15 17:36:34 +02:00
|
|
|
|
|
|
|
bool
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_tcp_conn_async_write(struct llarp_tcp_conn* conn, const llarp_buffer_t& b)
|
2018-08-15 17:36:34 +02:00
|
|
|
{
|
2019-02-03 01:48:10 +01:00
|
|
|
ManagedBuffer buf{b};
|
2019-06-03 14:02:54 +02:00
|
|
|
|
2020-04-07 20:38:56 +02:00
|
|
|
size_t sz = buf.underlying.sz;
|
2019-02-03 00:12:42 +01:00
|
|
|
buf.underlying.cur = buf.underlying.base;
|
2020-04-07 20:38:56 +02:00
|
|
|
while (sz > EV_WRITE_BUF_SZ)
|
2018-10-25 14:09:29 +02:00
|
|
|
{
|
2019-06-03 14:02:54 +02:00
|
|
|
ssize_t amount = conn->write(conn, buf.underlying.cur, EV_WRITE_BUF_SZ);
|
2020-04-07 20:38:56 +02:00
|
|
|
if (amount <= 0)
|
2019-06-03 16:03:59 +02:00
|
|
|
{
|
|
|
|
llarp::LogError("write underrun");
|
2020-01-07 12:59:17 +01:00
|
|
|
llarp_tcp_conn_close(conn);
|
2018-10-25 14:09:29 +02:00
|
|
|
return false;
|
2019-06-03 16:03:59 +02:00
|
|
|
}
|
2019-06-03 14:02:54 +02:00
|
|
|
buf.underlying.cur += amount;
|
|
|
|
sz -= amount;
|
2018-10-25 14:09:29 +02:00
|
|
|
}
|
2019-06-03 16:03:59 +02:00
|
|
|
return conn->write(conn, buf.underlying.cur, sz) > 0;
|
2018-08-17 00:36:15 +02:00
|
|
|
}
|
2018-10-09 14:06:30 +02:00
|
|
|
|
2018-11-01 13:47:14 +01:00
|
|
|
void
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_tcp_async_try_connect(struct llarp_ev_loop* loop, struct llarp_tcp_connecter* tcp)
|
2018-11-01 13:47:14 +01:00
|
|
|
{
|
|
|
|
tcp->loop = loop;
|
|
|
|
llarp::string_view addr_str, port_str;
|
|
|
|
// try parsing address
|
2020-04-07 20:38:56 +02:00
|
|
|
const char* begin = tcp->remote;
|
|
|
|
const char* ptr = strstr(tcp->remote, ":");
|
2018-11-01 13:47:14 +01:00
|
|
|
// get end of address
|
|
|
|
|
2020-04-07 20:38:56 +02:00
|
|
|
if (ptr == nullptr)
|
2018-11-01 13:47:14 +01:00
|
|
|
{
|
|
|
|
llarp::LogError("bad address: ", tcp->remote);
|
2020-04-07 20:38:56 +02:00
|
|
|
if (tcp->error)
|
2018-11-01 13:47:14 +01:00
|
|
|
tcp->error(tcp);
|
|
|
|
return;
|
|
|
|
}
|
2020-04-07 20:38:56 +02:00
|
|
|
const char* end = ptr;
|
|
|
|
while (*end && ((end - begin) < static_cast<ptrdiff_t>(sizeof tcp->remote)))
|
2018-11-01 13:47:14 +01:00
|
|
|
{
|
|
|
|
++end;
|
|
|
|
}
|
|
|
|
addr_str = llarp::string_view(begin, ptr - begin);
|
|
|
|
++ptr;
|
|
|
|
port_str = llarp::string_view(ptr, end - ptr);
|
|
|
|
// actually parse address
|
|
|
|
llarp::Addr addr(addr_str, port_str);
|
|
|
|
|
2020-04-07 20:38:56 +02:00
|
|
|
if (!loop->tcp_connect(tcp, addr))
|
2018-11-01 13:47:14 +01:00
|
|
|
{
|
2018-11-02 13:35:20 +01:00
|
|
|
llarp::LogError("async connect failed");
|
2020-04-07 20:38:56 +02:00
|
|
|
if (tcp->error)
|
2018-11-02 13:35:20 +01:00
|
|
|
tcp->error(tcp);
|
2018-11-01 13:47:14 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-10-09 14:06:30 +02:00
|
|
|
bool
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_tcp_serve(
|
|
|
|
struct llarp_ev_loop* loop, struct llarp_tcp_acceptor* tcp, const struct sockaddr* bindaddr)
|
2018-10-09 14:06:30 +02:00
|
|
|
{
|
2019-06-02 23:17:05 +02:00
|
|
|
tcp->loop = loop;
|
|
|
|
return loop->tcp_listen(tcp, bindaddr);
|
2018-10-09 14:06:30 +02:00
|
|
|
}
|
|
|
|
|
2018-10-19 13:41:36 +02:00
|
|
|
void
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_tcp_acceptor_close(struct llarp_tcp_acceptor* tcp)
|
2018-10-19 13:41:36 +02:00
|
|
|
{
|
2019-06-03 16:03:59 +02:00
|
|
|
tcp->close(tcp);
|
2018-10-24 20:02:42 +02:00
|
|
|
}
|
|
|
|
|
2018-10-09 14:06:30 +02:00
|
|
|
void
|
2020-04-07 20:38:56 +02:00
|
|
|
llarp_tcp_conn_close(struct llarp_tcp_conn* conn)
|
2018-10-09 14:06:30 +02:00
|
|
|
{
|
2019-06-02 23:17:05 +02:00
|
|
|
conn->close(conn);
|
2018-10-09 14:06:30 +02:00
|
|
|
}
|
2018-10-25 14:00:29 +02:00
|
|
|
|
|
|
|
namespace llarp
|
|
|
|
{
|
2018-10-25 14:39:32 +02:00
|
|
|
bool
|
|
|
|
tcp_conn::tick()
|
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
if (_shouldClose)
|
2018-10-25 14:39:32 +02:00
|
|
|
{
|
2020-04-07 20:38:56 +02:00
|
|
|
if (tcp.closed)
|
2018-11-01 13:47:14 +01:00
|
|
|
tcp.closed(&tcp);
|
2019-03-25 16:41:37 +01:00
|
|
|
::shutdown(fd, SHUT_RDWR);
|
2018-10-25 14:39:32 +02:00
|
|
|
return false;
|
|
|
|
}
|
2020-04-07 20:38:56 +02:00
|
|
|
if (tcp.tick)
|
2018-11-01 13:47:14 +01:00
|
|
|
tcp.tick(&tcp);
|
2018-10-25 14:39:32 +02:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2018-11-06 23:48:17 +01:00
|
|
|
} // namespace llarp
|