3 files changed, 4 insertions, 4 deletions

diff --git a/net/sctp/output.c b/net/sctp/output.c
index 89cee1482d35..9bf9d84a96b7 100644
--- a/net/sctp/output.c
+++ b/net/sctp/output.c
@@ -402,7 +402,7 @@ static void sctp_packet_set_owner_w(struct sk_buff *skb, struct sock *sk)
          * therefore only reserve a single byte to keep socket around until
          * the packet has been transmitted.
          */
-        atomic_inc(&sk->sk_wmem_alloc);
+        refcount_inc(&sk->sk_wmem_alloc);
 }
 
 static int sctp_packet_pack(struct sctp_packet *packet,
diff --git a/net/sctp/proc.c b/net/sctp/proc.c
index 8e34db56bc1d..26b4be6b4172 100644
--- a/net/sctp/proc.c
+++ b/net/sctp/proc.c
@@ -363,7 +363,7 @@ static int sctp_assocs_seq_show(struct seq_file *seq, void *v)
                 assoc->stream.outcnt, assoc->max_retrans,
                 assoc->init_retries, assoc->shutdown_retries,
                 assoc->rtx_data_chunks,
-                atomic_read(&sk->sk_wmem_alloc),
+                refcount_read(&sk->sk_wmem_alloc),
                 sk->sk_wmem_queued,
                 sk->sk_sndbuf,
                 sk->sk_rcvbuf);
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