ip6_tables.c 39.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4
/*
 * Packet matching code.
 *
 * Copyright (C) 1999 Paul `Rusty' Russell & Michael J. Neuling
5
 * Copyright (C) 2000-2005 Netfilter Core Team <coreteam@netfilter.org>
L
Linus Torvalds 已提交
6 7 8 9 10
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */
11 12

#include <linux/capability.h>
13
#include <linux/in.h>
L
Linus Torvalds 已提交
14 15 16 17 18
#include <linux/skbuff.h>
#include <linux/kmod.h>
#include <linux/vmalloc.h>
#include <linux/netdevice.h>
#include <linux/module.h>
R
Randy Dunlap 已提交
19
#include <linux/poison.h>
L
Linus Torvalds 已提交
20 21 22
#include <linux/icmpv6.h>
#include <net/ipv6.h>
#include <asm/uaccess.h>
I
Ingo Molnar 已提交
23
#include <linux/mutex.h>
L
Linus Torvalds 已提交
24
#include <linux/proc_fs.h>
25
#include <linux/cpumask.h>
L
Linus Torvalds 已提交
26 27

#include <linux/netfilter_ipv6/ip6_tables.h>
28
#include <linux/netfilter/x_tables.h>
L
Linus Torvalds 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Netfilter Core Team <coreteam@netfilter.org>");
MODULE_DESCRIPTION("IPv6 packet filter");

/*#define DEBUG_IP_FIREWALL*/
/*#define DEBUG_ALLOW_ALL*/ /* Useful for remote debugging */
/*#define DEBUG_IP_FIREWALL_USER*/

#ifdef DEBUG_IP_FIREWALL
#define dprintf(format, args...)  printk(format , ## args)
#else
#define dprintf(format, args...)
#endif

#ifdef DEBUG_IP_FIREWALL_USER
#define duprintf(format, args...) printk(format , ## args)
#else
#define duprintf(format, args...)
#endif

#ifdef CONFIG_NETFILTER_DEBUG
#define IP_NF_ASSERT(x)						\
do {								\
	if (!(x))						\
		printk("IP_NF_ASSERT: %s:%s:%u\n",		\
		       __FUNCTION__, __FILE__, __LINE__);	\
} while(0)
#else
#define IP_NF_ASSERT(x)
#endif

#if 0
/* All the better to debug you with... */
#define static
#define inline
#endif

67
/*
L
Linus Torvalds 已提交
68
   We keep a set of rules for each CPU, so we can avoid write-locking
69 70 71 72
   them in the softirq when updating the counters and therefore
   only need to read-lock in the softirq; doing a write_lock_bh() in user
   context stops packets coming through and allows user context to read
   the counters or update the rules.
L
Linus Torvalds 已提交
73 74 75 76

   Hence the start of any table is given by get_table() below.  */

/* Check for an extension */
77
int
L
Linus Torvalds 已提交
78 79
ip6t_ext_hdr(u8 nexthdr)
{
80 81 82 83 84 85 86
	return ( (nexthdr == IPPROTO_HOPOPTS)   ||
		 (nexthdr == IPPROTO_ROUTING)   ||
		 (nexthdr == IPPROTO_FRAGMENT)  ||
		 (nexthdr == IPPROTO_ESP)       ||
		 (nexthdr == IPPROTO_AH)        ||
		 (nexthdr == IPPROTO_NONE)      ||
		 (nexthdr == IPPROTO_DSTOPTS) );
L
Linus Torvalds 已提交
87 88 89
}

/* Returns whether matches rule or not. */
90
static inline bool
L
Linus Torvalds 已提交
91 92 93 94 95
ip6_packet_match(const struct sk_buff *skb,
		 const char *indev,
		 const char *outdev,
		 const struct ip6t_ip6 *ip6info,
		 unsigned int *protoff,
96
		 int *fragoff, bool *hotdrop)
L
Linus Torvalds 已提交
97 98 99
{
	size_t i;
	unsigned long ret;
100
	const struct ipv6hdr *ipv6 = ipv6_hdr(skb);
L
Linus Torvalds 已提交
101 102 103

#define FWINV(bool,invflg) ((bool) ^ !!(ip6info->invflags & invflg))

104
	if (FWINV(ipv6_masked_addr_cmp(&ipv6->saddr, &ip6info->smsk,
105
				       &ip6info->src), IP6T_INV_SRCIP)
106
	    || FWINV(ipv6_masked_addr_cmp(&ipv6->daddr, &ip6info->dmsk,
107
					  &ip6info->dst), IP6T_INV_DSTIP)) {
L
Linus Torvalds 已提交
108 109 110 111 112 113 114 115
		dprintf("Source or dest mismatch.\n");
/*
		dprintf("SRC: %u. Mask: %u. Target: %u.%s\n", ip->saddr,
			ipinfo->smsk.s_addr, ipinfo->src.s_addr,
			ipinfo->invflags & IP6T_INV_SRCIP ? " (INV)" : "");
		dprintf("DST: %u. Mask: %u. Target: %u.%s\n", ip->daddr,
			ipinfo->dmsk.s_addr, ipinfo->dst.s_addr,
			ipinfo->invflags & IP6T_INV_DSTIP ? " (INV)" : "");*/
116
		return false;
L
Linus Torvalds 已提交
117 118 119 120 121 122 123 124 125 126 127 128 129
	}

	/* Look for ifname matches; this should unroll nicely. */
	for (i = 0, ret = 0; i < IFNAMSIZ/sizeof(unsigned long); i++) {
		ret |= (((const unsigned long *)indev)[i]
			^ ((const unsigned long *)ip6info->iniface)[i])
			& ((const unsigned long *)ip6info->iniface_mask)[i];
	}

	if (FWINV(ret != 0, IP6T_INV_VIA_IN)) {
		dprintf("VIA in mismatch (%s vs %s).%s\n",
			indev, ip6info->iniface,
			ip6info->invflags&IP6T_INV_VIA_IN ?" (INV)":"");
130
		return false;
L
Linus Torvalds 已提交
131 132 133 134 135 136 137 138 139 140 141 142
	}

	for (i = 0, ret = 0; i < IFNAMSIZ/sizeof(unsigned long); i++) {
		ret |= (((const unsigned long *)outdev)[i]
			^ ((const unsigned long *)ip6info->outiface)[i])
			& ((const unsigned long *)ip6info->outiface_mask)[i];
	}

	if (FWINV(ret != 0, IP6T_INV_VIA_OUT)) {
		dprintf("VIA out mismatch (%s vs %s).%s\n",
			outdev, ip6info->outiface,
			ip6info->invflags&IP6T_INV_VIA_OUT ?" (INV)":"");
143
		return false;
L
Linus Torvalds 已提交
144 145 146 147 148 149
	}

/* ... might want to do something with class and flowlabel here ... */

	/* look for the desired protocol header */
	if((ip6info->flags & IP6T_F_PROTO)) {
150 151
		int protohdr;
		unsigned short _frag_off;
L
Linus Torvalds 已提交
152

153
		protohdr = ipv6_find_hdr(skb, protoff, -1, &_frag_off);
154 155
		if (protohdr < 0) {
			if (_frag_off == 0)
156
				*hotdrop = true;
157
			return false;
158
		}
159
		*fragoff = _frag_off;
L
Linus Torvalds 已提交
160 161

		dprintf("Packet protocol %hi ?= %s%hi.\n",
162
				protohdr,
L
Linus Torvalds 已提交
163 164 165
				ip6info->invflags & IP6T_INV_PROTO ? "!":"",
				ip6info->proto);

166
		if (ip6info->proto == protohdr) {
L
Linus Torvalds 已提交
167
			if(ip6info->invflags & IP6T_INV_PROTO) {
168
				return false;
L
Linus Torvalds 已提交
169
			}
170
			return true;
L
Linus Torvalds 已提交
171 172 173 174 175
		}

		/* We need match for the '-p all', too! */
		if ((ip6info->proto != 0) &&
			!(ip6info->invflags & IP6T_INV_PROTO))
176
			return false;
L
Linus Torvalds 已提交
177
	}
178
	return true;
L
Linus Torvalds 已提交
179 180 181
}

