xprtrdma: Simplify rpcrdma_convert_kvec() and frwr_map()
[linux-block.git] / net / sunrpc / xprtrdma / rpc_rdma.c
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /*
3  * Copyright (c) 2014-2020, Oracle and/or its affiliates.
4  * Copyright (c) 2003-2007 Network Appliance, Inc. All rights reserved.
5  *
6  * This software is available to you under a choice of one of two
7  * licenses.  You may choose to be licensed under the terms of the GNU
8  * General Public License (GPL) Version 2, available from the file
9  * COPYING in the main directory of this source tree, or the BSD-type
10  * license below:
11  *
12  * Redistribution and use in source and binary forms, with or without
13  * modification, are permitted provided that the following conditions
14  * are met:
15  *
16  *      Redistributions of source code must retain the above copyright
17  *      notice, this list of conditions and the following disclaimer.
18  *
19  *      Redistributions in binary form must reproduce the above
20  *      copyright notice, this list of conditions and the following
21  *      disclaimer in the documentation and/or other materials provided
22  *      with the distribution.
23  *
24  *      Neither the name of the Network Appliance, Inc. nor the names of
25  *      its contributors may be used to endorse or promote products
26  *      derived from this software without specific prior written
27  *      permission.
28  *
29  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
30  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
31  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
32  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
33  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
34  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
35  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
36  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
37  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
38  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
39  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
40  */
41
42 /*
43  * rpc_rdma.c
44  *
45  * This file contains the guts of the RPC RDMA protocol, and
46  * does marshaling/unmarshaling, etc. It is also where interfacing
47  * to the Linux RPC framework lives.
48  */
49
50 #include <linux/highmem.h>
51
52 #include <linux/sunrpc/svc_rdma.h>
53
54 #include "xprt_rdma.h"
55 #include <trace/events/rpcrdma.h>
56
57 #if IS_ENABLED(CONFIG_SUNRPC_DEBUG)
58 # define RPCDBG_FACILITY        RPCDBG_TRANS
59 #endif
60
61 /* Returns size of largest RPC-over-RDMA header in a Call message
62  *
63  * The largest Call header contains a full-size Read list and a
64  * minimal Reply chunk.
65  */
66 static unsigned int rpcrdma_max_call_header_size(unsigned int maxsegs)
67 {
68         unsigned int size;
69
70         /* Fixed header fields and list discriminators */
71         size = RPCRDMA_HDRLEN_MIN;
72
73         /* Maximum Read list size */
74         size += maxsegs * rpcrdma_readchunk_maxsz * sizeof(__be32);
75
76         /* Minimal Read chunk size */
77         size += sizeof(__be32); /* segment count */
78         size += rpcrdma_segment_maxsz * sizeof(__be32);
79         size += sizeof(__be32); /* list discriminator */
80
81         return size;
82 }
83
84 /* Returns size of largest RPC-over-RDMA header in a Reply message
85  *
86  * There is only one Write list or one Reply chunk per Reply
87  * message.  The larger list is the Write list.
88  */
89 static unsigned int rpcrdma_max_reply_header_size(unsigned int maxsegs)
90 {
91         unsigned int size;
92
93         /* Fixed header fields and list discriminators */
94         size = RPCRDMA_HDRLEN_MIN;
95
96         /* Maximum Write list size */
97         size += sizeof(__be32);         /* segment count */
98         size += maxsegs * rpcrdma_segment_maxsz * sizeof(__be32);
99         size += sizeof(__be32); /* list discriminator */
100
101         return size;
102 }
103
104 /**
105  * rpcrdma_set_max_header_sizes - Initialize inline payload sizes
106  * @ep: endpoint to initialize
107  *
108  * The max_inline fields contain the maximum size of an RPC message
109  * so the marshaling code doesn't have to repeat this calculation
110  * for every RPC.
111  */
112 void rpcrdma_set_max_header_sizes(struct rpcrdma_ep *ep)
113 {
114         unsigned int maxsegs = ep->re_max_rdma_segs;
115
116         ep->re_max_inline_send =
117                 ep->re_inline_send - rpcrdma_max_call_header_size(maxsegs);
118         ep->re_max_inline_recv =
119                 ep->re_inline_recv - rpcrdma_max_reply_header_size(maxsegs);
120 }
121
122 /* The client can send a request inline as long as the RPCRDMA header
123  * plus the RPC call fit under the transport's inline limit. If the
124  * combined call message size exceeds that limit, the client must use
125  * a Read chunk for this operation.
126  *
127  * A Read chunk is also required if sending the RPC call inline would
128  * exceed this device's max_sge limit.
129  */
130 static bool rpcrdma_args_inline(struct rpcrdma_xprt *r_xprt,
131                                 struct rpc_rqst *rqst)
132 {
133         struct xdr_buf *xdr = &rqst->rq_snd_buf;
134         struct rpcrdma_ep *ep = r_xprt->rx_ep;
135         unsigned int count, remaining, offset;
136
137         if (xdr->len > ep->re_max_inline_send)
138                 return false;
139
140         if (xdr->page_len) {
141                 remaining = xdr->page_len;
142                 offset = offset_in_page(xdr->page_base);
143                 count = RPCRDMA_MIN_SEND_SGES;
144                 while (remaining) {
145                         remaining -= min_t(unsigned int,
146                                            PAGE_SIZE - offset, remaining);
147                         offset = 0;
148                         if (++count > ep->re_attr.cap.max_send_sge)
149                                 return false;
150                 }
151         }
152
153         return true;
154 }
155
156 /* The client can't know how large the actual reply will be. Thus it
157  * plans for the largest possible reply for that particular ULP
158  * operation. If the maximum combined reply message size exceeds that
159  * limit, the client must provide a write list or a reply chunk for
160  * this request.
161  */
162 static bool rpcrdma_results_inline(struct rpcrdma_xprt *r_xprt,
163                                    struct rpc_rqst *rqst)
164 {
165         return rqst->rq_rcv_buf.buflen <= r_xprt->rx_ep->re_max_inline_recv;
166 }
167
168 /* The client is required to provide a Reply chunk if the maximum
169  * size of the non-payload part of the RPC Reply is larger than
170  * the inline threshold.
171  */
172 static bool
173 rpcrdma_nonpayload_inline(const struct rpcrdma_xprt *r_xprt,
174                           const struct rpc_rqst *rqst)
175 {
176         const struct xdr_buf *buf = &rqst->rq_rcv_buf;
177
178         return (buf->head[0].iov_len + buf->tail[0].iov_len) <
179                 r_xprt->rx_ep->re_max_inline_recv;
180 }
181
182 /* ACL likes to be lazy in allocating pages. For TCP, these
183  * pages can be allocated during receive processing. Not true
184  * for RDMA, which must always provision receive buffers
185  * up front.
186  */
187 static noinline int
188 rpcrdma_alloc_sparse_pages(struct xdr_buf *buf)
189 {
190         struct page **ppages;
191         int len;
192
193         len = buf->page_len;
194         ppages = buf->pages + (buf->page_base >> PAGE_SHIFT);
195         while (len > 0) {
196                 if (!*ppages)
197                         *ppages = alloc_page(GFP_NOWAIT | __GFP_NOWARN);
198                 if (!*ppages)
199                         return -ENOBUFS;
200                 ppages++;
201                 len -= PAGE_SIZE;
202         }
203
204         return 0;
205 }
206
207 /* Convert @vec to a single SGL element.
208  *
209  * Returns pointer to next available SGE, and bumps the total number
210  * of SGEs consumed.
211  */
212 static struct rpcrdma_mr_seg *
213 rpcrdma_convert_kvec(struct kvec *vec, struct rpcrdma_mr_seg *seg,
214                      unsigned int *n)
215 {
216         seg->mr_page = virt_to_page(vec->iov_base);
217         seg->mr_offset = vec->iov_base;
218         seg->mr_len = vec->iov_len;
219         ++seg;
220         ++(*n);
221         return seg;
222 }
223
224 /* Convert @xdrbuf into SGEs no larger than a page each. As they
225  * are registered, these SGEs are then coalesced into RDMA segments
226  * when the selected memreg mode supports it.
227  *
228  * Returns positive number of SGEs consumed, or a negative errno.
229  */
230
231 static int
232 rpcrdma_convert_iovs(struct rpcrdma_xprt *r_xprt, struct xdr_buf *xdrbuf,
233                      unsigned int pos, enum rpcrdma_chunktype type,
234                      struct rpcrdma_mr_seg *seg)
235 {
236         unsigned long page_base;
237         unsigned int len, n;
238         struct page **ppages;
239
240         n = 0;
241         if (pos == 0)
242                 seg = rpcrdma_convert_kvec(&xdrbuf->head[0], seg, &n);
243
244         len = xdrbuf->page_len;
245         ppages = xdrbuf->pages + (xdrbuf->page_base >> PAGE_SHIFT);
246         page_base = offset_in_page(xdrbuf->page_base);
247         while (len) {
248                 seg->mr_page = *ppages;
249                 seg->mr_offset = (char *)page_base;
250                 seg->mr_len = min_t(u32, PAGE_SIZE - page_base, len);
251                 len -= seg->mr_len;
252                 ++ppages;
253                 ++seg;
254                 ++n;
255                 page_base = 0;
256         }
257
258         /* When encoding a Read chunk, the tail iovec contains an
259          * XDR pad and may be omitted.
260          */
261         if (type == rpcrdma_readch && r_xprt->rx_ep->re_implicit_roundup)
262                 goto out;
263
264         /* When encoding a Write chunk, some servers need to see an
265          * extra segment for non-XDR-aligned Write chunks. The upper
266          * layer provides space in the tail iovec that may be used
267          * for this purpose.
268          */
269         if (type == rpcrdma_writech && r_xprt->rx_ep->re_implicit_roundup)
270                 goto out;
271
272         if (xdrbuf->tail[0].iov_len)
273                 rpcrdma_convert_kvec(&xdrbuf->tail[0], seg, &n);
274
275 out:
276         if (unlikely(n > RPCRDMA_MAX_SEGS))
277                 return -EIO;
278         return n;
279 }
280
281 static int
282 encode_rdma_segment(struct xdr_stream *xdr, struct rpcrdma_mr *mr)
283 {
284         __be32 *p;
285
286         p = xdr_reserve_space(xdr, 4 * sizeof(*p));
287         if (unlikely(!p))
288                 return -EMSGSIZE;
289
290         xdr_encode_rdma_segment(p, mr->mr_handle, mr->mr_length, mr->mr_offset);
291         return 0;
292 }
293
294 static int
295 encode_read_segment(struct xdr_stream *xdr, struct rpcrdma_mr *mr,
296                     u32 position)
297 {
298         __be32 *p;
299
300         p = xdr_reserve_space(xdr, 6 * sizeof(*p));
301         if (unlikely(!p))
302                 return -EMSGSIZE;
303
304         *p++ = xdr_one;                 /* Item present */
305         xdr_encode_read_segment(p, position, mr->mr_handle, mr->mr_length,
306                                 mr->mr_offset);
307         return 0;
308 }
309
310 static struct rpcrdma_mr_seg *rpcrdma_mr_prepare(struct rpcrdma_xprt *r_xprt,
311                                                  struct rpcrdma_req *req,
312                                                  struct rpcrdma_mr_seg *seg,
313                                                  int nsegs, bool writing,
314                                                  struct rpcrdma_mr **mr)
315 {
316         *mr = rpcrdma_mr_pop(&req->rl_free_mrs);
317         if (!*mr) {
318                 *mr = rpcrdma_mr_get(r_xprt);
319                 if (!*mr)
320                         goto out_getmr_err;
321                 (*mr)->mr_req = req;
322         }
323
324         rpcrdma_mr_push(*mr, &req->rl_registered);
325         return frwr_map(r_xprt, seg, nsegs, writing, req->rl_slot.rq_xid, *mr);
326
327 out_getmr_err:
328         trace_xprtrdma_nomrs_err(r_xprt, req);
329         xprt_wait_for_buffer_space(&r_xprt->rx_xprt);
330         rpcrdma_mrs_refresh(r_xprt);
331         return ERR_PTR(-EAGAIN);
332 }
333
334 /* Register and XDR encode the Read list. Supports encoding a list of read
335  * segments that belong to a single read chunk.
336  *
337  * Encoding key for single-list chunks (HLOO = Handle32 Length32 Offset64):
338  *
339  *  Read chunklist (a linked list):
340  *   N elements, position P (same P for all chunks of same arg!):
341  *    1 - PHLOO - 1 - PHLOO - ... - 1 - PHLOO - 0
342  *
343  * Returns zero on success, or a negative errno if a failure occurred.
344  * @xdr is advanced to the next position in the stream.
345  *
346  * Only a single @pos value is currently supported.
347  */
348 static int rpcrdma_encode_read_list(struct rpcrdma_xprt *r_xprt,
349                                     struct rpcrdma_req *req,
350                                     struct rpc_rqst *rqst,
351                                     enum rpcrdma_chunktype rtype)
352 {
353         struct xdr_stream *xdr = &req->rl_stream;
354         struct rpcrdma_mr_seg *seg;
355         struct rpcrdma_mr *mr;
356         unsigned int pos;
357         int nsegs;
358
359         if (rtype == rpcrdma_noch_pullup || rtype == rpcrdma_noch_mapped)
360                 goto done;
361
362         pos = rqst->rq_snd_buf.head[0].iov_len;
363         if (rtype == rpcrdma_areadch)
364                 pos = 0;
365         seg = req->rl_segments;
366         nsegs = rpcrdma_convert_iovs(r_xprt, &rqst->rq_snd_buf, pos,
367                                      rtype, seg);
368         if (nsegs < 0)
369                 return nsegs;
370
371         do {
372                 seg = rpcrdma_mr_prepare(r_xprt, req, seg, nsegs, false, &mr);
373                 if (IS_ERR(seg))
374                         return PTR_ERR(seg);
375
376                 if (encode_read_segment(xdr, mr, pos) < 0)
377                         return -EMSGSIZE;
378
379                 trace_xprtrdma_chunk_read(rqst->rq_task, pos, mr, nsegs);
380                 r_xprt->rx_stats.read_chunk_count++;
381                 nsegs -= mr->mr_nents;
382         } while (nsegs);
383
384 done:
385         if (xdr_stream_encode_item_absent(xdr) < 0)
386                 return -EMSGSIZE;
387         return 0;
388 }
389
390 /* Register and XDR encode the Write list. Supports encoding a list
391  * containing one array of plain segments that belong to a single
392  * write chunk.
393  *
394  * Encoding key for single-list chunks (HLOO = Handle32 Length32 Offset64):
395  *
396  *  Write chunklist (a list of (one) counted array):
397  *   N elements:
398  *    1 - N - HLOO - HLOO - ... - HLOO - 0
399  *
400  * Returns zero on success, or a negative errno if a failure occurred.
401  * @xdr is advanced to the next position in the stream.
402  *
403  * Only a single Write chunk is currently supported.
404  */
405 static int rpcrdma_encode_write_list(struct rpcrdma_xprt *r_xprt,
406                                      struct rpcrdma_req *req,
407                                      struct rpc_rqst *rqst,
408                                      enum rpcrdma_chunktype wtype)
409 {
410         struct xdr_stream *xdr = &req->rl_stream;
411         struct rpcrdma_mr_seg *seg;
412         struct rpcrdma_mr *mr;
413         int nsegs, nchunks;
414         __be32 *segcount;
415
416         if (wtype != rpcrdma_writech)
417                 goto done;
418
419         seg = req->rl_segments;
420         nsegs = rpcrdma_convert_iovs(r_xprt, &rqst->rq_rcv_buf,
421                                      rqst->rq_rcv_buf.head[0].iov_len,
422                                      wtype, seg);
423         if (nsegs < 0)
424                 return nsegs;
425
426         if (xdr_stream_encode_item_present(xdr) < 0)
427                 return -EMSGSIZE;
428         segcount = xdr_reserve_space(xdr, sizeof(*segcount));
429         if (unlikely(!segcount))
430                 return -EMSGSIZE;
431         /* Actual value encoded below */
432
433         nchunks = 0;
434         do {
435                 seg = rpcrdma_mr_prepare(r_xprt, req, seg, nsegs, true, &mr);
436                 if (IS_ERR(seg))
437                         return PTR_ERR(seg);
438
439                 if (encode_rdma_segment(xdr, mr) < 0)
440                         return -EMSGSIZE;
441
442                 trace_xprtrdma_chunk_write(rqst->rq_task, mr, nsegs);
443                 r_xprt->rx_stats.write_chunk_count++;
444                 r_xprt->rx_stats.total_rdma_request += mr->mr_length;
445                 nchunks++;
446                 nsegs -= mr->mr_nents;
447         } while (nsegs);
448
449         /* Update count of segments in this Write chunk */
450         *segcount = cpu_to_be32(nchunks);
451
452 done:
453         if (xdr_stream_encode_item_absent(xdr) < 0)
454                 return -EMSGSIZE;
455         return 0;
456 }
457
458 /* Register and XDR encode the Reply chunk. Supports encoding an array
459  * of plain segments that belong to a single write (reply) chunk.
460  *
461  * Encoding key for single-list chunks (HLOO = Handle32 Length32 Offset64):
462  *
463  *  Reply chunk (a counted array):
464  *   N elements:
465  *    1 - N - HLOO - HLOO - ... - HLOO
466  *
467  * Returns zero on success, or a negative errno if a failure occurred.
468  * @xdr is advanced to the next position in the stream.
469  */
470 static int rpcrdma_encode_reply_chunk(struct rpcrdma_xprt *r_xprt,
471                                       struct rpcrdma_req *req,
472                                       struct rpc_rqst *rqst,
473                                       enum rpcrdma_chunktype wtype)
474 {
475         struct xdr_stream *xdr = &req->rl_stream;
476         struct rpcrdma_mr_seg *seg;
477         struct rpcrdma_mr *mr;
478         int nsegs, nchunks;
479         __be32 *segcount;
480
481         if (wtype != rpcrdma_replych) {
482                 if (xdr_stream_encode_item_absent(xdr) < 0)
483                         return -EMSGSIZE;
484                 return 0;
485         }
486
487         seg = req->rl_segments;
488         nsegs = rpcrdma_convert_iovs(r_xprt, &rqst->rq_rcv_buf, 0, wtype, seg);
489         if (nsegs < 0)
490                 return nsegs;
491
492         if (xdr_stream_encode_item_present(xdr) < 0)
493                 return -EMSGSIZE;
494         segcount = xdr_reserve_space(xdr, sizeof(*segcount));
495         if (unlikely(!segcount))
496                 return -EMSGSIZE;
497         /* Actual value encoded below */
498
499         nchunks = 0;
500         do {
501                 seg = rpcrdma_mr_prepare(r_xprt, req, seg, nsegs, true, &mr);
502                 if (IS_ERR(seg))
503                         return PTR_ERR(seg);
504
505                 if (encode_rdma_segment(xdr, mr) < 0)
506                         return -EMSGSIZE;
507
508                 trace_xprtrdma_chunk_reply(rqst->rq_task, mr, nsegs);
509                 r_xprt->rx_stats.reply_chunk_count++;
510                 r_xprt->rx_stats.total_rdma_request += mr->mr_length;
511                 nchunks++;
512                 nsegs -= mr->mr_nents;
513         } while (nsegs);
514
515         /* Update count of segments in the Reply chunk */
516         *segcount = cpu_to_be32(nchunks);
517
518         return 0;
519 }
520
521 static void rpcrdma_sendctx_done(struct kref *kref)
522 {
523         struct rpcrdma_req *req =
524                 container_of(kref, struct rpcrdma_req, rl_kref);
525         struct rpcrdma_rep *rep = req->rl_reply;
526
527         rpcrdma_complete_rqst(rep);
528         rep->rr_rxprt->rx_stats.reply_waits_for_send++;
529 }
530
531 /**
532  * rpcrdma_sendctx_unmap - DMA-unmap Send buffer
533  * @sc: sendctx containing SGEs to unmap
534  *
535  */
536 void rpcrdma_sendctx_unmap(struct rpcrdma_sendctx *sc)
537 {
538         struct rpcrdma_regbuf *rb = sc->sc_req->rl_sendbuf;
539         struct ib_sge *sge;
540
541         if (!sc->sc_unmap_count)
542                 return;
543
544         /* The first two SGEs contain the transport header and
545          * the inline buffer. These are always left mapped so
546          * they can be cheaply re-used.
547          */
548         for (sge = &sc->sc_sges[2]; sc->sc_unmap_count;
549              ++sge, --sc->sc_unmap_count)
550                 ib_dma_unmap_page(rdmab_device(rb), sge->addr, sge->length,
551                                   DMA_TO_DEVICE);
552
553         kref_put(&sc->sc_req->rl_kref, rpcrdma_sendctx_done);
554 }
555
556 /* Prepare an SGE for the RPC-over-RDMA transport header.
557  */
558 static void rpcrdma_prepare_hdr_sge(struct rpcrdma_xprt *r_xprt,
559                                     struct rpcrdma_req *req, u32 len)
560 {
561         struct rpcrdma_sendctx *sc = req->rl_sendctx;
562         struct rpcrdma_regbuf *rb = req->rl_rdmabuf;
563         struct ib_sge *sge = &sc->sc_sges[req->rl_wr.num_sge++];
564
565         sge->addr = rdmab_addr(rb);
566         sge->length = len;
567         sge->lkey = rdmab_lkey(rb);
568
569         ib_dma_sync_single_for_device(rdmab_device(rb), sge->addr, sge->length,
570                                       DMA_TO_DEVICE);
571 }
572
573 /* The head iovec is straightforward, as it is usually already
574  * DMA-mapped. Sync the content that has changed.
575  */
576 static bool rpcrdma_prepare_head_iov(struct rpcrdma_xprt *r_xprt,
577                                      struct rpcrdma_req *req, unsigned int len)
578 {
579         struct rpcrdma_sendctx *sc = req->rl_sendctx;
580         struct ib_sge *sge = &sc->sc_sges[req->rl_wr.num_sge++];
581         struct rpcrdma_regbuf *rb = req->rl_sendbuf;
582
583         if (!rpcrdma_regbuf_dma_map(r_xprt, rb))
584                 return false;
585
586         sge->addr = rdmab_addr(rb);
587         sge->length = len;
588         sge->lkey = rdmab_lkey(rb);
589
590         ib_dma_sync_single_for_device(rdmab_device(rb), sge->addr, sge->length,
591                                       DMA_TO_DEVICE);
592         return true;
593 }
594
595 /* If there is a page list present, DMA map and prepare an
596  * SGE for each page to be sent.
597  */
598 static bool rpcrdma_prepare_pagelist(struct rpcrdma_req *req,
599                                      struct xdr_buf *xdr)
600 {
601         struct rpcrdma_sendctx *sc = req->rl_sendctx;
602         struct rpcrdma_regbuf *rb = req->rl_sendbuf;
603         unsigned int page_base, len, remaining;
604         struct page **ppages;
605         struct ib_sge *sge;
606
607         ppages = xdr->pages + (xdr->page_base >> PAGE_SHIFT);
608         page_base = offset_in_page(xdr->page_base);
609         remaining = xdr->page_len;
610         while (remaining) {
611                 sge = &sc->sc_sges[req->rl_wr.num_sge++];
612                 len = min_t(unsigned int, PAGE_SIZE - page_base, remaining);
613                 sge->addr = ib_dma_map_page(rdmab_device(rb), *ppages,
614                                             page_base, len, DMA_TO_DEVICE);
615                 if (ib_dma_mapping_error(rdmab_device(rb), sge->addr))
616                         goto out_mapping_err;
617
618                 sge->length = len;
619                 sge->lkey = rdmab_lkey(rb);
620
621                 sc->sc_unmap_count++;
622                 ppages++;
623                 remaining -= len;
624                 page_base = 0;
625         }
626
627         return true;
628
629 out_mapping_err:
630         trace_xprtrdma_dma_maperr(sge->addr);
631         return false;
632 }
633
634 /* The tail iovec may include an XDR pad for the page list,
635  * as well as additional content, and may not reside in the
636  * same page as the head iovec.
637  */
638 static bool rpcrdma_prepare_tail_iov(struct rpcrdma_req *req,
639                                      struct xdr_buf *xdr,
640                                      unsigned int page_base, unsigned int len)
641 {
642         struct rpcrdma_sendctx *sc = req->rl_sendctx;
643         struct ib_sge *sge = &sc->sc_sges[req->rl_wr.num_sge++];
644         struct rpcrdma_regbuf *rb = req->rl_sendbuf;
645         struct page *page = virt_to_page(xdr->tail[0].iov_base);
646
647         sge->addr = ib_dma_map_page(rdmab_device(rb), page, page_base, len,
648                                     DMA_TO_DEVICE);
649         if (ib_dma_mapping_error(rdmab_device(rb), sge->addr))
650                 goto out_mapping_err;
651
652         sge->length = len;
653         sge->lkey = rdmab_lkey(rb);
654         ++sc->sc_unmap_count;
655         return true;
656
657 out_mapping_err:
658         trace_xprtrdma_dma_maperr(sge->addr);
659         return false;
660 }
661
662 /* Copy the tail to the end of the head buffer.
663  */
664 static void rpcrdma_pullup_tail_iov(struct rpcrdma_xprt *r_xprt,
665                                     struct rpcrdma_req *req,
666                                     struct xdr_buf *xdr)
667 {
668         unsigned char *dst;
669
670         dst = (unsigned char *)xdr->head[0].iov_base;
671         dst += xdr->head[0].iov_len + xdr->page_len;
672         memmove(dst, xdr->tail[0].iov_base, xdr->tail[0].iov_len);
673         r_xprt->rx_stats.pullup_copy_count += xdr->tail[0].iov_len;
674 }
675
676 /* Copy pagelist content into the head buffer.
677  */
678 static void rpcrdma_pullup_pagelist(struct rpcrdma_xprt *r_xprt,
679                                     struct rpcrdma_req *req,
680                                     struct xdr_buf *xdr)
681 {
682         unsigned int len, page_base, remaining;
683         struct page **ppages;
684         unsigned char *src, *dst;
685
686         dst = (unsigned char *)xdr->head[0].iov_base;
687         dst += xdr->head[0].iov_len;
688         ppages = xdr->pages + (xdr->page_base >> PAGE_SHIFT);
689         page_base = offset_in_page(xdr->page_base);
690         remaining = xdr->page_len;
691         while (remaining) {
692                 src = page_address(*ppages);
693                 src += page_base;
694                 len = min_t(unsigned int, PAGE_SIZE - page_base, remaining);
695                 memcpy(dst, src, len);
696                 r_xprt->rx_stats.pullup_copy_count += len;
697
698                 ppages++;
699                 dst += len;
700                 remaining -= len;
701                 page_base = 0;
702         }
703 }
704
705 /* Copy the contents of @xdr into @rl_sendbuf and DMA sync it.
706  * When the head, pagelist, and tail are small, a pull-up copy
707  * is considerably less costly than DMA mapping the components
708  * of @xdr.
709  *
710  * Assumptions:
711  *  - the caller has already verified that the total length
712  *    of the RPC Call body will fit into @rl_sendbuf.
713  */
714 static bool rpcrdma_prepare_noch_pullup(struct rpcrdma_xprt *r_xprt,
715                                         struct rpcrdma_req *req,
716                                         struct xdr_buf *xdr)
717 {
718         if (unlikely(xdr->tail[0].iov_len))
719                 rpcrdma_pullup_tail_iov(r_xprt, req, xdr);
720
721         if (unlikely(xdr->page_len))
722                 rpcrdma_pullup_pagelist(r_xprt, req, xdr);
723
724         /* The whole RPC message resides in the head iovec now */
725         return rpcrdma_prepare_head_iov(r_xprt, req, xdr->len);
726 }
727
728 static bool rpcrdma_prepare_noch_mapped(struct rpcrdma_xprt *r_xprt,
729                                         struct rpcrdma_req *req,
730                                         struct xdr_buf *xdr)
731 {
732         struct kvec *tail = &xdr->tail[0];
733
734         if (!rpcrdma_prepare_head_iov(r_xprt, req, xdr->head[0].iov_len))
735                 return false;
736         if (xdr->page_len)
737                 if (!rpcrdma_prepare_pagelist(req, xdr))
738                         return false;
739         if (tail->iov_len)
740                 if (!rpcrdma_prepare_tail_iov(req, xdr,
741                                               offset_in_page(tail->iov_base),
742                                               tail->iov_len))
743                         return false;
744
745         if (req->rl_sendctx->sc_unmap_count)
746                 kref_get(&req->rl_kref);
747         return true;
748 }
749
750 static bool rpcrdma_prepare_readch(struct rpcrdma_xprt *r_xprt,
751                                    struct rpcrdma_req *req,
752                                    struct xdr_buf *xdr)
753 {
754         if (!rpcrdma_prepare_head_iov(r_xprt, req, xdr->head[0].iov_len))
755                 return false;
756
757         /* If there is a Read chunk, the page list is being handled
758          * via explicit RDMA, and thus is skipped here.
759          */
760
761         /* Do not include the tail if it is only an XDR pad */
762         if (xdr->tail[0].iov_len > 3) {
763                 unsigned int page_base, len;
764
765                 /* If the content in the page list is an odd length,
766                  * xdr_write_pages() adds a pad at the beginning of
767                  * the tail iovec. Force the tail's non-pad content to
768                  * land at the next XDR position in the Send message.
769                  */
770                 page_base = offset_in_page(xdr->tail[0].iov_base);
771                 len = xdr->tail[0].iov_len;
772                 page_base += len & 3;
773                 len -= len & 3;
774                 if (!rpcrdma_prepare_tail_iov(req, xdr, page_base, len))
775                         return false;
776                 kref_get(&req->rl_kref);
777         }
778
779         return true;
780 }
781
782 /**
783  * rpcrdma_prepare_send_sges - Construct SGEs for a Send WR
784  * @r_xprt: controlling transport
785  * @req: context of RPC Call being marshalled
786  * @hdrlen: size of transport header, in bytes
787  * @xdr: xdr_buf containing RPC Call
788  * @rtype: chunk type being encoded
789  *
790  * Returns 0 on success; otherwise a negative errno is returned.
791  */
792 inline int rpcrdma_prepare_send_sges(struct rpcrdma_xprt *r_xprt,
793                                      struct rpcrdma_req *req, u32 hdrlen,
794                                      struct xdr_buf *xdr,
795                                      enum rpcrdma_chunktype rtype)
796 {
797         int ret;
798
799         ret = -EAGAIN;
800         req->rl_sendctx = rpcrdma_sendctx_get_locked(r_xprt);
801         if (!req->rl_sendctx)
802                 goto out_nosc;
803         req->rl_sendctx->sc_unmap_count = 0;
804         req->rl_sendctx->sc_req = req;
805         kref_init(&req->rl_kref);
806         req->rl_wr.wr_cqe = &req->rl_sendctx->sc_cqe;
807         req->rl_wr.sg_list = req->rl_sendctx->sc_sges;
808         req->rl_wr.num_sge = 0;
809         req->rl_wr.opcode = IB_WR_SEND;
810
811         rpcrdma_prepare_hdr_sge(r_xprt, req, hdrlen);
812
813         ret = -EIO;
814         switch (rtype) {
815         case rpcrdma_noch_pullup:
816                 if (!rpcrdma_prepare_noch_pullup(r_xprt, req, xdr))
817                         goto out_unmap;
818                 break;
819         case rpcrdma_noch_mapped:
820                 if (!rpcrdma_prepare_noch_mapped(r_xprt, req, xdr))
821                         goto out_unmap;
822                 break;
823         case rpcrdma_readch:
824                 if (!rpcrdma_prepare_readch(r_xprt, req, xdr))
825                         goto out_unmap;
826                 break;
827         case rpcrdma_areadch:
828                 break;
829         default:
830                 goto out_unmap;
831         }
832
833         return 0;
834
835 out_unmap:
836         rpcrdma_sendctx_unmap(req->rl_sendctx);
837 out_nosc:
838         trace_xprtrdma_prepsend_failed(&req->rl_slot, ret);
839         return ret;
840 }
841
842 /**
843  * rpcrdma_marshal_req - Marshal and send one RPC request
844  * @r_xprt: controlling transport
845  * @rqst: RPC request to be marshaled
846  *
847  * For the RPC in "rqst", this function:
848  *  - Chooses the transfer mode (eg., RDMA_MSG or RDMA_NOMSG)
849  *  - Registers Read, Write, and Reply chunks
850  *  - Constructs the transport header
851  *  - Posts a Send WR to send the transport header and request
852  *
853  * Returns:
854  *      %0 if the RPC was sent successfully,
855  *      %-ENOTCONN if the connection was lost,
856  *      %-EAGAIN if the caller should call again with the same arguments,
857  *      %-ENOBUFS if the caller should call again after a delay,
858  *      %-EMSGSIZE if the transport header is too small,
859  *      %-EIO if a permanent problem occurred while marshaling.
860  */
861 int
862 rpcrdma_marshal_req(struct rpcrdma_xprt *r_xprt, struct rpc_rqst *rqst)
863 {
864         struct rpcrdma_req *req = rpcr_to_rdmar(rqst);
865         struct xdr_stream *xdr = &req->rl_stream;
866         enum rpcrdma_chunktype rtype, wtype;
867         struct xdr_buf *buf = &rqst->rq_snd_buf;
868         bool ddp_allowed;
869         __be32 *p;
870         int ret;
871
872         if (unlikely(rqst->rq_rcv_buf.flags & XDRBUF_SPARSE_PAGES)) {
873                 ret = rpcrdma_alloc_sparse_pages(&rqst->rq_rcv_buf);
874                 if (ret)
875                         return ret;
876         }
877
878         rpcrdma_set_xdrlen(&req->rl_hdrbuf, 0);
879         xdr_init_encode(xdr, &req->rl_hdrbuf, rdmab_data(req->rl_rdmabuf),
880                         rqst);
881
882         /* Fixed header fields */
883         ret = -EMSGSIZE;
884         p = xdr_reserve_space(xdr, 4 * sizeof(*p));
885         if (!p)
886                 goto out_err;
887         *p++ = rqst->rq_xid;
888         *p++ = rpcrdma_version;
889         *p++ = r_xprt->rx_buf.rb_max_requests;
890
891         /* When the ULP employs a GSS flavor that guarantees integrity
892          * or privacy, direct data placement of individual data items
893          * is not allowed.
894          */
895         ddp_allowed = !test_bit(RPCAUTH_AUTH_DATATOUCH,
896                                 &rqst->rq_cred->cr_auth->au_flags);
897
898         /*
899          * Chunks needed for results?
900          *
901          * o If the expected result is under the inline threshold, all ops
902          *   return as inline.
903          * o Large read ops return data as write chunk(s), header as
904          *   inline.
905          * o Large non-read ops return as a single reply chunk.
906          */
907         if (rpcrdma_results_inline(r_xprt, rqst))
908                 wtype = rpcrdma_noch;
909         else if ((ddp_allowed && rqst->rq_rcv_buf.flags & XDRBUF_READ) &&
910                  rpcrdma_nonpayload_inline(r_xprt, rqst))
911                 wtype = rpcrdma_writech;
912         else
913                 wtype = rpcrdma_replych;
914
915         /*
916          * Chunks needed for arguments?
917          *
918          * o If the total request is under the inline threshold, all ops
919          *   are sent as inline.
920          * o Large write ops transmit data as read chunk(s), header as
921          *   inline.
922          * o Large non-write ops are sent with the entire message as a
923          *   single read chunk (protocol 0-position special case).
924          *
925          * This assumes that the upper layer does not present a request
926          * that both has a data payload, and whose non-data arguments
927          * by themselves are larger than the inline threshold.
928          */
929         if (rpcrdma_args_inline(r_xprt, rqst)) {
930                 *p++ = rdma_msg;
931                 rtype = buf->len < rdmab_length(req->rl_sendbuf) ?
932                         rpcrdma_noch_pullup : rpcrdma_noch_mapped;
933         } else if (ddp_allowed && buf->flags & XDRBUF_WRITE) {
934                 *p++ = rdma_msg;
935                 rtype = rpcrdma_readch;
936         } else {
937                 r_xprt->rx_stats.nomsg_call_count++;
938                 *p++ = rdma_nomsg;
939                 rtype = rpcrdma_areadch;
940         }
941
942         /* This implementation supports the following combinations
943          * of chunk lists in one RPC-over-RDMA Call message:
944          *
945          *   - Read list
946          *   - Write list
947          *   - Reply chunk
948          *   - Read list + Reply chunk
949          *
950          * It might not yet support the following combinations:
951          *
952          *   - Read list + Write list
953          *
954          * It does not support the following combinations:
955          *
956          *   - Write list + Reply chunk
957          *   - Read list + Write list + Reply chunk
958          *
959          * This implementation supports only a single chunk in each
960          * Read or Write list. Thus for example the client cannot
961          * send a Call message with a Position Zero Read chunk and a
962          * regular Read chunk at the same time.
963          */
964         ret = rpcrdma_encode_read_list(r_xprt, req, rqst, rtype);
965         if (ret)
966                 goto out_err;
967         ret = rpcrdma_encode_write_list(r_xprt, req, rqst, wtype);
968         if (ret)
969                 goto out_err;
970         ret = rpcrdma_encode_reply_chunk(r_xprt, req, rqst, wtype);
971         if (ret)
972                 goto out_err;
973
974         ret = rpcrdma_prepare_send_sges(r_xprt, req, req->rl_hdrbuf.len,
975                                         buf, rtype);
976         if (ret)
977                 goto out_err;
978
979         trace_xprtrdma_marshal(req, rtype, wtype);
980         return 0;
981
982 out_err:
983         trace_xprtrdma_marshal_failed(rqst, ret);
984         r_xprt->rx_stats.failed_marshal_count++;
985         frwr_reset(req);
986         return ret;
987 }
988
989 static void __rpcrdma_update_cwnd_locked(struct rpc_xprt *xprt,
990                                          struct rpcrdma_buffer *buf,
991                                          u32 grant)
992 {
993         buf->rb_credits = grant;
994         xprt->cwnd = grant << RPC_CWNDSHIFT;
995 }
996
997 static void rpcrdma_update_cwnd(struct rpcrdma_xprt *r_xprt, u32 grant)
998 {
999         struct rpc_xprt *xprt = &r_xprt->rx_xprt;
1000
1001         spin_lock(&xprt->transport_lock);
1002         __rpcrdma_update_cwnd_locked(xprt, &r_xprt->rx_buf, grant);
1003         spin_unlock(&xprt->transport_lock);
1004 }
1005
1006 /**
1007  * rpcrdma_reset_cwnd - Reset the xprt's congestion window
1008  * @r_xprt: controlling transport instance
1009  *
1010  * Prepare @r_xprt for the next connection by reinitializing
1011  * its credit grant to one (see RFC 8166, Section 3.3.3).
1012  */
1013 void rpcrdma_reset_cwnd(struct rpcrdma_xprt *r_xprt)
1014 {
1015         struct rpc_xprt *xprt = &r_xprt->rx_xprt;
1016
1017         spin_lock(&xprt->transport_lock);
1018         xprt->cong = 0;
1019         __rpcrdma_update_cwnd_locked(xprt, &r_xprt->rx_buf, 1);
1020         spin_unlock(&xprt->transport_lock);
1021 }
1022
1023 /**
1024  * rpcrdma_inline_fixup - Scatter inline received data into rqst's iovecs
1025  * @rqst: controlling RPC request
1026  * @srcp: points to RPC message payload in receive buffer
1027  * @copy_len: remaining length of receive buffer content
1028  * @pad: Write chunk pad bytes needed (zero for pure inline)
1029  *
1030  * The upper layer has set the maximum number of bytes it can
1031  * receive in each component of rq_rcv_buf. These values are set in
1032  * the head.iov_len, page_len, tail.iov_len, and buflen fields.
1033  *
1034  * Unlike the TCP equivalent (xdr_partial_copy_from_skb), in
1035  * many cases this function simply updates iov_base pointers in
1036  * rq_rcv_buf to point directly to the received reply data, to
1037  * avoid copying reply data.
1038  *
1039  * Returns the count of bytes which had to be memcopied.
1040  */
1041 static unsigned long
1042 rpcrdma_inline_fixup(struct rpc_rqst *rqst, char *srcp, int copy_len, int pad)
1043 {
1044         unsigned long fixup_copy_count;
1045         int i, npages, curlen;
1046         char *destp;
1047         struct page **ppages;
1048         int page_base;
1049
1050         /* The head iovec is redirected to the RPC reply message
1051          * in the receive buffer, to avoid a memcopy.
1052          */
1053         rqst->rq_rcv_buf.head[0].iov_base = srcp;
1054         rqst->rq_private_buf.head[0].iov_base = srcp;
1055
1056         /* The contents of the receive buffer that follow
1057          * head.iov_len bytes are copied into the page list.
1058          */
1059         curlen = rqst->rq_rcv_buf.head[0].iov_len;
1060         if (curlen > copy_len)
1061                 curlen = copy_len;
1062         srcp += curlen;
1063         copy_len -= curlen;
1064
1065         ppages = rqst->rq_rcv_buf.pages +
1066                 (rqst->rq_rcv_buf.page_base >> PAGE_SHIFT);
1067         page_base = offset_in_page(rqst->rq_rcv_buf.page_base);
1068         fixup_copy_count = 0;
1069         if (copy_len && rqst->rq_rcv_buf.page_len) {
1070                 int pagelist_len;
1071
1072                 pagelist_len = rqst->rq_rcv_buf.page_len;
1073                 if (pagelist_len > copy_len)
1074                         pagelist_len = copy_len;
1075                 npages = PAGE_ALIGN(page_base + pagelist_len) >> PAGE_SHIFT;
1076                 for (i = 0; i < npages; i++) {
1077                         curlen = PAGE_SIZE - page_base;
1078                         if (curlen > pagelist_len)
1079                                 curlen = pagelist_len;
1080
1081                         destp = kmap_atomic(ppages[i]);
1082                         memcpy(destp + page_base, srcp, curlen);
1083                         flush_dcache_page(ppages[i]);
1084                         kunmap_atomic(destp);
1085                         srcp += curlen;
1086                         copy_len -= curlen;
1087                         fixup_copy_count += curlen;
1088                         pagelist_len -= curlen;
1089                         if (!pagelist_len)
1090                                 break;
1091                         page_base = 0;
1092                 }
1093
1094                 /* Implicit padding for the last segment in a Write
1095                  * chunk is inserted inline at the front of the tail
1096                  * iovec. The upper layer ignores the content of
1097                  * the pad. Simply ensure inline content in the tail
1098                  * that follows the Write chunk is properly aligned.
1099                  */
1100                 if (pad)
1101                         srcp -= pad;
1102         }
1103
1104         /* The tail iovec is redirected to the remaining data
1105          * in the receive buffer, to avoid a memcopy.
1106          */
1107         if (copy_len || pad) {
1108                 rqst->rq_rcv_buf.tail[0].iov_base = srcp;
1109                 rqst->rq_private_buf.tail[0].iov_base = srcp;
1110         }
1111
1112         if (fixup_copy_count)
1113                 trace_xprtrdma_fixup(rqst, fixup_copy_count);
1114         return fixup_copy_count;
1115 }
1116
1117 /* By convention, backchannel calls arrive via rdma_msg type
1118  * messages, and never populate the chunk lists. This makes
1119  * the RPC/RDMA header small and fixed in size, so it is
1120  * straightforward to check the RPC header's direction field.
1121  */
1122 static bool
1123 rpcrdma_is_bcall(struct rpcrdma_xprt *r_xprt, struct rpcrdma_rep *rep)
1124 #if defined(CONFIG_SUNRPC_BACKCHANNEL)
1125 {
1126         struct xdr_stream *xdr = &rep->rr_stream;
1127         __be32 *p;
1128
1129         if (rep->rr_proc != rdma_msg)
1130                 return false;
1131
1132         /* Peek at stream contents without advancing. */
1133         p = xdr_inline_decode(xdr, 0);
1134
1135         /* Chunk lists */
1136         if (xdr_item_is_present(p++))
1137                 return false;
1138         if (xdr_item_is_present(p++))
1139                 return false;
1140         if (xdr_item_is_present(p++))
1141                 return false;
1142
1143         /* RPC header */
1144         if (*p++ != rep->rr_xid)
1145                 return false;
1146         if (*p != cpu_to_be32(RPC_CALL))
1147                 return false;
1148
1149         /* Now that we are sure this is a backchannel call,
1150          * advance to the RPC header.
1151          */
1152         p = xdr_inline_decode(xdr, 3 * sizeof(*p));
1153         if (unlikely(!p))
1154                 goto out_short;
1155
1156         rpcrdma_bc_receive_call(r_xprt, rep);
1157         return true;
1158
1159 out_short:
1160         pr_warn("RPC/RDMA short backward direction call\n");
1161         return true;
1162 }
1163 #else   /* CONFIG_SUNRPC_BACKCHANNEL */
1164 {
1165         return false;
1166 }
1167 #endif  /* CONFIG_SUNRPC_BACKCHANNEL */
1168
1169 static int decode_rdma_segment(struct xdr_stream *xdr, u32 *length)
1170 {
1171         u32 handle;
1172         u64 offset;
1173         __be32 *p;
1174
1175         p = xdr_inline_decode(xdr, 4 * sizeof(*p));
1176         if (unlikely(!p))
1177                 return -EIO;
1178
1179         xdr_decode_rdma_segment(p, &handle, length, &offset);
1180         trace_xprtrdma_decode_seg(handle, *length, offset);
1181         return 0;
1182 }
1183
1184 static int decode_write_chunk(struct xdr_stream *xdr, u32 *length)
1185 {
1186         u32 segcount, seglength;
1187         __be32 *p;
1188
1189         p = xdr_inline_decode(xdr, sizeof(*p));
1190         if (unlikely(!p))
1191                 return -EIO;
1192
1193         *length = 0;
1194         segcount = be32_to_cpup(p);
1195         while (segcount--) {
1196                 if (decode_rdma_segment(xdr, &seglength))
1197                         return -EIO;
1198                 *length += seglength;
1199         }
1200
1201         return 0;
1202 }
1203
1204 /* In RPC-over-RDMA Version One replies, a Read list is never
1205  * expected. This decoder is a stub that returns an error if
1206  * a Read list is present.
1207  */
1208 static int decode_read_list(struct xdr_stream *xdr)
1209 {
1210         __be32 *p;
1211
1212         p = xdr_inline_decode(xdr, sizeof(*p));
1213         if (unlikely(!p))
1214                 return -EIO;
1215         if (unlikely(xdr_item_is_present(p)))
1216                 return -EIO;
1217         return 0;
1218 }
1219
1220 /* Supports only one Write chunk in the Write list
1221  */
1222 static int decode_write_list(struct xdr_stream *xdr, u32 *length)
1223 {
1224         u32 chunklen;
1225         bool first;
1226         __be32 *p;
1227
1228         *length = 0;
1229         first = true;
1230         do {
1231                 p = xdr_inline_decode(xdr, sizeof(*p));
1232                 if (unlikely(!p))
1233                         return -EIO;
1234                 if (xdr_item_is_absent(p))
1235                         break;
1236                 if (!first)
1237                         return -EIO;
1238
1239                 if (decode_write_chunk(xdr, &chunklen))
1240                         return -EIO;
1241                 *length += chunklen;
1242                 first = false;
1243         } while (true);
1244         return 0;
1245 }
1246
1247 static int decode_reply_chunk(struct xdr_stream *xdr, u32 *length)
1248 {
1249         __be32 *p;
1250
1251         p = xdr_inline_decode(xdr, sizeof(*p));
1252         if (unlikely(!p))
1253                 return -EIO;
1254
1255         *length = 0;
1256         if (xdr_item_is_present(p))
1257                 if (decode_write_chunk(xdr, length))
1258                         return -EIO;
1259         return 0;
1260 }
1261
1262 static int
1263 rpcrdma_decode_msg(struct rpcrdma_xprt *r_xprt, struct rpcrdma_rep *rep,
1264                    struct rpc_rqst *rqst)
1265 {
1266         struct xdr_stream *xdr = &rep->rr_stream;
1267         u32 writelist, replychunk, rpclen;
1268         char *base;
1269
1270         /* Decode the chunk lists */
1271         if (decode_read_list(xdr))
1272                 return -EIO;
1273         if (decode_write_list(xdr, &writelist))
1274                 return -EIO;
1275         if (decode_reply_chunk(xdr, &replychunk))
1276                 return -EIO;
1277
1278         /* RDMA_MSG sanity checks */
1279         if (unlikely(replychunk))
1280                 return -EIO;
1281
1282         /* Build the RPC reply's Payload stream in rqst->rq_rcv_buf */
1283         base = (char *)xdr_inline_decode(xdr, 0);
1284         rpclen = xdr_stream_remaining(xdr);
1285         r_xprt->rx_stats.fixup_copy_count +=
1286                 rpcrdma_inline_fixup(rqst, base, rpclen, writelist & 3);
1287
1288         r_xprt->rx_stats.total_rdma_reply += writelist;
1289         return rpclen + xdr_align_size(writelist);
1290 }
1291
1292 static noinline int
1293 rpcrdma_decode_nomsg(struct rpcrdma_xprt *r_xprt, struct rpcrdma_rep *rep)
1294 {
1295         struct xdr_stream *xdr = &rep->rr_stream;
1296         u32 writelist, replychunk;
1297
1298         /* Decode the chunk lists */
1299         if (decode_read_list(xdr))
1300                 return -EIO;
1301         if (decode_write_list(xdr, &writelist))
1302                 return -EIO;
1303         if (decode_reply_chunk(xdr, &replychunk))
1304                 return -EIO;
1305
1306         /* RDMA_NOMSG sanity checks */
1307         if (unlikely(writelist))
1308                 return -EIO;
1309         if (unlikely(!replychunk))
1310                 return -EIO;
1311
1312         /* Reply chunk buffer already is the reply vector */
1313         r_xprt->rx_stats.total_rdma_reply += replychunk;
1314         return replychunk;
1315 }
1316
1317 static noinline int
1318 rpcrdma_decode_error(struct rpcrdma_xprt *r_xprt, struct rpcrdma_rep *rep,
1319                      struct rpc_rqst *rqst)
1320 {
1321         struct xdr_stream *xdr = &rep->rr_stream;
1322         __be32 *p;
1323
1324         p = xdr_inline_decode(xdr, sizeof(*p));
1325         if (unlikely(!p))
1326                 return -EIO;
1327
1328         switch (*p) {
1329         case err_vers:
1330                 p = xdr_inline_decode(xdr, 2 * sizeof(*p));
1331                 if (!p)
1332                         break;
1333                 trace_xprtrdma_err_vers(rqst, p, p + 1);
1334                 break;
1335         case err_chunk:
1336                 trace_xprtrdma_err_chunk(rqst);
1337                 break;
1338         default:
1339                 trace_xprtrdma_err_unrecognized(rqst, p);
1340         }
1341
1342         return -EIO;
1343 }
1344
1345 /* Perform XID lookup, reconstruction of the RPC reply, and
1346  * RPC completion while holding the transport lock to ensure
1347  * the rep, rqst, and rq_task pointers remain stable.
1348  */
1349 void rpcrdma_complete_rqst(struct rpcrdma_rep *rep)
1350 {
1351         struct rpcrdma_xprt *r_xprt = rep->rr_rxprt;
1352         struct rpc_xprt *xprt = &r_xprt->rx_xprt;
1353         struct rpc_rqst *rqst = rep->rr_rqst;
1354         int status;
1355
1356         switch (rep->rr_proc) {
1357         case rdma_msg:
1358                 status = rpcrdma_decode_msg(r_xprt, rep, rqst);
1359                 break;
1360         case rdma_nomsg:
1361                 status = rpcrdma_decode_nomsg(r_xprt, rep);
1362                 break;
1363         case rdma_error:
1364                 status = rpcrdma_decode_error(r_xprt, rep, rqst);
1365                 break;
1366         default:
1367                 status = -EIO;
1368         }
1369         if (status < 0)
1370                 goto out_badheader;
1371
1372 out:
1373         spin_lock(&xprt->queue_lock);
1374         xprt_complete_rqst(rqst->rq_task, status);
1375         xprt_unpin_rqst(rqst);
1376         spin_unlock(&xprt->queue_lock);
1377         return;
1378
1379 out_badheader:
1380         trace_xprtrdma_reply_hdr_err(rep);
1381         r_xprt->rx_stats.bad_reply_count++;
1382         rqst->rq_task->tk_status = status;
1383         status = 0;
1384         goto out;
1385 }
1386
1387 static void rpcrdma_reply_done(struct kref *kref)
1388 {
1389         struct rpcrdma_req *req =
1390                 container_of(kref, struct rpcrdma_req, rl_kref);
1391
1392         rpcrdma_complete_rqst(req->rl_reply);
1393 }
1394
1395 /**
1396  * rpcrdma_reply_handler - Process received RPC/RDMA messages
1397  * @rep: Incoming rpcrdma_rep object to process
1398  *
1399  * Errors must result in the RPC task either being awakened, or
1400  * allowed to timeout, to discover the errors at that time.
1401  */
1402 void rpcrdma_reply_handler(struct rpcrdma_rep *rep)
1403 {
1404         struct rpcrdma_xprt *r_xprt = rep->rr_rxprt;
1405         struct rpc_xprt *xprt = &r_xprt->rx_xprt;
1406         struct rpcrdma_buffer *buf = &r_xprt->rx_buf;
1407         struct rpcrdma_req *req;
1408         struct rpc_rqst *rqst;
1409         u32 credits;
1410         __be32 *p;
1411
1412         /* Any data means we had a useful conversation, so
1413          * then we don't need to delay the next reconnect.
1414          */
1415         if (xprt->reestablish_timeout)
1416                 xprt->reestablish_timeout = 0;
1417
1418         /* Fixed transport header fields */
1419         xdr_init_decode(&rep->rr_stream, &rep->rr_hdrbuf,
1420                         rep->rr_hdrbuf.head[0].iov_base, NULL);
1421         p = xdr_inline_decode(&rep->rr_stream, 4 * sizeof(*p));
1422         if (unlikely(!p))
1423                 goto out_shortreply;
1424         rep->rr_xid = *p++;
1425         rep->rr_vers = *p++;
1426         credits = be32_to_cpu(*p++);
1427         rep->rr_proc = *p++;
1428
1429         if (rep->rr_vers != rpcrdma_version)
1430                 goto out_badversion;
1431
1432         if (rpcrdma_is_bcall(r_xprt, rep))
1433                 return;
1434
1435         /* Match incoming rpcrdma_rep to an rpcrdma_req to
1436          * get context for handling any incoming chunks.
1437          */
1438         spin_lock(&xprt->queue_lock);
1439         rqst = xprt_lookup_rqst(xprt, rep->rr_xid);
1440         if (!rqst)
1441                 goto out_norqst;
1442         xprt_pin_rqst(rqst);
1443         spin_unlock(&xprt->queue_lock);
1444
1445         if (credits == 0)
1446                 credits = 1;    /* don't deadlock */
1447         else if (credits > r_xprt->rx_ep->re_max_requests)
1448                 credits = r_xprt->rx_ep->re_max_requests;
1449         if (buf->rb_credits != credits)
1450                 rpcrdma_update_cwnd(r_xprt, credits);
1451         rpcrdma_post_recvs(r_xprt, false);
1452
1453         req = rpcr_to_rdmar(rqst);
1454         if (unlikely(req->rl_reply))
1455                 rpcrdma_recv_buffer_put(req->rl_reply);
1456         req->rl_reply = rep;
1457         rep->rr_rqst = rqst;
1458
1459         trace_xprtrdma_reply(rqst->rq_task, rep, credits);
1460
1461         if (rep->rr_wc_flags & IB_WC_WITH_INVALIDATE)
1462                 frwr_reminv(rep, &req->rl_registered);
1463         if (!list_empty(&req->rl_registered))
1464                 frwr_unmap_async(r_xprt, req);
1465                 /* LocalInv completion will complete the RPC */
1466         else
1467                 kref_put(&req->rl_kref, rpcrdma_reply_done);
1468         return;
1469
1470 out_badversion:
1471         trace_xprtrdma_reply_vers_err(rep);
1472         goto out;
1473
1474 out_norqst:
1475         spin_unlock(&xprt->queue_lock);
1476         trace_xprtrdma_reply_rqst_err(rep);
1477         goto out;
1478
1479 out_shortreply:
1480         trace_xprtrdma_reply_short_err(rep);
1481
1482 out:
1483         rpcrdma_recv_buffer_put(rep);
1484 }