1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22#define KMSG_COMPONENT "IPVS"
23#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
24
25#include <linux/module.h>
26#include <linux/kernel.h>
27#include <linux/ip.h>
28#include <linux/tcp.h>
29#include <linux/sctp.h>
30#include <linux/icmp.h>
31#include <linux/slab.h>
32
33#include <net/ip.h>
34#include <net/tcp.h>
35#include <net/udp.h>
36#include <net/icmp.h>
37#include <net/gue.h>
38#include <net/gre.h>
39#include <net/route.h>
40#include <net/ip6_checksum.h>
41#include <net/netns/generic.h>
42
43#include <linux/netfilter.h>
44#include <linux/netfilter_ipv4.h>
45
46#ifdef CONFIG_IP_VS_IPV6
47#include <net/ipv6.h>
48#include <linux/netfilter_ipv6.h>
49#include <net/ip6_route.h>
50#endif
51
52#include <net/ip_vs.h>
53#include <linux/indirect_call_wrapper.h>
54
55
56EXPORT_SYMBOL(register_ip_vs_scheduler);
57EXPORT_SYMBOL(unregister_ip_vs_scheduler);
58EXPORT_SYMBOL(ip_vs_proto_name);
59EXPORT_SYMBOL(ip_vs_conn_new);
60EXPORT_SYMBOL(ip_vs_conn_in_get);
61EXPORT_SYMBOL(ip_vs_conn_out_get);
62#ifdef CONFIG_IP_VS_PROTO_TCP
63EXPORT_SYMBOL(ip_vs_tcp_conn_listen);
64#endif
65EXPORT_SYMBOL(ip_vs_conn_put);
66#ifdef CONFIG_IP_VS_DEBUG
67EXPORT_SYMBOL(ip_vs_get_debug_level);
68#endif
69EXPORT_SYMBOL(ip_vs_new_conn_out);
70
71#if defined(CONFIG_IP_VS_PROTO_TCP) && defined(CONFIG_IP_VS_PROTO_UDP)
72#define SNAT_CALL(f, ...) \
73 INDIRECT_CALL_2(f, tcp_snat_handler, udp_snat_handler, __VA_ARGS__)
74#elif defined(CONFIG_IP_VS_PROTO_TCP)
75#define SNAT_CALL(f, ...) INDIRECT_CALL_1(f, tcp_snat_handler, __VA_ARGS__)
76#elif defined(CONFIG_IP_VS_PROTO_UDP)
77#define SNAT_CALL(f, ...) INDIRECT_CALL_1(f, udp_snat_handler, __VA_ARGS__)
78#else
79#define SNAT_CALL(f, ...) f(__VA_ARGS__)
80#endif
81
82static unsigned int ip_vs_net_id __read_mostly;
83
84static atomic_t ipvs_netns_cnt = ATOMIC_INIT(0);
85
86
87#define icmp_id(icmph) (((icmph)->un).echo.id)
88#define icmpv6_id(icmph) (icmph->icmp6_dataun.u_echo.identifier)
89
90const char *ip_vs_proto_name(unsigned int proto)
91{
92 static char buf[20];
93
94 switch (proto) {
95 case IPPROTO_IP:
96 return "IP";
97 case IPPROTO_UDP:
98 return "UDP";
99 case IPPROTO_TCP:
100 return "TCP";
101 case IPPROTO_SCTP:
102 return "SCTP";
103 case IPPROTO_ICMP:
104 return "ICMP";
105#ifdef CONFIG_IP_VS_IPV6
106 case IPPROTO_ICMPV6:
107 return "ICMPv6";
108#endif
109 default:
110 sprintf(buf, "IP_%u", proto);
111 return buf;
112 }
113}
114
115void ip_vs_init_hash_table(struct list_head *table, int rows)
116{
117 while (--rows >= 0)
118 INIT_LIST_HEAD(&table[rows]);
119}
120
121static inline void
122ip_vs_in_stats(struct ip_vs_conn *cp, struct sk_buff *skb)
123{
124 struct ip_vs_dest *dest = cp->dest;
125 struct netns_ipvs *ipvs = cp->ipvs;
126
127 if (dest && (dest->flags & IP_VS_DEST_F_AVAILABLE)) {
128 struct ip_vs_cpu_stats *s;
129 struct ip_vs_service *svc;
130
131 local_bh_disable();
132
133 s = this_cpu_ptr(dest->stats.cpustats);
134 u64_stats_update_begin(&s->syncp);
135 s->cnt.inpkts++;
136 s->cnt.inbytes += skb->len;
137 u64_stats_update_end(&s->syncp);
138
139 svc = rcu_dereference(dest->svc);
140 s = this_cpu_ptr(svc->stats.cpustats);
141 u64_stats_update_begin(&s->syncp);
142 s->cnt.inpkts++;
143 s->cnt.inbytes += skb->len;
144 u64_stats_update_end(&s->syncp);
145
146 s = this_cpu_ptr(ipvs->tot_stats.cpustats);
147 u64_stats_update_begin(&s->syncp);
148 s->cnt.inpkts++;
149 s->cnt.inbytes += skb->len;
150 u64_stats_update_end(&s->syncp);
151
152 local_bh_enable();
153 }
154}
155
156
157static inline void
158ip_vs_out_stats(struct ip_vs_conn *cp, struct sk_buff *skb)
159{
160 struct ip_vs_dest *dest = cp->dest;
161 struct netns_ipvs *ipvs = cp->ipvs;
162
163 if (dest && (dest->flags & IP_VS_DEST_F_AVAILABLE)) {
164 struct ip_vs_cpu_stats *s;
165 struct ip_vs_service *svc;
166
167 local_bh_disable();
168
169 s = this_cpu_ptr(dest->stats.cpustats);
170 u64_stats_update_begin(&s->syncp);
171 s->cnt.outpkts++;
172 s->cnt.outbytes += skb->len;
173 u64_stats_update_end(&s->syncp);
174
175 svc = rcu_dereference(dest->svc);
176 s = this_cpu_ptr(svc->stats.cpustats);
177 u64_stats_update_begin(&s->syncp);
178 s->cnt.outpkts++;
179 s->cnt.outbytes += skb->len;
180 u64_stats_update_end(&s->syncp);
181
182 s = this_cpu_ptr(ipvs->tot_stats.cpustats);
183 u64_stats_update_begin(&s->syncp);
184 s->cnt.outpkts++;
185 s->cnt.outbytes += skb->len;
186 u64_stats_update_end(&s->syncp);
187
188 local_bh_enable();
189 }
190}
191
192
193static inline void
194ip_vs_conn_stats(struct ip_vs_conn *cp, struct ip_vs_service *svc)
195{
196 struct netns_ipvs *ipvs = svc->ipvs;
197 struct ip_vs_cpu_stats *s;
198
199 local_bh_disable();
200
201 s = this_cpu_ptr(cp->dest->stats.cpustats);
202 u64_stats_update_begin(&s->syncp);
203 s->cnt.conns++;
204 u64_stats_update_end(&s->syncp);
205
206 s = this_cpu_ptr(svc->stats.cpustats);
207 u64_stats_update_begin(&s->syncp);
208 s->cnt.conns++;
209 u64_stats_update_end(&s->syncp);
210
211 s = this_cpu_ptr(ipvs->tot_stats.cpustats);
212 u64_stats_update_begin(&s->syncp);
213 s->cnt.conns++;
214 u64_stats_update_end(&s->syncp);
215
216 local_bh_enable();
217}
218
219
220static inline void
221ip_vs_set_state(struct ip_vs_conn *cp, int direction,
222 const struct sk_buff *skb,
223 struct ip_vs_proto_data *pd)
224{
225 if (likely(pd->pp->state_transition))
226 pd->pp->state_transition(cp, direction, skb, pd);
227}
228
229static inline int
230ip_vs_conn_fill_param_persist(const struct ip_vs_service *svc,
231 struct sk_buff *skb, int protocol,
232 const union nf_inet_addr *caddr, __be16 cport,
233 const union nf_inet_addr *vaddr, __be16 vport,
234 struct ip_vs_conn_param *p)
235{
236 ip_vs_conn_fill_param(svc->ipvs, svc->af, protocol, caddr, cport, vaddr,
237 vport, p);
238 p->pe = rcu_dereference(svc->pe);
239 if (p->pe && p->pe->fill_param)
240 return p->pe->fill_param(p, skb);
241
242 return 0;
243}
244
245
246
247
248
249
250
251
252static struct ip_vs_conn *
253ip_vs_sched_persist(struct ip_vs_service *svc,
254 struct sk_buff *skb, __be16 src_port, __be16 dst_port,
255 int *ignored, struct ip_vs_iphdr *iph)
256{
257 struct ip_vs_conn *cp = NULL;
258 struct ip_vs_dest *dest;
259 struct ip_vs_conn *ct;
260 __be16 dport = 0;
261 unsigned int flags;
262 struct ip_vs_conn_param param;
263 const union nf_inet_addr fwmark = { .ip = htonl(svc->fwmark) };
264 union nf_inet_addr snet;
265
266 const union nf_inet_addr *src_addr, *dst_addr;
267
268 if (likely(!ip_vs_iph_inverse(iph))) {
269 src_addr = &iph->saddr;
270 dst_addr = &iph->daddr;
271 } else {
272 src_addr = &iph->daddr;
273 dst_addr = &iph->saddr;
274 }
275
276
277
278#ifdef CONFIG_IP_VS_IPV6
279 if (svc->af == AF_INET6)
280 ipv6_addr_prefix(&snet.in6, &src_addr->in6,
281 (__force __u32) svc->netmask);
282 else
283#endif
284 snet.ip = src_addr->ip & svc->netmask;
285
286 IP_VS_DBG_BUF(6, "p-schedule: src %s:%u dest %s:%u "
287 "mnet %s\n",
288 IP_VS_DBG_ADDR(svc->af, src_addr), ntohs(src_port),
289 IP_VS_DBG_ADDR(svc->af, dst_addr), ntohs(dst_port),
290 IP_VS_DBG_ADDR(svc->af, &snet));
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305 {
306 int protocol = iph->protocol;
307 const union nf_inet_addr *vaddr = dst_addr;
308 __be16 vport = 0;
309
310 if (dst_port == svc->port) {
311
312
313
314
315
316 if (svc->port != FTPPORT)
317 vport = dst_port;
318 } else {
319
320
321
322
323
324
325
326 if (svc->fwmark) {
327 protocol = IPPROTO_IP;
328 vaddr = &fwmark;
329 }
330 }
331
332 if (ip_vs_conn_fill_param_persist(svc, skb, protocol, &snet, 0,
333 vaddr, vport, ¶m) < 0) {
334 *ignored = -1;
335 return NULL;
336 }
337 }
338
339
340 ct = ip_vs_ct_in_get(¶m);
341 if (!ct || !ip_vs_check_template(ct, NULL)) {
342 struct ip_vs_scheduler *sched;
343
344
345
346
347
348
349 sched = rcu_dereference(svc->scheduler);
350 if (sched) {
351
352 smp_rmb();
353 dest = sched->schedule(svc, skb, iph);
354 } else {
355 dest = NULL;
356 }
357 if (!dest) {
358 IP_VS_DBG(1, "p-schedule: no dest found.\n");
359 kfree(param.pe_data);
360 *ignored = 0;
361 return NULL;
362 }
363
364 if (dst_port == svc->port && svc->port != FTPPORT)
365 dport = dest->port;
366
367
368
369
370
371 ct = ip_vs_conn_new(¶m, dest->af, &dest->addr, dport,
372 IP_VS_CONN_F_TEMPLATE, dest, skb->mark);
373 if (ct == NULL) {
374 kfree(param.pe_data);
375 *ignored = -1;
376 return NULL;
377 }
378
379 ct->timeout = svc->timeout;
380 } else {
381
382 dest = ct->dest;
383 kfree(param.pe_data);
384 }
385
386 dport = dst_port;
387 if (dport == svc->port && dest->port)
388 dport = dest->port;
389
390 flags = (svc->flags & IP_VS_SVC_F_ONEPACKET
391 && iph->protocol == IPPROTO_UDP) ?
392 IP_VS_CONN_F_ONE_PACKET : 0;
393
394
395
396
397 ip_vs_conn_fill_param(svc->ipvs, svc->af, iph->protocol, src_addr,
398 src_port, dst_addr, dst_port, ¶m);
399
400 cp = ip_vs_conn_new(¶m, dest->af, &dest->addr, dport, flags, dest,
401 skb->mark);
402 if (cp == NULL) {
403 ip_vs_conn_put(ct);
404 *ignored = -1;
405 return NULL;
406 }
407
408
409
410
411 ip_vs_control_add(cp, ct);
412 ip_vs_conn_put(ct);
413
414 ip_vs_conn_stats(cp, svc);
415 return cp;
416}
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440struct ip_vs_conn *
441ip_vs_schedule(struct ip_vs_service *svc, struct sk_buff *skb,
442 struct ip_vs_proto_data *pd, int *ignored,
443 struct ip_vs_iphdr *iph)
444{
445 struct ip_vs_protocol *pp = pd->pp;
446 struct ip_vs_conn *cp = NULL;
447 struct ip_vs_scheduler *sched;
448 struct ip_vs_dest *dest;
449 __be16 _ports[2], *pptr, cport, vport;
450 const void *caddr, *vaddr;
451 unsigned int flags;
452
453 *ignored = 1;
454
455
456
457 pptr = frag_safe_skb_hp(skb, iph->len, sizeof(_ports), _ports);
458 if (pptr == NULL)
459 return NULL;
460
461 if (likely(!ip_vs_iph_inverse(iph))) {
462 cport = pptr[0];
463 caddr = &iph->saddr;
464 vport = pptr[1];
465 vaddr = &iph->daddr;
466 } else {
467 cport = pptr[1];
468 caddr = &iph->daddr;
469 vport = pptr[0];
470 vaddr = &iph->saddr;
471 }
472
473
474
475
476
477
478
479 if (cport == FTPDATA) {
480 IP_VS_DBG_PKT(12, svc->af, pp, skb, iph->off,
481 "Not scheduling FTPDATA");
482 return NULL;
483 }
484
485
486
487
488 if ((!skb->dev || skb->dev->flags & IFF_LOOPBACK)) {
489 iph->hdr_flags ^= IP_VS_HDR_INVERSE;
490 cp = INDIRECT_CALL_1(pp->conn_in_get,
491 ip_vs_conn_in_get_proto, svc->ipvs,
492 svc->af, skb, iph);
493 iph->hdr_flags ^= IP_VS_HDR_INVERSE;
494
495 if (cp) {
496 IP_VS_DBG_PKT(12, svc->af, pp, skb, iph->off,
497 "Not scheduling reply for existing"
498 " connection");
499 __ip_vs_conn_put(cp);
500 return NULL;
501 }
502 }
503
504
505
506
507 if (svc->flags & IP_VS_SVC_F_PERSISTENT)
508 return ip_vs_sched_persist(svc, skb, cport, vport, ignored,
509 iph);
510
511 *ignored = 0;
512
513
514
515
516 if (!svc->fwmark && vport != svc->port) {
517 if (!svc->port)
518 pr_err("Schedule: port zero only supported "
519 "in persistent services, "
520 "check your ipvs configuration\n");
521 return NULL;
522 }
523
524 sched = rcu_dereference(svc->scheduler);
525 if (sched) {
526
527 smp_rmb();
528 dest = sched->schedule(svc, skb, iph);
529 } else {
530 dest = NULL;
531 }
532 if (dest == NULL) {
533 IP_VS_DBG(1, "Schedule: no dest found.\n");
534 return NULL;
535 }
536
537 flags = (svc->flags & IP_VS_SVC_F_ONEPACKET
538 && iph->protocol == IPPROTO_UDP) ?
539 IP_VS_CONN_F_ONE_PACKET : 0;
540
541
542
543
544 {
545 struct ip_vs_conn_param p;
546
547 ip_vs_conn_fill_param(svc->ipvs, svc->af, iph->protocol,
548 caddr, cport, vaddr, vport, &p);
549 cp = ip_vs_conn_new(&p, dest->af, &dest->addr,
550 dest->port ? dest->port : vport,
551 flags, dest, skb->mark);
552 if (!cp) {
553 *ignored = -1;
554 return NULL;
555 }
556 }
557
558 IP_VS_DBG_BUF(6, "Schedule fwd:%c c:%s:%u v:%s:%u "
559 "d:%s:%u conn->flags:%X conn->refcnt:%d\n",
560 ip_vs_fwd_tag(cp),
561 IP_VS_DBG_ADDR(cp->af, &cp->caddr), ntohs(cp->cport),
562 IP_VS_DBG_ADDR(cp->af, &cp->vaddr), ntohs(cp->vport),
563 IP_VS_DBG_ADDR(cp->daf, &cp->daddr), ntohs(cp->dport),
564 cp->flags, refcount_read(&cp->refcnt));
565
566 ip_vs_conn_stats(cp, svc);
567 return cp;
568}
569
570static inline int ip_vs_addr_is_unicast(struct net *net, int af,
571 union nf_inet_addr *addr)
572{
573#ifdef CONFIG_IP_VS_IPV6
574 if (af == AF_INET6)
575 return ipv6_addr_type(&addr->in6) & IPV6_ADDR_UNICAST;
576#endif
577 return (inet_addr_type(net, addr->ip) == RTN_UNICAST);
578}
579
580
581
582
583
584
585int ip_vs_leave(struct ip_vs_service *svc, struct sk_buff *skb,
586 struct ip_vs_proto_data *pd, struct ip_vs_iphdr *iph)
587{
588 __be16 _ports[2], *pptr, dport;
589 struct netns_ipvs *ipvs = svc->ipvs;
590 struct net *net = ipvs->net;
591
592 pptr = frag_safe_skb_hp(skb, iph->len, sizeof(_ports), _ports);
593 if (!pptr)
594 return NF_DROP;
595 dport = likely(!ip_vs_iph_inverse(iph)) ? pptr[1] : pptr[0];
596
597
598
599
600 if (sysctl_cache_bypass(ipvs) && svc->fwmark &&
601 !(iph->hdr_flags & (IP_VS_HDR_INVERSE | IP_VS_HDR_ICMP)) &&
602 ip_vs_addr_is_unicast(net, svc->af, &iph->daddr)) {
603 int ret;
604 struct ip_vs_conn *cp;
605 unsigned int flags = (svc->flags & IP_VS_SVC_F_ONEPACKET &&
606 iph->protocol == IPPROTO_UDP) ?
607 IP_VS_CONN_F_ONE_PACKET : 0;
608 union nf_inet_addr daddr = { .all = { 0, 0, 0, 0 } };
609
610
611 IP_VS_DBG(6, "%s(): create a cache_bypass entry\n", __func__);
612 {
613 struct ip_vs_conn_param p;
614 ip_vs_conn_fill_param(svc->ipvs, svc->af, iph->protocol,
615 &iph->saddr, pptr[0],
616 &iph->daddr, pptr[1], &p);
617 cp = ip_vs_conn_new(&p, svc->af, &daddr, 0,
618 IP_VS_CONN_F_BYPASS | flags,
619 NULL, skb->mark);
620 if (!cp)
621 return NF_DROP;
622 }
623
624
625 ip_vs_in_stats(cp, skb);
626
627
628 ip_vs_set_state(cp, IP_VS_DIR_INPUT, skb, pd);
629
630
631 ret = cp->packet_xmit(skb, cp, pd->pp, iph);
632
633
634 if ((cp->flags & IP_VS_CONN_F_ONE_PACKET) && cp->control)
635 atomic_inc(&cp->control->in_pkts);
636 else
637 atomic_inc(&cp->in_pkts);
638 ip_vs_conn_put(cp);
639 return ret;
640 }
641
642
643
644
645
646
647
648 if (svc->port == FTPPORT && dport != FTPPORT)
649 return NF_ACCEPT;
650
651 if (unlikely(ip_vs_iph_icmp(iph)))
652 return NF_DROP;
653
654
655
656
657
658
659
660
661#ifdef CONFIG_IP_VS_IPV6
662 if (svc->af == AF_INET6) {
663 if (!skb->dev)
664 skb->dev = net->loopback_dev;
665 icmpv6_send(skb, ICMPV6_DEST_UNREACH, ICMPV6_PORT_UNREACH, 0);
666 } else
667#endif
668 icmp_send(skb, ICMP_DEST_UNREACH, ICMP_PORT_UNREACH, 0);
669
670 return NF_DROP;
671}
672
673#ifdef CONFIG_SYSCTL
674
675static int sysctl_snat_reroute(struct netns_ipvs *ipvs)
676{
677 return ipvs->sysctl_snat_reroute;
678}
679
680static int sysctl_nat_icmp_send(struct netns_ipvs *ipvs)
681{
682 return ipvs->sysctl_nat_icmp_send;
683}
684
685#else
686
687static int sysctl_snat_reroute(struct netns_ipvs *ipvs) { return 0; }
688static int sysctl_nat_icmp_send(struct netns_ipvs *ipvs) { return 0; }
689
690#endif
691
692__sum16 ip_vs_checksum_complete(struct sk_buff *skb, int offset)
693{
694 return csum_fold(skb_checksum(skb, offset, skb->len - offset, 0));
695}
696
697static inline enum ip_defrag_users ip_vs_defrag_user(unsigned int hooknum)
698{
699 if (NF_INET_LOCAL_IN == hooknum)
700 return IP_DEFRAG_VS_IN;
701 if (NF_INET_FORWARD == hooknum)
702 return IP_DEFRAG_VS_FWD;
703 return IP_DEFRAG_VS_OUT;
704}
705
706static inline int ip_vs_gather_frags(struct netns_ipvs *ipvs,
707 struct sk_buff *skb, u_int32_t user)
708{
709 int err;
710
711 local_bh_disable();
712 err = ip_defrag(ipvs->net, skb, user);
713 local_bh_enable();
714 if (!err)
715 ip_send_check(ip_hdr(skb));
716
717 return err;
718}
719
720static int ip_vs_route_me_harder(struct netns_ipvs *ipvs, int af,
721 struct sk_buff *skb, unsigned int hooknum)
722{
723 if (!sysctl_snat_reroute(ipvs))
724 return 0;
725
726 if (NF_INET_LOCAL_IN == hooknum)
727 return 0;
728#ifdef CONFIG_IP_VS_IPV6
729 if (af == AF_INET6) {
730 struct dst_entry *dst = skb_dst(skb);
731
732 if (dst->dev && !(dst->dev->flags & IFF_LOOPBACK) &&
733 ip6_route_me_harder(ipvs->net, skb->sk, skb) != 0)
734 return 1;
735 } else
736#endif
737 if (!(skb_rtable(skb)->rt_flags & RTCF_LOCAL) &&
738 ip_route_me_harder(ipvs->net, skb->sk, skb, RTN_LOCAL) != 0)
739 return 1;
740
741 return 0;
742}
743
744
745
746
747
748void ip_vs_nat_icmp(struct sk_buff *skb, struct ip_vs_protocol *pp,
749 struct ip_vs_conn *cp, int inout)
750{
751 struct iphdr *iph = ip_hdr(skb);
752 unsigned int icmp_offset = iph->ihl*4;
753 struct icmphdr *icmph = (struct icmphdr *)(skb_network_header(skb) +
754 icmp_offset);
755 struct iphdr *ciph = (struct iphdr *)(icmph + 1);
756
757 if (inout) {
758 iph->saddr = cp->vaddr.ip;
759 ip_send_check(iph);
760 ciph->daddr = cp->vaddr.ip;
761 ip_send_check(ciph);
762 } else {
763 iph->daddr = cp->daddr.ip;
764 ip_send_check(iph);
765 ciph->saddr = cp->daddr.ip;
766 ip_send_check(ciph);
767 }
768
769
770 if (IPPROTO_TCP == ciph->protocol || IPPROTO_UDP == ciph->protocol ||
771 IPPROTO_SCTP == ciph->protocol) {
772 __be16 *ports = (void *)ciph + ciph->ihl*4;
773
774 if (inout)
775 ports[1] = cp->vport;
776 else
777 ports[0] = cp->dport;
778 }
779
780
781 icmph->checksum = 0;
782 icmph->checksum = ip_vs_checksum_complete(skb, icmp_offset);
783 skb->ip_summed = CHECKSUM_UNNECESSARY;
784
785 if (inout)
786 IP_VS_DBG_PKT(11, AF_INET, pp, skb, (void *)ciph - (void *)iph,
787 "Forwarding altered outgoing ICMP");
788 else
789 IP_VS_DBG_PKT(11, AF_INET, pp, skb, (void *)ciph - (void *)iph,
790 "Forwarding altered incoming ICMP");
791}
792
793#ifdef CONFIG_IP_VS_IPV6
794void ip_vs_nat_icmp_v6(struct sk_buff *skb, struct ip_vs_protocol *pp,
795 struct ip_vs_conn *cp, int inout)
796{
797 struct ipv6hdr *iph = ipv6_hdr(skb);
798 unsigned int icmp_offset = 0;
799 unsigned int offs = 0;
800 int protocol;
801 struct icmp6hdr *icmph;
802 struct ipv6hdr *ciph;
803 unsigned short fragoffs;
804
805 ipv6_find_hdr(skb, &icmp_offset, IPPROTO_ICMPV6, &fragoffs, NULL);
806 icmph = (struct icmp6hdr *)(skb_network_header(skb) + icmp_offset);
807 offs = icmp_offset + sizeof(struct icmp6hdr);
808 ciph = (struct ipv6hdr *)(skb_network_header(skb) + offs);
809
810 protocol = ipv6_find_hdr(skb, &offs, -1, &fragoffs, NULL);
811
812 if (inout) {
813 iph->saddr = cp->vaddr.in6;
814 ciph->daddr = cp->vaddr.in6;
815 } else {
816 iph->daddr = cp->daddr.in6;
817 ciph->saddr = cp->daddr.in6;
818 }
819
820
821 if (!fragoffs && (IPPROTO_TCP == protocol || IPPROTO_UDP == protocol ||
822 IPPROTO_SCTP == protocol)) {
823 __be16 *ports = (void *)(skb_network_header(skb) + offs);
824
825 IP_VS_DBG(11, "%s() changed port %d to %d\n", __func__,
826 ntohs(inout ? ports[1] : ports[0]),
827 ntohs(inout ? cp->vport : cp->dport));
828 if (inout)
829 ports[1] = cp->vport;
830 else
831 ports[0] = cp->dport;
832 }
833
834
835 icmph->icmp6_cksum = ~csum_ipv6_magic(&iph->saddr, &iph->daddr,
836 skb->len - icmp_offset,
837 IPPROTO_ICMPV6, 0);
838 skb->csum_start = skb_network_header(skb) - skb->head + icmp_offset;
839 skb->csum_offset = offsetof(struct icmp6hdr, icmp6_cksum);
840 skb->ip_summed = CHECKSUM_PARTIAL;
841
842 if (inout)
843 IP_VS_DBG_PKT(11, AF_INET6, pp, skb,
844 (void *)ciph - (void *)iph,
845 "Forwarding altered outgoing ICMPv6");
846 else
847 IP_VS_DBG_PKT(11, AF_INET6, pp, skb,
848 (void *)ciph - (void *)iph,
849 "Forwarding altered incoming ICMPv6");
850}
851#endif
852
853
854
855
856static int handle_response_icmp(int af, struct sk_buff *skb,
857 union nf_inet_addr *snet,
858 __u8 protocol, struct ip_vs_conn *cp,
859 struct ip_vs_protocol *pp,
860 unsigned int offset, unsigned int ihl,
861 unsigned int hooknum)
862{
863 unsigned int verdict = NF_DROP;
864
865 if (IP_VS_FWD_METHOD(cp) != IP_VS_CONN_F_MASQ)
866 goto after_nat;
867
868
869 if (!skb_csum_unnecessary(skb) && ip_vs_checksum_complete(skb, ihl)) {
870
871 IP_VS_DBG_BUF(1, "Forward ICMP: failed checksum from %s!\n",
872 IP_VS_DBG_ADDR(af, snet));
873 goto out;
874 }
875
876 if (IPPROTO_TCP == protocol || IPPROTO_UDP == protocol ||
877 IPPROTO_SCTP == protocol)
878 offset += 2 * sizeof(__u16);
879 if (skb_ensure_writable(skb, offset))
880 goto out;
881
882#ifdef CONFIG_IP_VS_IPV6
883 if (af == AF_INET6)
884 ip_vs_nat_icmp_v6(skb, pp, cp, 1);
885 else
886#endif
887 ip_vs_nat_icmp(skb, pp, cp, 1);
888
889 if (ip_vs_route_me_harder(cp->ipvs, af, skb, hooknum))
890 goto out;
891
892after_nat:
893
894 ip_vs_out_stats(cp, skb);
895
896 skb->ipvs_property = 1;
897 if (!(cp->flags & IP_VS_CONN_F_NFCT))
898 ip_vs_notrack(skb);
899 else
900 ip_vs_update_conntrack(skb, cp, 0);
901 verdict = NF_ACCEPT;
902
903out:
904 __ip_vs_conn_put(cp);
905
906 return verdict;
907}
908
909
910
911
912
913
914static int ip_vs_out_icmp(struct netns_ipvs *ipvs, struct sk_buff *skb,
915 int *related, unsigned int hooknum)
916{
917 struct iphdr *iph;
918 struct icmphdr _icmph, *ic;
919 struct iphdr _ciph, *cih;
920 struct ip_vs_iphdr ciph;
921 struct ip_vs_conn *cp;
922 struct ip_vs_protocol *pp;
923 unsigned int offset, ihl;
924 union nf_inet_addr snet;
925
926 *related = 1;
927
928
929 if (ip_is_fragment(ip_hdr(skb))) {
930 if (ip_vs_gather_frags(ipvs, skb, ip_vs_defrag_user(hooknum)))
931 return NF_STOLEN;
932 }
933
934 iph = ip_hdr(skb);
935 offset = ihl = iph->ihl * 4;
936 ic = skb_header_pointer(skb, offset, sizeof(_icmph), &_icmph);
937 if (ic == NULL)
938 return NF_DROP;
939
940 IP_VS_DBG(12, "Outgoing ICMP (%d,%d) %pI4->%pI4\n",
941 ic->type, ntohs(icmp_id(ic)),
942 &iph->saddr, &iph->daddr);
943
944
945
946
947
948
949
950
951 if ((ic->type != ICMP_DEST_UNREACH) &&
952 (ic->type != ICMP_SOURCE_QUENCH) &&
953 (ic->type != ICMP_TIME_EXCEEDED)) {
954 *related = 0;
955 return NF_ACCEPT;
956 }
957
958
959 offset += sizeof(_icmph);
960 cih = skb_header_pointer(skb, offset, sizeof(_ciph), &_ciph);
961 if (cih == NULL)
962 return NF_ACCEPT;
963
964 pp = ip_vs_proto_get(cih->protocol);
965 if (!pp)
966 return NF_ACCEPT;
967
968
969 if (unlikely(cih->frag_off & htons(IP_OFFSET) &&
970 pp->dont_defrag))
971 return NF_ACCEPT;
972
973 IP_VS_DBG_PKT(11, AF_INET, pp, skb, offset,
974 "Checking outgoing ICMP for");
975
976 ip_vs_fill_iph_skb_icmp(AF_INET, skb, offset, true, &ciph);
977
978
979 cp = INDIRECT_CALL_1(pp->conn_out_get, ip_vs_conn_out_get_proto,
980 ipvs, AF_INET, skb, &ciph);
981 if (!cp)
982 return NF_ACCEPT;
983
984 snet.ip = iph->saddr;
985 return handle_response_icmp(AF_INET, skb, &snet, cih->protocol, cp,
986 pp, ciph.len, ihl, hooknum);
987}
988
989#ifdef CONFIG_IP_VS_IPV6
990static int ip_vs_out_icmp_v6(struct netns_ipvs *ipvs, struct sk_buff *skb,
991 int *related, unsigned int hooknum,
992 struct ip_vs_iphdr *ipvsh)
993{
994 struct icmp6hdr _icmph, *ic;
995 struct ip_vs_iphdr ciph = {.flags = 0, .fragoffs = 0};
996 struct ip_vs_conn *cp;
997 struct ip_vs_protocol *pp;
998 union nf_inet_addr snet;
999 unsigned int offset;
1000
1001 *related = 1;
1002 ic = frag_safe_skb_hp(skb, ipvsh->len, sizeof(_icmph), &_icmph);
1003 if (ic == NULL)
1004 return NF_DROP;
1005
1006
1007
1008
1009
1010
1011
1012
1013 if (ic->icmp6_type & ICMPV6_INFOMSG_MASK) {
1014 *related = 0;
1015 return NF_ACCEPT;
1016 }
1017
1018
1019
1020 if (ipvsh->flags & IP6_FH_F_FRAG)
1021 return NF_DROP;
1022
1023 IP_VS_DBG(8, "Outgoing ICMPv6 (%d,%d) %pI6c->%pI6c\n",
1024 ic->icmp6_type, ntohs(icmpv6_id(ic)),
1025 &ipvsh->saddr, &ipvsh->daddr);
1026
1027 if (!ip_vs_fill_iph_skb_icmp(AF_INET6, skb, ipvsh->len + sizeof(_icmph),
1028 true, &ciph))
1029 return NF_ACCEPT;
1030
1031 pp = ip_vs_proto_get(ciph.protocol);
1032 if (!pp)
1033 return NF_ACCEPT;
1034
1035
1036 cp = INDIRECT_CALL_1(pp->conn_out_get, ip_vs_conn_out_get_proto,
1037 ipvs, AF_INET6, skb, &ciph);
1038 if (!cp)
1039 return NF_ACCEPT;
1040
1041 snet.in6 = ciph.saddr.in6;
1042 offset = ciph.len;
1043 return handle_response_icmp(AF_INET6, skb, &snet, ciph.protocol, cp,
1044 pp, offset, sizeof(struct ipv6hdr),
1045 hooknum);
1046}
1047#endif
1048
1049
1050
1051
1052static inline int is_sctp_abort(const struct sk_buff *skb, int nh_len)
1053{
1054 struct sctp_chunkhdr *sch, schunk;
1055 sch = skb_header_pointer(skb, nh_len + sizeof(struct sctphdr),
1056 sizeof(schunk), &schunk);
1057 if (sch == NULL)
1058 return 0;
1059 if (sch->type == SCTP_CID_ABORT)
1060 return 1;
1061 return 0;
1062}
1063
1064static inline int is_tcp_reset(const struct sk_buff *skb, int nh_len)
1065{
1066 struct tcphdr _tcph, *th;
1067
1068 th = skb_header_pointer(skb, nh_len, sizeof(_tcph), &_tcph);
1069 if (th == NULL)
1070 return 0;
1071 return th->rst;
1072}
1073
1074static inline bool is_new_conn(const struct sk_buff *skb,
1075 struct ip_vs_iphdr *iph)
1076{
1077 switch (iph->protocol) {
1078 case IPPROTO_TCP: {
1079 struct tcphdr _tcph, *th;
1080
1081 th = skb_header_pointer(skb, iph->len, sizeof(_tcph), &_tcph);
1082 if (th == NULL)
1083 return false;
1084 return th->syn;
1085 }
1086 case IPPROTO_SCTP: {
1087 struct sctp_chunkhdr *sch, schunk;
1088
1089 sch = skb_header_pointer(skb, iph->len + sizeof(struct sctphdr),
1090 sizeof(schunk), &schunk);
1091 if (sch == NULL)
1092 return false;
1093 return sch->type == SCTP_CID_INIT;
1094 }
1095 default:
1096 return false;
1097 }
1098}
1099
1100static inline bool is_new_conn_expected(const struct ip_vs_conn *cp,
1101 int conn_reuse_mode)
1102{
1103
1104 if (cp->control)
1105 return false;
1106
1107 switch (cp->protocol) {
1108 case IPPROTO_TCP:
1109 return (cp->state == IP_VS_TCP_S_TIME_WAIT) ||
1110 (cp->state == IP_VS_TCP_S_CLOSE) ||
1111 ((conn_reuse_mode & 2) &&
1112 (cp->state == IP_VS_TCP_S_FIN_WAIT) &&
1113 (cp->flags & IP_VS_CONN_F_NOOUTPUT));
1114 case IPPROTO_SCTP:
1115 return cp->state == IP_VS_SCTP_S_CLOSED;
1116 default:
1117 return false;
1118 }
1119}
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129struct ip_vs_conn *ip_vs_new_conn_out(struct ip_vs_service *svc,
1130 struct ip_vs_dest *dest,
1131 struct sk_buff *skb,
1132 const struct ip_vs_iphdr *iph,
1133 __be16 dport,
1134 __be16 cport)
1135{
1136 struct ip_vs_conn_param param;
1137 struct ip_vs_conn *ct = NULL, *cp = NULL;
1138 const union nf_inet_addr *vaddr, *daddr, *caddr;
1139 union nf_inet_addr snet;
1140 __be16 vport;
1141 unsigned int flags;
1142
1143 EnterFunction(12);
1144 vaddr = &svc->addr;
1145 vport = svc->port;
1146 daddr = &iph->saddr;
1147 caddr = &iph->daddr;
1148
1149
1150 if (svc->fwmark)
1151 return NULL;
1152 if (!vport || !dport)
1153 return NULL;
1154
1155
1156 if (svc->flags & IP_VS_SVC_F_PERSISTENT) {
1157
1158#ifdef CONFIG_IP_VS_IPV6
1159 if (svc->af == AF_INET6)
1160 ipv6_addr_prefix(&snet.in6, &caddr->in6,
1161 (__force __u32)svc->netmask);
1162 else
1163#endif
1164 snet.ip = caddr->ip & svc->netmask;
1165
1166 if (ip_vs_conn_fill_param_persist(svc, skb, iph->protocol,
1167 &snet, 0, vaddr,
1168 vport, ¶m) < 0)
1169 return NULL;
1170 ct = ip_vs_ct_in_get(¶m);
1171
1172 if (!ct || !ip_vs_check_template(ct, dest)) {
1173 ct = ip_vs_conn_new(¶m, dest->af, daddr, dport,
1174 IP_VS_CONN_F_TEMPLATE, dest, 0);
1175 if (!ct) {
1176 kfree(param.pe_data);
1177 return NULL;
1178 }
1179 ct->timeout = svc->timeout;
1180 } else {
1181 kfree(param.pe_data);
1182 }
1183 }
1184
1185
1186 flags = ((svc->flags & IP_VS_SVC_F_ONEPACKET) &&
1187 iph->protocol == IPPROTO_UDP) ? IP_VS_CONN_F_ONE_PACKET : 0;
1188
1189 ip_vs_conn_fill_param(svc->ipvs, svc->af, iph->protocol,
1190 caddr, cport, vaddr, vport, ¶m);
1191 cp = ip_vs_conn_new(¶m, dest->af, daddr, dport, flags, dest, 0);
1192 if (!cp) {
1193 if (ct)
1194 ip_vs_conn_put(ct);
1195 return NULL;
1196 }
1197 if (ct) {
1198 ip_vs_control_add(cp, ct);
1199 ip_vs_conn_put(ct);
1200 }
1201 ip_vs_conn_stats(cp, svc);
1202
1203
1204 IP_VS_DBG_BUF(6, "New connection RS-initiated:%c c:%s:%u v:%s:%u "
1205 "d:%s:%u conn->flags:%X conn->refcnt:%d\n",
1206 ip_vs_fwd_tag(cp),
1207 IP_VS_DBG_ADDR(cp->af, &cp->caddr), ntohs(cp->cport),
1208 IP_VS_DBG_ADDR(cp->af, &cp->vaddr), ntohs(cp->vport),
1209 IP_VS_DBG_ADDR(cp->af, &cp->daddr), ntohs(cp->dport),
1210 cp->flags, refcount_read(&cp->refcnt));
1211 LeaveFunction(12);
1212 return cp;
1213}
1214
1215
1216
1217
1218
1219
1220
1221
1222static struct ip_vs_conn *__ip_vs_rs_conn_out(unsigned int hooknum,
1223 struct netns_ipvs *ipvs,
1224 int af, struct sk_buff *skb,
1225 const struct ip_vs_iphdr *iph)
1226{
1227 struct ip_vs_dest *dest;
1228 struct ip_vs_conn *cp = NULL;
1229 __be16 _ports[2], *pptr;
1230
1231 if (hooknum == NF_INET_LOCAL_IN)
1232 return NULL;
1233
1234 pptr = frag_safe_skb_hp(skb, iph->len,
1235 sizeof(_ports), _ports);
1236 if (!pptr)
1237 return NULL;
1238
1239 dest = ip_vs_find_real_service(ipvs, af, iph->protocol,
1240 &iph->saddr, pptr[0]);
1241 if (dest) {
1242 struct ip_vs_service *svc;
1243 struct ip_vs_pe *pe;
1244
1245 svc = rcu_dereference(dest->svc);
1246 if (svc) {
1247 pe = rcu_dereference(svc->pe);
1248 if (pe && pe->conn_out)
1249 cp = pe->conn_out(svc, dest, skb, iph,
1250 pptr[0], pptr[1]);
1251 }
1252 }
1253
1254 return cp;
1255}
1256
1257
1258
1259static unsigned int
1260handle_response(int af, struct sk_buff *skb, struct ip_vs_proto_data *pd,
1261 struct ip_vs_conn *cp, struct ip_vs_iphdr *iph,
1262 unsigned int hooknum)
1263{
1264 struct ip_vs_protocol *pp = pd->pp;
1265
1266 if (IP_VS_FWD_METHOD(cp) != IP_VS_CONN_F_MASQ)
1267 goto after_nat;
1268
1269 IP_VS_DBG_PKT(11, af, pp, skb, iph->off, "Outgoing packet");
1270
1271 if (skb_ensure_writable(skb, iph->len))
1272 goto drop;
1273
1274
1275 if (pp->snat_handler &&
1276 !SNAT_CALL(pp->snat_handler, skb, pp, cp, iph))
1277 goto drop;
1278
1279#ifdef CONFIG_IP_VS_IPV6
1280 if (af == AF_INET6)
1281 ipv6_hdr(skb)->saddr = cp->vaddr.in6;
1282 else
1283#endif
1284 {
1285 ip_hdr(skb)->saddr = cp->vaddr.ip;
1286 ip_send_check(ip_hdr(skb));
1287 }
1288
1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
1300
1301
1302
1303
1304 if (ip_vs_route_me_harder(cp->ipvs, af, skb, hooknum))
1305 goto drop;
1306
1307 IP_VS_DBG_PKT(10, af, pp, skb, iph->off, "After SNAT");
1308
1309after_nat:
1310 ip_vs_out_stats(cp, skb);
1311 ip_vs_set_state(cp, IP_VS_DIR_OUTPUT, skb, pd);
1312 skb->ipvs_property = 1;
1313 if (!(cp->flags & IP_VS_CONN_F_NFCT))
1314 ip_vs_notrack(skb);
1315 else
1316 ip_vs_update_conntrack(skb, cp, 0);
1317 ip_vs_conn_put(cp);
1318
1319 LeaveFunction(11);
1320 return NF_ACCEPT;
1321
1322drop:
1323 ip_vs_conn_put(cp);
1324 kfree_skb(skb);
1325 LeaveFunction(11);
1326 return NF_STOLEN;
1327}
1328
1329
1330
1331
1332static unsigned int
1333ip_vs_out_hook(void *priv, struct sk_buff *skb, const struct nf_hook_state *state)
1334{
1335 struct netns_ipvs *ipvs = net_ipvs(state->net);
1336 unsigned int hooknum = state->hook;
1337 struct ip_vs_iphdr iph;
1338 struct ip_vs_protocol *pp;
1339 struct ip_vs_proto_data *pd;
1340 struct ip_vs_conn *cp;
1341 int af = state->pf;
1342 struct sock *sk;
1343
1344 EnterFunction(11);
1345
1346
1347 if (skb->ipvs_property)
1348 return NF_ACCEPT;
1349
1350 sk = skb_to_full_sk(skb);
1351
1352 if (unlikely(sk && hooknum == NF_INET_LOCAL_OUT &&
1353 af == AF_INET)) {
1354
1355 if (sk->sk_family == PF_INET && inet_sk(sk)->nodefrag)
1356 return NF_ACCEPT;
1357 }
1358
1359 if (unlikely(!skb_dst(skb)))
1360 return NF_ACCEPT;
1361
1362 if (!ipvs->enable)
1363 return NF_ACCEPT;
1364
1365 ip_vs_fill_iph_skb(af, skb, false, &iph);
1366#ifdef CONFIG_IP_VS_IPV6
1367 if (af == AF_INET6) {
1368 if (unlikely(iph.protocol == IPPROTO_ICMPV6)) {
1369 int related;
1370 int verdict = ip_vs_out_icmp_v6(ipvs, skb, &related,
1371 hooknum, &iph);
1372
1373 if (related)
1374 return verdict;
1375 }
1376 } else
1377#endif
1378 if (unlikely(iph.protocol == IPPROTO_ICMP)) {
1379 int related;
1380 int verdict = ip_vs_out_icmp(ipvs, skb, &related, hooknum);
1381
1382 if (related)
1383 return verdict;
1384 }
1385
1386 pd = ip_vs_proto_data_get(ipvs, iph.protocol);
1387 if (unlikely(!pd))
1388 return NF_ACCEPT;
1389 pp = pd->pp;
1390
1391
1392#ifdef CONFIG_IP_VS_IPV6
1393 if (af == AF_INET)
1394#endif
1395 if (unlikely(ip_is_fragment(ip_hdr(skb)) && !pp->dont_defrag)) {
1396 if (ip_vs_gather_frags(ipvs, skb,
1397 ip_vs_defrag_user(hooknum)))
1398 return NF_STOLEN;
1399
1400 ip_vs_fill_iph_skb(AF_INET, skb, false, &iph);
1401 }
1402
1403
1404
1405
1406 cp = INDIRECT_CALL_1(pp->conn_out_get, ip_vs_conn_out_get_proto,
1407 ipvs, af, skb, &iph);
1408
1409 if (likely(cp))
1410 return handle_response(af, skb, pd, cp, &iph, hooknum);
1411
1412
1413 if (atomic_read(&ipvs->conn_out_counter)) {
1414
1415
1416
1417
1418
1419 if (pp->protocol == IPPROTO_UDP) {
1420 cp = __ip_vs_rs_conn_out(hooknum, ipvs, af, skb, &iph);
1421 if (likely(cp))
1422 return handle_response(af, skb, pd, cp, &iph,
1423 hooknum);
1424 }
1425 }
1426
1427 if (sysctl_nat_icmp_send(ipvs) &&
1428 (pp->protocol == IPPROTO_TCP ||
1429 pp->protocol == IPPROTO_UDP ||
1430 pp->protocol == IPPROTO_SCTP)) {
1431 __be16 _ports[2], *pptr;
1432
1433 pptr = frag_safe_skb_hp(skb, iph.len,
1434 sizeof(_ports), _ports);
1435 if (pptr == NULL)
1436 return NF_ACCEPT;
1437 if (ip_vs_has_real_service(ipvs, af, iph.protocol, &iph.saddr,
1438 pptr[0])) {
1439
1440
1441
1442
1443
1444 if ((iph.protocol != IPPROTO_TCP &&
1445 iph.protocol != IPPROTO_SCTP)
1446 || ((iph.protocol == IPPROTO_TCP
1447 && !is_tcp_reset(skb, iph.len))
1448 || (iph.protocol == IPPROTO_SCTP
1449 && !is_sctp_abort(skb,
1450 iph.len)))) {
1451#ifdef CONFIG_IP_VS_IPV6
1452 if (af == AF_INET6) {
1453 if (!skb->dev)
1454 skb->dev = ipvs->net->loopback_dev;
1455 icmpv6_send(skb,
1456 ICMPV6_DEST_UNREACH,
1457 ICMPV6_PORT_UNREACH,
1458 0);
1459 } else
1460#endif
1461 icmp_send(skb,
1462 ICMP_DEST_UNREACH,
1463 ICMP_PORT_UNREACH, 0);
1464 return NF_DROP;
1465 }
1466 }
1467 }
1468
1469 IP_VS_DBG_PKT(12, af, pp, skb, iph.off,
1470 "ip_vs_out: packet continues traversal as normal");
1471 return NF_ACCEPT;
1472}
1473
1474static unsigned int
1475ip_vs_try_to_schedule(struct netns_ipvs *ipvs, int af, struct sk_buff *skb,
1476 struct ip_vs_proto_data *pd,
1477 int *verdict, struct ip_vs_conn **cpp,
1478 struct ip_vs_iphdr *iph)
1479{
1480 struct ip_vs_protocol *pp = pd->pp;
1481
1482 if (!iph->fragoffs) {
1483
1484
1485
1486
1487
1488 if (!pp->conn_schedule(ipvs, af, skb, pd, verdict, cpp, iph))
1489 return 0;
1490 }
1491
1492 if (unlikely(!*cpp)) {
1493
1494 IP_VS_DBG_PKT(12, af, pp, skb, iph->off,
1495 "ip_vs_in: packet continues traversal as normal");
1496
1497
1498 if (iph->fragoffs)
1499 IP_VS_DBG_PKT(7, af, pp, skb, iph->off,
1500 "unhandled fragment");
1501
1502 *verdict = NF_ACCEPT;
1503 return 0;
1504 }
1505
1506 return 1;
1507}
1508
1509
1510static int ipvs_udp_decap(struct netns_ipvs *ipvs, struct sk_buff *skb,
1511 unsigned int offset, __u16 af,
1512 const union nf_inet_addr *daddr, __u8 *proto)
1513{
1514 struct udphdr _udph, *udph;
1515 struct ip_vs_dest *dest;
1516
1517 udph = skb_header_pointer(skb, offset, sizeof(_udph), &_udph);
1518 if (!udph)
1519 goto unk;
1520 offset += sizeof(struct udphdr);
1521 dest = ip_vs_find_tunnel(ipvs, af, daddr, udph->dest);
1522 if (!dest)
1523 goto unk;
1524 if (dest->tun_type == IP_VS_CONN_F_TUNNEL_TYPE_GUE) {
1525 struct guehdr _gueh, *gueh;
1526
1527 gueh = skb_header_pointer(skb, offset, sizeof(_gueh), &_gueh);
1528 if (!gueh)
1529 goto unk;
1530 if (gueh->control != 0 || gueh->version != 0)
1531 goto unk;
1532
1533 if (gueh->proto_ctype != IPPROTO_IPIP)
1534 goto unk;
1535 *proto = gueh->proto_ctype;
1536 return sizeof(struct udphdr) + sizeof(struct guehdr) +
1537 (gueh->hlen << 2);
1538 }
1539
1540unk:
1541 return 0;
1542}
1543
1544
1545static int ipvs_gre_decap(struct netns_ipvs *ipvs, struct sk_buff *skb,
1546 unsigned int offset, __u16 af,
1547 const union nf_inet_addr *daddr, __u8 *proto)
1548{
1549 struct gre_base_hdr _greh, *greh;
1550 struct ip_vs_dest *dest;
1551
1552 greh = skb_header_pointer(skb, offset, sizeof(_greh), &_greh);
1553 if (!greh)
1554 goto unk;
1555 dest = ip_vs_find_tunnel(ipvs, af, daddr, 0);
1556 if (!dest)
1557 goto unk;
1558 if (dest->tun_type == IP_VS_CONN_F_TUNNEL_TYPE_GRE) {
1559 __be16 type;
1560
1561
1562 if ((greh->flags & ~GRE_CSUM) != 0)
1563 goto unk;
1564 type = greh->protocol;
1565
1566 if (type != htons(ETH_P_IP))
1567 goto unk;
1568 *proto = IPPROTO_IPIP;
1569 return gre_calc_hlen(gre_flags_to_tnl_flags(greh->flags));
1570 }
1571
1572unk:
1573 return 0;
1574}
1575
1576
1577
1578
1579
1580
1581
1582static int
1583ip_vs_in_icmp(struct netns_ipvs *ipvs, struct sk_buff *skb, int *related,
1584 unsigned int hooknum)
1585{
1586 struct iphdr *iph;
1587 struct icmphdr _icmph, *ic;
1588 struct iphdr _ciph, *cih;
1589 struct ip_vs_iphdr ciph;
1590 struct ip_vs_conn *cp;
1591 struct ip_vs_protocol *pp;
1592 struct ip_vs_proto_data *pd;
1593 unsigned int offset, offset2, ihl, verdict;
1594 bool tunnel, new_cp = false;
1595 union nf_inet_addr *raddr;
1596 char *outer_proto = "IPIP";
1597
1598 *related = 1;
1599
1600
1601 if (ip_is_fragment(ip_hdr(skb))) {
1602 if (ip_vs_gather_frags(ipvs, skb, ip_vs_defrag_user(hooknum)))
1603 return NF_STOLEN;
1604 }
1605
1606 iph = ip_hdr(skb);
1607 offset = ihl = iph->ihl * 4;
1608 ic = skb_header_pointer(skb, offset, sizeof(_icmph), &_icmph);
1609 if (ic == NULL)
1610 return NF_DROP;
1611
1612 IP_VS_DBG(12, "Incoming ICMP (%d,%d) %pI4->%pI4\n",
1613 ic->type, ntohs(icmp_id(ic)),
1614 &iph->saddr, &iph->daddr);
1615
1616
1617
1618
1619
1620
1621
1622
1623 if ((ic->type != ICMP_DEST_UNREACH) &&
1624 (ic->type != ICMP_SOURCE_QUENCH) &&
1625 (ic->type != ICMP_TIME_EXCEEDED)) {
1626 *related = 0;
1627 return NF_ACCEPT;
1628 }
1629
1630
1631 offset += sizeof(_icmph);
1632 cih = skb_header_pointer(skb, offset, sizeof(_ciph), &_ciph);
1633 if (cih == NULL)
1634 return NF_ACCEPT;
1635 raddr = (union nf_inet_addr *)&cih->daddr;
1636
1637
1638 tunnel = false;
1639 if (cih->protocol == IPPROTO_IPIP) {
1640 struct ip_vs_dest *dest;
1641
1642 if (unlikely(cih->frag_off & htons(IP_OFFSET)))
1643 return NF_ACCEPT;
1644
1645 if (!(skb_rtable(skb)->rt_flags & RTCF_LOCAL))
1646 return NF_ACCEPT;
1647 dest = ip_vs_find_tunnel(ipvs, AF_INET, raddr, 0);
1648
1649 if (!dest || dest->tun_type != IP_VS_CONN_F_TUNNEL_TYPE_IPIP)
1650 return NF_ACCEPT;
1651 offset += cih->ihl * 4;
1652 cih = skb_header_pointer(skb, offset, sizeof(_ciph), &_ciph);
1653 if (cih == NULL)
1654 return NF_ACCEPT;
1655 tunnel = true;
1656 } else if ((cih->protocol == IPPROTO_UDP ||
1657 cih->protocol == IPPROTO_GRE) &&
1658
1659 (skb_rtable(skb)->rt_flags & RTCF_LOCAL)) {
1660 __u8 iproto;
1661 int ulen;
1662
1663
1664 if (unlikely(cih->frag_off & htons(IP_OFFSET)))
1665 return NF_ACCEPT;
1666 offset2 = offset + cih->ihl * 4;
1667 if (cih->protocol == IPPROTO_UDP) {
1668 ulen = ipvs_udp_decap(ipvs, skb, offset2, AF_INET,
1669 raddr, &iproto);
1670 outer_proto = "UDP";
1671 } else {
1672 ulen = ipvs_gre_decap(ipvs, skb, offset2, AF_INET,
1673 raddr, &iproto);
1674 outer_proto = "GRE";
1675 }
1676 if (ulen > 0) {
1677
1678 offset = offset2 + ulen;
1679
1680 cih = skb_header_pointer(skb, offset, sizeof(_ciph),
1681 &_ciph);
1682 if (cih && cih->version == 4 && cih->ihl >= 5 &&
1683 iproto == IPPROTO_IPIP)
1684 tunnel = true;
1685 else
1686 return NF_ACCEPT;
1687 }
1688 }
1689
1690 pd = ip_vs_proto_data_get(ipvs, cih->protocol);
1691 if (!pd)
1692 return NF_ACCEPT;
1693 pp = pd->pp;
1694
1695
1696 if (unlikely(cih->frag_off & htons(IP_OFFSET) &&
1697 pp->dont_defrag))
1698 return NF_ACCEPT;
1699
1700 IP_VS_DBG_PKT(11, AF_INET, pp, skb, offset,
1701 "Checking incoming ICMP for");
1702
1703 offset2 = offset;
1704 ip_vs_fill_iph_skb_icmp(AF_INET, skb, offset, !tunnel, &ciph);
1705 offset = ciph.len;
1706
1707
1708
1709
1710 cp = INDIRECT_CALL_1(pp->conn_in_get, ip_vs_conn_in_get_proto,
1711 ipvs, AF_INET, skb, &ciph);
1712
1713 if (!cp) {
1714 int v;
1715
1716 if (tunnel || !sysctl_schedule_icmp(ipvs))
1717 return NF_ACCEPT;
1718
1719 if (!ip_vs_try_to_schedule(ipvs, AF_INET, skb, pd, &v, &cp, &ciph))
1720 return v;
1721 new_cp = true;
1722 }
1723
1724 verdict = NF_DROP;
1725
1726
1727 if (!skb_csum_unnecessary(skb) && ip_vs_checksum_complete(skb, ihl)) {
1728
1729 IP_VS_DBG(1, "Incoming ICMP: failed checksum from %pI4!\n",
1730 &iph->saddr);
1731 goto out;
1732 }
1733
1734 if (tunnel) {
1735 __be32 info = ic->un.gateway;
1736 __u8 type = ic->type;
1737 __u8 code = ic->code;
1738
1739
1740 if (ic->type == ICMP_DEST_UNREACH &&
1741 ic->code == ICMP_FRAG_NEEDED) {
1742 struct ip_vs_dest *dest = cp->dest;
1743 u32 mtu = ntohs(ic->un.frag.mtu);
1744 __be16 frag_off = cih->frag_off;
1745
1746
1747 if (pskb_pull(skb, ihl + sizeof(_icmph)) == NULL)
1748 goto ignore_tunnel;
1749 offset2 -= ihl + sizeof(_icmph);
1750 skb_reset_network_header(skb);
1751 IP_VS_DBG(12, "ICMP for %s %pI4->%pI4: mtu=%u\n",
1752 outer_proto, &ip_hdr(skb)->saddr,
1753 &ip_hdr(skb)->daddr, mtu);
1754 ipv4_update_pmtu(skb, ipvs->net, mtu, 0, 0);
1755
1756 if (!(frag_off & htons(IP_DF)))
1757 goto ignore_tunnel;
1758
1759 if (dest) {
1760 struct ip_vs_dest_dst *dest_dst;
1761
1762 dest_dst = rcu_dereference(dest->dest_dst);
1763 if (dest_dst)
1764 mtu = dst_mtu(dest_dst->dst_cache);
1765 }
1766 if (mtu > 68 + sizeof(struct iphdr))
1767 mtu -= sizeof(struct iphdr);
1768 info = htonl(mtu);
1769 }
1770
1771
1772
1773 if (pskb_pull(skb, offset2) == NULL)
1774 goto ignore_tunnel;
1775 skb_reset_network_header(skb);
1776 IP_VS_DBG(12, "Sending ICMP for %pI4->%pI4: t=%u, c=%u, i=%u\n",
1777 &ip_hdr(skb)->saddr, &ip_hdr(skb)->daddr,
1778 type, code, ntohl(info));
1779 icmp_send(skb, type, code, info);
1780
1781 ip_vs_out_stats(cp, skb);
1782
1783ignore_tunnel:
1784 consume_skb(skb);
1785 verdict = NF_STOLEN;
1786 goto out;
1787 }
1788
1789
1790 ip_vs_in_stats(cp, skb);
1791 if (IPPROTO_TCP == cih->protocol || IPPROTO_UDP == cih->protocol ||
1792 IPPROTO_SCTP == cih->protocol)
1793 offset += 2 * sizeof(__u16);
1794 verdict = ip_vs_icmp_xmit(skb, cp, pp, offset, hooknum, &ciph);
1795
1796out:
1797 if (likely(!new_cp))
1798 __ip_vs_conn_put(cp);
1799 else
1800 ip_vs_conn_put(cp);
1801
1802 return verdict;
1803}
1804
1805#ifdef CONFIG_IP_VS_IPV6
1806static int ip_vs_in_icmp_v6(struct netns_ipvs *ipvs, struct sk_buff *skb,
1807 int *related, unsigned int hooknum,
1808 struct ip_vs_iphdr *iph)
1809{
1810 struct icmp6hdr _icmph, *ic;
1811 struct ip_vs_iphdr ciph = {.flags = 0, .fragoffs = 0};
1812 struct ip_vs_conn *cp;
1813 struct ip_vs_protocol *pp;
1814 struct ip_vs_proto_data *pd;
1815 unsigned int offset, verdict;
1816 bool new_cp = false;
1817
1818 *related = 1;
1819
1820 ic = frag_safe_skb_hp(skb, iph->len, sizeof(_icmph), &_icmph);
1821 if (ic == NULL)
1822 return NF_DROP;
1823
1824
1825
1826
1827
1828
1829
1830
1831 if (ic->icmp6_type & ICMPV6_INFOMSG_MASK) {
1832 *related = 0;
1833 return NF_ACCEPT;
1834 }
1835
1836
1837
1838 if (iph->flags & IP6_FH_F_FRAG)
1839 return NF_DROP;
1840
1841 IP_VS_DBG(8, "Incoming ICMPv6 (%d,%d) %pI6c->%pI6c\n",
1842 ic->icmp6_type, ntohs(icmpv6_id(ic)),
1843 &iph->saddr, &iph->daddr);
1844
1845 offset = iph->len + sizeof(_icmph);
1846 if (!ip_vs_fill_iph_skb_icmp(AF_INET6, skb, offset, true, &ciph))
1847 return NF_ACCEPT;
1848
1849 pd = ip_vs_proto_data_get(ipvs, ciph.protocol);
1850 if (!pd)
1851 return NF_ACCEPT;
1852 pp = pd->pp;
1853
1854
1855 if (ciph.fragoffs)
1856 return NF_ACCEPT;
1857
1858 IP_VS_DBG_PKT(11, AF_INET6, pp, skb, offset,
1859 "Checking incoming ICMPv6 for");
1860
1861
1862
1863
1864 cp = INDIRECT_CALL_1(pp->conn_in_get, ip_vs_conn_in_get_proto,
1865 ipvs, AF_INET6, skb, &ciph);
1866
1867 if (!cp) {
1868 int v;
1869
1870 if (!sysctl_schedule_icmp(ipvs))
1871 return NF_ACCEPT;
1872
1873 if (!ip_vs_try_to_schedule(ipvs, AF_INET6, skb, pd, &v, &cp, &ciph))
1874 return v;
1875
1876 new_cp = true;
1877 }
1878
1879
1880 if ((hooknum == NF_INET_LOCAL_OUT) &&
1881 (IP_VS_FWD_METHOD(cp) != IP_VS_CONN_F_MASQ)) {
1882 verdict = NF_ACCEPT;
1883 goto out;
1884 }
1885
1886
1887 ip_vs_in_stats(cp, skb);
1888
1889
1890 offset = ciph.len;
1891 if (IPPROTO_TCP == ciph.protocol || IPPROTO_UDP == ciph.protocol ||
1892 IPPROTO_SCTP == ciph.protocol)
1893 offset += 2 * sizeof(__u16);
1894
1895 verdict = ip_vs_icmp_xmit_v6(skb, cp, pp, offset, hooknum, &ciph);
1896
1897out:
1898 if (likely(!new_cp))
1899 __ip_vs_conn_put(cp);
1900 else
1901 ip_vs_conn_put(cp);
1902
1903 return verdict;
1904}
1905#endif
1906
1907
1908
1909
1910
1911
1912static unsigned int
1913ip_vs_in_hook(void *priv, struct sk_buff *skb, const struct nf_hook_state *state)
1914{
1915 struct netns_ipvs *ipvs = net_ipvs(state->net);
1916 unsigned int hooknum = state->hook;
1917 struct ip_vs_iphdr iph;
1918 struct ip_vs_protocol *pp;
1919 struct ip_vs_proto_data *pd;
1920 struct ip_vs_conn *cp;
1921 int ret, pkts;
1922 struct sock *sk;
1923 int af = state->pf;
1924
1925
1926 if (skb->ipvs_property)
1927 return NF_ACCEPT;
1928
1929
1930
1931
1932
1933
1934 if (unlikely((skb->pkt_type != PACKET_HOST &&
1935 hooknum != NF_INET_LOCAL_OUT) ||
1936 !skb_dst(skb))) {
1937 ip_vs_fill_iph_skb(af, skb, false, &iph);
1938 IP_VS_DBG_BUF(12, "packet type=%d proto=%d daddr=%s"
1939 " ignored in hook %u\n",
1940 skb->pkt_type, iph.protocol,
1941 IP_VS_DBG_ADDR(af, &iph.daddr), hooknum);
1942 return NF_ACCEPT;
1943 }
1944
1945 if (unlikely(sysctl_backup_only(ipvs) || !ipvs->enable))
1946 return NF_ACCEPT;
1947
1948 ip_vs_fill_iph_skb(af, skb, false, &iph);
1949
1950
1951 sk = skb_to_full_sk(skb);
1952 if (unlikely(sk && hooknum == NF_INET_LOCAL_OUT &&
1953 af == AF_INET)) {
1954
1955 if (sk->sk_family == PF_INET && inet_sk(sk)->nodefrag)
1956 return NF_ACCEPT;
1957 }
1958
1959#ifdef CONFIG_IP_VS_IPV6
1960 if (af == AF_INET6) {
1961 if (unlikely(iph.protocol == IPPROTO_ICMPV6)) {
1962 int related;
1963 int verdict = ip_vs_in_icmp_v6(ipvs, skb, &related,
1964 hooknum, &iph);
1965
1966 if (related)
1967 return verdict;
1968 }
1969 } else
1970#endif
1971 if (unlikely(iph.protocol == IPPROTO_ICMP)) {
1972 int related;
1973 int verdict = ip_vs_in_icmp(ipvs, skb, &related,
1974 hooknum);
1975
1976 if (related)
1977 return verdict;
1978 }
1979
1980
1981 pd = ip_vs_proto_data_get(ipvs, iph.protocol);
1982 if (unlikely(!pd)) {
1983
1984
1985
1986
1987 if (sysctl_ignore_tunneled(ipvs))
1988 skb->ipvs_property = 1;
1989
1990 return NF_ACCEPT;
1991 }
1992 pp = pd->pp;
1993
1994
1995
1996 cp = INDIRECT_CALL_1(pp->conn_in_get, ip_vs_conn_in_get_proto,
1997 ipvs, af, skb, &iph);
1998
1999 if (!iph.fragoffs && is_new_conn(skb, &iph) && cp) {
2000 int conn_reuse_mode = sysctl_conn_reuse_mode(ipvs);
2001 bool old_ct = false, resched = false;
2002
2003 if (unlikely(sysctl_expire_nodest_conn(ipvs)) && cp->dest &&
2004 unlikely(!atomic_read(&cp->dest->weight))) {
2005 resched = true;
2006 old_ct = ip_vs_conn_uses_old_conntrack(cp, skb);
2007 } else if (conn_reuse_mode &&
2008 is_new_conn_expected(cp, conn_reuse_mode)) {
2009 old_ct = ip_vs_conn_uses_old_conntrack(cp, skb);
2010 if (!atomic_read(&cp->n_control)) {
2011 resched = true;
2012 } else {
2013
2014
2015
2016
2017 resched = !old_ct;
2018 }
2019 }
2020
2021 if (resched) {
2022 if (!old_ct)
2023 cp->flags &= ~IP_VS_CONN_F_NFCT;
2024 if (!atomic_read(&cp->n_control))
2025 ip_vs_conn_expire_now(cp);
2026 __ip_vs_conn_put(cp);
2027 if (old_ct)
2028 return NF_DROP;
2029 cp = NULL;
2030 }
2031 }
2032
2033
2034 if (cp && cp->dest && !(cp->dest->flags & IP_VS_DEST_F_AVAILABLE)) {
2035
2036 if (sysctl_expire_nodest_conn(ipvs)) {
2037 bool old_ct = ip_vs_conn_uses_old_conntrack(cp, skb);
2038
2039 if (!old_ct)
2040 cp->flags &= ~IP_VS_CONN_F_NFCT;
2041
2042 ip_vs_conn_expire_now(cp);
2043 __ip_vs_conn_put(cp);
2044 if (old_ct)
2045 return NF_DROP;
2046 cp = NULL;
2047 } else {
2048 __ip_vs_conn_put(cp);
2049 return NF_DROP;
2050 }
2051 }
2052
2053 if (unlikely(!cp)) {
2054 int v;
2055
2056 if (!ip_vs_try_to_schedule(ipvs, af, skb, pd, &v, &cp, &iph))
2057 return v;
2058 }
2059
2060 IP_VS_DBG_PKT(11, af, pp, skb, iph.off, "Incoming packet");
2061
2062 ip_vs_in_stats(cp, skb);
2063 ip_vs_set_state(cp, IP_VS_DIR_INPUT, skb, pd);
2064 if (cp->packet_xmit)
2065 ret = cp->packet_xmit(skb, cp, pp, &iph);
2066
2067 else {
2068 IP_VS_DBG_RL("warning: packet_xmit is null");
2069 ret = NF_ACCEPT;
2070 }
2071
2072
2073
2074
2075
2076
2077
2078
2079
2080
2081 if (cp->flags & IP_VS_CONN_F_ONE_PACKET)
2082 pkts = sysctl_sync_threshold(ipvs);
2083 else
2084 pkts = atomic_inc_return(&cp->in_pkts);
2085
2086 if (ipvs->sync_state & IP_VS_STATE_MASTER)
2087 ip_vs_sync_conn(ipvs, cp, pkts);
2088 else if ((cp->flags & IP_VS_CONN_F_ONE_PACKET) && cp->control)
2089
2090 atomic_inc(&cp->control->in_pkts);
2091
2092 ip_vs_conn_put(cp);
2093 return ret;
2094}
2095
2096
2097
2098
2099
2100
2101
2102
2103
2104
2105static unsigned int
2106ip_vs_forward_icmp(void *priv, struct sk_buff *skb,
2107 const struct nf_hook_state *state)
2108{
2109 struct netns_ipvs *ipvs = net_ipvs(state->net);
2110 int r;
2111
2112
2113 if (unlikely(sysctl_backup_only(ipvs) || !ipvs->enable))
2114 return NF_ACCEPT;
2115
2116 if (state->pf == NFPROTO_IPV4) {
2117 if (ip_hdr(skb)->protocol != IPPROTO_ICMP)
2118 return NF_ACCEPT;
2119#ifdef CONFIG_IP_VS_IPV6
2120 } else {
2121 struct ip_vs_iphdr iphdr;
2122
2123 ip_vs_fill_iph_skb(AF_INET6, skb, false, &iphdr);
2124
2125 if (iphdr.protocol != IPPROTO_ICMPV6)
2126 return NF_ACCEPT;
2127
2128 return ip_vs_in_icmp_v6(ipvs, skb, &r, state->hook, &iphdr);
2129#endif
2130 }
2131
2132 return ip_vs_in_icmp(ipvs, skb, &r, state->hook);
2133}
2134
2135static const struct nf_hook_ops ip_vs_ops4[] = {
2136
2137 {
2138 .hook = ip_vs_out_hook,
2139 .pf = NFPROTO_IPV4,
2140 .hooknum = NF_INET_LOCAL_IN,
2141 .priority = NF_IP_PRI_NAT_SRC - 2,
2142 },
2143
2144
2145
2146 {
2147 .hook = ip_vs_in_hook,
2148 .pf = NFPROTO_IPV4,
2149 .hooknum = NF_INET_LOCAL_IN,
2150 .priority = NF_IP_PRI_NAT_SRC - 1,
2151 },
2152
2153 {
2154 .hook = ip_vs_out_hook,
2155 .pf = NFPROTO_IPV4,
2156 .hooknum = NF_INET_LOCAL_OUT,
2157 .priority = NF_IP_PRI_NAT_DST + 1,
2158 },
2159
2160 {
2161 .hook = ip_vs_in_hook,
2162 .pf = NFPROTO_IPV4,
2163 .hooknum = NF_INET_LOCAL_OUT,
2164 .priority = NF_IP_PRI_NAT_DST + 2,
2165 },
2166
2167
2168 {
2169 .hook = ip_vs_forward_icmp,
2170 .pf = NFPROTO_IPV4,
2171 .hooknum = NF_INET_FORWARD,
2172 .priority = 99,
2173 },
2174
2175 {
2176 .hook = ip_vs_out_hook,
2177 .pf = NFPROTO_IPV4,
2178 .hooknum = NF_INET_FORWARD,
2179 .priority = 100,
2180 },
2181};
2182
2183#ifdef CONFIG_IP_VS_IPV6
2184static const struct nf_hook_ops ip_vs_ops6[] = {
2185
2186 {
2187 .hook = ip_vs_out_hook,
2188 .pf = NFPROTO_IPV6,
2189 .hooknum = NF_INET_LOCAL_IN,
2190 .priority = NF_IP6_PRI_NAT_SRC - 2,
2191 },
2192
2193
2194
2195 {
2196 .hook = ip_vs_in_hook,
2197 .pf = NFPROTO_IPV6,
2198 .hooknum = NF_INET_LOCAL_IN,
2199 .priority = NF_IP6_PRI_NAT_SRC - 1,
2200 },
2201
2202 {
2203 .hook = ip_vs_out_hook,
2204 .pf = NFPROTO_IPV6,
2205 .hooknum = NF_INET_LOCAL_OUT,
2206 .priority = NF_IP6_PRI_NAT_DST + 1,
2207 },
2208
2209 {
2210 .hook = ip_vs_in_hook,
2211 .pf = NFPROTO_IPV6,
2212 .hooknum = NF_INET_LOCAL_OUT,
2213 .priority = NF_IP6_PRI_NAT_DST + 2,
2214 },
2215
2216
2217 {
2218 .hook = ip_vs_forward_icmp,
2219 .pf = NFPROTO_IPV6,
2220 .hooknum = NF_INET_FORWARD,
2221 .priority = 99,
2222 },
2223
2224 {
2225 .hook = ip_vs_out_hook,
2226 .pf = NFPROTO_IPV6,
2227 .hooknum = NF_INET_FORWARD,
2228 .priority = 100,
2229 },
2230};
2231#endif
2232
2233int ip_vs_register_hooks(struct netns_ipvs *ipvs, unsigned int af)
2234{
2235 const struct nf_hook_ops *ops;
2236 unsigned int count;
2237 unsigned int afmask;
2238 int ret = 0;
2239
2240 if (af == AF_INET6) {
2241#ifdef CONFIG_IP_VS_IPV6
2242 ops = ip_vs_ops6;
2243 count = ARRAY_SIZE(ip_vs_ops6);
2244 afmask = 2;
2245#else
2246 return -EINVAL;
2247#endif
2248 } else {
2249 ops = ip_vs_ops4;
2250 count = ARRAY_SIZE(ip_vs_ops4);
2251 afmask = 1;
2252 }
2253
2254 if (!(ipvs->hooks_afmask & afmask)) {
2255 ret = nf_register_net_hooks(ipvs->net, ops, count);
2256 if (ret >= 0)
2257 ipvs->hooks_afmask |= afmask;
2258 }
2259 return ret;
2260}
2261
2262void ip_vs_unregister_hooks(struct netns_ipvs *ipvs, unsigned int af)
2263{
2264 const struct nf_hook_ops *ops;
2265 unsigned int count;
2266 unsigned int afmask;
2267
2268 if (af == AF_INET6) {
2269#ifdef CONFIG_IP_VS_IPV6
2270 ops = ip_vs_ops6;
2271 count = ARRAY_SIZE(ip_vs_ops6);
2272 afmask = 2;
2273#else
2274 return;
2275#endif
2276 } else {
2277 ops = ip_vs_ops4;
2278 count = ARRAY_SIZE(ip_vs_ops4);
2279 afmask = 1;
2280 }
2281
2282 if (ipvs->hooks_afmask & afmask) {
2283 nf_unregister_net_hooks(ipvs->net, ops, count);
2284 ipvs->hooks_afmask &= ~afmask;
2285 }
2286}
2287
2288
2289
2290
2291static int __net_init __ip_vs_init(struct net *net)
2292{
2293 struct netns_ipvs *ipvs;
2294
2295 ipvs = net_generic(net, ip_vs_net_id);
2296 if (ipvs == NULL)
2297 return -ENOMEM;
2298
2299
2300 ipvs->enable = 0;
2301 ipvs->net = net;
2302
2303 ipvs->gen = atomic_read(&ipvs_netns_cnt);
2304 atomic_inc(&ipvs_netns_cnt);
2305 net->ipvs = ipvs;
2306
2307 if (ip_vs_estimator_net_init(ipvs) < 0)
2308 goto estimator_fail;
2309
2310 if (ip_vs_control_net_init(ipvs) < 0)
2311 goto control_fail;
2312
2313 if (ip_vs_protocol_net_init(ipvs) < 0)
2314 goto protocol_fail;
2315
2316 if (ip_vs_app_net_init(ipvs) < 0)
2317 goto app_fail;
2318
2319 if (ip_vs_conn_net_init(ipvs) < 0)
2320 goto conn_fail;
2321
2322 if (ip_vs_sync_net_init(ipvs) < 0)
2323 goto sync_fail;
2324
2325 return 0;
2326
2327
2328
2329
2330sync_fail:
2331 ip_vs_conn_net_cleanup(ipvs);
2332conn_fail:
2333 ip_vs_app_net_cleanup(ipvs);
2334app_fail:
2335 ip_vs_protocol_net_cleanup(ipvs);
2336protocol_fail:
2337 ip_vs_control_net_cleanup(ipvs);
2338control_fail:
2339 ip_vs_estimator_net_cleanup(ipvs);
2340estimator_fail:
2341 net->ipvs = NULL;
2342 return -ENOMEM;
2343}
2344
2345static void __net_exit __ip_vs_cleanup_batch(struct list_head *net_list)
2346{
2347 struct netns_ipvs *ipvs;
2348 struct net *net;
2349
2350 ip_vs_service_nets_cleanup(net_list);
2351 list_for_each_entry(net, net_list, exit_list) {
2352 ipvs = net_ipvs(net);
2353 ip_vs_conn_net_cleanup(ipvs);
2354 ip_vs_app_net_cleanup(ipvs);
2355 ip_vs_protocol_net_cleanup(ipvs);
2356 ip_vs_control_net_cleanup(ipvs);
2357 ip_vs_estimator_net_cleanup(ipvs);
2358 IP_VS_DBG(2, "ipvs netns %d released\n", ipvs->gen);
2359 net->ipvs = NULL;
2360 }
2361}
2362
2363static void __net_exit __ip_vs_dev_cleanup_batch(struct list_head *net_list)
2364{
2365 struct netns_ipvs *ipvs;
2366 struct net *net;
2367
2368 EnterFunction(2);
2369 list_for_each_entry(net, net_list, exit_list) {
2370 ipvs = net_ipvs(net);
2371 ip_vs_unregister_hooks(ipvs, AF_INET);
2372 ip_vs_unregister_hooks(ipvs, AF_INET6);
2373 ipvs->enable = 0;
2374 smp_wmb();
2375 ip_vs_sync_net_cleanup(ipvs);
2376 }
2377 LeaveFunction(2);
2378}
2379
2380static struct pernet_operations ipvs_core_ops = {
2381 .init = __ip_vs_init,
2382 .exit_batch = __ip_vs_cleanup_batch,
2383 .id = &ip_vs_net_id,
2384 .size = sizeof(struct netns_ipvs),
2385};
2386
2387static struct pernet_operations ipvs_core_dev_ops = {
2388 .exit_batch = __ip_vs_dev_cleanup_batch,
2389};
2390
2391
2392
2393
2394static int __init ip_vs_init(void)
2395{
2396 int ret;
2397
2398 ret = ip_vs_control_init();
2399 if (ret < 0) {
2400 pr_err("can't setup control.\n");
2401 goto exit;
2402 }
2403
2404 ip_vs_protocol_init();
2405
2406 ret = ip_vs_conn_init();
2407 if (ret < 0) {
2408 pr_err("can't setup connection table.\n");
2409 goto cleanup_protocol;
2410 }
2411
2412 ret = register_pernet_subsys(&ipvs_core_ops);
2413 if (ret < 0)
2414 goto cleanup_conn;
2415
2416 ret = register_pernet_device(&ipvs_core_dev_ops);
2417 if (ret < 0)
2418 goto cleanup_sub;
2419
2420 ret = ip_vs_register_nl_ioctl();
2421 if (ret < 0) {
2422 pr_err("can't register netlink/ioctl.\n");
2423 goto cleanup_dev;
2424 }
2425
2426 pr_info("ipvs loaded.\n");
2427
2428 return ret;
2429
2430cleanup_dev:
2431 unregister_pernet_device(&ipvs_core_dev_ops);
2432cleanup_sub:
2433 unregister_pernet_subsys(&ipvs_core_ops);
2434cleanup_conn:
2435 ip_vs_conn_cleanup();
2436cleanup_protocol:
2437 ip_vs_protocol_cleanup();
2438 ip_vs_control_cleanup();
2439exit:
2440 return ret;
2441}
2442
2443static void __exit ip_vs_cleanup(void)
2444{
2445 ip_vs_unregister_nl_ioctl();
2446 unregister_pernet_device(&ipvs_core_dev_ops);
2447 unregister_pernet_subsys(&ipvs_core_ops);
2448 ip_vs_conn_cleanup();
2449 ip_vs_protocol_cleanup();
2450 ip_vs_control_cleanup();
2451 pr_info("ipvs unloaded.\n");
2452}
2453
2454module_init(ip_vs_init);
2455module_exit(ip_vs_cleanup);
2456MODULE_LICENSE("GPL");
2457