summaryrefslogtreecommitdiffstats
path: root/src/conntrack/grp_getter.c
blob: a8d43cf865692e098c8d87e695fd0119167f040d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/*
 * (C) 2005-2011 by Pablo Neira Ayuso <pablo@netfilter.org>
 *
 * 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 2 of the License, or
 * (at your option) any later version.
 */

#include "internal/internal.h"

static void get_attr_grp_orig_ipv4(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ipv4 *this = data;
	this->src = ct->head.orig.src.v4;
	this->dst = ct->head.orig.dst.v4;
}

static void get_attr_grp_repl_ipv4(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ipv4 *this = data;
	this->src = ct->repl.src.v4;
	this->dst = ct->repl.dst.v4;
}

static void get_attr_grp_orig_ipv6(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ipv6 *this = data;
	memcpy(this->src, &ct->head.orig.src.v6, sizeof(u_int32_t)*4);
	memcpy(this->dst, &ct->head.orig.dst.v6, sizeof(u_int32_t)*4);
}

static void get_attr_grp_repl_ipv6(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ipv6 *this = data;
	memcpy(this->src, &ct->repl.src.v6, sizeof(u_int32_t)*4);
	memcpy(this->dst, &ct->repl.dst.v6, sizeof(u_int32_t)*4);
}

static void get_attr_grp_orig_port(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_port *this = data;
	this->sport = ct->head.orig.l4src.all;
	this->dport = ct->head.orig.l4dst.all;
}

static void get_attr_grp_repl_port(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_port *this = data;
	this->sport = ct->repl.l4src.all;
	this->dport = ct->repl.l4dst.all;
}

static void get_attr_grp_icmp(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_icmp *this = data;
	this->type = ct->head.orig.l4dst.icmp.type;
	this->code = ct->head.orig.l4dst.icmp.code;
	this->id = ct->head.orig.l4src.icmp.id;
}

static void get_attr_grp_master_ipv4(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ipv4 *this = data;
	this->src = ct->master.src.v4;
	this->dst = ct->master.dst.v4;
}

static void get_attr_grp_master_ipv6(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ipv6 *this = data;
	memcpy(this->src, &ct->master.src.v6, sizeof(u_int32_t)*4);
	memcpy(this->dst, &ct->master.dst.v6, sizeof(u_int32_t)*4);
}

static void get_attr_grp_master_port(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_port *this = data;
	this->sport = ct->master.l4src.all;
	this->dport = ct->master.l4dst.all;
}

static void get_attr_grp_orig_ctrs(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ctrs *this = data;
	this->packets = ct->counters[__DIR_ORIG].packets;
	this->bytes = ct->counters[__DIR_ORIG].bytes;
}

static void get_attr_grp_repl_ctrs(const struct nf_conntrack *ct, void *data)
{
	struct nfct_attr_grp_ctrs *this = data;
	this->packets = ct->counters[__DIR_REPL].packets;
	this->bytes = ct->counters[__DIR_REPL].bytes;
}

const get_attr_grp get_attr_grp_array[ATTR_GRP_MAX] = {
	[ATTR_GRP_ORIG_IPV4]		= get_attr_grp_orig_ipv4,
	[ATTR_GRP_REPL_IPV4]		= get_attr_grp_repl_ipv4,
	[ATTR_GRP_ORIG_IPV6]		= get_attr_grp_orig_ipv6,
	[ATTR_GRP_REPL_IPV6]		= get_attr_grp_repl_ipv6,
	[ATTR_GRP_ORIG_PORT]		= get_attr_grp_orig_port,
	[ATTR_GRP_REPL_PORT]		= get_attr_grp_repl_port,
	[ATTR_GRP_ICMP]			= get_attr_grp_icmp,
	[ATTR_GRP_MASTER_IPV4]		= get_attr_grp_master_ipv4,
	[ATTR_GRP_MASTER_IPV6]		= get_attr_grp_master_ipv6,
	[ATTR_GRP_MASTER_PORT]		= get_attr_grp_master_port,
	[ATTR_GRP_ORIG_COUNTERS]	= get_attr_grp_orig_ctrs,
	[ATTR_GRP_REPL_COUNTERS]	= get_attr_grp_repl_ctrs
};