/* should be ip6 safe */
182
static inline bool
L
Linus Torvalds 已提交
183 184 185 186 187
ip6_checkentry(const struct ip6t_ip6 *ipv6)
{
	if (ipv6->flags & ~IP6T_F_MASK) {
		duprintf("Unknown flag bits set: %08X\n",
			 ipv6->flags & ~IP6T_F_MASK);
188
		return false;
L
Linus Torvalds 已提交
189 190 191 192
	}
	if (ipv6->invflags & ~IP6T_INV_MASK) {
		duprintf("Unknown invflag bits set: %08X\n",
			 ipv6->invflags & ~IP6T_INV_MASK);
193
		return false;
L
Linus Torvalds 已提交
194
	}
195
	return true;
L
Linus Torvalds 已提交
196 197 198
}

static unsigned int
199
ip6t_error(struct sk_buff *skb,
L
Linus Torvalds 已提交
200 201 202
	  const struct net_device *in,
	  const struct net_device *out,
	  unsigned int hooknum,
203
	  const struct xt_target *target,
204
	  const void *targinfo)
L
Linus Torvalds 已提交
205 206 207 208 209 210 211 212
{
	if (net_ratelimit())
		printk("ip6_tables: error: `%s'\n", (char *)targinfo);

	return NF_DROP;
}

static inline
213 214 215 216 217 218 219
bool do_match(struct ip6t_entry_match *m,
	      const struct sk_buff *skb,
	      const struct net_device *in,
	      const struct net_device *out,
	      int offset,
	      unsigned int protoff,
	      bool *hotdrop)
L
Linus Torvalds 已提交
220 221
{
	/* Stop iteration if it doesn't match */
222
	if (!m->u.kernel.match->match(skb, in, out, m->u.kernel.match, m->data,
L
Linus Torvalds 已提交
223
				      offset, protoff, hotdrop))
224
		return true;
L
Linus Torvalds 已提交
225
	else
226
		return false;
L
Linus Torvalds 已提交
227 228 229 230 231 232 233 234
}

static inline struct ip6t_entry *
get_entry(void *base, unsigned int offset)
{
	return (struct ip6t_entry *)(base + offset);
}

235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
/* All zeroes == unconditional rule. */
static inline int
unconditional(const struct ip6t_ip6 *ipv6)
{
	unsigned int i;

	for (i = 0; i < sizeof(*ipv6); i++)
		if (((char *)ipv6)[i])
			break;

	return (i == sizeof(*ipv6));
}

#if defined(CONFIG_NETFILTER_XT_TARGET_TRACE) || \
    defined(CONFIG_NETFILTER_XT_TARGET_TRACE_MODULE)
/* This cries for unification! */
static const char *hooknames[] = {
252 253 254 255 256
	[NF_INET_PRE_ROUTING]		= "PREROUTING",
	[NF_INET_LOCAL_IN]		= "INPUT",
	[NF_INET_FORWARD]		= "FORWARD",
	[NF_INET_LOCAL_OUT]		= "OUTPUT",
	[NF_INET_POST_ROUTING]		= "POSTROUTING",
257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341
};

enum nf_ip_trace_comments {
	NF_IP6_TRACE_COMMENT_RULE,
	NF_IP6_TRACE_COMMENT_RETURN,
	NF_IP6_TRACE_COMMENT_POLICY,
};

static const char *comments[] = {
	[NF_IP6_TRACE_COMMENT_RULE]	= "rule",
	[NF_IP6_TRACE_COMMENT_RETURN]	= "return",
	[NF_IP6_TRACE_COMMENT_POLICY]	= "policy",
};

static struct nf_loginfo trace_loginfo = {
	.type = NF_LOG_TYPE_LOG,
	.u = {
		.log = {
			.level = 4,
			.logflags = NF_LOG_MASK,
		},
	},
};

static inline int
get_chainname_rulenum(struct ip6t_entry *s, struct ip6t_entry *e,
		      char *hookname, char **chainname,
		      char **comment, unsigned int *rulenum)
{
	struct ip6t_standard_target *t = (void *)ip6t_get_target(s);

	if (strcmp(t->target.u.kernel.target->name, IP6T_ERROR_TARGET) == 0) {
		/* Head of user chain: ERROR target with chainname */
		*chainname = t->target.data;
		(*rulenum) = 0;
	} else if (s == e) {
		(*rulenum)++;

		if (s->target_offset == sizeof(struct ip6t_entry)
		   && strcmp(t->target.u.kernel.target->name,
			     IP6T_STANDARD_TARGET) == 0
		   && t->verdict < 0
		   && unconditional(&s->ipv6)) {
			/* Tail of chains: STANDARD target (return/policy) */
			*comment = *chainname == hookname
				? (char *)comments[NF_IP6_TRACE_COMMENT_POLICY]
				: (char *)comments[NF_IP6_TRACE_COMMENT_RETURN];
		}
		return 1;
	} else
		(*rulenum)++;

	return 0;
}

static void trace_packet(struct sk_buff *skb,
			 unsigned int hook,
			 const struct net_device *in,
			 const struct net_device *out,
			 char *tablename,
			 struct xt_table_info *private,
			 struct ip6t_entry *e)
{
	void *table_base;
	struct ip6t_entry *root;
	char *hookname, *chainname, *comment;
	unsigned int rulenum = 0;

	table_base = (void *)private->entries[smp_processor_id()];
	root = get_entry(table_base, private->hook_entry[hook]);

	hookname = chainname = (char *)hooknames[hook];
	comment = (char *)comments[NF_IP6_TRACE_COMMENT_RULE];

	IP6T_ENTRY_ITERATE(root,
			   private->size - private->hook_entry[hook],
			   get_chainname_rulenum,
			   e, hookname, &chainname, &comment, &rulenum);

	nf_log_packet(AF_INET6, hook, skb, in, out, &trace_loginfo,
		      "TRACE: %s:%s:%s:%u ",
		      tablename, chainname, comment, rulenum);
}
#endif

L
Linus Torvalds 已提交
342 343
/* Returns one of the generic firewall policies, like NF_ACCEPT. */
unsigned int
344
ip6t_do_table(struct sk_buff *skb,
L
Linus Torvalds 已提交
345 346 347
	      unsigned int hook,
	      const struct net_device *in,
	      const struct net_device *out,
348
	      struct xt_table *table)
