Merge branch 'origin'
[linux-2.6] / net / ipv4 / netfilter / ipt_policy.c
1 /* IP tables module for matching IPsec policy
2  *
3  * Copyright (c) 2004,2005 Patrick McHardy, <kaber@trash.net>
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
10 #include <linux/kernel.h>
11 #include <linux/config.h>
12 #include <linux/module.h>
13 #include <linux/skbuff.h>
14 #include <linux/init.h>
15 #include <net/xfrm.h>
16
17 #include <linux/netfilter_ipv4.h>
18 #include <linux/netfilter_ipv4/ip_tables.h>
19 #include <linux/netfilter_ipv4/ipt_policy.h>
20
21 MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
22 MODULE_DESCRIPTION("IPtables IPsec policy matching module");
23 MODULE_LICENSE("GPL");
24
25
26 static inline int
27 match_xfrm_state(struct xfrm_state *x, const struct ipt_policy_elem *e)
28 {
29 #define MATCH_ADDR(x,y,z)       (!e->match.x ||                              \
30                                  ((e->x.a4.s_addr == (e->y.a4.s_addr & (z))) \
31                                   ^ e->invert.x))
32 #define MATCH(x,y)              (!e->match.x || ((e->x == (y)) ^ e->invert.x))
33
34         return MATCH_ADDR(saddr, smask, x->props.saddr.a4) &&
35                MATCH_ADDR(daddr, dmask, x->id.daddr.a4) &&
36                MATCH(proto, x->id.proto) &&
37                MATCH(mode, x->props.mode) &&
38                MATCH(spi, x->id.spi) &&
39                MATCH(reqid, x->props.reqid);
40 }
41
42 static int
43 match_policy_in(const struct sk_buff *skb, const struct ipt_policy_info *info)
44 {
45         const struct ipt_policy_elem *e;
46         struct sec_path *sp = skb->sp;
47         int strict = info->flags & IPT_POLICY_MATCH_STRICT;
48         int i, pos;
49
50         if (sp == NULL)
51                 return -1;
52         if (strict && info->len != sp->len)
53                 return 0;
54
55         for (i = sp->len - 1; i >= 0; i--) {
56                 pos = strict ? i - sp->len + 1 : 0;
57                 if (pos >= info->len)
58                         return 0;
59                 e = &info->pol[pos];
60
61                 if (match_xfrm_state(sp->x[i].xvec, e)) {
62                         if (!strict)
63                                 return 1;
64                 } else if (strict)
65                         return 0;
66         }
67
68         return strict ? 1 : 0;
69 }
70
71 static int
72 match_policy_out(const struct sk_buff *skb, const struct ipt_policy_info *info)
73 {
74         const struct ipt_policy_elem *e;
75         struct dst_entry *dst = skb->dst;
76         int strict = info->flags & IPT_POLICY_MATCH_STRICT;
77         int i, pos;
78
79         if (dst->xfrm == NULL)
80                 return -1;
81
82         for (i = 0; dst && dst->xfrm; dst = dst->child, i++) {
83                 pos = strict ? i : 0;
84                 if (pos >= info->len)
85                         return 0;
86                 e = &info->pol[pos];
87
88                 if (match_xfrm_state(dst->xfrm, e)) {
89                         if (!strict)
90                                 return 1;
91                 } else if (strict)
92                         return 0;
93         }
94
95         return strict ? i == info->len : 0;
96 }
97
98 static int match(const struct sk_buff *skb,
99                  const struct net_device *in,
100                  const struct net_device *out,
101                  const void *matchinfo,
102                  int offset,
103                  unsigned int protoff,
104                  int *hotdrop)
105 {
106         const struct ipt_policy_info *info = matchinfo;
107         int ret;
108
109         if (info->flags & IPT_POLICY_MATCH_IN)
110                 ret = match_policy_in(skb, info);
111         else
112                 ret = match_policy_out(skb, info);
113
114         if (ret < 0)
115                 ret = info->flags & IPT_POLICY_MATCH_NONE ? 1 : 0;
116         else if (info->flags & IPT_POLICY_MATCH_NONE)
117                 ret = 0;
118
119         return ret;
120 }
121
122 static int checkentry(const char *tablename, const void *ip_void,
123                       void *matchinfo, unsigned int matchsize,
124                       unsigned int hook_mask)
125 {
126         struct ipt_policy_info *info = matchinfo;
127
128         if (matchsize != IPT_ALIGN(sizeof(*info))) {
129                 printk(KERN_ERR "ipt_policy: matchsize %u != %zu\n",
130                        matchsize, IPT_ALIGN(sizeof(*info)));
131                 return 0;
132         }
133         if (!(info->flags & (IPT_POLICY_MATCH_IN|IPT_POLICY_MATCH_OUT))) {
134                 printk(KERN_ERR "ipt_policy: neither incoming nor "
135                                 "outgoing policy selected\n");
136                 return 0;
137         }
138         if (hook_mask & (1 << NF_IP_PRE_ROUTING | 1 << NF_IP_LOCAL_IN)
139             && info->flags & IPT_POLICY_MATCH_OUT) {
140                 printk(KERN_ERR "ipt_policy: output policy not valid in "
141                                 "PRE_ROUTING and INPUT\n");
142                 return 0;
143         }
144         if (hook_mask & (1 << NF_IP_POST_ROUTING | 1 << NF_IP_LOCAL_OUT)
145             && info->flags & IPT_POLICY_MATCH_IN) {
146                 printk(KERN_ERR "ipt_policy: input policy not valid in "
147                                 "POST_ROUTING and OUTPUT\n");
148                 return 0;
149         }
150         if (info->len > IPT_POLICY_MAX_ELEM) {
151                 printk(KERN_ERR "ipt_policy: too many policy elements\n");
152                 return 0;
153         }
154
155         return 1;
156 }
157
158 static struct ipt_match policy_match = {
159         .name           = "policy",
160         .match          = match,
161         .checkentry     = checkentry,
162         .me             = THIS_MODULE,
163 };
164
165 static int __init init(void)
166 {
167         return ipt_register_match(&policy_match);
168 }
169
170 static void __exit fini(void)
171 {
172         ipt_unregister_match(&policy_match);
173 }
174
175 module_init(init);
176 module_exit(fini);