transport.c 22 KB

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