Linux Audio

Check our new training course

Loading...
v5.14.15
  1// SPDX-License-Identifier: GPL-2.0-only
  2/* Copyright (C) 2000-2002 Joakim Axelsson <gozem@linux.nu>
  3 *                         Patrick Schaaf <bof@bof.de>
  4 *                         Martin Josefsson <gandalf@wlug.westbo.se>
  5 * Copyright (C) 2003-2013 Jozsef Kadlecsik <kadlec@netfilter.org>
 
 
 
 
  6 */
  7
  8/* Kernel module which implements the set match and SET target
  9 * for netfilter/iptables.
 10 */
 11
 12#include <linux/module.h>
 13#include <linux/skbuff.h>
 14
 15#include <linux/netfilter/x_tables.h>
 16#include <linux/netfilter/ipset/ip_set.h>
 
 17#include <uapi/linux/netfilter/xt_set.h>
 18
 19MODULE_LICENSE("GPL");
 20MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@netfilter.org>");
 21MODULE_DESCRIPTION("Xtables: IP set match and target module");
 22MODULE_ALIAS("xt_SET");
 23MODULE_ALIAS("ipt_set");
 24MODULE_ALIAS("ip6t_set");
 25MODULE_ALIAS("ipt_SET");
 26MODULE_ALIAS("ip6t_SET");
 27
 28static inline int
 29match_set(ip_set_id_t index, const struct sk_buff *skb,
 30	  const struct xt_action_param *par,
 31	  struct ip_set_adt_opt *opt, int inv)
 32{
 33	if (ip_set_test(index, skb, par, opt))
 34		inv = !inv;
 35	return inv;
 36}
 37
 38#define ADT_OPT(n, f, d, fs, cfs, t, p, b, po, bo)	\
 39struct ip_set_adt_opt n = {				\
 40	.family	= f,					\
 41	.dim = d,					\
 42	.flags = fs,					\
 43	.cmdflags = cfs,				\
 44	.ext.timeout = t,				\
 45	.ext.packets = p,				\
 46	.ext.bytes = b,					\
 47	.ext.packets_op = po,				\
 48	.ext.bytes_op = bo,				\
 49}
 50
 51/* Revision 0 interface: backward compatible with netfilter/iptables */
 52
 53static bool
 54set_match_v0(const struct sk_buff *skb, struct xt_action_param *par)
 55{
 56	const struct xt_set_info_match_v0 *info = par->matchinfo;
 57
 58	ADT_OPT(opt, xt_family(par), info->match_set.u.compat.dim,
 59		info->match_set.u.compat.flags, 0, UINT_MAX,
 60		0, 0, 0, 0);
 61
 62	return match_set(info->match_set.index, skb, par, &opt,
 63			 info->match_set.u.compat.flags & IPSET_INV_MATCH);
 64}
 65
 66static void
 67compat_flags(struct xt_set_info_v0 *info)
 68{
 69	u_int8_t i;
 70
 71	/* Fill out compatibility data according to enum ip_set_kopt */
 72	info->u.compat.dim = IPSET_DIM_ZERO;
 73	if (info->u.flags[0] & IPSET_MATCH_INV)
 74		info->u.compat.flags |= IPSET_INV_MATCH;
 75	for (i = 0; i < IPSET_DIM_MAX - 1 && info->u.flags[i]; i++) {
 76		info->u.compat.dim++;
 77		if (info->u.flags[i] & IPSET_SRC)
 78			info->u.compat.flags |= (1 << info->u.compat.dim);
 79	}
 80}
 81
 82static int
 83set_match_v0_checkentry(const struct xt_mtchk_param *par)
 84{
 85	struct xt_set_info_match_v0 *info = par->matchinfo;
 86	ip_set_id_t index;
 87
 88	index = ip_set_nfnl_get_byindex(par->net, info->match_set.index);
 89
 90	if (index == IPSET_INVALID_ID) {
 91		pr_info_ratelimited("Cannot find set identified by id %u to match\n",
 92				    info->match_set.index);
 93		return -ENOENT;
 94	}
 95	if (info->match_set.u.flags[IPSET_DIM_MAX - 1] != 0) {
 96		pr_info_ratelimited("set match dimension is over the limit!\n");
 97		ip_set_nfnl_put(par->net, info->match_set.index);
 98		return -ERANGE;
 99	}
100
101	/* Fill out compatibility data */
102	compat_flags(&info->match_set);
103
104	return 0;
105}
106
107static void
108set_match_v0_destroy(const struct xt_mtdtor_param *par)
109{
110	struct xt_set_info_match_v0 *info = par->matchinfo;
111
112	ip_set_nfnl_put(par->net, info->match_set.index);
113}
114
115/* Revision 1 match */
116
117static bool
118set_match_v1(const struct sk_buff *skb, struct xt_action_param *par)
119{
120	const struct xt_set_info_match_v1 *info = par->matchinfo;
121
122	ADT_OPT(opt, xt_family(par), info->match_set.dim,
123		info->match_set.flags, 0, UINT_MAX,
124		0, 0, 0, 0);
125
126	if (opt.flags & IPSET_RETURN_NOMATCH)
127		opt.cmdflags |= IPSET_FLAG_RETURN_NOMATCH;
128
129	return match_set(info->match_set.index, skb, par, &opt,
130			 info->match_set.flags & IPSET_INV_MATCH);
131}
132
133static int
134set_match_v1_checkentry(const struct xt_mtchk_param *par)
135{
136	struct xt_set_info_match_v1 *info = par->matchinfo;
137	ip_set_id_t index;
138
139	index = ip_set_nfnl_get_byindex(par->net, info->match_set.index);
140
141	if (index == IPSET_INVALID_ID) {
142		pr_info_ratelimited("Cannot find set identified by id %u to match\n",
143				    info->match_set.index);
144		return -ENOENT;
145	}
146	if (info->match_set.dim > IPSET_DIM_MAX) {
147		pr_info_ratelimited("set match dimension is over the limit!\n");
148		ip_set_nfnl_put(par->net, info->match_set.index);
149		return -ERANGE;
150	}
151
152	return 0;
153}
154
155static void
156set_match_v1_destroy(const struct xt_mtdtor_param *par)
157{
158	struct xt_set_info_match_v1 *info = par->matchinfo;
159
160	ip_set_nfnl_put(par->net, info->match_set.index);
161}
162
163/* Revision 3 match */
164
165static bool
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
166set_match_v3(const struct sk_buff *skb, struct xt_action_param *par)
167{
168	const struct xt_set_info_match_v3 *info = par->matchinfo;
 
169
170	ADT_OPT(opt, xt_family(par), info->match_set.dim,
171		info->match_set.flags, info->flags, UINT_MAX,
172		info->packets.value, info->bytes.value,
173		info->packets.op, info->bytes.op);
174
175	if (info->packets.op != IPSET_COUNTER_NONE ||
176	    info->bytes.op != IPSET_COUNTER_NONE)
177		opt.cmdflags |= IPSET_FLAG_MATCH_COUNTERS;
178
179	return match_set(info->match_set.index, skb, par, &opt,
180			 info->match_set.flags & IPSET_INV_MATCH);
 
 
 
 
 
 
 
181}
182
183#define set_match_v3_checkentry	set_match_v1_checkentry
184#define set_match_v3_destroy	set_match_v1_destroy
185
186/* Revision 4 match */
187
188static bool
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
189set_match_v4(const struct sk_buff *skb, struct xt_action_param *par)
190{
191	const struct xt_set_info_match_v4 *info = par->matchinfo;
 
192
193	ADT_OPT(opt, xt_family(par), info->match_set.dim,
194		info->match_set.flags, info->flags, UINT_MAX,
195		info->packets.value, info->bytes.value,
196		info->packets.op, info->bytes.op);
197
198	if (info->packets.op != IPSET_COUNTER_NONE ||
199	    info->bytes.op != IPSET_COUNTER_NONE)
200		opt.cmdflags |= IPSET_FLAG_MATCH_COUNTERS;
201
202	return match_set(info->match_set.index, skb, par, &opt,
203			 info->match_set.flags & IPSET_INV_MATCH);
 
 
 
 
 
 
 
204}
205
206#define set_match_v4_checkentry	set_match_v1_checkentry
207#define set_match_v4_destroy	set_match_v1_destroy
208
209/* Revision 0 interface: backward compatible with netfilter/iptables */
210
211static unsigned int
212set_target_v0(struct sk_buff *skb, const struct xt_action_param *par)
213{
214	const struct xt_set_info_target_v0 *info = par->targinfo;
215
216	ADT_OPT(add_opt, xt_family(par), info->add_set.u.compat.dim,
217		info->add_set.u.compat.flags, 0, UINT_MAX,
218		0, 0, 0, 0);
219	ADT_OPT(del_opt, xt_family(par), info->del_set.u.compat.dim,
220		info->del_set.u.compat.flags, 0, UINT_MAX,
221		0, 0, 0, 0);
222
223	if (info->add_set.index != IPSET_INVALID_ID)
224		ip_set_add(info->add_set.index, skb, par, &add_opt);
225	if (info->del_set.index != IPSET_INVALID_ID)
226		ip_set_del(info->del_set.index, skb, par, &del_opt);
227
228	return XT_CONTINUE;
229}
230
231static int
232set_target_v0_checkentry(const struct xt_tgchk_param *par)
233{
234	struct xt_set_info_target_v0 *info = par->targinfo;
235	ip_set_id_t index;
236
237	if (info->add_set.index != IPSET_INVALID_ID) {
238		index = ip_set_nfnl_get_byindex(par->net, info->add_set.index);
239		if (index == IPSET_INVALID_ID) {
240			pr_info_ratelimited("Cannot find add_set index %u as target\n",
241					    info->add_set.index);
242			return -ENOENT;
243		}
244	}
245
246	if (info->del_set.index != IPSET_INVALID_ID) {
247		index = ip_set_nfnl_get_byindex(par->net, info->del_set.index);
248		if (index == IPSET_INVALID_ID) {
249			pr_info_ratelimited("Cannot find del_set index %u as target\n",
250					    info->del_set.index);
251			if (info->add_set.index != IPSET_INVALID_ID)
252				ip_set_nfnl_put(par->net, info->add_set.index);
253			return -ENOENT;
254		}
255	}
256	if (info->add_set.u.flags[IPSET_DIM_MAX - 1] != 0 ||
257	    info->del_set.u.flags[IPSET_DIM_MAX - 1] != 0) {
258		pr_info_ratelimited("SET target dimension over the limit!\n");
259		if (info->add_set.index != IPSET_INVALID_ID)
260			ip_set_nfnl_put(par->net, info->add_set.index);
261		if (info->del_set.index != IPSET_INVALID_ID)
262			ip_set_nfnl_put(par->net, info->del_set.index);
263		return -ERANGE;
264	}
265
266	/* Fill out compatibility data */
267	compat_flags(&info->add_set);
268	compat_flags(&info->del_set);
269
270	return 0;
271}
272
273static void
274set_target_v0_destroy(const struct xt_tgdtor_param *par)
275{
276	const struct xt_set_info_target_v0 *info = par->targinfo;
277
278	if (info->add_set.index != IPSET_INVALID_ID)
279		ip_set_nfnl_put(par->net, info->add_set.index);
280	if (info->del_set.index != IPSET_INVALID_ID)
281		ip_set_nfnl_put(par->net, info->del_set.index);
282}
283
284/* Revision 1 target */
285
286static unsigned int
287set_target_v1(struct sk_buff *skb, const struct xt_action_param *par)
288{
289	const struct xt_set_info_target_v1 *info = par->targinfo;
290
291	ADT_OPT(add_opt, xt_family(par), info->add_set.dim,
292		info->add_set.flags, 0, UINT_MAX,
293		0, 0, 0, 0);
294	ADT_OPT(del_opt, xt_family(par), info->del_set.dim,
295		info->del_set.flags, 0, UINT_MAX,
296		0, 0, 0, 0);
297
298	if (info->add_set.index != IPSET_INVALID_ID)
299		ip_set_add(info->add_set.index, skb, par, &add_opt);
300	if (info->del_set.index != IPSET_INVALID_ID)
301		ip_set_del(info->del_set.index, skb, par, &del_opt);
302
303	return XT_CONTINUE;
304}
305
306static int
307set_target_v1_checkentry(const struct xt_tgchk_param *par)
308{
309	const struct xt_set_info_target_v1 *info = par->targinfo;
310	ip_set_id_t index;
311
312	if (info->add_set.index != IPSET_INVALID_ID) {
313		index = ip_set_nfnl_get_byindex(par->net, info->add_set.index);
314		if (index == IPSET_INVALID_ID) {
315			pr_info_ratelimited("Cannot find add_set index %u as target\n",
316					    info->add_set.index);
317			return -ENOENT;
318		}
319	}
320
321	if (info->del_set.index != IPSET_INVALID_ID) {
322		index = ip_set_nfnl_get_byindex(par->net, info->del_set.index);
323		if (index == IPSET_INVALID_ID) {
324			pr_info_ratelimited("Cannot find del_set index %u as target\n",
325					    info->del_set.index);
326			if (info->add_set.index != IPSET_INVALID_ID)
327				ip_set_nfnl_put(par->net, info->add_set.index);
328			return -ENOENT;
329		}
330	}
331	if (info->add_set.dim > IPSET_DIM_MAX ||
332	    info->del_set.dim > IPSET_DIM_MAX) {
333		pr_info_ratelimited("SET target dimension over the limit!\n");
334		if (info->add_set.index != IPSET_INVALID_ID)
335			ip_set_nfnl_put(par->net, info->add_set.index);
336		if (info->del_set.index != IPSET_INVALID_ID)
337			ip_set_nfnl_put(par->net, info->del_set.index);
338		return -ERANGE;
339	}
340
341	return 0;
342}
343
344static void
345set_target_v1_destroy(const struct xt_tgdtor_param *par)
346{
347	const struct xt_set_info_target_v1 *info = par->targinfo;
348
349	if (info->add_set.index != IPSET_INVALID_ID)
350		ip_set_nfnl_put(par->net, info->add_set.index);
351	if (info->del_set.index != IPSET_INVALID_ID)
352		ip_set_nfnl_put(par->net, info->del_set.index);
353}
354
355/* Revision 2 target */
356
357static unsigned int
358set_target_v2(struct sk_buff *skb, const struct xt_action_param *par)
359{
360	const struct xt_set_info_target_v2 *info = par->targinfo;
361
362	ADT_OPT(add_opt, xt_family(par), info->add_set.dim,
363		info->add_set.flags, info->flags, info->timeout,
364		0, 0, 0, 0);
365	ADT_OPT(del_opt, xt_family(par), info->del_set.dim,
366		info->del_set.flags, 0, UINT_MAX,
367		0, 0, 0, 0);
368
369	/* Normalize to fit into jiffies */
370	if (add_opt.ext.timeout != IPSET_NO_TIMEOUT &&
371	    add_opt.ext.timeout > IPSET_MAX_TIMEOUT)
372		add_opt.ext.timeout = IPSET_MAX_TIMEOUT;
373	if (info->add_set.index != IPSET_INVALID_ID)
374		ip_set_add(info->add_set.index, skb, par, &add_opt);
375	if (info->del_set.index != IPSET_INVALID_ID)
376		ip_set_del(info->del_set.index, skb, par, &del_opt);
377
378	return XT_CONTINUE;
379}
380
381#define set_target_v2_checkentry	set_target_v1_checkentry
382#define set_target_v2_destroy		set_target_v1_destroy
383
384/* Revision 3 target */
385
386#define MOPT(opt, member)	((opt).ext.skbinfo.member)
387
388static unsigned int
389set_target_v3(struct sk_buff *skb, const struct xt_action_param *par)
390{
391	const struct xt_set_info_target_v3 *info = par->targinfo;
392	int ret;
393
394	ADT_OPT(add_opt, xt_family(par), info->add_set.dim,
395		info->add_set.flags, info->flags, info->timeout,
396		0, 0, 0, 0);
397	ADT_OPT(del_opt, xt_family(par), info->del_set.dim,
398		info->del_set.flags, 0, UINT_MAX,
399		0, 0, 0, 0);
400	ADT_OPT(map_opt, xt_family(par), info->map_set.dim,
401		info->map_set.flags, 0, UINT_MAX,
402		0, 0, 0, 0);
403
404	/* Normalize to fit into jiffies */
405	if (add_opt.ext.timeout != IPSET_NO_TIMEOUT &&
406	    add_opt.ext.timeout > IPSET_MAX_TIMEOUT)
407		add_opt.ext.timeout = IPSET_MAX_TIMEOUT;
408	if (info->add_set.index != IPSET_INVALID_ID)
409		ip_set_add(info->add_set.index, skb, par, &add_opt);
410	if (info->del_set.index != IPSET_INVALID_ID)
411		ip_set_del(info->del_set.index, skb, par, &del_opt);
412	if (info->map_set.index != IPSET_INVALID_ID) {
413		map_opt.cmdflags |= info->flags & (IPSET_FLAG_MAP_SKBMARK |
414						   IPSET_FLAG_MAP_SKBPRIO |
415						   IPSET_FLAG_MAP_SKBQUEUE);
416		ret = match_set(info->map_set.index, skb, par, &map_opt,
417				info->map_set.flags & IPSET_INV_MATCH);
418		if (!ret)
419			return XT_CONTINUE;
420		if (map_opt.cmdflags & IPSET_FLAG_MAP_SKBMARK)
421			skb->mark = (skb->mark & ~MOPT(map_opt,skbmarkmask))
422				    ^ MOPT(map_opt, skbmark);
423		if (map_opt.cmdflags & IPSET_FLAG_MAP_SKBPRIO)
424			skb->priority = MOPT(map_opt, skbprio);
425		if ((map_opt.cmdflags & IPSET_FLAG_MAP_SKBQUEUE) &&
426		    skb->dev &&
427		    skb->dev->real_num_tx_queues > MOPT(map_opt, skbqueue))
428			skb_set_queue_mapping(skb, MOPT(map_opt, skbqueue));
429	}
430	return XT_CONTINUE;
431}
432
433static int
434set_target_v3_checkentry(const struct xt_tgchk_param *par)
435{
436	const struct xt_set_info_target_v3 *info = par->targinfo;
437	ip_set_id_t index;
438	int ret = 0;
439
440	if (info->add_set.index != IPSET_INVALID_ID) {
441		index = ip_set_nfnl_get_byindex(par->net,
442						info->add_set.index);
443		if (index == IPSET_INVALID_ID) {
444			pr_info_ratelimited("Cannot find add_set index %u as target\n",
445					    info->add_set.index);
446			return -ENOENT;
447		}
448	}
449
450	if (info->del_set.index != IPSET_INVALID_ID) {
451		index = ip_set_nfnl_get_byindex(par->net,
452						info->del_set.index);
453		if (index == IPSET_INVALID_ID) {
454			pr_info_ratelimited("Cannot find del_set index %u as target\n",
455					    info->del_set.index);
456			ret = -ENOENT;
457			goto cleanup_add;
 
 
458		}
459	}
460
461	if (info->map_set.index != IPSET_INVALID_ID) {
462		if (strncmp(par->table, "mangle", 7)) {
463			pr_info_ratelimited("--map-set only usable from mangle table\n");
464			ret = -EINVAL;
465			goto cleanup_del;
466		}
467		if (((info->flags & IPSET_FLAG_MAP_SKBPRIO) |
468		     (info->flags & IPSET_FLAG_MAP_SKBQUEUE)) &&
469		     (par->hook_mask & ~(1 << NF_INET_FORWARD |
470					 1 << NF_INET_LOCAL_OUT |
471					 1 << NF_INET_POST_ROUTING))) {
472			pr_info_ratelimited("mapping of prio or/and queue is allowed only from OUTPUT/FORWARD/POSTROUTING chains\n");
473			ret = -EINVAL;
474			goto cleanup_del;
475		}
476		index = ip_set_nfnl_get_byindex(par->net,
477						info->map_set.index);
478		if (index == IPSET_INVALID_ID) {
479			pr_info_ratelimited("Cannot find map_set index %u as target\n",
480					    info->map_set.index);
481			ret = -ENOENT;
482			goto cleanup_del;
 
 
 
 
 
483		}
484	}
485
486	if (info->add_set.dim > IPSET_DIM_MAX ||
487	    info->del_set.dim > IPSET_DIM_MAX ||
488	    info->map_set.dim > IPSET_DIM_MAX) {
489		pr_info_ratelimited("SET target dimension over the limit!\n");
490		ret = -ERANGE;
491		goto cleanup_mark;
 
 
 
 
 
492	}
493
494	return 0;
495cleanup_mark:
496	if (info->map_set.index != IPSET_INVALID_ID)
497		ip_set_nfnl_put(par->net, info->map_set.index);
498cleanup_del:
499	if (info->del_set.index != IPSET_INVALID_ID)
500		ip_set_nfnl_put(par->net, info->del_set.index);
501cleanup_add:
502	if (info->add_set.index != IPSET_INVALID_ID)
503		ip_set_nfnl_put(par->net, info->add_set.index);
504	return ret;
505}
506
507static void
508set_target_v3_destroy(const struct xt_tgdtor_param *par)
509{
510	const struct xt_set_info_target_v3 *info = par->targinfo;
511
512	if (info->add_set.index != IPSET_INVALID_ID)
513		ip_set_nfnl_put(par->net, info->add_set.index);
514	if (info->del_set.index != IPSET_INVALID_ID)
515		ip_set_nfnl_put(par->net, info->del_set.index);
516	if (info->map_set.index != IPSET_INVALID_ID)
517		ip_set_nfnl_put(par->net, info->map_set.index);
518}
519
520static struct xt_match set_matches[] __read_mostly = {
521	{
522		.name		= "set",
523		.family		= NFPROTO_IPV4,
524		.revision	= 0,
525		.match		= set_match_v0,
526		.matchsize	= sizeof(struct xt_set_info_match_v0),
527		.checkentry	= set_match_v0_checkentry,
528		.destroy	= set_match_v0_destroy,
529		.me		= THIS_MODULE
530	},
531	{
532		.name		= "set",
533		.family		= NFPROTO_IPV4,
534		.revision	= 1,
535		.match		= set_match_v1,
536		.matchsize	= sizeof(struct xt_set_info_match_v1),
537		.checkentry	= set_match_v1_checkentry,
538		.destroy	= set_match_v1_destroy,
539		.me		= THIS_MODULE
540	},
541	{
542		.name		= "set",
543		.family		= NFPROTO_IPV6,
544		.revision	= 1,
545		.match		= set_match_v1,
546		.matchsize	= sizeof(struct xt_set_info_match_v1),
547		.checkentry	= set_match_v1_checkentry,
548		.destroy	= set_match_v1_destroy,
549		.me		= THIS_MODULE
550	},
551	/* --return-nomatch flag support */
552	{
553		.name		= "set",
554		.family		= NFPROTO_IPV4,
555		.revision	= 2,
556		.match		= set_match_v1,
557		.matchsize	= sizeof(struct xt_set_info_match_v1),
558		.checkentry	= set_match_v1_checkentry,
559		.destroy	= set_match_v1_destroy,
560		.me		= THIS_MODULE
561	},
562	{
563		.name		= "set",
564		.family		= NFPROTO_IPV6,
565		.revision	= 2,
566		.match		= set_match_v1,
567		.matchsize	= sizeof(struct xt_set_info_match_v1),
568		.checkentry	= set_match_v1_checkentry,
569		.destroy	= set_match_v1_destroy,
570		.me		= THIS_MODULE
571	},
572	/* counters support: update, match */
573	{
574		.name		= "set",
575		.family		= NFPROTO_IPV4,
576		.revision	= 3,
577		.match		= set_match_v3,
578		.matchsize	= sizeof(struct xt_set_info_match_v3),
579		.checkentry	= set_match_v3_checkentry,
580		.destroy	= set_match_v3_destroy,
581		.me		= THIS_MODULE
582	},
583	{
584		.name		= "set",
585		.family		= NFPROTO_IPV6,
586		.revision	= 3,
587		.match		= set_match_v3,
588		.matchsize	= sizeof(struct xt_set_info_match_v3),
589		.checkentry	= set_match_v3_checkentry,
590		.destroy	= set_match_v3_destroy,
591		.me		= THIS_MODULE
592	},
593	/* new revision for counters support: update, match */
594	{
595		.name		= "set",
596		.family		= NFPROTO_IPV4,
597		.revision	= 4,
598		.match		= set_match_v4,
599		.matchsize	= sizeof(struct xt_set_info_match_v4),
600		.checkentry	= set_match_v4_checkentry,
601		.destroy	= set_match_v4_destroy,
602		.me		= THIS_MODULE
603	},
604	{
605		.name		= "set",
606		.family		= NFPROTO_IPV6,
607		.revision	= 4,
608		.match		= set_match_v4,
609		.matchsize	= sizeof(struct xt_set_info_match_v4),
610		.checkentry	= set_match_v4_checkentry,
611		.destroy	= set_match_v4_destroy,
612		.me		= THIS_MODULE
613	},
614};
615
616static struct xt_target set_targets[] __read_mostly = {
617	{
618		.name		= "SET",
619		.revision	= 0,
620		.family		= NFPROTO_IPV4,
621		.target		= set_target_v0,
622		.targetsize	= sizeof(struct xt_set_info_target_v0),
623		.checkentry	= set_target_v0_checkentry,
624		.destroy	= set_target_v0_destroy,
625		.me		= THIS_MODULE
626	},
627	{
628		.name		= "SET",
629		.revision	= 1,
630		.family		= NFPROTO_IPV4,
631		.target		= set_target_v1,
632		.targetsize	= sizeof(struct xt_set_info_target_v1),
633		.checkentry	= set_target_v1_checkentry,
634		.destroy	= set_target_v1_destroy,
635		.me		= THIS_MODULE
636	},
637	{
638		.name		= "SET",
639		.revision	= 1,
640		.family		= NFPROTO_IPV6,
641		.target		= set_target_v1,
642		.targetsize	= sizeof(struct xt_set_info_target_v1),
643		.checkentry	= set_target_v1_checkentry,
644		.destroy	= set_target_v1_destroy,
645		.me		= THIS_MODULE
646	},
647	/* --timeout and --exist flags support */
648	{
649		.name		= "SET",
650		.revision	= 2,
651		.family		= NFPROTO_IPV4,
652		.target		= set_target_v2,
653		.targetsize	= sizeof(struct xt_set_info_target_v2),
654		.checkentry	= set_target_v2_checkentry,
655		.destroy	= set_target_v2_destroy,
656		.me		= THIS_MODULE
657	},
658	{
659		.name		= "SET",
660		.revision	= 2,
661		.family		= NFPROTO_IPV6,
662		.target		= set_target_v2,
663		.targetsize	= sizeof(struct xt_set_info_target_v2),
664		.checkentry	= set_target_v2_checkentry,
665		.destroy	= set_target_v2_destroy,
666		.me		= THIS_MODULE
667	},
668	/* --map-set support */
669	{
670		.name		= "SET",
671		.revision	= 3,
672		.family		= NFPROTO_IPV4,
673		.target		= set_target_v3,
674		.targetsize	= sizeof(struct xt_set_info_target_v3),
675		.checkentry	= set_target_v3_checkentry,
676		.destroy	= set_target_v3_destroy,
677		.me		= THIS_MODULE
678	},
679	{
680		.name		= "SET",
681		.revision	= 3,
682		.family		= NFPROTO_IPV6,
683		.target		= set_target_v3,
684		.targetsize	= sizeof(struct xt_set_info_target_v3),
685		.checkentry	= set_target_v3_checkentry,
686		.destroy	= set_target_v3_destroy,
687		.me		= THIS_MODULE
688	},
689};
690
691static int __init xt_set_init(void)
692{
693	int ret = xt_register_matches(set_matches, ARRAY_SIZE(set_matches));
694
695	if (!ret) {
696		ret = xt_register_targets(set_targets,
697					  ARRAY_SIZE(set_targets));
698		if (ret)
699			xt_unregister_matches(set_matches,
700					      ARRAY_SIZE(set_matches));
701	}
702	return ret;
703}
704
705static void __exit xt_set_fini(void)
706{
707	xt_unregister_matches(set_matches, ARRAY_SIZE(set_matches));
708	xt_unregister_targets(set_targets, ARRAY_SIZE(set_targets));
709}
710
711module_init(xt_set_init);
712module_exit(xt_set_fini);
v4.10.11
 
  1/* Copyright (C) 2000-2002 Joakim Axelsson <gozem@linux.nu>
  2 *                         Patrick Schaaf <bof@bof.de>
  3 *                         Martin Josefsson <gandalf@wlug.westbo.se>
  4 * Copyright (C) 2003-2013 Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>
  5 *
  6 * This program is free software; you can redistribute it and/or modify
  7 * it under the terms of the GNU General Public License version 2 as
  8 * published by the Free Software Foundation.
  9 */
 10
 11/* Kernel module which implements the set match and SET target
 12 * for netfilter/iptables.
 13 */
 14
 15#include <linux/module.h>
 16#include <linux/skbuff.h>
 17
 18#include <linux/netfilter/x_tables.h>
 19#include <linux/netfilter/ipset/ip_set.h>
 20#include <linux/netfilter/ipset/ip_set_timeout.h>
 21#include <uapi/linux/netfilter/xt_set.h>
 22
 23MODULE_LICENSE("GPL");
 24MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>");
 25MODULE_DESCRIPTION("Xtables: IP set match and target module");
 26MODULE_ALIAS("xt_SET");
 27MODULE_ALIAS("ipt_set");
 28MODULE_ALIAS("ip6t_set");
 29MODULE_ALIAS("ipt_SET");
 30MODULE_ALIAS("ip6t_SET");
 31
 32static inline int
 33match_set(ip_set_id_t index, const struct sk_buff *skb,
 34	  const struct xt_action_param *par,
 35	  struct ip_set_adt_opt *opt, int inv)
 36{
 37	if (ip_set_test(index, skb, par, opt))
 38		inv = !inv;
 39	return inv;
 40}
 41
 42#define ADT_OPT(n, f, d, fs, cfs, t)	\
 43struct ip_set_adt_opt n = {		\
 44	.family	= f,			\
 45	.dim = d,			\
 46	.flags = fs,			\
 47	.cmdflags = cfs,		\
 48	.ext.timeout = t,		\
 
 
 
 
 49}
 50
 51/* Revision 0 interface: backward compatible with netfilter/iptables */
 52
 53static bool
 54set_match_v0(const struct sk_buff *skb, struct xt_action_param *par)
 55{
 56	const struct xt_set_info_match_v0 *info = par->matchinfo;
 57
 58	ADT_OPT(opt, xt_family(par), info->match_set.u.compat.dim,
 59		info->match_set.u.compat.flags, 0, UINT_MAX);
 
 60
 61	return match_set(info->match_set.index, skb, par, &opt,
 62			 info->match_set.u.compat.flags & IPSET_INV_MATCH);
 63}
 64
 65static void
 66compat_flags(struct xt_set_info_v0 *info)
 67{
 68	u_int8_t i;
 69
 70	/* Fill out compatibility data according to enum ip_set_kopt */
 71	info->u.compat.dim = IPSET_DIM_ZERO;
 72	if (info->u.flags[0] & IPSET_MATCH_INV)
 73		info->u.compat.flags |= IPSET_INV_MATCH;
 74	for (i = 0; i < IPSET_DIM_MAX - 1 && info->u.flags[i]; i++) {
 75		info->u.compat.dim++;
 76		if (info->u.flags[i] & IPSET_SRC)
 77			info->u.compat.flags |= (1 << info->u.compat.dim);
 78	}
 79}
 80
 81static int
 82set_match_v0_checkentry(const struct xt_mtchk_param *par)
 83{
 84	struct xt_set_info_match_v0 *info = par->matchinfo;
 85	ip_set_id_t index;
 86
 87	index = ip_set_nfnl_get_byindex(par->net, info->match_set.index);
 88
 89	if (index == IPSET_INVALID_ID) {
 90		pr_warn("Cannot find set identified by id %u to match\n",
 91			info->match_set.index);
 92		return -ENOENT;
 93	}
 94	if (info->match_set.u.flags[IPSET_DIM_MAX - 1] != 0) {
 95		pr_warn("Protocol error: set match dimension is over the limit!\n");
 96		ip_set_nfnl_put(par->net, info->match_set.index);
 97		return -ERANGE;
 98	}
 99
100	/* Fill out compatibility data */
101	compat_flags(&info->match_set);
102
103	return 0;
104}
105
106static void
107set_match_v0_destroy(const struct xt_mtdtor_param *par)
108{
109	struct xt_set_info_match_v0 *info = par->matchinfo;
110
111	ip_set_nfnl_put(par->net, info->match_set.index);
112}
113
114/* Revision 1 match */
115
116static bool
117set_match_v1(const struct sk_buff *skb, struct xt_action_param *par)
118{
119	const struct xt_set_info_match_v1 *info = par->matchinfo;
120
121	ADT_OPT(opt, xt_family(par), info->match_set.dim,
122		info->match_set.flags, 0, UINT_MAX);
 
123
124	if (opt.flags & IPSET_RETURN_NOMATCH)
125		opt.cmdflags |= IPSET_FLAG_RETURN_NOMATCH;
126
127	return match_set(info->match_set.index, skb, par, &opt,
128			 info->match_set.flags & IPSET_INV_MATCH);
129}
130
131static int
132set_match_v1_checkentry(const struct xt_mtchk_param *par)
133{
134	struct xt_set_info_match_v1 *info = par->matchinfo;
135	ip_set_id_t index;
136
137	index = ip_set_nfnl_get_byindex(par->net, info->match_set.index);
138
139	if (index == IPSET_INVALID_ID) {
140		pr_warn("Cannot find set identified by id %u to match\n",
141			info->match_set.index);
142		return -ENOENT;
143	}
144	if (info->match_set.dim > IPSET_DIM_MAX) {
145		pr_warn("Protocol error: set match dimension is over the limit!\n");
146		ip_set_nfnl_put(par->net, info->match_set.index);
147		return -ERANGE;
148	}
149
150	return 0;
151}
152
153static void
154set_match_v1_destroy(const struct xt_mtdtor_param *par)
155{
156	struct xt_set_info_match_v1 *info = par->matchinfo;
157
158	ip_set_nfnl_put(par->net, info->match_set.index);
159}
160
161/* Revision 3 match */
162
163static bool
164match_counter0(u64 counter, const struct ip_set_counter_match0 *info)
165{
166	switch (info->op) {
167	case IPSET_COUNTER_NONE:
168		return true;
169	case IPSET_COUNTER_EQ:
170		return counter == info->value;
171	case IPSET_COUNTER_NE:
172		return counter != info->value;
173	case IPSET_COUNTER_LT:
174		return counter < info->value;
175	case IPSET_COUNTER_GT:
176		return counter > info->value;
177	}
178	return false;
179}
180
181static bool
182set_match_v3(const struct sk_buff *skb, struct xt_action_param *par)
183{
184	const struct xt_set_info_match_v3 *info = par->matchinfo;
185	int ret;
186
187	ADT_OPT(opt, xt_family(par), info->match_set.dim,
188		info->match_set.flags, info->flags, UINT_MAX);
 
 
189
190	if (info->packets.op != IPSET_COUNTER_NONE ||
191	    info->bytes.op != IPSET_COUNTER_NONE)
192		opt.cmdflags |= IPSET_FLAG_MATCH_COUNTERS;
193
194	ret = match_set(info->match_set.index, skb, par, &opt,
195			info->match_set.flags & IPSET_INV_MATCH);
196
197	if (!(ret && opt.cmdflags & IPSET_FLAG_MATCH_COUNTERS))
198		return ret;
199
200	if (!match_counter0(opt.ext.packets, &info->packets))
201		return false;
202	return match_counter0(opt.ext.bytes, &info->bytes);
203}
204
205#define set_match_v3_checkentry	set_match_v1_checkentry
206#define set_match_v3_destroy	set_match_v1_destroy
207
208/* Revision 4 match */
209
210static bool
211match_counter(u64 counter, const struct ip_set_counter_match *info)
212{
213	switch (info->op) {
214	case IPSET_COUNTER_NONE:
215		return true;
216	case IPSET_COUNTER_EQ:
217		return counter == info->value;
218	case IPSET_COUNTER_NE:
219		return counter != info->value;
220	case IPSET_COUNTER_LT:
221		return counter < info->value;
222	case IPSET_COUNTER_GT:
223		return counter > info->value;
224	}
225	return false;
226}
227
228static bool
229set_match_v4(const struct sk_buff *skb, struct xt_action_param *par)
230{
231	const struct xt_set_info_match_v4 *info = par->matchinfo;
232	int ret;
233
234	ADT_OPT(opt, xt_family(par), info->match_set.dim,
235		info->match_set.flags, info->flags, UINT_MAX);
 
 
236
237	if (info->packets.op != IPSET_COUNTER_NONE ||
238	    info->bytes.op != IPSET_COUNTER_NONE)
239		opt.cmdflags |= IPSET_FLAG_MATCH_COUNTERS;
240
241	ret = match_set(info->match_set.index, skb, par, &opt,
242			info->match_set.flags & IPSET_INV_MATCH);
243
244	if (!(ret && opt.cmdflags & IPSET_FLAG_MATCH_COUNTERS))
245		return ret;
246
247	if (!match_counter(opt.ext.packets, &info->packets))
248		return false;
249	return match_counter(opt.ext.bytes, &info->bytes);
250}
251
252#define set_match_v4_checkentry	set_match_v1_checkentry
253#define set_match_v4_destroy	set_match_v1_destroy
254
255/* Revision 0 interface: backward compatible with netfilter/iptables */
256
257static unsigned int
258set_target_v0(struct sk_buff *skb, const struct xt_action_param *par)
259{
260	const struct xt_set_info_target_v0 *info = par->targinfo;
261
262	ADT_OPT(add_opt, xt_family(par), info->add_set.u.compat.dim,
263		info->add_set.u.compat.flags, 0, UINT_MAX);
 
264	ADT_OPT(del_opt, xt_family(par), info->del_set.u.compat.dim,
265		info->del_set.u.compat.flags, 0, UINT_MAX);
 
266
267	if (info->add_set.index != IPSET_INVALID_ID)
268		ip_set_add(info->add_set.index, skb, par, &add_opt);
269	if (info->del_set.index != IPSET_INVALID_ID)
270		ip_set_del(info->del_set.index, skb, par, &del_opt);
271
272	return XT_CONTINUE;
273}
274
275static int
276set_target_v0_checkentry(const struct xt_tgchk_param *par)
277{
278	struct xt_set_info_target_v0 *info = par->targinfo;
279	ip_set_id_t index;
280
281	if (info->add_set.index != IPSET_INVALID_ID) {
282		index = ip_set_nfnl_get_byindex(par->net, info->add_set.index);
283		if (index == IPSET_INVALID_ID) {
284			pr_warn("Cannot find add_set index %u as target\n",
285				info->add_set.index);
286			return -ENOENT;
287		}
288	}
289
290	if (info->del_set.index != IPSET_INVALID_ID) {
291		index = ip_set_nfnl_get_byindex(par->net, info->del_set.index);
292		if (index == IPSET_INVALID_ID) {
293			pr_warn("Cannot find del_set index %u as target\n",
294				info->del_set.index);
295			if (info->add_set.index != IPSET_INVALID_ID)
296				ip_set_nfnl_put(par->net, info->add_set.index);
297			return -ENOENT;
298		}
299	}
300	if (info->add_set.u.flags[IPSET_DIM_MAX - 1] != 0 ||
301	    info->del_set.u.flags[IPSET_DIM_MAX - 1] != 0) {
302		pr_warn("Protocol error: SET target dimension is over the limit!\n");
303		if (info->add_set.index != IPSET_INVALID_ID)
304			ip_set_nfnl_put(par->net, info->add_set.index);
305		if (info->del_set.index != IPSET_INVALID_ID)
306			ip_set_nfnl_put(par->net, info->del_set.index);
307		return -ERANGE;
308	}
309
310	/* Fill out compatibility data */
311	compat_flags(&info->add_set);
312	compat_flags(&info->del_set);
313
314	return 0;
315}
316
317static void
318set_target_v0_destroy(const struct xt_tgdtor_param *par)
319{
320	const struct xt_set_info_target_v0 *info = par->targinfo;
321
322	if (info->add_set.index != IPSET_INVALID_ID)
323		ip_set_nfnl_put(par->net, info->add_set.index);
324	if (info->del_set.index != IPSET_INVALID_ID)
325		ip_set_nfnl_put(par->net, info->del_set.index);
326}
327
328/* Revision 1 target */
329
330static unsigned int
331set_target_v1(struct sk_buff *skb, const struct xt_action_param *par)
332{
333	const struct xt_set_info_target_v1 *info = par->targinfo;
334
335	ADT_OPT(add_opt, xt_family(par), info->add_set.dim,
336		info->add_set.flags, 0, UINT_MAX);
 
337	ADT_OPT(del_opt, xt_family(par), info->del_set.dim,
338		info->del_set.flags, 0, UINT_MAX);
 
339
340	if (info->add_set.index != IPSET_INVALID_ID)
341		ip_set_add(info->add_set.index, skb, par, &add_opt);
342	if (info->del_set.index != IPSET_INVALID_ID)
343		ip_set_del(info->del_set.index, skb, par, &del_opt);
344
345	return XT_CONTINUE;
346}
347
348static int
349set_target_v1_checkentry(const struct xt_tgchk_param *par)
350{
351	const struct xt_set_info_target_v1 *info = par->targinfo;
352	ip_set_id_t index;
353
354	if (info->add_set.index != IPSET_INVALID_ID) {
355		index = ip_set_nfnl_get_byindex(par->net, info->add_set.index);
356		if (index == IPSET_INVALID_ID) {
357			pr_warn("Cannot find add_set index %u as target\n",
358				info->add_set.index);
359			return -ENOENT;
360		}
361	}
362
363	if (info->del_set.index != IPSET_INVALID_ID) {
364		index = ip_set_nfnl_get_byindex(par->net, info->del_set.index);
365		if (index == IPSET_INVALID_ID) {
366			pr_warn("Cannot find del_set index %u as target\n",
367				info->del_set.index);
368			if (info->add_set.index != IPSET_INVALID_ID)
369				ip_set_nfnl_put(par->net, info->add_set.index);
370			return -ENOENT;
371		}
372	}
373	if (info->add_set.dim > IPSET_DIM_MAX ||
374	    info->del_set.dim > IPSET_DIM_MAX) {
375		pr_warn("Protocol error: SET target dimension is over the limit!\n");
376		if (info->add_set.index != IPSET_INVALID_ID)
377			ip_set_nfnl_put(par->net, info->add_set.index);
378		if (info->del_set.index != IPSET_INVALID_ID)
379			ip_set_nfnl_put(par->net, info->del_set.index);
380		return -ERANGE;
381	}
382
383	return 0;
384}
385
386static void
387set_target_v1_destroy(const struct xt_tgdtor_param *par)
388{
389	const struct xt_set_info_target_v1 *info = par->targinfo;
390
391	if (info->add_set.index != IPSET_INVALID_ID)
392		ip_set_nfnl_put(par->net, info->add_set.index);
393	if (info->del_set.index != IPSET_INVALID_ID)
394		ip_set_nfnl_put(par->net, info->del_set.index);
395}
396
397/* Revision 2 target */
398
399static unsigned int
400set_target_v2(struct sk_buff *skb, const struct xt_action_param *par)
401{
402	const struct xt_set_info_target_v2 *info = par->targinfo;
403
404	ADT_OPT(add_opt, xt_family(par), info->add_set.dim,
405		info->add_set.flags, info->flags, info->timeout);
 
406	ADT_OPT(del_opt, xt_family(par), info->del_set.dim,
407		info->del_set.flags, 0, UINT_MAX);
 
408
409	/* Normalize to fit into jiffies */
410	if (add_opt.ext.timeout != IPSET_NO_TIMEOUT &&
411	    add_opt.ext.timeout > UINT_MAX / MSEC_PER_SEC)
412		add_opt.ext.timeout = UINT_MAX / MSEC_PER_SEC;
413	if (info->add_set.index != IPSET_INVALID_ID)
414		ip_set_add(info->add_set.index, skb, par, &add_opt);
415	if (info->del_set.index != IPSET_INVALID_ID)
416		ip_set_del(info->del_set.index, skb, par, &del_opt);
417
418	return XT_CONTINUE;
419}
420
421#define set_target_v2_checkentry	set_target_v1_checkentry
422#define set_target_v2_destroy		set_target_v1_destroy
423
424/* Revision 3 target */
425
426#define MOPT(opt, member)	((opt).ext.skbinfo.member)
427
428static unsigned int
429set_target_v3(struct sk_buff *skb, const struct xt_action_param *par)
430{
431	const struct xt_set_info_target_v3 *info = par->targinfo;
432	int ret;
433
434	ADT_OPT(add_opt, xt_family(par), info->add_set.dim,
435		info->add_set.flags, info->flags, info->timeout);
 
436	ADT_OPT(del_opt, xt_family(par), info->del_set.dim,
437		info->del_set.flags, 0, UINT_MAX);
 
438	ADT_OPT(map_opt, xt_family(par), info->map_set.dim,
439		info->map_set.flags, 0, UINT_MAX);
 
440
441	/* Normalize to fit into jiffies */
442	if (add_opt.ext.timeout != IPSET_NO_TIMEOUT &&
443	    add_opt.ext.timeout > UINT_MAX / MSEC_PER_SEC)
444		add_opt.ext.timeout = UINT_MAX / MSEC_PER_SEC;
445	if (info->add_set.index != IPSET_INVALID_ID)
446		ip_set_add(info->add_set.index, skb, par, &add_opt);
447	if (info->del_set.index != IPSET_INVALID_ID)
448		ip_set_del(info->del_set.index, skb, par, &del_opt);
449	if (info->map_set.index != IPSET_INVALID_ID) {
450		map_opt.cmdflags |= info->flags & (IPSET_FLAG_MAP_SKBMARK |
451						   IPSET_FLAG_MAP_SKBPRIO |
452						   IPSET_FLAG_MAP_SKBQUEUE);
453		ret = match_set(info->map_set.index, skb, par, &map_opt,
454				info->map_set.flags & IPSET_INV_MATCH);
455		if (!ret)
456			return XT_CONTINUE;
457		if (map_opt.cmdflags & IPSET_FLAG_MAP_SKBMARK)
458			skb->mark = (skb->mark & ~MOPT(map_opt,skbmarkmask))
459				    ^ MOPT(map_opt, skbmark);
460		if (map_opt.cmdflags & IPSET_FLAG_MAP_SKBPRIO)
461			skb->priority = MOPT(map_opt, skbprio);
462		if ((map_opt.cmdflags & IPSET_FLAG_MAP_SKBQUEUE) &&
463		    skb->dev &&
464		    skb->dev->real_num_tx_queues > MOPT(map_opt, skbqueue))
465			skb_set_queue_mapping(skb, MOPT(map_opt, skbqueue));
466	}
467	return XT_CONTINUE;
468}
469
470static int
471set_target_v3_checkentry(const struct xt_tgchk_param *par)
472{
473	const struct xt_set_info_target_v3 *info = par->targinfo;
474	ip_set_id_t index;
 
475
476	if (info->add_set.index != IPSET_INVALID_ID) {
477		index = ip_set_nfnl_get_byindex(par->net,
478						info->add_set.index);
479		if (index == IPSET_INVALID_ID) {
480			pr_warn("Cannot find add_set index %u as target\n",
481				info->add_set.index);
482			return -ENOENT;
483		}
484	}
485
486	if (info->del_set.index != IPSET_INVALID_ID) {
487		index = ip_set_nfnl_get_byindex(par->net,
488						info->del_set.index);
489		if (index == IPSET_INVALID_ID) {
490			pr_warn("Cannot find del_set index %u as target\n",
491				info->del_set.index);
492			if (info->add_set.index != IPSET_INVALID_ID)
493				ip_set_nfnl_put(par->net,
494						info->add_set.index);
495			return -ENOENT;
496		}
497	}
498
499	if (info->map_set.index != IPSET_INVALID_ID) {
500		if (strncmp(par->table, "mangle", 7)) {
501			pr_warn("--map-set only usable from mangle table\n");
502			return -EINVAL;
 
503		}
504		if (((info->flags & IPSET_FLAG_MAP_SKBPRIO) |
505		     (info->flags & IPSET_FLAG_MAP_SKBQUEUE)) &&
506		     !(par->hook_mask & (1 << NF_INET_FORWARD |
507					 1 << NF_INET_LOCAL_OUT |
508					 1 << NF_INET_POST_ROUTING))) {
509			pr_warn("mapping of prio or/and queue is allowed only from OUTPUT/FORWARD/POSTROUTING chains\n");
510			return -EINVAL;
 
511		}
512		index = ip_set_nfnl_get_byindex(par->net,
513						info->map_set.index);
514		if (index == IPSET_INVALID_ID) {
515			pr_warn("Cannot find map_set index %u as target\n",
516				info->map_set.index);
517			if (info->add_set.index != IPSET_INVALID_ID)
518				ip_set_nfnl_put(par->net,
519						info->add_set.index);
520			if (info->del_set.index != IPSET_INVALID_ID)
521				ip_set_nfnl_put(par->net,
522						info->del_set.index);
523			return -ENOENT;
524		}
525	}
526
527	if (info->add_set.dim > IPSET_DIM_MAX ||
528	    info->del_set.dim > IPSET_DIM_MAX ||
529	    info->map_set.dim > IPSET_DIM_MAX) {
530		pr_warn("Protocol error: SET target dimension is over the limit!\n");
531		if (info->add_set.index != IPSET_INVALID_ID)
532			ip_set_nfnl_put(par->net, info->add_set.index);
533		if (info->del_set.index != IPSET_INVALID_ID)
534			ip_set_nfnl_put(par->net, info->del_set.index);
535		if (info->map_set.index != IPSET_INVALID_ID)
536			ip_set_nfnl_put(par->net, info->map_set.index);
537		return -ERANGE;
538	}
539
540	return 0;
 
 
 
 
 
 
 
 
 
 
541}
542
543static void
544set_target_v3_destroy(const struct xt_tgdtor_param *par)
545{
546	const struct xt_set_info_target_v3 *info = par->targinfo;
547
548	if (info->add_set.index != IPSET_INVALID_ID)
549		ip_set_nfnl_put(par->net, info->add_set.index);
550	if (info->del_set.index != IPSET_INVALID_ID)
551		ip_set_nfnl_put(par->net, info->del_set.index);
552	if (info->map_set.index != IPSET_INVALID_ID)
553		ip_set_nfnl_put(par->net, info->map_set.index);
554}
555
556static struct xt_match set_matches[] __read_mostly = {
557	{
558		.name		= "set",
559		.family		= NFPROTO_IPV4,
560		.revision	= 0,
561		.match		= set_match_v0,
562		.matchsize	= sizeof(struct xt_set_info_match_v0),
563		.checkentry	= set_match_v0_checkentry,
564		.destroy	= set_match_v0_destroy,
565		.me		= THIS_MODULE
566	},
567	{
568		.name		= "set",
569		.family		= NFPROTO_IPV4,
570		.revision	= 1,
571		.match		= set_match_v1,
572		.matchsize	= sizeof(struct xt_set_info_match_v1),
573		.checkentry	= set_match_v1_checkentry,
574		.destroy	= set_match_v1_destroy,
575		.me		= THIS_MODULE
576	},
577	{
578		.name		= "set",
579		.family		= NFPROTO_IPV6,
580		.revision	= 1,
581		.match		= set_match_v1,
582		.matchsize	= sizeof(struct xt_set_info_match_v1),
583		.checkentry	= set_match_v1_checkentry,
584		.destroy	= set_match_v1_destroy,
585		.me		= THIS_MODULE
586	},
587	/* --return-nomatch flag support */
588	{
589		.name		= "set",
590		.family		= NFPROTO_IPV4,
591		.revision	= 2,
592		.match		= set_match_v1,
593		.matchsize	= sizeof(struct xt_set_info_match_v1),
594		.checkentry	= set_match_v1_checkentry,
595		.destroy	= set_match_v1_destroy,
596		.me		= THIS_MODULE
597	},
598	{
599		.name		= "set",
600		.family		= NFPROTO_IPV6,
601		.revision	= 2,
602		.match		= set_match_v1,
603		.matchsize	= sizeof(struct xt_set_info_match_v1),
604		.checkentry	= set_match_v1_checkentry,
605		.destroy	= set_match_v1_destroy,
606		.me		= THIS_MODULE
607	},
608	/* counters support: update, match */
609	{
610		.name		= "set",
611		.family		= NFPROTO_IPV4,
612		.revision	= 3,
613		.match		= set_match_v3,
614		.matchsize	= sizeof(struct xt_set_info_match_v3),
615		.checkentry	= set_match_v3_checkentry,
616		.destroy	= set_match_v3_destroy,
617		.me		= THIS_MODULE
618	},
619	{
620		.name		= "set",
621		.family		= NFPROTO_IPV6,
622		.revision	= 3,
623		.match		= set_match_v3,
624		.matchsize	= sizeof(struct xt_set_info_match_v3),
625		.checkentry	= set_match_v3_checkentry,
626		.destroy	= set_match_v3_destroy,
627		.me		= THIS_MODULE
628	},
629	/* new revision for counters support: update, match */
630	{
631		.name		= "set",
632		.family		= NFPROTO_IPV4,
633		.revision	= 4,
634		.match		= set_match_v4,
635		.matchsize	= sizeof(struct xt_set_info_match_v4),
636		.checkentry	= set_match_v4_checkentry,
637		.destroy	= set_match_v4_destroy,
638		.me		= THIS_MODULE
639	},
640	{
641		.name		= "set",
642		.family		= NFPROTO_IPV6,
643		.revision	= 4,
644		.match		= set_match_v4,
645		.matchsize	= sizeof(struct xt_set_info_match_v4),
646		.checkentry	= set_match_v4_checkentry,
647		.destroy	= set_match_v4_destroy,
648		.me		= THIS_MODULE
649	},
650};
651
652static struct xt_target set_targets[] __read_mostly = {
653	{
654		.name		= "SET",
655		.revision	= 0,
656		.family		= NFPROTO_IPV4,
657		.target		= set_target_v0,
658		.targetsize	= sizeof(struct xt_set_info_target_v0),
659		.checkentry	= set_target_v0_checkentry,
660		.destroy	= set_target_v0_destroy,
661		.me		= THIS_MODULE
662	},
663	{
664		.name		= "SET",
665		.revision	= 1,
666		.family		= NFPROTO_IPV4,
667		.target		= set_target_v1,
668		.targetsize	= sizeof(struct xt_set_info_target_v1),
669		.checkentry	= set_target_v1_checkentry,
670		.destroy	= set_target_v1_destroy,
671		.me		= THIS_MODULE
672	},
673	{
674		.name		= "SET",
675		.revision	= 1,
676		.family		= NFPROTO_IPV6,
677		.target		= set_target_v1,
678		.targetsize	= sizeof(struct xt_set_info_target_v1),
679		.checkentry	= set_target_v1_checkentry,
680		.destroy	= set_target_v1_destroy,
681		.me		= THIS_MODULE
682	},
683	/* --timeout and --exist flags support */
684	{
685		.name		= "SET",
686		.revision	= 2,
687		.family		= NFPROTO_IPV4,
688		.target		= set_target_v2,
689		.targetsize	= sizeof(struct xt_set_info_target_v2),
690		.checkentry	= set_target_v2_checkentry,
691		.destroy	= set_target_v2_destroy,
692		.me		= THIS_MODULE
693	},
694	{
695		.name		= "SET",
696		.revision	= 2,
697		.family		= NFPROTO_IPV6,
698		.target		= set_target_v2,
699		.targetsize	= sizeof(struct xt_set_info_target_v2),
700		.checkentry	= set_target_v2_checkentry,
701		.destroy	= set_target_v2_destroy,
702		.me		= THIS_MODULE
703	},
704	/* --map-set support */
705	{
706		.name		= "SET",
707		.revision	= 3,
708		.family		= NFPROTO_IPV4,
709		.target		= set_target_v3,
710		.targetsize	= sizeof(struct xt_set_info_target_v3),
711		.checkentry	= set_target_v3_checkentry,
712		.destroy	= set_target_v3_destroy,
713		.me		= THIS_MODULE
714	},
715	{
716		.name		= "SET",
717		.revision	= 3,
718		.family		= NFPROTO_IPV6,
719		.target		= set_target_v3,
720		.targetsize	= sizeof(struct xt_set_info_target_v3),
721		.checkentry	= set_target_v3_checkentry,
722		.destroy	= set_target_v3_destroy,
723		.me		= THIS_MODULE
724	},
725};
726
727static int __init xt_set_init(void)
728{
729	int ret = xt_register_matches(set_matches, ARRAY_SIZE(set_matches));
730
731	if (!ret) {
732		ret = xt_register_targets(set_targets,
733					  ARRAY_SIZE(set_targets));
734		if (ret)
735			xt_unregister_matches(set_matches,
736					      ARRAY_SIZE(set_matches));
737	}
738	return ret;
739}
740
741static void __exit xt_set_fini(void)
742{
743	xt_unregister_matches(set_matches, ARRAY_SIZE(set_matches));
744	xt_unregister_targets(set_targets, ARRAY_SIZE(set_targets));
745}
746
747module_init(xt_set_init);
748module_exit(xt_set_fini);