2 * Copyright (c) 2005 Topspin Communications. All rights reserved.
3 * Copyright (c) 2005 Intel Corporation. All rights reserved.
5 * This software is available to you under a choice of one of two
6 * licenses. You may choose to be licensed under the terms of the GNU
7 * General Public License (GPL) Version 2, available from the file
8 * COPYING in the main directory of this source tree, or the
9 * OpenIB.org BSD license below:
11 * Redistribution and use in source and binary forms, with or
12 * without modification, are permitted provided that the following
15 * - Redistributions of source code must retain the above
16 * copyright notice, this list of conditions and the following
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
22 * provided with the distribution.
24 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
25 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
26 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
27 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
28 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
29 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
30 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
34 #include <linux/completion.h>
35 #include <linux/init.h>
37 #include <linux/module.h>
38 #include <linux/device.h>
39 #include <linux/err.h>
40 #include <linux/poll.h>
41 #include <linux/sched.h>
42 #include <linux/file.h>
43 #include <linux/mount.h>
44 #include <linux/cdev.h>
45 #include <linux/idr.h>
46 #include <linux/mutex.h>
48 #include <asm/uaccess.h>
50 #include <rdma/ib_cm.h>
51 #include <rdma/ib_user_cm.h>
52 #include <rdma/ib_marshall.h>
54 MODULE_AUTHOR("Libor Michalek");
55 MODULE_DESCRIPTION("InfiniBand userspace Connection Manager access");
56 MODULE_LICENSE("Dual BSD/GPL");
58 struct ib_ucm_device
{
62 struct ib_device
*ib_dev
;
66 struct mutex file_mutex
;
68 struct ib_ucm_device
*device
;
70 struct list_head ctxs
;
71 struct list_head events
;
72 wait_queue_head_t poll_wait
;
75 struct ib_ucm_context
{
77 struct completion comp
;
81 struct ib_ucm_file
*file
;
82 struct ib_cm_id
*cm_id
;
85 struct list_head events
; /* list of pending events. */
86 struct list_head file_list
; /* member in file ctx list */
90 struct ib_ucm_context
*ctx
;
91 struct list_head file_list
; /* member in file event list */
92 struct list_head ctx_list
; /* member in ctx event list */
94 struct ib_cm_id
*cm_id
;
95 struct ib_ucm_event_resp resp
;
104 IB_UCM_BASE_MINOR
= 224,
105 IB_UCM_MAX_DEVICES
= 32
108 /* ib_cm and ib_user_cm modules share /sys/class/infiniband_cm */
109 extern struct class cm_class
;
111 #define IB_UCM_BASE_DEV MKDEV(IB_UCM_MAJOR, IB_UCM_BASE_MINOR)
113 static void ib_ucm_add_one(struct ib_device
*device
);
114 static void ib_ucm_remove_one(struct ib_device
*device
);
116 static struct ib_client ucm_client
= {
118 .add
= ib_ucm_add_one
,
119 .remove
= ib_ucm_remove_one
122 static DEFINE_MUTEX(ctx_id_mutex
);
123 static DEFINE_IDR(ctx_id_table
);
124 static DECLARE_BITMAP(dev_map
, IB_UCM_MAX_DEVICES
);
126 static struct ib_ucm_context
*ib_ucm_ctx_get(struct ib_ucm_file
*file
, int id
)
128 struct ib_ucm_context
*ctx
;
130 mutex_lock(&ctx_id_mutex
);
131 ctx
= idr_find(&ctx_id_table
, id
);
133 ctx
= ERR_PTR(-ENOENT
);
134 else if (ctx
->file
!= file
)
135 ctx
= ERR_PTR(-EINVAL
);
137 atomic_inc(&ctx
->ref
);
138 mutex_unlock(&ctx_id_mutex
);
143 static void ib_ucm_ctx_put(struct ib_ucm_context
*ctx
)
145 if (atomic_dec_and_test(&ctx
->ref
))
146 complete(&ctx
->comp
);
149 static inline int ib_ucm_new_cm_id(int event
)
151 return event
== IB_CM_REQ_RECEIVED
|| event
== IB_CM_SIDR_REQ_RECEIVED
;
154 static void ib_ucm_cleanup_events(struct ib_ucm_context
*ctx
)
156 struct ib_ucm_event
*uevent
;
158 mutex_lock(&ctx
->file
->file_mutex
);
159 list_del(&ctx
->file_list
);
160 while (!list_empty(&ctx
->events
)) {
162 uevent
= list_entry(ctx
->events
.next
,
163 struct ib_ucm_event
, ctx_list
);
164 list_del(&uevent
->file_list
);
165 list_del(&uevent
->ctx_list
);
166 mutex_unlock(&ctx
->file
->file_mutex
);
168 /* clear incoming connections. */
169 if (ib_ucm_new_cm_id(uevent
->resp
.event
))
170 ib_destroy_cm_id(uevent
->cm_id
);
173 mutex_lock(&ctx
->file
->file_mutex
);
175 mutex_unlock(&ctx
->file
->file_mutex
);
178 static struct ib_ucm_context
*ib_ucm_ctx_alloc(struct ib_ucm_file
*file
)
180 struct ib_ucm_context
*ctx
;
183 ctx
= kzalloc(sizeof *ctx
, GFP_KERNEL
);
187 atomic_set(&ctx
->ref
, 1);
188 init_completion(&ctx
->comp
);
190 INIT_LIST_HEAD(&ctx
->events
);
193 result
= idr_pre_get(&ctx_id_table
, GFP_KERNEL
);
197 mutex_lock(&ctx_id_mutex
);
198 result
= idr_get_new(&ctx_id_table
, ctx
, &ctx
->id
);
199 mutex_unlock(&ctx_id_mutex
);
200 } while (result
== -EAGAIN
);
205 list_add_tail(&ctx
->file_list
, &file
->ctxs
);
213 static void ib_ucm_event_req_get(struct ib_ucm_req_event_resp
*ureq
,
214 struct ib_cm_req_event_param
*kreq
)
216 ureq
->remote_ca_guid
= kreq
->remote_ca_guid
;
217 ureq
->remote_qkey
= kreq
->remote_qkey
;
218 ureq
->remote_qpn
= kreq
->remote_qpn
;
219 ureq
->qp_type
= kreq
->qp_type
;
220 ureq
->starting_psn
= kreq
->starting_psn
;
221 ureq
->responder_resources
= kreq
->responder_resources
;
222 ureq
->initiator_depth
= kreq
->initiator_depth
;
223 ureq
->local_cm_response_timeout
= kreq
->local_cm_response_timeout
;
224 ureq
->flow_control
= kreq
->flow_control
;
225 ureq
->remote_cm_response_timeout
= kreq
->remote_cm_response_timeout
;
226 ureq
->retry_count
= kreq
->retry_count
;
227 ureq
->rnr_retry_count
= kreq
->rnr_retry_count
;
228 ureq
->srq
= kreq
->srq
;
229 ureq
->port
= kreq
->port
;
231 ib_copy_path_rec_to_user(&ureq
->primary_path
, kreq
->primary_path
);
232 if (kreq
->alternate_path
)
233 ib_copy_path_rec_to_user(&ureq
->alternate_path
,
234 kreq
->alternate_path
);
237 static void ib_ucm_event_rep_get(struct ib_ucm_rep_event_resp
*urep
,
238 struct ib_cm_rep_event_param
*krep
)
240 urep
->remote_ca_guid
= krep
->remote_ca_guid
;
241 urep
->remote_qkey
= krep
->remote_qkey
;
242 urep
->remote_qpn
= krep
->remote_qpn
;
243 urep
->starting_psn
= krep
->starting_psn
;
244 urep
->responder_resources
= krep
->responder_resources
;
245 urep
->initiator_depth
= krep
->initiator_depth
;
246 urep
->target_ack_delay
= krep
->target_ack_delay
;
247 urep
->failover_accepted
= krep
->failover_accepted
;
248 urep
->flow_control
= krep
->flow_control
;
249 urep
->rnr_retry_count
= krep
->rnr_retry_count
;
250 urep
->srq
= krep
->srq
;
253 static void ib_ucm_event_sidr_rep_get(struct ib_ucm_sidr_rep_event_resp
*urep
,
254 struct ib_cm_sidr_rep_event_param
*krep
)
256 urep
->status
= krep
->status
;
257 urep
->qkey
= krep
->qkey
;
258 urep
->qpn
= krep
->qpn
;
261 static int ib_ucm_event_process(struct ib_cm_event
*evt
,
262 struct ib_ucm_event
*uvt
)
266 switch (evt
->event
) {
267 case IB_CM_REQ_RECEIVED
:
268 ib_ucm_event_req_get(&uvt
->resp
.u
.req_resp
,
269 &evt
->param
.req_rcvd
);
270 uvt
->data_len
= IB_CM_REQ_PRIVATE_DATA_SIZE
;
271 uvt
->resp
.present
= IB_UCM_PRES_PRIMARY
;
272 uvt
->resp
.present
|= (evt
->param
.req_rcvd
.alternate_path
?
273 IB_UCM_PRES_ALTERNATE
: 0);
275 case IB_CM_REP_RECEIVED
:
276 ib_ucm_event_rep_get(&uvt
->resp
.u
.rep_resp
,
277 &evt
->param
.rep_rcvd
);
278 uvt
->data_len
= IB_CM_REP_PRIVATE_DATA_SIZE
;
280 case IB_CM_RTU_RECEIVED
:
281 uvt
->data_len
= IB_CM_RTU_PRIVATE_DATA_SIZE
;
282 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
284 case IB_CM_DREQ_RECEIVED
:
285 uvt
->data_len
= IB_CM_DREQ_PRIVATE_DATA_SIZE
;
286 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
288 case IB_CM_DREP_RECEIVED
:
289 uvt
->data_len
= IB_CM_DREP_PRIVATE_DATA_SIZE
;
290 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
292 case IB_CM_MRA_RECEIVED
:
293 uvt
->resp
.u
.mra_resp
.timeout
=
294 evt
->param
.mra_rcvd
.service_timeout
;
295 uvt
->data_len
= IB_CM_MRA_PRIVATE_DATA_SIZE
;
297 case IB_CM_REJ_RECEIVED
:
298 uvt
->resp
.u
.rej_resp
.reason
= evt
->param
.rej_rcvd
.reason
;
299 uvt
->data_len
= IB_CM_REJ_PRIVATE_DATA_SIZE
;
300 uvt
->info_len
= evt
->param
.rej_rcvd
.ari_length
;
301 info
= evt
->param
.rej_rcvd
.ari
;
303 case IB_CM_LAP_RECEIVED
:
304 ib_copy_path_rec_to_user(&uvt
->resp
.u
.lap_resp
.path
,
305 evt
->param
.lap_rcvd
.alternate_path
);
306 uvt
->data_len
= IB_CM_LAP_PRIVATE_DATA_SIZE
;
307 uvt
->resp
.present
= IB_UCM_PRES_ALTERNATE
;
309 case IB_CM_APR_RECEIVED
:
310 uvt
->resp
.u
.apr_resp
.status
= evt
->param
.apr_rcvd
.ap_status
;
311 uvt
->data_len
= IB_CM_APR_PRIVATE_DATA_SIZE
;
312 uvt
->info_len
= evt
->param
.apr_rcvd
.info_len
;
313 info
= evt
->param
.apr_rcvd
.apr_info
;
315 case IB_CM_SIDR_REQ_RECEIVED
:
316 uvt
->resp
.u
.sidr_req_resp
.pkey
=
317 evt
->param
.sidr_req_rcvd
.pkey
;
318 uvt
->resp
.u
.sidr_req_resp
.port
=
319 evt
->param
.sidr_req_rcvd
.port
;
320 uvt
->data_len
= IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE
;
322 case IB_CM_SIDR_REP_RECEIVED
:
323 ib_ucm_event_sidr_rep_get(&uvt
->resp
.u
.sidr_rep_resp
,
324 &evt
->param
.sidr_rep_rcvd
);
325 uvt
->data_len
= IB_CM_SIDR_REP_PRIVATE_DATA_SIZE
;
326 uvt
->info_len
= evt
->param
.sidr_rep_rcvd
.info_len
;
327 info
= evt
->param
.sidr_rep_rcvd
.info
;
330 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
335 uvt
->data
= kmemdup(evt
->private_data
, uvt
->data_len
, GFP_KERNEL
);
339 uvt
->resp
.present
|= IB_UCM_PRES_DATA
;
343 uvt
->info
= kmemdup(info
, uvt
->info_len
, GFP_KERNEL
);
347 uvt
->resp
.present
|= IB_UCM_PRES_INFO
;
357 static int ib_ucm_event_handler(struct ib_cm_id
*cm_id
,
358 struct ib_cm_event
*event
)
360 struct ib_ucm_event
*uevent
;
361 struct ib_ucm_context
*ctx
;
364 ctx
= cm_id
->context
;
366 uevent
= kzalloc(sizeof *uevent
, GFP_KERNEL
);
371 uevent
->cm_id
= cm_id
;
372 uevent
->resp
.uid
= ctx
->uid
;
373 uevent
->resp
.id
= ctx
->id
;
374 uevent
->resp
.event
= event
->event
;
376 result
= ib_ucm_event_process(event
, uevent
);
380 mutex_lock(&ctx
->file
->file_mutex
);
381 list_add_tail(&uevent
->file_list
, &ctx
->file
->events
);
382 list_add_tail(&uevent
->ctx_list
, &ctx
->events
);
383 wake_up_interruptible(&ctx
->file
->poll_wait
);
384 mutex_unlock(&ctx
->file
->file_mutex
);
390 /* Destroy new cm_id's */
391 return ib_ucm_new_cm_id(event
->event
);
394 static ssize_t
ib_ucm_event(struct ib_ucm_file
*file
,
395 const char __user
*inbuf
,
396 int in_len
, int out_len
)
398 struct ib_ucm_context
*ctx
;
399 struct ib_ucm_event_get cmd
;
400 struct ib_ucm_event
*uevent
;
404 if (out_len
< sizeof(struct ib_ucm_event_resp
))
407 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
410 mutex_lock(&file
->file_mutex
);
411 while (list_empty(&file
->events
)) {
412 mutex_unlock(&file
->file_mutex
);
414 if (file
->filp
->f_flags
& O_NONBLOCK
)
417 if (wait_event_interruptible(file
->poll_wait
,
418 !list_empty(&file
->events
)))
421 mutex_lock(&file
->file_mutex
);
424 uevent
= list_entry(file
->events
.next
, struct ib_ucm_event
, file_list
);
426 if (ib_ucm_new_cm_id(uevent
->resp
.event
)) {
427 ctx
= ib_ucm_ctx_alloc(file
);
433 ctx
->cm_id
= uevent
->cm_id
;
434 ctx
->cm_id
->context
= ctx
;
435 uevent
->resp
.id
= ctx
->id
;
438 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
439 &uevent
->resp
, sizeof(uevent
->resp
))) {
445 if (cmd
.data_len
< uevent
->data_len
) {
449 if (copy_to_user((void __user
*)(unsigned long)cmd
.data
,
450 uevent
->data
, uevent
->data_len
)) {
457 if (cmd
.info_len
< uevent
->info_len
) {
461 if (copy_to_user((void __user
*)(unsigned long)cmd
.info
,
462 uevent
->info
, uevent
->info_len
)) {
468 list_del(&uevent
->file_list
);
469 list_del(&uevent
->ctx_list
);
470 uevent
->ctx
->events_reported
++;
476 mutex_unlock(&file
->file_mutex
);
480 static ssize_t
ib_ucm_create_id(struct ib_ucm_file
*file
,
481 const char __user
*inbuf
,
482 int in_len
, int out_len
)
484 struct ib_ucm_create_id cmd
;
485 struct ib_ucm_create_id_resp resp
;
486 struct ib_ucm_context
*ctx
;
489 if (out_len
< sizeof(resp
))
492 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
495 mutex_lock(&file
->file_mutex
);
496 ctx
= ib_ucm_ctx_alloc(file
);
497 mutex_unlock(&file
->file_mutex
);
502 ctx
->cm_id
= ib_create_cm_id(file
->device
->ib_dev
,
503 ib_ucm_event_handler
, ctx
);
504 if (IS_ERR(ctx
->cm_id
)) {
505 result
= PTR_ERR(ctx
->cm_id
);
510 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
511 &resp
, sizeof(resp
))) {
518 ib_destroy_cm_id(ctx
->cm_id
);
520 mutex_lock(&ctx_id_mutex
);
521 idr_remove(&ctx_id_table
, ctx
->id
);
522 mutex_unlock(&ctx_id_mutex
);
527 static ssize_t
ib_ucm_destroy_id(struct ib_ucm_file
*file
,
528 const char __user
*inbuf
,
529 int in_len
, int out_len
)
531 struct ib_ucm_destroy_id cmd
;
532 struct ib_ucm_destroy_id_resp resp
;
533 struct ib_ucm_context
*ctx
;
536 if (out_len
< sizeof(resp
))
539 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
542 mutex_lock(&ctx_id_mutex
);
543 ctx
= idr_find(&ctx_id_table
, cmd
.id
);
545 ctx
= ERR_PTR(-ENOENT
);
546 else if (ctx
->file
!= file
)
547 ctx
= ERR_PTR(-EINVAL
);
549 idr_remove(&ctx_id_table
, ctx
->id
);
550 mutex_unlock(&ctx_id_mutex
);
556 wait_for_completion(&ctx
->comp
);
558 /* No new events will be generated after destroying the cm_id. */
559 ib_destroy_cm_id(ctx
->cm_id
);
560 /* Cleanup events not yet reported to the user. */
561 ib_ucm_cleanup_events(ctx
);
563 resp
.events_reported
= ctx
->events_reported
;
564 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
565 &resp
, sizeof(resp
)))
572 static ssize_t
ib_ucm_attr_id(struct ib_ucm_file
*file
,
573 const char __user
*inbuf
,
574 int in_len
, int out_len
)
576 struct ib_ucm_attr_id_resp resp
;
577 struct ib_ucm_attr_id cmd
;
578 struct ib_ucm_context
*ctx
;
581 if (out_len
< sizeof(resp
))
584 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
587 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
591 resp
.service_id
= ctx
->cm_id
->service_id
;
592 resp
.service_mask
= ctx
->cm_id
->service_mask
;
593 resp
.local_id
= ctx
->cm_id
->local_id
;
594 resp
.remote_id
= ctx
->cm_id
->remote_id
;
596 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
597 &resp
, sizeof(resp
)))
604 static ssize_t
ib_ucm_init_qp_attr(struct ib_ucm_file
*file
,
605 const char __user
*inbuf
,
606 int in_len
, int out_len
)
608 struct ib_uverbs_qp_attr resp
;
609 struct ib_ucm_init_qp_attr cmd
;
610 struct ib_ucm_context
*ctx
;
611 struct ib_qp_attr qp_attr
;
614 if (out_len
< sizeof(resp
))
617 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
620 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
624 resp
.qp_attr_mask
= 0;
625 memset(&qp_attr
, 0, sizeof qp_attr
);
626 qp_attr
.qp_state
= cmd
.qp_state
;
627 result
= ib_cm_init_qp_attr(ctx
->cm_id
, &qp_attr
, &resp
.qp_attr_mask
);
631 ib_copy_qp_attr_to_user(&resp
, &qp_attr
);
633 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
634 &resp
, sizeof(resp
)))
642 static int ucm_validate_listen(__be64 service_id
, __be64 service_mask
)
644 service_id
&= service_mask
;
646 if (((service_id
& IB_CMA_SERVICE_ID_MASK
) == IB_CMA_SERVICE_ID
) ||
647 ((service_id
& IB_SDP_SERVICE_ID_MASK
) == IB_SDP_SERVICE_ID
))
653 static ssize_t
ib_ucm_listen(struct ib_ucm_file
*file
,
654 const char __user
*inbuf
,
655 int in_len
, int out_len
)
657 struct ib_ucm_listen cmd
;
658 struct ib_ucm_context
*ctx
;
661 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
664 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
668 result
= ucm_validate_listen(cmd
.service_id
, cmd
.service_mask
);
672 result
= ib_cm_listen(ctx
->cm_id
, cmd
.service_id
, cmd
.service_mask
,
679 static ssize_t
ib_ucm_notify(struct ib_ucm_file
*file
,
680 const char __user
*inbuf
,
681 int in_len
, int out_len
)
683 struct ib_ucm_notify cmd
;
684 struct ib_ucm_context
*ctx
;
687 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
690 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
694 result
= ib_cm_notify(ctx
->cm_id
, (enum ib_event_type
) cmd
.event
);
699 static int ib_ucm_alloc_data(const void **dest
, u64 src
, u32 len
)
708 data
= kmalloc(len
, GFP_KERNEL
);
712 if (copy_from_user(data
, (void __user
*)(unsigned long)src
, len
)) {
721 static int ib_ucm_path_get(struct ib_sa_path_rec
**path
, u64 src
)
723 struct ib_user_path_rec upath
;
724 struct ib_sa_path_rec
*sa_path
;
731 sa_path
= kmalloc(sizeof(*sa_path
), GFP_KERNEL
);
735 if (copy_from_user(&upath
, (void __user
*)(unsigned long)src
,
742 ib_copy_path_rec_from_user(sa_path
, &upath
);
747 static ssize_t
ib_ucm_send_req(struct ib_ucm_file
*file
,
748 const char __user
*inbuf
,
749 int in_len
, int out_len
)
751 struct ib_cm_req_param param
;
752 struct ib_ucm_context
*ctx
;
753 struct ib_ucm_req cmd
;
756 param
.private_data
= NULL
;
757 param
.primary_path
= NULL
;
758 param
.alternate_path
= NULL
;
760 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
763 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
767 result
= ib_ucm_path_get(¶m
.primary_path
, cmd
.primary_path
);
771 result
= ib_ucm_path_get(¶m
.alternate_path
, cmd
.alternate_path
);
775 param
.private_data_len
= cmd
.len
;
776 param
.service_id
= cmd
.sid
;
777 param
.qp_num
= cmd
.qpn
;
778 param
.qp_type
= cmd
.qp_type
;
779 param
.starting_psn
= cmd
.psn
;
780 param
.peer_to_peer
= cmd
.peer_to_peer
;
781 param
.responder_resources
= cmd
.responder_resources
;
782 param
.initiator_depth
= cmd
.initiator_depth
;
783 param
.remote_cm_response_timeout
= cmd
.remote_cm_response_timeout
;
784 param
.flow_control
= cmd
.flow_control
;
785 param
.local_cm_response_timeout
= cmd
.local_cm_response_timeout
;
786 param
.retry_count
= cmd
.retry_count
;
787 param
.rnr_retry_count
= cmd
.rnr_retry_count
;
788 param
.max_cm_retries
= cmd
.max_cm_retries
;
791 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
793 result
= ib_send_cm_req(ctx
->cm_id
, ¶m
);
796 result
= PTR_ERR(ctx
);
799 kfree(param
.private_data
);
800 kfree(param
.primary_path
);
801 kfree(param
.alternate_path
);
805 static ssize_t
ib_ucm_send_rep(struct ib_ucm_file
*file
,
806 const char __user
*inbuf
,
807 int in_len
, int out_len
)
809 struct ib_cm_rep_param param
;
810 struct ib_ucm_context
*ctx
;
811 struct ib_ucm_rep cmd
;
814 param
.private_data
= NULL
;
816 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
819 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
823 param
.qp_num
= cmd
.qpn
;
824 param
.starting_psn
= cmd
.psn
;
825 param
.private_data_len
= cmd
.len
;
826 param
.responder_resources
= cmd
.responder_resources
;
827 param
.initiator_depth
= cmd
.initiator_depth
;
828 param
.failover_accepted
= cmd
.failover_accepted
;
829 param
.flow_control
= cmd
.flow_control
;
830 param
.rnr_retry_count
= cmd
.rnr_retry_count
;
833 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
836 result
= ib_send_cm_rep(ctx
->cm_id
, ¶m
);
839 result
= PTR_ERR(ctx
);
841 kfree(param
.private_data
);
845 static ssize_t
ib_ucm_send_private_data(struct ib_ucm_file
*file
,
846 const char __user
*inbuf
, int in_len
,
847 int (*func
)(struct ib_cm_id
*cm_id
,
848 const void *private_data
,
849 u8 private_data_len
))
851 struct ib_ucm_private_data cmd
;
852 struct ib_ucm_context
*ctx
;
853 const void *private_data
= NULL
;
856 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
859 result
= ib_ucm_alloc_data(&private_data
, cmd
.data
, cmd
.len
);
863 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
865 result
= func(ctx
->cm_id
, private_data
, cmd
.len
);
868 result
= PTR_ERR(ctx
);
874 static ssize_t
ib_ucm_send_rtu(struct ib_ucm_file
*file
,
875 const char __user
*inbuf
,
876 int in_len
, int out_len
)
878 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_rtu
);
881 static ssize_t
ib_ucm_send_dreq(struct ib_ucm_file
*file
,
882 const char __user
*inbuf
,
883 int in_len
, int out_len
)
885 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_dreq
);
888 static ssize_t
ib_ucm_send_drep(struct ib_ucm_file
*file
,
889 const char __user
*inbuf
,
890 int in_len
, int out_len
)
892 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_drep
);
895 static ssize_t
ib_ucm_send_info(struct ib_ucm_file
*file
,
896 const char __user
*inbuf
, int in_len
,
897 int (*func
)(struct ib_cm_id
*cm_id
,
904 struct ib_ucm_context
*ctx
;
905 struct ib_ucm_info cmd
;
906 const void *data
= NULL
;
907 const void *info
= NULL
;
910 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
913 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.data_len
);
917 result
= ib_ucm_alloc_data(&info
, cmd
.info
, cmd
.info_len
);
921 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
923 result
= func(ctx
->cm_id
, cmd
.status
, info
, cmd
.info_len
,
927 result
= PTR_ERR(ctx
);
935 static ssize_t
ib_ucm_send_rej(struct ib_ucm_file
*file
,
936 const char __user
*inbuf
,
937 int in_len
, int out_len
)
939 return ib_ucm_send_info(file
, inbuf
, in_len
, (void *)ib_send_cm_rej
);
942 static ssize_t
ib_ucm_send_apr(struct ib_ucm_file
*file
,
943 const char __user
*inbuf
,
944 int in_len
, int out_len
)
946 return ib_ucm_send_info(file
, inbuf
, in_len
, (void *)ib_send_cm_apr
);
949 static ssize_t
ib_ucm_send_mra(struct ib_ucm_file
*file
,
950 const char __user
*inbuf
,
951 int in_len
, int out_len
)
953 struct ib_ucm_context
*ctx
;
954 struct ib_ucm_mra cmd
;
955 const void *data
= NULL
;
958 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
961 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.len
);
965 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
967 result
= ib_send_cm_mra(ctx
->cm_id
, cmd
.timeout
, data
, cmd
.len
);
970 result
= PTR_ERR(ctx
);
976 static ssize_t
ib_ucm_send_lap(struct ib_ucm_file
*file
,
977 const char __user
*inbuf
,
978 int in_len
, int out_len
)
980 struct ib_ucm_context
*ctx
;
981 struct ib_sa_path_rec
*path
= NULL
;
982 struct ib_ucm_lap cmd
;
983 const void *data
= NULL
;
986 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
989 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.len
);
993 result
= ib_ucm_path_get(&path
, cmd
.path
);
997 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
999 result
= ib_send_cm_lap(ctx
->cm_id
, path
, data
, cmd
.len
);
1000 ib_ucm_ctx_put(ctx
);
1002 result
= PTR_ERR(ctx
);
1010 static ssize_t
ib_ucm_send_sidr_req(struct ib_ucm_file
*file
,
1011 const char __user
*inbuf
,
1012 int in_len
, int out_len
)
1014 struct ib_cm_sidr_req_param param
;
1015 struct ib_ucm_context
*ctx
;
1016 struct ib_ucm_sidr_req cmd
;
1019 param
.private_data
= NULL
;
1022 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
1025 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
1029 result
= ib_ucm_path_get(¶m
.path
, cmd
.path
);
1033 param
.private_data_len
= cmd
.len
;
1034 param
.service_id
= cmd
.sid
;
1035 param
.timeout_ms
= cmd
.timeout
;
1036 param
.max_cm_retries
= cmd
.max_cm_retries
;
1038 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
1040 result
= ib_send_cm_sidr_req(ctx
->cm_id
, ¶m
);
1041 ib_ucm_ctx_put(ctx
);
1043 result
= PTR_ERR(ctx
);
1046 kfree(param
.private_data
);
1051 static ssize_t
ib_ucm_send_sidr_rep(struct ib_ucm_file
*file
,
1052 const char __user
*inbuf
,
1053 int in_len
, int out_len
)
1055 struct ib_cm_sidr_rep_param param
;
1056 struct ib_ucm_sidr_rep cmd
;
1057 struct ib_ucm_context
*ctx
;
1062 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
1065 result
= ib_ucm_alloc_data(¶m
.private_data
,
1066 cmd
.data
, cmd
.data_len
);
1070 result
= ib_ucm_alloc_data(¶m
.info
, cmd
.info
, cmd
.info_len
);
1074 param
.qp_num
= cmd
.qpn
;
1075 param
.qkey
= cmd
.qkey
;
1076 param
.status
= cmd
.status
;
1077 param
.info_length
= cmd
.info_len
;
1078 param
.private_data_len
= cmd
.data_len
;
1080 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
1082 result
= ib_send_cm_sidr_rep(ctx
->cm_id
, ¶m
);
1083 ib_ucm_ctx_put(ctx
);
1085 result
= PTR_ERR(ctx
);
1088 kfree(param
.private_data
);
1093 static ssize_t (*ucm_cmd_table
[])(struct ib_ucm_file
*file
,
1094 const char __user
*inbuf
,
1095 int in_len
, int out_len
) = {
1096 [IB_USER_CM_CMD_CREATE_ID
] = ib_ucm_create_id
,
1097 [IB_USER_CM_CMD_DESTROY_ID
] = ib_ucm_destroy_id
,
1098 [IB_USER_CM_CMD_ATTR_ID
] = ib_ucm_attr_id
,
1099 [IB_USER_CM_CMD_LISTEN
] = ib_ucm_listen
,
1100 [IB_USER_CM_CMD_NOTIFY
] = ib_ucm_notify
,
1101 [IB_USER_CM_CMD_SEND_REQ
] = ib_ucm_send_req
,
1102 [IB_USER_CM_CMD_SEND_REP
] = ib_ucm_send_rep
,
1103 [IB_USER_CM_CMD_SEND_RTU
] = ib_ucm_send_rtu
,
1104 [IB_USER_CM_CMD_SEND_DREQ
] = ib_ucm_send_dreq
,
1105 [IB_USER_CM_CMD_SEND_DREP
] = ib_ucm_send_drep
,
1106 [IB_USER_CM_CMD_SEND_REJ
] = ib_ucm_send_rej
,
1107 [IB_USER_CM_CMD_SEND_MRA
] = ib_ucm_send_mra
,
1108 [IB_USER_CM_CMD_SEND_LAP
] = ib_ucm_send_lap
,
1109 [IB_USER_CM_CMD_SEND_APR
] = ib_ucm_send_apr
,
1110 [IB_USER_CM_CMD_SEND_SIDR_REQ
] = ib_ucm_send_sidr_req
,
1111 [IB_USER_CM_CMD_SEND_SIDR_REP
] = ib_ucm_send_sidr_rep
,
1112 [IB_USER_CM_CMD_EVENT
] = ib_ucm_event
,
1113 [IB_USER_CM_CMD_INIT_QP_ATTR
] = ib_ucm_init_qp_attr
,
1116 static ssize_t
ib_ucm_write(struct file
*filp
, const char __user
*buf
,
1117 size_t len
, loff_t
*pos
)
1119 struct ib_ucm_file
*file
= filp
->private_data
;
1120 struct ib_ucm_cmd_hdr hdr
;
1123 if (len
< sizeof(hdr
))
1126 if (copy_from_user(&hdr
, buf
, sizeof(hdr
)))
1129 if (hdr
.cmd
< 0 || hdr
.cmd
>= ARRAY_SIZE(ucm_cmd_table
))
1132 if (hdr
.in
+ sizeof(hdr
) > len
)
1135 result
= ucm_cmd_table
[hdr
.cmd
](file
, buf
+ sizeof(hdr
),
1143 static unsigned int ib_ucm_poll(struct file
*filp
,
1144 struct poll_table_struct
*wait
)
1146 struct ib_ucm_file
*file
= filp
->private_data
;
1147 unsigned int mask
= 0;
1149 poll_wait(filp
, &file
->poll_wait
, wait
);
1151 if (!list_empty(&file
->events
))
1152 mask
= POLLIN
| POLLRDNORM
;
1158 * ib_ucm_open() does not need the BKL:
1160 * - no global state is referred to;
1161 * - there is no ioctl method to race against;
1162 * - no further module initialization is required for open to work
1163 * after the device is registered.
1165 static int ib_ucm_open(struct inode
*inode
, struct file
*filp
)
1167 struct ib_ucm_file
*file
;
1169 file
= kmalloc(sizeof(*file
), GFP_KERNEL
);
1173 INIT_LIST_HEAD(&file
->events
);
1174 INIT_LIST_HEAD(&file
->ctxs
);
1175 init_waitqueue_head(&file
->poll_wait
);
1177 mutex_init(&file
->file_mutex
);
1179 filp
->private_data
= file
;
1181 file
->device
= container_of(inode
->i_cdev
, struct ib_ucm_device
, cdev
);
1186 static int ib_ucm_close(struct inode
*inode
, struct file
*filp
)
1188 struct ib_ucm_file
*file
= filp
->private_data
;
1189 struct ib_ucm_context
*ctx
;
1191 mutex_lock(&file
->file_mutex
);
1192 while (!list_empty(&file
->ctxs
)) {
1193 ctx
= list_entry(file
->ctxs
.next
,
1194 struct ib_ucm_context
, file_list
);
1195 mutex_unlock(&file
->file_mutex
);
1197 mutex_lock(&ctx_id_mutex
);
1198 idr_remove(&ctx_id_table
, ctx
->id
);
1199 mutex_unlock(&ctx_id_mutex
);
1201 ib_destroy_cm_id(ctx
->cm_id
);
1202 ib_ucm_cleanup_events(ctx
);
1205 mutex_lock(&file
->file_mutex
);
1207 mutex_unlock(&file
->file_mutex
);
1212 static void ib_ucm_release_dev(struct device
*dev
)
1214 struct ib_ucm_device
*ucm_dev
;
1216 ucm_dev
= container_of(dev
, struct ib_ucm_device
, dev
);
1217 cdev_del(&ucm_dev
->cdev
);
1218 clear_bit(ucm_dev
->devnum
, dev_map
);
1222 static const struct file_operations ucm_fops
= {
1223 .owner
= THIS_MODULE
,
1224 .open
= ib_ucm_open
,
1225 .release
= ib_ucm_close
,
1226 .write
= ib_ucm_write
,
1227 .poll
= ib_ucm_poll
,
1230 static ssize_t
show_ibdev(struct device
*dev
, struct device_attribute
*attr
,
1233 struct ib_ucm_device
*ucm_dev
;
1235 ucm_dev
= container_of(dev
, struct ib_ucm_device
, dev
);
1236 return sprintf(buf
, "%s\n", ucm_dev
->ib_dev
->name
);
1238 static DEVICE_ATTR(ibdev
, S_IRUGO
, show_ibdev
, NULL
);
1240 static void ib_ucm_add_one(struct ib_device
*device
)
1242 struct ib_ucm_device
*ucm_dev
;
1244 if (!device
->alloc_ucontext
||
1245 rdma_node_get_transport(device
->node_type
) != RDMA_TRANSPORT_IB
)
1248 ucm_dev
= kzalloc(sizeof *ucm_dev
, GFP_KERNEL
);
1252 ucm_dev
->ib_dev
= device
;
1254 ucm_dev
->devnum
= find_first_zero_bit(dev_map
, IB_UCM_MAX_DEVICES
);
1255 if (ucm_dev
->devnum
>= IB_UCM_MAX_DEVICES
)
1258 set_bit(ucm_dev
->devnum
, dev_map
);
1260 cdev_init(&ucm_dev
->cdev
, &ucm_fops
);
1261 ucm_dev
->cdev
.owner
= THIS_MODULE
;
1262 kobject_set_name(&ucm_dev
->cdev
.kobj
, "ucm%d", ucm_dev
->devnum
);
1263 if (cdev_add(&ucm_dev
->cdev
, IB_UCM_BASE_DEV
+ ucm_dev
->devnum
, 1))
1266 ucm_dev
->dev
.class = &cm_class
;
1267 ucm_dev
->dev
.parent
= device
->dma_device
;
1268 ucm_dev
->dev
.devt
= ucm_dev
->cdev
.dev
;
1269 ucm_dev
->dev
.release
= ib_ucm_release_dev
;
1270 dev_set_name(&ucm_dev
->dev
, "ucm%d", ucm_dev
->devnum
);
1271 if (device_register(&ucm_dev
->dev
))
1274 if (device_create_file(&ucm_dev
->dev
, &dev_attr_ibdev
))
1277 ib_set_client_data(device
, &ucm_client
, ucm_dev
);
1281 device_unregister(&ucm_dev
->dev
);
1283 cdev_del(&ucm_dev
->cdev
);
1284 clear_bit(ucm_dev
->devnum
, dev_map
);
1290 static void ib_ucm_remove_one(struct ib_device
*device
)
1292 struct ib_ucm_device
*ucm_dev
= ib_get_client_data(device
, &ucm_client
);
1297 device_unregister(&ucm_dev
->dev
);
1300 static ssize_t
show_abi_version(struct class *class, char *buf
)
1302 return sprintf(buf
, "%d\n", IB_USER_CM_ABI_VERSION
);
1304 static CLASS_ATTR(abi_version
, S_IRUGO
, show_abi_version
, NULL
);
1306 static int __init
ib_ucm_init(void)
1310 ret
= register_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
,
1313 printk(KERN_ERR
"ucm: couldn't register device number\n");
1317 ret
= class_create_file(&cm_class
, &class_attr_abi_version
);
1319 printk(KERN_ERR
"ucm: couldn't create abi_version attribute\n");
1323 ret
= ib_register_client(&ucm_client
);
1325 printk(KERN_ERR
"ucm: couldn't register client\n");
1331 class_remove_file(&cm_class
, &class_attr_abi_version
);
1333 unregister_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
);
1338 static void __exit
ib_ucm_cleanup(void)
1340 ib_unregister_client(&ucm_client
);
1341 class_remove_file(&cm_class
, &class_attr_abi_version
);
1342 unregister_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
);
1343 idr_destroy(&ctx_id_table
);
1346 module_init(ib_ucm_init
);
1347 module_exit(ib_ucm_cleanup
);