netfilter: x_tables: use NFPROTO_* in extensions
[safe/jmp/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_ipv4.h>
19 #include <linux/netfilter_ipv6.h>
20 #include <linux/netfilter/x_tables.h>
21 #include <linux/netfilter/xt_CLASSIFY.h>
22
23 MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
24 MODULE_LICENSE("GPL");
25 MODULE_DESCRIPTION("Xtables: Qdisc classification");
26 MODULE_ALIAS("ipt_CLASSIFY");
27 MODULE_ALIAS("ip6t_CLASSIFY");
28
29 static unsigned int
30 classify_tg(struct sk_buff *skb, const struct net_device *in,
31             const struct net_device *out, unsigned int hooknum,
32             const struct xt_target *target, const void *targinfo)
33 {
34         const struct xt_classify_target_info *clinfo = targinfo;
35
36         skb->priority = clinfo->priority;
37         return XT_CONTINUE;
38 }
39
40 static struct xt_target classify_tg_reg[] __read_mostly = {
41         {
42                 .family         = NFPROTO_IPV4,
43                 .name           = "CLASSIFY",
44                 .target         = classify_tg,
45                 .targetsize     = sizeof(struct xt_classify_target_info),
46                 .table          = "mangle",
47                 .hooks          = (1 << NF_INET_LOCAL_OUT) |
48                                   (1 << NF_INET_FORWARD) |
49                                   (1 << NF_INET_POST_ROUTING),
50                 .me             = THIS_MODULE,
51         },
52         {
53                 .name           = "CLASSIFY",
54                 .family         = NFPROTO_IPV6,
55                 .target         = classify_tg,
56                 .targetsize     = sizeof(struct xt_classify_target_info),
57                 .table          = "mangle",
58                 .hooks          = (1 << NF_INET_LOCAL_OUT) |
59                                   (1 << NF_INET_FORWARD) |
60                                   (1 << NF_INET_POST_ROUTING),
61                 .me             = THIS_MODULE,
62         },
63 };
64
65 static int __init classify_tg_init(void)
66 {
67         return xt_register_targets(classify_tg_reg,
68                ARRAY_SIZE(classify_tg_reg));
69 }
70
71 static void __exit classify_tg_exit(void)
72 {
73         xt_unregister_targets(classify_tg_reg, ARRAY_SIZE(classify_tg_reg));
74 }
75
76 module_init(classify_tg_init);
77 module_exit(classify_tg_exit);