L
Linus Torvalds 已提交
349
{
350
	static const char nulldevname[IFNAMSIZ] __attribute__((aligned(sizeof(long))));
L
Linus Torvalds 已提交
351 352
	int offset = 0;
	unsigned int protoff = 0;
353
	bool hotdrop = false;
L
Linus Torvalds 已提交
354 355 356 357 358
	/* Initializing verdict to NF_DROP keeps gcc happy. */
	unsigned int verdict = NF_DROP;
	const char *indev, *outdev;
	void *table_base;
	struct ip6t_entry *e, *back;
359
	struct xt_table_info *private;
L
Linus Torvalds 已提交
360 361 362 363 364 365 366 367 368 369 370 371

	/* Initialization */
	indev = in ? in->name : nulldevname;
	outdev = out ? out->name : nulldevname;
	/* We handle fragments by dealing with the first fragment as
	 * if it was a normal packet.  All other fragments are treated
	 * normally, except that they will NEVER match rules that ask
	 * things we don't know, ie. tcp syn flag or ports).  If the
	 * rule is also a fragment-specific rule, non-fragments won't
	 * match it. */

	read_lock_bh(&table->lock);
372
	private = table->private;
L
Linus Torvalds 已提交
373
	IP_NF_ASSERT(table->valid_hooks & (1 << hook));
374 375
	table_base = (void *)private->entries[smp_processor_id()];
	e = get_entry(table_base, private->hook_entry[hook]);
L
Linus Torvalds 已提交
376 377

	/* For return from builtin chain */
378
	back = get_entry(table_base, private->underflow[hook]);
L
Linus Torvalds 已提交
379 380 381 382

	do {
		IP_NF_ASSERT(e);
		IP_NF_ASSERT(back);
383
		if (ip6_packet_match(skb, indev, outdev, &e->ipv6,
384
			&protoff, &offset, &hotdrop)) {
L
Linus Torvalds 已提交
385 386 387
			struct ip6t_entry_target *t;

			if (IP6T_MATCH_ITERATE(e, do_match,
388
					       skb, in, out,
L
Linus Torvalds 已提交
389 390 391 392
					       offset, protoff, &hotdrop) != 0)
				goto no_match;

			ADD_COUNTER(e->counters,
393 394
				    ntohs(ipv6_hdr(skb)->payload_len) +
				    sizeof(struct ipv6hdr), 1);
L
Linus Torvalds 已提交
395 396 397

			t = ip6t_get_target(e);
			IP_NF_ASSERT(t->u.kernel.target);
398 399 400 401

#if defined(CONFIG_NETFILTER_XT_TARGET_TRACE) || \
    defined(CONFIG_NETFILTER_XT_TARGET_TRACE_MODULE)
			/* The packet is traced: log it */
402 403
			if (unlikely(skb->nf_trace))
				trace_packet(skb, hook, in, out,
404 405
					     table->name, private, e);
#endif
L
Linus Torvalds 已提交
406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421
			/* Standard target? */
			if (!t->u.kernel.target->target) {
				int v;

				v = ((struct ip6t_standard_target *)t)->verdict;
				if (v < 0) {
					/* Pop from stack? */
					if (v != IP6T_RETURN) {
						verdict = (unsigned)(-v) - 1;
						break;
					}
					e = back;
					back = get_entry(table_base,
							 back->comefrom);
					continue;
				}
P
Patrick McHardy 已提交
422 423
				if (table_base + v != (void *)e + e->next_offset
				    && !(e->ipv6.flags & IP6T_F_GOTO)) {
L
Linus Torvalds 已提交
424 425 426 427 428 429 430 431 432 433 434 435
					/* Save old back ptr in next entry */
					struct ip6t_entry *next
						= (void *)e + e->next_offset;
					next->comefrom
						= (void *)back - table_base;
					/* set back pointer to next entry */
					back = next;
				}

				e = get_entry(table_base, v);
			} else {
				/* Targets which reenter must return
436
				   abs. verdicts */
L
Linus Torvalds 已提交
437 438 439 440
#ifdef CONFIG_NETFILTER_DEBUG
				((struct ip6t_entry *)table_base)->comefrom
					= 0xeeeeeeec;
#endif
441
				verdict = t->u.kernel.target->target(skb,
L
Linus Torvalds 已提交
442 443
								     in, out,
								     hook,
444
								     t->u.kernel.target,
445
								     t->data);
L
Linus Torvalds 已提交
446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471

#ifdef CONFIG_NETFILTER_DEBUG
				if (((struct ip6t_entry *)table_base)->comefrom
				    != 0xeeeeeeec
				    && verdict == IP6T_CONTINUE) {
					printk("Target %s reentered!\n",
					       t->u.kernel.target->name);
					verdict = NF_DROP;
				}
				((struct ip6t_entry *)table_base)->comefrom
					= 0x57acc001;
#endif
				if (verdict == IP6T_CONTINUE)
					e = (void *)e + e->next_offset;
				else
					/* Verdict */
					break;
			}
		} else {

		no_match:
			e = (void *)e + e->next_offset;
		}
	} while (!hotdrop);

#ifdef CONFIG_NETFILTER_DEBUG
R
Randy Dunlap 已提交
472
	((struct ip6t_entry *)table_base)->comefrom = NETFILTER_LINK_POISON;
L
Linus Torvalds 已提交
473 474 475 476 477 478 479 480 481 482 483 484 485 486 487
#endif
	read_unlock_bh(&table->lock);

#ifdef DEBUG_ALLOW_ALL
	return NF_ACCEPT;
#else
	if (hotdrop)
		return NF_DROP;
	else return verdict;
#endif
}

/* Figures out from what hook each rule can be called: returns 0 if
   there are loops.  Puts hook bitmask in comefrom. */
static int
488
mark_source_chains(struct xt_table_info *newinfo,
489
		   unsigned int valid_hooks, void *entry0)
L
Linus Torvalds 已提交
490 491 492 493 494
{
	unsigned int hook;

	/* No recursion; use packet counter to save back ptrs (reset
	   to 0 as we leave), and comefrom to save source hook bitmask */
495
	for (hook = 0; hook < NF_INET_NUMHOOKS; hook++) {
L
Linus Torvalds 已提交
496 497
		unsigned int pos = newinfo->hook_entry[hook];
		struct ip6t_entry *e
498
			= (struct ip6t_entry *)(entry0 + pos);
499
		int visited = e->comefrom & (1 << hook);
L
Linus Torvalds 已提交
500 501 502 503 504 505 506 507 508 509 510

		if (!(valid_hooks & (1 << hook)))
			continue;

		/* Set initial back pointer. */
		e->counters.pcnt = pos;

		for (;;) {
			struct ip6t_standard_target *t
				= (void *)ip6t_get_target(e);

511
			if (e->comefrom & (1 << NF_INET_NUMHOOKS)) {
L
Linus Torvalds 已提交
512 513 514 515 516
				printk("iptables: loop hook %u pos %u %08X.\n",
				       hook, pos, e->comefrom);
				return 0;
			}
			e->comefrom
517
				|= ((1 << hook) | (1 << NF_INET_NUMHOOKS));
L
Linus Torvalds 已提交
518 519

			/* Unconditional return/END. */
520
			if ((e->target_offset == sizeof(struct ip6t_entry)
L
Linus Torvalds 已提交
521 522 523
			    && (strcmp(t->target.u.user.name,
				       IP6T_STANDARD_TARGET) == 0)
			    && t->verdict < 0
524
			    && unconditional(&e->ipv6)) || visited) {
L
Linus Torvalds 已提交
525 526
				unsigned int oldpos, size;

527 528 529 530 531 532 533
				if (t->verdict < -NF_MAX_VERDICT - 1) {
					duprintf("mark_source_chains: bad "
						"negative verdict (%i)\n",
								t->verdict);
					return 0;
				}

L
Linus Torvalds 已提交
534 535 536
				/* Return: backtrack through the last
				   big jump. */
				do {
537
					e->comefrom ^= (1<<NF_INET_NUMHOOKS);
L
Linus Torvalds 已提交
538 539
#ifdef DEBUG_IP_FIREWALL_USER
					if (e->comefrom
540
					    & (1 << NF_INET_NUMHOOKS)) {
L
Linus Torvalds 已提交
541 542 543 544 545 546 547 548 549 550 551 552 553 554 555
						duprintf("Back unset "
							 "on hook %u "
							 "rule %u\n",
							 hook, pos);
					}
#endif
					oldpos = pos;
					pos = e->counters.pcnt;
					e->counters.pcnt = 0;

					/* We're at the start. */
					if (pos == oldpos)
						goto next;

					e = (struct ip6t_entry *)
556
						(entry0 + pos);
L
Linus Torvalds 已提交
557 558 559 560 561
				} while (oldpos == pos + e->next_offset);

				/* Move along one */
				size = e->next_offset;
				e = (struct ip6t_entry *)
562
					(entry0 + pos + size);
L
Linus Torvalds 已提交
563 564 565 566 567 568 569 570
				e->counters.pcnt = pos;
				pos += size;
			} else {
				int newpos = t->verdict;

				if (strcmp(t->target.u.user.name,
					   IP6T_STANDARD_TARGET) == 0
				    && newpos >= 0) {
571 572 573 574 575 576 577
					if (newpos > newinfo->size -
						sizeof(struct ip6t_entry)) {
						duprintf("mark_source_chains: "
							"bad verdict (%i)\n",
								newpos);
						return 0;
					}
L
Linus Torvalds 已提交
578 579 580 581 582 583 584 585
					/* This a jump; chase it. */
					duprintf("Jump rule %u -> %u\n",
						 pos, newpos);
				} else {
					/* ... this is a fallthru */
					newpos = pos + e->next_offset;
				}
				e = (struct ip6t_entry *)
586
					(entry0 + newpos);
L
Linus Torvalds 已提交
587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603
				e->counters.pcnt = pos;
				pos = newpos;
			}
		}
		next:
		duprintf("Finished chain %u\n", hook);
	}
	return 1;
}

static inline int
cleanup_match(struct ip6t_entry_match *m, unsigned int *i)
{
	if (i && (*i)-- == 0)
		return 1;

	if (m->u.kernel.match->destroy)
604
		m->u.kernel.match->destroy(m->u.kernel.match, m->data);
L
Linus Torvalds 已提交
605 606 607 608 609
	module_put(m->u.kernel.match->me);
	return 0;
}

