net: skb_free_datagram_locked() fix

[deliverable/linux.git] / net / core / datagram.c

diff --git a/net/core/datagram.c b/net/core/datagram.c
index 95b851f3d71380638f9c1415e919becf9db57a22..e0097531417aeeced3650a7f166ac648ae2fa798 100644 (file)
--- a/net/core/datagram.c
+++ b/net/core/datagram.c
@@ -229,13 +229,18 @@ EXPORT_SYMBOL(skb_free_datagram);
 
 void skb_free_datagram_locked(struct sock *sk, struct sk_buff *skb)
 {
+       if (likely(atomic_read(&skb->users) == 1))
+               smp_rmb();
+       else if (likely(!atomic_dec_and_test(&skb->users)))
+               return;
+
        lock_sock_bh(sk);
        skb_orphan(skb);
        sk_mem_reclaim_partial(sk);
        unlock_sock_bh(sk);
 
-       /* skb is now orphaned, might be freed outside of locked section */
-       consume_skb(skb);
+       /* skb is now orphaned, can be freed outside of locked section */
+       __kfree_skb(skb);
 }
 EXPORT_SYMBOL(skb_free_datagram_locked);