d2bf2f34cc
This also adds NF_CT_LABELS_MAX_SIZE so it can be re-used as BUILD_BUG_ON in nft_ct. At this time, nft doesn't yet support writing to the label area; when this changes the label->words handling needs to be moved out of xt_connlabel.c into nf_conntrack_labels.c. Also removes a useless run-time check: words cannot grow beyond 4 (32 bit) or 2 (64bit) since xt_connlabel enforces a maximum of 128 labels. Signed-off-by: Florian Westphal <fw@strlen.de> Acked-by: Patrick McHardy <kaber@trash.net> Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
61 lines
1.5 KiB
C
61 lines
1.5 KiB
C
#include <linux/types.h>
|
|
#include <net/net_namespace.h>
|
|
#include <linux/netfilter/nf_conntrack_common.h>
|
|
#include <linux/netfilter/nf_conntrack_tuple_common.h>
|
|
#include <net/netfilter/nf_conntrack.h>
|
|
#include <net/netfilter/nf_conntrack_extend.h>
|
|
|
|
#include <uapi/linux/netfilter/xt_connlabel.h>
|
|
|
|
#define NF_CT_LABELS_MAX_SIZE ((XT_CONNLABEL_MAXBIT + 1) / BITS_PER_BYTE)
|
|
|
|
struct nf_conn_labels {
|
|
u8 words;
|
|
unsigned long bits[];
|
|
};
|
|
|
|
static inline struct nf_conn_labels *nf_ct_labels_find(const struct nf_conn *ct)
|
|
{
|
|
#ifdef CONFIG_NF_CONNTRACK_LABELS
|
|
return nf_ct_ext_find(ct, NF_CT_EXT_LABELS);
|
|
#else
|
|
return NULL;
|
|
#endif
|
|
}
|
|
|
|
static inline struct nf_conn_labels *nf_ct_labels_ext_add(struct nf_conn *ct)
|
|
{
|
|
#ifdef CONFIG_NF_CONNTRACK_LABELS
|
|
struct nf_conn_labels *cl_ext;
|
|
struct net *net = nf_ct_net(ct);
|
|
u8 words;
|
|
|
|
words = ACCESS_ONCE(net->ct.label_words);
|
|
if (words == 0)
|
|
return NULL;
|
|
|
|
cl_ext = nf_ct_ext_add_length(ct, NF_CT_EXT_LABELS,
|
|
words * sizeof(long), GFP_ATOMIC);
|
|
if (cl_ext != NULL)
|
|
cl_ext->words = words;
|
|
|
|
return cl_ext;
|
|
#else
|
|
return NULL;
|
|
#endif
|
|
}
|
|
|
|
bool nf_connlabel_match(const struct nf_conn *ct, u16 bit);
|
|
int nf_connlabel_set(struct nf_conn *ct, u16 bit);
|
|
|
|
int nf_connlabels_replace(struct nf_conn *ct,
|
|
const u32 *data, const u32 *mask, unsigned int words);
|
|
|
|
#ifdef CONFIG_NF_CONNTRACK_LABELS
|
|
int nf_conntrack_labels_init(void);
|
|
void nf_conntrack_labels_fini(void);
|
|
#else
|
|
static inline int nf_conntrack_labels_init(void) { return 0; }
|
|
static inline void nf_conntrack_labels_fini(void) {}
|
|
#endif
|