]>
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> |
1cde6436 | 7 | * (C) 2005-2006 by Pablo Neira Ayuso <pablo@eurodev.net> |
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 PNA |
977 | struct nf_conntrack_tuple *otuple, |
978 | struct nf_conntrack_tuple *rtuple) | |
979 | { | |
980 | struct nf_conn *ct; | |
981 | int err = -EINVAL; | |
dafc741c | 982 | struct nf_conn_help *help; |
ceceae1b | 983 | struct nf_conntrack_helper *helper; |
c1d10adb | 984 | |
c1d10adb PNA |
985 | ct = nf_conntrack_alloc(otuple, rtuple); |
986 | if (ct == NULL || IS_ERR(ct)) | |
601e68e1 | 987 | return -ENOMEM; |
c1d10adb | 988 | |
df6fb868 | 989 | if (!cda[CTA_TIMEOUT]) |
c1d10adb | 990 | goto err; |
df6fb868 | 991 | ct->timeout.expires = ntohl(*(__be32 *)nla_data(cda[CTA_TIMEOUT])); |
c1d10adb PNA |
992 | |
993 | ct->timeout.expires = jiffies + ct->timeout.expires * HZ; | |
994 | ct->status |= IPS_CONFIRMED; | |
995 | ||
df6fb868 | 996 | if (cda[CTA_STATUS]) { |
bbb3357d PNA |
997 | err = ctnetlink_change_status(ct, cda); |
998 | if (err < 0) | |
999 | goto err; | |
1000 | } | |
c1d10adb | 1001 | |
df6fb868 | 1002 | if (cda[CTA_PROTOINFO]) { |
c1d10adb PNA |
1003 | err = ctnetlink_change_protoinfo(ct, cda); |
1004 | if (err < 0) | |
c54ea3b9 | 1005 | goto err; |
c1d10adb PNA |
1006 | } |
1007 | ||
bcd1e830 | 1008 | #if defined(CONFIG_NF_CONNTRACK_MARK) |
df6fb868 PM |
1009 | if (cda[CTA_MARK]) |
1010 | ct->mark = ntohl(*(__be32 *)nla_data(cda[CTA_MARK])); | |
c1d10adb PNA |
1011 | #endif |
1012 | ||
ceceae1b YK |
1013 | helper = nf_ct_helper_find_get(rtuple); |
1014 | if (helper) { | |
b560580a | 1015 | help = nf_ct_helper_ext_add(ct, GFP_KERNEL); |
ceceae1b YK |
1016 | if (help == NULL) { |
1017 | nf_ct_helper_put(helper); | |
1018 | err = -ENOMEM; | |
1019 | goto err; | |
1020 | } | |
3c158f7f PM |
1021 | /* not in hash table yet so not strictly necessary */ |
1022 | rcu_assign_pointer(help->helper, helper); | |
1023 | } | |
dafc741c | 1024 | |
c1d10adb PNA |
1025 | add_timer(&ct->timeout); |
1026 | nf_conntrack_hash_insert(ct); | |
1027 | ||
3c158f7f PM |
1028 | if (helper) |
1029 | nf_ct_helper_put(helper); | |
dafc741c | 1030 | |
c1d10adb PNA |
1031 | return 0; |
1032 | ||
601e68e1 | 1033 | err: |
c1d10adb PNA |
1034 | nf_conntrack_free(ct); |
1035 | return err; | |
1036 | } | |
1037 | ||
601e68e1 YH |
1038 | static int |
1039 | ctnetlink_new_conntrack(struct sock *ctnl, struct sk_buff *skb, | |
df6fb868 | 1040 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
1041 | { |
1042 | struct nf_conntrack_tuple otuple, rtuple; | |
1043 | struct nf_conntrack_tuple_hash *h = NULL; | |
1044 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
1045 | u_int8_t u3 = nfmsg->nfgen_family; | |
1046 | int err = 0; | |
1047 | ||
df6fb868 | 1048 | if (cda[CTA_TUPLE_ORIG]) { |
c1d10adb PNA |
1049 | err = ctnetlink_parse_tuple(cda, &otuple, CTA_TUPLE_ORIG, u3); |
1050 | if (err < 0) | |
1051 | return err; | |
1052 | } | |
1053 | ||
df6fb868 | 1054 | if (cda[CTA_TUPLE_REPLY]) { |
c1d10adb PNA |
1055 | err = ctnetlink_parse_tuple(cda, &rtuple, CTA_TUPLE_REPLY, u3); |
1056 | if (err < 0) | |
1057 | return err; | |
1058 | } | |
1059 | ||
1060 | write_lock_bh(&nf_conntrack_lock); | |
df6fb868 | 1061 | if (cda[CTA_TUPLE_ORIG]) |
c1d10adb | 1062 | h = __nf_conntrack_find(&otuple, NULL); |
df6fb868 | 1063 | else if (cda[CTA_TUPLE_REPLY]) |
c1d10adb PNA |
1064 | h = __nf_conntrack_find(&rtuple, NULL); |
1065 | ||
1066 | if (h == NULL) { | |
1067 | write_unlock_bh(&nf_conntrack_lock); | |
c1d10adb PNA |
1068 | err = -ENOENT; |
1069 | if (nlh->nlmsg_flags & NLM_F_CREATE) | |
1070 | err = ctnetlink_create_conntrack(cda, &otuple, &rtuple); | |
1071 | return err; | |
1072 | } | |
1073 | /* implicit 'else' */ | |
1074 | ||
c1d10adb PNA |
1075 | /* We manipulate the conntrack inside the global conntrack table lock, |
1076 | * so there's no need to increase the refcount */ | |
c1d10adb | 1077 | err = -EEXIST; |
ff4ca827 PNA |
1078 | if (!(nlh->nlmsg_flags & NLM_F_EXCL)) { |
1079 | /* we only allow nat config for new conntracks */ | |
df6fb868 | 1080 | if (cda[CTA_NAT_SRC] || cda[CTA_NAT_DST]) { |
ff4ca827 PNA |
1081 | err = -EINVAL; |
1082 | goto out_unlock; | |
1083 | } | |
1084 | err = ctnetlink_change_conntrack(nf_ct_tuplehash_to_ctrack(h), | |
1085 | cda); | |
1086 | } | |
c1d10adb PNA |
1087 | |
1088 | out_unlock: | |
1089 | write_unlock_bh(&nf_conntrack_lock); | |
1090 | return err; | |
1091 | } | |
1092 | ||
601e68e1 YH |
1093 | /*********************************************************************** |
1094 | * EXPECT | |
1095 | ***********************************************************************/ | |
c1d10adb PNA |
1096 | |
1097 | static inline int | |
1098 | ctnetlink_exp_dump_tuple(struct sk_buff *skb, | |
1099 | const struct nf_conntrack_tuple *tuple, | |
1100 | enum ctattr_expect type) | |
1101 | { | |
df6fb868 | 1102 | struct nlattr *nest_parms; |
601e68e1 | 1103 | |
df6fb868 PM |
1104 | nest_parms = nla_nest_start(skb, type | NLA_F_NESTED); |
1105 | if (!nest_parms) | |
1106 | goto nla_put_failure; | |
c1d10adb | 1107 | if (ctnetlink_dump_tuples(skb, tuple) < 0) |
df6fb868 PM |
1108 | goto nla_put_failure; |
1109 | nla_nest_end(skb, nest_parms); | |
c1d10adb PNA |
1110 | |
1111 | return 0; | |
1112 | ||
df6fb868 | 1113 | nla_put_failure: |
c1d10adb | 1114 | return -1; |
601e68e1 | 1115 | } |
c1d10adb | 1116 | |
1cde6436 PNA |
1117 | static inline int |
1118 | ctnetlink_exp_dump_mask(struct sk_buff *skb, | |
1119 | const struct nf_conntrack_tuple *tuple, | |
d4156e8c | 1120 | const struct nf_conntrack_tuple_mask *mask) |
1cde6436 PNA |
1121 | { |
1122 | int ret; | |
1123 | struct nf_conntrack_l3proto *l3proto; | |
605dcad6 | 1124 | struct nf_conntrack_l4proto *l4proto; |
d4156e8c | 1125 | struct nf_conntrack_tuple m; |
df6fb868 | 1126 | struct nlattr *nest_parms; |
d4156e8c PM |
1127 | |
1128 | memset(&m, 0xFF, sizeof(m)); | |
1129 | m.src.u.all = mask->src.u.all; | |
1130 | memcpy(&m.src.u3, &mask->src.u3, sizeof(m.src.u3)); | |
1131 | ||
df6fb868 PM |
1132 | nest_parms = nla_nest_start(skb, CTA_EXPECT_MASK | NLA_F_NESTED); |
1133 | if (!nest_parms) | |
1134 | goto nla_put_failure; | |
1cde6436 PNA |
1135 | |
1136 | l3proto = nf_ct_l3proto_find_get(tuple->src.l3num); | |
d4156e8c | 1137 | ret = ctnetlink_dump_tuples_ip(skb, &m, l3proto); |
1cde6436 PNA |
1138 | nf_ct_l3proto_put(l3proto); |
1139 | ||
1140 | if (unlikely(ret < 0)) | |
df6fb868 | 1141 | goto nla_put_failure; |
1cde6436 | 1142 | |
605dcad6 | 1143 | l4proto = nf_ct_l4proto_find_get(tuple->src.l3num, tuple->dst.protonum); |
d4156e8c | 1144 | ret = ctnetlink_dump_tuples_proto(skb, &m, l4proto); |
605dcad6 | 1145 | nf_ct_l4proto_put(l4proto); |
1cde6436 | 1146 | if (unlikely(ret < 0)) |
df6fb868 | 1147 | goto nla_put_failure; |
1cde6436 | 1148 | |
df6fb868 | 1149 | nla_nest_end(skb, nest_parms); |
1cde6436 PNA |
1150 | |
1151 | return 0; | |
1152 | ||
df6fb868 | 1153 | nla_put_failure: |
1cde6436 PNA |
1154 | return -1; |
1155 | } | |
1156 | ||
c1d10adb PNA |
1157 | static inline int |
1158 | ctnetlink_exp_dump_expect(struct sk_buff *skb, | |
601e68e1 | 1159 | const struct nf_conntrack_expect *exp) |
c1d10adb PNA |
1160 | { |
1161 | struct nf_conn *master = exp->master; | |
bff9a89b | 1162 | __be32 timeout = htonl((exp->timeout.expires - jiffies) / HZ); |
35832402 | 1163 | __be32 id = htonl((unsigned long)exp); |
c1d10adb PNA |
1164 | |
1165 | if (ctnetlink_exp_dump_tuple(skb, &exp->tuple, CTA_EXPECT_TUPLE) < 0) | |
df6fb868 | 1166 | goto nla_put_failure; |
1cde6436 | 1167 | if (ctnetlink_exp_dump_mask(skb, &exp->tuple, &exp->mask) < 0) |
df6fb868 | 1168 | goto nla_put_failure; |
c1d10adb PNA |
1169 | if (ctnetlink_exp_dump_tuple(skb, |
1170 | &master->tuplehash[IP_CT_DIR_ORIGINAL].tuple, | |
1171 | CTA_EXPECT_MASTER) < 0) | |
df6fb868 | 1172 | goto nla_put_failure; |
601e68e1 | 1173 | |
df6fb868 PM |
1174 | NLA_PUT(skb, CTA_EXPECT_TIMEOUT, sizeof(timeout), &timeout); |
1175 | NLA_PUT(skb, CTA_EXPECT_ID, sizeof(u_int32_t), &id); | |
c1d10adb PNA |
1176 | |
1177 | return 0; | |
601e68e1 | 1178 | |
df6fb868 | 1179 | nla_put_failure: |
c1d10adb PNA |
1180 | return -1; |
1181 | } | |
1182 | ||
1183 | static int | |
1184 | ctnetlink_exp_fill_info(struct sk_buff *skb, u32 pid, u32 seq, | |
601e68e1 YH |
1185 | int event, |
1186 | int nowait, | |
c1d10adb PNA |
1187 | const struct nf_conntrack_expect *exp) |
1188 | { | |
1189 | struct nlmsghdr *nlh; | |
1190 | struct nfgenmsg *nfmsg; | |
27a884dc | 1191 | unsigned char *b = skb_tail_pointer(skb); |
c1d10adb PNA |
1192 | |
1193 | event |= NFNL_SUBSYS_CTNETLINK_EXP << 8; | |
1194 | nlh = NLMSG_PUT(skb, pid, seq, event, sizeof(struct nfgenmsg)); | |
1195 | nfmsg = NLMSG_DATA(nlh); | |
1196 | ||
1197 | nlh->nlmsg_flags = (nowait && pid) ? NLM_F_MULTI : 0; | |
1198 | nfmsg->nfgen_family = exp->tuple.src.l3num; | |
1199 | nfmsg->version = NFNETLINK_V0; | |
1200 | nfmsg->res_id = 0; | |
1201 | ||
1202 | if (ctnetlink_exp_dump_expect(skb, exp) < 0) | |
df6fb868 | 1203 | goto nla_put_failure; |
c1d10adb | 1204 | |
27a884dc | 1205 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; |
c1d10adb PNA |
1206 | return skb->len; |
1207 | ||
1208 | nlmsg_failure: | |
df6fb868 | 1209 | nla_put_failure: |
dc5fc579 | 1210 | nlmsg_trim(skb, b); |
c1d10adb PNA |
1211 | return -1; |
1212 | } | |
1213 | ||
1214 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1215 | static int ctnetlink_expect_event(struct notifier_block *this, | |
1216 | unsigned long events, void *ptr) | |
1217 | { | |
1218 | struct nlmsghdr *nlh; | |
1219 | struct nfgenmsg *nfmsg; | |
1220 | struct nf_conntrack_expect *exp = (struct nf_conntrack_expect *)ptr; | |
1221 | struct sk_buff *skb; | |
1222 | unsigned int type; | |
27a884dc | 1223 | sk_buff_data_t b; |
c1d10adb PNA |
1224 | int flags = 0; |
1225 | ||
1226 | if (events & IPEXP_NEW) { | |
1227 | type = IPCTNL_MSG_EXP_NEW; | |
1228 | flags = NLM_F_CREATE|NLM_F_EXCL; | |
1229 | } else | |
1230 | return NOTIFY_DONE; | |
1231 | ||
b3a27bfb PNA |
1232 | if (!nfnetlink_has_listeners(NFNLGRP_CONNTRACK_EXP_NEW)) |
1233 | return NOTIFY_DONE; | |
1234 | ||
c1d10adb PNA |
1235 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC); |
1236 | if (!skb) | |
1237 | return NOTIFY_DONE; | |
1238 | ||
1239 | b = skb->tail; | |
1240 | ||
b633ad5f | 1241 | type |= NFNL_SUBSYS_CTNETLINK_EXP << 8; |
c1d10adb PNA |
1242 | nlh = NLMSG_PUT(skb, 0, 0, type, sizeof(struct nfgenmsg)); |
1243 | nfmsg = NLMSG_DATA(nlh); | |
1244 | ||
1245 | nlh->nlmsg_flags = flags; | |
1246 | nfmsg->nfgen_family = exp->tuple.src.l3num; | |
1247 | nfmsg->version = NFNETLINK_V0; | |
1248 | nfmsg->res_id = 0; | |
1249 | ||
1250 | if (ctnetlink_exp_dump_expect(skb, exp) < 0) | |
df6fb868 | 1251 | goto nla_put_failure; |
c1d10adb PNA |
1252 | |
1253 | nlh->nlmsg_len = skb->tail - b; | |
1254 | nfnetlink_send(skb, 0, NFNLGRP_CONNTRACK_EXP_NEW, 0); | |
1255 | return NOTIFY_DONE; | |
1256 | ||
1257 | nlmsg_failure: | |
df6fb868 | 1258 | nla_put_failure: |
c1d10adb PNA |
1259 | kfree_skb(skb); |
1260 | return NOTIFY_DONE; | |
1261 | } | |
1262 | #endif | |
cf6994c2 PM |
1263 | static int ctnetlink_exp_done(struct netlink_callback *cb) |
1264 | { | |
31f15875 PM |
1265 | if (cb->args[1]) |
1266 | nf_ct_expect_put((struct nf_conntrack_expect *)cb->args[1]); | |
cf6994c2 PM |
1267 | return 0; |
1268 | } | |
c1d10adb PNA |
1269 | |
1270 | static int | |
1271 | ctnetlink_exp_dump_table(struct sk_buff *skb, struct netlink_callback *cb) | |
1272 | { | |
cf6994c2 | 1273 | struct nf_conntrack_expect *exp, *last; |
87711cb8 | 1274 | struct nfgenmsg *nfmsg = NLMSG_DATA(cb->nlh); |
31f15875 | 1275 | struct hlist_node *n; |
87711cb8 | 1276 | u_int8_t l3proto = nfmsg->nfgen_family; |
c1d10adb | 1277 | |
c1d10adb | 1278 | read_lock_bh(&nf_conntrack_lock); |
31f15875 PM |
1279 | last = (struct nf_conntrack_expect *)cb->args[1]; |
1280 | for (; cb->args[0] < nf_ct_expect_hsize; cb->args[0]++) { | |
cf6994c2 | 1281 | restart: |
31f15875 PM |
1282 | hlist_for_each_entry(exp, n, &nf_ct_expect_hash[cb->args[0]], |
1283 | hnode) { | |
1284 | if (l3proto && exp->tuple.src.l3num != l3proto) | |
cf6994c2 | 1285 | continue; |
31f15875 PM |
1286 | if (cb->args[1]) { |
1287 | if (exp != last) | |
1288 | continue; | |
1289 | cb->args[1] = 0; | |
1290 | } | |
1291 | if (ctnetlink_exp_fill_info(skb, NETLINK_CB(cb->skb).pid, | |
1292 | cb->nlh->nlmsg_seq, | |
1293 | IPCTNL_MSG_EXP_NEW, | |
1294 | 1, exp) < 0) { | |
1295 | atomic_inc(&exp->use); | |
1296 | cb->args[1] = (unsigned long)exp; | |
1297 | goto out; | |
1298 | } | |
cf6994c2 | 1299 | } |
31f15875 PM |
1300 | if (cb->args[1]) { |
1301 | cb->args[1] = 0; | |
1302 | goto restart; | |
cf6994c2 PM |
1303 | } |
1304 | } | |
601e68e1 | 1305 | out: |
c1d10adb | 1306 | read_unlock_bh(&nf_conntrack_lock); |
cf6994c2 PM |
1307 | if (last) |
1308 | nf_ct_expect_put(last); | |
c1d10adb | 1309 | |
c1d10adb PNA |
1310 | return skb->len; |
1311 | } | |
1312 | ||
f73e924c PM |
1313 | static const struct nla_policy exp_nla_policy[CTA_EXPECT_MAX+1] = { |
1314 | [CTA_EXPECT_TIMEOUT] = { .type = NLA_U32 }, | |
1315 | [CTA_EXPECT_ID] = { .type = NLA_U32 }, | |
c1d10adb PNA |
1316 | }; |
1317 | ||
1318 | static int | |
601e68e1 | 1319 | ctnetlink_get_expect(struct sock *ctnl, struct sk_buff *skb, |
df6fb868 | 1320 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
1321 | { |
1322 | struct nf_conntrack_tuple tuple; | |
1323 | struct nf_conntrack_expect *exp; | |
1324 | struct sk_buff *skb2; | |
1325 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
1326 | u_int8_t u3 = nfmsg->nfgen_family; | |
1327 | int err = 0; | |
1328 | ||
c1d10adb | 1329 | if (nlh->nlmsg_flags & NLM_F_DUMP) { |
c702e804 TG |
1330 | return netlink_dump_start(ctnl, skb, nlh, |
1331 | ctnetlink_exp_dump_table, | |
cf6994c2 | 1332 | ctnetlink_exp_done); |
c1d10adb PNA |
1333 | } |
1334 | ||
df6fb868 | 1335 | if (cda[CTA_EXPECT_MASTER]) |
c1d10adb PNA |
1336 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_MASTER, u3); |
1337 | else | |
1338 | return -EINVAL; | |
1339 | ||
1340 | if (err < 0) | |
1341 | return err; | |
1342 | ||
6823645d | 1343 | exp = nf_ct_expect_find_get(&tuple); |
c1d10adb PNA |
1344 | if (!exp) |
1345 | return -ENOENT; | |
1346 | ||
df6fb868 PM |
1347 | if (cda[CTA_EXPECT_ID]) { |
1348 | __be32 id = *(__be32 *)nla_data(cda[CTA_EXPECT_ID]); | |
35832402 | 1349 | if (ntohl(id) != (u32)(unsigned long)exp) { |
6823645d | 1350 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1351 | return -ENOENT; |
1352 | } | |
601e68e1 | 1353 | } |
c1d10adb PNA |
1354 | |
1355 | err = -ENOMEM; | |
1356 | skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | |
1357 | if (!skb2) | |
1358 | goto out; | |
4e9b8269 | 1359 | |
601e68e1 | 1360 | err = ctnetlink_exp_fill_info(skb2, NETLINK_CB(skb).pid, |
c1d10adb PNA |
1361 | nlh->nlmsg_seq, IPCTNL_MSG_EXP_NEW, |
1362 | 1, exp); | |
1363 | if (err <= 0) | |
1364 | goto free; | |
1365 | ||
6823645d | 1366 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1367 | |
1368 | return netlink_unicast(ctnl, skb2, NETLINK_CB(skb).pid, MSG_DONTWAIT); | |
1369 | ||
1370 | free: | |
1371 | kfree_skb(skb2); | |
1372 | out: | |
6823645d | 1373 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1374 | return err; |
1375 | } | |
1376 | ||
1377 | static int | |
601e68e1 | 1378 | ctnetlink_del_expect(struct sock *ctnl, struct sk_buff *skb, |
df6fb868 | 1379 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb | 1380 | { |
31f15875 | 1381 | struct nf_conntrack_expect *exp; |
c1d10adb PNA |
1382 | struct nf_conntrack_tuple tuple; |
1383 | struct nf_conntrack_helper *h; | |
1384 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
31f15875 | 1385 | struct hlist_node *n, *next; |
c1d10adb | 1386 | u_int8_t u3 = nfmsg->nfgen_family; |
31f15875 | 1387 | unsigned int i; |
c1d10adb PNA |
1388 | int err; |
1389 | ||
df6fb868 | 1390 | if (cda[CTA_EXPECT_TUPLE]) { |
c1d10adb PNA |
1391 | /* delete a single expect by tuple */ |
1392 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3); | |
1393 | if (err < 0) | |
1394 | return err; | |
1395 | ||
1396 | /* bump usage count to 2 */ | |
6823645d | 1397 | exp = nf_ct_expect_find_get(&tuple); |
c1d10adb PNA |
1398 | if (!exp) |
1399 | return -ENOENT; | |
1400 | ||
df6fb868 PM |
1401 | if (cda[CTA_EXPECT_ID]) { |
1402 | __be32 id = *(__be32 *)nla_data(cda[CTA_EXPECT_ID]); | |
35832402 | 1403 | if (ntohl(id) != (u32)(unsigned long)exp) { |
6823645d | 1404 | nf_ct_expect_put(exp); |
c1d10adb PNA |
1405 | return -ENOENT; |
1406 | } | |
1407 | } | |
1408 | ||
1409 | /* after list removal, usage count == 1 */ | |
6823645d | 1410 | nf_ct_unexpect_related(exp); |
601e68e1 | 1411 | /* have to put what we 'get' above. |
c1d10adb | 1412 | * after this line usage count == 0 */ |
6823645d | 1413 | nf_ct_expect_put(exp); |
df6fb868 PM |
1414 | } else if (cda[CTA_EXPECT_HELP_NAME]) { |
1415 | char *name = nla_data(cda[CTA_EXPECT_HELP_NAME]); | |
31f15875 | 1416 | struct nf_conn_help *m_help; |
c1d10adb PNA |
1417 | |
1418 | /* delete all expectations for this helper */ | |
1419 | write_lock_bh(&nf_conntrack_lock); | |
1420 | h = __nf_conntrack_helper_find_byname(name); | |
1421 | if (!h) { | |
1422 | write_unlock_bh(&nf_conntrack_lock); | |
1423 | return -EINVAL; | |
1424 | } | |
31f15875 PM |
1425 | for (i = 0; i < nf_ct_expect_hsize; i++) { |
1426 | hlist_for_each_entry_safe(exp, n, next, | |
1427 | &nf_ct_expect_hash[i], | |
1428 | hnode) { | |
1429 | m_help = nfct_help(exp->master); | |
1430 | if (m_help->helper == h | |
1431 | && del_timer(&exp->timeout)) { | |
1432 | nf_ct_unlink_expect(exp); | |
1433 | nf_ct_expect_put(exp); | |
1434 | } | |
c1d10adb PNA |
1435 | } |
1436 | } | |
1437 | write_unlock_bh(&nf_conntrack_lock); | |
1438 | } else { | |
1439 | /* This basically means we have to flush everything*/ | |
1440 | write_lock_bh(&nf_conntrack_lock); | |
31f15875 PM |
1441 | for (i = 0; i < nf_ct_expect_hsize; i++) { |
1442 | hlist_for_each_entry_safe(exp, n, next, | |
1443 | &nf_ct_expect_hash[i], | |
1444 | hnode) { | |
1445 | if (del_timer(&exp->timeout)) { | |
1446 | nf_ct_unlink_expect(exp); | |
1447 | nf_ct_expect_put(exp); | |
1448 | } | |
c1d10adb PNA |
1449 | } |
1450 | } | |
1451 | write_unlock_bh(&nf_conntrack_lock); | |
1452 | } | |
1453 | ||
1454 | return 0; | |
1455 | } | |
1456 | static int | |
df6fb868 | 1457 | ctnetlink_change_expect(struct nf_conntrack_expect *x, struct nlattr *cda[]) |
c1d10adb PNA |
1458 | { |
1459 | return -EOPNOTSUPP; | |
1460 | } | |
1461 | ||
1462 | static int | |
df6fb868 | 1463 | ctnetlink_create_expect(struct nlattr *cda[], u_int8_t u3) |
c1d10adb PNA |
1464 | { |
1465 | struct nf_conntrack_tuple tuple, mask, master_tuple; | |
1466 | struct nf_conntrack_tuple_hash *h = NULL; | |
1467 | struct nf_conntrack_expect *exp; | |
1468 | struct nf_conn *ct; | |
dc808fe2 | 1469 | struct nf_conn_help *help; |
c1d10adb PNA |
1470 | int err = 0; |
1471 | ||
c1d10adb PNA |
1472 | /* caller guarantees that those three CTA_EXPECT_* exist */ |
1473 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3); | |
1474 | if (err < 0) | |
1475 | return err; | |
1476 | err = ctnetlink_parse_tuple(cda, &mask, CTA_EXPECT_MASK, u3); | |
1477 | if (err < 0) | |
1478 | return err; | |
1479 | err = ctnetlink_parse_tuple(cda, &master_tuple, CTA_EXPECT_MASTER, u3); | |
1480 | if (err < 0) | |
1481 | return err; | |
1482 | ||
1483 | /* Look for master conntrack of this expectation */ | |
330f7db5 | 1484 | h = nf_conntrack_find_get(&master_tuple); |
c1d10adb PNA |
1485 | if (!h) |
1486 | return -ENOENT; | |
1487 | ct = nf_ct_tuplehash_to_ctrack(h); | |
dc808fe2 | 1488 | help = nfct_help(ct); |
c1d10adb | 1489 | |
dc808fe2 | 1490 | if (!help || !help->helper) { |
c1d10adb PNA |
1491 | /* such conntrack hasn't got any helper, abort */ |
1492 | err = -EINVAL; | |
1493 | goto out; | |
1494 | } | |
1495 | ||
6823645d | 1496 | exp = nf_ct_expect_alloc(ct); |
c1d10adb PNA |
1497 | if (!exp) { |
1498 | err = -ENOMEM; | |
1499 | goto out; | |
1500 | } | |
601e68e1 | 1501 | |
c1d10adb PNA |
1502 | exp->expectfn = NULL; |
1503 | exp->flags = 0; | |
1504 | exp->master = ct; | |
9457d851 | 1505 | exp->helper = NULL; |
c1d10adb | 1506 | memcpy(&exp->tuple, &tuple, sizeof(struct nf_conntrack_tuple)); |
d4156e8c PM |
1507 | memcpy(&exp->mask.src.u3, &mask.src.u3, sizeof(exp->mask.src.u3)); |
1508 | exp->mask.src.u.all = mask.src.u.all; | |
c1d10adb | 1509 | |
6823645d PM |
1510 | err = nf_ct_expect_related(exp); |
1511 | nf_ct_expect_put(exp); | |
c1d10adb | 1512 | |
601e68e1 | 1513 | out: |
c1d10adb PNA |
1514 | nf_ct_put(nf_ct_tuplehash_to_ctrack(h)); |
1515 | return err; | |
1516 | } | |
1517 | ||
1518 | static int | |
1519 | ctnetlink_new_expect(struct sock *ctnl, struct sk_buff *skb, | |
df6fb868 | 1520 | struct nlmsghdr *nlh, struct nlattr *cda[]) |
c1d10adb PNA |
1521 | { |
1522 | struct nf_conntrack_tuple tuple; | |
1523 | struct nf_conntrack_expect *exp; | |
1524 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
1525 | u_int8_t u3 = nfmsg->nfgen_family; | |
1526 | int err = 0; | |
1527 | ||
df6fb868 PM |
1528 | if (!cda[CTA_EXPECT_TUPLE] |
1529 | || !cda[CTA_EXPECT_MASK] | |
1530 | || !cda[CTA_EXPECT_MASTER]) | |
c1d10adb PNA |
1531 | return -EINVAL; |
1532 | ||
1533 | err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3); | |
1534 | if (err < 0) | |
1535 | return err; | |
1536 | ||
1537 | write_lock_bh(&nf_conntrack_lock); | |
6823645d | 1538 | exp = __nf_ct_expect_find(&tuple); |
c1d10adb PNA |
1539 | |
1540 | if (!exp) { | |
1541 | write_unlock_bh(&nf_conntrack_lock); | |
1542 | err = -ENOENT; | |
1543 | if (nlh->nlmsg_flags & NLM_F_CREATE) | |
1544 | err = ctnetlink_create_expect(cda, u3); | |
1545 | return err; | |
1546 | } | |
1547 | ||
1548 | err = -EEXIST; | |
1549 | if (!(nlh->nlmsg_flags & NLM_F_EXCL)) | |
1550 | err = ctnetlink_change_expect(exp, cda); | |
1551 | write_unlock_bh(&nf_conntrack_lock); | |
1552 | ||
c1d10adb PNA |
1553 | return err; |
1554 | } | |
1555 | ||
1556 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1557 | static struct notifier_block ctnl_notifier = { | |
1558 | .notifier_call = ctnetlink_conntrack_event, | |
1559 | }; | |
1560 | ||
1561 | static struct notifier_block ctnl_notifier_exp = { | |
1562 | .notifier_call = ctnetlink_expect_event, | |
1563 | }; | |
1564 | #endif | |
1565 | ||
7c8d4cb4 | 1566 | static const struct nfnl_callback ctnl_cb[IPCTNL_MSG_MAX] = { |
c1d10adb | 1567 | [IPCTNL_MSG_CT_NEW] = { .call = ctnetlink_new_conntrack, |
f73e924c PM |
1568 | .attr_count = CTA_MAX, |
1569 | .policy = ct_nla_policy }, | |
c1d10adb | 1570 | [IPCTNL_MSG_CT_GET] = { .call = ctnetlink_get_conntrack, |
f73e924c PM |
1571 | .attr_count = CTA_MAX, |
1572 | .policy = ct_nla_policy }, | |
c1d10adb | 1573 | [IPCTNL_MSG_CT_DELETE] = { .call = ctnetlink_del_conntrack, |
f73e924c PM |
1574 | .attr_count = CTA_MAX, |
1575 | .policy = ct_nla_policy }, | |
c1d10adb | 1576 | [IPCTNL_MSG_CT_GET_CTRZERO] = { .call = ctnetlink_get_conntrack, |
f73e924c PM |
1577 | .attr_count = CTA_MAX, |
1578 | .policy = ct_nla_policy }, | |
c1d10adb PNA |
1579 | }; |
1580 | ||
7c8d4cb4 | 1581 | static const struct nfnl_callback ctnl_exp_cb[IPCTNL_MSG_EXP_MAX] = { |
c1d10adb | 1582 | [IPCTNL_MSG_EXP_GET] = { .call = ctnetlink_get_expect, |
f73e924c PM |
1583 | .attr_count = CTA_EXPECT_MAX, |
1584 | .policy = exp_nla_policy }, | |
c1d10adb | 1585 | [IPCTNL_MSG_EXP_NEW] = { .call = ctnetlink_new_expect, |
f73e924c PM |
1586 | .attr_count = CTA_EXPECT_MAX, |
1587 | .policy = exp_nla_policy }, | |
c1d10adb | 1588 | [IPCTNL_MSG_EXP_DELETE] = { .call = ctnetlink_del_expect, |
f73e924c PM |
1589 | .attr_count = CTA_EXPECT_MAX, |
1590 | .policy = exp_nla_policy }, | |
c1d10adb PNA |
1591 | }; |
1592 | ||
7c8d4cb4 | 1593 | static const struct nfnetlink_subsystem ctnl_subsys = { |
c1d10adb PNA |
1594 | .name = "conntrack", |
1595 | .subsys_id = NFNL_SUBSYS_CTNETLINK, | |
1596 | .cb_count = IPCTNL_MSG_MAX, | |
1597 | .cb = ctnl_cb, | |
1598 | }; | |
1599 | ||
7c8d4cb4 | 1600 | static const struct nfnetlink_subsystem ctnl_exp_subsys = { |
c1d10adb PNA |
1601 | .name = "conntrack_expect", |
1602 | .subsys_id = NFNL_SUBSYS_CTNETLINK_EXP, | |
1603 | .cb_count = IPCTNL_MSG_EXP_MAX, | |
1604 | .cb = ctnl_exp_cb, | |
1605 | }; | |
1606 | ||
d2483dde | 1607 | MODULE_ALIAS("ip_conntrack_netlink"); |
c1d10adb | 1608 | MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK); |
34f9a2e4 | 1609 | MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK_EXP); |
c1d10adb PNA |
1610 | |
1611 | static int __init ctnetlink_init(void) | |
1612 | { | |
1613 | int ret; | |
1614 | ||
1615 | printk("ctnetlink v%s: registering with nfnetlink.\n", version); | |
1616 | ret = nfnetlink_subsys_register(&ctnl_subsys); | |
1617 | if (ret < 0) { | |
1618 | printk("ctnetlink_init: cannot register with nfnetlink.\n"); | |
1619 | goto err_out; | |
1620 | } | |
1621 | ||
1622 | ret = nfnetlink_subsys_register(&ctnl_exp_subsys); | |
1623 | if (ret < 0) { | |
1624 | printk("ctnetlink_init: cannot register exp with nfnetlink.\n"); | |
1625 | goto err_unreg_subsys; | |
1626 | } | |
1627 | ||
1628 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1629 | ret = nf_conntrack_register_notifier(&ctnl_notifier); | |
1630 | if (ret < 0) { | |
1631 | printk("ctnetlink_init: cannot register notifier.\n"); | |
1632 | goto err_unreg_exp_subsys; | |
1633 | } | |
1634 | ||
6823645d | 1635 | ret = nf_ct_expect_register_notifier(&ctnl_notifier_exp); |
c1d10adb PNA |
1636 | if (ret < 0) { |
1637 | printk("ctnetlink_init: cannot expect register notifier.\n"); | |
1638 | goto err_unreg_notifier; | |
1639 | } | |
1640 | #endif | |
1641 | ||
1642 | return 0; | |
1643 | ||
1644 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
1645 | err_unreg_notifier: | |
1646 | nf_conntrack_unregister_notifier(&ctnl_notifier); | |
1647 | err_unreg_exp_subsys: | |
1648 | nfnetlink_subsys_unregister(&ctnl_exp_subsys); | |
1649 | #endif | |
1650 | err_unreg_subsys: | |
1651 | nfnetlink_subsys_unregister(&ctnl_subsys); | |
1652 | err_out: | |
1653 | return ret; | |
1654 | } | |
1655 | ||
1656 | static void __exit ctnetlink_exit(void) | |
1657 | { | |
1658 | printk("ctnetlink: unregistering from nfnetlink.\n"); | |
1659 | ||
1660 | #ifdef CONFIG_NF_CONNTRACK_EVENTS | |
6823645d | 1661 | nf_ct_expect_unregister_notifier(&ctnl_notifier_exp); |
c1d10adb PNA |
1662 | nf_conntrack_unregister_notifier(&ctnl_notifier); |
1663 | #endif | |
1664 | ||
1665 | nfnetlink_subsys_unregister(&ctnl_exp_subsys); | |
1666 | nfnetlink_subsys_unregister(&ctnl_subsys); | |
1667 | return; | |
1668 | } | |
1669 | ||
1670 | module_init(ctnetlink_init); | |
1671 | module_exit(ctnetlink_exit); |