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
11 #include <net/net_namespace.h>
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 {
18 	u_int64_t packets;
19 	u_int64_t bytes;
20 };
21 
22 static inline
nf_conn_acct_find(const struct nf_conn * ct)23 struct nf_conn_counter *nf_conn_acct_find(const struct nf_conn *ct)
24 {
25 	return nf_ct_ext_find(ct, NF_CT_EXT_ACCT);
26 }
27 
28 static inline
nf_ct_acct_ext_add(struct nf_conn * ct,gfp_t gfp)29 struct nf_conn_counter *nf_ct_acct_ext_add(struct nf_conn *ct, gfp_t gfp)
30 {
31 	struct net *net = nf_ct_net(ct);
32 	struct nf_conn_counter *acct;
33 
34 	if (!net->ct.sysctl_acct)
35 		return NULL;
36 
37 	acct = nf_ct_ext_add(ct, NF_CT_EXT_ACCT, gfp);
38 	if (!acct)
39 		pr_debug("failed to add accounting extension area");
40 
41 
42 	return acct;
43 };
44 
45 extern unsigned int
46 seq_print_acct(struct seq_file *s, const struct nf_conn *ct, int dir);
47 
48 /* Check if connection tracking accounting is enabled */
nf_ct_acct_enabled(struct net * net)49 static inline bool nf_ct_acct_enabled(struct net *net)
50 {
51 	return net->ct.sysctl_acct != 0;
52 }
53 
54 /* Enable/disable connection tracking accounting */
nf_ct_set_acct(struct net * net,bool enable)55 static inline void nf_ct_set_acct(struct net *net, bool enable)
56 {
57 	net->ct.sysctl_acct = enable;
58 }
59 
60 extern int nf_conntrack_acct_init(struct net *net);
61 extern void nf_conntrack_acct_fini(struct net *net);
62 
63 #endif /* _NF_CONNTRACK_ACCT_H */
64