static inline int
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658
check_entry(struct ip6t_entry *e, const char *name)
{
	struct ip6t_entry_target *t;

	if (!ip6_checkentry(&e->ipv6)) {
		duprintf("ip_tables: ip check failed %p %s.\n", e, name);
		return -EINVAL;
	}

	if (e->target_offset + sizeof(struct ip6t_entry_target) >
	    e->next_offset)
		return -EINVAL;

	t = ip6t_get_target(e);
	if (e->target_offset + t->u.target_size > e->next_offset)
		return -EINVAL;

	return 0;
}

static inline int check_match(struct ip6t_entry_match *m, const char *name,
			      const struct ip6t_ip6 *ipv6,
			      unsigned int hookmask, unsigned int *i)
{
	struct xt_match *match;
	int ret;

	match = m->u.kernel.match;
	ret = xt_check_match(match, AF_INET6, m->u.match_size - sizeof(*m),
			     name, hookmask, ipv6->proto,
			     ipv6->invflags & IP6T_INV_PROTO);
	if (!ret && m->u.kernel.match->checkentry
	    && !m->u.kernel.match->checkentry(name, ipv6, match, m->data,
					      hookmask)) {
		duprintf("ip_tables: check failed for `%s'.\n",
			 m->u.kernel.match->name);
		ret = -EINVAL;
	}
	if (!ret)
		(*i)++;
	return ret;
}

static inline int
find_check_match(struct ip6t_entry_match *m,
		 const char *name,
		 const struct ip6t_ip6 *ipv6,
		 unsigned int hookmask,
		 unsigned int *i)
L
Linus Torvalds 已提交
659
{
660
	struct xt_match *match;
661
	int ret;
L
Linus Torvalds 已提交
662

663
	match = try_then_request_module(xt_find_match(AF_INET6, m->u.user.name,
664
					m->u.user.revision),
665 666
					"ip6t_%s", m->u.user.name);
	if (IS_ERR(match) || !match) {
667
		duprintf("find_check_match: `%s' not found\n", m->u.user.name);
668
		return match ? PTR_ERR(match) : -ENOENT;
L
Linus Torvalds 已提交
669 670 671
	}
	m->u.kernel.match = match;

672
	ret = check_match(m, name, ipv6, hookmask, i);
673 674 675
	if (ret)
		goto err;

L
Linus Torvalds 已提交
676
	return 0;
677 678 679
err:
	module_put(m->u.kernel.match->me);
	return ret;
L
Linus Torvalds 已提交
680 681
}

682
static inline int check_target(struct ip6t_entry *e, const char *name)
L
Linus Torvalds 已提交
683 684
{
	struct ip6t_entry_target *t;
685
	struct xt_target *target;
L
Linus Torvalds 已提交
686 687
	int ret;

688 689 690 691 692 693 694 695 696 697 698
	t = ip6t_get_target(e);
	target = t->u.kernel.target;
	ret = xt_check_target(target, AF_INET6, t->u.target_size - sizeof(*t),
			      name, e->comefrom, e->ipv6.proto,
			      e->ipv6.invflags & IP6T_INV_PROTO);
	if (!ret && t->u.kernel.target->checkentry
	    && !t->u.kernel.target->checkentry(name, e, target, t->data,
					       e->comefrom)) {
		duprintf("ip_tables: check failed for `%s'.\n",
			 t->u.kernel.target->name);
		ret = -EINVAL;
L
Linus Torvalds 已提交
699
	}
700 701
	return ret;
}
L
Linus Torvalds 已提交
702

703 704 705 706 707 708 709 710 711 712 713 714
static inline int
find_check_entry(struct ip6t_entry *e, const char *name, unsigned int size,
		 unsigned int *i)
{
	struct ip6t_entry_target *t;
	struct xt_target *target;
	int ret;
	unsigned int j;

	ret = check_entry(e, name);
	if (ret)
		return ret;
715

L
Linus Torvalds 已提交
716
	j = 0;
717 718
	ret = IP6T_MATCH_ITERATE(e, find_check_match, name, &e->ipv6,
				 e->comefrom, &j);
L
Linus Torvalds 已提交
719 720 721 722
	if (ret != 0)
		goto cleanup_matches;

	t = ip6t_get_target(e);
723 724 725
	target = try_then_request_module(xt_find_target(AF_INET6,
							t->u.user.name,
							t->u.user.revision),
726 727
					 "ip6t_%s", t->u.user.name);
	if (IS_ERR(target) || !target) {
728
		duprintf("find_check_entry: `%s' not found\n", t->u.user.name);
729
		ret = target ? PTR_ERR(target) : -ENOENT;
L
Linus Torvalds 已提交
730 731 732
		goto cleanup_matches;
	}
	t->u.kernel.target = target;
733

734
	ret = check_target(e, name);
735 736 737
	if (ret)
		goto err;

L
Linus Torvalds 已提交
738 739
	(*i)++;
	return 0;
740 741
 err:
	module_put(t->u.kernel.target->me);
L
Linus Torvalds 已提交
742 743 744 745 746 747 748
 cleanup_matches:
	IP6T_MATCH_ITERATE(e, cleanup_match, &j);
	return ret;
}

static inline int
check_entry_size_and_hooks(struct ip6t_entry *e,
749
			   struct xt_table_info *newinfo,
L
Linus Torvalds 已提交
750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771
			   unsigned char *base,
			   unsigned char *limit,
			   const unsigned int *hook_entries,
			   const unsigned int *underflows,
			   unsigned int *i)
{
	unsigned int h;

	if ((unsigned long)e % __alignof__(struct ip6t_entry) != 0
	    || (unsigned char *)e + sizeof(struct ip6t_entry) >= limit) {
		duprintf("Bad offset %p\n", e);
		return -EINVAL;
	}

	if (e->next_offset
	    < sizeof(struct ip6t_entry) + sizeof(struct ip6t_entry_target)) {
		duprintf("checking: element %p size %u\n",
			 e, e->next_offset);
		return -EINVAL;
	}

	/* Check hooks & underflows */
772
	for (h = 0; h < NF_INET_NUMHOOKS; h++) {
L
Linus Torvalds 已提交
773 774 775 776 777 778 779
		if ((unsigned char *)e - base == hook_entries[h])
			newinfo->hook_entry[h] = hook_entries[h];
		if ((unsigned char *)e - base == underflows[h])
			newinfo->underflow[h] = underflows[h];
	}

	/* FIXME: underflows must be unconditional, standard verdicts
780
	   < 0 (not IP6T_RETURN). --RR */
L
Linus Torvalds 已提交
781 782

	/* Clear counters and comefrom */
783
	e->counters = ((struct xt_counters) { 0, 0 });
L
Linus Torvalds 已提交
784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801
	e->comefrom = 0;

	(*i)++;
	return 0;
}

static inline int
cleanup_entry(struct ip6t_entry *e, unsigned int *i)
{
	struct ip6t_entry_target *t;

	if (i && (*i)-- == 0)
		return 1;

	/* Cleanup all matches */
	IP6T_MATCH_ITERATE(e, cleanup_match, NULL);
	t = ip6t_get_target(e);
	if (t->u.kernel.target->destroy)
802
		t->u.kernel.target->destroy(t->u.kernel.target, t->data);
L
Linus Torvalds 已提交
803 804 805 806 807 808 809 810 811
	module_put(t->u.kernel.target->me);
	return 0;
}

/* Checks and translates the user-supplied table segment (held in
   newinfo) */
