xref: /openbmc/linux/net/ipv6/ila/ila_lwt.c (revision fa0a497b)
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>
14 #include "ila.h"
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 
22 static int ila_output(struct net *net, struct sock *sk, struct sk_buff *skb)
23 {
24 	struct dst_entry *dst = skb_dst(skb);
25 
26 	if (skb->protocol != htons(ETH_P_IPV6))
27 		goto drop;
28 
29 	update_ipv6_locator(skb, ila_params_lwtunnel(dst->lwtstate));
30 
31 	return dst->lwtstate->orig_output(net, sk, skb);
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);
41 
42 	if (skb->protocol != htons(ETH_P_IPV6))
43 		goto drop;
44 
45 	update_ipv6_locator(skb, ila_params_lwtunnel(dst->lwtstate));
46 
47 	return dst->lwtstate->orig_input(skb);
48 
49 drop:
50 	kfree_skb(skb);
51 	return -EINVAL;
52 }
53 
54 static struct nla_policy ila_nl_policy[ILA_ATTR_MAX + 1] = {
55 	[ILA_ATTR_LOCATOR] = { .type = NLA_U64, },
56 };
57 
58 static int ila_build_state(struct net_device *dev, struct nlattr *nla,
59 			   unsigned int family, const void *cfg,
60 			   struct lwtunnel_state **ts)
61 {
62 	struct ila_params *p;
63 	struct nlattr *tb[ILA_ATTR_MAX + 1];
64 	size_t encap_len = sizeof(*p);
65 	struct lwtunnel_state *newts;
66 	const struct fib6_config *cfg6 = cfg;
67 	int ret;
68 
69 	if (family != AF_INET6)
70 		return -EINVAL;
71 
72 	ret = nla_parse_nested(tb, ILA_ATTR_MAX, nla,
73 			       ila_nl_policy);
74 	if (ret < 0)
75 		return ret;
76 
77 	if (!tb[ILA_ATTR_LOCATOR])
78 		return -EINVAL;
79 
80 	newts = lwtunnel_state_alloc(encap_len);
81 	if (!newts)
82 		return -ENOMEM;
83 
84 	newts->len = encap_len;
85 	p = ila_params_lwtunnel(newts);
86 
87 	p->locator = (__force __be64)nla_get_u64(tb[ILA_ATTR_LOCATOR]);
88 
89 	if (cfg6->fc_dst_len > sizeof(__be64)) {
90 		/* Precompute checksum difference for translation since we
91 		 * know both the old locator and the new one.
92 		 */
93 		p->locator_match = *(__be64 *)&cfg6->fc_dst;
94 		p->csum_diff = compute_csum_diff8(
95 			(__be32 *)&p->locator_match, (__be32 *)&p->locator);
96 	}
97 
98 	newts->type = LWTUNNEL_ENCAP_ILA;
99 	newts->flags |= LWTUNNEL_STATE_OUTPUT_REDIRECT |
100 			LWTUNNEL_STATE_INPUT_REDIRECT;
101 
102 	*ts = newts;
103 
104 	return 0;
105 }
106 
107 static int ila_fill_encap_info(struct sk_buff *skb,
108 			       struct lwtunnel_state *lwtstate)
109 {
110 	struct ila_params *p = ila_params_lwtunnel(lwtstate);
111 
112 	if (nla_put_u64(skb, ILA_ATTR_LOCATOR, (__force u64)p->locator))
113 		goto nla_put_failure;
114 
115 	return 0;
116 
117 nla_put_failure:
118 	return -EMSGSIZE;
119 }
120 
121 static int ila_encap_nlsize(struct lwtunnel_state *lwtstate)
122 {
123 	return nla_total_size(sizeof(u64)); /* ILA_ATTR_LOCATOR */
124 }
125 
126 static int ila_encap_cmp(struct lwtunnel_state *a, struct lwtunnel_state *b)
127 {
128 	struct ila_params *a_p = ila_params_lwtunnel(a);
129 	struct ila_params *b_p = ila_params_lwtunnel(b);
130 
131 	return (a_p->locator != b_p->locator);
132 }
133 
134 static const struct lwtunnel_encap_ops ila_encap_ops = {
135 	.build_state = ila_build_state,
136 	.output = ila_output,
137 	.input = ila_input,
138 	.fill_encap = ila_fill_encap_info,
139 	.get_encap_size = ila_encap_nlsize,
140 	.cmp_encap = ila_encap_cmp,
141 };
142 
143 int ila_lwt_init(void)
144 {
145 	return lwtunnel_encap_add_ops(&ila_encap_ops, LWTUNNEL_ENCAP_ILA);
146 }
147 
148 void ila_lwt_fini(void)
149 {
150 	lwtunnel_encap_del_ops(&ila_encap_ops, LWTUNNEL_ENCAP_ILA);
151 }
152