Annotation of embedaddon/dnsmasq/src/conntrack.c, revision 1.1.1.4
1.1.1.4 ! misho 1: /* dnsmasq is Copyright (c) 2000-2021 Simon Kelley
1.1 misho 2:
3: This program is free software; you can redistribute it and/or modify
4: it under the terms of the GNU General Public License as published by
5: the Free Software Foundation; version 2 dated June, 1991, or
6: (at your option) version 3 dated 29 June, 2007.
7:
8: This program is distributed in the hope that it will be useful,
9: but WITHOUT ANY WARRANTY; without even the implied warranty of
10: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11: GNU General Public License for more details.
12:
13: You should have received a copy of the GNU General Public License
14: along with this program. If not, see <http://www.gnu.org/licenses/>.
15: */
16:
17: #include "dnsmasq.h"
18:
19: #ifdef HAVE_CONNTRACK
20:
21: #include <libnetfilter_conntrack/libnetfilter_conntrack.h>
22:
23: static int gotit = 0; /* yuck */
24:
25: static int callback(enum nf_conntrack_msg_type type, struct nf_conntrack *ct, void *data);
26:
1.1.1.4 ! misho 27: int get_incoming_mark(union mysockaddr *peer_addr, union all_addr *local_addr, int istcp, unsigned int *markp)
1.1 misho 28: {
29: struct nf_conntrack *ct;
30: struct nfct_handle *h;
31:
32: gotit = 0;
33:
34: if ((ct = nfct_new()))
35: {
36: nfct_set_attr_u8(ct, ATTR_L4PROTO, istcp ? IPPROTO_TCP : IPPROTO_UDP);
37: nfct_set_attr_u16(ct, ATTR_PORT_DST, htons(daemon->port));
38:
39: if (peer_addr->sa.sa_family == AF_INET6)
40: {
41: nfct_set_attr_u8(ct, ATTR_L3PROTO, AF_INET6);
42: nfct_set_attr(ct, ATTR_IPV6_SRC, peer_addr->in6.sin6_addr.s6_addr);
43: nfct_set_attr_u16(ct, ATTR_PORT_SRC, peer_addr->in6.sin6_port);
1.1.1.4 ! misho 44: nfct_set_attr(ct, ATTR_IPV6_DST, local_addr->addr6.s6_addr);
1.1 misho 45: }
46: else
47: {
48: nfct_set_attr_u8(ct, ATTR_L3PROTO, AF_INET);
49: nfct_set_attr_u32(ct, ATTR_IPV4_SRC, peer_addr->in.sin_addr.s_addr);
50: nfct_set_attr_u16(ct, ATTR_PORT_SRC, peer_addr->in.sin_port);
1.1.1.4 ! misho 51: nfct_set_attr_u32(ct, ATTR_IPV4_DST, local_addr->addr4.s_addr);
1.1 misho 52: }
53:
54:
55: if ((h = nfct_open(CONNTRACK, 0)))
56: {
57: nfct_callback_register(h, NFCT_T_ALL, callback, (void *)markp);
58: if (nfct_query(h, NFCT_Q_GET, ct) == -1)
59: {
60: static int warned = 0;
61: if (!warned)
62: {
63: my_syslog(LOG_ERR, _("Conntrack connection mark retrieval failed: %s"), strerror(errno));
64: warned = 1;
65: }
66: }
67: nfct_close(h);
68: }
69: nfct_destroy(ct);
70: }
71:
72: return gotit;
73: }
74:
75: static int callback(enum nf_conntrack_msg_type type, struct nf_conntrack *ct, void *data)
76: {
77: unsigned int *ret = (unsigned int *)data;
78: *ret = nfct_get_attr_u32(ct, ATTR_MARK);
79: (void)type; /* eliminate warning */
80: gotit = 1;
81:
82: return NFCT_CB_CONTINUE;
83: }
84:
85: #endif
86:
87:
88:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>