static int
translate_table(const char *name,
		unsigned int valid_hooks,
812
		struct xt_table_info *newinfo,
813
		void *entry0,
L
Linus Torvalds 已提交
814 815 816 817 818 819 820 821 822 823 824 825
		unsigned int size,
		unsigned int number,
		const unsigned int *hook_entries,
		const unsigned int *underflows)
{
	unsigned int i;
	int ret;

	newinfo->size = size;
	newinfo->number = number;

	/* Init all hooks to impossible value. */
826
	for (i = 0; i < NF_INET_NUMHOOKS; i++) {
L
Linus Torvalds 已提交
827 828 829 830 831 832 833
		newinfo->hook_entry[i] = 0xFFFFFFFF;
		newinfo->underflow[i] = 0xFFFFFFFF;
	}

	duprintf("translate_table: size %u\n", newinfo->size);
	i = 0;
	/* Walk through entries, checking offsets. */
834
	ret = IP6T_ENTRY_ITERATE(entry0, newinfo->size,
L
Linus Torvalds 已提交
835 836
				check_entry_size_and_hooks,
				newinfo,
837 838
				entry0,
				entry0 + size,
L
Linus Torvalds 已提交
839 840 841 842 843 844 845 846 847 848 849
				hook_entries, underflows, &i);
	if (ret != 0)
		return ret;

	if (i != number) {
		duprintf("translate_table: %u not %u entries\n",
			 i, number);
		return -EINVAL;
	}

	/* Check hooks all assigned */
850
	for (i = 0; i < NF_INET_NUMHOOKS; i++) {
L
Linus Torvalds 已提交
851 852 853 854 855 856 857 858 859 860 861 862 863 864 865
		/* Only hooks which are valid */
		if (!(valid_hooks & (1 << i)))
			continue;
		if (newinfo->hook_entry[i] == 0xFFFFFFFF) {
			duprintf("Invalid hook entry %u %u\n",
				 i, hook_entries[i]);
			return -EINVAL;
		}
		if (newinfo->underflow[i] == 0xFFFFFFFF) {
			duprintf("Invalid underflow %u %u\n",
				 i, underflows[i]);
			return -EINVAL;
		}
	}

866 867 868
	if (!mark_source_chains(newinfo, valid_hooks, entry0))
		return -ELOOP;

L
Linus Torvalds 已提交
869 870
	/* Finally, each sanity check must pass */
	i = 0;
871
	ret = IP6T_ENTRY_ITERATE(entry0, newinfo->size,
872
				find_check_entry, name, size, &i);
L
Linus Torvalds 已提交
873

874 875 876 877 878
	if (ret != 0) {
		IP6T_ENTRY_ITERATE(entry0, newinfo->size,
				   cleanup_entry, &i);
		return ret;
	}
L
Linus Torvalds 已提交
879 880

	/* And one copy for every other CPU */
881
	for_each_possible_cpu(i) {
882 883
		if (newinfo->entries[i] && newinfo->entries[i] != entry0)
			memcpy(newinfo->entries[i], entry0, newinfo->size);
L
Linus Torvalds 已提交
884 885
	}

886
	return 0;
L
Linus Torvalds 已提交
887 888 889 890 891
}

/* Gets counters. */
static inline int
add_entry_to_counter(const struct ip6t_entry *e,
892
		     struct xt_counters total[],
L
Linus Torvalds 已提交
893 894 895 896 897 898 899 900
		     unsigned int *i)
{
	ADD_COUNTER(total[*i], e->counters.bcnt, e->counters.pcnt);

	(*i)++;
	return 0;
}

901 902 903 904 905 906 907 908 909 910 911
static inline int
set_entry_to_counter(const struct ip6t_entry *e,
		     struct ip6t_counters total[],
		     unsigned int *i)
{
	SET_COUNTER(total[*i], e->counters.bcnt, e->counters.pcnt);

	(*i)++;
	return 0;
}

L
Linus Torvalds 已提交
912
static void
913 914
get_counters(const struct xt_table_info *t,
	     struct xt_counters counters[])
L
Linus Torvalds 已提交
915 916 917
{
	unsigned int cpu;
	unsigned int i;
918 919 920 921 922 923 924 925 926 927 928 929 930 931 932
	unsigned int curcpu;

	/* Instead of clearing (by a previous call to memset())
	 * the counters and using adds, we set the counters
	 * with data used by 'current' CPU
	 * We dont care about preemption here.
	 */
	curcpu = raw_smp_processor_id();

	i = 0;
	IP6T_ENTRY_ITERATE(t->entries[curcpu],
			   t->size,
			   set_entry_to_counter,
			   counters,
			   &i);
L
Linus Torvalds 已提交
933

934
	for_each_possible_cpu(cpu) {
935 936
		if (cpu == curcpu)
			continue;
L
Linus Torvalds 已提交
937
		i = 0;
938
		IP6T_ENTRY_ITERATE(t->entries[cpu],
L
Linus Torvalds 已提交
939 940 941 942 943 944 945
				  t->size,
				  add_entry_to_counter,
				  counters,
				  &i);
	}
}

946
static inline struct xt_counters *alloc_counters(struct xt_table *table)
L
Linus Torvalds 已提交
947
{
948
	unsigned int countersize;
949 950
	struct xt_counters *counters;
	struct xt_table_info *private = table->private;
L
Linus Torvalds 已提交
951 952 953 954

	/* We need atomic snapshot of counters: rest doesn't change
	   (other than comefrom, which userspace doesn't care
	   about). */
955
	countersize = sizeof(struct xt_counters) * private->number;
956
	counters = vmalloc_node(countersize, numa_node_id());
L
Linus Torvalds 已提交
957 958

	if (counters == NULL)
959
		return ERR_PTR(-ENOMEM);
L
Linus Torvalds 已提交
960 961 962

	/* First, sum counters... */
	write_lock_bh(&table->lock);
963
	get_counters(private, counters);
L
Linus Torvalds 已提交
964 965
	write_unlock_bh(&table->lock);

966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984
	return counters;
}

static int
copy_entries_to_user(unsigned int total_size,
		     struct xt_table *table,
		     void __user *userptr)
{
	unsigned int off, num;
	struct ip6t_entry *e;
	struct xt_counters *counters;
	struct xt_table_info *private = table->private;
	int ret = 0;
	void *loc_cpu_entry;

	counters = alloc_counters(table);
	if (IS_ERR(counters))
		return PTR_ERR(counters);

985
	/* choose the copy that is on ourc node/cpu */
986
	loc_cpu_entry = private->entries[raw_smp_processor_id()];
987
	if (copy_to_user(userptr, loc_cpu_entry, total_size) != 0) {
L
Linus Torvalds 已提交
988 989 990 991 992 993 994 995 996 997 998
		ret = -EFAULT;
		goto free_counters;
	}

	/* FIXME: use iterator macros --RR */
	/* ... then go back and fix counters and names */
	for (off = 0, num = 0; off < total_size; off += e->next_offset, num++){
		unsigned int i;
		struct ip6t_entry_match *m;
		struct ip6t_entry_target *t;

999
		e = (struct ip6t_entry *)(loc_cpu_entry + off);
L
Linus Torvalds 已提交
1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039
		if (copy_to_user(userptr + off
				 + offsetof(struct ip6t_entry, counters),
				 &counters[num],
				 sizeof(counters[num])) != 0) {
			ret = -EFAULT;
			goto free_counters;
		}

		for (i = sizeof(struct ip6t_entry);
		     i < e->target_offset;
		     i += m->u.match_size) {
			m = (void *)e + i;

			if (copy_to_user(userptr + off + i
					 + offsetof(struct ip6t_entry_match,
						    u.user.name),
					 m->u.kernel.match->name,
					 strlen(m->u.kernel.match->name)+1)
			    != 0) {
				ret = -EFAULT;
				goto free_counters;
			}
		}

		t = ip6t_get_target(e);
		if (copy_to_user(userptr + off + e->target_offset
				 + offsetof(struct ip6t_entry_target,
					    u.user.name),
				 t->u.kernel.target->name,
				 strlen(t->u.kernel.target->name)+1) != 0) {
			ret = -EFAULT;
			goto free_counters;
		}
	}

 free_counters:
	vfree(counters);
	return ret;
}

