git.oblomov.eu Git - linux-2.6/blob - net/netfilter/xt_CLASSIFY.c

   1 /*
   2  * This is a module which is used for setting the skb->priority field
   3  * of an skb for qdisc classification.
   4  */
   5
   6 /* (C) 2001-2002 Patrick McHardy <kaber@trash.net>
   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/module.h>
  14 #include <linux/skbuff.h>
  15 #include <linux/ip.h>
  16 #include <net/checksum.h>
  17
  18 #include <linux/netfilter/x_tables.h>
  19 #include <linux/netfilter/xt_CLASSIFY.h>
  20
  21 MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
  22 MODULE_LICENSE("GPL");
  23 MODULE_DESCRIPTION("iptables qdisc classification target module");
  24 MODULE_ALIAS("ipt_CLASSIFY");
  25
  26 static unsigned int
  27 target(struct sk_buff **pskb,
  28        const struct net_device *in,
  29        const struct net_device *out,
  30        unsigned int hooknum,
  31        const struct xt_target *target,
  32        const void *targinfo,
  33        void *userinfo)
  34 {
  35         const struct xt_classify_target_info *clinfo = targinfo;
  36
  37         if ((*pskb)->priority != clinfo->priority)
  38                 (*pskb)->priority = clinfo->priority;
  39
  40         return XT_CONTINUE;
  41 }
  42
  43 static struct xt_target classify_reg = {
  44         .name           = "CLASSIFY",
  45         .target         = target,
  46         .targetsize     = sizeof(struct xt_classify_target_info),
  47         .table          = "mangle",
  48         .hooks          = (1 << NF_IP_LOCAL_OUT) | (1 << NF_IP_FORWARD) |
  49                           (1 << NF_IP_POST_ROUTING),
  50         .me             = THIS_MODULE,
  51 };
  52 static struct xt_target classify6_reg = {
  53         .name           = "CLASSIFY",
  54         .target         = target,
  55         .targetsize     = sizeof(struct xt_classify_target_info),
  56         .table          = "mangle",
  57         .hooks          = (1 << NF_IP_LOCAL_OUT) | (1 << NF_IP_FORWARD) |
  58                           (1 << NF_IP_POST_ROUTING),
  59         .me             = THIS_MODULE,
  60 };
  61
  62
  63 static int __init init(void)
  64 {
  65         int ret;
  66
  67         ret = xt_register_target(AF_INET, &classify_reg);
  68         if (ret)
  69                 return ret;
  70
  71         ret = xt_register_target(AF_INET6, &classify6_reg);
  72         if (ret)
  73                 xt_unregister_target(AF_INET, &classify_reg);
  74
  75         return ret;
  76 }
  77
  78 static void __exit fini(void)
  79 {
  80         xt_unregister_target(AF_INET, &classify_reg);
  81         xt_unregister_target(AF_INET6, &classify6_reg);
  82 }
  83
  84 module_init(init);
  85 module_exit(fini);