blob: f0d9ffd4c916fd1f15e4c2be88336e06a318f63b [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
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
18/* used for print_string */
19#include <linux/sched.h>
20#include <linux/tty.h>
21
22#include <linux/kmod.h>
23#include <linux/module.h>
24#include <linux/vmalloc.h>
25#include <linux/netfilter_bridge/ebtables.h>
26#include <linux/spinlock.h>
Patrick McHardydf0933d2006-09-20 11:57:53 -070027#include <linux/mutex.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070028#include <asm/uaccess.h>
29#include <linux/smp.h>
David S. Millerc8923c62005-10-13 14:41:23 -070030#include <linux/cpumask.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070031#include <net/sock.h>
32/* needed for logical [in,out]-dev filtering */
33#include "../br_private.h"
34
Linus Torvalds1da177e2005-04-16 15:20:36 -070035#define BUGPRINT(format, args...) printk("kernel msg: ebtables bug: please "\
36 "report to author: "format, ## args)
37/* #define BUGPRINT(format, args...) */
Linus Torvalds1da177e2005-04-16 15:20:36 -070038#define MEMPRINT(format, args...) printk("kernel msg: ebtables "\
39 ": out of memory: "format, ## args)
40/* #define MEMPRINT(format, args...) */
41
42
43
44/*
45 * Each cpu has its own set of counters, so there is no need for write_lock in
46 * the softirq
47 * For reading or updating the counters, the user context needs to
48 * get a write_lock
49 */
50
51/* The size of each set of counters is altered to get cache alignment */
52#define SMP_ALIGN(x) (((x) + SMP_CACHE_BYTES-1) & ~(SMP_CACHE_BYTES-1))
53#define COUNTER_OFFSET(n) (SMP_ALIGN(n * sizeof(struct ebt_counter)))
54#define COUNTER_BASE(c, n, cpu) ((struct ebt_counter *)(((char *)c) + \
55 COUNTER_OFFSET(n) * cpu))
56
57
58
Ingo Molnar57b47a52006-03-20 22:35:41 -080059static DEFINE_MUTEX(ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -070060static LIST_HEAD(ebt_tables);
61static LIST_HEAD(ebt_targets);
62static LIST_HEAD(ebt_matches);
63static LIST_HEAD(ebt_watchers);
64
65static struct ebt_target ebt_standard_target =
66{ {NULL, NULL}, EBT_STANDARD_TARGET, NULL, NULL, NULL, NULL};
67
68static inline int ebt_do_watcher (struct ebt_entry_watcher *w,
69 const struct sk_buff *skb, unsigned int hooknr, const struct net_device *in,
70 const struct net_device *out)
71{
72 w->u.watcher->watcher(skb, hooknr, in, out, w->data,
73 w->watcher_size);
74 /* watchers don't give a verdict */
75 return 0;
76}
77
78static inline int ebt_do_match (struct ebt_entry_match *m,
79 const struct sk_buff *skb, const struct net_device *in,
80 const struct net_device *out)
81{
82 return m->u.match->match(skb, in, out, m->data,
83 m->match_size);
84}
85
86static inline int ebt_dev_check(char *entry, const struct net_device *device)
87{
88 int i = 0;
Meelis Roos6f5b7ef2006-11-01 18:07:27 -080089 const char *devname = device->name;
Linus Torvalds1da177e2005-04-16 15:20:36 -070090
91 if (*entry == '\0')
92 return 0;
93 if (!device)
94 return 1;
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
145/* Do some firewalling */
146unsigned int ebt_do_table (unsigned int hook, struct sk_buff **pskb,
147 const struct net_device *in, const struct net_device *out,
148 struct ebt_table *table)
149{
150 int i, nentries;
151 struct ebt_entry *point;
152 struct ebt_counter *counter_base, *cb_base;
153 struct ebt_entry_target *t;
154 int verdict, sp = 0;
155 struct ebt_chainstack *cs;
156 struct ebt_entries *chaininfo;
157 char *base;
158 struct ebt_table_info *private;
159
160 read_lock_bh(&table->lock);
161 private = table->private;
162 cb_base = COUNTER_BASE(private->counters, private->nentries,
163 smp_processor_id());
164 if (private->chainstack)
165 cs = private->chainstack[smp_processor_id()];
166 else
167 cs = NULL;
168 chaininfo = private->hook_entry[hook];
169 nentries = private->hook_entry[hook]->nentries;
170 point = (struct ebt_entry *)(private->hook_entry[hook]->data);
171 counter_base = cb_base + private->hook_entry[hook]->counter_offset;
172 /* base for chain jumps */
173 base = private->entries;
174 i = 0;
175 while (i < nentries) {
176 if (ebt_basic_match(point, eth_hdr(*pskb), in, out))
177 goto letscontinue;
178
179 if (EBT_MATCH_ITERATE(point, ebt_do_match, *pskb, in, out) != 0)
180 goto letscontinue;
181
182 /* increase counter */
183 (*(counter_base + i)).pcnt++;
184 (*(counter_base + i)).bcnt+=(**pskb).len;
185
186 /* these should only watch: not modify, nor tell us
187 what to do with the packet */
188 EBT_WATCHER_ITERATE(point, ebt_do_watcher, *pskb, hook, in,
189 out);
190
191 t = (struct ebt_entry_target *)
192 (((char *)point) + point->target_offset);
193 /* standard target */
194 if (!t->u.target->target)
195 verdict = ((struct ebt_standard_target *)t)->verdict;
196 else
197 verdict = t->u.target->target(pskb, hook,
198 in, out, t->data, t->target_size);
199 if (verdict == EBT_ACCEPT) {
200 read_unlock_bh(&table->lock);
201 return NF_ACCEPT;
202 }
203 if (verdict == EBT_DROP) {
204 read_unlock_bh(&table->lock);
205 return NF_DROP;
206 }
207 if (verdict == EBT_RETURN) {
208letsreturn:
209#ifdef CONFIG_NETFILTER_DEBUG
210 if (sp == 0) {
211 BUGPRINT("RETURN on base chain");
212 /* act like this is EBT_CONTINUE */
213 goto letscontinue;
214 }
215#endif
216 sp--;
217 /* put all the local variables right */
218 i = cs[sp].n;
219 chaininfo = cs[sp].chaininfo;
220 nentries = chaininfo->nentries;
221 point = cs[sp].e;
222 counter_base = cb_base +
223 chaininfo->counter_offset;
224 continue;
225 }
226 if (verdict == EBT_CONTINUE)
227 goto letscontinue;
228#ifdef CONFIG_NETFILTER_DEBUG
229 if (verdict < 0) {
230 BUGPRINT("bogus standard verdict\n");
231 read_unlock_bh(&table->lock);
232 return NF_DROP;
233 }
234#endif
235 /* jump to a udc */
236 cs[sp].n = i + 1;
237 cs[sp].chaininfo = chaininfo;
238 cs[sp].e = (struct ebt_entry *)
239 (((char *)point) + point->next_offset);
240 i = 0;
241 chaininfo = (struct ebt_entries *) (base + verdict);
242#ifdef CONFIG_NETFILTER_DEBUG
243 if (chaininfo->distinguisher) {
244 BUGPRINT("jump to non-chain\n");
245 read_unlock_bh(&table->lock);
246 return NF_DROP;
247 }
248#endif
249 nentries = chaininfo->nentries;
250 point = (struct ebt_entry *)chaininfo->data;
251 counter_base = cb_base + chaininfo->counter_offset;
252 sp++;
253 continue;
254letscontinue:
255 point = (struct ebt_entry *)
256 (((char *)point) + point->next_offset);
257 i++;
258 }
259
260 /* I actually like this :) */
261 if (chaininfo->policy == EBT_RETURN)
262 goto letsreturn;
263 if (chaininfo->policy == EBT_ACCEPT) {
264 read_unlock_bh(&table->lock);
265 return NF_ACCEPT;
266 }
267 read_unlock_bh(&table->lock);
268 return NF_DROP;
269}
270
271/* If it succeeds, returns element and locks mutex */
272static inline void *
273find_inlist_lock_noload(struct list_head *head, const char *name, int *error,
Ingo Molnar57b47a52006-03-20 22:35:41 -0800274 struct mutex *mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700275{
Patrick McHardydf0933d2006-09-20 11:57:53 -0700276 struct {
277 struct list_head list;
278 char name[EBT_FUNCTION_MAXNAMELEN];
279 } *e;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700280
Ingo Molnar57b47a52006-03-20 22:35:41 -0800281 *error = mutex_lock_interruptible(mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700282 if (*error != 0)
283 return NULL;
284
Patrick McHardydf0933d2006-09-20 11:57:53 -0700285 list_for_each_entry(e, head, list) {
286 if (strcmp(e->name, name) == 0)
287 return e;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700288 }
Patrick McHardydf0933d2006-09-20 11:57:53 -0700289 *error = -ENOENT;
290 mutex_unlock(mutex);
291 return NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700292}
293
294#ifndef CONFIG_KMOD
295#define find_inlist_lock(h,n,p,e,m) find_inlist_lock_noload((h),(n),(e),(m))
296#else
297static void *
298find_inlist_lock(struct list_head *head, const char *name, const char *prefix,
Ingo Molnar57b47a52006-03-20 22:35:41 -0800299 int *error, struct mutex *mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700300{
301 void *ret;
302
303 ret = find_inlist_lock_noload(head, name, error, mutex);
304 if (!ret) {
305 request_module("%s%s", prefix, name);
306 ret = find_inlist_lock_noload(head, name, error, mutex);
307 }
308 return ret;
309}
310#endif
311
312static inline struct ebt_table *
Ingo Molnar57b47a52006-03-20 22:35:41 -0800313find_table_lock(const char *name, int *error, struct mutex *mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700314{
315 return find_inlist_lock(&ebt_tables, name, "ebtable_", error, mutex);
316}
317
318static inline struct ebt_match *
Ingo Molnar57b47a52006-03-20 22:35:41 -0800319find_match_lock(const char *name, int *error, struct mutex *mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700320{
321 return find_inlist_lock(&ebt_matches, name, "ebt_", error, mutex);
322}
323
324static inline struct ebt_watcher *
Ingo Molnar57b47a52006-03-20 22:35:41 -0800325find_watcher_lock(const char *name, int *error, struct mutex *mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700326{
327 return find_inlist_lock(&ebt_watchers, name, "ebt_", error, mutex);
328}
329
330static inline struct ebt_target *
Ingo Molnar57b47a52006-03-20 22:35:41 -0800331find_target_lock(const char *name, int *error, struct mutex *mutex)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700332{
333 return find_inlist_lock(&ebt_targets, name, "ebt_", error, mutex);
334}
335
336static inline int
337ebt_check_match(struct ebt_entry_match *m, struct ebt_entry *e,
338 const char *name, unsigned int hookmask, unsigned int *cnt)
339{
340 struct ebt_match *match;
Al Viro14197d52006-11-30 19:25:21 -0800341 size_t left = ((char *)e + e->watchers_offset) - (char *)m;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700342 int ret;
343
Al Viro14197d52006-11-30 19:25:21 -0800344 if (left < sizeof(struct ebt_entry_match) ||
345 left - sizeof(struct ebt_entry_match) < m->match_size)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700346 return -EINVAL;
347 match = find_match_lock(m->u.name, &ret, &ebt_mutex);
348 if (!match)
349 return ret;
350 m->u.match = match;
351 if (!try_module_get(match->me)) {
Ingo Molnar57b47a52006-03-20 22:35:41 -0800352 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700353 return -ENOENT;
354 }
Ingo Molnar57b47a52006-03-20 22:35:41 -0800355 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700356 if (match->check &&
357 match->check(name, hookmask, e, m->data, m->match_size) != 0) {
358 BUGPRINT("match->check failed\n");
359 module_put(match->me);
360 return -EINVAL;
361 }
362 (*cnt)++;
363 return 0;
364}
365
366static inline int
367ebt_check_watcher(struct ebt_entry_watcher *w, struct ebt_entry *e,
368 const char *name, unsigned int hookmask, unsigned int *cnt)
369{
370 struct ebt_watcher *watcher;
Al Viro14197d52006-11-30 19:25:21 -0800371 size_t left = ((char *)e + e->target_offset) - (char *)w;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700372 int ret;
373
Al Viro14197d52006-11-30 19:25:21 -0800374 if (left < sizeof(struct ebt_entry_watcher) ||
375 left - sizeof(struct ebt_entry_watcher) < w->watcher_size)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700376 return -EINVAL;
377 watcher = find_watcher_lock(w->u.name, &ret, &ebt_mutex);
378 if (!watcher)
379 return ret;
380 w->u.watcher = watcher;
381 if (!try_module_get(watcher->me)) {
Ingo Molnar57b47a52006-03-20 22:35:41 -0800382 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700383 return -ENOENT;
384 }
Ingo Molnar57b47a52006-03-20 22:35:41 -0800385 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700386 if (watcher->check &&
387 watcher->check(name, hookmask, e, w->data, w->watcher_size) != 0) {
388 BUGPRINT("watcher->check failed\n");
389 module_put(watcher->me);
390 return -EINVAL;
391 }
392 (*cnt)++;
393 return 0;
394}
395
Al Viro70fe9af2006-11-30 19:26:14 -0800396static int ebt_verify_pointers(struct ebt_replace *repl,
397 struct ebt_table_info *newinfo)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700398{
Al Viro70fe9af2006-11-30 19:26:14 -0800399 unsigned int limit = repl->entries_size;
400 unsigned int valid_hooks = repl->valid_hooks;
401 unsigned int offset = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700402 int i;
403
Al Viroe4fd77d2006-11-30 19:26:35 -0800404 for (i = 0; i < NF_BR_NUMHOOKS; i++)
405 newinfo->hook_entry[i] = NULL;
406
407 newinfo->entries_size = repl->entries_size;
408 newinfo->nentries = repl->nentries;
409
Al Viro70fe9af2006-11-30 19:26:14 -0800410 while (offset < limit) {
411 size_t left = limit - offset;
412 struct ebt_entry *e = (void *)newinfo->entries + offset;
Al Virobb2ef252006-11-30 19:22:42 -0800413
Al Viro70fe9af2006-11-30 19:26:14 -0800414 if (left < sizeof(unsigned int))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700415 break;
Al Viro22b440b2006-11-30 19:25:51 -0800416
Al Viro70fe9af2006-11-30 19:26:14 -0800417 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
418 if ((valid_hooks & (1 << i)) == 0)
419 continue;
420 if ((char *)repl->hook_entry[i] == repl->entries + offset)
421 break;
422 }
423
424 if (i != NF_BR_NUMHOOKS || !(e->bitmask & EBT_ENTRY_OR_ENTRIES)) {
425 if (e->bitmask != 0) {
426 /* we make userspace set this right,
427 so there is no misunderstanding */
428 BUGPRINT("EBT_ENTRY_OR_ENTRIES shouldn't be set "
429 "in distinguisher\n");
430 return -EINVAL;
431 }
432 if (i != NF_BR_NUMHOOKS)
433 newinfo->hook_entry[i] = (struct ebt_entries *)e;
434 if (left < sizeof(struct ebt_entries))
435 break;
436 offset += sizeof(struct ebt_entries);
437 } else {
438 if (left < sizeof(struct ebt_entry))
439 break;
440 if (left < e->next_offset)
441 break;
442 offset += e->next_offset;
443 }
444 }
445 if (offset != limit) {
446 BUGPRINT("entries_size too small\n");
447 return -EINVAL;
448 }
Al Viroe4fd77d2006-11-30 19:26:35 -0800449
450 /* check if all valid hooks have a chain */
451 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
452 if (!newinfo->hook_entry[i] &&
453 (valid_hooks & (1 << i))) {
454 BUGPRINT("Valid hook without chain\n");
455 return -EINVAL;
456 }
457 }
Al Viro70fe9af2006-11-30 19:26:14 -0800458 return 0;
Al Viro22b440b2006-11-30 19:25:51 -0800459}
460
461/*
462 * this one is very careful, as it is the first function
463 * to parse the userspace data
464 */
465static inline int
466ebt_check_entry_size_and_hooks(struct ebt_entry *e,
Al Viro0e795532006-11-30 19:27:13 -0800467 struct ebt_table_info *newinfo,
468 unsigned int *n, unsigned int *cnt,
469 unsigned int *totalcnt, unsigned int *udc_cnt)
Al Viro22b440b2006-11-30 19:25:51 -0800470{
Al Viro22b440b2006-11-30 19:25:51 -0800471 int i;
472
473 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
Al Viro0e795532006-11-30 19:27:13 -0800474 if ((void *)e == (void *)newinfo->hook_entry[i])
Al Viro22b440b2006-11-30 19:25:51 -0800475 break;
476 }
477 /* beginning of a new chain
478 if i == NF_BR_NUMHOOKS it must be a user defined chain */
479 if (i != NF_BR_NUMHOOKS || !e->bitmask) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700480 /* this checks if the previous chain has as many entries
481 as it said it has */
482 if (*n != *cnt) {
483 BUGPRINT("nentries does not equal the nr of entries "
484 "in the chain\n");
485 return -EINVAL;
486 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700487 if (((struct ebt_entries *)e)->policy != EBT_DROP &&
488 ((struct ebt_entries *)e)->policy != EBT_ACCEPT) {
489 /* only RETURN from udc */
490 if (i != NF_BR_NUMHOOKS ||
491 ((struct ebt_entries *)e)->policy != EBT_RETURN) {
492 BUGPRINT("bad policy\n");
493 return -EINVAL;
494 }
495 }
496 if (i == NF_BR_NUMHOOKS) /* it's a user defined chain */
497 (*udc_cnt)++;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700498 if (((struct ebt_entries *)e)->counter_offset != *totalcnt) {
499 BUGPRINT("counter_offset != totalcnt");
500 return -EINVAL;
501 }
502 *n = ((struct ebt_entries *)e)->nentries;
503 *cnt = 0;
504 return 0;
505 }
506 /* a plain old entry, heh */
507 if (sizeof(struct ebt_entry) > e->watchers_offset ||
508 e->watchers_offset > e->target_offset ||
509 e->target_offset >= e->next_offset) {
510 BUGPRINT("entry offsets not in right order\n");
511 return -EINVAL;
512 }
513 /* this is not checked anywhere else */
514 if (e->next_offset - e->target_offset < sizeof(struct ebt_entry_target)) {
515 BUGPRINT("target size too small\n");
516 return -EINVAL;
517 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700518 (*cnt)++;
519 (*totalcnt)++;
520 return 0;
521}
522
523struct ebt_cl_stack
524{
525 struct ebt_chainstack cs;
526 int from;
527 unsigned int hookmask;
528};
529
530/*
531 * we need these positions to check that the jumps to a different part of the
532 * entries is a jump to the beginning of a new chain.
533 */
534static inline int
535ebt_get_udc_positions(struct ebt_entry *e, struct ebt_table_info *newinfo,
Al Viro177abc32006-11-30 19:27:32 -0800536 unsigned int *n, struct ebt_cl_stack *udc)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700537{
538 int i;
539
540 /* we're only interested in chain starts */
Al Viro40642f92006-11-30 19:24:12 -0800541 if (e->bitmask)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700542 return 0;
543 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700544 if (newinfo->hook_entry[i] == (struct ebt_entries *)e)
545 break;
546 }
547 /* only care about udc */
548 if (i != NF_BR_NUMHOOKS)
549 return 0;
550
551 udc[*n].cs.chaininfo = (struct ebt_entries *)e;
552 /* these initialisations are depended on later in check_chainloops() */
553 udc[*n].cs.n = 0;
554 udc[*n].hookmask = 0;
555
556 (*n)++;
557 return 0;
558}
559
560static inline int
561ebt_cleanup_match(struct ebt_entry_match *m, unsigned int *i)
562{
563 if (i && (*i)-- == 0)
564 return 1;
565 if (m->u.match->destroy)
566 m->u.match->destroy(m->data, m->match_size);
567 module_put(m->u.match->me);
568
569 return 0;
570}
571
572static inline int
573ebt_cleanup_watcher(struct ebt_entry_watcher *w, unsigned int *i)
574{
575 if (i && (*i)-- == 0)
576 return 1;
577 if (w->u.watcher->destroy)
578 w->u.watcher->destroy(w->data, w->watcher_size);
579 module_put(w->u.watcher->me);
580
581 return 0;
582}
583
584static inline int
585ebt_cleanup_entry(struct ebt_entry *e, unsigned int *cnt)
586{
587 struct ebt_entry_target *t;
588
Al Viro40642f92006-11-30 19:24:12 -0800589 if (e->bitmask == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700590 return 0;
591 /* we're done */
592 if (cnt && (*cnt)-- == 0)
593 return 1;
594 EBT_WATCHER_ITERATE(e, ebt_cleanup_watcher, NULL);
595 EBT_MATCH_ITERATE(e, ebt_cleanup_match, NULL);
596 t = (struct ebt_entry_target *)(((char *)e) + e->target_offset);
597 if (t->u.target->destroy)
598 t->u.target->destroy(t->data, t->target_size);
599 module_put(t->u.target->me);
600
601 return 0;
602}
603
604static inline int
605ebt_check_entry(struct ebt_entry *e, struct ebt_table_info *newinfo,
Al Virof7da79d2006-11-30 19:27:48 -0800606 const char *name, unsigned int *cnt,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700607 struct ebt_cl_stack *cl_s, unsigned int udc_cnt)
608{
609 struct ebt_entry_target *t;
610 struct ebt_target *target;
611 unsigned int i, j, hook = 0, hookmask = 0;
Al Viro14197d52006-11-30 19:25:21 -0800612 size_t gap = e->next_offset - e->target_offset;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700613 int ret;
614
615 /* don't mess with the struct ebt_entries */
Al Viro40642f92006-11-30 19:24:12 -0800616 if (e->bitmask == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700617 return 0;
618
619 if (e->bitmask & ~EBT_F_MASK) {
620 BUGPRINT("Unknown flag for bitmask\n");
621 return -EINVAL;
622 }
623 if (e->invflags & ~EBT_INV_MASK) {
624 BUGPRINT("Unknown flag for inv bitmask\n");
625 return -EINVAL;
626 }
627 if ( (e->bitmask & EBT_NOPROTO) && (e->bitmask & EBT_802_3) ) {
628 BUGPRINT("NOPROTO & 802_3 not allowed\n");
629 return -EINVAL;
630 }
631 /* what hook do we belong to? */
632 for (i = 0; i < NF_BR_NUMHOOKS; i++) {
Al Virof7da79d2006-11-30 19:27:48 -0800633 if (!newinfo->hook_entry[i])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700634 continue;
635 if ((char *)newinfo->hook_entry[i] < (char *)e)
636 hook = i;
637 else
638 break;
639 }
640 /* (1 << NF_BR_NUMHOOKS) tells the check functions the rule is on
641 a base chain */
642 if (i < NF_BR_NUMHOOKS)
643 hookmask = (1 << hook) | (1 << NF_BR_NUMHOOKS);
644 else {
645 for (i = 0; i < udc_cnt; i++)
646 if ((char *)(cl_s[i].cs.chaininfo) > (char *)e)
647 break;
648 if (i == 0)
649 hookmask = (1 << hook) | (1 << NF_BR_NUMHOOKS);
650 else
651 hookmask = cl_s[i - 1].hookmask;
652 }
653 i = 0;
654 ret = EBT_MATCH_ITERATE(e, ebt_check_match, e, name, hookmask, &i);
655 if (ret != 0)
656 goto cleanup_matches;
657 j = 0;
658 ret = EBT_WATCHER_ITERATE(e, ebt_check_watcher, e, name, hookmask, &j);
659 if (ret != 0)
660 goto cleanup_watchers;
661 t = (struct ebt_entry_target *)(((char *)e) + e->target_offset);
662 target = find_target_lock(t->u.name, &ret, &ebt_mutex);
663 if (!target)
664 goto cleanup_watchers;
665 if (!try_module_get(target->me)) {
Ingo Molnar57b47a52006-03-20 22:35:41 -0800666 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700667 ret = -ENOENT;
668 goto cleanup_watchers;
669 }
Ingo Molnar57b47a52006-03-20 22:35:41 -0800670 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700671
672 t->u.target = target;
673 if (t->u.target == &ebt_standard_target) {
Al Viro14197d52006-11-30 19:25:21 -0800674 if (gap < sizeof(struct ebt_standard_target)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700675 BUGPRINT("Standard target size too big\n");
676 ret = -EFAULT;
677 goto cleanup_watchers;
678 }
679 if (((struct ebt_standard_target *)t)->verdict <
680 -NUM_STANDARD_TARGETS) {
681 BUGPRINT("Invalid standard target\n");
682 ret = -EFAULT;
683 goto cleanup_watchers;
684 }
Al Viro14197d52006-11-30 19:25:21 -0800685 } else if (t->target_size > gap - sizeof(struct ebt_entry_target) ||
Linus Torvalds1da177e2005-04-16 15:20:36 -0700686 (t->u.target->check &&
687 t->u.target->check(name, hookmask, e, t->data, t->target_size) != 0)){
688 module_put(t->u.target->me);
689 ret = -EFAULT;
690 goto cleanup_watchers;
691 }
692 (*cnt)++;
693 return 0;
694cleanup_watchers:
695 EBT_WATCHER_ITERATE(e, ebt_cleanup_watcher, &j);
696cleanup_matches:
697 EBT_MATCH_ITERATE(e, ebt_cleanup_match, &i);
698 return ret;
699}
700
701/*
702 * checks for loops and sets the hook mask for udc
703 * the hook mask for udc tells us from which base chains the udc can be
704 * accessed. This mask is a parameter to the check() functions of the extensions
705 */
706static int check_chainloops(struct ebt_entries *chain, struct ebt_cl_stack *cl_s,
707 unsigned int udc_cnt, unsigned int hooknr, char *base)
708{
709 int i, chain_nr = -1, pos = 0, nentries = chain->nentries, verdict;
710 struct ebt_entry *e = (struct ebt_entry *)chain->data;
711 struct ebt_entry_target *t;
712
713 while (pos < nentries || chain_nr != -1) {
714 /* end of udc, go back one 'recursion' step */
715 if (pos == nentries) {
716 /* put back values of the time when this chain was called */
717 e = cl_s[chain_nr].cs.e;
718 if (cl_s[chain_nr].from != -1)
719 nentries =
720 cl_s[cl_s[chain_nr].from].cs.chaininfo->nentries;
721 else
722 nentries = chain->nentries;
723 pos = cl_s[chain_nr].cs.n;
724 /* make sure we won't see a loop that isn't one */
725 cl_s[chain_nr].cs.n = 0;
726 chain_nr = cl_s[chain_nr].from;
727 if (pos == nentries)
728 continue;
729 }
730 t = (struct ebt_entry_target *)
731 (((char *)e) + e->target_offset);
732 if (strcmp(t->u.name, EBT_STANDARD_TARGET))
733 goto letscontinue;
734 if (e->target_offset + sizeof(struct ebt_standard_target) >
735 e->next_offset) {
736 BUGPRINT("Standard target size too big\n");
737 return -1;
738 }
739 verdict = ((struct ebt_standard_target *)t)->verdict;
740 if (verdict >= 0) { /* jump to another chain */
741 struct ebt_entries *hlp2 =
742 (struct ebt_entries *)(base + verdict);
743 for (i = 0; i < udc_cnt; i++)
744 if (hlp2 == cl_s[i].cs.chaininfo)
745 break;
746 /* bad destination or loop */
747 if (i == udc_cnt) {
748 BUGPRINT("bad destination\n");
749 return -1;
750 }
751 if (cl_s[i].cs.n) {
752 BUGPRINT("loop\n");
753 return -1;
754 }
Al Viro98a08242006-11-30 19:24:49 -0800755 if (cl_s[i].hookmask & (1 << hooknr))
756 goto letscontinue;
757 /* this can't be 0, so the loop test is correct */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700758 cl_s[i].cs.n = pos + 1;
759 pos = 0;
760 cl_s[i].cs.e = ((void *)e + e->next_offset);
761 e = (struct ebt_entry *)(hlp2->data);
762 nentries = hlp2->nentries;
763 cl_s[i].from = chain_nr;
764 chain_nr = i;
765 /* this udc is accessible from the base chain for hooknr */
766 cl_s[i].hookmask |= (1 << hooknr);
767 continue;
768 }
769letscontinue:
770 e = (void *)e + e->next_offset;
771 pos++;
772 }
773 return 0;
774}
775
776/* do the parsing of the table/chains/entries/matches/watchers/targets, heh */
Al Viro1bc23262006-11-30 19:28:08 -0800777static int translate_table(char *name, struct ebt_table_info *newinfo)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700778{
779 unsigned int i, j, k, udc_cnt;
780 int ret;
781 struct ebt_cl_stack *cl_s = NULL; /* used in the checking for chain loops */
782
783 i = 0;
Al Viro1f072c92006-11-30 19:26:53 -0800784 while (i < NF_BR_NUMHOOKS && !newinfo->hook_entry[i])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700785 i++;
786 if (i == NF_BR_NUMHOOKS) {
787 BUGPRINT("No valid hooks specified\n");
788 return -EINVAL;
789 }
Al Viro1f072c92006-11-30 19:26:53 -0800790 if (newinfo->hook_entry[i] != (struct ebt_entries *)newinfo->entries) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700791 BUGPRINT("Chains don't start at beginning\n");
792 return -EINVAL;
793 }
794 /* make sure chains are ordered after each other in same order
795 as their corresponding hooks */
796 for (j = i + 1; j < NF_BR_NUMHOOKS; j++) {
Al Viro1f072c92006-11-30 19:26:53 -0800797 if (!newinfo->hook_entry[j])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700798 continue;
Al Viro1f072c92006-11-30 19:26:53 -0800799 if (newinfo->hook_entry[j] <= newinfo->hook_entry[i]) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700800 BUGPRINT("Hook order must be followed\n");
801 return -EINVAL;
802 }
803 i = j;
804 }
805
Linus Torvalds1da177e2005-04-16 15:20:36 -0700806 /* do some early checkings and initialize some things */
807 i = 0; /* holds the expected nr. of entries for the chain */
808 j = 0; /* holds the up to now counted entries for the chain */
809 k = 0; /* holds the total nr. of entries, should equal
810 newinfo->nentries afterwards */
811 udc_cnt = 0; /* will hold the nr. of user defined chains (udc) */
812 ret = EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
Al Viro0e795532006-11-30 19:27:13 -0800813 ebt_check_entry_size_and_hooks, newinfo,
814 &i, &j, &k, &udc_cnt);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700815
816 if (ret != 0)
817 return ret;
818
819 if (i != j) {
820 BUGPRINT("nentries does not equal the nr of entries in the "
821 "(last) chain\n");
822 return -EINVAL;
823 }
824 if (k != newinfo->nentries) {
825 BUGPRINT("Total nentries is wrong\n");
826 return -EINVAL;
827 }
828
Linus Torvalds1da177e2005-04-16 15:20:36 -0700829 /* get the location of the udc, put them in an array
830 while we're at it, allocate the chainstack */
831 if (udc_cnt) {
832 /* this will get free'd in do_replace()/ebt_register_table()
833 if an error occurs */
Jayachandran C7ad4d2f2006-04-11 17:25:38 -0700834 newinfo->chainstack =
835 vmalloc((highest_possible_processor_id()+1)
836 * sizeof(*(newinfo->chainstack)));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700837 if (!newinfo->chainstack)
838 return -ENOMEM;
KAMEZAWA Hiroyuki6f912042006-04-10 22:52:50 -0700839 for_each_possible_cpu(i) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700840 newinfo->chainstack[i] =
Jayachandran C18bc89a2006-04-20 00:14:49 -0700841 vmalloc(udc_cnt * sizeof(*(newinfo->chainstack[0])));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700842 if (!newinfo->chainstack[i]) {
843 while (i)
844 vfree(newinfo->chainstack[--i]);
845 vfree(newinfo->chainstack);
846 newinfo->chainstack = NULL;
847 return -ENOMEM;
848 }
849 }
850
Jayachandran C18bc89a2006-04-20 00:14:49 -0700851 cl_s = vmalloc(udc_cnt * sizeof(*cl_s));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700852 if (!cl_s)
853 return -ENOMEM;
854 i = 0; /* the i'th udc */
855 EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
Al Viro177abc32006-11-30 19:27:32 -0800856 ebt_get_udc_positions, newinfo, &i, cl_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700857 /* sanity check */
858 if (i != udc_cnt) {
859 BUGPRINT("i != udc_cnt\n");
860 vfree(cl_s);
861 return -EFAULT;
862 }
863 }
864
865 /* Check for loops */
866 for (i = 0; i < NF_BR_NUMHOOKS; i++)
Al Viro1f072c92006-11-30 19:26:53 -0800867 if (newinfo->hook_entry[i])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700868 if (check_chainloops(newinfo->hook_entry[i],
869 cl_s, udc_cnt, i, newinfo->entries)) {
James Lamanna68d31872005-06-22 22:12:57 -0700870 vfree(cl_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700871 return -EINVAL;
872 }
873
874 /* we now know the following (along with E=mc²):
875 - the nr of entries in each chain is right
876 - the size of the allocated space is right
877 - all valid hooks have a corresponding chain
878 - there are no loops
879 - wrong data can still be on the level of a single entry
880 - could be there are jumps to places that are not the
881 beginning of a chain. This can only occur in chains that
882 are not accessible from any base chains, so we don't care. */
883
884 /* used to know what we need to clean up if something goes wrong */
885 i = 0;
886 ret = EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
Al Viro1bc23262006-11-30 19:28:08 -0800887 ebt_check_entry, newinfo, name, &i, cl_s, udc_cnt);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700888 if (ret != 0) {
889 EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
890 ebt_cleanup_entry, &i);
891 }
James Lamanna68d31872005-06-22 22:12:57 -0700892 vfree(cl_s);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700893 return ret;
894}
895
896/* called under write_lock */
897static void get_counters(struct ebt_counter *oldcounters,
898 struct ebt_counter *counters, unsigned int nentries)
899{
900 int i, cpu;
901 struct ebt_counter *counter_base;
902
903 /* counters of cpu 0 */
904 memcpy(counters, oldcounters,
David S. Millerc8923c62005-10-13 14:41:23 -0700905 sizeof(struct ebt_counter) * nentries);
906
Linus Torvalds1da177e2005-04-16 15:20:36 -0700907 /* add other counters to those of cpu 0 */
KAMEZAWA Hiroyuki6f912042006-04-10 22:52:50 -0700908 for_each_possible_cpu(cpu) {
David S. Millerc8923c62005-10-13 14:41:23 -0700909 if (cpu == 0)
910 continue;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700911 counter_base = COUNTER_BASE(oldcounters, nentries, cpu);
912 for (i = 0; i < nentries; i++) {
913 counters[i].pcnt += counter_base[i].pcnt;
914 counters[i].bcnt += counter_base[i].bcnt;
915 }
916 }
917}
918
919/* replace the table */
920static int do_replace(void __user *user, unsigned int len)
921{
922 int ret, i, countersize;
923 struct ebt_table_info *newinfo;
924 struct ebt_replace tmp;
925 struct ebt_table *t;
926 struct ebt_counter *counterstmp = NULL;
927 /* used to be able to unlock earlier */
928 struct ebt_table_info *table;
929
930 if (copy_from_user(&tmp, user, sizeof(tmp)) != 0)
931 return -EFAULT;
932
933 if (len != sizeof(tmp) + tmp.entries_size) {
934 BUGPRINT("Wrong len argument\n");
935 return -EINVAL;
936 }
937
938 if (tmp.entries_size == 0) {
939 BUGPRINT("Entries_size never zero\n");
940 return -EINVAL;
941 }
Kirill Korotaevee4bb812006-02-04 02:16:56 -0800942 /* overflow check */
943 if (tmp.nentries >= ((INT_MAX - sizeof(struct ebt_table_info)) / NR_CPUS -
944 SMP_CACHE_BYTES) / sizeof(struct ebt_counter))
945 return -ENOMEM;
946 if (tmp.num_counters >= INT_MAX / sizeof(struct ebt_counter))
947 return -ENOMEM;
948
David S. Millerc8923c62005-10-13 14:41:23 -0700949 countersize = COUNTER_OFFSET(tmp.nentries) *
950 (highest_possible_processor_id()+1);
Jayachandran C18bc89a2006-04-20 00:14:49 -0700951 newinfo = vmalloc(sizeof(*newinfo) + countersize);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700952 if (!newinfo)
953 return -ENOMEM;
954
955 if (countersize)
956 memset(newinfo->counters, 0, countersize);
957
Kris Katterjohn8b3a7002006-01-11 15:56:43 -0800958 newinfo->entries = vmalloc(tmp.entries_size);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700959 if (!newinfo->entries) {
960 ret = -ENOMEM;
961 goto free_newinfo;
962 }
963 if (copy_from_user(
964 newinfo->entries, tmp.entries, tmp.entries_size) != 0) {
965 BUGPRINT("Couldn't copy entries from userspace\n");
966 ret = -EFAULT;
967 goto free_entries;
968 }
969
970 /* the user wants counters back
971 the check on the size is done later, when we have the lock */
972 if (tmp.num_counters) {
Jayachandran C18bc89a2006-04-20 00:14:49 -0700973 counterstmp = vmalloc(tmp.num_counters * sizeof(*counterstmp));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700974 if (!counterstmp) {
975 ret = -ENOMEM;
976 goto free_entries;
977 }
978 }
979 else
980 counterstmp = NULL;
981
982 /* this can get initialized by translate_table() */
983 newinfo->chainstack = NULL;
Al Viro1bc23262006-11-30 19:28:08 -0800984 ret = ebt_verify_pointers(&tmp, newinfo);
985 if (ret != 0)
986 goto free_counterstmp;
987
988 ret = translate_table(tmp.name, newinfo);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700989
990 if (ret != 0)
991 goto free_counterstmp;
992
993 t = find_table_lock(tmp.name, &ret, &ebt_mutex);
994 if (!t) {
995 ret = -ENOENT;
996 goto free_iterate;
997 }
998
999 /* the table doesn't like it */
1000 if (t->check && (ret = t->check(newinfo, tmp.valid_hooks)))
1001 goto free_unlock;
1002
1003 if (tmp.num_counters && tmp.num_counters != t->private->nentries) {
1004 BUGPRINT("Wrong nr. of counters requested\n");
1005 ret = -EINVAL;
1006 goto free_unlock;
1007 }
1008
1009 /* we have the mutex lock, so no danger in reading this pointer */
1010 table = t->private;
1011 /* make sure the table can only be rmmod'ed if it contains no rules */
1012 if (!table->nentries && newinfo->nentries && !try_module_get(t->me)) {
1013 ret = -ENOENT;
1014 goto free_unlock;
1015 } else if (table->nentries && !newinfo->nentries)
1016 module_put(t->me);
1017 /* we need an atomic snapshot of the counters */
1018 write_lock_bh(&t->lock);
1019 if (tmp.num_counters)
1020 get_counters(t->private->counters, counterstmp,
1021 t->private->nentries);
1022
1023 t->private = newinfo;
1024 write_unlock_bh(&t->lock);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001025 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001026 /* so, a user can change the chains while having messed up her counter
1027 allocation. Only reason why this is done is because this way the lock
1028 is held only once, while this doesn't bring the kernel into a
1029 dangerous state. */
1030 if (tmp.num_counters &&
1031 copy_to_user(tmp.counters, counterstmp,
1032 tmp.num_counters * sizeof(struct ebt_counter))) {
1033 BUGPRINT("Couldn't copy counters to userspace\n");
1034 ret = -EFAULT;
1035 }
1036 else
1037 ret = 0;
1038
1039 /* decrease module count and free resources */
1040 EBT_ENTRY_ITERATE(table->entries, table->entries_size,
1041 ebt_cleanup_entry, NULL);
1042
1043 vfree(table->entries);
1044 if (table->chainstack) {
KAMEZAWA Hiroyuki6f912042006-04-10 22:52:50 -07001045 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001046 vfree(table->chainstack[i]);
1047 vfree(table->chainstack);
1048 }
1049 vfree(table);
1050
James Lamanna68d31872005-06-22 22:12:57 -07001051 vfree(counterstmp);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001052 return ret;
1053
1054free_unlock:
Ingo Molnar57b47a52006-03-20 22:35:41 -08001055 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001056free_iterate:
1057 EBT_ENTRY_ITERATE(newinfo->entries, newinfo->entries_size,
1058 ebt_cleanup_entry, NULL);
1059free_counterstmp:
James Lamanna68d31872005-06-22 22:12:57 -07001060 vfree(counterstmp);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001061 /* can be initialized in translate_table() */
1062 if (newinfo->chainstack) {
KAMEZAWA Hiroyuki6f912042006-04-10 22:52:50 -07001063 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001064 vfree(newinfo->chainstack[i]);
1065 vfree(newinfo->chainstack);
1066 }
1067free_entries:
James Lamanna68d31872005-06-22 22:12:57 -07001068 vfree(newinfo->entries);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001069free_newinfo:
James Lamanna68d31872005-06-22 22:12:57 -07001070 vfree(newinfo);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001071 return ret;
1072}
1073
1074int ebt_register_target(struct ebt_target *target)
1075{
Patrick McHardydf0933d2006-09-20 11:57:53 -07001076 struct ebt_target *t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001077 int ret;
1078
Ingo Molnar57b47a52006-03-20 22:35:41 -08001079 ret = mutex_lock_interruptible(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001080 if (ret != 0)
1081 return ret;
Patrick McHardydf0933d2006-09-20 11:57:53 -07001082 list_for_each_entry(t, &ebt_targets, list) {
1083 if (strcmp(t->name, target->name) == 0) {
1084 mutex_unlock(&ebt_mutex);
1085 return -EEXIST;
1086 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001087 }
Patrick McHardydf0933d2006-09-20 11:57:53 -07001088 list_add(&target->list, &ebt_targets);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001089 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001090
1091 return 0;
1092}
1093
1094void ebt_unregister_target(struct ebt_target *target)
1095{
Ingo Molnar57b47a52006-03-20 22:35:41 -08001096 mutex_lock(&ebt_mutex);
Patrick McHardydf0933d2006-09-20 11:57:53 -07001097 list_del(&target->list);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001098 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001099}
1100
1101int ebt_register_match(struct ebt_match *match)
1102{
Patrick McHardydf0933d2006-09-20 11:57:53 -07001103 struct ebt_match *m;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001104 int ret;
1105
Ingo Molnar57b47a52006-03-20 22:35:41 -08001106 ret = mutex_lock_interruptible(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001107 if (ret != 0)
1108 return ret;
Patrick McHardydf0933d2006-09-20 11:57:53 -07001109 list_for_each_entry(m, &ebt_matches, list) {
1110 if (strcmp(m->name, match->name) == 0) {
1111 mutex_unlock(&ebt_mutex);
1112 return -EEXIST;
1113 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001114 }
Patrick McHardydf0933d2006-09-20 11:57:53 -07001115 list_add(&match->list, &ebt_matches);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001116 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001117
1118 return 0;
1119}
1120
1121void ebt_unregister_match(struct ebt_match *match)
1122{
Ingo Molnar57b47a52006-03-20 22:35:41 -08001123 mutex_lock(&ebt_mutex);
Patrick McHardydf0933d2006-09-20 11:57:53 -07001124 list_del(&match->list);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001125 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001126}
1127
1128int ebt_register_watcher(struct ebt_watcher *watcher)
1129{
Patrick McHardydf0933d2006-09-20 11:57:53 -07001130 struct ebt_watcher *w;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001131 int ret;
1132
Ingo Molnar57b47a52006-03-20 22:35:41 -08001133 ret = mutex_lock_interruptible(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001134 if (ret != 0)
1135 return ret;
Patrick McHardydf0933d2006-09-20 11:57:53 -07001136 list_for_each_entry(w, &ebt_watchers, list) {
1137 if (strcmp(w->name, watcher->name) == 0) {
1138 mutex_unlock(&ebt_mutex);
1139 return -EEXIST;
1140 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001141 }
Patrick McHardydf0933d2006-09-20 11:57:53 -07001142 list_add(&watcher->list, &ebt_watchers);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001143 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001144
1145 return 0;
1146}
1147
1148void ebt_unregister_watcher(struct ebt_watcher *watcher)
1149{
Ingo Molnar57b47a52006-03-20 22:35:41 -08001150 mutex_lock(&ebt_mutex);
Patrick McHardydf0933d2006-09-20 11:57:53 -07001151 list_del(&watcher->list);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001152 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001153}
1154
1155int ebt_register_table(struct ebt_table *table)
1156{
1157 struct ebt_table_info *newinfo;
Patrick McHardydf0933d2006-09-20 11:57:53 -07001158 struct ebt_table *t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001159 int ret, i, countersize;
1160
1161 if (!table || !table->table ||!table->table->entries ||
1162 table->table->entries_size == 0 ||
1163 table->table->counters || table->private) {
1164 BUGPRINT("Bad table data for ebt_register_table!!!\n");
1165 return -EINVAL;
1166 }
1167
David S. Millerc8923c62005-10-13 14:41:23 -07001168 countersize = COUNTER_OFFSET(table->table->nentries) *
1169 (highest_possible_processor_id()+1);
Jayachandran C18bc89a2006-04-20 00:14:49 -07001170 newinfo = vmalloc(sizeof(*newinfo) + countersize);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001171 ret = -ENOMEM;
1172 if (!newinfo)
1173 return -ENOMEM;
1174
Kris Katterjohn8b3a7002006-01-11 15:56:43 -08001175 newinfo->entries = vmalloc(table->table->entries_size);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001176 if (!(newinfo->entries))
1177 goto free_newinfo;
1178
1179 memcpy(newinfo->entries, table->table->entries,
1180 table->table->entries_size);
1181
1182 if (countersize)
1183 memset(newinfo->counters, 0, countersize);
1184
1185 /* fill in newinfo and parse the entries */
1186 newinfo->chainstack = NULL;
Al Viro1bc23262006-11-30 19:28:08 -08001187 ret = ebt_verify_pointers(table->table, newinfo);
1188 if (ret != 0)
1189 goto free_chainstack;
1190 ret = translate_table(table->table->name, newinfo);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001191 if (ret != 0) {
1192 BUGPRINT("Translate_table failed\n");
1193 goto free_chainstack;
1194 }
1195
1196 if (table->check && table->check(newinfo, table->valid_hooks)) {
1197 BUGPRINT("The table doesn't like its own initial data, lol\n");
1198 return -EINVAL;
1199 }
1200
1201 table->private = newinfo;
1202 rwlock_init(&table->lock);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001203 ret = mutex_lock_interruptible(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001204 if (ret != 0)
1205 goto free_chainstack;
1206
Patrick McHardydf0933d2006-09-20 11:57:53 -07001207 list_for_each_entry(t, &ebt_tables, list) {
1208 if (strcmp(t->name, table->name) == 0) {
1209 ret = -EEXIST;
1210 BUGPRINT("Table name already exists\n");
1211 goto free_unlock;
1212 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001213 }
1214
1215 /* Hold a reference count if the chains aren't empty */
1216 if (newinfo->nentries && !try_module_get(table->me)) {
1217 ret = -ENOENT;
1218 goto free_unlock;
1219 }
Patrick McHardydf0933d2006-09-20 11:57:53 -07001220 list_add(&table->list, &ebt_tables);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001221 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001222 return 0;
1223free_unlock:
Ingo Molnar57b47a52006-03-20 22:35:41 -08001224 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001225free_chainstack:
1226 if (newinfo->chainstack) {
KAMEZAWA Hiroyuki6f912042006-04-10 22:52:50 -07001227 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001228 vfree(newinfo->chainstack[i]);
1229 vfree(newinfo->chainstack);
1230 }
1231 vfree(newinfo->entries);
1232free_newinfo:
1233 vfree(newinfo);
1234 return ret;
1235}
1236
1237void ebt_unregister_table(struct ebt_table *table)
1238{
1239 int i;
1240
1241 if (!table) {
1242 BUGPRINT("Request to unregister NULL table!!!\n");
1243 return;
1244 }
Ingo Molnar57b47a52006-03-20 22:35:41 -08001245 mutex_lock(&ebt_mutex);
Patrick McHardydf0933d2006-09-20 11:57:53 -07001246 list_del(&table->list);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001247 mutex_unlock(&ebt_mutex);
James Lamanna68d31872005-06-22 22:12:57 -07001248 vfree(table->private->entries);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001249 if (table->private->chainstack) {
KAMEZAWA Hiroyuki6f912042006-04-10 22:52:50 -07001250 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001251 vfree(table->private->chainstack[i]);
1252 vfree(table->private->chainstack);
1253 }
1254 vfree(table->private);
1255}
1256
1257/* userspace just supplied us with counters */
1258static int update_counters(void __user *user, unsigned int len)
1259{
1260 int i, ret;
1261 struct ebt_counter *tmp;
1262 struct ebt_replace hlp;
1263 struct ebt_table *t;
1264
1265 if (copy_from_user(&hlp, user, sizeof(hlp)))
1266 return -EFAULT;
1267
1268 if (len != sizeof(hlp) + hlp.num_counters * sizeof(struct ebt_counter))
1269 return -EINVAL;
1270 if (hlp.num_counters == 0)
1271 return -EINVAL;
1272
Jayachandran C18bc89a2006-04-20 00:14:49 -07001273 if (!(tmp = vmalloc(hlp.num_counters * sizeof(*tmp)))) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001274 MEMPRINT("Update_counters && nomemory\n");
1275 return -ENOMEM;
1276 }
1277
1278 t = find_table_lock(hlp.name, &ret, &ebt_mutex);
1279 if (!t)
1280 goto free_tmp;
1281
1282 if (hlp.num_counters != t->private->nentries) {
1283 BUGPRINT("Wrong nr of counters\n");
1284 ret = -EINVAL;
1285 goto unlock_mutex;
1286 }
1287
1288 if ( copy_from_user(tmp, hlp.counters,
1289 hlp.num_counters * sizeof(struct ebt_counter)) ) {
1290 BUGPRINT("Updata_counters && !cfu\n");
1291 ret = -EFAULT;
1292 goto unlock_mutex;
1293 }
1294
1295 /* we want an atomic add of the counters */
1296 write_lock_bh(&t->lock);
1297
1298 /* we add to the counters of the first cpu */
1299 for (i = 0; i < hlp.num_counters; i++) {
1300 t->private->counters[i].pcnt += tmp[i].pcnt;
1301 t->private->counters[i].bcnt += tmp[i].bcnt;
1302 }
1303
1304 write_unlock_bh(&t->lock);
1305 ret = 0;
1306unlock_mutex:
Ingo Molnar57b47a52006-03-20 22:35:41 -08001307 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001308free_tmp:
1309 vfree(tmp);
1310 return ret;
1311}
1312
1313static inline int ebt_make_matchname(struct ebt_entry_match *m,
1314 char *base, char *ubase)
1315{
1316 char *hlp = ubase - base + (char *)m;
1317 if (copy_to_user(hlp, m->u.match->name, EBT_FUNCTION_MAXNAMELEN))
1318 return -EFAULT;
1319 return 0;
1320}
1321
1322static inline int ebt_make_watchername(struct ebt_entry_watcher *w,
1323 char *base, char *ubase)
1324{
1325 char *hlp = ubase - base + (char *)w;
1326 if (copy_to_user(hlp , w->u.watcher->name, EBT_FUNCTION_MAXNAMELEN))
1327 return -EFAULT;
1328 return 0;
1329}
1330
1331static inline int ebt_make_names(struct ebt_entry *e, char *base, char *ubase)
1332{
1333 int ret;
1334 char *hlp;
1335 struct ebt_entry_target *t;
1336
Al Viro40642f92006-11-30 19:24:12 -08001337 if (e->bitmask == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001338 return 0;
1339
1340 hlp = ubase - base + (char *)e + e->target_offset;
1341 t = (struct ebt_entry_target *)(((char *)e) + e->target_offset);
1342
1343 ret = EBT_MATCH_ITERATE(e, ebt_make_matchname, base, ubase);
1344 if (ret != 0)
1345 return ret;
1346 ret = EBT_WATCHER_ITERATE(e, ebt_make_watchername, base, ubase);
1347 if (ret != 0)
1348 return ret;
1349 if (copy_to_user(hlp, t->u.target->name, EBT_FUNCTION_MAXNAMELEN))
1350 return -EFAULT;
1351 return 0;
1352}
1353
Ingo Molnar57b47a52006-03-20 22:35:41 -08001354/* called with ebt_mutex locked */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001355static int copy_everything_to_user(struct ebt_table *t, void __user *user,
1356 int *len, int cmd)
1357{
1358 struct ebt_replace tmp;
1359 struct ebt_counter *counterstmp, *oldcounters;
1360 unsigned int entries_size, nentries;
1361 char *entries;
1362
1363 if (cmd == EBT_SO_GET_ENTRIES) {
1364 entries_size = t->private->entries_size;
1365 nentries = t->private->nentries;
1366 entries = t->private->entries;
1367 oldcounters = t->private->counters;
1368 } else {
1369 entries_size = t->table->entries_size;
1370 nentries = t->table->nentries;
1371 entries = t->table->entries;
1372 oldcounters = t->table->counters;
1373 }
1374
1375 if (copy_from_user(&tmp, user, sizeof(tmp))) {
1376 BUGPRINT("Cfu didn't work\n");
1377 return -EFAULT;
1378 }
1379
1380 if (*len != sizeof(struct ebt_replace) + entries_size +
1381 (tmp.num_counters? nentries * sizeof(struct ebt_counter): 0)) {
1382 BUGPRINT("Wrong size\n");
1383 return -EINVAL;
1384 }
1385
1386 if (tmp.nentries != nentries) {
1387 BUGPRINT("Nentries wrong\n");
1388 return -EINVAL;
1389 }
1390
1391 if (tmp.entries_size != entries_size) {
1392 BUGPRINT("Wrong size\n");
1393 return -EINVAL;
1394 }
1395
1396 /* userspace might not need the counters */
1397 if (tmp.num_counters) {
1398 if (tmp.num_counters != nentries) {
1399 BUGPRINT("Num_counters wrong\n");
1400 return -EINVAL;
1401 }
Jayachandran C18bc89a2006-04-20 00:14:49 -07001402 counterstmp = vmalloc(nentries * sizeof(*counterstmp));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001403 if (!counterstmp) {
1404 MEMPRINT("Couldn't copy counters, out of memory\n");
1405 return -ENOMEM;
1406 }
1407 write_lock_bh(&t->lock);
1408 get_counters(oldcounters, counterstmp, nentries);
1409 write_unlock_bh(&t->lock);
1410
1411 if (copy_to_user(tmp.counters, counterstmp,
1412 nentries * sizeof(struct ebt_counter))) {
1413 BUGPRINT("Couldn't copy counters to userspace\n");
1414 vfree(counterstmp);
1415 return -EFAULT;
1416 }
1417 vfree(counterstmp);
1418 }
1419
1420 if (copy_to_user(tmp.entries, entries, entries_size)) {
1421 BUGPRINT("Couldn't copy entries to userspace\n");
1422 return -EFAULT;
1423 }
1424 /* set the match/watcher/target names right */
1425 return EBT_ENTRY_ITERATE(entries, entries_size,
1426 ebt_make_names, entries, tmp.entries);
1427}
1428
1429static int do_ebt_set_ctl(struct sock *sk,
1430 int cmd, void __user *user, unsigned int len)
1431{
1432 int ret;
1433
1434 switch(cmd) {
1435 case EBT_SO_SET_ENTRIES:
1436 ret = do_replace(user, len);
1437 break;
1438 case EBT_SO_SET_COUNTERS:
1439 ret = update_counters(user, len);
1440 break;
1441 default:
1442 ret = -EINVAL;
1443 }
1444 return ret;
1445}
1446
1447static int do_ebt_get_ctl(struct sock *sk, int cmd, void __user *user, int *len)
1448{
1449 int ret;
1450 struct ebt_replace tmp;
1451 struct ebt_table *t;
1452
1453 if (copy_from_user(&tmp, user, sizeof(tmp)))
1454 return -EFAULT;
1455
1456 t = find_table_lock(tmp.name, &ret, &ebt_mutex);
1457 if (!t)
1458 return ret;
1459
1460 switch(cmd) {
1461 case EBT_SO_GET_INFO:
1462 case EBT_SO_GET_INIT_INFO:
1463 if (*len != sizeof(struct ebt_replace)){
1464 ret = -EINVAL;
Ingo Molnar57b47a52006-03-20 22:35:41 -08001465 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001466 break;
1467 }
1468 if (cmd == EBT_SO_GET_INFO) {
1469 tmp.nentries = t->private->nentries;
1470 tmp.entries_size = t->private->entries_size;
1471 tmp.valid_hooks = t->valid_hooks;
1472 } else {
1473 tmp.nentries = t->table->nentries;
1474 tmp.entries_size = t->table->entries_size;
1475 tmp.valid_hooks = t->table->valid_hooks;
1476 }
Ingo Molnar57b47a52006-03-20 22:35:41 -08001477 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001478 if (copy_to_user(user, &tmp, *len) != 0){
1479 BUGPRINT("c2u Didn't work\n");
1480 ret = -EFAULT;
1481 break;
1482 }
1483 ret = 0;
1484 break;
1485
1486 case EBT_SO_GET_ENTRIES:
1487 case EBT_SO_GET_INIT_ENTRIES:
1488 ret = copy_everything_to_user(t, user, len, cmd);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001489 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001490 break;
1491
1492 default:
Ingo Molnar57b47a52006-03-20 22:35:41 -08001493 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001494 ret = -EINVAL;
1495 }
1496
1497 return ret;
1498}
1499
1500static struct nf_sockopt_ops ebt_sockopts =
Andrew Morton74ca4e5a2006-03-20 22:55:02 -08001501{
1502 .pf = PF_INET,
1503 .set_optmin = EBT_BASE_CTL,
1504 .set_optmax = EBT_SO_SET_MAX + 1,
1505 .set = do_ebt_set_ctl,
1506 .get_optmin = EBT_BASE_CTL,
1507 .get_optmax = EBT_SO_GET_MAX + 1,
1508 .get = do_ebt_get_ctl,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001509};
1510
Andrew Morton65b4b4e2006-03-28 16:37:06 -08001511static int __init ebtables_init(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001512{
1513 int ret;
1514
Ingo Molnar57b47a52006-03-20 22:35:41 -08001515 mutex_lock(&ebt_mutex);
Patrick McHardydf0933d2006-09-20 11:57:53 -07001516 list_add(&ebt_standard_target.list, &ebt_targets);
Ingo Molnar57b47a52006-03-20 22:35:41 -08001517 mutex_unlock(&ebt_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001518 if ((ret = nf_register_sockopt(&ebt_sockopts)) < 0)
1519 return ret;
1520
1521 printk(KERN_NOTICE "Ebtables v2.0 registered\n");
1522 return 0;
1523}
1524
Andrew Morton65b4b4e2006-03-28 16:37:06 -08001525static void __exit ebtables_fini(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001526{
1527 nf_unregister_sockopt(&ebt_sockopts);
1528 printk(KERN_NOTICE "Ebtables v2.0 unregistered\n");
1529}
1530
1531EXPORT_SYMBOL(ebt_register_table);
1532EXPORT_SYMBOL(ebt_unregister_table);
1533EXPORT_SYMBOL(ebt_register_match);
1534EXPORT_SYMBOL(ebt_unregister_match);
1535EXPORT_SYMBOL(ebt_register_watcher);
1536EXPORT_SYMBOL(ebt_unregister_watcher);
1537EXPORT_SYMBOL(ebt_register_target);
1538EXPORT_SYMBOL(ebt_unregister_target);
1539EXPORT_SYMBOL(ebt_do_table);
Andrew Morton65b4b4e2006-03-28 16:37:06 -08001540module_init(ebtables_init);
1541module_exit(ebtables_fini);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001542MODULE_LICENSE("GPL");