]>
Commit | Line | Data |
---|---|---|
1 | // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause | |
2 | /* | |
3 | * Copyright (c) 2014-2017 Oracle. All rights reserved. | |
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 | * transport.c | |
44 | * | |
45 | * This file contains the top-level implementation of an RPC RDMA | |
46 | * transport. | |
47 | * | |
48 | * Naming convention: functions beginning with xprt_ are part of the | |
49 | * transport switch. All others are RPC RDMA internal. | |
50 | */ | |
51 | ||
52 | #include <linux/module.h> | |
53 | #include <linux/slab.h> | |
54 | #include <linux/seq_file.h> | |
55 | #include <linux/smp.h> | |
56 | ||
57 | #include <linux/sunrpc/addr.h> | |
58 | #include <linux/sunrpc/svc_rdma.h> | |
59 | ||
60 | #include "xprt_rdma.h" | |
61 | #include <trace/events/rpcrdma.h> | |
62 | ||
63 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) | |
64 | # define RPCDBG_FACILITY RPCDBG_TRANS | |
65 | #endif | |
66 | ||
67 | /* | |
68 | * tunables | |
69 | */ | |
70 | ||
71 | static unsigned int xprt_rdma_slot_table_entries = RPCRDMA_DEF_SLOT_TABLE; | |
72 | unsigned int xprt_rdma_max_inline_read = RPCRDMA_DEF_INLINE; | |
73 | unsigned int xprt_rdma_max_inline_write = RPCRDMA_DEF_INLINE; | |
74 | unsigned int xprt_rdma_memreg_strategy = RPCRDMA_FRWR; | |
75 | int xprt_rdma_pad_optimize; | |
76 | static struct xprt_class xprt_rdma; | |
77 | ||
78 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) | |
79 | ||
80 | static unsigned int min_slot_table_size = RPCRDMA_MIN_SLOT_TABLE; | |
81 | static unsigned int max_slot_table_size = RPCRDMA_MAX_SLOT_TABLE; | |
82 | static unsigned int min_inline_size = RPCRDMA_MIN_INLINE; | |
83 | static unsigned int max_inline_size = RPCRDMA_MAX_INLINE; | |
84 | static unsigned int max_padding = PAGE_SIZE; | |
85 | static unsigned int min_memreg = RPCRDMA_BOUNCEBUFFERS; | |
86 | static unsigned int max_memreg = RPCRDMA_LAST - 1; | |
87 | static unsigned int dummy; | |
88 | ||
89 | static struct ctl_table_header *sunrpc_table_header; | |
90 | ||
91 | static struct ctl_table xr_tunables_table[] = { | |
92 | { | |
93 | .procname = "rdma_slot_table_entries", | |
94 | .data = &xprt_rdma_slot_table_entries, | |
95 | .maxlen = sizeof(unsigned int), | |
96 | .mode = 0644, | |
97 | .proc_handler = proc_dointvec_minmax, | |
98 | .extra1 = &min_slot_table_size, | |
99 | .extra2 = &max_slot_table_size | |
100 | }, | |
101 | { | |
102 | .procname = "rdma_max_inline_read", | |
103 | .data = &xprt_rdma_max_inline_read, | |
104 | .maxlen = sizeof(unsigned int), | |
105 | .mode = 0644, | |
106 | .proc_handler = proc_dointvec_minmax, | |
107 | .extra1 = &min_inline_size, | |
108 | .extra2 = &max_inline_size, | |
109 | }, | |
110 | { | |
111 | .procname = "rdma_max_inline_write", | |
112 | .data = &xprt_rdma_max_inline_write, | |
113 | .maxlen = sizeof(unsigned int), | |
114 | .mode = 0644, | |
115 | .proc_handler = proc_dointvec_minmax, | |
116 | .extra1 = &min_inline_size, | |
117 | .extra2 = &max_inline_size, | |
118 | }, | |
119 | { | |
120 | .procname = "rdma_inline_write_padding", | |
121 | .data = &dummy, | |
122 | .maxlen = sizeof(unsigned int), | |
123 | .mode = 0644, | |
124 | .proc_handler = proc_dointvec_minmax, | |
125 | .extra1 = SYSCTL_ZERO, | |
126 | .extra2 = &max_padding, | |
127 | }, | |
128 | { | |
129 | .procname = "rdma_memreg_strategy", | |
130 | .data = &xprt_rdma_memreg_strategy, | |
131 | .maxlen = sizeof(unsigned int), | |
132 | .mode = 0644, | |
133 | .proc_handler = proc_dointvec_minmax, | |
134 | .extra1 = &min_memreg, | |
135 | .extra2 = &max_memreg, | |
136 | }, | |
137 | { | |
138 | .procname = "rdma_pad_optimize", | |
139 | .data = &xprt_rdma_pad_optimize, | |
140 | .maxlen = sizeof(unsigned int), | |
141 | .mode = 0644, | |
142 | .proc_handler = proc_dointvec, | |
143 | }, | |
144 | { }, | |
145 | }; | |
146 | ||
147 | static struct ctl_table sunrpc_table[] = { | |
148 | { | |
149 | .procname = "sunrpc", | |
150 | .mode = 0555, | |
151 | .child = xr_tunables_table | |
152 | }, | |
153 | { }, | |
154 | }; | |
155 | ||
156 | #endif | |
157 | ||
158 | static const struct rpc_xprt_ops xprt_rdma_procs; | |
159 | ||
160 | static void | |
161 | xprt_rdma_format_addresses4(struct rpc_xprt *xprt, struct sockaddr *sap) | |
162 | { | |
163 | struct sockaddr_in *sin = (struct sockaddr_in *)sap; | |
164 | char buf[20]; | |
165 | ||
166 | snprintf(buf, sizeof(buf), "%08x", ntohl(sin->sin_addr.s_addr)); | |
167 | xprt->address_strings[RPC_DISPLAY_HEX_ADDR] = kstrdup(buf, GFP_KERNEL); | |
168 | ||
169 | xprt->address_strings[RPC_DISPLAY_NETID] = RPCBIND_NETID_RDMA; | |
170 | } | |
171 | ||
172 | static void | |
173 | xprt_rdma_format_addresses6(struct rpc_xprt *xprt, struct sockaddr *sap) | |
174 | { | |
175 | struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sap; | |
176 | char buf[40]; | |
177 | ||
178 | snprintf(buf, sizeof(buf), "%pi6", &sin6->sin6_addr); | |
179 | xprt->address_strings[RPC_DISPLAY_HEX_ADDR] = kstrdup(buf, GFP_KERNEL); | |
180 | ||
181 | xprt->address_strings[RPC_DISPLAY_NETID] = RPCBIND_NETID_RDMA6; | |
182 | } | |
183 | ||
184 | void | |
185 | xprt_rdma_format_addresses(struct rpc_xprt *xprt, struct sockaddr *sap) | |
186 | { | |
187 | char buf[128]; | |
188 | ||
189 | switch (sap->sa_family) { | |
190 | case AF_INET: | |
191 | xprt_rdma_format_addresses4(xprt, sap); | |
192 | break; | |
193 | case AF_INET6: | |
194 | xprt_rdma_format_addresses6(xprt, sap); | |
195 | break; | |
196 | default: | |
197 | pr_err("rpcrdma: Unrecognized address family\n"); | |
198 | return; | |
199 | } | |
200 | ||
201 | (void)rpc_ntop(sap, buf, sizeof(buf)); | |
202 | xprt->address_strings[RPC_DISPLAY_ADDR] = kstrdup(buf, GFP_KERNEL); | |
203 | ||
204 | snprintf(buf, sizeof(buf), "%u", rpc_get_port(sap)); | |
205 | xprt->address_strings[RPC_DISPLAY_PORT] = kstrdup(buf, GFP_KERNEL); | |
206 | ||
207 | snprintf(buf, sizeof(buf), "%4hx", rpc_get_port(sap)); | |
208 | xprt->address_strings[RPC_DISPLAY_HEX_PORT] = kstrdup(buf, GFP_KERNEL); | |
209 | ||
210 | xprt->address_strings[RPC_DISPLAY_PROTO] = "rdma"; | |
211 | } | |
212 | ||
213 | void | |
214 | xprt_rdma_free_addresses(struct rpc_xprt *xprt) | |
215 | { | |
216 | unsigned int i; | |
217 | ||
218 | for (i = 0; i < RPC_DISPLAY_MAX; i++) | |
219 | switch (i) { | |
220 | case RPC_DISPLAY_PROTO: | |
221 | case RPC_DISPLAY_NETID: | |
222 | continue; | |
223 | default: | |
224 | kfree(xprt->address_strings[i]); | |
225 | } | |
226 | } | |
227 | ||
228 | /** | |
229 | * xprt_rdma_connect_worker - establish connection in the background | |
230 | * @work: worker thread context | |
231 | * | |
232 | * Requester holds the xprt's send lock to prevent activity on this | |
233 | * transport while a fresh connection is being established. RPC tasks | |
234 | * sleep on the xprt's pending queue waiting for connect to complete. | |
235 | */ | |
236 | static void | |
237 | xprt_rdma_connect_worker(struct work_struct *work) | |
238 | { | |
239 | struct rpcrdma_xprt *r_xprt = container_of(work, struct rpcrdma_xprt, | |
240 | rx_connect_worker.work); | |
241 | struct rpc_xprt *xprt = &r_xprt->rx_xprt; | |
242 | int rc; | |
243 | ||
244 | rc = rpcrdma_xprt_connect(r_xprt); | |
245 | xprt_clear_connecting(xprt); | |
246 | if (!rc) { | |
247 | xprt->connect_cookie++; | |
248 | xprt->stat.connect_count++; | |
249 | xprt->stat.connect_time += (long)jiffies - | |
250 | xprt->stat.connect_start; | |
251 | xprt_set_connected(xprt); | |
252 | rc = -EAGAIN; | |
253 | } else | |
254 | rpcrdma_xprt_disconnect(r_xprt); | |
255 | xprt_unlock_connect(xprt, r_xprt); | |
256 | xprt_wake_pending_tasks(xprt, rc); | |
257 | } | |
258 | ||
259 | /** | |
260 | * xprt_rdma_inject_disconnect - inject a connection fault | |
261 | * @xprt: transport context | |
262 | * | |
263 | * If @xprt is connected, disconnect it to simulate spurious | |
264 | * connection loss. Caller must hold @xprt's send lock to | |
265 | * ensure that data structures and hardware resources are | |
266 | * stable during the rdma_disconnect() call. | |
267 | */ | |
268 | static void | |
269 | xprt_rdma_inject_disconnect(struct rpc_xprt *xprt) | |
270 | { | |
271 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
272 | ||
273 | trace_xprtrdma_op_inject_dsc(r_xprt); | |
274 | rdma_disconnect(r_xprt->rx_ep->re_id); | |
275 | } | |
276 | ||
277 | /** | |
278 | * xprt_rdma_destroy - Full tear down of transport | |
279 | * @xprt: doomed transport context | |
280 | * | |
281 | * Caller guarantees there will be no more calls to us with | |
282 | * this @xprt. | |
283 | */ | |
284 | static void | |
285 | xprt_rdma_destroy(struct rpc_xprt *xprt) | |
286 | { | |
287 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
288 | ||
289 | cancel_delayed_work_sync(&r_xprt->rx_connect_worker); | |
290 | ||
291 | rpcrdma_xprt_disconnect(r_xprt); | |
292 | rpcrdma_buffer_destroy(&r_xprt->rx_buf); | |
293 | ||
294 | xprt_rdma_free_addresses(xprt); | |
295 | xprt_free(xprt); | |
296 | ||
297 | module_put(THIS_MODULE); | |
298 | } | |
299 | ||
300 | /* 60 second timeout, no retries */ | |
301 | static const struct rpc_timeout xprt_rdma_default_timeout = { | |
302 | .to_initval = 60 * HZ, | |
303 | .to_maxval = 60 * HZ, | |
304 | }; | |
305 | ||
306 | /** | |
307 | * xprt_setup_rdma - Set up transport to use RDMA | |
308 | * | |
309 | * @args: rpc transport arguments | |
310 | */ | |
311 | static struct rpc_xprt * | |
312 | xprt_setup_rdma(struct xprt_create *args) | |
313 | { | |
314 | struct rpc_xprt *xprt; | |
315 | struct rpcrdma_xprt *new_xprt; | |
316 | struct sockaddr *sap; | |
317 | int rc; | |
318 | ||
319 | if (args->addrlen > sizeof(xprt->addr)) | |
320 | return ERR_PTR(-EBADF); | |
321 | ||
322 | if (!try_module_get(THIS_MODULE)) | |
323 | return ERR_PTR(-EIO); | |
324 | ||
325 | xprt = xprt_alloc(args->net, sizeof(struct rpcrdma_xprt), 0, | |
326 | xprt_rdma_slot_table_entries); | |
327 | if (!xprt) { | |
328 | module_put(THIS_MODULE); | |
329 | return ERR_PTR(-ENOMEM); | |
330 | } | |
331 | ||
332 | xprt->timeout = &xprt_rdma_default_timeout; | |
333 | xprt->connect_timeout = xprt->timeout->to_initval; | |
334 | xprt->max_reconnect_timeout = xprt->timeout->to_maxval; | |
335 | xprt->bind_timeout = RPCRDMA_BIND_TO; | |
336 | xprt->reestablish_timeout = RPCRDMA_INIT_REEST_TO; | |
337 | xprt->idle_timeout = RPCRDMA_IDLE_DISC_TO; | |
338 | ||
339 | xprt->resvport = 0; /* privileged port not needed */ | |
340 | xprt->ops = &xprt_rdma_procs; | |
341 | ||
342 | /* | |
343 | * Set up RDMA-specific connect data. | |
344 | */ | |
345 | sap = args->dstaddr; | |
346 | ||
347 | /* Ensure xprt->addr holds valid server TCP (not RDMA) | |
348 | * address, for any side protocols which peek at it */ | |
349 | xprt->prot = IPPROTO_TCP; | |
350 | xprt->xprt_class = &xprt_rdma; | |
351 | xprt->addrlen = args->addrlen; | |
352 | memcpy(&xprt->addr, sap, xprt->addrlen); | |
353 | ||
354 | if (rpc_get_port(sap)) | |
355 | xprt_set_bound(xprt); | |
356 | xprt_rdma_format_addresses(xprt, sap); | |
357 | ||
358 | new_xprt = rpcx_to_rdmax(xprt); | |
359 | rc = rpcrdma_buffer_create(new_xprt); | |
360 | if (rc) { | |
361 | xprt_rdma_free_addresses(xprt); | |
362 | xprt_free(xprt); | |
363 | module_put(THIS_MODULE); | |
364 | return ERR_PTR(rc); | |
365 | } | |
366 | ||
367 | INIT_DELAYED_WORK(&new_xprt->rx_connect_worker, | |
368 | xprt_rdma_connect_worker); | |
369 | ||
370 | xprt->max_payload = RPCRDMA_MAX_DATA_SEGS << PAGE_SHIFT; | |
371 | ||
372 | return xprt; | |
373 | } | |
374 | ||
375 | /** | |
376 | * xprt_rdma_close - close a transport connection | |
377 | * @xprt: transport context | |
378 | * | |
379 | * Called during autoclose or device removal. | |
380 | * | |
381 | * Caller holds @xprt's send lock to prevent activity on this | |
382 | * transport while the connection is torn down. | |
383 | */ | |
384 | void xprt_rdma_close(struct rpc_xprt *xprt) | |
385 | { | |
386 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
387 | ||
388 | rpcrdma_xprt_disconnect(r_xprt); | |
389 | ||
390 | xprt->reestablish_timeout = 0; | |
391 | ++xprt->connect_cookie; | |
392 | xprt_disconnect_done(xprt); | |
393 | } | |
394 | ||
395 | /** | |
396 | * xprt_rdma_set_port - update server port with rpcbind result | |
397 | * @xprt: controlling RPC transport | |
398 | * @port: new port value | |
399 | * | |
400 | * Transport connect status is unchanged. | |
401 | */ | |
402 | static void | |
403 | xprt_rdma_set_port(struct rpc_xprt *xprt, u16 port) | |
404 | { | |
405 | struct sockaddr *sap = (struct sockaddr *)&xprt->addr; | |
406 | char buf[8]; | |
407 | ||
408 | rpc_set_port(sap, port); | |
409 | ||
410 | kfree(xprt->address_strings[RPC_DISPLAY_PORT]); | |
411 | snprintf(buf, sizeof(buf), "%u", port); | |
412 | xprt->address_strings[RPC_DISPLAY_PORT] = kstrdup(buf, GFP_KERNEL); | |
413 | ||
414 | kfree(xprt->address_strings[RPC_DISPLAY_HEX_PORT]); | |
415 | snprintf(buf, sizeof(buf), "%4hx", port); | |
416 | xprt->address_strings[RPC_DISPLAY_HEX_PORT] = kstrdup(buf, GFP_KERNEL); | |
417 | } | |
418 | ||
419 | /** | |
420 | * xprt_rdma_timer - invoked when an RPC times out | |
421 | * @xprt: controlling RPC transport | |
422 | * @task: RPC task that timed out | |
423 | * | |
424 | * Invoked when the transport is still connected, but an RPC | |
425 | * retransmit timeout occurs. | |
426 | * | |
427 | * Since RDMA connections don't have a keep-alive, forcibly | |
428 | * disconnect and retry to connect. This drives full | |
429 | * detection of the network path, and retransmissions of | |
430 | * all pending RPCs. | |
431 | */ | |
432 | static void | |
433 | xprt_rdma_timer(struct rpc_xprt *xprt, struct rpc_task *task) | |
434 | { | |
435 | xprt_force_disconnect(xprt); | |
436 | } | |
437 | ||
438 | /** | |
439 | * xprt_rdma_set_connect_timeout - set timeouts for establishing a connection | |
440 | * @xprt: controlling transport instance | |
441 | * @connect_timeout: reconnect timeout after client disconnects | |
442 | * @reconnect_timeout: reconnect timeout after server disconnects | |
443 | * | |
444 | */ | |
445 | static void xprt_rdma_set_connect_timeout(struct rpc_xprt *xprt, | |
446 | unsigned long connect_timeout, | |
447 | unsigned long reconnect_timeout) | |
448 | { | |
449 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
450 | ||
451 | trace_xprtrdma_op_set_cto(r_xprt, connect_timeout, reconnect_timeout); | |
452 | ||
453 | spin_lock(&xprt->transport_lock); | |
454 | ||
455 | if (connect_timeout < xprt->connect_timeout) { | |
456 | struct rpc_timeout to; | |
457 | unsigned long initval; | |
458 | ||
459 | to = *xprt->timeout; | |
460 | initval = connect_timeout; | |
461 | if (initval < RPCRDMA_INIT_REEST_TO << 1) | |
462 | initval = RPCRDMA_INIT_REEST_TO << 1; | |
463 | to.to_initval = initval; | |
464 | to.to_maxval = initval; | |
465 | r_xprt->rx_timeout = to; | |
466 | xprt->timeout = &r_xprt->rx_timeout; | |
467 | xprt->connect_timeout = connect_timeout; | |
468 | } | |
469 | ||
470 | if (reconnect_timeout < xprt->max_reconnect_timeout) | |
471 | xprt->max_reconnect_timeout = reconnect_timeout; | |
472 | ||
473 | spin_unlock(&xprt->transport_lock); | |
474 | } | |
475 | ||
476 | /** | |
477 | * xprt_rdma_connect - schedule an attempt to reconnect | |
478 | * @xprt: transport state | |
479 | * @task: RPC scheduler context (unused) | |
480 | * | |
481 | */ | |
482 | static void | |
483 | xprt_rdma_connect(struct rpc_xprt *xprt, struct rpc_task *task) | |
484 | { | |
485 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
486 | struct rpcrdma_ep *ep = r_xprt->rx_ep; | |
487 | unsigned long delay; | |
488 | ||
489 | WARN_ON_ONCE(!xprt_lock_connect(xprt, task, r_xprt)); | |
490 | ||
491 | delay = 0; | |
492 | if (ep && ep->re_connect_status != 0) { | |
493 | delay = xprt_reconnect_delay(xprt); | |
494 | xprt_reconnect_backoff(xprt, RPCRDMA_INIT_REEST_TO); | |
495 | } | |
496 | trace_xprtrdma_op_connect(r_xprt, delay); | |
497 | queue_delayed_work(xprtiod_workqueue, &r_xprt->rx_connect_worker, | |
498 | delay); | |
499 | } | |
500 | ||
501 | /** | |
502 | * xprt_rdma_alloc_slot - allocate an rpc_rqst | |
503 | * @xprt: controlling RPC transport | |
504 | * @task: RPC task requesting a fresh rpc_rqst | |
505 | * | |
506 | * tk_status values: | |
507 | * %0 if task->tk_rqstp points to a fresh rpc_rqst | |
508 | * %-EAGAIN if no rpc_rqst is available; queued on backlog | |
509 | */ | |
510 | static void | |
511 | xprt_rdma_alloc_slot(struct rpc_xprt *xprt, struct rpc_task *task) | |
512 | { | |
513 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
514 | struct rpcrdma_req *req; | |
515 | ||
516 | req = rpcrdma_buffer_get(&r_xprt->rx_buf); | |
517 | if (!req) | |
518 | goto out_sleep; | |
519 | task->tk_rqstp = &req->rl_slot; | |
520 | task->tk_status = 0; | |
521 | return; | |
522 | ||
523 | out_sleep: | |
524 | task->tk_status = -ENOMEM; | |
525 | xprt_add_backlog(xprt, task); | |
526 | } | |
527 | ||
528 | /** | |
529 | * xprt_rdma_free_slot - release an rpc_rqst | |
530 | * @xprt: controlling RPC transport | |
531 | * @rqst: rpc_rqst to release | |
532 | * | |
533 | */ | |
534 | static void | |
535 | xprt_rdma_free_slot(struct rpc_xprt *xprt, struct rpc_rqst *rqst) | |
536 | { | |
537 | struct rpcrdma_xprt *r_xprt = | |
538 | container_of(xprt, struct rpcrdma_xprt, rx_xprt); | |
539 | ||
540 | rpcrdma_reply_put(&r_xprt->rx_buf, rpcr_to_rdmar(rqst)); | |
541 | if (!xprt_wake_up_backlog(xprt, rqst)) { | |
542 | memset(rqst, 0, sizeof(*rqst)); | |
543 | rpcrdma_buffer_put(&r_xprt->rx_buf, rpcr_to_rdmar(rqst)); | |
544 | } | |
545 | } | |
546 | ||
547 | static bool rpcrdma_check_regbuf(struct rpcrdma_xprt *r_xprt, | |
548 | struct rpcrdma_regbuf *rb, size_t size, | |
549 | gfp_t flags) | |
550 | { | |
551 | if (unlikely(rdmab_length(rb) < size)) { | |
552 | if (!rpcrdma_regbuf_realloc(rb, size, flags)) | |
553 | return false; | |
554 | r_xprt->rx_stats.hardway_register_count += size; | |
555 | } | |
556 | return true; | |
557 | } | |
558 | ||
559 | /** | |
560 | * xprt_rdma_allocate - allocate transport resources for an RPC | |
561 | * @task: RPC task | |
562 | * | |
563 | * Return values: | |
564 | * 0: Success; rq_buffer points to RPC buffer to use | |
565 | * ENOMEM: Out of memory, call again later | |
566 | * EIO: A permanent error occurred, do not retry | |
567 | */ | |
568 | static int | |
569 | xprt_rdma_allocate(struct rpc_task *task) | |
570 | { | |
571 | struct rpc_rqst *rqst = task->tk_rqstp; | |
572 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(rqst->rq_xprt); | |
573 | struct rpcrdma_req *req = rpcr_to_rdmar(rqst); | |
574 | gfp_t flags; | |
575 | ||
576 | flags = RPCRDMA_DEF_GFP; | |
577 | if (RPC_IS_ASYNC(task)) | |
578 | flags = GFP_NOWAIT | __GFP_NOWARN; | |
579 | if (RPC_IS_SWAPPER(task)) | |
580 | flags |= __GFP_MEMALLOC; | |
581 | ||
582 | if (!rpcrdma_check_regbuf(r_xprt, req->rl_sendbuf, rqst->rq_callsize, | |
583 | flags)) | |
584 | goto out_fail; | |
585 | if (!rpcrdma_check_regbuf(r_xprt, req->rl_recvbuf, rqst->rq_rcvsize, | |
586 | flags)) | |
587 | goto out_fail; | |
588 | ||
589 | rqst->rq_buffer = rdmab_data(req->rl_sendbuf); | |
590 | rqst->rq_rbuffer = rdmab_data(req->rl_recvbuf); | |
591 | return 0; | |
592 | ||
593 | out_fail: | |
594 | return -ENOMEM; | |
595 | } | |
596 | ||
597 | /** | |
598 | * xprt_rdma_free - release resources allocated by xprt_rdma_allocate | |
599 | * @task: RPC task | |
600 | * | |
601 | * Caller guarantees rqst->rq_buffer is non-NULL. | |
602 | */ | |
603 | static void | |
604 | xprt_rdma_free(struct rpc_task *task) | |
605 | { | |
606 | struct rpc_rqst *rqst = task->tk_rqstp; | |
607 | struct rpcrdma_req *req = rpcr_to_rdmar(rqst); | |
608 | ||
609 | if (unlikely(!list_empty(&req->rl_registered))) { | |
610 | trace_xprtrdma_mrs_zap(task); | |
611 | frwr_unmap_sync(rpcx_to_rdmax(rqst->rq_xprt), req); | |
612 | } | |
613 | ||
614 | /* XXX: If the RPC is completing because of a signal and | |
615 | * not because a reply was received, we ought to ensure | |
616 | * that the Send completion has fired, so that memory | |
617 | * involved with the Send is not still visible to the NIC. | |
618 | */ | |
619 | } | |
620 | ||
621 | /** | |
622 | * xprt_rdma_send_request - marshal and send an RPC request | |
623 | * @rqst: RPC message in rq_snd_buf | |
624 | * | |
625 | * Caller holds the transport's write lock. | |
626 | * | |
627 | * Returns: | |
628 | * %0 if the RPC message has been sent | |
629 | * %-ENOTCONN if the caller should reconnect and call again | |
630 | * %-EAGAIN if the caller should call again | |
631 | * %-ENOBUFS if the caller should call again after a delay | |
632 | * %-EMSGSIZE if encoding ran out of buffer space. The request | |
633 | * was not sent. Do not try to send this message again. | |
634 | * %-EIO if an I/O error occurred. The request was not sent. | |
635 | * Do not try to send this message again. | |
636 | */ | |
637 | static int | |
638 | xprt_rdma_send_request(struct rpc_rqst *rqst) | |
639 | { | |
640 | struct rpc_xprt *xprt = rqst->rq_xprt; | |
641 | struct rpcrdma_req *req = rpcr_to_rdmar(rqst); | |
642 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
643 | int rc = 0; | |
644 | ||
645 | #if defined(CONFIG_SUNRPC_BACKCHANNEL) | |
646 | if (unlikely(!rqst->rq_buffer)) | |
647 | return xprt_rdma_bc_send_reply(rqst); | |
648 | #endif /* CONFIG_SUNRPC_BACKCHANNEL */ | |
649 | ||
650 | if (!xprt_connected(xprt)) | |
651 | return -ENOTCONN; | |
652 | ||
653 | if (!xprt_request_get_cong(xprt, rqst)) | |
654 | return -EBADSLT; | |
655 | ||
656 | rc = rpcrdma_marshal_req(r_xprt, rqst); | |
657 | if (rc < 0) | |
658 | goto failed_marshal; | |
659 | ||
660 | /* Must suppress retransmit to maintain credits */ | |
661 | if (rqst->rq_connect_cookie == xprt->connect_cookie) | |
662 | goto drop_connection; | |
663 | rqst->rq_xtime = ktime_get(); | |
664 | ||
665 | if (frwr_send(r_xprt, req)) | |
666 | goto drop_connection; | |
667 | ||
668 | rqst->rq_xmit_bytes_sent += rqst->rq_snd_buf.len; | |
669 | ||
670 | /* An RPC with no reply will throw off credit accounting, | |
671 | * so drop the connection to reset the credit grant. | |
672 | */ | |
673 | if (!rpc_reply_expected(rqst->rq_task)) | |
674 | goto drop_connection; | |
675 | return 0; | |
676 | ||
677 | failed_marshal: | |
678 | if (rc != -ENOTCONN) | |
679 | return rc; | |
680 | drop_connection: | |
681 | xprt_rdma_close(xprt); | |
682 | return -ENOTCONN; | |
683 | } | |
684 | ||
685 | void xprt_rdma_print_stats(struct rpc_xprt *xprt, struct seq_file *seq) | |
686 | { | |
687 | struct rpcrdma_xprt *r_xprt = rpcx_to_rdmax(xprt); | |
688 | long idle_time = 0; | |
689 | ||
690 | if (xprt_connected(xprt)) | |
691 | idle_time = (long)(jiffies - xprt->last_used) / HZ; | |
692 | ||
693 | seq_puts(seq, "\txprt:\trdma "); | |
694 | seq_printf(seq, "%u %lu %lu %lu %ld %lu %lu %lu %llu %llu ", | |
695 | 0, /* need a local port? */ | |
696 | xprt->stat.bind_count, | |
697 | xprt->stat.connect_count, | |
698 | xprt->stat.connect_time / HZ, | |
699 | idle_time, | |
700 | xprt->stat.sends, | |
701 | xprt->stat.recvs, | |
702 | xprt->stat.bad_xids, | |
703 | xprt->stat.req_u, | |
704 | xprt->stat.bklog_u); | |
705 | seq_printf(seq, "%lu %lu %lu %llu %llu %llu %llu %lu %lu %lu %lu ", | |
706 | r_xprt->rx_stats.read_chunk_count, | |
707 | r_xprt->rx_stats.write_chunk_count, | |
708 | r_xprt->rx_stats.reply_chunk_count, | |
709 | r_xprt->rx_stats.total_rdma_request, | |
710 | r_xprt->rx_stats.total_rdma_reply, | |
711 | r_xprt->rx_stats.pullup_copy_count, | |
712 | r_xprt->rx_stats.fixup_copy_count, | |
713 | r_xprt->rx_stats.hardway_register_count, | |
714 | r_xprt->rx_stats.failed_marshal_count, | |
715 | r_xprt->rx_stats.bad_reply_count, | |
716 | r_xprt->rx_stats.nomsg_call_count); | |
717 | seq_printf(seq, "%lu %lu %lu %lu %lu %lu\n", | |
718 | r_xprt->rx_stats.mrs_recycled, | |
719 | r_xprt->rx_stats.mrs_orphaned, | |
720 | r_xprt->rx_stats.mrs_allocated, | |
721 | r_xprt->rx_stats.local_inv_needed, | |
722 | r_xprt->rx_stats.empty_sendctx_q, | |
723 | r_xprt->rx_stats.reply_waits_for_send); | |
724 | } | |
725 | ||
726 | static int | |
727 | xprt_rdma_enable_swap(struct rpc_xprt *xprt) | |
728 | { | |
729 | return 0; | |
730 | } | |
731 | ||
732 | static void | |
733 | xprt_rdma_disable_swap(struct rpc_xprt *xprt) | |
734 | { | |
735 | } | |
736 | ||
737 | /* | |
738 | * Plumbing for rpc transport switch and kernel module | |
739 | */ | |
740 | ||
741 | static const struct rpc_xprt_ops xprt_rdma_procs = { | |
742 | .reserve_xprt = xprt_reserve_xprt_cong, | |
743 | .release_xprt = xprt_release_xprt_cong, /* sunrpc/xprt.c */ | |
744 | .alloc_slot = xprt_rdma_alloc_slot, | |
745 | .free_slot = xprt_rdma_free_slot, | |
746 | .release_request = xprt_release_rqst_cong, /* ditto */ | |
747 | .wait_for_reply_request = xprt_wait_for_reply_request_def, /* ditto */ | |
748 | .timer = xprt_rdma_timer, | |
749 | .rpcbind = rpcb_getport_async, /* sunrpc/rpcb_clnt.c */ | |
750 | .set_port = xprt_rdma_set_port, | |
751 | .connect = xprt_rdma_connect, | |
752 | .buf_alloc = xprt_rdma_allocate, | |
753 | .buf_free = xprt_rdma_free, | |
754 | .send_request = xprt_rdma_send_request, | |
755 | .close = xprt_rdma_close, | |
756 | .destroy = xprt_rdma_destroy, | |
757 | .set_connect_timeout = xprt_rdma_set_connect_timeout, | |
758 | .print_stats = xprt_rdma_print_stats, | |
759 | .enable_swap = xprt_rdma_enable_swap, | |
760 | .disable_swap = xprt_rdma_disable_swap, | |
761 | .inject_disconnect = xprt_rdma_inject_disconnect, | |
762 | #if defined(CONFIG_SUNRPC_BACKCHANNEL) | |
763 | .bc_setup = xprt_rdma_bc_setup, | |
764 | .bc_maxpayload = xprt_rdma_bc_maxpayload, | |
765 | .bc_num_slots = xprt_rdma_bc_max_slots, | |
766 | .bc_free_rqst = xprt_rdma_bc_free_rqst, | |
767 | .bc_destroy = xprt_rdma_bc_destroy, | |
768 | #endif | |
769 | }; | |
770 | ||
771 | static struct xprt_class xprt_rdma = { | |
772 | .list = LIST_HEAD_INIT(xprt_rdma.list), | |
773 | .name = "rdma", | |
774 | .owner = THIS_MODULE, | |
775 | .ident = XPRT_TRANSPORT_RDMA, | |
776 | .setup = xprt_setup_rdma, | |
777 | .netid = { "rdma", "rdma6", "" }, | |
778 | }; | |
779 | ||
780 | void xprt_rdma_cleanup(void) | |
781 | { | |
782 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) | |
783 | if (sunrpc_table_header) { | |
784 | unregister_sysctl_table(sunrpc_table_header); | |
785 | sunrpc_table_header = NULL; | |
786 | } | |
787 | #endif | |
788 | ||
789 | xprt_unregister_transport(&xprt_rdma); | |
790 | xprt_unregister_transport(&xprt_rdma_bc); | |
791 | } | |
792 | ||
793 | int xprt_rdma_init(void) | |
794 | { | |
795 | int rc; | |
796 | ||
797 | rc = xprt_register_transport(&xprt_rdma); | |
798 | if (rc) | |
799 | return rc; | |
800 | ||
801 | rc = xprt_register_transport(&xprt_rdma_bc); | |
802 | if (rc) { | |
803 | xprt_unregister_transport(&xprt_rdma); | |
804 | return rc; | |
805 | } | |
806 | ||
807 | #if IS_ENABLED(CONFIG_SUNRPC_DEBUG) | |
808 | if (!sunrpc_table_header) | |
809 | sunrpc_table_header = register_sysctl_table(sunrpc_table); | |
810 | #endif | |
811 | return 0; | |
812 | } |