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