[SK_BUFF]: Introduce skb_set_transport_header
[linux-2.6] / net / ipv6 / xfrm6_mode_transport.c
1 /*
2  * xfrm6_mode_transport.c - Transport mode encapsulation for IPv6.
3  *
4  * Copyright (C) 2002 USAGI/WIDE Project
5  * Copyright (c) 2004-2006 Herbert Xu <herbert@gondor.apana.org.au>
6  */
7
8 #include <linux/init.h>
9 #include <linux/kernel.h>
10 #include <linux/module.h>
11 #include <linux/skbuff.h>
12 #include <linux/stringify.h>
13 #include <net/dst.h>
14 #include <net/ipv6.h>
15 #include <net/xfrm.h>
16
17 /* Add encapsulation header.
18  *
19  * The IP header and mutable extension headers will be moved forward to make
20  * space for the encapsulation header.
21  *
22  * On exit, skb->h will be set to the start of the encapsulation header to be
23  * filled in by x->type->output and skb->nh will be set to the nextheader field
24  * of the extension header directly preceding the encapsulation header, or in
25  * its absence, that of the top IP header.  The value of skb->data will always
26  * point to the top IP header.
27  */
28 static int xfrm6_transport_output(struct xfrm_state *x, struct sk_buff *skb)
29 {
30         struct ipv6hdr *iph;
31         u8 *prevhdr;
32         int hdr_len;
33
34         skb_push(skb, x->props.header_len);
35         iph = ipv6_hdr(skb);
36
37         hdr_len = x->type->hdr_offset(x, skb, &prevhdr);
38         skb->nh.raw = prevhdr - x->props.header_len;
39         skb_set_transport_header(skb, hdr_len);
40         memmove(skb->data, iph, hdr_len);
41         return 0;
42 }
43
44 /* Remove encapsulation header.
45  *
46  * The IP header will be moved over the top of the encapsulation header.
47  *
48  * On entry, skb->h shall point to where the IP header should be and skb->nh
49  * shall be set to where the IP header currently is.  skb->data shall point
50  * to the start of the payload.
51  */
52 static int xfrm6_transport_input(struct xfrm_state *x, struct sk_buff *skb)
53 {
54         int ihl = skb->data - skb->h.raw;
55
56         if (skb->h.raw != skb->nh.raw) {
57                 memmove(skb->h.raw, skb_network_header(skb), ihl);
58                 skb->nh.raw = skb->h.raw;
59         }
60         ipv6_hdr(skb)->payload_len = htons(skb->len + ihl -
61                                            sizeof(struct ipv6hdr));
62         skb_reset_transport_header(skb);
63         return 0;
64 }
65
66 static struct xfrm_mode xfrm6_transport_mode = {
67         .input = xfrm6_transport_input,
68         .output = xfrm6_transport_output,
69         .owner = THIS_MODULE,
70         .encap = XFRM_MODE_TRANSPORT,
71 };
72
73 static int __init xfrm6_transport_init(void)
74 {
75         return xfrm_register_mode(&xfrm6_transport_mode, AF_INET6);
76 }
77
78 static void __exit xfrm6_transport_exit(void)
79 {
80         int err;
81
82         err = xfrm_unregister_mode(&xfrm6_transport_mode, AF_INET6);
83         BUG_ON(err);
84 }
85
86 module_init(xfrm6_transport_init);
87 module_exit(xfrm6_transport_exit);
88 MODULE_LICENSE("GPL");
89 MODULE_ALIAS_XFRM_MODE(AF_INET6, XFRM_MODE_TRANSPORT);