mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-10-29 11:05:43 +00:00
105 lines
1.9 KiB
C++
105 lines
1.9 KiB
C++
#ifndef EV_WIN32_H
|
|
#define EV_WIN32_H
|
|
#ifdef _WIN32
|
|
#include <ev/ev.hpp>
|
|
#include <net/net.h>
|
|
#include <net/net.hpp>
|
|
#include <util/buffer.hpp>
|
|
#include <util/thread/logic.hpp>
|
|
|
|
#include <windows.h>
|
|
#include <process.h>
|
|
|
|
#include <cstdio>
|
|
|
|
// io packet for TUN read/write
|
|
struct asio_evt_pkt
|
|
{
|
|
OVERLAPPED pkt = {
|
|
0, 0, 0, 0, nullptr}; // must be first, since this is part of the IO call
|
|
bool write = false; // true, or false if read pkt
|
|
size_t sz; // should match the queued data size, if not try again?
|
|
void* buf; // must remain valid until we get notification; this is _supposed_
|
|
// to be zero-copy
|
|
};
|
|
|
|
extern "C" DWORD FAR PASCAL
|
|
tun_ev_loop(void* unused);
|
|
|
|
void
|
|
exit_tun_loop();
|
|
|
|
void
|
|
begin_tun_loop(int nThreads);
|
|
|
|
namespace llarp
|
|
{
|
|
struct udp_listener : public ev_io
|
|
{
|
|
llarp_udp_io* udp;
|
|
llarp_pkt_list m_RecvPackets;
|
|
|
|
udp_listener(int fd, llarp_udp_io* u) : ev_io(fd), udp(u){};
|
|
|
|
~udp_listener()
|
|
{
|
|
}
|
|
|
|
bool
|
|
RecvMany(llarp_pkt_list*);
|
|
|
|
bool
|
|
tick();
|
|
|
|
int
|
|
read(byte_t* buf, size_t sz);
|
|
|
|
int
|
|
sendto(const sockaddr* to, const void* data, size_t sz);
|
|
};
|
|
} // namespace llarp
|
|
|
|
// A different kind of event loop,
|
|
// more suited for the native Windows NT
|
|
// event model
|
|
struct win32_tun_io
|
|
{
|
|
llarp_tun_io* t;
|
|
device* tunif;
|
|
byte_t readbuf[EV_READ_BUF_SZ] = {0};
|
|
|
|
win32_tun_io(llarp_tun_io* tio) : t(tio), tunif(tuntap_init()){};
|
|
|
|
bool
|
|
queue_write(const byte_t* buf, size_t sz);
|
|
|
|
bool
|
|
setup();
|
|
|
|
// first TUN device gets to set up the event port
|
|
bool
|
|
add_ev(llarp_ev_loop* l);
|
|
|
|
// places data in event queue for kernel to process
|
|
void
|
|
do_write(void* data, size_t sz);
|
|
|
|
// we call this one when we get a packet in the event port
|
|
// which then kicks off another write
|
|
void
|
|
flush_write();
|
|
|
|
void
|
|
read(byte_t* buf, size_t sz);
|
|
|
|
~win32_tun_io()
|
|
{
|
|
CancelIo(tunif->tun_fd);
|
|
if(tunif->tun_fd)
|
|
tuntap_destroy(tunif);
|
|
}
|
|
};
|
|
|
|
#endif
|
|
#endif
|