xref: /OK3568_Linux_fs/kernel/net/bridge/netfilter/ebtables.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *  ebtables
4  *
5  *  Author:
6  *  Bart De Schuymer		<bdschuym@pandora.be>
7  *
8  *  ebtables.c,v 2.0, July, 2002
9  *
10  *  This code is strongly inspired by the iptables code which is
11  *  Copyright (C) 1999 Paul `Rusty' Russell & Michael J. Neuling
12  */
13 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
14 #include <linux/kmod.h>
15 #include <linux/module.h>
16 #include <linux/vmalloc.h>
17 #include <linux/netfilter/x_tables.h>
18 #include <linux/netfilter_bridge/ebtables.h>
19 #include <linux/spinlock.h>
20 #include <linux/mutex.h>
21 #include <linux/slab.h>
22 #include <linux/uaccess.h>
23 #include <linux/smp.h>
24 #include <linux/cpumask.h>
25 #include <linux/audit.h>
26 #include <net/sock.h>
27 /* needed for logical [in,out]-dev filtering */
28 #include "../br_private.h"
29 
30 /* Each cpu has its own set of counters, so there is no need for write_lock in
31  * the softirq
32  * For reading or updating the counters, the user context needs to
33  * get a write_lock
34  */
35 
36 /* The size of each set of counters is altered to get cache alignment */
37 #define SMP_ALIGN(x) (((x) + SMP_CACHE_BYTES-1) & ~(SMP_CACHE_BYTES-1))
38 #define COUNTER_OFFSET(n) (SMP_ALIGN(n * sizeof(struct ebt_counter)))
39 #define COUNTER_BASE(c, n, cpu) ((struct ebt_counter *)(((char *)c) + \
40 				 COUNTER_OFFSET(n) * cpu))
41 
42 
43 
44 static DEFINE_MUTEX(ebt_mutex);
45 
46 #ifdef CONFIG_COMPAT
ebt_standard_compat_from_user(void * dst,const void * src)47 static void ebt_standard_compat_from_user(void *dst, const void *src)
48 {
49 	int v = *(compat_int_t *)src;
50 
51 	if (v >= 0)
52 		v += xt_compat_calc_jump(NFPROTO_BRIDGE, v);
53 	memcpy(dst, &v, sizeof(v));
54 }
55 
ebt_standard_compat_to_user(void __user * dst,const void * src)56 static int ebt_standard_compat_to_user(void __user *dst, const void *src)
57 {
58 	compat_int_t cv = *(int *)src;
59 
60 	if (cv >= 0)
61 		cv -= xt_compat_calc_jump(NFPROTO_BRIDGE, cv);
62 	return copy_to_user(dst, &cv, sizeof(cv)) ? -EFAULT : 0;
63 }
64 #endif
65 
66 
67 static struct xt_target ebt_standard_target = {
68 	.name       = "standard",
69 	.revision   = 0,
70 	.family     = NFPROTO_BRIDGE,
71 	.targetsize = sizeof(int),
72 #ifdef CONFIG_COMPAT
73 	.compatsize = sizeof(compat_int_t),
74 	.compat_from_user = ebt_standard_compat_from_user,
75 	.compat_to_user =  ebt_standard_compat_to_user,
76 #endif
77 };
78 
79 static inline int
ebt_do_watcher(const struct ebt_entry_watcher * w,struct sk_buff * skb,struct xt_action_param * par)80 ebt_do_watcher(const struct ebt_entry_watcher *w, struct sk_buff *skb,
81 	       struct xt_action_param *par)
82 {
83 	par->target   = w->u.watcher;
84 	par->targinfo = w->data;
85 	w->u.watcher->target(skb, par);
86 	/* watchers don't give a verdict */
87 	return 0;
88 }
89 
90 static inline int
ebt_do_match(struct ebt_entry_match * m,const struct sk_buff * skb,struct xt_action_param * par)91 ebt_do_match(struct ebt_entry_match *m, const struct sk_buff *skb,
92 	     struct xt_action_param *par)
93 {
94 	par->match     = m->u.match;
95 	par->matchinfo = m->data;
96 	return !m->u.match->match(skb, par);
97 }
98 
99 static inline int
ebt_dev_check(const char * entry,const struct net_device * device)100 ebt_dev_check(const char *entry, const struct net_device *device)
101 {
102 	int i = 0;
103 	const char *devname;
104 
105 	if (*entry == '\0')
106 		return 0;
107 	if (!device)
108 		return 1;
109 	devname = device->name;
110 	/* 1 is the wildcard token */
111 	while (entry[i] != '\0' && entry[i] != 1 && entry[i] == devname[i])
112 		i++;
113 	return devname[i] != entry[i] && entry[i] != 1;
114 }
115 
116 /* process standard matches */
117 static inline int
ebt_basic_match(const struct ebt_entry * e,const struct sk_buff * skb,const struct net_device * in,const struct net_device * out)118 ebt_basic_match(const struct ebt_entry *e, const struct sk_buff *skb,
119 		const struct net_device *in, const struct net_device *out)
120 {
121 	const struct ethhdr *h = eth_hdr(skb);
122 	const struct net_bridge_port *p;
123 	__be16 ethproto;
124 
125 	if (skb_vlan_tag_present(skb))
126 		ethproto = htons(ETH_P_8021Q);
127 	else
128 		ethproto = h->h_proto;
129 
130 	if (e->bitmask & EBT_802_3) {
131 		if (NF_INVF(e, EBT_IPROTO, eth_proto_is_802_3(ethproto)))
132 			return 1;
133 	} else if (!(e->bitmask & EBT_NOPROTO) &&
134 		   NF_INVF(e, EBT_IPROTO, e->ethproto != ethproto))
135 		return 1;
136 
137 	if (NF_INVF(e, EBT_IIN, ebt_dev_check(e->in, in)))
138 		return 1;
139 	if (NF_INVF(e, EBT_IOUT, ebt_dev_check(e->out, out)))
140 		return 1;
141 	/* rcu_read_lock()ed by nf_hook_thresh */
142 	if (in && (p = br_port_get_rcu(in)) != NULL &&
143 	    NF_INVF(e, EBT_ILOGICALIN,
144 		    ebt_dev_check(e->logical_in, p->br->dev)))
145 		return 1;
146 	if (out && (p = br_port_get_rcu(out)) != NULL &&
147 	    NF_INVF(e, EBT_ILOGICALOUT,
148 		    ebt_dev_check(e->logical_out, p->br->dev)))
149 		return 1;
150 
151 	if (e->bitmask & EBT_SOURCEMAC) {
152 		if (NF_INVF(e, EBT_ISOURCE,
153 			    !ether_addr_equal_masked(h->h_source, e->sourcemac,
154 						     e->sourcemsk)))
155 			return 1;
156 	}
157 	if (e->bitmask & EBT_DESTMAC) {
158 		if (NF_INVF(e, EBT_IDEST,
159 			    !ether_addr_equal_masked(h->h_dest, e->destmac,
160 						     e->destmsk)))
161 			return 1;
162 	}
163 	return 0;
164 }
165 
166 static inline
ebt_next_entry(const struct ebt_entry * entry)167 struct ebt_entry *ebt_next_entry(const struct ebt_entry *entry)
168 {
169 	return (void *)entry + entry->next_offset;
170 }
171 
172 static inline const struct ebt_entry_target *
ebt_get_target_c(const struct ebt_entry * e)173 ebt_get_target_c(const struct ebt_entry *e)
174 {
175 	return ebt_get_target((struct ebt_entry *)e);
176 }
177 
178 /* Do some firewalling */
ebt_do_table(struct sk_buff * skb,const struct nf_hook_state * state,struct ebt_table * table)179 unsigned int ebt_do_table(struct sk_buff *skb,
180 			  const struct nf_hook_state *state,
181 			  struct ebt_table *table)
182 {
183 	unsigned int hook = state->hook;
184 	int i, nentries;
185 	struct ebt_entry *point;
186 	struct ebt_counter *counter_base, *cb_base;
187 	const struct ebt_entry_target *t;
188 	int verdict, sp = 0;
189 	struct ebt_chainstack *cs;
190 	struct ebt_entries *chaininfo;
191 	const char *base;
192 	const struct ebt_table_info *private;
193 	struct xt_action_param acpar;
194 
195 	acpar.state   = state;
196 	acpar.hotdrop = false;
197 
198 	read_lock_bh(&table->lock);
199 	private = table->private;
200 	cb_base = COUNTER_BASE(private->counters, private->nentries,
201 	   smp_processor_id());
202 	if (private->chainstack)
203 		cs = private->chainstack[smp_processor_id()];
204 	else
205 		cs = NULL;
206 	chaininfo = private->hook_entry[hook];
207 	nentries = private->hook_entry[hook]->nentries;
208 	point = (struct ebt_entry *)(private->hook_entry[hook]->data);
209 	counter_base = cb_base + private->hook_entry[hook]->counter_offset;
210 	/* base for chain jumps */
211 	base = private->entries;
212 	i = 0;
213 	while (i < nentries) {
214 		if (ebt_basic_match(point, skb, state->in, state->out))
215 			goto letscontinue;
216 
217 		if (EBT_MATCH_ITERATE(point, ebt_do_match, skb, &acpar) != 0)
218 			goto letscontinue;
219 		if (acpar.hotdrop) {
220 			read_unlock_bh(&table->lock);
221 			return NF_DROP;
222 		}
223 
224 		ADD_COUNTER(*(counter_base + i), skb->len, 1);
225 
226 		/* these should only watch: not modify, nor tell us
227 		 * what to do with the packet
228 		 */
229 		EBT_WATCHER_ITERATE(point, ebt_do_watcher, skb, &acpar);
230 
231 		t = ebt_get_target_c(point);
232 		/* standard target */
233 		if (!t->u.target->target)
234 			verdict = ((struct ebt_standard_target *)t)->verdict;
235 		else {
236 			acpar.target   = t->u.target;
237 			acpar.targinfo = t->data;
238 			verdict = t->u.target->target(skb, &acpar);
239 		}
240 		if (verdict == EBT_ACCEPT) {
241 			read_unlock_bh(&table->lock);
242 			return NF_ACCEPT;
243 		}
244 		if (verdict == EBT_DROP) {
245 			read_unlock_bh(&table->lock);
246 			return NF_DROP;
247 		}
248 		if (verdict == EBT_RETURN) {
249 letsreturn:
250 			if (WARN(sp == 0, "RETURN on base chain")) {
251 				/* act like this is EBT_CONTINUE */
252 				goto letscontinue;
253 			}
254 
255 			sp--;
256 			/* put all the local variables right */
257 			i = cs[sp].n;
258 			chaininfo = cs[sp].chaininfo;
259 			nentries = chaininfo->nentries;
260 			point = cs[sp].e;
261 			counter_base = cb_base +
262 			   chaininfo->counter_offset;
263 			continue;
264 		}
265 		if (verdict == EBT_CONTINUE)
266 			goto letscontinue;
267 
268 		if (WARN(verdict < 0, "bogus standard verdict\n")) {
269 			read_unlock_bh(&table->lock);
270 			return NF_DROP;
271 		}
272 
273 		/* jump to a udc */
274 		cs[sp].n = i + 1;
275 		cs[sp].chaininfo = chaininfo;
276 		cs[sp].e = ebt_next_entry(point);
277 		i = 0;
278 		chaininfo = (struct ebt_entries *) (base + verdict);
279 
280 		if (WARN(chaininfo->distinguisher, "jump to non-chain\n")) {
281 			read_unlock_bh(&table->lock);
282 			return NF_DROP;
283 		}
284 
285 		nentries = chaininfo->nentries;
286 		point = (struct ebt_entry *)chaininfo->data;
287 		counter_base = cb_base + chaininfo->counter_offset;
288 		sp++;
289 		continue;
290 letscontinue:
291 		point = ebt_next_entry(point);
292 		i++;
293 	}
294 
295 	/* I actually like this :) */
296 	if (chaininfo->policy == EBT_RETURN)
297 		goto letsreturn;
298 	if (chaininfo->policy == EBT_ACCEPT) {
299 		read_unlock_bh(&table->lock);
300 		return NF_ACCEPT;
301 	}
302 	read_unlock_bh(&table->lock);
303 	return NF_DROP;
304 }
305 
306 /* If it succeeds, returns element and locks mutex */
307 static inline void *
find_inlist_lock_noload(struct list_head * head,const char * name,int * error,struct mutex * mutex)308 find_inlist_lock_noload(struct list_head *head, const char *name, int *error,
309 			struct mutex *mutex)
310 {
311 	struct {
312 		struct list_head list;
313 		char name[EBT_FUNCTION_MAXNAMELEN];
314 	} *e;
315 
316 	mutex_lock(mutex);
317 	list_for_each_entry(e, head, list) {
318 		if (strcmp(e->name, name) == 0)
319 			return e;
320 	}
321 	*error = -ENOENT;
322 	mutex_unlock(mutex);
323 	return NULL;
324 }
325 
326 static void *
find_inlist_lock(struct list_head * head,const char * name,const char * prefix,int * error,struct mutex * mutex)327 find_inlist_lock(struct list_head *head, const char *name, const char *prefix,
328 		 int *error, struct mutex *mutex)
329 {
330 	return try_then_request_module(
331 			find_inlist_lock_noload(head, name, error, mutex),
332 			"%s%s", prefix, name);
333 }
334 
335 static inline struct ebt_table *
find_table_lock(struct net * net,const char * name,int * error,struct mutex * mutex)336 find_table_lock(struct net *net, const char *name, int *error,
337 		struct mutex *mutex)
338 {
339 	return find_inlist_lock(&net->xt.tables[NFPROTO_BRIDGE], name,
340 				"ebtable_", error, mutex);
341 }
342 
ebt_free_table_info(struct ebt_table_info * info)343 static inline void ebt_free_table_info(struct ebt_table_info *info)
344 {
345 	int i;
346 
347 	if (info->chainstack) {
348 		for_each_possible_cpu(i)
349 			vfree(info->chainstack[i]);
350 		vfree(info->chainstack);
351 	}
352 }
353 static inline int
ebt_check_match(struct ebt_entry_match * m,struct xt_mtchk_param * par,unsigned int * cnt)354 ebt_check_match(struct ebt_entry_match *m, struct xt_mtchk_param *par,
355 		unsigned int *cnt)
356 {
357 	const struct ebt_entry *e = par->entryinfo;
358 	struct xt_match *match;
359 	size_t left = ((char *)e + e->watchers_offset) - (char *)m;
360 	int ret;
361 
362 	if (left < sizeof(struct ebt_entry_match) ||
363 	    left - sizeof(struct ebt_entry_match) < m->match_size)
364 		return -EINVAL;
365 
366 	match = xt_find_match(NFPROTO_BRIDGE, m->u.name, m->u.revision);
367 	if (IS_ERR(match) || match->family != NFPROTO_BRIDGE) {
368 		if (!IS_ERR(match))
369 			module_put(match->me);
370 		request_module("ebt_%s", m->u.name);
371 		match = xt_find_match(NFPROTO_BRIDGE, m->u.name, m->u.revision);
372 	}
373 	if (IS_ERR(match))
374 		return PTR_ERR(match);
375 	m->u.match = match;
376 
377 	par->match     = match;
378 	par->matchinfo = m->data;
379 	ret = xt_check_match(par, m->match_size,
380 	      ntohs(e->ethproto), e->invflags & EBT_IPROTO);
381 	if (ret < 0) {
382 		module_put(match->me);
383 		return ret;
384 	}
385 
386 	(*cnt)++;
387 	return 0;
388 }
389 
390 static inline int
ebt_check_watcher(struct ebt_entry_watcher * w,struct xt_tgchk_param * par,unsigned int * cnt)391 ebt_check_watcher(struct ebt_entry_watcher *w, struct xt_tgchk_param *par,
392 		  unsigned int *cnt)
393 {
394 	const struct ebt_entry *e = par->entryinfo;
395 	struct xt_target *watcher;
396 	size_t left = ((char *)e + e->target_offset) - (char *)w;
397 	int ret;
398 
399 	if (left < sizeof(struct ebt_entry_watcher) ||
400 	   left - sizeof(struct ebt_entry_watcher) < w->watcher_size)
401 		return -EINVAL;
402 
403 	watcher = xt_request_find_target(NFPROTO_BRIDGE, w->u.name, 0);
404 	if (IS_ERR(watcher))
405 		return PTR_ERR(watcher);
406 
407 	if (watcher->family != NFPROTO_BRIDGE) {
408 		module_put(watcher->me);
409 		return -ENOENT;
410 	}
411 
412 	w->u.watcher = watcher;
413 
414 	par->target   = watcher;
415 	par->targinfo = w->data;
416 	ret = xt_check_target(par, w->watcher_size,
417 	      ntohs(e->ethproto), e->invflags & EBT_IPROTO);
418 	if (ret < 0) {
419 		module_put(watcher->me);
420 		return ret;
421 	}
422 
423 	(*cnt)++;
424 	return 0;
425 }
426 
ebt_verify_pointers(const struct ebt_replace * repl,struct ebt_table_info * newinfo)427 static int ebt_verify_pointers(const struct ebt_replace *repl,
428 			       struct ebt_table_info *newinfo)
429 {
430 	unsigned int limit = repl->entries_size;
431 	unsigned int valid_hooks = repl->valid_hooks;
432 	unsigned int offset = 0;
433 	int i;
434 
435 	for (i = 0; i < NF_BR_NUMHOOKS; i++)
436 		newinfo->hook_entry[i] = NULL;
437 
438 	newinfo->entries_size = repl->entries_size;
439 	newinfo->nentries = repl->nentries;
440 
441 	while (offset < limit) {
442 		size_t left = limit - offset;
443 		struct ebt_entry *e = (void *)newinfo->entries + offset;
444 
445 		if (left < sizeof(unsigned int))
446 			break;
447 
448 		for (i = 0; i < NF_BR_NUMHOOKS; i++) {
449 			if ((valid_hooks & (1 << i)) == 0)
450 				continue;
451 			if ((char __user *)repl->hook_entry[i] ==
452 			     repl->entries + offset)
453 				break;
454 		}
455 
456 		if (i != NF_BR_NUMHOOKS || !(e->bitmask & EBT_ENTRY_OR_ENTRIES)) {
457 			if (e->bitmask != 0) {
458 				/* we make userspace set this right,
459 				 * so there is no misunderstanding
460 				 */
461 				return -EINVAL;
462 			}
463 			if (i != NF_BR_NUMHOOKS)
464 				newinfo->hook_entry[i] = (struct ebt_entries *)e;
465 			if (left < sizeof(struct ebt_entries))
466 				break;
467 			offset += sizeof(struct ebt_entries);
468 		} else {
469 			if (left < sizeof(struct ebt_entry))
470 				break;
471 			if (left < e->next_offset)
472 				break;
473 			if (e->next_offset < sizeof(struct ebt_entry))
474 				return -EINVAL;
475 			offset += e->next_offset;
476 		}
477 	}
478 	if (offset != limit)
479 		return -EINVAL;
480 
481 	/* check if all valid hooks have a chain */
482 	for (i = 0; i < NF_BR_NUMHOOKS; i++) {
483 		if (!newinfo->hook_entry[i] &&
484 		   (valid_hooks & (1 << i)))
485 			return -EINVAL;
486 	}
487 	return 0;
488 }
489 
490 /* this one is very careful, as it is the first function
491  * to parse the userspace data
492  */
493 static inline int
ebt_check_entry_size_and_hooks(const struct ebt_entry * e,const struct ebt_table_info * newinfo,unsigned int * n,unsigned int * cnt,unsigned int * totalcnt,unsigned int * udc_cnt)494 ebt_check_entry_size_and_hooks(const struct ebt_entry *e,
495 			       const struct ebt_table_info *newinfo,
496 			       unsigned int *n, unsigned int *cnt,
497 			       unsigned int *totalcnt, unsigned int *udc_cnt)
498 {
499 	int i;
500 
501 	for (i = 0; i < NF_BR_NUMHOOKS; i++) {
502 		if ((void *)e == (void *)newinfo->hook_entry[i])
503 			break;
504 	}
505 	/* beginning of a new chain
506 	 * if i == NF_BR_NUMHOOKS it must be a user defined chain
507 	 */
508 	if (i != NF_BR_NUMHOOKS || !e->bitmask) {
509 		/* this checks if the previous chain has as many entries
510 		 * as it said it has
511 		 */
512 		if (*n != *cnt)
513 			return -EINVAL;
514 
515 		if (((struct ebt_entries *)e)->policy != EBT_DROP &&
516 		   ((struct ebt_entries *)e)->policy != EBT_ACCEPT) {
517 			/* only RETURN from udc */
518 			if (i != NF_BR_NUMHOOKS ||
519 			   ((struct ebt_entries *)e)->policy != EBT_RETURN)
520 				return -EINVAL;
521 		}
522 		if (i == NF_BR_NUMHOOKS) /* it's a user defined chain */
523 			(*udc_cnt)++;
524 		if (((struct ebt_entries *)e)->counter_offset != *totalcnt)
525 			return -EINVAL;
526 		*n = ((struct ebt_entries *)e)->nentries;
527 		*cnt = 0;
528 		return 0;
529 	}
530 	/* a plain old entry, heh */
531 	if (sizeof(struct ebt_entry) > e->watchers_offset ||
532 	   e->watchers_offset > e->target_offset ||
533 	   e->target_offset >= e->next_offset)
534 		return -EINVAL;
535 
536 	/* this is not checked anywhere else */
537 	if (e->next_offset - e->target_offset < sizeof(struct ebt_entry_target))
538 		return -EINVAL;
539 
540 	(*cnt)++;
541 	(*totalcnt)++;
542 	return 0;
543 }
544 
545 struct ebt_cl_stack {
546 	struct ebt_chainstack cs;
547 	int from;
548 	unsigned int hookmask;
549 };
550 
551 /* We need these positions to check that the jumps to a different part of the
552  * entries is a jump to the beginning of a new chain.
553  */
554 static inline int
ebt_get_udc_positions(struct ebt_entry * e,struct ebt_table_info * newinfo,unsigned int * n,struct ebt_cl_stack * udc)555 ebt_get_udc_positions(struct ebt_entry *e, struct ebt_table_info *newinfo,
556 		      unsigned int *n, struct ebt_cl_stack *udc)
557 {
558 	int i;
559 
560 	/* we're only interested in chain starts */
561 	if (e->bitmask)
562 		return 0;
563 	for (i = 0; i < NF_BR_NUMHOOKS; i++) {
564 		if (newinfo->hook_entry[i] == (struct ebt_entries *)e)
565 			break;
566 	}
567 	/* only care about udc */
568 	if (i != NF_BR_NUMHOOKS)
569 		return 0;
570 
571 	udc[*n].cs.chaininfo = (struct ebt_entries *)e;
572 	/* these initialisations are depended on later in check_chainloops() */
573 	udc[*n].cs.n = 0;
574 	udc[*n].hookmask = 0;
575 
576 	(*n)++;
577 	return 0;
578 }
579 
580 static inline int
ebt_cleanup_match(struct ebt_entry_match * m,struct net * net,unsigned int * i)581 ebt_cleanup_match(struct ebt_entry_match *m, struct net *net, unsigned int *i)
582 {
583 	struct xt_mtdtor_param par;
584 
585 	if (i && (*i)-- == 0)
586 		return 1;
587 
588 	par.net       = net;
589 	par.match     = m->u.match;
590 	par.matchinfo = m->data;
591 	par.family    = NFPROTO_BRIDGE;
592 	if (par.match->destroy != NULL)
593 		par.match->destroy(&par);
594 	module_put(par.match->me);
595 	return 0;
596 }
597 
598 static inline int
ebt_cleanup_watcher(struct ebt_entry_watcher * w,struct net * net,unsigned int * i)599 ebt_cleanup_watcher(struct ebt_entry_watcher *w, struct net *net, unsigned int *i)
600 {
601 	struct xt_tgdtor_param par;
602 
603 	if (i && (*i)-- == 0)
604 		return 1;
605 
606 	par.net      = net;
607 	par.target   = w->u.watcher;
608 	par.targinfo = w->data;
609 	par.family   = NFPROTO_BRIDGE;
610 	if (par.target->destroy != NULL)
611 		par.target->destroy(&par);
612 	module_put(par.target->me);
613 	return 0;
614 }
615 
616 static inline int
ebt_cleanup_entry(struct ebt_entry * e,struct net * net,unsigned int * cnt)617 ebt_cleanup_entry(struct ebt_entry *e, struct net *net, unsigned int *cnt)
618 {
619 	struct xt_tgdtor_param par;
620 	struct ebt_entry_target *t;
621 
622 	if (e->bitmask == 0)
623 		return 0;
624 	/* we're done */
625 	if (cnt && (*cnt)-- == 0)
626 		return 1;
627 	EBT_WATCHER_ITERATE(e, ebt_cleanup_watcher, net, NULL);
628 	EBT_MATCH_ITERATE(e, ebt_cleanup_match, net, NULL);
629 	t = ebt_get_target(e);
630 
631 	par.net      = net;
632 	par.target   = t->u.target;
633 	par.targinfo = t->data;
634 	par.family   = NFPROTO_BRIDGE;
635 	if (par.target->destroy != NULL)
636 		par.target->destroy(&par);
637 	module_put(par.target->me);
638 	return 0;
639 }
640 
641 static inline int
ebt_check_entry(struct ebt_entry * e,struct net * net,const struct ebt_table_info * newinfo,const char * name,unsigned int * cnt,struct ebt_cl_stack * cl_s,unsigned int udc_cnt)642 ebt_check_entry(struct ebt_entry *e, struct net *net,
643 		const struct ebt_table_info *newinfo,
644 		const char *name, unsigned int *cnt,
645 		struct ebt_cl_stack *cl_s, unsigned int udc_cnt)
646 {
647 	struct ebt_entry_target *t;
648 	struct xt_target *target;
649 	unsigned int i, j, hook = 0, hookmask = 0;
650 	size_t gap;
651 	int ret;
652 	struct xt_mtchk_param mtpar;
653 	struct xt_tgchk_param tgpar;
654 
655 	/* don't mess with the struct ebt_entries */
656 	if (e->bitmask == 0)
657 		return 0;
658 
659 	if (e->bitmask & ~EBT_F_MASK)
660 		return -EINVAL;
661 
662 	if (e->invflags & ~EBT_INV_MASK)
663 		return -EINVAL;
664 
665 	if ((e->bitmask & EBT_NOPROTO) && (e->bitmask & EBT_802_3))
666 		return -EINVAL;
667 
668 	/* what hook do we belong to? */
669 	for (i = 0; i < NF_BR_NUMHOOKS; i++) {
670 		if (!newinfo->hook_entry[i])
671 			continue;
672 		if ((char *)newinfo->hook_entry[i] < (char *)e)
673 			hook = i;
674 		else
675 			break;
676 	}
677 	/* (1 << NF_BR_NUMHOOKS) tells the check functions the rule is on
678 	 * a base chain
679 	 */
680 	if (i < NF_BR_NUMHOOKS)
681 		hookmask = (1 << hook) | (1 << NF_BR_NUMHOOKS);
682 	else {
683 		for (i = 0; i < udc_cnt; i++)
684 			if ((char *)(cl_s[i].cs.chaininfo) > (char *)e)
685 				break;
686 		if (i == 0)
687 			hookmask = (1 << hook) | (1 << NF_BR_NUMHOOKS);
688 		else
689 			hookmask = cl_s[i - 1].hookmask;
690 	}
691 	i = 0;
692 
693 	memset(&mtpar, 0, sizeof(mtpar));
694 	memset(&tgpar, 0, sizeof(tgpar));
695 	mtpar.net	= tgpar.net       = net;
696 	mtpar.table     = tgpar.table     = name;
697 	mtpar.entryinfo = tgpar.entryinfo = e;
698 	mtpar.hook_mask = tgpar.hook_mask = hookmask;
699 	mtpar.family    = tgpar.family    = NFPROTO_BRIDGE;
700 	ret = EBT_MATCH_ITERATE(e, ebt_check_match, &mtpar, &i);
701 	if (ret != 0)
702 		goto cleanup_matches;
703 	j = 0;
704 	ret = EBT_WATCHER_ITERATE(e, ebt_check_watcher, &tgpar, &j);
705 	if (ret != 0)
706 		goto cleanup_watchers;
707 	t = ebt_get_target(e);
708 	gap = e->next_offset - e->target_offset;
709 
710 	target = xt_request_find_target(NFPROTO_BRIDGE, t->u.name, 0);
711 	if (IS_ERR(target)) {
712 		ret = PTR_ERR(target);
713 		goto cleanup_watchers;
714 	}
715 
716 	/* Reject UNSPEC, xtables verdicts/return values are incompatible */
717 	if (target->family != NFPROTO_BRIDGE) {
718 		module_put(target->me);
719 		ret = -ENOENT;
720 		goto cleanup_watchers;
721 	}
722 
723 	t->u.target = target;
724 	if (t->u.target == &ebt_standard_target) {
725 		if (gap < sizeof(struct ebt_standard_target)) {
726 			ret = -EFAULT;
727 			goto cleanup_watchers;
728 		}
729 		if (((struct ebt_standard_target *)t)->verdict <
730 		   -NUM_STANDARD_TARGETS) {
731 			ret = -EFAULT;
732 			goto cleanup_watchers;
733 		}
734 	} else if (t->target_size > gap - sizeof(struct ebt_entry_target)) {
735 		module_put(t->u.target->me);
736 		ret = -EFAULT;
737 		goto cleanup_watchers;
738 	}
739 
740 	tgpar.target   = target;
741 	tgpar.targinfo = t->data;
742 	ret = xt_check_target(&tgpar, t->target_size,
743 	      ntohs(e->ethproto), e->invflags & EBT_IPROTO);
744 	if (ret < 0) {
745 		module_put(target->me);
746 		goto cleanup_watchers;
747 	}
748 	(*cnt)++;
749 	return 0;
750 cleanup_watchers:
751 	EBT_WATCHER_ITERATE(e, ebt_cleanup_watcher, net, &j);
752 cleanup_matches:
753 	EBT_MATCH_ITERATE(e, ebt_cleanup_match, net, &i);
754 	return ret;
755 }
756 
757 /* checks for loops and sets the hook mask for udc
758  * the hook mask for udc tells us from which base chains the udc can be
759  * accessed. This mask is a parameter to the check() functions of the extensions
760  */
check_chainloops(const struct ebt_entries * chain,struct ebt_cl_stack * cl_s,unsigned int udc_cnt,unsigned int hooknr,char * base)761 static int check_chainloops(const struct ebt_entries *chain, struct ebt_cl_stack *cl_s,
762 			    unsigned int udc_cnt, unsigned int hooknr, char *base)
763 {
764 	int i, chain_nr = -1, pos = 0, nentries = chain->nentries, verdict;
765 	const struct ebt_entry *e = (struct ebt_entry *)chain->data;
766 	const struct ebt_entry_target *t;
767 
768 	while (pos < nentries || chain_nr != -1) {
769 		/* end of udc, go back one 'recursion' step */
770 		if (pos == nentries) {
771 			/* put back values of the time when this chain was called */
772 			e = cl_s[chain_nr].cs.e;
773 			if (cl_s[chain_nr].from != -1)
774 				nentries =
775 				cl_s[cl_s[chain_nr].from].cs.chaininfo->nentries;
776 			else
777 				nentries = chain->nentries;
778 			pos = cl_s[chain_nr].cs.n;
779 			/* make sure we won't see a loop that isn't one */
780 			cl_s[chain_nr].cs.n = 0;
781 			chain_nr = cl_s[chain_nr].from;
782 			if (pos == nentries)
783 				continue;
784 		}
785 		t = ebt_get_target_c(e);
786 		if (strcmp(t->u.name, EBT_STANDARD_TARGET))
787 			goto letscontinue;
788 		if (e->target_offset + sizeof(struct ebt_standard_target) >
789 		   e->next_offset)
790 			return -1;
791 
792 		verdict = ((struct ebt_standard_target *)t)->verdict;
793 		if (verdict >= 0) { /* jump to another chain */
794 			struct ebt_entries *hlp2 =
795 			   (struct ebt_entries *)(base + verdict);
796 			for (i = 0; i < udc_cnt; i++)
797 				if (hlp2 == cl_s[i].cs.chaininfo)
798 					break;
799 			/* bad destination or loop */
800 			if (i == udc_cnt)
801 				return -1;
802 
803 			if (cl_s[i].cs.n)
804 				return -1;
805 
806 			if (cl_s[i].hookmask & (1 << hooknr))
807 				goto letscontinue;
808 			/* this can't be 0, so the loop test is correct */
809 			cl_s[i].cs.n = pos + 1;
810 			pos = 0;
811 			cl_s[i].cs.e = ebt_next_entry(e);
812 			e = (struct ebt_entry *)(hlp2->data);
813 			nentries = hlp2->nentries;
814 			cl_s[i].from = chain_nr;
815 			chain_nr = i;
816 			/* this udc is accessible from the base chain for hooknr */
817 			cl_s[i].hookmask |= (1 << hooknr);
818 			continue;
819 		}
820 letscontinue:
821 		e = ebt_next_entry(e);
822 		pos++;
823 	}
824 	return 0;
825 }
826 
827 /* do the parsing of the table/chains/entries/matches/watchers/targets, heh */
translate_table(struct net * net,const char * name,struct ebt_table_info * newinfo)828 static int translate_table(struct net *net, const char *name,
829 			   struct ebt_table_info *newinfo)
830 {
831 	unsigned int i, j, k, udc_cnt;
832 	int ret;
833 	struct ebt_cl_stack *cl_s = NULL; /* used in the checking for chain loops */
834 
835 	i = 0;
836 	while (i < NF_BR_NUMHOOKS && !newinfo->hook_entry[i])
837 		i++;
838 	if (i == NF_BR_NUMHOOKS)
839 		return -EINVAL;
840 
841 	if (newinfo->hook_entry[i] != (struct ebt_entries *)newinfo->entries)
842 		return -EINVAL;
843 
844 	/* make sure chains are ordered after each other in same order
845 	 * as their corresponding hooks
846 	 */
847 	for (j = i + 1; j < NF_BR_NUMHOOKS; j++) {
848 		if (!newinfo->hook_entry[j])
849 			continue;
850 		if (newinfo->hook_entry[j] <= newinfo->hook_entry[i])
851 			return -EINVAL;
852 
853 		i = j;
854 	}
855 
856 	/* do some early checkings and initialize some things */
857 	i = 0; /* holds the expected nr. of entries for the chain */
858 	j = 0; /* holds the up to now counted entries for the chain */
859 	k = 0; /* holds the total nr. of entries, should equal
860 		* newinfo->nentries afterwards
861 		*/
862 	udc_cnt = 0; /* will hold the nr. of user defined chains (udc) */
863 	ret = EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
864 	   ebt_check_entry_size_and_hooks, newinfo,
865 	   &i, &j, &k, &udc_cnt);
866 
867 	if (ret != 0)
868 		return ret;
869 
870 	if (i != j)
871 		return -EINVAL;
872 
873 	if (k != newinfo->nentries)
874 		return -EINVAL;
875 
876 	/* get the location of the udc, put them in an array
877 	 * while we're at it, allocate the chainstack
878 	 */
879 	if (udc_cnt) {
880 		/* this will get free'd in do_replace()/ebt_register_table()
881 		 * if an error occurs
882 		 */
883 		newinfo->chainstack =
884 			vmalloc(array_size(nr_cpu_ids,
885 					   sizeof(*(newinfo->chainstack))));
886 		if (!newinfo->chainstack)
887 			return -ENOMEM;
888 		for_each_possible_cpu(i) {
889 			newinfo->chainstack[i] =
890 			  vmalloc(array_size(udc_cnt, sizeof(*(newinfo->chainstack[0]))));
891 			if (!newinfo->chainstack[i]) {
892 				while (i)
893 					vfree(newinfo->chainstack[--i]);
894 				vfree(newinfo->chainstack);
895 				newinfo->chainstack = NULL;
896 				return -ENOMEM;
897 			}
898 		}
899 
900 		cl_s = vmalloc(array_size(udc_cnt, sizeof(*cl_s)));
901 		if (!cl_s)
902 			return -ENOMEM;
903 		i = 0; /* the i'th udc */
904 		EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
905 		   ebt_get_udc_positions, newinfo, &i, cl_s);
906 		/* sanity check */
907 		if (i != udc_cnt) {
908 			vfree(cl_s);
909 			return -EFAULT;
910 		}
911 	}
912 
913 	/* Check for loops */
914 	for (i = 0; i < NF_BR_NUMHOOKS; i++)
915 		if (newinfo->hook_entry[i])
916 			if (check_chainloops(newinfo->hook_entry[i],
917 			   cl_s, udc_cnt, i, newinfo->entries)) {
918 				vfree(cl_s);
919 				return -EINVAL;
920 			}
921 
922 	/* we now know the following (along with E=mc²):
923 	 *  - the nr of entries in each chain is right
924 	 *  - the size of the allocated space is right
925 	 *  - all valid hooks have a corresponding chain
926 	 *  - there are no loops
927 	 *  - wrong data can still be on the level of a single entry
928 	 *  - could be there are jumps to places that are not the
929 	 *    beginning of a chain. This can only occur in chains that
930 	 *    are not accessible from any base chains, so we don't care.
931 	 */
932 
933 	/* used to know what we need to clean up if something goes wrong */
934 	i = 0;
935 	ret = EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
936 	   ebt_check_entry, net, newinfo, name, &i, cl_s, udc_cnt);
937 	if (ret != 0) {
938 		EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
939 				  ebt_cleanup_entry, net, &i);
940 	}
941 	vfree(cl_s);
942 	return ret;
943 }
944 
945 /* called under write_lock */
get_counters(const struct ebt_counter * oldcounters,struct ebt_counter * counters,unsigned int nentries)946 static void get_counters(const struct ebt_counter *oldcounters,
947 			 struct ebt_counter *counters, unsigned int nentries)
948 {
949 	int i, cpu;
950 	struct ebt_counter *counter_base;
951 
952 	/* counters of cpu 0 */
953 	memcpy(counters, oldcounters,
954 	       sizeof(struct ebt_counter) * nentries);
955 
956 	/* add other counters to those of cpu 0 */
957 	for_each_possible_cpu(cpu) {
958 		if (cpu == 0)
959 			continue;
960 		counter_base = COUNTER_BASE(oldcounters, nentries, cpu);
961 		for (i = 0; i < nentries; i++)
962 			ADD_COUNTER(counters[i], counter_base[i].bcnt,
963 				    counter_base[i].pcnt);
964 	}
965 }
966 
do_replace_finish(struct net * net,struct ebt_replace * repl,struct ebt_table_info * newinfo)967 static int do_replace_finish(struct net *net, struct ebt_replace *repl,
968 			      struct ebt_table_info *newinfo)
969 {
970 	int ret;
971 	struct ebt_counter *counterstmp = NULL;
972 	/* used to be able to unlock earlier */
973 	struct ebt_table_info *table;
974 	struct ebt_table *t;
975 
976 	/* the user wants counters back
977 	 * the check on the size is done later, when we have the lock
978 	 */
979 	if (repl->num_counters) {
980 		unsigned long size = repl->num_counters * sizeof(*counterstmp);
981 		counterstmp = vmalloc(size);
982 		if (!counterstmp)
983 			return -ENOMEM;
984 	}
985 
986 	newinfo->chainstack = NULL;
987 	ret = ebt_verify_pointers(repl, newinfo);
988 	if (ret != 0)
989 		goto free_counterstmp;
990 
991 	ret = translate_table(net, repl->name, newinfo);
992 
993 	if (ret != 0)
994 		goto free_counterstmp;
995 
996 	t = find_table_lock(net, repl->name, &ret, &ebt_mutex);
997 	if (!t) {
998 		ret = -ENOENT;
999 		goto free_iterate;
1000 	}
1001 
1002 	if (repl->valid_hooks != t->valid_hooks) {
1003 		ret = -EINVAL;
1004 		goto free_unlock;
1005 	}
1006 
1007 	if (repl->num_counters && repl->num_counters != t->private->nentries) {
1008 		ret = -EINVAL;
1009 		goto free_unlock;
1010 	}
1011 
1012 	/* we have the mutex lock, so no danger in reading this pointer */
1013 	table = t->private;
1014 	/* make sure the table can only be rmmod'ed if it contains no rules */
1015 	if (!table->nentries && newinfo->nentries && !try_module_get(t->me)) {
1016 		ret = -ENOENT;
1017 		goto free_unlock;
1018 	} else if (table->nentries && !newinfo->nentries)
1019 		module_put(t->me);
1020 	/* we need an atomic snapshot of the counters */
1021 	write_lock_bh(&t->lock);
1022 	if (repl->num_counters)
1023 		get_counters(t->private->counters, counterstmp,
1024 		   t->private->nentries);
1025 
1026 	t->private = newinfo;
1027 	write_unlock_bh(&t->lock);
1028 	mutex_unlock(&ebt_mutex);
1029 	/* so, a user can change the chains while having messed up her counter
1030 	 * allocation. Only reason why this is done is because this way the lock
1031 	 * is held only once, while this doesn't bring the kernel into a
1032 	 * dangerous state.
1033 	 */
1034 	if (repl->num_counters &&
1035 	   copy_to_user(repl->counters, counterstmp,
1036 	   repl->num_counters * sizeof(struct ebt_counter))) {
1037 		/* Silent error, can't fail, new table is already in place */
1038 		net_warn_ratelimited("ebtables: counters copy to user failed while replacing table\n");
1039 	}
1040 
1041 	/* decrease module count and free resources */
1042 	EBT_ENTRY_ITERATE(table->entries, table->entries_size,
1043 			  ebt_cleanup_entry, net, NULL);
1044 
1045 	vfree(table->entries);
1046 	ebt_free_table_info(table);
1047 	vfree(table);
1048 	vfree(counterstmp);
1049 
1050 	audit_log_nfcfg(repl->name, AF_BRIDGE, repl->nentries,
1051 			AUDIT_XT_OP_REPLACE, GFP_KERNEL);
1052 	return ret;
1053 
1054 free_unlock:
1055 	mutex_unlock(&ebt_mutex);
1056 free_iterate:
1057 	EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
1058 			  ebt_cleanup_entry, net, NULL);
1059 free_counterstmp:
1060 	vfree(counterstmp);
1061 	/* can be initialized in translate_table() */
1062 	ebt_free_table_info(newinfo);
1063 	return ret;
1064 }
1065 
1066 /* replace the table */
do_replace(struct net * net,sockptr_t arg,unsigned int len)1067 static int do_replace(struct net *net, sockptr_t arg, unsigned int len)
1068 {
1069 	int ret, countersize;
1070 	struct ebt_table_info *newinfo;
1071 	struct ebt_replace tmp;
1072 
1073 	if (copy_from_sockptr(&tmp, arg, sizeof(tmp)) != 0)
1074 		return -EFAULT;
1075 
1076 	if (len != sizeof(tmp) + tmp.entries_size)
1077 		return -EINVAL;
1078 
1079 	if (tmp.entries_size == 0)
1080 		return -EINVAL;
1081 
1082 	/* overflow check */
1083 	if (tmp.nentries >= ((INT_MAX - sizeof(struct ebt_table_info)) /
1084 			NR_CPUS - SMP_CACHE_BYTES) / sizeof(struct ebt_counter))
1085 		return -ENOMEM;
1086 	if (tmp.num_counters >= INT_MAX / sizeof(struct ebt_counter))
1087 		return -ENOMEM;
1088 
1089 	tmp.name[sizeof(tmp.name) - 1] = 0;
1090 
1091 	countersize = COUNTER_OFFSET(tmp.nentries) * nr_cpu_ids;
1092 	newinfo = __vmalloc(sizeof(*newinfo) + countersize, GFP_KERNEL_ACCOUNT);
1093 	if (!newinfo)
1094 		return -ENOMEM;
1095 
1096 	if (countersize)
1097 		memset(newinfo->counters, 0, countersize);
1098 
1099 	newinfo->entries = __vmalloc(tmp.entries_size, GFP_KERNEL_ACCOUNT);
1100 	if (!newinfo->entries) {
1101 		ret = -ENOMEM;
1102 		goto free_newinfo;
1103 	}
1104 	if (copy_from_user(
1105 	   newinfo->entries, tmp.entries, tmp.entries_size) != 0) {
1106 		ret = -EFAULT;
1107 		goto free_entries;
1108 	}
1109 
1110 	ret = do_replace_finish(net, &tmp, newinfo);
1111 	if (ret == 0)
1112 		return ret;
1113 free_entries:
1114 	vfree(newinfo->entries);
1115 free_newinfo:
1116 	vfree(newinfo);
1117 	return ret;
1118 }
1119 
__ebt_unregister_table(struct net * net,struct ebt_table * table)1120 static void __ebt_unregister_table(struct net *net, struct ebt_table *table)
1121 {
1122 	mutex_lock(&ebt_mutex);
1123 	list_del(&table->list);
1124 	mutex_unlock(&ebt_mutex);
1125 	audit_log_nfcfg(table->name, AF_BRIDGE, table->private->nentries,
1126 			AUDIT_XT_OP_UNREGISTER, GFP_KERNEL);
1127 	EBT_ENTRY_ITERATE(table->private->entries, table->private->entries_size,
1128 			  ebt_cleanup_entry, net, NULL);
1129 	if (table->private->nentries)
1130 		module_put(table->me);
1131 	vfree(table->private->entries);
1132 	ebt_free_table_info(table->private);
1133 	vfree(table->private);
1134 	kfree(table);
1135 }
1136 
ebt_register_table(struct net * net,const struct ebt_table * input_table,const struct nf_hook_ops * ops,struct ebt_table ** res)1137 int ebt_register_table(struct net *net, const struct ebt_table *input_table,
1138 		       const struct nf_hook_ops *ops, struct ebt_table **res)
1139 {
1140 	struct ebt_table_info *newinfo;
1141 	struct ebt_table *t, *table;
1142 	struct ebt_replace_kernel *repl;
1143 	int ret, i, countersize;
1144 	void *p;
1145 
1146 	if (input_table == NULL || (repl = input_table->table) == NULL ||
1147 	    repl->entries == NULL || repl->entries_size == 0 ||
1148 	    repl->counters != NULL || input_table->private != NULL)
1149 		return -EINVAL;
1150 
1151 	/* Don't add one table to multiple lists. */
1152 	table = kmemdup(input_table, sizeof(struct ebt_table), GFP_KERNEL);
1153 	if (!table) {
1154 		ret = -ENOMEM;
1155 		goto out;
1156 	}
1157 
1158 	countersize = COUNTER_OFFSET(repl->nentries) * nr_cpu_ids;
1159 	newinfo = vmalloc(sizeof(*newinfo) + countersize);
1160 	ret = -ENOMEM;
1161 	if (!newinfo)
1162 		goto free_table;
1163 
1164 	p = vmalloc(repl->entries_size);
1165 	if (!p)
1166 		goto free_newinfo;
1167 
1168 	memcpy(p, repl->entries, repl->entries_size);
1169 	newinfo->entries = p;
1170 
1171 	newinfo->entries_size = repl->entries_size;
1172 	newinfo->nentries = repl->nentries;
1173 
1174 	if (countersize)
1175 		memset(newinfo->counters, 0, countersize);
1176 
1177 	/* fill in newinfo and parse the entries */
1178 	newinfo->chainstack = NULL;
1179 	for (i = 0; i < NF_BR_NUMHOOKS; i++) {
1180 		if ((repl->valid_hooks & (1 << i)) == 0)
1181 			newinfo->hook_entry[i] = NULL;
1182 		else
1183 			newinfo->hook_entry[i] = p +
1184 				((char *)repl->hook_entry[i] - repl->entries);
1185 	}
1186 	ret = translate_table(net, repl->name, newinfo);
1187 	if (ret != 0)
1188 		goto free_chainstack;
1189 
1190 	table->private = newinfo;
1191 	rwlock_init(&table->lock);
1192 	mutex_lock(&ebt_mutex);
1193 	list_for_each_entry(t, &net->xt.tables[NFPROTO_BRIDGE], list) {
1194 		if (strcmp(t->name, table->name) == 0) {
1195 			ret = -EEXIST;
1196 			goto free_unlock;
1197 		}
1198 	}
1199 
1200 	/* Hold a reference count if the chains aren't empty */
1201 	if (newinfo->nentries && !try_module_get(table->me)) {
1202 		ret = -ENOENT;
1203 		goto free_unlock;
1204 	}
1205 	list_add(&table->list, &net->xt.tables[NFPROTO_BRIDGE]);
1206 	mutex_unlock(&ebt_mutex);
1207 
1208 	WRITE_ONCE(*res, table);
1209 	ret = nf_register_net_hooks(net, ops, hweight32(table->valid_hooks));
1210 	if (ret) {
1211 		__ebt_unregister_table(net, table);
1212 		*res = NULL;
1213 	}
1214 
1215 	audit_log_nfcfg(repl->name, AF_BRIDGE, repl->nentries,
1216 			AUDIT_XT_OP_REGISTER, GFP_KERNEL);
1217 	return ret;
1218 free_unlock:
1219 	mutex_unlock(&ebt_mutex);
1220 free_chainstack:
1221 	ebt_free_table_info(newinfo);
1222 	vfree(newinfo->entries);
1223 free_newinfo:
1224 	vfree(newinfo);
1225 free_table:
1226 	kfree(table);
1227 out:
1228 	return ret;
1229 }
1230 
__ebt_find_table(struct net * net,const char * name)1231 static struct ebt_table *__ebt_find_table(struct net *net, const char *name)
1232 {
1233 	struct ebt_table *t;
1234 
1235 	mutex_lock(&ebt_mutex);
1236 
1237 	list_for_each_entry(t, &net->xt.tables[NFPROTO_BRIDGE], list) {
1238 		if (strcmp(t->name, name) == 0) {
1239 			mutex_unlock(&ebt_mutex);
1240 			return t;
1241 		}
1242 	}
1243 
1244 	mutex_unlock(&ebt_mutex);
1245 	return NULL;
1246 }
1247 
ebt_unregister_table_pre_exit(struct net * net,const char * name,const struct nf_hook_ops * ops)1248 void ebt_unregister_table_pre_exit(struct net *net, const char *name, const struct nf_hook_ops *ops)
1249 {
1250 	struct ebt_table *table = __ebt_find_table(net, name);
1251 
1252 	if (table)
1253 		nf_unregister_net_hooks(net, ops, hweight32(table->valid_hooks));
1254 }
1255 EXPORT_SYMBOL(ebt_unregister_table_pre_exit);
1256 
ebt_unregister_table(struct net * net,struct ebt_table * table)1257 void ebt_unregister_table(struct net *net, struct ebt_table *table)
1258 {
1259 	__ebt_unregister_table(net, table);
1260 }
1261 
1262 /* userspace just supplied us with counters */
do_update_counters(struct net * net,const char * name,struct ebt_counter __user * counters,unsigned int num_counters,unsigned int len)1263 static int do_update_counters(struct net *net, const char *name,
1264 			      struct ebt_counter __user *counters,
1265 			      unsigned int num_counters, unsigned int len)
1266 {
1267 	int i, ret;
1268 	struct ebt_counter *tmp;
1269 	struct ebt_table *t;
1270 
1271 	if (num_counters == 0)
1272 		return -EINVAL;
1273 
1274 	tmp = vmalloc(array_size(num_counters, sizeof(*tmp)));
1275 	if (!tmp)
1276 		return -ENOMEM;
1277 
1278 	t = find_table_lock(net, name, &ret, &ebt_mutex);
1279 	if (!t)
1280 		goto free_tmp;
1281 
1282 	if (num_counters != t->private->nentries) {
1283 		ret = -EINVAL;
1284 		goto unlock_mutex;
1285 	}
1286 
1287 	if (copy_from_user(tmp, counters, num_counters * sizeof(*counters))) {
1288 		ret = -EFAULT;
1289 		goto unlock_mutex;
1290 	}
1291 
1292 	/* we want an atomic add of the counters */
1293 	write_lock_bh(&t->lock);
1294 
1295 	/* we add to the counters of the first cpu */
1296 	for (i = 0; i < num_counters; i++)
1297 		ADD_COUNTER(t->private->counters[i], tmp[i].bcnt, tmp[i].pcnt);
1298 
1299 	write_unlock_bh(&t->lock);
1300 	ret = 0;
1301 unlock_mutex:
1302 	mutex_unlock(&ebt_mutex);
1303 free_tmp:
1304 	vfree(tmp);
1305 	return ret;
1306 }
1307 
update_counters(struct net * net,sockptr_t arg,unsigned int len)1308 static int update_counters(struct net *net, sockptr_t arg, unsigned int len)
1309 {
1310 	struct ebt_replace hlp;
1311 
1312 	if (copy_from_sockptr(&hlp, arg, sizeof(hlp)))
1313 		return -EFAULT;
1314 
1315 	if (len != sizeof(hlp) + hlp.num_counters * sizeof(struct ebt_counter))
1316 		return -EINVAL;
1317 
1318 	return do_update_counters(net, hlp.name, hlp.counters,
1319 				  hlp.num_counters, len);
1320 }
1321 
ebt_obj_to_user(char __user * um,const char * _name,const char * data,int entrysize,int usersize,int datasize,u8 revision)1322 static inline int ebt_obj_to_user(char __user *um, const char *_name,
1323 				  const char *data, int entrysize,
1324 				  int usersize, int datasize, u8 revision)
1325 {
1326 	char name[EBT_EXTENSION_MAXNAMELEN] = {0};
1327 
1328 	/* ebtables expects 31 bytes long names but xt_match names are 29 bytes
1329 	 * long. Copy 29 bytes and fill remaining bytes with zeroes.
1330 	 */
1331 	strlcpy(name, _name, sizeof(name));
1332 	if (copy_to_user(um, name, EBT_EXTENSION_MAXNAMELEN) ||
1333 	    put_user(revision, (u8 __user *)(um + EBT_EXTENSION_MAXNAMELEN)) ||
1334 	    put_user(datasize, (int __user *)(um + EBT_EXTENSION_MAXNAMELEN + 1)) ||
1335 	    xt_data_to_user(um + entrysize, data, usersize, datasize,
1336 			    XT_ALIGN(datasize)))
1337 		return -EFAULT;
1338 
1339 	return 0;
1340 }
1341 
ebt_match_to_user(const struct ebt_entry_match * m,const char * base,char __user * ubase)1342 static inline int ebt_match_to_user(const struct ebt_entry_match *m,
1343 				    const char *base, char __user *ubase)
1344 {
1345 	return ebt_obj_to_user(ubase + ((char *)m - base),
1346 			       m->u.match->name, m->data, sizeof(*m),
1347 			       m->u.match->usersize, m->match_size,
1348 			       m->u.match->revision);
1349 }
1350 
ebt_watcher_to_user(const struct ebt_entry_watcher * w,const char * base,char __user * ubase)1351 static inline int ebt_watcher_to_user(const struct ebt_entry_watcher *w,
1352 				      const char *base, char __user *ubase)
1353 {
1354 	return ebt_obj_to_user(ubase + ((char *)w - base),
1355 			       w->u.watcher->name, w->data, sizeof(*w),
1356 			       w->u.watcher->usersize, w->watcher_size,
1357 			       w->u.watcher->revision);
1358 }
1359 
ebt_entry_to_user(struct ebt_entry * e,const char * base,char __user * ubase)1360 static inline int ebt_entry_to_user(struct ebt_entry *e, const char *base,
1361 				    char __user *ubase)
1362 {
1363 	int ret;
1364 	char __user *hlp;
1365 	const struct ebt_entry_target *t;
1366 
1367 	if (e->bitmask == 0) {
1368 		/* special case !EBT_ENTRY_OR_ENTRIES */
1369 		if (copy_to_user(ubase + ((char *)e - base), e,
1370 				 sizeof(struct ebt_entries)))
1371 			return -EFAULT;
1372 		return 0;
1373 	}
1374 
1375 	if (copy_to_user(ubase + ((char *)e - base), e, sizeof(*e)))
1376 		return -EFAULT;
1377 
1378 	hlp = ubase + (((char *)e + e->target_offset) - base);
1379 	t = ebt_get_target_c(e);
1380 
1381 	ret = EBT_MATCH_ITERATE(e, ebt_match_to_user, base, ubase);
1382 	if (ret != 0)
1383 		return ret;
1384 	ret = EBT_WATCHER_ITERATE(e, ebt_watcher_to_user, base, ubase);
1385 	if (ret != 0)
1386 		return ret;
1387 	ret = ebt_obj_to_user(hlp, t->u.target->name, t->data, sizeof(*t),
1388 			      t->u.target->usersize, t->target_size,
1389 			      t->u.target->revision);
1390 	if (ret != 0)
1391 		return ret;
1392 
1393 	return 0;
1394 }
1395 
copy_counters_to_user(struct ebt_table * t,const struct ebt_counter * oldcounters,void __user * user,unsigned int num_counters,unsigned int nentries)1396 static int copy_counters_to_user(struct ebt_table *t,
1397 				 const struct ebt_counter *oldcounters,
1398 				 void __user *user, unsigned int num_counters,
1399 				 unsigned int nentries)
1400 {
1401 	struct ebt_counter *counterstmp;
1402 	int ret = 0;
1403 
1404 	/* userspace might not need the counters */
1405 	if (num_counters == 0)
1406 		return 0;
1407 
1408 	if (num_counters != nentries)
1409 		return -EINVAL;
1410 
1411 	counterstmp = vmalloc(array_size(nentries, sizeof(*counterstmp)));
1412 	if (!counterstmp)
1413 		return -ENOMEM;
1414 
1415 	write_lock_bh(&t->lock);
1416 	get_counters(oldcounters, counterstmp, nentries);
1417 	write_unlock_bh(&t->lock);
1418 
1419 	if (copy_to_user(user, counterstmp,
1420 	   nentries * sizeof(struct ebt_counter)))
1421 		ret = -EFAULT;
1422 	vfree(counterstmp);
1423 	return ret;
1424 }
1425 
1426 /* called with ebt_mutex locked */
copy_everything_to_user(struct ebt_table * t,void __user * user,const int * len,int cmd)1427 static int copy_everything_to_user(struct ebt_table *t, void __user *user,
1428 				   const int *len, int cmd)
1429 {
1430 	struct ebt_replace tmp;
1431 	const struct ebt_counter *oldcounters;
1432 	unsigned int entries_size, nentries;
1433 	int ret;
1434 	char *entries;
1435 
1436 	if (cmd == EBT_SO_GET_ENTRIES) {
1437 		entries_size = t->private->entries_size;
1438 		nentries = t->private->nentries;
1439 		entries = t->private->entries;
1440 		oldcounters = t->private->counters;
1441 	} else {
1442 		entries_size = t->table->entries_size;
1443 		nentries = t->table->nentries;
1444 		entries = t->table->entries;
1445 		oldcounters = t->table->counters;
1446 	}
1447 
1448 	if (copy_from_user(&tmp, user, sizeof(tmp)))
1449 		return -EFAULT;
1450 
1451 	if (*len != sizeof(struct ebt_replace) + entries_size +
1452 	   (tmp.num_counters ? nentries * sizeof(struct ebt_counter) : 0))
1453 		return -EINVAL;
1454 
1455 	if (tmp.nentries != nentries)
1456 		return -EINVAL;
1457 
1458 	if (tmp.entries_size != entries_size)
1459 		return -EINVAL;
1460 
1461 	ret = copy_counters_to_user(t, oldcounters, tmp.counters,
1462 					tmp.num_counters, nentries);
1463 	if (ret)
1464 		return ret;
1465 
1466 	/* set the match/watcher/target names right */
1467 	return EBT_ENTRY_ITERATE(entries, entries_size,
1468 	   ebt_entry_to_user, entries, tmp.entries);
1469 }
1470 
1471 #ifdef CONFIG_COMPAT
1472 /* 32 bit-userspace compatibility definitions. */
1473 struct compat_ebt_replace {
1474 	char name[EBT_TABLE_MAXNAMELEN];
1475 	compat_uint_t valid_hooks;
1476 	compat_uint_t nentries;
1477 	compat_uint_t entries_size;
1478 	/* start of the chains */
1479 	compat_uptr_t hook_entry[NF_BR_NUMHOOKS];
1480 	/* nr of counters userspace expects back */
1481 	compat_uint_t num_counters;
1482 	/* where the kernel will put the old counters. */
1483 	compat_uptr_t counters;
1484 	compat_uptr_t entries;
1485 };
1486 
1487 /* struct ebt_entry_match, _target and _watcher have same layout */
1488 struct compat_ebt_entry_mwt {
1489 	union {
1490 		struct {
1491 			char name[EBT_EXTENSION_MAXNAMELEN];
1492 			u8 revision;
1493 		};
1494 		compat_uptr_t ptr;
1495 	} u;
1496 	compat_uint_t match_size;
1497 	compat_uint_t data[] __aligned(__alignof__(struct compat_ebt_replace));
1498 };
1499 
1500 /* account for possible padding between match_size and ->data */
ebt_compat_entry_padsize(void)1501 static int ebt_compat_entry_padsize(void)
1502 {
1503 	BUILD_BUG_ON(sizeof(struct ebt_entry_match) <
1504 			sizeof(struct compat_ebt_entry_mwt));
1505 	return (int) sizeof(struct ebt_entry_match) -
1506 			sizeof(struct compat_ebt_entry_mwt);
1507 }
1508 
ebt_compat_match_offset(const struct xt_match * match,unsigned int userlen)1509 static int ebt_compat_match_offset(const struct xt_match *match,
1510 				   unsigned int userlen)
1511 {
1512 	/* ebt_among needs special handling. The kernel .matchsize is
1513 	 * set to -1 at registration time; at runtime an EBT_ALIGN()ed
1514 	 * value is expected.
1515 	 * Example: userspace sends 4500, ebt_among.c wants 4504.
1516 	 */
1517 	if (unlikely(match->matchsize == -1))
1518 		return XT_ALIGN(userlen) - COMPAT_XT_ALIGN(userlen);
1519 	return xt_compat_match_offset(match);
1520 }
1521 
compat_match_to_user(struct ebt_entry_match * m,void __user ** dstptr,unsigned int * size)1522 static int compat_match_to_user(struct ebt_entry_match *m, void __user **dstptr,
1523 				unsigned int *size)
1524 {
1525 	const struct xt_match *match = m->u.match;
1526 	struct compat_ebt_entry_mwt __user *cm = *dstptr;
1527 	int off = ebt_compat_match_offset(match, m->match_size);
1528 	compat_uint_t msize = m->match_size - off;
1529 
1530 	if (WARN_ON(off >= m->match_size))
1531 		return -EINVAL;
1532 
1533 	if (copy_to_user(cm->u.name, match->name, strlen(match->name) + 1) ||
1534 	    put_user(match->revision, &cm->u.revision) ||
1535 	    put_user(msize, &cm->match_size))
1536 		return -EFAULT;
1537 
1538 	if (match->compat_to_user) {
1539 		if (match->compat_to_user(cm->data, m->data))
1540 			return -EFAULT;
1541 	} else {
1542 		if (xt_data_to_user(cm->data, m->data, match->usersize, msize,
1543 				    COMPAT_XT_ALIGN(msize)))
1544 			return -EFAULT;
1545 	}
1546 
1547 	*size -= ebt_compat_entry_padsize() + off;
1548 	*dstptr = cm->data;
1549 	*dstptr += msize;
1550 	return 0;
1551 }
1552 
compat_target_to_user(struct ebt_entry_target * t,void __user ** dstptr,unsigned int * size)1553 static int compat_target_to_user(struct ebt_entry_target *t,
1554 				 void __user **dstptr,
1555 				 unsigned int *size)
1556 {
1557 	const struct xt_target *target = t->u.target;
1558 	struct compat_ebt_entry_mwt __user *cm = *dstptr;
1559 	int off = xt_compat_target_offset(target);
1560 	compat_uint_t tsize = t->target_size - off;
1561 
1562 	if (WARN_ON(off >= t->target_size))
1563 		return -EINVAL;
1564 
1565 	if (copy_to_user(cm->u.name, target->name, strlen(target->name) + 1) ||
1566 	    put_user(target->revision, &cm->u.revision) ||
1567 	    put_user(tsize, &cm->match_size))
1568 		return -EFAULT;
1569 
1570 	if (target->compat_to_user) {
1571 		if (target->compat_to_user(cm->data, t->data))
1572 			return -EFAULT;
1573 	} else {
1574 		if (xt_data_to_user(cm->data, t->data, target->usersize, tsize,
1575 				    COMPAT_XT_ALIGN(tsize)))
1576 			return -EFAULT;
1577 	}
1578 
1579 	*size -= ebt_compat_entry_padsize() + off;
1580 	*dstptr = cm->data;
1581 	*dstptr += tsize;
1582 	return 0;
1583 }
1584 
compat_watcher_to_user(struct ebt_entry_watcher * w,void __user ** dstptr,unsigned int * size)1585 static int compat_watcher_to_user(struct ebt_entry_watcher *w,
1586 				  void __user **dstptr,
1587 				  unsigned int *size)
1588 {
1589 	return compat_target_to_user((struct ebt_entry_target *)w,
1590 							dstptr, size);
1591 }
1592 
compat_copy_entry_to_user(struct ebt_entry * e,void __user ** dstptr,unsigned int * size)1593 static int compat_copy_entry_to_user(struct ebt_entry *e, void __user **dstptr,
1594 				unsigned int *size)
1595 {
1596 	struct ebt_entry_target *t;
1597 	struct ebt_entry __user *ce;
1598 	u32 watchers_offset, target_offset, next_offset;
1599 	compat_uint_t origsize;
1600 	int ret;
1601 
1602 	if (e->bitmask == 0) {
1603 		if (*size < sizeof(struct ebt_entries))
1604 			return -EINVAL;
1605 		if (copy_to_user(*dstptr, e, sizeof(struct ebt_entries)))
1606 			return -EFAULT;
1607 
1608 		*dstptr += sizeof(struct ebt_entries);
1609 		*size -= sizeof(struct ebt_entries);
1610 		return 0;
1611 	}
1612 
1613 	if (*size < sizeof(*ce))
1614 		return -EINVAL;
1615 
1616 	ce = *dstptr;
1617 	if (copy_to_user(ce, e, sizeof(*ce)))
1618 		return -EFAULT;
1619 
1620 	origsize = *size;
1621 	*dstptr += sizeof(*ce);
1622 
1623 	ret = EBT_MATCH_ITERATE(e, compat_match_to_user, dstptr, size);
1624 	if (ret)
1625 		return ret;
1626 	watchers_offset = e->watchers_offset - (origsize - *size);
1627 
1628 	ret = EBT_WATCHER_ITERATE(e, compat_watcher_to_user, dstptr, size);
1629 	if (ret)
1630 		return ret;
1631 	target_offset = e->target_offset - (origsize - *size);
1632 
1633 	t = ebt_get_target(e);
1634 
1635 	ret = compat_target_to_user(t, dstptr, size);
1636 	if (ret)
1637 		return ret;
1638 	next_offset = e->next_offset - (origsize - *size);
1639 
1640 	if (put_user(watchers_offset, &ce->watchers_offset) ||
1641 	    put_user(target_offset, &ce->target_offset) ||
1642 	    put_user(next_offset, &ce->next_offset))
1643 		return -EFAULT;
1644 
1645 	*size -= sizeof(*ce);
1646 	return 0;
1647 }
1648 
compat_calc_match(struct ebt_entry_match * m,int * off)1649 static int compat_calc_match(struct ebt_entry_match *m, int *off)
1650 {
1651 	*off += ebt_compat_match_offset(m->u.match, m->match_size);
1652 	*off += ebt_compat_entry_padsize();
1653 	return 0;
1654 }
1655 
compat_calc_watcher(struct ebt_entry_watcher * w,int * off)1656 static int compat_calc_watcher(struct ebt_entry_watcher *w, int *off)
1657 {
1658 	*off += xt_compat_target_offset(w->u.watcher);
1659 	*off += ebt_compat_entry_padsize();
1660 	return 0;
1661 }
1662 
compat_calc_entry(const struct ebt_entry * e,const struct ebt_table_info * info,const void * base,struct compat_ebt_replace * newinfo)1663 static int compat_calc_entry(const struct ebt_entry *e,
1664 			     const struct ebt_table_info *info,
1665 			     const void *base,
1666 			     struct compat_ebt_replace *newinfo)
1667 {
1668 	const struct ebt_entry_target *t;
1669 	unsigned int entry_offset;
1670 	int off, ret, i;
1671 
1672 	if (e->bitmask == 0)
1673 		return 0;
1674 
1675 	off = 0;
1676 	entry_offset = (void *)e - base;
1677 
1678 	EBT_MATCH_ITERATE(e, compat_calc_match, &off);
1679 	EBT_WATCHER_ITERATE(e, compat_calc_watcher, &off);
1680 
1681 	t = ebt_get_target_c(e);
1682 
1683 	off += xt_compat_target_offset(t->u.target);
1684 	off += ebt_compat_entry_padsize();
1685 
1686 	newinfo->entries_size -= off;
1687 
1688 	ret = xt_compat_add_offset(NFPROTO_BRIDGE, entry_offset, off);
1689 	if (ret)
1690 		return ret;
1691 
1692 	for (i = 0; i < NF_BR_NUMHOOKS; i++) {
1693 		const void *hookptr = info->hook_entry[i];
1694 		if (info->hook_entry[i] &&
1695 		    (e < (struct ebt_entry *)(base - hookptr))) {
1696 			newinfo->hook_entry[i] -= off;
1697 			pr_debug("0x%08X -> 0x%08X\n",
1698 					newinfo->hook_entry[i] + off,
1699 					newinfo->hook_entry[i]);
1700 		}
1701 	}
1702 
1703 	return 0;
1704 }
1705 
ebt_compat_init_offsets(unsigned int number)1706 static int ebt_compat_init_offsets(unsigned int number)
1707 {
1708 	if (number > INT_MAX)
1709 		return -EINVAL;
1710 
1711 	/* also count the base chain policies */
1712 	number += NF_BR_NUMHOOKS;
1713 
1714 	return xt_compat_init_offsets(NFPROTO_BRIDGE, number);
1715 }
1716 
compat_table_info(const struct ebt_table_info * info,struct compat_ebt_replace * newinfo)1717 static int compat_table_info(const struct ebt_table_info *info,
1718 			     struct compat_ebt_replace *newinfo)
1719 {
1720 	unsigned int size = info->entries_size;
1721 	const void *entries = info->entries;
1722 	int ret;
1723 
1724 	newinfo->entries_size = size;
1725 	ret = ebt_compat_init_offsets(info->nentries);
1726 	if (ret)
1727 		return ret;
1728 
1729 	return EBT_ENTRY_ITERATE(entries, size, compat_calc_entry, info,
1730 							entries, newinfo);
1731 }
1732 
compat_copy_everything_to_user(struct ebt_table * t,void __user * user,int * len,int cmd)1733 static int compat_copy_everything_to_user(struct ebt_table *t,
1734 					  void __user *user, int *len, int cmd)
1735 {
1736 	struct compat_ebt_replace repl, tmp;
1737 	struct ebt_counter *oldcounters;
1738 	struct ebt_table_info tinfo;
1739 	int ret;
1740 	void __user *pos;
1741 
1742 	memset(&tinfo, 0, sizeof(tinfo));
1743 
1744 	if (cmd == EBT_SO_GET_ENTRIES) {
1745 		tinfo.entries_size = t->private->entries_size;
1746 		tinfo.nentries = t->private->nentries;
1747 		tinfo.entries = t->private->entries;
1748 		oldcounters = t->private->counters;
1749 	} else {
1750 		tinfo.entries_size = t->table->entries_size;
1751 		tinfo.nentries = t->table->nentries;
1752 		tinfo.entries = t->table->entries;
1753 		oldcounters = t->table->counters;
1754 	}
1755 
1756 	if (copy_from_user(&tmp, user, sizeof(tmp)))
1757 		return -EFAULT;
1758 
1759 	if (tmp.nentries != tinfo.nentries ||
1760 	   (tmp.num_counters && tmp.num_counters != tinfo.nentries))
1761 		return -EINVAL;
1762 
1763 	memcpy(&repl, &tmp, sizeof(repl));
1764 	if (cmd == EBT_SO_GET_ENTRIES)
1765 		ret = compat_table_info(t->private, &repl);
1766 	else
1767 		ret = compat_table_info(&tinfo, &repl);
1768 	if (ret)
1769 		return ret;
1770 
1771 	if (*len != sizeof(tmp) + repl.entries_size +
1772 	   (tmp.num_counters? tinfo.nentries * sizeof(struct ebt_counter): 0)) {
1773 		pr_err("wrong size: *len %d, entries_size %u, replsz %d\n",
1774 				*len, tinfo.entries_size, repl.entries_size);
1775 		return -EINVAL;
1776 	}
1777 
1778 	/* userspace might not need the counters */
1779 	ret = copy_counters_to_user(t, oldcounters, compat_ptr(tmp.counters),
1780 					tmp.num_counters, tinfo.nentries);
1781 	if (ret)
1782 		return ret;
1783 
1784 	pos = compat_ptr(tmp.entries);
1785 	return EBT_ENTRY_ITERATE(tinfo.entries, tinfo.entries_size,
1786 			compat_copy_entry_to_user, &pos, &tmp.entries_size);
1787 }
1788 
1789 struct ebt_entries_buf_state {
1790 	char *buf_kern_start;	/* kernel buffer to copy (translated) data to */
1791 	u32 buf_kern_len;	/* total size of kernel buffer */
1792 	u32 buf_kern_offset;	/* amount of data copied so far */
1793 	u32 buf_user_offset;	/* read position in userspace buffer */
1794 };
1795 
ebt_buf_count(struct ebt_entries_buf_state * state,unsigned int sz)1796 static int ebt_buf_count(struct ebt_entries_buf_state *state, unsigned int sz)
1797 {
1798 	state->buf_kern_offset += sz;
1799 	return state->buf_kern_offset >= sz ? 0 : -EINVAL;
1800 }
1801 
ebt_buf_add(struct ebt_entries_buf_state * state,const void * data,unsigned int sz)1802 static int ebt_buf_add(struct ebt_entries_buf_state *state,
1803 		       const void *data, unsigned int sz)
1804 {
1805 	if (state->buf_kern_start == NULL)
1806 		goto count_only;
1807 
1808 	if (WARN_ON(state->buf_kern_offset + sz > state->buf_kern_len))
1809 		return -EINVAL;
1810 
1811 	memcpy(state->buf_kern_start + state->buf_kern_offset, data, sz);
1812 
1813  count_only:
1814 	state->buf_user_offset += sz;
1815 	return ebt_buf_count(state, sz);
1816 }
1817 
ebt_buf_add_pad(struct ebt_entries_buf_state * state,unsigned int sz)1818 static int ebt_buf_add_pad(struct ebt_entries_buf_state *state, unsigned int sz)
1819 {
1820 	char *b = state->buf_kern_start;
1821 
1822 	if (WARN_ON(b && state->buf_kern_offset > state->buf_kern_len))
1823 		return -EINVAL;
1824 
1825 	if (b != NULL && sz > 0)
1826 		memset(b + state->buf_kern_offset, 0, sz);
1827 	/* do not adjust ->buf_user_offset here, we added kernel-side padding */
1828 	return ebt_buf_count(state, sz);
1829 }
1830 
1831 enum compat_mwt {
1832 	EBT_COMPAT_MATCH,
1833 	EBT_COMPAT_WATCHER,
1834 	EBT_COMPAT_TARGET,
1835 };
1836 
compat_mtw_from_user(const struct compat_ebt_entry_mwt * mwt,enum compat_mwt compat_mwt,struct ebt_entries_buf_state * state,const unsigned char * base)1837 static int compat_mtw_from_user(const struct compat_ebt_entry_mwt *mwt,
1838 				enum compat_mwt compat_mwt,
1839 				struct ebt_entries_buf_state *state,
1840 				const unsigned char *base)
1841 {
1842 	char name[EBT_EXTENSION_MAXNAMELEN];
1843 	struct xt_match *match;
1844 	struct xt_target *wt;
1845 	void *dst = NULL;
1846 	int off, pad = 0;
1847 	unsigned int size_kern, match_size = mwt->match_size;
1848 
1849 	if (strscpy(name, mwt->u.name, sizeof(name)) < 0)
1850 		return -EINVAL;
1851 
1852 	if (state->buf_kern_start)
1853 		dst = state->buf_kern_start + state->buf_kern_offset;
1854 
1855 	switch (compat_mwt) {
1856 	case EBT_COMPAT_MATCH:
1857 		match = xt_request_find_match(NFPROTO_BRIDGE, name,
1858 					      mwt->u.revision);
1859 		if (IS_ERR(match))
1860 			return PTR_ERR(match);
1861 
1862 		off = ebt_compat_match_offset(match, match_size);
1863 		if (dst) {
1864 			if (match->compat_from_user)
1865 				match->compat_from_user(dst, mwt->data);
1866 			else
1867 				memcpy(dst, mwt->data, match_size);
1868 		}
1869 
1870 		size_kern = match->matchsize;
1871 		if (unlikely(size_kern == -1))
1872 			size_kern = match_size;
1873 		module_put(match->me);
1874 		break;
1875 	case EBT_COMPAT_WATCHER:
1876 	case EBT_COMPAT_TARGET:
1877 		wt = xt_request_find_target(NFPROTO_BRIDGE, name,
1878 					    mwt->u.revision);
1879 		if (IS_ERR(wt))
1880 			return PTR_ERR(wt);
1881 		off = xt_compat_target_offset(wt);
1882 
1883 		if (dst) {
1884 			if (wt->compat_from_user)
1885 				wt->compat_from_user(dst, mwt->data);
1886 			else
1887 				memcpy(dst, mwt->data, match_size);
1888 		}
1889 
1890 		size_kern = wt->targetsize;
1891 		module_put(wt->me);
1892 		break;
1893 
1894 	default:
1895 		return -EINVAL;
1896 	}
1897 
1898 	state->buf_kern_offset += match_size + off;
1899 	state->buf_user_offset += match_size;
1900 	pad = XT_ALIGN(size_kern) - size_kern;
1901 
1902 	if (pad > 0 && dst) {
1903 		if (WARN_ON(state->buf_kern_len <= pad))
1904 			return -EINVAL;
1905 		if (WARN_ON(state->buf_kern_offset - (match_size + off) + size_kern > state->buf_kern_len - pad))
1906 			return -EINVAL;
1907 		memset(dst + size_kern, 0, pad);
1908 	}
1909 	return off + match_size;
1910 }
1911 
1912 /* return size of all matches, watchers or target, including necessary
1913  * alignment and padding.
1914  */
ebt_size_mwt(const struct compat_ebt_entry_mwt * match32,unsigned int size_left,enum compat_mwt type,struct ebt_entries_buf_state * state,const void * base)1915 static int ebt_size_mwt(const struct compat_ebt_entry_mwt *match32,
1916 			unsigned int size_left, enum compat_mwt type,
1917 			struct ebt_entries_buf_state *state, const void *base)
1918 {
1919 	const char *buf = (const char *)match32;
1920 	int growth = 0;
1921 
1922 	if (size_left == 0)
1923 		return 0;
1924 
1925 	do {
1926 		struct ebt_entry_match *match_kern;
1927 		int ret;
1928 
1929 		if (size_left < sizeof(*match32))
1930 			return -EINVAL;
1931 
1932 		match_kern = (struct ebt_entry_match *) state->buf_kern_start;
1933 		if (match_kern) {
1934 			char *tmp;
1935 			tmp = state->buf_kern_start + state->buf_kern_offset;
1936 			match_kern = (struct ebt_entry_match *) tmp;
1937 		}
1938 		ret = ebt_buf_add(state, buf, sizeof(*match32));
1939 		if (ret < 0)
1940 			return ret;
1941 		size_left -= sizeof(*match32);
1942 
1943 		/* add padding before match->data (if any) */
1944 		ret = ebt_buf_add_pad(state, ebt_compat_entry_padsize());
1945 		if (ret < 0)
1946 			return ret;
1947 
1948 		if (match32->match_size > size_left)
1949 			return -EINVAL;
1950 
1951 		size_left -= match32->match_size;
1952 
1953 		ret = compat_mtw_from_user(match32, type, state, base);
1954 		if (ret < 0)
1955 			return ret;
1956 
1957 		if (WARN_ON(ret < match32->match_size))
1958 			return -EINVAL;
1959 		growth += ret - match32->match_size;
1960 		growth += ebt_compat_entry_padsize();
1961 
1962 		buf += sizeof(*match32);
1963 		buf += match32->match_size;
1964 
1965 		if (match_kern)
1966 			match_kern->match_size = ret;
1967 
1968 		match32 = (struct compat_ebt_entry_mwt *) buf;
1969 	} while (size_left);
1970 
1971 	return growth;
1972 }
1973 
1974 /* called for all ebt_entry structures. */
size_entry_mwt(const struct ebt_entry * entry,const unsigned char * base,unsigned int * total,struct ebt_entries_buf_state * state)1975 static int size_entry_mwt(const struct ebt_entry *entry, const unsigned char *base,
1976 			  unsigned int *total,
1977 			  struct ebt_entries_buf_state *state)
1978 {
1979 	unsigned int i, j, startoff, next_expected_off, new_offset = 0;
1980 	/* stores match/watchers/targets & offset of next struct ebt_entry: */
1981 	unsigned int offsets[4];
1982 	unsigned int *offsets_update = NULL;
1983 	int ret;
1984 	char *buf_start;
1985 
1986 	if (*total < sizeof(struct ebt_entries))
1987 		return -EINVAL;
1988 
1989 	if (!entry->bitmask) {
1990 		*total -= sizeof(struct ebt_entries);
1991 		return ebt_buf_add(state, entry, sizeof(struct ebt_entries));
1992 	}
1993 	if (*total < sizeof(*entry) || entry->next_offset < sizeof(*entry))
1994 		return -EINVAL;
1995 
1996 	startoff = state->buf_user_offset;
1997 	/* pull in most part of ebt_entry, it does not need to be changed. */
1998 	ret = ebt_buf_add(state, entry,
1999 			offsetof(struct ebt_entry, watchers_offset));
2000 	if (ret < 0)
2001 		return ret;
2002 
2003 	offsets[0] = sizeof(struct ebt_entry); /* matches come first */
2004 	memcpy(&offsets[1], &entry->watchers_offset,
2005 			sizeof(offsets) - sizeof(offsets[0]));
2006 
2007 	if (state->buf_kern_start) {
2008 		buf_start = state->buf_kern_start + state->buf_kern_offset;
2009 		offsets_update = (unsigned int *) buf_start;
2010 	}
2011 	ret = ebt_buf_add(state, &offsets[1],
2012 			sizeof(offsets) - sizeof(offsets[0]));
2013 	if (ret < 0)
2014 		return ret;
2015 	buf_start = (char *) entry;
2016 	/* 0: matches offset, always follows ebt_entry.
2017 	 * 1: watchers offset, from ebt_entry structure
2018 	 * 2: target offset, from ebt_entry structure
2019 	 * 3: next ebt_entry offset, from ebt_entry structure
2020 	 *
2021 	 * offsets are relative to beginning of struct ebt_entry (i.e., 0).
2022 	 */
2023 	for (i = 0; i < 4 ; ++i) {
2024 		if (offsets[i] > *total)
2025 			return -EINVAL;
2026 
2027 		if (i < 3 && offsets[i] == *total)
2028 			return -EINVAL;
2029 
2030 		if (i == 0)
2031 			continue;
2032 		if (offsets[i-1] > offsets[i])
2033 			return -EINVAL;
2034 	}
2035 
2036 	for (i = 0, j = 1 ; j < 4 ; j++, i++) {
2037 		struct compat_ebt_entry_mwt *match32;
2038 		unsigned int size;
2039 		char *buf = buf_start + offsets[i];
2040 
2041 		if (offsets[i] > offsets[j])
2042 			return -EINVAL;
2043 
2044 		match32 = (struct compat_ebt_entry_mwt *) buf;
2045 		size = offsets[j] - offsets[i];
2046 		ret = ebt_size_mwt(match32, size, i, state, base);
2047 		if (ret < 0)
2048 			return ret;
2049 		new_offset += ret;
2050 		if (offsets_update && new_offset) {
2051 			pr_debug("change offset %d to %d\n",
2052 				offsets_update[i], offsets[j] + new_offset);
2053 			offsets_update[i] = offsets[j] + new_offset;
2054 		}
2055 	}
2056 
2057 	if (state->buf_kern_start == NULL) {
2058 		unsigned int offset = buf_start - (char *) base;
2059 
2060 		ret = xt_compat_add_offset(NFPROTO_BRIDGE, offset, new_offset);
2061 		if (ret < 0)
2062 			return ret;
2063 	}
2064 
2065 	next_expected_off = state->buf_user_offset - startoff;
2066 	if (next_expected_off != entry->next_offset)
2067 		return -EINVAL;
2068 
2069 	if (*total < entry->next_offset)
2070 		return -EINVAL;
2071 	*total -= entry->next_offset;
2072 	return 0;
2073 }
2074 
2075 /* repl->entries_size is the size of the ebt_entry blob in userspace.
2076  * It might need more memory when copied to a 64 bit kernel in case
2077  * userspace is 32-bit. So, first task: find out how much memory is needed.
2078  *
2079  * Called before validation is performed.
2080  */
compat_copy_entries(unsigned char * data,unsigned int size_user,struct ebt_entries_buf_state * state)2081 static int compat_copy_entries(unsigned char *data, unsigned int size_user,
2082 				struct ebt_entries_buf_state *state)
2083 {
2084 	unsigned int size_remaining = size_user;
2085 	int ret;
2086 
2087 	ret = EBT_ENTRY_ITERATE(data, size_user, size_entry_mwt, data,
2088 					&size_remaining, state);
2089 	if (ret < 0)
2090 		return ret;
2091 
2092 	if (size_remaining)
2093 		return -EINVAL;
2094 
2095 	return state->buf_kern_offset;
2096 }
2097 
2098 
compat_copy_ebt_replace_from_user(struct ebt_replace * repl,sockptr_t arg,unsigned int len)2099 static int compat_copy_ebt_replace_from_user(struct ebt_replace *repl,
2100 					     sockptr_t arg, unsigned int len)
2101 {
2102 	struct compat_ebt_replace tmp;
2103 	int i;
2104 
2105 	if (len < sizeof(tmp))
2106 		return -EINVAL;
2107 
2108 	if (copy_from_sockptr(&tmp, arg, sizeof(tmp)))
2109 		return -EFAULT;
2110 
2111 	if (len != sizeof(tmp) + tmp.entries_size)
2112 		return -EINVAL;
2113 
2114 	if (tmp.entries_size == 0)
2115 		return -EINVAL;
2116 
2117 	if (tmp.nentries >= ((INT_MAX - sizeof(struct ebt_table_info)) /
2118 			NR_CPUS - SMP_CACHE_BYTES) / sizeof(struct ebt_counter))
2119 		return -ENOMEM;
2120 	if (tmp.num_counters >= INT_MAX / sizeof(struct ebt_counter))
2121 		return -ENOMEM;
2122 
2123 	memcpy(repl, &tmp, offsetof(struct ebt_replace, hook_entry));
2124 
2125 	/* starting with hook_entry, 32 vs. 64 bit structures are different */
2126 	for (i = 0; i < NF_BR_NUMHOOKS; i++)
2127 		repl->hook_entry[i] = compat_ptr(tmp.hook_entry[i]);
2128 
2129 	repl->num_counters = tmp.num_counters;
2130 	repl->counters = compat_ptr(tmp.counters);
2131 	repl->entries = compat_ptr(tmp.entries);
2132 	return 0;
2133 }
2134 
compat_do_replace(struct net * net,sockptr_t arg,unsigned int len)2135 static int compat_do_replace(struct net *net, sockptr_t arg, unsigned int len)
2136 {
2137 	int ret, i, countersize, size64;
2138 	struct ebt_table_info *newinfo;
2139 	struct ebt_replace tmp;
2140 	struct ebt_entries_buf_state state;
2141 	void *entries_tmp;
2142 
2143 	ret = compat_copy_ebt_replace_from_user(&tmp, arg, len);
2144 	if (ret) {
2145 		/* try real handler in case userland supplied needed padding */
2146 		if (ret == -EINVAL && do_replace(net, arg, len) == 0)
2147 			ret = 0;
2148 		return ret;
2149 	}
2150 
2151 	countersize = COUNTER_OFFSET(tmp.nentries) * nr_cpu_ids;
2152 	newinfo = vmalloc(sizeof(*newinfo) + countersize);
2153 	if (!newinfo)
2154 		return -ENOMEM;
2155 
2156 	if (countersize)
2157 		memset(newinfo->counters, 0, countersize);
2158 
2159 	memset(&state, 0, sizeof(state));
2160 
2161 	newinfo->entries = vmalloc(tmp.entries_size);
2162 	if (!newinfo->entries) {
2163 		ret = -ENOMEM;
2164 		goto free_newinfo;
2165 	}
2166 	if (copy_from_user(
2167 	   newinfo->entries, tmp.entries, tmp.entries_size) != 0) {
2168 		ret = -EFAULT;
2169 		goto free_entries;
2170 	}
2171 
2172 	entries_tmp = newinfo->entries;
2173 
2174 	xt_compat_lock(NFPROTO_BRIDGE);
2175 
2176 	ret = ebt_compat_init_offsets(tmp.nentries);
2177 	if (ret < 0)
2178 		goto out_unlock;
2179 
2180 	ret = compat_copy_entries(entries_tmp, tmp.entries_size, &state);
2181 	if (ret < 0)
2182 		goto out_unlock;
2183 
2184 	pr_debug("tmp.entries_size %d, kern off %d, user off %d delta %d\n",
2185 		tmp.entries_size, state.buf_kern_offset, state.buf_user_offset,
2186 		xt_compat_calc_jump(NFPROTO_BRIDGE, tmp.entries_size));
2187 
2188 	size64 = ret;
2189 	newinfo->entries = vmalloc(size64);
2190 	if (!newinfo->entries) {
2191 		vfree(entries_tmp);
2192 		ret = -ENOMEM;
2193 		goto out_unlock;
2194 	}
2195 
2196 	memset(&state, 0, sizeof(state));
2197 	state.buf_kern_start = newinfo->entries;
2198 	state.buf_kern_len = size64;
2199 
2200 	ret = compat_copy_entries(entries_tmp, tmp.entries_size, &state);
2201 	if (WARN_ON(ret < 0)) {
2202 		vfree(entries_tmp);
2203 		goto out_unlock;
2204 	}
2205 
2206 	vfree(entries_tmp);
2207 	tmp.entries_size = size64;
2208 
2209 	for (i = 0; i < NF_BR_NUMHOOKS; i++) {
2210 		char __user *usrptr;
2211 		if (tmp.hook_entry[i]) {
2212 			unsigned int delta;
2213 			usrptr = (char __user *) tmp.hook_entry[i];
2214 			delta = usrptr - tmp.entries;
2215 			usrptr += xt_compat_calc_jump(NFPROTO_BRIDGE, delta);
2216 			tmp.hook_entry[i] = (struct ebt_entries __user *)usrptr;
2217 		}
2218 	}
2219 
2220 	xt_compat_flush_offsets(NFPROTO_BRIDGE);
2221 	xt_compat_unlock(NFPROTO_BRIDGE);
2222 
2223 	ret = do_replace_finish(net, &tmp, newinfo);
2224 	if (ret == 0)
2225 		return ret;
2226 free_entries:
2227 	vfree(newinfo->entries);
2228 free_newinfo:
2229 	vfree(newinfo);
2230 	return ret;
2231 out_unlock:
2232 	xt_compat_flush_offsets(NFPROTO_BRIDGE);
2233 	xt_compat_unlock(NFPROTO_BRIDGE);
2234 	goto free_entries;
2235 }
2236 
compat_update_counters(struct net * net,sockptr_t arg,unsigned int len)2237 static int compat_update_counters(struct net *net, sockptr_t arg,
2238 				  unsigned int len)
2239 {
2240 	struct compat_ebt_replace hlp;
2241 
2242 	if (copy_from_sockptr(&hlp, arg, sizeof(hlp)))
2243 		return -EFAULT;
2244 
2245 	/* try real handler in case userland supplied needed padding */
2246 	if (len != sizeof(hlp) + hlp.num_counters * sizeof(struct ebt_counter))
2247 		return update_counters(net, arg, len);
2248 
2249 	return do_update_counters(net, hlp.name, compat_ptr(hlp.counters),
2250 				  hlp.num_counters, len);
2251 }
2252 
compat_do_ebt_get_ctl(struct sock * sk,int cmd,void __user * user,int * len)2253 static int compat_do_ebt_get_ctl(struct sock *sk, int cmd,
2254 		void __user *user, int *len)
2255 {
2256 	int ret;
2257 	struct compat_ebt_replace tmp;
2258 	struct ebt_table *t;
2259 	struct net *net = sock_net(sk);
2260 
2261 	if ((cmd == EBT_SO_GET_INFO || cmd == EBT_SO_GET_INIT_INFO) &&
2262 	    *len != sizeof(struct compat_ebt_replace))
2263 		return -EINVAL;
2264 
2265 	if (copy_from_user(&tmp, user, sizeof(tmp)))
2266 		return -EFAULT;
2267 
2268 	tmp.name[sizeof(tmp.name) - 1] = '\0';
2269 
2270 	t = find_table_lock(net, tmp.name, &ret, &ebt_mutex);
2271 	if (!t)
2272 		return ret;
2273 
2274 	xt_compat_lock(NFPROTO_BRIDGE);
2275 	switch (cmd) {
2276 	case EBT_SO_GET_INFO:
2277 		tmp.nentries = t->private->nentries;
2278 		ret = compat_table_info(t->private, &tmp);
2279 		if (ret)
2280 			goto out;
2281 		tmp.valid_hooks = t->valid_hooks;
2282 
2283 		if (copy_to_user(user, &tmp, *len) != 0) {
2284 			ret = -EFAULT;
2285 			break;
2286 		}
2287 		ret = 0;
2288 		break;
2289 	case EBT_SO_GET_INIT_INFO:
2290 		tmp.nentries = t->table->nentries;
2291 		tmp.entries_size = t->table->entries_size;
2292 		tmp.valid_hooks = t->table->valid_hooks;
2293 
2294 		if (copy_to_user(user, &tmp, *len) != 0) {
2295 			ret = -EFAULT;
2296 			break;
2297 		}
2298 		ret = 0;
2299 		break;
2300 	case EBT_SO_GET_ENTRIES:
2301 	case EBT_SO_GET_INIT_ENTRIES:
2302 		/* try real handler first in case of userland-side padding.
2303 		 * in case we are dealing with an 'ordinary' 32 bit binary
2304 		 * without 64bit compatibility padding, this will fail right
2305 		 * after copy_from_user when the *len argument is validated.
2306 		 *
2307 		 * the compat_ variant needs to do one pass over the kernel
2308 		 * data set to adjust for size differences before it the check.
2309 		 */
2310 		if (copy_everything_to_user(t, user, len, cmd) == 0)
2311 			ret = 0;
2312 		else
2313 			ret = compat_copy_everything_to_user(t, user, len, cmd);
2314 		break;
2315 	default:
2316 		ret = -EINVAL;
2317 	}
2318  out:
2319 	xt_compat_flush_offsets(NFPROTO_BRIDGE);
2320 	xt_compat_unlock(NFPROTO_BRIDGE);
2321 	mutex_unlock(&ebt_mutex);
2322 	return ret;
2323 }
2324 #endif
2325 
do_ebt_get_ctl(struct sock * sk,int cmd,void __user * user,int * len)2326 static int do_ebt_get_ctl(struct sock *sk, int cmd, void __user *user, int *len)
2327 {
2328 	struct net *net = sock_net(sk);
2329 	struct ebt_replace tmp;
2330 	struct ebt_table *t;
2331 	int ret;
2332 
2333 	if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2334 		return -EPERM;
2335 
2336 #ifdef CONFIG_COMPAT
2337 	/* try real handler in case userland supplied needed padding */
2338 	if (in_compat_syscall() &&
2339 	    ((cmd != EBT_SO_GET_INFO && cmd != EBT_SO_GET_INIT_INFO) ||
2340 	     *len != sizeof(tmp)))
2341 		return compat_do_ebt_get_ctl(sk, cmd, user, len);
2342 #endif
2343 
2344 	if (copy_from_user(&tmp, user, sizeof(tmp)))
2345 		return -EFAULT;
2346 
2347 	tmp.name[sizeof(tmp.name) - 1] = '\0';
2348 
2349 	t = find_table_lock(net, tmp.name, &ret, &ebt_mutex);
2350 	if (!t)
2351 		return ret;
2352 
2353 	switch (cmd) {
2354 	case EBT_SO_GET_INFO:
2355 	case EBT_SO_GET_INIT_INFO:
2356 		if (*len != sizeof(struct ebt_replace)) {
2357 			ret = -EINVAL;
2358 			mutex_unlock(&ebt_mutex);
2359 			break;
2360 		}
2361 		if (cmd == EBT_SO_GET_INFO) {
2362 			tmp.nentries = t->private->nentries;
2363 			tmp.entries_size = t->private->entries_size;
2364 			tmp.valid_hooks = t->valid_hooks;
2365 		} else {
2366 			tmp.nentries = t->table->nentries;
2367 			tmp.entries_size = t->table->entries_size;
2368 			tmp.valid_hooks = t->table->valid_hooks;
2369 		}
2370 		mutex_unlock(&ebt_mutex);
2371 		if (copy_to_user(user, &tmp, *len) != 0) {
2372 			ret = -EFAULT;
2373 			break;
2374 		}
2375 		ret = 0;
2376 		break;
2377 
2378 	case EBT_SO_GET_ENTRIES:
2379 	case EBT_SO_GET_INIT_ENTRIES:
2380 		ret = copy_everything_to_user(t, user, len, cmd);
2381 		mutex_unlock(&ebt_mutex);
2382 		break;
2383 
2384 	default:
2385 		mutex_unlock(&ebt_mutex);
2386 		ret = -EINVAL;
2387 	}
2388 
2389 	return ret;
2390 }
2391 
do_ebt_set_ctl(struct sock * sk,int cmd,sockptr_t arg,unsigned int len)2392 static int do_ebt_set_ctl(struct sock *sk, int cmd, sockptr_t arg,
2393 		unsigned int len)
2394 {
2395 	struct net *net = sock_net(sk);
2396 	int ret;
2397 
2398 	if (!ns_capable(net->user_ns, CAP_NET_ADMIN))
2399 		return -EPERM;
2400 
2401 	switch (cmd) {
2402 	case EBT_SO_SET_ENTRIES:
2403 #ifdef CONFIG_COMPAT
2404 		if (in_compat_syscall())
2405 			ret = compat_do_replace(net, arg, len);
2406 		else
2407 #endif
2408 			ret = do_replace(net, arg, len);
2409 		break;
2410 	case EBT_SO_SET_COUNTERS:
2411 #ifdef CONFIG_COMPAT
2412 		if (in_compat_syscall())
2413 			ret = compat_update_counters(net, arg, len);
2414 		else
2415 #endif
2416 			ret = update_counters(net, arg, len);
2417 		break;
2418 	default:
2419 		ret = -EINVAL;
2420 	}
2421 	return ret;
2422 }
2423 
2424 static struct nf_sockopt_ops ebt_sockopts = {
2425 	.pf		= PF_INET,
2426 	.set_optmin	= EBT_BASE_CTL,
2427 	.set_optmax	= EBT_SO_SET_MAX + 1,
2428 	.set		= do_ebt_set_ctl,
2429 	.get_optmin	= EBT_BASE_CTL,
2430 	.get_optmax	= EBT_SO_GET_MAX + 1,
2431 	.get		= do_ebt_get_ctl,
2432 	.owner		= THIS_MODULE,
2433 };
2434 
ebtables_init(void)2435 static int __init ebtables_init(void)
2436 {
2437 	int ret;
2438 
2439 	ret = xt_register_target(&ebt_standard_target);
2440 	if (ret < 0)
2441 		return ret;
2442 	ret = nf_register_sockopt(&ebt_sockopts);
2443 	if (ret < 0) {
2444 		xt_unregister_target(&ebt_standard_target);
2445 		return ret;
2446 	}
2447 
2448 	return 0;
2449 }
2450 
ebtables_fini(void)2451 static void __exit ebtables_fini(void)
2452 {
2453 	nf_unregister_sockopt(&ebt_sockopts);
2454 	xt_unregister_target(&ebt_standard_target);
2455 }
2456 
2457 EXPORT_SYMBOL(ebt_register_table);
2458 EXPORT_SYMBOL(ebt_unregister_table);
2459 EXPORT_SYMBOL(ebt_do_table);
2460 module_init(ebtables_init);
2461 module_exit(ebtables_fini);
2462 MODULE_LICENSE("GPL");
2463