Add per-ucontext list tracking for PD objects. Each PD is added to the ucontext's pd_list on creation and removed on destruction. This enables iterating over all PDs belonging to a ucontext, which will be needed for service reset cleanup.
Signed-off-by: Long Li <[email protected]> --- drivers/infiniband/hw/mana/main.c | 21 +++++++++++++++++++++ drivers/infiniband/hw/mana/mana_ib.h | 2 ++ 2 files changed, 23 insertions(+) diff --git a/drivers/infiniband/hw/mana/main.c b/drivers/infiniband/hw/mana/main.c index fc28bdafcfd6..62d89ca06ba1 100644 --- a/drivers/infiniband/hw/mana/main.c +++ b/drivers/infiniband/hw/mana/main.c @@ -72,6 +72,7 @@ int mana_ib_alloc_pd(struct ib_pd *ibpd, struct ib_udata *udata) struct ib_device *ibdev = ibpd->device; struct gdma_create_pd_resp resp = {}; struct gdma_create_pd_req req = {}; + struct mana_ib_ucontext *mana_ucontext; enum gdma_pd_flags flags = 0; struct mana_ib_dev *dev; struct gdma_context *gc; @@ -107,6 +108,16 @@ int mana_ib_alloc_pd(struct ib_pd *ibpd, struct ib_udata *udata) mutex_init(&pd->vport_mutex); pd->vport_use_count = 0; + + INIT_LIST_HEAD(&pd->ucontext_list); + if (udata) { + mana_ucontext = rdma_udata_to_drv_context(udata, + struct mana_ib_ucontext, ibucontext); + mutex_lock(&mana_ucontext->lock); + list_add_tail(&pd->ucontext_list, &mana_ucontext->pd_list); + mutex_unlock(&mana_ucontext->lock); + } + return 0; } @@ -123,6 +134,15 @@ int mana_ib_dealloc_pd(struct ib_pd *ibpd, struct ib_udata *udata) dev = container_of(ibdev, struct mana_ib_dev, ib_dev); gc = mdev_to_gc(dev); + if (udata) { + struct mana_ib_ucontext *mana_ucontext = + rdma_udata_to_drv_context(udata, + struct mana_ib_ucontext, ibucontext); + mutex_lock(&mana_ucontext->lock); + list_del_init(&pd->ucontext_list); + mutex_unlock(&mana_ucontext->lock); + } + mana_gd_init_req_hdr(&req.hdr, GDMA_DESTROY_PD, sizeof(req), sizeof(resp)); @@ -222,6 +242,7 @@ int mana_ib_alloc_ucontext(struct ib_ucontext *ibcontext, ucontext->doorbell = doorbell_page; mutex_init(&ucontext->lock); + INIT_LIST_HEAD(&ucontext->pd_list); mutex_lock(&mdev->ucontext_lock); list_add_tail(&ucontext->dev_list, &mdev->ucontext_list); diff --git a/drivers/infiniband/hw/mana/mana_ib.h b/drivers/infiniband/hw/mana/mana_ib.h index c7e333d3e9d8..6dba08bccc18 100644 --- a/drivers/infiniband/hw/mana/mana_ib.h +++ b/drivers/infiniband/hw/mana/mana_ib.h @@ -107,6 +107,7 @@ struct mana_ib_pd { bool tx_shortform_allowed; u32 tx_vp_offset; + struct list_head ucontext_list; }; struct mana_ib_av { @@ -203,6 +204,7 @@ struct mana_ib_ucontext { struct list_head dev_list; /* Protects resource lists below */ struct mutex lock; + struct list_head pd_list; }; struct mana_ib_rwq_ind_table { -- 2.43.0
