iw.c 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330
  1. /*
  2. * Copyright (c) 2006 Oracle. All rights reserved.
  3. *
  4. * This software is available to you under a choice of one of two
  5. * licenses. You may choose to be licensed under the terms of the GNU
  6. * General Public License (GPL) Version 2, available from the file
  7. * COPYING in the main directory of this source tree, or the
  8. * OpenIB.org BSD license below:
  9. *
  10. * Redistribution and use in source and binary forms, with or
  11. * without modification, are permitted provided that the following
  12. * conditions are met:
  13. *
  14. * - Redistributions of source code must retain the above
  15. * copyright notice, this list of conditions and the following
  16. * disclaimer.
  17. *
  18. * - Redistributions in binary form must reproduce the above
  19. * copyright notice, this list of conditions and the following
  20. * disclaimer in the documentation and/or other materials
  21. * provided with the distribution.
  22. *
  23. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  24. * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  25. * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  26. * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  27. * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  28. * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  29. * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  30. * SOFTWARE.
  31. *
  32. */
  33. #include <linux/kernel.h>
  34. #include <linux/in.h>
  35. #include <linux/if.h>
  36. #include <linux/netdevice.h>
  37. #include <linux/inetdevice.h>
  38. #include <linux/if_arp.h>
  39. #include <linux/delay.h>
  40. #include <linux/slab.h>
  41. #include <linux/module.h>
  42. #include "rds.h"
  43. #include "iw.h"
  44. unsigned int fastreg_pool_size = RDS_FASTREG_POOL_SIZE;
  45. unsigned int fastreg_message_size = RDS_FASTREG_SIZE + 1; /* +1 allows for unaligned MRs */
  46. module_param(fastreg_pool_size, int, 0444);
  47. MODULE_PARM_DESC(fastreg_pool_size, " Max number of fastreg MRs per device");
  48. module_param(fastreg_message_size, int, 0444);
  49. MODULE_PARM_DESC(fastreg_message_size, " Max size of a RDMA transfer (fastreg MRs)");
  50. struct list_head rds_iw_devices;
  51. /* NOTE: if also grabbing iwdev lock, grab this first */
  52. DEFINE_SPINLOCK(iw_nodev_conns_lock);
  53. LIST_HEAD(iw_nodev_conns);
  54. static void rds_iw_add_one(struct ib_device *device)
  55. {
  56. struct rds_iw_device *rds_iwdev;
  57. struct ib_device_attr *dev_attr;
  58. /* Only handle iwarp devices */
  59. if (device->node_type != RDMA_NODE_RNIC)
  60. return;
  61. dev_attr = kmalloc(sizeof *dev_attr, GFP_KERNEL);
  62. if (!dev_attr)
  63. return;
  64. if (ib_query_device(device, dev_attr)) {
  65. rdsdebug("Query device failed for %s\n", device->name);
  66. goto free_attr;
  67. }
  68. rds_iwdev = kmalloc(sizeof *rds_iwdev, GFP_KERNEL);
  69. if (!rds_iwdev)
  70. goto free_attr;
  71. spin_lock_init(&rds_iwdev->spinlock);
  72. rds_iwdev->dma_local_lkey = !!(dev_attr->device_cap_flags & IB_DEVICE_LOCAL_DMA_LKEY);
  73. rds_iwdev->max_wrs = dev_attr->max_qp_wr;
  74. rds_iwdev->max_sge = min(dev_attr->max_sge, RDS_IW_MAX_SGE);
  75. rds_iwdev->dev = device;
  76. rds_iwdev->pd = ib_alloc_pd(device);
  77. if (IS_ERR(rds_iwdev->pd))
  78. goto free_dev;
  79. if (!rds_iwdev->dma_local_lkey) {
  80. rds_iwdev->mr = ib_get_dma_mr(rds_iwdev->pd,
  81. IB_ACCESS_REMOTE_READ |
  82. IB_ACCESS_REMOTE_WRITE |
  83. IB_ACCESS_LOCAL_WRITE);
  84. if (IS_ERR(rds_iwdev->mr))
  85. goto err_pd;
  86. } else
  87. rds_iwdev->mr = NULL;
  88. rds_iwdev->mr_pool = rds_iw_create_mr_pool(rds_iwdev);
  89. if (IS_ERR(rds_iwdev->mr_pool)) {
  90. rds_iwdev->mr_pool = NULL;
  91. goto err_mr;
  92. }
  93. INIT_LIST_HEAD(&rds_iwdev->cm_id_list);
  94. INIT_LIST_HEAD(&rds_iwdev->conn_list);
  95. list_add_tail(&rds_iwdev->list, &rds_iw_devices);
  96. ib_set_client_data(device, &rds_iw_client, rds_iwdev);
  97. goto free_attr;
  98. err_mr:
  99. if (rds_iwdev->mr)
  100. ib_dereg_mr(rds_iwdev->mr);
  101. err_pd:
  102. ib_dealloc_pd(rds_iwdev->pd);
  103. free_dev:
  104. kfree(rds_iwdev);
  105. free_attr:
  106. kfree(dev_attr);
  107. }
  108. static void rds_iw_remove_one(struct ib_device *device)
  109. {
  110. struct rds_iw_device *rds_iwdev;
  111. struct rds_iw_cm_id *i_cm_id, *next;
  112. rds_iwdev = ib_get_client_data(device, &rds_iw_client);
  113. if (!rds_iwdev)
  114. return;
  115. spin_lock_irq(&rds_iwdev->spinlock);
  116. list_for_each_entry_safe(i_cm_id, next, &rds_iwdev->cm_id_list, list) {
  117. list_del(&i_cm_id->list);
  118. kfree(i_cm_id);
  119. }
  120. spin_unlock_irq(&rds_iwdev->spinlock);
  121. rds_iw_destroy_conns(rds_iwdev);
  122. if (rds_iwdev->mr_pool)
  123. rds_iw_destroy_mr_pool(rds_iwdev->mr_pool);
  124. if (rds_iwdev->mr)
  125. ib_dereg_mr(rds_iwdev->mr);
  126. while (ib_dealloc_pd(rds_iwdev->pd)) {
  127. rdsdebug("Failed to dealloc pd %p\n", rds_iwdev->pd);
  128. msleep(1);
  129. }
  130. list_del(&rds_iwdev->list);
  131. kfree(rds_iwdev);
  132. }
  133. struct ib_client rds_iw_client = {
  134. .name = "rds_iw",
  135. .add = rds_iw_add_one,
  136. .remove = rds_iw_remove_one
  137. };
  138. static int rds_iw_conn_info_visitor(struct rds_connection *conn,
  139. void *buffer)
  140. {
  141. struct rds_info_rdma_connection *iinfo = buffer;
  142. struct rds_iw_connection *ic;
  143. /* We will only ever look at IB transports */
  144. if (conn->c_trans != &rds_iw_transport)
  145. return 0;
  146. iinfo->src_addr = conn->c_laddr;
  147. iinfo->dst_addr = conn->c_faddr;
  148. memset(&iinfo->src_gid, 0, sizeof(iinfo->src_gid));
  149. memset(&iinfo->dst_gid, 0, sizeof(iinfo->dst_gid));
  150. if (rds_conn_state(conn) == RDS_CONN_UP) {
  151. struct rds_iw_device *rds_iwdev;
  152. struct rdma_dev_addr *dev_addr;
  153. ic = conn->c_transport_data;
  154. dev_addr = &ic->i_cm_id->route.addr.dev_addr;
  155. rdma_addr_get_sgid(dev_addr, (union ib_gid *) &iinfo->src_gid);
  156. rdma_addr_get_dgid(dev_addr, (union ib_gid *) &iinfo->dst_gid);
  157. rds_iwdev = ib_get_client_data(ic->i_cm_id->device, &rds_iw_client);
  158. iinfo->max_send_wr = ic->i_send_ring.w_nr;
  159. iinfo->max_recv_wr = ic->i_recv_ring.w_nr;
  160. iinfo->max_send_sge = rds_iwdev->max_sge;
  161. rds_iw_get_mr_info(rds_iwdev, iinfo);
  162. }
  163. return 1;
  164. }
  165. static void rds_iw_ic_info(struct socket *sock, unsigned int len,
  166. struct rds_info_iterator *iter,
  167. struct rds_info_lengths *lens)
  168. {
  169. rds_for_each_conn_info(sock, len, iter, lens,
  170. rds_iw_conn_info_visitor,
  171. sizeof(struct rds_info_rdma_connection));
  172. }
  173. /*
  174. * Early RDS/IB was built to only bind to an address if there is an IPoIB
  175. * device with that address set.
  176. *
  177. * If it were me, I'd advocate for something more flexible. Sending and
  178. * receiving should be device-agnostic. Transports would try and maintain
  179. * connections between peers who have messages queued. Userspace would be
  180. * allowed to influence which paths have priority. We could call userspace
  181. * asserting this policy "routing".
  182. */
  183. static int rds_iw_laddr_check(__be32 addr)
  184. {
  185. int ret;
  186. struct rdma_cm_id *cm_id;
  187. struct sockaddr_in sin;
  188. /* Create a CMA ID and try to bind it. This catches both
  189. * IB and iWARP capable NICs.
  190. */
  191. cm_id = rdma_create_id(NULL, NULL, RDMA_PS_TCP, IB_QPT_RC);
  192. if (IS_ERR(cm_id))
  193. return PTR_ERR(cm_id);
  194. memset(&sin, 0, sizeof(sin));
  195. sin.sin_family = AF_INET;
  196. sin.sin_addr.s_addr = addr;
  197. /* rdma_bind_addr will only succeed for IB & iWARP devices */
  198. ret = rdma_bind_addr(cm_id, (struct sockaddr *)&sin);
  199. /* due to this, we will claim to support IB devices unless we
  200. check node_type. */
  201. if (ret || !cm_id->device ||
  202. cm_id->device->node_type != RDMA_NODE_RNIC)
  203. ret = -EADDRNOTAVAIL;
  204. rdsdebug("addr %pI4 ret %d node type %d\n",
  205. &addr, ret,
  206. cm_id->device ? cm_id->device->node_type : -1);
  207. rdma_destroy_id(cm_id);
  208. return ret;
  209. }
  210. void rds_iw_exit(void)
  211. {
  212. rds_info_deregister_func(RDS_INFO_IWARP_CONNECTIONS, rds_iw_ic_info);
  213. rds_iw_destroy_nodev_conns();
  214. ib_unregister_client(&rds_iw_client);
  215. rds_iw_sysctl_exit();
  216. rds_iw_recv_exit();
  217. rds_trans_unregister(&rds_iw_transport);
  218. }
  219. struct rds_transport rds_iw_transport = {
  220. .laddr_check = rds_iw_laddr_check,
  221. .xmit_complete = rds_iw_xmit_complete,
  222. .xmit = rds_iw_xmit,
  223. .xmit_rdma = rds_iw_xmit_rdma,
  224. .recv = rds_iw_recv,
  225. .conn_alloc = rds_iw_conn_alloc,
  226. .conn_free = rds_iw_conn_free,
  227. .conn_connect = rds_iw_conn_connect,
  228. .conn_shutdown = rds_iw_conn_shutdown,
  229. .inc_copy_to_user = rds_iw_inc_copy_to_user,
  230. .inc_free = rds_iw_inc_free,
  231. .cm_initiate_connect = rds_iw_cm_initiate_connect,
  232. .cm_handle_connect = rds_iw_cm_handle_connect,
  233. .cm_connect_complete = rds_iw_cm_connect_complete,
  234. .stats_info_copy = rds_iw_stats_info_copy,
  235. .exit = rds_iw_exit,
  236. .get_mr = rds_iw_get_mr,
  237. .sync_mr = rds_iw_sync_mr,
  238. .free_mr = rds_iw_free_mr,
  239. .flush_mrs = rds_iw_flush_mrs,
  240. .t_owner = THIS_MODULE,
  241. .t_name = "iwarp",
  242. .t_type = RDS_TRANS_IWARP,
  243. .t_prefer_loopback = 1,
  244. };
  245. int rds_iw_init(void)
  246. {
  247. int ret;
  248. INIT_LIST_HEAD(&rds_iw_devices);
  249. ret = ib_register_client(&rds_iw_client);
  250. if (ret)
  251. goto out;
  252. ret = rds_iw_sysctl_init();
  253. if (ret)
  254. goto out_ibreg;
  255. ret = rds_iw_recv_init();
  256. if (ret)
  257. goto out_sysctl;
  258. ret = rds_trans_register(&rds_iw_transport);
  259. if (ret)
  260. goto out_recv;
  261. rds_info_register_func(RDS_INFO_IWARP_CONNECTIONS, rds_iw_ic_info);
  262. goto out;
  263. out_recv:
  264. rds_iw_recv_exit();
  265. out_sysctl:
  266. rds_iw_sysctl_exit();
  267. out_ibreg:
  268. ib_unregister_client(&rds_iw_client);
  269. out:
  270. return ret;
  271. }
  272. MODULE_LICENSE("GPL");