include/net/netfilter/nf_conntrack_acct.h

   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 static inline
  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
  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 */
  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 */
  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_pernet_init(struct net *net);
  61 extern void nf_conntrack_acct_pernet_fini(struct net *net);
  62
  63 extern int nf_conntrack_acct_init(void);
  64 extern void nf_conntrack_acct_fini(void);
  65 #endif /* _NF_CONNTRACK_ACCT_H */