net, atm: convert atm_dev.refcnt from atomic_t to refcount_t

refcount_t type and corresponding API should be
used instead of atomic_t when the variable is used as
a reference counter. This allows to avoid accidental
refcounter overflows that might lead to use-after-free
situations.

Signed-off-by: Elena Reshetova <elena.reshetova@intel.com>
Signed-off-by: Hans Liljestrand <ishkamiel@gmail.com>
Signed-off-by: Kees Cook <keescook@chromium.org>
Signed-off-by: David Windsor <dwindsor@gmail.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

2 files changed, 2 insertions, 2 deletions

diff --git a/net/atm/proc.c b/net/atm/proc.c
index 27c9c01c537d..4caca2a90ec4 100644
--- a/net/atm/proc.c
+++ b/net/atm/proc.c
@@ -61,7 +61,7 @@ static void atm_dev_info(struct seq_file *seq, const struct atm_dev *dev)
         add_stats(seq, "0", &dev->stats.aal0);
         seq_puts(seq, "  ");
         add_stats(seq, "5", &dev->stats.aal5);
-        seq_printf(seq, "\t[%d]", atomic_read(&dev->refcnt));
+        seq_printf(seq, "\t[%d]", refcount_read(&dev->refcnt));
         seq_putc(seq, '\n');
 }
 
diff --git a/net/atm/resources.c b/net/atm/resources.c
index 0447d5d0b639..918244757b7d 100644
--- a/net/atm/resources.c
+++ b/net/atm/resources.c
@@ -109,7 +109,7 @@ struct atm_dev *atm_dev_register(const char *type, struct device *parent,
         else
                 memset(&dev->flags, 0, sizeof(dev->flags));
         memset(&dev->stats, 0, sizeof(dev->stats));
-        atomic_set(&dev->refcnt, 1);
+        refcount_set(&dev->refcnt, 1);
 
         if (atm_proc_dev_register(dev) < 0) {
                 pr_err("atm_proc_dev_register failed for dev %s\n", type);