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 <linux/uaccess.h>
52 #include <rdma/ib_cm.h>
53 #include <rdma/ib_user_cm.h>
54 #include <rdma/ib_marshall.h>
56 MODULE_AUTHOR("Libor Michalek");
57 MODULE_DESCRIPTION("InfiniBand userspace Connection Manager access");
58 MODULE_LICENSE("Dual BSD/GPL");
60 struct ib_ucm_device
{
64 struct ib_device
*ib_dev
;
68 struct mutex file_mutex
;
70 struct ib_ucm_device
*device
;
72 struct list_head ctxs
;
73 struct list_head events
;
74 wait_queue_head_t poll_wait
;
77 struct ib_ucm_context
{
79 struct completion comp
;
83 struct ib_ucm_file
*file
;
84 struct ib_cm_id
*cm_id
;
87 struct list_head events
; /* list of pending events. */
88 struct list_head file_list
; /* member in file ctx list */
92 struct ib_ucm_context
*ctx
;
93 struct list_head file_list
; /* member in file event list */
94 struct list_head ctx_list
; /* member in ctx event list */
96 struct ib_cm_id
*cm_id
;
97 struct ib_ucm_event_resp resp
;
106 IB_UCM_BASE_MINOR
= 224,
107 IB_UCM_MAX_DEVICES
= 32
110 #define IB_UCM_BASE_DEV MKDEV(IB_UCM_MAJOR, IB_UCM_BASE_MINOR)
112 static void ib_ucm_add_one(struct ib_device
*device
);
113 static void ib_ucm_remove_one(struct ib_device
*device
, void *client_data
);
115 static struct ib_client ucm_client
= {
117 .add
= ib_ucm_add_one
,
118 .remove
= ib_ucm_remove_one
121 static DEFINE_MUTEX(ctx_id_mutex
);
122 static DEFINE_IDR(ctx_id_table
);
123 static DECLARE_BITMAP(dev_map
, IB_UCM_MAX_DEVICES
);
125 static struct ib_ucm_context
*ib_ucm_ctx_get(struct ib_ucm_file
*file
, int id
)
127 struct ib_ucm_context
*ctx
;
129 mutex_lock(&ctx_id_mutex
);
130 ctx
= idr_find(&ctx_id_table
, id
);
132 ctx
= ERR_PTR(-ENOENT
);
133 else if (ctx
->file
!= file
)
134 ctx
= ERR_PTR(-EINVAL
);
136 atomic_inc(&ctx
->ref
);
137 mutex_unlock(&ctx_id_mutex
);
142 static void ib_ucm_ctx_put(struct ib_ucm_context
*ctx
)
144 if (atomic_dec_and_test(&ctx
->ref
))
145 complete(&ctx
->comp
);
148 static inline int ib_ucm_new_cm_id(int event
)
150 return event
== IB_CM_REQ_RECEIVED
|| event
== IB_CM_SIDR_REQ_RECEIVED
;
153 static void ib_ucm_cleanup_events(struct ib_ucm_context
*ctx
)
155 struct ib_ucm_event
*uevent
;
157 mutex_lock(&ctx
->file
->file_mutex
);
158 list_del(&ctx
->file_list
);
159 while (!list_empty(&ctx
->events
)) {
161 uevent
= list_entry(ctx
->events
.next
,
162 struct ib_ucm_event
, ctx_list
);
163 list_del(&uevent
->file_list
);
164 list_del(&uevent
->ctx_list
);
165 mutex_unlock(&ctx
->file
->file_mutex
);
167 /* clear incoming connections. */
168 if (ib_ucm_new_cm_id(uevent
->resp
.event
))
169 ib_destroy_cm_id(uevent
->cm_id
);
172 mutex_lock(&ctx
->file
->file_mutex
);
174 mutex_unlock(&ctx
->file
->file_mutex
);
177 static struct ib_ucm_context
*ib_ucm_ctx_alloc(struct ib_ucm_file
*file
)
179 struct ib_ucm_context
*ctx
;
181 ctx
= kzalloc(sizeof *ctx
, GFP_KERNEL
);
185 atomic_set(&ctx
->ref
, 1);
186 init_completion(&ctx
->comp
);
188 INIT_LIST_HEAD(&ctx
->events
);
190 mutex_lock(&ctx_id_mutex
);
191 ctx
->id
= idr_alloc(&ctx_id_table
, ctx
, 0, 0, GFP_KERNEL
);
192 mutex_unlock(&ctx_id_mutex
);
196 list_add_tail(&ctx
->file_list
, &file
->ctxs
);
204 static void ib_ucm_event_req_get(struct ib_ucm_req_event_resp
*ureq
,
205 struct ib_cm_req_event_param
*kreq
)
207 ureq
->remote_ca_guid
= kreq
->remote_ca_guid
;
208 ureq
->remote_qkey
= kreq
->remote_qkey
;
209 ureq
->remote_qpn
= kreq
->remote_qpn
;
210 ureq
->qp_type
= kreq
->qp_type
;
211 ureq
->starting_psn
= kreq
->starting_psn
;
212 ureq
->responder_resources
= kreq
->responder_resources
;
213 ureq
->initiator_depth
= kreq
->initiator_depth
;
214 ureq
->local_cm_response_timeout
= kreq
->local_cm_response_timeout
;
215 ureq
->flow_control
= kreq
->flow_control
;
216 ureq
->remote_cm_response_timeout
= kreq
->remote_cm_response_timeout
;
217 ureq
->retry_count
= kreq
->retry_count
;
218 ureq
->rnr_retry_count
= kreq
->rnr_retry_count
;
219 ureq
->srq
= kreq
->srq
;
220 ureq
->port
= kreq
->port
;
222 ib_copy_path_rec_to_user(&ureq
->primary_path
, kreq
->primary_path
);
223 if (kreq
->alternate_path
)
224 ib_copy_path_rec_to_user(&ureq
->alternate_path
,
225 kreq
->alternate_path
);
228 static void ib_ucm_event_rep_get(struct ib_ucm_rep_event_resp
*urep
,
229 struct ib_cm_rep_event_param
*krep
)
231 urep
->remote_ca_guid
= krep
->remote_ca_guid
;
232 urep
->remote_qkey
= krep
->remote_qkey
;
233 urep
->remote_qpn
= krep
->remote_qpn
;
234 urep
->starting_psn
= krep
->starting_psn
;
235 urep
->responder_resources
= krep
->responder_resources
;
236 urep
->initiator_depth
= krep
->initiator_depth
;
237 urep
->target_ack_delay
= krep
->target_ack_delay
;
238 urep
->failover_accepted
= krep
->failover_accepted
;
239 urep
->flow_control
= krep
->flow_control
;
240 urep
->rnr_retry_count
= krep
->rnr_retry_count
;
241 urep
->srq
= krep
->srq
;
244 static void ib_ucm_event_sidr_rep_get(struct ib_ucm_sidr_rep_event_resp
*urep
,
245 struct ib_cm_sidr_rep_event_param
*krep
)
247 urep
->status
= krep
->status
;
248 urep
->qkey
= krep
->qkey
;
249 urep
->qpn
= krep
->qpn
;
252 static int ib_ucm_event_process(struct ib_cm_event
*evt
,
253 struct ib_ucm_event
*uvt
)
257 switch (evt
->event
) {
258 case IB_CM_REQ_RECEIVED
:
259 ib_ucm_event_req_get(&uvt
->resp
.u
.req_resp
,
260 &evt
->param
.req_rcvd
);
261 uvt
->data_len
= IB_CM_REQ_PRIVATE_DATA_SIZE
;
262 uvt
->resp
.present
= IB_UCM_PRES_PRIMARY
;
263 uvt
->resp
.present
|= (evt
->param
.req_rcvd
.alternate_path
?
264 IB_UCM_PRES_ALTERNATE
: 0);
266 case IB_CM_REP_RECEIVED
:
267 ib_ucm_event_rep_get(&uvt
->resp
.u
.rep_resp
,
268 &evt
->param
.rep_rcvd
);
269 uvt
->data_len
= IB_CM_REP_PRIVATE_DATA_SIZE
;
271 case IB_CM_RTU_RECEIVED
:
272 uvt
->data_len
= IB_CM_RTU_PRIVATE_DATA_SIZE
;
273 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
275 case IB_CM_DREQ_RECEIVED
:
276 uvt
->data_len
= IB_CM_DREQ_PRIVATE_DATA_SIZE
;
277 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
279 case IB_CM_DREP_RECEIVED
:
280 uvt
->data_len
= IB_CM_DREP_PRIVATE_DATA_SIZE
;
281 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
283 case IB_CM_MRA_RECEIVED
:
284 uvt
->resp
.u
.mra_resp
.timeout
=
285 evt
->param
.mra_rcvd
.service_timeout
;
286 uvt
->data_len
= IB_CM_MRA_PRIVATE_DATA_SIZE
;
288 case IB_CM_REJ_RECEIVED
:
289 uvt
->resp
.u
.rej_resp
.reason
= evt
->param
.rej_rcvd
.reason
;
290 uvt
->data_len
= IB_CM_REJ_PRIVATE_DATA_SIZE
;
291 uvt
->info_len
= evt
->param
.rej_rcvd
.ari_length
;
292 info
= evt
->param
.rej_rcvd
.ari
;
294 case IB_CM_LAP_RECEIVED
:
295 ib_copy_path_rec_to_user(&uvt
->resp
.u
.lap_resp
.path
,
296 evt
->param
.lap_rcvd
.alternate_path
);
297 uvt
->data_len
= IB_CM_LAP_PRIVATE_DATA_SIZE
;
298 uvt
->resp
.present
= IB_UCM_PRES_ALTERNATE
;
300 case IB_CM_APR_RECEIVED
:
301 uvt
->resp
.u
.apr_resp
.status
= evt
->param
.apr_rcvd
.ap_status
;
302 uvt
->data_len
= IB_CM_APR_PRIVATE_DATA_SIZE
;
303 uvt
->info_len
= evt
->param
.apr_rcvd
.info_len
;
304 info
= evt
->param
.apr_rcvd
.apr_info
;
306 case IB_CM_SIDR_REQ_RECEIVED
:
307 uvt
->resp
.u
.sidr_req_resp
.pkey
=
308 evt
->param
.sidr_req_rcvd
.pkey
;
309 uvt
->resp
.u
.sidr_req_resp
.port
=
310 evt
->param
.sidr_req_rcvd
.port
;
311 uvt
->data_len
= IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE
;
313 case IB_CM_SIDR_REP_RECEIVED
:
314 ib_ucm_event_sidr_rep_get(&uvt
->resp
.u
.sidr_rep_resp
,
315 &evt
->param
.sidr_rep_rcvd
);
316 uvt
->data_len
= IB_CM_SIDR_REP_PRIVATE_DATA_SIZE
;
317 uvt
->info_len
= evt
->param
.sidr_rep_rcvd
.info_len
;
318 info
= evt
->param
.sidr_rep_rcvd
.info
;
321 uvt
->resp
.u
.send_status
= evt
->param
.send_status
;
326 uvt
->data
= kmemdup(evt
->private_data
, uvt
->data_len
, GFP_KERNEL
);
330 uvt
->resp
.present
|= IB_UCM_PRES_DATA
;
334 uvt
->info
= kmemdup(info
, uvt
->info_len
, GFP_KERNEL
);
338 uvt
->resp
.present
|= IB_UCM_PRES_INFO
;
348 static int ib_ucm_event_handler(struct ib_cm_id
*cm_id
,
349 struct ib_cm_event
*event
)
351 struct ib_ucm_event
*uevent
;
352 struct ib_ucm_context
*ctx
;
355 ctx
= cm_id
->context
;
357 uevent
= kzalloc(sizeof *uevent
, GFP_KERNEL
);
362 uevent
->cm_id
= cm_id
;
363 uevent
->resp
.uid
= ctx
->uid
;
364 uevent
->resp
.id
= ctx
->id
;
365 uevent
->resp
.event
= event
->event
;
367 result
= ib_ucm_event_process(event
, uevent
);
371 mutex_lock(&ctx
->file
->file_mutex
);
372 list_add_tail(&uevent
->file_list
, &ctx
->file
->events
);
373 list_add_tail(&uevent
->ctx_list
, &ctx
->events
);
374 wake_up_interruptible(&ctx
->file
->poll_wait
);
375 mutex_unlock(&ctx
->file
->file_mutex
);
381 /* Destroy new cm_id's */
382 return ib_ucm_new_cm_id(event
->event
);
385 static ssize_t
ib_ucm_event(struct ib_ucm_file
*file
,
386 const char __user
*inbuf
,
387 int in_len
, int out_len
)
389 struct ib_ucm_context
*ctx
;
390 struct ib_ucm_event_get cmd
;
391 struct ib_ucm_event
*uevent
;
394 if (out_len
< sizeof(struct ib_ucm_event_resp
))
397 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
400 mutex_lock(&file
->file_mutex
);
401 while (list_empty(&file
->events
)) {
402 mutex_unlock(&file
->file_mutex
);
404 if (file
->filp
->f_flags
& O_NONBLOCK
)
407 if (wait_event_interruptible(file
->poll_wait
,
408 !list_empty(&file
->events
)))
411 mutex_lock(&file
->file_mutex
);
414 uevent
= list_entry(file
->events
.next
, struct ib_ucm_event
, file_list
);
416 if (ib_ucm_new_cm_id(uevent
->resp
.event
)) {
417 ctx
= ib_ucm_ctx_alloc(file
);
423 ctx
->cm_id
= uevent
->cm_id
;
424 ctx
->cm_id
->context
= ctx
;
425 uevent
->resp
.id
= ctx
->id
;
428 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
429 &uevent
->resp
, sizeof(uevent
->resp
))) {
435 if (cmd
.data_len
< uevent
->data_len
) {
439 if (copy_to_user((void __user
*)(unsigned long)cmd
.data
,
440 uevent
->data
, uevent
->data_len
)) {
447 if (cmd
.info_len
< uevent
->info_len
) {
451 if (copy_to_user((void __user
*)(unsigned long)cmd
.info
,
452 uevent
->info
, uevent
->info_len
)) {
458 list_del(&uevent
->file_list
);
459 list_del(&uevent
->ctx_list
);
460 uevent
->ctx
->events_reported
++;
466 mutex_unlock(&file
->file_mutex
);
470 static ssize_t
ib_ucm_create_id(struct ib_ucm_file
*file
,
471 const char __user
*inbuf
,
472 int in_len
, int out_len
)
474 struct ib_ucm_create_id cmd
;
475 struct ib_ucm_create_id_resp resp
;
476 struct ib_ucm_context
*ctx
;
479 if (out_len
< sizeof(resp
))
482 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
485 mutex_lock(&file
->file_mutex
);
486 ctx
= ib_ucm_ctx_alloc(file
);
487 mutex_unlock(&file
->file_mutex
);
492 ctx
->cm_id
= ib_create_cm_id(file
->device
->ib_dev
,
493 ib_ucm_event_handler
, ctx
);
494 if (IS_ERR(ctx
->cm_id
)) {
495 result
= PTR_ERR(ctx
->cm_id
);
500 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
501 &resp
, sizeof(resp
))) {
508 ib_destroy_cm_id(ctx
->cm_id
);
510 mutex_lock(&ctx_id_mutex
);
511 idr_remove(&ctx_id_table
, ctx
->id
);
512 mutex_unlock(&ctx_id_mutex
);
517 static ssize_t
ib_ucm_destroy_id(struct ib_ucm_file
*file
,
518 const char __user
*inbuf
,
519 int in_len
, int out_len
)
521 struct ib_ucm_destroy_id cmd
;
522 struct ib_ucm_destroy_id_resp resp
;
523 struct ib_ucm_context
*ctx
;
526 if (out_len
< sizeof(resp
))
529 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
532 mutex_lock(&ctx_id_mutex
);
533 ctx
= idr_find(&ctx_id_table
, cmd
.id
);
535 ctx
= ERR_PTR(-ENOENT
);
536 else if (ctx
->file
!= file
)
537 ctx
= ERR_PTR(-EINVAL
);
539 idr_remove(&ctx_id_table
, ctx
->id
);
540 mutex_unlock(&ctx_id_mutex
);
546 wait_for_completion(&ctx
->comp
);
548 /* No new events will be generated after destroying the cm_id. */
549 ib_destroy_cm_id(ctx
->cm_id
);
550 /* Cleanup events not yet reported to the user. */
551 ib_ucm_cleanup_events(ctx
);
553 resp
.events_reported
= ctx
->events_reported
;
554 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
555 &resp
, sizeof(resp
)))
562 static ssize_t
ib_ucm_attr_id(struct ib_ucm_file
*file
,
563 const char __user
*inbuf
,
564 int in_len
, int out_len
)
566 struct ib_ucm_attr_id_resp resp
;
567 struct ib_ucm_attr_id cmd
;
568 struct ib_ucm_context
*ctx
;
571 if (out_len
< sizeof(resp
))
574 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
577 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
581 resp
.service_id
= ctx
->cm_id
->service_id
;
582 resp
.service_mask
= ctx
->cm_id
->service_mask
;
583 resp
.local_id
= ctx
->cm_id
->local_id
;
584 resp
.remote_id
= ctx
->cm_id
->remote_id
;
586 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
587 &resp
, sizeof(resp
)))
594 static ssize_t
ib_ucm_init_qp_attr(struct ib_ucm_file
*file
,
595 const char __user
*inbuf
,
596 int in_len
, int out_len
)
598 struct ib_uverbs_qp_attr resp
;
599 struct ib_ucm_init_qp_attr cmd
;
600 struct ib_ucm_context
*ctx
;
601 struct ib_qp_attr qp_attr
;
604 if (out_len
< sizeof(resp
))
607 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
610 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
614 resp
.qp_attr_mask
= 0;
615 memset(&qp_attr
, 0, sizeof qp_attr
);
616 qp_attr
.qp_state
= cmd
.qp_state
;
617 result
= ib_cm_init_qp_attr(ctx
->cm_id
, &qp_attr
, &resp
.qp_attr_mask
);
621 ib_copy_qp_attr_to_user(&resp
, &qp_attr
);
623 if (copy_to_user((void __user
*)(unsigned long)cmd
.response
,
624 &resp
, sizeof(resp
)))
632 static int ucm_validate_listen(__be64 service_id
, __be64 service_mask
)
634 service_id
&= service_mask
;
636 if (((service_id
& IB_CMA_SERVICE_ID_MASK
) == IB_CMA_SERVICE_ID
) ||
637 ((service_id
& IB_SDP_SERVICE_ID_MASK
) == IB_SDP_SERVICE_ID
))
643 static ssize_t
ib_ucm_listen(struct ib_ucm_file
*file
,
644 const char __user
*inbuf
,
645 int in_len
, int out_len
)
647 struct ib_ucm_listen cmd
;
648 struct ib_ucm_context
*ctx
;
651 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
654 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
658 result
= ucm_validate_listen(cmd
.service_id
, cmd
.service_mask
);
662 result
= ib_cm_listen(ctx
->cm_id
, cmd
.service_id
, cmd
.service_mask
);
668 static ssize_t
ib_ucm_notify(struct ib_ucm_file
*file
,
669 const char __user
*inbuf
,
670 int in_len
, int out_len
)
672 struct ib_ucm_notify cmd
;
673 struct ib_ucm_context
*ctx
;
676 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
679 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
683 result
= ib_cm_notify(ctx
->cm_id
, (enum ib_event_type
) cmd
.event
);
688 static int ib_ucm_alloc_data(const void **dest
, u64 src
, u32 len
)
697 data
= memdup_user((void __user
*)(unsigned long)src
, len
);
699 return PTR_ERR(data
);
705 static int ib_ucm_path_get(struct ib_sa_path_rec
**path
, u64 src
)
707 struct ib_user_path_rec upath
;
708 struct ib_sa_path_rec
*sa_path
;
715 sa_path
= kmalloc(sizeof(*sa_path
), GFP_KERNEL
);
719 if (copy_from_user(&upath
, (void __user
*)(unsigned long)src
,
726 ib_copy_path_rec_from_user(sa_path
, &upath
);
731 static ssize_t
ib_ucm_send_req(struct ib_ucm_file
*file
,
732 const char __user
*inbuf
,
733 int in_len
, int out_len
)
735 struct ib_cm_req_param param
;
736 struct ib_ucm_context
*ctx
;
737 struct ib_ucm_req cmd
;
740 param
.private_data
= NULL
;
741 param
.primary_path
= NULL
;
742 param
.alternate_path
= NULL
;
744 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
747 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
751 result
= ib_ucm_path_get(¶m
.primary_path
, cmd
.primary_path
);
755 result
= ib_ucm_path_get(¶m
.alternate_path
, cmd
.alternate_path
);
759 param
.private_data_len
= cmd
.len
;
760 param
.service_id
= cmd
.sid
;
761 param
.qp_num
= cmd
.qpn
;
762 param
.qp_type
= cmd
.qp_type
;
763 param
.starting_psn
= cmd
.psn
;
764 param
.peer_to_peer
= cmd
.peer_to_peer
;
765 param
.responder_resources
= cmd
.responder_resources
;
766 param
.initiator_depth
= cmd
.initiator_depth
;
767 param
.remote_cm_response_timeout
= cmd
.remote_cm_response_timeout
;
768 param
.flow_control
= cmd
.flow_control
;
769 param
.local_cm_response_timeout
= cmd
.local_cm_response_timeout
;
770 param
.retry_count
= cmd
.retry_count
;
771 param
.rnr_retry_count
= cmd
.rnr_retry_count
;
772 param
.max_cm_retries
= cmd
.max_cm_retries
;
775 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
777 result
= ib_send_cm_req(ctx
->cm_id
, ¶m
);
780 result
= PTR_ERR(ctx
);
783 kfree(param
.private_data
);
784 kfree(param
.primary_path
);
785 kfree(param
.alternate_path
);
789 static ssize_t
ib_ucm_send_rep(struct ib_ucm_file
*file
,
790 const char __user
*inbuf
,
791 int in_len
, int out_len
)
793 struct ib_cm_rep_param param
;
794 struct ib_ucm_context
*ctx
;
795 struct ib_ucm_rep cmd
;
798 param
.private_data
= NULL
;
800 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
803 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
807 param
.qp_num
= cmd
.qpn
;
808 param
.starting_psn
= cmd
.psn
;
809 param
.private_data_len
= cmd
.len
;
810 param
.responder_resources
= cmd
.responder_resources
;
811 param
.initiator_depth
= cmd
.initiator_depth
;
812 param
.failover_accepted
= cmd
.failover_accepted
;
813 param
.flow_control
= cmd
.flow_control
;
814 param
.rnr_retry_count
= cmd
.rnr_retry_count
;
817 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
820 result
= ib_send_cm_rep(ctx
->cm_id
, ¶m
);
823 result
= PTR_ERR(ctx
);
825 kfree(param
.private_data
);
829 static ssize_t
ib_ucm_send_private_data(struct ib_ucm_file
*file
,
830 const char __user
*inbuf
, int in_len
,
831 int (*func
)(struct ib_cm_id
*cm_id
,
832 const void *private_data
,
833 u8 private_data_len
))
835 struct ib_ucm_private_data cmd
;
836 struct ib_ucm_context
*ctx
;
837 const void *private_data
= NULL
;
840 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
843 result
= ib_ucm_alloc_data(&private_data
, cmd
.data
, cmd
.len
);
847 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
849 result
= func(ctx
->cm_id
, private_data
, cmd
.len
);
852 result
= PTR_ERR(ctx
);
858 static ssize_t
ib_ucm_send_rtu(struct ib_ucm_file
*file
,
859 const char __user
*inbuf
,
860 int in_len
, int out_len
)
862 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_rtu
);
865 static ssize_t
ib_ucm_send_dreq(struct ib_ucm_file
*file
,
866 const char __user
*inbuf
,
867 int in_len
, int out_len
)
869 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_dreq
);
872 static ssize_t
ib_ucm_send_drep(struct ib_ucm_file
*file
,
873 const char __user
*inbuf
,
874 int in_len
, int out_len
)
876 return ib_ucm_send_private_data(file
, inbuf
, in_len
, ib_send_cm_drep
);
879 static ssize_t
ib_ucm_send_info(struct ib_ucm_file
*file
,
880 const char __user
*inbuf
, int in_len
,
881 int (*func
)(struct ib_cm_id
*cm_id
,
888 struct ib_ucm_context
*ctx
;
889 struct ib_ucm_info cmd
;
890 const void *data
= NULL
;
891 const void *info
= NULL
;
894 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
897 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.data_len
);
901 result
= ib_ucm_alloc_data(&info
, cmd
.info
, cmd
.info_len
);
905 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
907 result
= func(ctx
->cm_id
, cmd
.status
, info
, cmd
.info_len
,
911 result
= PTR_ERR(ctx
);
919 static ssize_t
ib_ucm_send_rej(struct ib_ucm_file
*file
,
920 const char __user
*inbuf
,
921 int in_len
, int out_len
)
923 return ib_ucm_send_info(file
, inbuf
, in_len
, (void *)ib_send_cm_rej
);
926 static ssize_t
ib_ucm_send_apr(struct ib_ucm_file
*file
,
927 const char __user
*inbuf
,
928 int in_len
, int out_len
)
930 return ib_ucm_send_info(file
, inbuf
, in_len
, (void *)ib_send_cm_apr
);
933 static ssize_t
ib_ucm_send_mra(struct ib_ucm_file
*file
,
934 const char __user
*inbuf
,
935 int in_len
, int out_len
)
937 struct ib_ucm_context
*ctx
;
938 struct ib_ucm_mra cmd
;
939 const void *data
= NULL
;
942 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
945 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.len
);
949 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
951 result
= ib_send_cm_mra(ctx
->cm_id
, cmd
.timeout
, data
, cmd
.len
);
954 result
= PTR_ERR(ctx
);
960 static ssize_t
ib_ucm_send_lap(struct ib_ucm_file
*file
,
961 const char __user
*inbuf
,
962 int in_len
, int out_len
)
964 struct ib_ucm_context
*ctx
;
965 struct ib_sa_path_rec
*path
= NULL
;
966 struct ib_ucm_lap cmd
;
967 const void *data
= NULL
;
970 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
973 result
= ib_ucm_alloc_data(&data
, cmd
.data
, cmd
.len
);
977 result
= ib_ucm_path_get(&path
, cmd
.path
);
981 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
983 result
= ib_send_cm_lap(ctx
->cm_id
, path
, data
, cmd
.len
);
986 result
= PTR_ERR(ctx
);
994 static ssize_t
ib_ucm_send_sidr_req(struct ib_ucm_file
*file
,
995 const char __user
*inbuf
,
996 int in_len
, int out_len
)
998 struct ib_cm_sidr_req_param param
;
999 struct ib_ucm_context
*ctx
;
1000 struct ib_ucm_sidr_req cmd
;
1003 param
.private_data
= NULL
;
1006 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
1009 result
= ib_ucm_alloc_data(¶m
.private_data
, cmd
.data
, cmd
.len
);
1013 result
= ib_ucm_path_get(¶m
.path
, cmd
.path
);
1017 param
.private_data_len
= cmd
.len
;
1018 param
.service_id
= cmd
.sid
;
1019 param
.timeout_ms
= cmd
.timeout
;
1020 param
.max_cm_retries
= cmd
.max_cm_retries
;
1022 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
1024 result
= ib_send_cm_sidr_req(ctx
->cm_id
, ¶m
);
1025 ib_ucm_ctx_put(ctx
);
1027 result
= PTR_ERR(ctx
);
1030 kfree(param
.private_data
);
1035 static ssize_t
ib_ucm_send_sidr_rep(struct ib_ucm_file
*file
,
1036 const char __user
*inbuf
,
1037 int in_len
, int out_len
)
1039 struct ib_cm_sidr_rep_param param
;
1040 struct ib_ucm_sidr_rep cmd
;
1041 struct ib_ucm_context
*ctx
;
1046 if (copy_from_user(&cmd
, inbuf
, sizeof(cmd
)))
1049 result
= ib_ucm_alloc_data(¶m
.private_data
,
1050 cmd
.data
, cmd
.data_len
);
1054 result
= ib_ucm_alloc_data(¶m
.info
, cmd
.info
, cmd
.info_len
);
1058 param
.qp_num
= cmd
.qpn
;
1059 param
.qkey
= cmd
.qkey
;
1060 param
.status
= cmd
.status
;
1061 param
.info_length
= cmd
.info_len
;
1062 param
.private_data_len
= cmd
.data_len
;
1064 ctx
= ib_ucm_ctx_get(file
, cmd
.id
);
1066 result
= ib_send_cm_sidr_rep(ctx
->cm_id
, ¶m
);
1067 ib_ucm_ctx_put(ctx
);
1069 result
= PTR_ERR(ctx
);
1072 kfree(param
.private_data
);
1077 static ssize_t (*ucm_cmd_table
[])(struct ib_ucm_file
*file
,
1078 const char __user
*inbuf
,
1079 int in_len
, int out_len
) = {
1080 [IB_USER_CM_CMD_CREATE_ID
] = ib_ucm_create_id
,
1081 [IB_USER_CM_CMD_DESTROY_ID
] = ib_ucm_destroy_id
,
1082 [IB_USER_CM_CMD_ATTR_ID
] = ib_ucm_attr_id
,
1083 [IB_USER_CM_CMD_LISTEN
] = ib_ucm_listen
,
1084 [IB_USER_CM_CMD_NOTIFY
] = ib_ucm_notify
,
1085 [IB_USER_CM_CMD_SEND_REQ
] = ib_ucm_send_req
,
1086 [IB_USER_CM_CMD_SEND_REP
] = ib_ucm_send_rep
,
1087 [IB_USER_CM_CMD_SEND_RTU
] = ib_ucm_send_rtu
,
1088 [IB_USER_CM_CMD_SEND_DREQ
] = ib_ucm_send_dreq
,
1089 [IB_USER_CM_CMD_SEND_DREP
] = ib_ucm_send_drep
,
1090 [IB_USER_CM_CMD_SEND_REJ
] = ib_ucm_send_rej
,
1091 [IB_USER_CM_CMD_SEND_MRA
] = ib_ucm_send_mra
,
1092 [IB_USER_CM_CMD_SEND_LAP
] = ib_ucm_send_lap
,
1093 [IB_USER_CM_CMD_SEND_APR
] = ib_ucm_send_apr
,
1094 [IB_USER_CM_CMD_SEND_SIDR_REQ
] = ib_ucm_send_sidr_req
,
1095 [IB_USER_CM_CMD_SEND_SIDR_REP
] = ib_ucm_send_sidr_rep
,
1096 [IB_USER_CM_CMD_EVENT
] = ib_ucm_event
,
1097 [IB_USER_CM_CMD_INIT_QP_ATTR
] = ib_ucm_init_qp_attr
,
1100 static ssize_t
ib_ucm_write(struct file
*filp
, const char __user
*buf
,
1101 size_t len
, loff_t
*pos
)
1103 struct ib_ucm_file
*file
= filp
->private_data
;
1104 struct ib_ucm_cmd_hdr hdr
;
1107 if (!ib_safe_file_access(filp
)) {
1108 pr_err_once("ucm_write: process %d (%s) changed security contexts after opening file descriptor, this is not allowed.\n",
1109 task_tgid_vnr(current
), current
->comm
);
1113 if (len
< sizeof(hdr
))
1116 if (copy_from_user(&hdr
, buf
, sizeof(hdr
)))
1119 if (hdr
.cmd
>= ARRAY_SIZE(ucm_cmd_table
))
1122 if (hdr
.in
+ sizeof(hdr
) > len
)
1125 result
= ucm_cmd_table
[hdr
.cmd
](file
, buf
+ sizeof(hdr
),
1133 static unsigned int ib_ucm_poll(struct file
*filp
,
1134 struct poll_table_struct
*wait
)
1136 struct ib_ucm_file
*file
= filp
->private_data
;
1137 unsigned int mask
= 0;
1139 poll_wait(filp
, &file
->poll_wait
, wait
);
1141 if (!list_empty(&file
->events
))
1142 mask
= POLLIN
| POLLRDNORM
;
1148 * ib_ucm_open() does not need the BKL:
1150 * - no global state is referred to;
1151 * - there is no ioctl method to race against;
1152 * - no further module initialization is required for open to work
1153 * after the device is registered.
1155 static int ib_ucm_open(struct inode
*inode
, struct file
*filp
)
1157 struct ib_ucm_file
*file
;
1159 file
= kmalloc(sizeof(*file
), GFP_KERNEL
);
1163 INIT_LIST_HEAD(&file
->events
);
1164 INIT_LIST_HEAD(&file
->ctxs
);
1165 init_waitqueue_head(&file
->poll_wait
);
1167 mutex_init(&file
->file_mutex
);
1169 filp
->private_data
= file
;
1171 file
->device
= container_of(inode
->i_cdev
, struct ib_ucm_device
, cdev
);
1173 return nonseekable_open(inode
, filp
);
1176 static int ib_ucm_close(struct inode
*inode
, struct file
*filp
)
1178 struct ib_ucm_file
*file
= filp
->private_data
;
1179 struct ib_ucm_context
*ctx
;
1181 mutex_lock(&file
->file_mutex
);
1182 while (!list_empty(&file
->ctxs
)) {
1183 ctx
= list_entry(file
->ctxs
.next
,
1184 struct ib_ucm_context
, file_list
);
1185 mutex_unlock(&file
->file_mutex
);
1187 mutex_lock(&ctx_id_mutex
);
1188 idr_remove(&ctx_id_table
, ctx
->id
);
1189 mutex_unlock(&ctx_id_mutex
);
1191 ib_destroy_cm_id(ctx
->cm_id
);
1192 ib_ucm_cleanup_events(ctx
);
1195 mutex_lock(&file
->file_mutex
);
1197 mutex_unlock(&file
->file_mutex
);
1202 static DECLARE_BITMAP(overflow_map
, IB_UCM_MAX_DEVICES
);
1203 static void ib_ucm_release_dev(struct device
*dev
)
1205 struct ib_ucm_device
*ucm_dev
;
1207 ucm_dev
= container_of(dev
, struct ib_ucm_device
, dev
);
1208 cdev_del(&ucm_dev
->cdev
);
1209 if (ucm_dev
->devnum
< IB_UCM_MAX_DEVICES
)
1210 clear_bit(ucm_dev
->devnum
, dev_map
);
1212 clear_bit(ucm_dev
->devnum
- IB_UCM_MAX_DEVICES
, overflow_map
);
1216 static const struct file_operations ucm_fops
= {
1217 .owner
= THIS_MODULE
,
1218 .open
= ib_ucm_open
,
1219 .release
= ib_ucm_close
,
1220 .write
= ib_ucm_write
,
1221 .poll
= ib_ucm_poll
,
1222 .llseek
= no_llseek
,
1225 static ssize_t
show_ibdev(struct device
*dev
, struct device_attribute
*attr
,
1228 struct ib_ucm_device
*ucm_dev
;
1230 ucm_dev
= container_of(dev
, struct ib_ucm_device
, dev
);
1231 return sprintf(buf
, "%s\n", ucm_dev
->ib_dev
->name
);
1233 static DEVICE_ATTR(ibdev
, S_IRUGO
, show_ibdev
, NULL
);
1235 static dev_t overflow_maj
;
1236 static int find_overflow_devnum(void)
1240 if (!overflow_maj
) {
1241 ret
= alloc_chrdev_region(&overflow_maj
, 0, IB_UCM_MAX_DEVICES
,
1244 pr_err("ucm: couldn't register dynamic device number\n");
1249 ret
= find_first_zero_bit(overflow_map
, IB_UCM_MAX_DEVICES
);
1250 if (ret
>= IB_UCM_MAX_DEVICES
)
1256 static void ib_ucm_add_one(struct ib_device
*device
)
1260 struct ib_ucm_device
*ucm_dev
;
1262 if (!device
->alloc_ucontext
|| !rdma_cap_ib_cm(device
, 1))
1265 ucm_dev
= kzalloc(sizeof *ucm_dev
, GFP_KERNEL
);
1269 ucm_dev
->ib_dev
= device
;
1271 devnum
= find_first_zero_bit(dev_map
, IB_UCM_MAX_DEVICES
);
1272 if (devnum
>= IB_UCM_MAX_DEVICES
) {
1273 devnum
= find_overflow_devnum();
1277 ucm_dev
->devnum
= devnum
+ IB_UCM_MAX_DEVICES
;
1278 base
= devnum
+ overflow_maj
;
1279 set_bit(devnum
, overflow_map
);
1281 ucm_dev
->devnum
= devnum
;
1282 base
= devnum
+ IB_UCM_BASE_DEV
;
1283 set_bit(devnum
, dev_map
);
1286 cdev_init(&ucm_dev
->cdev
, &ucm_fops
);
1287 ucm_dev
->cdev
.owner
= THIS_MODULE
;
1288 kobject_set_name(&ucm_dev
->cdev
.kobj
, "ucm%d", ucm_dev
->devnum
);
1289 if (cdev_add(&ucm_dev
->cdev
, base
, 1))
1292 ucm_dev
->dev
.class = &cm_class
;
1293 ucm_dev
->dev
.parent
= device
->dma_device
;
1294 ucm_dev
->dev
.devt
= ucm_dev
->cdev
.dev
;
1295 ucm_dev
->dev
.release
= ib_ucm_release_dev
;
1296 dev_set_name(&ucm_dev
->dev
, "ucm%d", ucm_dev
->devnum
);
1297 if (device_register(&ucm_dev
->dev
))
1300 if (device_create_file(&ucm_dev
->dev
, &dev_attr_ibdev
))
1303 ib_set_client_data(device
, &ucm_client
, ucm_dev
);
1307 device_unregister(&ucm_dev
->dev
);
1309 cdev_del(&ucm_dev
->cdev
);
1310 if (ucm_dev
->devnum
< IB_UCM_MAX_DEVICES
)
1311 clear_bit(devnum
, dev_map
);
1313 clear_bit(devnum
, overflow_map
);
1319 static void ib_ucm_remove_one(struct ib_device
*device
, void *client_data
)
1321 struct ib_ucm_device
*ucm_dev
= client_data
;
1326 device_unregister(&ucm_dev
->dev
);
1329 static CLASS_ATTR_STRING(abi_version
, S_IRUGO
,
1330 __stringify(IB_USER_CM_ABI_VERSION
));
1332 static int __init
ib_ucm_init(void)
1336 ret
= register_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
,
1339 pr_err("ucm: couldn't register device number\n");
1343 ret
= class_create_file(&cm_class
, &class_attr_abi_version
.attr
);
1345 pr_err("ucm: couldn't create abi_version attribute\n");
1349 ret
= ib_register_client(&ucm_client
);
1351 pr_err("ucm: couldn't register client\n");
1357 class_remove_file(&cm_class
, &class_attr_abi_version
.attr
);
1359 unregister_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
);
1364 static void __exit
ib_ucm_cleanup(void)
1366 ib_unregister_client(&ucm_client
);
1367 class_remove_file(&cm_class
, &class_attr_abi_version
.attr
);
1368 unregister_chrdev_region(IB_UCM_BASE_DEV
, IB_UCM_MAX_DEVICES
);
1370 unregister_chrdev_region(overflow_maj
, IB_UCM_MAX_DEVICES
);
1371 idr_destroy(&ctx_id_table
);
1374 module_init(ib_ucm_init
);
1375 module_exit(ib_ucm_cleanup
);