netfilter/nf_conntrack_labels.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2
   3 #ifndef _NF_CONNTRACK_LABELS_H
   4 #define _NF_CONNTRACK_LABELS_H
   5
   6 #include <linux/netfilter/nf_conntrack_common.h>
   7 #include <linux/netfilter/nf_conntrack_tuple_common.h>
   8 #include <linux/types.h>
   9 #include <net/net_namespace.h>
  10 #include <net/netfilter/nf_conntrack.h>
  11 #include <net/netfilter/nf_conntrack_extend.h>
  12 #include <uapi/linux/netfilter/xt_connlabel.h>
  13
  14 #define NF_CT_LABELS_MAX_SIZE ((XT_CONNLABEL_MAXBIT + 1) / BITS_PER_BYTE)
  15
  16 struct nf_conn_labels {
  17         unsigned long bits[NF_CT_LABELS_MAX_SIZE / sizeof(long)];
  18 };
  19
  20 /* Can't use nf_ct_ext_find(), flow dissector cannot use symbols
  21  * exported by nf_conntrack module.
  22  */
  23 static inline struct nf_conn_labels *nf_ct_labels_find(const struct nf_conn *ct)
  24 {
  25 #ifdef CONFIG_NF_CONNTRACK_LABELS
  26         struct nf_ct_ext *ext = ct->ext;
  27
  28         if (!ext || !__nf_ct_ext_exist(ext, NF_CT_EXT_LABELS))
  29                 return NULL;
  30
  31         return (void *)ct->ext + ct->ext->offset[NF_CT_EXT_LABELS];
  32 #else
  33         return NULL;
  34 #endif
  35 }
  36
  37 static inline struct nf_conn_labels *nf_ct_labels_ext_add(struct nf_conn *ct)
  38 {
  39 #ifdef CONFIG_NF_CONNTRACK_LABELS
  40         struct net *net = nf_ct_net(ct);
  41
  42         if (atomic_read(&net->ct.labels_used) == 0)
  43                 return NULL;
  44
  45         return nf_ct_ext_add(ct, NF_CT_EXT_LABELS, GFP_ATOMIC);
  46 #else
  47         return NULL;
  48 #endif
  49 }
  50
  51 int nf_connlabels_replace(struct nf_conn *ct,
  52                           const u32 *data, const u32 *mask, unsigned int words);
  53
  54 #ifdef CONFIG_NF_CONNTRACK_LABELS
  55 int nf_connlabels_get(struct net *net, unsigned int bit);
  56 void nf_connlabels_put(struct net *net);
  57 #else
  58 static inline int nf_connlabels_get(struct net *net, unsigned int bit) { return 0; }
  59 static inline void nf_connlabels_put(struct net *net) {}
  60 #endif
  61
  62 #endif /* _NF_CONNTRACK_LABELS_H */