xref: /openbmc/linux/include/net/netfilter/nf_conntrack_acct.h (revision c900529f3d9161bfde5cca0754f83b4d3c3e0220)
1  /* SPDX-License-Identifier: GPL-2.0-only */
2  /*
3   * (C) 2008 Krzysztof Piotr Oledzki <ole@ans.pl>
4   */
5  
6  #ifndef _NF_CONNTRACK_ACCT_H
7  #define _NF_CONNTRACK_ACCT_H
8  #include <net/net_namespace.h>
9  #include <linux/netfilter/nf_conntrack_common.h>
10  #include <linux/netfilter/nf_conntrack_tuple_common.h>
11  #include <net/netfilter/nf_conntrack.h>
12  #include <net/netfilter/nf_conntrack_extend.h>
13  
14  struct nf_conn_counter {
15  	atomic64_t packets;
16  	atomic64_t bytes;
17  };
18  
19  struct nf_conn_acct {
20  	struct nf_conn_counter counter[IP_CT_DIR_MAX];
21  };
22  
23  static inline
nf_conn_acct_find(const struct nf_conn * ct)24  struct nf_conn_acct *nf_conn_acct_find(const struct nf_conn *ct)
25  {
26  	return nf_ct_ext_find(ct, NF_CT_EXT_ACCT);
27  }
28  
29  static inline
nf_ct_acct_ext_add(struct nf_conn * ct,gfp_t gfp)30  struct nf_conn_acct *nf_ct_acct_ext_add(struct nf_conn *ct, gfp_t gfp)
31  {
32  #if IS_ENABLED(CONFIG_NF_CONNTRACK)
33  	struct net *net = nf_ct_net(ct);
34  	struct nf_conn_acct *acct;
35  
36  	if (!net->ct.sysctl_acct)
37  		return NULL;
38  
39  	acct = nf_ct_ext_add(ct, NF_CT_EXT_ACCT, gfp);
40  	if (!acct)
41  		pr_debug("failed to add accounting extension area");
42  
43  
44  	return acct;
45  #else
46  	return NULL;
47  #endif
48  }
49  
50  /* Check if connection tracking accounting is enabled */
nf_ct_acct_enabled(struct net * net)51  static inline bool nf_ct_acct_enabled(struct net *net)
52  {
53  #if IS_ENABLED(CONFIG_NF_CONNTRACK)
54  	return net->ct.sysctl_acct != 0;
55  #else
56  	return false;
57  #endif
58  }
59  
60  /* Enable/disable connection tracking accounting */
nf_ct_set_acct(struct net * net,bool enable)61  static inline void nf_ct_set_acct(struct net *net, bool enable)
62  {
63  #if IS_ENABLED(CONFIG_NF_CONNTRACK)
64  	net->ct.sysctl_acct = enable;
65  #endif
66  }
67  
68  void nf_ct_acct_add(struct nf_conn *ct, u32 dir, unsigned int packets,
69  		    unsigned int bytes);
70  
nf_ct_acct_update(struct nf_conn * ct,u32 dir,unsigned int bytes)71  static inline void nf_ct_acct_update(struct nf_conn *ct, u32 dir,
72  				     unsigned int bytes)
73  {
74  #if IS_ENABLED(CONFIG_NF_CONNTRACK)
75  	nf_ct_acct_add(ct, dir, 1, bytes);
76  #endif
77  }
78  
79  void nf_conntrack_acct_pernet_init(struct net *net);
80  
81  #endif /* _NF_CONNTRACK_ACCT_H */
82