hollalinux
/
linux-sg2042
mirror of https://github.com/sophgo/linux-riscv.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ipvs: add ip_vs_dest_hold and ip_vs_dest_put 

ip_vs_dest_hold will be used under RCU lock
while ip_vs_dest_put can be called even after dest
is removed from service, as it happens for conns and
some schedulers.

Signed-off-by: Julian Anastasov <ja@ssi.bg>
Signed-off-by: Simon Horman <horms@verge.net.au>
This commit is contained in:
Julian Anastasov 2013-03-22 11:46:38 +02:00 committed by Pablo Neira Ayuso
parent 6b6df46663
commit fca9c20ae1
4 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
include/net/ip_vs.h
View File

@ -1427,6 +1427,16 @@ ip_vs_find_dest(struct net *net, int af, const union nf_inet_addr *daddr,
__u16 protocol, __u32 fwmark, __u32 flags); __u16 protocol, __u32 fwmark, __u32 flags);
extern struct ip_vs_dest *ip_vs_try_bind_dest(struct ip_vs_conn *cp); extern struct ip_vs_dest *ip_vs_try_bind_dest(struct ip_vs_conn *cp);
static inline void ip_vs_dest_hold(struct ip_vs_dest *dest)
{
atomic_inc(&dest->refcnt);
}
static inline void ip_vs_dest_put(struct ip_vs_dest *dest)
{
smp_mb__before_atomic_dec();
atomic_dec(&dest->refcnt);
}
/* /*
* IPVS sync daemon data and function prototypes * IPVS sync daemon data and function prototypes

9
net/netfilter/ipvs/ip_vs_conn.c
View File

@ -554,7 +554,7 @@ ip_vs_bind_dest(struct ip_vs_conn *cp, struct ip_vs_dest *dest)
return; return;
/* Increase the refcnt counter of the dest */ /* Increase the refcnt counter of the dest */
atomic_inc(&dest->refcnt); ip_vs_dest_hold(dest);
conn_flags = atomic_read(&dest->conn_flags); conn_flags = atomic_read(&dest->conn_flags);
if (cp->protocol != IPPROTO_UDP) if (cp->protocol != IPPROTO_UDP)
@ -700,12 +700,7 @@ static inline void ip_vs_unbind_dest(struct ip_vs_conn *cp)
dest->flags &= ~IP_VS_DEST_F_OVERLOAD; dest->flags &= ~IP_VS_DEST_F_OVERLOAD;
} }
/* ip_vs_dest_put(dest);
* Simply decrease the refcnt of the dest, because the
* dest will be either in service's destination list
* or in the trash.
*/
atomic_dec(&dest->refcnt);
} }
static int expire_quiescent_template(struct netns_ipvs *ipvs, static int expire_quiescent_template(struct netns_ipvs *ipvs,

4
net/netfilter/ipvs/ip_vs_ctl.c
View File

@ -616,7 +616,7 @@ struct ip_vs_dest *ip_vs_find_dest(struct net *net, int af,
if (!dest) if (!dest)
dest = ip_vs_lookup_dest(svc, daddr, port ^ dport); dest = ip_vs_lookup_dest(svc, daddr, port ^ dport);
if (dest) if (dest)
atomic_inc(&dest->refcnt); ip_vs_dest_hold(dest);
ip_vs_service_put(svc); ip_vs_service_put(svc);
return dest; return dest;
} }
@ -1056,7 +1056,7 @@ static void __ip_vs_del_dest(struct net *net, struct ip_vs_dest *dest)
ntohs(dest->port), ntohs(dest->port),
atomic_read(&dest->refcnt)); atomic_read(&dest->refcnt));
list_add(&dest->n_list, &ipvs->dest_trash); list_add(&dest->n_list, &ipvs->dest_trash);
atomic_inc(&dest->refcnt); ip_vs_dest_hold(dest);
} }
} }

4
net/netfilter/ipvs/ip_vs_sync.c
View File

@ -861,7 +861,7 @@ static void ip_vs_proc_conn(struct net *net, struct ip_vs_conn_param *param,
if (!dest) { if (!dest) {
dest = ip_vs_try_bind_dest(cp); dest = ip_vs_try_bind_dest(cp);
if (dest) if (dest)
atomic_dec(&dest->refcnt); ip_vs_dest_put(dest);
} }
} else { } else {
/* /*
@ -874,7 +874,7 @@ static void ip_vs_proc_conn(struct net *net, struct ip_vs_conn_param *param,
cp = ip_vs_conn_new(param, daddr, dport, flags, dest, fwmark); cp = ip_vs_conn_new(param, daddr, dport, flags, dest, fwmark);
if (dest) if (dest)
atomic_dec(&dest->refcnt); ip_vs_dest_put(dest);
if (!cp) { if (!cp) {
if (param->pe_data) if (param->pe_data)
kfree(param->pe_data); kfree(param->pe_data);