AuxXxilium/linux_dsm_epyc7002
1
0
Fork 0
mirror of https://github.com/AuxXxilium/linux_dsm_epyc7002.git synced 2024-12-17 03:59:01 +07:00
Code Issues Actions Packages Projects Releases Wiki Activity

RDMA/cm: Use refcount_t type for refcount variable

Browse Source 
This atomic in struct cm_id_private is being used as a refcount, change it
to refcount_t for better clarity and to get the refcount protections.

Link: https://lore.kernel.org/r/1573997601-4502-1-git-send-email-danitg@mellanox.com
Signed-off-by: Danit Goldberg <danitg@mellanox.com>
Reviewed-by: Leon Romanovsky <leonro@mellanox.com>
Reviewed-by: Jason Gunthorpe <jgg@mellanox.com>
Signed-off-by: Jason Gunthorpe <jgg@mellanox.com>
This commit is contained in:
Danit Goldberg 2019-11-17 15:33:21 +02:00 committed by Jason Gunthorpe
parent c16339b69c
commit 0acc637dac
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
drivers/infiniband/core/cm.c
View File

@ -237,7 +237,7 @@ struct cm_id_private {
struct rb_node sidr_id_node; struct rb_node sidr_id_node;
spinlock_t lock; /* Do not acquire inside cm.lock */ spinlock_t lock; /* Do not acquire inside cm.lock */
struct completion comp; struct completion comp;
atomic_t refcount; refcount_t refcount;
/* Number of clients sharing this ib_cm_id. Only valid for listeners. /* Number of clients sharing this ib_cm_id. Only valid for listeners.
* Protected by the cm.lock spinlock. */ * Protected by the cm.lock spinlock. */
int listen_sharecount; int listen_sharecount;
@ -282,7 +282,7 @@ static void cm_work_handler(struct work_struct *work);
static inline void cm_deref_id(struct cm_id_private *cm_id_priv) static inline void cm_deref_id(struct cm_id_private *cm_id_priv)
{ {
if (atomic_dec_and_test(&cm_id_priv->refcount)) if (refcount_dec_and_test(&cm_id_priv->refcount))
complete(&cm_id_priv->comp); complete(&cm_id_priv->comp);
} }
@ -339,7 +339,7 @@ static int cm_alloc_msg(struct cm_id_private *cm_id_priv,
m->ah = ah; m->ah = ah;
m->retries = cm_id_priv->max_cm_retries; m->retries = cm_id_priv->max_cm_retries;
atomic_inc(&cm_id_priv->refcount); refcount_inc(&cm_id_priv->refcount);
m->context[0] = cm_id_priv; m->context[0] = cm_id_priv;
*msg = m; *msg = m;
@ -600,7 +600,7 @@ static struct cm_id_private * cm_get_id(__be32 local_id, __be32 remote_id)
cm_id_priv = xa_load(&cm.local_id_table, cm_local_id(local_id)); cm_id_priv = xa_load(&cm.local_id_table, cm_local_id(local_id));
if (cm_id_priv) { if (cm_id_priv) {
if (cm_id_priv->id.remote_id == remote_id) if (cm_id_priv->id.remote_id == remote_id)
atomic_inc(&cm_id_priv->refcount); refcount_inc(&cm_id_priv->refcount);
else else
cm_id_priv = NULL; cm_id_priv = NULL;
} }
@ -857,7 +857,7 @@ struct ib_cm_id *ib_create_cm_id(struct ib_device *device,
INIT_LIST_HEAD(&cm_id_priv->prim_list); INIT_LIST_HEAD(&cm_id_priv->prim_list);
INIT_LIST_HEAD(&cm_id_priv->altr_list); INIT_LIST_HEAD(&cm_id_priv->altr_list);
atomic_set(&cm_id_priv->work_count, -1); atomic_set(&cm_id_priv->work_count, -1);
atomic_set(&cm_id_priv->refcount, 1); refcount_set(&cm_id_priv->refcount, 1);
return &cm_id_priv->id; return &cm_id_priv->id;
error: error:
@ -1204,7 +1204,7 @@ struct ib_cm_id *ib_cm_insert_listen(struct ib_device *device,
spin_unlock_irqrestore(&cm.lock, flags); spin_unlock_irqrestore(&cm.lock, flags);
return ERR_PTR(-EINVAL); return ERR_PTR(-EINVAL);
} }
atomic_inc(&cm_id_priv->refcount); refcount_inc(&cm_id_priv->refcount);
++cm_id_priv->listen_sharecount; ++cm_id_priv->listen_sharecount;
spin_unlock_irqrestore(&cm.lock, flags); spin_unlock_irqrestore(&cm.lock, flags);
@ -1861,8 +1861,8 @@ static struct cm_id_private * cm_match_req(struct cm_work *work,
NULL, 0); NULL, 0);
goto out; goto out;
} }
atomic_inc(&listen_cm_id_priv->refcount); refcount_inc(&listen_cm_id_priv->refcount);
atomic_inc(&cm_id_priv->refcount); refcount_inc(&cm_id_priv->refcount);
cm_id_priv->id.state = IB_CM_REQ_RCVD; cm_id_priv->id.state = IB_CM_REQ_RCVD;
atomic_inc(&cm_id_priv->work_count); atomic_inc(&cm_id_priv->work_count);
spin_unlock_irq(&cm.lock); spin_unlock_irq(&cm.lock);
@ -2018,7 +2018,7 @@ static int cm_req_handler(struct cm_work *work)
return 0; return 0;
rejected: rejected:
atomic_dec(&cm_id_priv->refcount); refcount_dec(&cm_id_priv->refcount);
cm_deref_id(listen_cm_id_priv); cm_deref_id(listen_cm_id_priv);
free_timeinfo: free_timeinfo:
kfree(cm_id_priv->timewait_info); kfree(cm_id_priv->timewait_info);
@ -2792,7 +2792,7 @@ static struct cm_id_private * cm_acquire_rejected_id(struct cm_rej_msg *rej_msg)
cm_local_id(timewait_info->work.local_id)); cm_local_id(timewait_info->work.local_id));
if (cm_id_priv) { if (cm_id_priv) {
if (cm_id_priv->id.remote_id == remote_id) if (cm_id_priv->id.remote_id == remote_id)
atomic_inc(&cm_id_priv->refcount); refcount_inc(&cm_id_priv->refcount);
else else
cm_id_priv = NULL; cm_id_priv = NULL;
} }
@ -3562,8 +3562,8 @@ static int cm_sidr_req_handler(struct cm_work *work)
cm_reject_sidr_req(cm_id_priv, IB_SIDR_UNSUPPORTED); cm_reject_sidr_req(cm_id_priv, IB_SIDR_UNSUPPORTED);
goto out; /* No match. */ goto out; /* No match. */
} }
atomic_inc(&cur_cm_id_priv->refcount); refcount_inc(&cur_cm_id_priv->refcount);
atomic_inc(&cm_id_priv->refcount); refcount_inc(&cm_id_priv->refcount);
spin_unlock_irq(&cm.lock); spin_unlock_irq(&cm.lock);
cm_id_priv->id.cm_handler = cur_cm_id_priv->id.cm_handler; cm_id_priv->id.cm_handler = cur_cm_id_priv->id.cm_handler;