[IPSEC]: Avoid undefined shift operation when testing algorithm ID
[linux-2.6/linux-acpi-2.6/ibm-acpi-2.6.git] / net / key / af_key.c
blob82938b9b53d1b1b3850e5dd68361e74f2374e23c
1 /*
2 * net/key/af_key.c An implementation of PF_KEYv2 sockets.
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version
7 * 2 of the License, or (at your option) any later version.
9 * Authors: Maxim Giryaev <gem@asplinux.ru>
10 * David S. Miller <davem@redhat.com>
11 * Alexey Kuznetsov <kuznet@ms2.inr.ac.ru>
12 * Kunihiro Ishiguro <kunihiro@ipinfusion.com>
13 * Kazunori MIYAZAWA / USAGI Project <miyazawa@linux-ipv6.org>
14 * Derek Atkins <derek@ihtfp.com>
17 #include <linux/config.h>
18 #include <linux/capability.h>
19 #include <linux/module.h>
20 #include <linux/kernel.h>
21 #include <linux/socket.h>
22 #include <linux/pfkeyv2.h>
23 #include <linux/ipsec.h>
24 #include <linux/skbuff.h>
25 #include <linux/rtnetlink.h>
26 #include <linux/in.h>
27 #include <linux/in6.h>
28 #include <linux/proc_fs.h>
29 #include <linux/init.h>
30 #include <net/xfrm.h>
32 #include <net/sock.h>
34 #define _X2KEY(x) ((x) == XFRM_INF ? 0 : (x))
35 #define _KEY2X(x) ((x) == 0 ? XFRM_INF : (x))
38 /* List of all pfkey sockets. */
39 static HLIST_HEAD(pfkey_table);
40 static DECLARE_WAIT_QUEUE_HEAD(pfkey_table_wait);
41 static DEFINE_RWLOCK(pfkey_table_lock);
42 static atomic_t pfkey_table_users = ATOMIC_INIT(0);
44 static atomic_t pfkey_socks_nr = ATOMIC_INIT(0);
46 struct pfkey_sock {
47 /* struct sock must be the first member of struct pfkey_sock */
48 struct sock sk;
49 int registered;
50 int promisc;
53 static inline struct pfkey_sock *pfkey_sk(struct sock *sk)
55 return (struct pfkey_sock *)sk;
58 static void pfkey_sock_destruct(struct sock *sk)
60 skb_queue_purge(&sk->sk_receive_queue);
62 if (!sock_flag(sk, SOCK_DEAD)) {
63 printk("Attempt to release alive pfkey socket: %p\n", sk);
64 return;
67 BUG_TRAP(!atomic_read(&sk->sk_rmem_alloc));
68 BUG_TRAP(!atomic_read(&sk->sk_wmem_alloc));
70 atomic_dec(&pfkey_socks_nr);
73 static void pfkey_table_grab(void)
75 write_lock_bh(&pfkey_table_lock);
77 if (atomic_read(&pfkey_table_users)) {
78 DECLARE_WAITQUEUE(wait, current);
80 add_wait_queue_exclusive(&pfkey_table_wait, &wait);
81 for(;;) {
82 set_current_state(TASK_UNINTERRUPTIBLE);
83 if (atomic_read(&pfkey_table_users) == 0)
84 break;
85 write_unlock_bh(&pfkey_table_lock);
86 schedule();
87 write_lock_bh(&pfkey_table_lock);
90 __set_current_state(TASK_RUNNING);
91 remove_wait_queue(&pfkey_table_wait, &wait);
95 static __inline__ void pfkey_table_ungrab(void)
97 write_unlock_bh(&pfkey_table_lock);
98 wake_up(&pfkey_table_wait);
101 static __inline__ void pfkey_lock_table(void)
103 /* read_lock() synchronizes us to pfkey_table_grab */
105 read_lock(&pfkey_table_lock);
106 atomic_inc(&pfkey_table_users);
107 read_unlock(&pfkey_table_lock);
110 static __inline__ void pfkey_unlock_table(void)
112 if (atomic_dec_and_test(&pfkey_table_users))
113 wake_up(&pfkey_table_wait);
117 static const struct proto_ops pfkey_ops;
119 static void pfkey_insert(struct sock *sk)
121 pfkey_table_grab();
122 sk_add_node(sk, &pfkey_table);
123 pfkey_table_ungrab();
126 static void pfkey_remove(struct sock *sk)
128 pfkey_table_grab();
129 sk_del_node_init(sk);
130 pfkey_table_ungrab();
133 static struct proto key_proto = {
134 .name = "KEY",
135 .owner = THIS_MODULE,
136 .obj_size = sizeof(struct pfkey_sock),
139 static int pfkey_create(struct socket *sock, int protocol)
141 struct sock *sk;
142 int err;
144 if (!capable(CAP_NET_ADMIN))
145 return -EPERM;
146 if (sock->type != SOCK_RAW)
147 return -ESOCKTNOSUPPORT;
148 if (protocol != PF_KEY_V2)
149 return -EPROTONOSUPPORT;
151 err = -ENOMEM;
152 sk = sk_alloc(PF_KEY, GFP_KERNEL, &key_proto, 1);
153 if (sk == NULL)
154 goto out;
156 sock->ops = &pfkey_ops;
157 sock_init_data(sock, sk);
159 sk->sk_family = PF_KEY;
160 sk->sk_destruct = pfkey_sock_destruct;
162 atomic_inc(&pfkey_socks_nr);
164 pfkey_insert(sk);
166 return 0;
167 out:
168 return err;
171 static int pfkey_release(struct socket *sock)
173 struct sock *sk = sock->sk;
175 if (!sk)
176 return 0;
178 pfkey_remove(sk);
180 sock_orphan(sk);
181 sock->sk = NULL;
182 skb_queue_purge(&sk->sk_write_queue);
183 sock_put(sk);
185 return 0;
188 static int pfkey_broadcast_one(struct sk_buff *skb, struct sk_buff **skb2,
189 gfp_t allocation, struct sock *sk)
191 int err = -ENOBUFS;
193 sock_hold(sk);
194 if (*skb2 == NULL) {
195 if (atomic_read(&skb->users) != 1) {
196 *skb2 = skb_clone(skb, allocation);
197 } else {
198 *skb2 = skb;
199 atomic_inc(&skb->users);
202 if (*skb2 != NULL) {
203 if (atomic_read(&sk->sk_rmem_alloc) <= sk->sk_rcvbuf) {
204 skb_orphan(*skb2);
205 skb_set_owner_r(*skb2, sk);
206 skb_queue_tail(&sk->sk_receive_queue, *skb2);
207 sk->sk_data_ready(sk, (*skb2)->len);
208 *skb2 = NULL;
209 err = 0;
212 sock_put(sk);
213 return err;
216 /* Send SKB to all pfkey sockets matching selected criteria. */
217 #define BROADCAST_ALL 0
218 #define BROADCAST_ONE 1
219 #define BROADCAST_REGISTERED 2
220 #define BROADCAST_PROMISC_ONLY 4
221 static int pfkey_broadcast(struct sk_buff *skb, gfp_t allocation,
222 int broadcast_flags, struct sock *one_sk)
224 struct sock *sk;
225 struct hlist_node *node;
226 struct sk_buff *skb2 = NULL;
227 int err = -ESRCH;
229 /* XXX Do we need something like netlink_overrun? I think
230 * XXX PF_KEY socket apps will not mind current behavior.
232 if (!skb)
233 return -ENOMEM;
235 pfkey_lock_table();
236 sk_for_each(sk, node, &pfkey_table) {
237 struct pfkey_sock *pfk = pfkey_sk(sk);
238 int err2;
240 /* Yes, it means that if you are meant to receive this
241 * pfkey message you receive it twice as promiscuous
242 * socket.
244 if (pfk->promisc)
245 pfkey_broadcast_one(skb, &skb2, allocation, sk);
247 /* the exact target will be processed later */
248 if (sk == one_sk)
249 continue;
250 if (broadcast_flags != BROADCAST_ALL) {
251 if (broadcast_flags & BROADCAST_PROMISC_ONLY)
252 continue;
253 if ((broadcast_flags & BROADCAST_REGISTERED) &&
254 !pfk->registered)
255 continue;
256 if (broadcast_flags & BROADCAST_ONE)
257 continue;
260 err2 = pfkey_broadcast_one(skb, &skb2, allocation, sk);
262 /* Error is cleare after succecful sending to at least one
263 * registered KM */
264 if ((broadcast_flags & BROADCAST_REGISTERED) && err)
265 err = err2;
267 pfkey_unlock_table();
269 if (one_sk != NULL)
270 err = pfkey_broadcast_one(skb, &skb2, allocation, one_sk);
272 if (skb2)
273 kfree_skb(skb2);
274 kfree_skb(skb);
275 return err;
278 static inline void pfkey_hdr_dup(struct sadb_msg *new, struct sadb_msg *orig)
280 *new = *orig;
283 static int pfkey_error(struct sadb_msg *orig, int err, struct sock *sk)
285 struct sk_buff *skb = alloc_skb(sizeof(struct sadb_msg) + 16, GFP_KERNEL);
286 struct sadb_msg *hdr;
288 if (!skb)
289 return -ENOBUFS;
291 /* Woe be to the platform trying to support PFKEY yet
292 * having normal errnos outside the 1-255 range, inclusive.
294 err = -err;
295 if (err == ERESTARTSYS ||
296 err == ERESTARTNOHAND ||
297 err == ERESTARTNOINTR)
298 err = EINTR;
299 if (err >= 512)
300 err = EINVAL;
301 BUG_ON(err <= 0 || err >= 256);
303 hdr = (struct sadb_msg *) skb_put(skb, sizeof(struct sadb_msg));
304 pfkey_hdr_dup(hdr, orig);
305 hdr->sadb_msg_errno = (uint8_t) err;
306 hdr->sadb_msg_len = (sizeof(struct sadb_msg) /
307 sizeof(uint64_t));
309 pfkey_broadcast(skb, GFP_KERNEL, BROADCAST_ONE, sk);
311 return 0;
314 static u8 sadb_ext_min_len[] = {
315 [SADB_EXT_RESERVED] = (u8) 0,
316 [SADB_EXT_SA] = (u8) sizeof(struct sadb_sa),
317 [SADB_EXT_LIFETIME_CURRENT] = (u8) sizeof(struct sadb_lifetime),
318 [SADB_EXT_LIFETIME_HARD] = (u8) sizeof(struct sadb_lifetime),
319 [SADB_EXT_LIFETIME_SOFT] = (u8) sizeof(struct sadb_lifetime),
320 [SADB_EXT_ADDRESS_SRC] = (u8) sizeof(struct sadb_address),
321 [SADB_EXT_ADDRESS_DST] = (u8) sizeof(struct sadb_address),
322 [SADB_EXT_ADDRESS_PROXY] = (u8) sizeof(struct sadb_address),
323 [SADB_EXT_KEY_AUTH] = (u8) sizeof(struct sadb_key),
324 [SADB_EXT_KEY_ENCRYPT] = (u8) sizeof(struct sadb_key),
325 [SADB_EXT_IDENTITY_SRC] = (u8) sizeof(struct sadb_ident),
326 [SADB_EXT_IDENTITY_DST] = (u8) sizeof(struct sadb_ident),
327 [SADB_EXT_SENSITIVITY] = (u8) sizeof(struct sadb_sens),
328 [SADB_EXT_PROPOSAL] = (u8) sizeof(struct sadb_prop),
329 [SADB_EXT_SUPPORTED_AUTH] = (u8) sizeof(struct sadb_supported),
330 [SADB_EXT_SUPPORTED_ENCRYPT] = (u8) sizeof(struct sadb_supported),
331 [SADB_EXT_SPIRANGE] = (u8) sizeof(struct sadb_spirange),
332 [SADB_X_EXT_KMPRIVATE] = (u8) sizeof(struct sadb_x_kmprivate),
333 [SADB_X_EXT_POLICY] = (u8) sizeof(struct sadb_x_policy),
334 [SADB_X_EXT_SA2] = (u8) sizeof(struct sadb_x_sa2),
335 [SADB_X_EXT_NAT_T_TYPE] = (u8) sizeof(struct sadb_x_nat_t_type),
336 [SADB_X_EXT_NAT_T_SPORT] = (u8) sizeof(struct sadb_x_nat_t_port),
337 [SADB_X_EXT_NAT_T_DPORT] = (u8) sizeof(struct sadb_x_nat_t_port),
338 [SADB_X_EXT_NAT_T_OA] = (u8) sizeof(struct sadb_address),
339 [SADB_X_EXT_SEC_CTX] = (u8) sizeof(struct sadb_x_sec_ctx),
342 /* Verify sadb_address_{len,prefixlen} against sa_family. */
343 static int verify_address_len(void *p)
345 struct sadb_address *sp = p;
346 struct sockaddr *addr = (struct sockaddr *)(sp + 1);
347 struct sockaddr_in *sin;
348 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
349 struct sockaddr_in6 *sin6;
350 #endif
351 int len;
353 switch (addr->sa_family) {
354 case AF_INET:
355 len = sizeof(*sp) + sizeof(*sin) + (sizeof(uint64_t) - 1);
356 len /= sizeof(uint64_t);
357 if (sp->sadb_address_len != len ||
358 sp->sadb_address_prefixlen > 32)
359 return -EINVAL;
360 break;
361 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
362 case AF_INET6:
363 len = sizeof(*sp) + sizeof(*sin6) + (sizeof(uint64_t) - 1);
364 len /= sizeof(uint64_t);
365 if (sp->sadb_address_len != len ||
366 sp->sadb_address_prefixlen > 128)
367 return -EINVAL;
368 break;
369 #endif
370 default:
371 /* It is user using kernel to keep track of security
372 * associations for another protocol, such as
373 * OSPF/RSVP/RIPV2/MIP. It is user's job to verify
374 * lengths.
376 * XXX Actually, association/policy database is not yet
377 * XXX able to cope with arbitrary sockaddr families.
378 * XXX When it can, remove this -EINVAL. -DaveM
380 return -EINVAL;
381 break;
384 return 0;
387 static inline int pfkey_sec_ctx_len(struct sadb_x_sec_ctx *sec_ctx)
389 int len = 0;
391 len += sizeof(struct sadb_x_sec_ctx);
392 len += sec_ctx->sadb_x_ctx_len;
393 len += sizeof(uint64_t) - 1;
394 len /= sizeof(uint64_t);
396 return len;
399 static inline int verify_sec_ctx_len(void *p)
401 struct sadb_x_sec_ctx *sec_ctx = (struct sadb_x_sec_ctx *)p;
402 int len;
404 if (sec_ctx->sadb_x_ctx_len > PAGE_SIZE)
405 return -EINVAL;
407 len = pfkey_sec_ctx_len(sec_ctx);
409 if (sec_ctx->sadb_x_sec_len != len)
410 return -EINVAL;
412 return 0;
415 static inline struct xfrm_user_sec_ctx *pfkey_sadb2xfrm_user_sec_ctx(struct sadb_x_sec_ctx *sec_ctx)
417 struct xfrm_user_sec_ctx *uctx = NULL;
418 int ctx_size = sec_ctx->sadb_x_ctx_len;
420 uctx = kmalloc((sizeof(*uctx)+ctx_size), GFP_KERNEL);
422 if (!uctx)
423 return NULL;
425 uctx->len = pfkey_sec_ctx_len(sec_ctx);
426 uctx->exttype = sec_ctx->sadb_x_sec_exttype;
427 uctx->ctx_doi = sec_ctx->sadb_x_ctx_doi;
428 uctx->ctx_alg = sec_ctx->sadb_x_ctx_alg;
429 uctx->ctx_len = sec_ctx->sadb_x_ctx_len;
430 memcpy(uctx + 1, sec_ctx + 1,
431 uctx->ctx_len);
433 return uctx;
436 static int present_and_same_family(struct sadb_address *src,
437 struct sadb_address *dst)
439 struct sockaddr *s_addr, *d_addr;
441 if (!src || !dst)
442 return 0;
444 s_addr = (struct sockaddr *)(src + 1);
445 d_addr = (struct sockaddr *)(dst + 1);
446 if (s_addr->sa_family != d_addr->sa_family)
447 return 0;
448 if (s_addr->sa_family != AF_INET
449 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
450 && s_addr->sa_family != AF_INET6
451 #endif
453 return 0;
455 return 1;
458 static int parse_exthdrs(struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
460 char *p = (char *) hdr;
461 int len = skb->len;
463 len -= sizeof(*hdr);
464 p += sizeof(*hdr);
465 while (len > 0) {
466 struct sadb_ext *ehdr = (struct sadb_ext *) p;
467 uint16_t ext_type;
468 int ext_len;
470 ext_len = ehdr->sadb_ext_len;
471 ext_len *= sizeof(uint64_t);
472 ext_type = ehdr->sadb_ext_type;
473 if (ext_len < sizeof(uint64_t) ||
474 ext_len > len ||
475 ext_type == SADB_EXT_RESERVED)
476 return -EINVAL;
478 if (ext_type <= SADB_EXT_MAX) {
479 int min = (int) sadb_ext_min_len[ext_type];
480 if (ext_len < min)
481 return -EINVAL;
482 if (ext_hdrs[ext_type-1] != NULL)
483 return -EINVAL;
484 if (ext_type == SADB_EXT_ADDRESS_SRC ||
485 ext_type == SADB_EXT_ADDRESS_DST ||
486 ext_type == SADB_EXT_ADDRESS_PROXY ||
487 ext_type == SADB_X_EXT_NAT_T_OA) {
488 if (verify_address_len(p))
489 return -EINVAL;
491 if (ext_type == SADB_X_EXT_SEC_CTX) {
492 if (verify_sec_ctx_len(p))
493 return -EINVAL;
495 ext_hdrs[ext_type-1] = p;
497 p += ext_len;
498 len -= ext_len;
501 return 0;
504 static uint16_t
505 pfkey_satype2proto(uint8_t satype)
507 switch (satype) {
508 case SADB_SATYPE_UNSPEC:
509 return IPSEC_PROTO_ANY;
510 case SADB_SATYPE_AH:
511 return IPPROTO_AH;
512 case SADB_SATYPE_ESP:
513 return IPPROTO_ESP;
514 case SADB_X_SATYPE_IPCOMP:
515 return IPPROTO_COMP;
516 break;
517 default:
518 return 0;
520 /* NOTREACHED */
523 static uint8_t
524 pfkey_proto2satype(uint16_t proto)
526 switch (proto) {
527 case IPPROTO_AH:
528 return SADB_SATYPE_AH;
529 case IPPROTO_ESP:
530 return SADB_SATYPE_ESP;
531 case IPPROTO_COMP:
532 return SADB_X_SATYPE_IPCOMP;
533 break;
534 default:
535 return 0;
537 /* NOTREACHED */
540 /* BTW, this scheme means that there is no way with PFKEY2 sockets to
541 * say specifically 'just raw sockets' as we encode them as 255.
544 static uint8_t pfkey_proto_to_xfrm(uint8_t proto)
546 return (proto == IPSEC_PROTO_ANY ? 0 : proto);
549 static uint8_t pfkey_proto_from_xfrm(uint8_t proto)
551 return (proto ? proto : IPSEC_PROTO_ANY);
554 static int pfkey_sadb_addr2xfrm_addr(struct sadb_address *addr,
555 xfrm_address_t *xaddr)
557 switch (((struct sockaddr*)(addr + 1))->sa_family) {
558 case AF_INET:
559 xaddr->a4 =
560 ((struct sockaddr_in *)(addr + 1))->sin_addr.s_addr;
561 return AF_INET;
562 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
563 case AF_INET6:
564 memcpy(xaddr->a6,
565 &((struct sockaddr_in6 *)(addr + 1))->sin6_addr,
566 sizeof(struct in6_addr));
567 return AF_INET6;
568 #endif
569 default:
570 return 0;
572 /* NOTREACHED */
575 static struct xfrm_state *pfkey_xfrm_state_lookup(struct sadb_msg *hdr, void **ext_hdrs)
577 struct sadb_sa *sa;
578 struct sadb_address *addr;
579 uint16_t proto;
580 unsigned short family;
581 xfrm_address_t *xaddr;
583 sa = (struct sadb_sa *) ext_hdrs[SADB_EXT_SA-1];
584 if (sa == NULL)
585 return NULL;
587 proto = pfkey_satype2proto(hdr->sadb_msg_satype);
588 if (proto == 0)
589 return NULL;
591 /* sadb_address_len should be checked by caller */
592 addr = (struct sadb_address *) ext_hdrs[SADB_EXT_ADDRESS_DST-1];
593 if (addr == NULL)
594 return NULL;
596 family = ((struct sockaddr *)(addr + 1))->sa_family;
597 switch (family) {
598 case AF_INET:
599 xaddr = (xfrm_address_t *)&((struct sockaddr_in *)(addr + 1))->sin_addr;
600 break;
601 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
602 case AF_INET6:
603 xaddr = (xfrm_address_t *)&((struct sockaddr_in6 *)(addr + 1))->sin6_addr;
604 break;
605 #endif
606 default:
607 xaddr = NULL;
610 if (!xaddr)
611 return NULL;
613 return xfrm_state_lookup(xaddr, sa->sadb_sa_spi, proto, family);
616 #define PFKEY_ALIGN8(a) (1 + (((a) - 1) | (8 - 1)))
617 static int
618 pfkey_sockaddr_size(sa_family_t family)
620 switch (family) {
621 case AF_INET:
622 return PFKEY_ALIGN8(sizeof(struct sockaddr_in));
623 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
624 case AF_INET6:
625 return PFKEY_ALIGN8(sizeof(struct sockaddr_in6));
626 #endif
627 default:
628 return 0;
630 /* NOTREACHED */
633 static struct sk_buff * pfkey_xfrm_state2msg(struct xfrm_state *x, int add_keys, int hsc)
635 struct sk_buff *skb;
636 struct sadb_msg *hdr;
637 struct sadb_sa *sa;
638 struct sadb_lifetime *lifetime;
639 struct sadb_address *addr;
640 struct sadb_key *key;
641 struct sadb_x_sa2 *sa2;
642 struct sockaddr_in *sin;
643 struct sadb_x_sec_ctx *sec_ctx;
644 struct xfrm_sec_ctx *xfrm_ctx;
645 int ctx_size = 0;
646 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
647 struct sockaddr_in6 *sin6;
648 #endif
649 int size;
650 int auth_key_size = 0;
651 int encrypt_key_size = 0;
652 int sockaddr_size;
653 struct xfrm_encap_tmpl *natt = NULL;
655 /* address family check */
656 sockaddr_size = pfkey_sockaddr_size(x->props.family);
657 if (!sockaddr_size)
658 return ERR_PTR(-EINVAL);
660 /* base, SA, (lifetime (HSC),) address(SD), (address(P),)
661 key(AE), (identity(SD),) (sensitivity)> */
662 size = sizeof(struct sadb_msg) +sizeof(struct sadb_sa) +
663 sizeof(struct sadb_lifetime) +
664 ((hsc & 1) ? sizeof(struct sadb_lifetime) : 0) +
665 ((hsc & 2) ? sizeof(struct sadb_lifetime) : 0) +
666 sizeof(struct sadb_address)*2 +
667 sockaddr_size*2 +
668 sizeof(struct sadb_x_sa2);
670 if ((xfrm_ctx = x->security)) {
671 ctx_size = PFKEY_ALIGN8(xfrm_ctx->ctx_len);
672 size += sizeof(struct sadb_x_sec_ctx) + ctx_size;
675 /* identity & sensitivity */
677 if ((x->props.family == AF_INET &&
678 x->sel.saddr.a4 != x->props.saddr.a4)
679 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
680 || (x->props.family == AF_INET6 &&
681 memcmp (x->sel.saddr.a6, x->props.saddr.a6, sizeof (struct in6_addr)))
682 #endif
684 size += sizeof(struct sadb_address) + sockaddr_size;
686 if (add_keys) {
687 if (x->aalg && x->aalg->alg_key_len) {
688 auth_key_size =
689 PFKEY_ALIGN8((x->aalg->alg_key_len + 7) / 8);
690 size += sizeof(struct sadb_key) + auth_key_size;
692 if (x->ealg && x->ealg->alg_key_len) {
693 encrypt_key_size =
694 PFKEY_ALIGN8((x->ealg->alg_key_len+7) / 8);
695 size += sizeof(struct sadb_key) + encrypt_key_size;
698 if (x->encap)
699 natt = x->encap;
701 if (natt && natt->encap_type) {
702 size += sizeof(struct sadb_x_nat_t_type);
703 size += sizeof(struct sadb_x_nat_t_port);
704 size += sizeof(struct sadb_x_nat_t_port);
707 skb = alloc_skb(size + 16, GFP_ATOMIC);
708 if (skb == NULL)
709 return ERR_PTR(-ENOBUFS);
711 /* call should fill header later */
712 hdr = (struct sadb_msg *) skb_put(skb, sizeof(struct sadb_msg));
713 memset(hdr, 0, size); /* XXX do we need this ? */
714 hdr->sadb_msg_len = size / sizeof(uint64_t);
716 /* sa */
717 sa = (struct sadb_sa *) skb_put(skb, sizeof(struct sadb_sa));
718 sa->sadb_sa_len = sizeof(struct sadb_sa)/sizeof(uint64_t);
719 sa->sadb_sa_exttype = SADB_EXT_SA;
720 sa->sadb_sa_spi = x->id.spi;
721 sa->sadb_sa_replay = x->props.replay_window;
722 switch (x->km.state) {
723 case XFRM_STATE_VALID:
724 sa->sadb_sa_state = x->km.dying ?
725 SADB_SASTATE_DYING : SADB_SASTATE_MATURE;
726 break;
727 case XFRM_STATE_ACQ:
728 sa->sadb_sa_state = SADB_SASTATE_LARVAL;
729 break;
730 default:
731 sa->sadb_sa_state = SADB_SASTATE_DEAD;
732 break;
734 sa->sadb_sa_auth = 0;
735 if (x->aalg) {
736 struct xfrm_algo_desc *a = xfrm_aalg_get_byname(x->aalg->alg_name, 0);
737 sa->sadb_sa_auth = a ? a->desc.sadb_alg_id : 0;
739 sa->sadb_sa_encrypt = 0;
740 BUG_ON(x->ealg && x->calg);
741 if (x->ealg) {
742 struct xfrm_algo_desc *a = xfrm_ealg_get_byname(x->ealg->alg_name, 0);
743 sa->sadb_sa_encrypt = a ? a->desc.sadb_alg_id : 0;
745 /* KAME compatible: sadb_sa_encrypt is overloaded with calg id */
746 if (x->calg) {
747 struct xfrm_algo_desc *a = xfrm_calg_get_byname(x->calg->alg_name, 0);
748 sa->sadb_sa_encrypt = a ? a->desc.sadb_alg_id : 0;
751 sa->sadb_sa_flags = 0;
752 if (x->props.flags & XFRM_STATE_NOECN)
753 sa->sadb_sa_flags |= SADB_SAFLAGS_NOECN;
754 if (x->props.flags & XFRM_STATE_DECAP_DSCP)
755 sa->sadb_sa_flags |= SADB_SAFLAGS_DECAP_DSCP;
756 if (x->props.flags & XFRM_STATE_NOPMTUDISC)
757 sa->sadb_sa_flags |= SADB_SAFLAGS_NOPMTUDISC;
759 /* hard time */
760 if (hsc & 2) {
761 lifetime = (struct sadb_lifetime *) skb_put(skb,
762 sizeof(struct sadb_lifetime));
763 lifetime->sadb_lifetime_len =
764 sizeof(struct sadb_lifetime)/sizeof(uint64_t);
765 lifetime->sadb_lifetime_exttype = SADB_EXT_LIFETIME_HARD;
766 lifetime->sadb_lifetime_allocations = _X2KEY(x->lft.hard_packet_limit);
767 lifetime->sadb_lifetime_bytes = _X2KEY(x->lft.hard_byte_limit);
768 lifetime->sadb_lifetime_addtime = x->lft.hard_add_expires_seconds;
769 lifetime->sadb_lifetime_usetime = x->lft.hard_use_expires_seconds;
771 /* soft time */
772 if (hsc & 1) {
773 lifetime = (struct sadb_lifetime *) skb_put(skb,
774 sizeof(struct sadb_lifetime));
775 lifetime->sadb_lifetime_len =
776 sizeof(struct sadb_lifetime)/sizeof(uint64_t);
777 lifetime->sadb_lifetime_exttype = SADB_EXT_LIFETIME_SOFT;
778 lifetime->sadb_lifetime_allocations = _X2KEY(x->lft.soft_packet_limit);
779 lifetime->sadb_lifetime_bytes = _X2KEY(x->lft.soft_byte_limit);
780 lifetime->sadb_lifetime_addtime = x->lft.soft_add_expires_seconds;
781 lifetime->sadb_lifetime_usetime = x->lft.soft_use_expires_seconds;
783 /* current time */
784 lifetime = (struct sadb_lifetime *) skb_put(skb,
785 sizeof(struct sadb_lifetime));
786 lifetime->sadb_lifetime_len =
787 sizeof(struct sadb_lifetime)/sizeof(uint64_t);
788 lifetime->sadb_lifetime_exttype = SADB_EXT_LIFETIME_CURRENT;
789 lifetime->sadb_lifetime_allocations = x->curlft.packets;
790 lifetime->sadb_lifetime_bytes = x->curlft.bytes;
791 lifetime->sadb_lifetime_addtime = x->curlft.add_time;
792 lifetime->sadb_lifetime_usetime = x->curlft.use_time;
793 /* src address */
794 addr = (struct sadb_address*) skb_put(skb,
795 sizeof(struct sadb_address)+sockaddr_size);
796 addr->sadb_address_len =
797 (sizeof(struct sadb_address)+sockaddr_size)/
798 sizeof(uint64_t);
799 addr->sadb_address_exttype = SADB_EXT_ADDRESS_SRC;
800 /* "if the ports are non-zero, then the sadb_address_proto field,
801 normally zero, MUST be filled in with the transport
802 protocol's number." - RFC2367 */
803 addr->sadb_address_proto = 0;
804 addr->sadb_address_reserved = 0;
805 if (x->props.family == AF_INET) {
806 addr->sadb_address_prefixlen = 32;
808 sin = (struct sockaddr_in *) (addr + 1);
809 sin->sin_family = AF_INET;
810 sin->sin_addr.s_addr = x->props.saddr.a4;
811 sin->sin_port = 0;
812 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
814 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
815 else if (x->props.family == AF_INET6) {
816 addr->sadb_address_prefixlen = 128;
818 sin6 = (struct sockaddr_in6 *) (addr + 1);
819 sin6->sin6_family = AF_INET6;
820 sin6->sin6_port = 0;
821 sin6->sin6_flowinfo = 0;
822 memcpy(&sin6->sin6_addr, x->props.saddr.a6,
823 sizeof(struct in6_addr));
824 sin6->sin6_scope_id = 0;
826 #endif
827 else
828 BUG();
830 /* dst address */
831 addr = (struct sadb_address*) skb_put(skb,
832 sizeof(struct sadb_address)+sockaddr_size);
833 addr->sadb_address_len =
834 (sizeof(struct sadb_address)+sockaddr_size)/
835 sizeof(uint64_t);
836 addr->sadb_address_exttype = SADB_EXT_ADDRESS_DST;
837 addr->sadb_address_proto = 0;
838 addr->sadb_address_prefixlen = 32; /* XXX */
839 addr->sadb_address_reserved = 0;
840 if (x->props.family == AF_INET) {
841 sin = (struct sockaddr_in *) (addr + 1);
842 sin->sin_family = AF_INET;
843 sin->sin_addr.s_addr = x->id.daddr.a4;
844 sin->sin_port = 0;
845 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
847 if (x->sel.saddr.a4 != x->props.saddr.a4) {
848 addr = (struct sadb_address*) skb_put(skb,
849 sizeof(struct sadb_address)+sockaddr_size);
850 addr->sadb_address_len =
851 (sizeof(struct sadb_address)+sockaddr_size)/
852 sizeof(uint64_t);
853 addr->sadb_address_exttype = SADB_EXT_ADDRESS_PROXY;
854 addr->sadb_address_proto =
855 pfkey_proto_from_xfrm(x->sel.proto);
856 addr->sadb_address_prefixlen = x->sel.prefixlen_s;
857 addr->sadb_address_reserved = 0;
859 sin = (struct sockaddr_in *) (addr + 1);
860 sin->sin_family = AF_INET;
861 sin->sin_addr.s_addr = x->sel.saddr.a4;
862 sin->sin_port = x->sel.sport;
863 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
866 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
867 else if (x->props.family == AF_INET6) {
868 addr->sadb_address_prefixlen = 128;
870 sin6 = (struct sockaddr_in6 *) (addr + 1);
871 sin6->sin6_family = AF_INET6;
872 sin6->sin6_port = 0;
873 sin6->sin6_flowinfo = 0;
874 memcpy(&sin6->sin6_addr, x->id.daddr.a6, sizeof(struct in6_addr));
875 sin6->sin6_scope_id = 0;
877 if (memcmp (x->sel.saddr.a6, x->props.saddr.a6,
878 sizeof(struct in6_addr))) {
879 addr = (struct sadb_address *) skb_put(skb,
880 sizeof(struct sadb_address)+sockaddr_size);
881 addr->sadb_address_len =
882 (sizeof(struct sadb_address)+sockaddr_size)/
883 sizeof(uint64_t);
884 addr->sadb_address_exttype = SADB_EXT_ADDRESS_PROXY;
885 addr->sadb_address_proto =
886 pfkey_proto_from_xfrm(x->sel.proto);
887 addr->sadb_address_prefixlen = x->sel.prefixlen_s;
888 addr->sadb_address_reserved = 0;
890 sin6 = (struct sockaddr_in6 *) (addr + 1);
891 sin6->sin6_family = AF_INET6;
892 sin6->sin6_port = x->sel.sport;
893 sin6->sin6_flowinfo = 0;
894 memcpy(&sin6->sin6_addr, x->sel.saddr.a6,
895 sizeof(struct in6_addr));
896 sin6->sin6_scope_id = 0;
899 #endif
900 else
901 BUG();
903 /* auth key */
904 if (add_keys && auth_key_size) {
905 key = (struct sadb_key *) skb_put(skb,
906 sizeof(struct sadb_key)+auth_key_size);
907 key->sadb_key_len = (sizeof(struct sadb_key) + auth_key_size) /
908 sizeof(uint64_t);
909 key->sadb_key_exttype = SADB_EXT_KEY_AUTH;
910 key->sadb_key_bits = x->aalg->alg_key_len;
911 key->sadb_key_reserved = 0;
912 memcpy(key + 1, x->aalg->alg_key, (x->aalg->alg_key_len+7)/8);
914 /* encrypt key */
915 if (add_keys && encrypt_key_size) {
916 key = (struct sadb_key *) skb_put(skb,
917 sizeof(struct sadb_key)+encrypt_key_size);
918 key->sadb_key_len = (sizeof(struct sadb_key) +
919 encrypt_key_size) / sizeof(uint64_t);
920 key->sadb_key_exttype = SADB_EXT_KEY_ENCRYPT;
921 key->sadb_key_bits = x->ealg->alg_key_len;
922 key->sadb_key_reserved = 0;
923 memcpy(key + 1, x->ealg->alg_key,
924 (x->ealg->alg_key_len+7)/8);
927 /* sa */
928 sa2 = (struct sadb_x_sa2 *) skb_put(skb, sizeof(struct sadb_x_sa2));
929 sa2->sadb_x_sa2_len = sizeof(struct sadb_x_sa2)/sizeof(uint64_t);
930 sa2->sadb_x_sa2_exttype = SADB_X_EXT_SA2;
931 sa2->sadb_x_sa2_mode = x->props.mode + 1;
932 sa2->sadb_x_sa2_reserved1 = 0;
933 sa2->sadb_x_sa2_reserved2 = 0;
934 sa2->sadb_x_sa2_sequence = 0;
935 sa2->sadb_x_sa2_reqid = x->props.reqid;
937 if (natt && natt->encap_type) {
938 struct sadb_x_nat_t_type *n_type;
939 struct sadb_x_nat_t_port *n_port;
941 /* type */
942 n_type = (struct sadb_x_nat_t_type*) skb_put(skb, sizeof(*n_type));
943 n_type->sadb_x_nat_t_type_len = sizeof(*n_type)/sizeof(uint64_t);
944 n_type->sadb_x_nat_t_type_exttype = SADB_X_EXT_NAT_T_TYPE;
945 n_type->sadb_x_nat_t_type_type = natt->encap_type;
946 n_type->sadb_x_nat_t_type_reserved[0] = 0;
947 n_type->sadb_x_nat_t_type_reserved[1] = 0;
948 n_type->sadb_x_nat_t_type_reserved[2] = 0;
950 /* source port */
951 n_port = (struct sadb_x_nat_t_port*) skb_put(skb, sizeof (*n_port));
952 n_port->sadb_x_nat_t_port_len = sizeof(*n_port)/sizeof(uint64_t);
953 n_port->sadb_x_nat_t_port_exttype = SADB_X_EXT_NAT_T_SPORT;
954 n_port->sadb_x_nat_t_port_port = natt->encap_sport;
955 n_port->sadb_x_nat_t_port_reserved = 0;
957 /* dest port */
958 n_port = (struct sadb_x_nat_t_port*) skb_put(skb, sizeof (*n_port));
959 n_port->sadb_x_nat_t_port_len = sizeof(*n_port)/sizeof(uint64_t);
960 n_port->sadb_x_nat_t_port_exttype = SADB_X_EXT_NAT_T_DPORT;
961 n_port->sadb_x_nat_t_port_port = natt->encap_dport;
962 n_port->sadb_x_nat_t_port_reserved = 0;
965 /* security context */
966 if (xfrm_ctx) {
967 sec_ctx = (struct sadb_x_sec_ctx *) skb_put(skb,
968 sizeof(struct sadb_x_sec_ctx) + ctx_size);
969 sec_ctx->sadb_x_sec_len =
970 (sizeof(struct sadb_x_sec_ctx) + ctx_size) / sizeof(uint64_t);
971 sec_ctx->sadb_x_sec_exttype = SADB_X_EXT_SEC_CTX;
972 sec_ctx->sadb_x_ctx_doi = xfrm_ctx->ctx_doi;
973 sec_ctx->sadb_x_ctx_alg = xfrm_ctx->ctx_alg;
974 sec_ctx->sadb_x_ctx_len = xfrm_ctx->ctx_len;
975 memcpy(sec_ctx + 1, xfrm_ctx->ctx_str,
976 xfrm_ctx->ctx_len);
979 return skb;
982 static struct xfrm_state * pfkey_msg2xfrm_state(struct sadb_msg *hdr,
983 void **ext_hdrs)
985 struct xfrm_state *x;
986 struct sadb_lifetime *lifetime;
987 struct sadb_sa *sa;
988 struct sadb_key *key;
989 struct sadb_x_sec_ctx *sec_ctx;
990 uint16_t proto;
991 int err;
994 sa = (struct sadb_sa *) ext_hdrs[SADB_EXT_SA-1];
995 if (!sa ||
996 !present_and_same_family(ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
997 ext_hdrs[SADB_EXT_ADDRESS_DST-1]))
998 return ERR_PTR(-EINVAL);
999 if (hdr->sadb_msg_satype == SADB_SATYPE_ESP &&
1000 !ext_hdrs[SADB_EXT_KEY_ENCRYPT-1])
1001 return ERR_PTR(-EINVAL);
1002 if (hdr->sadb_msg_satype == SADB_SATYPE_AH &&
1003 !ext_hdrs[SADB_EXT_KEY_AUTH-1])
1004 return ERR_PTR(-EINVAL);
1005 if (!!ext_hdrs[SADB_EXT_LIFETIME_HARD-1] !=
1006 !!ext_hdrs[SADB_EXT_LIFETIME_SOFT-1])
1007 return ERR_PTR(-EINVAL);
1009 proto = pfkey_satype2proto(hdr->sadb_msg_satype);
1010 if (proto == 0)
1011 return ERR_PTR(-EINVAL);
1013 /* default error is no buffer space */
1014 err = -ENOBUFS;
1016 /* RFC2367:
1018 Only SADB_SASTATE_MATURE SAs may be submitted in an SADB_ADD message.
1019 SADB_SASTATE_LARVAL SAs are created by SADB_GETSPI and it is not
1020 sensible to add a new SA in the DYING or SADB_SASTATE_DEAD state.
1021 Therefore, the sadb_sa_state field of all submitted SAs MUST be
1022 SADB_SASTATE_MATURE and the kernel MUST return an error if this is
1023 not true.
1025 However, KAME setkey always uses SADB_SASTATE_LARVAL.
1026 Hence, we have to _ignore_ sadb_sa_state, which is also reasonable.
1028 if (sa->sadb_sa_auth > SADB_AALG_MAX ||
1029 (hdr->sadb_msg_satype == SADB_X_SATYPE_IPCOMP &&
1030 sa->sadb_sa_encrypt > SADB_X_CALG_MAX) ||
1031 sa->sadb_sa_encrypt > SADB_EALG_MAX)
1032 return ERR_PTR(-EINVAL);
1033 key = (struct sadb_key*) ext_hdrs[SADB_EXT_KEY_AUTH-1];
1034 if (key != NULL &&
1035 sa->sadb_sa_auth != SADB_X_AALG_NULL &&
1036 ((key->sadb_key_bits+7) / 8 == 0 ||
1037 (key->sadb_key_bits+7) / 8 > key->sadb_key_len * sizeof(uint64_t)))
1038 return ERR_PTR(-EINVAL);
1039 key = ext_hdrs[SADB_EXT_KEY_ENCRYPT-1];
1040 if (key != NULL &&
1041 sa->sadb_sa_encrypt != SADB_EALG_NULL &&
1042 ((key->sadb_key_bits+7) / 8 == 0 ||
1043 (key->sadb_key_bits+7) / 8 > key->sadb_key_len * sizeof(uint64_t)))
1044 return ERR_PTR(-EINVAL);
1046 x = xfrm_state_alloc();
1047 if (x == NULL)
1048 return ERR_PTR(-ENOBUFS);
1050 x->id.proto = proto;
1051 x->id.spi = sa->sadb_sa_spi;
1052 x->props.replay_window = sa->sadb_sa_replay;
1053 if (sa->sadb_sa_flags & SADB_SAFLAGS_NOECN)
1054 x->props.flags |= XFRM_STATE_NOECN;
1055 if (sa->sadb_sa_flags & SADB_SAFLAGS_DECAP_DSCP)
1056 x->props.flags |= XFRM_STATE_DECAP_DSCP;
1057 if (sa->sadb_sa_flags & SADB_SAFLAGS_NOPMTUDISC)
1058 x->props.flags |= XFRM_STATE_NOPMTUDISC;
1060 lifetime = (struct sadb_lifetime*) ext_hdrs[SADB_EXT_LIFETIME_HARD-1];
1061 if (lifetime != NULL) {
1062 x->lft.hard_packet_limit = _KEY2X(lifetime->sadb_lifetime_allocations);
1063 x->lft.hard_byte_limit = _KEY2X(lifetime->sadb_lifetime_bytes);
1064 x->lft.hard_add_expires_seconds = lifetime->sadb_lifetime_addtime;
1065 x->lft.hard_use_expires_seconds = lifetime->sadb_lifetime_usetime;
1067 lifetime = (struct sadb_lifetime*) ext_hdrs[SADB_EXT_LIFETIME_SOFT-1];
1068 if (lifetime != NULL) {
1069 x->lft.soft_packet_limit = _KEY2X(lifetime->sadb_lifetime_allocations);
1070 x->lft.soft_byte_limit = _KEY2X(lifetime->sadb_lifetime_bytes);
1071 x->lft.soft_add_expires_seconds = lifetime->sadb_lifetime_addtime;
1072 x->lft.soft_use_expires_seconds = lifetime->sadb_lifetime_usetime;
1075 sec_ctx = (struct sadb_x_sec_ctx *) ext_hdrs[SADB_X_EXT_SEC_CTX-1];
1076 if (sec_ctx != NULL) {
1077 struct xfrm_user_sec_ctx *uctx = pfkey_sadb2xfrm_user_sec_ctx(sec_ctx);
1079 if (!uctx)
1080 goto out;
1082 err = security_xfrm_state_alloc(x, uctx);
1083 kfree(uctx);
1085 if (err)
1086 goto out;
1089 key = (struct sadb_key*) ext_hdrs[SADB_EXT_KEY_AUTH-1];
1090 if (sa->sadb_sa_auth) {
1091 int keysize = 0;
1092 struct xfrm_algo_desc *a = xfrm_aalg_get_byid(sa->sadb_sa_auth);
1093 if (!a) {
1094 err = -ENOSYS;
1095 goto out;
1097 if (key)
1098 keysize = (key->sadb_key_bits + 7) / 8;
1099 x->aalg = kmalloc(sizeof(*x->aalg) + keysize, GFP_KERNEL);
1100 if (!x->aalg)
1101 goto out;
1102 strcpy(x->aalg->alg_name, a->name);
1103 x->aalg->alg_key_len = 0;
1104 if (key) {
1105 x->aalg->alg_key_len = key->sadb_key_bits;
1106 memcpy(x->aalg->alg_key, key+1, keysize);
1108 x->props.aalgo = sa->sadb_sa_auth;
1109 /* x->algo.flags = sa->sadb_sa_flags; */
1111 if (sa->sadb_sa_encrypt) {
1112 if (hdr->sadb_msg_satype == SADB_X_SATYPE_IPCOMP) {
1113 struct xfrm_algo_desc *a = xfrm_calg_get_byid(sa->sadb_sa_encrypt);
1114 if (!a) {
1115 err = -ENOSYS;
1116 goto out;
1118 x->calg = kmalloc(sizeof(*x->calg), GFP_KERNEL);
1119 if (!x->calg)
1120 goto out;
1121 strcpy(x->calg->alg_name, a->name);
1122 x->props.calgo = sa->sadb_sa_encrypt;
1123 } else {
1124 int keysize = 0;
1125 struct xfrm_algo_desc *a = xfrm_ealg_get_byid(sa->sadb_sa_encrypt);
1126 if (!a) {
1127 err = -ENOSYS;
1128 goto out;
1130 key = (struct sadb_key*) ext_hdrs[SADB_EXT_KEY_ENCRYPT-1];
1131 if (key)
1132 keysize = (key->sadb_key_bits + 7) / 8;
1133 x->ealg = kmalloc(sizeof(*x->ealg) + keysize, GFP_KERNEL);
1134 if (!x->ealg)
1135 goto out;
1136 strcpy(x->ealg->alg_name, a->name);
1137 x->ealg->alg_key_len = 0;
1138 if (key) {
1139 x->ealg->alg_key_len = key->sadb_key_bits;
1140 memcpy(x->ealg->alg_key, key+1, keysize);
1142 x->props.ealgo = sa->sadb_sa_encrypt;
1145 /* x->algo.flags = sa->sadb_sa_flags; */
1147 x->props.family = pfkey_sadb_addr2xfrm_addr((struct sadb_address *) ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
1148 &x->props.saddr);
1149 if (!x->props.family) {
1150 err = -EAFNOSUPPORT;
1151 goto out;
1153 pfkey_sadb_addr2xfrm_addr((struct sadb_address *) ext_hdrs[SADB_EXT_ADDRESS_DST-1],
1154 &x->id.daddr);
1156 if (ext_hdrs[SADB_X_EXT_SA2-1]) {
1157 struct sadb_x_sa2 *sa2 = (void*)ext_hdrs[SADB_X_EXT_SA2-1];
1158 x->props.mode = sa2->sadb_x_sa2_mode;
1159 if (x->props.mode)
1160 x->props.mode--;
1161 x->props.reqid = sa2->sadb_x_sa2_reqid;
1164 if (ext_hdrs[SADB_EXT_ADDRESS_PROXY-1]) {
1165 struct sadb_address *addr = ext_hdrs[SADB_EXT_ADDRESS_PROXY-1];
1167 /* Nobody uses this, but we try. */
1168 x->sel.family = pfkey_sadb_addr2xfrm_addr(addr, &x->sel.saddr);
1169 x->sel.prefixlen_s = addr->sadb_address_prefixlen;
1172 if (ext_hdrs[SADB_X_EXT_NAT_T_TYPE-1]) {
1173 struct sadb_x_nat_t_type* n_type;
1174 struct xfrm_encap_tmpl *natt;
1176 x->encap = kmalloc(sizeof(*x->encap), GFP_KERNEL);
1177 if (!x->encap)
1178 goto out;
1180 natt = x->encap;
1181 n_type = ext_hdrs[SADB_X_EXT_NAT_T_TYPE-1];
1182 natt->encap_type = n_type->sadb_x_nat_t_type_type;
1184 if (ext_hdrs[SADB_X_EXT_NAT_T_SPORT-1]) {
1185 struct sadb_x_nat_t_port* n_port =
1186 ext_hdrs[SADB_X_EXT_NAT_T_SPORT-1];
1187 natt->encap_sport = n_port->sadb_x_nat_t_port_port;
1189 if (ext_hdrs[SADB_X_EXT_NAT_T_DPORT-1]) {
1190 struct sadb_x_nat_t_port* n_port =
1191 ext_hdrs[SADB_X_EXT_NAT_T_DPORT-1];
1192 natt->encap_dport = n_port->sadb_x_nat_t_port_port;
1196 err = xfrm_init_state(x);
1197 if (err)
1198 goto out;
1200 x->km.seq = hdr->sadb_msg_seq;
1201 return x;
1203 out:
1204 x->km.state = XFRM_STATE_DEAD;
1205 xfrm_state_put(x);
1206 return ERR_PTR(err);
1209 static int pfkey_reserved(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1211 return -EOPNOTSUPP;
1214 static int pfkey_getspi(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1216 struct sk_buff *resp_skb;
1217 struct sadb_x_sa2 *sa2;
1218 struct sadb_address *saddr, *daddr;
1219 struct sadb_msg *out_hdr;
1220 struct xfrm_state *x = NULL;
1221 u8 mode;
1222 u32 reqid;
1223 u8 proto;
1224 unsigned short family;
1225 xfrm_address_t *xsaddr = NULL, *xdaddr = NULL;
1227 if (!present_and_same_family(ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
1228 ext_hdrs[SADB_EXT_ADDRESS_DST-1]))
1229 return -EINVAL;
1231 proto = pfkey_satype2proto(hdr->sadb_msg_satype);
1232 if (proto == 0)
1233 return -EINVAL;
1235 if ((sa2 = ext_hdrs[SADB_X_EXT_SA2-1]) != NULL) {
1236 mode = sa2->sadb_x_sa2_mode - 1;
1237 reqid = sa2->sadb_x_sa2_reqid;
1238 } else {
1239 mode = 0;
1240 reqid = 0;
1243 saddr = ext_hdrs[SADB_EXT_ADDRESS_SRC-1];
1244 daddr = ext_hdrs[SADB_EXT_ADDRESS_DST-1];
1246 family = ((struct sockaddr *)(saddr + 1))->sa_family;
1247 switch (family) {
1248 case AF_INET:
1249 xdaddr = (xfrm_address_t *)&((struct sockaddr_in *)(daddr + 1))->sin_addr.s_addr;
1250 xsaddr = (xfrm_address_t *)&((struct sockaddr_in *)(saddr + 1))->sin_addr.s_addr;
1251 break;
1252 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1253 case AF_INET6:
1254 xdaddr = (xfrm_address_t *)&((struct sockaddr_in6 *)(daddr + 1))->sin6_addr;
1255 xsaddr = (xfrm_address_t *)&((struct sockaddr_in6 *)(saddr + 1))->sin6_addr;
1256 break;
1257 #endif
1260 if (hdr->sadb_msg_seq) {
1261 x = xfrm_find_acq_byseq(hdr->sadb_msg_seq);
1262 if (x && xfrm_addr_cmp(&x->id.daddr, xdaddr, family)) {
1263 xfrm_state_put(x);
1264 x = NULL;
1268 if (!x)
1269 x = xfrm_find_acq(mode, reqid, proto, xdaddr, xsaddr, 1, family);
1271 if (x == NULL)
1272 return -ENOENT;
1274 resp_skb = ERR_PTR(-ENOENT);
1276 spin_lock_bh(&x->lock);
1277 if (x->km.state != XFRM_STATE_DEAD) {
1278 struct sadb_spirange *range = ext_hdrs[SADB_EXT_SPIRANGE-1];
1279 u32 min_spi, max_spi;
1281 if (range != NULL) {
1282 min_spi = range->sadb_spirange_min;
1283 max_spi = range->sadb_spirange_max;
1284 } else {
1285 min_spi = 0x100;
1286 max_spi = 0x0fffffff;
1288 xfrm_alloc_spi(x, htonl(min_spi), htonl(max_spi));
1289 if (x->id.spi)
1290 resp_skb = pfkey_xfrm_state2msg(x, 0, 3);
1292 spin_unlock_bh(&x->lock);
1294 if (IS_ERR(resp_skb)) {
1295 xfrm_state_put(x);
1296 return PTR_ERR(resp_skb);
1299 out_hdr = (struct sadb_msg *) resp_skb->data;
1300 out_hdr->sadb_msg_version = hdr->sadb_msg_version;
1301 out_hdr->sadb_msg_type = SADB_GETSPI;
1302 out_hdr->sadb_msg_satype = pfkey_proto2satype(proto);
1303 out_hdr->sadb_msg_errno = 0;
1304 out_hdr->sadb_msg_reserved = 0;
1305 out_hdr->sadb_msg_seq = hdr->sadb_msg_seq;
1306 out_hdr->sadb_msg_pid = hdr->sadb_msg_pid;
1308 xfrm_state_put(x);
1310 pfkey_broadcast(resp_skb, GFP_KERNEL, BROADCAST_ONE, sk);
1312 return 0;
1315 static int pfkey_acquire(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1317 struct xfrm_state *x;
1319 if (hdr->sadb_msg_len != sizeof(struct sadb_msg)/8)
1320 return -EOPNOTSUPP;
1322 if (hdr->sadb_msg_seq == 0 || hdr->sadb_msg_errno == 0)
1323 return 0;
1325 x = xfrm_find_acq_byseq(hdr->sadb_msg_seq);
1326 if (x == NULL)
1327 return 0;
1329 spin_lock_bh(&x->lock);
1330 if (x->km.state == XFRM_STATE_ACQ) {
1331 x->km.state = XFRM_STATE_ERROR;
1332 wake_up(&km_waitq);
1334 spin_unlock_bh(&x->lock);
1335 xfrm_state_put(x);
1336 return 0;
1339 static inline int event2poltype(int event)
1341 switch (event) {
1342 case XFRM_MSG_DELPOLICY:
1343 return SADB_X_SPDDELETE;
1344 case XFRM_MSG_NEWPOLICY:
1345 return SADB_X_SPDADD;
1346 case XFRM_MSG_UPDPOLICY:
1347 return SADB_X_SPDUPDATE;
1348 case XFRM_MSG_POLEXPIRE:
1349 // return SADB_X_SPDEXPIRE;
1350 default:
1351 printk("pfkey: Unknown policy event %d\n", event);
1352 break;
1355 return 0;
1358 static inline int event2keytype(int event)
1360 switch (event) {
1361 case XFRM_MSG_DELSA:
1362 return SADB_DELETE;
1363 case XFRM_MSG_NEWSA:
1364 return SADB_ADD;
1365 case XFRM_MSG_UPDSA:
1366 return SADB_UPDATE;
1367 case XFRM_MSG_EXPIRE:
1368 return SADB_EXPIRE;
1369 default:
1370 printk("pfkey: Unknown SA event %d\n", event);
1371 break;
1374 return 0;
1377 /* ADD/UPD/DEL */
1378 static int key_notify_sa(struct xfrm_state *x, struct km_event *c)
1380 struct sk_buff *skb;
1381 struct sadb_msg *hdr;
1382 int hsc = 3;
1384 if (c->event == XFRM_MSG_DELSA)
1385 hsc = 0;
1387 skb = pfkey_xfrm_state2msg(x, 0, hsc);
1389 if (IS_ERR(skb))
1390 return PTR_ERR(skb);
1392 hdr = (struct sadb_msg *) skb->data;
1393 hdr->sadb_msg_version = PF_KEY_V2;
1394 hdr->sadb_msg_type = event2keytype(c->event);
1395 hdr->sadb_msg_satype = pfkey_proto2satype(x->id.proto);
1396 hdr->sadb_msg_errno = 0;
1397 hdr->sadb_msg_reserved = 0;
1398 hdr->sadb_msg_seq = c->seq;
1399 hdr->sadb_msg_pid = c->pid;
1401 pfkey_broadcast(skb, GFP_ATOMIC, BROADCAST_ALL, NULL);
1403 return 0;
1406 static int pfkey_add(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1408 struct xfrm_state *x;
1409 int err;
1410 struct km_event c;
1412 xfrm_probe_algs();
1414 x = pfkey_msg2xfrm_state(hdr, ext_hdrs);
1415 if (IS_ERR(x))
1416 return PTR_ERR(x);
1418 xfrm_state_hold(x);
1419 if (hdr->sadb_msg_type == SADB_ADD)
1420 err = xfrm_state_add(x);
1421 else
1422 err = xfrm_state_update(x);
1424 if (err < 0) {
1425 x->km.state = XFRM_STATE_DEAD;
1426 __xfrm_state_put(x);
1427 goto out;
1430 if (hdr->sadb_msg_type == SADB_ADD)
1431 c.event = XFRM_MSG_NEWSA;
1432 else
1433 c.event = XFRM_MSG_UPDSA;
1434 c.seq = hdr->sadb_msg_seq;
1435 c.pid = hdr->sadb_msg_pid;
1436 km_state_notify(x, &c);
1437 out:
1438 xfrm_state_put(x);
1439 return err;
1442 static int pfkey_delete(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1444 struct xfrm_state *x;
1445 struct km_event c;
1446 int err;
1448 if (!ext_hdrs[SADB_EXT_SA-1] ||
1449 !present_and_same_family(ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
1450 ext_hdrs[SADB_EXT_ADDRESS_DST-1]))
1451 return -EINVAL;
1453 x = pfkey_xfrm_state_lookup(hdr, ext_hdrs);
1454 if (x == NULL)
1455 return -ESRCH;
1457 if (xfrm_state_kern(x)) {
1458 xfrm_state_put(x);
1459 return -EPERM;
1462 err = xfrm_state_delete(x);
1463 if (err < 0) {
1464 xfrm_state_put(x);
1465 return err;
1468 c.seq = hdr->sadb_msg_seq;
1469 c.pid = hdr->sadb_msg_pid;
1470 c.event = XFRM_MSG_DELSA;
1471 km_state_notify(x, &c);
1472 xfrm_state_put(x);
1474 return err;
1477 static int pfkey_get(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1479 __u8 proto;
1480 struct sk_buff *out_skb;
1481 struct sadb_msg *out_hdr;
1482 struct xfrm_state *x;
1484 if (!ext_hdrs[SADB_EXT_SA-1] ||
1485 !present_and_same_family(ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
1486 ext_hdrs[SADB_EXT_ADDRESS_DST-1]))
1487 return -EINVAL;
1489 x = pfkey_xfrm_state_lookup(hdr, ext_hdrs);
1490 if (x == NULL)
1491 return -ESRCH;
1493 out_skb = pfkey_xfrm_state2msg(x, 1, 3);
1494 proto = x->id.proto;
1495 xfrm_state_put(x);
1496 if (IS_ERR(out_skb))
1497 return PTR_ERR(out_skb);
1499 out_hdr = (struct sadb_msg *) out_skb->data;
1500 out_hdr->sadb_msg_version = hdr->sadb_msg_version;
1501 out_hdr->sadb_msg_type = SADB_GET;
1502 out_hdr->sadb_msg_satype = pfkey_proto2satype(proto);
1503 out_hdr->sadb_msg_errno = 0;
1504 out_hdr->sadb_msg_reserved = 0;
1505 out_hdr->sadb_msg_seq = hdr->sadb_msg_seq;
1506 out_hdr->sadb_msg_pid = hdr->sadb_msg_pid;
1507 pfkey_broadcast(out_skb, GFP_ATOMIC, BROADCAST_ONE, sk);
1509 return 0;
1512 static struct sk_buff *compose_sadb_supported(struct sadb_msg *orig,
1513 gfp_t allocation)
1515 struct sk_buff *skb;
1516 struct sadb_msg *hdr;
1517 int len, auth_len, enc_len, i;
1519 auth_len = xfrm_count_auth_supported();
1520 if (auth_len) {
1521 auth_len *= sizeof(struct sadb_alg);
1522 auth_len += sizeof(struct sadb_supported);
1525 enc_len = xfrm_count_enc_supported();
1526 if (enc_len) {
1527 enc_len *= sizeof(struct sadb_alg);
1528 enc_len += sizeof(struct sadb_supported);
1531 len = enc_len + auth_len + sizeof(struct sadb_msg);
1533 skb = alloc_skb(len + 16, allocation);
1534 if (!skb)
1535 goto out_put_algs;
1537 hdr = (struct sadb_msg *) skb_put(skb, sizeof(*hdr));
1538 pfkey_hdr_dup(hdr, orig);
1539 hdr->sadb_msg_errno = 0;
1540 hdr->sadb_msg_len = len / sizeof(uint64_t);
1542 if (auth_len) {
1543 struct sadb_supported *sp;
1544 struct sadb_alg *ap;
1546 sp = (struct sadb_supported *) skb_put(skb, auth_len);
1547 ap = (struct sadb_alg *) (sp + 1);
1549 sp->sadb_supported_len = auth_len / sizeof(uint64_t);
1550 sp->sadb_supported_exttype = SADB_EXT_SUPPORTED_AUTH;
1552 for (i = 0; ; i++) {
1553 struct xfrm_algo_desc *aalg = xfrm_aalg_get_byidx(i);
1554 if (!aalg)
1555 break;
1556 if (aalg->available)
1557 *ap++ = aalg->desc;
1561 if (enc_len) {
1562 struct sadb_supported *sp;
1563 struct sadb_alg *ap;
1565 sp = (struct sadb_supported *) skb_put(skb, enc_len);
1566 ap = (struct sadb_alg *) (sp + 1);
1568 sp->sadb_supported_len = enc_len / sizeof(uint64_t);
1569 sp->sadb_supported_exttype = SADB_EXT_SUPPORTED_ENCRYPT;
1571 for (i = 0; ; i++) {
1572 struct xfrm_algo_desc *ealg = xfrm_ealg_get_byidx(i);
1573 if (!ealg)
1574 break;
1575 if (ealg->available)
1576 *ap++ = ealg->desc;
1580 out_put_algs:
1581 return skb;
1584 static int pfkey_register(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1586 struct pfkey_sock *pfk = pfkey_sk(sk);
1587 struct sk_buff *supp_skb;
1589 if (hdr->sadb_msg_satype > SADB_SATYPE_MAX)
1590 return -EINVAL;
1592 if (hdr->sadb_msg_satype != SADB_SATYPE_UNSPEC) {
1593 if (pfk->registered&(1<<hdr->sadb_msg_satype))
1594 return -EEXIST;
1595 pfk->registered |= (1<<hdr->sadb_msg_satype);
1598 xfrm_probe_algs();
1600 supp_skb = compose_sadb_supported(hdr, GFP_KERNEL);
1601 if (!supp_skb) {
1602 if (hdr->sadb_msg_satype != SADB_SATYPE_UNSPEC)
1603 pfk->registered &= ~(1<<hdr->sadb_msg_satype);
1605 return -ENOBUFS;
1608 pfkey_broadcast(supp_skb, GFP_KERNEL, BROADCAST_REGISTERED, sk);
1610 return 0;
1613 static int key_notify_sa_flush(struct km_event *c)
1615 struct sk_buff *skb;
1616 struct sadb_msg *hdr;
1618 skb = alloc_skb(sizeof(struct sadb_msg) + 16, GFP_ATOMIC);
1619 if (!skb)
1620 return -ENOBUFS;
1621 hdr = (struct sadb_msg *) skb_put(skb, sizeof(struct sadb_msg));
1622 hdr->sadb_msg_satype = pfkey_proto2satype(c->data.proto);
1623 hdr->sadb_msg_type = SADB_FLUSH;
1624 hdr->sadb_msg_seq = c->seq;
1625 hdr->sadb_msg_pid = c->pid;
1626 hdr->sadb_msg_version = PF_KEY_V2;
1627 hdr->sadb_msg_errno = (uint8_t) 0;
1628 hdr->sadb_msg_len = (sizeof(struct sadb_msg) / sizeof(uint64_t));
1630 pfkey_broadcast(skb, GFP_ATOMIC, BROADCAST_ALL, NULL);
1632 return 0;
1635 static int pfkey_flush(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1637 unsigned proto;
1638 struct km_event c;
1640 proto = pfkey_satype2proto(hdr->sadb_msg_satype);
1641 if (proto == 0)
1642 return -EINVAL;
1644 xfrm_state_flush(proto);
1645 c.data.proto = proto;
1646 c.seq = hdr->sadb_msg_seq;
1647 c.pid = hdr->sadb_msg_pid;
1648 c.event = XFRM_MSG_FLUSHSA;
1649 km_state_notify(NULL, &c);
1651 return 0;
1654 struct pfkey_dump_data
1656 struct sk_buff *skb;
1657 struct sadb_msg *hdr;
1658 struct sock *sk;
1661 static int dump_sa(struct xfrm_state *x, int count, void *ptr)
1663 struct pfkey_dump_data *data = ptr;
1664 struct sk_buff *out_skb;
1665 struct sadb_msg *out_hdr;
1667 out_skb = pfkey_xfrm_state2msg(x, 1, 3);
1668 if (IS_ERR(out_skb))
1669 return PTR_ERR(out_skb);
1671 out_hdr = (struct sadb_msg *) out_skb->data;
1672 out_hdr->sadb_msg_version = data->hdr->sadb_msg_version;
1673 out_hdr->sadb_msg_type = SADB_DUMP;
1674 out_hdr->sadb_msg_satype = pfkey_proto2satype(x->id.proto);
1675 out_hdr->sadb_msg_errno = 0;
1676 out_hdr->sadb_msg_reserved = 0;
1677 out_hdr->sadb_msg_seq = count;
1678 out_hdr->sadb_msg_pid = data->hdr->sadb_msg_pid;
1679 pfkey_broadcast(out_skb, GFP_ATOMIC, BROADCAST_ONE, data->sk);
1680 return 0;
1683 static int pfkey_dump(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1685 u8 proto;
1686 struct pfkey_dump_data data = { .skb = skb, .hdr = hdr, .sk = sk };
1688 proto = pfkey_satype2proto(hdr->sadb_msg_satype);
1689 if (proto == 0)
1690 return -EINVAL;
1692 return xfrm_state_walk(proto, dump_sa, &data);
1695 static int pfkey_promisc(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
1697 struct pfkey_sock *pfk = pfkey_sk(sk);
1698 int satype = hdr->sadb_msg_satype;
1700 if (hdr->sadb_msg_len == (sizeof(*hdr) / sizeof(uint64_t))) {
1701 /* XXX we mangle packet... */
1702 hdr->sadb_msg_errno = 0;
1703 if (satype != 0 && satype != 1)
1704 return -EINVAL;
1705 pfk->promisc = satype;
1707 pfkey_broadcast(skb_clone(skb, GFP_KERNEL), GFP_KERNEL, BROADCAST_ALL, NULL);
1708 return 0;
1711 static int check_reqid(struct xfrm_policy *xp, int dir, int count, void *ptr)
1713 int i;
1714 u32 reqid = *(u32*)ptr;
1716 for (i=0; i<xp->xfrm_nr; i++) {
1717 if (xp->xfrm_vec[i].reqid == reqid)
1718 return -EEXIST;
1720 return 0;
1723 static u32 gen_reqid(void)
1725 u32 start;
1726 static u32 reqid = IPSEC_MANUAL_REQID_MAX;
1728 start = reqid;
1729 do {
1730 ++reqid;
1731 if (reqid == 0)
1732 reqid = IPSEC_MANUAL_REQID_MAX+1;
1733 if (xfrm_policy_walk(check_reqid, (void*)&reqid) != -EEXIST)
1734 return reqid;
1735 } while (reqid != start);
1736 return 0;
1739 static int
1740 parse_ipsecrequest(struct xfrm_policy *xp, struct sadb_x_ipsecrequest *rq)
1742 struct xfrm_tmpl *t = xp->xfrm_vec + xp->xfrm_nr;
1743 struct sockaddr_in *sin;
1744 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1745 struct sockaddr_in6 *sin6;
1746 #endif
1748 if (xp->xfrm_nr >= XFRM_MAX_DEPTH)
1749 return -ELOOP;
1751 if (rq->sadb_x_ipsecrequest_mode == 0)
1752 return -EINVAL;
1754 t->id.proto = rq->sadb_x_ipsecrequest_proto; /* XXX check proto */
1755 t->mode = rq->sadb_x_ipsecrequest_mode-1;
1756 if (rq->sadb_x_ipsecrequest_level == IPSEC_LEVEL_USE)
1757 t->optional = 1;
1758 else if (rq->sadb_x_ipsecrequest_level == IPSEC_LEVEL_UNIQUE) {
1759 t->reqid = rq->sadb_x_ipsecrequest_reqid;
1760 if (t->reqid > IPSEC_MANUAL_REQID_MAX)
1761 t->reqid = 0;
1762 if (!t->reqid && !(t->reqid = gen_reqid()))
1763 return -ENOBUFS;
1766 /* addresses present only in tunnel mode */
1767 if (t->mode) {
1768 switch (xp->family) {
1769 case AF_INET:
1770 sin = (void*)(rq+1);
1771 if (sin->sin_family != AF_INET)
1772 return -EINVAL;
1773 t->saddr.a4 = sin->sin_addr.s_addr;
1774 sin++;
1775 if (sin->sin_family != AF_INET)
1776 return -EINVAL;
1777 t->id.daddr.a4 = sin->sin_addr.s_addr;
1778 break;
1779 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1780 case AF_INET6:
1781 sin6 = (void *)(rq+1);
1782 if (sin6->sin6_family != AF_INET6)
1783 return -EINVAL;
1784 memcpy(t->saddr.a6, &sin6->sin6_addr, sizeof(struct in6_addr));
1785 sin6++;
1786 if (sin6->sin6_family != AF_INET6)
1787 return -EINVAL;
1788 memcpy(t->id.daddr.a6, &sin6->sin6_addr, sizeof(struct in6_addr));
1789 break;
1790 #endif
1791 default:
1792 return -EINVAL;
1795 /* No way to set this via kame pfkey */
1796 t->aalgos = t->ealgos = t->calgos = ~0;
1797 xp->xfrm_nr++;
1798 return 0;
1801 static int
1802 parse_ipsecrequests(struct xfrm_policy *xp, struct sadb_x_policy *pol)
1804 int err;
1805 int len = pol->sadb_x_policy_len*8 - sizeof(struct sadb_x_policy);
1806 struct sadb_x_ipsecrequest *rq = (void*)(pol+1);
1808 while (len >= sizeof(struct sadb_x_ipsecrequest)) {
1809 if ((err = parse_ipsecrequest(xp, rq)) < 0)
1810 return err;
1811 len -= rq->sadb_x_ipsecrequest_len;
1812 rq = (void*)((u8*)rq + rq->sadb_x_ipsecrequest_len);
1814 return 0;
1817 static inline int pfkey_xfrm_policy2sec_ctx_size(struct xfrm_policy *xp)
1819 struct xfrm_sec_ctx *xfrm_ctx = xp->security;
1821 if (xfrm_ctx) {
1822 int len = sizeof(struct sadb_x_sec_ctx);
1823 len += xfrm_ctx->ctx_len;
1824 return PFKEY_ALIGN8(len);
1826 return 0;
1829 static int pfkey_xfrm_policy2msg_size(struct xfrm_policy *xp)
1831 int sockaddr_size = pfkey_sockaddr_size(xp->family);
1832 int socklen = (xp->family == AF_INET ?
1833 sizeof(struct sockaddr_in) :
1834 sizeof(struct sockaddr_in6));
1836 return sizeof(struct sadb_msg) +
1837 (sizeof(struct sadb_lifetime) * 3) +
1838 (sizeof(struct sadb_address) * 2) +
1839 (sockaddr_size * 2) +
1840 sizeof(struct sadb_x_policy) +
1841 (xp->xfrm_nr * (sizeof(struct sadb_x_ipsecrequest) +
1842 (socklen * 2))) +
1843 pfkey_xfrm_policy2sec_ctx_size(xp);
1846 static struct sk_buff * pfkey_xfrm_policy2msg_prep(struct xfrm_policy *xp)
1848 struct sk_buff *skb;
1849 int size;
1851 size = pfkey_xfrm_policy2msg_size(xp);
1853 skb = alloc_skb(size + 16, GFP_ATOMIC);
1854 if (skb == NULL)
1855 return ERR_PTR(-ENOBUFS);
1857 return skb;
1860 static void pfkey_xfrm_policy2msg(struct sk_buff *skb, struct xfrm_policy *xp, int dir)
1862 struct sadb_msg *hdr;
1863 struct sadb_address *addr;
1864 struct sadb_lifetime *lifetime;
1865 struct sadb_x_policy *pol;
1866 struct sockaddr_in *sin;
1867 struct sadb_x_sec_ctx *sec_ctx;
1868 struct xfrm_sec_ctx *xfrm_ctx;
1869 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1870 struct sockaddr_in6 *sin6;
1871 #endif
1872 int i;
1873 int size;
1874 int sockaddr_size = pfkey_sockaddr_size(xp->family);
1875 int socklen = (xp->family == AF_INET ?
1876 sizeof(struct sockaddr_in) :
1877 sizeof(struct sockaddr_in6));
1879 size = pfkey_xfrm_policy2msg_size(xp);
1881 /* call should fill header later */
1882 hdr = (struct sadb_msg *) skb_put(skb, sizeof(struct sadb_msg));
1883 memset(hdr, 0, size); /* XXX do we need this ? */
1885 /* src address */
1886 addr = (struct sadb_address*) skb_put(skb,
1887 sizeof(struct sadb_address)+sockaddr_size);
1888 addr->sadb_address_len =
1889 (sizeof(struct sadb_address)+sockaddr_size)/
1890 sizeof(uint64_t);
1891 addr->sadb_address_exttype = SADB_EXT_ADDRESS_SRC;
1892 addr->sadb_address_proto = pfkey_proto_from_xfrm(xp->selector.proto);
1893 addr->sadb_address_prefixlen = xp->selector.prefixlen_s;
1894 addr->sadb_address_reserved = 0;
1895 /* src address */
1896 if (xp->family == AF_INET) {
1897 sin = (struct sockaddr_in *) (addr + 1);
1898 sin->sin_family = AF_INET;
1899 sin->sin_addr.s_addr = xp->selector.saddr.a4;
1900 sin->sin_port = xp->selector.sport;
1901 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
1903 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1904 else if (xp->family == AF_INET6) {
1905 sin6 = (struct sockaddr_in6 *) (addr + 1);
1906 sin6->sin6_family = AF_INET6;
1907 sin6->sin6_port = xp->selector.sport;
1908 sin6->sin6_flowinfo = 0;
1909 memcpy(&sin6->sin6_addr, xp->selector.saddr.a6,
1910 sizeof(struct in6_addr));
1911 sin6->sin6_scope_id = 0;
1913 #endif
1914 else
1915 BUG();
1917 /* dst address */
1918 addr = (struct sadb_address*) skb_put(skb,
1919 sizeof(struct sadb_address)+sockaddr_size);
1920 addr->sadb_address_len =
1921 (sizeof(struct sadb_address)+sockaddr_size)/
1922 sizeof(uint64_t);
1923 addr->sadb_address_exttype = SADB_EXT_ADDRESS_DST;
1924 addr->sadb_address_proto = pfkey_proto_from_xfrm(xp->selector.proto);
1925 addr->sadb_address_prefixlen = xp->selector.prefixlen_d;
1926 addr->sadb_address_reserved = 0;
1927 if (xp->family == AF_INET) {
1928 sin = (struct sockaddr_in *) (addr + 1);
1929 sin->sin_family = AF_INET;
1930 sin->sin_addr.s_addr = xp->selector.daddr.a4;
1931 sin->sin_port = xp->selector.dport;
1932 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
1934 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
1935 else if (xp->family == AF_INET6) {
1936 sin6 = (struct sockaddr_in6 *) (addr + 1);
1937 sin6->sin6_family = AF_INET6;
1938 sin6->sin6_port = xp->selector.dport;
1939 sin6->sin6_flowinfo = 0;
1940 memcpy(&sin6->sin6_addr, xp->selector.daddr.a6,
1941 sizeof(struct in6_addr));
1942 sin6->sin6_scope_id = 0;
1944 #endif
1945 else
1946 BUG();
1948 /* hard time */
1949 lifetime = (struct sadb_lifetime *) skb_put(skb,
1950 sizeof(struct sadb_lifetime));
1951 lifetime->sadb_lifetime_len =
1952 sizeof(struct sadb_lifetime)/sizeof(uint64_t);
1953 lifetime->sadb_lifetime_exttype = SADB_EXT_LIFETIME_HARD;
1954 lifetime->sadb_lifetime_allocations = _X2KEY(xp->lft.hard_packet_limit);
1955 lifetime->sadb_lifetime_bytes = _X2KEY(xp->lft.hard_byte_limit);
1956 lifetime->sadb_lifetime_addtime = xp->lft.hard_add_expires_seconds;
1957 lifetime->sadb_lifetime_usetime = xp->lft.hard_use_expires_seconds;
1958 /* soft time */
1959 lifetime = (struct sadb_lifetime *) skb_put(skb,
1960 sizeof(struct sadb_lifetime));
1961 lifetime->sadb_lifetime_len =
1962 sizeof(struct sadb_lifetime)/sizeof(uint64_t);
1963 lifetime->sadb_lifetime_exttype = SADB_EXT_LIFETIME_SOFT;
1964 lifetime->sadb_lifetime_allocations = _X2KEY(xp->lft.soft_packet_limit);
1965 lifetime->sadb_lifetime_bytes = _X2KEY(xp->lft.soft_byte_limit);
1966 lifetime->sadb_lifetime_addtime = xp->lft.soft_add_expires_seconds;
1967 lifetime->sadb_lifetime_usetime = xp->lft.soft_use_expires_seconds;
1968 /* current time */
1969 lifetime = (struct sadb_lifetime *) skb_put(skb,
1970 sizeof(struct sadb_lifetime));
1971 lifetime->sadb_lifetime_len =
1972 sizeof(struct sadb_lifetime)/sizeof(uint64_t);
1973 lifetime->sadb_lifetime_exttype = SADB_EXT_LIFETIME_CURRENT;
1974 lifetime->sadb_lifetime_allocations = xp->curlft.packets;
1975 lifetime->sadb_lifetime_bytes = xp->curlft.bytes;
1976 lifetime->sadb_lifetime_addtime = xp->curlft.add_time;
1977 lifetime->sadb_lifetime_usetime = xp->curlft.use_time;
1979 pol = (struct sadb_x_policy *) skb_put(skb, sizeof(struct sadb_x_policy));
1980 pol->sadb_x_policy_len = sizeof(struct sadb_x_policy)/sizeof(uint64_t);
1981 pol->sadb_x_policy_exttype = SADB_X_EXT_POLICY;
1982 pol->sadb_x_policy_type = IPSEC_POLICY_DISCARD;
1983 if (xp->action == XFRM_POLICY_ALLOW) {
1984 if (xp->xfrm_nr)
1985 pol->sadb_x_policy_type = IPSEC_POLICY_IPSEC;
1986 else
1987 pol->sadb_x_policy_type = IPSEC_POLICY_NONE;
1989 pol->sadb_x_policy_dir = dir+1;
1990 pol->sadb_x_policy_id = xp->index;
1991 pol->sadb_x_policy_priority = xp->priority;
1993 for (i=0; i<xp->xfrm_nr; i++) {
1994 struct sadb_x_ipsecrequest *rq;
1995 struct xfrm_tmpl *t = xp->xfrm_vec + i;
1996 int req_size;
1998 req_size = sizeof(struct sadb_x_ipsecrequest);
1999 if (t->mode)
2000 req_size += 2*socklen;
2001 else
2002 size -= 2*socklen;
2003 rq = (void*)skb_put(skb, req_size);
2004 pol->sadb_x_policy_len += req_size/8;
2005 memset(rq, 0, sizeof(*rq));
2006 rq->sadb_x_ipsecrequest_len = req_size;
2007 rq->sadb_x_ipsecrequest_proto = t->id.proto;
2008 rq->sadb_x_ipsecrequest_mode = t->mode+1;
2009 rq->sadb_x_ipsecrequest_level = IPSEC_LEVEL_REQUIRE;
2010 if (t->reqid)
2011 rq->sadb_x_ipsecrequest_level = IPSEC_LEVEL_UNIQUE;
2012 if (t->optional)
2013 rq->sadb_x_ipsecrequest_level = IPSEC_LEVEL_USE;
2014 rq->sadb_x_ipsecrequest_reqid = t->reqid;
2015 if (t->mode) {
2016 switch (xp->family) {
2017 case AF_INET:
2018 sin = (void*)(rq+1);
2019 sin->sin_family = AF_INET;
2020 sin->sin_addr.s_addr = t->saddr.a4;
2021 sin->sin_port = 0;
2022 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
2023 sin++;
2024 sin->sin_family = AF_INET;
2025 sin->sin_addr.s_addr = t->id.daddr.a4;
2026 sin->sin_port = 0;
2027 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
2028 break;
2029 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2030 case AF_INET6:
2031 sin6 = (void*)(rq+1);
2032 sin6->sin6_family = AF_INET6;
2033 sin6->sin6_port = 0;
2034 sin6->sin6_flowinfo = 0;
2035 memcpy(&sin6->sin6_addr, t->saddr.a6,
2036 sizeof(struct in6_addr));
2037 sin6->sin6_scope_id = 0;
2039 sin6++;
2040 sin6->sin6_family = AF_INET6;
2041 sin6->sin6_port = 0;
2042 sin6->sin6_flowinfo = 0;
2043 memcpy(&sin6->sin6_addr, t->id.daddr.a6,
2044 sizeof(struct in6_addr));
2045 sin6->sin6_scope_id = 0;
2046 break;
2047 #endif
2048 default:
2049 break;
2054 /* security context */
2055 if ((xfrm_ctx = xp->security)) {
2056 int ctx_size = pfkey_xfrm_policy2sec_ctx_size(xp);
2058 sec_ctx = (struct sadb_x_sec_ctx *) skb_put(skb, ctx_size);
2059 sec_ctx->sadb_x_sec_len = ctx_size / sizeof(uint64_t);
2060 sec_ctx->sadb_x_sec_exttype = SADB_X_EXT_SEC_CTX;
2061 sec_ctx->sadb_x_ctx_doi = xfrm_ctx->ctx_doi;
2062 sec_ctx->sadb_x_ctx_alg = xfrm_ctx->ctx_alg;
2063 sec_ctx->sadb_x_ctx_len = xfrm_ctx->ctx_len;
2064 memcpy(sec_ctx + 1, xfrm_ctx->ctx_str,
2065 xfrm_ctx->ctx_len);
2068 hdr->sadb_msg_len = size / sizeof(uint64_t);
2069 hdr->sadb_msg_reserved = atomic_read(&xp->refcnt);
2072 static int key_notify_policy(struct xfrm_policy *xp, int dir, struct km_event *c)
2074 struct sk_buff *out_skb;
2075 struct sadb_msg *out_hdr;
2076 int err;
2078 out_skb = pfkey_xfrm_policy2msg_prep(xp);
2079 if (IS_ERR(out_skb)) {
2080 err = PTR_ERR(out_skb);
2081 goto out;
2083 pfkey_xfrm_policy2msg(out_skb, xp, dir);
2085 out_hdr = (struct sadb_msg *) out_skb->data;
2086 out_hdr->sadb_msg_version = PF_KEY_V2;
2088 if (c->data.byid && c->event == XFRM_MSG_DELPOLICY)
2089 out_hdr->sadb_msg_type = SADB_X_SPDDELETE2;
2090 else
2091 out_hdr->sadb_msg_type = event2poltype(c->event);
2092 out_hdr->sadb_msg_errno = 0;
2093 out_hdr->sadb_msg_seq = c->seq;
2094 out_hdr->sadb_msg_pid = c->pid;
2095 pfkey_broadcast(out_skb, GFP_ATOMIC, BROADCAST_ALL, NULL);
2096 out:
2097 return 0;
2101 static int pfkey_spdadd(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
2103 int err = 0;
2104 struct sadb_lifetime *lifetime;
2105 struct sadb_address *sa;
2106 struct sadb_x_policy *pol;
2107 struct xfrm_policy *xp;
2108 struct km_event c;
2109 struct sadb_x_sec_ctx *sec_ctx;
2111 if (!present_and_same_family(ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
2112 ext_hdrs[SADB_EXT_ADDRESS_DST-1]) ||
2113 !ext_hdrs[SADB_X_EXT_POLICY-1])
2114 return -EINVAL;
2116 pol = ext_hdrs[SADB_X_EXT_POLICY-1];
2117 if (pol->sadb_x_policy_type > IPSEC_POLICY_IPSEC)
2118 return -EINVAL;
2119 if (!pol->sadb_x_policy_dir || pol->sadb_x_policy_dir >= IPSEC_DIR_MAX)
2120 return -EINVAL;
2122 xp = xfrm_policy_alloc(GFP_KERNEL);
2123 if (xp == NULL)
2124 return -ENOBUFS;
2126 xp->action = (pol->sadb_x_policy_type == IPSEC_POLICY_DISCARD ?
2127 XFRM_POLICY_BLOCK : XFRM_POLICY_ALLOW);
2128 xp->priority = pol->sadb_x_policy_priority;
2130 sa = ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
2131 xp->family = pfkey_sadb_addr2xfrm_addr(sa, &xp->selector.saddr);
2132 if (!xp->family) {
2133 err = -EINVAL;
2134 goto out;
2136 xp->selector.family = xp->family;
2137 xp->selector.prefixlen_s = sa->sadb_address_prefixlen;
2138 xp->selector.proto = pfkey_proto_to_xfrm(sa->sadb_address_proto);
2139 xp->selector.sport = ((struct sockaddr_in *)(sa+1))->sin_port;
2140 if (xp->selector.sport)
2141 xp->selector.sport_mask = ~0;
2143 sa = ext_hdrs[SADB_EXT_ADDRESS_DST-1],
2144 pfkey_sadb_addr2xfrm_addr(sa, &xp->selector.daddr);
2145 xp->selector.prefixlen_d = sa->sadb_address_prefixlen;
2147 /* Amusing, we set this twice. KAME apps appear to set same value
2148 * in both addresses.
2150 xp->selector.proto = pfkey_proto_to_xfrm(sa->sadb_address_proto);
2152 xp->selector.dport = ((struct sockaddr_in *)(sa+1))->sin_port;
2153 if (xp->selector.dport)
2154 xp->selector.dport_mask = ~0;
2156 sec_ctx = (struct sadb_x_sec_ctx *) ext_hdrs[SADB_X_EXT_SEC_CTX-1];
2157 if (sec_ctx != NULL) {
2158 struct xfrm_user_sec_ctx *uctx = pfkey_sadb2xfrm_user_sec_ctx(sec_ctx);
2160 if (!uctx) {
2161 err = -ENOBUFS;
2162 goto out;
2165 err = security_xfrm_policy_alloc(xp, uctx);
2166 kfree(uctx);
2168 if (err)
2169 goto out;
2172 xp->lft.soft_byte_limit = XFRM_INF;
2173 xp->lft.hard_byte_limit = XFRM_INF;
2174 xp->lft.soft_packet_limit = XFRM_INF;
2175 xp->lft.hard_packet_limit = XFRM_INF;
2176 if ((lifetime = ext_hdrs[SADB_EXT_LIFETIME_HARD-1]) != NULL) {
2177 xp->lft.hard_packet_limit = _KEY2X(lifetime->sadb_lifetime_allocations);
2178 xp->lft.hard_byte_limit = _KEY2X(lifetime->sadb_lifetime_bytes);
2179 xp->lft.hard_add_expires_seconds = lifetime->sadb_lifetime_addtime;
2180 xp->lft.hard_use_expires_seconds = lifetime->sadb_lifetime_usetime;
2182 if ((lifetime = ext_hdrs[SADB_EXT_LIFETIME_SOFT-1]) != NULL) {
2183 xp->lft.soft_packet_limit = _KEY2X(lifetime->sadb_lifetime_allocations);
2184 xp->lft.soft_byte_limit = _KEY2X(lifetime->sadb_lifetime_bytes);
2185 xp->lft.soft_add_expires_seconds = lifetime->sadb_lifetime_addtime;
2186 xp->lft.soft_use_expires_seconds = lifetime->sadb_lifetime_usetime;
2188 xp->xfrm_nr = 0;
2189 if (pol->sadb_x_policy_type == IPSEC_POLICY_IPSEC &&
2190 (err = parse_ipsecrequests(xp, pol)) < 0)
2191 goto out;
2193 err = xfrm_policy_insert(pol->sadb_x_policy_dir-1, xp,
2194 hdr->sadb_msg_type != SADB_X_SPDUPDATE);
2196 if (err)
2197 goto out;
2199 if (hdr->sadb_msg_type == SADB_X_SPDUPDATE)
2200 c.event = XFRM_MSG_UPDPOLICY;
2201 else
2202 c.event = XFRM_MSG_NEWPOLICY;
2204 c.seq = hdr->sadb_msg_seq;
2205 c.pid = hdr->sadb_msg_pid;
2207 km_policy_notify(xp, pol->sadb_x_policy_dir-1, &c);
2208 xfrm_pol_put(xp);
2209 return 0;
2211 out:
2212 security_xfrm_policy_free(xp);
2213 kfree(xp);
2214 return err;
2217 static int pfkey_spddelete(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
2219 int err;
2220 struct sadb_address *sa;
2221 struct sadb_x_policy *pol;
2222 struct xfrm_policy *xp, tmp;
2223 struct xfrm_selector sel;
2224 struct km_event c;
2225 struct sadb_x_sec_ctx *sec_ctx;
2227 if (!present_and_same_family(ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
2228 ext_hdrs[SADB_EXT_ADDRESS_DST-1]) ||
2229 !ext_hdrs[SADB_X_EXT_POLICY-1])
2230 return -EINVAL;
2232 pol = ext_hdrs[SADB_X_EXT_POLICY-1];
2233 if (!pol->sadb_x_policy_dir || pol->sadb_x_policy_dir >= IPSEC_DIR_MAX)
2234 return -EINVAL;
2236 memset(&sel, 0, sizeof(sel));
2238 sa = ext_hdrs[SADB_EXT_ADDRESS_SRC-1],
2239 sel.family = pfkey_sadb_addr2xfrm_addr(sa, &sel.saddr);
2240 sel.prefixlen_s = sa->sadb_address_prefixlen;
2241 sel.proto = pfkey_proto_to_xfrm(sa->sadb_address_proto);
2242 sel.sport = ((struct sockaddr_in *)(sa+1))->sin_port;
2243 if (sel.sport)
2244 sel.sport_mask = ~0;
2246 sa = ext_hdrs[SADB_EXT_ADDRESS_DST-1],
2247 pfkey_sadb_addr2xfrm_addr(sa, &sel.daddr);
2248 sel.prefixlen_d = sa->sadb_address_prefixlen;
2249 sel.proto = pfkey_proto_to_xfrm(sa->sadb_address_proto);
2250 sel.dport = ((struct sockaddr_in *)(sa+1))->sin_port;
2251 if (sel.dport)
2252 sel.dport_mask = ~0;
2254 sec_ctx = (struct sadb_x_sec_ctx *) ext_hdrs[SADB_X_EXT_SEC_CTX-1];
2255 memset(&tmp, 0, sizeof(struct xfrm_policy));
2257 if (sec_ctx != NULL) {
2258 struct xfrm_user_sec_ctx *uctx = pfkey_sadb2xfrm_user_sec_ctx(sec_ctx);
2260 if (!uctx)
2261 return -ENOMEM;
2263 err = security_xfrm_policy_alloc(&tmp, uctx);
2264 kfree(uctx);
2266 if (err)
2267 return err;
2270 xp = xfrm_policy_bysel_ctx(pol->sadb_x_policy_dir-1, &sel, tmp.security, 1);
2271 security_xfrm_policy_free(&tmp);
2272 if (xp == NULL)
2273 return -ENOENT;
2275 err = 0;
2277 c.seq = hdr->sadb_msg_seq;
2278 c.pid = hdr->sadb_msg_pid;
2279 c.event = XFRM_MSG_DELPOLICY;
2280 km_policy_notify(xp, pol->sadb_x_policy_dir-1, &c);
2282 xfrm_pol_put(xp);
2283 return err;
2286 static int key_pol_get_resp(struct sock *sk, struct xfrm_policy *xp, struct sadb_msg *hdr, int dir)
2288 int err;
2289 struct sk_buff *out_skb;
2290 struct sadb_msg *out_hdr;
2291 err = 0;
2293 out_skb = pfkey_xfrm_policy2msg_prep(xp);
2294 if (IS_ERR(out_skb)) {
2295 err = PTR_ERR(out_skb);
2296 goto out;
2298 pfkey_xfrm_policy2msg(out_skb, xp, dir);
2300 out_hdr = (struct sadb_msg *) out_skb->data;
2301 out_hdr->sadb_msg_version = hdr->sadb_msg_version;
2302 out_hdr->sadb_msg_type = hdr->sadb_msg_type;
2303 out_hdr->sadb_msg_satype = 0;
2304 out_hdr->sadb_msg_errno = 0;
2305 out_hdr->sadb_msg_seq = hdr->sadb_msg_seq;
2306 out_hdr->sadb_msg_pid = hdr->sadb_msg_pid;
2307 pfkey_broadcast(out_skb, GFP_ATOMIC, BROADCAST_ONE, sk);
2308 err = 0;
2310 out:
2311 return err;
2314 static int pfkey_spdget(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
2316 unsigned int dir;
2317 int err;
2318 struct sadb_x_policy *pol;
2319 struct xfrm_policy *xp;
2320 struct km_event c;
2322 if ((pol = ext_hdrs[SADB_X_EXT_POLICY-1]) == NULL)
2323 return -EINVAL;
2325 dir = xfrm_policy_id2dir(pol->sadb_x_policy_id);
2326 if (dir >= XFRM_POLICY_MAX)
2327 return -EINVAL;
2329 xp = xfrm_policy_byid(dir, pol->sadb_x_policy_id,
2330 hdr->sadb_msg_type == SADB_X_SPDDELETE2);
2331 if (xp == NULL)
2332 return -ENOENT;
2334 err = 0;
2336 c.seq = hdr->sadb_msg_seq;
2337 c.pid = hdr->sadb_msg_pid;
2338 if (hdr->sadb_msg_type == SADB_X_SPDDELETE2) {
2339 c.data.byid = 1;
2340 c.event = XFRM_MSG_DELPOLICY;
2341 km_policy_notify(xp, dir, &c);
2342 } else {
2343 err = key_pol_get_resp(sk, xp, hdr, dir);
2346 xfrm_pol_put(xp);
2347 return err;
2350 static int dump_sp(struct xfrm_policy *xp, int dir, int count, void *ptr)
2352 struct pfkey_dump_data *data = ptr;
2353 struct sk_buff *out_skb;
2354 struct sadb_msg *out_hdr;
2356 out_skb = pfkey_xfrm_policy2msg_prep(xp);
2357 if (IS_ERR(out_skb))
2358 return PTR_ERR(out_skb);
2360 pfkey_xfrm_policy2msg(out_skb, xp, dir);
2362 out_hdr = (struct sadb_msg *) out_skb->data;
2363 out_hdr->sadb_msg_version = data->hdr->sadb_msg_version;
2364 out_hdr->sadb_msg_type = SADB_X_SPDDUMP;
2365 out_hdr->sadb_msg_satype = SADB_SATYPE_UNSPEC;
2366 out_hdr->sadb_msg_errno = 0;
2367 out_hdr->sadb_msg_seq = count;
2368 out_hdr->sadb_msg_pid = data->hdr->sadb_msg_pid;
2369 pfkey_broadcast(out_skb, GFP_ATOMIC, BROADCAST_ONE, data->sk);
2370 return 0;
2373 static int pfkey_spddump(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
2375 struct pfkey_dump_data data = { .skb = skb, .hdr = hdr, .sk = sk };
2377 return xfrm_policy_walk(dump_sp, &data);
2380 static int key_notify_policy_flush(struct km_event *c)
2382 struct sk_buff *skb_out;
2383 struct sadb_msg *hdr;
2385 skb_out = alloc_skb(sizeof(struct sadb_msg) + 16, GFP_ATOMIC);
2386 if (!skb_out)
2387 return -ENOBUFS;
2388 hdr = (struct sadb_msg *) skb_put(skb_out, sizeof(struct sadb_msg));
2389 hdr->sadb_msg_type = SADB_X_SPDFLUSH;
2390 hdr->sadb_msg_seq = c->seq;
2391 hdr->sadb_msg_pid = c->pid;
2392 hdr->sadb_msg_version = PF_KEY_V2;
2393 hdr->sadb_msg_errno = (uint8_t) 0;
2394 hdr->sadb_msg_len = (sizeof(struct sadb_msg) / sizeof(uint64_t));
2395 pfkey_broadcast(skb_out, GFP_ATOMIC, BROADCAST_ALL, NULL);
2396 return 0;
2400 static int pfkey_spdflush(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr, void **ext_hdrs)
2402 struct km_event c;
2404 xfrm_policy_flush();
2405 c.event = XFRM_MSG_FLUSHPOLICY;
2406 c.pid = hdr->sadb_msg_pid;
2407 c.seq = hdr->sadb_msg_seq;
2408 km_policy_notify(NULL, 0, &c);
2410 return 0;
2413 typedef int (*pfkey_handler)(struct sock *sk, struct sk_buff *skb,
2414 struct sadb_msg *hdr, void **ext_hdrs);
2415 static pfkey_handler pfkey_funcs[SADB_MAX + 1] = {
2416 [SADB_RESERVED] = pfkey_reserved,
2417 [SADB_GETSPI] = pfkey_getspi,
2418 [SADB_UPDATE] = pfkey_add,
2419 [SADB_ADD] = pfkey_add,
2420 [SADB_DELETE] = pfkey_delete,
2421 [SADB_GET] = pfkey_get,
2422 [SADB_ACQUIRE] = pfkey_acquire,
2423 [SADB_REGISTER] = pfkey_register,
2424 [SADB_EXPIRE] = NULL,
2425 [SADB_FLUSH] = pfkey_flush,
2426 [SADB_DUMP] = pfkey_dump,
2427 [SADB_X_PROMISC] = pfkey_promisc,
2428 [SADB_X_PCHANGE] = NULL,
2429 [SADB_X_SPDUPDATE] = pfkey_spdadd,
2430 [SADB_X_SPDADD] = pfkey_spdadd,
2431 [SADB_X_SPDDELETE] = pfkey_spddelete,
2432 [SADB_X_SPDGET] = pfkey_spdget,
2433 [SADB_X_SPDACQUIRE] = NULL,
2434 [SADB_X_SPDDUMP] = pfkey_spddump,
2435 [SADB_X_SPDFLUSH] = pfkey_spdflush,
2436 [SADB_X_SPDSETIDX] = pfkey_spdadd,
2437 [SADB_X_SPDDELETE2] = pfkey_spdget,
2440 static int pfkey_process(struct sock *sk, struct sk_buff *skb, struct sadb_msg *hdr)
2442 void *ext_hdrs[SADB_EXT_MAX];
2443 int err;
2445 pfkey_broadcast(skb_clone(skb, GFP_KERNEL), GFP_KERNEL,
2446 BROADCAST_PROMISC_ONLY, NULL);
2448 memset(ext_hdrs, 0, sizeof(ext_hdrs));
2449 err = parse_exthdrs(skb, hdr, ext_hdrs);
2450 if (!err) {
2451 err = -EOPNOTSUPP;
2452 if (pfkey_funcs[hdr->sadb_msg_type])
2453 err = pfkey_funcs[hdr->sadb_msg_type](sk, skb, hdr, ext_hdrs);
2455 return err;
2458 static struct sadb_msg *pfkey_get_base_msg(struct sk_buff *skb, int *errp)
2460 struct sadb_msg *hdr = NULL;
2462 if (skb->len < sizeof(*hdr)) {
2463 *errp = -EMSGSIZE;
2464 } else {
2465 hdr = (struct sadb_msg *) skb->data;
2466 if (hdr->sadb_msg_version != PF_KEY_V2 ||
2467 hdr->sadb_msg_reserved != 0 ||
2468 (hdr->sadb_msg_type <= SADB_RESERVED ||
2469 hdr->sadb_msg_type > SADB_MAX)) {
2470 hdr = NULL;
2471 *errp = -EINVAL;
2472 } else if (hdr->sadb_msg_len != (skb->len /
2473 sizeof(uint64_t)) ||
2474 hdr->sadb_msg_len < (sizeof(struct sadb_msg) /
2475 sizeof(uint64_t))) {
2476 hdr = NULL;
2477 *errp = -EMSGSIZE;
2478 } else {
2479 *errp = 0;
2482 return hdr;
2485 static inline int aalg_tmpl_set(struct xfrm_tmpl *t, struct xfrm_algo_desc *d)
2487 unsigned int id = d->desc.sadb_alg_id;
2489 if (id >= sizeof(t->aalgos) * 8)
2490 return 0;
2492 return (t->aalgos >> id) & 1;
2495 static inline int ealg_tmpl_set(struct xfrm_tmpl *t, struct xfrm_algo_desc *d)
2497 unsigned int id = d->desc.sadb_alg_id;
2499 if (id >= sizeof(t->ealgos) * 8)
2500 return 0;
2502 return (t->ealgos >> id) & 1;
2505 static int count_ah_combs(struct xfrm_tmpl *t)
2507 int i, sz = 0;
2509 for (i = 0; ; i++) {
2510 struct xfrm_algo_desc *aalg = xfrm_aalg_get_byidx(i);
2511 if (!aalg)
2512 break;
2513 if (aalg_tmpl_set(t, aalg) && aalg->available)
2514 sz += sizeof(struct sadb_comb);
2516 return sz + sizeof(struct sadb_prop);
2519 static int count_esp_combs(struct xfrm_tmpl *t)
2521 int i, k, sz = 0;
2523 for (i = 0; ; i++) {
2524 struct xfrm_algo_desc *ealg = xfrm_ealg_get_byidx(i);
2525 if (!ealg)
2526 break;
2528 if (!(ealg_tmpl_set(t, ealg) && ealg->available))
2529 continue;
2531 for (k = 1; ; k++) {
2532 struct xfrm_algo_desc *aalg = xfrm_aalg_get_byidx(k);
2533 if (!aalg)
2534 break;
2536 if (aalg_tmpl_set(t, aalg) && aalg->available)
2537 sz += sizeof(struct sadb_comb);
2540 return sz + sizeof(struct sadb_prop);
2543 static void dump_ah_combs(struct sk_buff *skb, struct xfrm_tmpl *t)
2545 struct sadb_prop *p;
2546 int i;
2548 p = (struct sadb_prop*)skb_put(skb, sizeof(struct sadb_prop));
2549 p->sadb_prop_len = sizeof(struct sadb_prop)/8;
2550 p->sadb_prop_exttype = SADB_EXT_PROPOSAL;
2551 p->sadb_prop_replay = 32;
2552 memset(p->sadb_prop_reserved, 0, sizeof(p->sadb_prop_reserved));
2554 for (i = 0; ; i++) {
2555 struct xfrm_algo_desc *aalg = xfrm_aalg_get_byidx(i);
2556 if (!aalg)
2557 break;
2559 if (aalg_tmpl_set(t, aalg) && aalg->available) {
2560 struct sadb_comb *c;
2561 c = (struct sadb_comb*)skb_put(skb, sizeof(struct sadb_comb));
2562 memset(c, 0, sizeof(*c));
2563 p->sadb_prop_len += sizeof(struct sadb_comb)/8;
2564 c->sadb_comb_auth = aalg->desc.sadb_alg_id;
2565 c->sadb_comb_auth_minbits = aalg->desc.sadb_alg_minbits;
2566 c->sadb_comb_auth_maxbits = aalg->desc.sadb_alg_maxbits;
2567 c->sadb_comb_hard_addtime = 24*60*60;
2568 c->sadb_comb_soft_addtime = 20*60*60;
2569 c->sadb_comb_hard_usetime = 8*60*60;
2570 c->sadb_comb_soft_usetime = 7*60*60;
2575 static void dump_esp_combs(struct sk_buff *skb, struct xfrm_tmpl *t)
2577 struct sadb_prop *p;
2578 int i, k;
2580 p = (struct sadb_prop*)skb_put(skb, sizeof(struct sadb_prop));
2581 p->sadb_prop_len = sizeof(struct sadb_prop)/8;
2582 p->sadb_prop_exttype = SADB_EXT_PROPOSAL;
2583 p->sadb_prop_replay = 32;
2584 memset(p->sadb_prop_reserved, 0, sizeof(p->sadb_prop_reserved));
2586 for (i=0; ; i++) {
2587 struct xfrm_algo_desc *ealg = xfrm_ealg_get_byidx(i);
2588 if (!ealg)
2589 break;
2591 if (!(ealg_tmpl_set(t, ealg) && ealg->available))
2592 continue;
2594 for (k = 1; ; k++) {
2595 struct sadb_comb *c;
2596 struct xfrm_algo_desc *aalg = xfrm_aalg_get_byidx(k);
2597 if (!aalg)
2598 break;
2599 if (!(aalg_tmpl_set(t, aalg) && aalg->available))
2600 continue;
2601 c = (struct sadb_comb*)skb_put(skb, sizeof(struct sadb_comb));
2602 memset(c, 0, sizeof(*c));
2603 p->sadb_prop_len += sizeof(struct sadb_comb)/8;
2604 c->sadb_comb_auth = aalg->desc.sadb_alg_id;
2605 c->sadb_comb_auth_minbits = aalg->desc.sadb_alg_minbits;
2606 c->sadb_comb_auth_maxbits = aalg->desc.sadb_alg_maxbits;
2607 c->sadb_comb_encrypt = ealg->desc.sadb_alg_id;
2608 c->sadb_comb_encrypt_minbits = ealg->desc.sadb_alg_minbits;
2609 c->sadb_comb_encrypt_maxbits = ealg->desc.sadb_alg_maxbits;
2610 c->sadb_comb_hard_addtime = 24*60*60;
2611 c->sadb_comb_soft_addtime = 20*60*60;
2612 c->sadb_comb_hard_usetime = 8*60*60;
2613 c->sadb_comb_soft_usetime = 7*60*60;
2618 static int key_notify_policy_expire(struct xfrm_policy *xp, struct km_event *c)
2620 return 0;
2623 static int key_notify_sa_expire(struct xfrm_state *x, struct km_event *c)
2625 struct sk_buff *out_skb;
2626 struct sadb_msg *out_hdr;
2627 int hard;
2628 int hsc;
2630 hard = c->data.hard;
2631 if (hard)
2632 hsc = 2;
2633 else
2634 hsc = 1;
2636 out_skb = pfkey_xfrm_state2msg(x, 0, hsc);
2637 if (IS_ERR(out_skb))
2638 return PTR_ERR(out_skb);
2640 out_hdr = (struct sadb_msg *) out_skb->data;
2641 out_hdr->sadb_msg_version = PF_KEY_V2;
2642 out_hdr->sadb_msg_type = SADB_EXPIRE;
2643 out_hdr->sadb_msg_satype = pfkey_proto2satype(x->id.proto);
2644 out_hdr->sadb_msg_errno = 0;
2645 out_hdr->sadb_msg_reserved = 0;
2646 out_hdr->sadb_msg_seq = 0;
2647 out_hdr->sadb_msg_pid = 0;
2649 pfkey_broadcast(out_skb, GFP_ATOMIC, BROADCAST_REGISTERED, NULL);
2650 return 0;
2653 static int pfkey_send_notify(struct xfrm_state *x, struct km_event *c)
2655 switch (c->event) {
2656 case XFRM_MSG_EXPIRE:
2657 return key_notify_sa_expire(x, c);
2658 case XFRM_MSG_DELSA:
2659 case XFRM_MSG_NEWSA:
2660 case XFRM_MSG_UPDSA:
2661 return key_notify_sa(x, c);
2662 case XFRM_MSG_FLUSHSA:
2663 return key_notify_sa_flush(c);
2664 default:
2665 printk("pfkey: Unknown SA event %d\n", c->event);
2666 break;
2669 return 0;
2672 static int pfkey_send_policy_notify(struct xfrm_policy *xp, int dir, struct km_event *c)
2674 switch (c->event) {
2675 case XFRM_MSG_POLEXPIRE:
2676 return key_notify_policy_expire(xp, c);
2677 case XFRM_MSG_DELPOLICY:
2678 case XFRM_MSG_NEWPOLICY:
2679 case XFRM_MSG_UPDPOLICY:
2680 return key_notify_policy(xp, dir, c);
2681 case XFRM_MSG_FLUSHPOLICY:
2682 return key_notify_policy_flush(c);
2683 default:
2684 printk("pfkey: Unknown policy event %d\n", c->event);
2685 break;
2688 return 0;
2691 static u32 get_acqseq(void)
2693 u32 res;
2694 static u32 acqseq;
2695 static DEFINE_SPINLOCK(acqseq_lock);
2697 spin_lock_bh(&acqseq_lock);
2698 res = (++acqseq ? : ++acqseq);
2699 spin_unlock_bh(&acqseq_lock);
2700 return res;
2703 static int pfkey_send_acquire(struct xfrm_state *x, struct xfrm_tmpl *t, struct xfrm_policy *xp, int dir)
2705 struct sk_buff *skb;
2706 struct sadb_msg *hdr;
2707 struct sadb_address *addr;
2708 struct sadb_x_policy *pol;
2709 struct sockaddr_in *sin;
2710 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2711 struct sockaddr_in6 *sin6;
2712 #endif
2713 int sockaddr_size;
2714 int size;
2716 sockaddr_size = pfkey_sockaddr_size(x->props.family);
2717 if (!sockaddr_size)
2718 return -EINVAL;
2720 size = sizeof(struct sadb_msg) +
2721 (sizeof(struct sadb_address) * 2) +
2722 (sockaddr_size * 2) +
2723 sizeof(struct sadb_x_policy);
2725 if (x->id.proto == IPPROTO_AH)
2726 size += count_ah_combs(t);
2727 else if (x->id.proto == IPPROTO_ESP)
2728 size += count_esp_combs(t);
2730 skb = alloc_skb(size + 16, GFP_ATOMIC);
2731 if (skb == NULL)
2732 return -ENOMEM;
2734 hdr = (struct sadb_msg *) skb_put(skb, sizeof(struct sadb_msg));
2735 hdr->sadb_msg_version = PF_KEY_V2;
2736 hdr->sadb_msg_type = SADB_ACQUIRE;
2737 hdr->sadb_msg_satype = pfkey_proto2satype(x->id.proto);
2738 hdr->sadb_msg_len = size / sizeof(uint64_t);
2739 hdr->sadb_msg_errno = 0;
2740 hdr->sadb_msg_reserved = 0;
2741 hdr->sadb_msg_seq = x->km.seq = get_acqseq();
2742 hdr->sadb_msg_pid = 0;
2744 /* src address */
2745 addr = (struct sadb_address*) skb_put(skb,
2746 sizeof(struct sadb_address)+sockaddr_size);
2747 addr->sadb_address_len =
2748 (sizeof(struct sadb_address)+sockaddr_size)/
2749 sizeof(uint64_t);
2750 addr->sadb_address_exttype = SADB_EXT_ADDRESS_SRC;
2751 addr->sadb_address_proto = 0;
2752 addr->sadb_address_reserved = 0;
2753 if (x->props.family == AF_INET) {
2754 addr->sadb_address_prefixlen = 32;
2756 sin = (struct sockaddr_in *) (addr + 1);
2757 sin->sin_family = AF_INET;
2758 sin->sin_addr.s_addr = x->props.saddr.a4;
2759 sin->sin_port = 0;
2760 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
2762 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2763 else if (x->props.family == AF_INET6) {
2764 addr->sadb_address_prefixlen = 128;
2766 sin6 = (struct sockaddr_in6 *) (addr + 1);
2767 sin6->sin6_family = AF_INET6;
2768 sin6->sin6_port = 0;
2769 sin6->sin6_flowinfo = 0;
2770 memcpy(&sin6->sin6_addr,
2771 x->props.saddr.a6, sizeof(struct in6_addr));
2772 sin6->sin6_scope_id = 0;
2774 #endif
2775 else
2776 BUG();
2778 /* dst address */
2779 addr = (struct sadb_address*) skb_put(skb,
2780 sizeof(struct sadb_address)+sockaddr_size);
2781 addr->sadb_address_len =
2782 (sizeof(struct sadb_address)+sockaddr_size)/
2783 sizeof(uint64_t);
2784 addr->sadb_address_exttype = SADB_EXT_ADDRESS_DST;
2785 addr->sadb_address_proto = 0;
2786 addr->sadb_address_reserved = 0;
2787 if (x->props.family == AF_INET) {
2788 addr->sadb_address_prefixlen = 32;
2790 sin = (struct sockaddr_in *) (addr + 1);
2791 sin->sin_family = AF_INET;
2792 sin->sin_addr.s_addr = x->id.daddr.a4;
2793 sin->sin_port = 0;
2794 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
2796 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2797 else if (x->props.family == AF_INET6) {
2798 addr->sadb_address_prefixlen = 128;
2800 sin6 = (struct sockaddr_in6 *) (addr + 1);
2801 sin6->sin6_family = AF_INET6;
2802 sin6->sin6_port = 0;
2803 sin6->sin6_flowinfo = 0;
2804 memcpy(&sin6->sin6_addr,
2805 x->id.daddr.a6, sizeof(struct in6_addr));
2806 sin6->sin6_scope_id = 0;
2808 #endif
2809 else
2810 BUG();
2812 pol = (struct sadb_x_policy *) skb_put(skb, sizeof(struct sadb_x_policy));
2813 pol->sadb_x_policy_len = sizeof(struct sadb_x_policy)/sizeof(uint64_t);
2814 pol->sadb_x_policy_exttype = SADB_X_EXT_POLICY;
2815 pol->sadb_x_policy_type = IPSEC_POLICY_IPSEC;
2816 pol->sadb_x_policy_dir = dir+1;
2817 pol->sadb_x_policy_id = xp->index;
2819 /* Set sadb_comb's. */
2820 if (x->id.proto == IPPROTO_AH)
2821 dump_ah_combs(skb, t);
2822 else if (x->id.proto == IPPROTO_ESP)
2823 dump_esp_combs(skb, t);
2825 return pfkey_broadcast(skb, GFP_ATOMIC, BROADCAST_REGISTERED, NULL);
2828 static struct xfrm_policy *pfkey_compile_policy(u16 family, int opt,
2829 u8 *data, int len, int *dir)
2831 struct xfrm_policy *xp;
2832 struct sadb_x_policy *pol = (struct sadb_x_policy*)data;
2833 struct sadb_x_sec_ctx *sec_ctx;
2835 switch (family) {
2836 case AF_INET:
2837 if (opt != IP_IPSEC_POLICY) {
2838 *dir = -EOPNOTSUPP;
2839 return NULL;
2841 break;
2842 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2843 case AF_INET6:
2844 if (opt != IPV6_IPSEC_POLICY) {
2845 *dir = -EOPNOTSUPP;
2846 return NULL;
2848 break;
2849 #endif
2850 default:
2851 *dir = -EINVAL;
2852 return NULL;
2855 *dir = -EINVAL;
2857 if (len < sizeof(struct sadb_x_policy) ||
2858 pol->sadb_x_policy_len*8 > len ||
2859 pol->sadb_x_policy_type > IPSEC_POLICY_BYPASS ||
2860 (!pol->sadb_x_policy_dir || pol->sadb_x_policy_dir > IPSEC_DIR_OUTBOUND))
2861 return NULL;
2863 xp = xfrm_policy_alloc(GFP_ATOMIC);
2864 if (xp == NULL) {
2865 *dir = -ENOBUFS;
2866 return NULL;
2869 xp->action = (pol->sadb_x_policy_type == IPSEC_POLICY_DISCARD ?
2870 XFRM_POLICY_BLOCK : XFRM_POLICY_ALLOW);
2872 xp->lft.soft_byte_limit = XFRM_INF;
2873 xp->lft.hard_byte_limit = XFRM_INF;
2874 xp->lft.soft_packet_limit = XFRM_INF;
2875 xp->lft.hard_packet_limit = XFRM_INF;
2876 xp->family = family;
2878 xp->xfrm_nr = 0;
2879 if (pol->sadb_x_policy_type == IPSEC_POLICY_IPSEC &&
2880 (*dir = parse_ipsecrequests(xp, pol)) < 0)
2881 goto out;
2883 /* security context too */
2884 if (len >= (pol->sadb_x_policy_len*8 +
2885 sizeof(struct sadb_x_sec_ctx))) {
2886 char *p = (char *)pol;
2887 struct xfrm_user_sec_ctx *uctx;
2889 p += pol->sadb_x_policy_len*8;
2890 sec_ctx = (struct sadb_x_sec_ctx *)p;
2891 if (len < pol->sadb_x_policy_len*8 +
2892 sec_ctx->sadb_x_sec_len)
2893 goto out;
2894 if ((*dir = verify_sec_ctx_len(p)))
2895 goto out;
2896 uctx = pfkey_sadb2xfrm_user_sec_ctx(sec_ctx);
2897 *dir = security_xfrm_policy_alloc(xp, uctx);
2898 kfree(uctx);
2900 if (*dir)
2901 goto out;
2904 *dir = pol->sadb_x_policy_dir-1;
2905 return xp;
2907 out:
2908 security_xfrm_policy_free(xp);
2909 kfree(xp);
2910 return NULL;
2913 static int pfkey_send_new_mapping(struct xfrm_state *x, xfrm_address_t *ipaddr, u16 sport)
2915 struct sk_buff *skb;
2916 struct sadb_msg *hdr;
2917 struct sadb_sa *sa;
2918 struct sadb_address *addr;
2919 struct sadb_x_nat_t_port *n_port;
2920 struct sockaddr_in *sin;
2921 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2922 struct sockaddr_in6 *sin6;
2923 #endif
2924 int sockaddr_size;
2925 int size;
2926 __u8 satype = (x->id.proto == IPPROTO_ESP ? SADB_SATYPE_ESP : 0);
2927 struct xfrm_encap_tmpl *natt = NULL;
2929 sockaddr_size = pfkey_sockaddr_size(x->props.family);
2930 if (!sockaddr_size)
2931 return -EINVAL;
2933 if (!satype)
2934 return -EINVAL;
2936 if (!x->encap)
2937 return -EINVAL;
2939 natt = x->encap;
2941 /* Build an SADB_X_NAT_T_NEW_MAPPING message:
2943 * HDR | SA | ADDRESS_SRC (old addr) | NAT_T_SPORT (old port) |
2944 * ADDRESS_DST (new addr) | NAT_T_DPORT (new port)
2947 size = sizeof(struct sadb_msg) +
2948 sizeof(struct sadb_sa) +
2949 (sizeof(struct sadb_address) * 2) +
2950 (sockaddr_size * 2) +
2951 (sizeof(struct sadb_x_nat_t_port) * 2);
2953 skb = alloc_skb(size + 16, GFP_ATOMIC);
2954 if (skb == NULL)
2955 return -ENOMEM;
2957 hdr = (struct sadb_msg *) skb_put(skb, sizeof(struct sadb_msg));
2958 hdr->sadb_msg_version = PF_KEY_V2;
2959 hdr->sadb_msg_type = SADB_X_NAT_T_NEW_MAPPING;
2960 hdr->sadb_msg_satype = satype;
2961 hdr->sadb_msg_len = size / sizeof(uint64_t);
2962 hdr->sadb_msg_errno = 0;
2963 hdr->sadb_msg_reserved = 0;
2964 hdr->sadb_msg_seq = x->km.seq = get_acqseq();
2965 hdr->sadb_msg_pid = 0;
2967 /* SA */
2968 sa = (struct sadb_sa *) skb_put(skb, sizeof(struct sadb_sa));
2969 sa->sadb_sa_len = sizeof(struct sadb_sa)/sizeof(uint64_t);
2970 sa->sadb_sa_exttype = SADB_EXT_SA;
2971 sa->sadb_sa_spi = x->id.spi;
2972 sa->sadb_sa_replay = 0;
2973 sa->sadb_sa_state = 0;
2974 sa->sadb_sa_auth = 0;
2975 sa->sadb_sa_encrypt = 0;
2976 sa->sadb_sa_flags = 0;
2978 /* ADDRESS_SRC (old addr) */
2979 addr = (struct sadb_address*)
2980 skb_put(skb, sizeof(struct sadb_address)+sockaddr_size);
2981 addr->sadb_address_len =
2982 (sizeof(struct sadb_address)+sockaddr_size)/
2983 sizeof(uint64_t);
2984 addr->sadb_address_exttype = SADB_EXT_ADDRESS_SRC;
2985 addr->sadb_address_proto = 0;
2986 addr->sadb_address_reserved = 0;
2987 if (x->props.family == AF_INET) {
2988 addr->sadb_address_prefixlen = 32;
2990 sin = (struct sockaddr_in *) (addr + 1);
2991 sin->sin_family = AF_INET;
2992 sin->sin_addr.s_addr = x->props.saddr.a4;
2993 sin->sin_port = 0;
2994 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
2996 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
2997 else if (x->props.family == AF_INET6) {
2998 addr->sadb_address_prefixlen = 128;
3000 sin6 = (struct sockaddr_in6 *) (addr + 1);
3001 sin6->sin6_family = AF_INET6;
3002 sin6->sin6_port = 0;
3003 sin6->sin6_flowinfo = 0;
3004 memcpy(&sin6->sin6_addr,
3005 x->props.saddr.a6, sizeof(struct in6_addr));
3006 sin6->sin6_scope_id = 0;
3008 #endif
3009 else
3010 BUG();
3012 /* NAT_T_SPORT (old port) */
3013 n_port = (struct sadb_x_nat_t_port*) skb_put(skb, sizeof (*n_port));
3014 n_port->sadb_x_nat_t_port_len = sizeof(*n_port)/sizeof(uint64_t);
3015 n_port->sadb_x_nat_t_port_exttype = SADB_X_EXT_NAT_T_SPORT;
3016 n_port->sadb_x_nat_t_port_port = natt->encap_sport;
3017 n_port->sadb_x_nat_t_port_reserved = 0;
3019 /* ADDRESS_DST (new addr) */
3020 addr = (struct sadb_address*)
3021 skb_put(skb, sizeof(struct sadb_address)+sockaddr_size);
3022 addr->sadb_address_len =
3023 (sizeof(struct sadb_address)+sockaddr_size)/
3024 sizeof(uint64_t);
3025 addr->sadb_address_exttype = SADB_EXT_ADDRESS_DST;
3026 addr->sadb_address_proto = 0;
3027 addr->sadb_address_reserved = 0;
3028 if (x->props.family == AF_INET) {
3029 addr->sadb_address_prefixlen = 32;
3031 sin = (struct sockaddr_in *) (addr + 1);
3032 sin->sin_family = AF_INET;
3033 sin->sin_addr.s_addr = ipaddr->a4;
3034 sin->sin_port = 0;
3035 memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
3037 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
3038 else if (x->props.family == AF_INET6) {
3039 addr->sadb_address_prefixlen = 128;
3041 sin6 = (struct sockaddr_in6 *) (addr + 1);
3042 sin6->sin6_family = AF_INET6;
3043 sin6->sin6_port = 0;
3044 sin6->sin6_flowinfo = 0;
3045 memcpy(&sin6->sin6_addr, &ipaddr->a6, sizeof(struct in6_addr));
3046 sin6->sin6_scope_id = 0;
3048 #endif
3049 else
3050 BUG();
3052 /* NAT_T_DPORT (new port) */
3053 n_port = (struct sadb_x_nat_t_port*) skb_put(skb, sizeof (*n_port));
3054 n_port->sadb_x_nat_t_port_len = sizeof(*n_port)/sizeof(uint64_t);
3055 n_port->sadb_x_nat_t_port_exttype = SADB_X_EXT_NAT_T_DPORT;
3056 n_port->sadb_x_nat_t_port_port = sport;
3057 n_port->sadb_x_nat_t_port_reserved = 0;
3059 return pfkey_broadcast(skb, GFP_ATOMIC, BROADCAST_REGISTERED, NULL);
3062 static int pfkey_sendmsg(struct kiocb *kiocb,
3063 struct socket *sock, struct msghdr *msg, size_t len)
3065 struct sock *sk = sock->sk;
3066 struct sk_buff *skb = NULL;
3067 struct sadb_msg *hdr = NULL;
3068 int err;
3070 err = -EOPNOTSUPP;
3071 if (msg->msg_flags & MSG_OOB)
3072 goto out;
3074 err = -EMSGSIZE;
3075 if ((unsigned)len > sk->sk_sndbuf - 32)
3076 goto out;
3078 err = -ENOBUFS;
3079 skb = alloc_skb(len, GFP_KERNEL);
3080 if (skb == NULL)
3081 goto out;
3083 err = -EFAULT;
3084 if (memcpy_fromiovec(skb_put(skb,len), msg->msg_iov, len))
3085 goto out;
3087 hdr = pfkey_get_base_msg(skb, &err);
3088 if (!hdr)
3089 goto out;
3091 down(&xfrm_cfg_sem);
3092 err = pfkey_process(sk, skb, hdr);
3093 up(&xfrm_cfg_sem);
3095 out:
3096 if (err && hdr && pfkey_error(hdr, err, sk) == 0)
3097 err = 0;
3098 if (skb)
3099 kfree_skb(skb);
3101 return err ? : len;
3104 static int pfkey_recvmsg(struct kiocb *kiocb,
3105 struct socket *sock, struct msghdr *msg, size_t len,
3106 int flags)
3108 struct sock *sk = sock->sk;
3109 struct sk_buff *skb;
3110 int copied, err;
3112 err = -EINVAL;
3113 if (flags & ~(MSG_PEEK|MSG_DONTWAIT|MSG_TRUNC|MSG_CMSG_COMPAT))
3114 goto out;
3116 msg->msg_namelen = 0;
3117 skb = skb_recv_datagram(sk, flags, flags & MSG_DONTWAIT, &err);
3118 if (skb == NULL)
3119 goto out;
3121 copied = skb->len;
3122 if (copied > len) {
3123 msg->msg_flags |= MSG_TRUNC;
3124 copied = len;
3127 skb->h.raw = skb->data;
3128 err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
3129 if (err)
3130 goto out_free;
3132 sock_recv_timestamp(msg, sk, skb);
3134 err = (flags & MSG_TRUNC) ? skb->len : copied;
3136 out_free:
3137 skb_free_datagram(sk, skb);
3138 out:
3139 return err;
3142 static const struct proto_ops pfkey_ops = {
3143 .family = PF_KEY,
3144 .owner = THIS_MODULE,
3145 /* Operations that make no sense on pfkey sockets. */
3146 .bind = sock_no_bind,
3147 .connect = sock_no_connect,
3148 .socketpair = sock_no_socketpair,
3149 .accept = sock_no_accept,
3150 .getname = sock_no_getname,
3151 .ioctl = sock_no_ioctl,
3152 .listen = sock_no_listen,
3153 .shutdown = sock_no_shutdown,
3154 .setsockopt = sock_no_setsockopt,
3155 .getsockopt = sock_no_getsockopt,
3156 .mmap = sock_no_mmap,
3157 .sendpage = sock_no_sendpage,
3159 /* Now the operations that really occur. */
3160 .release = pfkey_release,
3161 .poll = datagram_poll,
3162 .sendmsg = pfkey_sendmsg,
3163 .recvmsg = pfkey_recvmsg,
3166 static struct net_proto_family pfkey_family_ops = {
3167 .family = PF_KEY,
3168 .create = pfkey_create,
3169 .owner = THIS_MODULE,
3172 #ifdef CONFIG_PROC_FS
3173 static int pfkey_read_proc(char *buffer, char **start, off_t offset,
3174 int length, int *eof, void *data)
3176 off_t pos = 0;
3177 off_t begin = 0;
3178 int len = 0;
3179 struct sock *s;
3180 struct hlist_node *node;
3182 len += sprintf(buffer,"sk RefCnt Rmem Wmem User Inode\n");
3184 read_lock(&pfkey_table_lock);
3186 sk_for_each(s, node, &pfkey_table) {
3187 len += sprintf(buffer+len,"%p %-6d %-6u %-6u %-6u %-6lu",
3189 atomic_read(&s->sk_refcnt),
3190 atomic_read(&s->sk_rmem_alloc),
3191 atomic_read(&s->sk_wmem_alloc),
3192 sock_i_uid(s),
3193 sock_i_ino(s)
3196 buffer[len++] = '\n';
3198 pos = begin + len;
3199 if (pos < offset) {
3200 len = 0;
3201 begin = pos;
3203 if(pos > offset + length)
3204 goto done;
3206 *eof = 1;
3208 done:
3209 read_unlock(&pfkey_table_lock);
3211 *start = buffer + (offset - begin);
3212 len -= (offset - begin);
3214 if (len > length)
3215 len = length;
3216 if (len < 0)
3217 len = 0;
3219 return len;
3221 #endif
3223 static struct xfrm_mgr pfkeyv2_mgr =
3225 .id = "pfkeyv2",
3226 .notify = pfkey_send_notify,
3227 .acquire = pfkey_send_acquire,
3228 .compile_policy = pfkey_compile_policy,
3229 .new_mapping = pfkey_send_new_mapping,
3230 .notify_policy = pfkey_send_policy_notify,
3233 static void __exit ipsec_pfkey_exit(void)
3235 xfrm_unregister_km(&pfkeyv2_mgr);
3236 remove_proc_entry("net/pfkey", NULL);
3237 sock_unregister(PF_KEY);
3238 proto_unregister(&key_proto);
3241 static int __init ipsec_pfkey_init(void)
3243 int err = proto_register(&key_proto, 0);
3245 if (err != 0)
3246 goto out;
3248 err = sock_register(&pfkey_family_ops);
3249 if (err != 0)
3250 goto out_unregister_key_proto;
3251 #ifdef CONFIG_PROC_FS
3252 err = -ENOMEM;
3253 if (create_proc_read_entry("net/pfkey", 0, NULL, pfkey_read_proc, NULL) == NULL)
3254 goto out_sock_unregister;
3255 #endif
3256 err = xfrm_register_km(&pfkeyv2_mgr);
3257 if (err != 0)
3258 goto out_remove_proc_entry;
3259 out:
3260 return err;
3261 out_remove_proc_entry:
3262 #ifdef CONFIG_PROC_FS
3263 remove_proc_entry("net/pfkey", NULL);
3264 out_sock_unregister:
3265 #endif
3266 sock_unregister(PF_KEY);
3267 out_unregister_key_proto:
3268 proto_unregister(&key_proto);
3269 goto out;
3272 module_init(ipsec_pfkey_init);
3273 module_exit(ipsec_pfkey_exit);
3274 MODULE_LICENSE("GPL");
3275 MODULE_ALIAS_NETPROTO(PF_KEY);