Commit | Line | Data |
---|---|---|
58401572 KPO |
1 | /* |
2 | * (C) 2008 Krzysztof Piotr Oledzki <ole@ans.pl> | |
3 | * | |
4 | * This program is free software; you can redistribute it and/or modify | |
5 | * it under the terms of the GNU General Public License version 2 as | |
6 | * published by the Free Software Foundation. | |
7 | */ | |
8 | ||
9 | #ifndef _NF_CONNTRACK_ACCT_H | |
10 | #define _NF_CONNTRACK_ACCT_H | |
d716a4df | 11 | #include <net/net_namespace.h> |
58401572 KPO |
12 | #include <linux/netfilter/nf_conntrack_common.h> |
13 | #include <linux/netfilter/nf_conntrack_tuple_common.h> | |
14 | #include <net/netfilter/nf_conntrack.h> | |
15 | #include <net/netfilter/nf_conntrack_extend.h> | |
16 | ||
17 | struct nf_conn_counter { | |
b3e0bfa7 ED |
18 | atomic64_t packets; |
19 | atomic64_t bytes; | |
58401572 KPO |
20 | }; |
21 | ||
f7b13e43 HE |
22 | struct nf_conn_acct { |
23 | struct nf_conn_counter counter[IP_CT_DIR_MAX]; | |
24 | }; | |
25 | ||
58401572 | 26 | static inline |
f7b13e43 | 27 | struct nf_conn_acct *nf_conn_acct_find(const struct nf_conn *ct) |
58401572 KPO |
28 | { |
29 | return nf_ct_ext_find(ct, NF_CT_EXT_ACCT); | |
30 | } | |
31 | ||
32 | static inline | |
f7b13e43 | 33 | struct nf_conn_acct *nf_ct_acct_ext_add(struct nf_conn *ct, gfp_t gfp) |
58401572 | 34 | { |
d716a4df | 35 | struct net *net = nf_ct_net(ct); |
f7b13e43 | 36 | struct nf_conn_acct *acct; |
58401572 | 37 | |
d716a4df | 38 | if (!net->ct.sysctl_acct) |
58401572 KPO |
39 | return NULL; |
40 | ||
41 | acct = nf_ct_ext_add(ct, NF_CT_EXT_ACCT, gfp); | |
42 | if (!acct) | |
43 | pr_debug("failed to add accounting extension area"); | |
44 | ||
45 | ||
46 | return acct; | |
47 | }; | |
48 | ||
a8756201 TG |
49 | /* Check if connection tracking accounting is enabled */ |
50 | static inline bool nf_ct_acct_enabled(struct net *net) | |
51 | { | |
52 | return net->ct.sysctl_acct != 0; | |
53 | } | |
54 | ||
55 | /* Enable/disable connection tracking accounting */ | |
56 | static inline void nf_ct_set_acct(struct net *net, bool enable) | |
57 | { | |
58 | net->ct.sysctl_acct = enable; | |
59 | } | |
60 | ||
fc3893fd | 61 | void nf_conntrack_acct_pernet_init(struct net *net); |
58401572 | 62 | |
4e77be46 JP |
63 | int nf_conntrack_acct_init(void); |
64 | void nf_conntrack_acct_fini(void); | |
58401572 | 65 | #endif /* _NF_CONNTRACK_ACCT_H */ |