1 files changed, 2 insertions, 2 deletions

diff --git a/net/sctp/socket.c b/net/sctp/socket.c
index b497ee8ae279..15401d09efc4 100644
--- a/net/sctp/socket.c
+++ b/net/sctp/socket.c
@@ -164,7 +164,7 @@ static inline void sctp_set_owner_w(struct sctp_chunk *chunk)
                                 sizeof(struct sk_buff) +
                                 sizeof(struct sctp_chunk);
 
-        atomic_add(sizeof(struct sctp_chunk), &sk->sk_wmem_alloc);
+        refcount_add(sizeof(struct sctp_chunk), &sk->sk_wmem_alloc);
         sk->sk_wmem_queued += chunk->skb->truesize;
         sk_mem_charge(sk, chunk->skb->truesize);
 }
@@ -7684,7 +7684,7 @@ static void sctp_wfree(struct sk_buff *skb)
                                 sizeof(struct sk_buff) +
                                 sizeof(struct sctp_chunk);
 
-        atomic_sub(sizeof(struct sctp_chunk), &sk->sk_wmem_alloc);
+        WARN_ON(refcount_sub_and_test(sizeof(struct sctp_chunk), &sk->sk_wmem_alloc));
 
         /*
          * This undoes what is done via sctp_set_owner_w and sk_mem_charge