diff options
Diffstat (limited to 'daemon/io.h')
| -rw-r--r-- | daemon/io.h | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/daemon/io.h b/daemon/io.h new file mode 100644 index 0000000..0e88dc1 --- /dev/null +++ b/daemon/io.h @@ -0,0 +1,80 @@ +/* Copyright (C) CZ.NIC, z.s.p.o. <knot-resolver@labs.nic.cz> + * SPDX-License-Identifier: GPL-3.0-or-later + */ + +#pragma once + +#include <lua.h> +#include <uv.h> +#include <libknot/packet/pkt.h> +#include <gnutls/gnutls.h> +#include "lib/generic/array.h" +#include "daemon/worker.h" +#include "daemon/engine.h" + +struct tls_ctx; +struct tls_client_ctx; +struct io_stream_data; + +/** Communication data. */ +struct io_comm_data { + /** The original address the data came from. May be that of a proxied + * client, if they came through a proxy. May be `NULL` if + * the communication did not come from network. */ + const struct sockaddr *src_addr; + + /** The actual address the resolver is communicating with. May be + * the address of a proxy if the communication came through one, + * otherwise it will be the same as `src_addr`. May be `NULL` if + * the communication did not come from network. */ + const struct sockaddr *comm_addr; + + /** The original destination address. May be the resolver's address, or + * the address of a proxy if the communication came through one. May be + * `NULL` if the communication did not come from network. */ + const struct sockaddr *dst_addr; + + /** Data parsed from a PROXY header. May be `NULL` if the communication + * did not come through a proxy, or if the PROXYv2 protocol was not used. */ + const struct proxy_result *proxy; +}; + +/** Bind address into a file-descriptor (only, no libuv). type is e.g. SOCK_DGRAM */ +int io_bind(const struct sockaddr *addr, int type, const endpoint_flags_t *flags); +/** Initialize a UDP handle and start listening. */ +int io_listen_udp(uv_loop_t *loop, uv_udp_t *handle, int fd); +/** Initialize a TCP handle and start listening. */ +int io_listen_tcp(uv_loop_t *loop, uv_tcp_t *handle, int fd, int tcp_backlog, bool has_tls, bool has_http); +/** Initialize a pipe handle and start listening. */ +int io_listen_pipe(uv_loop_t *loop, uv_pipe_t *handle, int fd); +/** Initialize a poll handle (ep->handle) and start listening over AF_XDP on ifname. + * Sets ep->session. */ +int io_listen_xdp(uv_loop_t *loop, struct endpoint *ep, const char *ifname); + +/** Control socket / TTY - related functions. */ +void io_tty_process_input(uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf); +void io_tty_alloc(uv_handle_t *handle, size_t suggested, uv_buf_t *buf); +void io_tty_accept(uv_stream_t *master, int status); +struct io_stream_data *io_tty_alloc_data(void); + +void tcp_timeout_trigger(uv_timer_t *timer); + +/** Initialize the handle, incl. ->data = struct session * instance. + * \param type = SOCK_* + * \param family = AF_* + * \param has_tls has meanings only when type is SOCK_STREAM */ +int io_create(uv_loop_t *loop, uv_handle_t *handle, int type, + unsigned family, bool has_tls, bool has_http); +void io_free(uv_handle_t *handle); + +int io_start_read(uv_handle_t *handle); +int io_stop_read(uv_handle_t *handle); + +/** When uv_handle_t::type == UV_POLL, ::data points to this malloc-ed helper. + * (Other cases store a direct struct session pointer in ::data.) */ +typedef struct { + struct knot_xdp_socket *socket; + struct session *session; + uv_idle_t tx_waker; +} xdp_handle_data_t; + |