net/ipv4/netfilter/ipt_ah.c

   1 /* Kernel module to match AH parameters. */
   2 /* (C) 1999-2000 Yon Uriarte <yon@astaro.de>
   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
   9 #include <linux/module.h>
  10 #include <linux/skbuff.h>
  11 #include <linux/ip.h>
  12
  13 #include <linux/netfilter_ipv4/ipt_ah.h>
  14 #include <linux/netfilter_ipv4/ip_tables.h>
  15
  16 MODULE_LICENSE("GPL");
  17 MODULE_AUTHOR("Yon Uriarte <yon@astaro.de>");
  18 MODULE_DESCRIPTION("iptables AH SPI match module");
  19
  20 #ifdef DEBUG_CONNTRACK
  21 #define duprintf(format, args...) printk(format , ## args)
  22 #else
  23 #define duprintf(format, args...)
  24 #endif
  25
  26 /* Returns 1 if the spi is matched by the range, 0 otherwise */
  27 static inline int
  28 spi_match(u_int32_t min, u_int32_t max, u_int32_t spi, int invert)
  29 {
  30         int r=0;
  31         duprintf("ah spi_match:%c 0x%x <= 0x%x <= 0x%x",invert? '!':' ',
  32                 min,spi,max);
  33         r=(spi >= min && spi <= max) ^ invert;
  34         duprintf(" result %s\n",r? "PASS" : "FAILED");
  35         return r;
  36 }
  37
  38 static int
  39 match(const struct sk_buff *skb,
  40       const struct net_device *in,
  41       const struct net_device *out,
  42       const void *matchinfo,
  43       int offset,
  44       unsigned int protoff,
  45       int *hotdrop)
  46 {
  47         struct ip_auth_hdr _ahdr, *ah;
  48         const struct ipt_ah *ahinfo = matchinfo;
  49
  50         /* Must not be a fragment. */
  51         if (offset)
  52                 return 0;
  53
  54         ah = skb_header_pointer(skb, protoff,
  55                                 sizeof(_ahdr), &_ahdr);
  56         if (ah == NULL) {
  57                 /* We've been asked to examine this packet, and we
  58                  * can't.  Hence, no choice but to drop.
  59                  */
  60                 duprintf("Dropping evil AH tinygram.\n");
  61                 *hotdrop = 1;
  62                 return 0;
  63         }
  64
  65         return spi_match(ahinfo->spis[0], ahinfo->spis[1],
  66                          ntohl(ah->spi),
  67                          !!(ahinfo->invflags & IPT_AH_INV_SPI));
  68 }
  69
  70 /* Called when user tries to insert an entry of this type. */
  71 static int
  72 checkentry(const char *tablename,
  73            const void *ip_void,
  74            void *matchinfo,
  75            unsigned int matchinfosize,
  76            unsigned int hook_mask)
  77 {
  78         const struct ipt_ah *ahinfo = matchinfo;
  79         const struct ipt_ip *ip = ip_void;
  80
  81         /* Must specify proto == AH, and no unknown invflags */
  82         if (ip->proto != IPPROTO_AH || (ip->invflags & IPT_INV_PROTO)) {
  83                 duprintf("ipt_ah: Protocol %u != %u\n", ip->proto,
  84                          IPPROTO_AH);
  85                 return 0;
  86         }
  87         if (matchinfosize != IPT_ALIGN(sizeof(struct ipt_ah))) {
  88                 duprintf("ipt_ah: matchsize %u != %u\n",
  89                          matchinfosize, IPT_ALIGN(sizeof(struct ipt_ah)));
  90                 return 0;
  91         }
  92         if (ahinfo->invflags & ~IPT_AH_INV_MASK) {
  93                 duprintf("ipt_ah: unknown flags %X\n",
  94                          ahinfo->invflags);
  95                 return 0;
  96         }
  97
  98         return 1;
  99 }
 100
 101 static struct ipt_match ah_match = {
 102         .name           = "ah",
 103         .match          = &match,
 104         .checkentry     = &checkentry,
 105         .me             = THIS_MODULE,
 106 };
 107
 108 static int __init init(void)
 109 {
 110         return ipt_register_match(&ah_match);
 111 }
 112
 113 static void __exit cleanup(void)
 114 {
 115         ipt_unregister_match(&ah_match);
 116 }
 117
 118 module_init(init);
 119 module_exit(cleanup);