Commit 458bc30c authored by Reshetova, Elena's avatar Reshetova, Elena Committed by David S. Miller

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: default avatarElena Reshetova <elena.reshetova@intel.com>
Signed-off-by: default avatarHans Liljestrand <ishkamiel@gmail.com>
Signed-off-by: default avatarKees Cook <keescook@chromium.org>
Signed-off-by: default avatarDavid Windsor <dwindsor@gmail.com>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent e0542dd5
...@@ -11,6 +11,7 @@ ...@@ -11,6 +11,7 @@
#include <linux/uio.h> #include <linux/uio.h>
#include <net/sock.h> #include <net/sock.h>
#include <linux/atomic.h> #include <linux/atomic.h>
#include <linux/refcount.h>
#include <uapi/linux/atmdev.h> #include <uapi/linux/atmdev.h>
#ifdef CONFIG_PROC_FS #ifdef CONFIG_PROC_FS
...@@ -158,7 +159,7 @@ struct atm_dev { ...@@ -158,7 +159,7 @@ struct atm_dev {
struct k_atm_dev_stats stats; /* statistics */ struct k_atm_dev_stats stats; /* statistics */
char signal; /* signal status (ATM_PHY_SIG_*) */ char signal; /* signal status (ATM_PHY_SIG_*) */
int link_rate; /* link rate (default: OC3) */ int link_rate; /* link rate (default: OC3) */
atomic_t refcnt; /* reference count */ refcount_t refcnt; /* reference count */
spinlock_t lock; /* protect internal members */ spinlock_t lock; /* protect internal members */
#ifdef CONFIG_PROC_FS #ifdef CONFIG_PROC_FS
struct proc_dir_entry *proc_entry; /* proc entry */ struct proc_dir_entry *proc_entry; /* proc entry */
...@@ -261,13 +262,13 @@ static inline int atm_may_send(struct atm_vcc *vcc,unsigned int size) ...@@ -261,13 +262,13 @@ static inline int atm_may_send(struct atm_vcc *vcc,unsigned int size)
static inline void atm_dev_hold(struct atm_dev *dev) static inline void atm_dev_hold(struct atm_dev *dev)
{ {
atomic_inc(&dev->refcnt); refcount_inc(&dev->refcnt);
} }
static inline void atm_dev_put(struct atm_dev *dev) static inline void atm_dev_put(struct atm_dev *dev)
{ {
if (atomic_dec_and_test(&dev->refcnt)) { if (refcount_dec_and_test(&dev->refcnt)) {
BUG_ON(!test_bit(ATM_DF_REMOVED, &dev->flags)); BUG_ON(!test_bit(ATM_DF_REMOVED, &dev->flags));
if (dev->ops->dev_close) if (dev->ops->dev_close)
dev->ops->dev_close(dev); dev->ops->dev_close(dev);
......
...@@ -61,7 +61,7 @@ static void atm_dev_info(struct seq_file *seq, const struct atm_dev *dev) ...@@ -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); add_stats(seq, "0", &dev->stats.aal0);
seq_puts(seq, " "); seq_puts(seq, " ");
add_stats(seq, "5", &dev->stats.aal5); 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'); seq_putc(seq, '\n');
} }
......
...@@ -109,7 +109,7 @@ struct atm_dev *atm_dev_register(const char *type, struct device *parent, ...@@ -109,7 +109,7 @@ struct atm_dev *atm_dev_register(const char *type, struct device *parent,
else else
memset(&dev->flags, 0, sizeof(dev->flags)); memset(&dev->flags, 0, sizeof(dev->flags));
memset(&dev->stats, 0, sizeof(dev->stats)); memset(&dev->stats, 0, sizeof(dev->stats));
atomic_set(&dev->refcnt, 1); refcount_set(&dev->refcnt, 1);
if (atm_proc_dev_register(dev) < 0) { if (atm_proc_dev_register(dev) < 0) {
pr_err("atm_proc_dev_register failed for dev %s\n", type); pr_err("atm_proc_dev_register failed for dev %s\n", type);
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment