80 lines
3.1 KiB
C
80 lines
3.1 KiB
C
/* 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;
|
|
|