grp_getter.c revision 8a44513648da0c5f5551f96b329cf56b66f5b303 
1/*
2 * (C) 2005-2011 by Pablo Neira Ayuso <pablo@netfilter.org>
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 */
9
10#include "internal/internal.h"
11
12static void get_attr_grp_orig_ipv4(const struct nf_conntrack *ct, void *data)
13{
14	struct nfct_attr_grp_ipv4 *this = data;
15	this->src = ct->head.orig.src.v4;
16	this->dst = ct->head.orig.dst.v4;
17}
18
19static void get_attr_grp_repl_ipv4(const struct nf_conntrack *ct, void *data)
20{
21	struct nfct_attr_grp_ipv4 *this = data;
22	this->src = ct->repl.src.v4;
23	this->dst = ct->repl.dst.v4;
24}
25
26static void get_attr_grp_orig_ipv6(const struct nf_conntrack *ct, void *data)
27{
28	struct nfct_attr_grp_ipv6 *this = data;
29	memcpy(this->src, &ct->head.orig.src.v6, sizeof(uint32_t)*4);
30	memcpy(this->dst, &ct->head.orig.dst.v6, sizeof(uint32_t)*4);
31}
32
33static void get_attr_grp_repl_ipv6(const struct nf_conntrack *ct, void *data)
34{
35	struct nfct_attr_grp_ipv6 *this = data;
36	memcpy(this->src, &ct->repl.src.v6, sizeof(uint32_t)*4);
37	memcpy(this->dst, &ct->repl.dst.v6, sizeof(uint32_t)*4);
38}
39
40static void get_attr_grp_orig_port(const struct nf_conntrack *ct, void *data)
41{
42	struct nfct_attr_grp_port *this = data;
43	this->sport = ct->head.orig.l4src.all;
44	this->dport = ct->head.orig.l4dst.all;
45}
46
47static void get_attr_grp_repl_port(const struct nf_conntrack *ct, void *data)
48{
49	struct nfct_attr_grp_port *this = data;
50	this->sport = ct->repl.l4src.all;
51	this->dport = ct->repl.l4dst.all;
52}
53
54static void get_attr_grp_icmp(const struct nf_conntrack *ct, void *data)
55{
56	struct nfct_attr_grp_icmp *this = data;
57	this->type = ct->head.orig.l4dst.icmp.type;
58	this->code = ct->head.orig.l4dst.icmp.code;
59	this->id = ct->head.orig.l4src.icmp.id;
60}
61
62static void get_attr_grp_master_ipv4(const struct nf_conntrack *ct, void *data)
63{
64	struct nfct_attr_grp_ipv4 *this = data;
65	this->src = ct->master.src.v4;
66	this->dst = ct->master.dst.v4;
67}
68
69static void get_attr_grp_master_ipv6(const struct nf_conntrack *ct, void *data)
70{
71	struct nfct_attr_grp_ipv6 *this = data;
72	memcpy(this->src, &ct->master.src.v6, sizeof(uint32_t)*4);
73	memcpy(this->dst, &ct->master.dst.v6, sizeof(uint32_t)*4);
74}
75
76static void get_attr_grp_master_port(const struct nf_conntrack *ct, void *data)
77{
78	struct nfct_attr_grp_port *this = data;
79	this->sport = ct->master.l4src.all;
80	this->dport = ct->master.l4dst.all;
81}
82
83static void get_attr_grp_orig_ctrs(const struct nf_conntrack *ct, void *data)
84{
85	struct nfct_attr_grp_ctrs *this = data;
86	this->packets = ct->counters[__DIR_ORIG].packets;
87	this->bytes = ct->counters[__DIR_ORIG].bytes;
88}
89
90static void get_attr_grp_repl_ctrs(const struct nf_conntrack *ct, void *data)
91{
92	struct nfct_attr_grp_ctrs *this = data;
93	this->packets = ct->counters[__DIR_REPL].packets;
94	this->bytes = ct->counters[__DIR_REPL].bytes;
95}
96
97static void
98get_attr_grp_orig_addr_src(const struct nf_conntrack *ct, void *data)
99{
100	union nfct_attr_grp_addr *this = data;
101	memcpy(&this->addr, &ct->head.orig.src, sizeof(ct->head.orig.src));
102}
103
104static void
105get_attr_grp_orig_addr_dst(const struct nf_conntrack *ct, void *data)
106{
107	union nfct_attr_grp_addr *this = data;
108	memcpy(&this->addr, &ct->head.orig.dst, sizeof(ct->head.orig.dst));
109}
110
111static void
112get_attr_grp_repl_addr_src(const struct nf_conntrack *ct, void *data)
113{
114	union nfct_attr_grp_addr *this = data;
115	memcpy(&this->addr, &ct->repl.src, sizeof(ct->repl.src));
116}
117
118static void
119get_attr_grp_repl_addr_dst(const struct nf_conntrack *ct, void *data)
120{
121	union nfct_attr_grp_addr *this = data;
122	memcpy(&this->addr, &ct->repl.dst, sizeof(ct->repl.dst));
123}
124
125const get_attr_grp get_attr_grp_array[ATTR_GRP_MAX] = {
126	[ATTR_GRP_ORIG_IPV4]		= get_attr_grp_orig_ipv4,
127	[ATTR_GRP_REPL_IPV4]		= get_attr_grp_repl_ipv4,
128	[ATTR_GRP_ORIG_IPV6]		= get_attr_grp_orig_ipv6,
129	[ATTR_GRP_REPL_IPV6]		= get_attr_grp_repl_ipv6,
130	[ATTR_GRP_ORIG_PORT]		= get_attr_grp_orig_port,
131	[ATTR_GRP_REPL_PORT]		= get_attr_grp_repl_port,
132	[ATTR_GRP_ICMP]			= get_attr_grp_icmp,
133	[ATTR_GRP_MASTER_IPV4]		= get_attr_grp_master_ipv4,
134	[ATTR_GRP_MASTER_IPV6]		= get_attr_grp_master_ipv6,
135	[ATTR_GRP_MASTER_PORT]		= get_attr_grp_master_port,
136	[ATTR_GRP_ORIG_COUNTERS]	= get_attr_grp_orig_ctrs,
137	[ATTR_GRP_REPL_COUNTERS]	= get_attr_grp_repl_ctrs,
138	[ATTR_GRP_ORIG_ADDR_SRC]	= get_attr_grp_orig_addr_src,
139	[ATTR_GRP_ORIG_ADDR_DST]	= get_attr_grp_orig_addr_dst,
140	[ATTR_GRP_REPL_ADDR_SRC]	= get_attr_grp_repl_addr_src,
141	[ATTR_GRP_REPL_ADDR_DST]	= get_attr_grp_repl_addr_dst,
142};
143