[NETFILTER]: C99 initizalizers for NAT protocols
[safe/jmp/linux-2.6] / net / ipv4 / netfilter / ipt_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/ip_tables.h>
19 #include <linux/netfilter_ipv4/ipt_CLASSIFY.h>
20
21 MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
22 MODULE_LICENSE("GPL");
23 MODULE_DESCRIPTION("iptables qdisc classification target module");
24
25 static unsigned int
26 target(struct sk_buff **pskb,
27        const struct net_device *in,
28        const struct net_device *out,
29        unsigned int hooknum,
30        const void *targinfo,
31        void *userinfo)
32 {
33         const struct ipt_classify_target_info *clinfo = targinfo;
34
35         if((*pskb)->priority != clinfo->priority) 
36                 (*pskb)->priority = clinfo->priority;
37
38         return IPT_CONTINUE;
39 }
40
41 static int
42 checkentry(const char *tablename,
43            const struct ipt_entry *e,
44            void *targinfo,
45            unsigned int targinfosize,
46            unsigned int hook_mask)
47 {
48         if (targinfosize != IPT_ALIGN(sizeof(struct ipt_classify_target_info))){
49                 printk(KERN_ERR "CLASSIFY: invalid size (%u != %Zu).\n",
50                        targinfosize,
51                        IPT_ALIGN(sizeof(struct ipt_classify_target_info)));
52                 return 0;
53         }
54         
55         if (hook_mask & ~((1 << NF_IP_LOCAL_OUT) | (1 << NF_IP_FORWARD) |
56                           (1 << NF_IP_POST_ROUTING))) {
57                 printk(KERN_ERR "CLASSIFY: only valid in LOCAL_OUT, FORWARD "
58                                 "and POST_ROUTING.\n");
59                 return 0;
60         }
61
62         if (strcmp(tablename, "mangle") != 0) {
63                 printk(KERN_ERR "CLASSIFY: can only be called from "
64                                 "\"mangle\" table, not \"%s\".\n",
65                                 tablename);
66                 return 0;
67         }
68
69         return 1;
70 }
71
72 static struct ipt_target ipt_classify_reg = { 
73         .name           = "CLASSIFY", 
74         .target         = target,
75         .checkentry     = checkentry,
76         .me             = THIS_MODULE,
77 };
78
79 static int __init init(void)
80 {
81         return ipt_register_target(&ipt_classify_reg);
82 }
83
84 static void __exit fini(void)
85 {
86         ipt_unregister_target(&ipt_classify_reg);
87 }
88
89 module_init(init);
90 module_exit(fini);