1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
static int get_info(void __user *user, int *len)
{
	char name[IP6T_TABLE_MAXNAMELEN];
	struct xt_table *t;
	int ret;

	if (*len != sizeof(struct ip6t_getinfo)) {
		duprintf("length %u != %u\n", *len,
			 sizeof(struct ip6t_getinfo));
		return -EINVAL;
	}

	if (copy_from_user(name, user, sizeof(name)) != 0)
		return -EFAULT;

	name[IP6T_TABLE_MAXNAMELEN-1] = '\0';

	t = try_then_request_module(xt_find_table_lock(AF_INET6, name),
				    "ip6table_%s", name);
	if (t && !IS_ERR(t)) {
		struct ip6t_getinfo info;
		struct xt_table_info *private = t->private;

		info.valid_hooks = t->valid_hooks;
		memcpy(info.hook_entry, private->hook_entry,
		       sizeof(info.hook_entry));
		memcpy(info.underflow, private->underflow,
		       sizeof(info.underflow));
		info.num_entries = private->number;
		info.size = private->size;
		memcpy(info.name, name, sizeof(info.name));

		if (copy_to_user(user, &info, *len) != 0)
			ret = -EFAULT;
		else
			ret = 0;

		xt_table_unlock(t);
		module_put(t->me);
	} else
		ret = t ? PTR_ERR(t) : -ENOENT;
	return ret;
}

L
Linus Torvalds 已提交
1084
static int
1085
get_entries(struct ip6t_get_entries __user *uptr, int *len)
L
Linus Torvalds 已提交
1086 1087
{
	int ret;
1088
	struct ip6t_get_entries get;
1089
	struct xt_table *t;
L
Linus Torvalds 已提交
1090

1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103
	if (*len < sizeof(get)) {
		duprintf("get_entries: %u < %u\n", *len, sizeof(get));
		return -EINVAL;
	}
	if (copy_from_user(&get, uptr, sizeof(get)) != 0)
		return -EFAULT;
	if (*len != sizeof(struct ip6t_get_entries) + get.size) {
		duprintf("get_entries: %u != %u\n", *len,
			 sizeof(struct ip6t_get_entries) + get.size);
		return -EINVAL;
	}

	t = xt_find_table_lock(AF_INET6, get.name);
1104
	if (t && !IS_ERR(t)) {
1105 1106
		struct xt_table_info *private = t->private;
		duprintf("t->private->number = %u\n", private->number);
1107
		if (get.size == private->size)
1108
			ret = copy_entries_to_user(private->size,
L
Linus Torvalds 已提交
1109 1110 1111
						   t, uptr->entrytable);
		else {
			duprintf("get_entries: I've got %u not %u!\n",
1112
				 private->size, entries->size);
L
Linus Torvalds 已提交
1113 1114
			ret = -EINVAL;
		}
1115
		module_put(t->me);
1116
		xt_table_unlock(t);
L
Linus Torvalds 已提交
1117
	} else
1118
		ret = t ? PTR_ERR(t) : -ENOENT;
L
Linus Torvalds 已提交
1119 1120 1121 1122 1123 1124 1125 1126 1127

	return ret;
}

static int
do_replace(void __user *user, unsigned int len)
{
	int ret;
	struct ip6t_replace tmp;
1128 1129 1130
	struct xt_table *t;
	struct xt_table_info *newinfo, *oldinfo;
	struct xt_counters *counters;
1131
	void *loc_cpu_entry, *loc_cpu_old_entry;
L
Linus Torvalds 已提交
1132 1133 1134 1135

	if (copy_from_user(&tmp, user, sizeof(tmp)) != 0)
		return -EFAULT;

1136 1137 1138 1139
	/* overflow check */
	if (tmp.num_counters >= INT_MAX / sizeof(struct xt_counters))
		return -ENOMEM;

1140
	newinfo = xt_alloc_table_info(tmp.size);
L
Linus Torvalds 已提交
1141 1142 1143
	if (!newinfo)
		return -ENOMEM;

1144 1145 1146
	/* choose the copy that is on our node/cpu */
	loc_cpu_entry = newinfo->entries[raw_smp_processor_id()];
	if (copy_from_user(loc_cpu_entry, user + sizeof(tmp),
L
Linus Torvalds 已提交
1147 1148 1149 1150 1151
			   tmp.size) != 0) {
		ret = -EFAULT;
		goto free_newinfo;
	}

1152 1153
	counters = vmalloc_node(tmp.num_counters * sizeof(struct xt_counters),
				numa_node_id());
L
Linus Torvalds 已提交
1154 1155 1156 1157 1158 1159
	if (!counters) {
		ret = -ENOMEM;
		goto free_newinfo;
	}

	ret = translate_table(tmp.name, tmp.valid_hooks,
1160
			      newinfo, loc_cpu_entry, tmp.size, tmp.num_entries,
L
Linus Torvalds 已提交
1161 1162 1163 1164 1165 1166
			      tmp.hook_entry, tmp.underflow);
	if (ret != 0)
		goto free_newinfo_counters;

	duprintf("ip_tables: Translated table\n");

1167
	t = try_then_request_module(xt_find_table_lock(AF_INET6, tmp.name),
1168 1169 1170
				    "ip6table_%s", tmp.name);
	if (!t || IS_ERR(t)) {
		ret = t ? PTR_ERR(t) : -ENOENT;
L
Linus Torvalds 已提交
1171
		goto free_newinfo_counters_untrans;
1172
	}
L
Linus Torvalds 已提交
1173 1174 1175 1176 1177 1178

	/* You lied! */
	if (tmp.valid_hooks != t->valid_hooks) {
		duprintf("Valid hook crap: %08X vs %08X\n",
			 tmp.valid_hooks, t->valid_hooks);
		ret = -EINVAL;
1179
		goto put_module;
L
Linus Torvalds 已提交
1180 1181
	}

1182
	oldinfo = xt_replace_table(t, tmp.num_counters, newinfo, &ret);
L
Linus Torvalds 已提交
1183 1184 1185 1186 1187 1188
	if (!oldinfo)
		goto put_module;

	/* Update module usage count based on number of rules */
	duprintf("do_replace: oldnum=%u, initnum=%u, newnum=%u\n",
		oldinfo->number, oldinfo->initial_entries, newinfo->number);
1189 1190
	if ((oldinfo->number > oldinfo->initial_entries) ||
	    (newinfo->number <= oldinfo->initial_entries))
L
Linus Torvalds 已提交
1191 1192 1193 1194 1195 1196 1197 1198
		module_put(t->me);
	if ((oldinfo->number > oldinfo->initial_entries) &&
	    (newinfo->number <= oldinfo->initial_entries))
		module_put(t->me);

	/* Get the old counters. */
	get_counters(oldinfo, counters);
	/* Decrease module usage counts and free resource */
1199 1200
	loc_cpu_old_entry = oldinfo->entries[raw_smp_processor_id()];
	IP6T_ENTRY_ITERATE(loc_cpu_old_entry, oldinfo->size, cleanup_entry,NULL);
1201
	xt_free_table_info(oldinfo);
L
Linus Torvalds 已提交
1202
	if (copy_to_user(tmp.counters, counters,
1203
			 sizeof(struct xt_counters) * tmp.num_counters) != 0)
L
Linus Torvalds 已提交
1204 1205
		ret = -EFAULT;
	vfree(counters);
1206
	xt_table_unlock(t);
L
Linus Torvalds 已提交
1207 1208 1209 1210
	return ret;

 put_module:
	module_put(t->me);
1211
	xt_table_unlock(t);
L
Linus Torvalds 已提交
1212
 free_newinfo_counters_untrans:
1213
	IP6T_ENTRY_ITERATE(loc_cpu_entry, newinfo->size, cleanup_entry,NULL);
L
Linus Torvalds 已提交
1214 1215 1216
 free_newinfo_counters:
	vfree(counters);
 free_newinfo:
1217
	xt_free_table_info(newinfo);
L
Linus Torvalds 已提交
1218 1219 1220 1221 1222 1223 1224
	return ret;
}

/* We're lazy, and add to the first CPU; overflow works its fey magic
 * and everything is OK. */
static inline int
add_counter_to_entry(struct ip6t_entry *e,
1225
		     const struct xt_counters addme[],
L
Linus Torvalds 已提交
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
		     unsigned int *i)
{
#if 0
	duprintf("add_counter: Entry %u %lu/%lu + %lu/%lu\n",
		 *i,
		 (long unsigned int)e->counters.pcnt,
		 (long unsigned int)e->counters.bcnt,
		 (long unsigned int)addme[*i].pcnt,
		 (long unsigned int)addme[*i].bcnt);
#endif

	ADD_COUNTER(e->counters, addme[*i].bcnt, addme[*i].pcnt);

	(*i)++;
	return 0;
}

