]>
Commit | Line | Data |
---|---|---|
c1d10adb PNA |
1 | /* Connection tracking via netlink socket. Allows for user space |
2 | * protocol helpers and general trouble making from userspace. | |
3 | * | |
4 | * (C) 2001 by Jay Schulist <jschlst@samba.org> | |
dc808fe2 | 5 | * (C) 2002-2006 by Harald Welte <laforge@gnumonks.org> |
c1d10adb | 6 | * (C) 2003 by Patrick Mchardy <kaber@trash.net> |
5faa1f4c | 7 | * (C) 2005-2007 by Pablo Neira Ayuso <pablo@netfilter.org> |
c1d10adb | 8 | * |
601e68e1 | 9 | * Initial connection tracking via netlink development funded and |
c1d10adb PNA |
10 | * generally made possible by Network Robots, Inc. (www.networkrobots.com) |
11 | * | |
12 | * Further development of this code funded by Astaro AG (http://www.astaro.com) | |
13 | * | |
14 | * This software may be used and distributed according to the terms | |
15 | * of the GNU General Public License, incorporated herein by reference. | |
c1d10adb PNA |
16 | */ |
17 | ||
18 | #include <linux/init.h> | |
19 | #include <linux/module.h> | |
20 | #include <linux/kernel.h> | |
21 | #include <linux/types.h> | |
22 | #include <linux/timer.h> | |
23 | #include <linux/skbuff.h> | |
24 | #include <linux/errno.h> | |
25 | #include <linux/netlink.h> | |
26 | #include <linux/spinlock.h> | |
40a839fd | 27 | #include <linux/interrupt.h> |
c1d10adb PNA |
28 | #include <linux/notifier.h> |
29 | ||
30 | #include <linux/netfilter.h> | |
dc5fc579 | 31 | #include <net/netlink.h> |
c1d10adb PNA |
32 | #include <net/netfilter/nf_conntrack.h> |
33 | #include <net/netfilter/nf_conntrack_core.h> | |
77ab9cff | 34 | #include <net/netfilter/nf_conntrack_expect.h> |
c1d10adb PNA |
35 | #include <net/netfilter/nf_conntrack_helper.h> |
36 | #include <net/netfilter/nf_conntrack_l3proto.h> | |
605dcad6 | 37 | #include <net/netfilter/nf_conntrack_l4proto.h> |
5b1158e9 JK |
38 | #include <net/netfilter/nf_conntrack_tuple.h> |
39 | #ifdef CONFIG_NF_NAT_NEEDED | |
40 | #include <net/netfilter/nf_nat_core.h> | |
41 | #include <net/netfilter/nf_nat_protocol.h> | |
42 | #endif | |
c1d10adb PNA |
43 | |
44 | #include <linux/netfilter/nfnetlink.h> | |
45 | #include <linux/netfilter/nfnetlink_conntrack.h> | |
46 | ||
47 | MODULE_LICENSE("GPL"); | |
48 | ||
dc808fe2 | 49 | static char __initdata version[] = "0.93"; |
c1d10adb | 50 | |
c1d10adb | 51 | static inline int |
601e68e1 | 52 | ctnetlink_dump_tuples_proto(struct sk_buff *skb, |
1cde6436 | 53 | const struct nf_conntrack_tuple *tuple, |
605dcad6 | 54 | struct nf_conntrack_l4proto *l4proto) |
c1d10adb | 55 | { |
c1d10adb | 56 | int ret = 0; |
df6fb868 | 57 | struct nlattr *nest_parms; |
c1d10adb | 58 | |
df6fb868 PM |
59 | nest_parms = nla_nest_start(skb, CTA_TUPLE_PROTO | NLA_F_NESTED); |
60 | if (!nest_parms) | |
61 | goto nla_put_failure; | |
62 | NLA_PUT(skb, CTA_PROTO_NUM, sizeof(u_int8_t), &tuple->dst.protonum); | |
c1d10adb | 63 | |
fdf70832 PM |
64 | if (likely(l4proto->tuple_to_nlattr)) |
65 | ret = l4proto->tuple_to_nlattr(skb, tuple); | |
601e68e1 | 66 | |
df6fb868 | 67 | nla_nest_end(skb, nest_parms); |
c1d10adb PNA |
68 | |
69 | return ret; | |
70 | ||
df6fb868 | 71 | nla_put_failure: |
c1d10adb PNA |
72 | return -1; |
73 | } | |
74 | ||
75 | static inline int | |
1cde6436 PNA |
76 | ctnetlink_dump_tuples_ip(struct sk_buff *skb, |
77 | const struct nf_conntrack_tuple *tuple, | |
78 | struct nf_conntrack_l3proto *l3proto) | |
c1d10adb | 79 | { |
c1d10adb | 80 | int ret = 0; |
df6fb868 PM |
81 | struct nlattr *nest_parms; |
82 | ||
83 | nest_parms = nla_nest_start(skb, CTA_TUPLE_IP | NLA_F_NESTED); | |
84 | if (!nest_parms) | |
85 | goto nla_put_failure; | |
1cde6436 | 86 | |
fdf70832 PM |
87 | if (likely(l3proto->tuple_to_nlattr)) |
88 | ret = l3proto->tuple_to_nlattr(skb, tuple); | |
1cde6436 | 89 | |
df6fb868 | 90 | nla_nest_end(skb, nest_parms); |
c1d10adb | 91 | |
1cde6436 PNA |
92 | return ret; |
93 | ||
df6fb868 | 94 | nla_put_failure: |
1cde6436 PNA |
95 | return -1; |
96 | } | |
97 | ||
98 | static inline int | |
99 | ctnetlink_dump_tuples(struct sk_buff *skb, | |
100 | const struct nf_conntrack_tuple *tuple) | |
101 | { | |
102 | int ret; | |
103 | struct nf_conntrack_l3proto *l3proto; | |
605dcad6 | 104 | struct nf_conntrack_l4proto *l4proto; |
1cde6436 PNA |
105 | |
106 | l3proto = nf_ct_l3proto_find_get(tuple->src.l3num); | |
107 | ret = ctnetlink_dump_tuples_ip(skb, tuple, l3proto); | |
c1d10adb PNA |
108 | nf_ct_l3proto_put(l3proto); |
109 | ||
110 | if (unlikely(ret < 0)) | |
111 | return ret; | |
112 | ||
605dcad6 MJ |
113 | l4proto = nf_ct_l4proto_find_get(tuple->src.l3num, tuple->dst.protonum); |
114 | ret = ctnetlink_dump_tuples_proto(skb, tuple, l4proto); | |
115 | nf_ct_l4proto_put(l4proto); | |
c1d10adb PNA |
116 | |
117 | return ret; | |
c1d10adb PNA |
118 | } |
119 | ||
120 | static inline int | |
121 | ctnetlink_dump_status(struct sk_buff *skb, const struct nf_conn *ct) | |
122 | { | |
bff9a89b | 123 | __be32 status = htonl((u_int32_t) ct->status); |
df6fb868 | 124 | NLA_PUT(skb, CTA_STATUS, sizeof(status), &status); |
c1d10adb PNA |
125 | return 0; |
126 | ||
df6fb868 | 127 | nla_put_failure: |
c1d10adb PNA |
128 | return -1; |
129 | } | |
130 | ||
131 | static inline int | |
132 | ctnetlink_dump_timeout(struct sk_buff *skb, const struct nf_conn *ct) | |
133 | { | |
134 | long timeout_l = ct->timeout.expires - jiffies; | |
bff9a89b | 135 | __be32 timeout; |
c1d10adb PNA |
136 | |
137 | if (timeout_l < 0) | |
138 | timeout = 0; | |
139 | else | |
140 | timeout = htonl(timeout_l / HZ); | |
601e68e1 | 141 | |
df6fb868 | 142 | NLA_PUT(skb, CTA_TIMEOUT, sizeof(timeout), &timeout); |
c1d10adb PNA |
143 | return 0; |
144 | ||
df6fb868 | 145 | nla_put_failure: |
c1d10adb PNA |
146 | return -1; |
147 | } | |
148 | ||
149 | static inline int | |
150 | ctnetlink_dump_protoinfo(struct sk_buff *skb, const struct nf_conn *ct) | |
151 | { | |
605dcad6 | 152 | struct nf_conntrack_l4proto *l4proto = nf_ct_l4proto_find_get(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.l3num, ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.protonum); |
df6fb868 | 153 | struct nlattr *nest_proto; |
c1d10adb PNA |
154 | int ret; |
155 | ||
fdf70832 | 156 | if (!l4proto->to_nlattr) { |
605dcad6 | 157 | nf_ct_l4proto_put(l4proto); |
c1d10adb PNA |
158 | return 0; |
159 | } | |
601e68e1 | 160 | |
df6fb868 PM |
161 | nest_proto = nla_nest_start(skb, CTA_PROTOINFO | NLA_F_NESTED); |
162 | if (!nest_proto) | |
163 | goto nla_put_failure; | |
c1d10adb | 164 | |
fdf70832 | 165 | ret = l4proto->to_nlattr(skb, nest_proto, ct); |
c1d10adb | 166 | |
605dcad6 | 167 | nf_ct_l4proto_put(l4proto); |
c1d10adb | 168 | |
df6fb868 | 169 | nla_nest_end(skb, nest_proto); |
c1d10adb PNA |
170 | |
171 | return ret; | |
172 | ||
df6fb868 | 173 | nla_put_failure: |
605dcad6 | 174 | nf_ct_l4proto_put(l4proto); |
c1d10adb PNA |
175 | return -1; |
176 | } | |
177 | ||
178 | static inline int | |
179 | ctnetlink_dump_helpinfo(struct sk_buff *skb, const struct nf_conn *ct) | |
180 | { | |
df6fb868 | 181 | struct nlattr *nest_helper; |
dc808fe2 | 182 | const struct nf_conn_help *help = nfct_help(ct); |
3c158f7f | 183 | struct nf_conntrack_helper *helper; |
c1d10adb | 184 | |
3c158f7f | 185 | if (!help) |
c1d10adb | 186 | return 0; |
601e68e1 | 187 | |
3c158f7f PM |
188 | rcu_read_lock(); |
189 | helper = rcu_dereference(help->helper); | |
190 | if (!helper) | |
191 | goto out; | |
192 | ||
df6fb868 PM |
193 | nest_helper = nla_nest_start(skb, CTA_HELP | NLA_F_NESTED); |
194 | if (!nest_helper) | |
195 | goto nla_put_failure; | |
196 | NLA_PUT(skb, CTA_HELP_NAME, strlen(helper->name), helper->name); | |
c1d10adb | 197 | |
fdf70832 PM |
198 | if (helper->to_nlattr) |
199 | helper->to_nlattr(skb, ct); | |
c1d10adb | 200 | |
df6fb868 | 201 | nla_nest_end(skb, nest_helper); |
3c158f7f PM |
202 | out: |
203 | rcu_read_unlock(); | |
c1d10adb PNA |
204 | return 0; |
205 | ||
df6fb868 | 206 | nla_put_failure: |
3c158f7f | 207 | rcu_read_unlock(); |
c1d10adb PNA |
208 | return -1; |
209 | } | |
210 | ||
211 | #ifdef CONFIG_NF_CT_ACCT | |
212 | static inline int | |
213 | ctnetlink_dump_counters(struct sk_buff *skb, const struct nf_conn *ct, | |
214 | enum ip_conntrack_dir dir) | |
215 | { | |
216 | enum ctattr_type type = dir ? CTA_COUNTERS_REPLY: CTA_COUNTERS_ORIG; | |
df6fb868 | 217 | struct nlattr *nest_count; |
bff9a89b | 218 | __be32 tmp; |
c1d10adb | 219 | |
df6fb868 PM |
220 | nest_count = nla_nest_start(skb, type | NLA_F_NESTED); |
221 | if (!nest_count) | |
222 | goto nla_put_failure; | |
223 | ||
c1d10adb | 224 | tmp = htonl(ct->counters[dir].packets); |
df6fb868 | 225 | NLA_PUT(skb, CTA_COUNTERS32_PACKETS, sizeof(u_int32_t), &tmp); |
c1d10adb PNA |
226 | |
227 | tmp = htonl(ct->counters[dir].bytes); | |
df6fb868 | 228 | NLA_PUT(skb, CTA_COUNTERS32_BYTES, sizeof(u_int32_t), &tmp); |
c1d10adb | 229 | |
df6fb868 | 230 | nla_nest_end(skb, nest_count); |
c1d10adb PNA |
231 | |
232 | return 0; | |
233 | ||
df6fb868 | 234 | nla_put_failure: |
c1d10adb PNA |
235 | return -1; |
236 | } | |
237 | #else | |
238 | #define ctnetlink_dump_counters(a, b, c) (0) | |
239 | #endif | |
240 | ||
241 | #ifdef CONFIG_NF_CONNTRACK_MARK | |
242 | static inline int | |
243 | ctnetlink_dump_mark(struct sk_buff *skb, const struct nf_conn *ct) | |
244 | { | |
bff9a89b | 245 | __be32 mark = htonl(ct->mark); |
c1d10adb | 246 | |
df6fb868 | 247 | NLA_PUT(skb, CTA_MARK, sizeof(u_int32_t), &mark); |
c1d10adb PNA |
248 | return 0; |
249 | ||
df6fb868 | 250 | nla_put_failure: |
c1d10adb PNA |
251 | return -1; |
252 | } | |
253 | #else | |
254 | #define ctnetlink_dump_mark(a, b) (0) | |
255 | #endif | |
256 | ||
257 | static inline int | |
258 | ctnetlink_dump_id(struct sk_buff *skb, const struct nf_conn *ct) | |
259 | { | |
7f85f914 | 260 | __be32 id = htonl((unsigned long)ct); |
df6fb868 | 261 | NLA_PUT(skb, CTA_ID, sizeof(u_int32_t), &id); |
c1d10adb PNA |
262 | return 0; |
263 | ||
df6fb868 | 264 | nla_put_failure: |
c1d10adb PNA |
265 | return -1; |
266 | } | |
267 | ||
268 | static inline int | |
269 | ctnetlink_dump_use(struct sk_buff *skb, const struct nf_conn *ct) | |
270 | { | |
bff9a89b | 271 | __be32 use = htonl(atomic_read(&ct->ct_general.use)); |
601e68e1 | 272 | |
df6fb868 | 273 | NLA_PUT(skb, CTA_USE, sizeof(u_int32_t), &use); |
c1d10adb PNA |
274 | return 0; |
275 | ||
df6fb868 | 276 | nla_put_failure: |
c1d10adb PNA |
277 | return -1; |
278 | } | |
279 | ||
280 | #define tuple(ct, dir) (&(ct)->tuplehash[dir].tuple) | |
281 | ||
282 | static int | |
283 | ctnetlink_fill_info(struct sk_buff *skb, u32 pid, u32 seq, | |
601e68e1 | 284 | int event, int nowait, |
c1d10adb PNA |
285 | const struct nf_conn *ct) |
286 | { | |
287 | struct nlmsghdr *nlh; | |
288 | struct nfgenmsg *nfmsg; | |
df6fb868 | 289 | struct nlattr *nest_parms; |
27a884dc | 290 | unsigned char *b = skb_tail_pointer(skb); |
c1d10adb PNA |
291 | |
292 | event |= NFNL_SUBSYS_CTNETLINK << 8; | |
293 | nlh = NLMSG_PUT(skb, pid, seq, event, sizeof(struct nfgenmsg)); | |
294 | nfmsg = NLMSG_DATA(nlh); | |
295 | ||
296 | nlh->nlmsg_flags = (nowait && pid) ? NLM_F_MULTI : 0; | |
601e68e1 | 297 | nfmsg->nfgen_family = |
c1d10adb PNA |
298 | ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.l3num; |
299 | nfmsg->version = NFNETLINK_V0; | |
300 | nfmsg->res_id = 0; | |
301 | ||
df6fb868 PM |
302 | nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED); |
303 | if (!nest_parms) | |
304 | goto nla_put_failure; | |
c1d10adb | 305 | if (ctnetlink_dump_tuples(skb, tuple(ct, IP_CT_DIR_ORIGINAL)) < 0) |
df6fb868 PM |
306 | goto nla_put_failure; |
307 | nla_nest_end(skb, nest_parms); | |
601e68e1 | 308 | |
df6fb868 PM |
309 | nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED); |
310 | if (!nest_parms) | |
311 | goto nla_put_failure; | |
c1d10adb | 312 | if (ctnetlink_dump_tuples(skb, tuple(ct, IP_CT_DIR_REPLY)) < 0) |
df6fb868 PM |
313 | goto nla_put_failure; |
314 | nla_nest_end(skb, nest_parms); | |
c1d10adb PNA |
315 | |
316 | if (ctnetlink_dump_status(skb, ct) < 0 || | |
317 | ctnetlink_dump_timeout(skb, ct) < 0 || | |
318 | ctnetlink_dump_counters(skb, ct, IP_CT_DIR_ORIGINAL) < 0 || | |
319 | ctnetlink_dump_counters(skb, ct, IP_CT_DIR_REPLY) < 0 || | |
320 | ctnetlink_dump_protoinfo(skb, ct) < 0 || | |
321 | ctnetlink_dump_helpinfo(skb, ct) < 0 || | |
322 | ctnetlink_dump_mark(skb, ct) < 0 || | |
323 | ctnetlink_dump_id(skb, ct) < 0 || | |
324 | ctnetlink_dump_use(skb, ct) < 0) | |
df6fb868 | 325 | goto nla_put_failure; |
c1d10adb | 326 | |
27a884dc | 327 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; |
c1d10adb PNA |
328 | return skb->len; |
329 | ||
330 | nlmsg_failure: | |
df6fb868 | 331 | nla_put_failure: |
dc5fc579 | 332 | nlmsg_trim(skb, b); |
c1d10adb PNA |
333 | return -1; |
334 | } | |
335 | ||
336 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
337 | static int ctnetlink_conntrack_event(struct notifier_block *this, | |
601e68e1 | 338 | unsigned long events, void *ptr) |
c1d10adb PNA |
339 | { |
340 | struct nlmsghdr *nlh; | |
341 | struct nfgenmsg *nfmsg; | |
df6fb868 | 342 | struct nlattr *nest_parms; |
c1d10adb PNA |
343 | struct nf_conn *ct = (struct nf_conn *)ptr; |
344 | struct sk_buff *skb; | |
345 | unsigned int type; | |
27a884dc | 346 | sk_buff_data_t b; |
c1d10adb PNA |
347 | unsigned int flags = 0, group; |
348 | ||
349 | /* ignore our fake conntrack entry */ | |
350 | if (ct == &nf_conntrack_untracked) | |
351 | return NOTIFY_DONE; | |
352 | ||
353 | if (events & IPCT_DESTROY) { | |
354 | type = IPCTNL_MSG_CT_DELETE; | |
355 | group = NFNLGRP_CONNTRACK_DESTROY; | |
356 | } else if (events & (IPCT_NEW | IPCT_RELATED)) { | |
357 | type = IPCTNL_MSG_CT_NEW; | |
358 | flags = NLM_F_CREATE|NLM_F_EXCL; | |
c1d10adb | 359 | group = NFNLGRP_CONNTRACK_NEW; |
1a31526b | 360 | } else if (events & (IPCT_STATUS | IPCT_PROTOINFO)) { |
c1d10adb PNA |
361 | type = IPCTNL_MSG_CT_NEW; |
362 | group = NFNLGRP_CONNTRACK_UPDATE; | |
363 | } else | |
364 | return NOTIFY_DONE; | |
a2427692 PM |
365 | |
366 | if (!nfnetlink_has_listeners(group)) | |
367 | return NOTIFY_DONE; | |
368 | ||
c1d10adb PNA |
369 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC); |
370 | if (!skb) | |
371 | return NOTIFY_DONE; | |
372 | ||
373 | b = skb->tail; | |
374 | ||
375 | type |= NFNL_SUBSYS_CTNETLINK << 8; | |
376 | nlh = NLMSG_PUT(skb, 0, 0, type, sizeof(struct nfgenmsg)); | |
377 | nfmsg = NLMSG_DATA(nlh); | |
378 | ||
379 | nlh->nlmsg_flags = flags; | |
380 | nfmsg->nfgen_family = ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.l3num; | |
381 | nfmsg->version = NFNETLINK_V0; | |
382 | nfmsg->res_id = 0; | |
383 | ||
df6fb868 PM |
384 | nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED); |
385 | if (!nest_parms) | |
386 | goto nla_put_failure; | |
c1d10adb | 387 | if (ctnetlink_dump_tuples(skb, tuple(ct, IP_CT_DIR_ORIGINAL)) < 0) |
df6fb868 PM |
388 | goto nla_put_failure; |
389 | nla_nest_end(skb, nest_parms); | |
601e68e1 | 390 | |
df6fb868 PM |
391 | nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED); |
392 | if (!nest_parms) | |
393 | goto nla_put_failure; | |
c1d10adb | 394 | if (ctnetlink_dump_tuples(skb, tuple(ct, IP_CT_DIR_REPLY)) < 0) |
df6fb868 PM |
395 | goto nla_put_failure; |
396 | nla_nest_end(skb, nest_parms); | |
c1d10adb | 397 | |
7b621c1e PNA |
398 | if (events & IPCT_DESTROY) { |
399 | if (ctnetlink_dump_counters(skb, ct, IP_CT_DIR_ORIGINAL) < 0 || | |
400 | ctnetlink_dump_counters(skb, ct, IP_CT_DIR_REPLY) < 0) | |
df6fb868 | 401 | goto nla_put_failure; |
7b621c1e PNA |
402 | } else { |
403 | if (ctnetlink_dump_status(skb, ct) < 0) | |
df6fb868 | 404 | goto nla_put_failure; |
7b621c1e PNA |
405 | |
406 | if (ctnetlink_dump_timeout(skb, ct) < 0) | |
df6fb868 | 407 | goto nla_put_failure; |
7b621c1e PNA |
408 | |
409 | if (events & IPCT_PROTOINFO | |
410 | && ctnetlink_dump_protoinfo(skb, ct) < 0) | |
df6fb868 | 411 | goto nla_put_failure; |
7b621c1e PNA |
412 | |
413 | if ((events & IPCT_HELPER || nfct_help(ct)) | |
414 | && ctnetlink_dump_helpinfo(skb, ct) < 0) | |
df6fb868 | 415 | goto nla_put_failure; |
7b621c1e | 416 | |
40e0cb00 | 417 | #ifdef CONFIG_NF_CONNTRACK_MARK |
7b621c1e PNA |
418 | if ((events & IPCT_MARK || ct->mark) |
419 | && ctnetlink_dump_mark(skb, ct) < 0) | |
df6fb868 | 420 | goto nla_put_failure; |
40e0cb00 | 421 | #endif |
7b621c1e PNA |
422 | |
423 | if (events & IPCT_COUNTER_FILLING && | |
424 | (ctnetlink_dump_counters(skb, ct, IP_CT_DIR_ORIGINAL) < 0 || | |
425 | ctnetlink_dump_counters(skb, ct, IP_CT_DIR_REPLY) < 0)) | |
df6fb868 | 426 | goto nla_put_failure; |
7b621c1e | 427 | } |
b9a37e0c | 428 | |
c1d10adb PNA |
429 | nlh->nlmsg_len = skb->tail - b; |
430 | nfnetlink_send(skb, 0, group, 0); | |
431 | return NOTIFY_DONE; | |
432 | ||
433 | nlmsg_failure: | |
df6fb868 | 434 | nla_put_failure: |
c1d10adb PNA |
435 | kfree_skb(skb); |
436 | return NOTIFY_DONE; | |
437 | } | |
438 | #endif /* CONFIG_NF_CONNTRACK_EVENTS */ | |
439 | ||
440 | static int ctnetlink_done(struct netlink_callback *cb) | |
441 | { | |
89f2e218 PM |
442 | if (cb->args[1]) |
443 | nf_ct_put((struct nf_conn *)cb->args[1]); | |
c1d10adb PNA |
444 | return 0; |
445 | } | |
446 | ||
87711cb8 PNA |
447 | #define L3PROTO(ct) ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.l3num |
448 | ||
c1d10adb PNA |
449 | static int |
450 | ctnetlink_dump_table(struct sk_buff *skb, struct netlink_callback *cb) | |
451 | { | |
89f2e218 | 452 | struct nf_conn *ct, *last; |
c1d10adb | 453 | struct nf_conntrack_tuple_hash *h; |
f205c5e0 | 454 | struct hlist_node *n; |
87711cb8 PNA |
455 | struct nfgenmsg *nfmsg = NLMSG_DATA(cb->nlh); |
456 | u_int8_t l3proto = nfmsg->nfgen_family; | |
c1d10adb | 457 | |
c1d10adb | 458 | read_lock_bh(&nf_conntrack_lock); |
d205dc40 | 459 | last = (struct nf_conn *)cb->args[1]; |
89f2e218 PM |
460 | for (; cb->args[0] < nf_conntrack_htable_size; cb->args[0]++) { |
461 | restart: | |
f205c5e0 PM |
462 | hlist_for_each_entry(h, n, &nf_conntrack_hash[cb->args[0]], |
463 | hnode) { | |
5b1158e9 | 464 | if (NF_CT_DIRECTION(h) != IP_CT_DIR_ORIGINAL) |
c1d10adb PNA |
465 | continue; |
466 | ct = nf_ct_tuplehash_to_ctrack(h); | |
87711cb8 PNA |
467 | /* Dump entries of a given L3 protocol number. |
468 | * If it is not specified, ie. l3proto == 0, | |
469 | * then dump everything. */ | |
470 | if (l3proto && L3PROTO(ct) != l3proto) | |
471 | continue; | |
d205dc40 PM |
472 | if (cb->args[1]) { |
473 | if (ct != last) | |
89f2e218 | 474 | continue; |
d205dc40 | 475 | cb->args[1] = 0; |
89f2e218 | 476 | } |
c1d10adb | 477 | if (ctnetlink_fill_info(skb, NETLINK_CB(cb->skb).pid, |
601e68e1 | 478 | cb->nlh->nlmsg_seq, |
c1d10adb | 479 | IPCTNL_MSG_CT_NEW, |
89f2e218 PM |
480 | 1, ct) < 0) { |
481 | nf_conntrack_get(&ct->ct_general); | |
482 | cb->args[1] = (unsigned long)ct; | |
c1d10adb | 483 | goto out; |
89f2e218 | 484 | } |
01f34848 PNA |
485 | #ifdef CONFIG_NF_CT_ACCT |
486 | if (NFNL_MSG_TYPE(cb->nlh->nlmsg_type) == | |
487 | IPCTNL_MSG_CT_GET_CTRZERO) | |
488 | memset(&ct->counters, 0, sizeof(ct->counters)); | |
489 | #endif | |
89f2e218 | 490 | } |
d205dc40 | 491 | if (cb->args[1]) { |
89f2e218 PM |
492 | cb->args[1] = 0; |
493 | goto restart; | |
c1d10adb PNA |
494 | } |
495 | } | |
89f2e218 | 496 | out: |
c1d10adb | 497 | read_unlock_bh(&nf_conntrack_lock); |
d205dc40 PM |
498 | if (last) |
499 | nf_ct_put(last); | |
c1d10adb | 500 | |
c1d10adb PNA |
501 | return skb->len; |
502 | } | |
503 | ||
c1d10adb | 504 | static inline int |
df6fb868 | 505 | ctnetlink_parse_tuple_ip(struct nlattr *attr, struct nf_conntrack_tuple *tuple) |
c1d10adb | 506 | { |
df6fb868 | 507 | struct nlattr *tb[CTA_IP_MAX+1]; |
c1d10adb PNA |
508 | struct nf_conntrack_l3proto *l3proto; |
509 | int ret = 0; | |
510 | ||
df6fb868 | 511 | nla_parse_nested(tb, CTA_IP_MAX, attr, NULL); |
c1d10adb PNA |
512 | |
513 | l3proto = nf_ct_l3proto_find_get(tuple->src.l3num); | |
514 | ||
f73e924c PM |
515 | if (likely(l3proto->nlattr_to_tuple)) { |
516 | ret = nla_validate_nested(attr, CTA_IP_MAX, | |
517 | l3proto->nla_policy); | |
518 | if (ret == 0) | |
519 | ret = l3proto->nlattr_to_tuple(tb, tuple); | |
520 | } | |
c1d10adb PNA |
521 | |
522 | nf_ct_l3proto_put(l3proto); | |
523 | ||
c1d10adb PNA |
524 | return ret; |
525 | } | |
526 | ||
f73e924c PM |
527 | static const struct nla_policy proto_nla_policy[CTA_PROTO_MAX+1] = { |
528 | [CTA_PROTO_NUM] = { .type = NLA_U8 }, | |
c1d10adb PNA |
529 | }; |
530 | ||
531 | static inline int | |
df6fb868 | 532 | ctnetlink_parse_tuple_proto(struct nlattr *attr, |
c1d10adb PNA |
533 | struct nf_conntrack_tuple *tuple) |
534 | { | |
df6fb868 | 535 | struct nlattr *tb[CTA_PROTO_MAX+1]; |
605dcad6 | 536 | struct nf_conntrack_l4proto *l4proto; |
c1d10adb PNA |
537 | int ret = 0; |
538 | ||
f73e924c PM |
539 | ret = nla_parse_nested(tb, CTA_PROTO_MAX, attr, proto_nla_policy); |
540 | if (ret < 0) | |
541 | return ret; | |
c1d10adb | 542 | |
df6fb868 | 543 | if (!tb[CTA_PROTO_NUM]) |
c1d10adb | 544 | return -EINVAL; |
df6fb868 | 545 | tuple->dst.protonum = *(u_int8_t *)nla_data(tb[CTA_PROTO_NUM]); |
c1d10adb | 546 | |
605dcad6 | 547 | l4proto = nf_ct_l4proto_find_get(tuple->src.l3num, tuple->dst.protonum); |
c1d10adb | 548 | |
f73e924c PM |
549 | if (likely(l4proto->nlattr_to_tuple)) { |
550 | ret = nla_validate_nested(attr, CTA_PROTO_MAX, | |
551 | l4proto->nla_policy); | |
552 | if (ret == 0) | |
553 | ret = l4proto->nlattr_to_tuple(tb, tuple); | |
554 | } | |
c1d10adb | 555 | |
605dcad6 | 556 | nf_ct_l4proto_put(l4proto); |
601e68e1 | 557 | |
c1d10adb PNA |
558 | return ret; |
559 | } | |
560 | ||
561 | static inline int | |
df6fb868 | 562 | ctnetlink_parse_tuple(struct nlattr *cda[], struct nf_conntrack_tuple *tuple, |
c1d10adb PNA |
563 | enum ctattr_tuple type, u_int8_t l3num) |
564 | { | |
df6fb868 | 565 | struct nlattr *tb[CTA_TUPLE_MAX+1]; |
c1d10adb PNA |
566 | int err; |
567 | ||
c1d10adb PNA |
568 | memset(tuple, 0, sizeof(*tuple)); |
569 | ||
df6fb868 | 570 | nla_parse_nested(tb, CTA_TUPLE_MAX, cda[type], NULL); |
c1d10adb | 571 | |
df6fb868 | 572 | if (!tb[CTA_TUPLE_IP]) |
c1d10adb PNA |
573 | return -EINVAL; |
574 | ||
575 | tuple->src.l3num = l3num; | |
576 | ||
df6fb868 | 577 | err = ctnetlink_parse_tuple_ip(tb[CTA_TUPLE_IP], tuple); |
c1d10adb PNA |
578 | if (err < 0) |
579 | return err; | |
580 | ||
df6fb868 | 581 | if (!tb[CTA_TUPLE_PROTO]) |
c1d10adb PNA |
582 | return -EINVAL; |
583 | ||
df6fb868 | 584 | err = ctnetlink_parse_tuple_proto(tb[CTA_TUPLE_PROTO], tuple); |
c1d10adb PNA |
585 | if (err < 0) |
586 | return err; | |
587 | ||
588 | /* orig and expect tuples get DIR_ORIGINAL */ | |
589 | if (type == CTA_TUPLE_REPLY) | |
590 | tuple->dst.dir = IP_CT_DIR_REPLY; | |
591 | else | |
592 | tuple->dst.dir = IP_CT_DIR_ORIGINAL; | |
593 | ||
c1d10adb PNA |
594 | return 0; |
595 | } | |
596 | ||
5b1158e9 | 597 | #ifdef CONFIG_NF_NAT_NEEDED |
f73e924c PM |
598 | static const struct nla_policy protonat_nla_policy[CTA_PROTONAT_MAX+1] = { |
599 | [CTA_PROTONAT_PORT_MIN] = { .type = NLA_U16 }, | |
600 | [CTA_PROTONAT_PORT_MAX] = { .type = NLA_U16 }, | |
c1d10adb PNA |
601 | }; |
602 | ||
df6fb868 | 603 | static int nfnetlink_parse_nat_proto(struct nlattr *attr, |
c1d10adb | 604 | const struct nf_conn *ct, |
5b1158e9 | 605 | struct nf_nat_range *range) |
c1d10adb | 606 | { |
df6fb868 | 607 | struct nlattr *tb[CTA_PROTONAT_MAX+1]; |
5b1158e9 | 608 | struct nf_nat_protocol *npt; |
f73e924c | 609 | int err; |
c1d10adb | 610 | |
f73e924c PM |
611 | err = nla_parse_nested(tb, CTA_PROTONAT_MAX, attr, protonat_nla_policy); |
612 | if (err < 0) | |
613 | return err; | |
c1d10adb | 614 | |
5b1158e9 | 615 | npt = nf_nat_proto_find_get(ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.protonum); |
c1d10adb | 616 | |
fdf70832 | 617 | if (!npt->nlattr_to_range) { |
5b1158e9 | 618 | nf_nat_proto_put(npt); |
c1d10adb PNA |
619 | return 0; |
620 | } | |
621 | ||
fdf70832 PM |
622 | /* nlattr_to_range returns 1 if it parsed, 0 if not, neg. on error */ |
623 | if (npt->nlattr_to_range(tb, range) > 0) | |
c1d10adb PNA |
624 | range->flags |= IP_NAT_RANGE_PROTO_SPECIFIED; |
625 | ||
5b1158e9 | 626 | nf_nat_proto_put(npt); |
c1d10adb | 627 | |
c1d10adb PNA |
628 | return 0; |
629 | } | |
630 | ||
f73e924c PM |
631 | static const struct nla_policy nat_nla_policy[CTA_NAT_MAX+1] = { |
632 | [CTA_NAT_MINIP] = { .type = NLA_U32 }, | |
633 | [CTA_NAT_MAXIP] = { .type = NLA_U32 }, | |
c1d10adb PNA |
634 | }; |
635 | ||
636 | static inline int | |
df6fb868 | 637 | nfnetlink_parse_nat(struct nlattr *nat, |
5b1158e9 | 638 | const struct nf_conn *ct, struct nf_nat_range *range) |
c1d10adb | 639 | { |
df6fb868 | 640 | struct nlattr *tb[CTA_NAT_MAX+1]; |
c1d10adb PNA |
641 | int err; |
642 | ||
c1d10adb | 643 | memset(range, 0, sizeof(*range)); |
601e68e1 | 644 | |
f73e924c PM |
645 | err = nla_parse_nested(tb, CTA_NAT_MAX, nat, nat_nla_policy); |
646 | if (err < 0) | |
647 | return err; | |
c1d10adb | 648 | |
df6fb868 PM |
649 | if (tb[CTA_NAT_MINIP]) |
650 | range->min_ip = *(__be32 *)nla_data(tb[CTA_NAT_MINIP]); | |
c1d10adb | 651 | |
df6fb868 | 652 | if (!tb[CTA_NAT_MAXIP]) |
c1d10adb PNA |
653 | range->max_ip = range->min_ip; |
654 | else | |
df6fb868 | 655 | range->max_ip = *(__be32 *)nla_data(tb[CTA_NAT_MAXIP]); |
c1d10adb PNA |
656 | |
657 | if (range->min_ip) | |
658 | range->flags |= IP_NAT_RANGE_MAP_IPS; | |
659 | ||
df6fb868 | 660 | if (!tb[CTA_NAT_PROTO]) |
c1d10adb PNA |
661 | return 0; |
662 | ||
df6fb868 | 663 | err = nfnetlink_parse_nat_proto(tb[CTA_NAT_PROTO], ct, range); |
c1d10adb PNA |
664 | if (err < 0) |
665 | return err; | |
666 | ||
c1d10adb PNA |
667 | return 0; |
668 | } | |
669 | #endif | |
670 | ||
671 | static inline int | |
df6fb868 | 672 | ctnetlink_parse_help(struct nlattr *attr, char **helper_name) |
c1d10adb | 673 | { |
df6fb868 | 674 | struct nlattr *tb[CTA_HELP_MAX+1]; |
c1d10adb | 675 | |
df6fb868 | 676 | nla_parse_nested(tb, CTA_HELP_MAX, attr, NULL); |
c1d10adb | 677 | |
df6fb868 | 678 | if (!tb[CTA_HELP_NAME]) |
c1d10adb PNA |
679 | return -EINVAL; |
680 | ||
df6fb868 | 681 | *helper_name = nla_data(tb[CTA_HELP_NAME]); |
c1d10adb PNA |
682 | |
683 | return 0; | |
684 | } | |
685 | ||
f73e924c PM |
686 | static const struct nla_policy ct_nla_policy[CTA_MAX+1] = { |
687 | [CTA_STATUS] = { .type = NLA_U32 }, | |
688 | [CTA_TIMEOUT] = { .type = NLA_U32 }, | |
689 | [CTA_MARK] = { .type = NLA_U32 }, | |
690 | [CTA_USE] = { .type = NLA_U32 }, | |
691 | [CTA_ID] = { .type = NLA_U32 }, | |
c1d10adb PNA |
692 | }; |
693 | ||
694 | static int | |
601e68e1 | 695 | ctnetlink_del_conntrack(struct sock *ctnl, struct sk_buff *skb, |
df6fb868 | 696 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
697 | { |
698 | struct nf_conntrack_tuple_hash *h; | |
699 | struct nf_conntrack_tuple tuple; | |
700 | struct nf_conn *ct; | |
701 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
702 | u_int8_t u3 = nfmsg->nfgen_family; | |
703 | int err = 0; | |
704 | ||
df6fb868 | 705 | if (cda[CTA_TUPLE_ORIG]) |
c1d10adb | 706 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_ORIG, u3); |
df6fb868 | 707 | else if (cda[CTA_TUPLE_REPLY]) |
c1d10adb PNA |
708 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_REPLY, u3); |
709 | else { | |
710 | /* Flush the whole table */ | |
711 | nf_conntrack_flush(); | |
712 | return 0; | |
713 | } | |
714 | ||
715 | if (err < 0) | |
716 | return err; | |
717 | ||
330f7db5 | 718 | h = nf_conntrack_find_get(&tuple); |
9ea8cfd6 | 719 | if (!h) |
c1d10adb | 720 | return -ENOENT; |
c1d10adb PNA |
721 | |
722 | ct = nf_ct_tuplehash_to_ctrack(h); | |
601e68e1 | 723 | |
df6fb868 PM |
724 | if (cda[CTA_ID]) { |
725 | u_int32_t id = ntohl(*(__be32 *)nla_data(cda[CTA_ID])); | |
7f85f914 | 726 | if (id != (u32)(unsigned long)ct) { |
c1d10adb PNA |
727 | nf_ct_put(ct); |
728 | return -ENOENT; | |
729 | } | |
601e68e1 | 730 | } |
c1d10adb PNA |
731 | if (del_timer(&ct->timeout)) |
732 | ct->timeout.function((unsigned long)ct); | |
733 | ||
734 | nf_ct_put(ct); | |
c1d10adb PNA |
735 | |
736 | return 0; | |
737 | } | |
738 | ||
739 | static int | |
601e68e1 | 740 | ctnetlink_get_conntrack(struct sock *ctnl, struct sk_buff *skb, |
df6fb868 | 741 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
742 | { |
743 | struct nf_conntrack_tuple_hash *h; | |
744 | struct nf_conntrack_tuple tuple; | |
745 | struct nf_conn *ct; | |
746 | struct sk_buff *skb2 = NULL; | |
747 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
748 | u_int8_t u3 = nfmsg->nfgen_family; | |
749 | int err = 0; | |
750 | ||
c1d10adb | 751 | if (nlh->nlmsg_flags & NLM_F_DUMP) { |
01f34848 PNA |
752 | #ifndef CONFIG_NF_CT_ACCT |
753 | if (NFNL_MSG_TYPE(nlh->nlmsg_type) == IPCTNL_MSG_CT_GET_CTRZERO) | |
c1d10adb PNA |
754 | return -ENOTSUPP; |
755 | #endif | |
c702e804 TG |
756 | return netlink_dump_start(ctnl, skb, nlh, ctnetlink_dump_table, |
757 | ctnetlink_done); | |
c1d10adb PNA |
758 | } |
759 | ||
df6fb868 | 760 | if (cda[CTA_TUPLE_ORIG]) |
c1d10adb | 761 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_ORIG, u3); |
df6fb868 | 762 | else if (cda[CTA_TUPLE_REPLY]) |
c1d10adb PNA |
763 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_REPLY, u3); |
764 | else | |
765 | return -EINVAL; | |
766 | ||
767 | if (err < 0) | |
768 | return err; | |
769 | ||
330f7db5 | 770 | h = nf_conntrack_find_get(&tuple); |
9ea8cfd6 | 771 | if (!h) |
c1d10adb | 772 | return -ENOENT; |
9ea8cfd6 | 773 | |
c1d10adb PNA |
774 | ct = nf_ct_tuplehash_to_ctrack(h); |
775 | ||
776 | err = -ENOMEM; | |
777 | skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | |
778 | if (!skb2) { | |
779 | nf_ct_put(ct); | |
780 | return -ENOMEM; | |
781 | } | |
c1d10adb | 782 | |
601e68e1 | 783 | err = ctnetlink_fill_info(skb2, NETLINK_CB(skb).pid, nlh->nlmsg_seq, |
c1d10adb PNA |
784 | IPCTNL_MSG_CT_NEW, 1, ct); |
785 | nf_ct_put(ct); | |
786 | if (err <= 0) | |
787 | goto free; | |
788 | ||
789 | err = netlink_unicast(ctnl, skb2, NETLINK_CB(skb).pid, MSG_DONTWAIT); | |
790 | if (err < 0) | |
791 | goto out; | |
792 | ||
c1d10adb PNA |
793 | return 0; |
794 | ||
795 | free: | |
796 | kfree_skb(skb2); | |
797 | out: | |
798 | return err; | |
799 | } | |
800 | ||
801 | static inline int | |
df6fb868 | 802 | ctnetlink_change_status(struct nf_conn *ct, struct nlattr *cda[]) |
c1d10adb PNA |
803 | { |
804 | unsigned long d; | |
df6fb868 | 805 | unsigned int status = ntohl(*(__be32 *)nla_data(cda[CTA_STATUS])); |
c1d10adb PNA |
806 | d = ct->status ^ status; |
807 | ||
808 | if (d & (IPS_EXPECTED|IPS_CONFIRMED|IPS_DYING)) | |
809 | /* unchangeable */ | |
810 | return -EINVAL; | |
601e68e1 | 811 | |
c1d10adb PNA |
812 | if (d & IPS_SEEN_REPLY && !(status & IPS_SEEN_REPLY)) |
813 | /* SEEN_REPLY bit can only be set */ | |
814 | return -EINVAL; | |
815 | ||
601e68e1 | 816 | |
c1d10adb PNA |
817 | if (d & IPS_ASSURED && !(status & IPS_ASSURED)) |
818 | /* ASSURED bit can only be set */ | |
819 | return -EINVAL; | |
820 | ||
df6fb868 | 821 | if (cda[CTA_NAT_SRC] || cda[CTA_NAT_DST]) { |
5b1158e9 | 822 | #ifndef CONFIG_NF_NAT_NEEDED |
c1d10adb PNA |
823 | return -EINVAL; |
824 | #else | |
5b1158e9 | 825 | struct nf_nat_range range; |
c1d10adb | 826 | |
df6fb868 PM |
827 | if (cda[CTA_NAT_DST]) { |
828 | if (nfnetlink_parse_nat(cda[CTA_NAT_DST], ct, | |
3726add7 PM |
829 | &range) < 0) |
830 | return -EINVAL; | |
5b1158e9 | 831 | if (nf_nat_initialized(ct, |
3726add7 PM |
832 | HOOK2MANIP(NF_IP_PRE_ROUTING))) |
833 | return -EEXIST; | |
5b1158e9 | 834 | nf_nat_setup_info(ct, &range, NF_IP_PRE_ROUTING); |
3726add7 | 835 | } |
df6fb868 PM |
836 | if (cda[CTA_NAT_SRC]) { |
837 | if (nfnetlink_parse_nat(cda[CTA_NAT_SRC], ct, | |
3726add7 PM |
838 | &range) < 0) |
839 | return -EINVAL; | |
5b1158e9 | 840 | if (nf_nat_initialized(ct, |
3726add7 PM |
841 | HOOK2MANIP(NF_IP_POST_ROUTING))) |
842 | return -EEXIST; | |
5b1158e9 | 843 | nf_nat_setup_info(ct, &range, NF_IP_POST_ROUTING); |
3726add7 | 844 | } |
c1d10adb PNA |
845 | #endif |
846 | } | |
847 | ||
848 | /* Be careful here, modifying NAT bits can screw up things, | |
849 | * so don't let users modify them directly if they don't pass | |
5b1158e9 | 850 | * nf_nat_range. */ |
c1d10adb PNA |
851 | ct->status |= status & ~(IPS_NAT_DONE_MASK | IPS_NAT_MASK); |
852 | return 0; | |
853 | } | |
854 | ||
855 | ||
856 | static inline int | |
df6fb868 | 857 | ctnetlink_change_helper(struct nf_conn *ct, struct nlattr *cda[]) |
c1d10adb PNA |
858 | { |
859 | struct nf_conntrack_helper *helper; | |
dc808fe2 | 860 | struct nf_conn_help *help = nfct_help(ct); |
c1d10adb PNA |
861 | char *helpname; |
862 | int err; | |
863 | ||
c1d10adb PNA |
864 | /* don't change helper of sibling connections */ |
865 | if (ct->master) | |
866 | return -EINVAL; | |
867 | ||
df6fb868 | 868 | err = ctnetlink_parse_help(cda[CTA_HELP], &helpname); |
c1d10adb PNA |
869 | if (err < 0) |
870 | return err; | |
871 | ||
df293bbb YK |
872 | if (!strcmp(helpname, "")) { |
873 | if (help && help->helper) { | |
c1d10adb PNA |
874 | /* we had a helper before ... */ |
875 | nf_ct_remove_expectations(ct); | |
3c158f7f | 876 | rcu_assign_pointer(help->helper, NULL); |
c1d10adb | 877 | } |
df293bbb YK |
878 | |
879 | return 0; | |
c1d10adb | 880 | } |
601e68e1 | 881 | |
df293bbb YK |
882 | helper = __nf_conntrack_helper_find_byname(helpname); |
883 | if (helper == NULL) | |
884 | return -EINVAL; | |
885 | ||
ceceae1b YK |
886 | if (help) { |
887 | if (help->helper == helper) | |
888 | return 0; | |
889 | if (help->helper) | |
890 | return -EBUSY; | |
891 | /* need to zero data of old helper */ | |
892 | memset(&help->help, 0, sizeof(help->help)); | |
893 | } else { | |
b560580a | 894 | help = nf_ct_helper_ext_add(ct, GFP_KERNEL); |
ceceae1b YK |
895 | if (help == NULL) |
896 | return -ENOMEM; | |
897 | } | |
df293bbb | 898 | |
3c158f7f | 899 | rcu_assign_pointer(help->helper, helper); |
c1d10adb PNA |
900 | |
901 | return 0; | |
902 | } | |
903 | ||
904 | static inline int | |
df6fb868 | 905 | ctnetlink_change_timeout(struct nf_conn *ct, struct nlattr *cda[]) |
c1d10adb | 906 | { |
df6fb868 | 907 | u_int32_t timeout = ntohl(*(__be32 *)nla_data(cda[CTA_TIMEOUT])); |
601e68e1 | 908 | |
c1d10adb PNA |
909 | if (!del_timer(&ct->timeout)) |
910 | return -ETIME; | |
911 | ||
912 | ct->timeout.expires = jiffies + timeout * HZ; | |
913 | add_timer(&ct->timeout); | |
914 | ||
915 | return 0; | |
916 | } | |
917 | ||
918 | static inline int | |
df6fb868 | 919 | ctnetlink_change_protoinfo(struct nf_conn *ct, struct nlattr *cda[]) |
c1d10adb | 920 | { |
df6fb868 | 921 | struct nlattr *tb[CTA_PROTOINFO_MAX+1], *attr = cda[CTA_PROTOINFO]; |
605dcad6 | 922 | struct nf_conntrack_l4proto *l4proto; |
c1d10adb PNA |
923 | u_int16_t npt = ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.protonum; |
924 | u_int16_t l3num = ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.src.l3num; | |
925 | int err = 0; | |
926 | ||
df6fb868 | 927 | nla_parse_nested(tb, CTA_PROTOINFO_MAX, attr, NULL); |
c1d10adb | 928 | |
605dcad6 | 929 | l4proto = nf_ct_l4proto_find_get(l3num, npt); |
c1d10adb | 930 | |
fdf70832 PM |
931 | if (l4proto->from_nlattr) |
932 | err = l4proto->from_nlattr(tb, ct); | |
605dcad6 | 933 | nf_ct_l4proto_put(l4proto); |
c1d10adb PNA |
934 | |
935 | return err; | |
936 | } | |
937 | ||
938 | static int | |
df6fb868 | 939 | ctnetlink_change_conntrack(struct nf_conn *ct, struct nlattr *cda[]) |
c1d10adb PNA |
940 | { |
941 | int err; | |
942 | ||
df6fb868 | 943 | if (cda[CTA_HELP]) { |
c1d10adb PNA |
944 | err = ctnetlink_change_helper(ct, cda); |
945 | if (err < 0) | |
946 | return err; | |
947 | } | |
948 | ||
df6fb868 | 949 | if (cda[CTA_TIMEOUT]) { |
c1d10adb PNA |
950 | err = ctnetlink_change_timeout(ct, cda); |
951 | if (err < 0) | |
952 | return err; | |
953 | } | |
954 | ||
df6fb868 | 955 | if (cda[CTA_STATUS]) { |
c1d10adb PNA |
956 | err = ctnetlink_change_status(ct, cda); |
957 | if (err < 0) | |
958 | return err; | |
959 | } | |
960 | ||
df6fb868 | 961 | if (cda[CTA_PROTOINFO]) { |
c1d10adb PNA |
962 | err = ctnetlink_change_protoinfo(ct, cda); |
963 | if (err < 0) | |
964 | return err; | |
965 | } | |
966 | ||
bcd1e830 | 967 | #if defined(CONFIG_NF_CONNTRACK_MARK) |
df6fb868 PM |
968 | if (cda[CTA_MARK]) |
969 | ct->mark = ntohl(*(__be32 *)nla_data(cda[CTA_MARK])); | |
c1d10adb PNA |
970 | #endif |
971 | ||
c1d10adb PNA |
972 | return 0; |
973 | } | |
974 | ||
975 | static int | |
df6fb868 | 976 | ctnetlink_create_conntrack(struct nlattr *cda[], |
c1d10adb | 977 | struct nf_conntrack_tuple *otuple, |
5faa1f4c PNA |
978 | struct nf_conntrack_tuple *rtuple, |
979 | struct nf_conn *master_ct) | |
c1d10adb PNA |
980 | { |
981 | struct nf_conn *ct; | |
982 | int err = -EINVAL; | |
dafc741c | 983 | struct nf_conn_help *help; |
ceceae1b | 984 | struct nf_conntrack_helper *helper; |
c1d10adb | 985 | |
c1d10adb PNA |
986 | ct = nf_conntrack_alloc(otuple, rtuple); |
987 | if (ct == NULL || IS_ERR(ct)) | |
601e68e1 | 988 | return -ENOMEM; |
c1d10adb | 989 | |
df6fb868 | 990 | if (!cda[CTA_TIMEOUT]) |
c1d10adb | 991 | goto err; |
df6fb868 | 992 | ct->timeout.expires = ntohl(*(__be32 *)nla_data(cda[CTA_TIMEOUT])); |
c1d10adb PNA |
993 | |
994 | ct->timeout.expires = jiffies + ct->timeout.expires * HZ; | |
995 | ct->status |= IPS_CONFIRMED; | |
996 | ||
df6fb868 | 997 | if (cda[CTA_STATUS]) { |
bbb3357d PNA |
998 | err = ctnetlink_change_status(ct, cda); |
999 | if (err < 0) | |
1000 | goto err; | |
1001 | } | |
c1d10adb | 1002 | |
df6fb868 | 1003 | if (cda[CTA_PROTOINFO]) { |
c1d10adb PNA |
1004 | err = ctnetlink_change_protoinfo(ct, cda); |
1005 | if (err < 0) | |
c54ea3b9 | 1006 | goto err; |
c1d10adb PNA |
1007 | } |
1008 | ||
bcd1e830 | 1009 | #if defined(CONFIG_NF_CONNTRACK_MARK) |
df6fb868 PM |
1010 | if (cda[CTA_MARK]) |
1011 | ct->mark = ntohl(*(__be32 *)nla_data(cda[CTA_MARK])); | |
c1d10adb PNA |
1012 | #endif |
1013 | ||
ceceae1b YK |
1014 | helper = nf_ct_helper_find_get(rtuple); |
1015 | if (helper) { | |
b560580a | 1016 | help = nf_ct_helper_ext_add(ct, GFP_KERNEL); |
ceceae1b YK |
1017 | if (help == NULL) { |
1018 | nf_ct_helper_put(helper); | |
1019 | err = -ENOMEM; | |
1020 | goto err; | |
1021 | } | |
3c158f7f PM |
1022 | /* not in hash table yet so not strictly necessary */ |
1023 | rcu_assign_pointer(help->helper, helper); | |
1024 | } | |
dafc741c | 1025 | |
5faa1f4c PNA |
1026 | /* setup master conntrack: this is a confirmed expectation */ |
1027 | if (master_ct) | |
1028 | ct->master = master_ct; | |
1029 | ||
c1d10adb PNA |
1030 | add_timer(&ct->timeout); |
1031 | nf_conntrack_hash_insert(ct); | |
1032 | ||
3c158f7f PM |
1033 | if (helper) |
1034 | nf_ct_helper_put(helper); | |
dafc741c | 1035 | |
c1d10adb PNA |
1036 | return 0; |
1037 | ||
601e68e1 | 1038 | err: |
c1d10adb PNA |
1039 | nf_conntrack_free(ct); |
1040 | return err; | |
1041 | } | |
1042 | ||
601e68e1 YH |
1043 | static int |
1044 | ctnetlink_new_conntrack(struct sock *ctnl, struct sk_buff *skb, | |
df6fb868 | 1045 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
1046 | { |
1047 | struct nf_conntrack_tuple otuple, rtuple; | |
1048 | struct nf_conntrack_tuple_hash *h = NULL; | |
1049 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
1050 | u_int8_t u3 = nfmsg->nfgen_family; | |
1051 | int err = 0; | |
1052 | ||
df6fb868 | 1053 | if (cda[CTA_TUPLE_ORIG]) { |
c1d10adb PNA |
1054 | err = ctnetlink_parse_tuple(cda, &otuple, CTA_TUPLE_ORIG, u3); |
1055 | if (err < 0) | |
1056 | return err; | |
1057 | } | |
1058 | ||
df6fb868 | 1059 | if (cda[CTA_TUPLE_REPLY]) { |
c1d10adb PNA |
1060 | err = ctnetlink_parse_tuple(cda, &rtuple, CTA_TUPLE_REPLY, u3); |
1061 | if (err < 0) | |
1062 | return err; | |
1063 | } | |
1064 | ||
1065 | write_lock_bh(&nf_conntrack_lock); | |
df6fb868 | 1066 | if (cda[CTA_TUPLE_ORIG]) |
c1d10adb | 1067 | h = __nf_conntrack_find(&otuple, NULL); |
df6fb868 | 1068 | else if (cda[CTA_TUPLE_REPLY]) |
c1d10adb PNA |
1069 | h = __nf_conntrack_find(&rtuple, NULL); |
1070 | ||
1071 | if (h == NULL) { | |
5faa1f4c PNA |
1072 | struct nf_conntrack_tuple master; |
1073 | struct nf_conntrack_tuple_hash *master_h = NULL; | |
1074 | struct nf_conn *master_ct = NULL; | |
1075 | ||
1076 | if (cda[CTA_TUPLE_MASTER]) { | |
1077 | err = ctnetlink_parse_tuple(cda, | |
1078 | &master, | |
1079 | CTA_TUPLE_MASTER, | |
1080 | u3); | |
1081 | if (err < 0) | |
1082 | return err; | |
1083 | ||
1084 | master_h = __nf_conntrack_find(&master, NULL); | |
1085 | if (master_h == NULL) { | |
1086 | err = -ENOENT; | |
1087 | goto out_unlock; | |
1088 | } | |
1089 | master_ct = nf_ct_tuplehash_to_ctrack(master_h); | |
1090 | atomic_inc(&master_ct->ct_general.use); | |
1091 | } | |
1092 | ||
c1d10adb | 1093 | write_unlock_bh(&nf_conntrack_lock); |
c1d10adb PNA |
1094 | err = -ENOENT; |
1095 | if (nlh->nlmsg_flags & NLM_F_CREATE) | |
5faa1f4c PNA |
1096 | err = ctnetlink_create_conntrack(cda, |
1097 | &otuple, | |
1098 | &rtuple, | |
1099 | master_ct); | |
1100 | if (err < 0 && master_ct) | |
1101 | nf_ct_put(master_ct); | |
1102 | ||
c1d10adb PNA |
1103 | return err; |
1104 | } | |
1105 | /* implicit 'else' */ | |
1106 | ||
c1d10adb PNA |
1107 | /* We manipulate the conntrack inside the global conntrack table lock, |
1108 | * so there's no need to increase the refcount */ | |
c1d10adb | 1109 | err = -EEXIST; |
ff4ca827 PNA |
1110 | if (!(nlh->nlmsg_flags & NLM_F_EXCL)) { |
1111 | /* we only allow nat config for new conntracks */ | |
df6fb868 | 1112 | if (cda[CTA_NAT_SRC] || cda[CTA_NAT_DST]) { |
ff4ca827 PNA |
1113 | err = -EINVAL; |
1114 | goto out_unlock; | |
1115 | } | |
5faa1f4c PNA |
1116 | /* can't link an existing conntrack to a master */ |
1117 | if (cda[CTA_TUPLE_MASTER]) { | |
1118 | err = -EINVAL; | |
1119 | goto out_unlock; | |
1120 | } | |
ff4ca827 PNA |
1121 | err = ctnetlink_change_conntrack(nf_ct_tuplehash_to_ctrack(h), |
1122 | cda); | |
1123 | } | |
c1d10adb PNA |
1124 | |
1125 | out_unlock: | |
1126 | write_unlock_bh(&nf_conntrack_lock); | |
1127 | return err; | |
1128 | } | |
1129 | ||
601e68e1 YH |
1130 | /*********************************************************************** |
1131 | * EXPECT | |
1132 | ***********************************************************************/ | |
c1d10adb PNA |
1133 | |
1134 | static inline int | |
1135 | ctnetlink_exp_dump_tuple(struct sk_buff *skb, | |
1136 | const struct nf_conntrack_tuple *tuple, | |
1137 | enum ctattr_expect type) | |
1138 | { | |
df6fb868 | 1139 | struct nlattr *nest_parms; |
601e68e1 | 1140 | |
df6fb868 PM |
1141 | nest_parms = nla_nest_start(skb, type | NLA_F_NESTED); |
1142 | if (!nest_parms) | |
1143 | goto nla_put_failure; | |
c1d10adb | 1144 | if (ctnetlink_dump_tuples(skb, tuple) < 0) |
df6fb868 PM |
1145 | goto nla_put_failure; |
1146 | nla_nest_end(skb, nest_parms); | |
c1d10adb PNA |
1147 | |
1148 | return 0; | |
1149 | ||
df6fb868 | 1150 | nla_put_failure: |
c1d10adb | 1151 | return -1; |
601e68e1 | 1152 | } |
c1d10adb | 1153 | |
1cde6436 PNA |
1154 | static inline int |
1155 | ctnetlink_exp_dump_mask(struct sk_buff *skb, | |
1156 | const struct nf_conntrack_tuple *tuple, | |
d4156e8c | 1157 | const struct nf_conntrack_tuple_mask *mask) |
1cde6436 PNA |
1158 | { |
1159 | int ret; | |
1160 | struct nf_conntrack_l3proto *l3proto; | |
605dcad6 | 1161 | struct nf_conntrack_l4proto *l4proto; |
d4156e8c | 1162 | struct nf_conntrack_tuple m; |
df6fb868 | 1163 | struct nlattr *nest_parms; |
d4156e8c PM |
1164 | |
1165 | memset(&m, 0xFF, sizeof(m)); | |
1166 | m.src.u.all = mask->src.u.all; | |
1167 | memcpy(&m.src.u3, &mask->src.u3, sizeof(m.src.u3)); | |
1168 | ||
df6fb868 PM |
1169 | nest_parms = nla_nest_start(skb, CTA_EXPECT_MASK | NLA_F_NESTED); |
1170 | if (!nest_parms) | |
1171 | goto nla_put_failure; | |
1cde6436 PNA |
1172 | |
1173 | l3proto = nf_ct_l3proto_find_get(tuple->src.l3num); | |
d4156e8c | 1174 | ret = ctnetlink_dump_tuples_ip(skb, &m, l3proto); |
1cde6436 PNA |
1175 | nf_ct_l3proto_put(l3proto); |
1176 | ||
1177 | if (unlikely(ret < 0)) | |
df6fb868 | 1178 | goto nla_put_failure; |
1cde6436 | 1179 | |
605dcad6 | 1180 | l4proto = nf_ct_l4proto_find_get(tuple->src.l3num, tuple->dst.protonum); |
d4156e8c | 1181 | ret = ctnetlink_dump_tuples_proto(skb, &m, l4proto); |
605dcad6 | 1182 | nf_ct_l4proto_put(l4proto); |
1cde6436 | 1183 | if (unlikely(ret < 0)) |
df6fb868 | 1184 | goto nla_put_failure; |
1cde6436 | 1185 | |
df6fb868 | 1186 | nla_nest_end(skb, nest_parms); |
1cde6436 PNA |
1187 | |
1188 | return 0; | |
1189 | ||
df6fb868 | 1190 | nla_put_failure: |
1cde6436 PNA |
1191 | return -1; |
1192 | } | |
1193 | ||
c1d10adb PNA |
1194 | static inline int |
1195 | ctnetlink_exp_dump_expect(struct sk_buff *skb, | |
601e68e1 | 1196 | const struct nf_conntrack_expect *exp) |
c1d10adb PNA |
1197 | { |
1198 | struct nf_conn *master = exp->master; | |
bff9a89b | 1199 | __be32 timeout = htonl((exp->timeout.expires - jiffies) / HZ); |
35832402 | 1200 | __be32 id = htonl((unsigned long)exp); |
c1d10adb PNA |
1201 | |
1202 | if (ctnetlink_exp_dump_tuple(skb, &exp->tuple, CTA_EXPECT_TUPLE) < 0) | |
df6fb868 | 1203 | goto nla_put_failure; |
1cde6436 | 1204 | if (ctnetlink_exp_dump_mask(skb, &exp->tuple, &exp->mask) < 0) |
df6fb868 | 1205 | goto nla_put_failure; |
c1d10adb PNA |
1206 | if (ctnetlink_exp_dump_tuple(skb, |
1207 | &master->tuplehash[IP_CT_DIR_ORIGINAL].tuple, | |
1208 | CTA_EXPECT_MASTER) < 0) | |
df6fb868 | 1209 | goto nla_put_failure; |
601e68e1 | 1210 | |
df6fb868 PM |
1211 | NLA_PUT(skb, CTA_EXPECT_TIMEOUT, sizeof(timeout), &timeout); |
1212 | NLA_PUT(skb, CTA_EXPECT_ID, sizeof(u_int32_t), &id); | |
c1d10adb PNA |
1213 | |
1214 | return 0; | |
601e68e1 | 1215 | |
df6fb868 | 1216 | nla_put_failure: |
c1d10adb PNA |
1217 | return -1; |
1218 | } | |
1219 | ||
1220 | static int | |
1221 | ctnetlink_exp_fill_info(struct sk_buff *skb, u32 pid, u32 seq, | |
601e68e1 YH |
1222 | int event, |
1223 | int nowait, | |
c1d10adb PNA |
1224 | const struct nf_conntrack_expect *exp) |
1225 | { | |
1226 | struct nlmsghdr *nlh; | |
1227 | struct nfgenmsg *nfmsg; | |
27a884dc | 1228 | unsigned char *b = skb_tail_pointer(skb); |
c1d10adb PNA |
1229 | |
1230 | event |= NFNL_SUBSYS_CTNETLINK_EXP << 8; | |
1231 | nlh = NLMSG_PUT(skb, pid, seq, event, sizeof(struct nfgenmsg)); | |
1232 | nfmsg = NLMSG_DATA(nlh); | |
1233 | ||
1234 | nlh->nlmsg_flags = (nowait && pid) ? NLM_F_MULTI : 0; | |
1235 | nfmsg->nfgen_family = exp->tuple.src.l3num; | |
1236 | nfmsg->version = NFNETLINK_V0; | |
1237 | nfmsg->res_id = 0; | |
1238 | ||
1239 | if (ctnetlink_exp_dump_expect(skb, exp) < 0) | |
df6fb868 | 1240 | goto nla_put_failure; |
c1d10adb | 1241 | |
27a884dc | 1242 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; |
c1d10adb PNA |
1243 | return skb->len; |
1244 | ||
1245 | nlmsg_failure: | |
df6fb868 | 1246 | nla_put_failure: |
dc5fc579 | 1247 | nlmsg_trim(skb, b); |
c1d10adb PNA |
1248 | return -1; |
1249 | } | |
1250 | ||
1251 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1252 | static int ctnetlink_expect_event(struct notifier_block *this, | |
1253 | unsigned long events, void *ptr) | |
1254 | { | |
1255 | struct nlmsghdr *nlh; | |
1256 | struct nfgenmsg *nfmsg; | |
1257 | struct nf_conntrack_expect *exp = (struct nf_conntrack_expect *)ptr; | |
1258 | struct sk_buff *skb; | |
1259 | unsigned int type; | |
27a884dc | 1260 | sk_buff_data_t b; |
c1d10adb PNA |
1261 | int flags = 0; |
1262 | ||
1263 | if (events & IPEXP_NEW) { | |
1264 | type = IPCTNL_MSG_EXP_NEW; | |
1265 | flags = NLM_F_CREATE|NLM_F_EXCL; | |
1266 | } else | |
1267 | return NOTIFY_DONE; | |
1268 | ||
b3a27bfb PNA |
1269 | if (!nfnetlink_has_listeners(NFNLGRP_CONNTRACK_EXP_NEW)) |
1270 | return NOTIFY_DONE; | |
1271 | ||
c1d10adb PNA |
1272 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC); |
1273 | if (!skb) | |
1274 | return NOTIFY_DONE; | |
1275 | ||
1276 | b = skb->tail; | |
1277 | ||
b633ad5f | 1278 | type |= NFNL_SUBSYS_CTNETLINK_EXP << 8; |
c1d10adb PNA |
1279 | nlh = NLMSG_PUT(skb, 0, 0, type, sizeof(struct nfgenmsg)); |
1280 | nfmsg = NLMSG_DATA(nlh); | |
1281 | ||
1282 | nlh->nlmsg_flags = flags; | |
1283 | nfmsg->nfgen_family = exp->tuple.src.l3num; | |
1284 | nfmsg->version = NFNETLINK_V0; | |
1285 | nfmsg->res_id = 0; | |
1286 | ||
1287 | if (ctnetlink_exp_dump_expect(skb, exp) < 0) | |
df6fb868 | 1288 | goto nla_put_failure; |
c1d10adb PNA |
1289 | |
1290 | nlh->nlmsg_len = skb->tail - b; | |
1291 | nfnetlink_send(skb, 0, NFNLGRP_CONNTRACK_EXP_NEW, 0); | |
1292 | return NOTIFY_DONE; | |
1293 | ||
1294 | nlmsg_failure: | |
df6fb868 | 1295 | nla_put_failure: |
c1d10adb PNA |
1296 | kfree_skb(skb); |
1297 | return NOTIFY_DONE; | |
1298 | } | |
1299 | #endif | |
cf6994c2 PM |
1300 | static int ctnetlink_exp_done(struct netlink_callback *cb) |
1301 | { | |
31f15875 PM |
1302 | if (cb->args[1]) |
1303 | nf_ct_expect_put((struct nf_conntrack_expect *)cb->args[1]); | |
cf6994c2 PM |
1304 | return 0; |
1305 | } | |
c1d10adb PNA |
1306 | |
1307 | static int | |
1308 | ctnetlink_exp_dump_table(struct sk_buff *skb, struct netlink_callback *cb) | |
1309 | { | |
cf6994c2 | 1310 | struct nf_conntrack_expect *exp, *last; |
87711cb8 | 1311 | struct nfgenmsg *nfmsg = NLMSG_DATA(cb->nlh); |
31f15875 | 1312 | struct hlist_node *n; |
87711cb8 | 1313 | u_int8_t l3proto = nfmsg->nfgen_family; |
c1d10adb | 1314 | |
c1d10adb | 1315 | read_lock_bh(&nf_conntrack_lock); |
31f15875 PM |
1316 | last = (struct nf_conntrack_expect *)cb->args[1]; |
1317 | for (; cb->args[0] < nf_ct_expect_hsize; cb->args[0]++) { | |
cf6994c2 | 1318 | restart: |
31f15875 PM |
1319 | hlist_for_each_entry(exp, n, &nf_ct_expect_hash[cb->args[0]], |
1320 | hnode) { | |
1321 | if (l3proto && exp->tuple.src.l3num != l3proto) | |
cf6994c2 | 1322 | continue; |
31f15875 PM |
1323 | if (cb->args[1]) { |
1324 | if (exp != last) | |
1325 | continue; | |
1326 | cb->args[1] = 0; | |
1327 | } | |
1328 | if (ctnetlink_exp_fill_info(skb, NETLINK_CB(cb->skb).pid, | |
1329 | cb->nlh->nlmsg_seq, | |
1330 | IPCTNL_MSG_EXP_NEW, | |
1331 | 1, exp) < 0) { | |
1332 | atomic_inc(&exp->use); | |
1333 | cb->args[1] = (unsigned long)exp; | |
1334 | goto out; | |
1335 | } | |
cf6994c2 | 1336 | } |
31f15875 PM |
1337 | if (cb->args[1]) { |
1338 | cb->args[1] = 0; | |
1339 | goto restart; | |
cf6994c2 PM |
1340 | } |
1341 | } | |
601e68e1 | 1342 | out: |
c1d10adb | 1343 | read_unlock_bh(&nf_conntrack_lock); |
cf6994c2 PM |
1344 | if (last) |
1345 | nf_ct_expect_put(last); | |
c1d10adb | 1346 | |
c1d10adb PNA |
1347 | return skb->len; |
1348 | } | |
1349 | ||
f73e924c PM |
1350 | static const struct nla_policy exp_nla_policy[CTA_EXPECT_MAX+1] = { |
1351 | [CTA_EXPECT_TIMEOUT] = { .type = NLA_U32 }, | |
1352 | [CTA_EXPECT_ID] = { .type = NLA_U32 }, | |
c1d10adb PNA |
1353 | }; |
1354 | ||
1355 | static int | |
601e68e1 | 1356 | ctnetlink_get_expect(struct sock *ctnl, struct sk_buff *skb, |
df6fb868 | 1357 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
1358 | { |
1359 | struct nf_conntrack_tuple tuple; | |
1360 | struct nf_conntrack_expect *exp; | |
1361 | struct sk_buff *skb2; | |
1362 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
1363 | u_int8_t u3 = nfmsg->nfgen_family; | |
1364 | int err = 0; | |
1365 | ||
c1d10adb | 1366 | if (nlh->nlmsg_flags & NLM_F_DUMP) { |
c702e804 TG |
1367 | return netlink_dump_start(ctnl, skb, nlh, |
1368 | ctnetlink_exp_dump_table, | |
cf6994c2 | 1369 | ctnetlink_exp_done); |
c1d10adb PNA |
1370 | } |
1371 | ||
df6fb868 | 1372 | if (cda[CTA_EXPECT_MASTER]) |
c1d10adb PNA |
1373 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_MASTER, u3); |
1374 | else | |
1375 | return -EINVAL; | |
1376 | ||
1377 | if (err < 0) | |
1378 | return err; | |
1379 | ||
6823645d | 1380 | exp = nf_ct_expect_find_get(&tuple); |
c1d10adb PNA |
1381 | if (!exp) |
1382 | return -ENOENT; | |
1383 | ||
df6fb868 PM |
1384 | if (cda[CTA_EXPECT_ID]) { |
1385 | __be32 id = *(__be32 *)nla_data(cda[CTA_EXPECT_ID]); | |
35832402 | 1386 | if (ntohl(id) != (u32)(unsigned long)exp) { |
6823645d | 1387 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1388 | return -ENOENT; |
1389 | } | |
601e68e1 | 1390 | } |
c1d10adb PNA |
1391 | |
1392 | err = -ENOMEM; | |
1393 | skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | |
1394 | if (!skb2) | |
1395 | goto out; | |
4e9b8269 | 1396 | |
601e68e1 | 1397 | err = ctnetlink_exp_fill_info(skb2, NETLINK_CB(skb).pid, |
c1d10adb PNA |
1398 | nlh->nlmsg_seq, IPCTNL_MSG_EXP_NEW, |
1399 | 1, exp); | |
1400 | if (err <= 0) | |
1401 | goto free; | |
1402 | ||
6823645d | 1403 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1404 | |
1405 | return netlink_unicast(ctnl, skb2, NETLINK_CB(skb).pid, MSG_DONTWAIT); | |
1406 | ||
1407 | free: | |
1408 | kfree_skb(skb2); | |
1409 | out: | |
6823645d | 1410 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1411 | return err; |
1412 | } | |
1413 | ||
1414 | static int | |
601e68e1 | 1415 | ctnetlink_del_expect(struct sock *ctnl, struct sk_buff *skb, |
df6fb868 | 1416 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb | 1417 | { |
31f15875 | 1418 | struct nf_conntrack_expect *exp; |
c1d10adb PNA |
1419 | struct nf_conntrack_tuple tuple; |
1420 | struct nf_conntrack_helper *h; | |
1421 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
31f15875 | 1422 | struct hlist_node *n, *next; |
c1d10adb | 1423 | u_int8_t u3 = nfmsg->nfgen_family; |
31f15875 | 1424 | unsigned int i; |
c1d10adb PNA |
1425 | int err; |
1426 | ||
df6fb868 | 1427 | if (cda[CTA_EXPECT_TUPLE]) { |
c1d10adb PNA |
1428 | /* delete a single expect by tuple */ |
1429 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3); | |
1430 | if (err < 0) | |
1431 | return err; | |
1432 | ||
1433 | /* bump usage count to 2 */ | |
6823645d | 1434 | exp = nf_ct_expect_find_get(&tuple); |
c1d10adb PNA |
1435 | if (!exp) |
1436 | return -ENOENT; | |
1437 | ||
df6fb868 PM |
1438 | if (cda[CTA_EXPECT_ID]) { |
1439 | __be32 id = *(__be32 *)nla_data(cda[CTA_EXPECT_ID]); | |
35832402 | 1440 | if (ntohl(id) != (u32)(unsigned long)exp) { |
6823645d | 1441 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1442 | return -ENOENT; |
1443 | } | |
1444 | } | |
1445 | ||
1446 | /* after list removal, usage count == 1 */ | |
6823645d | 1447 | nf_ct_unexpect_related(exp); |
601e68e1 | 1448 | /* have to put what we 'get' above. |
c1d10adb | 1449 | * after this line usage count == 0 */ |
6823645d | 1450 | nf_ct_expect_put(exp); |
df6fb868 PM |
1451 | } else if (cda[CTA_EXPECT_HELP_NAME]) { |
1452 | char *name = nla_data(cda[CTA_EXPECT_HELP_NAME]); | |
31f15875 | 1453 | struct nf_conn_help *m_help; |
c1d10adb PNA |
1454 | |
1455 | /* delete all expectations for this helper */ | |
1456 | write_lock_bh(&nf_conntrack_lock); | |
1457 | h = __nf_conntrack_helper_find_byname(name); | |
1458 | if (!h) { | |
1459 | write_unlock_bh(&nf_conntrack_lock); | |
1460 | return -EINVAL; | |
1461 | } | |
31f15875 PM |
1462 | for (i = 0; i < nf_ct_expect_hsize; i++) { |
1463 | hlist_for_each_entry_safe(exp, n, next, | |
1464 | &nf_ct_expect_hash[i], | |
1465 | hnode) { | |
1466 | m_help = nfct_help(exp->master); | |
1467 | if (m_help->helper == h | |
1468 | && del_timer(&exp->timeout)) { | |
1469 | nf_ct_unlink_expect(exp); | |
1470 | nf_ct_expect_put(exp); | |
1471 | } | |
c1d10adb PNA |
1472 | } |
1473 | } | |
1474 | write_unlock_bh(&nf_conntrack_lock); | |
1475 | } else { | |
1476 | /* This basically means we have to flush everything*/ | |
1477 | write_lock_bh(&nf_conntrack_lock); | |
31f15875 PM |
1478 | for (i = 0; i < nf_ct_expect_hsize; i++) { |
1479 | hlist_for_each_entry_safe(exp, n, next, | |
1480 | &nf_ct_expect_hash[i], | |
1481 | hnode) { | |
1482 | if (del_timer(&exp->timeout)) { | |
1483 | nf_ct_unlink_expect(exp); | |
1484 | nf_ct_expect_put(exp); | |
1485 | } | |
c1d10adb PNA |
1486 | } |
1487 | } | |
1488 | write_unlock_bh(&nf_conntrack_lock); | |
1489 | } | |
1490 | ||
1491 | return 0; | |
1492 | } | |
1493 | static int | |
df6fb868 | 1494 | ctnetlink_change_expect(struct nf_conntrack_expect *x, struct nlattr *cda[]) |
c1d10adb PNA |
1495 | { |
1496 | return -EOPNOTSUPP; | |
1497 | } | |
1498 | ||
1499 | static int | |
df6fb868 | 1500 | ctnetlink_create_expect(struct nlattr *cda[], u_int8_t u3) |
c1d10adb PNA |
1501 | { |
1502 | struct nf_conntrack_tuple tuple, mask, master_tuple; | |
1503 | struct nf_conntrack_tuple_hash *h = NULL; | |
1504 | struct nf_conntrack_expect *exp; | |
1505 | struct nf_conn *ct; | |
dc808fe2 | 1506 | struct nf_conn_help *help; |
c1d10adb PNA |
1507 | int err = 0; |
1508 | ||
c1d10adb PNA |
1509 | /* caller guarantees that those three CTA_EXPECT_* exist */ |
1510 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3); | |
1511 | if (err < 0) | |
1512 | return err; | |
1513 | err = ctnetlink_parse_tuple(cda, &mask, CTA_EXPECT_MASK, u3); | |
1514 | if (err < 0) | |
1515 | return err; | |
1516 | err = ctnetlink_parse_tuple(cda, &master_tuple, CTA_EXPECT_MASTER, u3); | |
1517 | if (err < 0) | |
1518 | return err; | |
1519 | ||
1520 | /* Look for master conntrack of this expectation */ | |
330f7db5 | 1521 | h = nf_conntrack_find_get(&master_tuple); |
c1d10adb PNA |
1522 | if (!h) |
1523 | return -ENOENT; | |
1524 | ct = nf_ct_tuplehash_to_ctrack(h); | |
dc808fe2 | 1525 | help = nfct_help(ct); |
c1d10adb | 1526 | |
dc808fe2 | 1527 | if (!help || !help->helper) { |
c1d10adb PNA |
1528 | /* such conntrack hasn't got any helper, abort */ |
1529 | err = -EINVAL; | |
1530 | goto out; | |
1531 | } | |
1532 | ||
6823645d | 1533 | exp = nf_ct_expect_alloc(ct); |
c1d10adb PNA |
1534 | if (!exp) { |
1535 | err = -ENOMEM; | |
1536 | goto out; | |
1537 | } | |
601e68e1 | 1538 | |
c1d10adb PNA |
1539 | exp->expectfn = NULL; |
1540 | exp->flags = 0; | |
1541 | exp->master = ct; | |
9457d851 | 1542 | exp->helper = NULL; |
c1d10adb | 1543 | memcpy(&exp->tuple, &tuple, sizeof(struct nf_conntrack_tuple)); |
d4156e8c PM |
1544 | memcpy(&exp->mask.src.u3, &mask.src.u3, sizeof(exp->mask.src.u3)); |
1545 | exp->mask.src.u.all = mask.src.u.all; | |
c1d10adb | 1546 | |
6823645d PM |
1547 | err = nf_ct_expect_related(exp); |
1548 | nf_ct_expect_put(exp); | |
c1d10adb | 1549 | |
601e68e1 | 1550 | out: |
c1d10adb PNA |
1551 | nf_ct_put(nf_ct_tuplehash_to_ctrack(h)); |
1552 | return err; | |
1553 | } | |
1554 | ||
1555 | static int | |
1556 | ctnetlink_new_expect(struct sock *ctnl, struct sk_buff *skb, | |
df6fb868 | 1557 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
1558 | { |
1559 | struct nf_conntrack_tuple tuple; | |
1560 | struct nf_conntrack_expect *exp; | |
1561 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
1562 | u_int8_t u3 = nfmsg->nfgen_family; | |
1563 | int err = 0; | |
1564 | ||
df6fb868 PM |
1565 | if (!cda[CTA_EXPECT_TUPLE] |
1566 | || !cda[CTA_EXPECT_MASK] | |
1567 | || !cda[CTA_EXPECT_MASTER]) | |
c1d10adb PNA |
1568 | return -EINVAL; |
1569 | ||
1570 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3); | |
1571 | if (err < 0) | |
1572 | return err; | |
1573 | ||
1574 | write_lock_bh(&nf_conntrack_lock); | |
6823645d | 1575 | exp = __nf_ct_expect_find(&tuple); |
c1d10adb PNA |
1576 | |
1577 | if (!exp) { | |
1578 | write_unlock_bh(&nf_conntrack_lock); | |
1579 | err = -ENOENT; | |
1580 | if (nlh->nlmsg_flags & NLM_F_CREATE) | |
1581 | err = ctnetlink_create_expect(cda, u3); | |
1582 | return err; | |
1583 | } | |
1584 | ||
1585 | err = -EEXIST; | |
1586 | if (!(nlh->nlmsg_flags & NLM_F_EXCL)) | |
1587 | err = ctnetlink_change_expect(exp, cda); | |
1588 | write_unlock_bh(&nf_conntrack_lock); | |
1589 | ||
c1d10adb PNA |
1590 | return err; |
1591 | } | |
1592 | ||
1593 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1594 | static struct notifier_block ctnl_notifier = { | |
1595 | .notifier_call = ctnetlink_conntrack_event, | |
1596 | }; | |
1597 | ||
1598 | static struct notifier_block ctnl_notifier_exp = { | |
1599 | .notifier_call = ctnetlink_expect_event, | |
1600 | }; | |
1601 | #endif | |
1602 | ||
7c8d4cb4 | 1603 | static const struct nfnl_callback ctnl_cb[IPCTNL_MSG_MAX] = { |
c1d10adb | 1604 | [IPCTNL_MSG_CT_NEW] = { .call = ctnetlink_new_conntrack, |
f73e924c PM |
1605 | .attr_count = CTA_MAX, |
1606 | .policy = ct_nla_policy }, | |
c1d10adb | 1607 | [IPCTNL_MSG_CT_GET] = { .call = ctnetlink_get_conntrack, |
f73e924c PM |
1608 | .attr_count = CTA_MAX, |
1609 | .policy = ct_nla_policy }, | |
c1d10adb | 1610 | [IPCTNL_MSG_CT_DELETE] = { .call = ctnetlink_del_conntrack, |
f73e924c PM |
1611 | .attr_count = CTA_MAX, |
1612 | .policy = ct_nla_policy }, | |
c1d10adb | 1613 | [IPCTNL_MSG_CT_GET_CTRZERO] = { .call = ctnetlink_get_conntrack, |
f73e924c PM |
1614 | .attr_count = CTA_MAX, |
1615 | .policy = ct_nla_policy }, | |
c1d10adb PNA |
1616 | }; |
1617 | ||
7c8d4cb4 | 1618 | static const struct nfnl_callback ctnl_exp_cb[IPCTNL_MSG_EXP_MAX] = { |
c1d10adb | 1619 | [IPCTNL_MSG_EXP_GET] = { .call = ctnetlink_get_expect, |
f73e924c PM |
1620 | .attr_count = CTA_EXPECT_MAX, |
1621 | .policy = exp_nla_policy }, | |
c1d10adb | 1622 | [IPCTNL_MSG_EXP_NEW] = { .call = ctnetlink_new_expect, |
f73e924c PM |
1623 | .attr_count = CTA_EXPECT_MAX, |
1624 | .policy = exp_nla_policy }, | |
c1d10adb | 1625 | [IPCTNL_MSG_EXP_DELETE] = { .call = ctnetlink_del_expect, |
f73e924c PM |
1626 | .attr_count = CTA_EXPECT_MAX, |
1627 | .policy = exp_nla_policy }, | |
c1d10adb PNA |
1628 | }; |
1629 | ||
7c8d4cb4 | 1630 | static const struct nfnetlink_subsystem ctnl_subsys = { |
c1d10adb PNA |
1631 | .name = "conntrack", |
1632 | .subsys_id = NFNL_SUBSYS_CTNETLINK, | |
1633 | .cb_count = IPCTNL_MSG_MAX, | |
1634 | .cb = ctnl_cb, | |
1635 | }; | |
1636 | ||
7c8d4cb4 | 1637 | static const struct nfnetlink_subsystem ctnl_exp_subsys = { |
c1d10adb PNA |
1638 | .name = "conntrack_expect", |
1639 | .subsys_id = NFNL_SUBSYS_CTNETLINK_EXP, | |
1640 | .cb_count = IPCTNL_MSG_EXP_MAX, | |
1641 | .cb = ctnl_exp_cb, | |
1642 | }; | |
1643 | ||
d2483dde | 1644 | MODULE_ALIAS("ip_conntrack_netlink"); |
c1d10adb | 1645 | MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK); |
34f9a2e4 | 1646 | MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK_EXP); |
c1d10adb PNA |
1647 | |
1648 | static int __init ctnetlink_init(void) | |
1649 | { | |
1650 | int ret; | |
1651 | ||
1652 | printk("ctnetlink v%s: registering with nfnetlink.\n", version); | |
1653 | ret = nfnetlink_subsys_register(&ctnl_subsys); | |
1654 | if (ret < 0) { | |
1655 | printk("ctnetlink_init: cannot register with nfnetlink.\n"); | |
1656 | goto err_out; | |
1657 | } | |
1658 | ||
1659 | ret = nfnetlink_subsys_register(&ctnl_exp_subsys); | |
1660 | if (ret < 0) { | |
1661 | printk("ctnetlink_init: cannot register exp with nfnetlink.\n"); | |
1662 | goto err_unreg_subsys; | |
1663 | } | |
1664 | ||
1665 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1666 | ret = nf_conntrack_register_notifier(&ctnl_notifier); | |
1667 | if (ret < 0) { | |
1668 | printk("ctnetlink_init: cannot register notifier.\n"); | |
1669 | goto err_unreg_exp_subsys; | |
1670 | } | |
1671 | ||
6823645d | 1672 | ret = nf_ct_expect_register_notifier(&ctnl_notifier_exp); |
c1d10adb PNA |
1673 | if (ret < 0) { |
1674 | printk("ctnetlink_init: cannot expect register notifier.\n"); | |
1675 | goto err_unreg_notifier; | |
1676 | } | |
1677 | #endif | |
1678 | ||
1679 | return 0; | |
1680 | ||
1681 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1682 | err_unreg_notifier: | |
1683 | nf_conntrack_unregister_notifier(&ctnl_notifier); | |
1684 | err_unreg_exp_subsys: | |
1685 | nfnetlink_subsys_unregister(&ctnl_exp_subsys); | |
1686 | #endif | |
1687 | err_unreg_subsys: | |
1688 | nfnetlink_subsys_unregister(&ctnl_subsys); | |
1689 | err_out: | |
1690 | return ret; | |
1691 | } | |
1692 | ||
1693 | static void __exit ctnetlink_exit(void) | |
1694 | { | |
1695 | printk("ctnetlink: unregistering from nfnetlink.\n"); | |
1696 | ||
1697 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
6823645d | 1698 | nf_ct_expect_unregister_notifier(&ctnl_notifier_exp); |
c1d10adb PNA |
1699 | nf_conntrack_unregister_notifier(&ctnl_notifier); |
1700 | #endif | |
1701 | ||
1702 | nfnetlink_subsys_unregister(&ctnl_exp_subsys); | |
1703 | nfnetlink_subsys_unregister(&ctnl_subsys); | |
1704 | return; | |
1705 | } | |
1706 | ||
1707 | module_init(ctnetlink_init); | |
1708 | module_exit(ctnetlink_exit); |