ebt_snat.c 2.13 KB
Newer Older
Linus Torvalds's avatar
Linus Torvalds committed
1
2
3
4
5
6
7
8
9
10
/*
 *  ebt_snat
 *
 *	Authors:
 *	Bart De Schuymer <bdschuym@pandora.be>
 *
 *  June, 2002
 *
 */

11
#include <linux/netfilter.h>
Linus Torvalds's avatar
Linus Torvalds committed
12
13
14
15
#include <linux/netfilter_bridge/ebtables.h>
#include <linux/netfilter_bridge/ebt_nat.h>
#include <linux/module.h>
#include <net/sock.h>
16
17
#include <linux/if_arp.h>
#include <net/arp.h>
Linus Torvalds's avatar
Linus Torvalds committed
18

19
static int ebt_target_snat(struct sk_buff *skb, unsigned int hooknr,
Linus Torvalds's avatar
Linus Torvalds committed
20
21
22
   const struct net_device *in, const struct net_device *out,
   const void *data, unsigned int datalen)
{
23
	const struct ebt_nat_info *info = data;
Linus Torvalds's avatar
Linus Torvalds committed
24

25
	if (!skb_make_writable(skb, 0))
26
		return NF_DROP;
Linus Torvalds's avatar
Linus Torvalds committed
27

28
	memcpy(eth_hdr(skb)->h_source, info->mac, ETH_ALEN);
29
	if (!(info->target & NAT_ARP_BIT) &&
30
	    eth_hdr(skb)->h_proto == htons(ETH_P_ARP)) {
31
32
		const struct arphdr *ap;
		struct arphdr _ah;
33

34
		ap = skb_header_pointer(skb, 0, sizeof(_ah), &_ah);
35
36
37
38
		if (ap == NULL)
			return EBT_DROP;
		if (ap->ar_hln != ETH_ALEN)
			goto out;
39
		if (skb_store_bits(skb, sizeof(_ah), info->mac,ETH_ALEN))
40
41
42
43
			return EBT_DROP;
	}
out:
	return info->target | ~EBT_VERDICT_BITS;
Linus Torvalds's avatar
Linus Torvalds committed
44
45
46
47
48
}

static int ebt_target_snat_check(const char *tablename, unsigned int hookmask,
   const struct ebt_entry *e, void *data, unsigned int datalen)
{
49
	const struct ebt_nat_info *info = data;
50
	int tmp;
Linus Torvalds's avatar
Linus Torvalds committed
51
52
53

	if (datalen != EBT_ALIGN(sizeof(struct ebt_nat_info)))
		return -EINVAL;
54
55
	tmp = info->target | ~EBT_VERDICT_BITS;
	if (BASE_CHAIN && tmp == EBT_RETURN)
Linus Torvalds's avatar
Linus Torvalds committed
56
57
58
59
60
61
		return -EINVAL;
	CLEAR_BASE_CHAIN_BIT;
	if (strcmp(tablename, "nat"))
		return -EINVAL;
	if (hookmask & ~(1 << NF_BR_POST_ROUTING))
		return -EINVAL;
62
63
64
65
66

	if (tmp < -NUM_STANDARD_TARGETS || tmp >= 0)
		return -EINVAL;
	tmp = info->target | EBT_VERDICT_BITS;
	if ((tmp & ~NAT_ARP_BIT) != ~NAT_ARP_BIT)
Linus Torvalds's avatar
Linus Torvalds committed
67
68
69
70
		return -EINVAL;
	return 0;
}

71
static struct ebt_target snat __read_mostly = {
Linus Torvalds's avatar
Linus Torvalds committed
72
73
74
75
76
77
	.name		= EBT_SNAT_TARGET,
	.target		= ebt_target_snat,
	.check		= ebt_target_snat_check,
	.me		= THIS_MODULE,
};

78
static int __init ebt_snat_init(void)
Linus Torvalds's avatar
Linus Torvalds committed
79
80
81
82
{
	return ebt_register_target(&snat);
}

83
static void __exit ebt_snat_fini(void)
Linus Torvalds's avatar
Linus Torvalds committed
84
85
86
87
{
	ebt_unregister_target(&snat);
}

88
89
module_init(ebt_snat_init);
module_exit(ebt_snat_fini);
90
MODULE_DESCRIPTION("Ebtables: Source MAC address translation");
Linus Torvalds's avatar
Linus Torvalds committed
91
MODULE_LICENSE("GPL");