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