worker.h 1.99 KB
Newer Older
Marek Vavruša's avatar
Marek Vavruša committed
1
/*  Copyright (C) 2014 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz>
2

Marek Vavruša's avatar
Marek Vavruša committed
3 4 5 6
    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.
7

Marek Vavruša's avatar
Marek Vavruša committed
8 9 10 11
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
12

Marek Vavruša's avatar
Marek Vavruša committed
13 14 15
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
16 17 18

#pragma once

19
#include <libknot/internal/mempattern.h>
20

21
#include "daemon/engine.h"
22 23 24 25 26
#include "lib/generic/array.h"

/* @cond internal Freelist of available mempools. */
typedef array_t(void *) mp_freelist_t;
/* @endcond */
27

28 29
/**
 * Query resolution worker.
30
 */
31
struct worker_ctx {
32 33
	struct engine *engine;
	uv_loop_t *loop;
34 35 36 37 38
#if __linux__
	uint8_t wire_buf[RECVMMSG_BATCH * KNOT_WIRE_MAX_PKTSIZE];
#else
	uint8_t wire_buf[KNOT_WIRE_MAX_PKTSIZE];
#endif
39 40 41 42
	struct {
		size_t concurrent;
		size_t udp;
		size_t tcp;
43 44
		size_t ipv4;
		size_t ipv6;
45
	} stats;
46
	mp_freelist_t pools;
47
	mp_freelist_t ioreqs;
48
	mm_ctx_t pkt_pool;
49 50
};

51 52 53
/* Worker callback */
typedef void (*worker_cb_t)(struct worker_ctx *worker, struct kr_request *req, void *baton);

54
/**
55 56
 * Process incoming packet (query or answer to subrequest).
 * @return 0 or an error code
57
 */
58
int worker_exec(struct worker_ctx *worker, uv_handle_t *handle, knot_pkt_t *query, const struct sockaddr* addr);
59

60 61 62 63
/**
 * Schedule query for resolution.
 * @return 0 or an error code
 */
64
int worker_resolve(struct worker_ctx *worker, knot_pkt_t *query, unsigned options, worker_cb_t on_complete, void *baton);
65

66 67 68 69 70
/** Reserve worker buffers */
int worker_reserve(struct worker_ctx *worker, size_t ring_maxlen);

/** Collect worker mempools */
void worker_reclaim(struct worker_ctx *worker);