Merge master.kernel.org:/pub/scm/linux/kernel/git/gregkh/driver-2.6
[linux-2.6] / net / ipv4 / netfilter / ipt_NETMAP.c
1 /* NETMAP - static NAT mapping of IP network addresses (1:1).
2  * The mapping can be applied to source (POSTROUTING),
3  * destination (PREROUTING), or both (with separate rules).
4  */
5
6 /* (C) 2000-2001 Svenning Soerensen <svenning@post5.tele.dk>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License version 2 as
10  * published by the Free Software Foundation.
11  */
12
13 #include <linux/config.h>
14 #include <linux/ip.h>
15 #include <linux/module.h>
16 #include <linux/netdevice.h>
17 #include <linux/netfilter.h>
18 #include <linux/netfilter_ipv4.h>
19 #include <linux/netfilter_ipv4/ip_nat_rule.h>
20
21 #define MODULENAME "NETMAP"
22 MODULE_LICENSE("GPL");
23 MODULE_AUTHOR("Svenning Soerensen <svenning@post5.tele.dk>");
24 MODULE_DESCRIPTION("iptables 1:1 NAT mapping of IP networks target");
25
26 #if 0
27 #define DEBUGP printk
28 #else
29 #define DEBUGP(format, args...)
30 #endif
31
32 static int
33 check(const char *tablename,
34       const void *e,
35       void *targinfo,
36       unsigned int targinfosize,
37       unsigned int hook_mask)
38 {
39         const struct ip_nat_multi_range_compat *mr = targinfo;
40
41         if (strcmp(tablename, "nat") != 0) {
42                 DEBUGP(MODULENAME":check: bad table `%s'.\n", tablename);
43                 return 0;
44         }
45         if (targinfosize != IPT_ALIGN(sizeof(*mr))) {
46                 DEBUGP(MODULENAME":check: size %u.\n", targinfosize);
47                 return 0;
48         }
49         if (hook_mask & ~((1 << NF_IP_PRE_ROUTING) | (1 << NF_IP_POST_ROUTING) |
50                           (1 << NF_IP_LOCAL_OUT))) {
51                 DEBUGP(MODULENAME":check: bad hooks %x.\n", hook_mask);
52                 return 0;
53         }
54         if (!(mr->range[0].flags & IP_NAT_RANGE_MAP_IPS)) {
55                 DEBUGP(MODULENAME":check: bad MAP_IPS.\n");
56                 return 0;
57         }
58         if (mr->rangesize != 1) {
59                 DEBUGP(MODULENAME":check: bad rangesize %u.\n", mr->rangesize);
60                 return 0;
61         }
62         return 1;
63 }
64
65 static unsigned int
66 target(struct sk_buff **pskb,
67        const struct net_device *in,
68        const struct net_device *out,
69        unsigned int hooknum,
70        const void *targinfo,
71        void *userinfo)
72 {
73         struct ip_conntrack *ct;
74         enum ip_conntrack_info ctinfo;
75         u_int32_t new_ip, netmask;
76         const struct ip_nat_multi_range_compat *mr = targinfo;
77         struct ip_nat_range newrange;
78
79         IP_NF_ASSERT(hooknum == NF_IP_PRE_ROUTING
80                      || hooknum == NF_IP_POST_ROUTING
81                      || hooknum == NF_IP_LOCAL_OUT);
82         ct = ip_conntrack_get(*pskb, &ctinfo);
83
84         netmask = ~(mr->range[0].min_ip ^ mr->range[0].max_ip);
85
86         if (hooknum == NF_IP_PRE_ROUTING || hooknum == NF_IP_LOCAL_OUT)
87                 new_ip = (*pskb)->nh.iph->daddr & ~netmask;
88         else
89                 new_ip = (*pskb)->nh.iph->saddr & ~netmask;
90         new_ip |= mr->range[0].min_ip & netmask;
91
92         newrange = ((struct ip_nat_range)
93                 { mr->range[0].flags | IP_NAT_RANGE_MAP_IPS,
94                   new_ip, new_ip,
95                   mr->range[0].min, mr->range[0].max });
96
97         /* Hand modified range to generic setup. */
98         return ip_nat_setup_info(ct, &newrange, hooknum);
99 }
100
101 static struct ipt_target target_module = { 
102         .name           = MODULENAME,
103         .target         = target, 
104         .checkentry     = check,
105         .me             = THIS_MODULE 
106 };
107
108 static int __init init(void)
109 {
110         return ipt_register_target(&target_module);
111 }
112
113 static void __exit fini(void)
114 {
115         ipt_unregister_target(&target_module);
116 }
117
118 module_init(init);
119 module_exit(fini);