[net-next-2.6.git] / net / ipv6 / netfilter.c

#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/ipv6.h>
#include <linux/netfilter.h>
#include <linux/netfilter_ipv6.h>
#include <net/dst.h>
#include <net/ipv6.h>
#include <net/ip6_route.h>
#include <net/xfrm.h>
#include <net/ip6_checksum.h>

int ip6_route_me_harder(struct sk_buff *skb)
{
	struct ipv6hdr *iph = skb->nh.ipv6h;
	struct dst_entry *dst;
	struct flowi fl = {
		.oif = skb->sk ? skb->sk->sk_bound_dev_if : 0,
		.nl_u =
		{ .ip6_u =
		  { .daddr = iph->daddr,
		    .saddr = iph->saddr, } },
	};

	dst = ip6_route_output(skb->sk, &fl);

#ifdef CONFIG_XFRM
	if (!(IP6CB(skb)->flags & IP6SKB_XFRM_TRANSFORMED) &&
	    xfrm_decode_session(skb, &fl, AF_INET6) == 0)
		if (xfrm_lookup(&skb->dst, &fl, skb->sk, 0))
			return -1;
#endif

	if (dst->error) {
		IP6_INC_STATS(IPSTATS_MIB_OUTNOROUTES);
		LIMIT_NETDEBUG(KERN_DEBUG "ip6_route_me_harder: No more route.\n");
		dst_release(dst);
		return -EINVAL;
	}

	/* Drop old route. */
	dst_release(skb->dst);

	skb->dst = dst;
	return 0;
}
EXPORT_SYMBOL(ip6_route_me_harder);

/*
 * Extra routing may needed on local out, as the QUEUE target never
 * returns control to the table.
 */

struct ip6_rt_info {
	struct in6_addr daddr;
	struct in6_addr saddr;
};

static void nf_ip6_saveroute(const struct sk_buff *skb, struct nf_info *info)
{
	struct ip6_rt_info *rt_info = nf_info_reroute(info);

	if (info->hook == NF_IP6_LOCAL_OUT) {
		struct ipv6hdr *iph = skb->nh.ipv6h;

		rt_info->daddr = iph->daddr;
		rt_info->saddr = iph->saddr;
	}
}

static int nf_ip6_reroute(struct sk_buff **pskb, const struct nf_info *info)
{
	struct ip6_rt_info *rt_info = nf_info_reroute(info);

	if (info->hook == NF_IP6_LOCAL_OUT) {
		struct ipv6hdr *iph = (*pskb)->nh.ipv6h;
		if (!ipv6_addr_equal(&iph->daddr, &rt_info->daddr) ||
		    !ipv6_addr_equal(&iph->saddr, &rt_info->saddr))
			return ip6_route_me_harder(*pskb);
	}
	return 0;
}

unsigned int nf_ip6_checksum(struct sk_buff *skb, unsigned int hook,
			     unsigned int dataoff, u_int8_t protocol)
{
	struct ipv6hdr *ip6h = skb->nh.ipv6h;
	unsigned int csum = 0;

	switch (skb->ip_summed) {
	case CHECKSUM_HW:
		if (hook != NF_IP6_PRE_ROUTING && hook != NF_IP6_LOCAL_IN)
			break;
		if (!csum_ipv6_magic(&ip6h->saddr, &ip6h->daddr,
			    	     skb->len - dataoff, protocol,
				     csum_sub(skb->csum,
					      skb_checksum(skb, 0,
							   dataoff, 0)))) {
			skb->ip_summed = CHECKSUM_UNNECESSARY;
			break;
		}
		/* fall through */
	case CHECKSUM_NONE:
		skb->csum = ~csum_ipv6_magic(&ip6h->saddr, &ip6h->daddr,
					     skb->len - dataoff,
					     protocol,
					     csum_sub(0,
						      skb_checksum(skb, 0,
							           dataoff, 0)));
		csum = __skb_checksum_complete(skb);
	}
	return csum;
}

EXPORT_SYMBOL(nf_ip6_checksum);

static struct nf_afinfo nf_ip6_afinfo = {
	.family		= AF_INET6,
	.checksum	= nf_ip6_checksum,
	.saveroute	= nf_ip6_saveroute,
	.reroute	= nf_ip6_reroute,
	.route_key_size	= sizeof(struct ip6_rt_info),
};

int __init ipv6_netfilter_init(void)
{
	return nf_register_afinfo(&nf_ip6_afinfo);
}

/* This can be called from inet6_init() on errors, so it cannot
 * be marked __exit. -DaveM
 */
void ipv6_netfilter_fini(void)
{
	nf_unregister_afinfo(&nf_ip6_afinfo);
}
Commit	Line	Data
020b4c12	1	#include <linux/kernel.h>
bb94aa16	2	#include <linux/init.h>
020b4c12	3	#include <linux/ipv6.h>
2cc7d573 HW	4	#include <linux/netfilter.h>
2cc7d573 HW	5	#include <linux/netfilter_ipv6.h>
020b4c12 HW	6	#include <net/dst.h>
	7	#include <net/ipv6.h>
	8	#include <net/ip6_route.h>
