ipvs: use enum to instead of magic numbers
Signed-off-by: Changli Gao <xiaosuo@gmail.com> Signed-off-by: Simon Horman <horms@verge.net.au>
This commit is contained in:
parent
731109e784
commit
17a8f8e373
|
@ -43,6 +43,13 @@
|
|||
|
||||
#include <net/ip_vs.h>
|
||||
|
||||
enum {
|
||||
IP_VS_RT_MODE_LOCAL = 1, /* Allow local dest */
|
||||
IP_VS_RT_MODE_NON_LOCAL = 2, /* Allow non-local dest */
|
||||
IP_VS_RT_MODE_RDR = 4, /* Allow redirect from remote daddr to
|
||||
* local
|
||||
*/
|
||||
};
|
||||
|
||||
/*
|
||||
* Destination cache to speed up outgoing route lookup
|
||||
|
@ -77,11 +84,7 @@ __ip_vs_dst_check(struct ip_vs_dest *dest, u32 rtos)
|
|||
return dst;
|
||||
}
|
||||
|
||||
/*
|
||||
* Get route to destination or remote server
|
||||
* rt_mode: flags, &1=Allow local dest, &2=Allow non-local dest,
|
||||
* &4=Allow redirect from remote daddr to local
|
||||
*/
|
||||
/* Get route to destination or remote server */
|
||||
static struct rtable *
|
||||
__ip_vs_get_out_rt(struct sk_buff *skb, struct ip_vs_dest *dest,
|
||||
__be32 daddr, u32 rtos, int rt_mode)
|
||||
|
@ -126,15 +129,16 @@ __ip_vs_get_out_rt(struct sk_buff *skb, struct ip_vs_dest *dest,
|
|||
}
|
||||
|
||||
local = rt->rt_flags & RTCF_LOCAL;
|
||||
if (!((local ? 1 : 2) & rt_mode)) {
|
||||
if (!((local ? IP_VS_RT_MODE_LOCAL : IP_VS_RT_MODE_NON_LOCAL) &
|
||||
rt_mode)) {
|
||||
IP_VS_DBG_RL("Stopping traffic to %s address, dest: %pI4\n",
|
||||
(rt->rt_flags & RTCF_LOCAL) ?
|
||||
"local":"non-local", &rt->rt_dst);
|
||||
ip_rt_put(rt);
|
||||
return NULL;
|
||||
}
|
||||
if (local && !(rt_mode & 4) && !((ort = skb_rtable(skb)) &&
|
||||
ort->rt_flags & RTCF_LOCAL)) {
|
||||
if (local && !(rt_mode & IP_VS_RT_MODE_RDR) &&
|
||||
!((ort = skb_rtable(skb)) && ort->rt_flags & RTCF_LOCAL)) {
|
||||
IP_VS_DBG_RL("Redirect from non-local address %pI4 to local "
|
||||
"requires NAT method, dest: %pI4\n",
|
||||
&ip_hdr(skb)->daddr, &rt->rt_dst);
|
||||
|
@ -383,8 +387,8 @@ ip_vs_bypass_xmit(struct sk_buff *skb, struct ip_vs_conn *cp,
|
|||
|
||||
EnterFunction(10);
|
||||
|
||||
if (!(rt = __ip_vs_get_out_rt(skb, NULL, iph->daddr,
|
||||
RT_TOS(iph->tos), 2)))
|
||||
if (!(rt = __ip_vs_get_out_rt(skb, NULL, iph->daddr, RT_TOS(iph->tos),
|
||||
IP_VS_RT_MODE_NON_LOCAL)))
|
||||
goto tx_error_icmp;
|
||||
|
||||
/* MTU checking */
|
||||
|
@ -512,7 +516,10 @@ ip_vs_nat_xmit(struct sk_buff *skb, struct ip_vs_conn *cp,
|
|||
}
|
||||
|
||||
if (!(rt = __ip_vs_get_out_rt(skb, cp->dest, cp->daddr.ip,
|
||||
RT_TOS(iph->tos), 1|2|4)))
|
||||
RT_TOS(iph->tos),
|
||||
IP_VS_RT_MODE_LOCAL |
|
||||
IP_VS_RT_MODE_NON_LOCAL |
|
||||
IP_VS_RT_MODE_RDR)))
|
||||
goto tx_error_icmp;
|
||||
local = rt->rt_flags & RTCF_LOCAL;
|
||||
/*
|
||||
|
@ -755,7 +762,8 @@ ip_vs_tunnel_xmit(struct sk_buff *skb, struct ip_vs_conn *cp,
|
|||
EnterFunction(10);
|
||||
|
||||
if (!(rt = __ip_vs_get_out_rt(skb, cp->dest, cp->daddr.ip,
|
||||
RT_TOS(tos), 1|2)))
|
||||
RT_TOS(tos), IP_VS_RT_MODE_LOCAL |
|
||||
IP_VS_RT_MODE_NON_LOCAL)))
|
||||
goto tx_error_icmp;
|
||||
if (rt->rt_flags & RTCF_LOCAL) {
|
||||
ip_rt_put(rt);
|
||||
|
@ -984,7 +992,9 @@ ip_vs_dr_xmit(struct sk_buff *skb, struct ip_vs_conn *cp,
|
|||
EnterFunction(10);
|
||||
|
||||
if (!(rt = __ip_vs_get_out_rt(skb, cp->dest, cp->daddr.ip,
|
||||
RT_TOS(iph->tos), 1|2)))
|
||||
RT_TOS(iph->tos),
|
||||
IP_VS_RT_MODE_LOCAL |
|
||||
IP_VS_RT_MODE_NON_LOCAL)))
|
||||
goto tx_error_icmp;
|
||||
if (rt->rt_flags & RTCF_LOCAL) {
|
||||
ip_rt_put(rt);
|
||||
|
@ -1128,7 +1138,10 @@ ip_vs_icmp_xmit(struct sk_buff *skb, struct ip_vs_conn *cp,
|
|||
*/
|
||||
|
||||
if (!(rt = __ip_vs_get_out_rt(skb, cp->dest, cp->daddr.ip,
|
||||
RT_TOS(ip_hdr(skb)->tos), 1|2|4)))
|
||||
RT_TOS(ip_hdr(skb)->tos),
|
||||
IP_VS_RT_MODE_LOCAL |
|
||||
IP_VS_RT_MODE_NON_LOCAL |
|
||||
IP_VS_RT_MODE_RDR)))
|
||||
goto tx_error_icmp;
|
||||
local = rt->rt_flags & RTCF_LOCAL;
|
||||
|
||||
|
|
Loading…
Reference in New Issue