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
14struct nf_conn_counter {
15 atomic64_t packets;
16 atomic64_t bytes;
17};
18
19struct nf_conn_acct {
20 struct nf_conn_counter counter[IP_CT_DIR_MAX];
21};
22
23static inline
24struct nf_conn_acct *nf_conn_acct_find(const struct nf_conn *ct)
25{
26 return nf_ct_ext_find(ct, id: NF_CT_EXT_ACCT);
27}
28
29static inline
30struct 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, id: 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 */
51static 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 */
61static 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
68void nf_ct_acct_add(struct nf_conn *ct, u32 dir, unsigned int packets,
69 unsigned int bytes);
70
71static 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, packets: 1, bytes);
76#endif
77}
78
79void nf_conntrack_acct_pernet_init(struct net *net);
80
81#endif /* _NF_CONNTRACK_ACCT_H */
82