2 * ip_vs_xmit.c: various packet transmitters for IPVS
4 * Version: $Id: ip_vs_xmit.c,v 1.2 2002/11/30 01:50:35 wensong Exp $
6 * Authors: Wensong Zhang <wensong@linuxvirtualserver.org>
7 * Julian Anastasov <ja@ssi.bg>
9 * This program is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU General Public License
11 * as published by the Free Software Foundation; either version
12 * 2 of the License, or (at your option) any later version.
18 #include <linux/kernel.h>
19 #include <linux/tcp.h> /* for tcphdr */
21 #include <net/tcp.h> /* for csum_tcpudp_magic */
23 #include <net/icmp.h> /* for icmp_send */
24 #include <net/route.h> /* for ip_route_output */
25 #include <linux/netfilter.h>
26 #include <linux/netfilter_ipv4.h>
28 #include <net/ip_vs.h>
32 * Destination cache to speed up outgoing route lookup
35 __ip_vs_dst_set(struct ip_vs_dest
*dest
, u32 rtos
, struct dst_entry
*dst
)
37 struct dst_entry
*old_dst
;
39 old_dst
= dest
->dst_cache
;
40 dest
->dst_cache
= dst
;
41 dest
->dst_rtos
= rtos
;
45 static inline struct dst_entry
*
46 __ip_vs_dst_check(struct ip_vs_dest
*dest
, u32 rtos
, u32 cookie
)
48 struct dst_entry
*dst
= dest
->dst_cache
;
52 if ((dst
->obsolete
|| rtos
!= dest
->dst_rtos
) &&
53 dst
->ops
->check(dst
, cookie
) == NULL
) {
54 dest
->dst_cache
= NULL
;
62 static struct rtable
*
63 __ip_vs_get_out_rt(struct ip_vs_conn
*cp
, u32 rtos
)
65 struct rtable
*rt
; /* Route to the other host */
66 struct ip_vs_dest
*dest
= cp
->dest
;
69 spin_lock(&dest
->dst_lock
);
70 if (!(rt
= (struct rtable
*)
71 __ip_vs_dst_check(dest
, rtos
, 0))) {
81 if (ip_route_output_key(&init_net
, &rt
, &fl
)) {
82 spin_unlock(&dest
->dst_lock
);
83 IP_VS_DBG_RL("ip_route_output error, "
84 "dest: %u.%u.%u.%u\n",
88 __ip_vs_dst_set(dest
, rtos
, dst_clone(&rt
->u
.dst
));
89 IP_VS_DBG(10, "new dst %u.%u.%u.%u, refcnt=%d, rtos=%X\n",
91 atomic_read(&rt
->u
.dst
.__refcnt
), rtos
);
93 spin_unlock(&dest
->dst_lock
);
104 if (ip_route_output_key(&init_net
, &rt
, &fl
)) {
105 IP_VS_DBG_RL("ip_route_output error, dest: "
106 "%u.%u.%u.%u\n", NIPQUAD(cp
->daddr
));
116 * Release dest->dst_cache before a dest is removed
119 ip_vs_dst_reset(struct ip_vs_dest
*dest
)
121 struct dst_entry
*old_dst
;
123 old_dst
= dest
->dst_cache
;
124 dest
->dst_cache
= NULL
;
125 dst_release(old_dst
);
128 #define IP_VS_XMIT(skb, rt) \
130 (skb)->ipvs_property = 1; \
131 skb_forward_csum(skb); \
132 NF_HOOK(PF_INET, NF_INET_LOCAL_OUT, (skb), NULL, \
133 (rt)->u.dst.dev, dst_output); \
138 * NULL transmitter (do nothing except return NF_ACCEPT)
141 ip_vs_null_xmit(struct sk_buff
*skb
, struct ip_vs_conn
*cp
,
142 struct ip_vs_protocol
*pp
)
144 /* we do not touch skb and do not need pskb ptr */
151 * Let packets bypass the destination when the destination is not
152 * available, it may be only used in transparent cache cluster.
155 ip_vs_bypass_xmit(struct sk_buff
*skb
, struct ip_vs_conn
*cp
,
156 struct ip_vs_protocol
*pp
)
158 struct rtable
*rt
; /* Route to the other host */
159 struct iphdr
*iph
= ip_hdr(skb
);
168 .tos
= RT_TOS(tos
), } },
173 if (ip_route_output_key(&init_net
, &rt
, &fl
)) {
174 IP_VS_DBG_RL("ip_vs_bypass_xmit(): ip_route_output error, "
175 "dest: %u.%u.%u.%u\n", NIPQUAD(iph
->daddr
));
180 mtu
= dst_mtu(&rt
->u
.dst
);
181 if ((skb
->len
> mtu
) && (iph
->frag_off
& htons(IP_DF
))) {
183 icmp_send(skb
, ICMP_DEST_UNREACH
,ICMP_FRAG_NEEDED
, htonl(mtu
));
184 IP_VS_DBG_RL("ip_vs_bypass_xmit(): frag needed\n");
189 * Call ip_send_check because we are not sure it is called
190 * after ip_defrag. Is copy-on-write needed?
192 if (unlikely((skb
= skb_share_check(skb
, GFP_ATOMIC
)) == NULL
)) {
196 ip_send_check(ip_hdr(skb
));
199 dst_release(skb
->dst
);
200 skb
->dst
= &rt
->u
.dst
;
202 /* Another hack: avoid icmp_send in ip_fragment */
211 dst_link_failure(skb
);
220 * NAT transmitter (only for outside-to-inside nat forwarding)
221 * Not used for related ICMP
224 ip_vs_nat_xmit(struct sk_buff
*skb
, struct ip_vs_conn
*cp
,
225 struct ip_vs_protocol
*pp
)
227 struct rtable
*rt
; /* Route to the other host */
229 struct iphdr
*iph
= ip_hdr(skb
);
233 /* check if it is a connection of no-client-port */
234 if (unlikely(cp
->flags
& IP_VS_CONN_F_NO_CPORT
)) {
236 p
= skb_header_pointer(skb
, iph
->ihl
*4, sizeof(_pt
), &_pt
);
239 ip_vs_conn_fill_cport(cp
, *p
);
240 IP_VS_DBG(10, "filled cport=%d\n", ntohs(*p
));
243 if (!(rt
= __ip_vs_get_out_rt(cp
, RT_TOS(iph
->tos
))))
247 mtu
= dst_mtu(&rt
->u
.dst
);
248 if ((skb
->len
> mtu
) && (iph
->frag_off
& htons(IP_DF
))) {
250 icmp_send(skb
, ICMP_DEST_UNREACH
,ICMP_FRAG_NEEDED
, htonl(mtu
));
251 IP_VS_DBG_RL_PKT(0, pp
, skb
, 0, "ip_vs_nat_xmit(): frag needed for");
255 /* copy-on-write the packet before mangling it */
256 if (!skb_make_writable(skb
, sizeof(struct iphdr
)))
259 if (skb_cow(skb
, rt
->u
.dst
.dev
->hard_header_len
))
263 dst_release(skb
->dst
);
264 skb
->dst
= &rt
->u
.dst
;
266 /* mangle the packet */
267 if (pp
->dnat_handler
&& !pp
->dnat_handler(skb
, pp
, cp
))
269 ip_hdr(skb
)->daddr
= cp
->daddr
;
270 ip_send_check(ip_hdr(skb
));
272 IP_VS_DBG_PKT(10, pp
, skb
, 0, "After DNAT");
274 /* FIXME: when application helper enlarges the packet and the length
275 is larger than the MTU of outgoing device, there will be still
278 /* Another hack: avoid icmp_send in ip_fragment */
287 dst_link_failure(skb
);
299 * IP Tunneling transmitter
301 * This function encapsulates the packet in a new IP packet, its
302 * destination will be set to cp->daddr. Most code of this function
303 * is taken from ipip.c.
305 * It is used in VS/TUN cluster. The load balancer selects a real
306 * server from a cluster based on a scheduling algorithm,
307 * encapsulates the request packet and forwards it to the selected
308 * server. For example, all real servers are configured with
309 * "ifconfig tunl0 <Virtual IP Address> up". When the server receives
310 * the encapsulated packet, it will decapsulate the packet, processe
311 * the request and return the response packets directly to the client
312 * without passing the load balancer. This can greatly increase the
313 * scalability of virtual server.
315 * Used for ANY protocol
318 ip_vs_tunnel_xmit(struct sk_buff
*skb
, struct ip_vs_conn
*cp
,
319 struct ip_vs_protocol
*pp
)
321 struct rtable
*rt
; /* Route to the other host */
322 struct net_device
*tdev
; /* Device to other host */
323 struct iphdr
*old_iph
= ip_hdr(skb
);
324 u8 tos
= old_iph
->tos
;
325 __be16 df
= old_iph
->frag_off
;
326 sk_buff_data_t old_transport_header
= skb
->transport_header
;
327 struct iphdr
*iph
; /* Our new IP header */
328 unsigned int max_headroom
; /* The extra header space needed */
333 if (skb
->protocol
!= htons(ETH_P_IP
)) {
334 IP_VS_DBG_RL("ip_vs_tunnel_xmit(): protocol error, "
335 "ETH_P_IP: %d, skb protocol: %d\n",
336 htons(ETH_P_IP
), skb
->protocol
);
340 if (!(rt
= __ip_vs_get_out_rt(cp
, RT_TOS(tos
))))
343 tdev
= rt
->u
.dst
.dev
;
345 mtu
= dst_mtu(&rt
->u
.dst
) - sizeof(struct iphdr
);
348 IP_VS_DBG_RL("ip_vs_tunnel_xmit(): mtu less than 68\n");
352 skb
->dst
->ops
->update_pmtu(skb
->dst
, mtu
);
354 df
|= (old_iph
->frag_off
& htons(IP_DF
));
356 if ((old_iph
->frag_off
& htons(IP_DF
))
357 && mtu
< ntohs(old_iph
->tot_len
)) {
358 icmp_send(skb
, ICMP_DEST_UNREACH
,ICMP_FRAG_NEEDED
, htonl(mtu
));
360 IP_VS_DBG_RL("ip_vs_tunnel_xmit(): frag needed\n");
365 * Okay, now see if we can stuff it in the buffer as-is.
367 max_headroom
= LL_RESERVED_SPACE(tdev
) + sizeof(struct iphdr
);
369 if (skb_headroom(skb
) < max_headroom
370 || skb_cloned(skb
) || skb_shared(skb
)) {
371 struct sk_buff
*new_skb
=
372 skb_realloc_headroom(skb
, max_headroom
);
376 IP_VS_ERR_RL("ip_vs_tunnel_xmit(): no memory\n");
381 old_iph
= ip_hdr(skb
);
384 skb
->transport_header
= old_transport_header
;
386 /* fix old IP header checksum */
387 ip_send_check(old_iph
);
389 skb_push(skb
, sizeof(struct iphdr
));
390 skb_reset_network_header(skb
);
391 memset(&(IPCB(skb
)->opt
), 0, sizeof(IPCB(skb
)->opt
));
394 dst_release(skb
->dst
);
395 skb
->dst
= &rt
->u
.dst
;
398 * Push down and install the IPIP header.
402 iph
->ihl
= sizeof(struct iphdr
)>>2;
404 iph
->protocol
= IPPROTO_IPIP
;
406 iph
->daddr
= rt
->rt_dst
;
407 iph
->saddr
= rt
->rt_src
;
408 iph
->ttl
= old_iph
->ttl
;
409 ip_select_ident(iph
, &rt
->u
.dst
, NULL
);
411 /* Another hack: avoid icmp_send in ip_fragment */
421 dst_link_failure(skb
);
430 * Direct Routing transmitter
431 * Used for ANY protocol
434 ip_vs_dr_xmit(struct sk_buff
*skb
, struct ip_vs_conn
*cp
,
435 struct ip_vs_protocol
*pp
)
437 struct rtable
*rt
; /* Route to the other host */
438 struct iphdr
*iph
= ip_hdr(skb
);
443 if (!(rt
= __ip_vs_get_out_rt(cp
, RT_TOS(iph
->tos
))))
447 mtu
= dst_mtu(&rt
->u
.dst
);
448 if ((iph
->frag_off
& htons(IP_DF
)) && skb
->len
> mtu
) {
449 icmp_send(skb
, ICMP_DEST_UNREACH
,ICMP_FRAG_NEEDED
, htonl(mtu
));
451 IP_VS_DBG_RL("ip_vs_dr_xmit(): frag needed\n");
456 * Call ip_send_check because we are not sure it is called
457 * after ip_defrag. Is copy-on-write needed?
459 if (unlikely((skb
= skb_share_check(skb
, GFP_ATOMIC
)) == NULL
)) {
463 ip_send_check(ip_hdr(skb
));
466 dst_release(skb
->dst
);
467 skb
->dst
= &rt
->u
.dst
;
469 /* Another hack: avoid icmp_send in ip_fragment */
478 dst_link_failure(skb
);
487 * ICMP packet transmitter
488 * called by the ip_vs_in_icmp
491 ip_vs_icmp_xmit(struct sk_buff
*skb
, struct ip_vs_conn
*cp
,
492 struct ip_vs_protocol
*pp
, int offset
)
494 struct rtable
*rt
; /* Route to the other host */
500 /* The ICMP packet for VS/TUN, VS/DR and LOCALNODE will be
501 forwarded directly here, because there is no need to
502 translate address/port back */
503 if (IP_VS_FWD_METHOD(cp
) != IP_VS_CONN_F_MASQ
) {
505 rc
= cp
->packet_xmit(skb
, cp
, pp
);
508 /* do not touch skb anymore */
509 atomic_inc(&cp
->in_pkts
);
514 * mangle and send the packet here (only for VS/NAT)
517 if (!(rt
= __ip_vs_get_out_rt(cp
, RT_TOS(ip_hdr(skb
)->tos
))))
521 mtu
= dst_mtu(&rt
->u
.dst
);
522 if ((skb
->len
> mtu
) && (ip_hdr(skb
)->frag_off
& htons(IP_DF
))) {
524 icmp_send(skb
, ICMP_DEST_UNREACH
, ICMP_FRAG_NEEDED
, htonl(mtu
));
525 IP_VS_DBG_RL("ip_vs_in_icmp(): frag needed\n");
529 /* copy-on-write the packet before mangling it */
530 if (!skb_make_writable(skb
, offset
))
533 if (skb_cow(skb
, rt
->u
.dst
.dev
->hard_header_len
))
536 /* drop the old route when skb is not shared */
537 dst_release(skb
->dst
);
538 skb
->dst
= &rt
->u
.dst
;
540 ip_vs_nat_icmp(skb
, pp
, cp
, 0);
542 /* Another hack: avoid icmp_send in ip_fragment */
551 dst_link_failure(skb
);