xfrm_user: validate XFRM_MSG_NEWAE XFRMA_REPLAY_ESN_VAL replay_window
[pandora-kernel.git] / net / xfrm / xfrm_user.c
1 /* xfrm_user.c: User interface to configure xfrm engine.
2  *
3  * Copyright (C) 2002 David S. Miller (davem@redhat.com)
4  *
5  * Changes:
6  *      Mitsuru KANDA @USAGI
7  *      Kazunori MIYAZAWA @USAGI
8  *      Kunihiro Ishiguro <kunihiro@ipinfusion.com>
9  *              IPv6 support
10  *
11  */
12
13 #include <linux/crypto.h>
14 #include <linux/module.h>
15 #include <linux/kernel.h>
16 #include <linux/types.h>
17 #include <linux/slab.h>
18 #include <linux/socket.h>
19 #include <linux/string.h>
20 #include <linux/net.h>
21 #include <linux/skbuff.h>
22 #include <linux/pfkeyv2.h>
23 #include <linux/ipsec.h>
24 #include <linux/init.h>
25 #include <linux/security.h>
26 #include <net/sock.h>
27 #include <net/xfrm.h>
28 #include <net/netlink.h>
29 #include <net/ah.h>
30 #include <asm/uaccess.h>
31 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
32 #include <linux/in6.h>
33 #endif
34
35 static inline int aead_len(struct xfrm_algo_aead *alg)
36 {
37         return sizeof(*alg) + ((alg->alg_key_len + 7) / 8);
38 }
39
40 static int verify_one_alg(struct nlattr **attrs, enum xfrm_attr_type_t type)
41 {
42         struct nlattr *rt = attrs[type];
43         struct xfrm_algo *algp;
44
45         if (!rt)
46                 return 0;
47
48         algp = nla_data(rt);
49         if (nla_len(rt) < xfrm_alg_len(algp))
50                 return -EINVAL;
51
52         switch (type) {
53         case XFRMA_ALG_AUTH:
54         case XFRMA_ALG_CRYPT:
55         case XFRMA_ALG_COMP:
56                 break;
57
58         default:
59                 return -EINVAL;
60         }
61
62         algp->alg_name[CRYPTO_MAX_ALG_NAME - 1] = '\0';
63         return 0;
64 }
65
66 static int verify_auth_trunc(struct nlattr **attrs)
67 {
68         struct nlattr *rt = attrs[XFRMA_ALG_AUTH_TRUNC];
69         struct xfrm_algo_auth *algp;
70
71         if (!rt)
72                 return 0;
73
74         algp = nla_data(rt);
75         if (nla_len(rt) < xfrm_alg_auth_len(algp))
76                 return -EINVAL;
77
78         algp->alg_name[CRYPTO_MAX_ALG_NAME - 1] = '\0';
79         return 0;
80 }
81
82 static int verify_aead(struct nlattr **attrs)
83 {
84         struct nlattr *rt = attrs[XFRMA_ALG_AEAD];
85         struct xfrm_algo_aead *algp;
86
87         if (!rt)
88                 return 0;
89
90         algp = nla_data(rt);
91         if (nla_len(rt) < aead_len(algp))
92                 return -EINVAL;
93
94         algp->alg_name[CRYPTO_MAX_ALG_NAME - 1] = '\0';
95         return 0;
96 }
97
98 static void verify_one_addr(struct nlattr **attrs, enum xfrm_attr_type_t type,
99                            xfrm_address_t **addrp)
100 {
101         struct nlattr *rt = attrs[type];
102
103         if (rt && addrp)
104                 *addrp = nla_data(rt);
105 }
106
107 static inline int verify_sec_ctx_len(struct nlattr **attrs)
108 {
109         struct nlattr *rt = attrs[XFRMA_SEC_CTX];
110         struct xfrm_user_sec_ctx *uctx;
111
112         if (!rt)
113                 return 0;
114
115         uctx = nla_data(rt);
116         if (uctx->len != (sizeof(struct xfrm_user_sec_ctx) + uctx->ctx_len))
117                 return -EINVAL;
118
119         return 0;
120 }
121
122 static inline int verify_replay(struct xfrm_usersa_info *p,
123                                 struct nlattr **attrs)
124 {
125         struct nlattr *rt = attrs[XFRMA_REPLAY_ESN_VAL];
126         struct xfrm_replay_state_esn *rs;
127
128         if (p->flags & XFRM_STATE_ESN) {
129                 if (!rt)
130                         return -EINVAL;
131
132                 rs = nla_data(rt);
133
134                 if (rs->bmp_len > XFRMA_REPLAY_ESN_MAX / sizeof(rs->bmp[0]) / 8)
135                         return -EINVAL;
136
137                 if (nla_len(rt) < xfrm_replay_state_esn_len(rs) &&
138                     nla_len(rt) != sizeof(*rs))
139                         return -EINVAL;
140         }
141
142         if (!rt)
143                 return 0;
144
145         if (p->id.proto != IPPROTO_ESP)
146                 return -EINVAL;
147
148         if (p->replay_window != 0)
149                 return -EINVAL;
150
151         return 0;
152 }
153
154 static int verify_newsa_info(struct xfrm_usersa_info *p,
155                              struct nlattr **attrs)
156 {
157         int err;
158
159         err = -EINVAL;
160         switch (p->family) {
161         case AF_INET:
162                 break;
163
164         case AF_INET6:
165 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
166                 break;
167 #else
168                 err = -EAFNOSUPPORT;
169                 goto out;
170 #endif
171
172         default:
173                 goto out;
174         }
175
176         err = -EINVAL;
177         switch (p->id.proto) {
178         case IPPROTO_AH:
179                 if ((!attrs[XFRMA_ALG_AUTH]     &&
180                      !attrs[XFRMA_ALG_AUTH_TRUNC]) ||
181                     attrs[XFRMA_ALG_AEAD]       ||
182                     attrs[XFRMA_ALG_CRYPT]      ||
183                     attrs[XFRMA_ALG_COMP]       ||
184                     attrs[XFRMA_TFCPAD])
185                         goto out;
186                 break;
187
188         case IPPROTO_ESP:
189                 if (attrs[XFRMA_ALG_COMP])
190                         goto out;
191                 if (!attrs[XFRMA_ALG_AUTH] &&
192                     !attrs[XFRMA_ALG_AUTH_TRUNC] &&
193                     !attrs[XFRMA_ALG_CRYPT] &&
194                     !attrs[XFRMA_ALG_AEAD])
195                         goto out;
196                 if ((attrs[XFRMA_ALG_AUTH] ||
197                      attrs[XFRMA_ALG_AUTH_TRUNC] ||
198                      attrs[XFRMA_ALG_CRYPT]) &&
199                     attrs[XFRMA_ALG_AEAD])
200                         goto out;
201                 if (attrs[XFRMA_TFCPAD] &&
202                     p->mode != XFRM_MODE_TUNNEL)
203                         goto out;
204                 break;
205
206         case IPPROTO_COMP:
207                 if (!attrs[XFRMA_ALG_COMP]      ||
208                     attrs[XFRMA_ALG_AEAD]       ||
209                     attrs[XFRMA_ALG_AUTH]       ||
210                     attrs[XFRMA_ALG_AUTH_TRUNC] ||
211                     attrs[XFRMA_ALG_CRYPT]      ||
212                     attrs[XFRMA_TFCPAD])
213                         goto out;
214                 break;
215
216 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
217         case IPPROTO_DSTOPTS:
218         case IPPROTO_ROUTING:
219                 if (attrs[XFRMA_ALG_COMP]       ||
220                     attrs[XFRMA_ALG_AUTH]       ||
221                     attrs[XFRMA_ALG_AUTH_TRUNC] ||
222                     attrs[XFRMA_ALG_AEAD]       ||
223                     attrs[XFRMA_ALG_CRYPT]      ||
224                     attrs[XFRMA_ENCAP]          ||
225                     attrs[XFRMA_SEC_CTX]        ||
226                     attrs[XFRMA_TFCPAD]         ||
227                     !attrs[XFRMA_COADDR])
228                         goto out;
229                 break;
230 #endif
231
232         default:
233                 goto out;
234         }
235
236         if ((err = verify_aead(attrs)))
237                 goto out;
238         if ((err = verify_auth_trunc(attrs)))
239                 goto out;
240         if ((err = verify_one_alg(attrs, XFRMA_ALG_AUTH)))
241                 goto out;
242         if ((err = verify_one_alg(attrs, XFRMA_ALG_CRYPT)))
243                 goto out;
244         if ((err = verify_one_alg(attrs, XFRMA_ALG_COMP)))
245                 goto out;
246         if ((err = verify_sec_ctx_len(attrs)))
247                 goto out;
248         if ((err = verify_replay(p, attrs)))
249                 goto out;
250
251         err = -EINVAL;
252         switch (p->mode) {
253         case XFRM_MODE_TRANSPORT:
254         case XFRM_MODE_TUNNEL:
255         case XFRM_MODE_ROUTEOPTIMIZATION:
256         case XFRM_MODE_BEET:
257                 break;
258
259         default:
260                 goto out;
261         }
262
263         err = 0;
264
265 out:
266         return err;
267 }
268
269 static int attach_one_algo(struct xfrm_algo **algpp, u8 *props,
270                            struct xfrm_algo_desc *(*get_byname)(const char *, int),
271                            struct nlattr *rta)
272 {
273         struct xfrm_algo *p, *ualg;
274         struct xfrm_algo_desc *algo;
275
276         if (!rta)
277                 return 0;
278
279         ualg = nla_data(rta);
280
281         algo = get_byname(ualg->alg_name, 1);
282         if (!algo)
283                 return -ENOSYS;
284         *props = algo->desc.sadb_alg_id;
285
286         p = kmemdup(ualg, xfrm_alg_len(ualg), GFP_KERNEL);
287         if (!p)
288                 return -ENOMEM;
289
290         strcpy(p->alg_name, algo->name);
291         *algpp = p;
292         return 0;
293 }
294
295 static int attach_auth(struct xfrm_algo_auth **algpp, u8 *props,
296                        struct nlattr *rta)
297 {
298         struct xfrm_algo *ualg;
299         struct xfrm_algo_auth *p;
300         struct xfrm_algo_desc *algo;
301
302         if (!rta)
303                 return 0;
304
305         ualg = nla_data(rta);
306
307         algo = xfrm_aalg_get_byname(ualg->alg_name, 1);
308         if (!algo)
309                 return -ENOSYS;
310         *props = algo->desc.sadb_alg_id;
311
312         p = kmalloc(sizeof(*p) + (ualg->alg_key_len + 7) / 8, GFP_KERNEL);
313         if (!p)
314                 return -ENOMEM;
315
316         strcpy(p->alg_name, algo->name);
317         p->alg_key_len = ualg->alg_key_len;
318         p->alg_trunc_len = algo->uinfo.auth.icv_truncbits;
319         memcpy(p->alg_key, ualg->alg_key, (ualg->alg_key_len + 7) / 8);
320
321         *algpp = p;
322         return 0;
323 }
324
325 static int attach_auth_trunc(struct xfrm_algo_auth **algpp, u8 *props,
326                              struct nlattr *rta)
327 {
328         struct xfrm_algo_auth *p, *ualg;
329         struct xfrm_algo_desc *algo;
330
331         if (!rta)
332                 return 0;
333
334         ualg = nla_data(rta);
335
336         algo = xfrm_aalg_get_byname(ualg->alg_name, 1);
337         if (!algo)
338                 return -ENOSYS;
339         if ((ualg->alg_trunc_len / 8) > MAX_AH_AUTH_LEN ||
340             ualg->alg_trunc_len > algo->uinfo.auth.icv_fullbits)
341                 return -EINVAL;
342         *props = algo->desc.sadb_alg_id;
343
344         p = kmemdup(ualg, xfrm_alg_auth_len(ualg), GFP_KERNEL);
345         if (!p)
346                 return -ENOMEM;
347
348         strcpy(p->alg_name, algo->name);
349         if (!p->alg_trunc_len)
350                 p->alg_trunc_len = algo->uinfo.auth.icv_truncbits;
351
352         *algpp = p;
353         return 0;
354 }
355
356 static int attach_aead(struct xfrm_algo_aead **algpp, u8 *props,
357                        struct nlattr *rta)
358 {
359         struct xfrm_algo_aead *p, *ualg;
360         struct xfrm_algo_desc *algo;
361
362         if (!rta)
363                 return 0;
364
365         ualg = nla_data(rta);
366
367         algo = xfrm_aead_get_byname(ualg->alg_name, ualg->alg_icv_len, 1);
368         if (!algo)
369                 return -ENOSYS;
370         *props = algo->desc.sadb_alg_id;
371
372         p = kmemdup(ualg, aead_len(ualg), GFP_KERNEL);
373         if (!p)
374                 return -ENOMEM;
375
376         strcpy(p->alg_name, algo->name);
377         *algpp = p;
378         return 0;
379 }
380
381 static inline int xfrm_replay_verify_len(struct xfrm_replay_state_esn *replay_esn,
382                                          struct nlattr *rp)
383 {
384         struct xfrm_replay_state_esn *up;
385         int ulen;
386
387         if (!replay_esn || !rp)
388                 return 0;
389
390         up = nla_data(rp);
391         ulen = xfrm_replay_state_esn_len(up);
392
393         if (nla_len(rp) < ulen || xfrm_replay_state_esn_len(replay_esn) != ulen)
394                 return -EINVAL;
395
396         if (up->replay_window > up->bmp_len * sizeof(__u32) * 8)
397                 return -EINVAL;
398
399         return 0;
400 }
401
402 static int xfrm_alloc_replay_state_esn(struct xfrm_replay_state_esn **replay_esn,
403                                        struct xfrm_replay_state_esn **preplay_esn,
404                                        struct nlattr *rta)
405 {
406         struct xfrm_replay_state_esn *p, *pp, *up;
407         int klen, ulen;
408
409         if (!rta)
410                 return 0;
411
412         up = nla_data(rta);
413         klen = xfrm_replay_state_esn_len(up);
414         ulen = nla_len(rta) >= klen ? klen : sizeof(*up);
415
416         p = kzalloc(klen, GFP_KERNEL);
417         if (!p)
418                 return -ENOMEM;
419
420         pp = kzalloc(klen, GFP_KERNEL);
421         if (!pp) {
422                 kfree(p);
423                 return -ENOMEM;
424         }
425
426         memcpy(p, up, ulen);
427         memcpy(pp, up, ulen);
428
429         *replay_esn = p;
430         *preplay_esn = pp;
431
432         return 0;
433 }
434
435 static inline int xfrm_user_sec_ctx_size(struct xfrm_sec_ctx *xfrm_ctx)
436 {
437         int len = 0;
438
439         if (xfrm_ctx) {
440                 len += sizeof(struct xfrm_user_sec_ctx);
441                 len += xfrm_ctx->ctx_len;
442         }
443         return len;
444 }
445
446 static void copy_from_user_state(struct xfrm_state *x, struct xfrm_usersa_info *p)
447 {
448         memcpy(&x->id, &p->id, sizeof(x->id));
449         memcpy(&x->sel, &p->sel, sizeof(x->sel));
450         memcpy(&x->lft, &p->lft, sizeof(x->lft));
451         x->props.mode = p->mode;
452         x->props.replay_window = p->replay_window;
453         x->props.reqid = p->reqid;
454         x->props.family = p->family;
455         memcpy(&x->props.saddr, &p->saddr, sizeof(x->props.saddr));
456         x->props.flags = p->flags;
457
458         if (!x->sel.family && !(p->flags & XFRM_STATE_AF_UNSPEC))
459                 x->sel.family = p->family;
460 }
461
462 /*
463  * someday when pfkey also has support, we could have the code
464  * somehow made shareable and move it to xfrm_state.c - JHS
465  *
466 */
467 static void xfrm_update_ae_params(struct xfrm_state *x, struct nlattr **attrs,
468                                   int update_esn)
469 {
470         struct nlattr *rp = attrs[XFRMA_REPLAY_VAL];
471         struct nlattr *re = update_esn ? attrs[XFRMA_REPLAY_ESN_VAL] : NULL;
472         struct nlattr *lt = attrs[XFRMA_LTIME_VAL];
473         struct nlattr *et = attrs[XFRMA_ETIMER_THRESH];
474         struct nlattr *rt = attrs[XFRMA_REPLAY_THRESH];
475
476         if (re) {
477                 struct xfrm_replay_state_esn *replay_esn;
478                 replay_esn = nla_data(re);
479                 memcpy(x->replay_esn, replay_esn,
480                        xfrm_replay_state_esn_len(replay_esn));
481                 memcpy(x->preplay_esn, replay_esn,
482                        xfrm_replay_state_esn_len(replay_esn));
483         }
484
485         if (rp) {
486                 struct xfrm_replay_state *replay;
487                 replay = nla_data(rp);
488                 memcpy(&x->replay, replay, sizeof(*replay));
489                 memcpy(&x->preplay, replay, sizeof(*replay));
490         }
491
492         if (lt) {
493                 struct xfrm_lifetime_cur *ltime;
494                 ltime = nla_data(lt);
495                 x->curlft.bytes = ltime->bytes;
496                 x->curlft.packets = ltime->packets;
497                 x->curlft.add_time = ltime->add_time;
498                 x->curlft.use_time = ltime->use_time;
499         }
500
501         if (et)
502                 x->replay_maxage = nla_get_u32(et);
503
504         if (rt)
505                 x->replay_maxdiff = nla_get_u32(rt);
506 }
507
508 static struct xfrm_state *xfrm_state_construct(struct net *net,
509                                                struct xfrm_usersa_info *p,
510                                                struct nlattr **attrs,
511                                                int *errp)
512 {
513         struct xfrm_state *x = xfrm_state_alloc(net);
514         int err = -ENOMEM;
515
516         if (!x)
517                 goto error_no_put;
518
519         copy_from_user_state(x, p);
520
521         if ((err = attach_aead(&x->aead, &x->props.ealgo,
522                                attrs[XFRMA_ALG_AEAD])))
523                 goto error;
524         if ((err = attach_auth_trunc(&x->aalg, &x->props.aalgo,
525                                      attrs[XFRMA_ALG_AUTH_TRUNC])))
526                 goto error;
527         if (!x->props.aalgo) {
528                 if ((err = attach_auth(&x->aalg, &x->props.aalgo,
529                                        attrs[XFRMA_ALG_AUTH])))
530                         goto error;
531         }
532         if ((err = attach_one_algo(&x->ealg, &x->props.ealgo,
533                                    xfrm_ealg_get_byname,
534                                    attrs[XFRMA_ALG_CRYPT])))
535                 goto error;
536         if ((err = attach_one_algo(&x->calg, &x->props.calgo,
537                                    xfrm_calg_get_byname,
538                                    attrs[XFRMA_ALG_COMP])))
539                 goto error;
540
541         if (attrs[XFRMA_ENCAP]) {
542                 x->encap = kmemdup(nla_data(attrs[XFRMA_ENCAP]),
543                                    sizeof(*x->encap), GFP_KERNEL);
544                 if (x->encap == NULL)
545                         goto error;
546         }
547
548         if (attrs[XFRMA_TFCPAD])
549                 x->tfcpad = nla_get_u32(attrs[XFRMA_TFCPAD]);
550
551         if (attrs[XFRMA_COADDR]) {
552                 x->coaddr = kmemdup(nla_data(attrs[XFRMA_COADDR]),
553                                     sizeof(*x->coaddr), GFP_KERNEL);
554                 if (x->coaddr == NULL)
555                         goto error;
556         }
557
558         xfrm_mark_get(attrs, &x->mark);
559
560         err = __xfrm_init_state(x, false);
561         if (err)
562                 goto error;
563
564         if (attrs[XFRMA_SEC_CTX]) {
565                 err = security_xfrm_state_alloc(x,
566                                                 nla_data(attrs[XFRMA_SEC_CTX]));
567                 if (err)
568                         goto error;
569         }
570
571         if ((err = xfrm_alloc_replay_state_esn(&x->replay_esn, &x->preplay_esn,
572                                                attrs[XFRMA_REPLAY_ESN_VAL])))
573                 goto error;
574
575         x->km.seq = p->seq;
576         x->replay_maxdiff = net->xfrm.sysctl_aevent_rseqth;
577         /* sysctl_xfrm_aevent_etime is in 100ms units */
578         x->replay_maxage = (net->xfrm.sysctl_aevent_etime*HZ)/XFRM_AE_ETH_M;
579
580         if ((err = xfrm_init_replay(x)))
581                 goto error;
582
583         /* override default values from above */
584         xfrm_update_ae_params(x, attrs, 0);
585
586         return x;
587
588 error:
589         x->km.state = XFRM_STATE_DEAD;
590         xfrm_state_put(x);
591 error_no_put:
592         *errp = err;
593         return NULL;
594 }
595
596 static int xfrm_add_sa(struct sk_buff *skb, struct nlmsghdr *nlh,
597                 struct nlattr **attrs)
598 {
599         struct net *net = sock_net(skb->sk);
600         struct xfrm_usersa_info *p = nlmsg_data(nlh);
601         struct xfrm_state *x;
602         int err;
603         struct km_event c;
604         uid_t loginuid = audit_get_loginuid(current);
605         u32 sessionid = audit_get_sessionid(current);
606         u32 sid;
607
608         err = verify_newsa_info(p, attrs);
609         if (err)
610                 return err;
611
612         x = xfrm_state_construct(net, p, attrs, &err);
613         if (!x)
614                 return err;
615
616         xfrm_state_hold(x);
617         if (nlh->nlmsg_type == XFRM_MSG_NEWSA)
618                 err = xfrm_state_add(x);
619         else
620                 err = xfrm_state_update(x);
621
622         security_task_getsecid(current, &sid);
623         xfrm_audit_state_add(x, err ? 0 : 1, loginuid, sessionid, sid);
624
625         if (err < 0) {
626                 x->km.state = XFRM_STATE_DEAD;
627                 __xfrm_state_put(x);
628                 goto out;
629         }
630
631         c.seq = nlh->nlmsg_seq;
632         c.pid = nlh->nlmsg_pid;
633         c.event = nlh->nlmsg_type;
634
635         km_state_notify(x, &c);
636 out:
637         xfrm_state_put(x);
638         return err;
639 }
640
641 static struct xfrm_state *xfrm_user_state_lookup(struct net *net,
642                                                  struct xfrm_usersa_id *p,
643                                                  struct nlattr **attrs,
644                                                  int *errp)
645 {
646         struct xfrm_state *x = NULL;
647         struct xfrm_mark m;
648         int err;
649         u32 mark = xfrm_mark_get(attrs, &m);
650
651         if (xfrm_id_proto_match(p->proto, IPSEC_PROTO_ANY)) {
652                 err = -ESRCH;
653                 x = xfrm_state_lookup(net, mark, &p->daddr, p->spi, p->proto, p->family);
654         } else {
655                 xfrm_address_t *saddr = NULL;
656
657                 verify_one_addr(attrs, XFRMA_SRCADDR, &saddr);
658                 if (!saddr) {
659                         err = -EINVAL;
660                         goto out;
661                 }
662
663                 err = -ESRCH;
664                 x = xfrm_state_lookup_byaddr(net, mark,
665                                              &p->daddr, saddr,
666                                              p->proto, p->family);
667         }
668
669  out:
670         if (!x && errp)
671                 *errp = err;
672         return x;
673 }
674
675 static int xfrm_del_sa(struct sk_buff *skb, struct nlmsghdr *nlh,
676                 struct nlattr **attrs)
677 {
678         struct net *net = sock_net(skb->sk);
679         struct xfrm_state *x;
680         int err = -ESRCH;
681         struct km_event c;
682         struct xfrm_usersa_id *p = nlmsg_data(nlh);
683         uid_t loginuid = audit_get_loginuid(current);
684         u32 sessionid = audit_get_sessionid(current);
685         u32 sid;
686
687         x = xfrm_user_state_lookup(net, p, attrs, &err);
688         if (x == NULL)
689                 return err;
690
691         if ((err = security_xfrm_state_delete(x)) != 0)
692                 goto out;
693
694         if (xfrm_state_kern(x)) {
695                 err = -EPERM;
696                 goto out;
697         }
698
699         err = xfrm_state_delete(x);
700
701         if (err < 0)
702                 goto out;
703
704         c.seq = nlh->nlmsg_seq;
705         c.pid = nlh->nlmsg_pid;
706         c.event = nlh->nlmsg_type;
707         km_state_notify(x, &c);
708
709 out:
710         security_task_getsecid(current, &sid);
711         xfrm_audit_state_delete(x, err ? 0 : 1, loginuid, sessionid, sid);
712         xfrm_state_put(x);
713         return err;
714 }
715
716 static void copy_to_user_state(struct xfrm_state *x, struct xfrm_usersa_info *p)
717 {
718         memset(p, 0, sizeof(*p));
719         memcpy(&p->id, &x->id, sizeof(p->id));
720         memcpy(&p->sel, &x->sel, sizeof(p->sel));
721         memcpy(&p->lft, &x->lft, sizeof(p->lft));
722         memcpy(&p->curlft, &x->curlft, sizeof(p->curlft));
723         memcpy(&p->stats, &x->stats, sizeof(p->stats));
724         memcpy(&p->saddr, &x->props.saddr, sizeof(p->saddr));
725         p->mode = x->props.mode;
726         p->replay_window = x->props.replay_window;
727         p->reqid = x->props.reqid;
728         p->family = x->props.family;
729         p->flags = x->props.flags;
730         p->seq = x->km.seq;
731 }
732
733 struct xfrm_dump_info {
734         struct sk_buff *in_skb;
735         struct sk_buff *out_skb;
736         u32 nlmsg_seq;
737         u16 nlmsg_flags;
738 };
739
740 static int copy_sec_ctx(struct xfrm_sec_ctx *s, struct sk_buff *skb)
741 {
742         struct xfrm_user_sec_ctx *uctx;
743         struct nlattr *attr;
744         int ctx_size = sizeof(*uctx) + s->ctx_len;
745
746         attr = nla_reserve(skb, XFRMA_SEC_CTX, ctx_size);
747         if (attr == NULL)
748                 return -EMSGSIZE;
749
750         uctx = nla_data(attr);
751         uctx->exttype = XFRMA_SEC_CTX;
752         uctx->len = ctx_size;
753         uctx->ctx_doi = s->ctx_doi;
754         uctx->ctx_alg = s->ctx_alg;
755         uctx->ctx_len = s->ctx_len;
756         memcpy(uctx + 1, s->ctx_str, s->ctx_len);
757
758         return 0;
759 }
760
761 static int copy_to_user_auth(struct xfrm_algo_auth *auth, struct sk_buff *skb)
762 {
763         struct xfrm_algo *algo;
764         struct nlattr *nla;
765
766         nla = nla_reserve(skb, XFRMA_ALG_AUTH,
767                           sizeof(*algo) + (auth->alg_key_len + 7) / 8);
768         if (!nla)
769                 return -EMSGSIZE;
770
771         algo = nla_data(nla);
772         strncpy(algo->alg_name, auth->alg_name, sizeof(algo->alg_name));
773         memcpy(algo->alg_key, auth->alg_key, (auth->alg_key_len + 7) / 8);
774         algo->alg_key_len = auth->alg_key_len;
775
776         return 0;
777 }
778
779 /* Don't change this without updating xfrm_sa_len! */
780 static int copy_to_user_state_extra(struct xfrm_state *x,
781                                     struct xfrm_usersa_info *p,
782                                     struct sk_buff *skb)
783 {
784         copy_to_user_state(x, p);
785
786         if (x->coaddr)
787                 NLA_PUT(skb, XFRMA_COADDR, sizeof(*x->coaddr), x->coaddr);
788
789         if (x->lastused)
790                 NLA_PUT_U64(skb, XFRMA_LASTUSED, x->lastused);
791
792         if (x->aead)
793                 NLA_PUT(skb, XFRMA_ALG_AEAD, aead_len(x->aead), x->aead);
794         if (x->aalg) {
795                 if (copy_to_user_auth(x->aalg, skb))
796                         goto nla_put_failure;
797
798                 NLA_PUT(skb, XFRMA_ALG_AUTH_TRUNC,
799                         xfrm_alg_auth_len(x->aalg), x->aalg);
800         }
801         if (x->ealg)
802                 NLA_PUT(skb, XFRMA_ALG_CRYPT, xfrm_alg_len(x->ealg), x->ealg);
803         if (x->calg)
804                 NLA_PUT(skb, XFRMA_ALG_COMP, sizeof(*(x->calg)), x->calg);
805
806         if (x->encap)
807                 NLA_PUT(skb, XFRMA_ENCAP, sizeof(*x->encap), x->encap);
808
809         if (x->tfcpad)
810                 NLA_PUT_U32(skb, XFRMA_TFCPAD, x->tfcpad);
811
812         if (xfrm_mark_put(skb, &x->mark))
813                 goto nla_put_failure;
814
815         if (x->replay_esn)
816                 NLA_PUT(skb, XFRMA_REPLAY_ESN_VAL,
817                         xfrm_replay_state_esn_len(x->replay_esn), x->replay_esn);
818
819         if (x->security && copy_sec_ctx(x->security, skb) < 0)
820                 goto nla_put_failure;
821
822         return 0;
823
824 nla_put_failure:
825         return -EMSGSIZE;
826 }
827
828 static int dump_one_state(struct xfrm_state *x, int count, void *ptr)
829 {
830         struct xfrm_dump_info *sp = ptr;
831         struct sk_buff *in_skb = sp->in_skb;
832         struct sk_buff *skb = sp->out_skb;
833         struct xfrm_usersa_info *p;
834         struct nlmsghdr *nlh;
835         int err;
836
837         nlh = nlmsg_put(skb, NETLINK_CB(in_skb).pid, sp->nlmsg_seq,
838                         XFRM_MSG_NEWSA, sizeof(*p), sp->nlmsg_flags);
839         if (nlh == NULL)
840                 return -EMSGSIZE;
841
842         p = nlmsg_data(nlh);
843
844         err = copy_to_user_state_extra(x, p, skb);
845         if (err)
846                 goto nla_put_failure;
847
848         nlmsg_end(skb, nlh);
849         return 0;
850
851 nla_put_failure:
852         nlmsg_cancel(skb, nlh);
853         return err;
854 }
855
856 static int xfrm_dump_sa_done(struct netlink_callback *cb)
857 {
858         struct xfrm_state_walk *walk = (struct xfrm_state_walk *) &cb->args[1];
859         xfrm_state_walk_done(walk);
860         return 0;
861 }
862
863 static int xfrm_dump_sa(struct sk_buff *skb, struct netlink_callback *cb)
864 {
865         struct net *net = sock_net(skb->sk);
866         struct xfrm_state_walk *walk = (struct xfrm_state_walk *) &cb->args[1];
867         struct xfrm_dump_info info;
868
869         BUILD_BUG_ON(sizeof(struct xfrm_state_walk) >
870                      sizeof(cb->args) - sizeof(cb->args[0]));
871
872         info.in_skb = cb->skb;
873         info.out_skb = skb;
874         info.nlmsg_seq = cb->nlh->nlmsg_seq;
875         info.nlmsg_flags = NLM_F_MULTI;
876
877         if (!cb->args[0]) {
878                 cb->args[0] = 1;
879                 xfrm_state_walk_init(walk, 0);
880         }
881
882         (void) xfrm_state_walk(net, walk, dump_one_state, &info);
883
884         return skb->len;
885 }
886
887 static struct sk_buff *xfrm_state_netlink(struct sk_buff *in_skb,
888                                           struct xfrm_state *x, u32 seq)
889 {
890         struct xfrm_dump_info info;
891         struct sk_buff *skb;
892         int err;
893
894         skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
895         if (!skb)
896                 return ERR_PTR(-ENOMEM);
897
898         info.in_skb = in_skb;
899         info.out_skb = skb;
900         info.nlmsg_seq = seq;
901         info.nlmsg_flags = 0;
902
903         err = dump_one_state(x, 0, &info);
904         if (err) {
905                 kfree_skb(skb);
906                 return ERR_PTR(err);
907         }
908
909         return skb;
910 }
911
912 static inline size_t xfrm_spdinfo_msgsize(void)
913 {
914         return NLMSG_ALIGN(4)
915                + nla_total_size(sizeof(struct xfrmu_spdinfo))
916                + nla_total_size(sizeof(struct xfrmu_spdhinfo));
917 }
918
919 static int build_spdinfo(struct sk_buff *skb, struct net *net,
920                          u32 pid, u32 seq, u32 flags)
921 {
922         struct xfrmk_spdinfo si;
923         struct xfrmu_spdinfo spc;
924         struct xfrmu_spdhinfo sph;
925         struct nlmsghdr *nlh;
926         u32 *f;
927
928         nlh = nlmsg_put(skb, pid, seq, XFRM_MSG_NEWSPDINFO, sizeof(u32), 0);
929         if (nlh == NULL) /* shouldn't really happen ... */
930                 return -EMSGSIZE;
931
932         f = nlmsg_data(nlh);
933         *f = flags;
934         xfrm_spd_getinfo(net, &si);
935         spc.incnt = si.incnt;
936         spc.outcnt = si.outcnt;
937         spc.fwdcnt = si.fwdcnt;
938         spc.inscnt = si.inscnt;
939         spc.outscnt = si.outscnt;
940         spc.fwdscnt = si.fwdscnt;
941         sph.spdhcnt = si.spdhcnt;
942         sph.spdhmcnt = si.spdhmcnt;
943
944         NLA_PUT(skb, XFRMA_SPD_INFO, sizeof(spc), &spc);
945         NLA_PUT(skb, XFRMA_SPD_HINFO, sizeof(sph), &sph);
946
947         return nlmsg_end(skb, nlh);
948
949 nla_put_failure:
950         nlmsg_cancel(skb, nlh);
951         return -EMSGSIZE;
952 }
953
954 static int xfrm_get_spdinfo(struct sk_buff *skb, struct nlmsghdr *nlh,
955                 struct nlattr **attrs)
956 {
957         struct net *net = sock_net(skb->sk);
958         struct sk_buff *r_skb;
959         u32 *flags = nlmsg_data(nlh);
960         u32 spid = NETLINK_CB(skb).pid;
961         u32 seq = nlh->nlmsg_seq;
962
963         r_skb = nlmsg_new(xfrm_spdinfo_msgsize(), GFP_ATOMIC);
964         if (r_skb == NULL)
965                 return -ENOMEM;
966
967         if (build_spdinfo(r_skb, net, spid, seq, *flags) < 0)
968                 BUG();
969
970         return nlmsg_unicast(net->xfrm.nlsk, r_skb, spid);
971 }
972
973 static inline size_t xfrm_sadinfo_msgsize(void)
974 {
975         return NLMSG_ALIGN(4)
976                + nla_total_size(sizeof(struct xfrmu_sadhinfo))
977                + nla_total_size(4); /* XFRMA_SAD_CNT */
978 }
979
980 static int build_sadinfo(struct sk_buff *skb, struct net *net,
981                          u32 pid, u32 seq, u32 flags)
982 {
983         struct xfrmk_sadinfo si;
984         struct xfrmu_sadhinfo sh;
985         struct nlmsghdr *nlh;
986         u32 *f;
987
988         nlh = nlmsg_put(skb, pid, seq, XFRM_MSG_NEWSADINFO, sizeof(u32), 0);
989         if (nlh == NULL) /* shouldn't really happen ... */
990                 return -EMSGSIZE;
991
992         f = nlmsg_data(nlh);
993         *f = flags;
994         xfrm_sad_getinfo(net, &si);
995
996         sh.sadhmcnt = si.sadhmcnt;
997         sh.sadhcnt = si.sadhcnt;
998
999         NLA_PUT_U32(skb, XFRMA_SAD_CNT, si.sadcnt);
1000         NLA_PUT(skb, XFRMA_SAD_HINFO, sizeof(sh), &sh);
1001
1002         return nlmsg_end(skb, nlh);
1003
1004 nla_put_failure:
1005         nlmsg_cancel(skb, nlh);
1006         return -EMSGSIZE;
1007 }
1008
1009 static int xfrm_get_sadinfo(struct sk_buff *skb, struct nlmsghdr *nlh,
1010                 struct nlattr **attrs)
1011 {
1012         struct net *net = sock_net(skb->sk);
1013         struct sk_buff *r_skb;
1014         u32 *flags = nlmsg_data(nlh);
1015         u32 spid = NETLINK_CB(skb).pid;
1016         u32 seq = nlh->nlmsg_seq;
1017
1018         r_skb = nlmsg_new(xfrm_sadinfo_msgsize(), GFP_ATOMIC);
1019         if (r_skb == NULL)
1020                 return -ENOMEM;
1021
1022         if (build_sadinfo(r_skb, net, spid, seq, *flags) < 0)
1023                 BUG();
1024
1025         return nlmsg_unicast(net->xfrm.nlsk, r_skb, spid);
1026 }
1027
1028 static int xfrm_get_sa(struct sk_buff *skb, struct nlmsghdr *nlh,
1029                 struct nlattr **attrs)
1030 {
1031         struct net *net = sock_net(skb->sk);
1032         struct xfrm_usersa_id *p = nlmsg_data(nlh);
1033         struct xfrm_state *x;
1034         struct sk_buff *resp_skb;
1035         int err = -ESRCH;
1036
1037         x = xfrm_user_state_lookup(net, p, attrs, &err);
1038         if (x == NULL)
1039                 goto out_noput;
1040
1041         resp_skb = xfrm_state_netlink(skb, x, nlh->nlmsg_seq);
1042         if (IS_ERR(resp_skb)) {
1043                 err = PTR_ERR(resp_skb);
1044         } else {
1045                 err = nlmsg_unicast(net->xfrm.nlsk, resp_skb, NETLINK_CB(skb).pid);
1046         }
1047         xfrm_state_put(x);
1048 out_noput:
1049         return err;
1050 }
1051
1052 static int verify_userspi_info(struct xfrm_userspi_info *p)
1053 {
1054         switch (p->info.id.proto) {
1055         case IPPROTO_AH:
1056         case IPPROTO_ESP:
1057                 break;
1058
1059         case IPPROTO_COMP:
1060                 /* IPCOMP spi is 16-bits. */
1061                 if (p->max >= 0x10000)
1062                         return -EINVAL;
1063                 break;
1064
1065         default:
1066                 return -EINVAL;
1067         }
1068
1069         if (p->min > p->max)
1070                 return -EINVAL;
1071
1072         return 0;
1073 }
1074
1075 static int xfrm_alloc_userspi(struct sk_buff *skb, struct nlmsghdr *nlh,
1076                 struct nlattr **attrs)
1077 {
1078         struct net *net = sock_net(skb->sk);
1079         struct xfrm_state *x;
1080         struct xfrm_userspi_info *p;
1081         struct sk_buff *resp_skb;
1082         xfrm_address_t *daddr;
1083         int family;
1084         int err;
1085         u32 mark;
1086         struct xfrm_mark m;
1087
1088         p = nlmsg_data(nlh);
1089         err = verify_userspi_info(p);
1090         if (err)
1091                 goto out_noput;
1092
1093         family = p->info.family;
1094         daddr = &p->info.id.daddr;
1095
1096         x = NULL;
1097
1098         mark = xfrm_mark_get(attrs, &m);
1099         if (p->info.seq) {
1100                 x = xfrm_find_acq_byseq(net, mark, p->info.seq);
1101                 if (x && xfrm_addr_cmp(&x->id.daddr, daddr, family)) {
1102                         xfrm_state_put(x);
1103                         x = NULL;
1104                 }
1105         }
1106
1107         if (!x)
1108                 x = xfrm_find_acq(net, &m, p->info.mode, p->info.reqid,
1109                                   p->info.id.proto, daddr,
1110                                   &p->info.saddr, 1,
1111                                   family);
1112         err = -ENOENT;
1113         if (x == NULL)
1114                 goto out_noput;
1115
1116         err = xfrm_alloc_spi(x, p->min, p->max);
1117         if (err)
1118                 goto out;
1119
1120         resp_skb = xfrm_state_netlink(skb, x, nlh->nlmsg_seq);
1121         if (IS_ERR(resp_skb)) {
1122                 err = PTR_ERR(resp_skb);
1123                 goto out;
1124         }
1125
1126         err = nlmsg_unicast(net->xfrm.nlsk, resp_skb, NETLINK_CB(skb).pid);
1127
1128 out:
1129         xfrm_state_put(x);
1130 out_noput:
1131         return err;
1132 }
1133
1134 static int verify_policy_dir(u8 dir)
1135 {
1136         switch (dir) {
1137         case XFRM_POLICY_IN:
1138         case XFRM_POLICY_OUT:
1139         case XFRM_POLICY_FWD:
1140                 break;
1141
1142         default:
1143                 return -EINVAL;
1144         }
1145
1146         return 0;
1147 }
1148
1149 static int verify_policy_type(u8 type)
1150 {
1151         switch (type) {
1152         case XFRM_POLICY_TYPE_MAIN:
1153 #ifdef CONFIG_XFRM_SUB_POLICY
1154         case XFRM_POLICY_TYPE_SUB:
1155 #endif
1156                 break;
1157
1158         default:
1159                 return -EINVAL;
1160         }
1161
1162         return 0;
1163 }
1164
1165 static int verify_newpolicy_info(struct xfrm_userpolicy_info *p)
1166 {
1167         switch (p->share) {
1168         case XFRM_SHARE_ANY:
1169         case XFRM_SHARE_SESSION:
1170         case XFRM_SHARE_USER:
1171         case XFRM_SHARE_UNIQUE:
1172                 break;
1173
1174         default:
1175                 return -EINVAL;
1176         }
1177
1178         switch (p->action) {
1179         case XFRM_POLICY_ALLOW:
1180         case XFRM_POLICY_BLOCK:
1181                 break;
1182
1183         default:
1184                 return -EINVAL;
1185         }
1186
1187         switch (p->sel.family) {
1188         case AF_INET:
1189                 break;
1190
1191         case AF_INET6:
1192 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1193                 break;
1194 #else
1195                 return  -EAFNOSUPPORT;
1196 #endif
1197
1198         default:
1199                 return -EINVAL;
1200         }
1201
1202         return verify_policy_dir(p->dir);
1203 }
1204
1205 static int copy_from_user_sec_ctx(struct xfrm_policy *pol, struct nlattr **attrs)
1206 {
1207         struct nlattr *rt = attrs[XFRMA_SEC_CTX];
1208         struct xfrm_user_sec_ctx *uctx;
1209
1210         if (!rt)
1211                 return 0;
1212
1213         uctx = nla_data(rt);
1214         return security_xfrm_policy_alloc(&pol->security, uctx);
1215 }
1216
1217 static void copy_templates(struct xfrm_policy *xp, struct xfrm_user_tmpl *ut,
1218                            int nr)
1219 {
1220         int i;
1221
1222         xp->xfrm_nr = nr;
1223         for (i = 0; i < nr; i++, ut++) {
1224                 struct xfrm_tmpl *t = &xp->xfrm_vec[i];
1225
1226                 memcpy(&t->id, &ut->id, sizeof(struct xfrm_id));
1227                 memcpy(&t->saddr, &ut->saddr,
1228                        sizeof(xfrm_address_t));
1229                 t->reqid = ut->reqid;
1230                 t->mode = ut->mode;
1231                 t->share = ut->share;
1232                 t->optional = ut->optional;
1233                 t->aalgos = ut->aalgos;
1234                 t->ealgos = ut->ealgos;
1235                 t->calgos = ut->calgos;
1236                 /* If all masks are ~0, then we allow all algorithms. */
1237                 t->allalgs = !~(t->aalgos & t->ealgos & t->calgos);
1238                 t->encap_family = ut->family;
1239         }
1240 }
1241
1242 static int validate_tmpl(int nr, struct xfrm_user_tmpl *ut, u16 family)
1243 {
1244         int i;
1245
1246         if (nr > XFRM_MAX_DEPTH)
1247                 return -EINVAL;
1248
1249         for (i = 0; i < nr; i++) {
1250                 /* We never validated the ut->family value, so many
1251                  * applications simply leave it at zero.  The check was
1252                  * never made and ut->family was ignored because all
1253                  * templates could be assumed to have the same family as
1254                  * the policy itself.  Now that we will have ipv4-in-ipv6
1255                  * and ipv6-in-ipv4 tunnels, this is no longer true.
1256                  */
1257                 if (!ut[i].family)
1258                         ut[i].family = family;
1259
1260                 switch (ut[i].family) {
1261                 case AF_INET:
1262                         break;
1263 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1264                 case AF_INET6:
1265                         break;
1266 #endif
1267                 default:
1268                         return -EINVAL;
1269                 }
1270         }
1271
1272         return 0;
1273 }
1274
1275 static int copy_from_user_tmpl(struct xfrm_policy *pol, struct nlattr **attrs)
1276 {
1277         struct nlattr *rt = attrs[XFRMA_TMPL];
1278
1279         if (!rt) {
1280                 pol->xfrm_nr = 0;
1281         } else {
1282                 struct xfrm_user_tmpl *utmpl = nla_data(rt);
1283                 int nr = nla_len(rt) / sizeof(*utmpl);
1284                 int err;
1285
1286                 err = validate_tmpl(nr, utmpl, pol->family);
1287                 if (err)
1288                         return err;
1289
1290                 copy_templates(pol, utmpl, nr);
1291         }
1292         return 0;
1293 }
1294
1295 static int copy_from_user_policy_type(u8 *tp, struct nlattr **attrs)
1296 {
1297         struct nlattr *rt = attrs[XFRMA_POLICY_TYPE];
1298         struct xfrm_userpolicy_type *upt;
1299         u8 type = XFRM_POLICY_TYPE_MAIN;
1300         int err;
1301
1302         if (rt) {
1303                 upt = nla_data(rt);
1304                 type = upt->type;
1305         }
1306
1307         err = verify_policy_type(type);
1308         if (err)
1309                 return err;
1310
1311         *tp = type;
1312         return 0;
1313 }
1314
1315 static void copy_from_user_policy(struct xfrm_policy *xp, struct xfrm_userpolicy_info *p)
1316 {
1317         xp->priority = p->priority;
1318         xp->index = p->index;
1319         memcpy(&xp->selector, &p->sel, sizeof(xp->selector));
1320         memcpy(&xp->lft, &p->lft, sizeof(xp->lft));
1321         xp->action = p->action;
1322         xp->flags = p->flags;
1323         xp->family = p->sel.family;
1324         /* XXX xp->share = p->share; */
1325 }
1326
1327 static void copy_to_user_policy(struct xfrm_policy *xp, struct xfrm_userpolicy_info *p, int dir)
1328 {
1329         memset(p, 0, sizeof(*p));
1330         memcpy(&p->sel, &xp->selector, sizeof(p->sel));
1331         memcpy(&p->lft, &xp->lft, sizeof(p->lft));
1332         memcpy(&p->curlft, &xp->curlft, sizeof(p->curlft));
1333         p->priority = xp->priority;
1334         p->index = xp->index;
1335         p->sel.family = xp->family;
1336         p->dir = dir;
1337         p->action = xp->action;
1338         p->flags = xp->flags;
1339         p->share = XFRM_SHARE_ANY; /* XXX xp->share */
1340 }
1341
1342 static struct xfrm_policy *xfrm_policy_construct(struct net *net, struct xfrm_userpolicy_info *p, struct nlattr **attrs, int *errp)
1343 {
1344         struct xfrm_policy *xp = xfrm_policy_alloc(net, GFP_KERNEL);
1345         int err;
1346
1347         if (!xp) {
1348                 *errp = -ENOMEM;
1349                 return NULL;
1350         }
1351
1352         copy_from_user_policy(xp, p);
1353
1354         err = copy_from_user_policy_type(&xp->type, attrs);
1355         if (err)
1356                 goto error;
1357
1358         if (!(err = copy_from_user_tmpl(xp, attrs)))
1359                 err = copy_from_user_sec_ctx(xp, attrs);
1360         if (err)
1361                 goto error;
1362
1363         xfrm_mark_get(attrs, &xp->mark);
1364
1365         return xp;
1366  error:
1367         *errp = err;
1368         xp->walk.dead = 1;
1369         xfrm_policy_destroy(xp);
1370         return NULL;
1371 }
1372
1373 static int xfrm_add_policy(struct sk_buff *skb, struct nlmsghdr *nlh,
1374                 struct nlattr **attrs)
1375 {
1376         struct net *net = sock_net(skb->sk);
1377         struct xfrm_userpolicy_info *p = nlmsg_data(nlh);
1378         struct xfrm_policy *xp;
1379         struct km_event c;
1380         int err;
1381         int excl;
1382         uid_t loginuid = audit_get_loginuid(current);
1383         u32 sessionid = audit_get_sessionid(current);
1384         u32 sid;
1385
1386         err = verify_newpolicy_info(p);
1387         if (err)
1388                 return err;
1389         err = verify_sec_ctx_len(attrs);
1390         if (err)
1391                 return err;
1392
1393         xp = xfrm_policy_construct(net, p, attrs, &err);
1394         if (!xp)
1395                 return err;
1396
1397         /* shouldn't excl be based on nlh flags??
1398          * Aha! this is anti-netlink really i.e  more pfkey derived
1399          * in netlink excl is a flag and you wouldnt need
1400          * a type XFRM_MSG_UPDPOLICY - JHS */
1401         excl = nlh->nlmsg_type == XFRM_MSG_NEWPOLICY;
1402         err = xfrm_policy_insert(p->dir, xp, excl);
1403         security_task_getsecid(current, &sid);
1404         xfrm_audit_policy_add(xp, err ? 0 : 1, loginuid, sessionid, sid);
1405
1406         if (err) {
1407                 security_xfrm_policy_free(xp->security);
1408                 kfree(xp);
1409                 return err;
1410         }
1411
1412         c.event = nlh->nlmsg_type;
1413         c.seq = nlh->nlmsg_seq;
1414         c.pid = nlh->nlmsg_pid;
1415         km_policy_notify(xp, p->dir, &c);
1416
1417         xfrm_pol_put(xp);
1418
1419         return 0;
1420 }
1421
1422 static int copy_to_user_tmpl(struct xfrm_policy *xp, struct sk_buff *skb)
1423 {
1424         struct xfrm_user_tmpl vec[XFRM_MAX_DEPTH];
1425         int i;
1426
1427         if (xp->xfrm_nr == 0)
1428                 return 0;
1429
1430         for (i = 0; i < xp->xfrm_nr; i++) {
1431                 struct xfrm_user_tmpl *up = &vec[i];
1432                 struct xfrm_tmpl *kp = &xp->xfrm_vec[i];
1433
1434                 memset(up, 0, sizeof(*up));
1435                 memcpy(&up->id, &kp->id, sizeof(up->id));
1436                 up->family = kp->encap_family;
1437                 memcpy(&up->saddr, &kp->saddr, sizeof(up->saddr));
1438                 up->reqid = kp->reqid;
1439                 up->mode = kp->mode;
1440                 up->share = kp->share;
1441                 up->optional = kp->optional;
1442                 up->aalgos = kp->aalgos;
1443                 up->ealgos = kp->ealgos;
1444                 up->calgos = kp->calgos;
1445         }
1446
1447         return nla_put(skb, XFRMA_TMPL,
1448                        sizeof(struct xfrm_user_tmpl) * xp->xfrm_nr, vec);
1449 }
1450
1451 static inline int copy_to_user_state_sec_ctx(struct xfrm_state *x, struct sk_buff *skb)
1452 {
1453         if (x->security) {
1454                 return copy_sec_ctx(x->security, skb);
1455         }
1456         return 0;
1457 }
1458
1459 static inline int copy_to_user_sec_ctx(struct xfrm_policy *xp, struct sk_buff *skb)
1460 {
1461         if (xp->security) {
1462                 return copy_sec_ctx(xp->security, skb);
1463         }
1464         return 0;
1465 }
1466 static inline size_t userpolicy_type_attrsize(void)
1467 {
1468 #ifdef CONFIG_XFRM_SUB_POLICY
1469         return nla_total_size(sizeof(struct xfrm_userpolicy_type));
1470 #else
1471         return 0;
1472 #endif
1473 }
1474
1475 #ifdef CONFIG_XFRM_SUB_POLICY
1476 static int copy_to_user_policy_type(u8 type, struct sk_buff *skb)
1477 {
1478         struct xfrm_userpolicy_type upt = {
1479                 .type = type,
1480         };
1481
1482         return nla_put(skb, XFRMA_POLICY_TYPE, sizeof(upt), &upt);
1483 }
1484
1485 #else
1486 static inline int copy_to_user_policy_type(u8 type, struct sk_buff *skb)
1487 {
1488         return 0;
1489 }
1490 #endif
1491
1492 static int dump_one_policy(struct xfrm_policy *xp, int dir, int count, void *ptr)
1493 {
1494         struct xfrm_dump_info *sp = ptr;
1495         struct xfrm_userpolicy_info *p;
1496         struct sk_buff *in_skb = sp->in_skb;
1497         struct sk_buff *skb = sp->out_skb;
1498         struct nlmsghdr *nlh;
1499
1500         nlh = nlmsg_put(skb, NETLINK_CB(in_skb).pid, sp->nlmsg_seq,
1501                         XFRM_MSG_NEWPOLICY, sizeof(*p), sp->nlmsg_flags);
1502         if (nlh == NULL)
1503                 return -EMSGSIZE;
1504
1505         p = nlmsg_data(nlh);
1506         copy_to_user_policy(xp, p, dir);
1507         if (copy_to_user_tmpl(xp, skb) < 0)
1508                 goto nlmsg_failure;
1509         if (copy_to_user_sec_ctx(xp, skb))
1510                 goto nlmsg_failure;
1511         if (copy_to_user_policy_type(xp->type, skb) < 0)
1512                 goto nlmsg_failure;
1513         if (xfrm_mark_put(skb, &xp->mark))
1514                 goto nla_put_failure;
1515
1516         nlmsg_end(skb, nlh);
1517         return 0;
1518
1519 nla_put_failure:
1520 nlmsg_failure:
1521         nlmsg_cancel(skb, nlh);
1522         return -EMSGSIZE;
1523 }
1524
1525 static int xfrm_dump_policy_done(struct netlink_callback *cb)
1526 {
1527         struct xfrm_policy_walk *walk = (struct xfrm_policy_walk *) &cb->args[1];
1528
1529         xfrm_policy_walk_done(walk);
1530         return 0;
1531 }
1532
1533 static int xfrm_dump_policy(struct sk_buff *skb, struct netlink_callback *cb)
1534 {
1535         struct net *net = sock_net(skb->sk);
1536         struct xfrm_policy_walk *walk = (struct xfrm_policy_walk *) &cb->args[1];
1537         struct xfrm_dump_info info;
1538
1539         BUILD_BUG_ON(sizeof(struct xfrm_policy_walk) >
1540                      sizeof(cb->args) - sizeof(cb->args[0]));
1541
1542         info.in_skb = cb->skb;
1543         info.out_skb = skb;
1544         info.nlmsg_seq = cb->nlh->nlmsg_seq;
1545         info.nlmsg_flags = NLM_F_MULTI;
1546
1547         if (!cb->args[0]) {
1548                 cb->args[0] = 1;
1549                 xfrm_policy_walk_init(walk, XFRM_POLICY_TYPE_ANY);
1550         }
1551
1552         (void) xfrm_policy_walk(net, walk, dump_one_policy, &info);
1553
1554         return skb->len;
1555 }
1556
1557 static struct sk_buff *xfrm_policy_netlink(struct sk_buff *in_skb,
1558                                           struct xfrm_policy *xp,
1559                                           int dir, u32 seq)
1560 {
1561         struct xfrm_dump_info info;
1562         struct sk_buff *skb;
1563         int err;
1564
1565         skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1566         if (!skb)
1567                 return ERR_PTR(-ENOMEM);
1568
1569         info.in_skb = in_skb;
1570         info.out_skb = skb;
1571         info.nlmsg_seq = seq;
1572         info.nlmsg_flags = 0;
1573
1574         err = dump_one_policy(xp, dir, 0, &info);
1575         if (err) {
1576                 kfree_skb(skb);
1577                 return ERR_PTR(err);
1578         }
1579
1580         return skb;
1581 }
1582
1583 static int xfrm_get_policy(struct sk_buff *skb, struct nlmsghdr *nlh,
1584                 struct nlattr **attrs)
1585 {
1586         struct net *net = sock_net(skb->sk);
1587         struct xfrm_policy *xp;
1588         struct xfrm_userpolicy_id *p;
1589         u8 type = XFRM_POLICY_TYPE_MAIN;
1590         int err;
1591         struct km_event c;
1592         int delete;
1593         struct xfrm_mark m;
1594         u32 mark = xfrm_mark_get(attrs, &m);
1595
1596         p = nlmsg_data(nlh);
1597         delete = nlh->nlmsg_type == XFRM_MSG_DELPOLICY;
1598
1599         err = copy_from_user_policy_type(&type, attrs);
1600         if (err)
1601                 return err;
1602
1603         err = verify_policy_dir(p->dir);
1604         if (err)
1605                 return err;
1606
1607         if (p->index)
1608                 xp = xfrm_policy_byid(net, mark, type, p->dir, p->index, delete, &err);
1609         else {
1610                 struct nlattr *rt = attrs[XFRMA_SEC_CTX];
1611                 struct xfrm_sec_ctx *ctx;
1612
1613                 err = verify_sec_ctx_len(attrs);
1614                 if (err)
1615                         return err;
1616
1617                 ctx = NULL;
1618                 if (rt) {
1619                         struct xfrm_user_sec_ctx *uctx = nla_data(rt);
1620
1621                         err = security_xfrm_policy_alloc(&ctx, uctx);
1622                         if (err)
1623                                 return err;
1624                 }
1625                 xp = xfrm_policy_bysel_ctx(net, mark, type, p->dir, &p->sel,
1626                                            ctx, delete, &err);
1627                 security_xfrm_policy_free(ctx);
1628         }
1629         if (xp == NULL)
1630                 return -ENOENT;
1631
1632         if (!delete) {
1633                 struct sk_buff *resp_skb;
1634
1635                 resp_skb = xfrm_policy_netlink(skb, xp, p->dir, nlh->nlmsg_seq);
1636                 if (IS_ERR(resp_skb)) {
1637                         err = PTR_ERR(resp_skb);
1638                 } else {
1639                         err = nlmsg_unicast(net->xfrm.nlsk, resp_skb,
1640                                             NETLINK_CB(skb).pid);
1641                 }
1642         } else {
1643                 uid_t loginuid = audit_get_loginuid(current);
1644                 u32 sessionid = audit_get_sessionid(current);
1645                 u32 sid;
1646
1647                 security_task_getsecid(current, &sid);
1648                 xfrm_audit_policy_delete(xp, err ? 0 : 1, loginuid, sessionid,
1649                                          sid);
1650
1651                 if (err != 0)
1652                         goto out;
1653
1654                 c.data.byid = p->index;
1655                 c.event = nlh->nlmsg_type;
1656                 c.seq = nlh->nlmsg_seq;
1657                 c.pid = nlh->nlmsg_pid;
1658                 km_policy_notify(xp, p->dir, &c);
1659         }
1660
1661 out:
1662         xfrm_pol_put(xp);
1663         return err;
1664 }
1665
1666 static int xfrm_flush_sa(struct sk_buff *skb, struct nlmsghdr *nlh,
1667                 struct nlattr **attrs)
1668 {
1669         struct net *net = sock_net(skb->sk);
1670         struct km_event c;
1671         struct xfrm_usersa_flush *p = nlmsg_data(nlh);
1672         struct xfrm_audit audit_info;
1673         int err;
1674
1675         audit_info.loginuid = audit_get_loginuid(current);
1676         audit_info.sessionid = audit_get_sessionid(current);
1677         security_task_getsecid(current, &audit_info.secid);
1678         err = xfrm_state_flush(net, p->proto, &audit_info);
1679         if (err) {
1680                 if (err == -ESRCH) /* empty table */
1681                         return 0;
1682                 return err;
1683         }
1684         c.data.proto = p->proto;
1685         c.event = nlh->nlmsg_type;
1686         c.seq = nlh->nlmsg_seq;
1687         c.pid = nlh->nlmsg_pid;
1688         c.net = net;
1689         km_state_notify(NULL, &c);
1690
1691         return 0;
1692 }
1693
1694 static inline size_t xfrm_aevent_msgsize(struct xfrm_state *x)
1695 {
1696         size_t replay_size = x->replay_esn ?
1697                               xfrm_replay_state_esn_len(x->replay_esn) :
1698                               sizeof(struct xfrm_replay_state);
1699
1700         return NLMSG_ALIGN(sizeof(struct xfrm_aevent_id))
1701                + nla_total_size(replay_size)
1702                + nla_total_size(sizeof(struct xfrm_lifetime_cur))
1703                + nla_total_size(sizeof(struct xfrm_mark))
1704                + nla_total_size(4) /* XFRM_AE_RTHR */
1705                + nla_total_size(4); /* XFRM_AE_ETHR */
1706 }
1707
1708 static int build_aevent(struct sk_buff *skb, struct xfrm_state *x, const struct km_event *c)
1709 {
1710         struct xfrm_aevent_id *id;
1711         struct nlmsghdr *nlh;
1712
1713         nlh = nlmsg_put(skb, c->pid, c->seq, XFRM_MSG_NEWAE, sizeof(*id), 0);
1714         if (nlh == NULL)
1715                 return -EMSGSIZE;
1716
1717         id = nlmsg_data(nlh);
1718         memcpy(&id->sa_id.daddr, &x->id.daddr,sizeof(x->id.daddr));
1719         id->sa_id.spi = x->id.spi;
1720         id->sa_id.family = x->props.family;
1721         id->sa_id.proto = x->id.proto;
1722         memcpy(&id->saddr, &x->props.saddr,sizeof(x->props.saddr));
1723         id->reqid = x->props.reqid;
1724         id->flags = c->data.aevent;
1725
1726         if (x->replay_esn)
1727                 NLA_PUT(skb, XFRMA_REPLAY_ESN_VAL,
1728                         xfrm_replay_state_esn_len(x->replay_esn),
1729                         x->replay_esn);
1730         else
1731                 NLA_PUT(skb, XFRMA_REPLAY_VAL, sizeof(x->replay), &x->replay);
1732
1733         NLA_PUT(skb, XFRMA_LTIME_VAL, sizeof(x->curlft), &x->curlft);
1734
1735         if (id->flags & XFRM_AE_RTHR)
1736                 NLA_PUT_U32(skb, XFRMA_REPLAY_THRESH, x->replay_maxdiff);
1737
1738         if (id->flags & XFRM_AE_ETHR)
1739                 NLA_PUT_U32(skb, XFRMA_ETIMER_THRESH,
1740                             x->replay_maxage * 10 / HZ);
1741
1742         if (xfrm_mark_put(skb, &x->mark))
1743                 goto nla_put_failure;
1744
1745         return nlmsg_end(skb, nlh);
1746
1747 nla_put_failure:
1748         nlmsg_cancel(skb, nlh);
1749         return -EMSGSIZE;
1750 }
1751
1752 static int xfrm_get_ae(struct sk_buff *skb, struct nlmsghdr *nlh,
1753                 struct nlattr **attrs)
1754 {
1755         struct net *net = sock_net(skb->sk);
1756         struct xfrm_state *x;
1757         struct sk_buff *r_skb;
1758         int err;
1759         struct km_event c;
1760         u32 mark;
1761         struct xfrm_mark m;
1762         struct xfrm_aevent_id *p = nlmsg_data(nlh);
1763         struct xfrm_usersa_id *id = &p->sa_id;
1764
1765         mark = xfrm_mark_get(attrs, &m);
1766
1767         x = xfrm_state_lookup(net, mark, &id->daddr, id->spi, id->proto, id->family);
1768         if (x == NULL)
1769                 return -ESRCH;
1770
1771         r_skb = nlmsg_new(xfrm_aevent_msgsize(x), GFP_ATOMIC);
1772         if (r_skb == NULL) {
1773                 xfrm_state_put(x);
1774                 return -ENOMEM;
1775         }
1776
1777         /*
1778          * XXX: is this lock really needed - none of the other
1779          * gets lock (the concern is things getting updated
1780          * while we are still reading) - jhs
1781         */
1782         spin_lock_bh(&x->lock);
1783         c.data.aevent = p->flags;
1784         c.seq = nlh->nlmsg_seq;
1785         c.pid = nlh->nlmsg_pid;
1786
1787         if (build_aevent(r_skb, x, &c) < 0)
1788                 BUG();
1789         err = nlmsg_unicast(net->xfrm.nlsk, r_skb, NETLINK_CB(skb).pid);
1790         spin_unlock_bh(&x->lock);
1791         xfrm_state_put(x);
1792         return err;
1793 }
1794
1795 static int xfrm_new_ae(struct sk_buff *skb, struct nlmsghdr *nlh,
1796                 struct nlattr **attrs)
1797 {
1798         struct net *net = sock_net(skb->sk);
1799         struct xfrm_state *x;
1800         struct km_event c;
1801         int err = - EINVAL;
1802         u32 mark = 0;
1803         struct xfrm_mark m;
1804         struct xfrm_aevent_id *p = nlmsg_data(nlh);
1805         struct nlattr *rp = attrs[XFRMA_REPLAY_VAL];
1806         struct nlattr *re = attrs[XFRMA_REPLAY_ESN_VAL];
1807         struct nlattr *lt = attrs[XFRMA_LTIME_VAL];
1808
1809         if (!lt && !rp && !re)
1810                 return err;
1811
1812         /* pedantic mode - thou shalt sayeth replaceth */
1813         if (!(nlh->nlmsg_flags&NLM_F_REPLACE))
1814                 return err;
1815
1816         mark = xfrm_mark_get(attrs, &m);
1817
1818         x = xfrm_state_lookup(net, mark, &p->sa_id.daddr, p->sa_id.spi, p->sa_id.proto, p->sa_id.family);
1819         if (x == NULL)
1820                 return -ESRCH;
1821
1822         if (x->km.state != XFRM_STATE_VALID)
1823                 goto out;
1824
1825         err = xfrm_replay_verify_len(x->replay_esn, rp);
1826         if (err)
1827                 goto out;
1828
1829         spin_lock_bh(&x->lock);
1830         xfrm_update_ae_params(x, attrs, 1);
1831         spin_unlock_bh(&x->lock);
1832
1833         c.event = nlh->nlmsg_type;
1834         c.seq = nlh->nlmsg_seq;
1835         c.pid = nlh->nlmsg_pid;
1836         c.data.aevent = XFRM_AE_CU;
1837         km_state_notify(x, &c);
1838         err = 0;
1839 out:
1840         xfrm_state_put(x);
1841         return err;
1842 }
1843
1844 static int xfrm_flush_policy(struct sk_buff *skb, struct nlmsghdr *nlh,
1845                 struct nlattr **attrs)
1846 {
1847         struct net *net = sock_net(skb->sk);
1848         struct km_event c;
1849         u8 type = XFRM_POLICY_TYPE_MAIN;
1850         int err;
1851         struct xfrm_audit audit_info;
1852
1853         err = copy_from_user_policy_type(&type, attrs);
1854         if (err)
1855                 return err;
1856
1857         audit_info.loginuid = audit_get_loginuid(current);
1858         audit_info.sessionid = audit_get_sessionid(current);
1859         security_task_getsecid(current, &audit_info.secid);
1860         err = xfrm_policy_flush(net, type, &audit_info);
1861         if (err) {
1862                 if (err == -ESRCH) /* empty table */
1863                         return 0;
1864                 return err;
1865         }
1866
1867         c.data.type = type;
1868         c.event = nlh->nlmsg_type;
1869         c.seq = nlh->nlmsg_seq;
1870         c.pid = nlh->nlmsg_pid;
1871         c.net = net;
1872         km_policy_notify(NULL, 0, &c);
1873         return 0;
1874 }
1875
1876 static int xfrm_add_pol_expire(struct sk_buff *skb, struct nlmsghdr *nlh,
1877                 struct nlattr **attrs)
1878 {
1879         struct net *net = sock_net(skb->sk);
1880         struct xfrm_policy *xp;
1881         struct xfrm_user_polexpire *up = nlmsg_data(nlh);
1882         struct xfrm_userpolicy_info *p = &up->pol;
1883         u8 type = XFRM_POLICY_TYPE_MAIN;
1884         int err = -ENOENT;
1885         struct xfrm_mark m;
1886         u32 mark = xfrm_mark_get(attrs, &m);
1887
1888         err = copy_from_user_policy_type(&type, attrs);
1889         if (err)
1890                 return err;
1891
1892         err = verify_policy_dir(p->dir);
1893         if (err)
1894                 return err;
1895
1896         if (p->index)
1897                 xp = xfrm_policy_byid(net, mark, type, p->dir, p->index, 0, &err);
1898         else {
1899                 struct nlattr *rt = attrs[XFRMA_SEC_CTX];
1900                 struct xfrm_sec_ctx *ctx;
1901
1902                 err = verify_sec_ctx_len(attrs);
1903                 if (err)
1904                         return err;
1905
1906                 ctx = NULL;
1907                 if (rt) {
1908                         struct xfrm_user_sec_ctx *uctx = nla_data(rt);
1909
1910                         err = security_xfrm_policy_alloc(&ctx, uctx);
1911                         if (err)
1912                                 return err;
1913                 }
1914                 xp = xfrm_policy_bysel_ctx(net, mark, type, p->dir,
1915                                            &p->sel, ctx, 0, &err);
1916                 security_xfrm_policy_free(ctx);
1917         }
1918         if (xp == NULL)
1919                 return -ENOENT;
1920
1921         if (unlikely(xp->walk.dead))
1922                 goto out;
1923
1924         err = 0;
1925         if (up->hard) {
1926                 uid_t loginuid = audit_get_loginuid(current);
1927                 u32 sessionid = audit_get_sessionid(current);
1928                 u32 sid;
1929
1930                 security_task_getsecid(current, &sid);
1931                 xfrm_policy_delete(xp, p->dir);
1932                 xfrm_audit_policy_delete(xp, 1, loginuid, sessionid, sid);
1933
1934         } else {
1935                 // reset the timers here?
1936                 WARN(1, "Dont know what to do with soft policy expire\n");
1937         }
1938         km_policy_expired(xp, p->dir, up->hard, current->pid);
1939
1940 out:
1941         xfrm_pol_put(xp);
1942         return err;
1943 }
1944
1945 static int xfrm_add_sa_expire(struct sk_buff *skb, struct nlmsghdr *nlh,
1946                 struct nlattr **attrs)
1947 {
1948         struct net *net = sock_net(skb->sk);
1949         struct xfrm_state *x;
1950         int err;
1951         struct xfrm_user_expire *ue = nlmsg_data(nlh);
1952         struct xfrm_usersa_info *p = &ue->state;
1953         struct xfrm_mark m;
1954         u32 mark = xfrm_mark_get(attrs, &m);
1955
1956         x = xfrm_state_lookup(net, mark, &p->id.daddr, p->id.spi, p->id.proto, p->family);
1957
1958         err = -ENOENT;
1959         if (x == NULL)
1960                 return err;
1961
1962         spin_lock_bh(&x->lock);
1963         err = -EINVAL;
1964         if (x->km.state != XFRM_STATE_VALID)
1965                 goto out;
1966         km_state_expired(x, ue->hard, current->pid);
1967
1968         if (ue->hard) {
1969                 uid_t loginuid = audit_get_loginuid(current);
1970                 u32 sessionid = audit_get_sessionid(current);
1971                 u32 sid;
1972
1973                 security_task_getsecid(current, &sid);
1974                 __xfrm_state_delete(x);
1975                 xfrm_audit_state_delete(x, 1, loginuid, sessionid, sid);
1976         }
1977         err = 0;
1978 out:
1979         spin_unlock_bh(&x->lock);
1980         xfrm_state_put(x);
1981         return err;
1982 }
1983
1984 static int xfrm_add_acquire(struct sk_buff *skb, struct nlmsghdr *nlh,
1985                 struct nlattr **attrs)
1986 {
1987         struct net *net = sock_net(skb->sk);
1988         struct xfrm_policy *xp;
1989         struct xfrm_user_tmpl *ut;
1990         int i;
1991         struct nlattr *rt = attrs[XFRMA_TMPL];
1992         struct xfrm_mark mark;
1993
1994         struct xfrm_user_acquire *ua = nlmsg_data(nlh);
1995         struct xfrm_state *x = xfrm_state_alloc(net);
1996         int err = -ENOMEM;
1997
1998         if (!x)
1999                 goto nomem;
2000
2001         xfrm_mark_get(attrs, &mark);
2002
2003         err = verify_newpolicy_info(&ua->policy);
2004         if (err)
2005                 goto bad_policy;
2006
2007         /*   build an XP */
2008         xp = xfrm_policy_construct(net, &ua->policy, attrs, &err);
2009         if (!xp)
2010                 goto free_state;
2011
2012         memcpy(&x->id, &ua->id, sizeof(ua->id));
2013         memcpy(&x->props.saddr, &ua->saddr, sizeof(ua->saddr));
2014         memcpy(&x->sel, &ua->sel, sizeof(ua->sel));
2015         xp->mark.m = x->mark.m = mark.m;
2016         xp->mark.v = x->mark.v = mark.v;
2017         ut = nla_data(rt);
2018         /* extract the templates and for each call km_key */
2019         for (i = 0; i < xp->xfrm_nr; i++, ut++) {
2020                 struct xfrm_tmpl *t = &xp->xfrm_vec[i];
2021                 memcpy(&x->id, &t->id, sizeof(x->id));
2022                 x->props.mode = t->mode;
2023                 x->props.reqid = t->reqid;
2024                 x->props.family = ut->family;
2025                 t->aalgos = ua->aalgos;
2026                 t->ealgos = ua->ealgos;
2027                 t->calgos = ua->calgos;
2028                 err = km_query(x, t, xp);
2029
2030         }
2031
2032         kfree(x);
2033         kfree(xp);
2034
2035         return 0;
2036
2037 bad_policy:
2038         WARN(1, "BAD policy passed\n");
2039 free_state:
2040         kfree(x);
2041 nomem:
2042         return err;
2043 }
2044
2045 #ifdef CONFIG_XFRM_MIGRATE
2046 static int copy_from_user_migrate(struct xfrm_migrate *ma,
2047                                   struct xfrm_kmaddress *k,
2048                                   struct nlattr **attrs, int *num)
2049 {
2050         struct nlattr *rt = attrs[XFRMA_MIGRATE];
2051         struct xfrm_user_migrate *um;
2052         int i, num_migrate;
2053
2054         if (k != NULL) {
2055                 struct xfrm_user_kmaddress *uk;
2056
2057                 uk = nla_data(attrs[XFRMA_KMADDRESS]);
2058                 memcpy(&k->local, &uk->local, sizeof(k->local));
2059                 memcpy(&k->remote, &uk->remote, sizeof(k->remote));
2060                 k->family = uk->family;
2061                 k->reserved = uk->reserved;
2062         }
2063
2064         um = nla_data(rt);
2065         num_migrate = nla_len(rt) / sizeof(*um);
2066
2067         if (num_migrate <= 0 || num_migrate > XFRM_MAX_DEPTH)
2068                 return -EINVAL;
2069
2070         for (i = 0; i < num_migrate; i++, um++, ma++) {
2071                 memcpy(&ma->old_daddr, &um->old_daddr, sizeof(ma->old_daddr));
2072                 memcpy(&ma->old_saddr, &um->old_saddr, sizeof(ma->old_saddr));
2073                 memcpy(&ma->new_daddr, &um->new_daddr, sizeof(ma->new_daddr));
2074                 memcpy(&ma->new_saddr, &um->new_saddr, sizeof(ma->new_saddr));
2075
2076                 ma->proto = um->proto;
2077                 ma->mode = um->mode;
2078                 ma->reqid = um->reqid;
2079
2080                 ma->old_family = um->old_family;
2081                 ma->new_family = um->new_family;
2082         }
2083
2084         *num = i;
2085         return 0;
2086 }
2087
2088 static int xfrm_do_migrate(struct sk_buff *skb, struct nlmsghdr *nlh,
2089                            struct nlattr **attrs)
2090 {
2091         struct xfrm_userpolicy_id *pi = nlmsg_data(nlh);
2092         struct xfrm_migrate m[XFRM_MAX_DEPTH];
2093         struct xfrm_kmaddress km, *kmp;
2094         u8 type;
2095         int err;
2096         int n = 0;
2097
2098         if (attrs[XFRMA_MIGRATE] == NULL)
2099                 return -EINVAL;
2100
2101         kmp = attrs[XFRMA_KMADDRESS] ? &km : NULL;
2102
2103         err = copy_from_user_policy_type(&type, attrs);
2104         if (err)
2105                 return err;
2106
2107         err = copy_from_user_migrate((struct xfrm_migrate *)m, kmp, attrs, &n);
2108         if (err)
2109                 return err;
2110
2111         if (!n)
2112                 return 0;
2113
2114         xfrm_migrate(&pi->sel, pi->dir, type, m, n, kmp);
2115
2116         return 0;
2117 }
2118 #else
2119 static int xfrm_do_migrate(struct sk_buff *skb, struct nlmsghdr *nlh,
2120                            struct nlattr **attrs)
2121 {
2122         return -ENOPROTOOPT;
2123 }
2124 #endif
2125
2126 #ifdef CONFIG_XFRM_MIGRATE
2127 static int copy_to_user_migrate(const struct xfrm_migrate *m, struct sk_buff *skb)
2128 {
2129         struct xfrm_user_migrate um;
2130
2131         memset(&um, 0, sizeof(um));
2132         um.proto = m->proto;
2133         um.mode = m->mode;
2134         um.reqid = m->reqid;
2135         um.old_family = m->old_family;
2136         memcpy(&um.old_daddr, &m->old_daddr, sizeof(um.old_daddr));
2137         memcpy(&um.old_saddr, &m->old_saddr, sizeof(um.old_saddr));
2138         um.new_family = m->new_family;
2139         memcpy(&um.new_daddr, &m->new_daddr, sizeof(um.new_daddr));
2140         memcpy(&um.new_saddr, &m->new_saddr, sizeof(um.new_saddr));
2141
2142         return nla_put(skb, XFRMA_MIGRATE, sizeof(um), &um);
2143 }
2144
2145 static int copy_to_user_kmaddress(const struct xfrm_kmaddress *k, struct sk_buff *skb)
2146 {
2147         struct xfrm_user_kmaddress uk;
2148
2149         memset(&uk, 0, sizeof(uk));
2150         uk.family = k->family;
2151         uk.reserved = k->reserved;
2152         memcpy(&uk.local, &k->local, sizeof(uk.local));
2153         memcpy(&uk.remote, &k->remote, sizeof(uk.remote));
2154
2155         return nla_put(skb, XFRMA_KMADDRESS, sizeof(uk), &uk);
2156 }
2157
2158 static inline size_t xfrm_migrate_msgsize(int num_migrate, int with_kma)
2159 {
2160         return NLMSG_ALIGN(sizeof(struct xfrm_userpolicy_id))
2161               + (with_kma ? nla_total_size(sizeof(struct xfrm_kmaddress)) : 0)
2162               + nla_total_size(sizeof(struct xfrm_user_migrate) * num_migrate)
2163               + userpolicy_type_attrsize();
2164 }
2165
2166 static int build_migrate(struct sk_buff *skb, const struct xfrm_migrate *m,
2167                          int num_migrate, const struct xfrm_kmaddress *k,
2168                          const struct xfrm_selector *sel, u8 dir, u8 type)
2169 {
2170         const struct xfrm_migrate *mp;
2171         struct xfrm_userpolicy_id *pol_id;
2172         struct nlmsghdr *nlh;
2173         int i;
2174
2175         nlh = nlmsg_put(skb, 0, 0, XFRM_MSG_MIGRATE, sizeof(*pol_id), 0);
2176         if (nlh == NULL)
2177                 return -EMSGSIZE;
2178
2179         pol_id = nlmsg_data(nlh);
2180         /* copy data from selector, dir, and type to the pol_id */
2181         memset(pol_id, 0, sizeof(*pol_id));
2182         memcpy(&pol_id->sel, sel, sizeof(pol_id->sel));
2183         pol_id->dir = dir;
2184
2185         if (k != NULL && (copy_to_user_kmaddress(k, skb) < 0))
2186                         goto nlmsg_failure;
2187
2188         if (copy_to_user_policy_type(type, skb) < 0)
2189                 goto nlmsg_failure;
2190
2191         for (i = 0, mp = m ; i < num_migrate; i++, mp++) {
2192                 if (copy_to_user_migrate(mp, skb) < 0)
2193                         goto nlmsg_failure;
2194         }
2195
2196         return nlmsg_end(skb, nlh);
2197 nlmsg_failure:
2198         nlmsg_cancel(skb, nlh);
2199         return -EMSGSIZE;
2200 }
2201
2202 static int xfrm_send_migrate(const struct xfrm_selector *sel, u8 dir, u8 type,
2203                              const struct xfrm_migrate *m, int num_migrate,
2204                              const struct xfrm_kmaddress *k)
2205 {
2206         struct net *net = &init_net;
2207         struct sk_buff *skb;
2208
2209         skb = nlmsg_new(xfrm_migrate_msgsize(num_migrate, !!k), GFP_ATOMIC);
2210         if (skb == NULL)
2211                 return -ENOMEM;
2212
2213         /* build migrate */
2214         if (build_migrate(skb, m, num_migrate, k, sel, dir, type) < 0)
2215                 BUG();
2216
2217         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_MIGRATE, GFP_ATOMIC);
2218 }
2219 #else
2220 static int xfrm_send_migrate(const struct xfrm_selector *sel, u8 dir, u8 type,
2221                              const struct xfrm_migrate *m, int num_migrate,
2222                              const struct xfrm_kmaddress *k)
2223 {
2224         return -ENOPROTOOPT;
2225 }
2226 #endif
2227
2228 #define XMSGSIZE(type) sizeof(struct type)
2229
2230 static const int xfrm_msg_min[XFRM_NR_MSGTYPES] = {
2231         [XFRM_MSG_NEWSA       - XFRM_MSG_BASE] = XMSGSIZE(xfrm_usersa_info),
2232         [XFRM_MSG_DELSA       - XFRM_MSG_BASE] = XMSGSIZE(xfrm_usersa_id),
2233         [XFRM_MSG_GETSA       - XFRM_MSG_BASE] = XMSGSIZE(xfrm_usersa_id),
2234         [XFRM_MSG_NEWPOLICY   - XFRM_MSG_BASE] = XMSGSIZE(xfrm_userpolicy_info),
2235         [XFRM_MSG_DELPOLICY   - XFRM_MSG_BASE] = XMSGSIZE(xfrm_userpolicy_id),
2236         [XFRM_MSG_GETPOLICY   - XFRM_MSG_BASE] = XMSGSIZE(xfrm_userpolicy_id),
2237         [XFRM_MSG_ALLOCSPI    - XFRM_MSG_BASE] = XMSGSIZE(xfrm_userspi_info),
2238         [XFRM_MSG_ACQUIRE     - XFRM_MSG_BASE] = XMSGSIZE(xfrm_user_acquire),
2239         [XFRM_MSG_EXPIRE      - XFRM_MSG_BASE] = XMSGSIZE(xfrm_user_expire),
2240         [XFRM_MSG_UPDPOLICY   - XFRM_MSG_BASE] = XMSGSIZE(xfrm_userpolicy_info),
2241         [XFRM_MSG_UPDSA       - XFRM_MSG_BASE] = XMSGSIZE(xfrm_usersa_info),
2242         [XFRM_MSG_POLEXPIRE   - XFRM_MSG_BASE] = XMSGSIZE(xfrm_user_polexpire),
2243         [XFRM_MSG_FLUSHSA     - XFRM_MSG_BASE] = XMSGSIZE(xfrm_usersa_flush),
2244         [XFRM_MSG_FLUSHPOLICY - XFRM_MSG_BASE] = 0,
2245         [XFRM_MSG_NEWAE       - XFRM_MSG_BASE] = XMSGSIZE(xfrm_aevent_id),
2246         [XFRM_MSG_GETAE       - XFRM_MSG_BASE] = XMSGSIZE(xfrm_aevent_id),
2247         [XFRM_MSG_REPORT      - XFRM_MSG_BASE] = XMSGSIZE(xfrm_user_report),
2248         [XFRM_MSG_MIGRATE     - XFRM_MSG_BASE] = XMSGSIZE(xfrm_userpolicy_id),
2249         [XFRM_MSG_GETSADINFO  - XFRM_MSG_BASE] = sizeof(u32),
2250         [XFRM_MSG_GETSPDINFO  - XFRM_MSG_BASE] = sizeof(u32),
2251 };
2252
2253 #undef XMSGSIZE
2254
2255 static const struct nla_policy xfrma_policy[XFRMA_MAX+1] = {
2256         [XFRMA_SA]              = { .len = sizeof(struct xfrm_usersa_info)},
2257         [XFRMA_POLICY]          = { .len = sizeof(struct xfrm_userpolicy_info)},
2258         [XFRMA_LASTUSED]        = { .type = NLA_U64},
2259         [XFRMA_ALG_AUTH_TRUNC]  = { .len = sizeof(struct xfrm_algo_auth)},
2260         [XFRMA_ALG_AEAD]        = { .len = sizeof(struct xfrm_algo_aead) },
2261         [XFRMA_ALG_AUTH]        = { .len = sizeof(struct xfrm_algo) },
2262         [XFRMA_ALG_CRYPT]       = { .len = sizeof(struct xfrm_algo) },
2263         [XFRMA_ALG_COMP]        = { .len = sizeof(struct xfrm_algo) },
2264         [XFRMA_ENCAP]           = { .len = sizeof(struct xfrm_encap_tmpl) },
2265         [XFRMA_TMPL]            = { .len = sizeof(struct xfrm_user_tmpl) },
2266         [XFRMA_SEC_CTX]         = { .len = sizeof(struct xfrm_sec_ctx) },
2267         [XFRMA_LTIME_VAL]       = { .len = sizeof(struct xfrm_lifetime_cur) },
2268         [XFRMA_REPLAY_VAL]      = { .len = sizeof(struct xfrm_replay_state) },
2269         [XFRMA_REPLAY_THRESH]   = { .type = NLA_U32 },
2270         [XFRMA_ETIMER_THRESH]   = { .type = NLA_U32 },
2271         [XFRMA_SRCADDR]         = { .len = sizeof(xfrm_address_t) },
2272         [XFRMA_COADDR]          = { .len = sizeof(xfrm_address_t) },
2273         [XFRMA_POLICY_TYPE]     = { .len = sizeof(struct xfrm_userpolicy_type)},
2274         [XFRMA_MIGRATE]         = { .len = sizeof(struct xfrm_user_migrate) },
2275         [XFRMA_KMADDRESS]       = { .len = sizeof(struct xfrm_user_kmaddress) },
2276         [XFRMA_MARK]            = { .len = sizeof(struct xfrm_mark) },
2277         [XFRMA_TFCPAD]          = { .type = NLA_U32 },
2278         [XFRMA_REPLAY_ESN_VAL]  = { .len = sizeof(struct xfrm_replay_state_esn) },
2279 };
2280
2281 static struct xfrm_link {
2282         int (*doit)(struct sk_buff *, struct nlmsghdr *, struct nlattr **);
2283         int (*dump)(struct sk_buff *, struct netlink_callback *);
2284         int (*done)(struct netlink_callback *);
2285 } xfrm_dispatch[XFRM_NR_MSGTYPES] = {
2286         [XFRM_MSG_NEWSA       - XFRM_MSG_BASE] = { .doit = xfrm_add_sa        },
2287         [XFRM_MSG_DELSA       - XFRM_MSG_BASE] = { .doit = xfrm_del_sa        },
2288         [XFRM_MSG_GETSA       - XFRM_MSG_BASE] = { .doit = xfrm_get_sa,
2289                                                    .dump = xfrm_dump_sa,
2290                                                    .done = xfrm_dump_sa_done  },
2291         [XFRM_MSG_NEWPOLICY   - XFRM_MSG_BASE] = { .doit = xfrm_add_policy    },
2292         [XFRM_MSG_DELPOLICY   - XFRM_MSG_BASE] = { .doit = xfrm_get_policy    },
2293         [XFRM_MSG_GETPOLICY   - XFRM_MSG_BASE] = { .doit = xfrm_get_policy,
2294                                                    .dump = xfrm_dump_policy,
2295                                                    .done = xfrm_dump_policy_done },
2296         [XFRM_MSG_ALLOCSPI    - XFRM_MSG_BASE] = { .doit = xfrm_alloc_userspi },
2297         [XFRM_MSG_ACQUIRE     - XFRM_MSG_BASE] = { .doit = xfrm_add_acquire   },
2298         [XFRM_MSG_EXPIRE      - XFRM_MSG_BASE] = { .doit = xfrm_add_sa_expire },
2299         [XFRM_MSG_UPDPOLICY   - XFRM_MSG_BASE] = { .doit = xfrm_add_policy    },
2300         [XFRM_MSG_UPDSA       - XFRM_MSG_BASE] = { .doit = xfrm_add_sa        },
2301         [XFRM_MSG_POLEXPIRE   - XFRM_MSG_BASE] = { .doit = xfrm_add_pol_expire},
2302         [XFRM_MSG_FLUSHSA     - XFRM_MSG_BASE] = { .doit = xfrm_flush_sa      },
2303         [XFRM_MSG_FLUSHPOLICY - XFRM_MSG_BASE] = { .doit = xfrm_flush_policy  },
2304         [XFRM_MSG_NEWAE       - XFRM_MSG_BASE] = { .doit = xfrm_new_ae  },
2305         [XFRM_MSG_GETAE       - XFRM_MSG_BASE] = { .doit = xfrm_get_ae  },
2306         [XFRM_MSG_MIGRATE     - XFRM_MSG_BASE] = { .doit = xfrm_do_migrate    },
2307         [XFRM_MSG_GETSADINFO  - XFRM_MSG_BASE] = { .doit = xfrm_get_sadinfo   },
2308         [XFRM_MSG_GETSPDINFO  - XFRM_MSG_BASE] = { .doit = xfrm_get_spdinfo   },
2309 };
2310
2311 static int xfrm_user_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh)
2312 {
2313         struct net *net = sock_net(skb->sk);
2314         struct nlattr *attrs[XFRMA_MAX+1];
2315         struct xfrm_link *link;
2316         int type, err;
2317
2318         type = nlh->nlmsg_type;
2319         if (type > XFRM_MSG_MAX)
2320                 return -EINVAL;
2321
2322         type -= XFRM_MSG_BASE;
2323         link = &xfrm_dispatch[type];
2324
2325         /* All operations require privileges, even GET */
2326         if (security_netlink_recv(skb, CAP_NET_ADMIN))
2327                 return -EPERM;
2328
2329         if ((type == (XFRM_MSG_GETSA - XFRM_MSG_BASE) ||
2330              type == (XFRM_MSG_GETPOLICY - XFRM_MSG_BASE)) &&
2331             (nlh->nlmsg_flags & NLM_F_DUMP)) {
2332                 if (link->dump == NULL)
2333                         return -EINVAL;
2334
2335                 return netlink_dump_start(net->xfrm.nlsk, skb, nlh,
2336                                           link->dump, link->done, 0);
2337         }
2338
2339         err = nlmsg_parse(nlh, xfrm_msg_min[type], attrs, XFRMA_MAX,
2340                           xfrma_policy);
2341         if (err < 0)
2342                 return err;
2343
2344         if (link->doit == NULL)
2345                 return -EINVAL;
2346
2347         return link->doit(skb, nlh, attrs);
2348 }
2349
2350 static void xfrm_netlink_rcv(struct sk_buff *skb)
2351 {
2352         mutex_lock(&xfrm_cfg_mutex);
2353         netlink_rcv_skb(skb, &xfrm_user_rcv_msg);
2354         mutex_unlock(&xfrm_cfg_mutex);
2355 }
2356
2357 static inline size_t xfrm_expire_msgsize(void)
2358 {
2359         return NLMSG_ALIGN(sizeof(struct xfrm_user_expire))
2360                + nla_total_size(sizeof(struct xfrm_mark));
2361 }
2362
2363 static int build_expire(struct sk_buff *skb, struct xfrm_state *x, const struct km_event *c)
2364 {
2365         struct xfrm_user_expire *ue;
2366         struct nlmsghdr *nlh;
2367
2368         nlh = nlmsg_put(skb, c->pid, 0, XFRM_MSG_EXPIRE, sizeof(*ue), 0);
2369         if (nlh == NULL)
2370                 return -EMSGSIZE;
2371
2372         ue = nlmsg_data(nlh);
2373         copy_to_user_state(x, &ue->state);
2374         ue->hard = (c->data.hard != 0) ? 1 : 0;
2375
2376         if (xfrm_mark_put(skb, &x->mark))
2377                 goto nla_put_failure;
2378
2379         return nlmsg_end(skb, nlh);
2380
2381 nla_put_failure:
2382         return -EMSGSIZE;
2383 }
2384
2385 static int xfrm_exp_state_notify(struct xfrm_state *x, const struct km_event *c)
2386 {
2387         struct net *net = xs_net(x);
2388         struct sk_buff *skb;
2389
2390         skb = nlmsg_new(xfrm_expire_msgsize(), GFP_ATOMIC);
2391         if (skb == NULL)
2392                 return -ENOMEM;
2393
2394         if (build_expire(skb, x, c) < 0) {
2395                 kfree_skb(skb);
2396                 return -EMSGSIZE;
2397         }
2398
2399         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_EXPIRE, GFP_ATOMIC);
2400 }
2401
2402 static int xfrm_aevent_state_notify(struct xfrm_state *x, const struct km_event *c)
2403 {
2404         struct net *net = xs_net(x);
2405         struct sk_buff *skb;
2406
2407         skb = nlmsg_new(xfrm_aevent_msgsize(x), GFP_ATOMIC);
2408         if (skb == NULL)
2409                 return -ENOMEM;
2410
2411         if (build_aevent(skb, x, c) < 0)
2412                 BUG();
2413
2414         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_AEVENTS, GFP_ATOMIC);
2415 }
2416
2417 static int xfrm_notify_sa_flush(const struct km_event *c)
2418 {
2419         struct net *net = c->net;
2420         struct xfrm_usersa_flush *p;
2421         struct nlmsghdr *nlh;
2422         struct sk_buff *skb;
2423         int len = NLMSG_ALIGN(sizeof(struct xfrm_usersa_flush));
2424
2425         skb = nlmsg_new(len, GFP_ATOMIC);
2426         if (skb == NULL)
2427                 return -ENOMEM;
2428
2429         nlh = nlmsg_put(skb, c->pid, c->seq, XFRM_MSG_FLUSHSA, sizeof(*p), 0);
2430         if (nlh == NULL) {
2431                 kfree_skb(skb);
2432                 return -EMSGSIZE;
2433         }
2434
2435         p = nlmsg_data(nlh);
2436         p->proto = c->data.proto;
2437
2438         nlmsg_end(skb, nlh);
2439
2440         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_SA, GFP_ATOMIC);
2441 }
2442
2443 static inline size_t xfrm_sa_len(struct xfrm_state *x)
2444 {
2445         size_t l = 0;
2446         if (x->aead)
2447                 l += nla_total_size(aead_len(x->aead));
2448         if (x->aalg) {
2449                 l += nla_total_size(sizeof(struct xfrm_algo) +
2450                                     (x->aalg->alg_key_len + 7) / 8);
2451                 l += nla_total_size(xfrm_alg_auth_len(x->aalg));
2452         }
2453         if (x->ealg)
2454                 l += nla_total_size(xfrm_alg_len(x->ealg));
2455         if (x->calg)
2456                 l += nla_total_size(sizeof(*x->calg));
2457         if (x->encap)
2458                 l += nla_total_size(sizeof(*x->encap));
2459         if (x->tfcpad)
2460                 l += nla_total_size(sizeof(x->tfcpad));
2461         if (x->replay_esn)
2462                 l += nla_total_size(xfrm_replay_state_esn_len(x->replay_esn));
2463         if (x->security)
2464                 l += nla_total_size(sizeof(struct xfrm_user_sec_ctx) +
2465                                     x->security->ctx_len);
2466         if (x->coaddr)
2467                 l += nla_total_size(sizeof(*x->coaddr));
2468
2469         /* Must count x->lastused as it may become non-zero behind our back. */
2470         l += nla_total_size(sizeof(u64));
2471
2472         return l;
2473 }
2474
2475 static int xfrm_notify_sa(struct xfrm_state *x, const struct km_event *c)
2476 {
2477         struct net *net = xs_net(x);
2478         struct xfrm_usersa_info *p;
2479         struct xfrm_usersa_id *id;
2480         struct nlmsghdr *nlh;
2481         struct sk_buff *skb;
2482         int len = xfrm_sa_len(x);
2483         int headlen;
2484
2485         headlen = sizeof(*p);
2486         if (c->event == XFRM_MSG_DELSA) {
2487                 len += nla_total_size(headlen);
2488                 headlen = sizeof(*id);
2489                 len += nla_total_size(sizeof(struct xfrm_mark));
2490         }
2491         len += NLMSG_ALIGN(headlen);
2492
2493         skb = nlmsg_new(len, GFP_ATOMIC);
2494         if (skb == NULL)
2495                 return -ENOMEM;
2496
2497         nlh = nlmsg_put(skb, c->pid, c->seq, c->event, headlen, 0);
2498         if (nlh == NULL)
2499                 goto nla_put_failure;
2500
2501         p = nlmsg_data(nlh);
2502         if (c->event == XFRM_MSG_DELSA) {
2503                 struct nlattr *attr;
2504
2505                 id = nlmsg_data(nlh);
2506                 memcpy(&id->daddr, &x->id.daddr, sizeof(id->daddr));
2507                 id->spi = x->id.spi;
2508                 id->family = x->props.family;
2509                 id->proto = x->id.proto;
2510
2511                 attr = nla_reserve(skb, XFRMA_SA, sizeof(*p));
2512                 if (attr == NULL)
2513                         goto nla_put_failure;
2514
2515                 p = nla_data(attr);
2516         }
2517
2518         if (copy_to_user_state_extra(x, p, skb))
2519                 goto nla_put_failure;
2520
2521         nlmsg_end(skb, nlh);
2522
2523         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_SA, GFP_ATOMIC);
2524
2525 nla_put_failure:
2526         /* Somebody screwed up with xfrm_sa_len! */
2527         WARN_ON(1);
2528         kfree_skb(skb);
2529         return -1;
2530 }
2531
2532 static int xfrm_send_state_notify(struct xfrm_state *x, const struct km_event *c)
2533 {
2534
2535         switch (c->event) {
2536         case XFRM_MSG_EXPIRE:
2537                 return xfrm_exp_state_notify(x, c);
2538         case XFRM_MSG_NEWAE:
2539                 return xfrm_aevent_state_notify(x, c);
2540         case XFRM_MSG_DELSA:
2541         case XFRM_MSG_UPDSA:
2542         case XFRM_MSG_NEWSA:
2543                 return xfrm_notify_sa(x, c);
2544         case XFRM_MSG_FLUSHSA:
2545                 return xfrm_notify_sa_flush(c);
2546         default:
2547                 printk(KERN_NOTICE "xfrm_user: Unknown SA event %d\n",
2548                        c->event);
2549                 break;
2550         }
2551
2552         return 0;
2553
2554 }
2555
2556 static inline size_t xfrm_acquire_msgsize(struct xfrm_state *x,
2557                                           struct xfrm_policy *xp)
2558 {
2559         return NLMSG_ALIGN(sizeof(struct xfrm_user_acquire))
2560                + nla_total_size(sizeof(struct xfrm_user_tmpl) * xp->xfrm_nr)
2561                + nla_total_size(sizeof(struct xfrm_mark))
2562                + nla_total_size(xfrm_user_sec_ctx_size(x->security))
2563                + userpolicy_type_attrsize();
2564 }
2565
2566 static int build_acquire(struct sk_buff *skb, struct xfrm_state *x,
2567                          struct xfrm_tmpl *xt, struct xfrm_policy *xp,
2568                          int dir)
2569 {
2570         struct xfrm_user_acquire *ua;
2571         struct nlmsghdr *nlh;
2572         __u32 seq = xfrm_get_acqseq();
2573
2574         nlh = nlmsg_put(skb, 0, 0, XFRM_MSG_ACQUIRE, sizeof(*ua), 0);
2575         if (nlh == NULL)
2576                 return -EMSGSIZE;
2577
2578         ua = nlmsg_data(nlh);
2579         memcpy(&ua->id, &x->id, sizeof(ua->id));
2580         memcpy(&ua->saddr, &x->props.saddr, sizeof(ua->saddr));
2581         memcpy(&ua->sel, &x->sel, sizeof(ua->sel));
2582         copy_to_user_policy(xp, &ua->policy, dir);
2583         ua->aalgos = xt->aalgos;
2584         ua->ealgos = xt->ealgos;
2585         ua->calgos = xt->calgos;
2586         ua->seq = x->km.seq = seq;
2587
2588         if (copy_to_user_tmpl(xp, skb) < 0)
2589                 goto nlmsg_failure;
2590         if (copy_to_user_state_sec_ctx(x, skb))
2591                 goto nlmsg_failure;
2592         if (copy_to_user_policy_type(xp->type, skb) < 0)
2593                 goto nlmsg_failure;
2594         if (xfrm_mark_put(skb, &xp->mark))
2595                 goto nla_put_failure;
2596
2597         return nlmsg_end(skb, nlh);
2598
2599 nla_put_failure:
2600 nlmsg_failure:
2601         nlmsg_cancel(skb, nlh);
2602         return -EMSGSIZE;
2603 }
2604
2605 static int xfrm_send_acquire(struct xfrm_state *x, struct xfrm_tmpl *xt,
2606                              struct xfrm_policy *xp, int dir)
2607 {
2608         struct net *net = xs_net(x);
2609         struct sk_buff *skb;
2610
2611         skb = nlmsg_new(xfrm_acquire_msgsize(x, xp), GFP_ATOMIC);
2612         if (skb == NULL)
2613                 return -ENOMEM;
2614
2615         if (build_acquire(skb, x, xt, xp, dir) < 0)
2616                 BUG();
2617
2618         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_ACQUIRE, GFP_ATOMIC);
2619 }
2620
2621 /* User gives us xfrm_user_policy_info followed by an array of 0
2622  * or more templates.
2623  */
2624 static struct xfrm_policy *xfrm_compile_policy(struct sock *sk, int opt,
2625                                                u8 *data, int len, int *dir)
2626 {
2627         struct net *net = sock_net(sk);
2628         struct xfrm_userpolicy_info *p = (struct xfrm_userpolicy_info *)data;
2629         struct xfrm_user_tmpl *ut = (struct xfrm_user_tmpl *) (p + 1);
2630         struct xfrm_policy *xp;
2631         int nr;
2632
2633         switch (sk->sk_family) {
2634         case AF_INET:
2635                 if (opt != IP_XFRM_POLICY) {
2636                         *dir = -EOPNOTSUPP;
2637                         return NULL;
2638                 }
2639                 break;
2640 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2641         case AF_INET6:
2642                 if (opt != IPV6_XFRM_POLICY) {
2643                         *dir = -EOPNOTSUPP;
2644                         return NULL;
2645                 }
2646                 break;
2647 #endif
2648         default:
2649                 *dir = -EINVAL;
2650                 return NULL;
2651         }
2652
2653         *dir = -EINVAL;
2654
2655         if (len < sizeof(*p) ||
2656             verify_newpolicy_info(p))
2657                 return NULL;
2658
2659         nr = ((len - sizeof(*p)) / sizeof(*ut));
2660         if (validate_tmpl(nr, ut, p->sel.family))
2661                 return NULL;
2662
2663         if (p->dir > XFRM_POLICY_OUT)
2664                 return NULL;
2665
2666         xp = xfrm_policy_alloc(net, GFP_ATOMIC);
2667         if (xp == NULL) {
2668                 *dir = -ENOBUFS;
2669                 return NULL;
2670         }
2671
2672         copy_from_user_policy(xp, p);
2673         xp->type = XFRM_POLICY_TYPE_MAIN;
2674         copy_templates(xp, ut, nr);
2675
2676         *dir = p->dir;
2677
2678         return xp;
2679 }
2680
2681 static inline size_t xfrm_polexpire_msgsize(struct xfrm_policy *xp)
2682 {
2683         return NLMSG_ALIGN(sizeof(struct xfrm_user_polexpire))
2684                + nla_total_size(sizeof(struct xfrm_user_tmpl) * xp->xfrm_nr)
2685                + nla_total_size(xfrm_user_sec_ctx_size(xp->security))
2686                + nla_total_size(sizeof(struct xfrm_mark))
2687                + userpolicy_type_attrsize();
2688 }
2689
2690 static int build_polexpire(struct sk_buff *skb, struct xfrm_policy *xp,
2691                            int dir, const struct km_event *c)
2692 {
2693         struct xfrm_user_polexpire *upe;
2694         struct nlmsghdr *nlh;
2695         int hard = c->data.hard;
2696
2697         nlh = nlmsg_put(skb, c->pid, 0, XFRM_MSG_POLEXPIRE, sizeof(*upe), 0);
2698         if (nlh == NULL)
2699                 return -EMSGSIZE;
2700
2701         upe = nlmsg_data(nlh);
2702         copy_to_user_policy(xp, &upe->pol, dir);
2703         if (copy_to_user_tmpl(xp, skb) < 0)
2704                 goto nlmsg_failure;
2705         if (copy_to_user_sec_ctx(xp, skb))
2706                 goto nlmsg_failure;
2707         if (copy_to_user_policy_type(xp->type, skb) < 0)
2708                 goto nlmsg_failure;
2709         if (xfrm_mark_put(skb, &xp->mark))
2710                 goto nla_put_failure;
2711         upe->hard = !!hard;
2712
2713         return nlmsg_end(skb, nlh);
2714
2715 nla_put_failure:
2716 nlmsg_failure:
2717         nlmsg_cancel(skb, nlh);
2718         return -EMSGSIZE;
2719 }
2720
2721 static int xfrm_exp_policy_notify(struct xfrm_policy *xp, int dir, const struct km_event *c)
2722 {
2723         struct net *net = xp_net(xp);
2724         struct sk_buff *skb;
2725
2726         skb = nlmsg_new(xfrm_polexpire_msgsize(xp), GFP_ATOMIC);
2727         if (skb == NULL)
2728                 return -ENOMEM;
2729
2730         if (build_polexpire(skb, xp, dir, c) < 0)
2731                 BUG();
2732
2733         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_EXPIRE, GFP_ATOMIC);
2734 }
2735
2736 static int xfrm_notify_policy(struct xfrm_policy *xp, int dir, const struct km_event *c)
2737 {
2738         struct net *net = xp_net(xp);
2739         struct xfrm_userpolicy_info *p;
2740         struct xfrm_userpolicy_id *id;
2741         struct nlmsghdr *nlh;
2742         struct sk_buff *skb;
2743         int len = nla_total_size(sizeof(struct xfrm_user_tmpl) * xp->xfrm_nr);
2744         int headlen;
2745
2746         headlen = sizeof(*p);
2747         if (c->event == XFRM_MSG_DELPOLICY) {
2748                 len += nla_total_size(headlen);
2749                 headlen = sizeof(*id);
2750         }
2751         len += userpolicy_type_attrsize();
2752         len += nla_total_size(sizeof(struct xfrm_mark));
2753         len += NLMSG_ALIGN(headlen);
2754
2755         skb = nlmsg_new(len, GFP_ATOMIC);
2756         if (skb == NULL)
2757                 return -ENOMEM;
2758
2759         nlh = nlmsg_put(skb, c->pid, c->seq, c->event, headlen, 0);
2760         if (nlh == NULL)
2761                 goto nlmsg_failure;
2762
2763         p = nlmsg_data(nlh);
2764         if (c->event == XFRM_MSG_DELPOLICY) {
2765                 struct nlattr *attr;
2766
2767                 id = nlmsg_data(nlh);
2768                 memset(id, 0, sizeof(*id));
2769                 id->dir = dir;
2770                 if (c->data.byid)
2771                         id->index = xp->index;
2772                 else
2773                         memcpy(&id->sel, &xp->selector, sizeof(id->sel));
2774
2775                 attr = nla_reserve(skb, XFRMA_POLICY, sizeof(*p));
2776                 if (attr == NULL)
2777                         goto nlmsg_failure;
2778
2779                 p = nla_data(attr);
2780         }
2781
2782         copy_to_user_policy(xp, p, dir);
2783         if (copy_to_user_tmpl(xp, skb) < 0)
2784                 goto nlmsg_failure;
2785         if (copy_to_user_policy_type(xp->type, skb) < 0)
2786                 goto nlmsg_failure;
2787
2788         if (xfrm_mark_put(skb, &xp->mark))
2789                 goto nla_put_failure;
2790
2791         nlmsg_end(skb, nlh);
2792
2793         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_POLICY, GFP_ATOMIC);
2794
2795 nla_put_failure:
2796 nlmsg_failure:
2797         kfree_skb(skb);
2798         return -1;
2799 }
2800
2801 static int xfrm_notify_policy_flush(const struct km_event *c)
2802 {
2803         struct net *net = c->net;
2804         struct nlmsghdr *nlh;
2805         struct sk_buff *skb;
2806
2807         skb = nlmsg_new(userpolicy_type_attrsize(), GFP_ATOMIC);
2808         if (skb == NULL)
2809                 return -ENOMEM;
2810
2811         nlh = nlmsg_put(skb, c->pid, c->seq, XFRM_MSG_FLUSHPOLICY, 0, 0);
2812         if (nlh == NULL)
2813                 goto nlmsg_failure;
2814         if (copy_to_user_policy_type(c->data.type, skb) < 0)
2815                 goto nlmsg_failure;
2816
2817         nlmsg_end(skb, nlh);
2818
2819         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_POLICY, GFP_ATOMIC);
2820
2821 nlmsg_failure:
2822         kfree_skb(skb);
2823         return -1;
2824 }
2825
2826 static int xfrm_send_policy_notify(struct xfrm_policy *xp, int dir, const struct km_event *c)
2827 {
2828
2829         switch (c->event) {
2830         case XFRM_MSG_NEWPOLICY:
2831         case XFRM_MSG_UPDPOLICY:
2832         case XFRM_MSG_DELPOLICY:
2833                 return xfrm_notify_policy(xp, dir, c);
2834         case XFRM_MSG_FLUSHPOLICY:
2835                 return xfrm_notify_policy_flush(c);
2836         case XFRM_MSG_POLEXPIRE:
2837                 return xfrm_exp_policy_notify(xp, dir, c);
2838         default:
2839                 printk(KERN_NOTICE "xfrm_user: Unknown Policy event %d\n",
2840                        c->event);
2841         }
2842
2843         return 0;
2844
2845 }
2846
2847 static inline size_t xfrm_report_msgsize(void)
2848 {
2849         return NLMSG_ALIGN(sizeof(struct xfrm_user_report));
2850 }
2851
2852 static int build_report(struct sk_buff *skb, u8 proto,
2853                         struct xfrm_selector *sel, xfrm_address_t *addr)
2854 {
2855         struct xfrm_user_report *ur;
2856         struct nlmsghdr *nlh;
2857
2858         nlh = nlmsg_put(skb, 0, 0, XFRM_MSG_REPORT, sizeof(*ur), 0);
2859         if (nlh == NULL)
2860                 return -EMSGSIZE;
2861
2862         ur = nlmsg_data(nlh);
2863         ur->proto = proto;
2864         memcpy(&ur->sel, sel, sizeof(ur->sel));
2865
2866         if (addr)
2867                 NLA_PUT(skb, XFRMA_COADDR, sizeof(*addr), addr);
2868
2869         return nlmsg_end(skb, nlh);
2870
2871 nla_put_failure:
2872         nlmsg_cancel(skb, nlh);
2873         return -EMSGSIZE;
2874 }
2875
2876 static int xfrm_send_report(struct net *net, u8 proto,
2877                             struct xfrm_selector *sel, xfrm_address_t *addr)
2878 {
2879         struct sk_buff *skb;
2880
2881         skb = nlmsg_new(xfrm_report_msgsize(), GFP_ATOMIC);
2882         if (skb == NULL)
2883                 return -ENOMEM;
2884
2885         if (build_report(skb, proto, sel, addr) < 0)
2886                 BUG();
2887
2888         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_REPORT, GFP_ATOMIC);
2889 }
2890
2891 static inline size_t xfrm_mapping_msgsize(void)
2892 {
2893         return NLMSG_ALIGN(sizeof(struct xfrm_user_mapping));
2894 }
2895
2896 static int build_mapping(struct sk_buff *skb, struct xfrm_state *x,
2897                          xfrm_address_t *new_saddr, __be16 new_sport)
2898 {
2899         struct xfrm_user_mapping *um;
2900         struct nlmsghdr *nlh;
2901
2902         nlh = nlmsg_put(skb, 0, 0, XFRM_MSG_MAPPING, sizeof(*um), 0);
2903         if (nlh == NULL)
2904                 return -EMSGSIZE;
2905
2906         um = nlmsg_data(nlh);
2907
2908         memcpy(&um->id.daddr, &x->id.daddr, sizeof(um->id.daddr));
2909         um->id.spi = x->id.spi;
2910         um->id.family = x->props.family;
2911         um->id.proto = x->id.proto;
2912         memcpy(&um->new_saddr, new_saddr, sizeof(um->new_saddr));
2913         memcpy(&um->old_saddr, &x->props.saddr, sizeof(um->old_saddr));
2914         um->new_sport = new_sport;
2915         um->old_sport = x->encap->encap_sport;
2916         um->reqid = x->props.reqid;
2917
2918         return nlmsg_end(skb, nlh);
2919 }
2920
2921 static int xfrm_send_mapping(struct xfrm_state *x, xfrm_address_t *ipaddr,
2922                              __be16 sport)
2923 {
2924         struct net *net = xs_net(x);
2925         struct sk_buff *skb;
2926
2927         if (x->id.proto != IPPROTO_ESP)
2928                 return -EINVAL;
2929
2930         if (!x->encap)
2931                 return -EINVAL;
2932
2933         skb = nlmsg_new(xfrm_mapping_msgsize(), GFP_ATOMIC);
2934         if (skb == NULL)
2935                 return -ENOMEM;
2936
2937         if (build_mapping(skb, x, ipaddr, sport) < 0)
2938                 BUG();
2939
2940         return nlmsg_multicast(net->xfrm.nlsk, skb, 0, XFRMNLGRP_MAPPING, GFP_ATOMIC);
2941 }
2942
2943 static struct xfrm_mgr netlink_mgr = {
2944         .id             = "netlink",
2945         .notify         = xfrm_send_state_notify,
2946         .acquire        = xfrm_send_acquire,
2947         .compile_policy = xfrm_compile_policy,
2948         .notify_policy  = xfrm_send_policy_notify,
2949         .report         = xfrm_send_report,
2950         .migrate        = xfrm_send_migrate,
2951         .new_mapping    = xfrm_send_mapping,
2952 };
2953
2954 static int __net_init xfrm_user_net_init(struct net *net)
2955 {
2956         struct sock *nlsk;
2957
2958         nlsk = netlink_kernel_create(net, NETLINK_XFRM, XFRMNLGRP_MAX,
2959                                      xfrm_netlink_rcv, NULL, THIS_MODULE);
2960         if (nlsk == NULL)
2961                 return -ENOMEM;
2962         net->xfrm.nlsk_stash = nlsk; /* Don't set to NULL */
2963         rcu_assign_pointer(net->xfrm.nlsk, nlsk);
2964         return 0;
2965 }
2966
2967 static void __net_exit xfrm_user_net_exit(struct list_head *net_exit_list)
2968 {
2969         struct net *net;
2970         list_for_each_entry(net, net_exit_list, exit_list)
2971                 RCU_INIT_POINTER(net->xfrm.nlsk, NULL);
2972         synchronize_net();
2973         list_for_each_entry(net, net_exit_list, exit_list)
2974                 netlink_kernel_release(net->xfrm.nlsk_stash);
2975 }
2976
2977 static struct pernet_operations xfrm_user_net_ops = {
2978         .init       = xfrm_user_net_init,
2979         .exit_batch = xfrm_user_net_exit,
2980 };
2981
2982 static int __init xfrm_user_init(void)
2983 {
2984         int rv;
2985
2986         printk(KERN_INFO "Initializing XFRM netlink socket\n");
2987
2988         rv = register_pernet_subsys(&xfrm_user_net_ops);
2989         if (rv < 0)
2990                 return rv;
2991         rv = xfrm_register_km(&netlink_mgr);
2992         if (rv < 0)
2993                 unregister_pernet_subsys(&xfrm_user_net_ops);
2994         return rv;
2995 }
2996
2997 static void __exit xfrm_user_exit(void)
2998 {
2999         xfrm_unregister_km(&netlink_mgr);
3000         unregister_pernet_subsys(&xfrm_user_net_ops);
3001 }
3002
3003 module_init(xfrm_user_init);
3004 module_exit(xfrm_user_exit);
3005 MODULE_LICENSE("GPL");
3006 MODULE_ALIAS_NET_PF_PROTO(PF_NETLINK, NETLINK_XFRM);
3007