| // SPDX-License-Identifier: GPL-2.0 |
| /* |
| * Copyright (c) 2015, 2017 Oracle. All rights reserved. |
| * Copyright (c) 2003-2007 Network Appliance, Inc. All rights reserved. |
| */ |
| |
| /* Lightweight memory registration using Fast Registration Work |
| * Requests (FRWR). |
| * |
| * FRWR features ordered asynchronous registration and invalidation |
| * of arbitrarily-sized memory regions. This is the fastest and safest |
| * but most complex memory registration mode. |
| */ |
| |
| /* Normal operation |
| * |
| * A Memory Region is prepared for RDMA Read or Write using a FAST_REG |
| * Work Request (frwr_map). When the RDMA operation is finished, this |
| * Memory Region is invalidated using a LOCAL_INV Work Request |
| * (frwr_unmap_async and frwr_unmap_sync). |
| * |
| * Typically FAST_REG Work Requests are not signaled, and neither are |
| * RDMA Send Work Requests (with the exception of signaling occasionally |
| * to prevent provider work queue overflows). This greatly reduces HCA |
| * interrupt workload. |
| */ |
| |
| /* Transport recovery |
| * |
| * frwr_map and frwr_unmap_* cannot run at the same time the transport |
| * connect worker is running. The connect worker holds the transport |
| * send lock, just as ->send_request does. This prevents frwr_map and |
| * the connect worker from running concurrently. When a connection is |
| * closed, the Receive completion queue is drained before the allowing |
| * the connect worker to get control. This prevents frwr_unmap and the |
| * connect worker from running concurrently. |
| * |
| * When the underlying transport disconnects, MRs that are in flight |
| * are flushed and are likely unusable. Thus all MRs are destroyed. |
| * New MRs are created on demand. |
| */ |
| |
| #include <linux/sunrpc/rpc_rdma.h> |
| #include <linux/sunrpc/svc_rdma.h> |
| |
| #include "xprt_rdma.h" |
| #include <trace/events/rpcrdma.h> |
| |
| #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) |
| # define RPCDBG_FACILITY RPCDBG_TRANS |
| #endif |
| |
| /** |
| * frwr_release_mr - Destroy one MR |
| * @mr: MR allocated by frwr_mr_init |
| * |
| */ |
| void frwr_release_mr(struct rpcrdma_mr *mr) |
| { |
| int rc; |
| |
| rc = ib_dereg_mr(mr->frwr.fr_mr); |
| if (rc) |
| trace_xprtrdma_frwr_dereg(mr, rc); |
| kfree(mr->mr_sg); |
| kfree(mr); |
| } |
| |
| static void frwr_mr_recycle(struct rpcrdma_mr *mr) |
| { |
| struct rpcrdma_xprt *r_xprt = mr->mr_xprt; |
| |
| trace_xprtrdma_mr_recycle(mr); |
| |
| if (mr->mr_dir != DMA_NONE) { |
| trace_xprtrdma_mr_unmap(mr); |
| ib_dma_unmap_sg(r_xprt->rx_ep->re_id->device, |
| mr->mr_sg, mr->mr_nents, mr->mr_dir); |
| mr->mr_dir = DMA_NONE; |
| } |
| |
| spin_lock(&r_xprt->rx_buf.rb_lock); |
| list_del(&mr->mr_all); |
| r_xprt->rx_stats.mrs_recycled++; |
| spin_unlock(&r_xprt->rx_buf.rb_lock); |
| |
| frwr_release_mr(mr); |
| } |
| |
| /* frwr_reset - Place MRs back on the free list |
| * @req: request to reset |
| * |
| * Used after a failed marshal. For FRWR, this means the MRs |
| * don't have to be fully released and recreated. |
| * |
| * NB: This is safe only as long as none of @req's MRs are |
| * involved with an ongoing asynchronous FAST_REG or LOCAL_INV |
| * Work Request. |
| */ |
| void frwr_reset(struct rpcrdma_req *req) |
| { |
| struct rpcrdma_mr *mr; |
| |
| while ((mr = rpcrdma_mr_pop(&req->rl_registered))) |
| rpcrdma_mr_put(mr); |
| } |
| |
| /** |
| * frwr_mr_init - Initialize one MR |
| * @r_xprt: controlling transport instance |
| * @mr: generic MR to prepare for FRWR |
| * |
| * Returns zero if successful. Otherwise a negative errno |
| * is returned. |
| */ |
| int frwr_mr_init(struct rpcrdma_xprt *r_xprt, struct rpcrdma_mr *mr) |
| { |
| struct rpcrdma_ep *ep = r_xprt->rx_ep; |
| unsigned int depth = ep->re_max_fr_depth; |
| struct scatterlist *sg; |
| struct ib_mr *frmr; |
| int rc; |
| |
| frmr = ib_alloc_mr(ep->re_pd, ep->re_mrtype, depth); |
| if (IS_ERR(frmr)) |
| goto out_mr_err; |
| |
| sg = kcalloc(depth, sizeof(*sg), GFP_NOFS); |
| if (!sg) |
| goto out_list_err; |
| |
| mr->mr_xprt = r_xprt; |
| mr->frwr.fr_mr = frmr; |
| mr->mr_dir = DMA_NONE; |
| INIT_LIST_HEAD(&mr->mr_list); |
| init_completion(&mr->frwr.fr_linv_done); |
| |
| sg_init_table(sg, depth); |
| mr->mr_sg = sg; |
| return 0; |
| |
| out_mr_err: |
| rc = PTR_ERR(frmr); |
| trace_xprtrdma_frwr_alloc(mr, rc); |
| return rc; |
| |
| out_list_err: |
| ib_dereg_mr(frmr); |
| return -ENOMEM; |
| } |
| |
| /** |
| * frwr_query_device - Prepare a transport for use with FRWR |
| * @ep: endpoint to fill in |
| * @device: RDMA device to query |
| * |
| * On success, sets: |
| * ep->re_attr |
| * ep->re_max_requests |
| * ep->re_max_rdma_segs |
| * ep->re_max_fr_depth |
| * ep->re_mrtype |
| * |
| * Return values: |
| * On success, returns zero. |
| * %-EINVAL - the device does not support FRWR memory registration |
| * %-ENOMEM - the device is not sufficiently capable for NFS/RDMA |
| */ |
| int frwr_query_device(struct rpcrdma_ep *ep, const struct ib_device *device) |
| { |
| const struct ib_device_attr *attrs = &device->attrs; |
| int max_qp_wr, depth, delta; |
| unsigned int max_sge; |
| |
| if (!(attrs->device_cap_flags & IB_DEVICE_MEM_MGT_EXTENSIONS) || |
| attrs->max_fast_reg_page_list_len == 0) { |
| pr_err("rpcrdma: 'frwr' mode is not supported by device %s\n", |
| device->name); |
| return -EINVAL; |
| } |
| |
| max_sge = min_t(unsigned int, attrs->max_send_sge, |
| RPCRDMA_MAX_SEND_SGES); |
| if (max_sge < RPCRDMA_MIN_SEND_SGES) { |
| pr_err("rpcrdma: HCA provides only %u send SGEs\n", max_sge); |
| return -ENOMEM; |
| } |
| ep->re_attr.cap.max_send_sge = max_sge; |
| ep->re_attr.cap.max_recv_sge = 1; |
| |
| ep->re_mrtype = IB_MR_TYPE_MEM_REG; |
| if (attrs->device_cap_flags & IB_DEVICE_SG_GAPS_REG) |
| ep->re_mrtype = IB_MR_TYPE_SG_GAPS; |
| |
| /* Quirk: Some devices advertise a large max_fast_reg_page_list_len |
| * capability, but perform optimally when the MRs are not larger |
| * than a page. |
| */ |
| if (attrs->max_sge_rd > RPCRDMA_MAX_HDR_SEGS) |
| ep->re_max_fr_depth = attrs->max_sge_rd; |
| else |
| ep->re_max_fr_depth = attrs->max_fast_reg_page_list_len; |
| if (ep->re_max_fr_depth > RPCRDMA_MAX_DATA_SEGS) |
| ep->re_max_fr_depth = RPCRDMA_MAX_DATA_SEGS; |
| |
| /* Add room for frwr register and invalidate WRs. |
| * 1. FRWR reg WR for head |
| * 2. FRWR invalidate WR for head |
| * 3. N FRWR reg WRs for pagelist |
| * 4. N FRWR invalidate WRs for pagelist |
| * 5. FRWR reg WR for tail |
| * 6. FRWR invalidate WR for tail |
| * 7. The RDMA_SEND WR |
| */ |
| depth = 7; |
| |
| /* Calculate N if the device max FRWR depth is smaller than |
| * RPCRDMA_MAX_DATA_SEGS. |
| */ |
| if (ep->re_max_fr_depth < RPCRDMA_MAX_DATA_SEGS) { |
| delta = RPCRDMA_MAX_DATA_SEGS - ep->re_max_fr_depth; |
| do { |
| depth += 2; /* FRWR reg + invalidate */ |
| delta -= ep->re_max_fr_depth; |
| } while (delta > 0); |
| } |
| |
| max_qp_wr = attrs->max_qp_wr; |
| max_qp_wr -= RPCRDMA_BACKWARD_WRS; |
| max_qp_wr -= 1; |
| if (max_qp_wr < RPCRDMA_MIN_SLOT_TABLE) |
| return -ENOMEM; |
| if (ep->re_max_requests > max_qp_wr) |
| ep->re_max_requests = max_qp_wr; |
| ep->re_attr.cap.max_send_wr = ep->re_max_requests * depth; |
| if (ep->re_attr.cap.max_send_wr > max_qp_wr) { |
| ep->re_max_requests = max_qp_wr / depth; |
| if (!ep->re_max_requests) |
| return -ENOMEM; |
| ep->re_attr.cap.max_send_wr = ep->re_max_requests * depth; |
| } |
| ep->re_attr.cap.max_send_wr += RPCRDMA_BACKWARD_WRS; |
| ep->re_attr.cap.max_send_wr += 1; /* for ib_drain_sq */ |
| ep->re_attr.cap.max_recv_wr = ep->re_max_requests; |
| ep->re_attr.cap.max_recv_wr += RPCRDMA_BACKWARD_WRS; |
| ep->re_attr.cap.max_recv_wr += 1; /* for ib_drain_rq */ |
| |
| ep->re_max_rdma_segs = |
| DIV_ROUND_UP(RPCRDMA_MAX_DATA_SEGS, ep->re_max_fr_depth); |
| /* Reply chunks require segments for head and tail buffers */ |
| ep->re_max_rdma_segs += 2; |
| if (ep->re_max_rdma_segs > RPCRDMA_MAX_HDR_SEGS) |
| ep->re_max_rdma_segs = RPCRDMA_MAX_HDR_SEGS; |
| |
| /* Ensure the underlying device is capable of conveying the |
| * largest r/wsize NFS will ask for. This guarantees that |
| * failing over from one RDMA device to another will not |
| * break NFS I/O. |
| */ |
| if ((ep->re_max_rdma_segs * ep->re_max_fr_depth) < RPCRDMA_MAX_SEGS) |
| return -ENOMEM; |
| |
| return 0; |
| } |
| |
| /** |
| * frwr_map - Register a memory region |
| * @r_xprt: controlling transport |
| * @seg: memory region co-ordinates |
| * @nsegs: number of segments remaining |
| * @writing: true when RDMA Write will be used |
| * @xid: XID of RPC using the registered memory |
| * @mr: MR to fill in |
| * |
| * Prepare a REG_MR Work Request to register a memory region |
| * for remote access via RDMA READ or RDMA WRITE. |
| * |
| * Returns the next segment or a negative errno pointer. |
| * On success, @mr is filled in. |
| */ |
| struct rpcrdma_mr_seg *frwr_map(struct rpcrdma_xprt *r_xprt, |
| struct rpcrdma_mr_seg *seg, |
| int nsegs, bool writing, __be32 xid, |
| struct rpcrdma_mr *mr) |
| { |
| struct rpcrdma_ep *ep = r_xprt->rx_ep; |
| struct ib_reg_wr *reg_wr; |
| int i, n, dma_nents; |
| struct ib_mr *ibmr; |
| u8 key; |
| |
| if (nsegs > ep->re_max_fr_depth) |
| nsegs = ep->re_max_fr_depth; |
| for (i = 0; i < nsegs;) { |
| if (seg->mr_page) |
| sg_set_page(&mr->mr_sg[i], |
| seg->mr_page, |
| seg->mr_len, |
| offset_in_page(seg->mr_offset)); |
| else |
| sg_set_buf(&mr->mr_sg[i], seg->mr_offset, |
| seg->mr_len); |
| |
| ++seg; |
| ++i; |
| if (ep->re_mrtype == IB_MR_TYPE_SG_GAPS) |
| continue; |
| if ((i < nsegs && offset_in_page(seg->mr_offset)) || |
| offset_in_page((seg-1)->mr_offset + (seg-1)->mr_len)) |
| break; |
| } |
| mr->mr_dir = rpcrdma_data_dir(writing); |
| mr->mr_nents = i; |
| |
| dma_nents = ib_dma_map_sg(ep->re_id->device, mr->mr_sg, mr->mr_nents, |
| mr->mr_dir); |
| if (!dma_nents) |
| goto out_dmamap_err; |
| |
| ibmr = mr->frwr.fr_mr; |
| n = ib_map_mr_sg(ibmr, mr->mr_sg, dma_nents, NULL, PAGE_SIZE); |
| if (n != dma_nents) |
| goto out_mapmr_err; |
| |
| ibmr->iova &= 0x00000000ffffffff; |
| ibmr->iova |= ((u64)be32_to_cpu(xid)) << 32; |
| key = (u8)(ibmr->rkey & 0x000000FF); |
| ib_update_fast_reg_key(ibmr, ++key); |
| |
| reg_wr = &mr->frwr.fr_regwr; |
| reg_wr->mr = ibmr; |
| reg_wr->key = ibmr->rkey; |
| reg_wr->access = writing ? |
| IB_ACCESS_REMOTE_WRITE | IB_ACCESS_LOCAL_WRITE : |
| IB_ACCESS_REMOTE_READ; |
| |
| mr->mr_handle = ibmr->rkey; |
| mr->mr_length = ibmr->length; |
| mr->mr_offset = ibmr->iova; |
| trace_xprtrdma_mr_map(mr); |
| |
| return seg; |
| |
| out_dmamap_err: |
| mr->mr_dir = DMA_NONE; |
| trace_xprtrdma_frwr_sgerr(mr, i); |
| return ERR_PTR(-EIO); |
| |
| out_mapmr_err: |
| trace_xprtrdma_frwr_maperr(mr, n); |
| return ERR_PTR(-EIO); |
| } |
| |
| /** |
| * frwr_wc_fastreg - Invoked by RDMA provider for a flushed FastReg WC |
| * @cq: completion queue |
| * @wc: WCE for a completed FastReg WR |
| * |
| */ |
| static void frwr_wc_fastreg(struct ib_cq *cq, struct ib_wc *wc) |
| { |
| struct ib_cqe *cqe = wc->wr_cqe; |
| struct rpcrdma_frwr *frwr = |
| container_of(cqe, struct rpcrdma_frwr, fr_cqe); |
| |
| /* WARNING: Only wr_cqe and status are reliable at this point */ |
| trace_xprtrdma_wc_fastreg(wc, frwr); |
| /* The MR will get recycled when the associated req is retransmitted */ |
| |
| rpcrdma_flush_disconnect(cq->cq_context, wc); |
| } |
| |
| /** |
| * frwr_send - post Send WRs containing the RPC Call message |
| * @r_xprt: controlling transport instance |
| * @req: prepared RPC Call |
| * |
| * For FRWR, chain any FastReg WRs to the Send WR. Only a |
| * single ib_post_send call is needed to register memory |
| * and then post the Send WR. |
| * |
| * Returns the return code from ib_post_send. |
| * |
| * Caller must hold the transport send lock to ensure that the |
| * pointers to the transport's rdma_cm_id and QP are stable. |
| */ |
| int frwr_send(struct rpcrdma_xprt *r_xprt, struct rpcrdma_req *req) |
| { |
| struct ib_send_wr *post_wr; |
| struct rpcrdma_mr *mr; |
| |
| post_wr = &req->rl_wr; |
| list_for_each_entry(mr, &req->rl_registered, mr_list) { |
| struct rpcrdma_frwr *frwr; |
| |
| frwr = &mr->frwr; |
| |
| frwr->fr_cqe.done = frwr_wc_fastreg; |
| frwr->fr_regwr.wr.next = post_wr; |
| frwr->fr_regwr.wr.wr_cqe = &frwr->fr_cqe; |
| frwr->fr_regwr.wr.num_sge = 0; |
| frwr->fr_regwr.wr.opcode = IB_WR_REG_MR; |
| frwr->fr_regwr.wr.send_flags = 0; |
| |
| post_wr = &frwr->fr_regwr.wr; |
| } |
| |
| return ib_post_send(r_xprt->rx_ep->re_id->qp, post_wr, NULL); |
| } |
| |
| /** |
| * frwr_reminv - handle a remotely invalidated mr on the @mrs list |
| * @rep: Received reply |
| * @mrs: list of MRs to check |
| * |
| */ |
| void frwr_reminv(struct rpcrdma_rep *rep, struct list_head *mrs) |
| { |
| struct rpcrdma_mr *mr; |
| |
| list_for_each_entry(mr, mrs, mr_list) |
| if (mr->mr_handle == rep->rr_inv_rkey) { |
| list_del_init(&mr->mr_list); |
| trace_xprtrdma_mr_reminv(mr); |
| rpcrdma_mr_put(mr); |
| break; /* only one invalidated MR per RPC */ |
| } |
| } |
| |
| static void __frwr_release_mr(struct ib_wc *wc, struct rpcrdma_mr *mr) |
| { |
| if (wc->status != IB_WC_SUCCESS) |
| frwr_mr_recycle(mr); |
| else |
| rpcrdma_mr_put(mr); |
| } |
| |
| /** |
| * frwr_wc_localinv - Invoked by RDMA provider for a LOCAL_INV WC |
| * @cq: completion queue |
| * @wc: WCE for a completed LocalInv WR |
| * |
| */ |
| static void frwr_wc_localinv(struct ib_cq *cq, struct ib_wc *wc) |
| { |
| struct ib_cqe *cqe = wc->wr_cqe; |
| struct rpcrdma_frwr *frwr = |
| container_of(cqe, struct rpcrdma_frwr, fr_cqe); |
| struct rpcrdma_mr *mr = container_of(frwr, struct rpcrdma_mr, frwr); |
| |
| /* WARNING: Only wr_cqe and status are reliable at this point */ |
| trace_xprtrdma_wc_li(wc, frwr); |
| __frwr_release_mr(wc, mr); |
| |
| rpcrdma_flush_disconnect(cq->cq_context, wc); |
| } |
| |
| /** |
| * frwr_wc_localinv_wake - Invoked by RDMA provider for a LOCAL_INV WC |
| * @cq: completion queue |
| * @wc: WCE for a completed LocalInv WR |
| * |
| * Awaken anyone waiting for an MR to finish being fenced. |
| */ |
| static void frwr_wc_localinv_wake(struct ib_cq *cq, struct ib_wc *wc) |
| { |
| struct ib_cqe *cqe = wc->wr_cqe; |
| struct rpcrdma_frwr *frwr = |
| container_of(cqe, struct rpcrdma_frwr, fr_cqe); |
| struct rpcrdma_mr *mr = container_of(frwr, struct rpcrdma_mr, frwr); |
| |
| /* WARNING: Only wr_cqe and status are reliable at this point */ |
| trace_xprtrdma_wc_li_wake(wc, frwr); |
| __frwr_release_mr(wc, mr); |
| complete(&frwr->fr_linv_done); |
| |
| rpcrdma_flush_disconnect(cq->cq_context, wc); |
| } |
| |
| /** |
| * frwr_unmap_sync - invalidate memory regions that were registered for @req |
| * @r_xprt: controlling transport instance |
| * @req: rpcrdma_req with a non-empty list of MRs to process |
| * |
| * Sleeps until it is safe for the host CPU to access the previously mapped |
| * memory regions. This guarantees that registered MRs are properly fenced |
| * from the server before the RPC consumer accesses the data in them. It |
| * also ensures proper Send flow control: waking the next RPC waits until |
| * this RPC has relinquished all its Send Queue entries. |
| */ |
| void frwr_unmap_sync(struct rpcrdma_xprt *r_xprt, struct rpcrdma_req *req) |
| { |
| struct ib_send_wr *first, **prev, *last; |
| const struct ib_send_wr *bad_wr; |
| struct rpcrdma_frwr *frwr; |
| struct rpcrdma_mr *mr; |
| int rc; |
| |
| /* ORDER: Invalidate all of the MRs first |
| * |
| * Chain the LOCAL_INV Work Requests and post them with |
| * a single ib_post_send() call. |
| */ |
| frwr = NULL; |
| prev = &first; |
| while ((mr = rpcrdma_mr_pop(&req->rl_registered))) { |
| |
| trace_xprtrdma_mr_localinv(mr); |
| r_xprt->rx_stats.local_inv_needed++; |
| |
| frwr = &mr->frwr; |
| frwr->fr_cqe.done = frwr_wc_localinv; |
| last = &frwr->fr_invwr; |
| last->next = NULL; |
| last->wr_cqe = &frwr->fr_cqe; |
| last->sg_list = NULL; |
| last->num_sge = 0; |
| last->opcode = IB_WR_LOCAL_INV; |
| last->send_flags = IB_SEND_SIGNALED; |
| last->ex.invalidate_rkey = mr->mr_handle; |
| |
| *prev = last; |
| prev = &last->next; |
| } |
| |
| /* Strong send queue ordering guarantees that when the |
| * last WR in the chain completes, all WRs in the chain |
| * are complete. |
| */ |
| frwr->fr_cqe.done = frwr_wc_localinv_wake; |
| reinit_completion(&frwr->fr_linv_done); |
| |
| /* Transport disconnect drains the receive CQ before it |
| * replaces the QP. The RPC reply handler won't call us |
| * unless re_id->qp is a valid pointer. |
| */ |
| bad_wr = NULL; |
| rc = ib_post_send(r_xprt->rx_ep->re_id->qp, first, &bad_wr); |
| |
| /* The final LOCAL_INV WR in the chain is supposed to |
| * do the wake. If it was never posted, the wake will |
| * not happen, so don't wait in that case. |
| */ |
| if (bad_wr != first) |
| wait_for_completion(&frwr->fr_linv_done); |
| if (!rc) |
| return; |
| |
| /* Recycle MRs in the LOCAL_INV chain that did not get posted. |
| */ |
| trace_xprtrdma_post_linv(req, rc); |
| while (bad_wr) { |
| frwr = container_of(bad_wr, struct rpcrdma_frwr, |
| fr_invwr); |
| mr = container_of(frwr, struct rpcrdma_mr, frwr); |
| bad_wr = bad_wr->next; |
| |
| list_del_init(&mr->mr_list); |
| frwr_mr_recycle(mr); |
| } |
| } |
| |
| /** |
| * frwr_wc_localinv_done - Invoked by RDMA provider for a signaled LOCAL_INV WC |
| * @cq: completion queue |
| * @wc: WCE for a completed LocalInv WR |
| * |
| */ |
| static void frwr_wc_localinv_done(struct ib_cq *cq, struct ib_wc *wc) |
| { |
| struct ib_cqe *cqe = wc->wr_cqe; |
| struct rpcrdma_frwr *frwr = |
| container_of(cqe, struct rpcrdma_frwr, fr_cqe); |
| struct rpcrdma_mr *mr = container_of(frwr, struct rpcrdma_mr, frwr); |
| struct rpcrdma_rep *rep = mr->mr_req->rl_reply; |
| |
| /* WARNING: Only wr_cqe and status are reliable at this point */ |
| trace_xprtrdma_wc_li_done(wc, frwr); |
| __frwr_release_mr(wc, mr); |
| |
| /* Ensure @rep is generated before __frwr_release_mr */ |
| smp_rmb(); |
| rpcrdma_complete_rqst(rep); |
| |
| rpcrdma_flush_disconnect(cq->cq_context, wc); |
| } |
| |
| /** |
| * frwr_unmap_async - invalidate memory regions that were registered for @req |
| * @r_xprt: controlling transport instance |
| * @req: rpcrdma_req with a non-empty list of MRs to process |
| * |
| * This guarantees that registered MRs are properly fenced from the |
| * server before the RPC consumer accesses the data in them. It also |
| * ensures proper Send flow control: waking the next RPC waits until |
| * this RPC has relinquished all its Send Queue entries. |
| */ |
| void frwr_unmap_async(struct rpcrdma_xprt *r_xprt, struct rpcrdma_req *req) |
| { |
| struct ib_send_wr *first, *last, **prev; |
| const struct ib_send_wr *bad_wr; |
| struct rpcrdma_frwr *frwr; |
| struct rpcrdma_mr *mr; |
| int rc; |
| |
| /* Chain the LOCAL_INV Work Requests and post them with |
| * a single ib_post_send() call. |
| */ |
| frwr = NULL; |
| prev = &first; |
| while ((mr = rpcrdma_mr_pop(&req->rl_registered))) { |
| |
| trace_xprtrdma_mr_localinv(mr); |
| r_xprt->rx_stats.local_inv_needed++; |
| |
| frwr = &mr->frwr; |
| frwr->fr_cqe.done = frwr_wc_localinv; |
| last = &frwr->fr_invwr; |
| last->next = NULL; |
| last->wr_cqe = &frwr->fr_cqe; |
| last->sg_list = NULL; |
| last->num_sge = 0; |
| last->opcode = IB_WR_LOCAL_INV; |
| last->send_flags = IB_SEND_SIGNALED; |
| last->ex.invalidate_rkey = mr->mr_handle; |
| |
| *prev = last; |
| prev = &last->next; |
| } |
| |
| /* Strong send queue ordering guarantees that when the |
| * last WR in the chain completes, all WRs in the chain |
| * are complete. The last completion will wake up the |
| * RPC waiter. |
| */ |
| frwr->fr_cqe.done = frwr_wc_localinv_done; |
| |
| /* Transport disconnect drains the receive CQ before it |
| * replaces the QP. The RPC reply handler won't call us |
| * unless re_id->qp is a valid pointer. |
| */ |
| bad_wr = NULL; |
| rc = ib_post_send(r_xprt->rx_ep->re_id->qp, first, &bad_wr); |
| if (!rc) |
| return; |
| |
| /* Recycle MRs in the LOCAL_INV chain that did not get posted. |
| */ |
| trace_xprtrdma_post_linv(req, rc); |
| while (bad_wr) { |
| frwr = container_of(bad_wr, struct rpcrdma_frwr, fr_invwr); |
| mr = container_of(frwr, struct rpcrdma_mr, frwr); |
| bad_wr = bad_wr->next; |
| |
| frwr_mr_recycle(mr); |
| } |
| |
| /* The final LOCAL_INV WR in the chain is supposed to |
| * do the wake. If it was never posted, the wake will |
| * not happen, so wake here in that case. |
| */ |
| rpcrdma_complete_rqst(req->rl_reply); |
| } |