[deliverable/linux.git] / net / ipv6 / ila / ila_lwt.c

#include <linux/errno.h>
#include <linux/ip.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/skbuff.h>
#include <linux/socket.h>
#include <linux/types.h>
#include <net/checksum.h>
#include <net/ip.h>
#include <net/ip6_fib.h>
#include <net/lwtunnel.h>
#include <net/protocol.h>
#include <uapi/linux/ila.h>
#include "ila.h"

static inline struct ila_params *ila_params_lwtunnel(
	struct lwtunnel_state *lwstate)
{
	return (struct ila_params *)lwstate->data;
}

static int ila_output(struct net *net, struct sock *sk, struct sk_buff *skb)
{
	struct dst_entry *dst = skb_dst(skb);

	if (skb->protocol != htons(ETH_P_IPV6))
		goto drop;

	ila_update_ipv6_locator(skb, ila_params_lwtunnel(dst->lwtstate), true);

	return dst->lwtstate->orig_output(net, sk, skb);

drop:
	kfree_skb(skb);
	return -EINVAL;
}

static int ila_input(struct sk_buff *skb)
{
	struct dst_entry *dst = skb_dst(skb);

	if (skb->protocol != htons(ETH_P_IPV6))
		goto drop;

	ila_update_ipv6_locator(skb, ila_params_lwtunnel(dst->lwtstate), false);

	return dst->lwtstate->orig_input(skb);

drop:
	kfree_skb(skb);
	return -EINVAL;
}

static const struct nla_policy ila_nl_policy[ILA_ATTR_MAX + 1] = {
	[ILA_ATTR_LOCATOR] = { .type = NLA_U64, },
	[ILA_ATTR_CSUM_MODE] = { .type = NLA_U8, },
};

static int ila_build_state(struct net_device *dev, struct nlattr *nla,
			   unsigned int family, const void *cfg,
			   struct lwtunnel_state **ts)
{
	struct ila_params *p;
	struct nlattr *tb[ILA_ATTR_MAX + 1];
	size_t encap_len = sizeof(*p);
	struct lwtunnel_state *newts;
	const struct fib6_config *cfg6 = cfg;
	struct ila_addr *iaddr;
	int ret;

	if (family != AF_INET6)
		return -EINVAL;

	if (cfg6->fc_dst_len < sizeof(struct ila_locator) + 1) {
		/* Need to have full locator and at least type field
		 * included in destination
		 */
		return -EINVAL;
	}

	iaddr = (struct ila_addr *)&cfg6->fc_dst;

	if (!ila_addr_is_ila(iaddr) || ila_csum_neutral_set(iaddr->ident)) {
		/* Don't allow translation for a non-ILA address or checksum
		 * neutral flag to be set.
		 */
		return -EINVAL;
	}

	ret = nla_parse_nested(tb, ILA_ATTR_MAX, nla,
			       ila_nl_policy);
	if (ret < 0)
		return ret;

	if (!tb[ILA_ATTR_LOCATOR])
		return -EINVAL;

	newts = lwtunnel_state_alloc(encap_len);
	if (!newts)
		return -ENOMEM;

	newts->len = encap_len;
	p = ila_params_lwtunnel(newts);

	p->locator.v64 = (__force __be64)nla_get_u64(tb[ILA_ATTR_LOCATOR]);

	/* Precompute checksum difference for translation since we
	 * know both the old locator and the new one.
	 */
	p->locator_match = iaddr->loc;
	p->csum_diff = compute_csum_diff8(
		(__be32 *)&p->locator_match, (__be32 *)&p->locator);

	if (tb[ILA_ATTR_CSUM_MODE])
		p->csum_mode = nla_get_u8(tb[ILA_ATTR_CSUM_MODE]);

	ila_init_saved_csum(p);

	newts->type = LWTUNNEL_ENCAP_ILA;
	newts->flags |= LWTUNNEL_STATE_OUTPUT_REDIRECT |
			LWTUNNEL_STATE_INPUT_REDIRECT;

	*ts = newts;

	return 0;
}

static int ila_fill_encap_info(struct sk_buff *skb,
			       struct lwtunnel_state *lwtstate)
{
	struct ila_params *p = ila_params_lwtunnel(lwtstate);

	if (nla_put_u64_64bit(skb, ILA_ATTR_LOCATOR, (__force u64)p->locator.v64,
			      ILA_ATTR_PAD))
		goto nla_put_failure;
	if (nla_put_u8(skb, ILA_ATTR_CSUM_MODE, (__force u8)p->csum_mode))
		goto nla_put_failure;

