1/*
2 *  ebt_mark
3 *
4 *	Authors:
5 *	Bart De Schuymer <bdschuym@pandora.be>
6 *
7 *  July, 2002
8 *
9 */
10
11/* The mark target can be used in any chain,
12 * I believe adding a mangle table just for marking is total overkill.
13 * Marking a frame doesn't really change anything in the frame anyway.
14 */
15
16#include <linux/netfilter_bridge/ebtables.h>
17#include <linux/netfilter_bridge/ebt_mark_t.h>
18#include <linux/module.h>
19
20static int ebt_target_mark(struct sk_buff **pskb, unsigned int hooknr,
21   const struct net_device *in, const struct net_device *out,
22   const void *data, unsigned int datalen)
23{
24	struct ebt_mark_t_info *info = (struct ebt_mark_t_info *)data;
25	int action = info->target & -16;
26
27	if (action == MARK_SET_VALUE)
28		(*pskb)->mark = info->mark;
29	else if (action == MARK_OR_VALUE)
30		(*pskb)->mark |= info->mark;
31	else if (action == MARK_AND_VALUE)
32		(*pskb)->mark &= info->mark;
33	else
34		(*pskb)->mark ^= info->mark;
35
36	return info->target | ~EBT_VERDICT_BITS;
37}
38
39static int ebt_target_mark_check(const char *tablename, unsigned int hookmask,
40   const struct ebt_entry *e, void *data, unsigned int datalen)
41{
42	struct ebt_mark_t_info *info = (struct ebt_mark_t_info *)data;
43	int tmp;
44
45	if (datalen != EBT_ALIGN(sizeof(struct ebt_mark_t_info)))
46		return -EINVAL;
47	tmp = info->target | ~EBT_VERDICT_BITS;
48	if (BASE_CHAIN && tmp == EBT_RETURN)
49		return -EINVAL;
50	CLEAR_BASE_CHAIN_BIT;
51	if (tmp < -NUM_STANDARD_TARGETS || tmp >= 0)
52		return -EINVAL;
53	tmp = info->target & ~EBT_VERDICT_BITS;
54	if (tmp != MARK_SET_VALUE && tmp != MARK_OR_VALUE &&
55	    tmp != MARK_AND_VALUE && tmp != MARK_XOR_VALUE)
56		return -EINVAL;
57	return 0;
58}
59
60static struct ebt_target mark_target =
61{
62	.name		= EBT_MARK_TARGET,
63	.target		= ebt_target_mark,
64	.check		= ebt_target_mark_check,
65	.me		= THIS_MODULE,
66};
67
68static int __init ebt_mark_init(void)
69{
70	return ebt_register_target(&mark_target);
71}
72
73static void __exit ebt_mark_fini(void)
74{
75	ebt_unregister_target(&mark_target);
76}
77
78module_init(ebt_mark_init);
79module_exit(ebt_mark_fini);
80MODULE_LICENSE("GPL");
81