[XFRM] IPV6: Add sort functions to combine templates/states for IPsec.
[safe/jmp/linux-2.6] / net / ipv6 / xfrm6_state.c
1 /*
2  * xfrm6_state.c: based on xfrm4_state.c
3  *
4  * Authors:
5  *      Mitsuru KANDA @USAGI
6  *      Kazunori MIYAZAWA @USAGI
7  *      Kunihiro Ishiguro <kunihiro@ipinfusion.com>
8  *              IPv6 support
9  *      YOSHIFUJI Hideaki @USAGI
10  *              Split up af-specific portion
11  *      
12  */
13
14 #include <net/xfrm.h>
15 #include <linux/pfkeyv2.h>
16 #include <linux/ipsec.h>
17 #include <net/ipv6.h>
18 #include <net/addrconf.h>
19
20 static struct xfrm_state_afinfo xfrm6_state_afinfo;
21
22 static void
23 __xfrm6_init_tempsel(struct xfrm_state *x, struct flowi *fl,
24                      struct xfrm_tmpl *tmpl,
25                      xfrm_address_t *daddr, xfrm_address_t *saddr)
26 {
27         /* Initialize temporary selector matching only
28          * to current session. */
29         ipv6_addr_copy((struct in6_addr *)&x->sel.daddr, &fl->fl6_dst);
30         ipv6_addr_copy((struct in6_addr *)&x->sel.saddr, &fl->fl6_src);
31         x->sel.dport = xfrm_flowi_dport(fl);
32         x->sel.dport_mask = ~0;
33         x->sel.sport = xfrm_flowi_sport(fl);
34         x->sel.sport_mask = ~0;
35         x->sel.prefixlen_d = 128;
36         x->sel.prefixlen_s = 128;
37         x->sel.proto = fl->proto;
38         x->sel.ifindex = fl->oif;
39         x->id = tmpl->id;
40         if (ipv6_addr_any((struct in6_addr*)&x->id.daddr))
41                 memcpy(&x->id.daddr, daddr, sizeof(x->sel.daddr));
42         memcpy(&x->props.saddr, &tmpl->saddr, sizeof(x->props.saddr));
43         if (ipv6_addr_any((struct in6_addr*)&x->props.saddr))
44                 memcpy(&x->props.saddr, saddr, sizeof(x->props.saddr));
45         if (tmpl->mode == XFRM_MODE_TUNNEL && ipv6_addr_any((struct in6_addr*)&x->props.saddr)) {
46                 struct rt6_info *rt;
47                 struct flowi fl_tunnel = {
48                         .nl_u = {
49                                 .ip6_u = {
50                                         .daddr = *(struct in6_addr *)daddr,
51                                 }
52                         }
53                 };
54                 if (!xfrm_dst_lookup((struct xfrm_dst **)&rt,
55                                      &fl_tunnel, AF_INET6)) {
56                         ipv6_get_saddr(&rt->u.dst, (struct in6_addr *)daddr,
57                                        (struct in6_addr *)&x->props.saddr);
58                         dst_release(&rt->u.dst);
59                 }
60         }
61         x->props.mode = tmpl->mode;
62         x->props.reqid = tmpl->reqid;
63         x->props.family = AF_INET6;
64 }
65
66 static struct xfrm_state *
67 __xfrm6_state_lookup_byaddr(xfrm_address_t *daddr, xfrm_address_t *saddr,
68                             u8 proto)
69 {
70         struct xfrm_state *x = NULL;
71         unsigned h;
72
73         h = __xfrm6_src_hash(saddr);
74         list_for_each_entry(x, xfrm6_state_afinfo.state_bysrc+h, bysrc) {
75                 if (x->props.family == AF_INET6 &&
76                     ipv6_addr_equal((struct in6_addr *)daddr, (struct in6_addr *)x->id.daddr.a6) &&
77                     ipv6_addr_equal((struct in6_addr *)saddr, (struct in6_addr *)x->props.saddr.a6) &&
78                     proto == x->id.proto) {
79                         xfrm_state_hold(x);
80                         return x;
81                 }
82         }
83         return NULL;
84 }
85
86 static struct xfrm_state *
87 __xfrm6_state_lookup(xfrm_address_t *daddr, u32 spi, u8 proto)
88 {
89         unsigned h = __xfrm6_spi_hash(daddr, spi, proto);
90         struct xfrm_state *x;
91
92         list_for_each_entry(x, xfrm6_state_afinfo.state_byspi+h, byspi) {
93                 if (x->props.family == AF_INET6 &&
94                     spi == x->id.spi &&
95                     ipv6_addr_equal((struct in6_addr *)daddr, (struct in6_addr *)x->id.daddr.a6) &&
96                     proto == x->id.proto) {
97                         xfrm_state_hold(x);
98                         return x;
99                 }
100         }
101         return NULL;
102 }
103
104 static struct xfrm_state *
105 __xfrm6_find_acq(u8 mode, u32 reqid, u8 proto, 
106                  xfrm_address_t *daddr, xfrm_address_t *saddr, 
107                  int create)
108 {
109         struct xfrm_state *x, *x0;
110         unsigned h = __xfrm6_dst_hash(daddr);
111
112         x0 = NULL;
113
114         list_for_each_entry(x, xfrm6_state_afinfo.state_bydst+h, bydst) {
115                 if (x->props.family == AF_INET6 &&
116                     ipv6_addr_equal((struct in6_addr *)daddr, (struct in6_addr *)x->id.daddr.a6) &&
117                     mode == x->props.mode &&
118                     proto == x->id.proto &&
119                     ipv6_addr_equal((struct in6_addr *)saddr, (struct in6_addr *)x->props.saddr.a6) &&
120                     reqid == x->props.reqid &&
121                     x->km.state == XFRM_STATE_ACQ &&
122                     !x->id.spi) {
123                             x0 = x;
124                             break;
125                     }
126         }
127         if (!x0 && create && (x0 = xfrm_state_alloc()) != NULL) {
128                 ipv6_addr_copy((struct in6_addr *)x0->sel.daddr.a6,
129                                (struct in6_addr *)daddr);
130                 ipv6_addr_copy((struct in6_addr *)x0->sel.saddr.a6,
131                                (struct in6_addr *)saddr);
132                 x0->sel.prefixlen_d = 128;
133                 x0->sel.prefixlen_s = 128;
134                 ipv6_addr_copy((struct in6_addr *)x0->props.saddr.a6,
135                                (struct in6_addr *)saddr);
136                 x0->km.state = XFRM_STATE_ACQ;
137                 ipv6_addr_copy((struct in6_addr *)x0->id.daddr.a6,
138                                (struct in6_addr *)daddr);
139                 x0->id.proto = proto;
140                 x0->props.family = AF_INET6;
141                 x0->props.mode = mode;
142                 x0->props.reqid = reqid;
143                 x0->lft.hard_add_expires_seconds = XFRM_ACQ_EXPIRES;
144                 xfrm_state_hold(x0);
145                 x0->timer.expires = jiffies + XFRM_ACQ_EXPIRES*HZ;
146                 add_timer(&x0->timer);
147                 xfrm_state_hold(x0);
148                 list_add_tail(&x0->bydst, xfrm6_state_afinfo.state_bydst+h);
149                 h = __xfrm6_src_hash(saddr);
150                 xfrm_state_hold(x0);
151                 list_add_tail(&x0->bysrc, xfrm6_state_afinfo.state_bysrc+h);
152                 wake_up(&km_waitq);
153         }
154         if (x0)
155                 xfrm_state_hold(x0);
156         return x0;
157 }
158
159 static int
160 __xfrm6_state_sort(struct xfrm_state **dst, struct xfrm_state **src, int n)
161 {
162         int i;
163         int j = 0;
164
165         /* Rule 1: select IPsec transport except AH */
166         for (i = 0; i < n; i++) {
167                 if (src[i]->props.mode == XFRM_MODE_TRANSPORT &&
168                     src[i]->id.proto != IPPROTO_AH) {
169                         dst[j++] = src[i];
170                         src[i] = NULL;
171                 }
172         }
173         if (j == n)
174                 goto end;
175
176         /* XXX: Rule 2: select MIPv6 RO or inbound trigger */
177
178         /* Rule 3: select IPsec transport AH */
179         for (i = 0; i < n; i++) {
180                 if (src[i] &&
181                     src[i]->props.mode == XFRM_MODE_TRANSPORT &&
182                     src[i]->id.proto == IPPROTO_AH) {
183                         dst[j++] = src[i];
184                         src[i] = NULL;
185                 }
186         }
187         if (j == n)
188                 goto end;
189
190         /* Rule 4: select IPsec tunnel */
191         for (i = 0; i < n; i++) {
192                 if (src[i] &&
193                     src[i]->props.mode == XFRM_MODE_TUNNEL) {
194                         dst[j++] = src[i];
195                         src[i] = NULL;
196                 }
197         }
198         if (likely(j == n))
199                 goto end;
200
201         /* Final rule */
202         for (i = 0; i < n; i++) {
203                 if (src[i]) {
204                         dst[j++] = src[i];
205                         src[i] = NULL;
206                 }
207         }
208
209  end:
210         return 0;
211 }
212
213 static int
214 __xfrm6_tmpl_sort(struct xfrm_tmpl **dst, struct xfrm_tmpl **src, int n)
215 {
216         int i;
217         int j = 0;
218
219         /* Rule 1: select IPsec transport */
220         for (i = 0; i < n; i++) {
221                 if (src[i]->mode == XFRM_MODE_TRANSPORT) {
222                         dst[j++] = src[i];
223                         src[i] = NULL;
224                 }
225         }
226         if (j == n)
227                 goto end;
228
229         /* XXX: Rule 2: select MIPv6 RO or inbound trigger */
230
231         /* Rule 3: select IPsec tunnel */
232         for (i = 0; i < n; i++) {
233                 if (src[i] &&
234                     src[i]->mode == XFRM_MODE_TUNNEL) {
235                         dst[j++] = src[i];
236                         src[i] = NULL;
237                 }
238         }
239         if (likely(j == n))
240                 goto end;
241
242         /* Final rule */
243         for (i = 0; i < n; i++) {
244                 if (src[i]) {
245                         dst[j++] = src[i];
246                         src[i] = NULL;
247                 }
248         }
249
250  end:
251         return 0;
252 }
253
254 static struct xfrm_state_afinfo xfrm6_state_afinfo = {
255         .family                 = AF_INET6,
256         .init_tempsel           = __xfrm6_init_tempsel,
257         .state_lookup           = __xfrm6_state_lookup,
258         .state_lookup_byaddr    = __xfrm6_state_lookup_byaddr,
259         .find_acq               = __xfrm6_find_acq,
260         .tmpl_sort              = __xfrm6_tmpl_sort,
261         .state_sort             = __xfrm6_state_sort,
262 };
263
264 void __init xfrm6_state_init(void)
265 {
266         xfrm_state_register_afinfo(&xfrm6_state_afinfo);
267 }
268
269 void xfrm6_state_fini(void)
270 {
271         xfrm_state_unregister_afinfo(&xfrm6_state_afinfo);
272 }
273