projects / linux-2.6-block.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
netfilter: x_tables: introduce and use xt_copy_counters_from_user
[linux-2.6-block.git] / net / netfilter / nf_conntrack_netlink.c
CommitLineData
c1d10adb
PNA		 1/* Connection tracking via netlink socket. Allows for user space
2 * protocol helpers and general trouble making from userspace.
3 *
4 * (C) 2001 by Jay Schulist <jschlst@samba.org>
dc808fe2 5 * (C) 2002-2006 by Harald Welte <laforge@gnumonks.org>
c1d10adb 6 * (C) 2003 by Patrick Mchardy <kaber@trash.net>
392025f8 7 * (C) 2005-2012 by Pablo Neira Ayuso <pablo@netfilter.org>
c1d10adb 8 *
601e68e1 9 * Initial connection tracking via netlink development funded and
c1d10adb
PNA		 10 * generally made possible by Network Robots, Inc. (www.networkrobots.com)
11 *
12 * Further development of this code funded by Astaro AG (http://www.astaro.com)
13 *
14 * This software may be used and distributed according to the terms
15 * of the GNU General Public License, incorporated herein by reference.
c1d10adb
PNA		 16 */
17
18#include <linux/init.h>
19#include <linux/module.h>
20#include <linux/kernel.h>
711bbdd6 21#include <linux/rculist.h>
ea781f19 22#include <linux/rculist_nulls.h>
c1d10adb
PNA		 23#include <linux/types.h>
24#include <linux/timer.h>
1cc63249 25#include <linux/security.h>
c1d10adb
PNA		 26#include <linux/skbuff.h>
27#include <linux/errno.h>
28#include <linux/netlink.h>
29#include <linux/spinlock.h>
40a839fd 30#include <linux/interrupt.h>
5a0e3ad6 31#include <linux/slab.h>
c1d10adb
PNA		 32
33#include <linux/netfilter.h>
dc5fc579 34#include <net/netlink.h>
9592a5c0 35#include <net/sock.h>
c1d10adb
PNA		 36#include <net/netfilter/nf_conntrack.h>
37#include <net/netfilter/nf_conntrack_core.h>
77ab9cff 38#include <net/netfilter/nf_conntrack_expect.h>
c1d10adb 39#include <net/netfilter/nf_conntrack_helper.h>
41d73ec0 40#include <net/netfilter/nf_conntrack_seqadj.h>
c1d10adb 41#include <net/netfilter/nf_conntrack_l3proto.h>
605dcad6 42#include <net/netfilter/nf_conntrack_l4proto.h>
5b1158e9 43#include <net/netfilter/nf_conntrack_tuple.h>
58401572 44#include <net/netfilter/nf_conntrack_acct.h>
ef00f89f 45#include <net/netfilter/nf_conntrack_zones.h>
a992ca2a 46#include <net/netfilter/nf_conntrack_timestamp.h>
c539f017 47#include <net/netfilter/nf_conntrack_labels.h>
5b1158e9
JK		 48#ifdef CONFIG_NF_NAT_NEEDED
49#include <net/netfilter/nf_nat_core.h>
c7232c99 50#include <net/netfilter/nf_nat_l4proto.h>
8c88f87c 51#include <net/netfilter/nf_nat_helper.h>
5b1158e9 52#endif
c1d10adb
PNA		 53
54#include <linux/netfilter/nfnetlink.h>
55#include <linux/netfilter/nfnetlink_conntrack.h>
56
57MODULE_LICENSE("GPL");
58
dc808fe2 59static char __initdata version[] = "0.93";
c1d10adb 60
c1d10adb 61static inline int
601e68e1 62ctnetlink_dump_tuples_proto(struct sk_buff *skb,
1cde6436 63 const struct nf_conntrack_tuple *tuple,
605dcad6 64 struct nf_conntrack_l4proto *l4proto)
c1d10adb 65{
c1d10adb 66 int ret = 0;
df6fb868 67 struct nlattr *nest_parms;
c1d10adb 68
df6fb868
PM		 69 nest_parms = nla_nest_start(skb, CTA_TUPLE_PROTO | NLA_F_NESTED);
70 if (!nest_parms)
71 goto nla_put_failure;
cc1eb431
DM		 72 if (nla_put_u8(skb, CTA_PROTO_NUM, tuple->dst.protonum))
73 goto nla_put_failure;
c1d10adb 74
fdf70832
PM		 75 if (likely(l4proto->tuple_to_nlattr))
76 ret = l4proto->tuple_to_nlattr(skb, tuple);
601e68e1 77
df6fb868 78 nla_nest_end(skb, nest_parms);
c1d10adb
PNA		 79
80 return ret;
81
df6fb868 82nla_put_failure:
c1d10adb
PNA		 83 return -1;
84}
85
86static inline int
1cde6436
PNA		 87ctnetlink_dump_tuples_ip(struct sk_buff *skb,
88 const struct nf_conntrack_tuple *tuple,
89 struct nf_conntrack_l3proto *l3proto)
c1d10adb 90{
c1d10adb 91 int ret = 0;
df6fb868
PM		 92 struct nlattr *nest_parms;
93
94 nest_parms = nla_nest_start(skb, CTA_TUPLE_IP | NLA_F_NESTED);
95 if (!nest_parms)
96 goto nla_put_failure;
1cde6436 97
fdf70832
PM		 98 if (likely(l3proto->tuple_to_nlattr))
99 ret = l3proto->tuple_to_nlattr(skb, tuple);
1cde6436 100
df6fb868 101 nla_nest_end(skb, nest_parms);
c1d10adb 102
1cde6436
PNA		 103 return ret;
104
df6fb868 105nla_put_failure:
1cde6436
PNA		 106 return -1;
107}
108
bb5cf80e 109static int
1cde6436
PNA		 110ctnetlink_dump_tuples(struct sk_buff *skb,
111 const struct nf_conntrack_tuple *tuple)
112{
113 int ret;
114 struct nf_conntrack_l3proto *l3proto;
605dcad6 115 struct nf_conntrack_l4proto *l4proto;
1cde6436 116
3b988ece 117 rcu_read_lock();
528a3a6f 118 l3proto = __nf_ct_l3proto_find(tuple->src.l3num);
1cde6436 119 ret = ctnetlink_dump_tuples_ip(skb, tuple, l3proto);
c1d10adb 120
3b988ece
HS		 121 if (ret >= 0) {
122 l4proto = __nf_ct_l4proto_find(tuple->src.l3num,
123 tuple->dst.protonum);
124 ret = ctnetlink_dump_tuples_proto(skb, tuple, l4proto);
125 }
126 rcu_read_unlock();
c1d10adb 127 return ret;
c1d10adb
PNA		 128}
129
deedb590
DB		 130static inline int
131ctnetlink_dump_zone_id(struct sk_buff *skb, int attrtype,
132 const struct nf_conntrack_zone *zone, int dir)
133{
134 if (zone->id == NF_CT_DEFAULT_ZONE_ID || zone->dir != dir)
135 return 0;
136 if (nla_put_be16(skb, attrtype, htons(zone->id)))
137 goto nla_put_failure;
138 return 0;
139
140nla_put_failure:
141 return -1;
142}
143
c1d10adb
PNA		 144static inline int
145ctnetlink_dump_status(struct sk_buff *skb, const struct nf_conn *ct)
146{
cc1eb431
DM		 147 if (nla_put_be32(skb, CTA_STATUS, htonl(ct->status)))
148 goto nla_put_failure;
c1d10adb
PNA		 149 return 0;
150
df6fb868 151nla_put_failure:
c1d10adb
PNA		 152 return -1;
153}
154
155static inline int
156ctnetlink_dump_timeout(struct sk_buff *skb, const struct nf_conn *ct)
157{
c1216382 158 long timeout = ((long)ct->timeout.expires - (long)jiffies) / HZ;
c1d10adb 159
77236b6e 160 if (timeout < 0)
c1d10adb 161 timeout = 0;
601e68e1 162
cc1eb431
DM		 163 if (nla_put_be32(skb, CTA_TIMEOUT, htonl(timeout)))
164 goto nla_put_failure;
c1d10adb
PNA		 165 return 0;
166
df6fb868 167nla_put_failure:
c1d10adb
PNA		 168 return -1;
169}
170
171static inline int
440f0d58 172ctnetlink_dump_protoinfo(struct sk_buff *skb, struct nf_conn *ct)
c1d10adb 173{
5e8fbe2a 174 struct nf_conntrack_l4proto *l4proto;
df6fb868 175 struct nlattr *nest_proto;
c1d10adb
PNA		 176 int ret;
177
528a3a6f
PNA		 178 l4proto = __nf_ct_l4proto_find(nf_ct_l3num(ct), nf_ct_protonum(ct));
179 if (!l4proto->to_nlattr)
c1d10adb 180 return 0;
601e68e1 181
df6fb868
PM		 182 nest_proto = nla_nest_start(skb, CTA_PROTOINFO | NLA_F_NESTED);
183 if (!nest_proto)
184 goto nla_put_failure;
c1d10adb 185
fdf70832 186 ret = l4proto->to_nlattr(skb, nest_proto, ct);
c1d10adb 187
df6fb868 188 nla_nest_end(skb, nest_proto);
c1d10adb
PNA		 189
190 return ret;
191
df6fb868 192nla_put_failure:
c1d10adb
PNA		 193 return -1;
194}
195
196static inline int
197ctnetlink_dump_helpinfo(struct sk_buff *skb, const struct nf_conn *ct)
198{
df6fb868 199 struct nlattr *nest_helper;
dc808fe2 200 const struct nf_conn_help *help = nfct_help(ct);
3c158f7f 201 struct nf_conntrack_helper *helper;
c1d10adb 202
3c158f7f 203 if (!help)
c1d10adb 204 return 0;
601e68e1 205
3c158f7f
PM		 206 helper = rcu_dereference(help->helper);
207 if (!helper)
208 goto out;
209
df6fb868
PM		 210 nest_helper = nla_nest_start(skb, CTA_HELP | NLA_F_NESTED);
211 if (!nest_helper)
212 goto nla_put_failure;
cc1eb431
DM		 213 if (nla_put_string(skb, CTA_HELP_NAME, helper->name))
214 goto nla_put_failure;
c1d10adb 215
fdf70832
PM		 216 if (helper->to_nlattr)
217 helper->to_nlattr(skb, ct);
c1d10adb 218
df6fb868 219 nla_nest_end(skb, nest_helper);
3c158f7f 220out:
c1d10adb
PNA		 221 return 0;
222
df6fb868 223nla_put_failure:
c1d10adb
PNA		 224 return -1;
225}
226
bb5cf80e 227static int
4542fa47
HE		 228dump_counters(struct sk_buff *skb, struct nf_conn_acct *acct,
229 enum ip_conntrack_dir dir, int type)
c1d10adb 230{
4542fa47
HE		 231 enum ctattr_type attr = dir ? CTA_COUNTERS_REPLY: CTA_COUNTERS_ORIG;
232 struct nf_conn_counter *counter = acct->counter;
df6fb868 233 struct nlattr *nest_count;
4542fa47 234 u64 pkts, bytes;
c1d10adb 235
4542fa47
HE		 236 if (type == IPCTNL_MSG_CT_GET_CTRZERO) {
237 pkts = atomic64_xchg(&counter[dir].packets, 0);
238 bytes = atomic64_xchg(&counter[dir].bytes, 0);
239 } else {
240 pkts = atomic64_read(&counter[dir].packets);
241 bytes = atomic64_read(&counter[dir].bytes);
242 }
243
244 nest_count = nla_nest_start(skb, attr | NLA_F_NESTED);
df6fb868
PM		 245 if (!nest_count)
246 goto nla_put_failure;
247
cc1eb431
DM		 248 if (nla_put_be64(skb, CTA_COUNTERS_PACKETS, cpu_to_be64(pkts)) ||
249 nla_put_be64(skb, CTA_COUNTERS_BYTES, cpu_to_be64(bytes)))
250 goto nla_put_failure;
c1d10adb 251
df6fb868 252 nla_nest_end(skb, nest_count);
c1d10adb
PNA		 253
254 return 0;
255
df6fb868 256nla_put_failure:
c1d10adb
PNA		 257 return -1;
258}
c1d10adb 259
80e60e67 260static int
4542fa47 261ctnetlink_dump_acct(struct sk_buff *skb, const struct nf_conn *ct, int type)
80e60e67 262{
4542fa47 263 struct nf_conn_acct *acct = nf_conn_acct_find(ct);
80e60e67 264
80e60e67
PNA		 265 if (!acct)
266 return 0;
267
4542fa47
HE		 268 if (dump_counters(skb, acct, IP_CT_DIR_ORIGINAL, type) < 0)
269 return -1;
270 if (dump_counters(skb, acct, IP_CT_DIR_REPLY, type) < 0)
271 return -1;
272
273 return 0;
80e60e67
PNA		 274}
275
a992ca2a
PNA		 276static int
277ctnetlink_dump_timestamp(struct sk_buff *skb, const struct nf_conn *ct)
278{
279 struct nlattr *nest_count;
280 const struct nf_conn_tstamp *tstamp;
281
282 tstamp = nf_conn_tstamp_find(ct);
283 if (!tstamp)
284 return 0;
285
286 nest_count = nla_nest_start(skb, CTA_TIMESTAMP | NLA_F_NESTED);
287 if (!nest_count)
288 goto nla_put_failure;
289
cc1eb431
DM		 290 if (nla_put_be64(skb, CTA_TIMESTAMP_START, cpu_to_be64(tstamp->start)) ||
291 (tstamp->stop != 0 && nla_put_be64(skb, CTA_TIMESTAMP_STOP,
292 cpu_to_be64(tstamp->stop))))
293 goto nla_put_failure;
a992ca2a
PNA		 294 nla_nest_end(skb, nest_count);
295
296 return 0;
297
298nla_put_failure:
299 return -1;
300}
301
c1d10adb
PNA		 302#ifdef CONFIG_NF_CONNTRACK_MARK
303static inline int
304ctnetlink_dump_mark(struct sk_buff *skb, const struct nf_conn *ct)
305{
cc1eb431
DM		 306 if (nla_put_be32(skb, CTA_MARK, htonl(ct->mark)))
307 goto nla_put_failure;
c1d10adb
PNA		 308 return 0;
309
df6fb868 310nla_put_failure:
c1d10adb
PNA		 311 return -1;
312}
313#else
314#define ctnetlink_dump_mark(a, b) (0)
315#endif
316
37fccd85
PNA		 317#ifdef CONFIG_NF_CONNTRACK_SECMARK
318static inline int
1cc63249 319ctnetlink_dump_secctx(struct sk_buff *skb, const struct nf_conn *ct)
37fccd85 320{
1cc63249
EP		 321 struct nlattr *nest_secctx;
322 int len, ret;
323 char *secctx;
324
325 ret = security_secid_to_secctx(ct->secmark, &secctx, &len);
326 if (ret)
cba85b53 327 return 0;
1cc63249
EP		 328
329 ret = -1;
330 nest_secctx = nla_nest_start(skb, CTA_SECCTX | NLA_F_NESTED);
331 if (!nest_secctx)
332 goto nla_put_failure;
37fccd85 333
cc1eb431
DM		 334 if (nla_put_string(skb, CTA_SECCTX_NAME, secctx))
335 goto nla_put_failure;
1cc63249
EP		 336 nla_nest_end(skb, nest_secctx);
337
338 ret = 0;
37fccd85 339nla_put_failure:
1cc63249
EP		 340 security_release_secctx(secctx, len);
341 return ret;
37fccd85
PNA		 342}
343#else
1cc63249 344#define ctnetlink_dump_secctx(a, b) (0)
37fccd85
PNA		 345#endif
346
0ceabd83
FW		 347#ifdef CONFIG_NF_CONNTRACK_LABELS
348static int ctnetlink_label_size(const struct nf_conn *ct)
349{
350 struct nf_conn_labels *labels = nf_ct_labels_find(ct);
351
352 if (!labels)
353 return 0;
354 return nla_total_size(labels->words * sizeof(long));
355}
356
357static int
358ctnetlink_dump_labels(struct sk_buff *skb, const struct nf_conn *ct)
359{
360 struct nf_conn_labels *labels = nf_ct_labels_find(ct);
361 unsigned int len, i;
362
363 if (!labels)
364 return 0;
365
366 len = labels->words * sizeof(long);
367 i = 0;
368 do {
369 if (labels->bits[i] != 0)
370 return nla_put(skb, CTA_LABELS, len, labels->bits);
371 i++;
372 } while (i < labels->words);
373
374 return 0;
375}
376#else
377#define ctnetlink_dump_labels(a, b) (0)
378#define ctnetlink_label_size(a) (0)
379#endif
380
0f417ce9
PNA		 381#define master_tuple(ct) &(ct->master->tuplehash[IP_CT_DIR_ORIGINAL].tuple)
382
383static inline int
384ctnetlink_dump_master(struct sk_buff *skb, const struct nf_conn *ct)
385{
386 struct nlattr *nest_parms;
387
388 if (!(ct->status & IPS_EXPECTED))
389 return 0;
390
391 nest_parms = nla_nest_start(skb, CTA_TUPLE_MASTER | NLA_F_NESTED);
392 if (!nest_parms)
393 goto nla_put_failure;
394 if (ctnetlink_dump_tuples(skb, master_tuple(ct)) < 0)
395 goto nla_put_failure;
396 nla_nest_end(skb, nest_parms);
397
398 return 0;
399
400nla_put_failure:
401 return -1;
402}
403
bb5cf80e 404static int
41d73ec0 405dump_ct_seq_adj(struct sk_buff *skb, const struct nf_ct_seqadj *seq, int type)
13eae15a 406{
13eae15a
PNA		 407 struct nlattr *nest_parms;
408
409 nest_parms = nla_nest_start(skb, type | NLA_F_NESTED);
410 if (!nest_parms)
411 goto nla_put_failure;
412
41d73ec0
PM		 413 if (nla_put_be32(skb, CTA_SEQADJ_CORRECTION_POS,
414 htonl(seq->correction_pos)) ||
415 nla_put_be32(skb, CTA_SEQADJ_OFFSET_BEFORE,
416 htonl(seq->offset_before)) ||
417 nla_put_be32(skb, CTA_SEQADJ_OFFSET_AFTER,
418 htonl(seq->offset_after)))
cc1eb431 419 goto nla_put_failure;
13eae15a
PNA		 420
421 nla_nest_end(skb, nest_parms);
422
423 return 0;
424
425nla_put_failure:
426 return -1;
427}
428
429static inline int
41d73ec0 430ctnetlink_dump_ct_seq_adj(struct sk_buff *skb, const struct nf_conn *ct)
13eae15a 431{
41d73ec0
PM		 432 struct nf_conn_seqadj *seqadj = nfct_seqadj(ct);
433 struct nf_ct_seqadj *seq;
13eae15a 434
41d73ec0 435 if (!(ct->status & IPS_SEQ_ADJUST) || !seqadj)
13eae15a
PNA		 436 return 0;
437
41d73ec0
PM		 438 seq = &seqadj->seq[IP_CT_DIR_ORIGINAL];
439 if (dump_ct_seq_adj(skb, seq, CTA_SEQ_ADJ_ORIG) == -1)
13eae15a
PNA		 440 return -1;
441
41d73ec0
PM		 442 seq = &seqadj->seq[IP_CT_DIR_REPLY];
443 if (dump_ct_seq_adj(skb, seq, CTA_SEQ_ADJ_REPLY) == -1)
13eae15a
PNA		 444 return -1;
445
446 return 0;
447}
13eae15a 448
c1d10adb
PNA		 449static inline int
450ctnetlink_dump_id(struct sk_buff *skb, const struct nf_conn *ct)
451{
cc1eb431
DM		 452 if (nla_put_be32(skb, CTA_ID, htonl((unsigned long)ct)))
453 goto nla_put_failure;
c1d10adb
PNA		 454 return 0;
455
df6fb868 456nla_put_failure:
c1d10adb
PNA		 457 return -1;
458}
459
460static inline int
461ctnetlink_dump_use(struct sk_buff *skb, const struct nf_conn *ct)
462{
cc1eb431
DM		 463 if (nla_put_be32(skb, CTA_USE, htonl(atomic_read(&ct->ct_general.use))))
464 goto nla_put_failure;
c1d10adb
PNA		 465 return 0;
466
df6fb868 467nla_put_failure:
c1d10adb
PNA		 468 return -1;
469}
470
c1d10adb 471static int
15e47304 472ctnetlink_fill_info(struct sk_buff *skb, u32 portid, u32 seq, u32 type,
80e60e67 473 struct nf_conn *ct)
c1d10adb 474{
308ac914 475 const struct nf_conntrack_zone *zone;
c1d10adb
PNA		 476 struct nlmsghdr *nlh;
477 struct nfgenmsg *nfmsg;
df6fb868 478 struct nlattr *nest_parms;
15e47304 479 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
c1d10adb 480
80e60e67 481 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_CT_NEW);
15e47304 482 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
96bcf938
PNA		 483 if (nlh == NULL)
484 goto nlmsg_failure;
c1d10adb 485
96bcf938 486 nfmsg = nlmsg_data(nlh);
5e8fbe2a 487 nfmsg->nfgen_family = nf_ct_l3num(ct);
c1d10adb
PNA		 488 nfmsg->version = NFNETLINK_V0;
489 nfmsg->res_id = 0;
490
deedb590
DB		 491 zone = nf_ct_zone(ct);
492
df6fb868
PM		 493 nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED);
494 if (!nest_parms)
495 goto nla_put_failure;
f2f3e38c 496 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_ORIGINAL)) < 0)
df6fb868 497 goto nla_put_failure;
deedb590
DB		 498 if (ctnetlink_dump_zone_id(skb, CTA_TUPLE_ZONE, zone,
499 NF_CT_ZONE_DIR_ORIG) < 0)
500 goto nla_put_failure;
df6fb868 501 nla_nest_end(skb, nest_parms);
601e68e1 502
df6fb868
PM		 503 nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED);
504 if (!nest_parms)
505 goto nla_put_failure;
f2f3e38c 506 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_REPLY)) < 0)
df6fb868 507 goto nla_put_failure;
deedb590
DB		 508 if (ctnetlink_dump_zone_id(skb, CTA_TUPLE_ZONE, zone,
509 NF_CT_ZONE_DIR_REPL) < 0)
510 goto nla_put_failure;
df6fb868 511 nla_nest_end(skb, nest_parms);
c1d10adb 512
deedb590
DB		 513 if (ctnetlink_dump_zone_id(skb, CTA_ZONE, zone,
514 NF_CT_DEFAULT_ZONE_DIR) < 0)
cc1eb431 515 goto nla_put_failure;
ef00f89f 516
c1d10adb
PNA		 517 if (ctnetlink_dump_status(skb, ct) < 0 ||
518 ctnetlink_dump_timeout(skb, ct) < 0 ||
4542fa47 519 ctnetlink_dump_acct(skb, ct, type) < 0 ||
a992ca2a 520 ctnetlink_dump_timestamp(skb, ct) < 0 ||
c1d10adb
PNA		 521 ctnetlink_dump_protoinfo(skb, ct) < 0 ||
522 ctnetlink_dump_helpinfo(skb, ct) < 0 ||
523 ctnetlink_dump_mark(skb, ct) < 0 ||
1cc63249 524 ctnetlink_dump_secctx(skb, ct) < 0 ||
0ceabd83 525 ctnetlink_dump_labels(skb, ct) < 0 ||
c1d10adb 526 ctnetlink_dump_id(skb, ct) < 0 ||
13eae15a 527 ctnetlink_dump_use(skb, ct) < 0 ||
0f417ce9 528 ctnetlink_dump_master(skb, ct) < 0 ||
41d73ec0 529 ctnetlink_dump_ct_seq_adj(skb, ct) < 0)
df6fb868 530 goto nla_put_failure;
c1d10adb 531
96bcf938 532 nlmsg_end(skb, nlh);
c1d10adb
PNA		 533 return skb->len;
534
535nlmsg_failure:
df6fb868 536nla_put_failure:
96bcf938 537 nlmsg_cancel(skb, nlh);
c1d10adb
PNA		 538 return -1;
539}
540
03b64f51
PNA		 541static inline size_t
542ctnetlink_proto_size(const struct nf_conn *ct)
2732c4e4
HE		 543{
544 struct nf_conntrack_l3proto *l3proto;
545 struct nf_conntrack_l4proto *l4proto;
03b64f51
PNA		 546 size_t len = 0;
547
548 rcu_read_lock();
549 l3proto = __nf_ct_l3proto_find(nf_ct_l3num(ct));
550 len += l3proto->nla_size;
551
552 l4proto = __nf_ct_l4proto_find(nf_ct_l3num(ct), nf_ct_protonum(ct));
553 len += l4proto->nla_size;
554 rcu_read_unlock();
555
556 return len;
557}
558
d26e6a02 559static inline size_t
f7b13e43 560ctnetlink_acct_size(const struct nf_conn *ct)
d26e6a02
JP		 561{
562 if (!nf_ct_ext_exist(ct, NF_CT_EXT_ACCT))
563 return 0;
564 return 2 * nla_total_size(0) /* CTA_COUNTERS_ORIG|REPL */
565 + 2 * nla_total_size(sizeof(uint64_t)) /* CTA_COUNTERS_PACKETS */
566 + 2 * nla_total_size(sizeof(uint64_t)) /* CTA_COUNTERS_BYTES */
567 ;
568}
569
cba85b53
PNA		 570static inline int
571ctnetlink_secctx_size(const struct nf_conn *ct)
1cc63249 572{
cba85b53
PNA		 573#ifdef CONFIG_NF_CONNTRACK_SECMARK
574 int len, ret;
1cc63249 575
cba85b53
PNA		 576 ret = security_secid_to_secctx(ct->secmark, NULL, &len);
577 if (ret)
578 return 0;
1cc63249 579
cba85b53
PNA		 580 return nla_total_size(0) /* CTA_SECCTX */
581 + nla_total_size(sizeof(char) * len); /* CTA_SECCTX_NAME */
582#else
583 return 0;
1cc63249 584#endif
cba85b53 585}
1cc63249 586
a992ca2a
PNA		 587static inline size_t
588ctnetlink_timestamp_size(const struct nf_conn *ct)
589{
590#ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
591 if (!nf_ct_ext_exist(ct, NF_CT_EXT_TSTAMP))
592 return 0;
593 return nla_total_size(0) + 2 * nla_total_size(sizeof(uint64_t));
594#else
595 return 0;
596#endif
597}
598
03b64f51
PNA		 599static inline size_t
600ctnetlink_nlmsg_size(const struct nf_conn *ct)
601{
602 return NLMSG_ALIGN(sizeof(struct nfgenmsg))
603 + 3 * nla_total_size(0) /* CTA_TUPLE_ORIG|REPL|MASTER */
604 + 3 * nla_total_size(0) /* CTA_TUPLE_IP */
605 + 3 * nla_total_size(0) /* CTA_TUPLE_PROTO */
606 + 3 * nla_total_size(sizeof(u_int8_t)) /* CTA_PROTO_NUM */
607 + nla_total_size(sizeof(u_int32_t)) /* CTA_ID */
608 + nla_total_size(sizeof(u_int32_t)) /* CTA_STATUS */
f7b13e43 609 + ctnetlink_acct_size(ct)
a992ca2a 610 + ctnetlink_timestamp_size(ct)
03b64f51
PNA		 611 + nla_total_size(sizeof(u_int32_t)) /* CTA_TIMEOUT */
612 + nla_total_size(0) /* CTA_PROTOINFO */
613 + nla_total_size(0) /* CTA_HELP */
614 + nla_total_size(NF_CT_HELPER_NAME_LEN) /* CTA_HELP_NAME */
cba85b53 615 + ctnetlink_secctx_size(ct)
d271e8bd 616#ifdef CONFIG_NF_NAT_NEEDED
03b64f51
PNA		 617 + 2 * nla_total_size(0) /* CTA_NAT_SEQ_ADJ_ORIG|REPL */
618 + 6 * nla_total_size(sizeof(u_int32_t)) /* CTA_NAT_SEQ_OFFSET */
d271e8bd
HE		 619#endif
620#ifdef CONFIG_NF_CONNTRACK_MARK
03b64f51 621 + nla_total_size(sizeof(u_int32_t)) /* CTA_MARK */
4a001068
KM		 622#endif
623#ifdef CONFIG_NF_CONNTRACK_ZONES
deedb590 624 + nla_total_size(sizeof(u_int16_t)) /* CTA_ZONE|CTA_TUPLE_ZONE */
d271e8bd 625#endif
03b64f51 626 + ctnetlink_proto_size(ct)
0ceabd83 627 + ctnetlink_label_size(ct)
03b64f51 628 ;
2732c4e4
HE		 629}
630
8e36c4b5 631#ifdef CONFIG_NF_CONNTRACK_EVENTS
e34d5c1a
PNA		 632static int
633ctnetlink_conntrack_event(unsigned int events, struct nf_ct_event *item)
c1d10adb 634{
308ac914 635 const struct nf_conntrack_zone *zone;
9592a5c0 636 struct net *net;
c1d10adb
PNA		 637 struct nlmsghdr *nlh;
638 struct nfgenmsg *nfmsg;
df6fb868 639 struct nlattr *nest_parms;
19abb7b0 640 struct nf_conn *ct = item->ct;
c1d10adb
PNA		 641 struct sk_buff *skb;
642 unsigned int type;
c1d10adb 643 unsigned int flags = 0, group;
dd7669a9 644 int err;
c1d10adb
PNA		 645
646 /* ignore our fake conntrack entry */
5bfddbd4 647 if (nf_ct_is_untracked(ct))
e34d5c1a 648 return 0;
c1d10adb 649
a0891aa6 650 if (events & (1 << IPCT_DESTROY)) {
c1d10adb
PNA		 651 type = IPCTNL_MSG_CT_DELETE;
652 group = NFNLGRP_CONNTRACK_DESTROY;
a0891aa6 653 } else if (events & ((1 << IPCT_NEW) | (1 << IPCT_RELATED))) {
c1d10adb
PNA		 654 type = IPCTNL_MSG_CT_NEW;
655 flags = NLM_F_CREATE|NLM_F_EXCL;
c1d10adb 656 group = NFNLGRP_CONNTRACK_NEW;
17e6e4ea 657 } else if (events) {
c1d10adb
PNA		 658 type = IPCTNL_MSG_CT_NEW;
659 group = NFNLGRP_CONNTRACK_UPDATE;
660 } else
e34d5c1a 661 return 0;
a2427692 662
9592a5c0
AD		 663 net = nf_ct_net(ct);
664 if (!item->report && !nfnetlink_has_listeners(net, group))
e34d5c1a 665 return 0;
a2427692 666
03b64f51
PNA		 667 skb = nlmsg_new(ctnetlink_nlmsg_size(ct), GFP_ATOMIC);
668 if (skb == NULL)
150ace0d 669 goto errout;
c1d10adb 670
c1d10adb 671 type |= NFNL_SUBSYS_CTNETLINK << 8;
15e47304 672 nlh = nlmsg_put(skb, item->portid, 0, type, sizeof(*nfmsg), flags);
96bcf938
PNA		 673 if (nlh == NULL)
674 goto nlmsg_failure;
c1d10adb 675
96bcf938 676 nfmsg = nlmsg_data(nlh);
5e8fbe2a 677 nfmsg->nfgen_family = nf_ct_l3num(ct);
c1d10adb
PNA		 678 nfmsg->version = NFNETLINK_V0;
679 nfmsg->res_id = 0;
680
528a3a6f 681 rcu_read_lock();
deedb590
DB		 682 zone = nf_ct_zone(ct);
683
df6fb868
PM		 684 nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED);
685 if (!nest_parms)
686 goto nla_put_failure;
f2f3e38c 687 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_ORIGINAL)) < 0)
df6fb868 688 goto nla_put_failure;
deedb590
DB		 689 if (ctnetlink_dump_zone_id(skb, CTA_TUPLE_ZONE, zone,
690 NF_CT_ZONE_DIR_ORIG) < 0)
691 goto nla_put_failure;
df6fb868 692 nla_nest_end(skb, nest_parms);
601e68e1 693
df6fb868
PM		 694 nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED);
695 if (!nest_parms)
696 goto nla_put_failure;
f2f3e38c 697 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_REPLY)) < 0)
df6fb868 698 goto nla_put_failure;
deedb590
DB		 699 if (ctnetlink_dump_zone_id(skb, CTA_TUPLE_ZONE, zone,
700 NF_CT_ZONE_DIR_REPL) < 0)
701 goto nla_put_failure;
df6fb868 702 nla_nest_end(skb, nest_parms);
c1d10adb 703
deedb590
DB		 704 if (ctnetlink_dump_zone_id(skb, CTA_ZONE, zone,
705 NF_CT_DEFAULT_ZONE_DIR) < 0)
cc1eb431 706 goto nla_put_failure;
ef00f89f 707
1eedf699
EL		 708 if (ctnetlink_dump_id(skb, ct) < 0)
709 goto nla_put_failure;
710
e57dce60
FH		 711 if (ctnetlink_dump_status(skb, ct) < 0)
712 goto nla_put_failure;
713
a0891aa6 714 if (events & (1 << IPCT_DESTROY)) {
4542fa47 715 if (ctnetlink_dump_acct(skb, ct, type) < 0 ||
a992ca2a 716 ctnetlink_dump_timestamp(skb, ct) < 0)
df6fb868 717 goto nla_put_failure;
7b621c1e 718 } else {
7b621c1e 719 if (ctnetlink_dump_timeout(skb, ct) < 0)
df6fb868 720 goto nla_put_failure;
7b621c1e 721
a0891aa6 722 if (events & (1 << IPCT_PROTOINFO)
7b621c1e 723 && ctnetlink_dump_protoinfo(skb, ct) < 0)
df6fb868 724 goto nla_put_failure;
7b621c1e 725
a0891aa6 726 if ((events & (1 << IPCT_HELPER) || nfct_help(ct))
7b621c1e 727 && ctnetlink_dump_helpinfo(skb, ct) < 0)
df6fb868 728 goto nla_put_failure;
7b621c1e 729
ff660c80 730#ifdef CONFIG_NF_CONNTRACK_SECMARK
a0891aa6 731 if ((events & (1 << IPCT_SECMARK) || ct->secmark)
1cc63249 732 && ctnetlink_dump_secctx(skb, ct) < 0)
37fccd85 733 goto nla_put_failure;
ff660c80 734#endif
0ceabd83
FW		 735 if (events & (1 << IPCT_LABEL) &&
736 ctnetlink_dump_labels(skb, ct) < 0)
737 goto nla_put_failure;
7b621c1e 738
a0891aa6 739 if (events & (1 << IPCT_RELATED) &&
0f417ce9
PNA		 740 ctnetlink_dump_master(skb, ct) < 0)
741 goto nla_put_failure;
742
41d73ec0
PM		 743 if (events & (1 << IPCT_SEQADJ) &&
744 ctnetlink_dump_ct_seq_adj(skb, ct) < 0)
13eae15a 745 goto nla_put_failure;
7b621c1e 746 }
b9a37e0c 747
a83099a6 748#ifdef CONFIG_NF_CONNTRACK_MARK
a0891aa6 749 if ((events & (1 << IPCT_MARK) || ct->mark)
a83099a6
EL		 750 && ctnetlink_dump_mark(skb, ct) < 0)
751 goto nla_put_failure;
752#endif
528a3a6f 753 rcu_read_unlock();
a83099a6 754
96bcf938 755 nlmsg_end(skb, nlh);
15e47304 756 err = nfnetlink_send(skb, net, item->portid, group, item->report,
cd8c20b6 757 GFP_ATOMIC);
dd7669a9
PNA		 758 if (err == -ENOBUFS || err == -EAGAIN)
759 return -ENOBUFS;
760
e34d5c1a 761 return 0;
c1d10adb 762
df6fb868 763nla_put_failure:
528a3a6f 764 rcu_read_unlock();
96bcf938 765 nlmsg_cancel(skb, nlh);
528a3a6f 766nlmsg_failure:
c1d10adb 767 kfree_skb(skb);
150ace0d 768errout:
37b7ef72
PNA		 769 if (nfnetlink_set_err(net, 0, group, -ENOBUFS) > 0)
770 return -ENOBUFS;
771
e34d5c1a 772 return 0;
c1d10adb
PNA		 773}
774#endif /* CONFIG_NF_CONNTRACK_EVENTS */
775
776static int ctnetlink_done(struct netlink_callback *cb)
777{
89f2e218
PM		 778 if (cb->args[1])
779 nf_ct_put((struct nf_conn *)cb->args[1]);
397304b5 780 kfree(cb->data);
c1d10adb
PNA		 781 return 0;
782}
783
866476f3 784struct ctnetlink_filter {
0f298a28
PNA		 785 struct {
786 u_int32_t val;
787 u_int32_t mask;
788 } mark;
789};
790
866476f3
KE		 791static struct ctnetlink_filter *
792ctnetlink_alloc_filter(const struct nlattr * const cda[])
793{
794#ifdef CONFIG_NF_CONNTRACK_MARK
795 struct ctnetlink_filter *filter;
796
797 filter = kzalloc(sizeof(*filter), GFP_KERNEL);
798 if (filter == NULL)
799 return ERR_PTR(-ENOMEM);
800
801 filter->mark.val = ntohl(nla_get_be32(cda[CTA_MARK]));
802 filter->mark.mask = ntohl(nla_get_be32(cda[CTA_MARK_MASK]));
803
804 return filter;
805#else
806 return ERR_PTR(-EOPNOTSUPP);
807#endif
808}
809
810static int ctnetlink_filter_match(struct nf_conn *ct, void *data)
811{
812 struct ctnetlink_filter *filter = data;
813
814 if (filter == NULL)
815 return 1;
816
817#ifdef CONFIG_NF_CONNTRACK_MARK
818 if ((ct->mark & filter->mark.mask) == filter->mark.val)
819 return 1;
820#endif
821
822 return 0;
823}
824
c1d10adb
PNA		 825static int
826ctnetlink_dump_table(struct sk_buff *skb, struct netlink_callback *cb)
827{
9592a5c0 828 struct net *net = sock_net(skb->sk);
89f2e218 829 struct nf_conn *ct, *last;
c1d10adb 830 struct nf_conntrack_tuple_hash *h;
ea781f19 831 struct hlist_nulls_node *n;
96bcf938 832 struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
87711cb8 833 u_int8_t l3proto = nfmsg->nfgen_family;
3b988ece 834 int res;
93bb0ceb
JDB		 835 spinlock_t *lockp;
836
d205dc40 837 last = (struct nf_conn *)cb->args[1];
93bb0ceb
JDB		 838
839 local_bh_disable();
9ab99d5a 840 for (; cb->args[0] < net->ct.htable_size; cb->args[0]++) {
89f2e218 841restart:
93bb0ceb 842 lockp = &nf_conntrack_locks[cb->args[0] % CONNTRACK_LOCKS];
b16c2919 843 nf_conntrack_lock(lockp);
93bb0ceb
JDB		 844 if (cb->args[0] >= net->ct.htable_size) {
845 spin_unlock(lockp);
846 goto out;
847 }
13ee6ac5 848 hlist_nulls_for_each_entry(h, n, &net->ct.hash[cb->args[0]],
ea781f19 849 hnnode) {
5b1158e9 850 if (NF_CT_DIRECTION(h) != IP_CT_DIR_ORIGINAL)
c1d10adb
PNA		 851 continue;
852 ct = nf_ct_tuplehash_to_ctrack(h);
87711cb8
PNA		 853 /* Dump entries of a given L3 protocol number.
854 * If it is not specified, ie. l3proto == 0,
855 * then dump everything. */
5e8fbe2a 856 if (l3proto && nf_ct_l3num(ct) != l3proto)
13ee6ac5 857 continue;
d205dc40
PM		 858 if (cb->args[1]) {
859 if (ct != last)
13ee6ac5 860 continue;
d205dc40 861 cb->args[1] = 0;
89f2e218 862 }
866476f3 863 if (!ctnetlink_filter_match(ct, cb->data))
0f298a28 864 continue;
866476f3 865
3b988ece
HS		 866 rcu_read_lock();
867 res =
15e47304 868 ctnetlink_fill_info(skb, NETLINK_CB(cb->skb).portid,
3b988ece
HS		 869 cb->nlh->nlmsg_seq,
870 NFNL_MSG_TYPE(cb->nlh->nlmsg_type),
871 ct);
872 rcu_read_unlock();
873 if (res < 0) {
c71caf41 874 nf_conntrack_get(&ct->ct_general);
89f2e218 875 cb->args[1] = (unsigned long)ct;
93bb0ceb 876 spin_unlock(lockp);
c1d10adb 877 goto out;
89f2e218
PM		 878 }
879 }
93bb0ceb 880 spin_unlock(lockp);
d205dc40 881 if (cb->args[1]) {
89f2e218
PM		 882 cb->args[1] = 0;
883 goto restart;
c1d10adb
PNA		 884 }
885 }
89f2e218 886out:
93bb0ceb 887 local_bh_enable();
d205dc40
PM		 888 if (last)
889 nf_ct_put(last);
c1d10adb 890
c1d10adb
PNA		 891 return skb->len;
892}
893
c1d10adb 894static inline int
df6fb868 895ctnetlink_parse_tuple_ip(struct nlattr *attr, struct nf_conntrack_tuple *tuple)
c1d10adb 896{
df6fb868 897 struct nlattr *tb[CTA_IP_MAX+1];
c1d10adb
PNA		 898 struct nf_conntrack_l3proto *l3proto;
899 int ret = 0;
900
130ffbc2
DB		 901 ret = nla_parse_nested(tb, CTA_IP_MAX, attr, NULL);
902 if (ret < 0)
903 return ret;
c1d10adb 904
cd91566e
FW		 905 rcu_read_lock();
906 l3proto = __nf_ct_l3proto_find(tuple->src.l3num);
c1d10adb 907
f73e924c
PM		 908 if (likely(l3proto->nlattr_to_tuple)) {
909 ret = nla_validate_nested(attr, CTA_IP_MAX,
910 l3proto->nla_policy);
911 if (ret == 0)
912 ret = l3proto->nlattr_to_tuple(tb, tuple);
913 }
c1d10adb 914
cd91566e 915 rcu_read_unlock();
c1d10adb 916
c1d10adb
PNA		 917 return ret;
918}
919
f73e924c
PM		 920static const struct nla_policy proto_nla_policy[CTA_PROTO_MAX+1] = {
921 [CTA_PROTO_NUM] = { .type = NLA_U8 },
c1d10adb
PNA		 922};
923
924static inline int
df6fb868 925ctnetlink_parse_tuple_proto(struct nlattr *attr,
c1d10adb
PNA		 926 struct nf_conntrack_tuple *tuple)
927{
df6fb868 928 struct nlattr *tb[CTA_PROTO_MAX+1];
605dcad6 929 struct nf_conntrack_l4proto *l4proto;
c1d10adb
PNA		 930 int ret = 0;
931
f73e924c
PM		 932 ret = nla_parse_nested(tb, CTA_PROTO_MAX, attr, proto_nla_policy);
933 if (ret < 0)
934 return ret;
c1d10adb 935
df6fb868 936 if (!tb[CTA_PROTO_NUM])
c1d10adb 937 return -EINVAL;
77236b6e 938 tuple->dst.protonum = nla_get_u8(tb[CTA_PROTO_NUM]);
c1d10adb 939
cd91566e
FW		 940 rcu_read_lock();
941 l4proto = __nf_ct_l4proto_find(tuple->src.l3num, tuple->dst.protonum);
c1d10adb 942
f73e924c
PM		 943 if (likely(l4proto->nlattr_to_tuple)) {
944 ret = nla_validate_nested(attr, CTA_PROTO_MAX,
945 l4proto->nla_policy);
946 if (ret == 0)
947 ret = l4proto->nlattr_to_tuple(tb, tuple);
948 }
c1d10adb 949
cd91566e 950 rcu_read_unlock();
601e68e1 951
c1d10adb
PNA		 952 return ret;
953}
954
deedb590
DB		 955static int
956ctnetlink_parse_zone(const struct nlattr *attr,
957 struct nf_conntrack_zone *zone)
958{
5e8018fc
DB		 959 nf_ct_zone_init(zone, NF_CT_DEFAULT_ZONE_ID,
960 NF_CT_DEFAULT_ZONE_DIR, 0);
deedb590
DB		 961#ifdef CONFIG_NF_CONNTRACK_ZONES
962 if (attr)
963 zone->id = ntohs(nla_get_be16(attr));
964#else
965 if (attr)
966 return -EOPNOTSUPP;
967#endif
968 return 0;
969}
970
971static int
972ctnetlink_parse_tuple_zone(struct nlattr *attr, enum ctattr_type type,
973 struct nf_conntrack_zone *zone)
974{
975 int ret;
976
977 if (zone->id != NF_CT_DEFAULT_ZONE_ID)
978 return -EINVAL;
979
980 ret = ctnetlink_parse_zone(attr, zone);
981 if (ret < 0)
982 return ret;
983
984 if (type == CTA_TUPLE_REPLY)
985 zone->dir = NF_CT_ZONE_DIR_REPL;
986 else
987 zone->dir = NF_CT_ZONE_DIR_ORIG;
988
989 return 0;
990}
991
d0b0268f
PM		 992static const struct nla_policy tuple_nla_policy[CTA_TUPLE_MAX+1] = {
993 [CTA_TUPLE_IP] = { .type = NLA_NESTED },
994 [CTA_TUPLE_PROTO] = { .type = NLA_NESTED },
deedb590 995 [CTA_TUPLE_ZONE] = { .type = NLA_U16 },
d0b0268f
PM		 996};
997
bb5cf80e 998static int
39938324
PM		 999ctnetlink_parse_tuple(const struct nlattr * const cda[],
1000 struct nf_conntrack_tuple *tuple,
deedb590
DB		 1001 enum ctattr_type type, u_int8_t l3num,
1002 struct nf_conntrack_zone *zone)
c1d10adb 1003{
df6fb868 1004 struct nlattr *tb[CTA_TUPLE_MAX+1];
c1d10adb
PNA		 1005 int err;
1006
c1d10adb
PNA		 1007 memset(tuple, 0, sizeof(*tuple));
1008
130ffbc2
DB		 1009 err = nla_parse_nested(tb, CTA_TUPLE_MAX, cda[type], tuple_nla_policy);
1010 if (err < 0)
1011 return err;
c1d10adb 1012
df6fb868 1013 if (!tb[CTA_TUPLE_IP])
c1d10adb
PNA		 1014 return -EINVAL;
1015
1016 tuple->src.l3num = l3num;
1017
df6fb868 1018 err = ctnetlink_parse_tuple_ip(tb[CTA_TUPLE_IP], tuple);
c1d10adb
PNA		 1019 if (err < 0)
1020 return err;
1021
df6fb868 1022 if (!tb[CTA_TUPLE_PROTO])
c1d10adb
PNA		 1023 return -EINVAL;
1024
df6fb868 1025 err = ctnetlink_parse_tuple_proto(tb[CTA_TUPLE_PROTO], tuple);
c1d10adb
PNA		 1026 if (err < 0)
1027 return err;
1028
deedb590
DB		 1029 if (tb[CTA_TUPLE_ZONE]) {
1030 if (!zone)
1031 return -EINVAL;
1032
1033 err = ctnetlink_parse_tuple_zone(tb[CTA_TUPLE_ZONE],
1034 type, zone);
1035 if (err < 0)
1036 return err;
1037 }
1038
c1d10adb
PNA		 1039 /* orig and expect tuples get DIR_ORIGINAL */
1040 if (type == CTA_TUPLE_REPLY)
1041 tuple->dst.dir = IP_CT_DIR_REPLY;
1042 else
1043 tuple->dst.dir = IP_CT_DIR_ORIGINAL;
1044
c1d10adb
PNA		 1045 return 0;
1046}
1047
d0b0268f 1048static const struct nla_policy help_nla_policy[CTA_HELP_MAX+1] = {
6d1fafca
FW		 1049 [CTA_HELP_NAME] = { .type = NLA_NUL_STRING,
1050 .len = NF_CT_HELPER_NAME_LEN - 1 },
d0b0268f
PM		 1051};
1052
c1d10adb 1053static inline int
ae243bee
PNA		 1054ctnetlink_parse_help(const struct nlattr *attr, char **helper_name,
1055 struct nlattr **helpinfo)
c1d10adb 1056{
130ffbc2 1057 int err;
df6fb868 1058 struct nlattr *tb[CTA_HELP_MAX+1];
c1d10adb 1059
130ffbc2
DB		 1060 err = nla_parse_nested(tb, CTA_HELP_MAX, attr, help_nla_policy);
1061 if (err < 0)
1062 return err;
c1d10adb 1063
df6fb868 1064 if (!tb[CTA_HELP_NAME])
c1d10adb
PNA		 1065 return -EINVAL;
1066
df6fb868 1067 *helper_name = nla_data(tb[CTA_HELP_NAME]);
c1d10adb 1068
ae243bee
PNA		 1069 if (tb[CTA_HELP_INFO])
1070 *helpinfo = tb[CTA_HELP_INFO];
1071
c1d10adb
PNA		 1072 return 0;
1073}
1074
f73e924c 1075static const struct nla_policy ct_nla_policy[CTA_MAX+1] = {
d0b0268f
PM		 1076 [CTA_TUPLE_ORIG] = { .type = NLA_NESTED },
1077 [CTA_TUPLE_REPLY] = { .type = NLA_NESTED },
f73e924c 1078 [CTA_STATUS] = { .type = NLA_U32 },
d0b0268f
PM		 1079 [CTA_PROTOINFO] = { .type = NLA_NESTED },
1080 [CTA_HELP] = { .type = NLA_NESTED },
1081 [CTA_NAT_SRC] = { .type = NLA_NESTED },
f73e924c
PM		 1082 [CTA_TIMEOUT] = { .type = NLA_U32 },
1083 [CTA_MARK] = { .type = NLA_U32 },
f73e924c 1084 [CTA_ID] = { .type = NLA_U32 },
d0b0268f
PM		 1085 [CTA_NAT_DST] = { .type = NLA_NESTED },
1086 [CTA_TUPLE_MASTER] = { .type = NLA_NESTED },
6d1fafca
FW		 1087 [CTA_NAT_SEQ_ADJ_ORIG] = { .type = NLA_NESTED },
1088 [CTA_NAT_SEQ_ADJ_REPLY] = { .type = NLA_NESTED },
ef00f89f 1089 [CTA_ZONE] = { .type = NLA_U16 },
0f298a28 1090 [CTA_MARK_MASK] = { .type = NLA_U32 },
9b21f6a9 1091 [CTA_LABELS] = { .type = NLA_BINARY,
d2bf2f34 1092 .len = NF_CT_LABELS_MAX_SIZE },
9b21f6a9 1093 [CTA_LABELS_MASK] = { .type = NLA_BINARY,
d2bf2f34 1094 .len = NF_CT_LABELS_MAX_SIZE },
c1d10adb
PNA		 1095};
1096
866476f3
KE		 1097static int ctnetlink_flush_conntrack(struct net *net,
1098 const struct nlattr * const cda[],
1099 u32 portid, int report)
1100{
1101 struct ctnetlink_filter *filter = NULL;
1102
1103 if (cda[CTA_MARK] && cda[CTA_MARK_MASK]) {
1104 filter = ctnetlink_alloc_filter(cda);
1105 if (IS_ERR(filter))
1106 return PTR_ERR(filter);
1107 }
1108
1109 nf_ct_iterate_cleanup(net, ctnetlink_filter_match, filter,
1110 portid, report);
1111 kfree(filter);
1112
1113 return 0;
1114}
1115
7b8002a1
PNA		 1116static int ctnetlink_del_conntrack(struct net *net, struct sock *ctnl,
1117 struct sk_buff *skb,
1118 const struct nlmsghdr *nlh,
1119 const struct nlattr * const cda[])
c1d10adb
PNA		 1120{
1121 struct nf_conntrack_tuple_hash *h;
1122 struct nf_conntrack_tuple tuple;
1123 struct nf_conn *ct;
96bcf938 1124 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
c1d10adb 1125 u_int8_t u3 = nfmsg->nfgen_family;
308ac914 1126 struct nf_conntrack_zone zone;
ef00f89f
PM		 1127 int err;
1128
1129 err = ctnetlink_parse_zone(cda[CTA_ZONE], &zone);
1130 if (err < 0)
1131 return err;
c1d10adb 1132
df6fb868 1133 if (cda[CTA_TUPLE_ORIG])
deedb590
DB		 1134 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_ORIG,
1135 u3, &zone);
df6fb868 1136 else if (cda[CTA_TUPLE_REPLY])
deedb590
DB		 1137 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_REPLY,
1138 u3, &zone);
c1d10adb 1139 else {
866476f3
KE		 1140 return ctnetlink_flush_conntrack(net, cda,
1141 NETLINK_CB(skb).portid,
1142 nlmsg_report(nlh));
c1d10adb
PNA		 1143 }
1144
1145 if (err < 0)
1146 return err;
1147
308ac914 1148 h = nf_conntrack_find_get(net, &zone, &tuple);
9ea8cfd6 1149 if (!h)
c1d10adb 1150 return -ENOENT;
c1d10adb
PNA		 1151
1152 ct = nf_ct_tuplehash_to_ctrack(h);
601e68e1 1153
df6fb868 1154 if (cda[CTA_ID]) {
77236b6e 1155 u_int32_t id = ntohl(nla_get_be32(cda[CTA_ID]));
7f85f914 1156 if (id != (u32)(unsigned long)ct) {
c1d10adb
PNA		 1157 nf_ct_put(ct);
1158 return -ENOENT;
1159 }
601e68e1 1160 }
c1d10adb 1161
02982c27
FW		 1162 if (del_timer(&ct->timeout))
1163 nf_ct_delete(ct, NETLINK_CB(skb).portid, nlmsg_report(nlh));
1164
c1d10adb 1165 nf_ct_put(ct);
c1d10adb
PNA		 1166
1167 return 0;
1168}
1169
7b8002a1
PNA		 1170static int ctnetlink_get_conntrack(struct net *net, struct sock *ctnl,
1171 struct sk_buff *skb,
1172 const struct nlmsghdr *nlh,
1173 const struct nlattr * const cda[])
c1d10adb
PNA		 1174{
1175 struct nf_conntrack_tuple_hash *h;
1176 struct nf_conntrack_tuple tuple;
1177 struct nf_conn *ct;
1178 struct sk_buff *skb2 = NULL;
96bcf938 1179 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
c1d10adb 1180 u_int8_t u3 = nfmsg->nfgen_family;
308ac914 1181 struct nf_conntrack_zone zone;
ef00f89f 1182 int err;
c1d10adb 1183
80d326fa
PNA		 1184 if (nlh->nlmsg_flags & NLM_F_DUMP) {
1185 struct netlink_dump_control c = {
1186 .dump = ctnetlink_dump_table,
1187 .done = ctnetlink_done,
1188 };
866476f3 1189
0f298a28 1190 if (cda[CTA_MARK] && cda[CTA_MARK_MASK]) {
866476f3 1191 struct ctnetlink_filter *filter;
0f298a28 1192
866476f3
KE		 1193 filter = ctnetlink_alloc_filter(cda);
1194 if (IS_ERR(filter))
1195 return PTR_ERR(filter);
0f298a28 1196
0f298a28
PNA		 1197 c.data = filter;
1198 }
80d326fa
PNA		 1199 return netlink_dump_start(ctnl, skb, nlh, &c);
1200 }
c1d10adb 1201
ef00f89f
PM		 1202 err = ctnetlink_parse_zone(cda[CTA_ZONE], &zone);
1203 if (err < 0)
1204 return err;
1205
df6fb868 1206 if (cda[CTA_TUPLE_ORIG])
deedb590
DB		 1207 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_ORIG,
1208 u3, &zone);
df6fb868 1209 else if (cda[CTA_TUPLE_REPLY])
deedb590
DB		 1210 err = ctnetlink_parse_tuple(cda, &tuple, CTA_TUPLE_REPLY,
1211 u3, &zone);
c1d10adb
PNA		 1212 else
1213 return -EINVAL;
1214
1215 if (err < 0)
1216 return err;
1217
308ac914 1218 h = nf_conntrack_find_get(net, &zone, &tuple);
9ea8cfd6 1219 if (!h)
c1d10adb 1220 return -ENOENT;
9ea8cfd6 1221
c1d10adb
PNA		 1222 ct = nf_ct_tuplehash_to_ctrack(h);
1223
1224 err = -ENOMEM;
96bcf938
PNA		 1225 skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1226 if (skb2 == NULL) {
c1d10adb
PNA		 1227 nf_ct_put(ct);
1228 return -ENOMEM;
1229 }
c1d10adb 1230
528a3a6f 1231 rcu_read_lock();
15e47304 1232 err = ctnetlink_fill_info(skb2, NETLINK_CB(skb).portid, nlh->nlmsg_seq,
80e60e67 1233 NFNL_MSG_TYPE(nlh->nlmsg_type), ct);
528a3a6f 1234 rcu_read_unlock();
c1d10adb
PNA		 1235 nf_ct_put(ct);
1236 if (err <= 0)
1237 goto free;
1238
15e47304 1239 err = netlink_unicast(ctnl, skb2, NETLINK_CB(skb).portid, MSG_DONTWAIT);
c1d10adb
PNA		 1240 if (err < 0)
1241 goto out;
1242
c1d10adb
PNA		 1243 return 0;
1244
1245free:
1246 kfree_skb(skb2);
1247out:
f31e8d49
PNA		 1248 /* this avoids a loop in nfnetlink. */
1249 return err == -EAGAIN ? -ENOBUFS : err;
c1d10adb
PNA		 1250}
1251
d871befe
PNA		 1252static int ctnetlink_done_list(struct netlink_callback *cb)
1253{
1254 if (cb->args[1])
1255 nf_ct_put((struct nf_conn *)cb->args[1]);
1256 return 0;
1257}
1258
1259static int
b7779d06 1260ctnetlink_dump_list(struct sk_buff *skb, struct netlink_callback *cb, bool dying)
d871befe 1261{
cd5f336f 1262 struct nf_conn *ct, *last;
d871befe
PNA		 1263 struct nf_conntrack_tuple_hash *h;
1264 struct hlist_nulls_node *n;
1265 struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
1266 u_int8_t l3proto = nfmsg->nfgen_family;
1267 int res;
b7779d06
JDB		 1268 int cpu;
1269 struct hlist_nulls_head *list;
1270 struct net *net = sock_net(skb->sk);
d871befe
PNA		 1271
1272 if (cb->args[2])
1273 return 0;
1274
cd5f336f
FW		 1275 last = (struct nf_conn *)cb->args[1];
1276
b7779d06
JDB		 1277 for (cpu = cb->args[0]; cpu < nr_cpu_ids; cpu++) {
1278 struct ct_pcpu *pcpu;
1279
1280 if (!cpu_possible(cpu))
d871befe 1281 continue;
b7779d06
JDB		 1282
1283 pcpu = per_cpu_ptr(net->ct.pcpu_lists, cpu);
1284 spin_lock_bh(&pcpu->lock);
b7779d06
JDB		 1285 list = dying ? &pcpu->dying : &pcpu->unconfirmed;
1286restart:
1287 hlist_nulls_for_each_entry(h, n, list, hnnode) {
1288 ct = nf_ct_tuplehash_to_ctrack(h);
1289 if (l3proto && nf_ct_l3num(ct) != l3proto)
d871befe 1290 continue;
b7779d06
JDB		 1291 if (cb->args[1]) {
1292 if (ct != last)
1293 continue;
1294 cb->args[1] = 0;
1295 }
1296 rcu_read_lock();
1297 res = ctnetlink_fill_info(skb, NETLINK_CB(cb->skb).portid,
1298 cb->nlh->nlmsg_seq,
1299 NFNL_MSG_TYPE(cb->nlh->nlmsg_type),
1300 ct);
1301 rcu_read_unlock();
1302 if (res < 0) {
cd5f336f
FW		 1303 if (!atomic_inc_not_zero(&ct->ct_general.use))
1304 continue;
266155b2 1305 cb->args[0] = cpu;
b7779d06
JDB		 1306 cb->args[1] = (unsigned long)ct;
1307 spin_unlock_bh(&pcpu->lock);
1308 goto out;
1309 }
d871befe 1310 }
b7779d06
JDB		 1311 if (cb->args[1]) {
1312 cb->args[1] = 0;
1313 goto restart;
266155b2 1314 }
b7779d06 1315 spin_unlock_bh(&pcpu->lock);
d871befe 1316 }
266155b2 1317 cb->args[2] = 1;
d871befe 1318out:
d871befe
PNA		 1319 if (last)
1320 nf_ct_put(last);
1321
1322 return skb->len;
1323}
1324
1325static int
1326ctnetlink_dump_dying(struct sk_buff *skb, struct netlink_callback *cb)
1327{
b7779d06 1328 return ctnetlink_dump_list(skb, cb, true);
d871befe
PNA		 1329}
1330
7b8002a1
PNA		 1331static int ctnetlink_get_ct_dying(struct net *net, struct sock *ctnl,
1332 struct sk_buff *skb,
1333 const struct nlmsghdr *nlh,
1334 const struct nlattr * const cda[])
d871befe
PNA		 1335{
1336 if (nlh->nlmsg_flags & NLM_F_DUMP) {
1337 struct netlink_dump_control c = {
1338 .dump = ctnetlink_dump_dying,
1339 .done = ctnetlink_done_list,
1340 };
1341 return netlink_dump_start(ctnl, skb, nlh, &c);
1342 }
1343
1344 return -EOPNOTSUPP;
1345}
1346
1347static int
1348ctnetlink_dump_unconfirmed(struct sk_buff *skb, struct netlink_callback *cb)
1349{
b7779d06 1350 return ctnetlink_dump_list(skb, cb, false);
d871befe
PNA		 1351}
1352
7b8002a1
PNA		 1353static int ctnetlink_get_ct_unconfirmed(struct net *net, struct sock *ctnl,
1354 struct sk_buff *skb,
1355 const struct nlmsghdr *nlh,
1356 const struct nlattr * const cda[])
d871befe
PNA		 1357{
1358 if (nlh->nlmsg_flags & NLM_F_DUMP) {
1359 struct netlink_dump_control c = {
1360 .dump = ctnetlink_dump_unconfirmed,
1361 .done = ctnetlink_done_list,
1362 };
1363 return netlink_dump_start(ctnl, skb, nlh, &c);
1364 }
1365
1366 return -EOPNOTSUPP;
1367}
1368
67671841 1369#ifdef CONFIG_NF_NAT_NEEDED
e6a7d3c0
PNA		 1370static int
1371ctnetlink_parse_nat_setup(struct nf_conn *ct,
1372 enum nf_nat_manip_type manip,
39938324 1373 const struct nlattr *attr)
e6a7d3c0
PNA		 1374{
1375 typeof(nfnetlink_parse_nat_setup_hook) parse_nat_setup;
c7232c99 1376 int err;
e6a7d3c0
PNA		 1377
1378 parse_nat_setup = rcu_dereference(nfnetlink_parse_nat_setup_hook);
1379 if (!parse_nat_setup) {
95a5afca 1380#ifdef CONFIG_MODULES
e6a7d3c0 1381 rcu_read_unlock();
c14b78e7 1382 nfnl_unlock(NFNL_SUBSYS_CTNETLINK);
c7232c99 1383 if (request_module("nf-nat") < 0) {
c14b78e7 1384 nfnl_lock(NFNL_SUBSYS_CTNETLINK);
e6a7d3c0
PNA		 1385 rcu_read_lock();
1386 return -EOPNOTSUPP;
1387 }
c14b78e7 1388 nfnl_lock(NFNL_SUBSYS_CTNETLINK);
e6a7d3c0
PNA		 1389 rcu_read_lock();
1390 if (nfnetlink_parse_nat_setup_hook)
1391 return -EAGAIN;
1392#endif
1393 return -EOPNOTSUPP;
1394 }
1395
c7232c99
PM		 1396 err = parse_nat_setup(ct, manip, attr);
1397 if (err == -EAGAIN) {
1398#ifdef CONFIG_MODULES
1399 rcu_read_unlock();
c14b78e7 1400 nfnl_unlock(NFNL_SUBSYS_CTNETLINK);
c7232c99 1401 if (request_module("nf-nat-%u", nf_ct_l3num(ct)) < 0) {
c14b78e7 1402 nfnl_lock(NFNL_SUBSYS_CTNETLINK);
c7232c99
PM		 1403 rcu_read_lock();
1404 return -EOPNOTSUPP;
1405 }
c14b78e7 1406 nfnl_lock(NFNL_SUBSYS_CTNETLINK);
c7232c99
PM		 1407 rcu_read_lock();
1408#else
1409 err = -EOPNOTSUPP;
1410#endif
1411 }
1412 return err;
e6a7d3c0 1413}
67671841 1414#endif
e6a7d3c0 1415
bb5cf80e 1416static int
39938324 1417ctnetlink_change_status(struct nf_conn *ct, const struct nlattr * const cda[])
c1d10adb
PNA		 1418{
1419 unsigned long d;
77236b6e 1420 unsigned int status = ntohl(nla_get_be32(cda[CTA_STATUS]));
c1d10adb
PNA		 1421 d = ct->status ^ status;
1422
1423 if (d & (IPS_EXPECTED|IPS_CONFIRMED|IPS_DYING))
1424 /* unchangeable */
0adf9d67 1425 return -EBUSY;
601e68e1 1426
c1d10adb
PNA		 1427 if (d & IPS_SEEN_REPLY && !(status & IPS_SEEN_REPLY))
1428 /* SEEN_REPLY bit can only be set */
0adf9d67 1429 return -EBUSY;
601e68e1 1430
c1d10adb
PNA		 1431 if (d & IPS_ASSURED && !(status & IPS_ASSURED))
1432 /* ASSURED bit can only be set */
0adf9d67 1433 return -EBUSY;
c1d10adb 1434
c1d10adb
PNA		 1435 /* Be careful here, modifying NAT bits can screw up things,
1436 * so don't let users modify them directly if they don't pass
5b1158e9 1437 * nf_nat_range. */
c1d10adb
PNA		 1438 ct->status |= status & ~(IPS_NAT_DONE_MASK | IPS_NAT_MASK);
1439 return 0;
1440}
1441
e6a7d3c0 1442static int
0eba801b 1443ctnetlink_setup_nat(struct nf_conn *ct, const struct nlattr * const cda[])
e6a7d3c0
PNA		 1444{
1445#ifdef CONFIG_NF_NAT_NEEDED
1446 int ret;
1447
fe337ac2
FW		 1448 if (!cda[CTA_NAT_DST] && !cda[CTA_NAT_SRC])
1449 return 0;
1450
0eba801b
PNA		 1451 ret = ctnetlink_parse_nat_setup(ct, NF_NAT_MANIP_DST,
1452 cda[CTA_NAT_DST]);
1453 if (ret < 0)
1454 return ret;
1455
1456 ret = ctnetlink_parse_nat_setup(ct, NF_NAT_MANIP_SRC,
1457 cda[CTA_NAT_SRC]);
1458 return ret;
e6a7d3c0 1459#else
0eba801b
PNA		 1460 if (!cda[CTA_NAT_DST] && !cda[CTA_NAT_SRC])
1461 return 0;
e6a7d3c0
PNA		 1462 return -EOPNOTSUPP;
1463#endif
1464}
c1d10adb
PNA		 1465
1466static inline int
39938324 1467ctnetlink_change_helper(struct nf_conn *ct, const struct nlattr * const cda[])
c1d10adb
PNA		 1468{
1469 struct nf_conntrack_helper *helper;
dc808fe2 1470 struct nf_conn_help *help = nfct_help(ct);
29fe1b48 1471 char *helpname = NULL;
ae243bee 1472 struct nlattr *helpinfo = NULL;
c1d10adb
PNA		 1473 int err;
1474
c1d10adb
PNA		 1475 /* don't change helper of sibling connections */
1476 if (ct->master)
0adf9d67 1477 return -EBUSY;
c1d10adb 1478
ae243bee 1479 err = ctnetlink_parse_help(cda[CTA_HELP], &helpname, &helpinfo);
c1d10adb
PNA		 1480 if (err < 0)
1481 return err;
1482
df293bbb
YK		 1483 if (!strcmp(helpname, "")) {
1484 if (help && help->helper) {
c1d10adb
PNA		 1485 /* we had a helper before ... */
1486 nf_ct_remove_expectations(ct);
a9b3cd7f 1487 RCU_INIT_POINTER(help->helper, NULL);
c1d10adb 1488 }
df293bbb
YK		 1489
1490 return 0;
c1d10adb 1491 }
601e68e1 1492
794e6871
PM		 1493 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
1494 nf_ct_protonum(ct));
226c0c0e
PNA		 1495 if (helper == NULL) {
1496#ifdef CONFIG_MODULES
ca7433df 1497 spin_unlock_bh(&nf_conntrack_expect_lock);
226c0c0e
PNA		 1498
1499 if (request_module("nfct-helper-%s", helpname) < 0) {
ca7433df 1500 spin_lock_bh(&nf_conntrack_expect_lock);
226c0c0e
PNA		 1501 return -EOPNOTSUPP;
1502 }
1503
ca7433df 1504 spin_lock_bh(&nf_conntrack_expect_lock);
794e6871
PM		 1505 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
1506 nf_ct_protonum(ct));
226c0c0e
PNA		 1507 if (helper)
1508 return -EAGAIN;
1509#endif
0adf9d67 1510 return -EOPNOTSUPP;
226c0c0e 1511 }
df293bbb 1512
ceceae1b 1513 if (help) {
ae243bee
PNA		 1514 if (help->helper == helper) {
1515 /* update private helper data if allowed. */
7be54ca4 1516 if (helper->from_nlattr)
ae243bee 1517 helper->from_nlattr(helpinfo, ct);
ceceae1b 1518 return 0;
fd7462de 1519 } else
ceceae1b 1520 return -EBUSY;
ceceae1b 1521 }
df293bbb 1522
fd7462de
PNA		 1523 /* we cannot set a helper for an existing conntrack */
1524 return -EOPNOTSUPP;
c1d10adb
PNA		 1525}
1526
1527static inline int
39938324 1528ctnetlink_change_timeout(struct nf_conn *ct, const struct nlattr * const cda[])
c1d10adb 1529{
77236b6e 1530 u_int32_t timeout = ntohl(nla_get_be32(cda[CTA_TIMEOUT]));
601e68e1 1531
c1d10adb
PNA		 1532 if (!del_timer(&ct->timeout))
1533 return -ETIME;
1534
1535 ct->timeout.expires = jiffies + timeout * HZ;
1536 add_timer(&ct->timeout);
1537
1538 return 0;
1539}
1540
d0b0268f
PM		 1541static const struct nla_policy protoinfo_policy[CTA_PROTOINFO_MAX+1] = {
1542 [CTA_PROTOINFO_TCP] = { .type = NLA_NESTED },
1543 [CTA_PROTOINFO_DCCP] = { .type = NLA_NESTED },
1544 [CTA_PROTOINFO_SCTP] = { .type = NLA_NESTED },
1545};
1546
c1d10adb 1547static inline int
39938324 1548ctnetlink_change_protoinfo(struct nf_conn *ct, const struct nlattr * const cda[])
c1d10adb 1549{
39938324
PM		 1550 const struct nlattr *attr = cda[CTA_PROTOINFO];
1551 struct nlattr *tb[CTA_PROTOINFO_MAX+1];
605dcad6 1552 struct nf_conntrack_l4proto *l4proto;
c1d10adb
PNA		 1553 int err = 0;
1554
130ffbc2
DB		 1555 err = nla_parse_nested(tb, CTA_PROTOINFO_MAX, attr, protoinfo_policy);
1556 if (err < 0)
1557 return err;
c1d10adb 1558
cd91566e
FW		 1559 rcu_read_lock();
1560 l4proto = __nf_ct_l4proto_find(nf_ct_l3num(ct), nf_ct_protonum(ct));
fdf70832
PM		 1561 if (l4proto->from_nlattr)
1562 err = l4proto->from_nlattr(tb, ct);
cd91566e 1563 rcu_read_unlock();
c1d10adb
PNA		 1564
1565 return err;
1566}
1567
41d73ec0
PM		 1568static const struct nla_policy seqadj_policy[CTA_SEQADJ_MAX+1] = {
1569 [CTA_SEQADJ_CORRECTION_POS] = { .type = NLA_U32 },
1570 [CTA_SEQADJ_OFFSET_BEFORE] = { .type = NLA_U32 },
1571 [CTA_SEQADJ_OFFSET_AFTER] = { .type = NLA_U32 },
d0b0268f
PM		 1572};
1573
13eae15a 1574static inline int
41d73ec0 1575change_seq_adj(struct nf_ct_seqadj *seq, const struct nlattr * const attr)
13eae15a 1576{
130ffbc2 1577 int err;
41d73ec0 1578 struct nlattr *cda[CTA_SEQADJ_MAX+1];
13eae15a 1579
41d73ec0 1580 err = nla_parse_nested(cda, CTA_SEQADJ_MAX, attr, seqadj_policy);
130ffbc2
DB		 1581 if (err < 0)
1582 return err;
13eae15a 1583
41d73ec0 1584 if (!cda[CTA_SEQADJ_CORRECTION_POS])
13eae15a
PNA		 1585 return -EINVAL;
1586
41d73ec0
PM		 1587 seq->correction_pos =
1588 ntohl(nla_get_be32(cda[CTA_SEQADJ_CORRECTION_POS]));
13eae15a 1589
41d73ec0 1590 if (!cda[CTA_SEQADJ_OFFSET_BEFORE])
13eae15a
PNA		 1591 return -EINVAL;
1592
41d73ec0
PM		 1593 seq->offset_before =
1594 ntohl(nla_get_be32(cda[CTA_SEQADJ_OFFSET_BEFORE]));
13eae15a 1595
41d73ec0 1596 if (!cda[CTA_SEQADJ_OFFSET_AFTER])
13eae15a
PNA		 1597 return -EINVAL;
1598
41d73ec0
PM		 1599 seq->offset_after =
1600 ntohl(nla_get_be32(cda[CTA_SEQADJ_OFFSET_AFTER]));
13eae15a
PNA		 1601
1602 return 0;
1603}
1604
1605static int
41d73ec0
PM		 1606ctnetlink_change_seq_adj(struct nf_conn *ct,
1607 const struct nlattr * const cda[])
13eae15a 1608{
41d73ec0 1609 struct nf_conn_seqadj *seqadj = nfct_seqadj(ct);
13eae15a 1610 int ret = 0;
13eae15a 1611
41d73ec0 1612 if (!seqadj)
13eae15a
PNA		 1613 return 0;
1614
41d73ec0
PM		 1615 if (cda[CTA_SEQ_ADJ_ORIG]) {
1616 ret = change_seq_adj(&seqadj->seq[IP_CT_DIR_ORIGINAL],
1617 cda[CTA_SEQ_ADJ_ORIG]);
13eae15a
PNA		 1618 if (ret < 0)
1619 return ret;
1620
1621 ct->status |= IPS_SEQ_ADJUST;
1622 }
1623
41d73ec0
PM		 1624 if (cda[CTA_SEQ_ADJ_REPLY]) {
1625 ret = change_seq_adj(&seqadj->seq[IP_CT_DIR_REPLY],
1626 cda[CTA_SEQ_ADJ_REPLY]);
13eae15a
PNA		 1627 if (ret < 0)
1628 return ret;
1629
1630 ct->status |= IPS_SEQ_ADJUST;
1631 }
1632
1633 return 0;
1634}
13eae15a 1635
9b21f6a9
FW		 1636static int
1637ctnetlink_attach_labels(struct nf_conn *ct, const struct nlattr * const cda[])
1638{
1639#ifdef CONFIG_NF_CONNTRACK_LABELS
1640 size_t len = nla_len(cda[CTA_LABELS]);
1641 const void *mask = cda[CTA_LABELS_MASK];
1642
1643 if (len & (sizeof(u32)-1)) /* must be multiple of u32 */
1644 return -EINVAL;
1645
1646 if (mask) {
1647 if (nla_len(cda[CTA_LABELS_MASK]) == 0 ||
1648 nla_len(cda[CTA_LABELS_MASK]) != len)
1649 return -EINVAL;
1650 mask = nla_data(cda[CTA_LABELS_MASK]);
1651 }
1652
1653 len /= sizeof(u32);
1654
1655 return nf_connlabels_replace(ct, nla_data(cda[CTA_LABELS]), mask, len);
1656#else
1657 return -EOPNOTSUPP;
1658#endif
1659}
1660
c1d10adb 1661static int
39938324
PM		 1662ctnetlink_change_conntrack(struct nf_conn *ct,
1663 const struct nlattr * const cda[])
c1d10adb
PNA		 1664{
1665 int err;
1666
e098360f
PNA		 1667 /* only allow NAT changes and master assignation for new conntracks */
1668 if (cda[CTA_NAT_SRC] || cda[CTA_NAT_DST] || cda[CTA_TUPLE_MASTER])
1669 return -EOPNOTSUPP;
1670
df6fb868 1671 if (cda[CTA_HELP]) {
c1d10adb
PNA		 1672 err = ctnetlink_change_helper(ct, cda);
1673 if (err < 0)
1674 return err;
1675 }
1676
df6fb868 1677 if (cda[CTA_TIMEOUT]) {
c1d10adb
PNA		 1678 err = ctnetlink_change_timeout(ct, cda);
1679 if (err < 0)
1680 return err;
1681 }
1682
df6fb868 1683 if (cda[CTA_STATUS]) {
c1d10adb
PNA		 1684 err = ctnetlink_change_status(ct, cda);
1685 if (err < 0)
1686 return err;
1687 }
1688
df6fb868 1689 if (cda[CTA_PROTOINFO]) {
c1d10adb
PNA		 1690 err = ctnetlink_change_protoinfo(ct, cda);
1691 if (err < 0)
1692 return err;
1693 }
1694
bcd1e830 1695#if defined(CONFIG_NF_CONNTRACK_MARK)
df6fb868 1696 if (cda[CTA_MARK])
77236b6e 1697 ct->mark = ntohl(nla_get_be32(cda[CTA_MARK]));
c1d10adb
PNA		 1698#endif
1699
41d73ec0
PM		 1700 if (cda[CTA_SEQ_ADJ_ORIG] || cda[CTA_SEQ_ADJ_REPLY]) {
1701 err = ctnetlink_change_seq_adj(ct, cda);
13eae15a
PNA		 1702 if (err < 0)
1703 return err;
1704 }
41d73ec0 1705
9b21f6a9
FW		 1706 if (cda[CTA_LABELS]) {
1707 err = ctnetlink_attach_labels(ct, cda);
1708 if (err < 0)
1709 return err;
1710 }
13eae15a 1711
c1d10adb
PNA		 1712 return 0;
1713}
1714
f0a3c086 1715static struct nf_conn *
308ac914
DB		 1716ctnetlink_create_conntrack(struct net *net,
1717 const struct nf_conntrack_zone *zone,
9592a5c0 1718 const struct nlattr * const cda[],
c1d10adb 1719 struct nf_conntrack_tuple *otuple,
5faa1f4c 1720 struct nf_conntrack_tuple *rtuple,
7ec47496 1721 u8 u3)
c1d10adb
PNA		 1722{
1723 struct nf_conn *ct;
1724 int err = -EINVAL;
ceceae1b 1725 struct nf_conntrack_helper *helper;
315c34da 1726 struct nf_conn_tstamp *tstamp;
c1d10adb 1727
ef00f89f 1728 ct = nf_conntrack_alloc(net, zone, otuple, rtuple, GFP_ATOMIC);
cd7fcbf1 1729 if (IS_ERR(ct))
f0a3c086 1730 return ERR_PTR(-ENOMEM);
c1d10adb 1731
df6fb868 1732 if (!cda[CTA_TIMEOUT])
0f5b3e85 1733 goto err1;
77236b6e 1734 ct->timeout.expires = ntohl(nla_get_be32(cda[CTA_TIMEOUT]));
c1d10adb
PNA		 1735
1736 ct->timeout.expires = jiffies + ct->timeout.expires * HZ;
c1d10adb 1737
1575e7ea 1738 rcu_read_lock();
226c0c0e 1739 if (cda[CTA_HELP]) {
29fe1b48 1740 char *helpname = NULL;
ae243bee
PNA		 1741 struct nlattr *helpinfo = NULL;
1742
1743 err = ctnetlink_parse_help(cda[CTA_HELP], &helpname, &helpinfo);
0f5b3e85
PM		 1744 if (err < 0)
1745 goto err2;
226c0c0e 1746
794e6871
PM		 1747 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
1748 nf_ct_protonum(ct));
226c0c0e
PNA		 1749 if (helper == NULL) {
1750 rcu_read_unlock();
1751#ifdef CONFIG_MODULES
1752 if (request_module("nfct-helper-%s", helpname) < 0) {
1753 err = -EOPNOTSUPP;
0f5b3e85 1754 goto err1;
226c0c0e
PNA		 1755 }
1756
1757 rcu_read_lock();
794e6871
PM		 1758 helper = __nf_conntrack_helper_find(helpname,
1759 nf_ct_l3num(ct),
1760 nf_ct_protonum(ct));
226c0c0e 1761 if (helper) {
226c0c0e 1762 err = -EAGAIN;
0f5b3e85 1763 goto err2;
226c0c0e
PNA		 1764 }
1765 rcu_read_unlock();
1766#endif
1767 err = -EOPNOTSUPP;
0f5b3e85 1768 goto err1;
226c0c0e
PNA		 1769 } else {
1770 struct nf_conn_help *help;
1771
1afc5679 1772 help = nf_ct_helper_ext_add(ct, helper, GFP_ATOMIC);
226c0c0e 1773 if (help == NULL) {
226c0c0e 1774 err = -ENOMEM;
0f5b3e85 1775 goto err2;
226c0c0e 1776 }
ae243bee 1777 /* set private helper data if allowed. */
7be54ca4 1778 if (helper->from_nlattr)
ae243bee 1779 helper->from_nlattr(helpinfo, ct);
226c0c0e
PNA		 1780
1781 /* not in hash table yet so not strictly necessary */
a9b3cd7f 1782 RCU_INIT_POINTER(help->helper, helper);
226c0c0e
PNA		 1783 }
1784 } else {
1785 /* try an implicit helper assignation */
b2a15a60 1786 err = __nf_ct_try_assign_helper(ct, NULL, GFP_ATOMIC);
0f5b3e85
PM		 1787 if (err < 0)
1788 goto err2;
1575e7ea
PNA		 1789 }
1790
0eba801b
PNA		 1791 err = ctnetlink_setup_nat(ct, cda);
1792 if (err < 0)
1793 goto err2;
e6a7d3c0 1794
a88e22ad 1795 nf_ct_acct_ext_add(ct, GFP_ATOMIC);
a992ca2a 1796 nf_ct_tstamp_ext_add(ct, GFP_ATOMIC);
a88e22ad 1797 nf_ct_ecache_ext_add(ct, 0, 0, GFP_ATOMIC);
c539f017
FW		 1798 nf_ct_labels_ext_add(ct);
1799
a88e22ad
PNA		 1800 /* we must add conntrack extensions before confirmation. */
1801 ct->status |= IPS_CONFIRMED;
1802
1803 if (cda[CTA_STATUS]) {
1804 err = ctnetlink_change_status(ct, cda);
0f5b3e85
PM		 1805 if (err < 0)
1806 goto err2;
bbb3357d 1807 }
c1d10adb 1808
41d73ec0
PM		 1809 if (cda[CTA_SEQ_ADJ_ORIG] || cda[CTA_SEQ_ADJ_REPLY]) {
1810 err = ctnetlink_change_seq_adj(ct, cda);
0f5b3e85
PM		 1811 if (err < 0)
1812 goto err2;
c969aa7d 1813 }
c969aa7d 1814
e5fc9e7a 1815 memset(&ct->proto, 0, sizeof(ct->proto));
df6fb868 1816 if (cda[CTA_PROTOINFO]) {
c1d10adb 1817 err = ctnetlink_change_protoinfo(ct, cda);
0f5b3e85
PM		 1818 if (err < 0)
1819 goto err2;
c1d10adb
PNA		 1820 }
1821
bcd1e830 1822#if defined(CONFIG_NF_CONNTRACK_MARK)
df6fb868 1823 if (cda[CTA_MARK])
77236b6e 1824 ct->mark = ntohl(nla_get_be32(cda[CTA_MARK]));
c1d10adb
PNA		 1825#endif
1826
5faa1f4c 1827 /* setup master conntrack: this is a confirmed expectation */
7ec47496
PNA		 1828 if (cda[CTA_TUPLE_MASTER]) {
1829 struct nf_conntrack_tuple master;
1830 struct nf_conntrack_tuple_hash *master_h;
1831 struct nf_conn *master_ct;
1832
deedb590
DB		 1833 err = ctnetlink_parse_tuple(cda, &master, CTA_TUPLE_MASTER,
1834 u3, NULL);
7ec47496 1835 if (err < 0)
0f5b3e85 1836 goto err2;
7ec47496 1837
ef00f89f 1838 master_h = nf_conntrack_find_get(net, zone, &master);
7ec47496
PNA		 1839 if (master_h == NULL) {
1840 err = -ENOENT;
0f5b3e85 1841 goto err2;
7ec47496
PNA		 1842 }
1843 master_ct = nf_ct_tuplehash_to_ctrack(master_h);
f2a89004 1844 __set_bit(IPS_EXPECTED_BIT, &ct->status);
5faa1f4c 1845 ct->master = master_ct;
f2a89004 1846 }
315c34da
PNA		 1847 tstamp = nf_conn_tstamp_find(ct);
1848 if (tstamp)
d2de875c 1849 tstamp->start = ktime_get_real_ns();
5faa1f4c 1850
7d367e06
JK		 1851 err = nf_conntrack_hash_check_insert(ct);
1852 if (err < 0)
1853 goto err2;
1854
58a3c9bb 1855 rcu_read_unlock();
dafc741c 1856
f0a3c086 1857 return ct;
c1d10adb 1858
0f5b3e85
PM		 1859err2:
1860 rcu_read_unlock();
1861err1:
c1d10adb 1862 nf_conntrack_free(ct);
f0a3c086 1863 return ERR_PTR(err);
c1d10adb
PNA		 1864}
1865
7b8002a1
PNA		 1866static int ctnetlink_new_conntrack(struct net *net, struct sock *ctnl,
1867 struct sk_buff *skb,
1868 const struct nlmsghdr *nlh,
1869 const struct nlattr * const cda[])
c1d10adb
PNA		 1870{
1871 struct nf_conntrack_tuple otuple, rtuple;
1872 struct nf_conntrack_tuple_hash *h = NULL;
96bcf938 1873 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
7d367e06 1874 struct nf_conn *ct;
c1d10adb 1875 u_int8_t u3 = nfmsg->nfgen_family;
308ac914 1876 struct nf_conntrack_zone zone;
ef00f89f
PM		 1877 int err;
1878
1879 err = ctnetlink_parse_zone(cda[CTA_ZONE], &zone);
1880 if (err < 0)
1881 return err;
c1d10adb 1882
df6fb868 1883 if (cda[CTA_TUPLE_ORIG]) {
deedb590
DB		 1884 err = ctnetlink_parse_tuple(cda, &otuple, CTA_TUPLE_ORIG,
1885 u3, &zone);
c1d10adb
PNA		 1886 if (err < 0)
1887 return err;
1888 }
1889
df6fb868 1890 if (cda[CTA_TUPLE_REPLY]) {
deedb590
DB		 1891 err = ctnetlink_parse_tuple(cda, &rtuple, CTA_TUPLE_REPLY,
1892 u3, &zone);
c1d10adb
PNA		 1893 if (err < 0)
1894 return err;
1895 }
1896
df6fb868 1897 if (cda[CTA_TUPLE_ORIG])
308ac914 1898 h = nf_conntrack_find_get(net, &zone, &otuple);
df6fb868 1899 else if (cda[CTA_TUPLE_REPLY])
308ac914 1900 h = nf_conntrack_find_get(net, &zone, &rtuple);
c1d10adb
PNA		 1901
1902 if (h == NULL) {
c1d10adb 1903 err = -ENOENT;
f0a3c086 1904 if (nlh->nlmsg_flags & NLM_F_CREATE) {
fecc1133 1905 enum ip_conntrack_events events;
5faa1f4c 1906
442fad94
FW		 1907 if (!cda[CTA_TUPLE_ORIG] || !cda[CTA_TUPLE_REPLY])
1908 return -EINVAL;
1909
308ac914 1910 ct = ctnetlink_create_conntrack(net, &zone, cda, &otuple,
f0a3c086 1911 &rtuple, u3);
7d367e06
JK		 1912 if (IS_ERR(ct))
1913 return PTR_ERR(ct);
1914
f0a3c086 1915 err = 0;
fecc1133
PNA		 1916 if (test_bit(IPS_EXPECTED_BIT, &ct->status))
1917 events = IPCT_RELATED;
1918 else
1919 events = IPCT_NEW;
1920
9b21f6a9
FW		 1921 if (cda[CTA_LABELS] &&
1922 ctnetlink_attach_labels(ct, cda) == 0)
1923 events |= (1 << IPCT_LABEL);
1924
858b3133
PM		 1925 nf_conntrack_eventmask_report((1 << IPCT_REPLY) |
1926 (1 << IPCT_ASSURED) |
a0891aa6
PNA		 1927 (1 << IPCT_HELPER) |
1928 (1 << IPCT_PROTOINFO) |
41d73ec0 1929 (1 << IPCT_SEQADJ) |
a0891aa6 1930 (1 << IPCT_MARK) | events,
15e47304 1931 ct, NETLINK_CB(skb).portid,
a0891aa6 1932 nlmsg_report(nlh));
f0a3c086 1933 nf_ct_put(ct);
7d367e06 1934 }
5faa1f4c 1935
c1d10adb
PNA		 1936 return err;
1937 }
1938 /* implicit 'else' */
1939
c1d10adb 1940 err = -EEXIST;
7d367e06 1941 ct = nf_ct_tuplehash_to_ctrack(h);
ff4ca827 1942 if (!(nlh->nlmsg_flags & NLM_F_EXCL)) {
ca7433df 1943 spin_lock_bh(&nf_conntrack_expect_lock);
19abb7b0 1944 err = ctnetlink_change_conntrack(ct, cda);
ca7433df 1945 spin_unlock_bh(&nf_conntrack_expect_lock);
19abb7b0 1946 if (err == 0) {
858b3133
PM		 1947 nf_conntrack_eventmask_report((1 << IPCT_REPLY) |
1948 (1 << IPCT_ASSURED) |
a0891aa6 1949 (1 << IPCT_HELPER) |
797a7d66 1950 (1 << IPCT_LABEL) |
a0891aa6 1951 (1 << IPCT_PROTOINFO) |
41d73ec0 1952 (1 << IPCT_SEQADJ) |
a0891aa6 1953 (1 << IPCT_MARK),
15e47304 1954 ct, NETLINK_CB(skb).portid,
a0891aa6 1955 nlmsg_report(nlh));
7d367e06 1956 }
ff4ca827 1957 }
c1d10adb 1958
7d367e06 1959 nf_ct_put(ct);
c1d10adb
PNA		 1960 return err;
1961}
1962
392025f8 1963static int
15e47304 1964ctnetlink_ct_stat_cpu_fill_info(struct sk_buff *skb, u32 portid, u32 seq,
392025f8
PNA		 1965 __u16 cpu, const struct ip_conntrack_stat *st)
1966{
1967 struct nlmsghdr *nlh;
1968 struct nfgenmsg *nfmsg;
15e47304 1969 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
392025f8
PNA		 1970
1971 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_CT_GET_STATS_CPU);
15e47304 1972 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
392025f8
PNA		 1973 if (nlh == NULL)
1974 goto nlmsg_failure;
1975
1976 nfmsg = nlmsg_data(nlh);
1977 nfmsg->nfgen_family = AF_UNSPEC;
1978 nfmsg->version = NFNETLINK_V0;
1979 nfmsg->res_id = htons(cpu);
1980
1981 if (nla_put_be32(skb, CTA_STATS_SEARCHED, htonl(st->searched)) ||
1982 nla_put_be32(skb, CTA_STATS_FOUND, htonl(st->found)) ||
1983 nla_put_be32(skb, CTA_STATS_NEW, htonl(st->new)) ||
1984 nla_put_be32(skb, CTA_STATS_INVALID, htonl(st->invalid)) ||
1985 nla_put_be32(skb, CTA_STATS_IGNORE, htonl(st->ignore)) ||
1986 nla_put_be32(skb, CTA_STATS_DELETE, htonl(st->delete)) ||
1987 nla_put_be32(skb, CTA_STATS_DELETE_LIST, htonl(st->delete_list)) ||
1988 nla_put_be32(skb, CTA_STATS_INSERT, htonl(st->insert)) ||
1989 nla_put_be32(skb, CTA_STATS_INSERT_FAILED,
1990 htonl(st->insert_failed)) ||
1991 nla_put_be32(skb, CTA_STATS_DROP, htonl(st->drop)) ||
1992 nla_put_be32(skb, CTA_STATS_EARLY_DROP, htonl(st->early_drop)) ||
1993 nla_put_be32(skb, CTA_STATS_ERROR, htonl(st->error)) ||
1994 nla_put_be32(skb, CTA_STATS_SEARCH_RESTART,
1995 htonl(st->search_restart)))
1996 goto nla_put_failure;
1997
1998 nlmsg_end(skb, nlh);
1999 return skb->len;
2000
2001nla_put_failure:
2002nlmsg_failure:
2003 nlmsg_cancel(skb, nlh);
2004 return -1;
2005}
2006
2007static int
2008ctnetlink_ct_stat_cpu_dump(struct sk_buff *skb, struct netlink_callback *cb)
2009{
2010 int cpu;
2011 struct net *net = sock_net(skb->sk);
2012
2013 if (cb->args[0] == nr_cpu_ids)
2014 return 0;
2015
2016 for (cpu = cb->args[0]; cpu < nr_cpu_ids; cpu++) {
2017 const struct ip_conntrack_stat *st;
2018
2019 if (!cpu_possible(cpu))
2020 continue;
2021
2022 st = per_cpu_ptr(net->ct.stat, cpu);
2023 if (ctnetlink_ct_stat_cpu_fill_info(skb,
15e47304 2024 NETLINK_CB(cb->skb).portid,
392025f8
PNA		 2025 cb->nlh->nlmsg_seq,
2026 cpu, st) < 0)
2027 break;
2028 }
2029 cb->args[0] = cpu;
2030
2031 return skb->len;
2032}
2033
7b8002a1
PNA		 2034static int ctnetlink_stat_ct_cpu(struct net *net, struct sock *ctnl,
2035 struct sk_buff *skb,
2036 const struct nlmsghdr *nlh,
2037 const struct nlattr * const cda[])
392025f8
PNA		 2038{
2039 if (nlh->nlmsg_flags & NLM_F_DUMP) {
2040 struct netlink_dump_control c = {
2041 .dump = ctnetlink_ct_stat_cpu_dump,
2042 };
2043 return netlink_dump_start(ctnl, skb, nlh, &c);
2044 }
2045
2046 return 0;
2047}
2048
2049static int
15e47304 2050ctnetlink_stat_ct_fill_info(struct sk_buff *skb, u32 portid, u32 seq, u32 type,
392025f8
PNA		 2051 struct net *net)
2052{
2053 struct nlmsghdr *nlh;
2054 struct nfgenmsg *nfmsg;
15e47304 2055 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
392025f8
PNA		 2056 unsigned int nr_conntracks = atomic_read(&net->ct.count);
2057
2058 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_CT_GET_STATS);
15e47304 2059 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
392025f8
PNA		 2060 if (nlh == NULL)
2061 goto nlmsg_failure;
2062
2063 nfmsg = nlmsg_data(nlh);
2064 nfmsg->nfgen_family = AF_UNSPEC;
2065 nfmsg->version = NFNETLINK_V0;
2066 nfmsg->res_id = 0;
2067
2068 if (nla_put_be32(skb, CTA_STATS_GLOBAL_ENTRIES, htonl(nr_conntracks)))
2069 goto nla_put_failure;
2070
2071 nlmsg_end(skb, nlh);
2072 return skb->len;
2073
2074nla_put_failure:
2075nlmsg_failure:
2076 nlmsg_cancel(skb, nlh);
2077 return -1;
2078}
2079
7b8002a1
PNA		 2080static int ctnetlink_stat_ct(struct net *net, struct sock *ctnl,
2081 struct sk_buff *skb, const struct nlmsghdr *nlh,
2082 const struct nlattr * const cda[])
392025f8
PNA		 2083{
2084 struct sk_buff *skb2;
2085 int err;
2086
2087 skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
2088 if (skb2 == NULL)
2089 return -ENOMEM;
2090
15e47304 2091 err = ctnetlink_stat_ct_fill_info(skb2, NETLINK_CB(skb).portid,
392025f8
PNA		 2092 nlh->nlmsg_seq,
2093 NFNL_MSG_TYPE(nlh->nlmsg_type),
2094 sock_net(skb->sk));
2095 if (err <= 0)
2096 goto free;
2097
15e47304 2098 err = netlink_unicast(ctnl, skb2, NETLINK_CB(skb).portid, MSG_DONTWAIT);
392025f8
PNA		 2099 if (err < 0)
2100 goto out;
2101
2102 return 0;
2103
2104free:
2105 kfree_skb(skb2);
2106out:
2107 /* this avoids a loop in nfnetlink. */
2108 return err == -EAGAIN ? -ENOBUFS : err;
2109}
2110
bd077937
PNA		 2111static const struct nla_policy exp_nla_policy[CTA_EXPECT_MAX+1] = {
2112 [CTA_EXPECT_MASTER] = { .type = NLA_NESTED },
2113 [CTA_EXPECT_TUPLE] = { .type = NLA_NESTED },
2114 [CTA_EXPECT_MASK] = { .type = NLA_NESTED },
2115 [CTA_EXPECT_TIMEOUT] = { .type = NLA_U32 },
2116 [CTA_EXPECT_ID] = { .type = NLA_U32 },
2117 [CTA_EXPECT_HELP_NAME] = { .type = NLA_NUL_STRING,
2118 .len = NF_CT_HELPER_NAME_LEN - 1 },
2119 [CTA_EXPECT_ZONE] = { .type = NLA_U16 },
2120 [CTA_EXPECT_FLAGS] = { .type = NLA_U32 },
2121 [CTA_EXPECT_CLASS] = { .type = NLA_U32 },
2122 [CTA_EXPECT_NAT] = { .type = NLA_NESTED },
2123 [CTA_EXPECT_FN] = { .type = NLA_NUL_STRING },
2124};
2125
2126static struct nf_conntrack_expect *
2127ctnetlink_alloc_expect(const struct nlattr *const cda[], struct nf_conn *ct,
2128 struct nf_conntrack_helper *helper,
2129 struct nf_conntrack_tuple *tuple,
2130 struct nf_conntrack_tuple *mask);
2131
83f3e94d 2132#ifdef CONFIG_NETFILTER_NETLINK_GLUE_CT
9cb01766 2133static size_t
a4b4766c 2134ctnetlink_glue_build_size(const struct nf_conn *ct)
9cb01766
PNA		 2135{
2136 return 3 * nla_total_size(0) /* CTA_TUPLE_ORIG|REPL|MASTER */
2137 + 3 * nla_total_size(0) /* CTA_TUPLE_IP */
2138 + 3 * nla_total_size(0) /* CTA_TUPLE_PROTO */
2139 + 3 * nla_total_size(sizeof(u_int8_t)) /* CTA_PROTO_NUM */
2140 + nla_total_size(sizeof(u_int32_t)) /* CTA_ID */
2141 + nla_total_size(sizeof(u_int32_t)) /* CTA_STATUS */
2142 + nla_total_size(sizeof(u_int32_t)) /* CTA_TIMEOUT */
2143 + nla_total_size(0) /* CTA_PROTOINFO */
2144 + nla_total_size(0) /* CTA_HELP */
2145 + nla_total_size(NF_CT_HELPER_NAME_LEN) /* CTA_HELP_NAME */
2146 + ctnetlink_secctx_size(ct)
2147#ifdef CONFIG_NF_NAT_NEEDED
2148 + 2 * nla_total_size(0) /* CTA_NAT_SEQ_ADJ_ORIG|REPL */
2149 + 6 * nla_total_size(sizeof(u_int32_t)) /* CTA_NAT_SEQ_OFFSET */
2150#endif
2151#ifdef CONFIG_NF_CONNTRACK_MARK
2152 + nla_total_size(sizeof(u_int32_t)) /* CTA_MARK */
4a001068
KM		 2153#endif
2154#ifdef CONFIG_NF_CONNTRACK_ZONES
deedb590 2155 + nla_total_size(sizeof(u_int16_t)) /* CTA_ZONE|CTA_TUPLE_ZONE */
9cb01766
PNA		 2156#endif
2157 + ctnetlink_proto_size(ct)
2158 ;
2159}
2160
224a0597 2161static struct nf_conn *ctnetlink_glue_get_ct(const struct sk_buff *skb,
a4b4766c 2162 enum ip_conntrack_info *ctinfo)
b7bd1809
PNA		 2163{
2164 struct nf_conn *ct;
2165
2166 ct = nf_ct_get(skb, ctinfo);
2167 if (ct && nf_ct_is_untracked(ct))
2168 ct = NULL;
2169
2170 return ct;
2171}
2172
a4b4766c 2173static int __ctnetlink_glue_build(struct sk_buff *skb, struct nf_conn *ct)
9cb01766 2174{
308ac914 2175 const struct nf_conntrack_zone *zone;
9cb01766
PNA		 2176 struct nlattr *nest_parms;
2177
2178 rcu_read_lock();
deedb590
DB		 2179 zone = nf_ct_zone(ct);
2180
9cb01766
PNA		 2181 nest_parms = nla_nest_start(skb, CTA_TUPLE_ORIG | NLA_F_NESTED);
2182 if (!nest_parms)
2183 goto nla_put_failure;
2184 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_ORIGINAL)) < 0)
2185 goto nla_put_failure;
deedb590
DB		 2186 if (ctnetlink_dump_zone_id(skb, CTA_TUPLE_ZONE, zone,
2187 NF_CT_ZONE_DIR_ORIG) < 0)
2188 goto nla_put_failure;
9cb01766
PNA		 2189 nla_nest_end(skb, nest_parms);
2190
2191 nest_parms = nla_nest_start(skb, CTA_TUPLE_REPLY | NLA_F_NESTED);
2192 if (!nest_parms)
2193 goto nla_put_failure;
2194 if (ctnetlink_dump_tuples(skb, nf_ct_tuple(ct, IP_CT_DIR_REPLY)) < 0)
2195 goto nla_put_failure;
deedb590
DB		 2196 if (ctnetlink_dump_zone_id(skb, CTA_TUPLE_ZONE, zone,
2197 NF_CT_ZONE_DIR_REPL) < 0)
2198 goto nla_put_failure;
9cb01766
PNA		 2199 nla_nest_end(skb, nest_parms);
2200
deedb590
DB		 2201 if (ctnetlink_dump_zone_id(skb, CTA_ZONE, zone,
2202 NF_CT_DEFAULT_ZONE_DIR) < 0)
308ac914 2203 goto nla_put_failure;
9cb01766
PNA		 2204
2205 if (ctnetlink_dump_id(skb, ct) < 0)
2206 goto nla_put_failure;
2207
2208 if (ctnetlink_dump_status(skb, ct) < 0)
2209 goto nla_put_failure;
2210
2211 if (ctnetlink_dump_timeout(skb, ct) < 0)
2212 goto nla_put_failure;
2213
2214 if (ctnetlink_dump_protoinfo(skb, ct) < 0)
2215 goto nla_put_failure;
2216
2217 if (ctnetlink_dump_helpinfo(skb, ct) < 0)
2218 goto nla_put_failure;
2219
2220#ifdef CONFIG_NF_CONNTRACK_SECMARK
2221 if (ct->secmark && ctnetlink_dump_secctx(skb, ct) < 0)
2222 goto nla_put_failure;
2223#endif
2224 if (ct->master && ctnetlink_dump_master(skb, ct) < 0)
2225 goto nla_put_failure;
2226
2227 if ((ct->status & IPS_SEQ_ADJUST) &&
41d73ec0 2228 ctnetlink_dump_ct_seq_adj(skb, ct) < 0)
9cb01766
PNA		 2229 goto nla_put_failure;
2230
2231#ifdef CONFIG_NF_CONNTRACK_MARK
2232 if (ct->mark && ctnetlink_dump_mark(skb, ct) < 0)
2233 goto nla_put_failure;
2234#endif
0ceabd83
FW		 2235 if (ctnetlink_dump_labels(skb, ct) < 0)
2236 goto nla_put_failure;
9cb01766
PNA		 2237 rcu_read_unlock();
2238 return 0;
2239
2240nla_put_failure:
2241 rcu_read_unlock();
2242 return -ENOSPC;
2243}
2244
b7bd1809 2245static int
a4b4766c
KM		 2246ctnetlink_glue_build(struct sk_buff *skb, struct nf_conn *ct,
2247 enum ip_conntrack_info ctinfo,
2248 u_int16_t ct_attr, u_int16_t ct_info_attr)
b7bd1809
PNA		 2249{
2250 struct nlattr *nest_parms;
2251
2252 nest_parms = nla_nest_start(skb, ct_attr | NLA_F_NESTED);
2253 if (!nest_parms)
2254 goto nla_put_failure;
2255
a4b4766c 2256 if (__ctnetlink_glue_build(skb, ct) < 0)
b7bd1809
PNA		 2257 goto nla_put_failure;
2258
2259 nla_nest_end(skb, nest_parms);
2260
2261 if (nla_put_be32(skb, ct_info_attr, htonl(ctinfo)))
2262 goto nla_put_failure;
2263
2264 return 0;
2265
2266nla_put_failure:
2267 return -ENOSPC;
2268}
2269
9cb01766 2270static int
a4b4766c 2271ctnetlink_glue_parse_ct(const struct nlattr *cda[], struct nf_conn *ct)
9cb01766
PNA		 2272{
2273 int err;
2274
2275 if (cda[CTA_TIMEOUT]) {
2276 err = ctnetlink_change_timeout(ct, cda);
2277 if (err < 0)
2278 return err;
2279 }
2280 if (cda[CTA_STATUS]) {
2281 err = ctnetlink_change_status(ct, cda);
2282 if (err < 0)
2283 return err;
2284 }
2285 if (cda[CTA_HELP]) {
2286 err = ctnetlink_change_helper(ct, cda);
2287 if (err < 0)
2288 return err;
2289 }
9b21f6a9
FW		 2290 if (cda[CTA_LABELS]) {
2291 err = ctnetlink_attach_labels(ct, cda);
2292 if (err < 0)
2293 return err;
2294 }
9cb01766 2295#if defined(CONFIG_NF_CONNTRACK_MARK)
534473c6
FW		 2296 if (cda[CTA_MARK]) {
2297 u32 mask = 0, mark, newmark;
2298 if (cda[CTA_MARK_MASK])
2299 mask = ~ntohl(nla_get_be32(cda[CTA_MARK_MASK]));
2300
2301 mark = ntohl(nla_get_be32(cda[CTA_MARK]));
2302 newmark = (ct->mark & mask) ^ mark;
2303 if (newmark != ct->mark)
2304 ct->mark = newmark;
2305 }
9cb01766
PNA		 2306#endif
2307 return 0;
2308}
2309
2310static int
a4b4766c 2311ctnetlink_glue_parse(const struct nlattr *attr, struct nf_conn *ct)
9cb01766
PNA		 2312{
2313 struct nlattr *cda[CTA_MAX+1];
68e035c9 2314 int ret;
9cb01766 2315
130ffbc2
DB		 2316 ret = nla_parse_nested(cda, CTA_MAX, attr, ct_nla_policy);
2317 if (ret < 0)
2318 return ret;
9cb01766 2319
ca7433df 2320 spin_lock_bh(&nf_conntrack_expect_lock);
a4b4766c 2321 ret = ctnetlink_glue_parse_ct((const struct nlattr **)cda, ct);
ca7433df 2322 spin_unlock_bh(&nf_conntrack_expect_lock);
68e035c9
PNA		 2323
2324 return ret;
9cb01766
PNA		 2325}
2326
a4b4766c
KM		 2327static int ctnetlink_glue_exp_parse(const struct nlattr * const *cda,
2328 const struct nf_conn *ct,
2329 struct nf_conntrack_tuple *tuple,
2330 struct nf_conntrack_tuple *mask)
bd077937
PNA		 2331{
2332 int err;
2333
2334 err = ctnetlink_parse_tuple(cda, tuple, CTA_EXPECT_TUPLE,
deedb590 2335 nf_ct_l3num(ct), NULL);
bd077937
PNA		 2336 if (err < 0)
2337 return err;
2338
2339 return ctnetlink_parse_tuple(cda, mask, CTA_EXPECT_MASK,
deedb590 2340 nf_ct_l3num(ct), NULL);
bd077937
PNA		 2341}
2342
2343static int
a4b4766c
KM		 2344ctnetlink_glue_attach_expect(const struct nlattr *attr, struct nf_conn *ct,
2345 u32 portid, u32 report)
bd077937
PNA		 2346{
2347 struct nlattr *cda[CTA_EXPECT_MAX+1];
2348 struct nf_conntrack_tuple tuple, mask;
b7e092c0 2349 struct nf_conntrack_helper *helper = NULL;
bd077937
PNA		 2350 struct nf_conntrack_expect *exp;
2351 int err;
2352
2353 err = nla_parse_nested(cda, CTA_EXPECT_MAX, attr, exp_nla_policy);
2354 if (err < 0)
2355 return err;
2356
a4b4766c
KM		 2357 err = ctnetlink_glue_exp_parse((const struct nlattr * const *)cda,
2358 ct, &tuple, &mask);
bd077937
PNA		 2359 if (err < 0)
2360 return err;
2361
2362 if (cda[CTA_EXPECT_HELP_NAME]) {
2363 const char *helpname = nla_data(cda[CTA_EXPECT_HELP_NAME]);
2364
2365 helper = __nf_conntrack_helper_find(helpname, nf_ct_l3num(ct),
2366 nf_ct_protonum(ct));
2367 if (helper == NULL)
2368 return -EOPNOTSUPP;
2369 }
2370
2371 exp = ctnetlink_alloc_expect((const struct nlattr * const *)cda, ct,
2372 helper, &tuple, &mask);
2373 if (IS_ERR(exp))
2374 return PTR_ERR(exp);
2375
2376 err = nf_ct_expect_related_report(exp, portid, report);
2377 if (err < 0) {
2378 nf_ct_expect_put(exp);
2379 return err;
2380 }
2381
2382 return 0;
2383}
2384
a4b4766c
KM		 2385static void ctnetlink_glue_seqadj(struct sk_buff *skb, struct nf_conn *ct,
2386 enum ip_conntrack_info ctinfo, int diff)
b7bd1809
PNA		 2387{
2388 if (!(ct->status & IPS_NAT_MASK))
2389 return;
2390
2391 nf_ct_tcp_seqadj_set(skb, ct, ctinfo, diff);
2392}
2393
a4b4766c
KM		 2394static struct nfnl_ct_hook ctnetlink_glue_hook = {
2395 .get_ct = ctnetlink_glue_get_ct,
2396 .build_size = ctnetlink_glue_build_size,
2397 .build = ctnetlink_glue_build,
2398 .parse = ctnetlink_glue_parse,
2399 .attach_expect = ctnetlink_glue_attach_expect,
2400 .seq_adjust = ctnetlink_glue_seqadj,
9cb01766 2401};
83f3e94d 2402#endif /* CONFIG_NETFILTER_NETLINK_GLUE_CT */
9cb01766 2403
601e68e1
YH		 2404/***********************************************************************
2405 * EXPECT
2406 ***********************************************************************/
c1d10adb
PNA		 2407
2408static inline int
2409ctnetlink_exp_dump_tuple(struct sk_buff *skb,
2410 const struct nf_conntrack_tuple *tuple,
2411 enum ctattr_expect type)
2412{
df6fb868 2413 struct nlattr *nest_parms;
601e68e1 2414
df6fb868
PM		 2415 nest_parms = nla_nest_start(skb, type | NLA_F_NESTED);
2416 if (!nest_parms)
2417 goto nla_put_failure;
c1d10adb 2418 if (ctnetlink_dump_tuples(skb, tuple) < 0)
df6fb868
PM		 2419 goto nla_put_failure;
2420 nla_nest_end(skb, nest_parms);
c1d10adb
PNA		 2421
2422 return 0;
2423
df6fb868 2424nla_put_failure:
c1d10adb 2425 return -1;
601e68e1 2426}
c1d10adb 2427
1cde6436
PNA		 2428static inline int
2429ctnetlink_exp_dump_mask(struct sk_buff *skb,
2430 const struct nf_conntrack_tuple *tuple,
d4156e8c 2431 const struct nf_conntrack_tuple_mask *mask)
1cde6436
PNA		 2432{
2433 int ret;
2434 struct nf_conntrack_l3proto *l3proto;
605dcad6 2435 struct nf_conntrack_l4proto *l4proto;
d4156e8c 2436 struct nf_conntrack_tuple m;
df6fb868 2437 struct nlattr *nest_parms;
d4156e8c
PM		 2438
2439 memset(&m, 0xFF, sizeof(m));
d4156e8c 2440 memcpy(&m.src.u3, &mask->src.u3, sizeof(m.src.u3));
e578756c
PM		 2441 m.src.u.all = mask->src.u.all;
2442 m.dst.protonum = tuple->dst.protonum;
d4156e8c 2443
df6fb868
PM		 2444 nest_parms = nla_nest_start(skb, CTA_EXPECT_MASK | NLA_F_NESTED);
2445 if (!nest_parms)
2446 goto nla_put_failure;
1cde6436 2447
3b988ece 2448 rcu_read_lock();
528a3a6f 2449 l3proto = __nf_ct_l3proto_find(tuple->src.l3num);
d4156e8c 2450 ret = ctnetlink_dump_tuples_ip(skb, &m, l3proto);
3b988ece
HS		 2451 if (ret >= 0) {
2452 l4proto = __nf_ct_l4proto_find(tuple->src.l3num,
2453 tuple->dst.protonum);
d4156e8c 2454 ret = ctnetlink_dump_tuples_proto(skb, &m, l4proto);
3b988ece
HS		 2455 }
2456 rcu_read_unlock();
2457
1cde6436 2458 if (unlikely(ret < 0))
df6fb868 2459 goto nla_put_failure;
1cde6436 2460
df6fb868 2461 nla_nest_end(skb, nest_parms);
1cde6436
PNA		 2462
2463 return 0;
2464
df6fb868 2465nla_put_failure:
1cde6436
PNA		 2466 return -1;
2467}
2468
c7232c99
PM		 2469static const union nf_inet_addr any_addr;
2470
bb5cf80e 2471static int
c1d10adb 2472ctnetlink_exp_dump_expect(struct sk_buff *skb,
601e68e1 2473 const struct nf_conntrack_expect *exp)
c1d10adb
PNA		 2474{
2475 struct nf_conn *master = exp->master;
c1216382 2476 long timeout = ((long)exp->timeout.expires - (long)jiffies) / HZ;
bc01befd 2477 struct nf_conn_help *help;
076a0ca0
PNA		 2478#ifdef CONFIG_NF_NAT_NEEDED
2479 struct nlattr *nest_parms;
2480 struct nf_conntrack_tuple nat_tuple = {};
2481#endif
544d5c7d
PNA		 2482 struct nf_ct_helper_expectfn *expfn;
2483
d978e5da
PM		 2484 if (timeout < 0)
2485 timeout = 0;
c1d10adb
PNA		 2486
2487 if (ctnetlink_exp_dump_tuple(skb, &exp->tuple, CTA_EXPECT_TUPLE) < 0)
df6fb868 2488 goto nla_put_failure;
1cde6436 2489 if (ctnetlink_exp_dump_mask(skb, &exp->tuple, &exp->mask) < 0)
df6fb868 2490 goto nla_put_failure;
c1d10adb
PNA		 2491 if (ctnetlink_exp_dump_tuple(skb,
2492 &master->tuplehash[IP_CT_DIR_ORIGINAL].tuple,
2493 CTA_EXPECT_MASTER) < 0)
df6fb868 2494 goto nla_put_failure;
601e68e1 2495
076a0ca0 2496#ifdef CONFIG_NF_NAT_NEEDED
c7232c99
PM		 2497 if (!nf_inet_addr_cmp(&exp->saved_addr, &any_addr) ||
2498 exp->saved_proto.all) {
076a0ca0
PNA		 2499 nest_parms = nla_nest_start(skb, CTA_EXPECT_NAT | NLA_F_NESTED);
2500 if (!nest_parms)
2501 goto nla_put_failure;
2502
cc1eb431
DM		 2503 if (nla_put_be32(skb, CTA_EXPECT_NAT_DIR, htonl(exp->dir)))
2504 goto nla_put_failure;
076a0ca0
PNA		 2505
2506 nat_tuple.src.l3num = nf_ct_l3num(master);
c7232c99 2507 nat_tuple.src.u3 = exp->saved_addr;
076a0ca0
PNA		 2508 nat_tuple.dst.protonum = nf_ct_protonum(master);
2509 nat_tuple.src.u = exp->saved_proto;
2510
2511 if (ctnetlink_exp_dump_tuple(skb, &nat_tuple,
2512 CTA_EXPECT_NAT_TUPLE) < 0)
2513 goto nla_put_failure;
2514 nla_nest_end(skb, nest_parms);
2515 }
2516#endif
cc1eb431
DM		 2517 if (nla_put_be32(skb, CTA_EXPECT_TIMEOUT, htonl(timeout)) ||
2518 nla_put_be32(skb, CTA_EXPECT_ID, htonl((unsigned long)exp)) ||
2519 nla_put_be32(skb, CTA_EXPECT_FLAGS, htonl(exp->flags)) ||
2520 nla_put_be32(skb, CTA_EXPECT_CLASS, htonl(exp->class)))
2521 goto nla_put_failure;
bc01befd
PNA		 2522 help = nfct_help(master);
2523 if (help) {
2524 struct nf_conntrack_helper *helper;
2525
2526 helper = rcu_dereference(help->helper);
cc1eb431
DM		 2527 if (helper &&
2528 nla_put_string(skb, CTA_EXPECT_HELP_NAME, helper->name))
2529 goto nla_put_failure;
bc01befd 2530 }
544d5c7d 2531 expfn = nf_ct_helper_expectfn_find_by_symbol(exp->expectfn);
cc1eb431
DM		 2532 if (expfn != NULL &&
2533 nla_put_string(skb, CTA_EXPECT_FN, expfn->name))
2534 goto nla_put_failure;
c1d10adb
PNA		 2535
2536 return 0;
601e68e1 2537
df6fb868 2538nla_put_failure:
c1d10adb
PNA		 2539 return -1;
2540}
2541
2542static int
15e47304 2543ctnetlink_exp_fill_info(struct sk_buff *skb, u32 portid, u32 seq,
8b0a231d 2544 int event, const struct nf_conntrack_expect *exp)
c1d10adb
PNA		 2545{
2546 struct nlmsghdr *nlh;
2547 struct nfgenmsg *nfmsg;
15e47304 2548 unsigned int flags = portid ? NLM_F_MULTI : 0;
c1d10adb
PNA		 2549
2550 event |= NFNL_SUBSYS_CTNETLINK_EXP << 8;
15e47304 2551 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
96bcf938
PNA		 2552 if (nlh == NULL)
2553 goto nlmsg_failure;
c1d10adb 2554
96bcf938 2555 nfmsg = nlmsg_data(nlh);
c1d10adb
PNA		 2556 nfmsg->nfgen_family = exp->tuple.src.l3num;
2557 nfmsg->version = NFNETLINK_V0;
2558 nfmsg->res_id = 0;
2559
2560 if (ctnetlink_exp_dump_expect(skb, exp) < 0)
df6fb868 2561 goto nla_put_failure;
c1d10adb 2562
96bcf938 2563 nlmsg_end(skb, nlh);
c1d10adb
PNA		 2564 return skb->len;
2565
2566nlmsg_failure:
df6fb868 2567nla_put_failure:
96bcf938 2568 nlmsg_cancel(skb, nlh);
c1d10adb
PNA		 2569 return -1;
2570}
2571
2572#ifdef CONFIG_NF_CONNTRACK_EVENTS
e34d5c1a
PNA		 2573static int
2574ctnetlink_expect_event(unsigned int events, struct nf_exp_event *item)
c1d10adb 2575{
9592a5c0
AD		 2576 struct nf_conntrack_expect *exp = item->exp;
2577 struct net *net = nf_ct_exp_net(exp);
c1d10adb
PNA		 2578 struct nlmsghdr *nlh;
2579 struct nfgenmsg *nfmsg;
c1d10adb 2580 struct sk_buff *skb;
ebbf41df 2581 unsigned int type, group;
c1d10adb
PNA		 2582 int flags = 0;
2583
ebbf41df
PNA		 2584 if (events & (1 << IPEXP_DESTROY)) {
2585 type = IPCTNL_MSG_EXP_DELETE;
2586 group = NFNLGRP_CONNTRACK_EXP_DESTROY;
2587 } else if (events & (1 << IPEXP_NEW)) {
c1d10adb
PNA		 2588 type = IPCTNL_MSG_EXP_NEW;
2589 flags = NLM_F_CREATE|NLM_F_EXCL;
ebbf41df 2590 group = NFNLGRP_CONNTRACK_EXP_NEW;
c1d10adb 2591 } else
e34d5c1a 2592 return 0;
c1d10adb 2593
ebbf41df 2594 if (!item->report && !nfnetlink_has_listeners(net, group))
e34d5c1a 2595 return 0;
b3a27bfb 2596
96bcf938
PNA		 2597 skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_ATOMIC);
2598 if (skb == NULL)
150ace0d 2599 goto errout;
c1d10adb 2600
b633ad5f 2601 type |= NFNL_SUBSYS_CTNETLINK_EXP << 8;
15e47304 2602 nlh = nlmsg_put(skb, item->portid, 0, type, sizeof(*nfmsg), flags);
96bcf938
PNA		 2603 if (nlh == NULL)
2604 goto nlmsg_failure;
c1d10adb 2605
96bcf938 2606 nfmsg = nlmsg_data(nlh);
c1d10adb
PNA		 2607 nfmsg->nfgen_family = exp->tuple.src.l3num;
2608 nfmsg->version = NFNETLINK_V0;
2609 nfmsg->res_id = 0;
2610
528a3a6f 2611 rcu_read_lock();
c1d10adb 2612 if (ctnetlink_exp_dump_expect(skb, exp) < 0)
df6fb868 2613 goto nla_put_failure;
528a3a6f 2614 rcu_read_unlock();
c1d10adb 2615
96bcf938 2616 nlmsg_end(skb, nlh);
15e47304 2617 nfnetlink_send(skb, net, item->portid, group, item->report, GFP_ATOMIC);
e34d5c1a 2618 return 0;
c1d10adb 2619
df6fb868 2620nla_put_failure:
528a3a6f 2621 rcu_read_unlock();
96bcf938 2622 nlmsg_cancel(skb, nlh);
528a3a6f 2623nlmsg_failure:
c1d10adb 2624 kfree_skb(skb);
150ace0d 2625errout:
9592a5c0 2626 nfnetlink_set_err(net, 0, 0, -ENOBUFS);
e34d5c1a 2627 return 0;
c1d10adb
PNA		 2628}
2629#endif
cf6994c2
PM		 2630static int ctnetlink_exp_done(struct netlink_callback *cb)
2631{
31f15875
PM		 2632 if (cb->args[1])
2633 nf_ct_expect_put((struct nf_conntrack_expect *)cb->args[1]);
cf6994c2
PM		 2634 return 0;
2635}
c1d10adb
PNA		 2636
2637static int
2638ctnetlink_exp_dump_table(struct sk_buff *skb, struct netlink_callback *cb)
2639{
9592a5c0 2640 struct net *net = sock_net(skb->sk);
cf6994c2 2641 struct nf_conntrack_expect *exp, *last;
96bcf938 2642 struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
87711cb8 2643 u_int8_t l3proto = nfmsg->nfgen_family;
c1d10adb 2644
7d0742da 2645 rcu_read_lock();
31f15875
PM		 2646 last = (struct nf_conntrack_expect *)cb->args[1];
2647 for (; cb->args[0] < nf_ct_expect_hsize; cb->args[0]++) {
cf6994c2 2648restart:
b67bfe0d 2649 hlist_for_each_entry(exp, &net->ct.expect_hash[cb->args[0]],
31f15875
PM		 2650 hnode) {
2651 if (l3proto && exp->tuple.src.l3num != l3proto)
cf6994c2 2652 continue;
31f15875
PM		 2653 if (cb->args[1]) {
2654 if (exp != last)
2655 continue;
2656 cb->args[1] = 0;
2657 }
8b0a231d 2658 if (ctnetlink_exp_fill_info(skb,
15e47304 2659 NETLINK_CB(cb->skb).portid,
31f15875
PM		 2660 cb->nlh->nlmsg_seq,
2661 IPCTNL_MSG_EXP_NEW,
8b0a231d 2662 exp) < 0) {
7d0742da
PM		 2663 if (!atomic_inc_not_zero(&exp->use))
2664 continue;
31f15875
PM		 2665 cb->args[1] = (unsigned long)exp;
2666 goto out;
2667 }
cf6994c2 2668 }
31f15875
PM		 2669 if (cb->args[1]) {
2670 cb->args[1] = 0;
2671 goto restart;
cf6994c2
PM		 2672 }
2673 }
601e68e1 2674out:
7d0742da 2675 rcu_read_unlock();
cf6994c2
PM		 2676 if (last)
2677 nf_ct_expect_put(last);
c1d10adb 2678
c1d10adb
PNA		 2679 return skb->len;
2680}
2681
e844a928
PNA		 2682static int
2683ctnetlink_exp_ct_dump_table(struct sk_buff *skb, struct netlink_callback *cb)
2684{
2685 struct nf_conntrack_expect *exp, *last;
2686 struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
2687 struct nf_conn *ct = cb->data;
2688 struct nf_conn_help *help = nfct_help(ct);
2689 u_int8_t l3proto = nfmsg->nfgen_family;
2690
2691 if (cb->args[0])
2692 return 0;
2693
2694 rcu_read_lock();
2695 last = (struct nf_conntrack_expect *)cb->args[1];
2696restart:
2697 hlist_for_each_entry(exp, &help->expectations, lnode) {
2698 if (l3proto && exp->tuple.src.l3num != l3proto)
2699 continue;
2700 if (cb->args[1]) {
2701 if (exp != last)
2702 continue;
2703 cb->args[1] = 0;
2704 }
2705 if (ctnetlink_exp_fill_info(skb, NETLINK_CB(cb->skb).portid,
2706 cb->nlh->nlmsg_seq,
2707 IPCTNL_MSG_EXP_NEW,
2708 exp) < 0) {
2709 if (!atomic_inc_not_zero(&exp->use))
2710 continue;
2711 cb->args[1] = (unsigned long)exp;
2712 goto out;
2713 }
2714 }
2715 if (cb->args[1]) {
2716 cb->args[1] = 0;
2717 goto restart;
2718 }
2719 cb->args[0] = 1;
2720out:
2721 rcu_read_unlock();
2722 if (last)
2723 nf_ct_expect_put(last);
2724
2725 return skb->len;
2726}
2727
7b8002a1
PNA		 2728static int ctnetlink_dump_exp_ct(struct net *net, struct sock *ctnl,
2729 struct sk_buff *skb,
e844a928
PNA		 2730 const struct nlmsghdr *nlh,
2731 const struct nlattr * const cda[])
2732{
2733 int err;
e844a928
PNA		 2734 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2735 u_int8_t u3 = nfmsg->nfgen_family;
2736 struct nf_conntrack_tuple tuple;
2737 struct nf_conntrack_tuple_hash *h;
2738 struct nf_conn *ct;
308ac914 2739 struct nf_conntrack_zone zone;
e844a928
PNA		 2740 struct netlink_dump_control c = {
2741 .dump = ctnetlink_exp_ct_dump_table,
2742 .done = ctnetlink_exp_done,
2743 };
2744
deedb590
DB		 2745 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_MASTER,
2746 u3, NULL);
e844a928
PNA		 2747 if (err < 0)
2748 return err;
2749
308ac914
DB		 2750 err = ctnetlink_parse_zone(cda[CTA_EXPECT_ZONE], &zone);
2751 if (err < 0)
2752 return err;
e844a928 2753
308ac914 2754 h = nf_conntrack_find_get(net, &zone, &tuple);
e844a928
PNA		 2755 if (!h)
2756 return -ENOENT;
2757
2758 ct = nf_ct_tuplehash_to_ctrack(h);
2759 c.data = ct;
2760
2761 err = netlink_dump_start(ctnl, skb, nlh, &c);
2762 nf_ct_put(ct);
2763
2764 return err;
2765}
2766
7b8002a1
PNA		 2767static int ctnetlink_get_expect(struct net *net, struct sock *ctnl,
2768 struct sk_buff *skb, const struct nlmsghdr *nlh,
2769 const struct nlattr * const cda[])
c1d10adb
PNA		 2770{
2771 struct nf_conntrack_tuple tuple;
2772 struct nf_conntrack_expect *exp;
2773 struct sk_buff *skb2;
96bcf938 2774 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
c1d10adb 2775 u_int8_t u3 = nfmsg->nfgen_family;
308ac914 2776 struct nf_conntrack_zone zone;
ef00f89f 2777 int err;
c1d10adb 2778
b8f3ab42 2779 if (nlh->nlmsg_flags & NLM_F_DUMP) {
e844a928 2780 if (cda[CTA_EXPECT_MASTER])
7b8002a1 2781 return ctnetlink_dump_exp_ct(net, ctnl, skb, nlh, cda);
e844a928
PNA		 2782 else {
2783 struct netlink_dump_control c = {
2784 .dump = ctnetlink_exp_dump_table,
2785 .done = ctnetlink_exp_done,
2786 };
2787 return netlink_dump_start(ctnl, skb, nlh, &c);
2788 }
c1d10adb
PNA		 2789 }
2790
ef00f89f
PM		 2791 err = ctnetlink_parse_zone(cda[CTA_EXPECT_ZONE], &zone);
2792 if (err < 0)
2793 return err;
2794
35dba1d7 2795 if (cda[CTA_EXPECT_TUPLE])
deedb590
DB		 2796 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE,
2797 u3, NULL);
35dba1d7 2798 else if (cda[CTA_EXPECT_MASTER])
deedb590
DB		 2799 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_MASTER,
2800 u3, NULL);
c1d10adb
PNA		 2801 else
2802 return -EINVAL;
2803
2804 if (err < 0)
2805 return err;
2806
308ac914 2807 exp = nf_ct_expect_find_get(net, &zone, &tuple);
c1d10adb
PNA		 2808 if (!exp)
2809 return -ENOENT;
2810
df6fb868 2811 if (cda[CTA_EXPECT_ID]) {
77236b6e 2812 __be32 id = nla_get_be32(cda[CTA_EXPECT_ID]);
35832402 2813 if (ntohl(id) != (u32)(unsigned long)exp) {
6823645d 2814 nf_ct_expect_put(exp);
c1d10adb
PNA		 2815 return -ENOENT;
2816 }
601e68e1 2817 }
c1d10adb
PNA		 2818
2819 err = -ENOMEM;
96bcf938 2820 skb2 = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
81378f72
PNA		 2821 if (skb2 == NULL) {
2822 nf_ct_expect_put(exp);
c1d10adb 2823 goto out;
81378f72 2824 }
4e9b8269 2825
528a3a6f 2826 rcu_read_lock();
15e47304 2827 err = ctnetlink_exp_fill_info(skb2, NETLINK_CB(skb).portid,
8b0a231d 2828 nlh->nlmsg_seq, IPCTNL_MSG_EXP_NEW, exp);
528a3a6f 2829 rcu_read_unlock();
81378f72 2830 nf_ct_expect_put(exp);
c1d10adb
PNA		 2831 if (err <= 0)
2832 goto free;
2833
15e47304 2834 err = netlink_unicast(ctnl, skb2, NETLINK_CB(skb).portid, MSG_DONTWAIT);
81378f72
PNA		 2835 if (err < 0)
2836 goto out;
c1d10adb 2837
81378f72 2838 return 0;
c1d10adb
PNA		 2839
2840free:
2841 kfree_skb(skb2);
2842out:
81378f72
PNA		 2843 /* this avoids a loop in nfnetlink. */
2844 return err == -EAGAIN ? -ENOBUFS : err;
c1d10adb
PNA		 2845}
2846
7b8002a1
PNA		 2847static int ctnetlink_del_expect(struct net *net, struct sock *ctnl,
2848 struct sk_buff *skb, const struct nlmsghdr *nlh,
2849 const struct nlattr * const cda[])
c1d10adb 2850{
31f15875 2851 struct nf_conntrack_expect *exp;
c1d10adb 2852 struct nf_conntrack_tuple tuple;
96bcf938 2853 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
b67bfe0d 2854 struct hlist_node *next;
c1d10adb 2855 u_int8_t u3 = nfmsg->nfgen_family;
308ac914 2856 struct nf_conntrack_zone zone;
31f15875 2857 unsigned int i;
c1d10adb
PNA		 2858 int err;
2859
df6fb868 2860 if (cda[CTA_EXPECT_TUPLE]) {
c1d10adb 2861 /* delete a single expect by tuple */
ef00f89f
PM		 2862 err = ctnetlink_parse_zone(cda[CTA_EXPECT_ZONE], &zone);
2863 if (err < 0)
2864 return err;
2865
deedb590
DB		 2866 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE,
2867 u3, NULL);
c1d10adb
PNA		 2868 if (err < 0)
2869 return err;
2870
2871 /* bump usage count to 2 */
308ac914 2872 exp = nf_ct_expect_find_get(net, &zone, &tuple);
c1d10adb
PNA		 2873 if (!exp)
2874 return -ENOENT;
2875
df6fb868 2876 if (cda[CTA_EXPECT_ID]) {
77236b6e 2877 __be32 id = nla_get_be32(cda[CTA_EXPECT_ID]);
35832402 2878 if (ntohl(id) != (u32)(unsigned long)exp) {
6823645d 2879 nf_ct_expect_put(exp);
c1d10adb
PNA		 2880 return -ENOENT;
2881 }
2882 }
2883
2884 /* after list removal, usage count == 1 */
ca7433df 2885 spin_lock_bh(&nf_conntrack_expect_lock);
ebbf41df 2886 if (del_timer(&exp->timeout)) {
15e47304 2887 nf_ct_unlink_expect_report(exp, NETLINK_CB(skb).portid,
ebbf41df
PNA		 2888 nlmsg_report(nlh));
2889 nf_ct_expect_put(exp);
2890 }
ca7433df 2891 spin_unlock_bh(&nf_conntrack_expect_lock);
601e68e1 2892 /* have to put what we 'get' above.
c1d10adb 2893 * after this line usage count == 0 */
6823645d 2894 nf_ct_expect_put(exp);
df6fb868
PM		 2895 } else if (cda[CTA_EXPECT_HELP_NAME]) {
2896 char *name = nla_data(cda[CTA_EXPECT_HELP_NAME]);
31f15875 2897 struct nf_conn_help *m_help;
c1d10adb
PNA		 2898
2899 /* delete all expectations for this helper */
ca7433df 2900 spin_lock_bh(&nf_conntrack_expect_lock);
31f15875 2901 for (i = 0; i < nf_ct_expect_hsize; i++) {
b67bfe0d 2902 hlist_for_each_entry_safe(exp, next,
9592a5c0 2903 &net->ct.expect_hash[i],
31f15875
PM		 2904 hnode) {
2905 m_help = nfct_help(exp->master);
794e6871
PM		 2906 if (!strcmp(m_help->helper->name, name) &&
2907 del_timer(&exp->timeout)) {
ebbf41df 2908 nf_ct_unlink_expect_report(exp,
15e47304 2909 NETLINK_CB(skb).portid,
ebbf41df 2910 nlmsg_report(nlh));
31f15875
PM		 2911 nf_ct_expect_put(exp);
2912 }
c1d10adb
PNA		 2913 }
2914 }
ca7433df 2915 spin_unlock_bh(&nf_conntrack_expect_lock);
c1d10adb
PNA		 2916 } else {
2917 /* This basically means we have to flush everything*/
ca7433df 2918 spin_lock_bh(&nf_conntrack_expect_lock);
31f15875 2919 for (i = 0; i < nf_ct_expect_hsize; i++) {
b67bfe0d 2920 hlist_for_each_entry_safe(exp, next,
9592a5c0 2921 &net->ct.expect_hash[i],
31f15875
PM		 2922 hnode) {
2923 if (del_timer(&exp->timeout)) {
ebbf41df 2924 nf_ct_unlink_expect_report(exp,
15e47304 2925 NETLINK_CB(skb).portid,
ebbf41df 2926 nlmsg_report(nlh));
31f15875
PM		 2927 nf_ct_expect_put(exp);
2928 }
c1d10adb
PNA		 2929 }
2930 }
ca7433df 2931 spin_unlock_bh(&nf_conntrack_expect_lock);
c1d10adb
PNA		 2932 }
2933
2934 return 0;
2935}
2936static int
39938324
PM		 2937ctnetlink_change_expect(struct nf_conntrack_expect *x,
2938 const struct nlattr * const cda[])
c1d10adb 2939{
9768e1ac
KW		 2940 if (cda[CTA_EXPECT_TIMEOUT]) {
2941 if (!del_timer(&x->timeout))
2942 return -ETIME;
2943
2944 x->timeout.expires = jiffies +
2945 ntohl(nla_get_be32(cda[CTA_EXPECT_TIMEOUT])) * HZ;
2946 add_timer(&x->timeout);
2947 }
2948 return 0;
c1d10adb
PNA		 2949}
2950
076a0ca0
PNA		 2951static const struct nla_policy exp_nat_nla_policy[CTA_EXPECT_NAT_MAX+1] = {
2952 [CTA_EXPECT_NAT_DIR] = { .type = NLA_U32 },
2953 [CTA_EXPECT_NAT_TUPLE] = { .type = NLA_NESTED },
2954};
2955
2956static int
2957ctnetlink_parse_expect_nat(const struct nlattr *attr,
2958 struct nf_conntrack_expect *exp,
2959 u_int8_t u3)
2960{
2961#ifdef CONFIG_NF_NAT_NEEDED
2962 struct nlattr *tb[CTA_EXPECT_NAT_MAX+1];
2963 struct nf_conntrack_tuple nat_tuple = {};
2964 int err;
2965
130ffbc2
DB		 2966 err = nla_parse_nested(tb, CTA_EXPECT_NAT_MAX, attr, exp_nat_nla_policy);
2967 if (err < 0)
2968 return err;
076a0ca0
PNA		 2969
2970 if (!tb[CTA_EXPECT_NAT_DIR] || !tb[CTA_EXPECT_NAT_TUPLE])
2971 return -EINVAL;
2972
2973 err = ctnetlink_parse_tuple((const struct nlattr * const *)tb,
deedb590
DB		 2974 &nat_tuple, CTA_EXPECT_NAT_TUPLE,
2975 u3, NULL);
076a0ca0
PNA		 2976 if (err < 0)
2977 return err;
2978
c7232c99 2979 exp->saved_addr = nat_tuple.src.u3;
076a0ca0
PNA		 2980 exp->saved_proto = nat_tuple.src.u;
2981 exp->dir = ntohl(nla_get_be32(tb[CTA_EXPECT_NAT_DIR]));
2982
2983 return 0;
2984#else
2985 return -EOPNOTSUPP;
2986#endif
2987}
2988
0ef71ee1
PNA		 2989static struct nf_conntrack_expect *
2990ctnetlink_alloc_expect(const struct nlattr * const cda[], struct nf_conn *ct,
2991 struct nf_conntrack_helper *helper,
2992 struct nf_conntrack_tuple *tuple,
2993 struct nf_conntrack_tuple *mask)
c1d10adb 2994{
0ef71ee1 2995 u_int32_t class = 0;
c1d10adb 2996 struct nf_conntrack_expect *exp;
dc808fe2 2997 struct nf_conn_help *help;
0ef71ee1 2998 int err;
660fdb2a 2999
b8c5e52c
PNA		 3000 if (cda[CTA_EXPECT_CLASS] && helper) {
3001 class = ntohl(nla_get_be32(cda[CTA_EXPECT_CLASS]));
0ef71ee1
PNA		 3002 if (class > helper->expect_class_max)
3003 return ERR_PTR(-EINVAL);
b8c5e52c 3004 }
6823645d 3005 exp = nf_ct_expect_alloc(ct);
0ef71ee1
PNA		 3006 if (!exp)
3007 return ERR_PTR(-ENOMEM);
3008
bc01befd
PNA		 3009 help = nfct_help(ct);
3010 if (!help) {
3011 if (!cda[CTA_EXPECT_TIMEOUT]) {
3012 err = -EINVAL;
1310b955 3013 goto err_out;
bc01befd
PNA		 3014 }
3015 exp->timeout.expires =
3016 jiffies + ntohl(nla_get_be32(cda[CTA_EXPECT_TIMEOUT])) * HZ;
601e68e1 3017
bc01befd
PNA		 3018 exp->flags = NF_CT_EXPECT_USERSPACE;
3019 if (cda[CTA_EXPECT_FLAGS]) {
3020 exp->flags |=
3021 ntohl(nla_get_be32(cda[CTA_EXPECT_FLAGS]));
3022 }
3023 } else {
3024 if (cda[CTA_EXPECT_FLAGS]) {
3025 exp->flags = ntohl(nla_get_be32(cda[CTA_EXPECT_FLAGS]));
3026 exp->flags &= ~NF_CT_EXPECT_USERSPACE;
3027 } else
3028 exp->flags = 0;
3029 }
544d5c7d
PNA		 3030 if (cda[CTA_EXPECT_FN]) {
3031 const char *name = nla_data(cda[CTA_EXPECT_FN]);
3032 struct nf_ct_helper_expectfn *expfn;
3033
3034 expfn = nf_ct_helper_expectfn_find_by_name(name);
3035 if (expfn == NULL) {
3036 err = -EINVAL;
3037 goto err_out;
3038 }
3039 exp->expectfn = expfn->expectfn;
3040 } else
3041 exp->expectfn = NULL;
601e68e1 3042
b8c5e52c 3043 exp->class = class;
c1d10adb 3044 exp->master = ct;
660fdb2a 3045 exp->helper = helper;
0ef71ee1
PNA		 3046 exp->tuple = *tuple;
3047 exp->mask.src.u3 = mask->src.u3;
3048 exp->mask.src.u.all = mask->src.u.all;
c1d10adb 3049
076a0ca0
PNA		 3050 if (cda[CTA_EXPECT_NAT]) {
3051 err = ctnetlink_parse_expect_nat(cda[CTA_EXPECT_NAT],
0ef71ee1 3052 exp, nf_ct_l3num(ct));
076a0ca0
PNA		 3053 if (err < 0)
3054 goto err_out;
3055 }
0ef71ee1 3056 return exp;
076a0ca0 3057err_out:
6823645d 3058 nf_ct_expect_put(exp);
0ef71ee1
PNA		 3059 return ERR_PTR(err);
3060}
3061
3062static int
308ac914
DB		 3063ctnetlink_create_expect(struct net *net,
3064 const struct nf_conntrack_zone *zone,
0ef71ee1
PNA		 3065 const struct nlattr * const cda[],
3066 u_int8_t u3, u32 portid, int report)
3067{
3068 struct nf_conntrack_tuple tuple, mask, master_tuple;
3069 struct nf_conntrack_tuple_hash *h = NULL;
3070 struct nf_conntrack_helper *helper = NULL;
3071 struct nf_conntrack_expect *exp;
3072 struct nf_conn *ct;
3073 int err;
3074
3075 /* caller guarantees that those three CTA_EXPECT_* exist */
deedb590
DB		 3076 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE,
3077 u3, NULL);
0ef71ee1
PNA		 3078 if (err < 0)
3079 return err;
deedb590
DB		 3080 err = ctnetlink_parse_tuple(cda, &mask, CTA_EXPECT_MASK,
3081 u3, NULL);
0ef71ee1
PNA		 3082 if (err < 0)
3083 return err;
deedb590
DB		 3084 err = ctnetlink_parse_tuple(cda, &master_tuple, CTA_EXPECT_MASTER,
3085 u3, NULL);
0ef71ee1
PNA		 3086 if (err < 0)
3087 return err;
3088
3089 /* Look for master conntrack of this expectation */
3090 h = nf_conntrack_find_get(net, zone, &master_tuple);
3091 if (!h)
3092 return -ENOENT;
3093 ct = nf_ct_tuplehash_to_ctrack(h);
3094
3095 if (cda[CTA_EXPECT_HELP_NAME]) {
3096 const char *helpname = nla_data(cda[CTA_EXPECT_HELP_NAME]);
3097
3098 helper = __nf_conntrack_helper_find(helpname, u3,
3099 nf_ct_protonum(ct));
3100 if (helper == NULL) {
3101#ifdef CONFIG_MODULES
3102 if (request_module("nfct-helper-%s", helpname) < 0) {
3103 err = -EOPNOTSUPP;
3104 goto err_ct;
3105 }
3106 helper = __nf_conntrack_helper_find(helpname, u3,
3107 nf_ct_protonum(ct));
3108 if (helper) {
3109 err = -EAGAIN;
3110 goto err_ct;
3111 }
3112#endif
3113 err = -EOPNOTSUPP;
3114 goto err_ct;
3115 }
3116 }
3117
3118 exp = ctnetlink_alloc_expect(cda, ct, helper, &tuple, &mask);
3119 if (IS_ERR(exp)) {
3120 err = PTR_ERR(exp);
3121 goto err_ct;
3122 }
3123
3124 err = nf_ct_expect_related_report(exp, portid, report);
0ef71ee1
PNA		 3125 nf_ct_expect_put(exp);
3126err_ct:
3127 nf_ct_put(ct);
c1d10adb
PNA		 3128 return err;
3129}
3130
7b8002a1
PNA		 3131static int ctnetlink_new_expect(struct net *net, struct sock *ctnl,
3132 struct sk_buff *skb, const struct nlmsghdr *nlh,
3133 const struct nlattr * const cda[])
c1d10adb
PNA		 3134{
3135 struct nf_conntrack_tuple tuple;
3136 struct nf_conntrack_expect *exp;
96bcf938 3137 struct nfgenmsg *nfmsg = nlmsg_data(nlh);
c1d10adb 3138 u_int8_t u3 = nfmsg->nfgen_family;
308ac914 3139 struct nf_conntrack_zone zone;
ef00f89f 3140 int err;
c1d10adb 3141
df6fb868
PM		 3142 if (!cda[CTA_EXPECT_TUPLE]
3143 || !cda[CTA_EXPECT_MASK]
3144 || !cda[CTA_EXPECT_MASTER])
c1d10adb
PNA		 3145 return -EINVAL;
3146
ef00f89f
PM		 3147 err = ctnetlink_parse_zone(cda[CTA_EXPECT_ZONE], &zone);
3148 if (err < 0)
3149 return err;
3150
deedb590
DB		 3151 err = ctnetlink_parse_tuple(cda, &tuple, CTA_EXPECT_TUPLE,
3152 u3, NULL);
c1d10adb
PNA		 3153 if (err < 0)
3154 return err;
3155
ca7433df 3156 spin_lock_bh(&nf_conntrack_expect_lock);
308ac914 3157 exp = __nf_ct_expect_find(net, &zone, &tuple);
c1d10adb 3158 if (!exp) {
ca7433df 3159 spin_unlock_bh(&nf_conntrack_expect_lock);
c1d10adb 3160 err = -ENOENT;
19abb7b0 3161 if (nlh->nlmsg_flags & NLM_F_CREATE) {
308ac914 3162 err = ctnetlink_create_expect(net, &zone, cda, u3,
15e47304 3163 NETLINK_CB(skb).portid,
19abb7b0
PNA		 3164 nlmsg_report(nlh));
3165 }
c1d10adb
PNA		 3166 return err;
3167 }
3168
3169 err = -EEXIST;
3170 if (!(nlh->nlmsg_flags & NLM_F_EXCL))
3171 err = ctnetlink_change_expect(exp, cda);
ca7433df 3172 spin_unlock_bh(&nf_conntrack_expect_lock);
c1d10adb 3173
c1d10adb
PNA		 3174 return err;
3175}
3176
392025f8 3177static int
15e47304 3178ctnetlink_exp_stat_fill_info(struct sk_buff *skb, u32 portid, u32 seq, int cpu,
392025f8
PNA		 3179 const struct ip_conntrack_stat *st)
3180{
3181 struct nlmsghdr *nlh;
3182 struct nfgenmsg *nfmsg;
15e47304 3183 unsigned int flags = portid ? NLM_F_MULTI : 0, event;
392025f8
PNA		 3184
3185 event = (NFNL_SUBSYS_CTNETLINK << 8 | IPCTNL_MSG_EXP_GET_STATS_CPU);
15e47304 3186 nlh = nlmsg_put(skb, portid, seq, event, sizeof(*nfmsg), flags);
392025f8
PNA		 3187 if (nlh == NULL)
3188 goto nlmsg_failure;
3189
3190 nfmsg = nlmsg_data(nlh);
3191 nfmsg->nfgen_family = AF_UNSPEC;
3192 nfmsg->version = NFNETLINK_V0;
3193 nfmsg->res_id = htons(cpu);
3194
3195 if (nla_put_be32(skb, CTA_STATS_EXP_NEW, htonl(st->expect_new)) ||
3196 nla_put_be32(skb, CTA_STATS_EXP_CREATE, htonl(st->expect_create)) ||
3197 nla_put_be32(skb, CTA_STATS_EXP_DELETE, htonl(st->expect_delete)))
3198 goto nla_put_failure;
3199
3200 nlmsg_end(skb, nlh);
3201 return skb->len;
3202
3203nla_put_failure:
3204nlmsg_failure:
3205 nlmsg_cancel(skb, nlh);
3206 return -1;
3207}
3208
3209static int
3210ctnetlink_exp_stat_cpu_dump(struct sk_buff *skb, struct netlink_callback *cb)
3211{
3212 int cpu;
3213 struct net *net = sock_net(skb->sk);
3214
3215 if (cb->args[0] == nr_cpu_ids)
3216 return 0;
3217
3218 for (cpu = cb->args[0]; cpu < nr_cpu_ids; cpu++) {
3219 const struct ip_conntrack_stat *st;
3220
3221 if (!cpu_possible(cpu))
3222 continue;
3223
3224 st = per_cpu_ptr(net->ct.stat, cpu);
15e47304 3225 if (ctnetlink_exp_stat_fill_info(skb, NETLINK_CB(cb->skb).portid,
392025f8
PNA		 3226 cb->nlh->nlmsg_seq,
3227 cpu, st) < 0)
3228 break;
3229 }
3230 cb->args[0] = cpu;
3231
3232 return skb->len;
3233}
3234
7b8002a1
PNA		 3235static int ctnetlink_stat_exp_cpu(struct net *net, struct sock *ctnl,
3236 struct sk_buff *skb,
3237 const struct nlmsghdr *nlh,
3238 const struct nlattr * const cda[])
392025f8
PNA		 3239{
3240 if (nlh->nlmsg_flags & NLM_F_DUMP) {
3241 struct netlink_dump_control c = {
3242 .dump = ctnetlink_exp_stat_cpu_dump,
3243 };
3244 return netlink_dump_start(ctnl, skb, nlh, &c);
3245 }
3246
3247 return 0;
3248}
3249
c1d10adb 3250#ifdef CONFIG_NF_CONNTRACK_EVENTS
e34d5c1a
PNA		 3251static struct nf_ct_event_notifier ctnl_notifier = {
3252 .fcn = ctnetlink_conntrack_event,
c1d10adb
PNA		 3253};
3254
e34d5c1a
PNA		 3255static struct nf_exp_event_notifier ctnl_notifier_exp = {
3256 .fcn = ctnetlink_expect_event,
c1d10adb
PNA		 3257};
3258#endif
3259
7c8d4cb4 3260static const struct nfnl_callback ctnl_cb[IPCTNL_MSG_MAX] = {
c1d10adb 3261 [IPCTNL_MSG_CT_NEW] = { .call = ctnetlink_new_conntrack,
f73e924c
PM		 3262 .attr_count = CTA_MAX,
3263 .policy = ct_nla_policy },
c1d10adb 3264 [IPCTNL_MSG_CT_GET] = { .call = ctnetlink_get_conntrack,
f73e924c
PM		 3265 .attr_count = CTA_MAX,
3266 .policy = ct_nla_policy },
c1d10adb 3267 [IPCTNL_MSG_CT_DELETE] = { .call = ctnetlink_del_conntrack,
f73e924c
PM		 3268 .attr_count = CTA_MAX,
3269 .policy = ct_nla_policy },
c1d10adb 3270 [IPCTNL_MSG_CT_GET_CTRZERO] = { .call = ctnetlink_get_conntrack,
f73e924c
PM		 3271 .attr_count = CTA_MAX,
3272 .policy = ct_nla_policy },
392025f8
PNA		 3273 [IPCTNL_MSG_CT_GET_STATS_CPU] = { .call = ctnetlink_stat_ct_cpu },
3274 [IPCTNL_MSG_CT_GET_STATS] = { .call = ctnetlink_stat_ct },
d871befe
PNA		 3275 [IPCTNL_MSG_CT_GET_DYING] = { .call = ctnetlink_get_ct_dying },
3276 [IPCTNL_MSG_CT_GET_UNCONFIRMED] = { .call = ctnetlink_get_ct_unconfirmed },
c1d10adb
PNA		 3277};
3278
7c8d4cb4 3279static const struct nfnl_callback ctnl_exp_cb[IPCTNL_MSG_EXP_MAX] = {
c1d10adb 3280 [IPCTNL_MSG_EXP_GET] = { .call = ctnetlink_get_expect,
f73e924c
PM		 3281 .attr_count = CTA_EXPECT_MAX,
3282 .policy = exp_nla_policy },
c1d10adb 3283 [IPCTNL_MSG_EXP_NEW] = { .call = ctnetlink_new_expect,
f73e924c
PM		 3284 .attr_count = CTA_EXPECT_MAX,
3285 .policy = exp_nla_policy },
c1d10adb 3286 [IPCTNL_MSG_EXP_DELETE] = { .call = ctnetlink_del_expect,
f73e924c
PM		 3287 .attr_count = CTA_EXPECT_MAX,
3288 .policy = exp_nla_policy },
392025f8 3289 [IPCTNL_MSG_EXP_GET_STATS_CPU] = { .call = ctnetlink_stat_exp_cpu },
c1d10adb
PNA		 3290};
3291
7c8d4cb4 3292static const struct nfnetlink_subsystem ctnl_subsys = {
c1d10adb
PNA		 3293 .name = "conntrack",
3294 .subsys_id = NFNL_SUBSYS_CTNETLINK,
3295 .cb_count = IPCTNL_MSG_MAX,
3296 .cb = ctnl_cb,
3297};
3298
7c8d4cb4 3299static const struct nfnetlink_subsystem ctnl_exp_subsys = {
c1d10adb
PNA		 3300 .name = "conntrack_expect",
3301 .subsys_id = NFNL_SUBSYS_CTNETLINK_EXP,
3302 .cb_count = IPCTNL_MSG_EXP_MAX,
3303 .cb = ctnl_exp_cb,
3304};
3305
d2483dde 3306MODULE_ALIAS("ip_conntrack_netlink");
c1d10adb 3307MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK);
34f9a2e4 3308MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_CTNETLINK_EXP);
c1d10adb 3309
70e9942f
PNA		 3310static int __net_init ctnetlink_net_init(struct net *net)
3311{
3312#ifdef CONFIG_NF_CONNTRACK_EVENTS
3313 int ret;
3314
3315 ret = nf_conntrack_register_notifier(net, &ctnl_notifier);
3316 if (ret < 0) {
3317 pr_err("ctnetlink_init: cannot register notifier.\n");
3318 goto err_out;
3319 }
3320
3321 ret = nf_ct_expect_register_notifier(net, &ctnl_notifier_exp);
3322 if (ret < 0) {
3323 pr_err("ctnetlink_init: cannot expect register notifier.\n");
3324 goto err_unreg_notifier;
3325 }
3326#endif
3327 return 0;
3328
3329#ifdef CONFIG_NF_CONNTRACK_EVENTS
3330err_unreg_notifier:
3331 nf_conntrack_unregister_notifier(net, &ctnl_notifier);
3332err_out:
3333 return ret;
3334#endif
3335}
3336
3337static void ctnetlink_net_exit(struct net *net)
3338{
3339#ifdef CONFIG_NF_CONNTRACK_EVENTS
3340 nf_ct_expect_unregister_notifier(net, &ctnl_notifier_exp);
3341 nf_conntrack_unregister_notifier(net, &ctnl_notifier);
3342#endif
3343}
3344
3345static void __net_exit ctnetlink_net_exit_batch(struct list_head *net_exit_list)
3346{
3347 struct net *net;
3348
3349 list_for_each_entry(net, net_exit_list, exit_list)
3350 ctnetlink_net_exit(net);
3351}
3352
3353static struct pernet_operations ctnetlink_net_ops = {
3354 .init = ctnetlink_net_init,
3355 .exit_batch = ctnetlink_net_exit_batch,
3356};
3357
c1d10adb
PNA		 3358static int __init ctnetlink_init(void)
3359{
3360 int ret;
3361
654d0fbd 3362 pr_info("ctnetlink v%s: registering with nfnetlink.\n", version);
c1d10adb
PNA		 3363 ret = nfnetlink_subsys_register(&ctnl_subsys);
3364 if (ret < 0) {
654d0fbd 3365 pr_err("ctnetlink_init: cannot register with nfnetlink.\n");
c1d10adb
PNA		 3366 goto err_out;
3367 }
3368
3369 ret = nfnetlink_subsys_register(&ctnl_exp_subsys);
3370 if (ret < 0) {
654d0fbd 3371 pr_err("ctnetlink_init: cannot register exp with nfnetlink.\n");
c1d10adb
PNA		 3372 goto err_unreg_subsys;
3373 }
3374
ef6acf68
JL		 3375 ret = register_pernet_subsys(&ctnetlink_net_ops);
3376 if (ret < 0) {
70e9942f 3377 pr_err("ctnetlink_init: cannot register pernet operations\n");
c1d10adb
PNA		 3378 goto err_unreg_exp_subsys;
3379 }
83f3e94d 3380#ifdef CONFIG_NETFILTER_NETLINK_GLUE_CT
9cb01766 3381 /* setup interaction between nf_queue and nf_conntrack_netlink. */
a4b4766c 3382 RCU_INIT_POINTER(nfnl_ct_hook, &ctnetlink_glue_hook);
9cb01766 3383#endif
c1d10adb
PNA		 3384 return 0;
3385
c1d10adb
PNA		 3386err_unreg_exp_subsys:
3387 nfnetlink_subsys_unregister(&ctnl_exp_subsys);
c1d10adb
PNA		 3388err_unreg_subsys:
3389 nfnetlink_subsys_unregister(&ctnl_subsys);
3390err_out:
3391 return ret;
3392}
3393
3394static void __exit ctnetlink_exit(void)
3395{
654d0fbd 3396 pr_info("ctnetlink: unregistering from nfnetlink.\n");
c1d10adb 3397
70e9942f 3398 unregister_pernet_subsys(&ctnetlink_net_ops);
c1d10adb
PNA		 3399 nfnetlink_subsys_unregister(&ctnl_exp_subsys);
3400 nfnetlink_subsys_unregister(&ctnl_subsys);
83f3e94d 3401#ifdef CONFIG_NETFILTER_NETLINK_GLUE_CT
a4b4766c 3402 RCU_INIT_POINTER(nfnl_ct_hook, NULL);
9cb01766 3403#endif
c1d10adb
PNA		 3404}
3405
3406module_init(ctnetlink_init);
3407module_exit(ctnetlink_exit);
Linux 6.x block layer and io_uring tree(s)