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>
47 #include <linux/slab.h>
49 #include <asm/uaccess.h>
51 #include <rdma/ib_cm.h>
52 #include <rdma/ib_user_cm.h>
53 #include <rdma/ib_marshall.h>
55 MODULE_AUTHOR("Libor Michalek");
56 MODULE_DESCRIPTION("InfiniBand userspace Connection Manager access");
57 MODULE_LICENSE("Dual BSD/GPL");
59 struct ib_ucm_device
{
63 struct ib_device
*ib_dev
;
67 struct mutex file_mutex
;
69 struct ib_ucm_device
*device
;
71 struct list_head ctxs
;
72 struct list_head events
;
73 wait_queue_head_t poll_wait
;
76 struct ib_ucm_context
{
78 struct completion comp
;
82 struct ib_ucm_file
*file
;
83 struct ib_cm_id
*cm_id
;
86 struct list_head events
; /* list of pending events. */
87 struct list_head file_list
; /* member in file ctx list */
91 struct ib_ucm_context
*ctx
;
92 struct list_head file_list
; /* member in file event list */
93 struct list_head ctx_list
; /* member in ctx event list */
95 struct ib_cm_id
*cm_id
;
96 struct ib_ucm_event_resp resp
;
105 IB_UCM_BASE_MINOR
= 224,
106 IB_UCM_MAX_DEVICES
= 32
109 #define IB_UCM_BASE_DEV MKDEV(IB_UCM_MAJOR, IB_UCM_BASE_MINOR)
111 static void ib_ucm_add_one(struct ib_device
*device
);
112 static void ib_ucm_remove_one(struct ib_device
*device
, void *client_data
);
114 static struct ib_client ucm_client
= {
116 .add
= ib_ucm_add_one
,
117 .remove
= ib_ucm_remove_one
120 static DEFINE_MUTEX(ctx_id_mutex
);
121 static DEFINE_IDR(ctx_id_table
);
122 static DECLARE_BITMAP(dev_map
, IB_UCM_MAX_DEVICES
);
124 static struct ib_ucm_context
*ib_ucm_ctx_get(struct ib_ucm_file
*file
, int id
)
126 struct ib_ucm_context
*ctx
;
128 mutex_lock(&ctx_id_mutex
);
129 ctx
= idr_find(&ctx_id_table
, id
);
131 ctx
= ERR_PTR(-ENOENT
);
132 else if (ctx
->file
!= file
)
133 ctx
= ERR_PTR(-EINVAL
);
135 atomic_inc(&ctx
->ref
);
136 mutex_unlock(&ctx_id_mutex
);
141 static void ib_ucm_ctx_put(struct ib_ucm_context
*ctx
)
143 if (atomic_dec_and_test(&ctx
->ref
))
144 complete(&ctx
->comp
);
147 static inline int ib_ucm_new_cm_id(int event
)
149 return event
== IB_CM_REQ_RECEIVED
|| event
== IB_CM_SIDR_REQ_RECEIVED
;
152 static void ib_ucm_cleanup_events(struct ib_ucm_context
*ctx
)
154 struct ib_ucm_event
*uevent
;
156 mutex_lock(&ctx
->file
->file_mutex
);
157 list_del(&ctx
->file_list
);
158 while (!list_empty(&ctx
->events
)) {
160 uevent
= list_entry(ctx
->events
.next
,
161 struct ib_ucm_event
, ctx_list
);
162 list_del(&uevent
->file_list
);
163 list_del(&uevent
->ctx_list
);
164 mutex_unlock(&ctx
->file
->file_mutex
);
166 /* clear incoming connections. */
167 if (ib_ucm_new_cm_id(uevent
->resp
.event
))
168 ib_destroy_cm_id(uevent
->cm_id
);
171 mutex_lock(&ctx
->file
->file_mutex
);
173 mutex_unlock(&ctx
->file
->file_mutex
);
176 static struct ib_ucm_context
*ib_ucm_ctx_alloc(struct ib_ucm_file
*file
)
178 struct ib_ucm_context
*ctx
;
180 ctx
= kzalloc(sizeof *ctx
, GFP_KERNEL
);
184 atomic_set(&ctx
->ref
, 1);
185 init_completion(&ctx
->comp
);
187 INIT_LIST_HEAD(&ctx
->events
);
189 mutex_lock(&ctx_id_mutex
);
190 ctx
->id
= idr_alloc(&ctx_id_table
, ctx
, 0, 0, GFP_KERNEL
);
191 mutex_unlock(&ctx_id_mutex
);
195 list_add_tail(&ctx
->file_list
, &file
->ctxs
);
203 static void ib_ucm_event_req_get(struct ib_ucm_req_event_resp
*ureq
,
204 struct ib_cm_req_event_param
*kreq
)
206 ureq
->remote_ca_guid
= kreq
->remote_ca_guid
;
207 ureq
->remote_qkey
= kreq
->remote_qkey
;
208 ureq
->remote_qpn
= kreq
->remote_qpn
;
209 ureq
->qp_type
= kreq
->qp_type
;
210 ureq
->starting_psn
= kreq
->starting_psn
;
211 ureq
->responder_resources
= kreq
->responder_resources
;
212 ureq
->initiator_depth
= kreq
->initiator_depth
;
213 ureq
->local_cm_response_timeout
= kreq
->local_cm_response_timeout
;
214 ureq
->flow_control
= kreq
->flow_control
;
215 ureq
->remote_cm_response_timeout
= kreq
->remote_cm_response_timeout
;
216 ureq
->retry_count
= kreq
->retry_count
;
217 ureq
->rnr_retry_count
= kreq
->rnr_retry_count
;
218 ureq
->srq
= kreq
->srq
;
219 ureq
->port
= kreq
->port
;
221 ib_copy_path_rec_to_user(&ureq
->primary_path
, kreq
->primary_path
);
222 if (kreq
->alternate_path
)
223 ib_copy_path_rec_to_user(&ureq
->alternate_path
,
224 kreq
->alternate_path
);
227 static void ib_ucm_event_rep_get(struct ib_ucm_rep_event_resp
*urep
,
228 struct ib_cm_rep_event_param
*krep
)
230 urep
->remote_ca_guid
= krep
->remote_ca_guid
;
231 urep
->remote_qkey
= krep
->remote_qkey
;
232 urep
->remote_qpn
= krep
->remote_qpn
;
233 urep
->starting_psn
= krep
->starting_psn
;
234 urep
->responder_resources
= krep
->responder_resources
;
235 urep
->initiator_depth
= krep
->initiator_depth
;
236 urep
->target_ack_delay
= krep
->target_ack_delay
;
237 urep
->failover_accepted
= krep
->failover_accepted
;
238 urep
->flow_control
= krep
->flow_control
;
239 urep
->rnr_retry_count
= krep
->rnr_retry_count
;
240 urep
->srq
= krep
->srq
;
243 static void ib_ucm_event_sidr_rep_get(struct ib_ucm_sidr_rep_event_resp
*urep
,
244 struct ib_cm_sidr_rep_event_param
*krep
)
246 urep
->status
= krep
->status
;
247 urep
->qkey
= krep
->qkey
;
248 urep
->qpn
= krep
->qpn
;
251 static int ib_ucm_event_process(struct ib_cm_event
*evt
,
252 struct ib_ucm_event
*uvt
)
256 switch (evt
->event
) {
257 case IB_CM_REQ_RECEIVED
:
258 ib_ucm_event_req_get(&uvt
->resp
.u
.req_resp
,
259 &evt
->param
.req_rcvd
);
260 uvt
->data_len
= IB_CM_REQ_PRIVATE_DATA_SIZE
;
261 uvt
->resp
.present
= IB_UCM_PRES_PRIMARY
;
262 uvt
->resp
.present
|= (evt
->param
.req_rcvd
.alternate_path
?
263 IB_UCM_PRES_ALTERNATE
: 0);
265 case IB_CM_REP_RECEIVED
:
266 ib_ucm_event_rep_get(&uvt
->resp
.u
.rep_resp
,
267 &evt
->param
.rep_rcvd
);
268 uvt
->data_len
= IB_CM_REP_PRIVATE_DATA_SIZE
;
270 case IB_CM_RTU_RECEIVED
:
271 uvt
->data_len
= IB_CM_RTU_PRIVATE_DATA_SIZE
;
272 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
274 case IB_CM_DREQ_RECEIVED
:
275 uvt
->data_len
= IB_CM_DREQ_PRIVATE_DATA_SIZE
;
276 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
278 case IB_CM_DREP_RECEIVED
:
279 uvt
->data_len
= IB_CM_DREP_PRIVATE_DATA_SIZE
;
280 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
282 case IB_CM_MRA_RECEIVED
:
283 uvt
->resp
.u
.mra_resp
.timeout
=
284 evt
->param
.mra_rcvd
.service_timeout
;
285 uvt
->data_len
= IB_CM_MRA_PRIVATE_DATA_SIZE
;
287 case IB_CM_REJ_RECEIVED
:
288 uvt
->resp
.u
.rej_resp
.reason
= evt
->param
.rej_rcvd
.reason
;
289 uvt
->data_len
= IB_CM_REJ_PRIVATE_DATA_SIZE
;
290 uvt
->info_len
= evt
->param
.rej_rcvd
.ari_length
;
291 info
= evt
->param
.rej_rcvd
.ari
;
293 case IB_CM_LAP_RECEIVED
:
294 ib_copy_path_rec_to_user(&uvt
->resp
.u
.lap_resp
.path
,
295 evt
->param
.lap_rcvd
.alternate_path
);
296 uvt
->data_len
= IB_CM_LAP_PRIVATE_DATA_SIZE
;
297 uvt
->resp
.present
= IB_UCM_PRES_ALTERNATE
;
299 case IB_CM_APR_RECEIVED
:
300 uvt
->resp
.u
.apr_resp
.status
= evt
->param
.apr_rcvd
.ap_status
;
301 uvt
->data_len
= IB_CM_APR_PRIVATE_DATA_SIZE
;
302 uvt
->info_len
= evt
->param
.apr_rcvd
.info_len
;
303 info
= evt
->param
.apr_rcvd
.apr_info
;
305 case IB_CM_SIDR_REQ_RECEIVED
:
306 uvt
->resp
.u
.sidr_req_resp
.pkey
=
307 evt
->param
.sidr_req_rcvd
.pkey
;
308 uvt
->resp
.u
.sidr_req_resp
.port
=
309 evt
->param
.sidr_req_rcvd
.port
;
310 uvt
->data_len
= IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE
;
312 case IB_CM_SIDR_REP_RECEIVED
:
313 ib_ucm_event_sidr_rep_get(&uvt
->resp
.u
.sidr_rep_resp
,
314 &evt
->param
.sidr_rep_rcvd
);
315 uvt
->data_len
= IB_CM_SIDR_REP_PRIVATE_DATA_SIZE
;
316 uvt
->info_len
= evt
->param
.sidr_rep_rcvd
.info_len
;
317 info
= evt
->param
.sidr_rep_rcvd
.info
;
320 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
325 uvt
->data
= kmemdup(evt
->private_data
, uvt
->data_len
, GFP_KERNEL
);
329 uvt
->resp
.present
|= IB_UCM_PRES_DATA
;
333 uvt
->info
= kmemdup(info
, uvt
->info_len
, GFP_KERNEL
);
337 uvt
->resp
.present
|= IB_UCM_PRES_INFO
;
347 static int ib_ucm_event_handler(struct ib_cm_id
*cm_id
,
348 struct ib_cm_event
*event
)
350 struct ib_ucm_event
*uevent
;
351 struct ib_ucm_context
*ctx
;
354 ctx
= cm_id
->context
;
356 uevent
= kzalloc(sizeof *uevent
, GFP_KERNEL
);
361 uevent
->cm_id
= cm_id
;
362 uevent
->resp
.uid
= ctx
->uid
;
363 uevent
->resp
.id
= ctx
->id
;
364 uevent
->resp
.event
= event
->event
;
366 result
= ib_ucm_event_process(event
, uevent
);
370 mutex_lock(&ctx
->file
->file_mutex
);
371 list_add_tail(&uevent
->file_list
, &ctx
->file
->events
);
372 list_add_tail(&uevent
->ctx_list
, &ctx
->events
);
373 wake_up_interruptible(&ctx
->file
->poll_wait
);
374 mutex_unlock(&ctx
->file
->file_mutex
);
380 /* Destroy new cm_id's */
381 return ib_ucm_new_cm_id(event
->event
);
384 static ssize_t
ib_ucm_event(struct ib_ucm_file
*file
,
385 const char __user
*inbuf
,
386 int in_len
, int out_len
)
388 struct ib_ucm_context
*ctx
;
389 struct ib_ucm_event_get cmd
;
390 struct ib_ucm_event
*uevent
;
393 if (out_len
< sizeof(struct ib_ucm_event_resp
))
396 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
399 mutex_lock(&file
->file_mutex
);
400 while (list_empty(&file
->events
)) {
401 mutex_unlock(&file
->file_mutex
);
403 if (file
->filp
->f_flags
& O_NONBLOCK
)
406 if (wait_event_interruptible(file
->poll_wait
,
407 !list_empty(&file
->events
)))
410 mutex_lock(&file
->file_mutex
);
413 uevent
= list_entry(file
->events
.next
, struct ib_ucm_event
, file_list
);
415 if (ib_ucm_new_cm_id(uevent
->resp
.event
)) {
416 ctx
= ib_ucm_ctx_alloc(file
);
422 ctx
->cm_id
= uevent
->cm_id
;
423 ctx
->cm_id
->context
= ctx
;
424 uevent
->resp
.id
= ctx
->id
;
427 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
428 &uevent
->resp
, sizeof(uevent
->resp
))) {
434 if (cmd
.data_len
< uevent
->data_len
) {
438 if (copy_to_user((void __user
*)(unsigned long)cmd
.data
,
439 uevent
->data
, uevent
->data_len
)) {
446 if (cmd
.info_len
< uevent
->info_len
) {
450 if (copy_to_user((void __user
*)(unsigned long)cmd
.info
,
451 uevent
->info
, uevent
->info_len
)) {
457 list_del(&uevent
->file_list
);
458 list_del(&uevent
->ctx_list
);
459 uevent
->ctx
->events_reported
++;
465 mutex_unlock(&file
->file_mutex
);
469 static ssize_t
ib_ucm_create_id(struct ib_ucm_file
*file
,
470 const char __user
*inbuf
,
471 int in_len
, int out_len
)
473 struct ib_ucm_create_id cmd
;
474 struct ib_ucm_create_id_resp resp
;
475 struct ib_ucm_context
*ctx
;
478 if (out_len
< sizeof(resp
))
481 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
484 mutex_lock(&file
->file_mutex
);
485 ctx
= ib_ucm_ctx_alloc(file
);
486 mutex_unlock(&file
->file_mutex
);
491 ctx
->cm_id
= ib_create_cm_id(file
->device
->ib_dev
,
492 ib_ucm_event_handler
, ctx
);
493 if (IS_ERR(ctx
->cm_id
)) {
494 result
= PTR_ERR(ctx
->cm_id
);
499 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
500 &resp
, sizeof(resp
))) {
507 ib_destroy_cm_id(ctx
->cm_id
);
509 mutex_lock(&ctx_id_mutex
);
510 idr_remove(&ctx_id_table
, ctx
->id
);
511 mutex_unlock(&ctx_id_mutex
);
516 static ssize_t
ib_ucm_destroy_id(struct ib_ucm_file
*file
,
517 const char __user
*inbuf
,
518 int in_len
, int out_len
)
520 struct ib_ucm_destroy_id cmd
;
521 struct ib_ucm_destroy_id_resp resp
;
522 struct ib_ucm_context
*ctx
;
525 if (out_len
< sizeof(resp
))
528 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
531 mutex_lock(&ctx_id_mutex
);
532 ctx
= idr_find(&ctx_id_table
, cmd
.id
);
534 ctx
= ERR_PTR(-ENOENT
);
535 else if (ctx
->file
!= file
)
536 ctx
= ERR_PTR(-EINVAL
);
538 idr_remove(&ctx_id_table
, ctx
->id
);
539 mutex_unlock(&ctx_id_mutex
);
545 wait_for_completion(&ctx
->comp
);
547 /* No new events will be generated after destroying the cm_id. */
548 ib_destroy_cm_id(ctx
->cm_id
);
549 /* Cleanup events not yet reported to the user. */
550 ib_ucm_cleanup_events(ctx
);
552 resp
.events_reported
= ctx
->events_reported
;
553 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
554 &resp
, sizeof(resp
)))
561 static ssize_t
ib_ucm_attr_id(struct ib_ucm_file
*file
,
562 const char __user
*inbuf
,
563 int in_len
, int out_len
)
565 struct ib_ucm_attr_id_resp resp
;
566 struct ib_ucm_attr_id cmd
;
567 struct ib_ucm_context
*ctx
;
570 if (out_len
< sizeof(resp
))
573 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
576 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
580 resp
.service_id
= ctx
->cm_id
->service_id
;
581 resp
.service_mask
= ctx
->cm_id
->service_mask
;
582 resp
.local_id
= ctx
->cm_id
->local_id
;
583 resp
.remote_id
= ctx
->cm_id
->remote_id
;
585 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
586 &resp
, sizeof(resp
)))
593 static ssize_t
ib_ucm_init_qp_attr(struct ib_ucm_file
*file
,
594 const char __user
*inbuf
,
595 int in_len
, int out_len
)
597 struct ib_uverbs_qp_attr resp
;
598 struct ib_ucm_init_qp_attr cmd
;
599 struct ib_ucm_context
*ctx
;
600 struct ib_qp_attr qp_attr
;
603 if (out_len
< sizeof(resp
))
606 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
609 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
613 resp
.qp_attr_mask
= 0;
614 memset(&qp_attr
, 0, sizeof qp_attr
);
615 qp_attr
.qp_state
= cmd
.qp_state
;
616 result
= ib_cm_init_qp_attr(ctx
->cm_id
, &qp_attr
, &resp
.qp_attr_mask
);
620 ib_copy_qp_attr_to_user(&resp
, &qp_attr
);
622 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
623 &resp
, sizeof(resp
)))
631 static int ucm_validate_listen(__be64 service_id
, __be64 service_mask
)
633 service_id
&= service_mask
;
635 if (((service_id
& IB_CMA_SERVICE_ID_MASK
) == IB_CMA_SERVICE_ID
) ||
636 ((service_id
& IB_SDP_SERVICE_ID_MASK
) == IB_SDP_SERVICE_ID
))
642 static ssize_t
ib_ucm_listen(struct ib_ucm_file
*file
,
643 const char __user
*inbuf
,
644 int in_len
, int out_len
)
646 struct ib_ucm_listen cmd
;
647 struct ib_ucm_context
*ctx
;
650 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
653 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
657 result
= ucm_validate_listen(cmd
.service_id
, cmd
.service_mask
);
661 result
= ib_cm_listen(ctx
->cm_id
, cmd
.service_id
, cmd
.service_mask
);
667 static ssize_t
ib_ucm_notify(struct ib_ucm_file
*file
,
668 const char __user
*inbuf
,
669 int in_len
, int out_len
)
671 struct ib_ucm_notify cmd
;
672 struct ib_ucm_context
*ctx
;
675 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
678 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
682 result
= ib_cm_notify(ctx
->cm_id
, (enum ib_event_type
) cmd
.event
);
687 static int ib_ucm_alloc_data(const void **dest
, u64 src
, u32 len
)
696 data
= memdup_user((void __user
*)(unsigned long)src
, len
);
698 return PTR_ERR(data
);
704 static int ib_ucm_path_get(struct ib_sa_path_rec
**path
, u64 src
)
706 struct ib_user_path_rec upath
;
707 struct ib_sa_path_rec
*sa_path
;
714 sa_path
= kmalloc(sizeof(*sa_path
), GFP_KERNEL
);
718 if (copy_from_user(&upath
, (void __user
*)(unsigned long)src
,
725 ib_copy_path_rec_from_user(sa_path
, &upath
);
730 static ssize_t
ib_ucm_send_req(struct ib_ucm_file
*file
,
731 const char __user
*inbuf
,
732 int in_len
, int out_len
)
734 struct ib_cm_req_param param
;
735 struct ib_ucm_context
*ctx
;
736 struct ib_ucm_req cmd
;
739 param
.private_data
= NULL
;
740 param
.primary_path
= NULL
;
741 param
.alternate_path
= NULL
;
743 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
746 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
750 result
= ib_ucm_path_get(¶m
.primary_path
, cmd
.primary_path
);
754 result
= ib_ucm_path_get(¶m
.alternate_path
, cmd
.alternate_path
);
758 param
.private_data_len
= cmd
.len
;
759 param
.service_id
= cmd
.sid
;
760 param
.qp_num
= cmd
.qpn
;
761 param
.qp_type
= cmd
.qp_type
;
762 param
.starting_psn
= cmd
.psn
;
763 param
.peer_to_peer
= cmd
.peer_to_peer
;
764 param
.responder_resources
= cmd
.responder_resources
;
765 param
.initiator_depth
= cmd
.initiator_depth
;
766 param
.remote_cm_response_timeout
= cmd
.remote_cm_response_timeout
;
767 param
.flow_control
= cmd
.flow_control
;
768 param
.local_cm_response_timeout
= cmd
.local_cm_response_timeout
;
769 param
.retry_count
= cmd
.retry_count
;
770 param
.rnr_retry_count
= cmd
.rnr_retry_count
;
771 param
.max_cm_retries
= cmd
.max_cm_retries
;
774 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
776 result
= ib_send_cm_req(ctx
->cm_id
, ¶m
);
779 result
= PTR_ERR(ctx
);
782 kfree(param
.private_data
);
783 kfree(param
.primary_path
);
784 kfree(param
.alternate_path
);
788 static ssize_t
ib_ucm_send_rep(struct ib_ucm_file
*file
,
789 const char __user
*inbuf
,
790 int in_len
, int out_len
)
792 struct ib_cm_rep_param param
;
793 struct ib_ucm_context
*ctx
;
794 struct ib_ucm_rep cmd
;
797 param
.private_data
= NULL
;
799 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
802 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
806 param
.qp_num
= cmd
.qpn
;
807 param
.starting_psn
= cmd
.psn
;
808 param
.private_data_len
= cmd
.len
;
809 param
.responder_resources
= cmd
.responder_resources
;
810 param
.initiator_depth
= cmd
.initiator_depth
;
811 param
.failover_accepted
= cmd
.failover_accepted
;
812 param
.flow_control
= cmd
.flow_control
;
813 param
.rnr_retry_count
= cmd
.rnr_retry_count
;
816 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
819 result
= ib_send_cm_rep(ctx
->cm_id
, ¶m
);
822 result
= PTR_ERR(ctx
);
824 kfree(param
.private_data
);
828 static ssize_t
ib_ucm_send_private_data(struct ib_ucm_file
*file
,
829 const char __user
*inbuf
, int in_len
,
830 int (*func
)(struct ib_cm_id
*cm_id
,
831 const void *private_data
,
832 u8 private_data_len
))
834 struct ib_ucm_private_data cmd
;
835 struct ib_ucm_context
*ctx
;
836 const void *private_data
= NULL
;
839 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
842 result
= ib_ucm_alloc_data(&private_data
, cmd
.data
, cmd
.len
);
846 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
848 result
= func(ctx
->cm_id
, private_data
, cmd
.len
);
851 result
= PTR_ERR(ctx
);
857 static ssize_t
ib_ucm_send_rtu(struct ib_ucm_file
*file
,
858 const char __user
*inbuf
,
859 int in_len
, int out_len
)
861 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_rtu
);
864 static ssize_t
ib_ucm_send_dreq(struct ib_ucm_file
*file
,
865 const char __user
*inbuf
,
866 int in_len
, int out_len
)
868 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_dreq
);
871 static ssize_t
ib_ucm_send_drep(struct ib_ucm_file
*file
,
872 const char __user
*inbuf
,
873 int in_len
, int out_len
)
875 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_drep
);
878 static ssize_t
ib_ucm_send_info(struct ib_ucm_file
*file
,
879 const char __user
*inbuf
, int in_len
,
880 int (*func
)(struct ib_cm_id
*cm_id
,
887 struct ib_ucm_context
*ctx
;
888 struct ib_ucm_info cmd
;
889 const void *data
= NULL
;
890 const void *info
= NULL
;
893 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
896 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.data_len
);
900 result
= ib_ucm_alloc_data(&info
, cmd
.info
, cmd
.info_len
);
904 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
906 result
= func(ctx
->cm_id
, cmd
.status
, info
, cmd
.info_len
,
910 result
= PTR_ERR(ctx
);
918 static ssize_t
ib_ucm_send_rej(struct ib_ucm_file
*file
,
919 const char __user
*inbuf
,
920 int in_len
, int out_len
)
922 return ib_ucm_send_info(file
, inbuf
, in_len
, (void *)ib_send_cm_rej
);
925 static ssize_t
ib_ucm_send_apr(struct ib_ucm_file
*file
,
926 const char __user
*inbuf
,
927 int in_len
, int out_len
)
929 return ib_ucm_send_info(file
, inbuf
, in_len
, (void *)ib_send_cm_apr
);
932 static ssize_t
ib_ucm_send_mra(struct ib_ucm_file
*file
,
933 const char __user
*inbuf
,
934 int in_len
, int out_len
)
936 struct ib_ucm_context
*ctx
;
937 struct ib_ucm_mra cmd
;
938 const void *data
= NULL
;
941 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
944 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.len
);
948 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
950 result
= ib_send_cm_mra(ctx
->cm_id
, cmd
.timeout
, data
, cmd
.len
);
953 result
= PTR_ERR(ctx
);
959 static ssize_t
ib_ucm_send_lap(struct ib_ucm_file
*file
,
960 const char __user
*inbuf
,
961 int in_len
, int out_len
)
963 struct ib_ucm_context
*ctx
;
964 struct ib_sa_path_rec
*path
= NULL
;
965 struct ib_ucm_lap cmd
;
966 const void *data
= NULL
;
969 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
972 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.len
);
976 result
= ib_ucm_path_get(&path
, cmd
.path
);
980 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
982 result
= ib_send_cm_lap(ctx
->cm_id
, path
, data
, cmd
.len
);
985 result
= PTR_ERR(ctx
);
993 static ssize_t
ib_ucm_send_sidr_req(struct ib_ucm_file
*file
,
994 const char __user
*inbuf
,
995 int in_len
, int out_len
)
997 struct ib_cm_sidr_req_param param
;
998 struct ib_ucm_context
*ctx
;
999 struct ib_ucm_sidr_req cmd
;
1002 param
.private_data
= NULL
;
1005 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
1008 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
1012 result
= ib_ucm_path_get(¶m
.path
, cmd
.path
);
1016 param
.private_data_len
= cmd
.len
;
1017 param
.service_id
= cmd
.sid
;
1018 param
.timeout_ms
= cmd
.timeout
;
1019 param
.max_cm_retries
= cmd
.max_cm_retries
;
1021 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
1023 result
= ib_send_cm_sidr_req(ctx
->cm_id
, ¶m
);
1024 ib_ucm_ctx_put(ctx
);
1026 result
= PTR_ERR(ctx
);
1029 kfree(param
.private_data
);
1034 static ssize_t
ib_ucm_send_sidr_rep(struct ib_ucm_file
*file
,
1035 const char __user
*inbuf
,
1036 int in_len
, int out_len
)
1038 struct ib_cm_sidr_rep_param param
;
1039 struct ib_ucm_sidr_rep cmd
;
1040 struct ib_ucm_context
*ctx
;
1045 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
1048 result
= ib_ucm_alloc_data(¶m
.private_data
,
1049 cmd
.data
, cmd
.data_len
);
1053 result
= ib_ucm_alloc_data(¶m
.info
, cmd
.info
, cmd
.info_len
);
1057 param
.qp_num
= cmd
.qpn
;
1058 param
.qkey
= cmd
.qkey
;
1059 param
.status
= cmd
.status
;
1060 param
.info_length
= cmd
.info_len
;
1061 param
.private_data_len
= cmd
.data_len
;
1063 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
1065 result
= ib_send_cm_sidr_rep(ctx
->cm_id
, ¶m
);
1066 ib_ucm_ctx_put(ctx
);
1068 result
= PTR_ERR(ctx
);
1071 kfree(param
.private_data
);
1076 static ssize_t (*ucm_cmd_table
[])(struct ib_ucm_file
*file
,
1077 const char __user
*inbuf
,
1078 int in_len
, int out_len
) = {
1079 [IB_USER_CM_CMD_CREATE_ID
] = ib_ucm_create_id
,
1080 [IB_USER_CM_CMD_DESTROY_ID
] = ib_ucm_destroy_id
,
1081 [IB_USER_CM_CMD_ATTR_ID
] = ib_ucm_attr_id
,
1082 [IB_USER_CM_CMD_LISTEN
] = ib_ucm_listen
,
1083 [IB_USER_CM_CMD_NOTIFY
] = ib_ucm_notify
,
1084 [IB_USER_CM_CMD_SEND_REQ
] = ib_ucm_send_req
,
1085 [IB_USER_CM_CMD_SEND_REP
] = ib_ucm_send_rep
,
1086 [IB_USER_CM_CMD_SEND_RTU
] = ib_ucm_send_rtu
,
1087 [IB_USER_CM_CMD_SEND_DREQ
] = ib_ucm_send_dreq
,
1088 [IB_USER_CM_CMD_SEND_DREP
] = ib_ucm_send_drep
,
1089 [IB_USER_CM_CMD_SEND_REJ
] = ib_ucm_send_rej
,
1090 [IB_USER_CM_CMD_SEND_MRA
] = ib_ucm_send_mra
,
1091 [IB_USER_CM_CMD_SEND_LAP
] = ib_ucm_send_lap
,
1092 [IB_USER_CM_CMD_SEND_APR
] = ib_ucm_send_apr
,
1093 [IB_USER_CM_CMD_SEND_SIDR_REQ
] = ib_ucm_send_sidr_req
,
1094 [IB_USER_CM_CMD_SEND_SIDR_REP
] = ib_ucm_send_sidr_rep
,
1095 [IB_USER_CM_CMD_EVENT
] = ib_ucm_event
,
1096 [IB_USER_CM_CMD_INIT_QP_ATTR
] = ib_ucm_init_qp_attr
,
1099 static ssize_t
ib_ucm_write(struct file
*filp
, const char __user
*buf
,
1100 size_t len
, loff_t
*pos
)
1102 struct ib_ucm_file
*file
= filp
->private_data
;
1103 struct ib_ucm_cmd_hdr hdr
;
1106 if (len
< sizeof(hdr
))
1109 if (copy_from_user(&hdr
, buf
, sizeof(hdr
)))
1112 if (hdr
.cmd
>= ARRAY_SIZE(ucm_cmd_table
))
1115 if (hdr
.in
+ sizeof(hdr
) > len
)
1118 result
= ucm_cmd_table
[hdr
.cmd
](file
, buf
+ sizeof(hdr
),
1126 static unsigned int ib_ucm_poll(struct file
*filp
,
1127 struct poll_table_struct
*wait
)
1129 struct ib_ucm_file
*file
= filp
->private_data
;
1130 unsigned int mask
= 0;
1132 poll_wait(filp
, &file
->poll_wait
, wait
);
1134 if (!list_empty(&file
->events
))
1135 mask
= POLLIN
| POLLRDNORM
;
1141 * ib_ucm_open() does not need the BKL:
1143 * - no global state is referred to;
1144 * - there is no ioctl method to race against;
1145 * - no further module initialization is required for open to work
1146 * after the device is registered.
1148 static int ib_ucm_open(struct inode
*inode
, struct file
*filp
)
1150 struct ib_ucm_file
*file
;
1152 file
= kmalloc(sizeof(*file
), GFP_KERNEL
);
1156 INIT_LIST_HEAD(&file
->events
);
1157 INIT_LIST_HEAD(&file
->ctxs
);
1158 init_waitqueue_head(&file
->poll_wait
);
1160 mutex_init(&file
->file_mutex
);
1162 filp
->private_data
= file
;
1164 file
->device
= container_of(inode
->i_cdev
, struct ib_ucm_device
, cdev
);
1166 return nonseekable_open(inode
, filp
);
1169 static int ib_ucm_close(struct inode
*inode
, struct file
*filp
)
1171 struct ib_ucm_file
*file
= filp
->private_data
;
1172 struct ib_ucm_context
*ctx
;
1174 mutex_lock(&file
->file_mutex
);
1175 while (!list_empty(&file
->ctxs
)) {
1176 ctx
= list_entry(file
->ctxs
.next
,
1177 struct ib_ucm_context
, file_list
);
1178 mutex_unlock(&file
->file_mutex
);
1180 mutex_lock(&ctx_id_mutex
);
1181 idr_remove(&ctx_id_table
, ctx
->id
);
1182 mutex_unlock(&ctx_id_mutex
);
1184 ib_destroy_cm_id(ctx
->cm_id
);
1185 ib_ucm_cleanup_events(ctx
);
1188 mutex_lock(&file
->file_mutex
);
1190 mutex_unlock(&file
->file_mutex
);
1195 static DECLARE_BITMAP(overflow_map
, IB_UCM_MAX_DEVICES
);
1196 static void ib_ucm_release_dev(struct device
*dev
)
1198 struct ib_ucm_device
*ucm_dev
;
1200 ucm_dev
= container_of(dev
, struct ib_ucm_device
, dev
);
1201 cdev_del(&ucm_dev
->cdev
);
1202 if (ucm_dev
->devnum
< IB_UCM_MAX_DEVICES
)
1203 clear_bit(ucm_dev
->devnum
, dev_map
);
1205 clear_bit(ucm_dev
->devnum
- IB_UCM_MAX_DEVICES
, overflow_map
);
1209 static const struct file_operations ucm_fops
= {
1210 .owner
= THIS_MODULE
,
1211 .open
= ib_ucm_open
,
1212 .release
= ib_ucm_close
,
1213 .write
= ib_ucm_write
,
1214 .poll
= ib_ucm_poll
,
1215 .llseek
= no_llseek
,
1218 static ssize_t
show_ibdev(struct device
*dev
, struct device_attribute
*attr
,
1221 struct ib_ucm_device
*ucm_dev
;
1223 ucm_dev
= container_of(dev
, struct ib_ucm_device
, dev
);
1224 return sprintf(buf
, "%s\n", ucm_dev
->ib_dev
->name
);
1226 static DEVICE_ATTR(ibdev
, S_IRUGO
, show_ibdev
, NULL
);
1228 static dev_t overflow_maj
;
1229 static int find_overflow_devnum(void)
1233 if (!overflow_maj
) {
1234 ret
= alloc_chrdev_region(&overflow_maj
, 0, IB_UCM_MAX_DEVICES
,
1237 pr_err("ucm: couldn't register dynamic device number\n");
1242 ret
= find_first_zero_bit(overflow_map
, IB_UCM_MAX_DEVICES
);
1243 if (ret
>= IB_UCM_MAX_DEVICES
)
1249 static void ib_ucm_add_one(struct ib_device
*device
)
1253 struct ib_ucm_device
*ucm_dev
;
1255 if (!device
->alloc_ucontext
|| !rdma_cap_ib_cm(device
, 1))
1258 ucm_dev
= kzalloc(sizeof *ucm_dev
, GFP_KERNEL
);
1262 ucm_dev
->ib_dev
= device
;
1264 devnum
= find_first_zero_bit(dev_map
, IB_UCM_MAX_DEVICES
);
1265 if (devnum
>= IB_UCM_MAX_DEVICES
) {
1266 devnum
= find_overflow_devnum();
1270 ucm_dev
->devnum
= devnum
+ IB_UCM_MAX_DEVICES
;
1271 base
= devnum
+ overflow_maj
;
1272 set_bit(devnum
, overflow_map
);
1274 ucm_dev
->devnum
= devnum
;
1275 base
= devnum
+ IB_UCM_BASE_DEV
;
1276 set_bit(devnum
, dev_map
);
1279 cdev_init(&ucm_dev
->cdev
, &ucm_fops
);
1280 ucm_dev
->cdev
.owner
= THIS_MODULE
;
1281 kobject_set_name(&ucm_dev
->cdev
.kobj
, "ucm%d", ucm_dev
->devnum
);
1282 if (cdev_add(&ucm_dev
->cdev
, base
, 1))
1285 ucm_dev
->dev
.class = &cm_class
;
1286 ucm_dev
->dev
.parent
= device
->dma_device
;
1287 ucm_dev
->dev
.devt
= ucm_dev
->cdev
.dev
;
1288 ucm_dev
->dev
.release
= ib_ucm_release_dev
;
1289 dev_set_name(&ucm_dev
->dev
, "ucm%d", ucm_dev
->devnum
);
1290 if (device_register(&ucm_dev
->dev
))
1293 if (device_create_file(&ucm_dev
->dev
, &dev_attr_ibdev
))
1296 ib_set_client_data(device
, &ucm_client
, ucm_dev
);
1300 device_unregister(&ucm_dev
->dev
);
1302 cdev_del(&ucm_dev
->cdev
);
1303 if (ucm_dev
->devnum
< IB_UCM_MAX_DEVICES
)
1304 clear_bit(devnum
, dev_map
);
1306 clear_bit(devnum
, overflow_map
);
1312 static void ib_ucm_remove_one(struct ib_device
*device
, void *client_data
)
1314 struct ib_ucm_device
*ucm_dev
= client_data
;
1319 device_unregister(&ucm_dev
->dev
);
1322 static CLASS_ATTR_STRING(abi_version
, S_IRUGO
,
1323 __stringify(IB_USER_CM_ABI_VERSION
));
1325 static int __init
ib_ucm_init(void)
1329 ret
= register_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
,
1332 pr_err("ucm: couldn't register device number\n");
1336 ret
= class_create_file(&cm_class
, &class_attr_abi_version
.attr
);
1338 pr_err("ucm: couldn't create abi_version attribute\n");
1342 ret
= ib_register_client(&ucm_client
);
1344 pr_err("ucm: couldn't register client\n");
1350 class_remove_file(&cm_class
, &class_attr_abi_version
.attr
);
1352 unregister_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
);
1357 static void __exit
ib_ucm_cleanup(void)
1359 ib_unregister_client(&ucm_client
);
1360 class_remove_file(&cm_class
, &class_attr_abi_version
.attr
);
1361 unregister_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
);
1363 unregister_chrdev_region(overflow_maj
, IB_UCM_MAX_DEVICES
);
1364 idr_destroy(&ctx_id_table
);
1367 module_init(ib_ucm_init
);
1368 module_exit(ib_ucm_cleanup
);