2 * ip_vs_proto_udp.c: UDP load balancing support for IPVS
4 * Authors: Wensong Zhang <wensong@linuxvirtualserver.org>
5 * Julian Anastasov <ja@ssi.bg>
7 * This program is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version
10 * 2 of the License, or (at your option) any later version.
12 * Changes: Hans Schillstrom <hans.schillstrom@ericsson.com>
13 * Network name space (netns) aware.
17 #define KMSG_COMPONENT "IPVS"
18 #define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
22 #include <linux/kernel.h>
23 #include <linux/netfilter.h>
24 #include <linux/netfilter_ipv4.h>
25 #include <linux/udp.h>
27 #include <net/ip_vs.h>
29 #include <net/ip6_checksum.h>
32 udp_conn_schedule(int af
, struct sk_buff
*skb
, struct ip_vs_proto_data
*pd
,
33 int *verdict
, struct ip_vs_conn
**cpp
,
34 struct ip_vs_iphdr
*iph
)
37 struct ip_vs_service
*svc
;
38 struct udphdr _udph
, *uh
;
39 __be16 _ports
[2], *ports
= NULL
;
41 if (likely(!ip_vs_iph_icmp(iph
))) {
42 /* IPv6 fragments, only first fragment will hit this */
43 uh
= skb_header_pointer(skb
, iph
->len
, sizeof(_udph
), &_udph
);
47 ports
= skb_header_pointer(
48 skb
, iph
->len
, sizeof(_ports
), &_ports
);
58 if (likely(!ip_vs_iph_inverse(iph
)))
59 svc
= ip_vs_service_find(net
, af
, skb
->mark
, iph
->protocol
,
60 &iph
->daddr
, ports
[1]);
62 svc
= ip_vs_service_find(net
, af
, skb
->mark
, iph
->protocol
,
63 &iph
->saddr
, ports
[0]);
68 if (ip_vs_todrop(net_ipvs(net
))) {
70 * It seems that we are very loaded.
71 * We have to drop this packet :(
79 * Let the virtual server select a real server for the
80 * incoming connection, and create a connection entry.
82 *cpp
= ip_vs_schedule(svc
, skb
, pd
, &ignored
, iph
);
83 if (!*cpp
&& ignored
<= 0) {
85 *verdict
= ip_vs_leave(svc
, skb
, pd
, iph
);
99 udp_fast_csum_update(int af
, struct udphdr
*uhdr
,
100 const union nf_inet_addr
*oldip
,
101 const union nf_inet_addr
*newip
,
102 __be16 oldport
, __be16 newport
)
104 #ifdef CONFIG_IP_VS_IPV6
107 csum_fold(ip_vs_check_diff16(oldip
->ip6
, newip
->ip6
,
108 ip_vs_check_diff2(oldport
, newport
,
109 ~csum_unfold(uhdr
->check
))));
113 csum_fold(ip_vs_check_diff4(oldip
->ip
, newip
->ip
,
114 ip_vs_check_diff2(oldport
, newport
,
115 ~csum_unfold(uhdr
->check
))));
117 uhdr
->check
= CSUM_MANGLED_0
;
121 udp_partial_csum_update(int af
, struct udphdr
*uhdr
,
122 const union nf_inet_addr
*oldip
,
123 const union nf_inet_addr
*newip
,
124 __be16 oldlen
, __be16 newlen
)
126 #ifdef CONFIG_IP_VS_IPV6
129 ~csum_fold(ip_vs_check_diff16(oldip
->ip6
, newip
->ip6
,
130 ip_vs_check_diff2(oldlen
, newlen
,
131 csum_unfold(uhdr
->check
))));
135 ~csum_fold(ip_vs_check_diff4(oldip
->ip
, newip
->ip
,
136 ip_vs_check_diff2(oldlen
, newlen
,
137 csum_unfold(uhdr
->check
))));
142 udp_snat_handler(struct sk_buff
*skb
, struct ip_vs_protocol
*pp
,
143 struct ip_vs_conn
*cp
, struct ip_vs_iphdr
*iph
)
146 unsigned int udphoff
= iph
->len
;
148 int payload_csum
= 0;
150 #ifdef CONFIG_IP_VS_IPV6
151 if (cp
->af
== AF_INET6
&& iph
->fragoffs
)
154 oldlen
= skb
->len
- udphoff
;
156 /* csum_check requires unshared skb */
157 if (!skb_make_writable(skb
, udphoff
+sizeof(*udph
)))
160 if (unlikely(cp
->app
!= NULL
)) {
163 /* Some checks before mangling */
164 if (pp
->csum_check
&& !pp
->csum_check(cp
->af
, skb
, pp
))
168 * Call application helper if needed
170 if (!(ret
= ip_vs_app_pkt_out(cp
, skb
)))
172 /* ret=2: csum update is needed after payload mangling */
174 oldlen
= skb
->len
- udphoff
;
179 udph
= (void *)skb_network_header(skb
) + udphoff
;
180 udph
->source
= cp
->vport
;
183 * Adjust UDP checksums
185 if (skb
->ip_summed
== CHECKSUM_PARTIAL
) {
186 udp_partial_csum_update(cp
->af
, udph
, &cp
->daddr
, &cp
->vaddr
,
188 htons(skb
->len
- udphoff
));
189 } else if (!payload_csum
&& (udph
->check
!= 0)) {
190 /* Only port and addr are changed, do fast csum update */
191 udp_fast_csum_update(cp
->af
, udph
, &cp
->daddr
, &cp
->vaddr
,
192 cp
->dport
, cp
->vport
);
193 if (skb
->ip_summed
== CHECKSUM_COMPLETE
)
194 skb
->ip_summed
= (cp
->app
&& pp
->csum_check
) ?
195 CHECKSUM_UNNECESSARY
: CHECKSUM_NONE
;
197 /* full checksum calculation */
199 skb
->csum
= skb_checksum(skb
, udphoff
, skb
->len
- udphoff
, 0);
200 #ifdef CONFIG_IP_VS_IPV6
201 if (cp
->af
== AF_INET6
)
202 udph
->check
= csum_ipv6_magic(&cp
->vaddr
.in6
,
205 cp
->protocol
, skb
->csum
);
208 udph
->check
= csum_tcpudp_magic(cp
->vaddr
.ip
,
213 if (udph
->check
== 0)
214 udph
->check
= CSUM_MANGLED_0
;
215 skb
->ip_summed
= CHECKSUM_UNNECESSARY
;
216 IP_VS_DBG(11, "O-pkt: %s O-csum=%d (+%zd)\n",
217 pp
->name
, udph
->check
,
218 (char*)&(udph
->check
) - (char*)udph
);
225 udp_dnat_handler(struct sk_buff
*skb
, struct ip_vs_protocol
*pp
,
226 struct ip_vs_conn
*cp
, struct ip_vs_iphdr
*iph
)
229 unsigned int udphoff
= iph
->len
;
231 int payload_csum
= 0;
233 #ifdef CONFIG_IP_VS_IPV6
234 if (cp
->af
== AF_INET6
&& iph
->fragoffs
)
237 oldlen
= skb
->len
- udphoff
;
239 /* csum_check requires unshared skb */
240 if (!skb_make_writable(skb
, udphoff
+sizeof(*udph
)))
243 if (unlikely(cp
->app
!= NULL
)) {
246 /* Some checks before mangling */
247 if (pp
->csum_check
&& !pp
->csum_check(cp
->af
, skb
, pp
))
251 * Attempt ip_vs_app call.
252 * It will fix ip_vs_conn
254 if (!(ret
= ip_vs_app_pkt_in(cp
, skb
)))
256 /* ret=2: csum update is needed after payload mangling */
258 oldlen
= skb
->len
- udphoff
;
263 udph
= (void *)skb_network_header(skb
) + udphoff
;
264 udph
->dest
= cp
->dport
;
267 * Adjust UDP checksums
269 if (skb
->ip_summed
== CHECKSUM_PARTIAL
) {
270 udp_partial_csum_update(cp
->af
, udph
, &cp
->vaddr
, &cp
->daddr
,
272 htons(skb
->len
- udphoff
));
273 } else if (!payload_csum
&& (udph
->check
!= 0)) {
274 /* Only port and addr are changed, do fast csum update */
275 udp_fast_csum_update(cp
->af
, udph
, &cp
->vaddr
, &cp
->daddr
,
276 cp
->vport
, cp
->dport
);
277 if (skb
->ip_summed
== CHECKSUM_COMPLETE
)
278 skb
->ip_summed
= (cp
->app
&& pp
->csum_check
) ?
279 CHECKSUM_UNNECESSARY
: CHECKSUM_NONE
;
281 /* full checksum calculation */
283 skb
->csum
= skb_checksum(skb
, udphoff
, skb
->len
- udphoff
, 0);
284 #ifdef CONFIG_IP_VS_IPV6
285 if (cp
->af
== AF_INET6
)
286 udph
->check
= csum_ipv6_magic(&cp
->caddr
.in6
,
289 cp
->protocol
, skb
->csum
);
292 udph
->check
= csum_tcpudp_magic(cp
->caddr
.ip
,
297 if (udph
->check
== 0)
298 udph
->check
= CSUM_MANGLED_0
;
299 skb
->ip_summed
= CHECKSUM_UNNECESSARY
;
306 udp_csum_check(int af
, struct sk_buff
*skb
, struct ip_vs_protocol
*pp
)
308 struct udphdr _udph
, *uh
;
309 unsigned int udphoff
;
311 #ifdef CONFIG_IP_VS_IPV6
313 udphoff
= sizeof(struct ipv6hdr
);
316 udphoff
= ip_hdrlen(skb
);
318 uh
= skb_header_pointer(skb
, udphoff
, sizeof(_udph
), &_udph
);
322 if (uh
->check
!= 0) {
323 switch (skb
->ip_summed
) {
325 skb
->csum
= skb_checksum(skb
, udphoff
,
326 skb
->len
- udphoff
, 0);
327 case CHECKSUM_COMPLETE
:
328 #ifdef CONFIG_IP_VS_IPV6
329 if (af
== AF_INET6
) {
330 if (csum_ipv6_magic(&ipv6_hdr(skb
)->saddr
,
331 &ipv6_hdr(skb
)->daddr
,
333 ipv6_hdr(skb
)->nexthdr
,
335 IP_VS_DBG_RL_PKT(0, af
, pp
, skb
, 0,
336 "Failed checksum for");
341 if (csum_tcpudp_magic(ip_hdr(skb
)->saddr
,
344 ip_hdr(skb
)->protocol
,
346 IP_VS_DBG_RL_PKT(0, af
, pp
, skb
, 0,
347 "Failed checksum for");
352 /* No need to checksum. */
359 static inline __u16
udp_app_hashkey(__be16 port
)
361 return (((__force u16
)port
>> UDP_APP_TAB_BITS
) ^ (__force u16
)port
)
366 static int udp_register_app(struct net
*net
, struct ip_vs_app
*inc
)
370 __be16 port
= inc
->port
;
372 struct netns_ipvs
*ipvs
= net_ipvs(net
);
373 struct ip_vs_proto_data
*pd
= ip_vs_proto_data_get(net
, IPPROTO_UDP
);
375 hash
= udp_app_hashkey(port
);
377 list_for_each_entry(i
, &ipvs
->udp_apps
[hash
], p_list
) {
378 if (i
->port
== port
) {
383 list_add_rcu(&inc
->p_list
, &ipvs
->udp_apps
[hash
]);
384 atomic_inc(&pd
->appcnt
);
392 udp_unregister_app(struct net
*net
, struct ip_vs_app
*inc
)
394 struct ip_vs_proto_data
*pd
= ip_vs_proto_data_get(net
, IPPROTO_UDP
);
396 atomic_dec(&pd
->appcnt
);
397 list_del_rcu(&inc
->p_list
);
401 static int udp_app_conn_bind(struct ip_vs_conn
*cp
)
403 struct netns_ipvs
*ipvs
= net_ipvs(ip_vs_conn_net(cp
));
405 struct ip_vs_app
*inc
;
408 /* Default binding: bind app only for NAT */
409 if (IP_VS_FWD_METHOD(cp
) != IP_VS_CONN_F_MASQ
)
412 /* Lookup application incarnations and bind the right one */
413 hash
= udp_app_hashkey(cp
->vport
);
416 list_for_each_entry_rcu(inc
, &ipvs
->udp_apps
[hash
], p_list
) {
417 if (inc
->port
== cp
->vport
) {
418 if (unlikely(!ip_vs_app_inc_get(inc
)))
422 IP_VS_DBG_BUF(9, "%s(): Binding conn %s:%u->"
423 "%s:%u to app %s on port %u\n",
425 IP_VS_DBG_ADDR(cp
->af
, &cp
->caddr
),
427 IP_VS_DBG_ADDR(cp
->af
, &cp
->vaddr
),
429 inc
->name
, ntohs(inc
->port
));
433 result
= inc
->init_conn(inc
, cp
);
444 static const int udp_timeouts
[IP_VS_UDP_S_LAST
+1] = {
445 [IP_VS_UDP_S_NORMAL
] = 5*60*HZ
,
446 [IP_VS_UDP_S_LAST
] = 2*HZ
,
449 static const char *const udp_state_name_table
[IP_VS_UDP_S_LAST
+1] = {
450 [IP_VS_UDP_S_NORMAL
] = "UDP",
451 [IP_VS_UDP_S_LAST
] = "BUG!",
454 static const char * udp_state_name(int state
)
456 if (state
>= IP_VS_UDP_S_LAST
)
458 return udp_state_name_table
[state
] ? udp_state_name_table
[state
] : "?";
462 udp_state_transition(struct ip_vs_conn
*cp
, int direction
,
463 const struct sk_buff
*skb
,
464 struct ip_vs_proto_data
*pd
)
467 pr_err("UDP no ns data\n");
471 cp
->timeout
= pd
->timeout_table
[IP_VS_UDP_S_NORMAL
];
474 static int __udp_init(struct net
*net
, struct ip_vs_proto_data
*pd
)
476 struct netns_ipvs
*ipvs
= net_ipvs(net
);
478 ip_vs_init_hash_table(ipvs
->udp_apps
, UDP_APP_TAB_SIZE
);
479 pd
->timeout_table
= ip_vs_create_timeout_table((int *)udp_timeouts
,
480 sizeof(udp_timeouts
));
481 if (!pd
->timeout_table
)
486 static void __udp_exit(struct net
*net
, struct ip_vs_proto_data
*pd
)
488 kfree(pd
->timeout_table
);
492 struct ip_vs_protocol ip_vs_protocol_udp
= {
494 .protocol
= IPPROTO_UDP
,
495 .num_states
= IP_VS_UDP_S_LAST
,
499 .init_netns
= __udp_init
,
500 .exit_netns
= __udp_exit
,
501 .conn_schedule
= udp_conn_schedule
,
502 .conn_in_get
= ip_vs_conn_in_get_proto
,
503 .conn_out_get
= ip_vs_conn_out_get_proto
,
504 .snat_handler
= udp_snat_handler
,
505 .dnat_handler
= udp_dnat_handler
,
506 .csum_check
= udp_csum_check
,
507 .state_transition
= udp_state_transition
,
508 .state_name
= udp_state_name
,
509 .register_app
= udp_register_app
,
510 .unregister_app
= udp_unregister_app
,
511 .app_conn_bind
= udp_app_conn_bind
,
512 .debug_packet
= ip_vs_tcpudp_debug_packet
,
513 .timeout_change
= NULL
,