[NETFILTER]: nf_conntrack: use extension infrastructure for helper
[safe/jmp/linux-2.6] / net / ipv4 / netfilter / nf_nat_standalone.c
1 /* (C) 1999-2001 Paul `Rusty' Russell
2  * (C) 2002-2006 Netfilter Core Team <coreteam@netfilter.org>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License version 2 as
6  * published by the Free Software Foundation.
7  */
8 #include <linux/types.h>
9 #include <linux/icmp.h>
10 #include <linux/ip.h>
11 #include <linux/netfilter.h>
12 #include <linux/netfilter_ipv4.h>
13 #include <linux/module.h>
14 #include <linux/skbuff.h>
15 #include <linux/proc_fs.h>
16 #include <net/ip.h>
17 #include <net/checksum.h>
18 #include <linux/spinlock.h>
19
20 #include <net/netfilter/nf_conntrack.h>
21 #include <net/netfilter/nf_conntrack_core.h>
22 #include <net/netfilter/nf_nat.h>
23 #include <net/netfilter/nf_nat_rule.h>
24 #include <net/netfilter/nf_nat_protocol.h>
25 #include <net/netfilter/nf_nat_core.h>
26 #include <net/netfilter/nf_nat_helper.h>
27 #include <linux/netfilter_ipv4/ip_tables.h>
28
29 #if 0
30 #define DEBUGP printk
31 #else
32 #define DEBUGP(format, args...)
33 #endif
34
35 #ifdef CONFIG_XFRM
36 static void nat_decode_session(struct sk_buff *skb, struct flowi *fl)
37 {
38         struct nf_conn *ct;
39         struct nf_conntrack_tuple *t;
40         enum ip_conntrack_info ctinfo;
41         enum ip_conntrack_dir dir;
42         unsigned long statusbit;
43
44         ct = nf_ct_get(skb, &ctinfo);
45         if (ct == NULL)
46                 return;
47         dir = CTINFO2DIR(ctinfo);
48         t = &ct->tuplehash[dir].tuple;
49
50         if (dir == IP_CT_DIR_ORIGINAL)
51                 statusbit = IPS_DST_NAT;
52         else
53                 statusbit = IPS_SRC_NAT;
54
55         if (ct->status & statusbit) {
56                 fl->fl4_dst = t->dst.u3.ip;
57                 if (t->dst.protonum == IPPROTO_TCP ||
58                     t->dst.protonum == IPPROTO_UDP)
59                         fl->fl_ip_dport = t->dst.u.tcp.port;
60         }
61
62         statusbit ^= IPS_NAT_MASK;
63
64         if (ct->status & statusbit) {
65                 fl->fl4_src = t->src.u3.ip;
66                 if (t->dst.protonum == IPPROTO_TCP ||
67                     t->dst.protonum == IPPROTO_UDP)
68                         fl->fl_ip_sport = t->src.u.tcp.port;
69         }
70 }
71 #endif
72
73 static unsigned int
74 nf_nat_fn(unsigned int hooknum,
75           struct sk_buff **pskb,
76           const struct net_device *in,
77           const struct net_device *out,
78           int (*okfn)(struct sk_buff *))
79 {
80         struct nf_conn *ct;
81         enum ip_conntrack_info ctinfo;
82         struct nf_conn_nat *nat;
83         /* maniptype == SRC for postrouting. */
84         enum nf_nat_manip_type maniptype = HOOK2MANIP(hooknum);
85
86         /* We never see fragments: conntrack defrags on pre-routing
87            and local-out, and nf_nat_out protects post-routing. */
88         NF_CT_ASSERT(!(ip_hdr(*pskb)->frag_off & htons(IP_MF | IP_OFFSET)));
89
90         ct = nf_ct_get(*pskb, &ctinfo);
91         /* Can't track?  It's not due to stress, or conntrack would
92            have dropped it.  Hence it's the user's responsibilty to
93            packet filter it out, or implement conntrack/NAT for that
94            protocol. 8) --RR */
95         if (!ct) {
96                 /* Exception: ICMP redirect to new connection (not in
97                    hash table yet).  We must not let this through, in
98                    case we're doing NAT to the same network. */
99                 if (ip_hdr(*pskb)->protocol == IPPROTO_ICMP) {
100                         struct icmphdr _hdr, *hp;
101
102                         hp = skb_header_pointer(*pskb, ip_hdrlen(*pskb),
103                                                 sizeof(_hdr), &_hdr);
104                         if (hp != NULL &&
105                             hp->type == ICMP_REDIRECT)
106                                 return NF_DROP;
107                 }
108                 return NF_ACCEPT;
109         }
110
111         /* Don't try to NAT if this packet is not conntracked */
112         if (ct == &nf_conntrack_untracked)
113                 return NF_ACCEPT;
114
115         nat = nfct_nat(ct);
116         if (!nat)
117                 return NF_ACCEPT;
118
119         switch (ctinfo) {
120         case IP_CT_RELATED:
121         case IP_CT_RELATED+IP_CT_IS_REPLY:
122                 if (ip_hdr(*pskb)->protocol == IPPROTO_ICMP) {
123                         if (!nf_nat_icmp_reply_translation(ct, ctinfo,
124                                                            hooknum, pskb))
125                                 return NF_DROP;
126                         else
127                                 return NF_ACCEPT;
128                 }
129                 /* Fall thru... (Only ICMPs can be IP_CT_IS_REPLY) */
130         case IP_CT_NEW:
131
132                 /* Seen it before?  This can happen for loopback, retrans,
133                    or local packets.. */
134                 if (!nf_nat_initialized(ct, maniptype)) {
135                         unsigned int ret;
136
137                         if (unlikely(nf_ct_is_confirmed(ct)))
138                                 /* NAT module was loaded late */
139                                 ret = alloc_null_binding_confirmed(ct, hooknum);
140                         else if (hooknum == NF_IP_LOCAL_IN)
141                                 /* LOCAL_IN hook doesn't have a chain!  */
142                                 ret = alloc_null_binding(ct, hooknum);
143                         else
144                                 ret = nf_nat_rule_find(pskb, hooknum, in, out,
145                                                        ct);
146
147                         if (ret != NF_ACCEPT) {
148                                 return ret;
149                         }
150                 } else
151                         DEBUGP("Already setup manip %s for ct %p\n",
152                                maniptype == IP_NAT_MANIP_SRC ? "SRC" : "DST",
153                                ct);
154                 break;
155
156         default:
157                 /* ESTABLISHED */
158                 NF_CT_ASSERT(ctinfo == IP_CT_ESTABLISHED ||
159                              ctinfo == (IP_CT_ESTABLISHED+IP_CT_IS_REPLY));
160         }
161
162         return nf_nat_packet(ct, ctinfo, hooknum, pskb);
163 }
164
165 static unsigned int
166 nf_nat_in(unsigned int hooknum,
167           struct sk_buff **pskb,
168           const struct net_device *in,
169           const struct net_device *out,
170           int (*okfn)(struct sk_buff *))
171 {
172         unsigned int ret;
173         __be32 daddr = ip_hdr(*pskb)->daddr;
174
175         ret = nf_nat_fn(hooknum, pskb, in, out, okfn);
176         if (ret != NF_DROP && ret != NF_STOLEN &&
177             daddr != ip_hdr(*pskb)->daddr) {
178                 dst_release((*pskb)->dst);
179                 (*pskb)->dst = NULL;
180         }
181         return ret;
182 }
183
184 static unsigned int
185 nf_nat_out(unsigned int hooknum,
186            struct sk_buff **pskb,
187            const struct net_device *in,
188            const struct net_device *out,
189            int (*okfn)(struct sk_buff *))
190 {
191 #ifdef CONFIG_XFRM
192         struct nf_conn *ct;
193         enum ip_conntrack_info ctinfo;
194 #endif
195         unsigned int ret;
196
197         /* root is playing with raw sockets. */
198         if ((*pskb)->len < sizeof(struct iphdr) ||
199             ip_hdrlen(*pskb) < sizeof(struct iphdr))
200                 return NF_ACCEPT;
201
202         ret = nf_nat_fn(hooknum, pskb, in, out, okfn);
203 #ifdef CONFIG_XFRM
204         if (ret != NF_DROP && ret != NF_STOLEN &&
205             (ct = nf_ct_get(*pskb, &ctinfo)) != NULL) {
206                 enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
207
208                 if (ct->tuplehash[dir].tuple.src.u3.ip !=
209                     ct->tuplehash[!dir].tuple.dst.u3.ip
210                     || ct->tuplehash[dir].tuple.src.u.all !=
211                        ct->tuplehash[!dir].tuple.dst.u.all
212                     )
213                         return ip_xfrm_me_harder(pskb) == 0 ? ret : NF_DROP;
214         }
215 #endif
216         return ret;
217 }
218
219 static unsigned int
220 nf_nat_local_fn(unsigned int hooknum,
221                 struct sk_buff **pskb,
222                 const struct net_device *in,
223                 const struct net_device *out,
224                 int (*okfn)(struct sk_buff *))
225 {
226         struct nf_conn *ct;
227         enum ip_conntrack_info ctinfo;
228         unsigned int ret;
229
230         /* root is playing with raw sockets. */
231         if ((*pskb)->len < sizeof(struct iphdr) ||
232             ip_hdrlen(*pskb) < sizeof(struct iphdr))
233                 return NF_ACCEPT;
234
235         ret = nf_nat_fn(hooknum, pskb, in, out, okfn);
236         if (ret != NF_DROP && ret != NF_STOLEN &&
237             (ct = nf_ct_get(*pskb, &ctinfo)) != NULL) {
238                 enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
239
240                 if (ct->tuplehash[dir].tuple.dst.u3.ip !=
241                     ct->tuplehash[!dir].tuple.src.u3.ip) {
242                         if (ip_route_me_harder(pskb, RTN_UNSPEC))
243                                 ret = NF_DROP;
244                 }
245 #ifdef CONFIG_XFRM
246                 else if (ct->tuplehash[dir].tuple.dst.u.all !=
247                          ct->tuplehash[!dir].tuple.src.u.all)
248                         if (ip_xfrm_me_harder(pskb))
249                                 ret = NF_DROP;
250 #endif
251         }
252         return ret;
253 }
254
255 static unsigned int
256 nf_nat_adjust(unsigned int hooknum,
257               struct sk_buff **pskb,
258               const struct net_device *in,
259               const struct net_device *out,
260               int (*okfn)(struct sk_buff *))
261 {
262         struct nf_conn *ct;
263         enum ip_conntrack_info ctinfo;
264
265         ct = nf_ct_get(*pskb, &ctinfo);
266         if (ct && test_bit(IPS_SEQ_ADJUST_BIT, &ct->status)) {
267                 DEBUGP("nf_nat_standalone: adjusting sequence number\n");
268                 if (!nf_nat_seq_adjust(pskb, ct, ctinfo))
269                         return NF_DROP;
270         }
271         return NF_ACCEPT;
272 }
273
274 /* We must be after connection tracking and before packet filtering. */
275
276 static struct nf_hook_ops nf_nat_ops[] = {
277         /* Before packet filtering, change destination */
278         {
279                 .hook           = nf_nat_in,
280                 .owner          = THIS_MODULE,
281                 .pf             = PF_INET,
282                 .hooknum        = NF_IP_PRE_ROUTING,
283                 .priority       = NF_IP_PRI_NAT_DST,
284         },
285         /* After packet filtering, change source */
286         {
287                 .hook           = nf_nat_out,
288                 .owner          = THIS_MODULE,
289                 .pf             = PF_INET,
290                 .hooknum        = NF_IP_POST_ROUTING,
291                 .priority       = NF_IP_PRI_NAT_SRC,
292         },
293         /* After conntrack, adjust sequence number */
294         {
295                 .hook           = nf_nat_adjust,
296                 .owner          = THIS_MODULE,
297                 .pf             = PF_INET,
298                 .hooknum        = NF_IP_POST_ROUTING,
299                 .priority       = NF_IP_PRI_NAT_SEQ_ADJUST,
300         },
301         /* Before packet filtering, change destination */
302         {
303                 .hook           = nf_nat_local_fn,
304                 .owner          = THIS_MODULE,
305                 .pf             = PF_INET,
306                 .hooknum        = NF_IP_LOCAL_OUT,
307                 .priority       = NF_IP_PRI_NAT_DST,
308         },
309         /* After packet filtering, change source */
310         {
311                 .hook           = nf_nat_fn,
312                 .owner          = THIS_MODULE,
313                 .pf             = PF_INET,
314                 .hooknum        = NF_IP_LOCAL_IN,
315                 .priority       = NF_IP_PRI_NAT_SRC,
316         },
317         /* After conntrack, adjust sequence number */
318         {
319                 .hook           = nf_nat_adjust,
320                 .owner          = THIS_MODULE,
321                 .pf             = PF_INET,
322                 .hooknum        = NF_IP_LOCAL_IN,
323                 .priority       = NF_IP_PRI_NAT_SEQ_ADJUST,
324         },
325 };
326
327 static int __init nf_nat_standalone_init(void)
328 {
329         int size, ret = 0;
330
331         need_conntrack();
332
333         size = ALIGN(sizeof(struct nf_conn), __alignof__(struct nf_conn_nat)) +
334                sizeof(struct nf_conn_nat);
335         ret = nf_conntrack_register_cache(NF_CT_F_NAT, "nf_nat:base", size);
336         if (ret < 0) {
337                 printk(KERN_ERR "nf_nat_init: Unable to create slab cache\n");
338                 return ret;
339         }
340
341 #ifdef CONFIG_XFRM
342         BUG_ON(ip_nat_decode_session != NULL);
343         ip_nat_decode_session = nat_decode_session;
344 #endif
345         ret = nf_nat_rule_init();
346         if (ret < 0) {
347                 printk("nf_nat_init: can't setup rules.\n");
348                 goto cleanup_decode_session;
349         }
350         ret = nf_register_hooks(nf_nat_ops, ARRAY_SIZE(nf_nat_ops));
351         if (ret < 0) {
352                 printk("nf_nat_init: can't register hooks.\n");
353                 goto cleanup_rule_init;
354         }
355         nf_nat_module_is_loaded = 1;
356         return ret;
357
358  cleanup_rule_init:
359         nf_nat_rule_cleanup();
360  cleanup_decode_session:
361 #ifdef CONFIG_XFRM
362         ip_nat_decode_session = NULL;
363         synchronize_net();
364 #endif
365         nf_conntrack_unregister_cache(NF_CT_F_NAT);
366         return ret;
367 }
368
369 static void __exit nf_nat_standalone_fini(void)
370 {
371         nf_unregister_hooks(nf_nat_ops, ARRAY_SIZE(nf_nat_ops));
372         nf_nat_rule_cleanup();
373         nf_nat_module_is_loaded = 0;
374 #ifdef CONFIG_XFRM
375         ip_nat_decode_session = NULL;
376         synchronize_net();
377 #endif
378         /* Conntrack caches are unregistered in nf_conntrack_cleanup */
379 }
380
381 module_init(nf_nat_standalone_init);
382 module_exit(nf_nat_standalone_fini);
383
384 MODULE_LICENSE("GPL");
385 MODULE_ALIAS("ip_nat");