diff options
Diffstat (limited to 'include/net/inet_hashtables.h')
-rw-r--r-- | include/net/inet_hashtables.h | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/include/net/inet_hashtables.h b/include/net/inet_hashtables.h index 3522bbcd546d..5cc182f9ecae 100644 --- a/include/net/inet_hashtables.h +++ b/include/net/inet_hashtables.h | |||
@@ -378,11 +378,15 @@ static inline struct sock *__inet_lookup_skb(struct inet_hashinfo *hashinfo, | |||
378 | const __be16 sport, | 378 | const __be16 sport, |
379 | const __be16 dport) | 379 | const __be16 dport) |
380 | { | 380 | { |
381 | struct sock *sk; | ||
381 | const struct iphdr *iph = ip_hdr(skb); | 382 | const struct iphdr *iph = ip_hdr(skb); |
382 | 383 | ||
383 | return __inet_lookup(dev_net(skb->dst->dev), hashinfo, | 384 | if (unlikely(sk = skb_steal_sock(skb))) |
384 | iph->saddr, sport, | 385 | return sk; |
385 | iph->daddr, dport, inet_iif(skb)); | 386 | else |
387 | return __inet_lookup(dev_net(skb->dst->dev), hashinfo, | ||
388 | iph->saddr, sport, | ||
389 | iph->daddr, dport, inet_iif(skb)); | ||
386 | } | 390 | } |
387 | 391 | ||
388 | extern int __inet_hash_connect(struct inet_timewait_death_row *death_row, | 392 | extern int __inet_hash_connect(struct inet_timewait_death_row *death_row, |