netfilter: ctnetlink: only assign helpers for matching protocols
[linux-2.6-block.git] / net / bridge / netfilter / ebtables.c
CommitLineData
1da177e4
LT
1/*
2 * ebtables
3 *
4 * Author:
5 * Bart De Schuymer <bdschuym@pandora.be>
6 *
7 * ebtables.c,v 2.0, July, 2002
8 *
9 * This code is stongly inspired on the iptables code which is
10 * Copyright (C) 1999 Paul `Rusty' Russell & Michael J. Neuling
11 *
12 * This program is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU General Public License
14 * as published by the Free Software Foundation; either version
15 * 2 of the License, or (at your option) any later version.
16 */
17
1da177e4
LT
18
19#include <linux/kmod.h>
20#include <linux/module.h>
21#include <linux/vmalloc.h>
18219d3f 22#include <linux/netfilter/x_tables.h>
1da177e4
LT
23#include <linux/netfilter_bridge/ebtables.h>
24#include <linux/spinlock.h>
df0933dc 25#include <linux/mutex.h>
1da177e4
LT
26#include <asm/uaccess.h>
27#include <linux/smp.h>
c8923c6b 28#include <linux/cpumask.h>
1da177e4
LT
29#include <net/sock.h>
30/* needed for logical [in,out]-dev filtering */
31#include "../br_private.h"
32
1da177e4 33#define BUGPRINT(format, args...) printk("kernel msg: ebtables bug: please "\
9d6f229f 34 "report to author: "format, ## args)
1da177e4 35/* #define BUGPRINT(format, args...) */
1da177e4 36#define MEMPRINT(format, args...) printk("kernel msg: ebtables "\
9d6f229f 37 ": out of memory: "format, ## args)
1da177e4
LT
38/* #define MEMPRINT(format, args...) */
39
40
41
42/*
43 * Each cpu has its own set of counters, so there is no need for write_lock in
44 * the softirq
45 * For reading or updating the counters, the user context needs to
46 * get a write_lock
47 */
48
49/* The size of each set of counters is altered to get cache alignment */
50#define SMP_ALIGN(x) (((x) + SMP_CACHE_BYTES-1) & ~(SMP_CACHE_BYTES-1))
51#define COUNTER_OFFSET(n) (SMP_ALIGN(n * sizeof(struct ebt_counter)))
52#define COUNTER_BASE(c, n, cpu) ((struct ebt_counter *)(((char *)c) + \
53 COUNTER_OFFSET(n) * cpu))
54
55
56
57b47a53 57static DEFINE_MUTEX(ebt_mutex);
1da177e4 58
043ef46c 59static struct xt_target ebt_standard_target = {
001a18d3
JE
60 .name = "standard",
61 .revision = 0,
62 .family = NFPROTO_BRIDGE,
043ef46c 63 .targetsize = sizeof(int),
18219d3f 64};
1da177e4 65
7eb35586
JE
66static inline int
67ebt_do_watcher(const struct ebt_entry_watcher *w, struct sk_buff *skb,
68 struct xt_target_param *par)
1da177e4 69{
7eb35586
JE
70 par->target = w->u.watcher;
71 par->targinfo = w->data;
72 w->u.watcher->target(skb, par);
1da177e4
LT
73 /* watchers don't give a verdict */
74 return 0;
75}
76
77static inline int ebt_do_match (struct ebt_entry_match *m,
f7108a20 78 const struct sk_buff *skb, struct xt_match_param *par)
1da177e4 79{
f7108a20
JE
80 par->match = m->u.match;
81 par->matchinfo = m->data;
d61ba9fd 82 return m->u.match->match(skb, par) ? EBT_MATCH : EBT_NOMATCH;
1da177e4
LT
83}
84
85static inline int ebt_dev_check(char *entry, const struct net_device *device)
86{
87 int i = 0;
f3d8b2e4 88 const char *devname;
1da177e4
LT
89
90 if (*entry == '\0')
91 return 0;
92 if (!device)
93 return 1;
f3d8b2e4 94 devname = device->name;
1da177e4
LT
95 /* 1 is the wildcard token */
96 while (entry[i] != '\0' && entry[i] != 1 && entry[i] == devname[i])
97 i++;
98 return (devname[i] != entry[i] && entry[i] != 1);
99}
100
101#define FWINV2(bool,invflg) ((bool) ^ !!(e->invflags & invflg))
102/* process standard matches */
103static inline int ebt_basic_match(struct ebt_entry *e, struct ethhdr *h,
104 const struct net_device *in, const struct net_device *out)
105{
106 int verdict, i;
107
108 if (e->bitmask & EBT_802_3) {
109 if (FWINV2(ntohs(h->h_proto) >= 1536, EBT_IPROTO))
110 return 1;
111 } else if (!(e->bitmask & EBT_NOPROTO) &&
112 FWINV2(e->ethproto != h->h_proto, EBT_IPROTO))
113 return 1;
114
115 if (FWINV2(ebt_dev_check(e->in, in), EBT_IIN))
116 return 1;
117 if (FWINV2(ebt_dev_check(e->out, out), EBT_IOUT))
118 return 1;
119 if ((!in || !in->br_port) ? 0 : FWINV2(ebt_dev_check(
120 e->logical_in, in->br_port->br->dev), EBT_ILOGICALIN))
121 return 1;
122 if ((!out || !out->br_port) ? 0 : FWINV2(ebt_dev_check(
123 e->logical_out, out->br_port->br->dev), EBT_ILOGICALOUT))
124 return 1;
125
126 if (e->bitmask & EBT_SOURCEMAC) {
127 verdict = 0;
128 for (i = 0; i < 6; i++)
129 verdict |= (h->h_source[i] ^ e->sourcemac[i]) &
130 e->sourcemsk[i];
131 if (FWINV2(verdict != 0, EBT_ISOURCE) )
132 return 1;
133 }
134 if (e->bitmask & EBT_DESTMAC) {
135 verdict = 0;
136 for (i = 0; i < 6; i++)
137 verdict |= (h->h_dest[i] ^ e->destmac[i]) &
138 e->destmsk[i];
139 if (FWINV2(verdict != 0, EBT_IDEST) )
140 return 1;
141 }
142 return 0;
143}
144
98e86403
JE
145static inline __pure
146struct ebt_entry *ebt_next_entry(const struct ebt_entry *entry)
147{
148 return (void *)entry + entry->next_offset;
149}
150
1da177e4 151/* Do some firewalling */
3db05fea 152unsigned int ebt_do_table (unsigned int hook, struct sk_buff *skb,
1da177e4
LT
153 const struct net_device *in, const struct net_device *out,
154 struct ebt_table *table)
155{
156 int i, nentries;
157 struct ebt_entry *point;
158 struct ebt_counter *counter_base, *cb_base;
159 struct ebt_entry_target *t;
160 int verdict, sp = 0;
161 struct ebt_chainstack *cs;
162 struct ebt_entries *chaininfo;
163 char *base;
164 struct ebt_table_info *private;
5365f802 165 bool hotdrop = false;
f7108a20 166 struct xt_match_param mtpar;
7eb35586 167 struct xt_target_param tgpar;
f7108a20 168
916a917d 169 mtpar.family = tgpar.family = NFPROTO_BRIDGE;
7eb35586
JE
170 mtpar.in = tgpar.in = in;
171 mtpar.out = tgpar.out = out;
f7108a20 172 mtpar.hotdrop = &hotdrop;
a5e78820 173 mtpar.hooknum = tgpar.hooknum = hook;
1da177e4
LT
174
175 read_lock_bh(&table->lock);
176 private = table->private;
177 cb_base = COUNTER_BASE(private->counters, private->nentries,
178 smp_processor_id());
179 if (private->chainstack)
180 cs = private->chainstack[smp_processor_id()];
181 else
182 cs = NULL;
183 chaininfo = private->hook_entry[hook];
184 nentries = private->hook_entry[hook]->nentries;
185 point = (struct ebt_entry *)(private->hook_entry[hook]->data);
186 counter_base = cb_base + private->hook_entry[hook]->counter_offset;
187 /* base for chain jumps */
188 base = private->entries;
189 i = 0;
190 while (i < nentries) {
3db05fea 191 if (ebt_basic_match(point, eth_hdr(skb), in, out))
1da177e4
LT
192 goto letscontinue;
193
f7108a20 194 if (EBT_MATCH_ITERATE(point, ebt_do_match, skb, &mtpar) != 0)
1da177e4 195 goto letscontinue;
5365f802
JE
196 if (hotdrop) {
197 read_unlock_bh(&table->lock);
198 return NF_DROP;
199 }
1da177e4
LT
200
201 /* increase counter */
202 (*(counter_base + i)).pcnt++;
3db05fea 203 (*(counter_base + i)).bcnt += skb->len;
1da177e4
LT
204
205 /* these should only watch: not modify, nor tell us
206 what to do with the packet */
7eb35586 207 EBT_WATCHER_ITERATE(point, ebt_do_watcher, skb, &tgpar);
1da177e4
LT
208
209 t = (struct ebt_entry_target *)
210 (((char *)point) + point->target_offset);
211 /* standard target */
212 if (!t->u.target->target)
213 verdict = ((struct ebt_standard_target *)t)->verdict;
7eb35586
JE
214 else {
215 tgpar.target = t->u.target;
216 tgpar.targinfo = t->data;
217 verdict = t->u.target->target(skb, &tgpar);
218 }
1da177e4
LT
219 if (verdict == EBT_ACCEPT) {
220 read_unlock_bh(&table->lock);
221 return NF_ACCEPT;
222 }
223 if (verdict == EBT_DROP) {
224 read_unlock_bh(&table->lock);
225 return NF_DROP;
226 }
227 if (verdict == EBT_RETURN) {
228letsreturn:
229#ifdef CONFIG_NETFILTER_DEBUG
230 if (sp == 0) {
231 BUGPRINT("RETURN on base chain");
232 /* act like this is EBT_CONTINUE */
233 goto letscontinue;
234 }
235#endif
236 sp--;
237 /* put all the local variables right */
238 i = cs[sp].n;
239 chaininfo = cs[sp].chaininfo;
240 nentries = chaininfo->nentries;
241 point = cs[sp].e;
242 counter_base = cb_base +
243 chaininfo->counter_offset;
244 continue;
245 }
246 if (verdict == EBT_CONTINUE)
247 goto letscontinue;
248#ifdef CONFIG_NETFILTER_DEBUG
249 if (verdict < 0) {
250 BUGPRINT("bogus standard verdict\n");
251 read_unlock_bh(&table->lock);
252 return NF_DROP;
253 }
254#endif
255 /* jump to a udc */
256 cs[sp].n = i + 1;
257 cs[sp].chaininfo = chaininfo;
98e86403 258 cs[sp].e = ebt_next_entry(point);
1da177e4
LT
259 i = 0;
260 chaininfo = (struct ebt_entries *) (base + verdict);
261#ifdef CONFIG_NETFILTER_DEBUG
262 if (chaininfo->distinguisher) {
263 BUGPRINT("jump to non-chain\n");
264 read_unlock_bh(&table->lock);
265 return NF_DROP;
266 }
267#endif
268 nentries = chaininfo->nentries;
269 point = (struct ebt_entry *)chaininfo->data;
270 counter_base = cb_base + chaininfo->counter_offset;
271 sp++;
272 continue;
273letscontinue:
98e86403 274 point = ebt_next_entry(point);
1da177e4
LT
275 i++;
276 }
277
278 /* I actually like this :) */
279 if (chaininfo->policy == EBT_RETURN)
280 goto letsreturn;
281 if (chaininfo->policy == EBT_ACCEPT) {
282 read_unlock_bh(&table->lock);
283 return NF_ACCEPT;
284 }
285 read_unlock_bh(&table->lock);
286 return NF_DROP;
287}
288
289/* If it succeeds, returns element and locks mutex */
290static inline void *
291find_inlist_lock_noload(struct list_head *head, const char *name, int *error,
57b47a53 292 struct mutex *mutex)
1da177e4 293{
df0933dc
PM
294 struct {
295 struct list_head list;
296 char name[EBT_FUNCTION_MAXNAMELEN];
297 } *e;
1da177e4 298
57b47a53 299 *error = mutex_lock_interruptible(mutex);
1da177e4
LT
300 if (*error != 0)
301 return NULL;
302
df0933dc
PM
303 list_for_each_entry(e, head, list) {
304 if (strcmp(e->name, name) == 0)
305 return e;
1da177e4 306 }
df0933dc
PM
307 *error = -ENOENT;
308 mutex_unlock(mutex);
309 return NULL;
1da177e4
LT
310}
311
1da177e4
LT
312static void *
313find_inlist_lock(struct list_head *head, const char *name, const char *prefix,
57b47a53 314 int *error, struct mutex *mutex)
1da177e4 315{
95a5afca
JB
316 return try_then_request_module(
317 find_inlist_lock_noload(head, name, error, mutex),
318 "%s%s", prefix, name);
1da177e4 319}
1da177e4
LT
320
321static inline struct ebt_table *
511061e2
AD
322find_table_lock(struct net *net, const char *name, int *error,
323 struct mutex *mutex)
1da177e4 324{
511061e2
AD
325 return find_inlist_lock(&net->xt.tables[NFPROTO_BRIDGE], name,
326 "ebtable_", error, mutex);
1da177e4
LT
327}
328
1da177e4 329static inline int
9b4fce7a
JE
330ebt_check_match(struct ebt_entry_match *m, struct xt_mtchk_param *par,
331 unsigned int *cnt)
1da177e4 332{
9b4fce7a 333 const struct ebt_entry *e = par->entryinfo;
043ef46c 334 struct xt_match *match;
14197d54 335 size_t left = ((char *)e + e->watchers_offset) - (char *)m;
1da177e4
LT
336 int ret;
337
14197d54
AV
338 if (left < sizeof(struct ebt_entry_match) ||
339 left - sizeof(struct ebt_entry_match) < m->match_size)
1da177e4 340 return -EINVAL;
043ef46c
JE
341
342 match = try_then_request_module(xt_find_match(NFPROTO_BRIDGE,
343 m->u.name, 0), "ebt_%s", m->u.name);
344 if (IS_ERR(match))
345 return PTR_ERR(match);
346 if (match == NULL)
1da177e4 347 return -ENOENT;
043ef46c
JE
348 m->u.match = match;
349
9b4fce7a
JE
350 par->match = match;
351 par->matchinfo = m->data;
916a917d 352 ret = xt_check_match(par, m->match_size,
9b4fce7a 353 e->ethproto, e->invflags & EBT_IPROTO);
043ef46c
JE
354 if (ret < 0) {
355 module_put(match->me);
356 return ret;
1da177e4 357 }
043ef46c 358
1da177e4
LT
359 (*cnt)++;
360 return 0;
361}
362
363static inline int
af5d6dc2
JE
364ebt_check_watcher(struct ebt_entry_watcher *w, struct xt_tgchk_param *par,
365 unsigned int *cnt)
1da177e4 366{
af5d6dc2 367 const struct ebt_entry *e = par->entryinfo;
043ef46c 368 struct xt_target *watcher;
14197d54 369 size_t left = ((char *)e + e->target_offset) - (char *)w;
1da177e4
LT
370 int ret;
371
14197d54
AV
372 if (left < sizeof(struct ebt_entry_watcher) ||
373 left - sizeof(struct ebt_entry_watcher) < w->watcher_size)
1da177e4 374 return -EINVAL;
043ef46c
JE
375
376 watcher = try_then_request_module(
377 xt_find_target(NFPROTO_BRIDGE, w->u.name, 0),
378 "ebt_%s", w->u.name);
379 if (IS_ERR(watcher))
380 return PTR_ERR(watcher);
381 if (watcher == NULL)
1da177e4 382 return -ENOENT;
043ef46c
JE
383 w->u.watcher = watcher;
384
af5d6dc2
JE
385 par->target = watcher;
386 par->targinfo = w->data;
916a917d 387 ret = xt_check_target(par, w->watcher_size,
af5d6dc2 388 e->ethproto, e->invflags & EBT_IPROTO);
043ef46c
JE
389 if (ret < 0) {
390 module_put(watcher->me);
391 return ret;
1da177e4 392 }
043ef46c 393
1da177e4
LT
394 (*cnt)++;
395 return 0;
396}
397
70fe9af4
AV
398static int ebt_verify_pointers(struct ebt_replace *repl,
399 struct ebt_table_info *newinfo)
1da177e4 400{
70fe9af4
AV
401 unsigned int limit = repl->entries_size;
402 unsigned int valid_hooks = repl->valid_hooks;
403 unsigned int offset = 0;
1da177e4
LT
404 int i;
405
e4fd77de
AV
406 for (i = 0; i < NF_BR_NUMHOOKS; i++)
407 newinfo->hook_entry[i] = NULL;
408
409 newinfo->entries_size = repl->entries_size;
410 newinfo->nentries = repl->nentries;
411
70fe9af4
AV
412 while (offset < limit) {
413 size_t left = limit - offset;
414 struct ebt_entry *e = (void *)newinfo->entries + offset;
bb2ef25c 415
70fe9af4 416 if (left < sizeof(unsigned int))
1da177e4 417 break;
70fe9af4
AV
418
419 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
420 if ((valid_hooks & (1 << i)) == 0)
421 continue;
1e419cd9
AV
422 if ((char __user *)repl->hook_entry[i] ==
423 repl->entries + offset)
70fe9af4
AV
424 break;
425 }
426
427 if (i != NF_BR_NUMHOOKS || !(e->bitmask & EBT_ENTRY_OR_ENTRIES)) {
428 if (e->bitmask != 0) {
429 /* we make userspace set this right,
430 so there is no misunderstanding */
431 BUGPRINT("EBT_ENTRY_OR_ENTRIES shouldn't be set "
432 "in distinguisher\n");
433 return -EINVAL;
434 }
435 if (i != NF_BR_NUMHOOKS)
436 newinfo->hook_entry[i] = (struct ebt_entries *)e;
437 if (left < sizeof(struct ebt_entries))
438 break;
439 offset += sizeof(struct ebt_entries);
440 } else {
441 if (left < sizeof(struct ebt_entry))
442 break;
443 if (left < e->next_offset)
444 break;
445 offset += e->next_offset;
1da177e4 446 }
22b440bf 447 }
70fe9af4
AV
448 if (offset != limit) {
449 BUGPRINT("entries_size too small\n");
450 return -EINVAL;
451 }
e4fd77de
AV
452
453 /* check if all valid hooks have a chain */
454 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
455 if (!newinfo->hook_entry[i] &&
456 (valid_hooks & (1 << i))) {
457 BUGPRINT("Valid hook without chain\n");
458 return -EINVAL;
459 }
460 }
22b440bf 461 return 0;
22b440bf
AV
462}
463
464/*
465 * this one is very careful, as it is the first function
466 * to parse the userspace data
467 */
468static inline int
469ebt_check_entry_size_and_hooks(struct ebt_entry *e,
0e795531
AV
470 struct ebt_table_info *newinfo,
471 unsigned int *n, unsigned int *cnt,
472 unsigned int *totalcnt, unsigned int *udc_cnt)
22b440bf 473{
22b440bf
AV
474 int i;
475
476 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
0e795531 477 if ((void *)e == (void *)newinfo->hook_entry[i])
22b440bf
AV
478 break;
479 }
480 /* beginning of a new chain
481 if i == NF_BR_NUMHOOKS it must be a user defined chain */
482 if (i != NF_BR_NUMHOOKS || !e->bitmask) {
1da177e4
LT
483 /* this checks if the previous chain has as many entries
484 as it said it has */
485 if (*n != *cnt) {
486 BUGPRINT("nentries does not equal the nr of entries "
9d6f229f 487 "in the chain\n");
1da177e4
LT
488 return -EINVAL;
489 }
1da177e4
LT
490 if (((struct ebt_entries *)e)->policy != EBT_DROP &&
491 ((struct ebt_entries *)e)->policy != EBT_ACCEPT) {
492 /* only RETURN from udc */
493 if (i != NF_BR_NUMHOOKS ||
494 ((struct ebt_entries *)e)->policy != EBT_RETURN) {
495 BUGPRINT("bad policy\n");
496 return -EINVAL;
497 }
498 }
499 if (i == NF_BR_NUMHOOKS) /* it's a user defined chain */
500 (*udc_cnt)++;
1da177e4
LT
501 if (((struct ebt_entries *)e)->counter_offset != *totalcnt) {
502 BUGPRINT("counter_offset != totalcnt");
503 return -EINVAL;
504 }
505 *n = ((struct ebt_entries *)e)->nentries;
506 *cnt = 0;
507 return 0;
508 }
509 /* a plain old entry, heh */
510 if (sizeof(struct ebt_entry) > e->watchers_offset ||
511 e->watchers_offset > e->target_offset ||
512 e->target_offset >= e->next_offset) {
513 BUGPRINT("entry offsets not in right order\n");
514 return -EINVAL;
515 }
516 /* this is not checked anywhere else */
517 if (e->next_offset - e->target_offset < sizeof(struct ebt_entry_target)) {
518 BUGPRINT("target size too small\n");
519 return -EINVAL;
520 }
1da177e4
LT
521 (*cnt)++;
522 (*totalcnt)++;
523 return 0;
524}
525
526struct ebt_cl_stack
527{
528 struct ebt_chainstack cs;
529 int from;
530 unsigned int hookmask;
531};
532
533/*
534 * we need these positions to check that the jumps to a different part of the
535 * entries is a jump to the beginning of a new chain.
536 */
537static inline int
538ebt_get_udc_positions(struct ebt_entry *e, struct ebt_table_info *newinfo,
177abc34 539 unsigned int *n, struct ebt_cl_stack *udc)
1da177e4
LT
540{
541 int i;
542
543 /* we're only interested in chain starts */
40642f95 544 if (e->bitmask)
1da177e4
LT
545 return 0;
546 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
1da177e4
LT
547 if (newinfo->hook_entry[i] == (struct ebt_entries *)e)
548 break;
549 }
550 /* only care about udc */
551 if (i != NF_BR_NUMHOOKS)
552 return 0;
553
554 udc[*n].cs.chaininfo = (struct ebt_entries *)e;
555 /* these initialisations are depended on later in check_chainloops() */
556 udc[*n].cs.n = 0;
557 udc[*n].hookmask = 0;
558
559 (*n)++;
560 return 0;
561}
562
563static inline int
f54e9367 564ebt_cleanup_match(struct ebt_entry_match *m, struct net *net, unsigned int *i)
1da177e4 565{
6be3d859
JE
566 struct xt_mtdtor_param par;
567
1da177e4
LT
568 if (i && (*i)-- == 0)
569 return 1;
1da177e4 570
f54e9367 571 par.net = net;
6be3d859
JE
572 par.match = m->u.match;
573 par.matchinfo = m->data;
916a917d 574 par.family = NFPROTO_BRIDGE;
6be3d859
JE
575 if (par.match->destroy != NULL)
576 par.match->destroy(&par);
577 module_put(par.match->me);
1da177e4
LT
578 return 0;
579}
580
581static inline int
582ebt_cleanup_watcher(struct ebt_entry_watcher *w, unsigned int *i)
583{
a2df1648
JE
584 struct xt_tgdtor_param par;
585
1da177e4
LT
586 if (i && (*i)-- == 0)
587 return 1;
1da177e4 588
a2df1648
JE
589 par.target = w->u.watcher;
590 par.targinfo = w->data;
916a917d 591 par.family = NFPROTO_BRIDGE;
a2df1648
JE
592 if (par.target->destroy != NULL)
593 par.target->destroy(&par);
594 module_put(par.target->me);
1da177e4
LT
595 return 0;
596}
597
598static inline int
f54e9367 599ebt_cleanup_entry(struct ebt_entry *e, struct net *net, unsigned int *cnt)
1da177e4 600{
a2df1648 601 struct xt_tgdtor_param par;
1da177e4
LT
602 struct ebt_entry_target *t;
603
40642f95 604 if (e->bitmask == 0)
1da177e4
LT
605 return 0;
606 /* we're done */
607 if (cnt && (*cnt)-- == 0)
608 return 1;
609 EBT_WATCHER_ITERATE(e, ebt_cleanup_watcher, NULL);
f54e9367 610 EBT_MATCH_ITERATE(e, ebt_cleanup_match, net, NULL);
1da177e4 611 t = (struct ebt_entry_target *)(((char *)e) + e->target_offset);
1da177e4 612
a2df1648
JE
613 par.target = t->u.target;
614 par.targinfo = t->data;
916a917d 615 par.family = NFPROTO_BRIDGE;
a2df1648
JE
616 if (par.target->destroy != NULL)
617 par.target->destroy(&par);
618 module_put(par.target->me);
1da177e4
LT
619 return 0;
620}
621
622static inline int
a83d8e8d
AD
623ebt_check_entry(struct ebt_entry *e,
624 struct net *net,
625 struct ebt_table_info *newinfo,
f7da79d9 626 const char *name, unsigned int *cnt,
1da177e4
LT
627 struct ebt_cl_stack *cl_s, unsigned int udc_cnt)
628{
629 struct ebt_entry_target *t;
043ef46c 630 struct xt_target *target;
1da177e4 631 unsigned int i, j, hook = 0, hookmask = 0;
44f9a2fd 632 size_t gap;
1da177e4 633 int ret;
6be3d859 634 struct xt_mtchk_param mtpar;
af5d6dc2 635 struct xt_tgchk_param tgpar;
1da177e4
LT
636
637 /* don't mess with the struct ebt_entries */
40642f95 638 if (e->bitmask == 0)
1da177e4
LT
639 return 0;
640
641 if (e->bitmask & ~EBT_F_MASK) {
642 BUGPRINT("Unknown flag for bitmask\n");
643 return -EINVAL;
644 }
645 if (e->invflags & ~EBT_INV_MASK) {
646 BUGPRINT("Unknown flag for inv bitmask\n");
647 return -EINVAL;
648 }
649 if ( (e->bitmask & EBT_NOPROTO) && (e->bitmask & EBT_802_3) ) {
650 BUGPRINT("NOPROTO & 802_3 not allowed\n");
651 return -EINVAL;
652 }
653 /* what hook do we belong to? */
654 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
f7da79d9 655 if (!newinfo->hook_entry[i])
1da177e4
LT
656 continue;
657 if ((char *)newinfo->hook_entry[i] < (char *)e)
658 hook = i;
659 else
660 break;
661 }
662 /* (1 << NF_BR_NUMHOOKS) tells the check functions the rule is on
663 a base chain */
664 if (i < NF_BR_NUMHOOKS)
665 hookmask = (1 << hook) | (1 << NF_BR_NUMHOOKS);
666 else {
667 for (i = 0; i < udc_cnt; i++)
668 if ((char *)(cl_s[i].cs.chaininfo) > (char *)e)
669 break;
670 if (i == 0)
671 hookmask = (1 << hook) | (1 << NF_BR_NUMHOOKS);
672 else
673 hookmask = cl_s[i - 1].hookmask;
674 }
675 i = 0;
9b4fce7a 676
a83d8e8d 677 mtpar.net = net;
af5d6dc2
JE
678 mtpar.table = tgpar.table = name;
679 mtpar.entryinfo = tgpar.entryinfo = e;
680 mtpar.hook_mask = tgpar.hook_mask = hookmask;
916a917d 681 mtpar.family = tgpar.family = NFPROTO_BRIDGE;
6be3d859 682 ret = EBT_MATCH_ITERATE(e, ebt_check_match, &mtpar, &i);
1da177e4
LT
683 if (ret != 0)
684 goto cleanup_matches;
685 j = 0;
af5d6dc2 686 ret = EBT_WATCHER_ITERATE(e, ebt_check_watcher, &tgpar, &j);
1da177e4
LT
687 if (ret != 0)
688 goto cleanup_watchers;
689 t = (struct ebt_entry_target *)(((char *)e) + e->target_offset);
44f9a2fd 690 gap = e->next_offset - e->target_offset;
1da177e4 691
043ef46c
JE
692 target = try_then_request_module(
693 xt_find_target(NFPROTO_BRIDGE, t->u.name, 0),
694 "ebt_%s", t->u.name);
695 if (IS_ERR(target)) {
696 ret = PTR_ERR(target);
001a18d3 697 goto cleanup_watchers;
043ef46c
JE
698 } else if (target == NULL) {
699 ret = -ENOENT;
001a18d3
JE
700 goto cleanup_watchers;
701 }
702
1da177e4
LT
703 t->u.target = target;
704 if (t->u.target == &ebt_standard_target) {
14197d54 705 if (gap < sizeof(struct ebt_standard_target)) {
1da177e4
LT
706 BUGPRINT("Standard target size too big\n");
707 ret = -EFAULT;
708 goto cleanup_watchers;
709 }
710 if (((struct ebt_standard_target *)t)->verdict <
711 -NUM_STANDARD_TARGETS) {
712 BUGPRINT("Invalid standard target\n");
713 ret = -EFAULT;
714 goto cleanup_watchers;
715 }
18219d3f
JE
716 } else if (t->target_size > gap - sizeof(struct ebt_entry_target)) {
717 module_put(t->u.target->me);
718 ret = -EFAULT;
719 goto cleanup_watchers;
043ef46c
JE
720 }
721
af5d6dc2
JE
722 tgpar.target = target;
723 tgpar.targinfo = t->data;
916a917d 724 ret = xt_check_target(&tgpar, t->target_size,
af5d6dc2 725 e->ethproto, e->invflags & EBT_IPROTO);
043ef46c
JE
726 if (ret < 0) {
727 module_put(target->me);
18219d3f 728 goto cleanup_watchers;
1da177e4
LT
729 }
730 (*cnt)++;
731 return 0;
732cleanup_watchers:
733 EBT_WATCHER_ITERATE(e, ebt_cleanup_watcher, &j);
734cleanup_matches:
f54e9367 735 EBT_MATCH_ITERATE(e, ebt_cleanup_match, net, &i);
1da177e4
LT
736 return ret;
737}
738
739/*
740 * checks for loops and sets the hook mask for udc
741 * the hook mask for udc tells us from which base chains the udc can be
742 * accessed. This mask is a parameter to the check() functions of the extensions
743 */
744static int check_chainloops(struct ebt_entries *chain, struct ebt_cl_stack *cl_s,
745 unsigned int udc_cnt, unsigned int hooknr, char *base)
746{
747 int i, chain_nr = -1, pos = 0, nentries = chain->nentries, verdict;
748 struct ebt_entry *e = (struct ebt_entry *)chain->data;
749 struct ebt_entry_target *t;
750
751 while (pos < nentries || chain_nr != -1) {
752 /* end of udc, go back one 'recursion' step */
753 if (pos == nentries) {
754 /* put back values of the time when this chain was called */
755 e = cl_s[chain_nr].cs.e;
756 if (cl_s[chain_nr].from != -1)
757 nentries =
758 cl_s[cl_s[chain_nr].from].cs.chaininfo->nentries;
759 else
760 nentries = chain->nentries;
761 pos = cl_s[chain_nr].cs.n;
762 /* make sure we won't see a loop that isn't one */
763 cl_s[chain_nr].cs.n = 0;
764 chain_nr = cl_s[chain_nr].from;
765 if (pos == nentries)
766 continue;
767 }
768 t = (struct ebt_entry_target *)
769 (((char *)e) + e->target_offset);
770 if (strcmp(t->u.name, EBT_STANDARD_TARGET))
771 goto letscontinue;
772 if (e->target_offset + sizeof(struct ebt_standard_target) >
773 e->next_offset) {
774 BUGPRINT("Standard target size too big\n");
775 return -1;
776 }
777 verdict = ((struct ebt_standard_target *)t)->verdict;
778 if (verdict >= 0) { /* jump to another chain */
779 struct ebt_entries *hlp2 =
780 (struct ebt_entries *)(base + verdict);
781 for (i = 0; i < udc_cnt; i++)
782 if (hlp2 == cl_s[i].cs.chaininfo)
783 break;
784 /* bad destination or loop */
785 if (i == udc_cnt) {
786 BUGPRINT("bad destination\n");
787 return -1;
788 }
789 if (cl_s[i].cs.n) {
790 BUGPRINT("loop\n");
791 return -1;
792 }
98a0824a
AV
793 if (cl_s[i].hookmask & (1 << hooknr))
794 goto letscontinue;
795 /* this can't be 0, so the loop test is correct */
1da177e4
LT
796 cl_s[i].cs.n = pos + 1;
797 pos = 0;
98e86403 798 cl_s[i].cs.e = ebt_next_entry(e);
1da177e4
LT
799 e = (struct ebt_entry *)(hlp2->data);
800 nentries = hlp2->nentries;
801 cl_s[i].from = chain_nr;
802 chain_nr = i;
803 /* this udc is accessible from the base chain for hooknr */
804 cl_s[i].hookmask |= (1 << hooknr);
805 continue;
806 }
807letscontinue:
98e86403 808 e = ebt_next_entry(e);
1da177e4
LT
809 pos++;
810 }
811 return 0;
812}
813
814/* do the parsing of the table/chains/entries/matches/watchers/targets, heh */
a83d8e8d
AD
815static int translate_table(struct net *net, char *name,
816 struct ebt_table_info *newinfo)
1da177e4
LT
817{
818 unsigned int i, j, k, udc_cnt;
819 int ret;
820 struct ebt_cl_stack *cl_s = NULL; /* used in the checking for chain loops */
821
822 i = 0;
1f072c96 823 while (i < NF_BR_NUMHOOKS && !newinfo->hook_entry[i])
1da177e4
LT
824 i++;
825 if (i == NF_BR_NUMHOOKS) {
826 BUGPRINT("No valid hooks specified\n");
827 return -EINVAL;
828 }
1f072c96 829 if (newinfo->hook_entry[i] != (struct ebt_entries *)newinfo->entries) {
1da177e4
LT
830 BUGPRINT("Chains don't start at beginning\n");
831 return -EINVAL;
832 }
833 /* make sure chains are ordered after each other in same order
834 as their corresponding hooks */
835 for (j = i + 1; j < NF_BR_NUMHOOKS; j++) {
1f072c96 836 if (!newinfo->hook_entry[j])
1da177e4 837 continue;
1f072c96 838 if (newinfo->hook_entry[j] <= newinfo->hook_entry[i]) {
1da177e4
LT
839 BUGPRINT("Hook order must be followed\n");
840 return -EINVAL;
841 }
842 i = j;
843 }
844
1da177e4
LT
845 /* do some early checkings and initialize some things */
846 i = 0; /* holds the expected nr. of entries for the chain */
847 j = 0; /* holds the up to now counted entries for the chain */
848 k = 0; /* holds the total nr. of entries, should equal
9d6f229f 849 newinfo->nentries afterwards */
1da177e4
LT
850 udc_cnt = 0; /* will hold the nr. of user defined chains (udc) */
851 ret = EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
0e795531
AV
852 ebt_check_entry_size_and_hooks, newinfo,
853 &i, &j, &k, &udc_cnt);
1da177e4
LT
854
855 if (ret != 0)
856 return ret;
857
858 if (i != j) {
859 BUGPRINT("nentries does not equal the nr of entries in the "
9d6f229f 860 "(last) chain\n");
1da177e4
LT
861 return -EINVAL;
862 }
863 if (k != newinfo->nentries) {
864 BUGPRINT("Total nentries is wrong\n");
865 return -EINVAL;
866 }
867
1da177e4
LT
868 /* get the location of the udc, put them in an array
869 while we're at it, allocate the chainstack */
870 if (udc_cnt) {
871 /* this will get free'd in do_replace()/ebt_register_table()
872 if an error occurs */
7ad4d2f6 873 newinfo->chainstack =
53b8a315 874 vmalloc(nr_cpu_ids * sizeof(*(newinfo->chainstack)));
1da177e4
LT
875 if (!newinfo->chainstack)
876 return -ENOMEM;
6f912042 877 for_each_possible_cpu(i) {
1da177e4 878 newinfo->chainstack[i] =
18bc89aa 879 vmalloc(udc_cnt * sizeof(*(newinfo->chainstack[0])));
1da177e4
LT
880 if (!newinfo->chainstack[i]) {
881 while (i)
882 vfree(newinfo->chainstack[--i]);
883 vfree(newinfo->chainstack);
884 newinfo->chainstack = NULL;
885 return -ENOMEM;
886 }
887 }
888
18bc89aa 889 cl_s = vmalloc(udc_cnt * sizeof(*cl_s));
1da177e4
LT
890 if (!cl_s)
891 return -ENOMEM;
892 i = 0; /* the i'th udc */
893 EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
177abc34 894 ebt_get_udc_positions, newinfo, &i, cl_s);
1da177e4
LT
895 /* sanity check */
896 if (i != udc_cnt) {
897 BUGPRINT("i != udc_cnt\n");
898 vfree(cl_s);
899 return -EFAULT;
900 }
901 }
902
903 /* Check for loops */
904 for (i = 0; i < NF_BR_NUMHOOKS; i++)
1f072c96 905 if (newinfo->hook_entry[i])
1da177e4
LT
906 if (check_chainloops(newinfo->hook_entry[i],
907 cl_s, udc_cnt, i, newinfo->entries)) {
68d31872 908 vfree(cl_s);
1da177e4
LT
909 return -EINVAL;
910 }
911
96de0e25 912 /* we now know the following (along with E=mc²):
1da177e4
LT
913 - the nr of entries in each chain is right
914 - the size of the allocated space is right
915 - all valid hooks have a corresponding chain
916 - there are no loops
917 - wrong data can still be on the level of a single entry
918 - could be there are jumps to places that are not the
919 beginning of a chain. This can only occur in chains that
920 are not accessible from any base chains, so we don't care. */
921
922 /* used to know what we need to clean up if something goes wrong */
923 i = 0;
924 ret = EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
a83d8e8d 925 ebt_check_entry, net, newinfo, name, &i, cl_s, udc_cnt);
1da177e4
LT
926 if (ret != 0) {
927 EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
f54e9367 928 ebt_cleanup_entry, net, &i);
1da177e4 929 }
68d31872 930 vfree(cl_s);
1da177e4
LT
931 return ret;
932}
933
934/* called under write_lock */
935static void get_counters(struct ebt_counter *oldcounters,
936 struct ebt_counter *counters, unsigned int nentries)
937{
938 int i, cpu;
939 struct ebt_counter *counter_base;
940
941 /* counters of cpu 0 */
942 memcpy(counters, oldcounters,
c8923c6b
DM
943 sizeof(struct ebt_counter) * nentries);
944
1da177e4 945 /* add other counters to those of cpu 0 */
6f912042 946 for_each_possible_cpu(cpu) {
c8923c6b
DM
947 if (cpu == 0)
948 continue;
1da177e4
LT
949 counter_base = COUNTER_BASE(oldcounters, nentries, cpu);
950 for (i = 0; i < nentries; i++) {
951 counters[i].pcnt += counter_base[i].pcnt;
952 counters[i].bcnt += counter_base[i].bcnt;
953 }
954 }
955}
956
957/* replace the table */
511061e2 958static int do_replace(struct net *net, void __user *user, unsigned int len)
1da177e4
LT
959{
960 int ret, i, countersize;
961 struct ebt_table_info *newinfo;
962 struct ebt_replace tmp;
963 struct ebt_table *t;
964 struct ebt_counter *counterstmp = NULL;
965 /* used to be able to unlock earlier */
966 struct ebt_table_info *table;
967
968 if (copy_from_user(&tmp, user, sizeof(tmp)) != 0)
969 return -EFAULT;
970
971 if (len != sizeof(tmp) + tmp.entries_size) {
972 BUGPRINT("Wrong len argument\n");
973 return -EINVAL;
974 }
975
976 if (tmp.entries_size == 0) {
977 BUGPRINT("Entries_size never zero\n");
978 return -EINVAL;
979 }
ee4bb818
KK
980 /* overflow check */
981 if (tmp.nentries >= ((INT_MAX - sizeof(struct ebt_table_info)) / NR_CPUS -
982 SMP_CACHE_BYTES) / sizeof(struct ebt_counter))
983 return -ENOMEM;
984 if (tmp.num_counters >= INT_MAX / sizeof(struct ebt_counter))
985 return -ENOMEM;
986
53b8a315 987 countersize = COUNTER_OFFSET(tmp.nentries) * nr_cpu_ids;
18bc89aa 988 newinfo = vmalloc(sizeof(*newinfo) + countersize);
1da177e4
LT
989 if (!newinfo)
990 return -ENOMEM;
991
992 if (countersize)
993 memset(newinfo->counters, 0, countersize);
994
8b3a7005 995 newinfo->entries = vmalloc(tmp.entries_size);
1da177e4
LT
996 if (!newinfo->entries) {
997 ret = -ENOMEM;
998 goto free_newinfo;
999 }
1000 if (copy_from_user(
1001 newinfo->entries, tmp.entries, tmp.entries_size) != 0) {
1002 BUGPRINT("Couldn't copy entries from userspace\n");
1003 ret = -EFAULT;
1004 goto free_entries;
1005 }
1006
1007 /* the user wants counters back
1008 the check on the size is done later, when we have the lock */
1009 if (tmp.num_counters) {
18bc89aa 1010 counterstmp = vmalloc(tmp.num_counters * sizeof(*counterstmp));
1da177e4
LT
1011 if (!counterstmp) {
1012 ret = -ENOMEM;
1013 goto free_entries;
1014 }
1015 }
1016 else
1017 counterstmp = NULL;
1018
1019 /* this can get initialized by translate_table() */
1020 newinfo->chainstack = NULL;
1bc2326c
AV
1021 ret = ebt_verify_pointers(&tmp, newinfo);
1022 if (ret != 0)
1023 goto free_counterstmp;
1024
a83d8e8d 1025 ret = translate_table(net, tmp.name, newinfo);
1da177e4
LT
1026
1027 if (ret != 0)
1028 goto free_counterstmp;
1029
511061e2 1030 t = find_table_lock(net, tmp.name, &ret, &ebt_mutex);
1da177e4
LT
1031 if (!t) {
1032 ret = -ENOENT;
1033 goto free_iterate;
1034 }
1035
1036 /* the table doesn't like it */
1037 if (t->check && (ret = t->check(newinfo, tmp.valid_hooks)))
1038 goto free_unlock;
1039
1040 if (tmp.num_counters && tmp.num_counters != t->private->nentries) {
1041 BUGPRINT("Wrong nr. of counters requested\n");
1042 ret = -EINVAL;
1043 goto free_unlock;
1044 }
1045
1046 /* we have the mutex lock, so no danger in reading this pointer */
1047 table = t->private;
1048 /* make sure the table can only be rmmod'ed if it contains no rules */
1049 if (!table->nentries && newinfo->nentries && !try_module_get(t->me)) {
1050 ret = -ENOENT;
1051 goto free_unlock;
1052 } else if (table->nentries && !newinfo->nentries)
1053 module_put(t->me);
1054 /* we need an atomic snapshot of the counters */
1055 write_lock_bh(&t->lock);
1056 if (tmp.num_counters)
1057 get_counters(t->private->counters, counterstmp,
1058 t->private->nentries);
1059
1060 t->private = newinfo;
1061 write_unlock_bh(&t->lock);
57b47a53 1062 mutex_unlock(&ebt_mutex);
1da177e4
LT
1063 /* so, a user can change the chains while having messed up her counter
1064 allocation. Only reason why this is done is because this way the lock
1065 is held only once, while this doesn't bring the kernel into a
1066 dangerous state. */
1067 if (tmp.num_counters &&
1068 copy_to_user(tmp.counters, counterstmp,
1069 tmp.num_counters * sizeof(struct ebt_counter))) {
1070 BUGPRINT("Couldn't copy counters to userspace\n");
1071 ret = -EFAULT;
1072 }
1073 else
1074 ret = 0;
1075
1076 /* decrease module count and free resources */
1077 EBT_ENTRY_ITERATE(table->entries, table->entries_size,
f54e9367 1078 ebt_cleanup_entry, net, NULL);
1da177e4
LT
1079
1080 vfree(table->entries);
1081 if (table->chainstack) {
6f912042 1082 for_each_possible_cpu(i)
1da177e4
LT
1083 vfree(table->chainstack[i]);
1084 vfree(table->chainstack);
1085 }
1086 vfree(table);
1087
68d31872 1088 vfree(counterstmp);
1da177e4
LT
1089 return ret;
1090
1091free_unlock:
57b47a53 1092 mutex_unlock(&ebt_mutex);
1da177e4
LT
1093free_iterate:
1094 EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
f54e9367 1095 ebt_cleanup_entry, net, NULL);
1da177e4 1096free_counterstmp:
68d31872 1097 vfree(counterstmp);
1da177e4
LT
1098 /* can be initialized in translate_table() */
1099 if (newinfo->chainstack) {
6f912042 1100 for_each_possible_cpu(i)
1da177e4
LT
1101 vfree(newinfo->chainstack[i]);
1102 vfree(newinfo->chainstack);
1103 }
1104free_entries:
68d31872 1105 vfree(newinfo->entries);
1da177e4 1106free_newinfo:
68d31872 1107 vfree(newinfo);
1da177e4
LT
1108 return ret;
1109}
1110
35aad0ff
JE
1111struct ebt_table *
1112ebt_register_table(struct net *net, const struct ebt_table *input_table)
1da177e4
LT
1113{
1114 struct ebt_table_info *newinfo;
35aad0ff 1115 struct ebt_table *t, *table;
1e419cd9 1116 struct ebt_replace_kernel *repl;
1da177e4 1117 int ret, i, countersize;
df07a81e 1118 void *p;
1da177e4 1119
35aad0ff
JE
1120 if (input_table == NULL || (repl = input_table->table) == NULL ||
1121 repl->entries == 0 || repl->entries_size == 0 ||
1122 repl->counters != NULL || input_table->private != NULL) {
1da177e4 1123 BUGPRINT("Bad table data for ebt_register_table!!!\n");
6beceee5
AD
1124 return ERR_PTR(-EINVAL);
1125 }
1126
1127 /* Don't add one table to multiple lists. */
35aad0ff 1128 table = kmemdup(input_table, sizeof(struct ebt_table), GFP_KERNEL);
6beceee5
AD
1129 if (!table) {
1130 ret = -ENOMEM;
1131 goto out;
1da177e4
LT
1132 }
1133
53b8a315 1134 countersize = COUNTER_OFFSET(repl->nentries) * nr_cpu_ids;
18bc89aa 1135 newinfo = vmalloc(sizeof(*newinfo) + countersize);
1da177e4
LT
1136 ret = -ENOMEM;
1137 if (!newinfo)
6beceee5 1138 goto free_table;
1da177e4 1139
df07a81e
AV
1140 p = vmalloc(repl->entries_size);
1141 if (!p)
1da177e4
LT
1142 goto free_newinfo;
1143
df07a81e
AV
1144 memcpy(p, repl->entries, repl->entries_size);
1145 newinfo->entries = p;
1146
1147 newinfo->entries_size = repl->entries_size;
1148 newinfo->nentries = repl->nentries;
1da177e4
LT
1149
1150 if (countersize)
1151 memset(newinfo->counters, 0, countersize);
1152
1153 /* fill in newinfo and parse the entries */
1154 newinfo->chainstack = NULL;
df07a81e
AV
1155 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
1156 if ((repl->valid_hooks & (1 << i)) == 0)
1157 newinfo->hook_entry[i] = NULL;
1158 else
1159 newinfo->hook_entry[i] = p +
1160 ((char *)repl->hook_entry[i] - repl->entries);
1161 }
a83d8e8d 1162 ret = translate_table(net, repl->name, newinfo);
1da177e4
LT
1163 if (ret != 0) {
1164 BUGPRINT("Translate_table failed\n");
1165 goto free_chainstack;
1166 }
1167
1168 if (table->check && table->check(newinfo, table->valid_hooks)) {
1169 BUGPRINT("The table doesn't like its own initial data, lol\n");
6beceee5 1170 return ERR_PTR(-EINVAL);
1da177e4
LT
1171 }
1172
1173 table->private = newinfo;
1174 rwlock_init(&table->lock);
57b47a53 1175 ret = mutex_lock_interruptible(&ebt_mutex);
1da177e4
LT
1176 if (ret != 0)
1177 goto free_chainstack;
1178
511061e2 1179 list_for_each_entry(t, &net->xt.tables[NFPROTO_BRIDGE], list) {
df0933dc
PM
1180 if (strcmp(t->name, table->name) == 0) {
1181 ret = -EEXIST;
1182 BUGPRINT("Table name already exists\n");
1183 goto free_unlock;
1184 }
1da177e4
LT
1185 }
1186
1187 /* Hold a reference count if the chains aren't empty */
1188 if (newinfo->nentries && !try_module_get(table->me)) {
1189 ret = -ENOENT;
1190 goto free_unlock;
1191 }
511061e2 1192 list_add(&table->list, &net->xt.tables[NFPROTO_BRIDGE]);
57b47a53 1193 mutex_unlock(&ebt_mutex);
6beceee5 1194 return table;
1da177e4 1195free_unlock:
57b47a53 1196 mutex_unlock(&ebt_mutex);
1da177e4
LT
1197free_chainstack:
1198 if (newinfo->chainstack) {
6f912042 1199 for_each_possible_cpu(i)
1da177e4
LT
1200 vfree(newinfo->chainstack[i]);
1201 vfree(newinfo->chainstack);
1202 }
1203 vfree(newinfo->entries);
1204free_newinfo:
1205 vfree(newinfo);
6beceee5
AD
1206free_table:
1207 kfree(table);
1208out:
1209 return ERR_PTR(ret);
1da177e4
LT
1210}
1211
f54e9367 1212void ebt_unregister_table(struct net *net, struct ebt_table *table)
1da177e4
LT
1213{
1214 int i;
1215
1216 if (!table) {
1217 BUGPRINT("Request to unregister NULL table!!!\n");
1218 return;
1219 }
57b47a53 1220 mutex_lock(&ebt_mutex);
df0933dc 1221 list_del(&table->list);
57b47a53 1222 mutex_unlock(&ebt_mutex);
dbcdf85a 1223 EBT_ENTRY_ITERATE(table->private->entries, table->private->entries_size,
f54e9367 1224 ebt_cleanup_entry, net, NULL);
dbcdf85a
AD
1225 if (table->private->nentries)
1226 module_put(table->me);
68d31872 1227 vfree(table->private->entries);
1da177e4 1228 if (table->private->chainstack) {
6f912042 1229 for_each_possible_cpu(i)
1da177e4
LT
1230 vfree(table->private->chainstack[i]);
1231 vfree(table->private->chainstack);
1232 }
1233 vfree(table->private);
6beceee5 1234 kfree(table);
1da177e4
LT
1235}
1236
1237/* userspace just supplied us with counters */
511061e2 1238static int update_counters(struct net *net, void __user *user, unsigned int len)
1da177e4
LT
1239{
1240 int i, ret;
1241 struct ebt_counter *tmp;
1242 struct ebt_replace hlp;
1243 struct ebt_table *t;
1244
1245 if (copy_from_user(&hlp, user, sizeof(hlp)))
1246 return -EFAULT;
1247
1248 if (len != sizeof(hlp) + hlp.num_counters * sizeof(struct ebt_counter))
1249 return -EINVAL;
1250 if (hlp.num_counters == 0)
1251 return -EINVAL;
1252
18bc89aa 1253 if (!(tmp = vmalloc(hlp.num_counters * sizeof(*tmp)))) {
1da177e4
LT
1254 MEMPRINT("Update_counters && nomemory\n");
1255 return -ENOMEM;
1256 }
1257
511061e2 1258 t = find_table_lock(net, hlp.name, &ret, &ebt_mutex);
1da177e4
LT
1259 if (!t)
1260 goto free_tmp;
1261
1262 if (hlp.num_counters != t->private->nentries) {
1263 BUGPRINT("Wrong nr of counters\n");
1264 ret = -EINVAL;
1265 goto unlock_mutex;
1266 }
1267
1268 if ( copy_from_user(tmp, hlp.counters,
1269 hlp.num_counters * sizeof(struct ebt_counter)) ) {
1270 BUGPRINT("Updata_counters && !cfu\n");
1271 ret = -EFAULT;
1272 goto unlock_mutex;
1273 }
1274
1275 /* we want an atomic add of the counters */
1276 write_lock_bh(&t->lock);
1277
1278 /* we add to the counters of the first cpu */
1279 for (i = 0; i < hlp.num_counters; i++) {
1280 t->private->counters[i].pcnt += tmp[i].pcnt;
1281 t->private->counters[i].bcnt += tmp[i].bcnt;
1282 }
1283
1284 write_unlock_bh(&t->lock);
1285 ret = 0;
1286unlock_mutex:
57b47a53 1287 mutex_unlock(&ebt_mutex);
1da177e4
LT
1288free_tmp:
1289 vfree(tmp);
1290 return ret;
1291}
1292
1293static inline int ebt_make_matchname(struct ebt_entry_match *m,
1e419cd9 1294 char *base, char __user *ubase)
1da177e4 1295{
1e419cd9 1296 char __user *hlp = ubase + ((char *)m - base);
1da177e4
LT
1297 if (copy_to_user(hlp, m->u.match->name, EBT_FUNCTION_MAXNAMELEN))
1298 return -EFAULT;
1299 return 0;
1300}
1301
1302static inline int ebt_make_watchername(struct ebt_entry_watcher *w,
1e419cd9 1303 char *base, char __user *ubase)
1da177e4 1304{
1e419cd9 1305 char __user *hlp = ubase + ((char *)w - base);
1da177e4
LT
1306 if (copy_to_user(hlp , w->u.watcher->name, EBT_FUNCTION_MAXNAMELEN))
1307 return -EFAULT;
1308 return 0;
1309}
1310
1e419cd9 1311static inline int ebt_make_names(struct ebt_entry *e, char *base, char __user *ubase)
1da177e4
LT
1312{
1313 int ret;
1e419cd9 1314 char __user *hlp;
1da177e4
LT
1315 struct ebt_entry_target *t;
1316
40642f95 1317 if (e->bitmask == 0)
1da177e4
LT
1318 return 0;
1319
1e419cd9 1320 hlp = ubase + (((char *)e + e->target_offset) - base);
1da177e4 1321 t = (struct ebt_entry_target *)(((char *)e) + e->target_offset);
9d6f229f 1322
1da177e4
LT
1323 ret = EBT_MATCH_ITERATE(e, ebt_make_matchname, base, ubase);
1324 if (ret != 0)
1325 return ret;
1326 ret = EBT_WATCHER_ITERATE(e, ebt_make_watchername, base, ubase);
1327 if (ret != 0)
1328 return ret;
1329 if (copy_to_user(hlp, t->u.target->name, EBT_FUNCTION_MAXNAMELEN))
1330 return -EFAULT;
1331 return 0;
1332}
1333
57b47a53 1334/* called with ebt_mutex locked */
1da177e4
LT
1335static int copy_everything_to_user(struct ebt_table *t, void __user *user,
1336 int *len, int cmd)
1337{
1338 struct ebt_replace tmp;
1339 struct ebt_counter *counterstmp, *oldcounters;
1340 unsigned int entries_size, nentries;
1341 char *entries;
1342
1343 if (cmd == EBT_SO_GET_ENTRIES) {
1344 entries_size = t->private->entries_size;
1345 nentries = t->private->nentries;
1346 entries = t->private->entries;
1347 oldcounters = t->private->counters;
1348 } else {
1349 entries_size = t->table->entries_size;
1350 nentries = t->table->nentries;
1351 entries = t->table->entries;
1352 oldcounters = t->table->counters;
1353 }
1354
1355 if (copy_from_user(&tmp, user, sizeof(tmp))) {
1356 BUGPRINT("Cfu didn't work\n");
1357 return -EFAULT;
1358 }
1359
1360 if (*len != sizeof(struct ebt_replace) + entries_size +
1361 (tmp.num_counters? nentries * sizeof(struct ebt_counter): 0)) {
1362 BUGPRINT("Wrong size\n");
1363 return -EINVAL;
1364 }
1365
1366 if (tmp.nentries != nentries) {
1367 BUGPRINT("Nentries wrong\n");
1368 return -EINVAL;
1369 }
1370
1371 if (tmp.entries_size != entries_size) {
1372 BUGPRINT("Wrong size\n");
1373 return -EINVAL;
1374 }
1375
1376 /* userspace might not need the counters */
1377 if (tmp.num_counters) {
1378 if (tmp.num_counters != nentries) {
1379 BUGPRINT("Num_counters wrong\n");
1380 return -EINVAL;
1381 }
18bc89aa 1382 counterstmp = vmalloc(nentries * sizeof(*counterstmp));
1da177e4
LT
1383 if (!counterstmp) {
1384 MEMPRINT("Couldn't copy counters, out of memory\n");
1385 return -ENOMEM;
1386 }
1387 write_lock_bh(&t->lock);
1388 get_counters(oldcounters, counterstmp, nentries);
1389 write_unlock_bh(&t->lock);
1390
1391 if (copy_to_user(tmp.counters, counterstmp,
1392 nentries * sizeof(struct ebt_counter))) {
1393 BUGPRINT("Couldn't copy counters to userspace\n");
1394 vfree(counterstmp);
1395 return -EFAULT;
1396 }
1397 vfree(counterstmp);
1398 }
1399
1400 if (copy_to_user(tmp.entries, entries, entries_size)) {
1401 BUGPRINT("Couldn't copy entries to userspace\n");
1402 return -EFAULT;
1403 }
1404 /* set the match/watcher/target names right */
1405 return EBT_ENTRY_ITERATE(entries, entries_size,
1406 ebt_make_names, entries, tmp.entries);
1407}
1408
1409static int do_ebt_set_ctl(struct sock *sk,
1410 int cmd, void __user *user, unsigned int len)
1411{
1412 int ret;
1413
1414 switch(cmd) {
1415 case EBT_SO_SET_ENTRIES:
511061e2 1416 ret = do_replace(sock_net(sk), user, len);
1da177e4
LT
1417 break;
1418 case EBT_SO_SET_COUNTERS:
511061e2 1419 ret = update_counters(sock_net(sk), user, len);
1da177e4
LT
1420 break;
1421 default:
1422 ret = -EINVAL;
1423 }
1424 return ret;
1425}
1426
1427static int do_ebt_get_ctl(struct sock *sk, int cmd, void __user *user, int *len)
1428{
1429 int ret;
1430 struct ebt_replace tmp;
1431 struct ebt_table *t;
1432
1433 if (copy_from_user(&tmp, user, sizeof(tmp)))
1434 return -EFAULT;
1435
511061e2 1436 t = find_table_lock(sock_net(sk), tmp.name, &ret, &ebt_mutex);
1da177e4
LT
1437 if (!t)
1438 return ret;
1439
1440 switch(cmd) {
1441 case EBT_SO_GET_INFO:
1442 case EBT_SO_GET_INIT_INFO:
1443 if (*len != sizeof(struct ebt_replace)){
1444 ret = -EINVAL;
57b47a53 1445 mutex_unlock(&ebt_mutex);
1da177e4
LT
1446 break;
1447 }
1448 if (cmd == EBT_SO_GET_INFO) {
1449 tmp.nentries = t->private->nentries;
1450 tmp.entries_size = t->private->entries_size;
1451 tmp.valid_hooks = t->valid_hooks;
1452 } else {
1453 tmp.nentries = t->table->nentries;
1454 tmp.entries_size = t->table->entries_size;
1455 tmp.valid_hooks = t->table->valid_hooks;
1456 }
57b47a53 1457 mutex_unlock(&ebt_mutex);
1da177e4
LT
1458 if (copy_to_user(user, &tmp, *len) != 0){
1459 BUGPRINT("c2u Didn't work\n");
1460 ret = -EFAULT;
1461 break;
1462 }
1463 ret = 0;
1464 break;
1465
1466 case EBT_SO_GET_ENTRIES:
1467 case EBT_SO_GET_INIT_ENTRIES:
1468 ret = copy_everything_to_user(t, user, len, cmd);
57b47a53 1469 mutex_unlock(&ebt_mutex);
1da177e4
LT
1470 break;
1471
1472 default:
57b47a53 1473 mutex_unlock(&ebt_mutex);
1da177e4
LT
1474 ret = -EINVAL;
1475 }
1476
1477 return ret;
1478}
1479
1480static struct nf_sockopt_ops ebt_sockopts =
74ca4e5a
AM
1481{
1482 .pf = PF_INET,
1483 .set_optmin = EBT_BASE_CTL,
1484 .set_optmax = EBT_SO_SET_MAX + 1,
1485 .set = do_ebt_set_ctl,
1486 .get_optmin = EBT_BASE_CTL,
1487 .get_optmax = EBT_SO_GET_MAX + 1,
1488 .get = do_ebt_get_ctl,
16fcec35 1489 .owner = THIS_MODULE,
1da177e4
LT
1490};
1491
65b4b4e8 1492static int __init ebtables_init(void)
1da177e4
LT
1493{
1494 int ret;
1495
043ef46c
JE
1496 ret = xt_register_target(&ebt_standard_target);
1497 if (ret < 0)
1da177e4 1498 return ret;
043ef46c
JE
1499 ret = nf_register_sockopt(&ebt_sockopts);
1500 if (ret < 0) {
1501 xt_unregister_target(&ebt_standard_target);
1502 return ret;
1503 }
1da177e4 1504
a887c1c1 1505 printk(KERN_INFO "Ebtables v2.0 registered\n");
1da177e4
LT
1506 return 0;
1507}
1508
65b4b4e8 1509static void __exit ebtables_fini(void)
1da177e4
LT
1510{
1511 nf_unregister_sockopt(&ebt_sockopts);
043ef46c 1512 xt_unregister_target(&ebt_standard_target);
a887c1c1 1513 printk(KERN_INFO "Ebtables v2.0 unregistered\n");
1da177e4
LT
1514}
1515
1516EXPORT_SYMBOL(ebt_register_table);
1517EXPORT_SYMBOL(ebt_unregister_table);
1da177e4 1518EXPORT_SYMBOL(ebt_do_table);
65b4b4e8
AM
1519module_init(ebtables_init);
1520module_exit(ebtables_fini);
1da177e4 1521MODULE_LICENSE("GPL");