	return 0;

nla_put_failure:
	return -EMSGSIZE;
}

static int ila_encap_nlsize(struct lwtunnel_state *lwtstate)
{
	return nla_total_size_64bit(sizeof(u64)) + /* ILA_ATTR_LOCATOR */
	       nla_total_size(sizeof(u8)) +        /* ILA_ATTR_CSUM_MODE */
	       0;
}

static int ila_encap_cmp(struct lwtunnel_state *a, struct lwtunnel_state *b)
{
	struct ila_params *a_p = ila_params_lwtunnel(a);
	struct ila_params *b_p = ila_params_lwtunnel(b);

	return (a_p->locator.v64 != b_p->locator.v64);
}

static const struct lwtunnel_encap_ops ila_encap_ops = {
	.build_state = ila_build_state,
	.output = ila_output,
	.input = ila_input,
	.fill_encap = ila_fill_encap_info,
	.get_encap_size = ila_encap_nlsize,
	.cmp_encap = ila_encap_cmp,
};

int ila_lwt_init(void)
{
	return lwtunnel_encap_add_ops(&ila_encap_ops, LWTUNNEL_ENCAP_ILA);
}

void ila_lwt_fini(void)
{
	lwtunnel_encap_del_ops(&ila_encap_ops, LWTUNNEL_ENCAP_ILA);
}
Commit	Line	Data
65d7ab8d TH	1	#include <linux/errno.h>
	2	#include <linux/ip.h>
	3	#include <linux/kernel.h>
	4	#include <linux/module.h>
	5	#include <linux/skbuff.h>
	6	#include <linux/socket.h>
	7	#include <linux/types.h>
	8	#include <net/checksum.h>
	9	#include <net/ip.h>
	10	#include <net/ip6_fib.h>
	11	#include <net/lwtunnel.h>
	12	#include <net/protocol.h>
	13	#include <uapi/linux/ila.h>
33f11d16	14	#include "ila.h"
65d7ab8d TH	15
	16	static inline struct ila_params *ila_params_lwtunnel(
	17	struct lwtunnel_state *lwstate)
	18	{
	19	return (struct ila_params *)lwstate->data;
	20	}
	21
ede2059d	22	static int ila_output(struct net net, struct sock sk, struct sk_buff *skb)
65d7ab8d TH	23	{
65d7ab8d TH	24	struct dst_entry *dst = skb_dst(skb);
65d7ab8d TH	25
	26	if (skb->protocol != htons(ETH_P_IPV6))
	27	goto drop;
	28
707a2ca4	29	ila_update_ipv6_locator(skb, ila_params_lwtunnel(dst->lwtstate), true);
65d7ab8d	30
ede2059d	31	return dst->lwtstate->orig_output(net, sk, skb);
65d7ab8d TH	32
	33	drop:
	34	kfree_skb(skb);
	35	return -EINVAL;
	36	}
	37
	38	static int ila_input(struct sk_buff *skb)
	39	{
	40	struct dst_entry *dst = skb_dst(skb);
65d7ab8d TH	41
	42	if (skb->protocol != htons(ETH_P_IPV6))
	43	goto drop;
	44
707a2ca4	45	ila_update_ipv6_locator(skb, ila_params_lwtunnel(dst->lwtstate), false);
65d7ab8d	46
61adedf3	47	return dst->lwtstate->orig_input(skb);
65d7ab8d TH	48
	49	drop:
	50	kfree_skb(skb);
	51	return -EINVAL;
	52	}
	53