static int
do_add_counters(void __user *user, unsigned int len)
{
	unsigned int i;
1247 1248 1249
	struct xt_counters_info tmp, *paddc;
	struct xt_table_info *private;
	struct xt_table *t;
1250
	int ret = 0;
1251
	void *loc_cpu_entry;
L
Linus Torvalds 已提交
1252 1253 1254 1255

	if (copy_from_user(&tmp, user, sizeof(tmp)) != 0)
		return -EFAULT;

1256
	if (len != sizeof(tmp) + tmp.num_counters*sizeof(struct xt_counters))
L
Linus Torvalds 已提交
1257 1258
		return -EINVAL;

1259
	paddc = vmalloc_node(len, numa_node_id());
L
Linus Torvalds 已提交
1260 1261 1262 1263 1264 1265 1266 1267
	if (!paddc)
		return -ENOMEM;

	if (copy_from_user(paddc, user, len) != 0) {
		ret = -EFAULT;
		goto free;
	}

1268
	t = xt_find_table_lock(AF_INET6, tmp.name);
1269 1270
	if (!t || IS_ERR(t)) {
		ret = t ? PTR_ERR(t) : -ENOENT;
L
Linus Torvalds 已提交
1271
		goto free;
1272
	}
L
Linus Torvalds 已提交
1273 1274

	write_lock_bh(&t->lock);
1275
	private = t->private;
1276
	if (private->number != tmp.num_counters) {
L
Linus Torvalds 已提交
1277 1278 1279 1280 1281
		ret = -EINVAL;
		goto unlock_up_free;
	}

	i = 0;
1282
	/* Choose the copy that is on our node */
1283
	loc_cpu_entry = private->entries[smp_processor_id()];
1284
	IP6T_ENTRY_ITERATE(loc_cpu_entry,
1285
			  private->size,
L
Linus Torvalds 已提交
1286 1287 1288 1289 1290
			  add_counter_to_entry,
			  paddc->counters,
			  &i);
 unlock_up_free:
	write_unlock_bh(&t->lock);
1291
	xt_table_unlock(t);
1292
	module_put(t->me);
L
Linus Torvalds 已提交
1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
 free:
	vfree(paddc);

	return ret;
}

static int
do_ip6t_set_ctl(struct sock *sk, int cmd, void __user *user, unsigned int len)
{
	int ret;

	if (!capable(CAP_NET_ADMIN))
		return -EPERM;

	switch (cmd) {
	case IP6T_SO_SET_REPLACE:
		ret = do_replace(user, len);
		break;

	case IP6T_SO_SET_ADD_COUNTERS:
		ret = do_add_counters(user, len);
		break;

	default:
		duprintf("do_ip6t_set_ctl:  unknown request %i\n", cmd);
		ret = -EINVAL;
	}

	return ret;
}

static int
do_ip6t_get_ctl(struct sock *sk, int cmd, void __user *user, int *len)
{
	int ret;

	if (!capable(CAP_NET_ADMIN))
		return -EPERM;

	switch (cmd) {
1333 1334 1335
	case IP6T_SO_GET_INFO:
		ret = get_info(user, len);
		break;
L
Linus Torvalds 已提交
1336

1337 1338
	case IP6T_SO_GET_ENTRIES:
		ret = get_entries(user, len);
L
Linus Torvalds 已提交
1339 1340
		break;

1341 1342 1343
	case IP6T_SO_GET_REVISION_MATCH:
	case IP6T_SO_GET_REVISION_TARGET: {
		struct ip6t_get_revision rev;
1344
		int target;
1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355

		if (*len != sizeof(rev)) {
			ret = -EINVAL;
			break;
		}
		if (copy_from_user(&rev, user, sizeof(rev)) != 0) {
			ret = -EFAULT;
			break;
		}

		if (cmd == IP6T_SO_GET_REVISION_TARGET)
1356
			target = 1;
1357
		else
1358
			target = 0;
1359

1360 1361 1362
		try_then_request_module(xt_find_revision(AF_INET6, rev.name,
							 rev.revision,
							 target, &ret),
1363 1364 1365 1366
					"ip6t_%s", rev.name);
		break;
	}

L
Linus Torvalds 已提交
1367 1368 1369 1370 1371 1372 1373 1374
	default:
		duprintf("do_ip6t_get_ctl: unknown request %i\n", cmd);
		ret = -EINVAL;
	}

	return ret;
}

1375
int ip6t_register_table(struct xt_table *table,
L
Linus Torvalds 已提交
1376 1377 1378
			const struct ip6t_replace *repl)
{
	int ret;
1379
	struct xt_table_info *newinfo;
1380
	struct xt_table_info bootstrap
L
Linus Torvalds 已提交
1381
		= { 0, 0, 0, { 0 }, { 0 }, { } };
1382
	void *loc_cpu_entry;
L
Linus Torvalds 已提交
1383

1384
	newinfo = xt_alloc_table_info(repl->size);
L
Linus Torvalds 已提交
1385 1386 1387
	if (!newinfo)
		return -ENOMEM;

1388 1389 1390
	/* choose the copy on our node/cpu */
	loc_cpu_entry = newinfo->entries[raw_smp_processor_id()];
	memcpy(loc_cpu_entry, repl->entries, repl->size);
L
Linus Torvalds 已提交
1391 1392

	ret = translate_table(table->name, table->valid_hooks,
1393
			      newinfo, loc_cpu_entry, repl->size,
L
Linus Torvalds 已提交
1394 1395 1396 1397
			      repl->num_entries,
			      repl->hook_entry,
			      repl->underflow);
	if (ret != 0) {
1398
		xt_free_table_info(newinfo);
L
Linus Torvalds 已提交
1399 1400 1401
		return ret;
	}

1402 1403
	ret = xt_register_table(table, &bootstrap, newinfo);
	if (ret != 0) {
1404
		xt_free_table_info(newinfo);
L
Linus Torvalds 已提交
1405 1406 1407
		return ret;
	}

1408
	return 0;
L
Linus Torvalds 已提交
1409 1410
}

1411
void ip6t_unregister_table(struct xt_table *table)
L
Linus Torvalds 已提交
1412
{
1413
	struct xt_table_info *private;
1414 1415
	void *loc_cpu_entry;

1416
	private = xt_unregister_table(table);
L
Linus Torvalds 已提交
1417 1418

	/* Decrease module usage counts and free resources */
1419 1420 1421
	loc_cpu_entry = private->entries[raw_smp_processor_id()];
	IP6T_ENTRY_ITERATE(loc_cpu_entry, private->size, cleanup_entry, NULL);
	xt_free_table_info(private);
L
Linus Torvalds 已提交
1422 1423 1424
}

/* Returns 1 if the type and code is matched by the range, 0 otherwise */
1425
static inline bool
L
Linus Torvalds 已提交
1426 1427
icmp6_type_code_match(u_int8_t test_type, u_int8_t min_code, u_int8_t max_code,
		     u_int8_t type, u_int8_t code,
1428
		     bool invert)
L
Linus Torvalds 已提交
1429 1430 1431 1432 1433
{
	return (type == test_type && code >= min_code && code <= max_code)
		^ invert;
}

1434
static bool
L
Linus Torvalds 已提交
1435 1436 1437
icmp6_match(const struct sk_buff *skb,
	   const struct net_device *in,
	   const struct net_device *out,
1438
	   const struct xt_match *match,
L
Linus Torvalds 已提交
1439 1440 1441
	   const void *matchinfo,
	   int offset,
	   unsigned int protoff,
1442
	   bool *hotdrop)
L
Linus Torvalds 已提交
1443 1444 1445 1446 1447 1448
{
	struct icmp6hdr _icmp, *ic;
	const struct ip6t_icmp *icmpinfo = matchinfo;

	/* Must not be a fragment. */
	if (offset)
1449
		return false;
L
Linus Torvalds 已提交
1450 1451 1452 1453 1454 1455

	ic = skb_header_pointer(skb, protoff, sizeof(_icmp), &_icmp);
	if (ic == NULL) {
		/* We've been asked to examine this packet, and we
		   can't.  Hence, no choice but to drop. */
		duprintf("Dropping evil ICMP tinygram.\n");
1456
		*hotdrop = true;
1457
		return false;
L
Linus Torvalds 已提交
1458 1459 1460 1461 1462 1463 1464 1465 1466 1467
	}

	return icmp6_type_code_match(icmpinfo->type,
				     icmpinfo->code[0],
				     icmpinfo->code[1],
				     ic->icmp6_type, ic->icmp6_code,
				     !!(icmpinfo->invflags&IP6T_ICMP_INV));
}

