1 files changed, 4 insertions, 3 deletions

diff --git a/net/ipv4/route.c b/net/ipv4/route.c
index 1ac3ecaf36e8..78462658fccb 100644
--- a/net/ipv4/route.c
+++ b/net/ipv4/route.c
@@ -2720,7 +2720,7 @@ static int ipv4_dst_blackhole(struct net *net, struct rtable **rp, struct flowi
 }
 
 int ip_route_output_flow(struct net *net, struct rtable **rp, struct flowi *flp,
-                         struct sock *sk, bool can_sleep)
+                         struct sock *sk)
 {
         int err;
 
@@ -2733,7 +2733,8 @@ int ip_route_output_flow(struct net *net, struct rtable **rp, struct flowi *flp,
                 if (!flp->fl4_dst)
                         flp->fl4_dst = (*rp)->rt_dst;
                 err = __xfrm_lookup(net, (struct dst_entry **)rp, flp, sk,
-                                    can_sleep ? XFRM_LOOKUP_WAIT : 0);
+                                    ((flp->flags & FLOWI_FLAG_CAN_SLEEP) ?
+                                     XFRM_LOOKUP_WAIT : 0));
                 if (err == -EREMOTE)
                         err = ipv4_dst_blackhole(net, rp, flp);
 
@@ -2746,7 +2747,7 @@ EXPORT_SYMBOL_GPL(ip_route_output_flow);
 
 int ip_route_output_key(struct net *net, struct rtable **rp, struct flowi *flp)
 {
-        return ip_route_output_flow(net, rp, flp, NULL, false);
+        return ip_route_output_flow(net, rp, flp, NULL);
 }
 EXPORT_SYMBOL(ip_route_output_key);