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_conntrack_extend.h>
  23 #include <net/netfilter/nf_nat.h>
  24 #include <net/netfilter/nf_nat_rule.h>
  25 #include <net/netfilter/nf_nat_protocol.h>
  26 #include <net/netfilter/nf_nat_core.h>
  27 #include <net/netfilter/nf_nat_helper.h>
  28 #include <linux/netfilter_ipv4/ip_tables.h>
  29
  30 #ifdef CONFIG_XFRM
  31 static void nat_decode_session(struct sk_buff *skb, struct flowi *fl)
  32 {
  33         const struct nf_conn *ct;
  34         const struct nf_conntrack_tuple *t;
  35         enum ip_conntrack_info ctinfo;
  36         enum ip_conntrack_dir dir;
  37         unsigned long statusbit;
  38
  39         ct = nf_ct_get(skb, &ctinfo);
  40         if (ct == NULL)
  41                 return;
  42         dir = CTINFO2DIR(ctinfo);
  43         t = &ct->tuplehash[dir].tuple;
  44
  45         if (dir == IP_CT_DIR_ORIGINAL)
  46                 statusbit = IPS_DST_NAT;
  47         else
  48                 statusbit = IPS_SRC_NAT;
  49
  50         if (ct->status & statusbit) {
  51                 fl->fl4_dst = t->dst.u3.ip;
  52                 if (t->dst.protonum == IPPROTO_TCP ||
  53                     t->dst.protonum == IPPROTO_UDP ||
  54                     t->dst.protonum == IPPROTO_UDPLITE ||
  55                     t->dst.protonum == IPPROTO_DCCP ||
  56                     t->dst.protonum == IPPROTO_SCTP)
  57                         fl->fl_ip_dport = t->dst.u.tcp.port;
  58         }
  59
  60         statusbit ^= IPS_NAT_MASK;
  61
  62         if (ct->status & statusbit) {
  63                 fl->fl4_src = t->src.u3.ip;
  64                 if (t->dst.protonum == IPPROTO_TCP ||
  65                     t->dst.protonum == IPPROTO_UDP ||
  66                     t->dst.protonum == IPPROTO_UDPLITE ||
  67                     t->dst.protonum == IPPROTO_DCCP ||
  68                     t->dst.protonum == IPPROTO_SCTP)
  69                         fl->fl_ip_sport = t->src.u.tcp.port;
  70         }
  71 }
  72 #endif
  73
  74 static unsigned int
  75 nf_nat_fn(unsigned int hooknum,
  76           struct sk_buff *skb,
  77           const struct net_device *in,
  78           const struct net_device *out,
  79           int (*okfn)(struct sk_buff *))
  80 {
  81         struct nf_conn *ct;
  82         enum ip_conntrack_info ctinfo;
  83         struct nf_conn_nat *nat;
  84         /* maniptype == SRC for postrouting. */
  85         enum nf_nat_manip_type maniptype = HOOK2MANIP(hooknum);
  86
  87         /* We never see fragments: conntrack defrags on pre-routing
  88            and local-out, and nf_nat_out protects post-routing. */
  89         NF_CT_ASSERT(!(ip_hdr(skb)->frag_off & htons(IP_MF | IP_OFFSET)));
  90
  91         ct = nf_ct_get(skb, &ctinfo);
  92         /* Can't track?  It's not due to stress, or conntrack would
  93            have dropped it.  Hence it's the user's responsibilty to
  94            packet filter it out, or implement conntrack/NAT for that
  95            protocol. 8) --RR */
  96         if (!ct)
  97                 return NF_ACCEPT;
  98
  99         /* Don't try to NAT if this packet is not conntracked */
 100         if (ct == &nf_conntrack_untracked)
 101                 return NF_ACCEPT;
 102
 103         nat = nfct_nat(ct);
 104         if (!nat) {
 105                 /* NAT module was loaded late. */
 106                 if (nf_ct_is_confirmed(ct))
 107                         return NF_ACCEPT;
 108                 nat = nf_ct_ext_add(ct, NF_CT_EXT_NAT, GFP_ATOMIC);
 109                 if (nat == NULL) {
 110                         pr_debug("failed to add NAT extension\n");
 111                         return NF_ACCEPT;
 112                 }
 113         }
 114
 115         switch (ctinfo) {
 116         case IP_CT_RELATED:
 117         case IP_CT_RELATED+IP_CT_IS_REPLY:
 118                 if (ip_hdr(skb)->protocol == IPPROTO_ICMP) {
 119                         if (!nf_nat_icmp_reply_translation(ct, ctinfo,
 120                                                            hooknum, skb))
 121                                 return NF_DROP;
 122                         else
 123                                 return NF_ACCEPT;
 124                 }
 125                 /* Fall thru... (Only ICMPs can be IP_CT_IS_REPLY) */
 126         case IP_CT_NEW:
 127
 128                 /* Seen it before?  This can happen for loopback, retrans,
 129                    or local packets.. */
 130                 if (!nf_nat_initialized(ct, maniptype)) {
 131                         unsigned int ret;
 132
 133                         if (hooknum == NF_INET_LOCAL_IN)
 134                                 /* LOCAL_IN hook doesn't have a chain!  */
 135                                 ret = alloc_null_binding(ct, hooknum);
 136                         else
 137                                 ret = nf_nat_rule_find(skb, hooknum, in, out,
 138                                                        ct);
 139
 140                         if (ret != NF_ACCEPT) {
 141                                 return ret;
 142                         }
 143                 } else
 144                         pr_debug("Already setup manip %s for ct %p\n",
 145                                  maniptype == IP_NAT_MANIP_SRC ? "SRC" : "DST",
 146                                  ct);
 147                 break;
 148
 149         default:
 150                 /* ESTABLISHED */
 151                 NF_CT_ASSERT(ctinfo == IP_CT_ESTABLISHED ||
 152                              ctinfo == (IP_CT_ESTABLISHED+IP_CT_IS_REPLY));
 153         }
 154
 155         return nf_nat_packet(ct, ctinfo, hooknum, skb);
 156 }
 157
 158 static unsigned int
 159 nf_nat_in(unsigned int hooknum,
 160           struct sk_buff *skb,
 161           const struct net_device *in,
 162           const struct net_device *out,
 163           int (*okfn)(struct sk_buff *))
 164 {
 165         unsigned int ret;
 166         __be32 daddr = ip_hdr(skb)->daddr;
 167
 168         ret = nf_nat_fn(hooknum, skb, in, out, okfn);
 169         if (ret != NF_DROP && ret != NF_STOLEN &&
 170             daddr != ip_hdr(skb)->daddr)
 171                 skb_dst_drop(skb);
 172
 173         return ret;
 174 }
 175
 176 static unsigned int
 177 nf_nat_out(unsigned int hooknum,
 178            struct sk_buff *skb,
 179            const struct net_device *in,
 180            const struct net_device *out,
 181            int (*okfn)(struct sk_buff *))
 182 {
 183 #ifdef CONFIG_XFRM
 184         const struct nf_conn *ct;
 185         enum ip_conntrack_info ctinfo;
 186 #endif
 187         unsigned int ret;
 188
 189         /* root is playing with raw sockets. */
 190         if (skb->len < sizeof(struct iphdr) ||
 191             ip_hdrlen(skb) < sizeof(struct iphdr))
 192                 return NF_ACCEPT;
 193
 194         ret = nf_nat_fn(hooknum, skb, in, out, okfn);
 195 #ifdef CONFIG_XFRM
 196         if (ret != NF_DROP && ret != NF_STOLEN &&
 197             (ct = nf_ct_get(skb, &ctinfo)) != NULL) {
 198                 enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
 199
 200                 if (ct->tuplehash[dir].tuple.src.u3.ip !=
 201                     ct->tuplehash[!dir].tuple.dst.u3.ip
 202                     || ct->tuplehash[dir].tuple.src.u.all !=
 203                        ct->tuplehash[!dir].tuple.dst.u.all
 204                     )
 205                         return ip_xfrm_me_harder(skb) == 0 ? ret : NF_DROP;
 206         }
 207 #endif
 208         return ret;
 209 }
 210
 211 static unsigned int
 212 nf_nat_local_fn(unsigned int hooknum,
 213                 struct sk_buff *skb,
 214                 const struct net_device *in,
 215                 const struct net_device *out,
 216                 int (*okfn)(struct sk_buff *))
 217 {
 218         const struct nf_conn *ct;
 219         enum ip_conntrack_info ctinfo;
 220         unsigned int ret;
 221
 222         /* root is playing with raw sockets. */
 223         if (skb->len < sizeof(struct iphdr) ||
 224             ip_hdrlen(skb) < sizeof(struct iphdr))
 225                 return NF_ACCEPT;
 226
 227         ret = nf_nat_fn(hooknum, skb, in, out, okfn);
 228         if (ret != NF_DROP && ret != NF_STOLEN &&
 229             (ct = nf_ct_get(skb, &ctinfo)) != NULL) {
 230                 enum ip_conntrack_dir dir = CTINFO2DIR(ctinfo);
 231
 232                 if (ct->tuplehash[dir].tuple.dst.u3.ip !=
 233                     ct->tuplehash[!dir].tuple.src.u3.ip) {
 234                         if (ip_route_me_harder(skb, RTN_UNSPEC))
 235                                 ret = NF_DROP;
 236                 }
 237 #ifdef CONFIG_XFRM
 238                 else if (ct->tuplehash[dir].tuple.dst.u.all !=
 239                          ct->tuplehash[!dir].tuple.src.u.all)
 240                         if (ip_xfrm_me_harder(skb))
 241                                 ret = NF_DROP;
 242 #endif
 243         }
 244         return ret;
 245 }
 246
 247 /* We must be after connection tracking and before packet filtering. */
 248
 249 static struct nf_hook_ops nf_nat_ops[] __read_mostly = {
 250         /* Before packet filtering, change destination */
 251         {
 252                 .hook           = nf_nat_in,
 253                 .owner          = THIS_MODULE,
 254                 .pf             = PF_INET,
 255                 .hooknum        = NF_INET_PRE_ROUTING,
 256                 .priority       = NF_IP_PRI_NAT_DST,
 257         },
 258         /* After packet filtering, change source */
 259         {
 260                 .hook           = nf_nat_out,
 261                 .owner          = THIS_MODULE,
 262                 .pf             = PF_INET,
 263                 .hooknum        = NF_INET_POST_ROUTING,
 264                 .priority       = NF_IP_PRI_NAT_SRC,
 265         },
 266         /* Before packet filtering, change destination */
 267         {
 268                 .hook           = nf_nat_local_fn,
 269                 .owner          = THIS_MODULE,
 270                 .pf             = PF_INET,
 271                 .hooknum        = NF_INET_LOCAL_OUT,
 272                 .priority       = NF_IP_PRI_NAT_DST,
 273         },
 274         /* After packet filtering, change source */
 275         {
 276                 .hook           = nf_nat_fn,
 277                 .owner          = THIS_MODULE,
 278                 .pf             = PF_INET,
 279                 .hooknum        = NF_INET_LOCAL_IN,
 280                 .priority       = NF_IP_PRI_NAT_SRC,
 281         },
 282 };
 283
 284 static int __init nf_nat_standalone_init(void)
 285 {
 286         int ret = 0;
 287
 288         need_ipv4_conntrack();
 289
 290 #ifdef CONFIG_XFRM
 291         BUG_ON(ip_nat_decode_session != NULL);
 292         rcu_assign_pointer(ip_nat_decode_session, nat_decode_session);
 293 #endif
 294         ret = nf_nat_rule_init();
 295         if (ret < 0) {
 296                 printk("nf_nat_init: can't setup rules.\n");
 297                 goto cleanup_decode_session;
 298         }
 299         ret = nf_register_hooks(nf_nat_ops, ARRAY_SIZE(nf_nat_ops));
 300         if (ret < 0) {
 301                 printk("nf_nat_init: can't register hooks.\n");
 302                 goto cleanup_rule_init;
 303         }
 304         return ret;
 305
 306  cleanup_rule_init:
 307         nf_nat_rule_cleanup();
 308  cleanup_decode_session:
 309 #ifdef CONFIG_XFRM
 310         rcu_assign_pointer(ip_nat_decode_session, NULL);
 311         synchronize_net();
 312 #endif
 313         return ret;
 314 }
 315
 316 static void __exit nf_nat_standalone_fini(void)
 317 {
 318         nf_unregister_hooks(nf_nat_ops, ARRAY_SIZE(nf_nat_ops));
 319         nf_nat_rule_cleanup();
 320 #ifdef CONFIG_XFRM
 321         rcu_assign_pointer(ip_nat_decode_session, NULL);
 322         synchronize_net();
 323 #endif
 324         /* Conntrack caches are unregistered in nf_conntrack_cleanup */
 325 }
 326
 327 module_init(nf_nat_standalone_init);
 328 module_exit(nf_nat_standalone_fini);
 329
 330 MODULE_LICENSE("GPL");
 331 MODULE_ALIAS("ip_nat");