Merge master.kernel.org:/pub/scm/linux/kernel/git/lethal/sh-2.6
[linux-2.6] / net / ipv4 / xfrm4_mode_tunnel.c
1 /*
2  * xfrm4_mode_tunnel.c - Tunnel mode encapsulation for IPv4.
3  *
4  * Copyright (c) 2004-2006 Herbert Xu <herbert@gondor.apana.org.au>
5  */
6
7 #include <linux/init.h>
8 #include <linux/kernel.h>
9 #include <linux/module.h>
10 #include <linux/skbuff.h>
11 #include <linux/stringify.h>
12 #include <net/dst.h>
13 #include <net/inet_ecn.h>
14 #include <net/ip.h>
15 #include <net/xfrm.h>
16
17 static inline void ipip_ecn_decapsulate(struct sk_buff *skb)
18 {
19         struct iphdr *outer_iph = ip_hdr(skb);
20         struct iphdr *inner_iph = ipip_hdr(skb);
21
22         if (INET_ECN_is_ce(outer_iph->tos))
23                 IP_ECN_set_ce(inner_iph);
24 }
25
26 static inline void ipip6_ecn_decapsulate(struct iphdr *iph, struct sk_buff *skb)
27 {
28         if (INET_ECN_is_ce(iph->tos))
29                 IP6_ECN_set_ce(ipv6_hdr(skb));
30 }
31
32 /* Add encapsulation header.
33  *
34  * The top IP header will be constructed per RFC 2401.  The following fields
35  * in it shall be filled in by x->type->output:
36  *      tot_len
37  *      check
38  *
39  * On exit, skb->h will be set to the start of the payload to be processed
40  * by x->type->output and skb->nh will be set to the top IP header.
41  */
42 static int xfrm4_tunnel_output(struct xfrm_state *x, struct sk_buff *skb)
43 {
44         struct dst_entry *dst = skb->dst;
45         struct xfrm_dst *xdst = (struct xfrm_dst*)dst;
46         struct iphdr *iph, *top_iph;
47         int flags;
48
49         iph = ip_hdr(skb);
50         skb->transport_header = skb->network_header;
51
52         skb_push(skb, x->props.header_len);
53         skb_reset_network_header(skb);
54         top_iph = ip_hdr(skb);
55
56         top_iph->ihl = 5;
57         top_iph->version = 4;
58
59         flags = x->props.flags;
60
61         /* DS disclosed */
62         if (xdst->route->ops->family == AF_INET) {
63                 top_iph->protocol = IPPROTO_IPIP;
64                 top_iph->tos = INET_ECN_encapsulate(iph->tos, iph->tos);
65                 top_iph->frag_off = (flags & XFRM_STATE_NOPMTUDISC) ?
66                         0 : (iph->frag_off & htons(IP_DF));
67         }
68 #if defined(CONFIG_IPV6) || defined (CONFIG_IPV6_MODULE)
69         else {
70                 struct ipv6hdr *ipv6h = (struct ipv6hdr*)iph;
71                 top_iph->protocol = IPPROTO_IPV6;
72                 top_iph->tos = INET_ECN_encapsulate(iph->tos, ipv6_get_dsfield(ipv6h));
73                 top_iph->frag_off = 0;
74         }
75 #endif
76
77         if (flags & XFRM_STATE_NOECN)
78                 IP_ECN_clear(top_iph);
79
80         if (!top_iph->frag_off)
81                 __ip_select_ident(top_iph, dst->child, 0);
82
83         top_iph->ttl = dst_metric(dst->child, RTAX_HOPLIMIT);
84
85         top_iph->saddr = x->props.saddr.a4;
86         top_iph->daddr = x->id.daddr.a4;
87
88         skb->protocol = htons(ETH_P_IP);
89
90         memset(&(IPCB(skb)->opt), 0, sizeof(struct ip_options));
91         return 0;
92 }
93
94 static int xfrm4_tunnel_input(struct xfrm_state *x, struct sk_buff *skb)
95 {
96         struct iphdr *iph = ip_hdr(skb);
97         const unsigned char *old_mac;
98         int err = -EINVAL;
99
100         switch (iph->protocol){
101                 case IPPROTO_IPIP:
102                         break;
103 #if defined(CONFIG_IPV6) || defined (CONFIG_IPV6_MODULE)
104                 case IPPROTO_IPV6:
105                         break;
106 #endif
107                 default:
108                         goto out;
109         }
110
111         if (!pskb_may_pull(skb, sizeof(struct iphdr)))
112                 goto out;
113
114         if (skb_cloned(skb) &&
115             (err = pskb_expand_head(skb, 0, 0, GFP_ATOMIC)))
116                 goto out;
117
118         iph = ip_hdr(skb);
119         if (iph->protocol == IPPROTO_IPIP) {
120                 if (x->props.flags & XFRM_STATE_DECAP_DSCP)
121                         ipv4_copy_dscp(iph, ipip_hdr(skb));
122                 if (!(x->props.flags & XFRM_STATE_NOECN))
123                         ipip_ecn_decapsulate(skb);
124         }
125 #if defined(CONFIG_IPV6) || defined (CONFIG_IPV6_MODULE)
126         else {
127                 if (!(x->props.flags & XFRM_STATE_NOECN))
128                         ipip6_ecn_decapsulate(iph, skb);
129                 skb->protocol = htons(ETH_P_IPV6);
130         }
131 #endif
132         old_mac = skb_mac_header(skb);
133         skb_set_mac_header(skb, -skb->mac_len);
134         memmove(skb_mac_header(skb), old_mac, skb->mac_len);
135         skb_reset_network_header(skb);
136         err = 0;
137
138 out:
139         return err;
140 }
141
142 static struct xfrm_mode xfrm4_tunnel_mode = {
143         .input = xfrm4_tunnel_input,
144         .output = xfrm4_tunnel_output,
145         .owner = THIS_MODULE,
146         .encap = XFRM_MODE_TUNNEL,
147 };
148
149 static int __init xfrm4_tunnel_init(void)
150 {
151         return xfrm_register_mode(&xfrm4_tunnel_mode, AF_INET);
152 }
153
154 static void __exit xfrm4_tunnel_exit(void)
155 {
156         int err;
157
158         err = xfrm_unregister_mode(&xfrm4_tunnel_mode, AF_INET);
159         BUG_ON(err);
160 }
161
162 module_init(xfrm4_tunnel_init);
163 module_exit(xfrm4_tunnel_exit);
164 MODULE_LICENSE("GPL");
165 MODULE_ALIAS_XFRM_MODE(AF_INET, XFRM_MODE_TUNNEL);