Commit fbfe95a4 authored by David S. Miller's avatar David S. Miller

inet: Create and use rt{,6}_get_peer_create().

There's a lot of places that open-code rt{,6}_get_peer() only because
they want to set 'create' to one.  So add an rt{,6}_get_peer_create()
for their sake.

There were also a few spots open-coding plain rt{,6}_get_peer() and
those are transformed here as well.
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent 7123aaa3
...@@ -53,18 +53,27 @@ static inline unsigned int rt6_flags2srcprefs(int flags) ...@@ -53,18 +53,27 @@ static inline unsigned int rt6_flags2srcprefs(int flags)
return (flags >> 3) & 7; return (flags >> 3) & 7;
} }
extern void rt6_bind_peer(struct rt6_info *rt, extern void rt6_bind_peer(struct rt6_info *rt, int create);
int create);
static inline struct inet_peer *rt6_get_peer(struct rt6_info *rt) static inline struct inet_peer *__rt6_get_peer(struct rt6_info *rt, int create)
{ {
if (rt->rt6i_peer) if (rt->rt6i_peer)
return rt->rt6i_peer; return rt->rt6i_peer;
rt6_bind_peer(rt, 0); rt6_bind_peer(rt, create);
return rt->rt6i_peer; return rt->rt6i_peer;
} }
static inline struct inet_peer *rt6_get_peer(struct rt6_info *rt)
{
return __rt6_get_peer(rt, 0);
}
static inline struct inet_peer *rt6_get_peer_create(struct rt6_info *rt)
{
return __rt6_get_peer(rt, 1);
}
extern void ip6_route_input(struct sk_buff *skb); extern void ip6_route_input(struct sk_buff *skb);
extern struct dst_entry * ip6_route_output(struct net *net, extern struct dst_entry * ip6_route_output(struct net *net,
......
...@@ -296,15 +296,25 @@ static inline struct rtable *ip_route_newports(struct flowi4 *fl4, struct rtable ...@@ -296,15 +296,25 @@ static inline struct rtable *ip_route_newports(struct flowi4 *fl4, struct rtable
extern void rt_bind_peer(struct rtable *rt, __be32 daddr, int create); extern void rt_bind_peer(struct rtable *rt, __be32 daddr, int create);
static inline struct inet_peer *rt_get_peer(struct rtable *rt, __be32 daddr) static inline struct inet_peer *__rt_get_peer(struct rtable *rt, __be32 daddr, int create)
{ {
if (rt->peer) if (rt->peer)
return rt->peer; return rt->peer;
rt_bind_peer(rt, daddr, 0); rt_bind_peer(rt, daddr, create);
return rt->peer; return rt->peer;
} }
static inline struct inet_peer *rt_get_peer(struct rtable *rt, __be32 daddr)
{
return __rt_get_peer(rt, daddr, 0);
}
static inline struct inet_peer *rt_get_peer_create(struct rtable *rt, __be32 daddr)
{
return __rt_get_peer(rt, daddr, 1);
}
static inline int inet_iif(const struct sk_buff *skb) static inline int inet_iif(const struct sk_buff *skb)
{ {
return skb_rtable(skb)->rt_iif; return skb_rtable(skb)->rt_iif;
......
...@@ -253,9 +253,8 @@ static inline bool icmpv4_xrlim_allow(struct net *net, struct rtable *rt, ...@@ -253,9 +253,8 @@ static inline bool icmpv4_xrlim_allow(struct net *net, struct rtable *rt,
/* Limit if icmp type is enabled in ratemask. */ /* Limit if icmp type is enabled in ratemask. */
if ((1 << type) & net->ipv4.sysctl_icmp_ratemask) { if ((1 << type) & net->ipv4.sysctl_icmp_ratemask) {
if (!rt->peer) struct inet_peer *peer = rt_get_peer_create(rt, fl4->daddr);
rt_bind_peer(rt, fl4->daddr, 1); rc = inet_peer_xrlim_allow(peer,
rc = inet_peer_xrlim_allow(rt->peer,
net->ipv4.sysctl_icmp_ratelimit); net->ipv4.sysctl_icmp_ratelimit);
} }
out: out:
......
...@@ -162,10 +162,7 @@ static u32 *ipv4_cow_metrics(struct dst_entry *dst, unsigned long old) ...@@ -162,10 +162,7 @@ static u32 *ipv4_cow_metrics(struct dst_entry *dst, unsigned long old)
struct inet_peer *peer; struct inet_peer *peer;
u32 *p = NULL; u32 *p = NULL;
if (!rt->peer) peer = rt_get_peer_create(rt, rt->rt_dst);
rt_bind_peer(rt, rt->rt_dst, 1);
peer = rt->peer;
if (peer) { if (peer) {
u32 *old_p = __DST_METRICS_PTR(old); u32 *old_p = __DST_METRICS_PTR(old);
unsigned long prev, new; unsigned long prev, new;
...@@ -1364,14 +1361,13 @@ void __ip_select_ident(struct iphdr *iph, struct dst_entry *dst, int more) ...@@ -1364,14 +1361,13 @@ void __ip_select_ident(struct iphdr *iph, struct dst_entry *dst, int more)
struct rtable *rt = (struct rtable *) dst; struct rtable *rt = (struct rtable *) dst;
if (rt && !(rt->dst.flags & DST_NOPEER)) { if (rt && !(rt->dst.flags & DST_NOPEER)) {
if (rt->peer == NULL) struct inet_peer *peer = rt_get_peer_create(rt, rt->rt_dst);
rt_bind_peer(rt, rt->rt_dst, 1);
/* If peer is attached to destination, it is never detached, /* If peer is attached to destination, it is never detached,
so that we need not to grab a lock to dereference it. so that we need not to grab a lock to dereference it.
*/ */
if (rt->peer) { if (peer) {
iph->id = htons(inet_getid(rt->peer, more)); iph->id = htons(inet_getid(peer, more));
return; return;
} }
} else if (!rt) } else if (!rt)
...@@ -1481,10 +1477,7 @@ void ip_rt_redirect(__be32 old_gw, __be32 daddr, __be32 new_gw, ...@@ -1481,10 +1477,7 @@ void ip_rt_redirect(__be32 old_gw, __be32 daddr, __be32 new_gw,
rt->rt_gateway != old_gw) rt->rt_gateway != old_gw)
continue; continue;
if (!rt->peer) peer = rt_get_peer_create(rt, rt->rt_dst);
rt_bind_peer(rt, rt->rt_dst, 1);
peer = rt->peer;
if (peer) { if (peer) {
if (peer->redirect_learned.a4 != new_gw) { if (peer->redirect_learned.a4 != new_gw) {
peer->redirect_learned.a4 = new_gw; peer->redirect_learned.a4 = new_gw;
...@@ -1579,9 +1572,7 @@ void ip_rt_send_redirect(struct sk_buff *skb) ...@@ -1579,9 +1572,7 @@ void ip_rt_send_redirect(struct sk_buff *skb)
log_martians = IN_DEV_LOG_MARTIANS(in_dev); log_martians = IN_DEV_LOG_MARTIANS(in_dev);
rcu_read_unlock(); rcu_read_unlock();
if (!rt->peer) peer = rt_get_peer_create(rt, rt->rt_dst);
rt_bind_peer(rt, rt->rt_dst, 1);
peer = rt->peer;
if (!peer) { if (!peer) {
icmp_send(skb, ICMP_REDIRECT, ICMP_REDIR_HOST, rt->rt_gateway); icmp_send(skb, ICMP_REDIRECT, ICMP_REDIR_HOST, rt->rt_gateway);
return; return;
...@@ -1646,9 +1637,7 @@ static int ip_error(struct sk_buff *skb) ...@@ -1646,9 +1637,7 @@ static int ip_error(struct sk_buff *skb)
break; break;
} }
if (!rt->peer) peer = rt_get_peer_create(rt, rt->rt_dst);
rt_bind_peer(rt, rt->rt_dst, 1);
peer = rt->peer;
send = true; send = true;
if (peer) { if (peer) {
...@@ -1754,9 +1743,7 @@ static void ip_rt_update_pmtu(struct dst_entry *dst, u32 mtu) ...@@ -1754,9 +1743,7 @@ static void ip_rt_update_pmtu(struct dst_entry *dst, u32 mtu)
dst_confirm(dst); dst_confirm(dst);
if (!rt->peer) peer = rt_get_peer_create(rt, rt->rt_dst);
rt_bind_peer(rt, rt->rt_dst, 1);
peer = rt->peer;
if (peer) { if (peer) {
unsigned long pmtu_expires = ACCESS_ONCE(peer->pmtu_expires); unsigned long pmtu_expires = ACCESS_ONCE(peer->pmtu_expires);
...@@ -1782,12 +1769,8 @@ static void ip_rt_update_pmtu(struct dst_entry *dst, u32 mtu) ...@@ -1782,12 +1769,8 @@ static void ip_rt_update_pmtu(struct dst_entry *dst, u32 mtu)
static void ipv4_validate_peer(struct rtable *rt) static void ipv4_validate_peer(struct rtable *rt)
{ {
if (rt->rt_peer_genid != rt_peer_genid()) { if (rt->rt_peer_genid != rt_peer_genid()) {
struct inet_peer *peer; struct inet_peer *peer = rt_get_peer(rt, rt->rt_dst);
if (!rt->peer)
rt_bind_peer(rt, rt->rt_dst, 0);
peer = rt->peer;
if (peer) { if (peer) {
check_peer_pmtu(&rt->dst, peer); check_peer_pmtu(&rt->dst, peer);
......
...@@ -1832,9 +1832,7 @@ struct inet_peer *tcp_v4_get_peer(struct sock *sk, bool *release_it) ...@@ -1832,9 +1832,7 @@ struct inet_peer *tcp_v4_get_peer(struct sock *sk, bool *release_it)
peer = inet_getpeer_v4(net, inet->inet_daddr, 1); peer = inet_getpeer_v4(net, inet->inet_daddr, 1);
*release_it = true; *release_it = true;
} else { } else {
if (!rt->peer) peer = rt_get_peer_create(rt, inet->inet_daddr);
rt_bind_peer(rt, inet->inet_daddr, 1);
peer = rt->peer;
*release_it = false; *release_it = false;
} }
......
...@@ -188,14 +188,14 @@ static inline bool icmpv6_xrlim_allow(struct sock *sk, u8 type, ...@@ -188,14 +188,14 @@ static inline bool icmpv6_xrlim_allow(struct sock *sk, u8 type,
} else { } else {
struct rt6_info *rt = (struct rt6_info *)dst; struct rt6_info *rt = (struct rt6_info *)dst;
int tmo = net->ipv6.sysctl.icmpv6_time; int tmo = net->ipv6.sysctl.icmpv6_time;
struct inet_peer *peer;
/* Give more bandwidth to wider prefixes. */ /* Give more bandwidth to wider prefixes. */
if (rt->rt6i_dst.plen < 128) if (rt->rt6i_dst.plen < 128)
tmo >>= ((128 - rt->rt6i_dst.plen)>>5); tmo >>= ((128 - rt->rt6i_dst.plen)>>5);
if (!rt->rt6i_peer) peer = rt6_get_peer_create(rt);
rt6_bind_peer(rt, 1); res = inet_peer_xrlim_allow(peer, tmo);
res = inet_peer_xrlim_allow(rt->rt6i_peer, tmo);
} }
dst_release(dst); dst_release(dst);
return res; return res;
......
...@@ -463,6 +463,7 @@ int ip6_forward(struct sk_buff *skb) ...@@ -463,6 +463,7 @@ int ip6_forward(struct sk_buff *skb)
*/ */
if (skb->dev == dst->dev && opt->srcrt == 0 && !skb_sec_path(skb)) { if (skb->dev == dst->dev && opt->srcrt == 0 && !skb_sec_path(skb)) {
struct in6_addr *target = NULL; struct in6_addr *target = NULL;
struct inet_peer *peer;
struct rt6_info *rt; struct rt6_info *rt;
/* /*
...@@ -476,13 +477,12 @@ int ip6_forward(struct sk_buff *skb) ...@@ -476,13 +477,12 @@ int ip6_forward(struct sk_buff *skb)
else else
target = &hdr->daddr; target = &hdr->daddr;
if (!rt->rt6i_peer) peer = rt6_get_peer_create(rt);
rt6_bind_peer(rt, 1);
/* Limit redirects both by destination (here) /* Limit redirects both by destination (here)
and by source (inside ndisc_send_redirect) and by source (inside ndisc_send_redirect)
*/ */
if (inet_peer_xrlim_allow(rt->rt6i_peer, 1*HZ)) if (inet_peer_xrlim_allow(peer, 1*HZ))
ndisc_send_redirect(skb, target); ndisc_send_redirect(skb, target);
} else { } else {
int addrtype = ipv6_addr_type(&hdr->saddr); int addrtype = ipv6_addr_type(&hdr->saddr);
...@@ -602,11 +602,8 @@ void ipv6_select_ident(struct frag_hdr *fhdr, struct rt6_info *rt) ...@@ -602,11 +602,8 @@ void ipv6_select_ident(struct frag_hdr *fhdr, struct rt6_info *rt)
int old, new; int old, new;
if (rt && !(rt->dst.flags & DST_NOPEER)) { if (rt && !(rt->dst.flags & DST_NOPEER)) {
struct inet_peer *peer; struct inet_peer *peer = rt6_get_peer_create(rt);
if (!rt->rt6i_peer)
rt6_bind_peer(rt, 1);
peer = rt->rt6i_peer;
if (peer) { if (peer) {
fhdr->identification = htonl(inet_getid(peer, 0)); fhdr->identification = htonl(inet_getid(peer, 0));
return; return;
......
...@@ -1472,6 +1472,7 @@ void ndisc_send_redirect(struct sk_buff *skb, const struct in6_addr *target) ...@@ -1472,6 +1472,7 @@ void ndisc_send_redirect(struct sk_buff *skb, const struct in6_addr *target)
struct net *net = dev_net(dev); struct net *net = dev_net(dev);
struct sock *sk = net->ipv6.ndisc_sk; struct sock *sk = net->ipv6.ndisc_sk;
int len = sizeof(struct icmp6hdr) + 2 * sizeof(struct in6_addr); int len = sizeof(struct icmp6hdr) + 2 * sizeof(struct in6_addr);
struct inet_peer *peer;
struct sk_buff *buff; struct sk_buff *buff;
struct icmp6hdr *icmph; struct icmp6hdr *icmph;
struct in6_addr saddr_buf; struct in6_addr saddr_buf;
...@@ -1518,9 +1519,8 @@ void ndisc_send_redirect(struct sk_buff *skb, const struct in6_addr *target) ...@@ -1518,9 +1519,8 @@ void ndisc_send_redirect(struct sk_buff *skb, const struct in6_addr *target)
"Redirect: destination is not a neighbour\n"); "Redirect: destination is not a neighbour\n");
goto release; goto release;
} }
if (!rt->rt6i_peer) peer = rt6_get_peer_create(rt);
rt6_bind_peer(rt, 1); if (!inet_peer_xrlim_allow(peer, 1*HZ))
if (!inet_peer_xrlim_allow(rt->rt6i_peer, 1*HZ))
goto release; goto release;
if (dev->addr_len) { if (dev->addr_len) {
......
...@@ -99,10 +99,7 @@ static u32 *ipv6_cow_metrics(struct dst_entry *dst, unsigned long old) ...@@ -99,10 +99,7 @@ static u32 *ipv6_cow_metrics(struct dst_entry *dst, unsigned long old)
if (!(rt->dst.flags & DST_HOST)) if (!(rt->dst.flags & DST_HOST))
return NULL; return NULL;
if (!rt->rt6i_peer) peer = rt6_get_peer_create(rt);
rt6_bind_peer(rt, 1);
peer = rt->rt6i_peer;
if (peer) { if (peer) {
u32 *old_p = __DST_METRICS_PTR(old); u32 *old_p = __DST_METRICS_PTR(old);
unsigned long prev, new; unsigned long prev, new;
......
...@@ -1744,9 +1744,7 @@ static struct inet_peer *tcp_v6_get_peer(struct sock *sk, bool *release_it) ...@@ -1744,9 +1744,7 @@ static struct inet_peer *tcp_v6_get_peer(struct sock *sk, bool *release_it)
peer = inet_getpeer_v6(net, &np->daddr, 1); peer = inet_getpeer_v6(net, &np->daddr, 1);
*release_it = true; *release_it = true;
} else { } else {
if (!rt->rt6i_peer) peer = rt6_get_peer_create(rt);
rt6_bind_peer(rt, 1);
peer = rt->rt6i_peer;
*release_it = false; *release_it = false;
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment