net/netfilter/xt_helper.c

   1 /* iptables module to match on related connections */
   2 /*
   3  * (C) 2001 Martin Josefsson <gandalf@wlug.westbo.se>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2 as
   7  * published by the Free Software Foundation.
   8  *
   9  *   19 Mar 2002 Harald Welte <laforge@gnumonks.org>:
  10  *               - Port to newnat infrastructure
  11  */
  12
  13 #include <linux/module.h>
  14 #include <linux/skbuff.h>
  15 #include <linux/netfilter.h>
  16 #if defined(CONFIG_IP_NF_CONNTRACK) || defined(CONFIG_IP_NF_CONNTRACK_MODULE)
  17 #include <linux/netfilter_ipv4/ip_conntrack.h>
  18 #include <linux/netfilter_ipv4/ip_conntrack_core.h>
  19 #include <linux/netfilter_ipv4/ip_conntrack_helper.h>
  20 #else
  21 #include <net/netfilter/nf_conntrack.h>
  22 #include <net/netfilter/nf_conntrack_core.h>
  23 #include <net/netfilter/nf_conntrack_helper.h>
  24 #endif
  25 #include <linux/netfilter/x_tables.h>
  26 #include <linux/netfilter/xt_helper.h>
  27
  28 MODULE_LICENSE("GPL");
  29 MODULE_AUTHOR("Martin Josefsson <gandalf@netfilter.org>");
  30 MODULE_DESCRIPTION("iptables helper match module");
  31 MODULE_ALIAS("ipt_helper");
  32 MODULE_ALIAS("ip6t_helper");
  33
  34 #if 0
  35 #define DEBUGP printk
  36 #else
  37 #define DEBUGP(format, args...)
  38 #endif
  39
  40 #if defined(CONFIG_IP_NF_CONNTRACK) || defined(CONFIG_IP_NF_CONNTRACK_MODULE)
  41 static int
  42 match(const struct sk_buff *skb,
  43       const struct net_device *in,
  44       const struct net_device *out,
  45       const struct xt_match *match,
  46       const void *matchinfo,
  47       int offset,
  48       unsigned int protoff,
  49       int *hotdrop)
  50 {
  51         const struct xt_helper_info *info = matchinfo;
  52         struct ip_conntrack *ct;
  53         enum ip_conntrack_info ctinfo;
  54         int ret = info->invert;
  55
  56         ct = ip_conntrack_get((struct sk_buff *)skb, &ctinfo);
  57         if (!ct) {
  58                 DEBUGP("xt_helper: Eek! invalid conntrack?\n");
  59                 return ret;
  60         }
  61
  62         if (!ct->master) {
  63                 DEBUGP("xt_helper: conntrack %p has no master\n", ct);
  64                 return ret;
  65         }
  66
  67         read_lock_bh(&ip_conntrack_lock);
  68         if (!ct->master->helper) {
  69                 DEBUGP("xt_helper: master ct %p has no helper\n",
  70                         exp->expectant);
  71                 goto out_unlock;
  72         }
  73
  74         DEBUGP("master's name = %s , info->name = %s\n",
  75                 ct->master->helper->name, info->name);
  76
  77         if (info->name[0] == '\0')
  78                 ret ^= 1;
  79         else
  80                 ret ^= !strncmp(ct->master->helper->name, info->name,
  81                                 strlen(ct->master->helper->name));
  82 out_unlock:
  83         read_unlock_bh(&ip_conntrack_lock);
  84         return ret;
  85 }
  86
  87 #else /* CONFIG_IP_NF_CONNTRACK */
  88
  89 static int
  90 match(const struct sk_buff *skb,
  91       const struct net_device *in,
  92       const struct net_device *out,
  93       const struct xt_match *match,
  94       const void *matchinfo,
  95       int offset,
  96       unsigned int protoff,
  97       int *hotdrop)
  98 {
  99         const struct xt_helper_info *info = matchinfo;
 100         struct nf_conn *ct;
 101         struct nf_conn_help *master_help;
 102         enum ip_conntrack_info ctinfo;
 103         int ret = info->invert;
 104
 105         ct = nf_ct_get((struct sk_buff *)skb, &ctinfo);
 106         if (!ct) {
 107                 DEBUGP("xt_helper: Eek! invalid conntrack?\n");
 108                 return ret;
 109         }
 110
 111         if (!ct->master) {
 112                 DEBUGP("xt_helper: conntrack %p has no master\n", ct);
 113                 return ret;
 114         }
 115
 116         read_lock_bh(&nf_conntrack_lock);
 117         master_help = nfct_help(ct->master);
 118         if (!master_help || !master_help->helper) {
 119                 DEBUGP("xt_helper: master ct %p has no helper\n",
 120                         exp->expectant);
 121                 goto out_unlock;
 122         }
 123
 124         DEBUGP("master's name = %s , info->name = %s\n",
 125                 ct->master->helper->name, info->name);
 126
 127         if (info->name[0] == '\0')
 128                 ret ^= 1;
 129         else
 130                 ret ^= !strncmp(master_help->helper->name, info->name,
 131                                 strlen(master_help->helper->name));
 132 out_unlock:
 133         read_unlock_bh(&nf_conntrack_lock);
 134         return ret;
 135 }
 136 #endif
 137
 138 static int check(const char *tablename,
 139                  const void *inf,
 140                  const struct xt_match *match,
 141                  void *matchinfo,
 142                  unsigned int matchsize,
 143                  unsigned int hook_mask)
 144 {
 145         struct xt_helper_info *info = matchinfo;
 146
 147 #if defined(CONFIG_NF_CONNTRACK) || defined(CONFIG_NF_CONNTRACK_MODULE)
 148         if (nf_ct_l3proto_try_module_get(match->family) < 0) {
 149                 printk(KERN_WARNING "can't load nf_conntrack support for "
 150                                     "proto=%d\n", match->family);
 151                 return 0;
 152         }
 153 #endif
 154         info->name[29] = '\0';
 155         return 1;
 156 }
 157
 158 static void
 159 destroy(const struct xt_match *match, void *matchinfo, unsigned int matchsize)
 160 {
 161 #if defined(CONFIG_NF_CONNTRACK) || defined(CONFIG_NF_CONNTRACK_MODULE)
 162         nf_ct_l3proto_module_put(match->family);
 163 #endif
 164 }
 165
 166 static struct xt_match helper_match = {
 167         .name           = "helper",
 168         .match          = match,
 169         .matchsize      = sizeof(struct xt_helper_info),
 170         .checkentry     = check,
 171         .destroy        = destroy,
 172         .family         = AF_INET,
 173         .me             = THIS_MODULE,
 174 };
 175 static struct xt_match helper6_match = {
 176         .name           = "helper",
 177         .match          = match,
 178         .matchsize      = sizeof(struct xt_helper_info),
 179         .checkentry     = check,
 180         .destroy        = destroy,
 181         .family         = AF_INET6,
 182         .me             = THIS_MODULE,
 183 };
 184
 185 static int __init xt_helper_init(void)
 186 {
 187         int ret;
 188         need_conntrack();
 189
 190         ret = xt_register_match(&helper_match);
 191         if (ret < 0)
 192                 return ret;
 193
 194         ret = xt_register_match(&helper6_match);
 195         if (ret < 0)
 196                 xt_unregister_match(&helper_match);
 197
 198         return ret;
 199 }
 200
 201 static void __exit xt_helper_fini(void)
 202 {
 203         xt_unregister_match(&helper_match);
 204         xt_unregister_match(&helper6_match);
 205 }
 206
 207 module_init(xt_helper_init);
 208 module_exit(xt_helper_fini);
 209