ucma.c 36 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556
  1. /*
  2. * Copyright (c) 2005-2006 Intel Corporation. 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. #include <linux/completion.h>
  33. #include <linux/file.h>
  34. #include <linux/mutex.h>
  35. #include <linux/poll.h>
  36. #include <linux/sched.h>
  37. #include <linux/idr.h>
  38. #include <linux/in.h>
  39. #include <linux/in6.h>
  40. #include <linux/miscdevice.h>
  41. #include <linux/slab.h>
  42. #include <linux/sysctl.h>
  43. #include <linux/module.h>
  44. #include <rdma/rdma_user_cm.h>
  45. #include <rdma/ib_marshall.h>
  46. #include <rdma/rdma_cm.h>
  47. #include <rdma/rdma_cm_ib.h>
  48. #include <rdma/ib_addr.h>
  49. #include <rdma/ib.h>
  50. MODULE_AUTHOR("Sean Hefty");
  51. MODULE_DESCRIPTION("RDMA Userspace Connection Manager Access");
  52. MODULE_LICENSE("Dual BSD/GPL");
  53. static unsigned int max_backlog = 1024;
  54. static struct ctl_table_header *ucma_ctl_table_hdr;
  55. static ctl_table ucma_ctl_table[] = {
  56. {
  57. .procname = "max_backlog",
  58. .data = &max_backlog,
  59. .maxlen = sizeof max_backlog,
  60. .mode = 0644,
  61. .proc_handler = proc_dointvec,
  62. },
  63. { }
  64. };
  65. struct ucma_file {
  66. struct mutex mut;
  67. struct file *filp;
  68. struct list_head ctx_list;
  69. struct list_head event_list;
  70. wait_queue_head_t poll_wait;
  71. };
  72. struct ucma_context {
  73. int id;
  74. struct completion comp;
  75. atomic_t ref;
  76. int events_reported;
  77. int backlog;
  78. struct ucma_file *file;
  79. struct rdma_cm_id *cm_id;
  80. u64 uid;
  81. struct list_head list;
  82. struct list_head mc_list;
  83. };
  84. struct ucma_multicast {
  85. struct ucma_context *ctx;
  86. int id;
  87. int events_reported;
  88. u64 uid;
  89. struct list_head list;
  90. struct sockaddr_storage addr;
  91. };
  92. struct ucma_event {
  93. struct ucma_context *ctx;
  94. struct ucma_multicast *mc;
  95. struct list_head list;
  96. struct rdma_cm_id *cm_id;
  97. struct rdma_ucm_event_resp resp;
  98. };
  99. static DEFINE_MUTEX(mut);
  100. static DEFINE_IDR(ctx_idr);
  101. static DEFINE_IDR(multicast_idr);
  102. static inline struct ucma_context *_ucma_find_context(int id,
  103. struct ucma_file *file)
  104. {
  105. struct ucma_context *ctx;
  106. ctx = idr_find(&ctx_idr, id);
  107. if (!ctx)
  108. ctx = ERR_PTR(-ENOENT);
  109. else if (ctx->file != file)
  110. ctx = ERR_PTR(-EINVAL);
  111. return ctx;
  112. }
  113. static struct ucma_context *ucma_get_ctx(struct ucma_file *file, int id)
  114. {
  115. struct ucma_context *ctx;
  116. mutex_lock(&mut);
  117. ctx = _ucma_find_context(id, file);
  118. if (!IS_ERR(ctx))
  119. atomic_inc(&ctx->ref);
  120. mutex_unlock(&mut);
  121. return ctx;
  122. }
  123. static void ucma_put_ctx(struct ucma_context *ctx)
  124. {
  125. if (atomic_dec_and_test(&ctx->ref))
  126. complete(&ctx->comp);
  127. }
  128. static struct ucma_context *ucma_alloc_ctx(struct ucma_file *file)
  129. {
  130. struct ucma_context *ctx;
  131. ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
  132. if (!ctx)
  133. return NULL;
  134. atomic_set(&ctx->ref, 1);
  135. init_completion(&ctx->comp);
  136. INIT_LIST_HEAD(&ctx->mc_list);
  137. ctx->file = file;
  138. mutex_lock(&mut);
  139. ctx->id = idr_alloc(&ctx_idr, ctx, 0, 0, GFP_KERNEL);
  140. mutex_unlock(&mut);
  141. if (ctx->id < 0)
  142. goto error;
  143. list_add_tail(&ctx->list, &file->ctx_list);
  144. return ctx;
  145. error:
  146. kfree(ctx);
  147. return NULL;
  148. }
  149. static struct ucma_multicast* ucma_alloc_multicast(struct ucma_context *ctx)
  150. {
  151. struct ucma_multicast *mc;
  152. mc = kzalloc(sizeof(*mc), GFP_KERNEL);
  153. if (!mc)
  154. return NULL;
  155. mutex_lock(&mut);
  156. mc->id = idr_alloc(&multicast_idr, mc, 0, 0, GFP_KERNEL);
  157. mutex_unlock(&mut);
  158. if (mc->id < 0)
  159. goto error;
  160. mc->ctx = ctx;
  161. list_add_tail(&mc->list, &ctx->mc_list);
  162. return mc;
  163. error:
  164. kfree(mc);
  165. return NULL;
  166. }
  167. static void ucma_copy_conn_event(struct rdma_ucm_conn_param *dst,
  168. struct rdma_conn_param *src)
  169. {
  170. if (src->private_data_len)
  171. memcpy(dst->private_data, src->private_data,
  172. src->private_data_len);
  173. dst->private_data_len = src->private_data_len;
  174. dst->responder_resources =src->responder_resources;
  175. dst->initiator_depth = src->initiator_depth;
  176. dst->flow_control = src->flow_control;
  177. dst->retry_count = src->retry_count;
  178. dst->rnr_retry_count = src->rnr_retry_count;
  179. dst->srq = src->srq;
  180. dst->qp_num = src->qp_num;
  181. }
  182. static void ucma_copy_ud_event(struct rdma_ucm_ud_param *dst,
  183. struct rdma_ud_param *src)
  184. {
  185. if (src->private_data_len)
  186. memcpy(dst->private_data, src->private_data,
  187. src->private_data_len);
  188. dst->private_data_len = src->private_data_len;
  189. ib_copy_ah_attr_to_user(&dst->ah_attr, &src->ah_attr);
  190. dst->qp_num = src->qp_num;
  191. dst->qkey = src->qkey;
  192. }
  193. static void ucma_set_event_context(struct ucma_context *ctx,
  194. struct rdma_cm_event *event,
  195. struct ucma_event *uevent)
  196. {
  197. uevent->ctx = ctx;
  198. switch (event->event) {
  199. case RDMA_CM_EVENT_MULTICAST_JOIN:
  200. case RDMA_CM_EVENT_MULTICAST_ERROR:
  201. uevent->mc = (struct ucma_multicast *)
  202. event->param.ud.private_data;
  203. uevent->resp.uid = uevent->mc->uid;
  204. uevent->resp.id = uevent->mc->id;
  205. break;
  206. default:
  207. uevent->resp.uid = ctx->uid;
  208. uevent->resp.id = ctx->id;
  209. break;
  210. }
  211. }
  212. static int ucma_event_handler(struct rdma_cm_id *cm_id,
  213. struct rdma_cm_event *event)
  214. {
  215. struct ucma_event *uevent;
  216. struct ucma_context *ctx = cm_id->context;
  217. int ret = 0;
  218. uevent = kzalloc(sizeof(*uevent), GFP_KERNEL);
  219. if (!uevent)
  220. return event->event == RDMA_CM_EVENT_CONNECT_REQUEST;
  221. mutex_lock(&ctx->file->mut);
  222. uevent->cm_id = cm_id;
  223. ucma_set_event_context(ctx, event, uevent);
  224. uevent->resp.event = event->event;
  225. uevent->resp.status = event->status;
  226. if (cm_id->qp_type == IB_QPT_UD)
  227. ucma_copy_ud_event(&uevent->resp.param.ud, &event->param.ud);
  228. else
  229. ucma_copy_conn_event(&uevent->resp.param.conn,
  230. &event->param.conn);
  231. if (event->event == RDMA_CM_EVENT_CONNECT_REQUEST) {
  232. if (!ctx->backlog) {
  233. ret = -ENOMEM;
  234. kfree(uevent);
  235. goto out;
  236. }
  237. ctx->backlog--;
  238. } else if (!ctx->uid) {
  239. /*
  240. * We ignore events for new connections until userspace has set
  241. * their context. This can only happen if an error occurs on a
  242. * new connection before the user accepts it. This is okay,
  243. * since the accept will just fail later.
  244. */
  245. kfree(uevent);
  246. goto out;
  247. }
  248. list_add_tail(&uevent->list, &ctx->file->event_list);
  249. wake_up_interruptible(&ctx->file->poll_wait);
  250. out:
  251. mutex_unlock(&ctx->file->mut);
  252. return ret;
  253. }
  254. static ssize_t ucma_get_event(struct ucma_file *file, const char __user *inbuf,
  255. int in_len, int out_len)
  256. {
  257. struct ucma_context *ctx;
  258. struct rdma_ucm_get_event cmd;
  259. struct ucma_event *uevent;
  260. int ret = 0;
  261. if (out_len < sizeof uevent->resp)
  262. return -ENOSPC;
  263. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  264. return -EFAULT;
  265. mutex_lock(&file->mut);
  266. while (list_empty(&file->event_list)) {
  267. mutex_unlock(&file->mut);
  268. if (file->filp->f_flags & O_NONBLOCK)
  269. return -EAGAIN;
  270. if (wait_event_interruptible(file->poll_wait,
  271. !list_empty(&file->event_list)))
  272. return -ERESTARTSYS;
  273. mutex_lock(&file->mut);
  274. }
  275. uevent = list_entry(file->event_list.next, struct ucma_event, list);
  276. if (uevent->resp.event == RDMA_CM_EVENT_CONNECT_REQUEST) {
  277. ctx = ucma_alloc_ctx(file);
  278. if (!ctx) {
  279. ret = -ENOMEM;
  280. goto done;
  281. }
  282. uevent->ctx->backlog++;
  283. ctx->cm_id = uevent->cm_id;
  284. ctx->cm_id->context = ctx;
  285. uevent->resp.id = ctx->id;
  286. }
  287. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  288. &uevent->resp, sizeof uevent->resp)) {
  289. ret = -EFAULT;
  290. goto done;
  291. }
  292. list_del(&uevent->list);
  293. uevent->ctx->events_reported++;
  294. if (uevent->mc)
  295. uevent->mc->events_reported++;
  296. kfree(uevent);
  297. done:
  298. mutex_unlock(&file->mut);
  299. return ret;
  300. }
  301. static int ucma_get_qp_type(struct rdma_ucm_create_id *cmd, enum ib_qp_type *qp_type)
  302. {
  303. switch (cmd->ps) {
  304. case RDMA_PS_TCP:
  305. *qp_type = IB_QPT_RC;
  306. return 0;
  307. case RDMA_PS_UDP:
  308. case RDMA_PS_IPOIB:
  309. *qp_type = IB_QPT_UD;
  310. return 0;
  311. case RDMA_PS_IB:
  312. *qp_type = cmd->qp_type;
  313. return 0;
  314. default:
  315. return -EINVAL;
  316. }
  317. }
  318. static ssize_t ucma_create_id(struct ucma_file *file, const char __user *inbuf,
  319. int in_len, int out_len)
  320. {
  321. struct rdma_ucm_create_id cmd;
  322. struct rdma_ucm_create_id_resp resp;
  323. struct ucma_context *ctx;
  324. enum ib_qp_type qp_type;
  325. int ret;
  326. if (out_len < sizeof(resp))
  327. return -ENOSPC;
  328. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  329. return -EFAULT;
  330. ret = ucma_get_qp_type(&cmd, &qp_type);
  331. if (ret)
  332. return ret;
  333. mutex_lock(&file->mut);
  334. ctx = ucma_alloc_ctx(file);
  335. mutex_unlock(&file->mut);
  336. if (!ctx)
  337. return -ENOMEM;
  338. ctx->uid = cmd.uid;
  339. ctx->cm_id = rdma_create_id(ucma_event_handler, ctx, cmd.ps, qp_type);
  340. if (IS_ERR(ctx->cm_id)) {
  341. ret = PTR_ERR(ctx->cm_id);
  342. goto err1;
  343. }
  344. resp.id = ctx->id;
  345. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  346. &resp, sizeof(resp))) {
  347. ret = -EFAULT;
  348. goto err2;
  349. }
  350. return 0;
  351. err2:
  352. rdma_destroy_id(ctx->cm_id);
  353. err1:
  354. mutex_lock(&mut);
  355. idr_remove(&ctx_idr, ctx->id);
  356. mutex_unlock(&mut);
  357. kfree(ctx);
  358. return ret;
  359. }
  360. static void ucma_cleanup_multicast(struct ucma_context *ctx)
  361. {
  362. struct ucma_multicast *mc, *tmp;
  363. mutex_lock(&mut);
  364. list_for_each_entry_safe(mc, tmp, &ctx->mc_list, list) {
  365. list_del(&mc->list);
  366. idr_remove(&multicast_idr, mc->id);
  367. kfree(mc);
  368. }
  369. mutex_unlock(&mut);
  370. }
  371. static void ucma_cleanup_mc_events(struct ucma_multicast *mc)
  372. {
  373. struct ucma_event *uevent, *tmp;
  374. list_for_each_entry_safe(uevent, tmp, &mc->ctx->file->event_list, list) {
  375. if (uevent->mc != mc)
  376. continue;
  377. list_del(&uevent->list);
  378. kfree(uevent);
  379. }
  380. }
  381. /*
  382. * We cannot hold file->mut when calling rdma_destroy_id() or we can
  383. * deadlock. We also acquire file->mut in ucma_event_handler(), and
  384. * rdma_destroy_id() will wait until all callbacks have completed.
  385. */
  386. static int ucma_free_ctx(struct ucma_context *ctx)
  387. {
  388. int events_reported;
  389. struct ucma_event *uevent, *tmp;
  390. LIST_HEAD(list);
  391. /* No new events will be generated after destroying the id. */
  392. rdma_destroy_id(ctx->cm_id);
  393. ucma_cleanup_multicast(ctx);
  394. /* Cleanup events not yet reported to the user. */
  395. mutex_lock(&ctx->file->mut);
  396. list_for_each_entry_safe(uevent, tmp, &ctx->file->event_list, list) {
  397. if (uevent->ctx == ctx)
  398. list_move_tail(&uevent->list, &list);
  399. }
  400. list_del(&ctx->list);
  401. mutex_unlock(&ctx->file->mut);
  402. list_for_each_entry_safe(uevent, tmp, &list, list) {
  403. list_del(&uevent->list);
  404. if (uevent->resp.event == RDMA_CM_EVENT_CONNECT_REQUEST)
  405. rdma_destroy_id(uevent->cm_id);
  406. kfree(uevent);
  407. }
  408. events_reported = ctx->events_reported;
  409. kfree(ctx);
  410. return events_reported;
  411. }
  412. static ssize_t ucma_destroy_id(struct ucma_file *file, const char __user *inbuf,
  413. int in_len, int out_len)
  414. {
  415. struct rdma_ucm_destroy_id cmd;
  416. struct rdma_ucm_destroy_id_resp resp;
  417. struct ucma_context *ctx;
  418. int ret = 0;
  419. if (out_len < sizeof(resp))
  420. return -ENOSPC;
  421. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  422. return -EFAULT;
  423. mutex_lock(&mut);
  424. ctx = _ucma_find_context(cmd.id, file);
  425. if (!IS_ERR(ctx))
  426. idr_remove(&ctx_idr, ctx->id);
  427. mutex_unlock(&mut);
  428. if (IS_ERR(ctx))
  429. return PTR_ERR(ctx);
  430. ucma_put_ctx(ctx);
  431. wait_for_completion(&ctx->comp);
  432. resp.events_reported = ucma_free_ctx(ctx);
  433. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  434. &resp, sizeof(resp)))
  435. ret = -EFAULT;
  436. return ret;
  437. }
  438. static ssize_t ucma_bind_addr(struct ucma_file *file, const char __user *inbuf,
  439. int in_len, int out_len)
  440. {
  441. struct rdma_ucm_bind_addr cmd;
  442. struct ucma_context *ctx;
  443. int ret;
  444. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  445. return -EFAULT;
  446. ctx = ucma_get_ctx(file, cmd.id);
  447. if (IS_ERR(ctx))
  448. return PTR_ERR(ctx);
  449. ret = rdma_bind_addr(ctx->cm_id, (struct sockaddr *) &cmd.addr);
  450. ucma_put_ctx(ctx);
  451. return ret;
  452. }
  453. static ssize_t ucma_resolve_addr(struct ucma_file *file,
  454. const char __user *inbuf,
  455. int in_len, int out_len)
  456. {
  457. struct rdma_ucm_resolve_addr cmd;
  458. struct ucma_context *ctx;
  459. int ret;
  460. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  461. return -EFAULT;
  462. ctx = ucma_get_ctx(file, cmd.id);
  463. if (IS_ERR(ctx))
  464. return PTR_ERR(ctx);
  465. ret = rdma_resolve_addr(ctx->cm_id, (struct sockaddr *) &cmd.src_addr,
  466. (struct sockaddr *) &cmd.dst_addr,
  467. cmd.timeout_ms);
  468. ucma_put_ctx(ctx);
  469. return ret;
  470. }
  471. static ssize_t ucma_resolve_route(struct ucma_file *file,
  472. const char __user *inbuf,
  473. int in_len, int out_len)
  474. {
  475. struct rdma_ucm_resolve_route cmd;
  476. struct ucma_context *ctx;
  477. int ret;
  478. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  479. return -EFAULT;
  480. ctx = ucma_get_ctx(file, cmd.id);
  481. if (IS_ERR(ctx))
  482. return PTR_ERR(ctx);
  483. ret = rdma_resolve_route(ctx->cm_id, cmd.timeout_ms);
  484. ucma_put_ctx(ctx);
  485. return ret;
  486. }
  487. static void ucma_copy_ib_route(struct rdma_ucm_query_route_resp *resp,
  488. struct rdma_route *route)
  489. {
  490. struct rdma_dev_addr *dev_addr;
  491. resp->num_paths = route->num_paths;
  492. switch (route->num_paths) {
  493. case 0:
  494. dev_addr = &route->addr.dev_addr;
  495. rdma_addr_get_dgid(dev_addr,
  496. (union ib_gid *) &resp->ib_route[0].dgid);
  497. rdma_addr_get_sgid(dev_addr,
  498. (union ib_gid *) &resp->ib_route[0].sgid);
  499. resp->ib_route[0].pkey = cpu_to_be16(ib_addr_get_pkey(dev_addr));
  500. break;
  501. case 2:
  502. ib_copy_path_rec_to_user(&resp->ib_route[1],
  503. &route->path_rec[1]);
  504. /* fall through */
  505. case 1:
  506. ib_copy_path_rec_to_user(&resp->ib_route[0],
  507. &route->path_rec[0]);
  508. break;
  509. default:
  510. break;
  511. }
  512. }
  513. static void ucma_copy_iboe_route(struct rdma_ucm_query_route_resp *resp,
  514. struct rdma_route *route)
  515. {
  516. struct rdma_dev_addr *dev_addr;
  517. struct net_device *dev;
  518. u16 vid = 0;
  519. resp->num_paths = route->num_paths;
  520. switch (route->num_paths) {
  521. case 0:
  522. dev_addr = &route->addr.dev_addr;
  523. dev = dev_get_by_index(&init_net, dev_addr->bound_dev_if);
  524. if (dev) {
  525. vid = rdma_vlan_dev_vlan_id(dev);
  526. dev_put(dev);
  527. }
  528. iboe_mac_vlan_to_ll((union ib_gid *) &resp->ib_route[0].dgid,
  529. dev_addr->dst_dev_addr, vid);
  530. iboe_addr_get_sgid(dev_addr,
  531. (union ib_gid *) &resp->ib_route[0].sgid);
  532. resp->ib_route[0].pkey = cpu_to_be16(0xffff);
  533. break;
  534. case 2:
  535. ib_copy_path_rec_to_user(&resp->ib_route[1],
  536. &route->path_rec[1]);
  537. /* fall through */
  538. case 1:
  539. ib_copy_path_rec_to_user(&resp->ib_route[0],
  540. &route->path_rec[0]);
  541. break;
  542. default:
  543. break;
  544. }
  545. }
  546. static void ucma_copy_iw_route(struct rdma_ucm_query_route_resp *resp,
  547. struct rdma_route *route)
  548. {
  549. struct rdma_dev_addr *dev_addr;
  550. dev_addr = &route->addr.dev_addr;
  551. rdma_addr_get_dgid(dev_addr, (union ib_gid *) &resp->ib_route[0].dgid);
  552. rdma_addr_get_sgid(dev_addr, (union ib_gid *) &resp->ib_route[0].sgid);
  553. }
  554. static ssize_t ucma_query_route(struct ucma_file *file,
  555. const char __user *inbuf,
  556. int in_len, int out_len)
  557. {
  558. struct rdma_ucm_query cmd;
  559. struct rdma_ucm_query_route_resp resp;
  560. struct ucma_context *ctx;
  561. struct sockaddr *addr;
  562. int ret = 0;
  563. if (out_len < sizeof(resp))
  564. return -ENOSPC;
  565. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  566. return -EFAULT;
  567. ctx = ucma_get_ctx(file, cmd.id);
  568. if (IS_ERR(ctx))
  569. return PTR_ERR(ctx);
  570. memset(&resp, 0, sizeof resp);
  571. addr = (struct sockaddr *) &ctx->cm_id->route.addr.src_addr;
  572. memcpy(&resp.src_addr, addr, addr->sa_family == AF_INET ?
  573. sizeof(struct sockaddr_in) :
  574. sizeof(struct sockaddr_in6));
  575. addr = (struct sockaddr *) &ctx->cm_id->route.addr.dst_addr;
  576. memcpy(&resp.dst_addr, addr, addr->sa_family == AF_INET ?
  577. sizeof(struct sockaddr_in) :
  578. sizeof(struct sockaddr_in6));
  579. if (!ctx->cm_id->device)
  580. goto out;
  581. resp.node_guid = (__force __u64) ctx->cm_id->device->node_guid;
  582. resp.port_num = ctx->cm_id->port_num;
  583. switch (rdma_node_get_transport(ctx->cm_id->device->node_type)) {
  584. case RDMA_TRANSPORT_IB:
  585. switch (rdma_port_get_link_layer(ctx->cm_id->device,
  586. ctx->cm_id->port_num)) {
  587. case IB_LINK_LAYER_INFINIBAND:
  588. ucma_copy_ib_route(&resp, &ctx->cm_id->route);
  589. break;
  590. case IB_LINK_LAYER_ETHERNET:
  591. ucma_copy_iboe_route(&resp, &ctx->cm_id->route);
  592. break;
  593. default:
  594. break;
  595. }
  596. break;
  597. case RDMA_TRANSPORT_IWARP:
  598. ucma_copy_iw_route(&resp, &ctx->cm_id->route);
  599. break;
  600. default:
  601. break;
  602. }
  603. out:
  604. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  605. &resp, sizeof(resp)))
  606. ret = -EFAULT;
  607. ucma_put_ctx(ctx);
  608. return ret;
  609. }
  610. static void ucma_query_device_addr(struct rdma_cm_id *cm_id,
  611. struct rdma_ucm_query_addr_resp *resp)
  612. {
  613. if (!cm_id->device)
  614. return;
  615. resp->node_guid = (__force __u64) cm_id->device->node_guid;
  616. resp->port_num = cm_id->port_num;
  617. resp->pkey = (__force __u16) cpu_to_be16(
  618. ib_addr_get_pkey(&cm_id->route.addr.dev_addr));
  619. }
  620. static ssize_t ucma_query_addr(struct ucma_context *ctx,
  621. void __user *response, int out_len)
  622. {
  623. struct rdma_ucm_query_addr_resp resp;
  624. struct sockaddr *addr;
  625. int ret = 0;
  626. if (out_len < sizeof(resp))
  627. return -ENOSPC;
  628. memset(&resp, 0, sizeof resp);
  629. addr = (struct sockaddr *) &ctx->cm_id->route.addr.src_addr;
  630. resp.src_size = rdma_addr_size(addr);
  631. memcpy(&resp.src_addr, addr, resp.src_size);
  632. addr = (struct sockaddr *) &ctx->cm_id->route.addr.dst_addr;
  633. resp.dst_size = rdma_addr_size(addr);
  634. memcpy(&resp.dst_addr, addr, resp.dst_size);
  635. ucma_query_device_addr(ctx->cm_id, &resp);
  636. if (copy_to_user(response, &resp, sizeof(resp)))
  637. ret = -EFAULT;
  638. return ret;
  639. }
  640. static ssize_t ucma_query_path(struct ucma_context *ctx,
  641. void __user *response, int out_len)
  642. {
  643. struct rdma_ucm_query_path_resp *resp;
  644. int i, ret = 0;
  645. if (out_len < sizeof(*resp))
  646. return -ENOSPC;
  647. resp = kzalloc(out_len, GFP_KERNEL);
  648. if (!resp)
  649. return -ENOMEM;
  650. resp->num_paths = ctx->cm_id->route.num_paths;
  651. for (i = 0, out_len -= sizeof(*resp);
  652. i < resp->num_paths && out_len > sizeof(struct ib_path_rec_data);
  653. i++, out_len -= sizeof(struct ib_path_rec_data)) {
  654. resp->path_data[i].flags = IB_PATH_GMP | IB_PATH_PRIMARY |
  655. IB_PATH_BIDIRECTIONAL;
  656. ib_sa_pack_path(&ctx->cm_id->route.path_rec[i],
  657. &resp->path_data[i].path_rec);
  658. }
  659. if (copy_to_user(response, resp,
  660. sizeof(*resp) + (i * sizeof(struct ib_path_rec_data))))
  661. ret = -EFAULT;
  662. kfree(resp);
  663. return ret;
  664. }
  665. static ssize_t ucma_query_gid(struct ucma_context *ctx,
  666. void __user *response, int out_len)
  667. {
  668. struct rdma_ucm_query_addr_resp resp;
  669. struct sockaddr_ib *addr;
  670. int ret = 0;
  671. if (out_len < sizeof(resp))
  672. return -ENOSPC;
  673. memset(&resp, 0, sizeof resp);
  674. ucma_query_device_addr(ctx->cm_id, &resp);
  675. addr = (struct sockaddr_ib *) &resp.src_addr;
  676. resp.src_size = sizeof(*addr);
  677. if (ctx->cm_id->route.addr.src_addr.ss_family == AF_IB) {
  678. memcpy(addr, &ctx->cm_id->route.addr.src_addr, resp.src_size);
  679. } else {
  680. addr->sib_family = AF_IB;
  681. addr->sib_pkey = (__force __be16) resp.pkey;
  682. rdma_addr_get_sgid(&ctx->cm_id->route.addr.dev_addr,
  683. (union ib_gid *) &addr->sib_addr);
  684. addr->sib_sid = rdma_get_service_id(ctx->cm_id, (struct sockaddr *)
  685. &ctx->cm_id->route.addr.src_addr);
  686. }
  687. addr = (struct sockaddr_ib *) &resp.dst_addr;
  688. resp.dst_size = sizeof(*addr);
  689. if (ctx->cm_id->route.addr.dst_addr.ss_family == AF_IB) {
  690. memcpy(addr, &ctx->cm_id->route.addr.dst_addr, resp.dst_size);
  691. } else {
  692. addr->sib_family = AF_IB;
  693. addr->sib_pkey = (__force __be16) resp.pkey;
  694. rdma_addr_get_dgid(&ctx->cm_id->route.addr.dev_addr,
  695. (union ib_gid *) &addr->sib_addr);
  696. addr->sib_sid = rdma_get_service_id(ctx->cm_id, (struct sockaddr *)
  697. &ctx->cm_id->route.addr.dst_addr);
  698. }
  699. if (copy_to_user(response, &resp, sizeof(resp)))
  700. ret = -EFAULT;
  701. return ret;
  702. }
  703. static ssize_t ucma_query(struct ucma_file *file,
  704. const char __user *inbuf,
  705. int in_len, int out_len)
  706. {
  707. struct rdma_ucm_query cmd;
  708. struct ucma_context *ctx;
  709. void __user *response;
  710. int ret;
  711. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  712. return -EFAULT;
  713. response = (void __user *)(unsigned long) cmd.response;
  714. ctx = ucma_get_ctx(file, cmd.id);
  715. if (IS_ERR(ctx))
  716. return PTR_ERR(ctx);
  717. switch (cmd.option) {
  718. case RDMA_USER_CM_QUERY_ADDR:
  719. ret = ucma_query_addr(ctx, response, out_len);
  720. break;
  721. case RDMA_USER_CM_QUERY_PATH:
  722. ret = ucma_query_path(ctx, response, out_len);
  723. break;
  724. case RDMA_USER_CM_QUERY_GID:
  725. ret = ucma_query_gid(ctx, response, out_len);
  726. break;
  727. default:
  728. ret = -ENOSYS;
  729. break;
  730. }
  731. ucma_put_ctx(ctx);
  732. return ret;
  733. }
  734. static void ucma_copy_conn_param(struct rdma_cm_id *id,
  735. struct rdma_conn_param *dst,
  736. struct rdma_ucm_conn_param *src)
  737. {
  738. dst->private_data = src->private_data;
  739. dst->private_data_len = src->private_data_len;
  740. dst->responder_resources =src->responder_resources;
  741. dst->initiator_depth = src->initiator_depth;
  742. dst->flow_control = src->flow_control;
  743. dst->retry_count = src->retry_count;
  744. dst->rnr_retry_count = src->rnr_retry_count;
  745. dst->srq = src->srq;
  746. dst->qp_num = src->qp_num;
  747. dst->qkey = (id->route.addr.src_addr.ss_family == AF_IB) ? src->qkey : 0;
  748. }
  749. static ssize_t ucma_connect(struct ucma_file *file, const char __user *inbuf,
  750. int in_len, int out_len)
  751. {
  752. struct rdma_ucm_connect cmd;
  753. struct rdma_conn_param conn_param;
  754. struct ucma_context *ctx;
  755. int ret;
  756. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  757. return -EFAULT;
  758. if (!cmd.conn_param.valid)
  759. return -EINVAL;
  760. ctx = ucma_get_ctx(file, cmd.id);
  761. if (IS_ERR(ctx))
  762. return PTR_ERR(ctx);
  763. ucma_copy_conn_param(ctx->cm_id, &conn_param, &cmd.conn_param);
  764. ret = rdma_connect(ctx->cm_id, &conn_param);
  765. ucma_put_ctx(ctx);
  766. return ret;
  767. }
  768. static ssize_t ucma_listen(struct ucma_file *file, const char __user *inbuf,
  769. int in_len, int out_len)
  770. {
  771. struct rdma_ucm_listen cmd;
  772. struct ucma_context *ctx;
  773. int ret;
  774. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  775. return -EFAULT;
  776. ctx = ucma_get_ctx(file, cmd.id);
  777. if (IS_ERR(ctx))
  778. return PTR_ERR(ctx);
  779. ctx->backlog = cmd.backlog > 0 && cmd.backlog < max_backlog ?
  780. cmd.backlog : max_backlog;
  781. ret = rdma_listen(ctx->cm_id, ctx->backlog);
  782. ucma_put_ctx(ctx);
  783. return ret;
  784. }
  785. static ssize_t ucma_accept(struct ucma_file *file, const char __user *inbuf,
  786. int in_len, int out_len)
  787. {
  788. struct rdma_ucm_accept cmd;
  789. struct rdma_conn_param conn_param;
  790. struct ucma_context *ctx;
  791. int ret;
  792. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  793. return -EFAULT;
  794. ctx = ucma_get_ctx(file, cmd.id);
  795. if (IS_ERR(ctx))
  796. return PTR_ERR(ctx);
  797. if (cmd.conn_param.valid) {
  798. ucma_copy_conn_param(ctx->cm_id, &conn_param, &cmd.conn_param);
  799. mutex_lock(&file->mut);
  800. ret = rdma_accept(ctx->cm_id, &conn_param);
  801. if (!ret)
  802. ctx->uid = cmd.uid;
  803. mutex_unlock(&file->mut);
  804. } else
  805. ret = rdma_accept(ctx->cm_id, NULL);
  806. ucma_put_ctx(ctx);
  807. return ret;
  808. }
  809. static ssize_t ucma_reject(struct ucma_file *file, const char __user *inbuf,
  810. int in_len, int out_len)
  811. {
  812. struct rdma_ucm_reject cmd;
  813. struct ucma_context *ctx;
  814. int ret;
  815. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  816. return -EFAULT;
  817. ctx = ucma_get_ctx(file, cmd.id);
  818. if (IS_ERR(ctx))
  819. return PTR_ERR(ctx);
  820. ret = rdma_reject(ctx->cm_id, cmd.private_data, cmd.private_data_len);
  821. ucma_put_ctx(ctx);
  822. return ret;
  823. }
  824. static ssize_t ucma_disconnect(struct ucma_file *file, const char __user *inbuf,
  825. int in_len, int out_len)
  826. {
  827. struct rdma_ucm_disconnect cmd;
  828. struct ucma_context *ctx;
  829. int ret;
  830. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  831. return -EFAULT;
  832. ctx = ucma_get_ctx(file, cmd.id);
  833. if (IS_ERR(ctx))
  834. return PTR_ERR(ctx);
  835. ret = rdma_disconnect(ctx->cm_id);
  836. ucma_put_ctx(ctx);
  837. return ret;
  838. }
  839. static ssize_t ucma_init_qp_attr(struct ucma_file *file,
  840. const char __user *inbuf,
  841. int in_len, int out_len)
  842. {
  843. struct rdma_ucm_init_qp_attr cmd;
  844. struct ib_uverbs_qp_attr resp;
  845. struct ucma_context *ctx;
  846. struct ib_qp_attr qp_attr;
  847. int ret;
  848. if (out_len < sizeof(resp))
  849. return -ENOSPC;
  850. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  851. return -EFAULT;
  852. ctx = ucma_get_ctx(file, cmd.id);
  853. if (IS_ERR(ctx))
  854. return PTR_ERR(ctx);
  855. resp.qp_attr_mask = 0;
  856. memset(&qp_attr, 0, sizeof qp_attr);
  857. qp_attr.qp_state = cmd.qp_state;
  858. ret = rdma_init_qp_attr(ctx->cm_id, &qp_attr, &resp.qp_attr_mask);
  859. if (ret)
  860. goto out;
  861. ib_copy_qp_attr_to_user(&resp, &qp_attr);
  862. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  863. &resp, sizeof(resp)))
  864. ret = -EFAULT;
  865. out:
  866. ucma_put_ctx(ctx);
  867. return ret;
  868. }
  869. static int ucma_set_option_id(struct ucma_context *ctx, int optname,
  870. void *optval, size_t optlen)
  871. {
  872. int ret = 0;
  873. switch (optname) {
  874. case RDMA_OPTION_ID_TOS:
  875. if (optlen != sizeof(u8)) {
  876. ret = -EINVAL;
  877. break;
  878. }
  879. rdma_set_service_type(ctx->cm_id, *((u8 *) optval));
  880. break;
  881. case RDMA_OPTION_ID_REUSEADDR:
  882. if (optlen != sizeof(int)) {
  883. ret = -EINVAL;
  884. break;
  885. }
  886. ret = rdma_set_reuseaddr(ctx->cm_id, *((int *) optval) ? 1 : 0);
  887. break;
  888. case RDMA_OPTION_ID_AFONLY:
  889. if (optlen != sizeof(int)) {
  890. ret = -EINVAL;
  891. break;
  892. }
  893. ret = rdma_set_afonly(ctx->cm_id, *((int *) optval) ? 1 : 0);
  894. break;
  895. default:
  896. ret = -ENOSYS;
  897. }
  898. return ret;
  899. }
  900. static int ucma_set_ib_path(struct ucma_context *ctx,
  901. struct ib_path_rec_data *path_data, size_t optlen)
  902. {
  903. struct ib_sa_path_rec sa_path;
  904. struct rdma_cm_event event;
  905. int ret;
  906. if (optlen % sizeof(*path_data))
  907. return -EINVAL;
  908. for (; optlen; optlen -= sizeof(*path_data), path_data++) {
  909. if (path_data->flags == (IB_PATH_GMP | IB_PATH_PRIMARY |
  910. IB_PATH_BIDIRECTIONAL))
  911. break;
  912. }
  913. if (!optlen)
  914. return -EINVAL;
  915. ib_sa_unpack_path(path_data->path_rec, &sa_path);
  916. ret = rdma_set_ib_paths(ctx->cm_id, &sa_path, 1);
  917. if (ret)
  918. return ret;
  919. memset(&event, 0, sizeof event);
  920. event.event = RDMA_CM_EVENT_ROUTE_RESOLVED;
  921. return ucma_event_handler(ctx->cm_id, &event);
  922. }
  923. static int ucma_set_option_ib(struct ucma_context *ctx, int optname,
  924. void *optval, size_t optlen)
  925. {
  926. int ret;
  927. switch (optname) {
  928. case RDMA_OPTION_IB_PATH:
  929. ret = ucma_set_ib_path(ctx, optval, optlen);
  930. break;
  931. default:
  932. ret = -ENOSYS;
  933. }
  934. return ret;
  935. }
  936. static int ucma_set_option_level(struct ucma_context *ctx, int level,
  937. int optname, void *optval, size_t optlen)
  938. {
  939. int ret;
  940. switch (level) {
  941. case RDMA_OPTION_ID:
  942. ret = ucma_set_option_id(ctx, optname, optval, optlen);
  943. break;
  944. case RDMA_OPTION_IB:
  945. ret = ucma_set_option_ib(ctx, optname, optval, optlen);
  946. break;
  947. default:
  948. ret = -ENOSYS;
  949. }
  950. return ret;
  951. }
  952. static ssize_t ucma_set_option(struct ucma_file *file, const char __user *inbuf,
  953. int in_len, int out_len)
  954. {
  955. struct rdma_ucm_set_option cmd;
  956. struct ucma_context *ctx;
  957. void *optval;
  958. int ret;
  959. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  960. return -EFAULT;
  961. ctx = ucma_get_ctx(file, cmd.id);
  962. if (IS_ERR(ctx))
  963. return PTR_ERR(ctx);
  964. optval = memdup_user((void __user *) (unsigned long) cmd.optval,
  965. cmd.optlen);
  966. if (IS_ERR(optval)) {
  967. ret = PTR_ERR(optval);
  968. goto out;
  969. }
  970. ret = ucma_set_option_level(ctx, cmd.level, cmd.optname, optval,
  971. cmd.optlen);
  972. kfree(optval);
  973. out:
  974. ucma_put_ctx(ctx);
  975. return ret;
  976. }
  977. static ssize_t ucma_notify(struct ucma_file *file, const char __user *inbuf,
  978. int in_len, int out_len)
  979. {
  980. struct rdma_ucm_notify cmd;
  981. struct ucma_context *ctx;
  982. int ret;
  983. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  984. return -EFAULT;
  985. ctx = ucma_get_ctx(file, cmd.id);
  986. if (IS_ERR(ctx))
  987. return PTR_ERR(ctx);
  988. ret = rdma_notify(ctx->cm_id, (enum ib_event_type) cmd.event);
  989. ucma_put_ctx(ctx);
  990. return ret;
  991. }
  992. static ssize_t ucma_join_multicast(struct ucma_file *file,
  993. const char __user *inbuf,
  994. int in_len, int out_len)
  995. {
  996. struct rdma_ucm_join_mcast cmd;
  997. struct rdma_ucm_create_id_resp resp;
  998. struct ucma_context *ctx;
  999. struct ucma_multicast *mc;
  1000. int ret;
  1001. if (out_len < sizeof(resp))
  1002. return -ENOSPC;
  1003. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  1004. return -EFAULT;
  1005. ctx = ucma_get_ctx(file, cmd.id);
  1006. if (IS_ERR(ctx))
  1007. return PTR_ERR(ctx);
  1008. mutex_lock(&file->mut);
  1009. mc = ucma_alloc_multicast(ctx);
  1010. if (!mc) {
  1011. ret = -ENOMEM;
  1012. goto err1;
  1013. }
  1014. mc->uid = cmd.uid;
  1015. memcpy(&mc->addr, &cmd.addr, sizeof cmd.addr);
  1016. ret = rdma_join_multicast(ctx->cm_id, (struct sockaddr *) &mc->addr, mc);
  1017. if (ret)
  1018. goto err2;
  1019. resp.id = mc->id;
  1020. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  1021. &resp, sizeof(resp))) {
  1022. ret = -EFAULT;
  1023. goto err3;
  1024. }
  1025. mutex_unlock(&file->mut);
  1026. ucma_put_ctx(ctx);
  1027. return 0;
  1028. err3:
  1029. rdma_leave_multicast(ctx->cm_id, (struct sockaddr *) &mc->addr);
  1030. ucma_cleanup_mc_events(mc);
  1031. err2:
  1032. mutex_lock(&mut);
  1033. idr_remove(&multicast_idr, mc->id);
  1034. mutex_unlock(&mut);
  1035. list_del(&mc->list);
  1036. kfree(mc);
  1037. err1:
  1038. mutex_unlock(&file->mut);
  1039. ucma_put_ctx(ctx);
  1040. return ret;
  1041. }
  1042. static ssize_t ucma_leave_multicast(struct ucma_file *file,
  1043. const char __user *inbuf,
  1044. int in_len, int out_len)
  1045. {
  1046. struct rdma_ucm_destroy_id cmd;
  1047. struct rdma_ucm_destroy_id_resp resp;
  1048. struct ucma_multicast *mc;
  1049. int ret = 0;
  1050. if (out_len < sizeof(resp))
  1051. return -ENOSPC;
  1052. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  1053. return -EFAULT;
  1054. mutex_lock(&mut);
  1055. mc = idr_find(&multicast_idr, cmd.id);
  1056. if (!mc)
  1057. mc = ERR_PTR(-ENOENT);
  1058. else if (mc->ctx->file != file)
  1059. mc = ERR_PTR(-EINVAL);
  1060. else {
  1061. idr_remove(&multicast_idr, mc->id);
  1062. atomic_inc(&mc->ctx->ref);
  1063. }
  1064. mutex_unlock(&mut);
  1065. if (IS_ERR(mc)) {
  1066. ret = PTR_ERR(mc);
  1067. goto out;
  1068. }
  1069. rdma_leave_multicast(mc->ctx->cm_id, (struct sockaddr *) &mc->addr);
  1070. mutex_lock(&mc->ctx->file->mut);
  1071. ucma_cleanup_mc_events(mc);
  1072. list_del(&mc->list);
  1073. mutex_unlock(&mc->ctx->file->mut);
  1074. ucma_put_ctx(mc->ctx);
  1075. resp.events_reported = mc->events_reported;
  1076. kfree(mc);
  1077. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  1078. &resp, sizeof(resp)))
  1079. ret = -EFAULT;
  1080. out:
  1081. return ret;
  1082. }
  1083. static void ucma_lock_files(struct ucma_file *file1, struct ucma_file *file2)
  1084. {
  1085. /* Acquire mutex's based on pointer comparison to prevent deadlock. */
  1086. if (file1 < file2) {
  1087. mutex_lock(&file1->mut);
  1088. mutex_lock(&file2->mut);
  1089. } else {
  1090. mutex_lock(&file2->mut);
  1091. mutex_lock(&file1->mut);
  1092. }
  1093. }
  1094. static void ucma_unlock_files(struct ucma_file *file1, struct ucma_file *file2)
  1095. {
  1096. if (file1 < file2) {
  1097. mutex_unlock(&file2->mut);
  1098. mutex_unlock(&file1->mut);
  1099. } else {
  1100. mutex_unlock(&file1->mut);
  1101. mutex_unlock(&file2->mut);
  1102. }
  1103. }
  1104. static void ucma_move_events(struct ucma_context *ctx, struct ucma_file *file)
  1105. {
  1106. struct ucma_event *uevent, *tmp;
  1107. list_for_each_entry_safe(uevent, tmp, &ctx->file->event_list, list)
  1108. if (uevent->ctx == ctx)
  1109. list_move_tail(&uevent->list, &file->event_list);
  1110. }
  1111. static ssize_t ucma_migrate_id(struct ucma_file *new_file,
  1112. const char __user *inbuf,
  1113. int in_len, int out_len)
  1114. {
  1115. struct rdma_ucm_migrate_id cmd;
  1116. struct rdma_ucm_migrate_resp resp;
  1117. struct ucma_context *ctx;
  1118. struct fd f;
  1119. struct ucma_file *cur_file;
  1120. int ret = 0;
  1121. if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
  1122. return -EFAULT;
  1123. /* Get current fd to protect against it being closed */
  1124. f = fdget(cmd.fd);
  1125. if (!f.file)
  1126. return -ENOENT;
  1127. /* Validate current fd and prevent destruction of id. */
  1128. ctx = ucma_get_ctx(f.file->private_data, cmd.id);
  1129. if (IS_ERR(ctx)) {
  1130. ret = PTR_ERR(ctx);
  1131. goto file_put;
  1132. }
  1133. cur_file = ctx->file;
  1134. if (cur_file == new_file) {
  1135. resp.events_reported = ctx->events_reported;
  1136. goto response;
  1137. }
  1138. /*
  1139. * Migrate events between fd's, maintaining order, and avoiding new
  1140. * events being added before existing events.
  1141. */
  1142. ucma_lock_files(cur_file, new_file);
  1143. mutex_lock(&mut);
  1144. list_move_tail(&ctx->list, &new_file->ctx_list);
  1145. ucma_move_events(ctx, new_file);
  1146. ctx->file = new_file;
  1147. resp.events_reported = ctx->events_reported;
  1148. mutex_unlock(&mut);
  1149. ucma_unlock_files(cur_file, new_file);
  1150. response:
  1151. if (copy_to_user((void __user *)(unsigned long)cmd.response,
  1152. &resp, sizeof(resp)))
  1153. ret = -EFAULT;
  1154. ucma_put_ctx(ctx);
  1155. file_put:
  1156. fdput(f);
  1157. return ret;
  1158. }
  1159. static ssize_t (*ucma_cmd_table[])(struct ucma_file *file,
  1160. const char __user *inbuf,
  1161. int in_len, int out_len) = {
  1162. [RDMA_USER_CM_CMD_CREATE_ID] = ucma_create_id,
  1163. [RDMA_USER_CM_CMD_DESTROY_ID] = ucma_destroy_id,
  1164. [RDMA_USER_CM_CMD_BIND_ADDR] = ucma_bind_addr,
  1165. [RDMA_USER_CM_CMD_RESOLVE_ADDR] = ucma_resolve_addr,
  1166. [RDMA_USER_CM_CMD_RESOLVE_ROUTE]= ucma_resolve_route,
  1167. [RDMA_USER_CM_CMD_QUERY_ROUTE] = ucma_query_route,
  1168. [RDMA_USER_CM_CMD_CONNECT] = ucma_connect,
  1169. [RDMA_USER_CM_CMD_LISTEN] = ucma_listen,
  1170. [RDMA_USER_CM_CMD_ACCEPT] = ucma_accept,
  1171. [RDMA_USER_CM_CMD_REJECT] = ucma_reject,
  1172. [RDMA_USER_CM_CMD_DISCONNECT] = ucma_disconnect,
  1173. [RDMA_USER_CM_CMD_INIT_QP_ATTR] = ucma_init_qp_attr,
  1174. [RDMA_USER_CM_CMD_GET_EVENT] = ucma_get_event,
  1175. [RDMA_USER_CM_CMD_GET_OPTION] = NULL,
  1176. [RDMA_USER_CM_CMD_SET_OPTION] = ucma_set_option,
  1177. [RDMA_USER_CM_CMD_NOTIFY] = ucma_notify,
  1178. [RDMA_USER_CM_CMD_JOIN_MCAST] = ucma_join_multicast,
  1179. [RDMA_USER_CM_CMD_LEAVE_MCAST] = ucma_leave_multicast,
  1180. [RDMA_USER_CM_CMD_MIGRATE_ID] = ucma_migrate_id,
  1181. [RDMA_USER_CM_CMD_QUERY] = ucma_query
  1182. };
  1183. static ssize_t ucma_write(struct file *filp, const char __user *buf,
  1184. size_t len, loff_t *pos)
  1185. {
  1186. struct ucma_file *file = filp->private_data;
  1187. struct rdma_ucm_cmd_hdr hdr;
  1188. ssize_t ret;
  1189. if (len < sizeof(hdr))
  1190. return -EINVAL;
  1191. if (copy_from_user(&hdr, buf, sizeof(hdr)))
  1192. return -EFAULT;
  1193. if (hdr.cmd >= ARRAY_SIZE(ucma_cmd_table))
  1194. return -EINVAL;
  1195. if (hdr.in + sizeof(hdr) > len)
  1196. return -EINVAL;
  1197. if (!ucma_cmd_table[hdr.cmd])
  1198. return -ENOSYS;
  1199. ret = ucma_cmd_table[hdr.cmd](file, buf + sizeof(hdr), hdr.in, hdr.out);
  1200. if (!ret)
  1201. ret = len;
  1202. return ret;
  1203. }
  1204. static unsigned int ucma_poll(struct file *filp, struct poll_table_struct *wait)
  1205. {
  1206. struct ucma_file *file = filp->private_data;
  1207. unsigned int mask = 0;
  1208. poll_wait(filp, &file->poll_wait, wait);
  1209. if (!list_empty(&file->event_list))
  1210. mask = POLLIN | POLLRDNORM;
  1211. return mask;
  1212. }
  1213. /*
  1214. * ucma_open() does not need the BKL:
  1215. *
  1216. * - no global state is referred to;
  1217. * - there is no ioctl method to race against;
  1218. * - no further module initialization is required for open to work
  1219. * after the device is registered.
  1220. */
  1221. static int ucma_open(struct inode *inode, struct file *filp)
  1222. {
  1223. struct ucma_file *file;
  1224. file = kmalloc(sizeof *file, GFP_KERNEL);
  1225. if (!file)
  1226. return -ENOMEM;
  1227. INIT_LIST_HEAD(&file->event_list);
  1228. INIT_LIST_HEAD(&file->ctx_list);
  1229. init_waitqueue_head(&file->poll_wait);
  1230. mutex_init(&file->mut);
  1231. filp->private_data = file;
  1232. file->filp = filp;
  1233. return nonseekable_open(inode, filp);
  1234. }
  1235. static int ucma_close(struct inode *inode, struct file *filp)
  1236. {
  1237. struct ucma_file *file = filp->private_data;
  1238. struct ucma_context *ctx, *tmp;
  1239. mutex_lock(&file->mut);
  1240. list_for_each_entry_safe(ctx, tmp, &file->ctx_list, list) {
  1241. mutex_unlock(&file->mut);
  1242. mutex_lock(&mut);
  1243. idr_remove(&ctx_idr, ctx->id);
  1244. mutex_unlock(&mut);
  1245. ucma_free_ctx(ctx);
  1246. mutex_lock(&file->mut);
  1247. }
  1248. mutex_unlock(&file->mut);
  1249. kfree(file);
  1250. return 0;
  1251. }
  1252. static const struct file_operations ucma_fops = {
  1253. .owner = THIS_MODULE,
  1254. .open = ucma_open,
  1255. .release = ucma_close,
  1256. .write = ucma_write,
  1257. .poll = ucma_poll,
  1258. .llseek = no_llseek,
  1259. };
  1260. static struct miscdevice ucma_misc = {
  1261. .minor = MISC_DYNAMIC_MINOR,
  1262. .name = "rdma_cm",
  1263. .nodename = "infiniband/rdma_cm",
  1264. .mode = 0666,
  1265. .fops = &ucma_fops,
  1266. };
  1267. static ssize_t show_abi_version(struct device *dev,
  1268. struct device_attribute *attr,
  1269. char *buf)
  1270. {
  1271. return sprintf(buf, "%d\n", RDMA_USER_CM_ABI_VERSION);
  1272. }
  1273. static DEVICE_ATTR(abi_version, S_IRUGO, show_abi_version, NULL);
  1274. static int __init ucma_init(void)
  1275. {
  1276. int ret;
  1277. ret = misc_register(&ucma_misc);
  1278. if (ret)
  1279. return ret;
  1280. ret = device_create_file(ucma_misc.this_device, &dev_attr_abi_version);
  1281. if (ret) {
  1282. printk(KERN_ERR "rdma_ucm: couldn't create abi_version attr\n");
  1283. goto err1;
  1284. }
  1285. ucma_ctl_table_hdr = register_net_sysctl(&init_net, "net/rdma_ucm", ucma_ctl_table);
  1286. if (!ucma_ctl_table_hdr) {
  1287. printk(KERN_ERR "rdma_ucm: couldn't register sysctl paths\n");
  1288. ret = -ENOMEM;
  1289. goto err2;
  1290. }
  1291. return 0;
  1292. err2:
  1293. device_remove_file(ucma_misc.this_device, &dev_attr_abi_version);
  1294. err1:
  1295. misc_deregister(&ucma_misc);
  1296. return ret;
  1297. }
  1298. static void __exit ucma_cleanup(void)
  1299. {
  1300. unregister_net_sysctl_table(ucma_ctl_table_hdr);
  1301. device_remove_file(ucma_misc.this_device, &dev_attr_abi_version);
  1302. misc_deregister(&ucma_misc);
  1303. idr_destroy(&ctx_idr);
  1304. }
  1305. module_init(ucma_init);
  1306. module_exit(ucma_cleanup);