1 // SPDX-License-Identifier: GPL-2.0
2
3 #include <linux/types.h>
4 #include <linux/netfilter.h>
5 #include <linux/module.h>
6 #include <linux/slab.h>
7 #include <linux/mutex.h>
8 #include <linux/vmalloc.h>
9 #include <linux/stddef.h>
10 #include <linux/err.h>
11 #include <linux/percpu.h>
12 #include <linux/notifier.h>
13 #include <linux/kernel.h>
14 #include <linux/netdevice.h>
15
16 #include <net/netfilter/nf_conntrack.h>
17 #include <net/netfilter/nf_conntrack_l4proto.h>
18 #include <net/netfilter/nf_conntrack_core.h>
19 #include <net/netfilter/nf_conntrack_bridge.h>
20 #include <net/netfilter/nf_log.h>
21
22 #include <linux/ip.h>
23 #include <linux/icmp.h>
24 #include <linux/sysctl.h>
25 #include <net/route.h>
26 #include <net/ip.h>
27
28 #include <linux/netfilter_ipv4.h>
29 #include <linux/netfilter_ipv6.h>
30 #include <linux/netfilter_ipv6/ip6_tables.h>
31 #include <net/netfilter/nf_conntrack_helper.h>
32 #include <net/netfilter/nf_conntrack_zones.h>
33 #include <net/netfilter/nf_conntrack_seqadj.h>
34 #include <net/netfilter/ipv4/nf_conntrack_ipv4.h>
35 #include <net/netfilter/ipv6/nf_conntrack_ipv6.h>
36 #include <net/netfilter/nf_nat_helper.h>
37 #include <net/netfilter/ipv4/nf_defrag_ipv4.h>
38 #include <net/netfilter/ipv6/nf_defrag_ipv6.h>
39
40 #include <linux/ipv6.h>
41 #include <linux/in6.h>
42 #include <net/ipv6.h>
43 #include <net/inet_frag.h>
44
45 static DEFINE_MUTEX(nf_ct_proto_mutex);
46
47 #ifdef CONFIG_SYSCTL
48 __printf(4, 5)
nf_l4proto_log_invalid(const struct sk_buff * skb,const struct nf_hook_state * state,u8 protonum,const char * fmt,...)49 void nf_l4proto_log_invalid(const struct sk_buff *skb,
50 const struct nf_hook_state *state,
51 u8 protonum,
52 const char *fmt, ...)
53 {
54 struct net *net = state->net;
55 struct va_format vaf;
56 va_list args;
57
58 if (net->ct.sysctl_log_invalid != protonum &&
59 net->ct.sysctl_log_invalid != IPPROTO_RAW)
60 return;
61
62 va_start(args, fmt);
63 vaf.fmt = fmt;
64 vaf.va = &args;
65
66 nf_log_packet(net, state->pf, 0, skb, state->in, state->out,
67 NULL, "nf_ct_proto_%d: %pV ", protonum, &vaf);
68 va_end(args);
69 }
70 EXPORT_SYMBOL_GPL(nf_l4proto_log_invalid);
71
72 __printf(4, 5)
nf_ct_l4proto_log_invalid(const struct sk_buff * skb,const struct nf_conn * ct,const struct nf_hook_state * state,const char * fmt,...)73 void nf_ct_l4proto_log_invalid(const struct sk_buff *skb,
74 const struct nf_conn *ct,
75 const struct nf_hook_state *state,
76 const char *fmt, ...)
77 {
78 struct va_format vaf;
79 struct net *net;
80 va_list args;
81
82 net = nf_ct_net(ct);
83 if (likely(net->ct.sysctl_log_invalid == 0))
84 return;
85
86 va_start(args, fmt);
87 vaf.fmt = fmt;
88 vaf.va = &args;
89
90 nf_l4proto_log_invalid(skb, state,
91 nf_ct_protonum(ct), "%pV", &vaf);
92 va_end(args);
93 }
94 EXPORT_SYMBOL_GPL(nf_ct_l4proto_log_invalid);
95 #endif
96
nf_ct_l4proto_find(u8 l4proto)97 const struct nf_conntrack_l4proto *nf_ct_l4proto_find(u8 l4proto)
98 {
99 switch (l4proto) {
100 case IPPROTO_UDP: return &nf_conntrack_l4proto_udp;
101 case IPPROTO_TCP: return &nf_conntrack_l4proto_tcp;
102 case IPPROTO_ICMP: return &nf_conntrack_l4proto_icmp;
103 #ifdef CONFIG_NF_CT_PROTO_DCCP
104 case IPPROTO_DCCP: return &nf_conntrack_l4proto_dccp;
105 #endif
106 #ifdef CONFIG_NF_CT_PROTO_SCTP
107 case IPPROTO_SCTP: return &nf_conntrack_l4proto_sctp;
108 #endif
109 #ifdef CONFIG_NF_CT_PROTO_UDPLITE
110 case IPPROTO_UDPLITE: return &nf_conntrack_l4proto_udplite;
111 #endif
112 #ifdef CONFIG_NF_CT_PROTO_GRE
113 case IPPROTO_GRE: return &nf_conntrack_l4proto_gre;
114 #endif
115 #if IS_ENABLED(CONFIG_IPV6)
116 case IPPROTO_ICMPV6: return &nf_conntrack_l4proto_icmpv6;
117 #endif /* CONFIG_IPV6 */
118 }
119
120 return &nf_conntrack_l4proto_generic;
121 };
122 EXPORT_SYMBOL_GPL(nf_ct_l4proto_find);
123
nf_confirm(struct sk_buff * skb,unsigned int protoff,struct nf_conn * ct,enum ip_conntrack_info ctinfo)124 unsigned int nf_confirm(struct sk_buff *skb, unsigned int protoff,
125 struct nf_conn *ct, enum ip_conntrack_info ctinfo)
126 {
127 const struct nf_conn_help *help;
128
129 help = nfct_help(ct);
130 if (help) {
131 const struct nf_conntrack_helper *helper;
132 int ret;
133
134 /* rcu_read_lock()ed by nf_hook_thresh */
135 helper = rcu_dereference(help->helper);
136 if (helper) {
137 ret = helper->help(skb,
138 protoff,
139 ct, ctinfo);
140 if (ret != NF_ACCEPT)
141 return ret;
142 }
143 }
144
145 if (test_bit(IPS_SEQ_ADJUST_BIT, &ct->status) &&
146 !nf_is_loopback_packet(skb)) {
147 if (!nf_ct_seq_adjust(skb, ct, ctinfo, protoff)) {
148 NF_CT_STAT_INC_ATOMIC(nf_ct_net(ct), drop);
149 return NF_DROP;
150 }
151 }
152
153 /* We've seen it coming out the other side: confirm it */
154 return nf_conntrack_confirm(skb);
155 }
156 EXPORT_SYMBOL_GPL(nf_confirm);
157
in_vrf_postrouting(const struct nf_hook_state * state)158 static bool in_vrf_postrouting(const struct nf_hook_state *state)
159 {
160 #if IS_ENABLED(CONFIG_NET_L3_MASTER_DEV)
161 if (state->hook == NF_INET_POST_ROUTING &&
162 netif_is_l3_master(state->out))
163 return true;
164 #endif
165 return false;
166 }
167
ipv4_confirm(void * priv,struct sk_buff * skb,const struct nf_hook_state * state)168 static unsigned int ipv4_confirm(void *priv,
169 struct sk_buff *skb,
170 const struct nf_hook_state *state)
171 {
172 enum ip_conntrack_info ctinfo;
173 struct nf_conn *ct;
174
175 ct = nf_ct_get(skb, &ctinfo);
176 if (!ct || ctinfo == IP_CT_RELATED_REPLY)
177 return nf_conntrack_confirm(skb);
178
179 if (in_vrf_postrouting(state))
180 return NF_ACCEPT;
181
182 return nf_confirm(skb,
183 skb_network_offset(skb) + ip_hdrlen(skb),
184 ct, ctinfo);
185 }
186
ipv4_conntrack_in(void * priv,struct sk_buff * skb,const struct nf_hook_state * state)187 static unsigned int ipv4_conntrack_in(void *priv,
188 struct sk_buff *skb,
189 const struct nf_hook_state *state)
190 {
191 return nf_conntrack_in(skb, state);
192 }
193
ipv4_conntrack_local(void * priv,struct sk_buff * skb,const struct nf_hook_state * state)194 static unsigned int ipv4_conntrack_local(void *priv,
195 struct sk_buff *skb,
196 const struct nf_hook_state *state)
197 {
198 if (ip_is_fragment(ip_hdr(skb))) { /* IP_NODEFRAG setsockopt set */
199 enum ip_conntrack_info ctinfo;
200 struct nf_conn *tmpl;
201
202 tmpl = nf_ct_get(skb, &ctinfo);
203 if (tmpl && nf_ct_is_template(tmpl)) {
204 /* when skipping ct, clear templates to avoid fooling
205 * later targets/matches
206 */
207 skb->_nfct = 0;
208 nf_ct_put(tmpl);
209 }
210 return NF_ACCEPT;
211 }
212
213 return nf_conntrack_in(skb, state);
214 }
215
216 /* Connection tracking may drop packets, but never alters them, so
217 * make it the first hook.
218 */
219 static const struct nf_hook_ops ipv4_conntrack_ops[] = {
220 {
221 .hook = ipv4_conntrack_in,
222 .pf = NFPROTO_IPV4,
223 .hooknum = NF_INET_PRE_ROUTING,
224 .priority = NF_IP_PRI_CONNTRACK,
225 },
226 {
227 .hook = ipv4_conntrack_local,
228 .pf = NFPROTO_IPV4,
229 .hooknum = NF_INET_LOCAL_OUT,
230 .priority = NF_IP_PRI_CONNTRACK,
231 },
232 {
233 .hook = ipv4_confirm,
234 .pf = NFPROTO_IPV4,
235 .hooknum = NF_INET_POST_ROUTING,
236 .priority = NF_IP_PRI_CONNTRACK_CONFIRM,
237 },
238 {
239 .hook = ipv4_confirm,
240 .pf = NFPROTO_IPV4,
241 .hooknum = NF_INET_LOCAL_IN,
242 .priority = NF_IP_PRI_CONNTRACK_CONFIRM,
243 },
244 };
245
246 /* Fast function for those who don't want to parse /proc (and I don't
247 * blame them).
248 * Reversing the socket's dst/src point of view gives us the reply
249 * mapping.
250 */
251 static int
getorigdst(struct sock * sk,int optval,void __user * user,int * len)252 getorigdst(struct sock *sk, int optval, void __user *user, int *len)
253 {
254 const struct inet_sock *inet = inet_sk(sk);
255 const struct nf_conntrack_tuple_hash *h;
256 struct nf_conntrack_tuple tuple;
257
258 memset(&tuple, 0, sizeof(tuple));
259
260 lock_sock(sk);
261 tuple.src.u3.ip = inet->inet_rcv_saddr;
262 tuple.src.u.tcp.port = inet->inet_sport;
263 tuple.dst.u3.ip = inet->inet_daddr;
264 tuple.dst.u.tcp.port = inet->inet_dport;
265 tuple.src.l3num = PF_INET;
266 tuple.dst.protonum = sk->sk_protocol;
267 release_sock(sk);
268
269 /* We only do TCP and SCTP at the moment: is there a better way? */
270 if (tuple.dst.protonum != IPPROTO_TCP &&
271 tuple.dst.protonum != IPPROTO_SCTP) {
272 pr_debug("SO_ORIGINAL_DST: Not a TCP/SCTP socket\n");
273 return -ENOPROTOOPT;
274 }
275
276 if ((unsigned int)*len < sizeof(struct sockaddr_in)) {
277 pr_debug("SO_ORIGINAL_DST: len %d not %zu\n",
278 *len, sizeof(struct sockaddr_in));
279 return -EINVAL;
280 }
281
282 h = nf_conntrack_find_get(sock_net(sk), &nf_ct_zone_dflt, &tuple);
283 if (h) {
284 struct sockaddr_in sin;
285 struct nf_conn *ct = nf_ct_tuplehash_to_ctrack(h);
286
287 sin.sin_family = AF_INET;
288 sin.sin_port = ct->tuplehash[IP_CT_DIR_ORIGINAL]
289 .tuple.dst.u.tcp.port;
290 sin.sin_addr.s_addr = ct->tuplehash[IP_CT_DIR_ORIGINAL]
291 .tuple.dst.u3.ip;
292 memset(sin.sin_zero, 0, sizeof(sin.sin_zero));
293
294 pr_debug("SO_ORIGINAL_DST: %pI4 %u\n",
295 &sin.sin_addr.s_addr, ntohs(sin.sin_port));
296 nf_ct_put(ct);
297 if (copy_to_user(user, &sin, sizeof(sin)) != 0)
298 return -EFAULT;
299 else
300 return 0;
301 }
302 pr_debug("SO_ORIGINAL_DST: Can't find %pI4/%u-%pI4/%u.\n",
303 &tuple.src.u3.ip, ntohs(tuple.src.u.tcp.port),
304 &tuple.dst.u3.ip, ntohs(tuple.dst.u.tcp.port));
305 return -ENOENT;
306 }
307
308 static struct nf_sockopt_ops so_getorigdst = {
309 .pf = PF_INET,
310 .get_optmin = SO_ORIGINAL_DST,
311 .get_optmax = SO_ORIGINAL_DST + 1,
312 .get = getorigdst,
313 .owner = THIS_MODULE,
314 };
315
316 #if IS_ENABLED(CONFIG_IPV6)
317 static int
ipv6_getorigdst(struct sock * sk,int optval,void __user * user,int * len)318 ipv6_getorigdst(struct sock *sk, int optval, void __user *user, int *len)
319 {
320 struct nf_conntrack_tuple tuple = { .src.l3num = NFPROTO_IPV6 };
321 const struct ipv6_pinfo *inet6 = inet6_sk(sk);
322 const struct inet_sock *inet = inet_sk(sk);
323 const struct nf_conntrack_tuple_hash *h;
324 struct sockaddr_in6 sin6;
325 struct nf_conn *ct;
326 __be32 flow_label;
327 int bound_dev_if;
328
329 lock_sock(sk);
330 tuple.src.u3.in6 = sk->sk_v6_rcv_saddr;
331 tuple.src.u.tcp.port = inet->inet_sport;
332 tuple.dst.u3.in6 = sk->sk_v6_daddr;
333 tuple.dst.u.tcp.port = inet->inet_dport;
334 tuple.dst.protonum = sk->sk_protocol;
335 bound_dev_if = sk->sk_bound_dev_if;
336 flow_label = inet6->flow_label;
337 release_sock(sk);
338
339 if (tuple.dst.protonum != IPPROTO_TCP &&
340 tuple.dst.protonum != IPPROTO_SCTP)
341 return -ENOPROTOOPT;
342
343 if (*len < 0 || (unsigned int)*len < sizeof(sin6))
344 return -EINVAL;
345
346 h = nf_conntrack_find_get(sock_net(sk), &nf_ct_zone_dflt, &tuple);
347 if (!h) {
348 pr_debug("IP6T_SO_ORIGINAL_DST: Can't find %pI6c/%u-%pI6c/%u.\n",
349 &tuple.src.u3.ip6, ntohs(tuple.src.u.tcp.port),
350 &tuple.dst.u3.ip6, ntohs(tuple.dst.u.tcp.port));
351 return -ENOENT;
352 }
353
354 ct = nf_ct_tuplehash_to_ctrack(h);
355
356 sin6.sin6_family = AF_INET6;
357 sin6.sin6_port = ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.u.tcp.port;
358 sin6.sin6_flowinfo = flow_label & IPV6_FLOWINFO_MASK;
359 memcpy(&sin6.sin6_addr,
360 &ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple.dst.u3.in6,
361 sizeof(sin6.sin6_addr));
362
363 nf_ct_put(ct);
364 sin6.sin6_scope_id = ipv6_iface_scope_id(&sin6.sin6_addr, bound_dev_if);
365 return copy_to_user(user, &sin6, sizeof(sin6)) ? -EFAULT : 0;
366 }
367
368 static struct nf_sockopt_ops so_getorigdst6 = {
369 .pf = NFPROTO_IPV6,
370 .get_optmin = IP6T_SO_ORIGINAL_DST,
371 .get_optmax = IP6T_SO_ORIGINAL_DST + 1,
372 .get = ipv6_getorigdst,
373 .owner = THIS_MODULE,
374 };
375
ipv6_confirm(void * priv,struct sk_buff * skb,const struct nf_hook_state * state)376 static unsigned int ipv6_confirm(void *priv,
377 struct sk_buff *skb,
378 const struct nf_hook_state *state)
379 {
380 struct nf_conn *ct;
381 enum ip_conntrack_info ctinfo;
382 unsigned char pnum = ipv6_hdr(skb)->nexthdr;
383 __be16 frag_off;
384 int protoff;
385
386 ct = nf_ct_get(skb, &ctinfo);
387 if (!ct || ctinfo == IP_CT_RELATED_REPLY)
388 return nf_conntrack_confirm(skb);
389
390 if (in_vrf_postrouting(state))
391 return NF_ACCEPT;
392
393 protoff = ipv6_skip_exthdr(skb, sizeof(struct ipv6hdr), &pnum,
394 &frag_off);
395 if (protoff < 0 || (frag_off & htons(~0x7)) != 0) {
396 pr_debug("proto header not found\n");
397 return nf_conntrack_confirm(skb);
398 }
399
400 return nf_confirm(skb, protoff, ct, ctinfo);
401 }
402
ipv6_conntrack_in(void * priv,struct sk_buff * skb,const struct nf_hook_state * state)403 static unsigned int ipv6_conntrack_in(void *priv,
404 struct sk_buff *skb,
405 const struct nf_hook_state *state)
406 {
407 return nf_conntrack_in(skb, state);
408 }
409
ipv6_conntrack_local(void * priv,struct sk_buff * skb,const struct nf_hook_state * state)410 static unsigned int ipv6_conntrack_local(void *priv,
411 struct sk_buff *skb,
412 const struct nf_hook_state *state)
413 {
414 return nf_conntrack_in(skb, state);
415 }
416
417 static const struct nf_hook_ops ipv6_conntrack_ops[] = {
418 {
419 .hook = ipv6_conntrack_in,
420 .pf = NFPROTO_IPV6,
421 .hooknum = NF_INET_PRE_ROUTING,
422 .priority = NF_IP6_PRI_CONNTRACK,
423 },
424 {
425 .hook = ipv6_conntrack_local,
426 .pf = NFPROTO_IPV6,
427 .hooknum = NF_INET_LOCAL_OUT,
428 .priority = NF_IP6_PRI_CONNTRACK,
429 },
430 {
431 .hook = ipv6_confirm,
432 .pf = NFPROTO_IPV6,
433 .hooknum = NF_INET_POST_ROUTING,
434 .priority = NF_IP6_PRI_LAST,
435 },
436 {
437 .hook = ipv6_confirm,
438 .pf = NFPROTO_IPV6,
439 .hooknum = NF_INET_LOCAL_IN,
440 .priority = NF_IP6_PRI_LAST - 1,
441 },
442 };
443 #endif
444
nf_ct_tcp_fixup(struct nf_conn * ct,void * _nfproto)445 static int nf_ct_tcp_fixup(struct nf_conn *ct, void *_nfproto)
446 {
447 u8 nfproto = (unsigned long)_nfproto;
448
449 if (nf_ct_l3num(ct) != nfproto)
450 return 0;
451
452 if (nf_ct_protonum(ct) == IPPROTO_TCP &&
453 ct->proto.tcp.state == TCP_CONNTRACK_ESTABLISHED) {
454 ct->proto.tcp.seen[0].td_maxwin = 0;
455 ct->proto.tcp.seen[1].td_maxwin = 0;
456 }
457
458 return 0;
459 }
460
461 static struct nf_ct_bridge_info *nf_ct_bridge_info;
462
nf_ct_netns_do_get(struct net * net,u8 nfproto)463 static int nf_ct_netns_do_get(struct net *net, u8 nfproto)
464 {
465 struct nf_conntrack_net *cnet = nf_ct_pernet(net);
466 bool fixup_needed = false, retry = true;
467 int err = 0;
468 retry:
469 mutex_lock(&nf_ct_proto_mutex);
470
471 switch (nfproto) {
472 case NFPROTO_IPV4:
473 cnet->users4++;
474 if (cnet->users4 > 1)
475 goto out_unlock;
476 err = nf_defrag_ipv4_enable(net);
477 if (err) {
478 cnet->users4 = 0;
479 goto out_unlock;
480 }
481
482 err = nf_register_net_hooks(net, ipv4_conntrack_ops,
483 ARRAY_SIZE(ipv4_conntrack_ops));
484 if (err)
485 cnet->users4 = 0;
486 else
487 fixup_needed = true;
488 break;
489 #if IS_ENABLED(CONFIG_IPV6)
490 case NFPROTO_IPV6:
491 cnet->users6++;
492 if (cnet->users6 > 1)
493 goto out_unlock;
494 err = nf_defrag_ipv6_enable(net);
495 if (err < 0) {
496 cnet->users6 = 0;
497 goto out_unlock;
498 }
499
500 err = nf_register_net_hooks(net, ipv6_conntrack_ops,
501 ARRAY_SIZE(ipv6_conntrack_ops));
502 if (err)
503 cnet->users6 = 0;
504 else
505 fixup_needed = true;
506 break;
507 #endif
508 case NFPROTO_BRIDGE:
509 if (!nf_ct_bridge_info) {
510 if (!retry) {
511 err = -EPROTO;
512 goto out_unlock;
513 }
514 mutex_unlock(&nf_ct_proto_mutex);
515 request_module("nf_conntrack_bridge");
516 retry = false;
517 goto retry;
518 }
519 if (!try_module_get(nf_ct_bridge_info->me)) {
520 err = -EPROTO;
521 goto out_unlock;
522 }
523 cnet->users_bridge++;
524 if (cnet->users_bridge > 1)
525 goto out_unlock;
526
527 err = nf_register_net_hooks(net, nf_ct_bridge_info->ops,
528 nf_ct_bridge_info->ops_size);
529 if (err)
530 cnet->users_bridge = 0;
531 else
532 fixup_needed = true;
533 break;
534 default:
535 err = -EPROTO;
536 break;
537 }
538 out_unlock:
539 mutex_unlock(&nf_ct_proto_mutex);
540
541 if (fixup_needed) {
542 struct nf_ct_iter_data iter_data = {
543 .net = net,
544 .data = (void *)(unsigned long)nfproto,
545 };
546 nf_ct_iterate_cleanup_net(nf_ct_tcp_fixup, &iter_data);
547 }
548
549 return err;
550 }
551
nf_ct_netns_do_put(struct net * net,u8 nfproto)552 static void nf_ct_netns_do_put(struct net *net, u8 nfproto)
553 {
554 struct nf_conntrack_net *cnet = nf_ct_pernet(net);
555
556 mutex_lock(&nf_ct_proto_mutex);
557 switch (nfproto) {
558 case NFPROTO_IPV4:
559 if (cnet->users4 && (--cnet->users4 == 0)) {
560 nf_unregister_net_hooks(net, ipv4_conntrack_ops,
561 ARRAY_SIZE(ipv4_conntrack_ops));
562 nf_defrag_ipv4_disable(net);
563 }
564 break;
565 #if IS_ENABLED(CONFIG_IPV6)
566 case NFPROTO_IPV6:
567 if (cnet->users6 && (--cnet->users6 == 0)) {
568 nf_unregister_net_hooks(net, ipv6_conntrack_ops,
569 ARRAY_SIZE(ipv6_conntrack_ops));
570 nf_defrag_ipv6_disable(net);
571 }
572 break;
573 #endif
574 case NFPROTO_BRIDGE:
575 if (!nf_ct_bridge_info)
576 break;
577 if (cnet->users_bridge && (--cnet->users_bridge == 0))
578 nf_unregister_net_hooks(net, nf_ct_bridge_info->ops,
579 nf_ct_bridge_info->ops_size);
580
581 module_put(nf_ct_bridge_info->me);
582 break;
583 }
584 mutex_unlock(&nf_ct_proto_mutex);
585 }
586
nf_ct_netns_inet_get(struct net * net)587 static int nf_ct_netns_inet_get(struct net *net)
588 {
589 int err;
590
591 err = nf_ct_netns_do_get(net, NFPROTO_IPV4);
592 #if IS_ENABLED(CONFIG_IPV6)
593 if (err < 0)
594 goto err1;
595 err = nf_ct_netns_do_get(net, NFPROTO_IPV6);
596 if (err < 0)
597 goto err2;
598
599 return err;
600 err2:
601 nf_ct_netns_put(net, NFPROTO_IPV4);
602 err1:
603 #endif
604 return err;
605 }
606
nf_ct_netns_get(struct net * net,u8 nfproto)607 int nf_ct_netns_get(struct net *net, u8 nfproto)
608 {
609 int err;
610
611 switch (nfproto) {
612 case NFPROTO_INET:
613 err = nf_ct_netns_inet_get(net);
614 break;
615 case NFPROTO_BRIDGE:
616 err = nf_ct_netns_do_get(net, NFPROTO_BRIDGE);
617 if (err < 0)
618 return err;
619
620 err = nf_ct_netns_inet_get(net);
621 if (err < 0) {
622 nf_ct_netns_put(net, NFPROTO_BRIDGE);
623 return err;
624 }
625 break;
626 default:
627 err = nf_ct_netns_do_get(net, nfproto);
628 break;
629 }
630 return err;
631 }
632 EXPORT_SYMBOL_GPL(nf_ct_netns_get);
633
nf_ct_netns_put(struct net * net,uint8_t nfproto)634 void nf_ct_netns_put(struct net *net, uint8_t nfproto)
635 {
636 switch (nfproto) {
637 case NFPROTO_BRIDGE:
638 nf_ct_netns_do_put(net, NFPROTO_BRIDGE);
639 fallthrough;
640 case NFPROTO_INET:
641 nf_ct_netns_do_put(net, NFPROTO_IPV4);
642 nf_ct_netns_do_put(net, NFPROTO_IPV6);
643 break;
644 default:
645 nf_ct_netns_do_put(net, nfproto);
646 break;
647 }
648 }
649 EXPORT_SYMBOL_GPL(nf_ct_netns_put);
650
nf_ct_bridge_register(struct nf_ct_bridge_info * info)651 void nf_ct_bridge_register(struct nf_ct_bridge_info *info)
652 {
653 WARN_ON(nf_ct_bridge_info);
654 mutex_lock(&nf_ct_proto_mutex);
655 nf_ct_bridge_info = info;
656 mutex_unlock(&nf_ct_proto_mutex);
657 }
658 EXPORT_SYMBOL_GPL(nf_ct_bridge_register);
659
nf_ct_bridge_unregister(struct nf_ct_bridge_info * info)660 void nf_ct_bridge_unregister(struct nf_ct_bridge_info *info)
661 {
662 WARN_ON(!nf_ct_bridge_info);
663 mutex_lock(&nf_ct_proto_mutex);
664 nf_ct_bridge_info = NULL;
665 mutex_unlock(&nf_ct_proto_mutex);
666 }
667 EXPORT_SYMBOL_GPL(nf_ct_bridge_unregister);
668
nf_conntrack_proto_init(void)669 int nf_conntrack_proto_init(void)
670 {
671 int ret;
672
673 ret = nf_register_sockopt(&so_getorigdst);
674 if (ret < 0)
675 return ret;
676
677 #if IS_ENABLED(CONFIG_IPV6)
678 ret = nf_register_sockopt(&so_getorigdst6);
679 if (ret < 0)
680 goto cleanup_sockopt;
681 #endif
682
683 return ret;
684
685 #if IS_ENABLED(CONFIG_IPV6)
686 cleanup_sockopt:
687 nf_unregister_sockopt(&so_getorigdst);
688 #endif
689 return ret;
690 }
691
nf_conntrack_proto_fini(void)692 void nf_conntrack_proto_fini(void)
693 {
694 nf_unregister_sockopt(&so_getorigdst);
695 #if IS_ENABLED(CONFIG_IPV6)
696 nf_unregister_sockopt(&so_getorigdst6);
697 #endif
698 }
699
nf_conntrack_proto_pernet_init(struct net * net)700 void nf_conntrack_proto_pernet_init(struct net *net)
701 {
702 nf_conntrack_generic_init_net(net);
703 nf_conntrack_udp_init_net(net);
704 nf_conntrack_tcp_init_net(net);
705 nf_conntrack_icmp_init_net(net);
706 #if IS_ENABLED(CONFIG_IPV6)
707 nf_conntrack_icmpv6_init_net(net);
708 #endif
709 #ifdef CONFIG_NF_CT_PROTO_DCCP
710 nf_conntrack_dccp_init_net(net);
711 #endif
712 #ifdef CONFIG_NF_CT_PROTO_SCTP
713 nf_conntrack_sctp_init_net(net);
714 #endif
715 #ifdef CONFIG_NF_CT_PROTO_GRE
716 nf_conntrack_gre_init_net(net);
717 #endif
718 }
719
720 module_param_call(hashsize, nf_conntrack_set_hashsize, param_get_uint,
721 &nf_conntrack_htable_size, 0600);
722
723 MODULE_ALIAS("ip_conntrack");
724 MODULE_ALIAS("nf_conntrack-" __stringify(AF_INET));
725 MODULE_ALIAS("nf_conntrack-" __stringify(AF_INET6));
726 MODULE_LICENSE("GPL");
727