1 files changed, 4 insertions, 3 deletions

diff --git a/net/ipv4/inet_diag.c b/net/ipv4/inet_diag.c
index 6029157a19ed..5fdb02f5598e 100644
--- a/net/ipv4/inet_diag.c
+++ b/net/ipv4/inet_diag.c
@@ -357,18 +357,18 @@ struct sock *inet_diag_find_one_icsk(struct net *net,
         struct sock *sk;
 
         if (req->sdiag_family == AF_INET)
-                sk = inet_lookup(net, hashinfo, req->id.idiag_dst[0],
+                sk = inet_lookup(net, hashinfo, NULL, 0, req->id.idiag_dst[0],
                                  req->id.idiag_dport, req->id.idiag_src[0],
                                  req->id.idiag_sport, req->id.idiag_if);
 #if IS_ENABLED(CONFIG_IPV6)
         else if (req->sdiag_family == AF_INET6) {
                 if (ipv6_addr_v4mapped((struct in6_addr *)req->id.idiag_dst) &&
                     ipv6_addr_v4mapped((struct in6_addr *)req->id.idiag_src))
-                        sk = inet_lookup(net, hashinfo, req->id.idiag_dst[3],
+                        sk = inet_lookup(net, hashinfo, NULL, 0, req->id.idiag_dst[3],
                                          req->id.idiag_dport, req->id.idiag_src[3],
                                          req->id.idiag_sport, req->id.idiag_if);
                 else
-                        sk = inet6_lookup(net, hashinfo,
+                        sk = inet6_lookup(net, hashinfo, NULL, 0,
                                           (struct in6_addr *)req->id.idiag_dst,
                                           req->id.idiag_dport,
                                           (struct in6_addr *)req->id.idiag_src,
@@ -879,6 +879,7 @@ next_normal:
                 }
 
                 spin_unlock_bh(lock);
+                cond_resched();
         }
 
 done: