Merge master.kernel.org:/pub/scm/linux/kernel/git/davem/net-2.6
[linux-2.6] / 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);