/* Called when user tries to insert an entry of this type. */
1468
static bool
L
Linus Torvalds 已提交
1469
icmp6_checkentry(const char *tablename,
1470
	   const void *entry,
1471
	   const struct xt_match *match,
L
Linus Torvalds 已提交
1472 1473 1474 1475 1476
	   void *matchinfo,
	   unsigned int hook_mask)
{
	const struct ip6t_icmp *icmpinfo = matchinfo;

1477 1478
	/* Must specify no unknown invflags */
	return !(icmpinfo->invflags & ~IP6T_ICMP_INV);
L
Linus Torvalds 已提交
1479 1480 1481
}

/* The built-in targets: standard (NULL) and error. */
1482
static struct xt_target ip6t_standard_target __read_mostly = {
L
Linus Torvalds 已提交
1483
	.name		= IP6T_STANDARD_TARGET,
1484
	.targetsize	= sizeof(int),
1485
	.family		= AF_INET6,
L
Linus Torvalds 已提交
1486 1487
};

1488
static struct xt_target ip6t_error_target __read_mostly = {
L
Linus Torvalds 已提交
1489 1490
	.name		= IP6T_ERROR_TARGET,
	.target		= ip6t_error,
1491
	.targetsize	= IP6T_FUNCTION_MAXNAMELEN,
1492
	.family		= AF_INET6,
L
Linus Torvalds 已提交
1493 1494 1495 1496 1497 1498 1499 1500 1501 1502
};

static struct nf_sockopt_ops ip6t_sockopts = {
	.pf		= PF_INET6,
	.set_optmin	= IP6T_BASE_CTL,
	.set_optmax	= IP6T_SO_SET_MAX+1,
	.set		= do_ip6t_set_ctl,
	.get_optmin	= IP6T_BASE_CTL,
	.get_optmax	= IP6T_SO_GET_MAX+1,
	.get		= do_ip6t_get_ctl,
1503
	.owner		= THIS_MODULE,
L
Linus Torvalds 已提交
1504 1505
};

1506
static struct xt_match icmp6_matchstruct __read_mostly = {
L
Linus Torvalds 已提交
1507 1508
	.name		= "icmp6",
	.match		= &icmp6_match,
1509 1510 1511
	.matchsize	= sizeof(struct ip6t_icmp),
	.checkentry	= icmp6_checkentry,
	.proto		= IPPROTO_ICMPV6,
1512
	.family		= AF_INET6,
L
Linus Torvalds 已提交
1513 1514
};

1515
static int __init ip6_tables_init(void)
L
Linus Torvalds 已提交
1516 1517 1518
{
	int ret;

1519 1520 1521
	ret = xt_proto_init(AF_INET6);
	if (ret < 0)
		goto err1;
1522

L
Linus Torvalds 已提交
1523
	/* Noone else will be downing sem now, so we won't sleep */
1524 1525 1526 1527 1528 1529 1530 1531 1532
	ret = xt_register_target(&ip6t_standard_target);
	if (ret < 0)
		goto err2;
	ret = xt_register_target(&ip6t_error_target);
	if (ret < 0)
		goto err3;
	ret = xt_register_match(&icmp6_matchstruct);
	if (ret < 0)
		goto err4;
L
Linus Torvalds 已提交
1533 1534 1535

	/* Register setsockopt */
	ret = nf_register_sockopt(&ip6t_sockopts);
1536 1537
	if (ret < 0)
		goto err5;
L
Linus Torvalds 已提交
1538

1539
	printk(KERN_INFO "ip6_tables: (C) 2000-2006 Netfilter Core Team\n");
L
Linus Torvalds 已提交
1540
	return 0;
1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551

err5:
	xt_unregister_match(&icmp6_matchstruct);
err4:
	xt_unregister_target(&ip6t_error_target);
err3:
	xt_unregister_target(&ip6t_standard_target);
err2:
	xt_proto_fini(AF_INET6);
err1:
	return ret;
L
Linus Torvalds 已提交
1552 1553
}

1554
static void __exit ip6_tables_fini(void)
L
Linus Torvalds 已提交
1555 1556
{
	nf_unregister_sockopt(&ip6t_sockopts);
1557 1558 1559
	xt_unregister_match(&icmp6_matchstruct);
	xt_unregister_target(&ip6t_error_target);
	xt_unregister_target(&ip6t_standard_target);
1560
	xt_proto_fini(AF_INET6);
L
Linus Torvalds 已提交
1561 1562
}

1563
/*
1564 1565 1566 1567 1568 1569 1570
 * find the offset to specified header or the protocol number of last header
 * if target < 0. "last header" is transport protocol header, ESP, or
 * "No next header".
 *
 * If target header is found, its offset is set in *offset and return protocol
 * number. Otherwise, return -1.
 *
1571 1572 1573
 * If the first fragment doesn't contain the final protocol header or
 * NEXTHDR_NONE it is considered invalid.
 *
1574 1575 1576 1577
 * Note that non-1st fragment is special case that "the protocol number
 * of last header" is "next header" field in Fragment header. In this case,
 * *offset is meaningless and fragment offset is stored in *fragoff if fragoff
 * isn't NULL.
1578 1579
 *
 */
1580 1581
int ipv6_find_hdr(const struct sk_buff *skb, unsigned int *offset,
		  int target, unsigned short *fragoff)
1582
{
1583
	unsigned int start = skb_network_offset(skb) + sizeof(struct ipv6hdr);
1584
	u8 nexthdr = ipv6_hdr(skb)->nexthdr;
1585 1586
	unsigned int len = skb->len - start;

1587 1588 1589
	if (fragoff)
		*fragoff = 0;

1590 1591 1592 1593
	while (nexthdr != target) {
		struct ipv6_opt_hdr _hdr, *hp;
		unsigned int hdrlen;

1594 1595 1596
		if ((!ipv6_ext_hdr(nexthdr)) || nexthdr == NEXTHDR_NONE) {
			if (target < 0)
				break;
1597
			return -ENOENT;
1598 1599
		}

1600 1601
		hp = skb_header_pointer(skb, start, sizeof(_hdr), &_hdr);
		if (hp == NULL)
1602
			return -EBADMSG;
1603
		if (nexthdr == NEXTHDR_FRAGMENT) {
A
Al Viro 已提交
1604 1605
			unsigned short _frag_off;
			__be16 *fp;
1606 1607 1608 1609 1610 1611
			fp = skb_header_pointer(skb,
						start+offsetof(struct frag_hdr,
							       frag_off),
						sizeof(_frag_off),
						&_frag_off);
			if (fp == NULL)
1612
				return -EBADMSG;
1613

1614 1615 1616 1617
			_frag_off = ntohs(*fp) & ~0x7;
			if (_frag_off) {
				if (target < 0 &&
				    ((!ipv6_ext_hdr(hp->nexthdr)) ||
1618
				     hp->nexthdr == NEXTHDR_NONE)) {
1619 1620 1621 1622
					if (fragoff)
						*fragoff = _frag_off;
					return hp->nexthdr;
				}
1623
				return -ENOENT;
1624
			}
1625 1626
			hdrlen = 8;
		} else if (nexthdr == NEXTHDR_AUTH)
1627
			hdrlen = (hp->hdrlen + 2) << 2;
1628
		else
1629
			hdrlen = ipv6_optlen(hp);
1630 1631 1632 1633 1634 1635 1636

		nexthdr = hp->nexthdr;
		len -= hdrlen;
		start += hdrlen;
	}

	*offset = start;
1637
	return nexthdr;
1638 1639
}

L
Linus Torvalds 已提交
1640 1641 1642 1643
EXPORT_SYMBOL(ip6t_register_table);
EXPORT_SYMBOL(ip6t_unregister_table);
EXPORT_SYMBOL(ip6t_do_table);
EXPORT_SYMBOL(ip6t_ext_hdr);
1644
EXPORT_SYMBOL(ipv6_find_hdr);
L
Linus Torvalds 已提交
1645

1646 1647
module_init(ip6_tables_init);
module_exit(ip6_tables_fini);