include/net/netfilter/nf_conntrack_timestamp.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef _NF_CONNTRACK_TSTAMP_H
   3 #define _NF_CONNTRACK_TSTAMP_H
   4
   5 #include <net/net_namespace.h>
   6 #include <linux/netfilter/nf_conntrack_common.h>
   7 #include <linux/netfilter/nf_conntrack_tuple_common.h>
   8 #include <net/netfilter/nf_conntrack.h>
   9 #include <net/netfilter/nf_conntrack_extend.h>
  10
  11 struct nf_conn_tstamp {
  12         u_int64_t start;
  13         u_int64_t stop;
  14 };
  15
  16 static inline
  17 struct nf_conn_tstamp *nf_conn_tstamp_find(const struct nf_conn *ct)
  18 {
  19 #ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
  20         return nf_ct_ext_find(ct, NF_CT_EXT_TSTAMP);
  21 #else
  22         return NULL;
  23 #endif
  24 }
  25
  26 static inline
  27 struct nf_conn_tstamp *nf_ct_tstamp_ext_add(struct nf_conn *ct, gfp_t gfp)
  28 {
  29 #ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
  30         struct net *net = nf_ct_net(ct);
  31
  32         if (!net->ct.sysctl_tstamp)
  33                 return NULL;
  34
  35         return nf_ct_ext_add(ct, NF_CT_EXT_TSTAMP, gfp);
  36 #else
  37         return NULL;
  38 #endif
  39 };
  40
  41 static inline bool nf_ct_tstamp_enabled(struct net *net)
  42 {
  43         return net->ct.sysctl_tstamp != 0;
  44 }
  45
  46 static inline void nf_ct_set_tstamp(struct net *net, bool enable)
  47 {
  48         net->ct.sysctl_tstamp = enable;
  49 }
  50
  51 #ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
  52 int nf_conntrack_tstamp_pernet_init(struct net *net);
  53 void nf_conntrack_tstamp_pernet_fini(struct net *net);
  54
  55 int nf_conntrack_tstamp_init(void);
  56 void nf_conntrack_tstamp_fini(void);
  57 #else
  58 static inline int nf_conntrack_tstamp_pernet_init(struct net *net)
  59 {
  60         return 0;
  61 }
  62
  63 static inline void nf_conntrack_tstamp_pernet_fini(struct net *net)
  64 {
  65         return;
  66 }
  67
  68 static inline int nf_conntrack_tstamp_init(void)
  69 {
  70         return 0;
  71 }
  72
  73 static inline void nf_conntrack_tstamp_fini(void)
  74 {
  75         return;
  76 }
  77 #endif /* CONFIG_NF_CONNTRACK_TIMESTAMP */
  78
  79 #endif /* _NF_CONNTRACK_TSTAMP_H */