- make the siw_user_mmap_entry.address a void *, which naturally fits with remap_vmalloc_range. also avoids other casting during resource address assignment. - do not kfree SQ/RQ/CQ/SRQ in preparation of mmap. Those resources are always further needed ;) - Fix check for correct mmap range in siw_mmap(). - entry->length is the object length. We have to expand to PAGE_ALIGN(entry->length), since mmap comes with complete page(s) containing the object. - put mmap_entry if that check fails. Otherwise entry object ref counting screws up, and later crashes during context close. - simplify siw_mmap_free() - it must just free the entry. --- drivers/infiniband/sw/siw/siw.h | 2 +- drivers/infiniband/sw/siw/siw_verbs.c | 59 +++++++++------------------ 2 files changed, 20 insertions(+), 41 deletions(-) diff --git a/drivers/infiniband/sw/siw/siw.h b/drivers/infiniband/sw/siw/siw.h index d48cd42ae43e..d62f18f49ac5 100644 --- a/drivers/infiniband/sw/siw/siw.h +++ b/drivers/infiniband/sw/siw/siw.h @@ -505,7 +505,7 @@ struct iwarp_msg_info { struct siw_user_mmap_entry { struct rdma_user_mmap_entry rdma_entry; - u64 address; + void *address; u64 length; }; diff --git a/drivers/infiniband/sw/siw/siw_verbs.c b/drivers/infiniband/sw/siw/siw_verbs.c index 9e049241051e..24bdf5b508e6 100644 --- a/drivers/infiniband/sw/siw/siw_verbs.c +++ b/drivers/infiniband/sw/siw/siw_verbs.c @@ -36,10 +36,7 @@ static char ib_qp_state_to_string[IB_QPS_ERR + 1][sizeof("RESET")] = { void siw_mmap_free(struct rdma_user_mmap_entry *rdma_entry) { - struct siw_user_mmap_entry *entry = to_siw_mmap_entry(rdma_entry); - - vfree((void *)entry->address); - kfree(entry); + kfree(rdma_entry); } int siw_mmap(struct ib_ucontext *ctx, struct vm_area_struct *vma) @@ -56,29 +53,28 @@ int siw_mmap(struct ib_ucontext *ctx, struct vm_area_struct *vma) */ if (vma->vm_start & (PAGE_SIZE - 1)) { pr_warn("siw: mmap not page aligned\n"); - goto out; + return -EINVAL; } rdma_entry = rdma_user_mmap_entry_get(&uctx->base_ucontext, off, size, vma); if (!rdma_entry) { siw_dbg(&uctx->sdev->base_dev, "mmap lookup failed: %lu, %u\n", off, size); - goto out; + return -EINVAL; } entry = to_siw_mmap_entry(rdma_entry); - if (entry->length != size) { + if (PAGE_ALIGN(entry->length) != size) { siw_dbg(&uctx->sdev->base_dev, "key[%#lx] does not have valid length[%#x] expected[%#llx]\n", off, size, entry->length); + rdma_user_mmap_entry_put(&uctx->base_ucontext, rdma_entry); return -EINVAL; } - - rv = remap_vmalloc_range(vma, (void *)entry->address, 0); + rv = remap_vmalloc_range(vma, entry->address, 0); if (rv) { pr_warn("remap_vmalloc_range failed: %lu, %u\n", off, size); rdma_user_mmap_entry_put(&uctx->base_ucontext, rdma_entry); } -out: return rv; } @@ -270,7 +266,7 @@ void siw_qp_put_ref(struct ib_qp *base_qp) } static int siw_user_mmap_entry_insert(struct ib_ucontext *ucontext, - u64 address, u64 length, + void *address, u64 length, u64 *key) { struct siw_user_mmap_entry *entry = kzalloc(sizeof(*entry), GFP_KERNEL); @@ -461,30 +457,22 @@ struct ib_qp *siw_create_qp(struct ib_pd *pd, if (qp->sendq) { length = num_sqe * sizeof(struct siw_sqe); rv = siw_user_mmap_entry_insert(&uctx->base_ucontext, - (uintptr_t)qp->sendq, - length, &key); - if (!rv) + qp->sendq, length, + &key); + if (rv) goto err_out_xa; - /* If entry was inserted successfully, qp->sendq - * will be freed by siw_mmap_free - */ - qp->sendq = NULL; qp->sq_key = key; } if (qp->recvq) { length = num_rqe * sizeof(struct siw_rqe); rv = siw_user_mmap_entry_insert(&uctx->base_ucontext, - (uintptr_t)qp->recvq, - length, &key); - if (!rv) + qp->recvq, length, + &key); + if (rv) goto err_out_mmap_rem; - /* If entry was inserted successfully, qp->recvq will - * be freed by siw_mmap_free - */ - qp->recvq = NULL; qp->rq_key = key; } @@ -1078,16 +1066,11 @@ int siw_create_cq(struct ib_cq *base_cq, const struct ib_cq_init_attr *attr, sizeof(struct siw_cq_ctrl); rv = siw_user_mmap_entry_insert(&ctx->base_ucontext, - (uintptr_t)cq->queue, - length, &cq->cq_key); - if (!rv) + cq->queue, length, + &cq->cq_key); + if (rv) goto err_out; - /* If entry was inserted successfully, cq->queue will be freed - * by siw_mmap_free - */ - cq->queue = NULL; - uresp.cq_key = cq->cq_key; uresp.cq_id = cq->id; uresp.num_cqe = size; @@ -1535,15 +1518,11 @@ int siw_create_srq(struct ib_srq *base_srq, u64 length = srq->num_rqe * sizeof(struct siw_rqe); rv = siw_user_mmap_entry_insert(&ctx->base_ucontext, - (uintptr_t)srq->recvq, - length, &srq->srq_key); - if (!rv) + srq->recvq, length, + &srq->srq_key); + if (rv) goto err_out; - /* If entry was inserted successfully, srq->recvq will be freed - * by siw_mmap_free - */ - srq->recvq = NULL; uresp.srq_key = srq->srq_key; uresp.num_rqe = srq->num_rqe; -- 2.17.2