Merge commit 'origin/trunk@12184' into fedora

[iptables.git] / trunk / extensions / libipt_NETMAP.c

diff --git a/trunk/extensions/libipt_NETMAP.c b/trunk/extensions/libipt_NETMAP.c
new file mode 100644 (file)
index 0000000..2a78696
--- /dev/null
+++ b/trunk/extensions/libipt_NETMAP.c
@@ -0,0 +1,192 @@
+/* Shared library add-on to iptables to add static NAT support.
+   Author: Svenning Soerensen <svenning@post5.tele.dk>
+*/
+
+#include <stdio.h>
+#include <netdb.h>
+#include <string.h>
+#include <stdlib.h>
+#include <getopt.h>
+#include <iptables.h>
+#include <linux/netfilter_ipv4/ip_tables.h>
+#include <linux/netfilter/nf_nat.h>
+
+#define MODULENAME "NETMAP"
+
+static const struct option NETMAP_opts[] = {
+       { "to", 1, NULL, '1' },
+       { .name = NULL }
+};
+
+/* Function which prints out usage message. */
+static void NETMAP_help(void)
+{
+       printf(MODULENAME" target options:\n"
+              "  --%s address[/mask]\n"
+              "                                Network address to map to.\n\n",
+              NETMAP_opts[0].name);
+}
+
+static u_int32_t
+bits2netmask(int bits)
+{
+       u_int32_t netmask, bm;
+
+       if (bits >= 32 || bits < 0)
+               return(~0);
+       for (netmask = 0, bm = 0x80000000; bits; bits--, bm >>= 1)
+               netmask |= bm;
+       return htonl(netmask);
+}
+
+static int
+netmask2bits(u_int32_t netmask)
+{
+       u_int32_t bm;
+       int bits;
+
+       netmask = ntohl(netmask);
+       for (bits = 0, bm = 0x80000000; netmask & bm; netmask <<= 1)
+               bits++;
+       if (netmask)
+               return -1; /* holes in netmask */
+       return bits;
+}
+
+/* Initialize the target. */
+static void NETMAP_init(struct xt_entry_target *t)
+{
+       struct ip_nat_multi_range *mr = (struct ip_nat_multi_range *)t->data;
+
+       /* Actually, it's 0, but it's ignored at the moment. */
+       mr->rangesize = 1;
+
+}
+
+/* Parses network address */
+static void
+parse_to(char *arg, struct ip_nat_range *range)
+{
+       char *slash;
+       const struct in_addr *ip;
+       u_int32_t netmask;
+       unsigned int bits;
+
+       range->flags |= IP_NAT_RANGE_MAP_IPS;
+       slash = strchr(arg, '/');
+       if (slash)
+               *slash = '\0';
+
+       ip = numeric_to_ipaddr(arg);
+       if (!ip)
+               exit_error(PARAMETER_PROBLEM, "Bad IP address `%s'\n",
+                          arg);
+       range->min_ip = ip->s_addr;
+       if (slash) {
+               if (strchr(slash+1, '.')) {
+                       ip = numeric_to_ipmask(slash+1);
+                       if (!ip)
+                               exit_error(PARAMETER_PROBLEM, "Bad netmask `%s'\n",
+                                          slash+1);
+                       netmask = ip->s_addr;
+               }
+               else {
+                       if (string_to_number(slash+1, 0, 32, &bits) == -1)
+                               exit_error(PARAMETER_PROBLEM, "Bad netmask `%s'\n",
+                                          slash+1);
+                       netmask = bits2netmask(bits);
+               }
+               /* Don't allow /0 (/1 is probably insane, too) */
+               if (netmask == 0)
+                       exit_error(PARAMETER_PROBLEM, "Netmask needed\n");
+       }
+       else
+               netmask = ~0;
+
+       if (range->min_ip & ~netmask) {
+               if (slash)
+                       *slash = '/';
+               exit_error(PARAMETER_PROBLEM, "Bad network address `%s'\n",
+                          arg);
+       }
+       range->max_ip = range->min_ip | ~netmask;
+}
+
+/* Function which parses command options; returns true if it
+   ate an option */
+static int NETMAP_parse(int c, char **argv, int invert, unsigned int *flags,
+                        const void *entry, struct xt_entry_target **target)
+{
+       struct ip_nat_multi_range *mr
+               = (struct ip_nat_multi_range *)(*target)->data;
+
+       switch (c) {
+       case '1':
+               if (check_inverse(optarg, &invert, NULL, 0))
+                       exit_error(PARAMETER_PROBLEM,
+                                  "Unexpected `!' after --%s", NETMAP_opts[0].name);
+
+               parse_to(optarg, &mr->range[0]);
+               *flags = 1;
+               return 1;
+
+       default:
+               return 0;
+       }
+}
+
+/* Final check; need --to */
+static void NETMAP_check(unsigned int flags)
+{
+       if (!flags)
+               exit_error(PARAMETER_PROBLEM,
+                          MODULENAME" needs --%s", NETMAP_opts[0].name);
+}
+
+/* Prints out the targinfo. */
+static void NETMAP_print(const void *ip, const struct xt_entry_target *target,
+                         int numeric)
+{
+       struct ip_nat_multi_range *mr
+               = (struct ip_nat_multi_range *)target->data;
+       struct ip_nat_range *r = &mr->range[0];
+       struct in_addr a;
+       int bits;
+
+       a.s_addr = r->min_ip;
+       printf("%s", ipaddr_to_numeric(&a));
+       a.s_addr = ~(r->min_ip ^ r->max_ip);
+       bits = netmask2bits(a.s_addr);
+       if (bits < 0)
+               printf("/%s", ipaddr_to_numeric(&a));
+       else
+               printf("/%d", bits);
+}
+
+/* Saves the targinfo in parsable form to stdout. */
+static void NETMAP_save(const void *ip, const struct xt_entry_target *target)
+{
+       printf("--%s ", NETMAP_opts[0].name);
+       NETMAP_print(ip, target, 0);
+}
+
+static struct xtables_target netmap_tg_reg = {
+       .name           = MODULENAME,
+       .version        = XTABLES_VERSION,
+       .family         = PF_INET,
+       .size           = XT_ALIGN(sizeof(struct ip_nat_multi_range)),
+       .userspacesize  = XT_ALIGN(sizeof(struct ip_nat_multi_range)),
+       .help           = NETMAP_help,
+       .init           = NETMAP_init,
+       .parse          = NETMAP_parse,
+       .final_check    = NETMAP_check,
+       .print          = NETMAP_print,
+       .save           = NETMAP_save,
+       .extra_opts     = NETMAP_opts,
+};
+
+void _init(void)
+{
+       xtables_register_target(&netmap_tg_reg);
+}
+