1 /* SPDX-License-Identifier: GPL-2.0 */
3 #ifndef _NF_CONNTRACK_LABELS_H
4 #define _NF_CONNTRACK_LABELS_H
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>
14 #define NF_CT_LABELS_MAX_SIZE ((XT_CONNLABEL_MAXBIT + 1) / BITS_PER_BYTE)
16 struct nf_conn_labels {
17 unsigned long bits[NF_CT_LABELS_MAX_SIZE / sizeof(long)];
20 static inline struct nf_conn_labels *nf_ct_labels_find(const struct nf_conn *ct)
22 #ifdef CONFIG_NF_CONNTRACK_LABELS
23 return nf_ct_ext_find(ct, NF_CT_EXT_LABELS);
29 static inline struct nf_conn_labels *nf_ct_labels_ext_add(struct nf_conn *ct)
31 #ifdef CONFIG_NF_CONNTRACK_LABELS
32 struct net *net = nf_ct_net(ct);
34 if (net->ct.labels_used == 0)
37 return nf_ct_ext_add(ct, NF_CT_EXT_LABELS, GFP_ATOMIC);
43 int nf_connlabels_replace(struct nf_conn *ct,
44 const u32 *data, const u32 *mask, unsigned int words);
46 #ifdef CONFIG_NF_CONNTRACK_LABELS
47 int nf_conntrack_labels_init(void);
48 void nf_conntrack_labels_fini(void);
49 int nf_connlabels_get(struct net *net, unsigned int bit);
50 void nf_connlabels_put(struct net *net);
52 static inline int nf_conntrack_labels_init(void) { return 0; }
53 static inline void nf_conntrack_labels_fini(void) {}
54 static inline int nf_connlabels_get(struct net *net, unsigned int bit) { return 0; }
55 static inline void nf_connlabels_put(struct net *net) {}
58 #endif /* _NF_CONNTRACK_LABELS_H */