6501f34f	54	static const struct nla_policy ila_nl_policy[ILA_ATTR_MAX + 1] = {
65d7ab8d	55	[ILA_ATTR_LOCATOR] = { .type = NLA_U64, },
90bfe662	56	[ILA_ATTR_CSUM_MODE] = { .type = NLA_U8, },
65d7ab8d TH	57	};
	58
	59	static int ila_build_state(struct net_device dev, struct nlattr nla,
127eb7cd	60	unsigned int family, const void *cfg,
65d7ab8d TH	61	struct lwtunnel_state **ts)
	62	{
	63	struct ila_params *p;
	64	struct nlattr *tb[ILA_ATTR_MAX + 1];
	65	size_t encap_len = sizeof(*p);
	66	struct lwtunnel_state *newts;
92b78aff	67	const struct fib6_config *cfg6 = cfg;
351596aa	68	struct ila_addr *iaddr;
65d7ab8d TH	69	int ret;
65d7ab8d TH	70
92b78aff TH	71	if (family != AF_INET6)
	72	return -EINVAL;
	73
351596aa TH	74	if (cfg6->fc_dst_len < sizeof(struct ila_locator) + 1) {
	75	/* Need to have full locator and at least type field
	76	* included in destination
	77	*/
	78	return -EINVAL;
	79	}
	80
	81	iaddr = (struct ila_addr *)&cfg6->fc_dst;
	82
90bfe662 TH	83	if (!ila_addr_is_ila(iaddr) \|\| ila_csum_neutral_set(iaddr->ident)) {
	84	/* Don't allow translation for a non-ILA address or checksum
	85	* neutral flag to be set.
	86	*/
351596aa TH	87	return -EINVAL;
	88	}
	89
65d7ab8d TH	90	ret = nla_parse_nested(tb, ILA_ATTR_MAX, nla,
	91	ila_nl_policy);
	92	if (ret < 0)
	93	return ret;
	94
	95	if (!tb[ILA_ATTR_LOCATOR])
	96	return -EINVAL;
	97
	98	newts = lwtunnel_state_alloc(encap_len);
	99	if (!newts)
	100	return -ENOMEM;
	101
	102	newts->len = encap_len;
	103	p = ila_params_lwtunnel(newts);
	104
351596aa	105	p->locator.v64 = (__force __be64)nla_get_u64(tb[ILA_ATTR_LOCATOR]);
65d7ab8d	106
351596aa TH	107	/* Precompute checksum difference for translation since we
	108	* know both the old locator and the new one.
	109	*/
	110	p->locator_match = iaddr->loc;
	111	p->csum_diff = compute_csum_diff8(
	112	(__be32 )&p->locator_match, (__be32 )&p->locator);
92b78aff	113
90bfe662 TH	114	if (tb[ILA_ATTR_CSUM_MODE])
	115	p->csum_mode = nla_get_u8(tb[ILA_ATTR_CSUM_MODE]);
	116
	117	ila_init_saved_csum(p);
	118
65d7ab8d TH	119	newts->type = LWTUNNEL_ENCAP_ILA;
	120	newts->flags \|= LWTUNNEL_STATE_OUTPUT_REDIRECT \|
	121	LWTUNNEL_STATE_INPUT_REDIRECT;
	122
	123	*ts = newts;
	124
	125	return 0;
	126	}
	127
	128	static int ila_fill_encap_info(struct sk_buff *skb,
	129	struct lwtunnel_state *lwtstate)
	130	{
	131	struct ila_params *p = ila_params_lwtunnel(lwtstate);
	132
351596aa	133	if (nla_put_u64_64bit(skb, ILA_ATTR_LOCATOR, (__force u64)p->locator.v64,
f13a82d8	134	ILA_ATTR_PAD))
65d7ab8d	135	goto nla_put_failure;
1ddb6b71	136	if (nla_put_u8(skb, ILA_ATTR_CSUM_MODE, (__force u8)p->csum_mode))
90bfe662	137	goto nla_put_failure;
65d7ab8d TH	138
	139	return 0;
	140
	141	nla_put_failure:
	142	return -EMSGSIZE;
	143	}
	144
	145	static int ila_encap_nlsize(struct lwtunnel_state *lwtstate)
	146	{
1ddb6b71 TH	147	return nla_total_size_64bit(sizeof(u64)) + /* ILA_ATTR_LOCATOR */
	148	nla_total_size(sizeof(u8)) + /* ILA_ATTR_CSUM_MODE */
	149	0;
65d7ab8d TH	150	}
	151
	152	static int ila_encap_cmp(struct lwtunnel_state a, struct lwtunnel_state b)
	153	{
	154	struct ila_params *a_p = ila_params_lwtunnel(a);
	155	struct ila_params *b_p = ila_params_lwtunnel(b);
	156
351596aa	157	return (a_p->locator.v64 != b_p->locator.v64);
65d7ab8d TH	158	}
	159
	160	static const struct lwtunnel_encap_ops ila_encap_ops = {
	161	.build_state = ila_build_state,
	162	.output = ila_output,
	163	.input = ila_input,
	164	.fill_encap = ila_fill_encap_info,
	165	.get_encap_size = ila_encap_nlsize,
	166	.cmp_encap = ila_encap_cmp,
	167	};
	168
33f11d16	169	int ila_lwt_init(void)
65d7ab8d TH	170	{
	171	return lwtunnel_encap_add_ops(&ila_encap_ops, LWTUNNEL_ENCAP_ILA);
	172	}
	173
33f11d16	174	void ila_lwt_fini(void)
65d7ab8d TH	175	{
	176	lwtunnel_encap_del_ops(&ila_encap_ops, LWTUNNEL_ENCAP_ILA);
	177	}