dm thin: wake worker when discard is prepared
[linux-2.6.git] / net / netfilter / nf_conntrack_netlink.c
blob7bbfb3deea305e4d8434c3af9a68ee6e4531c80f
1 /* Connection tracking via netlink socket. Allows for user space
2 * protocol helpers and general trouble making from userspace.
4 * (C) 2001 by Jay Schulist <jschlst@samba.org>
5 * (C) 2002-2006 by Harald Welte <laforge@gnumonks.org>
6 * (C) 2003 by Patrick Mchardy <kaber@trash.net>
7 * (C) 2005-2012 by Pablo Neira Ayuso <pablo@netfilter.org>
9 * Initial connection tracking via netlink development funded and
10 * generally made possible by Network Robots, Inc. (www.networkrobots.com)
12 * Further development of this code funded by Astaro AG (http://www.astaro.com)
14 * This software may be used and distributed according to the terms
15 * of the GNU General Public License, incorporated herein by reference.
18 #include <linux/init.h>
19 #include <linux/module.h>
20 #include <linux/kernel.h>
21 #include <linux/rculist.h>
22 #include <linux/rculist_nulls.h>
23 #include <linux/types.h>
24 #include <linux/timer.h>
25 #include <linux/security.h>
26 #include <linux/skbuff.h>
27 #include <linux/errno.h>
28 #include <linux/netlink.h>
29 #include <linux/spinlock.h>
30 #include <linux/interrupt.h>
31 #include <linux/slab.h>
33 #include <linux/netfilter.h>
34 #include <net/netlink.h>
35 #include <net/sock.h>
36 #include <net/netfilter/nf_conntrack.h>
37 #include <net/netfilter/nf_conntrack_core.h>
38 #include <net/netfilter/nf_conntrack_expect.h>
39 #include <net/netfilter/nf_conntrack_helper.h>
40 #include <net/netfilter/nf_conntrack_l3proto.h>
41 #include <net/netfilter/nf_conntrack_l4proto.h>
42 #include <net/netfilter/nf_conntrack_tuple.h>
43 #include <net/netfilter/nf_conntrack_acct.h>
44 #include <net/netfilter/nf_conntrack_zones.h>
45 #include <net/netfilter/nf_conntrack_timestamp.h>
46 #ifdef CONFIG_NF_NAT_NEEDED
47 #include <net/netfilter/nf_nat_core.h>
48 #include <net/netfilter/nf_nat_l4proto.h>
49 #include <net/netfilter/nf_nat_helper.h>
50 #endif
52 #include <linux/netfilter/nfnetlink.h>
53 #include <linux/netfilter/nfnetlink_conntrack.h>
55 MODULE_LICENSE("GPL");
57 static char __initdata version[] = "0.93";
59 static inline int
60 ctnetlink_dump_tuples_proto(struct sk_buff *skb,
61 const struct nf_conntrack_tuple *tuple,
62 struct nf_conntrack_l4proto *l4proto)
64 int ret = 0;
65 struct nlattr *nest_parms;
67 nest_parms = nla_nest_start(skb, CTA_TUPLE_PROTO | NLA_F_NESTED);
68 if (!nest_parms)
69 goto nla_put_failure;
70 if (nla_put_u8(skb, CTA_PROTO_NUM, tuple->dst.protonum))
71 goto nla_put_failure;
73 if (likely(l4proto->tuple_to_nlattr))
74 ret = l4proto->tuple_to_nlattr(skb, tuple);
76 nla_nest_end(skb, nest_parms);
78 return ret;
80 nla_put_failure:
81 return -1;
84 static inline int
85 ctnetlink_dump_tuples_ip(struct sk_buff *skb,
86 const struct nf_conntrack_tuple *tuple,
87 struct nf_conntrack_l3proto *l3proto)
89 int ret = 0;
90 struct nlattr *nest_parms;
92 nest_parms = nla_nest_start(skb, CTA_TUPLE_IP | NLA_F_NESTED);
93 if (!nest_parms)
94 goto nla_put_failure;
96 if (likely(l3proto->tuple_to_nlattr))
97 ret = l3proto->tuple_to_nlattr(skb, tuple);
99 nla_nest_end(skb, nest_parms);
101 return ret;
103 nla_put_failure:
104 return -1;
107 static int
108 ctnetlink_dump_tuples(struct sk_buff *skb,
109 const struct nf_conntrack_tuple *tuple)
111 int ret;
112 struct nf_conntrack_l3proto *l3proto;
113 struct nf_conntrack_l4proto *l4proto;
115 rcu_read_lock();
116 l3proto = __nf_ct_l3proto_find(tuple->src.l3num);
117 ret = ctnetlink_dump_tuples_ip(skb, tuple, l3proto);
119 if (ret >= 0) {
120 l4proto = __nf_ct_l4proto_find(tuple->src.l3num,
121 tuple->dst.protonum);
122 ret = ctnetlink_dump_tuples_proto(skb, tuple, l4proto);
124 rcu_read_unlock();
125 return ret;
128 static inline int
129 ctnetlink_dump_status(struct sk_buff *skb, const struct nf_conn *ct)
131 if (nla_put_be32(skb, CTA_STATUS, htonl(ct->status)))
132 goto nla_put_failure;
133 return 0;
135 nla_put_failure:
136 return -1;
139 static inline int
140 ctnetlink_dump_timeout(struct sk_buff *skb, const struct nf_conn *ct)
142 long timeout = ((long)ct->timeout.expires - (long)jiffies) / HZ;
144 if (timeout < 0)
145 timeout = 0;
147 if (nla_put_be32(skb, CTA_TIMEOUT, htonl(timeout)))
148 goto nla_put_failure;
149 return 0;
151 nla_put_failure:
152 return -1;
155 static inline int
156 ctnetlink_dump_protoinfo(struct sk_buff *skb, struct nf_conn *ct)
158 struct nf_conntrack_l4proto *l4proto;
159 struct nlattr *nest_proto;
160 int ret;
162 l4proto = __nf_ct_l4proto_find(nf_ct_l3num(ct), nf_ct_protonum(ct));
163 if (!l4proto->to_nlattr)
164 return 0;
166 nest_proto = nla_nest_start(skb, CTA_PROTOINFO | NLA_F_NESTED);
167 if (!nest_proto)
168 goto nla_put_failure;
170 ret = l4proto->to_nlattr(skb, nest_proto, ct);
172 nla_nest_end(skb, nest_proto);
174 return ret;
176 nla_put_failure:
177 return -1;
180 static inline int
181 ctnetlink_dump_helpinfo(struct sk_buff *skb, const struct nf_conn *ct)
183 struct nlattr *nest_helper;
184 const struct nf_conn_help *help = nfct_help(ct);
185 struct nf_conntrack_helper *helper;
187 if (!help)
188 return 0;
190 helper = rcu_dereference(help->helper);
191 if (!helper)
192 goto out;
194 nest_helper = nla_nest_start(skb, CTA_HELP | NLA_F_NESTED);
195 if (!nest_helper)
196 goto nla_put_failure;
197 if (nla_put_string(skb, CTA_HELP_NAME, helper->name))
198 goto nla_put_failure;
200 if (helper->to_nlattr)
201 helper->to_nlattr(skb, ct);
203 nla_nest_end(skb, nest_helper);
204 out:
205 return 0;
207 nla_put_failure:
208 return -1;
211 static int
212 dump_counters(struct sk_buff *skb, u64 pkts, u64 bytes,
213 enum ip_conntrack_dir dir)
215 enum ctattr_type type = dir ? CTA_COUNTERS_REPLY: CTA_COUNTERS_ORIG;
216 struct nlattr *nest_count;
218 nest_count = nla_nest_start(skb, type | NLA_F_NESTED);
219 if (!nest_count)
220 goto nla_put_failure;
222 if (nla_put_be64(skb, CTA_COUNTERS_PACKETS, cpu_to_be64(pkts)) ||
223 nla_put_be64(skb, CTA_COUNTERS_BYTES, cpu_to_be64(bytes)))
224 goto nla_put_failure;
226 nla_nest_end(skb, nest_count);
228 return 0;
230 nla_put_failure:
231 return -1;
234 static int
235 ctnetlink_dump_counters(struct sk_buff *skb, const struct nf_conn *ct,
236 enum ip_conntrack_dir dir, int type)
238 struct nf_conn_counter *acct;
239 u64 pkts, bytes;
241 acct = nf_conn_acct_find(ct);
242 if (!acct)
243 return 0;
245 if (type == IPCTNL_MSG_CT_GET_CTRZERO) {
246 pkts = atomic64_xchg(&acct[dir].packets, 0);
247 bytes = atomic64_xchg(&acct[dir].bytes, 0);
248 } else {
249 pkts = atomic64_read(&acct[dir].packets);
250 bytes = atomic64_read(&acct[dir].bytes);
252 return dump_counters(skb, pkts, bytes, dir);
255 static int
256 ctnetlink_dump_timestamp(struct sk_buff *skb, const struct nf_conn *ct)
258 struct nlattr *nest_count;
259 const struct nf_conn_tstamp *tstamp;
261 tstamp = nf_conn_tstamp_find(ct);
262 if (!tstamp)
263 return 0;
265 nest_count = nla_nest_start(skb, CTA_TIMESTAMP | NLA_F_NESTED);
266 if (!nest_count)
267 goto nla_put_failure;
269 if (nla_put_be64(skb, CTA_TIMESTAMP_START, cpu_to_be64(tstamp->start)) ||
270 (tstamp->stop != 0 && nla_put_be64(skb, CTA_TIMESTAMP_STOP,
271 cpu_to_be64(tstamp->stop))))
272 goto nla_put_failure;
273 nla_nest_end(skb, nest_count);
275 return 0;
277 nla_put_failure:
278 return -1;
281 #ifdef CONFIG_NF_CONNTRACK_MARK
282 static inline int
283 ctnetlink_dump_mark(struct sk_buff *skb, const struct nf_conn *ct)
285 if (nla_put_be32(skb, CTA_MARK, htonl(ct->mark)))
286 goto nla_put_failure;
287 return 0;
289 nla_put_failure:
290 return -1;
292 #else
293 #define ctnetlink_dump_mark(a, b) (0)
294 #endif
296 #ifdef CONFIG_NF_CONNTRACK_SECMARK
297 static inline int
298 ctnetlink_dump_secctx(struct sk_buff *skb, const struct nf_conn *ct)
300 struct nlattr *nest_secctx;
301 int len, ret;
302 char *secctx;
304 ret = security_secid_to_secctx(ct->secmark, &secctx, &len);
305 if (ret)
306 return 0;
308 ret = -1;
309 nest_secctx = nla_nest_start(skb, CTA_SECCTX | NLA_F_NESTED);
310 if (!nest_secctx)
311 goto nla_put_failure;
313 if (nla_put_string(skb, CTA_SECCTX_NAME, secctx))
314 goto nla_put_failure;
315 nla_nest_end(skb, nest_secctx);
317 ret = 0;
318 nla_put_failure:
319 security_release_secctx(secctx, len);
320 return ret;
322 #else
323 #define ctnetlink_dump_secctx(a, b) (0)
324 #endif
326 #define master_tuple(ct) &(ct->master->tuplehash[IP_CT_DIR_ORIGINAL].tuple)
328 static inline int
329 ctnetlink_dump_master(struct sk_buff *skb, const struct nf_conn *ct)
331 struct nlattr *nest_parms;
333 if (!(ct->status & IPS_EXPECTED))
334 return 0;
336 nest_parms = nla_nest_start(skb, CTA_TUPLE_MASTER | NLA_F_NESTED);
337 if (!nest_parms)
338 goto nla_put_failure;
339 if (ctnetlink_dump_tuples(skb, master_tuple(ct)) < 0)
340 goto nla_put_failure;
341 nla_nest_end(skb, nest_parms);
343 return 0;
345 nla_put_failure:
346 return -1;
349 #ifdef CONFIG_NF_NAT_NEEDED
350 static int
351 dump_nat_seq_adj(struct sk_buff *skb, const struct nf_nat_seq *natseq, int type)
353 struct nlattr *nest_parms;
355 nest_parms = nla_nest_start(skb, type | NLA_F_NESTED);
356 if (!nest_parms)
357 goto nla_put_failure;
359 if (nla_put_be32(skb, CTA_NAT_SEQ_CORRECTION_POS,
360 htonl(natseq->correction_pos)) ||
361 nla_put_be32(skb, CTA_NAT_SEQ_OFFSET_BEFORE,
362 htonl(natseq->offset_before)) ||
363 nla_put_be32(skb, CTA_NAT_SEQ_OFFSET_AFTER,
364 htonl(natseq->offset_after)))
365 goto nla_put_failure;
367 nla_nest_end(skb, nest_parms);
369 return 0;
371 nla_put_failure:
372 return -1;
375 static inline int
376 ctnetlink_dump_nat_seq_adj(struct sk_buff *skb, const struct nf_conn *ct)
378 struct nf_nat_seq *natseq;
379 struct nf_conn_nat *nat = nfct_nat(ct);
381 if (!(ct->status & IPS_SEQ_ADJUST) || !nat)
382 return 0;
384 natseq = &nat->seq[IP_CT_DIR_ORIGINAL];
385 if (dump_nat_seq_adj(skb, natseq, CTA_NAT_SEQ_ADJ_ORIG) == -1)
386 return -1;
388 natseq = &nat->seq[IP_CT_DIR_REPLY];
389 if (dump_nat_seq_adj(skb, natseq, CTA_NAT_SEQ_ADJ_REPLY) == -1)
390 return -1;
392 return 0;
394 #else
395 #define ctnetlink_dump_nat_seq_adj(a, b) (0)
396 #endif
398 static inline int
399 ctnetlink_dump_id(struct sk_buff *skb, const struct nf_conn *ct)
401 if (nla_put_be32(skb, CTA_ID, htonl((unsigned long)ct)))
402 goto nla_put_failure;
403 return 0;
405 nla_put_failure:
406 return -1;
409 static inline int
410 ctnetlink_dump_use(struct sk_buff *skb, const struct nf_conn *ct)
412 if (nla_put_be32(skb, CTA_USE, htonl(atomic_read(&ct->ct_general.use))))
413 goto nla_put_failure;
414 return 0;
416 nla_put_failure:
417 return -1;
420 static int
421 ctnetlink_fill_info(struct sk_buff *skb, u32 portid, u32 seq, u32 type,
422 struct nf_conn *ct)
424 struct nlmsghdr *nlh;
425 struct nfgenmsg *nfmsg;
426 struct nlattr *nest_parms;
427 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
429 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_CT_NEW);
430 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
431 if (nlh == NULL)
432 goto nlmsg_failure;
434 nfmsg = nlmsg_data(nlh);
435 nfmsg->nfgen_family = nf_ct_l3num(ct);
436 nfmsg->version = NFNETLINK_V0;
437 nfmsg->res_id = 0;
439 nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED);
440 if (!nest_parms)
441 goto nla_put_failure;
442 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_ORIGINAL)) < 0)
443 goto nla_put_failure;
444 nla_nest_end(skb, nest_parms);
446 nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED);
447 if (!nest_parms)
448 goto nla_put_failure;
449 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_REPLY)) < 0)
450 goto nla_put_failure;
451 nla_nest_end(skb, nest_parms);
453 if (nf_ct_zone(ct) &&
454 nla_put_be16(skb, CTA_ZONE, htons(nf_ct_zone(ct))))
455 goto nla_put_failure;
457 if (ctnetlink_dump_status(skb, ct) < 0 ||
458 ctnetlink_dump_timeout(skb, ct) < 0 ||
459 ctnetlink_dump_counters(skb, ct, IP_CT_DIR_ORIGINAL, type) < 0 ||
460 ctnetlink_dump_counters(skb, ct, IP_CT_DIR_REPLY, type) < 0 ||
461 ctnetlink_dump_timestamp(skb, ct) < 0 ||
462 ctnetlink_dump_protoinfo(skb, ct) < 0 ||
463 ctnetlink_dump_helpinfo(skb, ct) < 0 ||
464 ctnetlink_dump_mark(skb, ct) < 0 ||
465 ctnetlink_dump_secctx(skb, ct) < 0 ||
466 ctnetlink_dump_id(skb, ct) < 0 ||
467 ctnetlink_dump_use(skb, ct) < 0 ||
468 ctnetlink_dump_master(skb, ct) < 0 ||
469 ctnetlink_dump_nat_seq_adj(skb, ct) < 0)
470 goto nla_put_failure;
472 nlmsg_end(skb, nlh);
473 return skb->len;
475 nlmsg_failure:
476 nla_put_failure:
477 nlmsg_cancel(skb, nlh);
478 return -1;
481 static inline size_t
482 ctnetlink_proto_size(const struct nf_conn *ct)
484 struct nf_conntrack_l3proto *l3proto;
485 struct nf_conntrack_l4proto *l4proto;
486 size_t len = 0;
488 rcu_read_lock();
489 l3proto = __nf_ct_l3proto_find(nf_ct_l3num(ct));
490 len += l3proto->nla_size;
492 l4proto = __nf_ct_l4proto_find(nf_ct_l3num(ct), nf_ct_protonum(ct));
493 len += l4proto->nla_size;
494 rcu_read_unlock();
496 return len;
499 static inline size_t
500 ctnetlink_counters_size(const struct nf_conn *ct)
502 if (!nf_ct_ext_exist(ct, NF_CT_EXT_ACCT))
503 return 0;
504 return 2 * nla_total_size(0) /* CTA_COUNTERS_ORIG|REPL */
505 + 2 * nla_total_size(sizeof(uint64_t)) /* CTA_COUNTERS_PACKETS */
506 + 2 * nla_total_size(sizeof(uint64_t)) /* CTA_COUNTERS_BYTES */
510 static inline int
511 ctnetlink_secctx_size(const struct nf_conn *ct)
513 #ifdef CONFIG_NF_CONNTRACK_SECMARK
514 int len, ret;
516 ret = security_secid_to_secctx(ct->secmark, NULL, &len);
517 if (ret)
518 return 0;
520 return nla_total_size(0) /* CTA_SECCTX */
521 + nla_total_size(sizeof(char) * len); /* CTA_SECCTX_NAME */
522 #else
523 return 0;
524 #endif
527 static inline size_t
528 ctnetlink_timestamp_size(const struct nf_conn *ct)
530 #ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
531 if (!nf_ct_ext_exist(ct, NF_CT_EXT_TSTAMP))
532 return 0;
533 return nla_total_size(0) + 2 * nla_total_size(sizeof(uint64_t));
534 #else
535 return 0;
536 #endif
539 static inline size_t
540 ctnetlink_nlmsg_size(const struct nf_conn *ct)
542 return NLMSG_ALIGN(sizeof(struct nfgenmsg))
543 + 3 * nla_total_size(0) /* CTA_TUPLE_ORIG|REPL|MASTER */
544 + 3 * nla_total_size(0) /* CTA_TUPLE_IP */
545 + 3 * nla_total_size(0) /* CTA_TUPLE_PROTO */
546 + 3 * nla_total_size(sizeof(u_int8_t)) /* CTA_PROTO_NUM */
547 + nla_total_size(sizeof(u_int32_t)) /* CTA_ID */
548 + nla_total_size(sizeof(u_int32_t)) /* CTA_STATUS */
549 + ctnetlink_counters_size(ct)
550 + ctnetlink_timestamp_size(ct)
551 + nla_total_size(sizeof(u_int32_t)) /* CTA_TIMEOUT */
552 + nla_total_size(0) /* CTA_PROTOINFO */
553 + nla_total_size(0) /* CTA_HELP */
554 + nla_total_size(NF_CT_HELPER_NAME_LEN) /* CTA_HELP_NAME */
555 + ctnetlink_secctx_size(ct)
556 #ifdef CONFIG_NF_NAT_NEEDED
557 + 2 * nla_total_size(0) /* CTA_NAT_SEQ_ADJ_ORIG|REPL */
558 + 6 * nla_total_size(sizeof(u_int32_t)) /* CTA_NAT_SEQ_OFFSET */
559 #endif
560 #ifdef CONFIG_NF_CONNTRACK_MARK
561 + nla_total_size(sizeof(u_int32_t)) /* CTA_MARK */
562 #endif
563 + ctnetlink_proto_size(ct)
567 #ifdef CONFIG_NF_CONNTRACK_EVENTS
568 static int
569 ctnetlink_conntrack_event(unsigned int events, struct nf_ct_event *item)
571 struct net *net;
572 struct nlmsghdr *nlh;
573 struct nfgenmsg *nfmsg;
574 struct nlattr *nest_parms;
575 struct nf_conn *ct = item->ct;
576 struct sk_buff *skb;
577 unsigned int type;
578 unsigned int flags = 0, group;
579 int err;
581 /* ignore our fake conntrack entry */
582 if (nf_ct_is_untracked(ct))
583 return 0;
585 if (events & (1 << IPCT_DESTROY)) {
586 type = IPCTNL_MSG_CT_DELETE;
587 group = NFNLGRP_CONNTRACK_DESTROY;
588 } else if (events & ((1 << IPCT_NEW) | (1 << IPCT_RELATED))) {
589 type = IPCTNL_MSG_CT_NEW;
590 flags = NLM_F_CREATE|NLM_F_EXCL;
591 group = NFNLGRP_CONNTRACK_NEW;
592 } else if (events) {
593 type = IPCTNL_MSG_CT_NEW;
594 group = NFNLGRP_CONNTRACK_UPDATE;
595 } else
596 return 0;
598 net = nf_ct_net(ct);
599 if (!item->report && !nfnetlink_has_listeners(net, group))
600 return 0;
602 skb = nlmsg_new(ctnetlink_nlmsg_size(ct), GFP_ATOMIC);
603 if (skb == NULL)
604 goto errout;
606 type |= NFNL_SUBSYS_CTNETLINK << 8;
607 nlh = nlmsg_put(skb, item->portid, 0, type, sizeof(*nfmsg), flags);
608 if (nlh == NULL)
609 goto nlmsg_failure;
611 nfmsg = nlmsg_data(nlh);
612 nfmsg->nfgen_family = nf_ct_l3num(ct);
613 nfmsg->version = NFNETLINK_V0;
614 nfmsg->res_id = 0;
616 rcu_read_lock();
617 nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED);
618 if (!nest_parms)
619 goto nla_put_failure;
620 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_ORIGINAL)) < 0)
621 goto nla_put_failure;
622 nla_nest_end(skb, nest_parms);
624 nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED);
625 if (!nest_parms)
626 goto nla_put_failure;
627 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_REPLY)) < 0)
628 goto nla_put_failure;
629 nla_nest_end(skb, nest_parms);
631 if (nf_ct_zone(ct) &&
632 nla_put_be16(skb, CTA_ZONE, htons(nf_ct_zone(ct))))
633 goto nla_put_failure;
635 if (ctnetlink_dump_id(skb, ct) < 0)
636 goto nla_put_failure;
638 if (ctnetlink_dump_status(skb, ct) < 0)
639 goto nla_put_failure;
641 if (events & (1 << IPCT_DESTROY)) {
642 if (ctnetlink_dump_counters(skb, ct,
643 IP_CT_DIR_ORIGINAL, type) < 0 ||
644 ctnetlink_dump_counters(skb, ct,
645 IP_CT_DIR_REPLY, type) < 0 ||
646 ctnetlink_dump_timestamp(skb, ct) < 0)
647 goto nla_put_failure;
648 } else {
649 if (ctnetlink_dump_timeout(skb, ct) < 0)
650 goto nla_put_failure;
652 if (events & (1 << IPCT_PROTOINFO)
653 && ctnetlink_dump_protoinfo(skb, ct) < 0)
654 goto nla_put_failure;
656 if ((events & (1 << IPCT_HELPER) || nfct_help(ct))
657 && ctnetlink_dump_helpinfo(skb, ct) < 0)
658 goto nla_put_failure;
660 #ifdef CONFIG_NF_CONNTRACK_SECMARK
661 if ((events & (1 << IPCT_SECMARK) || ct->secmark)
662 && ctnetlink_dump_secctx(skb, ct) < 0)
663 goto nla_put_failure;
664 #endif
666 if (events & (1 << IPCT_RELATED) &&
667 ctnetlink_dump_master(skb, ct) < 0)
668 goto nla_put_failure;
670 if (events & (1 << IPCT_NATSEQADJ) &&
671 ctnetlink_dump_nat_seq_adj(skb, ct) < 0)
672 goto nla_put_failure;
675 #ifdef CONFIG_NF_CONNTRACK_MARK
676 if ((events & (1 << IPCT_MARK) || ct->mark)
677 && ctnetlink_dump_mark(skb, ct) < 0)
678 goto nla_put_failure;
679 #endif
680 rcu_read_unlock();
682 nlmsg_end(skb, nlh);
683 err = nfnetlink_send(skb, net, item->portid, group, item->report,
684 GFP_ATOMIC);
685 if (err == -ENOBUFS || err == -EAGAIN)
686 return -ENOBUFS;
688 return 0;
690 nla_put_failure:
691 rcu_read_unlock();
692 nlmsg_cancel(skb, nlh);
693 nlmsg_failure:
694 kfree_skb(skb);
695 errout:
696 if (nfnetlink_set_err(net, 0, group, -ENOBUFS) > 0)
697 return -ENOBUFS;
699 return 0;
701 #endif /* CONFIG_NF_CONNTRACK_EVENTS */
703 static int ctnetlink_done(struct netlink_callback *cb)
705 if (cb->args[1])
706 nf_ct_put((struct nf_conn *)cb->args[1]);
707 if (cb->data)
708 kfree(cb->data);
709 return 0;
712 struct ctnetlink_dump_filter {
713 struct {
714 u_int32_t val;
715 u_int32_t mask;
716 } mark;
719 static int
720 ctnetlink_dump_table(struct sk_buff *skb, struct netlink_callback *cb)
722 struct net *net = sock_net(skb->sk);
723 struct nf_conn *ct, *last;
724 struct nf_conntrack_tuple_hash *h;
725 struct hlist_nulls_node *n;
726 struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
727 u_int8_t l3proto = nfmsg->nfgen_family;
728 int res;
729 #ifdef CONFIG_NF_CONNTRACK_MARK
730 const struct ctnetlink_dump_filter *filter = cb->data;
731 #endif
733 spin_lock_bh(&nf_conntrack_lock);
734 last = (struct nf_conn *)cb->args[1];
735 for (; cb->args[0] < net->ct.htable_size; cb->args[0]++) {
736 restart:
737 hlist_nulls_for_each_entry(h, n, &net->ct.hash[cb->args[0]],
738 hnnode) {
739 if (NF_CT_DIRECTION(h) != IP_CT_DIR_ORIGINAL)
740 continue;
741 ct = nf_ct_tuplehash_to_ctrack(h);
742 /* Dump entries of a given L3 protocol number.
743 * If it is not specified, ie. l3proto == 0,
744 * then dump everything. */
745 if (l3proto && nf_ct_l3num(ct) != l3proto)
746 continue;
747 if (cb->args[1]) {
748 if (ct != last)
749 continue;
750 cb->args[1] = 0;
752 #ifdef CONFIG_NF_CONNTRACK_MARK
753 if (filter && !((ct->mark & filter->mark.mask) ==
754 filter->mark.val)) {
755 continue;
757 #endif
758 rcu_read_lock();
759 res =
760 ctnetlink_fill_info(skb, NETLINK_CB(cb->skb).portid,
761 cb->nlh->nlmsg_seq,
762 NFNL_MSG_TYPE(cb->nlh->nlmsg_type),
763 ct);
764 rcu_read_unlock();
765 if (res < 0) {
766 nf_conntrack_get(&ct->ct_general);
767 cb->args[1] = (unsigned long)ct;
768 goto out;
771 if (cb->args[1]) {
772 cb->args[1] = 0;
773 goto restart;
776 out:
777 spin_unlock_bh(&nf_conntrack_lock);
778 if (last)
779 nf_ct_put(last);
781 return skb->len;
784 static inline int
785 ctnetlink_parse_tuple_ip(struct nlattr *attr, struct nf_conntrack_tuple *tuple)
787 struct nlattr *tb[CTA_IP_MAX+1];
788 struct nf_conntrack_l3proto *l3proto;
789 int ret = 0;
791 nla_parse_nested(tb, CTA_IP_MAX, attr, NULL);
793 rcu_read_lock();
794 l3proto = __nf_ct_l3proto_find(tuple->src.l3num);
796 if (likely(l3proto->nlattr_to_tuple)) {
797 ret = nla_validate_nested(attr, CTA_IP_MAX,
798 l3proto->nla_policy);
799 if (ret == 0)
800 ret = l3proto->nlattr_to_tuple(tb, tuple);
803 rcu_read_unlock();
805 return ret;
808 static const struct nla_policy proto_nla_policy[CTA_PROTO_MAX+1] = {
809 [CTA_PROTO_NUM] = { .type = NLA_U8 },
812 static inline int
813 ctnetlink_parse_tuple_proto(struct nlattr *attr,
814 struct nf_conntrack_tuple *tuple)
816 struct nlattr *tb[CTA_PROTO_MAX+1];
817 struct nf_conntrack_l4proto *l4proto;
818 int ret = 0;
820 ret = nla_parse_nested(tb, CTA_PROTO_MAX, attr, proto_nla_policy);
821 if (ret < 0)
822 return ret;
824 if (!tb[CTA_PROTO_NUM])
825 return -EINVAL;
826 tuple->dst.protonum = nla_get_u8(tb[CTA_PROTO_NUM]);
828 rcu_read_lock();
829 l4proto = __nf_ct_l4proto_find(tuple->src.l3num, tuple->dst.protonum);
831 if (likely(l4proto->nlattr_to_tuple)) {
832 ret = nla_validate_nested(attr, CTA_PROTO_MAX,
833 l4proto->nla_policy);
834 if (ret == 0)
835 ret = l4proto->nlattr_to_tuple(tb, tuple);
838 rcu_read_unlock();
840 return ret;
843 static const struct nla_policy tuple_nla_policy[CTA_TUPLE_MAX+1] = {
844 [CTA_TUPLE_IP] = { .type = NLA_NESTED },
845 [CTA_TUPLE_PROTO] = { .type = NLA_NESTED },
848 static int
849 ctnetlink_parse_tuple(const struct nlattr * const cda[],
850 struct nf_conntrack_tuple *tuple,
851 enum ctattr_type type, u_int8_t l3num)
853 struct nlattr *tb[CTA_TUPLE_MAX+1];
854 int err;
856 memset(tuple, 0, sizeof(*tuple));
858 nla_parse_nested(tb, CTA_TUPLE_MAX, cda[type], tuple_nla_policy);
860 if (!tb[CTA_TUPLE_IP])
861 return -EINVAL;
863 tuple->src.l3num = l3num;
865 err = ctnetlink_parse_tuple_ip(tb[CTA_TUPLE_IP], tuple);
866 if (err < 0)
867 return err;
869 if (!tb[CTA_TUPLE_PROTO])
870 return -EINVAL;
872 err = ctnetlink_parse_tuple_proto(tb[CTA_TUPLE_PROTO], tuple);
873 if (err < 0)
874 return err;
876 /* orig and expect tuples get DIR_ORIGINAL */
877 if (type == CTA_TUPLE_REPLY)
878 tuple->dst.dir = IP_CT_DIR_REPLY;
879 else
880 tuple->dst.dir = IP_CT_DIR_ORIGINAL;
882 return 0;
885 static int
886 ctnetlink_parse_zone(const struct nlattr *attr, u16 *zone)
888 if (attr)
889 #ifdef CONFIG_NF_CONNTRACK_ZONES
890 *zone = ntohs(nla_get_be16(attr));
891 #else
892 return -EOPNOTSUPP;
893 #endif
894 else
895 *zone = 0;
897 return 0;
900 static const struct nla_policy help_nla_policy[CTA_HELP_MAX+1] = {
901 [CTA_HELP_NAME] = { .type = NLA_NUL_STRING },
904 static inline int
905 ctnetlink_parse_help(const struct nlattr *attr, char **helper_name,
906 struct nlattr **helpinfo)
908 struct nlattr *tb[CTA_HELP_MAX+1];
910 nla_parse_nested(tb, CTA_HELP_MAX, attr, help_nla_policy);
912 if (!tb[CTA_HELP_NAME])
913 return -EINVAL;
915 *helper_name = nla_data(tb[CTA_HELP_NAME]);
917 if (tb[CTA_HELP_INFO])
918 *helpinfo = tb[CTA_HELP_INFO];
920 return 0;
923 static const struct nla_policy ct_nla_policy[CTA_MAX+1] = {
924 [CTA_TUPLE_ORIG] = { .type = NLA_NESTED },
925 [CTA_TUPLE_REPLY] = { .type = NLA_NESTED },
926 [CTA_STATUS] = { .type = NLA_U32 },
927 [CTA_PROTOINFO] = { .type = NLA_NESTED },
928 [CTA_HELP] = { .type = NLA_NESTED },
929 [CTA_NAT_SRC] = { .type = NLA_NESTED },
930 [CTA_TIMEOUT] = { .type = NLA_U32 },
931 [CTA_MARK] = { .type = NLA_U32 },
932 [CTA_ID] = { .type = NLA_U32 },
933 [CTA_NAT_DST] = { .type = NLA_NESTED },
934 [CTA_TUPLE_MASTER] = { .type = NLA_NESTED },
935 [CTA_ZONE] = { .type = NLA_U16 },
936 [CTA_MARK_MASK] = { .type = NLA_U32 },
939 static int
940 ctnetlink_del_conntrack(struct sock *ctnl, struct sk_buff *skb,
941 const struct nlmsghdr *nlh,
942 const struct nlattr * const cda[])
944 struct net *net = sock_net(ctnl);
945 struct nf_conntrack_tuple_hash *h;
946 struct nf_conntrack_tuple tuple;
947 struct nf_conn *ct;
948 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
949 u_int8_t u3 = nfmsg->nfgen_family;
950 u16 zone;
951 int err;
953 err = ctnetlink_parse_zone(cda[CTA_ZONE], &zone);
954 if (err < 0)
955 return err;
957 if (cda[CTA_TUPLE_ORIG])
958 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_ORIG, u3);
959 else if (cda[CTA_TUPLE_REPLY])
960 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_REPLY, u3);
961 else {
962 /* Flush the whole table */
963 nf_conntrack_flush_report(net,
964 NETLINK_CB(skb).portid,
965 nlmsg_report(nlh));
966 return 0;
969 if (err < 0)
970 return err;
972 h = nf_conntrack_find_get(net, zone, &tuple);
973 if (!h)
974 return -ENOENT;
976 ct = nf_ct_tuplehash_to_ctrack(h);
978 if (cda[CTA_ID]) {
979 u_int32_t id = ntohl(nla_get_be32(cda[CTA_ID]));
980 if (id != (u32)(unsigned long)ct) {
981 nf_ct_put(ct);
982 return -ENOENT;
986 if (del_timer(&ct->timeout)) {
987 if (nf_conntrack_event_report(IPCT_DESTROY, ct,
988 NETLINK_CB(skb).portid,
989 nlmsg_report(nlh)) < 0) {
990 nf_ct_delete_from_lists(ct);
991 /* we failed to report the event, try later */
992 nf_ct_insert_dying_list(ct);
993 nf_ct_put(ct);
994 return 0;
996 /* death_by_timeout would report the event again */
997 set_bit(IPS_DYING_BIT, &ct->status);
998 nf_ct_delete_from_lists(ct);
999 nf_ct_put(ct);
1001 nf_ct_put(ct);
1003 return 0;
1006 static int
1007 ctnetlink_get_conntrack(struct sock *ctnl, struct sk_buff *skb,
1008 const struct nlmsghdr *nlh,
1009 const struct nlattr * const cda[])
1011 struct net *net = sock_net(ctnl);
1012 struct nf_conntrack_tuple_hash *h;
1013 struct nf_conntrack_tuple tuple;
1014 struct nf_conn *ct;
1015 struct sk_buff *skb2 = NULL;
1016 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
1017 u_int8_t u3 = nfmsg->nfgen_family;
1018 u16 zone;
1019 int err;
1021 if (nlh->nlmsg_flags & NLM_F_DUMP) {
1022 struct netlink_dump_control c = {
1023 .dump = ctnetlink_dump_table,
1024 .done = ctnetlink_done,
1026 #ifdef CONFIG_NF_CONNTRACK_MARK
1027 if (cda[CTA_MARK] && cda[CTA_MARK_MASK]) {
1028 struct ctnetlink_dump_filter *filter;
1030 filter = kzalloc(sizeof(struct ctnetlink_dump_filter),
1031 GFP_ATOMIC);
1032 if (filter == NULL)
1033 return -ENOMEM;
1035 filter->mark.val = ntohl(nla_get_be32(cda[CTA_MARK]));
1036 filter->mark.mask =
1037 ntohl(nla_get_be32(cda[CTA_MARK_MASK]));
1038 c.data = filter;
1040 #endif
1041 return netlink_dump_start(ctnl, skb, nlh, &c);
1044 err = ctnetlink_parse_zone(cda[CTA_ZONE], &zone);
1045 if (err < 0)
1046 return err;
1048 if (cda[CTA_TUPLE_ORIG])
1049 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_ORIG, u3);
1050 else if (cda[CTA_TUPLE_REPLY])
1051 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_REPLY, u3);
1052 else
1053 return -EINVAL;
1055 if (err < 0)
1056 return err;
1058 h = nf_conntrack_find_get(net, zone, &tuple);
1059 if (!h)
1060 return -ENOENT;
1062 ct = nf_ct_tuplehash_to_ctrack(h);
1064 err = -ENOMEM;
1065 skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1066 if (skb2 == NULL) {
1067 nf_ct_put(ct);
1068 return -ENOMEM;
1071 rcu_read_lock();
1072 err = ctnetlink_fill_info(skb2, NETLINK_CB(skb).portid, nlh->nlmsg_seq,
1073 NFNL_MSG_TYPE(nlh->nlmsg_type), ct);
1074 rcu_read_unlock();
1075 nf_ct_put(ct);
1076 if (err <= 0)
1077 goto free;
1079 err = netlink_unicast(ctnl, skb2, NETLINK_CB(skb).portid, MSG_DONTWAIT);
1080 if (err < 0)
1081 goto out;
1083 return 0;
1085 free:
1086 kfree_skb(skb2);
1087 out:
1088 /* this avoids a loop in nfnetlink. */
1089 return err == -EAGAIN ? -ENOBUFS : err;
1092 #ifdef CONFIG_NF_NAT_NEEDED
1093 static int
1094 ctnetlink_parse_nat_setup(struct nf_conn *ct,
1095 enum nf_nat_manip_type manip,
1096 const struct nlattr *attr)
1098 typeof(nfnetlink_parse_nat_setup_hook) parse_nat_setup;
1099 int err;
1101 parse_nat_setup = rcu_dereference(nfnetlink_parse_nat_setup_hook);
1102 if (!parse_nat_setup) {
1103 #ifdef CONFIG_MODULES
1104 rcu_read_unlock();
1105 nfnl_unlock();
1106 if (request_module("nf-nat") < 0) {
1107 nfnl_lock();
1108 rcu_read_lock();
1109 return -EOPNOTSUPP;
1111 nfnl_lock();
1112 rcu_read_lock();
1113 if (nfnetlink_parse_nat_setup_hook)
1114 return -EAGAIN;
1115 #endif
1116 return -EOPNOTSUPP;
1119 err = parse_nat_setup(ct, manip, attr);
1120 if (err == -EAGAIN) {
1121 #ifdef CONFIG_MODULES
1122 rcu_read_unlock();
1123 nfnl_unlock();
1124 if (request_module("nf-nat-%u", nf_ct_l3num(ct)) < 0) {
1125 nfnl_lock();
1126 rcu_read_lock();
1127 return -EOPNOTSUPP;
1129 nfnl_lock();
1130 rcu_read_lock();
1131 #else
1132 err = -EOPNOTSUPP;
1133 #endif
1135 return err;
1137 #endif
1139 static int
1140 ctnetlink_change_status(struct nf_conn *ct, const struct nlattr * const cda[])
1142 unsigned long d;
1143 unsigned int status = ntohl(nla_get_be32(cda[CTA_STATUS]));
1144 d = ct->status ^ status;
1146 if (d & (IPS_EXPECTED|IPS_CONFIRMED|IPS_DYING))
1147 /* unchangeable */
1148 return -EBUSY;
1150 if (d & IPS_SEEN_REPLY && !(status & IPS_SEEN_REPLY))
1151 /* SEEN_REPLY bit can only be set */
1152 return -EBUSY;
1154 if (d & IPS_ASSURED && !(status & IPS_ASSURED))
1155 /* ASSURED bit can only be set */
1156 return -EBUSY;
1158 /* Be careful here, modifying NAT bits can screw up things,
1159 * so don't let users modify them directly if they don't pass
1160 * nf_nat_range. */
1161 ct->status |= status & ~(IPS_NAT_DONE_MASK | IPS_NAT_MASK);
1162 return 0;
1165 static int
1166 ctnetlink_change_nat(struct nf_conn *ct, const struct nlattr * const cda[])
1168 #ifdef CONFIG_NF_NAT_NEEDED
1169 int ret;
1171 if (cda[CTA_NAT_DST]) {
1172 ret = ctnetlink_parse_nat_setup(ct,
1173 NF_NAT_MANIP_DST,
1174 cda[CTA_NAT_DST]);
1175 if (ret < 0)
1176 return ret;
1178 if (cda[CTA_NAT_SRC]) {
1179 ret = ctnetlink_parse_nat_setup(ct,
1180 NF_NAT_MANIP_SRC,
1181 cda[CTA_NAT_SRC]);
1182 if (ret < 0)
1183 return ret;
1185 return 0;
1186 #else
1187 return -EOPNOTSUPP;
1188 #endif
1191 static inline int
1192 ctnetlink_change_helper(struct nf_conn *ct, const struct nlattr * const cda[])
1194 struct nf_conntrack_helper *helper;
1195 struct nf_conn_help *help = nfct_help(ct);
1196 char *helpname = NULL;
1197 struct nlattr *helpinfo = NULL;
1198 int err;
1200 /* don't change helper of sibling connections */
1201 if (ct->master)
1202 return -EBUSY;
1204 err = ctnetlink_parse_help(cda[CTA_HELP], &helpname, &helpinfo);
1205 if (err < 0)
1206 return err;
1208 if (!strcmp(helpname, "")) {
1209 if (help && help->helper) {
1210 /* we had a helper before ... */
1211 nf_ct_remove_expectations(ct);
1212 RCU_INIT_POINTER(help->helper, NULL);
1215 return 0;
1218 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
1219 nf_ct_protonum(ct));
1220 if (helper == NULL) {
1221 #ifdef CONFIG_MODULES
1222 spin_unlock_bh(&nf_conntrack_lock);
1224 if (request_module("nfct-helper-%s", helpname) < 0) {
1225 spin_lock_bh(&nf_conntrack_lock);
1226 return -EOPNOTSUPP;
1229 spin_lock_bh(&nf_conntrack_lock);
1230 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
1231 nf_ct_protonum(ct));
1232 if (helper)
1233 return -EAGAIN;
1234 #endif
1235 return -EOPNOTSUPP;
1238 if (help) {
1239 if (help->helper == helper) {
1240 /* update private helper data if allowed. */
1241 if (helper->from_nlattr)
1242 helper->from_nlattr(helpinfo, ct);
1243 return 0;
1244 } else
1245 return -EBUSY;
1248 /* we cannot set a helper for an existing conntrack */
1249 return -EOPNOTSUPP;
1252 static inline int
1253 ctnetlink_change_timeout(struct nf_conn *ct, const struct nlattr * const cda[])
1255 u_int32_t timeout = ntohl(nla_get_be32(cda[CTA_TIMEOUT]));
1257 if (!del_timer(&ct->timeout))
1258 return -ETIME;
1260 ct->timeout.expires = jiffies + timeout * HZ;
1261 add_timer(&ct->timeout);
1263 return 0;
1266 static const struct nla_policy protoinfo_policy[CTA_PROTOINFO_MAX+1] = {
1267 [CTA_PROTOINFO_TCP] = { .type = NLA_NESTED },
1268 [CTA_PROTOINFO_DCCP] = { .type = NLA_NESTED },
1269 [CTA_PROTOINFO_SCTP] = { .type = NLA_NESTED },
1272 static inline int
1273 ctnetlink_change_protoinfo(struct nf_conn *ct, const struct nlattr * const cda[])
1275 const struct nlattr *attr = cda[CTA_PROTOINFO];
1276 struct nlattr *tb[CTA_PROTOINFO_MAX+1];
1277 struct nf_conntrack_l4proto *l4proto;
1278 int err = 0;
1280 nla_parse_nested(tb, CTA_PROTOINFO_MAX, attr, protoinfo_policy);
1282 rcu_read_lock();
1283 l4proto = __nf_ct_l4proto_find(nf_ct_l3num(ct), nf_ct_protonum(ct));
1284 if (l4proto->from_nlattr)
1285 err = l4proto->from_nlattr(tb, ct);
1286 rcu_read_unlock();
1288 return err;
1291 #ifdef CONFIG_NF_NAT_NEEDED
1292 static const struct nla_policy nat_seq_policy[CTA_NAT_SEQ_MAX+1] = {
1293 [CTA_NAT_SEQ_CORRECTION_POS] = { .type = NLA_U32 },
1294 [CTA_NAT_SEQ_OFFSET_BEFORE] = { .type = NLA_U32 },
1295 [CTA_NAT_SEQ_OFFSET_AFTER] = { .type = NLA_U32 },
1298 static inline int
1299 change_nat_seq_adj(struct nf_nat_seq *natseq, const struct nlattr * const attr)
1301 struct nlattr *cda[CTA_NAT_SEQ_MAX+1];
1303 nla_parse_nested(cda, CTA_NAT_SEQ_MAX, attr, nat_seq_policy);
1305 if (!cda[CTA_NAT_SEQ_CORRECTION_POS])
1306 return -EINVAL;
1308 natseq->correction_pos =
1309 ntohl(nla_get_be32(cda[CTA_NAT_SEQ_CORRECTION_POS]));
1311 if (!cda[CTA_NAT_SEQ_OFFSET_BEFORE])
1312 return -EINVAL;
1314 natseq->offset_before =
1315 ntohl(nla_get_be32(cda[CTA_NAT_SEQ_OFFSET_BEFORE]));
1317 if (!cda[CTA_NAT_SEQ_OFFSET_AFTER])
1318 return -EINVAL;
1320 natseq->offset_after =
1321 ntohl(nla_get_be32(cda[CTA_NAT_SEQ_OFFSET_AFTER]));
1323 return 0;
1326 static int
1327 ctnetlink_change_nat_seq_adj(struct nf_conn *ct,
1328 const struct nlattr * const cda[])
1330 int ret = 0;
1331 struct nf_conn_nat *nat = nfct_nat(ct);
1333 if (!nat)
1334 return 0;
1336 if (cda[CTA_NAT_SEQ_ADJ_ORIG]) {
1337 ret = change_nat_seq_adj(&nat->seq[IP_CT_DIR_ORIGINAL],
1338 cda[CTA_NAT_SEQ_ADJ_ORIG]);
1339 if (ret < 0)
1340 return ret;
1342 ct->status |= IPS_SEQ_ADJUST;
1345 if (cda[CTA_NAT_SEQ_ADJ_REPLY]) {
1346 ret = change_nat_seq_adj(&nat->seq[IP_CT_DIR_REPLY],
1347 cda[CTA_NAT_SEQ_ADJ_REPLY]);
1348 if (ret < 0)
1349 return ret;
1351 ct->status |= IPS_SEQ_ADJUST;
1354 return 0;
1356 #endif
1358 static int
1359 ctnetlink_change_conntrack(struct nf_conn *ct,
1360 const struct nlattr * const cda[])
1362 int err;
1364 /* only allow NAT changes and master assignation for new conntracks */
1365 if (cda[CTA_NAT_SRC] || cda[CTA_NAT_DST] || cda[CTA_TUPLE_MASTER])
1366 return -EOPNOTSUPP;
1368 if (cda[CTA_HELP]) {
1369 err = ctnetlink_change_helper(ct, cda);
1370 if (err < 0)
1371 return err;
1374 if (cda[CTA_TIMEOUT]) {
1375 err = ctnetlink_change_timeout(ct, cda);
1376 if (err < 0)
1377 return err;
1380 if (cda[CTA_STATUS]) {
1381 err = ctnetlink_change_status(ct, cda);
1382 if (err < 0)
1383 return err;
1386 if (cda[CTA_PROTOINFO]) {
1387 err = ctnetlink_change_protoinfo(ct, cda);
1388 if (err < 0)
1389 return err;
1392 #if defined(CONFIG_NF_CONNTRACK_MARK)
1393 if (cda[CTA_MARK])
1394 ct->mark = ntohl(nla_get_be32(cda[CTA_MARK]));
1395 #endif
1397 #ifdef CONFIG_NF_NAT_NEEDED
1398 if (cda[CTA_NAT_SEQ_ADJ_ORIG] || cda[CTA_NAT_SEQ_ADJ_REPLY]) {
1399 err = ctnetlink_change_nat_seq_adj(ct, cda);
1400 if (err < 0)
1401 return err;
1403 #endif
1405 return 0;
1408 static struct nf_conn *
1409 ctnetlink_create_conntrack(struct net *net, u16 zone,
1410 const struct nlattr * const cda[],
1411 struct nf_conntrack_tuple *otuple,
1412 struct nf_conntrack_tuple *rtuple,
1413 u8 u3)
1415 struct nf_conn *ct;
1416 int err = -EINVAL;
1417 struct nf_conntrack_helper *helper;
1418 struct nf_conn_tstamp *tstamp;
1420 ct = nf_conntrack_alloc(net, zone, otuple, rtuple, GFP_ATOMIC);
1421 if (IS_ERR(ct))
1422 return ERR_PTR(-ENOMEM);
1424 if (!cda[CTA_TIMEOUT])
1425 goto err1;
1426 ct->timeout.expires = ntohl(nla_get_be32(cda[CTA_TIMEOUT]));
1428 ct->timeout.expires = jiffies + ct->timeout.expires * HZ;
1430 rcu_read_lock();
1431 if (cda[CTA_HELP]) {
1432 char *helpname = NULL;
1433 struct nlattr *helpinfo = NULL;
1435 err = ctnetlink_parse_help(cda[CTA_HELP], &helpname, &helpinfo);
1436 if (err < 0)
1437 goto err2;
1439 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
1440 nf_ct_protonum(ct));
1441 if (helper == NULL) {
1442 rcu_read_unlock();
1443 #ifdef CONFIG_MODULES
1444 if (request_module("nfct-helper-%s", helpname) < 0) {
1445 err = -EOPNOTSUPP;
1446 goto err1;
1449 rcu_read_lock();
1450 helper = __nf_conntrack_helper_find(helpname,
1451 nf_ct_l3num(ct),
1452 nf_ct_protonum(ct));
1453 if (helper) {
1454 err = -EAGAIN;
1455 goto err2;
1457 rcu_read_unlock();
1458 #endif
1459 err = -EOPNOTSUPP;
1460 goto err1;
1461 } else {
1462 struct nf_conn_help *help;
1464 help = nf_ct_helper_ext_add(ct, helper, GFP_ATOMIC);
1465 if (help == NULL) {
1466 err = -ENOMEM;
1467 goto err2;
1469 /* set private helper data if allowed. */
1470 if (helper->from_nlattr)
1471 helper->from_nlattr(helpinfo, ct);
1473 /* not in hash table yet so not strictly necessary */
1474 RCU_INIT_POINTER(help->helper, helper);
1476 } else {
1477 /* try an implicit helper assignation */
1478 err = __nf_ct_try_assign_helper(ct, NULL, GFP_ATOMIC);
1479 if (err < 0)
1480 goto err2;
1483 if (cda[CTA_NAT_SRC] || cda[CTA_NAT_DST]) {
1484 err = ctnetlink_change_nat(ct, cda);
1485 if (err < 0)
1486 goto err2;
1489 nf_ct_acct_ext_add(ct, GFP_ATOMIC);
1490 nf_ct_tstamp_ext_add(ct, GFP_ATOMIC);
1491 nf_ct_ecache_ext_add(ct, 0, 0, GFP_ATOMIC);
1492 /* we must add conntrack extensions before confirmation. */
1493 ct->status |= IPS_CONFIRMED;
1495 if (cda[CTA_STATUS]) {
1496 err = ctnetlink_change_status(ct, cda);
1497 if (err < 0)
1498 goto err2;
1501 #ifdef CONFIG_NF_NAT_NEEDED
1502 if (cda[CTA_NAT_SEQ_ADJ_ORIG] || cda[CTA_NAT_SEQ_ADJ_REPLY]) {
1503 err = ctnetlink_change_nat_seq_adj(ct, cda);
1504 if (err < 0)
1505 goto err2;
1507 #endif
1509 memset(&ct->proto, 0, sizeof(ct->proto));
1510 if (cda[CTA_PROTOINFO]) {
1511 err = ctnetlink_change_protoinfo(ct, cda);
1512 if (err < 0)
1513 goto err2;
1516 #if defined(CONFIG_NF_CONNTRACK_MARK)
1517 if (cda[CTA_MARK])
1518 ct->mark = ntohl(nla_get_be32(cda[CTA_MARK]));
1519 #endif
1521 /* setup master conntrack: this is a confirmed expectation */
1522 if (cda[CTA_TUPLE_MASTER]) {
1523 struct nf_conntrack_tuple master;
1524 struct nf_conntrack_tuple_hash *master_h;
1525 struct nf_conn *master_ct;
1527 err = ctnetlink_parse_tuple(cda, &master, CTA_TUPLE_MASTER, u3);
1528 if (err < 0)
1529 goto err2;
1531 master_h = nf_conntrack_find_get(net, zone, &master);
1532 if (master_h == NULL) {
1533 err = -ENOENT;
1534 goto err2;
1536 master_ct = nf_ct_tuplehash_to_ctrack(master_h);
1537 __set_bit(IPS_EXPECTED_BIT, &ct->status);
1538 ct->master = master_ct;
1540 tstamp = nf_conn_tstamp_find(ct);
1541 if (tstamp)
1542 tstamp->start = ktime_to_ns(ktime_get_real());
1544 err = nf_conntrack_hash_check_insert(ct);
1545 if (err < 0)
1546 goto err2;
1548 rcu_read_unlock();
1550 return ct;
1552 err2:
1553 rcu_read_unlock();
1554 err1:
1555 nf_conntrack_free(ct);
1556 return ERR_PTR(err);
1559 static int
1560 ctnetlink_new_conntrack(struct sock *ctnl, struct sk_buff *skb,
1561 const struct nlmsghdr *nlh,
1562 const struct nlattr * const cda[])
1564 struct net *net = sock_net(ctnl);
1565 struct nf_conntrack_tuple otuple, rtuple;
1566 struct nf_conntrack_tuple_hash *h = NULL;
1567 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
1568 struct nf_conn *ct;
1569 u_int8_t u3 = nfmsg->nfgen_family;
1570 u16 zone;
1571 int err;
1573 err = ctnetlink_parse_zone(cda[CTA_ZONE], &zone);
1574 if (err < 0)
1575 return err;
1577 if (cda[CTA_TUPLE_ORIG]) {
1578 err = ctnetlink_parse_tuple(cda, &otuple, CTA_TUPLE_ORIG, u3);
1579 if (err < 0)
1580 return err;
1583 if (cda[CTA_TUPLE_REPLY]) {
1584 err = ctnetlink_parse_tuple(cda, &rtuple, CTA_TUPLE_REPLY, u3);
1585 if (err < 0)
1586 return err;
1589 if (cda[CTA_TUPLE_ORIG])
1590 h = nf_conntrack_find_get(net, zone, &otuple);
1591 else if (cda[CTA_TUPLE_REPLY])
1592 h = nf_conntrack_find_get(net, zone, &rtuple);
1594 if (h == NULL) {
1595 err = -ENOENT;
1596 if (nlh->nlmsg_flags & NLM_F_CREATE) {
1597 enum ip_conntrack_events events;
1599 ct = ctnetlink_create_conntrack(net, zone, cda, &otuple,
1600 &rtuple, u3);
1601 if (IS_ERR(ct))
1602 return PTR_ERR(ct);
1604 err = 0;
1605 if (test_bit(IPS_EXPECTED_BIT, &ct->status))
1606 events = IPCT_RELATED;
1607 else
1608 events = IPCT_NEW;
1610 nf_conntrack_eventmask_report((1 << IPCT_REPLY) |
1611 (1 << IPCT_ASSURED) |
1612 (1 << IPCT_HELPER) |
1613 (1 << IPCT_PROTOINFO) |
1614 (1 << IPCT_NATSEQADJ) |
1615 (1 << IPCT_MARK) | events,
1616 ct, NETLINK_CB(skb).portid,
1617 nlmsg_report(nlh));
1618 nf_ct_put(ct);
1621 return err;
1623 /* implicit 'else' */
1625 err = -EEXIST;
1626 ct = nf_ct_tuplehash_to_ctrack(h);
1627 if (!(nlh->nlmsg_flags & NLM_F_EXCL)) {
1628 spin_lock_bh(&nf_conntrack_lock);
1629 err = ctnetlink_change_conntrack(ct, cda);
1630 spin_unlock_bh(&nf_conntrack_lock);
1631 if (err == 0) {
1632 nf_conntrack_eventmask_report((1 << IPCT_REPLY) |
1633 (1 << IPCT_ASSURED) |
1634 (1 << IPCT_HELPER) |
1635 (1 << IPCT_PROTOINFO) |
1636 (1 << IPCT_NATSEQADJ) |
1637 (1 << IPCT_MARK),
1638 ct, NETLINK_CB(skb).portid,
1639 nlmsg_report(nlh));
1643 nf_ct_put(ct);
1644 return err;
1647 static int
1648 ctnetlink_ct_stat_cpu_fill_info(struct sk_buff *skb, u32 portid, u32 seq,
1649 __u16 cpu, const struct ip_conntrack_stat *st)
1651 struct nlmsghdr *nlh;
1652 struct nfgenmsg *nfmsg;
1653 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
1655 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_CT_GET_STATS_CPU);
1656 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
1657 if (nlh == NULL)
1658 goto nlmsg_failure;
1660 nfmsg = nlmsg_data(nlh);
1661 nfmsg->nfgen_family = AF_UNSPEC;
1662 nfmsg->version = NFNETLINK_V0;
1663 nfmsg->res_id = htons(cpu);
1665 if (nla_put_be32(skb, CTA_STATS_SEARCHED, htonl(st->searched)) ||
1666 nla_put_be32(skb, CTA_STATS_FOUND, htonl(st->found)) ||
1667 nla_put_be32(skb, CTA_STATS_NEW, htonl(st->new)) ||
1668 nla_put_be32(skb, CTA_STATS_INVALID, htonl(st->invalid)) ||
1669 nla_put_be32(skb, CTA_STATS_IGNORE, htonl(st->ignore)) ||
1670 nla_put_be32(skb, CTA_STATS_DELETE, htonl(st->delete)) ||
1671 nla_put_be32(skb, CTA_STATS_DELETE_LIST, htonl(st->delete_list)) ||
1672 nla_put_be32(skb, CTA_STATS_INSERT, htonl(st->insert)) ||
1673 nla_put_be32(skb, CTA_STATS_INSERT_FAILED,
1674 htonl(st->insert_failed)) ||
1675 nla_put_be32(skb, CTA_STATS_DROP, htonl(st->drop)) ||
1676 nla_put_be32(skb, CTA_STATS_EARLY_DROP, htonl(st->early_drop)) ||
1677 nla_put_be32(skb, CTA_STATS_ERROR, htonl(st->error)) ||
1678 nla_put_be32(skb, CTA_STATS_SEARCH_RESTART,
1679 htonl(st->search_restart)))
1680 goto nla_put_failure;
1682 nlmsg_end(skb, nlh);
1683 return skb->len;
1685 nla_put_failure:
1686 nlmsg_failure:
1687 nlmsg_cancel(skb, nlh);
1688 return -1;
1691 static int
1692 ctnetlink_ct_stat_cpu_dump(struct sk_buff *skb, struct netlink_callback *cb)
1694 int cpu;
1695 struct net *net = sock_net(skb->sk);
1697 if (cb->args[0] == nr_cpu_ids)
1698 return 0;
1700 for (cpu = cb->args[0]; cpu < nr_cpu_ids; cpu++) {
1701 const struct ip_conntrack_stat *st;
1703 if (!cpu_possible(cpu))
1704 continue;
1706 st = per_cpu_ptr(net->ct.stat, cpu);
1707 if (ctnetlink_ct_stat_cpu_fill_info(skb,
1708 NETLINK_CB(cb->skb).portid,
1709 cb->nlh->nlmsg_seq,
1710 cpu, st) < 0)
1711 break;
1713 cb->args[0] = cpu;
1715 return skb->len;
1718 static int
1719 ctnetlink_stat_ct_cpu(struct sock *ctnl, struct sk_buff *skb,
1720 const struct nlmsghdr *nlh,
1721 const struct nlattr * const cda[])
1723 if (nlh->nlmsg_flags & NLM_F_DUMP) {
1724 struct netlink_dump_control c = {
1725 .dump = ctnetlink_ct_stat_cpu_dump,
1727 return netlink_dump_start(ctnl, skb, nlh, &c);
1730 return 0;
1733 static int
1734 ctnetlink_stat_ct_fill_info(struct sk_buff *skb, u32 portid, u32 seq, u32 type,
1735 struct net *net)
1737 struct nlmsghdr *nlh;
1738 struct nfgenmsg *nfmsg;
1739 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
1740 unsigned int nr_conntracks = atomic_read(&net->ct.count);
1742 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_CT_GET_STATS);
1743 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
1744 if (nlh == NULL)
1745 goto nlmsg_failure;
1747 nfmsg = nlmsg_data(nlh);
1748 nfmsg->nfgen_family = AF_UNSPEC;
1749 nfmsg->version = NFNETLINK_V0;
1750 nfmsg->res_id = 0;
1752 if (nla_put_be32(skb, CTA_STATS_GLOBAL_ENTRIES, htonl(nr_conntracks)))
1753 goto nla_put_failure;
1755 nlmsg_end(skb, nlh);
1756 return skb->len;
1758 nla_put_failure:
1759 nlmsg_failure:
1760 nlmsg_cancel(skb, nlh);
1761 return -1;
1764 static int
1765 ctnetlink_stat_ct(struct sock *ctnl, struct sk_buff *skb,
1766 const struct nlmsghdr *nlh,
1767 const struct nlattr * const cda[])
1769 struct sk_buff *skb2;
1770 int err;
1772 skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1773 if (skb2 == NULL)
1774 return -ENOMEM;
1776 err = ctnetlink_stat_ct_fill_info(skb2, NETLINK_CB(skb).portid,
1777 nlh->nlmsg_seq,
1778 NFNL_MSG_TYPE(nlh->nlmsg_type),
1779 sock_net(skb->sk));
1780 if (err <= 0)
1781 goto free;
1783 err = netlink_unicast(ctnl, skb2, NETLINK_CB(skb).portid, MSG_DONTWAIT);
1784 if (err < 0)
1785 goto out;
1787 return 0;
1789 free:
1790 kfree_skb(skb2);
1791 out:
1792 /* this avoids a loop in nfnetlink. */
1793 return err == -EAGAIN ? -ENOBUFS : err;
1796 #ifdef CONFIG_NETFILTER_NETLINK_QUEUE_CT
1797 static size_t
1798 ctnetlink_nfqueue_build_size(const struct nf_conn *ct)
1800 return 3 * nla_total_size(0) /* CTA_TUPLE_ORIG|REPL|MASTER */
1801 + 3 * nla_total_size(0) /* CTA_TUPLE_IP */
1802 + 3 * nla_total_size(0) /* CTA_TUPLE_PROTO */
1803 + 3 * nla_total_size(sizeof(u_int8_t)) /* CTA_PROTO_NUM */
1804 + nla_total_size(sizeof(u_int32_t)) /* CTA_ID */
1805 + nla_total_size(sizeof(u_int32_t)) /* CTA_STATUS */
1806 + nla_total_size(sizeof(u_int32_t)) /* CTA_TIMEOUT */
1807 + nla_total_size(0) /* CTA_PROTOINFO */
1808 + nla_total_size(0) /* CTA_HELP */
1809 + nla_total_size(NF_CT_HELPER_NAME_LEN) /* CTA_HELP_NAME */
1810 + ctnetlink_secctx_size(ct)
1811 #ifdef CONFIG_NF_NAT_NEEDED
1812 + 2 * nla_total_size(0) /* CTA_NAT_SEQ_ADJ_ORIG|REPL */
1813 + 6 * nla_total_size(sizeof(u_int32_t)) /* CTA_NAT_SEQ_OFFSET */
1814 #endif
1815 #ifdef CONFIG_NF_CONNTRACK_MARK
1816 + nla_total_size(sizeof(u_int32_t)) /* CTA_MARK */
1817 #endif
1818 + ctnetlink_proto_size(ct)
1822 static int
1823 ctnetlink_nfqueue_build(struct sk_buff *skb, struct nf_conn *ct)
1825 struct nlattr *nest_parms;
1827 rcu_read_lock();
1828 nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED);
1829 if (!nest_parms)
1830 goto nla_put_failure;
1831 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_ORIGINAL)) < 0)
1832 goto nla_put_failure;
1833 nla_nest_end(skb, nest_parms);
1835 nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED);
1836 if (!nest_parms)
1837 goto nla_put_failure;
1838 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_REPLY)) < 0)
1839 goto nla_put_failure;
1840 nla_nest_end(skb, nest_parms);
1842 if (nf_ct_zone(ct)) {
1843 if (nla_put_be16(skb, CTA_ZONE, htons(nf_ct_zone(ct))))
1844 goto nla_put_failure;
1847 if (ctnetlink_dump_id(skb, ct) < 0)
1848 goto nla_put_failure;
1850 if (ctnetlink_dump_status(skb, ct) < 0)
1851 goto nla_put_failure;
1853 if (ctnetlink_dump_timeout(skb, ct) < 0)
1854 goto nla_put_failure;
1856 if (ctnetlink_dump_protoinfo(skb, ct) < 0)
1857 goto nla_put_failure;
1859 if (ctnetlink_dump_helpinfo(skb, ct) < 0)
1860 goto nla_put_failure;
1862 #ifdef CONFIG_NF_CONNTRACK_SECMARK
1863 if (ct->secmark && ctnetlink_dump_secctx(skb, ct) < 0)
1864 goto nla_put_failure;
1865 #endif
1866 if (ct->master && ctnetlink_dump_master(skb, ct) < 0)
1867 goto nla_put_failure;
1869 if ((ct->status & IPS_SEQ_ADJUST) &&
1870 ctnetlink_dump_nat_seq_adj(skb, ct) < 0)
1871 goto nla_put_failure;
1873 #ifdef CONFIG_NF_CONNTRACK_MARK
1874 if (ct->mark && ctnetlink_dump_mark(skb, ct) < 0)
1875 goto nla_put_failure;
1876 #endif
1877 rcu_read_unlock();
1878 return 0;
1880 nla_put_failure:
1881 rcu_read_unlock();
1882 return -ENOSPC;
1885 static int
1886 ctnetlink_nfqueue_parse_ct(const struct nlattr *cda[], struct nf_conn *ct)
1888 int err;
1890 if (cda[CTA_TIMEOUT]) {
1891 err = ctnetlink_change_timeout(ct, cda);
1892 if (err < 0)
1893 return err;
1895 if (cda[CTA_STATUS]) {
1896 err = ctnetlink_change_status(ct, cda);
1897 if (err < 0)
1898 return err;
1900 if (cda[CTA_HELP]) {
1901 err = ctnetlink_change_helper(ct, cda);
1902 if (err < 0)
1903 return err;
1905 #if defined(CONFIG_NF_CONNTRACK_MARK)
1906 if (cda[CTA_MARK])
1907 ct->mark = ntohl(nla_get_be32(cda[CTA_MARK]));
1908 #endif
1909 return 0;
1912 static int
1913 ctnetlink_nfqueue_parse(const struct nlattr *attr, struct nf_conn *ct)
1915 struct nlattr *cda[CTA_MAX+1];
1916 int ret;
1918 nla_parse_nested(cda, CTA_MAX, attr, ct_nla_policy);
1920 spin_lock_bh(&nf_conntrack_lock);
1921 ret = ctnetlink_nfqueue_parse_ct((const struct nlattr **)cda, ct);
1922 spin_unlock_bh(&nf_conntrack_lock);
1924 return ret;
1927 static struct nfq_ct_hook ctnetlink_nfqueue_hook = {
1928 .build_size = ctnetlink_nfqueue_build_size,
1929 .build = ctnetlink_nfqueue_build,
1930 .parse = ctnetlink_nfqueue_parse,
1932 #endif /* CONFIG_NETFILTER_NETLINK_QUEUE_CT */
1934 /***********************************************************************
1935 * EXPECT
1936 ***********************************************************************/
1938 static inline int
1939 ctnetlink_exp_dump_tuple(struct sk_buff *skb,
1940 const struct nf_conntrack_tuple *tuple,
1941 enum ctattr_expect type)
1943 struct nlattr *nest_parms;
1945 nest_parms = nla_nest_start(skb, type | NLA_F_NESTED);
1946 if (!nest_parms)
1947 goto nla_put_failure;
1948 if (ctnetlink_dump_tuples(skb, tuple) < 0)
1949 goto nla_put_failure;
1950 nla_nest_end(skb, nest_parms);
1952 return 0;
1954 nla_put_failure:
1955 return -1;
1958 static inline int
1959 ctnetlink_exp_dump_mask(struct sk_buff *skb,
1960 const struct nf_conntrack_tuple *tuple,
1961 const struct nf_conntrack_tuple_mask *mask)
1963 int ret;
1964 struct nf_conntrack_l3proto *l3proto;
1965 struct nf_conntrack_l4proto *l4proto;
1966 struct nf_conntrack_tuple m;
1967 struct nlattr *nest_parms;
1969 memset(&m, 0xFF, sizeof(m));
1970 memcpy(&m.src.u3, &mask->src.u3, sizeof(m.src.u3));
1971 m.src.u.all = mask->src.u.all;
1972 m.dst.protonum = tuple->dst.protonum;
1974 nest_parms = nla_nest_start(skb, CTA_EXPECT_MASK | NLA_F_NESTED);
1975 if (!nest_parms)
1976 goto nla_put_failure;
1978 rcu_read_lock();
1979 l3proto = __nf_ct_l3proto_find(tuple->src.l3num);
1980 ret = ctnetlink_dump_tuples_ip(skb, &m, l3proto);
1981 if (ret >= 0) {
1982 l4proto = __nf_ct_l4proto_find(tuple->src.l3num,
1983 tuple->dst.protonum);
1984 ret = ctnetlink_dump_tuples_proto(skb, &m, l4proto);
1986 rcu_read_unlock();
1988 if (unlikely(ret < 0))
1989 goto nla_put_failure;
1991 nla_nest_end(skb, nest_parms);
1993 return 0;
1995 nla_put_failure:
1996 return -1;
1999 static const union nf_inet_addr any_addr;
2001 static int
2002 ctnetlink_exp_dump_expect(struct sk_buff *skb,
2003 const struct nf_conntrack_expect *exp)
2005 struct nf_conn *master = exp->master;
2006 long timeout = ((long)exp->timeout.expires - (long)jiffies) / HZ;
2007 struct nf_conn_help *help;
2008 #ifdef CONFIG_NF_NAT_NEEDED
2009 struct nlattr *nest_parms;
2010 struct nf_conntrack_tuple nat_tuple = {};
2011 #endif
2012 struct nf_ct_helper_expectfn *expfn;
2014 if (timeout < 0)
2015 timeout = 0;
2017 if (ctnetlink_exp_dump_tuple(skb, &exp->tuple, CTA_EXPECT_TUPLE) < 0)
2018 goto nla_put_failure;
2019 if (ctnetlink_exp_dump_mask(skb, &exp->tuple, &exp->mask) < 0)
2020 goto nla_put_failure;
2021 if (ctnetlink_exp_dump_tuple(skb,
2022 &master->tuplehash[IP_CT_DIR_ORIGINAL].tuple,
2023 CTA_EXPECT_MASTER) < 0)
2024 goto nla_put_failure;
2026 #ifdef CONFIG_NF_NAT_NEEDED
2027 if (!nf_inet_addr_cmp(&exp->saved_addr, &any_addr) ||
2028 exp->saved_proto.all) {
2029 nest_parms = nla_nest_start(skb, CTA_EXPECT_NAT | NLA_F_NESTED);
2030 if (!nest_parms)
2031 goto nla_put_failure;
2033 if (nla_put_be32(skb, CTA_EXPECT_NAT_DIR, htonl(exp->dir)))
2034 goto nla_put_failure;
2036 nat_tuple.src.l3num = nf_ct_l3num(master);
2037 nat_tuple.src.u3 = exp->saved_addr;
2038 nat_tuple.dst.protonum = nf_ct_protonum(master);
2039 nat_tuple.src.u = exp->saved_proto;
2041 if (ctnetlink_exp_dump_tuple(skb, &nat_tuple,
2042 CTA_EXPECT_NAT_TUPLE) < 0)
2043 goto nla_put_failure;
2044 nla_nest_end(skb, nest_parms);
2046 #endif
2047 if (nla_put_be32(skb, CTA_EXPECT_TIMEOUT, htonl(timeout)) ||
2048 nla_put_be32(skb, CTA_EXPECT_ID, htonl((unsigned long)exp)) ||
2049 nla_put_be32(skb, CTA_EXPECT_FLAGS, htonl(exp->flags)) ||
2050 nla_put_be32(skb, CTA_EXPECT_CLASS, htonl(exp->class)))
2051 goto nla_put_failure;
2052 help = nfct_help(master);
2053 if (help) {
2054 struct nf_conntrack_helper *helper;
2056 helper = rcu_dereference(help->helper);
2057 if (helper &&
2058 nla_put_string(skb, CTA_EXPECT_HELP_NAME, helper->name))
2059 goto nla_put_failure;
2061 expfn = nf_ct_helper_expectfn_find_by_symbol(exp->expectfn);
2062 if (expfn != NULL &&
2063 nla_put_string(skb, CTA_EXPECT_FN, expfn->name))
2064 goto nla_put_failure;
2066 return 0;
2068 nla_put_failure:
2069 return -1;
2072 static int
2073 ctnetlink_exp_fill_info(struct sk_buff *skb, u32 portid, u32 seq,
2074 int event, const struct nf_conntrack_expect *exp)
2076 struct nlmsghdr *nlh;
2077 struct nfgenmsg *nfmsg;
2078 unsigned int flags = portid ? NLM_F_MULTI : 0;
2080 event |= NFNL_SUBSYS_CTNETLINK_EXP << 8;
2081 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
2082 if (nlh == NULL)
2083 goto nlmsg_failure;
2085 nfmsg = nlmsg_data(nlh);
2086 nfmsg->nfgen_family = exp->tuple.src.l3num;
2087 nfmsg->version = NFNETLINK_V0;
2088 nfmsg->res_id = 0;
2090 if (ctnetlink_exp_dump_expect(skb, exp) < 0)
2091 goto nla_put_failure;
2093 nlmsg_end(skb, nlh);
2094 return skb->len;
2096 nlmsg_failure:
2097 nla_put_failure:
2098 nlmsg_cancel(skb, nlh);
2099 return -1;
2102 #ifdef CONFIG_NF_CONNTRACK_EVENTS
2103 static int
2104 ctnetlink_expect_event(unsigned int events, struct nf_exp_event *item)
2106 struct nf_conntrack_expect *exp = item->exp;
2107 struct net *net = nf_ct_exp_net(exp);
2108 struct nlmsghdr *nlh;
2109 struct nfgenmsg *nfmsg;
2110 struct sk_buff *skb;
2111 unsigned int type, group;
2112 int flags = 0;
2114 if (events & (1 << IPEXP_DESTROY)) {
2115 type = IPCTNL_MSG_EXP_DELETE;
2116 group = NFNLGRP_CONNTRACK_EXP_DESTROY;
2117 } else if (events & (1 << IPEXP_NEW)) {
2118 type = IPCTNL_MSG_EXP_NEW;
2119 flags = NLM_F_CREATE|NLM_F_EXCL;
2120 group = NFNLGRP_CONNTRACK_EXP_NEW;
2121 } else
2122 return 0;
2124 if (!item->report && !nfnetlink_has_listeners(net, group))
2125 return 0;
2127 skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
2128 if (skb == NULL)
2129 goto errout;
2131 type |= NFNL_SUBSYS_CTNETLINK_EXP << 8;
2132 nlh = nlmsg_put(skb, item->portid, 0, type, sizeof(*nfmsg), flags);
2133 if (nlh == NULL)
2134 goto nlmsg_failure;
2136 nfmsg = nlmsg_data(nlh);
2137 nfmsg->nfgen_family = exp->tuple.src.l3num;
2138 nfmsg->version = NFNETLINK_V0;
2139 nfmsg->res_id = 0;
2141 rcu_read_lock();
2142 if (ctnetlink_exp_dump_expect(skb, exp) < 0)
2143 goto nla_put_failure;
2144 rcu_read_unlock();
2146 nlmsg_end(skb, nlh);
2147 nfnetlink_send(skb, net, item->portid, group, item->report, GFP_ATOMIC);
2148 return 0;
2150 nla_put_failure:
2151 rcu_read_unlock();
2152 nlmsg_cancel(skb, nlh);
2153 nlmsg_failure:
2154 kfree_skb(skb);
2155 errout:
2156 nfnetlink_set_err(net, 0, 0, -ENOBUFS);
2157 return 0;
2159 #endif
2160 static int ctnetlink_exp_done(struct netlink_callback *cb)
2162 if (cb->args[1])
2163 nf_ct_expect_put((struct nf_conntrack_expect *)cb->args[1]);
2164 return 0;
2167 static int
2168 ctnetlink_exp_dump_table(struct sk_buff *skb, struct netlink_callback *cb)
2170 struct net *net = sock_net(skb->sk);
2171 struct nf_conntrack_expect *exp, *last;
2172 struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
2173 struct hlist_node *n;
2174 u_int8_t l3proto = nfmsg->nfgen_family;
2176 rcu_read_lock();
2177 last = (struct nf_conntrack_expect *)cb->args[1];
2178 for (; cb->args[0] < nf_ct_expect_hsize; cb->args[0]++) {
2179 restart:
2180 hlist_for_each_entry(exp, n, &net->ct.expect_hash[cb->args[0]],
2181 hnode) {
2182 if (l3proto && exp->tuple.src.l3num != l3proto)
2183 continue;
2184 if (cb->args[1]) {
2185 if (exp != last)
2186 continue;
2187 cb->args[1] = 0;
2189 if (ctnetlink_exp_fill_info(skb,
2190 NETLINK_CB(cb->skb).portid,
2191 cb->nlh->nlmsg_seq,
2192 IPCTNL_MSG_EXP_NEW,
2193 exp) < 0) {
2194 if (!atomic_inc_not_zero(&exp->use))
2195 continue;
2196 cb->args[1] = (unsigned long)exp;
2197 goto out;
2200 if (cb->args[1]) {
2201 cb->args[1] = 0;
2202 goto restart;
2205 out:
2206 rcu_read_unlock();
2207 if (last)
2208 nf_ct_expect_put(last);
2210 return skb->len;
2213 static const struct nla_policy exp_nla_policy[CTA_EXPECT_MAX+1] = {
2214 [CTA_EXPECT_MASTER] = { .type = NLA_NESTED },
2215 [CTA_EXPECT_TUPLE] = { .type = NLA_NESTED },
2216 [CTA_EXPECT_MASK] = { .type = NLA_NESTED },
2217 [CTA_EXPECT_TIMEOUT] = { .type = NLA_U32 },
2218 [CTA_EXPECT_ID] = { .type = NLA_U32 },
2219 [CTA_EXPECT_HELP_NAME] = { .type = NLA_NUL_STRING },
2220 [CTA_EXPECT_ZONE] = { .type = NLA_U16 },
2221 [CTA_EXPECT_FLAGS] = { .type = NLA_U32 },
2222 [CTA_EXPECT_CLASS] = { .type = NLA_U32 },
2223 [CTA_EXPECT_NAT] = { .type = NLA_NESTED },
2224 [CTA_EXPECT_FN] = { .type = NLA_NUL_STRING },
2227 static int
2228 ctnetlink_get_expect(struct sock *ctnl, struct sk_buff *skb,
2229 const struct nlmsghdr *nlh,
2230 const struct nlattr * const cda[])
2232 struct net *net = sock_net(ctnl);
2233 struct nf_conntrack_tuple tuple;
2234 struct nf_conntrack_expect *exp;
2235 struct sk_buff *skb2;
2236 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2237 u_int8_t u3 = nfmsg->nfgen_family;
2238 u16 zone;
2239 int err;
2241 if (nlh->nlmsg_flags & NLM_F_DUMP) {
2242 struct netlink_dump_control c = {
2243 .dump = ctnetlink_exp_dump_table,
2244 .done = ctnetlink_exp_done,
2246 return netlink_dump_start(ctnl, skb, nlh, &c);
2249 err = ctnetlink_parse_zone(cda[CTA_EXPECT_ZONE], &zone);
2250 if (err < 0)
2251 return err;
2253 if (cda[CTA_EXPECT_TUPLE])
2254 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3);
2255 else if (cda[CTA_EXPECT_MASTER])
2256 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_MASTER, u3);
2257 else
2258 return -EINVAL;
2260 if (err < 0)
2261 return err;
2263 exp = nf_ct_expect_find_get(net, zone, &tuple);
2264 if (!exp)
2265 return -ENOENT;
2267 if (cda[CTA_EXPECT_ID]) {
2268 __be32 id = nla_get_be32(cda[CTA_EXPECT_ID]);
2269 if (ntohl(id) != (u32)(unsigned long)exp) {
2270 nf_ct_expect_put(exp);
2271 return -ENOENT;
2275 err = -ENOMEM;
2276 skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2277 if (skb2 == NULL) {
2278 nf_ct_expect_put(exp);
2279 goto out;
2282 rcu_read_lock();
2283 err = ctnetlink_exp_fill_info(skb2, NETLINK_CB(skb).portid,
2284 nlh->nlmsg_seq, IPCTNL_MSG_EXP_NEW, exp);
2285 rcu_read_unlock();
2286 nf_ct_expect_put(exp);
2287 if (err <= 0)
2288 goto free;
2290 err = netlink_unicast(ctnl, skb2, NETLINK_CB(skb).portid, MSG_DONTWAIT);
2291 if (err < 0)
2292 goto out;
2294 return 0;
2296 free:
2297 kfree_skb(skb2);
2298 out:
2299 /* this avoids a loop in nfnetlink. */
2300 return err == -EAGAIN ? -ENOBUFS : err;
2303 static int
2304 ctnetlink_del_expect(struct sock *ctnl, struct sk_buff *skb,
2305 const struct nlmsghdr *nlh,
2306 const struct nlattr * const cda[])
2308 struct net *net = sock_net(ctnl);
2309 struct nf_conntrack_expect *exp;
2310 struct nf_conntrack_tuple tuple;
2311 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2312 struct hlist_node *n, *next;
2313 u_int8_t u3 = nfmsg->nfgen_family;
2314 unsigned int i;
2315 u16 zone;
2316 int err;
2318 if (cda[CTA_EXPECT_TUPLE]) {
2319 /* delete a single expect by tuple */
2320 err = ctnetlink_parse_zone(cda[CTA_EXPECT_ZONE], &zone);
2321 if (err < 0)
2322 return err;
2324 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3);
2325 if (err < 0)
2326 return err;
2328 /* bump usage count to 2 */
2329 exp = nf_ct_expect_find_get(net, zone, &tuple);
2330 if (!exp)
2331 return -ENOENT;
2333 if (cda[CTA_EXPECT_ID]) {
2334 __be32 id = nla_get_be32(cda[CTA_EXPECT_ID]);
2335 if (ntohl(id) != (u32)(unsigned long)exp) {
2336 nf_ct_expect_put(exp);
2337 return -ENOENT;
2341 /* after list removal, usage count == 1 */
2342 spin_lock_bh(&nf_conntrack_lock);
2343 if (del_timer(&exp->timeout)) {
2344 nf_ct_unlink_expect_report(exp, NETLINK_CB(skb).portid,
2345 nlmsg_report(nlh));
2346 nf_ct_expect_put(exp);
2348 spin_unlock_bh(&nf_conntrack_lock);
2349 /* have to put what we 'get' above.
2350 * after this line usage count == 0 */
2351 nf_ct_expect_put(exp);
2352 } else if (cda[CTA_EXPECT_HELP_NAME]) {
2353 char *name = nla_data(cda[CTA_EXPECT_HELP_NAME]);
2354 struct nf_conn_help *m_help;
2356 /* delete all expectations for this helper */
2357 spin_lock_bh(&nf_conntrack_lock);
2358 for (i = 0; i < nf_ct_expect_hsize; i++) {
2359 hlist_for_each_entry_safe(exp, n, next,
2360 &net->ct.expect_hash[i],
2361 hnode) {
2362 m_help = nfct_help(exp->master);
2363 if (!strcmp(m_help->helper->name, name) &&
2364 del_timer(&exp->timeout)) {
2365 nf_ct_unlink_expect_report(exp,
2366 NETLINK_CB(skb).portid,
2367 nlmsg_report(nlh));
2368 nf_ct_expect_put(exp);
2372 spin_unlock_bh(&nf_conntrack_lock);
2373 } else {
2374 /* This basically means we have to flush everything*/
2375 spin_lock_bh(&nf_conntrack_lock);
2376 for (i = 0; i < nf_ct_expect_hsize; i++) {
2377 hlist_for_each_entry_safe(exp, n, next,
2378 &net->ct.expect_hash[i],
2379 hnode) {
2380 if (del_timer(&exp->timeout)) {
2381 nf_ct_unlink_expect_report(exp,
2382 NETLINK_CB(skb).portid,
2383 nlmsg_report(nlh));
2384 nf_ct_expect_put(exp);
2388 spin_unlock_bh(&nf_conntrack_lock);
2391 return 0;
2393 static int
2394 ctnetlink_change_expect(struct nf_conntrack_expect *x,
2395 const struct nlattr * const cda[])
2397 if (cda[CTA_EXPECT_TIMEOUT]) {
2398 if (!del_timer(&x->timeout))
2399 return -ETIME;
2401 x->timeout.expires = jiffies +
2402 ntohl(nla_get_be32(cda[CTA_EXPECT_TIMEOUT])) * HZ;
2403 add_timer(&x->timeout);
2405 return 0;
2408 static const struct nla_policy exp_nat_nla_policy[CTA_EXPECT_NAT_MAX+1] = {
2409 [CTA_EXPECT_NAT_DIR] = { .type = NLA_U32 },
2410 [CTA_EXPECT_NAT_TUPLE] = { .type = NLA_NESTED },
2413 static int
2414 ctnetlink_parse_expect_nat(const struct nlattr *attr,
2415 struct nf_conntrack_expect *exp,
2416 u_int8_t u3)
2418 #ifdef CONFIG_NF_NAT_NEEDED
2419 struct nlattr *tb[CTA_EXPECT_NAT_MAX+1];
2420 struct nf_conntrack_tuple nat_tuple = {};
2421 int err;
2423 nla_parse_nested(tb, CTA_EXPECT_NAT_MAX, attr, exp_nat_nla_policy);
2425 if (!tb[CTA_EXPECT_NAT_DIR] || !tb[CTA_EXPECT_NAT_TUPLE])
2426 return -EINVAL;
2428 err = ctnetlink_parse_tuple((const struct nlattr * const *)tb,
2429 &nat_tuple, CTA_EXPECT_NAT_TUPLE, u3);
2430 if (err < 0)
2431 return err;
2433 exp->saved_addr = nat_tuple.src.u3;
2434 exp->saved_proto = nat_tuple.src.u;
2435 exp->dir = ntohl(nla_get_be32(tb[CTA_EXPECT_NAT_DIR]));
2437 return 0;
2438 #else
2439 return -EOPNOTSUPP;
2440 #endif
2443 static int
2444 ctnetlink_create_expect(struct net *net, u16 zone,
2445 const struct nlattr * const cda[],
2446 u_int8_t u3,
2447 u32 portid, int report)
2449 struct nf_conntrack_tuple tuple, mask, master_tuple;
2450 struct nf_conntrack_tuple_hash *h = NULL;
2451 struct nf_conntrack_expect *exp;
2452 struct nf_conn *ct;
2453 struct nf_conn_help *help;
2454 struct nf_conntrack_helper *helper = NULL;
2455 u_int32_t class = 0;
2456 int err = 0;
2458 /* caller guarantees that those three CTA_EXPECT_* exist */
2459 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3);
2460 if (err < 0)
2461 return err;
2462 err = ctnetlink_parse_tuple(cda, &mask, CTA_EXPECT_MASK, u3);
2463 if (err < 0)
2464 return err;
2465 err = ctnetlink_parse_tuple(cda, &master_tuple, CTA_EXPECT_MASTER, u3);
2466 if (err < 0)
2467 return err;
2469 /* Look for master conntrack of this expectation */
2470 h = nf_conntrack_find_get(net, zone, &master_tuple);
2471 if (!h)
2472 return -ENOENT;
2473 ct = nf_ct_tuplehash_to_ctrack(h);
2475 /* Look for helper of this expectation */
2476 if (cda[CTA_EXPECT_HELP_NAME]) {
2477 const char *helpname = nla_data(cda[CTA_EXPECT_HELP_NAME]);
2479 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
2480 nf_ct_protonum(ct));
2481 if (helper == NULL) {
2482 #ifdef CONFIG_MODULES
2483 if (request_module("nfct-helper-%s", helpname) < 0) {
2484 err = -EOPNOTSUPP;
2485 goto out;
2488 helper = __nf_conntrack_helper_find(helpname,
2489 nf_ct_l3num(ct),
2490 nf_ct_protonum(ct));
2491 if (helper) {
2492 err = -EAGAIN;
2493 goto out;
2495 #endif
2496 err = -EOPNOTSUPP;
2497 goto out;
2501 if (cda[CTA_EXPECT_CLASS] && helper) {
2502 class = ntohl(nla_get_be32(cda[CTA_EXPECT_CLASS]));
2503 if (class > helper->expect_class_max) {
2504 err = -EINVAL;
2505 goto out;
2508 exp = nf_ct_expect_alloc(ct);
2509 if (!exp) {
2510 err = -ENOMEM;
2511 goto out;
2513 help = nfct_help(ct);
2514 if (!help) {
2515 if (!cda[CTA_EXPECT_TIMEOUT]) {
2516 err = -EINVAL;
2517 goto out;
2519 exp->timeout.expires =
2520 jiffies + ntohl(nla_get_be32(cda[CTA_EXPECT_TIMEOUT])) * HZ;
2522 exp->flags = NF_CT_EXPECT_USERSPACE;
2523 if (cda[CTA_EXPECT_FLAGS]) {
2524 exp->flags |=
2525 ntohl(nla_get_be32(cda[CTA_EXPECT_FLAGS]));
2527 } else {
2528 if (cda[CTA_EXPECT_FLAGS]) {
2529 exp->flags = ntohl(nla_get_be32(cda[CTA_EXPECT_FLAGS]));
2530 exp->flags &= ~NF_CT_EXPECT_USERSPACE;
2531 } else
2532 exp->flags = 0;
2534 if (cda[CTA_EXPECT_FN]) {
2535 const char *name = nla_data(cda[CTA_EXPECT_FN]);
2536 struct nf_ct_helper_expectfn *expfn;
2538 expfn = nf_ct_helper_expectfn_find_by_name(name);
2539 if (expfn == NULL) {
2540 err = -EINVAL;
2541 goto err_out;
2543 exp->expectfn = expfn->expectfn;
2544 } else
2545 exp->expectfn = NULL;
2547 exp->class = class;
2548 exp->master = ct;
2549 exp->helper = helper;
2550 memcpy(&exp->tuple, &tuple, sizeof(struct nf_conntrack_tuple));
2551 memcpy(&exp->mask.src.u3, &mask.src.u3, sizeof(exp->mask.src.u3));
2552 exp->mask.src.u.all = mask.src.u.all;
2554 if (cda[CTA_EXPECT_NAT]) {
2555 err = ctnetlink_parse_expect_nat(cda[CTA_EXPECT_NAT],
2556 exp, u3);
2557 if (err < 0)
2558 goto err_out;
2560 err = nf_ct_expect_related_report(exp, portid, report);
2561 err_out:
2562 nf_ct_expect_put(exp);
2563 out:
2564 nf_ct_put(nf_ct_tuplehash_to_ctrack(h));
2565 return err;
2568 static int
2569 ctnetlink_new_expect(struct sock *ctnl, struct sk_buff *skb,
2570 const struct nlmsghdr *nlh,
2571 const struct nlattr * const cda[])
2573 struct net *net = sock_net(ctnl);
2574 struct nf_conntrack_tuple tuple;
2575 struct nf_conntrack_expect *exp;
2576 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2577 u_int8_t u3 = nfmsg->nfgen_family;
2578 u16 zone;
2579 int err;
2581 if (!cda[CTA_EXPECT_TUPLE]
2582 || !cda[CTA_EXPECT_MASK]
2583 || !cda[CTA_EXPECT_MASTER])
2584 return -EINVAL;
2586 err = ctnetlink_parse_zone(cda[CTA_EXPECT_ZONE], &zone);
2587 if (err < 0)
2588 return err;
2590 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE, u3);
2591 if (err < 0)
2592 return err;
2594 spin_lock_bh(&nf_conntrack_lock);
2595 exp = __nf_ct_expect_find(net, zone, &tuple);
2597 if (!exp) {
2598 spin_unlock_bh(&nf_conntrack_lock);
2599 err = -ENOENT;
2600 if (nlh->nlmsg_flags & NLM_F_CREATE) {
2601 err = ctnetlink_create_expect(net, zone, cda,
2603 NETLINK_CB(skb).portid,
2604 nlmsg_report(nlh));
2606 return err;
2609 err = -EEXIST;
2610 if (!(nlh->nlmsg_flags & NLM_F_EXCL))
2611 err = ctnetlink_change_expect(exp, cda);
2612 spin_unlock_bh(&nf_conntrack_lock);
2614 return err;
2617 static int
2618 ctnetlink_exp_stat_fill_info(struct sk_buff *skb, u32 portid, u32 seq, int cpu,
2619 const struct ip_conntrack_stat *st)
2621 struct nlmsghdr *nlh;
2622 struct nfgenmsg *nfmsg;
2623 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
2625 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_EXP_GET_STATS_CPU);
2626 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
2627 if (nlh == NULL)
2628 goto nlmsg_failure;
2630 nfmsg = nlmsg_data(nlh);
2631 nfmsg->nfgen_family = AF_UNSPEC;
2632 nfmsg->version = NFNETLINK_V0;
2633 nfmsg->res_id = htons(cpu);
2635 if (nla_put_be32(skb, CTA_STATS_EXP_NEW, htonl(st->expect_new)) ||
2636 nla_put_be32(skb, CTA_STATS_EXP_CREATE, htonl(st->expect_create)) ||
2637 nla_put_be32(skb, CTA_STATS_EXP_DELETE, htonl(st->expect_delete)))
2638 goto nla_put_failure;
2640 nlmsg_end(skb, nlh);
2641 return skb->len;
2643 nla_put_failure:
2644 nlmsg_failure:
2645 nlmsg_cancel(skb, nlh);
2646 return -1;
2649 static int
2650 ctnetlink_exp_stat_cpu_dump(struct sk_buff *skb, struct netlink_callback *cb)
2652 int cpu;
2653 struct net *net = sock_net(skb->sk);
2655 if (cb->args[0] == nr_cpu_ids)
2656 return 0;
2658 for (cpu = cb->args[0]; cpu < nr_cpu_ids; cpu++) {
2659 const struct ip_conntrack_stat *st;
2661 if (!cpu_possible(cpu))
2662 continue;
2664 st = per_cpu_ptr(net->ct.stat, cpu);
2665 if (ctnetlink_exp_stat_fill_info(skb, NETLINK_CB(cb->skb).portid,
2666 cb->nlh->nlmsg_seq,
2667 cpu, st) < 0)
2668 break;
2670 cb->args[0] = cpu;
2672 return skb->len;
2675 static int
2676 ctnetlink_stat_exp_cpu(struct sock *ctnl, struct sk_buff *skb,
2677 const struct nlmsghdr *nlh,
2678 const struct nlattr * const cda[])
2680 if (nlh->nlmsg_flags & NLM_F_DUMP) {
2681 struct netlink_dump_control c = {
2682 .dump = ctnetlink_exp_stat_cpu_dump,
2684 return netlink_dump_start(ctnl, skb, nlh, &c);
2687 return 0;
2690 #ifdef CONFIG_NF_CONNTRACK_EVENTS
2691 static struct nf_ct_event_notifier ctnl_notifier = {
2692 .fcn = ctnetlink_conntrack_event,
2695 static struct nf_exp_event_notifier ctnl_notifier_exp = {
2696 .fcn = ctnetlink_expect_event,
2698 #endif
2700 static const struct nfnl_callback ctnl_cb[IPCTNL_MSG_MAX] = {
2701 [IPCTNL_MSG_CT_NEW] = { .call = ctnetlink_new_conntrack,
2702 .attr_count = CTA_MAX,
2703 .policy = ct_nla_policy },
2704 [IPCTNL_MSG_CT_GET] = { .call = ctnetlink_get_conntrack,
2705 .attr_count = CTA_MAX,
2706 .policy = ct_nla_policy },
2707 [IPCTNL_MSG_CT_DELETE] = { .call = ctnetlink_del_conntrack,
2708 .attr_count = CTA_MAX,
2709 .policy = ct_nla_policy },
2710 [IPCTNL_MSG_CT_GET_CTRZERO] = { .call = ctnetlink_get_conntrack,
2711 .attr_count = CTA_MAX,
2712 .policy = ct_nla_policy },
2713 [IPCTNL_MSG_CT_GET_STATS_CPU] = { .call = ctnetlink_stat_ct_cpu },
2714 [IPCTNL_MSG_CT_GET_STATS] = { .call = ctnetlink_stat_ct },
2717 static const struct nfnl_callback ctnl_exp_cb[IPCTNL_MSG_EXP_MAX] = {
2718 [IPCTNL_MSG_EXP_GET] = { .call = ctnetlink_get_expect,
2719 .attr_count = CTA_EXPECT_MAX,
2720 .policy = exp_nla_policy },
2721 [IPCTNL_MSG_EXP_NEW] = { .call = ctnetlink_new_expect,
2722 .attr_count = CTA_EXPECT_MAX,
2723 .policy = exp_nla_policy },
2724 [IPCTNL_MSG_EXP_DELETE] = { .call = ctnetlink_del_expect,
2725 .attr_count = CTA_EXPECT_MAX,
2726 .policy = exp_nla_policy },
2727 [IPCTNL_MSG_EXP_GET_STATS_CPU] = { .call = ctnetlink_stat_exp_cpu },
2730 static const struct nfnetlink_subsystem ctnl_subsys = {
2731 .name = "conntrack",
2732 .subsys_id = NFNL_SUBSYS_CTNETLINK,
2733 .cb_count = IPCTNL_MSG_MAX,
2734 .cb = ctnl_cb,
2737 static const struct nfnetlink_subsystem ctnl_exp_subsys = {
2738 .name = "conntrack_expect",
2739 .subsys_id = NFNL_SUBSYS_CTNETLINK_EXP,
2740 .cb_count = IPCTNL_MSG_EXP_MAX,
2741 .cb = ctnl_exp_cb,
2744 MODULE_ALIAS("ip_conntrack_netlink");
2745 MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK);
2746 MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK_EXP);
2748 static int __net_init ctnetlink_net_init(struct net *net)
2750 #ifdef CONFIG_NF_CONNTRACK_EVENTS
2751 int ret;
2753 ret = nf_conntrack_register_notifier(net, &ctnl_notifier);
2754 if (ret < 0) {
2755 pr_err("ctnetlink_init: cannot register notifier.\n");
2756 goto err_out;
2759 ret = nf_ct_expect_register_notifier(net, &ctnl_notifier_exp);
2760 if (ret < 0) {
2761 pr_err("ctnetlink_init: cannot expect register notifier.\n");
2762 goto err_unreg_notifier;
2764 #endif
2765 return 0;
2767 #ifdef CONFIG_NF_CONNTRACK_EVENTS
2768 err_unreg_notifier:
2769 nf_conntrack_unregister_notifier(net, &ctnl_notifier);
2770 err_out:
2771 return ret;
2772 #endif
2775 static void ctnetlink_net_exit(struct net *net)
2777 #ifdef CONFIG_NF_CONNTRACK_EVENTS
2778 nf_ct_expect_unregister_notifier(net, &ctnl_notifier_exp);
2779 nf_conntrack_unregister_notifier(net, &ctnl_notifier);
2780 #endif
2783 static void __net_exit ctnetlink_net_exit_batch(struct list_head *net_exit_list)
2785 struct net *net;
2787 list_for_each_entry(net, net_exit_list, exit_list)
2788 ctnetlink_net_exit(net);
2791 static struct pernet_operations ctnetlink_net_ops = {
2792 .init = ctnetlink_net_init,
2793 .exit_batch = ctnetlink_net_exit_batch,
2796 static int __init ctnetlink_init(void)
2798 int ret;
2800 pr_info("ctnetlink v%s: registering with nfnetlink.\n", version);
2801 ret = nfnetlink_subsys_register(&ctnl_subsys);
2802 if (ret < 0) {
2803 pr_err("ctnetlink_init: cannot register with nfnetlink.\n");
2804 goto err_out;
2807 ret = nfnetlink_subsys_register(&ctnl_exp_subsys);
2808 if (ret < 0) {
2809 pr_err("ctnetlink_init: cannot register exp with nfnetlink.\n");
2810 goto err_unreg_subsys;
2813 ret = register_pernet_subsys(&ctnetlink_net_ops);
2814 if (ret < 0) {
2815 pr_err("ctnetlink_init: cannot register pernet operations\n");
2816 goto err_unreg_exp_subsys;
2818 #ifdef CONFIG_NETFILTER_NETLINK_QUEUE_CT
2819 /* setup interaction between nf_queue and nf_conntrack_netlink. */
2820 RCU_INIT_POINTER(nfq_ct_hook, &ctnetlink_nfqueue_hook);
2821 #endif
2822 return 0;
2824 err_unreg_exp_subsys:
2825 nfnetlink_subsys_unregister(&ctnl_exp_subsys);
2826 err_unreg_subsys:
2827 nfnetlink_subsys_unregister(&ctnl_subsys);
2828 err_out:
2829 return ret;
2832 static void __exit ctnetlink_exit(void)
2834 pr_info("ctnetlink: unregistering from nfnetlink.\n");
2836 unregister_pernet_subsys(&ctnetlink_net_ops);
2837 nfnetlink_subsys_unregister(&ctnl_exp_subsys);
2838 nfnetlink_subsys_unregister(&ctnl_subsys);
2839 #ifdef CONFIG_NETFILTER_NETLINK_QUEUE_CT
2840 RCU_INIT_POINTER(nfq_ct_hook, NULL);
2841 #endif
2844 module_init(ctnetlink_init);
2845 module_exit(ctnetlink_exit);