Commit | Line | Data |
---|---|---|
5d50e1d8 | 1 | /* Copyright (C) 2003-2013 Jozsef Kadlecsik <kadlec@blackhole.kfki.hu> |
5663bc30 JK |
2 | * |
3 | * This program is free software; you can redistribute it and/or modify | |
4 | * it under the terms of the GNU General Public License version 2 as | |
5 | * published by the Free Software Foundation. | |
6 | */ | |
7 | ||
8 | /* Kernel module implementing an IP set type: the hash:ip,port,ip type */ | |
9 | ||
10 | #include <linux/jhash.h> | |
11 | #include <linux/module.h> | |
12 | #include <linux/ip.h> | |
13 | #include <linux/skbuff.h> | |
14 | #include <linux/errno.h> | |
5663bc30 JK |
15 | #include <linux/random.h> |
16 | #include <net/ip.h> | |
17 | #include <net/ipv6.h> | |
18 | #include <net/netlink.h> | |
19 | #include <net/tcp.h> | |
20 | ||
21 | #include <linux/netfilter.h> | |
22 | #include <linux/netfilter/ipset/pfxlen.h> | |
23 | #include <linux/netfilter/ipset/ip_set.h> | |
5663bc30 JK |
24 | #include <linux/netfilter/ipset/ip_set_getport.h> |
25 | #include <linux/netfilter/ipset/ip_set_hash.h> | |
26 | ||
35b8dcf8 JK |
27 | #define IPSET_TYPE_REV_MIN 0 |
28 | /* 1 SCTP and UDPLITE support added */ | |
fda75c6d | 29 | /* 2 Counters support added */ |
07cf8f5a | 30 | /* 3 Comments support added */ |
af331419 AD |
31 | /* 4 Forceadd support added */ |
32 | #define IPSET_TYPE_REV_MAX 5 /* skbinfo support added */ | |
10111a6e | 33 | |
5663bc30 JK |
34 | MODULE_LICENSE("GPL"); |
35 | MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>"); | |
35b8dcf8 | 36 | IP_SET_MODULE_DESC("hash:ip,port,ip", IPSET_TYPE_REV_MIN, IPSET_TYPE_REV_MAX); |
5663bc30 JK |
37 | MODULE_ALIAS("ip_set_hash:ip,port,ip"); |
38 | ||
39 | /* Type specific function prefix */ | |
5d50e1d8 | 40 | #define HTYPE hash_ipportip |
5663bc30 | 41 | |
03c8b234 | 42 | /* IPv4 variant */ |
5663bc30 | 43 | |
5d50e1d8 | 44 | /* Member elements */ |
5663bc30 JK |
45 | struct hash_ipportip4_elem { |
46 | __be32 ip; | |
47 | __be32 ip2; | |
48 | __be16 port; | |
49 | u8 proto; | |
50 | u8 padding; | |
51 | }; | |
52 | ||
5663bc30 JK |
53 | static inline bool |
54 | hash_ipportip4_data_equal(const struct hash_ipportip4_elem *ip1, | |
89dc79b7 JK |
55 | const struct hash_ipportip4_elem *ip2, |
56 | u32 *multi) | |
5663bc30 JK |
57 | { |
58 | return ip1->ip == ip2->ip && | |
59 | ip1->ip2 == ip2->ip2 && | |
60 | ip1->port == ip2->port && | |
61 | ip1->proto == ip2->proto; | |
62 | } | |
63 | ||
5663bc30 JK |
64 | static bool |
65 | hash_ipportip4_data_list(struct sk_buff *skb, | |
ca0f6a5c | 66 | const struct hash_ipportip4_elem *data) |
5663bc30 | 67 | { |
7cf7899d DM |
68 | if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip) || |
69 | nla_put_ipaddr4(skb, IPSET_ATTR_IP2, data->ip2) || | |
70 | nla_put_net16(skb, IPSET_ATTR_PORT, data->port) || | |
71 | nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto)) | |
72 | goto nla_put_failure; | |
728a7e69 | 73 | return false; |
5663bc30 JK |
74 | |
75 | nla_put_failure: | |
728a7e69 | 76 | return true; |
5663bc30 JK |
77 | } |
78 | ||
5d50e1d8 JK |
79 | static inline void |
80 | hash_ipportip4_data_next(struct hash_ipportip4_elem *next, | |
81 | const struct hash_ipportip4_elem *d) | |
5663bc30 | 82 | { |
5d50e1d8 JK |
83 | next->ip = d->ip; |
84 | next->port = d->port; | |
5663bc30 JK |
85 | } |
86 | ||
5d50e1d8 JK |
87 | /* Common functions */ |
88 | #define MTYPE hash_ipportip4 | |
5663bc30 | 89 | #define HOST_MASK 32 |
5d50e1d8 | 90 | #include "ip_set_hash_gen.h" |
3d14b171 | 91 | |
5663bc30 JK |
92 | static int |
93 | hash_ipportip4_kadt(struct ip_set *set, const struct sk_buff *skb, | |
b66554cf | 94 | const struct xt_action_param *par, |
5d50e1d8 | 95 | enum ipset_adt adt, struct ip_set_adt_opt *opt) |
5663bc30 | 96 | { |
5663bc30 | 97 | ipset_adtfn adtfn = set->variant->adt[adt]; |
94729f8a | 98 | struct hash_ipportip4_elem e = { .ip = 0 }; |
ca134ce8 | 99 | struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set); |
5663bc30 | 100 | |
ac8cc925 | 101 | if (!ip_set_get_ip4_port(skb, opt->flags & IPSET_DIM_TWO_SRC, |
5d50e1d8 | 102 | &e.port, &e.proto)) |
5663bc30 JK |
103 | return -EINVAL; |
104 | ||
5d50e1d8 JK |
105 | ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip); |
106 | ip4addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip2); | |
107 | return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags); | |
5663bc30 JK |
108 | } |
109 | ||
110 | static int | |
111 | hash_ipportip4_uadt(struct ip_set *set, struct nlattr *tb[], | |
3d14b171 | 112 | enum ipset_adt adt, u32 *lineno, u32 flags, bool retried) |
5663bc30 | 113 | { |
5d50e1d8 | 114 | const struct hash_ipportip *h = set->data; |
5663bc30 | 115 | ipset_adtfn adtfn = set->variant->adt[adt]; |
94729f8a | 116 | struct hash_ipportip4_elem e = { .ip = 0 }; |
ca134ce8 | 117 | struct ip_set_ext ext = IP_SET_INIT_UEXT(set); |
20b2fab4 | 118 | u32 ip, ip_to = 0, p = 0, port, port_to; |
5e0c1eb7 | 119 | bool with_ports = false; |
5663bc30 JK |
120 | int ret; |
121 | ||
a212e08e SP |
122 | if (tb[IPSET_ATTR_LINENO]) |
123 | *lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]); | |
124 | ||
5663bc30 JK |
125 | if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] || |
126 | !ip_set_attr_netorder(tb, IPSET_ATTR_PORT) || | |
7dd37bc8 | 127 | !ip_set_optattr_netorder(tb, IPSET_ATTR_PORT_TO))) |
5663bc30 JK |
128 | return -IPSET_ERR_PROTOCOL; |
129 | ||
8e55d2e5 SP |
130 | ret = ip_set_get_ipaddr4(tb[IPSET_ATTR_IP], &e.ip); |
131 | if (ret) | |
132 | return ret; | |
133 | ||
134 | ret = ip_set_get_extensions(set, tb, &ext); | |
5663bc30 JK |
135 | if (ret) |
136 | return ret; | |
137 | ||
5d50e1d8 | 138 | ret = ip_set_get_ipaddr4(tb[IPSET_ATTR_IP2], &e.ip2); |
5663bc30 JK |
139 | if (ret) |
140 | return ret; | |
141 | ||
d25472e4 | 142 | e.port = nla_get_be16(tb[IPSET_ATTR_PORT]); |
5663bc30 JK |
143 | |
144 | if (tb[IPSET_ATTR_PROTO]) { | |
5d50e1d8 JK |
145 | e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]); |
146 | with_ports = ip_set_proto_with_ports(e.proto); | |
5663bc30 | 147 | |
5d50e1d8 | 148 | if (e.proto == 0) |
5663bc30 | 149 | return -IPSET_ERR_INVALID_PROTO; |
ca0f6a5c | 150 | } else { |
5663bc30 | 151 | return -IPSET_ERR_MISSING_PROTO; |
ca0f6a5c | 152 | } |
5663bc30 | 153 | |
5d50e1d8 JK |
154 | if (!(with_ports || e.proto == IPPROTO_ICMP)) |
155 | e.port = 0; | |
5663bc30 JK |
156 | |
157 | if (adt == IPSET_TEST || | |
5663bc30 JK |
158 | !(tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_CIDR] || |
159 | tb[IPSET_ATTR_PORT_TO])) { | |
5d50e1d8 | 160 | ret = adtfn(set, &e, &ext, &ext, flags); |
5663bc30 JK |
161 | return ip_set_eexist(ret, flags) ? 0 : ret; |
162 | } | |
163 | ||
5d50e1d8 | 164 | ip_to = ip = ntohl(e.ip); |
5663bc30 JK |
165 | if (tb[IPSET_ATTR_IP_TO]) { |
166 | ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP_TO], &ip_to); | |
167 | if (ret) | |
168 | return ret; | |
169 | if (ip > ip_to) | |
170 | swap(ip, ip_to); | |
171 | } else if (tb[IPSET_ATTR_CIDR]) { | |
172 | u8 cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]); | |
173 | ||
cabfd139 | 174 | if (!cidr || cidr > HOST_MASK) |
5663bc30 | 175 | return -IPSET_ERR_INVALID_CIDR; |
e6146e86 | 176 | ip_set_mask_from_to(ip, ip_to, cidr); |
4fe198e6 | 177 | } |
5663bc30 | 178 | |
5d50e1d8 | 179 | port_to = port = ntohs(e.port); |
5e0c1eb7 | 180 | if (with_ports && tb[IPSET_ATTR_PORT_TO]) { |
5663bc30 JK |
181 | port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]); |
182 | if (port > port_to) | |
183 | swap(port, port_to); | |
5e0c1eb7 | 184 | } |
5663bc30 | 185 | |
3d14b171 | 186 | if (retried) |
6e27c9b4 | 187 | ip = ntohl(h->next.ip); |
3d14b171 | 188 | for (; !before(ip_to, ip); ip++) { |
6e27c9b4 JK |
189 | p = retried && ip == ntohl(h->next.ip) ? ntohs(h->next.port) |
190 | : port; | |
3d14b171 | 191 | for (; p <= port_to; p++) { |
5d50e1d8 JK |
192 | e.ip = htonl(ip); |
193 | e.port = htons(p); | |
194 | ret = adtfn(set, &e, &ext, &ext, flags); | |
5663bc30 JK |
195 | |
196 | if (ret && !ip_set_eexist(ret, flags)) | |
197 | return ret; | |
ca0f6a5c JK |
198 | |
199 | ret = 0; | |
5663bc30 | 200 | } |
3d14b171 | 201 | } |
5663bc30 JK |
202 | return ret; |
203 | } | |
204 | ||
03c8b234 | 205 | /* IPv6 variant */ |
5663bc30 JK |
206 | |
207 | struct hash_ipportip6_elem { | |
208 | union nf_inet_addr ip; | |
209 | union nf_inet_addr ip2; | |
210 | __be16 port; | |
211 | u8 proto; | |
212 | u8 padding; | |
213 | }; | |
214 | ||
5d50e1d8 JK |
215 | /* Common functions */ |
216 | ||
5663bc30 JK |
217 | static inline bool |
218 | hash_ipportip6_data_equal(const struct hash_ipportip6_elem *ip1, | |
89dc79b7 JK |
219 | const struct hash_ipportip6_elem *ip2, |
220 | u32 *multi) | |
5663bc30 | 221 | { |
29e3b160 YH |
222 | return ipv6_addr_equal(&ip1->ip.in6, &ip2->ip.in6) && |
223 | ipv6_addr_equal(&ip1->ip2.in6, &ip2->ip2.in6) && | |
5663bc30 JK |
224 | ip1->port == ip2->port && |
225 | ip1->proto == ip2->proto; | |
226 | } | |
227 | ||
5663bc30 JK |
228 | static bool |
229 | hash_ipportip6_data_list(struct sk_buff *skb, | |
230 | const struct hash_ipportip6_elem *data) | |
231 | { | |
7cf7899d DM |
232 | if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip.in6) || |
233 | nla_put_ipaddr6(skb, IPSET_ATTR_IP2, &data->ip2.in6) || | |
234 | nla_put_net16(skb, IPSET_ATTR_PORT, data->port) || | |
235 | nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto)) | |
236 | goto nla_put_failure; | |
728a7e69 | 237 | return false; |
5663bc30 JK |
238 | |
239 | nla_put_failure: | |
728a7e69 | 240 | return true; |
5663bc30 JK |
241 | } |
242 | ||
5d50e1d8 JK |
243 | static inline void |
244 | hash_ipportip6_data_next(struct hash_ipportip4_elem *next, | |
245 | const struct hash_ipportip6_elem *d) | |
5663bc30 | 246 | { |
5d50e1d8 | 247 | next->port = d->port; |
5663bc30 JK |
248 | } |
249 | ||
5d50e1d8 | 250 | #undef MTYPE |
5663bc30 JK |
251 | #undef HOST_MASK |
252 | ||
5d50e1d8 | 253 | #define MTYPE hash_ipportip6 |
5663bc30 | 254 | #define HOST_MASK 128 |
5d50e1d8 JK |
255 | #define IP_SET_EMIT_CREATE |
256 | #include "ip_set_hash_gen.h" | |
3d14b171 | 257 | |
5663bc30 JK |
258 | static int |
259 | hash_ipportip6_kadt(struct ip_set *set, const struct sk_buff *skb, | |
b66554cf | 260 | const struct xt_action_param *par, |
5d50e1d8 | 261 | enum ipset_adt adt, struct ip_set_adt_opt *opt) |
5663bc30 | 262 | { |
5663bc30 | 263 | ipset_adtfn adtfn = set->variant->adt[adt]; |
94729f8a | 264 | struct hash_ipportip6_elem e = { .ip = { .all = { 0 } } }; |
ca134ce8 | 265 | struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set); |
5663bc30 | 266 | |
ac8cc925 | 267 | if (!ip_set_get_ip6_port(skb, opt->flags & IPSET_DIM_TWO_SRC, |
5d50e1d8 | 268 | &e.port, &e.proto)) |
5663bc30 JK |
269 | return -EINVAL; |
270 | ||
5d50e1d8 JK |
271 | ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip.in6); |
272 | ip6addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip2.in6); | |
273 | return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags); | |
5663bc30 JK |
274 | } |
275 | ||
276 | static int | |
277 | hash_ipportip6_uadt(struct ip_set *set, struct nlattr *tb[], | |
3d14b171 | 278 | enum ipset_adt adt, u32 *lineno, u32 flags, bool retried) |
5663bc30 | 279 | { |
5d50e1d8 | 280 | const struct hash_ipportip *h = set->data; |
5663bc30 | 281 | ipset_adtfn adtfn = set->variant->adt[adt]; |
94729f8a | 282 | struct hash_ipportip6_elem e = { .ip = { .all = { 0 } } }; |
ca134ce8 | 283 | struct ip_set_ext ext = IP_SET_INIT_UEXT(set); |
5663bc30 | 284 | u32 port, port_to; |
5e0c1eb7 | 285 | bool with_ports = false; |
5663bc30 JK |
286 | int ret; |
287 | ||
a212e08e SP |
288 | if (tb[IPSET_ATTR_LINENO]) |
289 | *lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]); | |
290 | ||
5663bc30 JK |
291 | if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] || |
292 | !ip_set_attr_netorder(tb, IPSET_ATTR_PORT) || | |
2c227f27 | 293 | !ip_set_optattr_netorder(tb, IPSET_ATTR_PORT_TO))) |
5663bc30 | 294 | return -IPSET_ERR_PROTOCOL; |
2c227f27 SP |
295 | if (unlikely(tb[IPSET_ATTR_IP_TO])) |
296 | return -IPSET_ERR_HASH_RANGE_UNSUPPORTED; | |
297 | if (unlikely(tb[IPSET_ATTR_CIDR])) { | |
298 | u8 cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]); | |
299 | ||
300 | if (cidr != HOST_MASK) | |
301 | return -IPSET_ERR_INVALID_CIDR; | |
302 | } | |
5663bc30 | 303 | |
8e55d2e5 SP |
304 | ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP], &e.ip); |
305 | if (ret) | |
306 | return ret; | |
307 | ||
308 | ret = ip_set_get_extensions(set, tb, &ext); | |
5663bc30 JK |
309 | if (ret) |
310 | return ret; | |
311 | ||
5d50e1d8 | 312 | ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP2], &e.ip2); |
5663bc30 JK |
313 | if (ret) |
314 | return ret; | |
315 | ||
d25472e4 | 316 | e.port = nla_get_be16(tb[IPSET_ATTR_PORT]); |
5663bc30 JK |
317 | |
318 | if (tb[IPSET_ATTR_PROTO]) { | |
5d50e1d8 JK |
319 | e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]); |
320 | with_ports = ip_set_proto_with_ports(e.proto); | |
5663bc30 | 321 | |
5d50e1d8 | 322 | if (e.proto == 0) |
5663bc30 | 323 | return -IPSET_ERR_INVALID_PROTO; |
ca0f6a5c | 324 | } else { |
5663bc30 | 325 | return -IPSET_ERR_MISSING_PROTO; |
ca0f6a5c | 326 | } |
5663bc30 | 327 | |
5d50e1d8 JK |
328 | if (!(with_ports || e.proto == IPPROTO_ICMPV6)) |
329 | e.port = 0; | |
5663bc30 | 330 | |
5e0c1eb7 | 331 | if (adt == IPSET_TEST || !with_ports || !tb[IPSET_ATTR_PORT_TO]) { |
5d50e1d8 | 332 | ret = adtfn(set, &e, &ext, &ext, flags); |
5663bc30 JK |
333 | return ip_set_eexist(ret, flags) ? 0 : ret; |
334 | } | |
335 | ||
5d50e1d8 | 336 | port = ntohs(e.port); |
5663bc30 JK |
337 | port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]); |
338 | if (port > port_to) | |
339 | swap(port, port_to); | |
340 | ||
3d14b171 | 341 | if (retried) |
6e27c9b4 | 342 | port = ntohs(h->next.port); |
5663bc30 | 343 | for (; port <= port_to; port++) { |
5d50e1d8 JK |
344 | e.port = htons(port); |
345 | ret = adtfn(set, &e, &ext, &ext, flags); | |
5663bc30 JK |
346 | |
347 | if (ret && !ip_set_eexist(ret, flags)) | |
348 | return ret; | |
ca0f6a5c JK |
349 | |
350 | ret = 0; | |
5663bc30 JK |
351 | } |
352 | return ret; | |
353 | } | |
354 | ||
5663bc30 JK |
355 | static struct ip_set_type hash_ipportip_type __read_mostly = { |
356 | .name = "hash:ip,port,ip", | |
357 | .protocol = IPSET_PROTOCOL, | |
358 | .features = IPSET_TYPE_IP | IPSET_TYPE_PORT | IPSET_TYPE_IP2, | |
359 | .dimension = IPSET_DIM_THREE, | |
c15f1c83 | 360 | .family = NFPROTO_UNSPEC, |
35b8dcf8 JK |
361 | .revision_min = IPSET_TYPE_REV_MIN, |
362 | .revision_max = IPSET_TYPE_REV_MAX, | |
5663bc30 JK |
363 | .create = hash_ipportip_create, |
364 | .create_policy = { | |
365 | [IPSET_ATTR_HASHSIZE] = { .type = NLA_U32 }, | |
366 | [IPSET_ATTR_MAXELEM] = { .type = NLA_U32 }, | |
367 | [IPSET_ATTR_PROBES] = { .type = NLA_U8 }, | |
368 | [IPSET_ATTR_RESIZE] = { .type = NLA_U8 }, | |
369 | [IPSET_ATTR_TIMEOUT] = { .type = NLA_U32 }, | |
00d71b27 | 370 | [IPSET_ATTR_CADT_FLAGS] = { .type = NLA_U32 }, |
5663bc30 JK |
371 | }, |
372 | .adt_policy = { | |
373 | [IPSET_ATTR_IP] = { .type = NLA_NESTED }, | |
374 | [IPSET_ATTR_IP_TO] = { .type = NLA_NESTED }, | |
375 | [IPSET_ATTR_IP2] = { .type = NLA_NESTED }, | |
376 | [IPSET_ATTR_PORT] = { .type = NLA_U16 }, | |
377 | [IPSET_ATTR_PORT_TO] = { .type = NLA_U16 }, | |
378 | [IPSET_ATTR_CIDR] = { .type = NLA_U8 }, | |
379 | [IPSET_ATTR_PROTO] = { .type = NLA_U8 }, | |
380 | [IPSET_ATTR_TIMEOUT] = { .type = NLA_U32 }, | |
381 | [IPSET_ATTR_LINENO] = { .type = NLA_U32 }, | |
00d71b27 JK |
382 | [IPSET_ATTR_BYTES] = { .type = NLA_U64 }, |
383 | [IPSET_ATTR_PACKETS] = { .type = NLA_U64 }, | |
03726186 SP |
384 | [IPSET_ATTR_COMMENT] = { .type = NLA_NUL_STRING, |
385 | .len = IPSET_MAX_COMMENT_SIZE }, | |
af331419 AD |
386 | [IPSET_ATTR_SKBMARK] = { .type = NLA_U64 }, |
387 | [IPSET_ATTR_SKBPRIO] = { .type = NLA_U32 }, | |
388 | [IPSET_ATTR_SKBQUEUE] = { .type = NLA_U16 }, | |
5663bc30 JK |
389 | }, |
390 | .me = THIS_MODULE, | |
391 | }; | |
392 | ||
393 | static int __init | |
394 | hash_ipportip_init(void) | |
395 | { | |
396 | return ip_set_type_register(&hash_ipportip_type); | |
397 | } | |
398 | ||
399 | static void __exit | |
400 | hash_ipportip_fini(void) | |
401 | { | |
18f84d41 | 402 | rcu_barrier(); |
5663bc30 JK |
403 | ip_set_type_unregister(&hash_ipportip_type); |
404 | } | |
405 | ||
406 | module_init(hash_ipportip_init); | |
407 | module_exit(hash_ipportip_fini); |