release/src-rt-6.x/linux/linux-2.6/net/ipv4/netfilter/ip_set_portmap.c

   1 /* Copyright (C) 2003-2008 Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>
   2  *
   3  * This program is free software; you can redistribute it and/or modify
   4  * it under the terms of the GNU General Public License version 2 as
   5  * published by the Free Software Foundation.
   6  */
   7
   8 /* Kernel module implementing a port set type as a bitmap */
   9
  10 #include <linux/module.h>
  11 #include <linux/ip.h>
  12 #include <linux/tcp.h>
  13 #include <linux/udp.h>
  14 #include <linux/skbuff.h>
  15 #include <linux/errno.h>
  16 #include <asm/uaccess.h>
  17 #include <asm/bitops.h>
  18 #include <linux/spinlock.h>
  19
  20 #include <net/ip.h>
  21
  22 #include <linux/netfilter_ipv4/ip_set_portmap.h>
  23 #include <linux/netfilter_ipv4/ip_set_getport.h>
  24
  25 static inline int
  26 portmap_test(const struct ip_set *set, ip_set_ip_t port)
  27 {
  28         const struct ip_set_portmap *map = set->data;
  29
  30         if (port < map->first_ip || port > map->last_ip)
  31                 return -ERANGE;
  32
  33         DP("set: %s, port: %u", set->name, port);
  34         return !!test_bit(port - map->first_ip, map->members);
  35 }
  36
  37 #define KADT_CONDITION                  \
  38         if (ip == INVALID_PORT)         \
  39                 return 0;
  40
  41 UADT(portmap, test)
  42 KADT(portmap, test, get_port)
  43
  44 static inline int
  45 portmap_add(struct ip_set *set, ip_set_ip_t port)
  46 {
  47         struct ip_set_portmap *map = set->data;
  48
  49         if (port < map->first_ip || port > map->last_ip)
  50                 return -ERANGE;
  51         if (test_and_set_bit(port - map->first_ip, map->members))
  52                 return -EEXIST;
  53
  54         DP("set: %s, port %u", set->name, port);
  55         return 0;
  56 }
  57
  58 UADT(portmap, add)
  59 KADT(portmap, add, get_port)
  60
  61 static inline int
  62 portmap_del(struct ip_set *set, ip_set_ip_t port)
  63 {
  64         struct ip_set_portmap *map = set->data;
  65
  66         if (port < map->first_ip || port > map->last_ip)
  67                 return -ERANGE;
  68         if (!test_and_clear_bit(port - map->first_ip, map->members))
  69                 return -EEXIST;
  70
  71         DP("set: %s, port %u", set->name, port);
  72         return 0;
  73 }
  74
  75 UADT(portmap, del)
  76 KADT(portmap, del, get_port)
  77
  78 static inline int
  79 __portmap_create(const struct ip_set_req_portmap_create *req,
  80                  struct ip_set_portmap *map)
  81 {
  82         if (req->to - req->from > MAX_RANGE) {
  83                 ip_set_printk("range too big, %d elements (max %d)",
  84                               req->to - req->from + 1, MAX_RANGE+1);
  85                 return -ENOEXEC;
  86         }
  87         return bitmap_bytes(req->from, req->to);
  88 }
  89
  90 BITMAP_CREATE(portmap)
  91 BITMAP_DESTROY(portmap)
  92 BITMAP_FLUSH(portmap)
  93
  94 static inline void
  95 __portmap_list_header(const struct ip_set_portmap *map,
  96                       struct ip_set_req_portmap_create *header)
  97 {
  98 }
  99
 100 BITMAP_LIST_HEADER(portmap)
 101 BITMAP_LIST_MEMBERS_SIZE(portmap, ip_set_ip_t, (map->last_ip - map->first_ip + 1),
 102                          test_bit(i, map->members))
 103
 104 static void
 105 portmap_list_members(const struct ip_set *set, void *data, char dont_align)
 106 {
 107         const struct ip_set_portmap *map = set->data;
 108         uint32_t i, n = 0;
 109         ip_set_ip_t *d;
 110
 111         if (dont_align) {
 112                 memcpy(data, map->members, map->size);
 113                 return;
 114         }
 115
 116         for (i = 0; i < map->last_ip - map->first_ip + 1; i++)
 117                 if (test_bit(i, map->members)) {
 118                         d = data + n * IPSET_ALIGN(sizeof(ip_set_ip_t));
 119                         *d = map->first_ip + i;
 120                         n++;
 121                 }
 122 }
 123
 124 IP_SET_TYPE(portmap, IPSET_TYPE_PORT | IPSET_DATA_SINGLE)
 125
 126 MODULE_LICENSE("GPL");
 127 MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>");
 128 MODULE_DESCRIPTION("portmap type of IP sets");
 129
 130 REGISTER_MODULE(portmap)