trans_rdma.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765
  1. /*
  2. * linux/fs/9p/trans_rdma.c
  3. *
  4. * RDMA transport layer based on the trans_fd.c implementation.
  5. *
  6. * Copyright (C) 2008 by Tom Tucker <tom@opengridcomputing.com>
  7. * Copyright (C) 2006 by Russ Cox <rsc@swtch.com>
  8. * Copyright (C) 2004-2005 by Latchesar Ionkov <lucho@ionkov.net>
  9. * Copyright (C) 2004-2008 by Eric Van Hensbergen <ericvh@gmail.com>
  10. * Copyright (C) 1997-2002 by Ron Minnich <rminnich@sarnoff.com>
  11. *
  12. * This program is free software; you can redistribute it and/or modify
  13. * it under the terms of the GNU General Public License version 2
  14. * as published by the Free Software Foundation.
  15. *
  16. * This program is distributed in the hope that it will be useful,
  17. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  19. * GNU General Public License for more details.
  20. *
  21. * You should have received a copy of the GNU General Public License
  22. * along with this program; if not, write to:
  23. * Free Software Foundation
  24. * 51 Franklin Street, Fifth Floor
  25. * Boston, MA 02111-1301 USA
  26. *
  27. */
  28. #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  29. #include <linux/in.h>
  30. #include <linux/module.h>
  31. #include <linux/net.h>
  32. #include <linux/ipv6.h>
  33. #include <linux/kthread.h>
  34. #include <linux/errno.h>
  35. #include <linux/kernel.h>
  36. #include <linux/un.h>
  37. #include <linux/uaccess.h>
  38. #include <linux/inet.h>
  39. #include <linux/idr.h>
  40. #include <linux/file.h>
  41. #include <linux/parser.h>
  42. #include <linux/semaphore.h>
  43. #include <linux/slab.h>
  44. #include <net/9p/9p.h>
  45. #include <net/9p/client.h>
  46. #include <net/9p/transport.h>
  47. #include <rdma/ib_verbs.h>
  48. #include <rdma/rdma_cm.h>
  49. #define P9_PORT 5640
  50. #define P9_RDMA_SQ_DEPTH 32
  51. #define P9_RDMA_RQ_DEPTH 32
  52. #define P9_RDMA_SEND_SGE 4
  53. #define P9_RDMA_RECV_SGE 4
  54. #define P9_RDMA_IRD 0
  55. #define P9_RDMA_ORD 0
  56. #define P9_RDMA_TIMEOUT 30000 /* 30 seconds */
  57. #define P9_RDMA_MAXSIZE (1024*1024) /* 1MB */
  58. /**
  59. * struct p9_trans_rdma - RDMA transport instance
  60. *
  61. * @state: tracks the transport state machine for connection setup and tear down
  62. * @cm_id: The RDMA CM ID
  63. * @pd: Protection Domain pointer
  64. * @qp: Queue Pair pointer
  65. * @cq: Completion Queue pointer
  66. * @dm_mr: DMA Memory Region pointer
  67. * @lkey: The local access only memory region key
  68. * @timeout: Number of uSecs to wait for connection management events
  69. * @sq_depth: The depth of the Send Queue
  70. * @sq_sem: Semaphore for the SQ
  71. * @rq_depth: The depth of the Receive Queue.
  72. * @rq_sem: Semaphore for the RQ
  73. * @excess_rc : Amount of posted Receive Contexts without a pending request.
  74. * See rdma_request()
  75. * @addr: The remote peer's address
  76. * @req_lock: Protects the active request list
  77. * @cm_done: Completion event for connection management tracking
  78. */
  79. struct p9_trans_rdma {
  80. enum {
  81. P9_RDMA_INIT,
  82. P9_RDMA_ADDR_RESOLVED,
  83. P9_RDMA_ROUTE_RESOLVED,
  84. P9_RDMA_CONNECTED,
  85. P9_RDMA_FLUSHING,
  86. P9_RDMA_CLOSING,
  87. P9_RDMA_CLOSED,
  88. } state;
  89. struct rdma_cm_id *cm_id;
  90. struct ib_pd *pd;
  91. struct ib_qp *qp;
  92. struct ib_cq *cq;
  93. struct ib_mr *dma_mr;
  94. u32 lkey;
  95. long timeout;
  96. int sq_depth;
  97. struct semaphore sq_sem;
  98. int rq_depth;
  99. struct semaphore rq_sem;
  100. atomic_t excess_rc;
  101. struct sockaddr_in addr;
  102. spinlock_t req_lock;
  103. struct completion cm_done;
  104. };
  105. /**
  106. * p9_rdma_context - Keeps track of in-process WR
  107. *
  108. * @wc_op: The original WR op for when the CQE completes in error.
  109. * @busa: Bus address to unmap when the WR completes
  110. * @req: Keeps track of requests (send)
  111. * @rc: Keepts track of replies (receive)
  112. */
  113. struct p9_rdma_req;
  114. struct p9_rdma_context {
  115. enum ib_wc_opcode wc_op;
  116. dma_addr_t busa;
  117. union {
  118. struct p9_req_t *req;
  119. struct p9_fcall *rc;
  120. };
  121. };
  122. /**
  123. * p9_rdma_opts - Collection of mount options
  124. * @port: port of connection
  125. * @sq_depth: The requested depth of the SQ. This really doesn't need
  126. * to be any deeper than the number of threads used in the client
  127. * @rq_depth: The depth of the RQ. Should be greater than or equal to SQ depth
  128. * @timeout: Time to wait in msecs for CM events
  129. */
  130. struct p9_rdma_opts {
  131. short port;
  132. int sq_depth;
  133. int rq_depth;
  134. long timeout;
  135. };
  136. /*
  137. * Option Parsing (code inspired by NFS code)
  138. */
  139. enum {
  140. /* Options that take integer arguments */
  141. Opt_port, Opt_rq_depth, Opt_sq_depth, Opt_timeout, Opt_err,
  142. };
  143. static match_table_t tokens = {
  144. {Opt_port, "port=%u"},
  145. {Opt_sq_depth, "sq=%u"},
  146. {Opt_rq_depth, "rq=%u"},
  147. {Opt_timeout, "timeout=%u"},
  148. {Opt_err, NULL},
  149. };
  150. /**
  151. * parse_opts - parse mount options into rdma options structure
  152. * @params: options string passed from mount
  153. * @opts: rdma transport-specific structure to parse options into
  154. *
  155. * Returns 0 upon success, -ERRNO upon failure
  156. */
  157. static int parse_opts(char *params, struct p9_rdma_opts *opts)
  158. {
  159. char *p;
  160. substring_t args[MAX_OPT_ARGS];
  161. int option;
  162. char *options, *tmp_options;
  163. opts->port = P9_PORT;
  164. opts->sq_depth = P9_RDMA_SQ_DEPTH;
  165. opts->rq_depth = P9_RDMA_RQ_DEPTH;
  166. opts->timeout = P9_RDMA_TIMEOUT;
  167. if (!params)
  168. return 0;
  169. tmp_options = kstrdup(params, GFP_KERNEL);
  170. if (!tmp_options) {
  171. p9_debug(P9_DEBUG_ERROR,
  172. "failed to allocate copy of option string\n");
  173. return -ENOMEM;
  174. }
  175. options = tmp_options;
  176. while ((p = strsep(&options, ",")) != NULL) {
  177. int token;
  178. int r;
  179. if (!*p)
  180. continue;
  181. token = match_token(p, tokens, args);
  182. if (token == Opt_err)
  183. continue;
  184. r = match_int(&args[0], &option);
  185. if (r < 0) {
  186. p9_debug(P9_DEBUG_ERROR,
  187. "integer field, but no integer?\n");
  188. continue;
  189. }
  190. switch (token) {
  191. case Opt_port:
  192. opts->port = option;
  193. break;
  194. case Opt_sq_depth:
  195. opts->sq_depth = option;
  196. break;
  197. case Opt_rq_depth:
  198. opts->rq_depth = option;
  199. break;
  200. case Opt_timeout:
  201. opts->timeout = option;
  202. break;
  203. default:
  204. continue;
  205. }
  206. }
  207. /* RQ must be at least as large as the SQ */
  208. opts->rq_depth = max(opts->rq_depth, opts->sq_depth);
  209. kfree(tmp_options);
  210. return 0;
  211. }
  212. static int
  213. p9_cm_event_handler(struct rdma_cm_id *id, struct rdma_cm_event *event)
  214. {
  215. struct p9_client *c = id->context;
  216. struct p9_trans_rdma *rdma = c->trans;
  217. switch (event->event) {
  218. case RDMA_CM_EVENT_ADDR_RESOLVED:
  219. BUG_ON(rdma->state != P9_RDMA_INIT);
  220. rdma->state = P9_RDMA_ADDR_RESOLVED;
  221. break;
  222. case RDMA_CM_EVENT_ROUTE_RESOLVED:
  223. BUG_ON(rdma->state != P9_RDMA_ADDR_RESOLVED);
  224. rdma->state = P9_RDMA_ROUTE_RESOLVED;
  225. break;
  226. case RDMA_CM_EVENT_ESTABLISHED:
  227. BUG_ON(rdma->state != P9_RDMA_ROUTE_RESOLVED);
  228. rdma->state = P9_RDMA_CONNECTED;
  229. break;
  230. case RDMA_CM_EVENT_DISCONNECTED:
  231. if (rdma)
  232. rdma->state = P9_RDMA_CLOSED;
  233. if (c)
  234. c->status = Disconnected;
  235. break;
  236. case RDMA_CM_EVENT_TIMEWAIT_EXIT:
  237. break;
  238. case RDMA_CM_EVENT_ADDR_CHANGE:
  239. case RDMA_CM_EVENT_ROUTE_ERROR:
  240. case RDMA_CM_EVENT_DEVICE_REMOVAL:
  241. case RDMA_CM_EVENT_MULTICAST_JOIN:
  242. case RDMA_CM_EVENT_MULTICAST_ERROR:
  243. case RDMA_CM_EVENT_REJECTED:
  244. case RDMA_CM_EVENT_CONNECT_REQUEST:
  245. case RDMA_CM_EVENT_CONNECT_RESPONSE:
  246. case RDMA_CM_EVENT_CONNECT_ERROR:
  247. case RDMA_CM_EVENT_ADDR_ERROR:
  248. case RDMA_CM_EVENT_UNREACHABLE:
  249. c->status = Disconnected;
  250. rdma_disconnect(rdma->cm_id);
  251. break;
  252. default:
  253. BUG();
  254. }
  255. complete(&rdma->cm_done);
  256. return 0;
  257. }
  258. static void
  259. handle_recv(struct p9_client *client, struct p9_trans_rdma *rdma,
  260. struct p9_rdma_context *c, enum ib_wc_status status, u32 byte_len)
  261. {
  262. struct p9_req_t *req;
  263. int err = 0;
  264. int16_t tag;
  265. req = NULL;
  266. ib_dma_unmap_single(rdma->cm_id->device, c->busa, client->msize,
  267. DMA_FROM_DEVICE);
  268. if (status != IB_WC_SUCCESS)
  269. goto err_out;
  270. err = p9_parse_header(c->rc, NULL, NULL, &tag, 1);
  271. if (err)
  272. goto err_out;
  273. req = p9_tag_lookup(client, tag);
  274. if (!req)
  275. goto err_out;
  276. /* Check that we have not yet received a reply for this request.
  277. */
  278. if (unlikely(req->rc)) {
  279. pr_err("Duplicate reply for request %d", tag);
  280. goto err_out;
  281. }
  282. req->rc = c->rc;
  283. p9_client_cb(client, req, REQ_STATUS_RCVD);
  284. return;
  285. err_out:
  286. p9_debug(P9_DEBUG_ERROR, "req %p err %d status %d\n", req, err, status);
  287. rdma->state = P9_RDMA_FLUSHING;
  288. client->status = Disconnected;
  289. }
  290. static void
  291. handle_send(struct p9_client *client, struct p9_trans_rdma *rdma,
  292. struct p9_rdma_context *c, enum ib_wc_status status, u32 byte_len)
  293. {
  294. ib_dma_unmap_single(rdma->cm_id->device,
  295. c->busa, c->req->tc->size,
  296. DMA_TO_DEVICE);
  297. }
  298. static void qp_event_handler(struct ib_event *event, void *context)
  299. {
  300. p9_debug(P9_DEBUG_ERROR, "QP event %d context %p\n",
  301. event->event, context);
  302. }
  303. static void cq_comp_handler(struct ib_cq *cq, void *cq_context)
  304. {
  305. struct p9_client *client = cq_context;
  306. struct p9_trans_rdma *rdma = client->trans;
  307. int ret;
  308. struct ib_wc wc;
  309. ib_req_notify_cq(rdma->cq, IB_CQ_NEXT_COMP);
  310. while ((ret = ib_poll_cq(cq, 1, &wc)) > 0) {
  311. struct p9_rdma_context *c = (void *) (unsigned long) wc.wr_id;
  312. switch (c->wc_op) {
  313. case IB_WC_RECV:
  314. handle_recv(client, rdma, c, wc.status, wc.byte_len);
  315. up(&rdma->rq_sem);
  316. break;
  317. case IB_WC_SEND:
  318. handle_send(client, rdma, c, wc.status, wc.byte_len);
  319. up(&rdma->sq_sem);
  320. break;
  321. default:
  322. pr_err("unexpected completion type, c->wc_op=%d, wc.opcode=%d, status=%d\n",
  323. c->wc_op, wc.opcode, wc.status);
  324. break;
  325. }
  326. kfree(c);
  327. }
  328. }
  329. static void cq_event_handler(struct ib_event *e, void *v)
  330. {
  331. p9_debug(P9_DEBUG_ERROR, "CQ event %d context %p\n", e->event, v);
  332. }
  333. static void rdma_destroy_trans(struct p9_trans_rdma *rdma)
  334. {
  335. if (!rdma)
  336. return;
  337. if (rdma->dma_mr && !IS_ERR(rdma->dma_mr))
  338. ib_dereg_mr(rdma->dma_mr);
  339. if (rdma->qp && !IS_ERR(rdma->qp))
  340. ib_destroy_qp(rdma->qp);
  341. if (rdma->pd && !IS_ERR(rdma->pd))
  342. ib_dealloc_pd(rdma->pd);
  343. if (rdma->cq && !IS_ERR(rdma->cq))
  344. ib_destroy_cq(rdma->cq);
  345. if (rdma->cm_id && !IS_ERR(rdma->cm_id))
  346. rdma_destroy_id(rdma->cm_id);
  347. kfree(rdma);
  348. }
  349. static int
  350. post_recv(struct p9_client *client, struct p9_rdma_context *c)
  351. {
  352. struct p9_trans_rdma *rdma = client->trans;
  353. struct ib_recv_wr wr, *bad_wr;
  354. struct ib_sge sge;
  355. c->busa = ib_dma_map_single(rdma->cm_id->device,
  356. c->rc->sdata, client->msize,
  357. DMA_FROM_DEVICE);
  358. if (ib_dma_mapping_error(rdma->cm_id->device, c->busa))
  359. goto error;
  360. sge.addr = c->busa;
  361. sge.length = client->msize;
  362. sge.lkey = rdma->lkey;
  363. wr.next = NULL;
  364. c->wc_op = IB_WC_RECV;
  365. wr.wr_id = (unsigned long) c;
  366. wr.sg_list = &sge;
  367. wr.num_sge = 1;
  368. return ib_post_recv(rdma->qp, &wr, &bad_wr);
  369. error:
  370. p9_debug(P9_DEBUG_ERROR, "EIO\n");
  371. return -EIO;
  372. }
  373. static int rdma_request(struct p9_client *client, struct p9_req_t *req)
  374. {
  375. struct p9_trans_rdma *rdma = client->trans;
  376. struct ib_send_wr wr, *bad_wr;
  377. struct ib_sge sge;
  378. int err = 0;
  379. unsigned long flags;
  380. struct p9_rdma_context *c = NULL;
  381. struct p9_rdma_context *rpl_context = NULL;
  382. /* When an error occurs between posting the recv and the send,
  383. * there will be a receive context posted without a pending request.
  384. * Since there is no way to "un-post" it, we remember it and skip
  385. * post_recv() for the next request.
  386. * So here,
  387. * see if we are this `next request' and need to absorb an excess rc.
  388. * If yes, then drop and free our own, and do not recv_post().
  389. **/
  390. if (unlikely(atomic_read(&rdma->excess_rc) > 0)) {
  391. if ((atomic_sub_return(1, &rdma->excess_rc) >= 0)) {
  392. /* Got one ! */
  393. kfree(req->rc);
  394. req->rc = NULL;
  395. goto dont_need_post_recv;
  396. } else {
  397. /* We raced and lost. */
  398. atomic_inc(&rdma->excess_rc);
  399. }
  400. }
  401. /* Allocate an fcall for the reply */
  402. rpl_context = kmalloc(sizeof *rpl_context, GFP_NOFS);
  403. if (!rpl_context) {
  404. err = -ENOMEM;
  405. goto recv_error;
  406. }
  407. rpl_context->rc = req->rc;
  408. /*
  409. * Post a receive buffer for this request. We need to ensure
  410. * there is a reply buffer available for every outstanding
  411. * request. A flushed request can result in no reply for an
  412. * outstanding request, so we must keep a count to avoid
  413. * overflowing the RQ.
  414. */
  415. if (down_interruptible(&rdma->rq_sem)) {
  416. err = -EINTR;
  417. goto recv_error;
  418. }
  419. err = post_recv(client, rpl_context);
  420. if (err) {
  421. p9_debug(P9_DEBUG_FCALL, "POST RECV failed\n");
  422. goto recv_error;
  423. }
  424. /* remove posted receive buffer from request structure */
  425. req->rc = NULL;
  426. dont_need_post_recv:
  427. /* Post the request */
  428. c = kmalloc(sizeof *c, GFP_NOFS);
  429. if (!c) {
  430. err = -ENOMEM;
  431. goto send_error;
  432. }
  433. c->req = req;
  434. c->busa = ib_dma_map_single(rdma->cm_id->device,
  435. c->req->tc->sdata, c->req->tc->size,
  436. DMA_TO_DEVICE);
  437. if (ib_dma_mapping_error(rdma->cm_id->device, c->busa)) {
  438. err = -EIO;
  439. goto send_error;
  440. }
  441. sge.addr = c->busa;
  442. sge.length = c->req->tc->size;
  443. sge.lkey = rdma->lkey;
  444. wr.next = NULL;
  445. c->wc_op = IB_WC_SEND;
  446. wr.wr_id = (unsigned long) c;
  447. wr.opcode = IB_WR_SEND;
  448. wr.send_flags = IB_SEND_SIGNALED;
  449. wr.sg_list = &sge;
  450. wr.num_sge = 1;
  451. if (down_interruptible(&rdma->sq_sem)) {
  452. err = -EINTR;
  453. goto send_error;
  454. }
  455. /* Mark request as `sent' *before* we actually send it,
  456. * because doing if after could erase the REQ_STATUS_RCVD
  457. * status in case of a very fast reply.
  458. */
  459. req->status = REQ_STATUS_SENT;
  460. err = ib_post_send(rdma->qp, &wr, &bad_wr);
  461. if (err)
  462. goto send_error;
  463. /* Success */
  464. return 0;
  465. /* Handle errors that happened during or while preparing the send: */
  466. send_error:
  467. req->status = REQ_STATUS_ERROR;
  468. kfree(c);
  469. p9_debug(P9_DEBUG_ERROR, "Error %d in rdma_request()\n", err);
  470. /* Ach.
  471. * We did recv_post(), but not send. We have one recv_post in excess.
  472. */
  473. atomic_inc(&rdma->excess_rc);
  474. return err;
  475. /* Handle errors that happened during or while preparing post_recv(): */
  476. recv_error:
  477. kfree(rpl_context);
  478. spin_lock_irqsave(&rdma->req_lock, flags);
  479. if (rdma->state < P9_RDMA_CLOSING) {
  480. rdma->state = P9_RDMA_CLOSING;
  481. spin_unlock_irqrestore(&rdma->req_lock, flags);
  482. rdma_disconnect(rdma->cm_id);
  483. } else
  484. spin_unlock_irqrestore(&rdma->req_lock, flags);
  485. return err;
  486. }
  487. static void rdma_close(struct p9_client *client)
  488. {
  489. struct p9_trans_rdma *rdma;
  490. if (!client)
  491. return;
  492. rdma = client->trans;
  493. if (!rdma)
  494. return;
  495. client->status = Disconnected;
  496. rdma_disconnect(rdma->cm_id);
  497. rdma_destroy_trans(rdma);
  498. }
  499. /**
  500. * alloc_rdma - Allocate and initialize the rdma transport structure
  501. * @opts: Mount options structure
  502. */
  503. static struct p9_trans_rdma *alloc_rdma(struct p9_rdma_opts *opts)
  504. {
  505. struct p9_trans_rdma *rdma;
  506. rdma = kzalloc(sizeof(struct p9_trans_rdma), GFP_KERNEL);
  507. if (!rdma)
  508. return NULL;
  509. rdma->sq_depth = opts->sq_depth;
  510. rdma->rq_depth = opts->rq_depth;
  511. rdma->timeout = opts->timeout;
  512. spin_lock_init(&rdma->req_lock);
  513. init_completion(&rdma->cm_done);
  514. sema_init(&rdma->sq_sem, rdma->sq_depth);
  515. sema_init(&rdma->rq_sem, rdma->rq_depth);
  516. atomic_set(&rdma->excess_rc, 0);
  517. return rdma;
  518. }
  519. static int rdma_cancel(struct p9_client *client, struct p9_req_t *req)
  520. {
  521. /* Nothing to do here.
  522. * We will take care of it (if we have to) in rdma_cancelled()
  523. */
  524. return 1;
  525. }
  526. /* A request has been fully flushed without a reply.
  527. * That means we have posted one buffer in excess.
  528. */
  529. static int rdma_cancelled(struct p9_client *client, struct p9_req_t *req)
  530. {
  531. struct p9_trans_rdma *rdma = client->trans;
  532. atomic_inc(&rdma->excess_rc);
  533. return 0;
  534. }
  535. /**
  536. * trans_create_rdma - Transport method for creating atransport instance
  537. * @client: client instance
  538. * @addr: IP address string
  539. * @args: Mount options string
  540. */
  541. static int
  542. rdma_create_trans(struct p9_client *client, const char *addr, char *args)
  543. {
  544. int err;
  545. struct p9_rdma_opts opts;
  546. struct p9_trans_rdma *rdma;
  547. struct rdma_conn_param conn_param;
  548. struct ib_qp_init_attr qp_attr;
  549. struct ib_device_attr devattr;
  550. /* Parse the transport specific mount options */
  551. err = parse_opts(args, &opts);
  552. if (err < 0)
  553. return err;
  554. /* Create and initialize the RDMA transport structure */
  555. rdma = alloc_rdma(&opts);
  556. if (!rdma)
  557. return -ENOMEM;
  558. /* Create the RDMA CM ID */
  559. rdma->cm_id = rdma_create_id(p9_cm_event_handler, client, RDMA_PS_TCP,
  560. IB_QPT_RC);
  561. if (IS_ERR(rdma->cm_id))
  562. goto error;
  563. /* Associate the client with the transport */
  564. client->trans = rdma;
  565. /* Resolve the server's address */
  566. rdma->addr.sin_family = AF_INET;
  567. rdma->addr.sin_addr.s_addr = in_aton(addr);
  568. rdma->addr.sin_port = htons(opts.port);
  569. err = rdma_resolve_addr(rdma->cm_id, NULL,
  570. (struct sockaddr *)&rdma->addr,
  571. rdma->timeout);
  572. if (err)
  573. goto error;
  574. err = wait_for_completion_interruptible(&rdma->cm_done);
  575. if (err || (rdma->state != P9_RDMA_ADDR_RESOLVED))
  576. goto error;
  577. /* Resolve the route to the server */
  578. err = rdma_resolve_route(rdma->cm_id, rdma->timeout);
  579. if (err)
  580. goto error;
  581. err = wait_for_completion_interruptible(&rdma->cm_done);
  582. if (err || (rdma->state != P9_RDMA_ROUTE_RESOLVED))
  583. goto error;
  584. /* Query the device attributes */
  585. err = ib_query_device(rdma->cm_id->device, &devattr);
  586. if (err)
  587. goto error;
  588. /* Create the Completion Queue */
  589. rdma->cq = ib_create_cq(rdma->cm_id->device, cq_comp_handler,
  590. cq_event_handler, client,
  591. opts.sq_depth + opts.rq_depth + 1, 0);
  592. if (IS_ERR(rdma->cq))
  593. goto error;
  594. ib_req_notify_cq(rdma->cq, IB_CQ_NEXT_COMP);
  595. /* Create the Protection Domain */
  596. rdma->pd = ib_alloc_pd(rdma->cm_id->device);
  597. if (IS_ERR(rdma->pd))
  598. goto error;
  599. /* Cache the DMA lkey in the transport */
  600. rdma->dma_mr = NULL;
  601. if (devattr.device_cap_flags & IB_DEVICE_LOCAL_DMA_LKEY)
  602. rdma->lkey = rdma->cm_id->device->local_dma_lkey;
  603. else {
  604. rdma->dma_mr = ib_get_dma_mr(rdma->pd, IB_ACCESS_LOCAL_WRITE);
  605. if (IS_ERR(rdma->dma_mr))
  606. goto error;
  607. rdma->lkey = rdma->dma_mr->lkey;
  608. }
  609. /* Create the Queue Pair */
  610. memset(&qp_attr, 0, sizeof qp_attr);
  611. qp_attr.event_handler = qp_event_handler;
  612. qp_attr.qp_context = client;
  613. qp_attr.cap.max_send_wr = opts.sq_depth;
  614. qp_attr.cap.max_recv_wr = opts.rq_depth;
  615. qp_attr.cap.max_send_sge = P9_RDMA_SEND_SGE;
  616. qp_attr.cap.max_recv_sge = P9_RDMA_RECV_SGE;
  617. qp_attr.sq_sig_type = IB_SIGNAL_REQ_WR;
  618. qp_attr.qp_type = IB_QPT_RC;
  619. qp_attr.send_cq = rdma->cq;
  620. qp_attr.recv_cq = rdma->cq;
  621. err = rdma_create_qp(rdma->cm_id, rdma->pd, &qp_attr);
  622. if (err)
  623. goto error;
  624. rdma->qp = rdma->cm_id->qp;
  625. /* Request a connection */
  626. memset(&conn_param, 0, sizeof(conn_param));
  627. conn_param.private_data = NULL;
  628. conn_param.private_data_len = 0;
  629. conn_param.responder_resources = P9_RDMA_IRD;
  630. conn_param.initiator_depth = P9_RDMA_ORD;
  631. err = rdma_connect(rdma->cm_id, &conn_param);
  632. if (err)
  633. goto error;
  634. err = wait_for_completion_interruptible(&rdma->cm_done);
  635. if (err || (rdma->state != P9_RDMA_CONNECTED))
  636. goto error;
  637. client->status = Connected;
  638. return 0;
  639. error:
  640. rdma_destroy_trans(rdma);
  641. return -ENOTCONN;
  642. }
  643. static struct p9_trans_module p9_rdma_trans = {
  644. .name = "rdma",
  645. .maxsize = P9_RDMA_MAXSIZE,
  646. .def = 0,
  647. .owner = THIS_MODULE,
  648. .create = rdma_create_trans,
  649. .close = rdma_close,
  650. .request = rdma_request,
  651. .cancel = rdma_cancel,
  652. .cancelled = rdma_cancelled,
  653. };
  654. /**
  655. * p9_trans_rdma_init - Register the 9P RDMA transport driver
  656. */
  657. static int __init p9_trans_rdma_init(void)
  658. {
  659. v9fs_register_trans(&p9_rdma_trans);
  660. return 0;
  661. }
  662. static void __exit p9_trans_rdma_exit(void)
  663. {
  664. v9fs_unregister_trans(&p9_rdma_trans);
  665. }
  666. module_init(p9_trans_rdma_init);
  667. module_exit(p9_trans_rdma_exit);
  668. MODULE_AUTHOR("Tom Tucker <tom@opengridcomputing.com>");
  669. MODULE_DESCRIPTION("RDMA Transport for 9P");
  670. MODULE_LICENSE("Dual BSD/GPL");