[IPSEC]: Store IPv6 nh pointer in mac_header on output
[safe/jmp/linux-2.6] / net / ipv6 / xfrm6_mode_tunnel.c
1 /*
2  * xfrm6_mode_tunnel.c - Tunnel 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/dsfield.h>
14 #include <net/dst.h>
15 #include <net/inet_ecn.h>
16 #include <net/ipv6.h>
17 #include <net/xfrm.h>
18
19 static inline void ipip6_ecn_decapsulate(struct sk_buff *skb)
20 {
21         struct ipv6hdr *outer_iph = ipv6_hdr(skb);
22         struct ipv6hdr *inner_iph = ipipv6_hdr(skb);
23
24         if (INET_ECN_is_ce(ipv6_get_dsfield(outer_iph)))
25                 IP6_ECN_set_ce(inner_iph);
26 }
27
28 static inline void ip6ip_ecn_decapsulate(struct sk_buff *skb)
29 {
30         if (INET_ECN_is_ce(ipv6_get_dsfield(ipv6_hdr(skb))))
31                         IP_ECN_set_ce(ipip_hdr(skb));
32 }
33
34 /* Add encapsulation header.
35  *
36  * The top IP header will be constructed per RFC 2401.  The following fields
37  * in it shall be filled in by x->type->output:
38  *      payload_len
39  *
40  * On exit, skb->h will be set to the start of the encapsulation header to be
41  * filled in by x->type->output and the mac header will be set to the
42  * nextheader field of the extension header directly preceding the
43  * encapsulation header, or in its absence, that of the top IP header.
44  * The value of skb->data and the network header will always point to the
45  * top IP header.
46  */
47 static int xfrm6_tunnel_output(struct xfrm_state *x, struct sk_buff *skb)
48 {
49         struct dst_entry *dst = skb->dst;
50         struct xfrm_dst *xdst = (struct xfrm_dst*)dst;
51         struct ipv6hdr *iph, *top_iph;
52         int dsfield;
53
54         skb_push(skb, x->props.header_len);
55         iph = ipv6_hdr(skb);
56
57         skb_set_mac_header(skb, offsetof(struct ipv6hdr, nexthdr));
58         skb_reset_network_header(skb);
59         skb_set_transport_header(skb, sizeof(struct ipv6hdr));
60         top_iph = ipv6_hdr(skb);
61
62         top_iph->version = 6;
63         if (xdst->route->ops->family == AF_INET6) {
64                 top_iph->priority = iph->priority;
65                 top_iph->flow_lbl[0] = iph->flow_lbl[0];
66                 top_iph->flow_lbl[1] = iph->flow_lbl[1];
67                 top_iph->flow_lbl[2] = iph->flow_lbl[2];
68                 top_iph->nexthdr = IPPROTO_IPV6;
69         } else {
70                 top_iph->priority = 0;
71                 top_iph->flow_lbl[0] = 0;
72                 top_iph->flow_lbl[1] = 0;
73                 top_iph->flow_lbl[2] = 0;
74                 top_iph->nexthdr = IPPROTO_IPIP;
75         }
76         dsfield = ipv6_get_dsfield(top_iph);
77         dsfield = INET_ECN_encapsulate(dsfield, dsfield);
78         if (x->props.flags & XFRM_STATE_NOECN)
79                 dsfield &= ~INET_ECN_MASK;
80         ipv6_change_dsfield(top_iph, 0, dsfield);
81         top_iph->hop_limit = dst_metric(dst->child, RTAX_HOPLIMIT);
82         ipv6_addr_copy(&top_iph->saddr, (struct in6_addr *)&x->props.saddr);
83         ipv6_addr_copy(&top_iph->daddr, (struct in6_addr *)&x->id.daddr);
84         skb->protocol = htons(ETH_P_IPV6);
85         return 0;
86 }
87
88 static int xfrm6_tunnel_input(struct xfrm_state *x, struct sk_buff *skb)
89 {
90         int err = -EINVAL;
91         const unsigned char *old_mac;
92         const unsigned char *nh = skb_network_header(skb);
93
94         if (nh[IP6CB(skb)->nhoff] != IPPROTO_IPV6 &&
95             nh[IP6CB(skb)->nhoff] != IPPROTO_IPIP)
96                 goto out;
97         if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
98                 goto out;
99
100         if (skb_cloned(skb) &&
101             (err = pskb_expand_head(skb, 0, 0, GFP_ATOMIC)))
102                 goto out;
103
104         nh = skb_network_header(skb);
105         if (nh[IP6CB(skb)->nhoff] == IPPROTO_IPV6) {
106                 if (x->props.flags & XFRM_STATE_DECAP_DSCP)
107                         ipv6_copy_dscp(ipv6_hdr(skb), ipipv6_hdr(skb));
108                 if (!(x->props.flags & XFRM_STATE_NOECN))
109                         ipip6_ecn_decapsulate(skb);
110         } else {
111                 if (!(x->props.flags & XFRM_STATE_NOECN))
112                         ip6ip_ecn_decapsulate(skb);
113                 skb->protocol = htons(ETH_P_IP);
114         }
115         old_mac = skb_mac_header(skb);
116         skb_set_mac_header(skb, -skb->mac_len);
117         memmove(skb_mac_header(skb), old_mac, skb->mac_len);
118         skb_reset_network_header(skb);
119         err = 0;
120
121 out:
122         return err;
123 }
124
125 static struct xfrm_mode xfrm6_tunnel_mode = {
126         .input = xfrm6_tunnel_input,
127         .output = xfrm6_tunnel_output,
128         .owner = THIS_MODULE,
129         .encap = XFRM_MODE_TUNNEL,
130 };
131
132 static int __init xfrm6_tunnel_init(void)
133 {
134         return xfrm_register_mode(&xfrm6_tunnel_mode, AF_INET6);
135 }
136
137 static void __exit xfrm6_tunnel_exit(void)
138 {
139         int err;
140
141         err = xfrm_unregister_mode(&xfrm6_tunnel_mode, AF_INET6);
142         BUG_ON(err);
143 }
144
145 module_init(xfrm6_tunnel_init);
146 module_exit(xfrm6_tunnel_exit);
147 MODULE_LICENSE("GPL");
148 MODULE_ALIAS_XFRM_MODE(AF_INET6, XFRM_MODE_TUNNEL);