3e3850e9	9	#include <net/xfrm.h>
503e4faa	10	#include <net/ip6_checksum.h>
020b4c12 HW	11
	12	int ip6_route_me_harder(struct sk_buff *skb)
	13	{
	14	struct ipv6hdr *iph = skb->nh.ipv6h;
	15	struct dst_entry *dst;
	16	struct flowi fl = {
	17	.oif = skb->sk ? skb->sk->sk_bound_dev_if : 0,
	18	.nl_u =
	19	{ .ip6_u =
	20	{ .daddr = iph->daddr,
	21	.saddr = iph->saddr, } },
020b4c12 HW	22	};
	23
	24	dst = ip6_route_output(skb->sk, &fl);
	25
3e3850e9 PM	26	#ifdef CONFIG_XFRM
	27	if (!(IP6CB(skb)->flags & IP6SKB_XFRM_TRANSFORMED) &&
	28	xfrm_decode_session(skb, &fl, AF_INET6) == 0)
	29	if (xfrm_lookup(&skb->dst, &fl, skb->sk, 0))
	30	return -1;
	31	#endif
	32
020b4c12 HW	33	if (dst->error) {
020b4c12 HW	34	IP6_INC_STATS(IPSTATS_MIB_OUTNOROUTES);
64ce2073	35	LIMIT_NETDEBUG(KERN_DEBUG "ip6_route_me_harder: No more route.\n");
020b4c12 HW	36	dst_release(dst);
	37	return -EINVAL;
	38	}
	39
	40	/* Drop old route. */
	41	dst_release(skb->dst);
	42
	43	skb->dst = dst;
	44	return 0;
	45	}
	46	EXPORT_SYMBOL(ip6_route_me_harder);
	47
2cc7d573 HW	48	/*
	49	* Extra routing may needed on local out, as the QUEUE target never
	50	* returns control to the table.
	51	*/
	52
	53	struct ip6_rt_info {
	54	struct in6_addr daddr;
	55	struct in6_addr saddr;
	56	};
	57
bce8032e	58	static void nf_ip6_saveroute(const struct sk_buff skb, struct nf_info info)
2cc7d573 HW	59	{
	60	struct ip6_rt_info *rt_info = nf_info_reroute(info);
	61
	62	if (info->hook == NF_IP6_LOCAL_OUT) {
	63	struct ipv6hdr *iph = skb->nh.ipv6h;
	64
	65	rt_info->daddr = iph->daddr;
	66	rt_info->saddr = iph->saddr;
	67	}
	68	}
	69
bce8032e	70	static int nf_ip6_reroute(struct sk_buff *pskb, const struct nf_info info)
2cc7d573 HW	71	{
	72	struct ip6_rt_info *rt_info = nf_info_reroute(info);
	73
	74	if (info->hook == NF_IP6_LOCAL_OUT) {
	75	struct ipv6hdr iph = (pskb)->nh.ipv6h;
	76	if (!ipv6_addr_equal(&iph->daddr, &rt_info->daddr) \|\|
	77	!ipv6_addr_equal(&iph->saddr, &rt_info->saddr))
	78	return ip6_route_me_harder(*pskb);
	79	}
	80	return 0;
	81	}
	82
422c346f PM	83	unsigned int nf_ip6_checksum(struct sk_buff *skb, unsigned int hook,
	84	unsigned int dataoff, u_int8_t protocol)
	85	{
	86	struct ipv6hdr *ip6h = skb->nh.ipv6h;
	87	unsigned int csum = 0;
	88
	89	switch (skb->ip_summed) {
	90	case CHECKSUM_HW:
	91	if (hook != NF_IP6_PRE_ROUTING && hook != NF_IP6_LOCAL_IN)
	92	break;
	93	if (!csum_ipv6_magic(&ip6h->saddr, &ip6h->daddr,
	94	skb->len - dataoff, protocol,
	95	csum_sub(skb->csum,
	96	skb_checksum(skb, 0,
	97	dataoff, 0)))) {
	98	skb->ip_summed = CHECKSUM_UNNECESSARY;
	99	break;
	100	}
	101	/* fall through */
	102	case CHECKSUM_NONE:
	103	skb->csum = ~csum_ipv6_magic(&ip6h->saddr, &ip6h->daddr,
	104	skb->len - dataoff,
	105	protocol,
	106	csum_sub(0,
	107	skb_checksum(skb, 0,
	108	dataoff, 0)));
	109	csum = __skb_checksum_complete(skb);
	110	}
	111	return csum;
	112	}
	113
	114	EXPORT_SYMBOL(nf_ip6_checksum);
	115
bce8032e PM	116	static struct nf_afinfo nf_ip6_afinfo = {
bce8032e PM	117	.family = AF_INET6,
422c346f	118	.checksum = nf_ip6_checksum,
bce8032e PM	119	.saveroute = nf_ip6_saveroute,
	120	.reroute = nf_ip6_reroute,
	121	.route_key_size = sizeof(struct ip6_rt_info),
2cc7d573 HW	122	};
	123
	124	int __init ipv6_netfilter_init(void)
	125	{
bce8032e	126	return nf_register_afinfo(&nf_ip6_afinfo);
2cc7d573 HW	127	}
2cc7d573 HW	128
5bf887f2 DM	129	/* This can be called from inet6_init() on errors, so it cannot
	130	* be marked __exit. -DaveM
	131	*/
	132	void ipv6_netfilter_fini(void)
2cc7d573	133	{
bce8032e	134	nf_unregister_afinfo(&nf_ip6_afinfo);
2cc7d573	135	}