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 	atomic64_t packets;
19 	atomic64_t bytes;
20 };
21 
22 struct nf_conn_acct {
23 	struct nf_conn_counter counter[IP_CT_DIR_MAX];
24 };
25 
26 static inline
nf_conn_acct_find(const struct nf_conn * ct)27 struct nf_conn_acct *nf_conn_acct_find(const struct nf_conn *ct)
28 {
29 	return nf_ct_ext_find(ct, NF_CT_EXT_ACCT);
30 }
31 
32 static inline
nf_ct_acct_ext_add(struct nf_conn * ct,gfp_t gfp)33 struct nf_conn_acct *nf_ct_acct_ext_add(struct nf_conn *ct, gfp_t gfp)
34 {
35 	struct net *net = nf_ct_net(ct);
36 	struct nf_conn_acct *acct;
37 
38 	if (!net->ct.sysctl_acct)
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 
49 unsigned int seq_print_acct(struct seq_file *s, const struct nf_conn *ct,
50 			    int dir);
51 
52 /* Check if connection tracking accounting is enabled */
nf_ct_acct_enabled(struct net * net)53 static inline bool nf_ct_acct_enabled(struct net *net)
54 {
55 	return net->ct.sysctl_acct != 0;
56 }
57 
58 /* Enable/disable connection tracking accounting */
nf_ct_set_acct(struct net * net,bool enable)59 static inline void nf_ct_set_acct(struct net *net, bool enable)
60 {
61 	net->ct.sysctl_acct = enable;
62 }
63 
64 int nf_conntrack_acct_pernet_init(struct net *net);
65 void nf_conntrack_acct_pernet_fini(struct net *net);
66 
67 int nf_conntrack_acct_init(void);
68 void nf_conntrack_acct_fini(void);
69 #endif /* _NF_CONNTRACK_ACCT_H */
70