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 struct nf_conn_acct {
  23         struct nf_conn_counter counter[IP_CT_DIR_MAX];
  24 };
  25
  26 static inline
  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
  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 */
  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 */
  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 */