2 * Copyright (c) 2005 Topspin Communications. All rights reserved.
3 * Copyright (c) 2005, 2006 Cisco Systems. All rights reserved.
4 * Copyright (c) 2005 Mellanox Technologies. All rights reserved.
5 * Copyright (c) 2005 Voltaire, Inc. All rights reserved.
6 * Copyright (c) 2005 PathScale, Inc. All rights reserved.
8 * This software is available to you under a choice of one of two
9 * licenses. You may choose to be licensed under the terms of the GNU
10 * General Public License (GPL) Version 2, available from the file
11 * COPYING in the main directory of this source tree, or the
12 * OpenIB.org BSD license below:
14 * Redistribution and use in source and binary forms, with or
15 * without modification, are permitted provided that the following
18 * - Redistributions of source code must retain the above
19 * copyright notice, this list of conditions and the following
22 * - Redistributions in binary form must reproduce the above
23 * copyright notice, this list of conditions and the following
24 * disclaimer in the documentation and/or other materials
25 * provided with the distribution.
27 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
28 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
29 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
30 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
31 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
32 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
33 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
37 #include <linux/module.h>
38 #include <linux/init.h>
39 #include <linux/device.h>
40 #include <linux/err.h>
42 #include <linux/poll.h>
43 #include <linux/sched.h>
44 #include <linux/file.h>
45 #include <linux/cdev.h>
46 #include <linux/anon_inodes.h>
47 #include <linux/slab.h>
49 #include <asm/uaccess.h>
53 MODULE_AUTHOR("Roland Dreier");
54 MODULE_DESCRIPTION("InfiniBand userspace verbs access");
55 MODULE_LICENSE("Dual BSD/GPL");
58 IB_UVERBS_MAJOR
= 231,
59 IB_UVERBS_BASE_MINOR
= 192,
60 IB_UVERBS_MAX_DEVICES
= 32
63 #define IB_UVERBS_BASE_DEV MKDEV(IB_UVERBS_MAJOR, IB_UVERBS_BASE_MINOR)
65 static struct class *uverbs_class
;
67 DEFINE_SPINLOCK(ib_uverbs_idr_lock
);
68 DEFINE_IDR(ib_uverbs_pd_idr
);
69 DEFINE_IDR(ib_uverbs_mr_idr
);
70 DEFINE_IDR(ib_uverbs_mw_idr
);
71 DEFINE_IDR(ib_uverbs_ah_idr
);
72 DEFINE_IDR(ib_uverbs_cq_idr
);
73 DEFINE_IDR(ib_uverbs_qp_idr
);
74 DEFINE_IDR(ib_uverbs_srq_idr
);
75 DEFINE_IDR(ib_uverbs_xrcd_idr
);
76 DEFINE_IDR(ib_uverbs_rule_idr
);
78 static DEFINE_SPINLOCK(map_lock
);
79 static DECLARE_BITMAP(dev_map
, IB_UVERBS_MAX_DEVICES
);
81 static ssize_t (*uverbs_cmd_table
[])(struct ib_uverbs_file
*file
,
82 const char __user
*buf
, int in_len
,
84 [IB_USER_VERBS_CMD_GET_CONTEXT
] = ib_uverbs_get_context
,
85 [IB_USER_VERBS_CMD_QUERY_DEVICE
] = ib_uverbs_query_device
,
86 [IB_USER_VERBS_CMD_QUERY_PORT
] = ib_uverbs_query_port
,
87 [IB_USER_VERBS_CMD_ALLOC_PD
] = ib_uverbs_alloc_pd
,
88 [IB_USER_VERBS_CMD_DEALLOC_PD
] = ib_uverbs_dealloc_pd
,
89 [IB_USER_VERBS_CMD_REG_MR
] = ib_uverbs_reg_mr
,
90 [IB_USER_VERBS_CMD_DEREG_MR
] = ib_uverbs_dereg_mr
,
91 [IB_USER_VERBS_CMD_ALLOC_MW
] = ib_uverbs_alloc_mw
,
92 [IB_USER_VERBS_CMD_DEALLOC_MW
] = ib_uverbs_dealloc_mw
,
93 [IB_USER_VERBS_CMD_CREATE_COMP_CHANNEL
] = ib_uverbs_create_comp_channel
,
94 [IB_USER_VERBS_CMD_CREATE_CQ
] = ib_uverbs_create_cq
,
95 [IB_USER_VERBS_CMD_RESIZE_CQ
] = ib_uverbs_resize_cq
,
96 [IB_USER_VERBS_CMD_POLL_CQ
] = ib_uverbs_poll_cq
,
97 [IB_USER_VERBS_CMD_REQ_NOTIFY_CQ
] = ib_uverbs_req_notify_cq
,
98 [IB_USER_VERBS_CMD_DESTROY_CQ
] = ib_uverbs_destroy_cq
,
99 [IB_USER_VERBS_CMD_CREATE_QP
] = ib_uverbs_create_qp
,
100 [IB_USER_VERBS_CMD_QUERY_QP
] = ib_uverbs_query_qp
,
101 [IB_USER_VERBS_CMD_MODIFY_QP
] = ib_uverbs_modify_qp
,
102 [IB_USER_VERBS_CMD_DESTROY_QP
] = ib_uverbs_destroy_qp
,
103 [IB_USER_VERBS_CMD_POST_SEND
] = ib_uverbs_post_send
,
104 [IB_USER_VERBS_CMD_POST_RECV
] = ib_uverbs_post_recv
,
105 [IB_USER_VERBS_CMD_POST_SRQ_RECV
] = ib_uverbs_post_srq_recv
,
106 [IB_USER_VERBS_CMD_CREATE_AH
] = ib_uverbs_create_ah
,
107 [IB_USER_VERBS_CMD_DESTROY_AH
] = ib_uverbs_destroy_ah
,
108 [IB_USER_VERBS_CMD_ATTACH_MCAST
] = ib_uverbs_attach_mcast
,
109 [IB_USER_VERBS_CMD_DETACH_MCAST
] = ib_uverbs_detach_mcast
,
110 [IB_USER_VERBS_CMD_CREATE_SRQ
] = ib_uverbs_create_srq
,
111 [IB_USER_VERBS_CMD_MODIFY_SRQ
] = ib_uverbs_modify_srq
,
112 [IB_USER_VERBS_CMD_QUERY_SRQ
] = ib_uverbs_query_srq
,
113 [IB_USER_VERBS_CMD_DESTROY_SRQ
] = ib_uverbs_destroy_srq
,
114 [IB_USER_VERBS_CMD_OPEN_XRCD
] = ib_uverbs_open_xrcd
,
115 [IB_USER_VERBS_CMD_CLOSE_XRCD
] = ib_uverbs_close_xrcd
,
116 [IB_USER_VERBS_CMD_CREATE_XSRQ
] = ib_uverbs_create_xsrq
,
117 [IB_USER_VERBS_CMD_OPEN_QP
] = ib_uverbs_open_qp
,
120 static int (*uverbs_ex_cmd_table
[])(struct ib_uverbs_file
*file
,
121 struct ib_udata
*ucore
,
122 struct ib_udata
*uhw
) = {
123 [IB_USER_VERBS_EX_CMD_CREATE_FLOW
] = ib_uverbs_ex_create_flow
,
124 [IB_USER_VERBS_EX_CMD_DESTROY_FLOW
] = ib_uverbs_ex_destroy_flow
127 static void ib_uverbs_add_one(struct ib_device
*device
);
128 static void ib_uverbs_remove_one(struct ib_device
*device
);
130 static void ib_uverbs_release_dev(struct kref
*ref
)
132 struct ib_uverbs_device
*dev
=
133 container_of(ref
, struct ib_uverbs_device
, ref
);
135 complete(&dev
->comp
);
138 static void ib_uverbs_release_event_file(struct kref
*ref
)
140 struct ib_uverbs_event_file
*file
=
141 container_of(ref
, struct ib_uverbs_event_file
, ref
);
146 void ib_uverbs_release_ucq(struct ib_uverbs_file
*file
,
147 struct ib_uverbs_event_file
*ev_file
,
148 struct ib_ucq_object
*uobj
)
150 struct ib_uverbs_event
*evt
, *tmp
;
153 spin_lock_irq(&ev_file
->lock
);
154 list_for_each_entry_safe(evt
, tmp
, &uobj
->comp_list
, obj_list
) {
155 list_del(&evt
->list
);
158 spin_unlock_irq(&ev_file
->lock
);
160 kref_put(&ev_file
->ref
, ib_uverbs_release_event_file
);
163 spin_lock_irq(&file
->async_file
->lock
);
164 list_for_each_entry_safe(evt
, tmp
, &uobj
->async_list
, obj_list
) {
165 list_del(&evt
->list
);
168 spin_unlock_irq(&file
->async_file
->lock
);
171 void ib_uverbs_release_uevent(struct ib_uverbs_file
*file
,
172 struct ib_uevent_object
*uobj
)
174 struct ib_uverbs_event
*evt
, *tmp
;
176 spin_lock_irq(&file
->async_file
->lock
);
177 list_for_each_entry_safe(evt
, tmp
, &uobj
->event_list
, obj_list
) {
178 list_del(&evt
->list
);
181 spin_unlock_irq(&file
->async_file
->lock
);
184 static void ib_uverbs_detach_umcast(struct ib_qp
*qp
,
185 struct ib_uqp_object
*uobj
)
187 struct ib_uverbs_mcast_entry
*mcast
, *tmp
;
189 list_for_each_entry_safe(mcast
, tmp
, &uobj
->mcast_list
, list
) {
190 ib_detach_mcast(qp
, &mcast
->gid
, mcast
->lid
);
191 list_del(&mcast
->list
);
196 static int ib_uverbs_cleanup_ucontext(struct ib_uverbs_file
*file
,
197 struct ib_ucontext
*context
)
199 struct ib_uobject
*uobj
, *tmp
;
204 context
->closing
= 1;
206 list_for_each_entry_safe(uobj
, tmp
, &context
->ah_list
, list
) {
207 struct ib_ah
*ah
= uobj
->object
;
209 idr_remove_uobj(&ib_uverbs_ah_idr
, uobj
);
214 /* Remove MWs before QPs, in order to support type 2A MWs. */
215 list_for_each_entry_safe(uobj
, tmp
, &context
->mw_list
, list
) {
216 struct ib_mw
*mw
= uobj
->object
;
218 idr_remove_uobj(&ib_uverbs_mw_idr
, uobj
);
223 list_for_each_entry_safe(uobj
, tmp
, &context
->rule_list
, list
) {
224 struct ib_flow
*flow_id
= uobj
->object
;
226 idr_remove_uobj(&ib_uverbs_rule_idr
, uobj
);
227 ib_destroy_flow(flow_id
);
231 list_for_each_entry_safe(uobj
, tmp
, &context
->qp_list
, list
) {
232 struct ib_qp
*qp
= uobj
->object
;
233 struct ib_uqp_object
*uqp
=
234 container_of(uobj
, struct ib_uqp_object
, uevent
.uobject
);
236 idr_remove_uobj(&ib_uverbs_qp_idr
, uobj
);
237 if (qp
!= qp
->real_qp
) {
240 ib_uverbs_detach_umcast(qp
, uqp
);
243 ib_uverbs_release_uevent(file
, &uqp
->uevent
);
247 list_for_each_entry_safe(uobj
, tmp
, &context
->cq_list
, list
) {
248 struct ib_cq
*cq
= uobj
->object
;
249 struct ib_uverbs_event_file
*ev_file
= cq
->cq_context
;
250 struct ib_ucq_object
*ucq
=
251 container_of(uobj
, struct ib_ucq_object
, uobject
);
253 idr_remove_uobj(&ib_uverbs_cq_idr
, uobj
);
255 ib_uverbs_release_ucq(file
, ev_file
, ucq
);
259 list_for_each_entry_safe(uobj
, tmp
, &context
->srq_list
, list
) {
260 struct ib_srq
*srq
= uobj
->object
;
261 struct ib_uevent_object
*uevent
=
262 container_of(uobj
, struct ib_uevent_object
, uobject
);
264 idr_remove_uobj(&ib_uverbs_srq_idr
, uobj
);
266 ib_uverbs_release_uevent(file
, uevent
);
270 list_for_each_entry_safe(uobj
, tmp
, &context
->mr_list
, list
) {
271 struct ib_mr
*mr
= uobj
->object
;
273 idr_remove_uobj(&ib_uverbs_mr_idr
, uobj
);
278 mutex_lock(&file
->device
->xrcd_tree_mutex
);
279 list_for_each_entry_safe(uobj
, tmp
, &context
->xrcd_list
, list
) {
280 struct ib_xrcd
*xrcd
= uobj
->object
;
281 struct ib_uxrcd_object
*uxrcd
=
282 container_of(uobj
, struct ib_uxrcd_object
, uobject
);
284 idr_remove_uobj(&ib_uverbs_xrcd_idr
, uobj
);
285 ib_uverbs_dealloc_xrcd(file
->device
, xrcd
);
288 mutex_unlock(&file
->device
->xrcd_tree_mutex
);
290 list_for_each_entry_safe(uobj
, tmp
, &context
->pd_list
, list
) {
291 struct ib_pd
*pd
= uobj
->object
;
293 idr_remove_uobj(&ib_uverbs_pd_idr
, uobj
);
298 return context
->device
->dealloc_ucontext(context
);
301 static void ib_uverbs_release_file(struct kref
*ref
)
303 struct ib_uverbs_file
*file
=
304 container_of(ref
, struct ib_uverbs_file
, ref
);
306 module_put(file
->device
->ib_dev
->owner
);
307 kref_put(&file
->device
->ref
, ib_uverbs_release_dev
);
312 static ssize_t
ib_uverbs_event_read(struct file
*filp
, char __user
*buf
,
313 size_t count
, loff_t
*pos
)
315 struct ib_uverbs_event_file
*file
= filp
->private_data
;
316 struct ib_uverbs_event
*event
;
320 spin_lock_irq(&file
->lock
);
322 while (list_empty(&file
->event_list
)) {
323 spin_unlock_irq(&file
->lock
);
325 if (filp
->f_flags
& O_NONBLOCK
)
328 if (wait_event_interruptible(file
->poll_wait
,
329 !list_empty(&file
->event_list
)))
332 spin_lock_irq(&file
->lock
);
335 event
= list_entry(file
->event_list
.next
, struct ib_uverbs_event
, list
);
338 eventsz
= sizeof (struct ib_uverbs_async_event_desc
);
340 eventsz
= sizeof (struct ib_uverbs_comp_event_desc
);
342 if (eventsz
> count
) {
346 list_del(file
->event_list
.next
);
347 if (event
->counter
) {
349 list_del(&event
->obj_list
);
353 spin_unlock_irq(&file
->lock
);
356 if (copy_to_user(buf
, event
, eventsz
))
367 static unsigned int ib_uverbs_event_poll(struct file
*filp
,
368 struct poll_table_struct
*wait
)
370 unsigned int pollflags
= 0;
371 struct ib_uverbs_event_file
*file
= filp
->private_data
;
373 poll_wait(filp
, &file
->poll_wait
, wait
);
375 spin_lock_irq(&file
->lock
);
376 if (!list_empty(&file
->event_list
))
377 pollflags
= POLLIN
| POLLRDNORM
;
378 spin_unlock_irq(&file
->lock
);
383 static int ib_uverbs_event_fasync(int fd
, struct file
*filp
, int on
)
385 struct ib_uverbs_event_file
*file
= filp
->private_data
;
387 return fasync_helper(fd
, filp
, on
, &file
->async_queue
);
390 static int ib_uverbs_event_close(struct inode
*inode
, struct file
*filp
)
392 struct ib_uverbs_event_file
*file
= filp
->private_data
;
393 struct ib_uverbs_event
*entry
, *tmp
;
395 spin_lock_irq(&file
->lock
);
397 list_for_each_entry_safe(entry
, tmp
, &file
->event_list
, list
) {
399 list_del(&entry
->obj_list
);
402 spin_unlock_irq(&file
->lock
);
404 if (file
->is_async
) {
405 ib_unregister_event_handler(&file
->uverbs_file
->event_handler
);
406 kref_put(&file
->uverbs_file
->ref
, ib_uverbs_release_file
);
408 kref_put(&file
->ref
, ib_uverbs_release_event_file
);
413 static const struct file_operations uverbs_event_fops
= {
414 .owner
= THIS_MODULE
,
415 .read
= ib_uverbs_event_read
,
416 .poll
= ib_uverbs_event_poll
,
417 .release
= ib_uverbs_event_close
,
418 .fasync
= ib_uverbs_event_fasync
,
422 void ib_uverbs_comp_handler(struct ib_cq
*cq
, void *cq_context
)
424 struct ib_uverbs_event_file
*file
= cq_context
;
425 struct ib_ucq_object
*uobj
;
426 struct ib_uverbs_event
*entry
;
432 spin_lock_irqsave(&file
->lock
, flags
);
433 if (file
->is_closed
) {
434 spin_unlock_irqrestore(&file
->lock
, flags
);
438 entry
= kmalloc(sizeof *entry
, GFP_ATOMIC
);
440 spin_unlock_irqrestore(&file
->lock
, flags
);
444 uobj
= container_of(cq
->uobject
, struct ib_ucq_object
, uobject
);
446 entry
->desc
.comp
.cq_handle
= cq
->uobject
->user_handle
;
447 entry
->counter
= &uobj
->comp_events_reported
;
449 list_add_tail(&entry
->list
, &file
->event_list
);
450 list_add_tail(&entry
->obj_list
, &uobj
->comp_list
);
451 spin_unlock_irqrestore(&file
->lock
, flags
);
453 wake_up_interruptible(&file
->poll_wait
);
454 kill_fasync(&file
->async_queue
, SIGIO
, POLL_IN
);
457 static void ib_uverbs_async_handler(struct ib_uverbs_file
*file
,
458 __u64 element
, __u64 event
,
459 struct list_head
*obj_list
,
462 struct ib_uverbs_event
*entry
;
465 spin_lock_irqsave(&file
->async_file
->lock
, flags
);
466 if (file
->async_file
->is_closed
) {
467 spin_unlock_irqrestore(&file
->async_file
->lock
, flags
);
471 entry
= kmalloc(sizeof *entry
, GFP_ATOMIC
);
473 spin_unlock_irqrestore(&file
->async_file
->lock
, flags
);
477 entry
->desc
.async
.element
= element
;
478 entry
->desc
.async
.event_type
= event
;
479 entry
->counter
= counter
;
481 list_add_tail(&entry
->list
, &file
->async_file
->event_list
);
483 list_add_tail(&entry
->obj_list
, obj_list
);
484 spin_unlock_irqrestore(&file
->async_file
->lock
, flags
);
486 wake_up_interruptible(&file
->async_file
->poll_wait
);
487 kill_fasync(&file
->async_file
->async_queue
, SIGIO
, POLL_IN
);
490 void ib_uverbs_cq_event_handler(struct ib_event
*event
, void *context_ptr
)
492 struct ib_ucq_object
*uobj
= container_of(event
->element
.cq
->uobject
,
493 struct ib_ucq_object
, uobject
);
495 ib_uverbs_async_handler(uobj
->uverbs_file
, uobj
->uobject
.user_handle
,
496 event
->event
, &uobj
->async_list
,
497 &uobj
->async_events_reported
);
500 void ib_uverbs_qp_event_handler(struct ib_event
*event
, void *context_ptr
)
502 struct ib_uevent_object
*uobj
;
504 uobj
= container_of(event
->element
.qp
->uobject
,
505 struct ib_uevent_object
, uobject
);
507 ib_uverbs_async_handler(context_ptr
, uobj
->uobject
.user_handle
,
508 event
->event
, &uobj
->event_list
,
509 &uobj
->events_reported
);
512 void ib_uverbs_srq_event_handler(struct ib_event
*event
, void *context_ptr
)
514 struct ib_uevent_object
*uobj
;
516 uobj
= container_of(event
->element
.srq
->uobject
,
517 struct ib_uevent_object
, uobject
);
519 ib_uverbs_async_handler(context_ptr
, uobj
->uobject
.user_handle
,
520 event
->event
, &uobj
->event_list
,
521 &uobj
->events_reported
);
524 void ib_uverbs_event_handler(struct ib_event_handler
*handler
,
525 struct ib_event
*event
)
527 struct ib_uverbs_file
*file
=
528 container_of(handler
, struct ib_uverbs_file
, event_handler
);
530 ib_uverbs_async_handler(file
, event
->element
.port_num
, event
->event
,
534 struct file
*ib_uverbs_alloc_event_file(struct ib_uverbs_file
*uverbs_file
,
537 struct ib_uverbs_event_file
*ev_file
;
540 ev_file
= kmalloc(sizeof *ev_file
, GFP_KERNEL
);
542 return ERR_PTR(-ENOMEM
);
544 kref_init(&ev_file
->ref
);
545 spin_lock_init(&ev_file
->lock
);
546 INIT_LIST_HEAD(&ev_file
->event_list
);
547 init_waitqueue_head(&ev_file
->poll_wait
);
548 ev_file
->uverbs_file
= uverbs_file
;
549 ev_file
->async_queue
= NULL
;
550 ev_file
->is_async
= is_async
;
551 ev_file
->is_closed
= 0;
553 filp
= anon_inode_getfile("[infinibandevent]", &uverbs_event_fops
,
562 * Look up a completion event file by FD. If lookup is successful,
563 * takes a ref to the event file struct that it returns; if
564 * unsuccessful, returns NULL.
566 struct ib_uverbs_event_file
*ib_uverbs_lookup_comp_file(int fd
)
568 struct ib_uverbs_event_file
*ev_file
= NULL
;
569 struct fd f
= fdget(fd
);
574 if (f
.file
->f_op
!= &uverbs_event_fops
)
577 ev_file
= f
.file
->private_data
;
578 if (ev_file
->is_async
) {
583 kref_get(&ev_file
->ref
);
590 static ssize_t
ib_uverbs_write(struct file
*filp
, const char __user
*buf
,
591 size_t count
, loff_t
*pos
)
593 struct ib_uverbs_file
*file
= filp
->private_data
;
594 struct ib_uverbs_cmd_hdr hdr
;
597 if (count
< sizeof hdr
)
600 if (copy_from_user(&hdr
, buf
, sizeof hdr
))
603 flags
= (hdr
.command
&
604 IB_USER_VERBS_CMD_FLAGS_MASK
) >> IB_USER_VERBS_CMD_FLAGS_SHIFT
;
609 if (hdr
.command
& ~(__u32
)(IB_USER_VERBS_CMD_FLAGS_MASK
|
610 IB_USER_VERBS_CMD_COMMAND_MASK
))
613 command
= hdr
.command
& IB_USER_VERBS_CMD_COMMAND_MASK
;
615 if (command
>= ARRAY_SIZE(uverbs_cmd_table
) ||
616 !uverbs_cmd_table
[command
])
619 if (!file
->ucontext
&&
620 command
!= IB_USER_VERBS_CMD_GET_CONTEXT
)
623 if (!(file
->device
->ib_dev
->uverbs_cmd_mask
& (1ull << command
)))
626 if (hdr
.in_words
* 4 != count
)
629 return uverbs_cmd_table
[command
](file
,
634 } else if (flags
== IB_USER_VERBS_CMD_FLAG_EXTENDED
) {
637 struct ib_uverbs_ex_cmd_hdr ex_hdr
;
638 struct ib_udata ucore
;
641 size_t written_count
= count
;
643 if (hdr
.command
& ~(__u32
)(IB_USER_VERBS_CMD_FLAGS_MASK
|
644 IB_USER_VERBS_CMD_COMMAND_MASK
))
647 command
= hdr
.command
& IB_USER_VERBS_CMD_COMMAND_MASK
;
649 if (command
>= ARRAY_SIZE(uverbs_ex_cmd_table
) ||
650 !uverbs_ex_cmd_table
[command
])
656 if (!(file
->device
->ib_dev
->uverbs_ex_cmd_mask
& (1ull << command
)))
659 if (count
< (sizeof(hdr
) + sizeof(ex_hdr
)))
662 if (copy_from_user(&ex_hdr
, buf
+ sizeof(hdr
), sizeof(ex_hdr
)))
665 count
-= sizeof(hdr
) + sizeof(ex_hdr
);
666 buf
+= sizeof(hdr
) + sizeof(ex_hdr
);
668 if ((hdr
.in_words
+ ex_hdr
.provider_in_words
) * 8 != count
)
671 if (ex_hdr
.cmd_hdr_reserved
)
674 if (ex_hdr
.response
) {
675 if (!hdr
.out_words
&& !ex_hdr
.provider_out_words
)
678 if (!access_ok(VERIFY_WRITE
,
679 (void __user
*) (unsigned long) ex_hdr
.response
,
680 (hdr
.out_words
+ ex_hdr
.provider_out_words
) * 8))
683 if (hdr
.out_words
|| ex_hdr
.provider_out_words
)
687 INIT_UDATA_BUF_OR_NULL(&ucore
, buf
, (unsigned long) ex_hdr
.response
,
688 hdr
.in_words
* 8, hdr
.out_words
* 8);
690 INIT_UDATA_BUF_OR_NULL(&uhw
,
692 (unsigned long) ex_hdr
.response
+ ucore
.outlen
,
693 ex_hdr
.provider_in_words
* 8,
694 ex_hdr
.provider_out_words
* 8);
696 err
= uverbs_ex_cmd_table
[command
](file
,
703 return written_count
;
709 static int ib_uverbs_mmap(struct file
*filp
, struct vm_area_struct
*vma
)
711 struct ib_uverbs_file
*file
= filp
->private_data
;
716 return file
->device
->ib_dev
->mmap(file
->ucontext
, vma
);
720 * ib_uverbs_open() does not need the BKL:
722 * - the ib_uverbs_device structures are properly reference counted and
723 * everything else is purely local to the file being created, so
724 * races against other open calls are not a problem;
725 * - there is no ioctl method to race against;
726 * - the open method will either immediately run -ENXIO, or all
727 * required initialization will be done.
729 static int ib_uverbs_open(struct inode
*inode
, struct file
*filp
)
731 struct ib_uverbs_device
*dev
;
732 struct ib_uverbs_file
*file
;
735 dev
= container_of(inode
->i_cdev
, struct ib_uverbs_device
, cdev
);
741 if (!try_module_get(dev
->ib_dev
->owner
)) {
746 file
= kmalloc(sizeof *file
, GFP_KERNEL
);
753 file
->ucontext
= NULL
;
754 file
->async_file
= NULL
;
755 kref_init(&file
->ref
);
756 mutex_init(&file
->mutex
);
758 filp
->private_data
= file
;
760 return nonseekable_open(inode
, filp
);
763 module_put(dev
->ib_dev
->owner
);
766 kref_put(&dev
->ref
, ib_uverbs_release_dev
);
770 static int ib_uverbs_close(struct inode
*inode
, struct file
*filp
)
772 struct ib_uverbs_file
*file
= filp
->private_data
;
774 ib_uverbs_cleanup_ucontext(file
, file
->ucontext
);
776 if (file
->async_file
)
777 kref_put(&file
->async_file
->ref
, ib_uverbs_release_event_file
);
779 kref_put(&file
->ref
, ib_uverbs_release_file
);
784 static const struct file_operations uverbs_fops
= {
785 .owner
= THIS_MODULE
,
786 .write
= ib_uverbs_write
,
787 .open
= ib_uverbs_open
,
788 .release
= ib_uverbs_close
,
792 static const struct file_operations uverbs_mmap_fops
= {
793 .owner
= THIS_MODULE
,
794 .write
= ib_uverbs_write
,
795 .mmap
= ib_uverbs_mmap
,
796 .open
= ib_uverbs_open
,
797 .release
= ib_uverbs_close
,
801 static struct ib_client uverbs_client
= {
803 .add
= ib_uverbs_add_one
,
804 .remove
= ib_uverbs_remove_one
807 static ssize_t
show_ibdev(struct device
*device
, struct device_attribute
*attr
,
810 struct ib_uverbs_device
*dev
= dev_get_drvdata(device
);
815 return sprintf(buf
, "%s\n", dev
->ib_dev
->name
);
817 static DEVICE_ATTR(ibdev
, S_IRUGO
, show_ibdev
, NULL
);
819 static ssize_t
show_dev_abi_version(struct device
*device
,
820 struct device_attribute
*attr
, char *buf
)
822 struct ib_uverbs_device
*dev
= dev_get_drvdata(device
);
827 return sprintf(buf
, "%d\n", dev
->ib_dev
->uverbs_abi_ver
);
829 static DEVICE_ATTR(abi_version
, S_IRUGO
, show_dev_abi_version
, NULL
);
831 static CLASS_ATTR_STRING(abi_version
, S_IRUGO
,
832 __stringify(IB_USER_VERBS_ABI_VERSION
));
834 static dev_t overflow_maj
;
835 static DECLARE_BITMAP(overflow_map
, IB_UVERBS_MAX_DEVICES
);
838 * If we have more than IB_UVERBS_MAX_DEVICES, dynamically overflow by
839 * requesting a new major number and doubling the number of max devices we
840 * support. It's stupid, but simple.
842 static int find_overflow_devnum(void)
847 ret
= alloc_chrdev_region(&overflow_maj
, 0, IB_UVERBS_MAX_DEVICES
,
850 printk(KERN_ERR
"user_verbs: couldn't register dynamic device number\n");
855 ret
= find_first_zero_bit(overflow_map
, IB_UVERBS_MAX_DEVICES
);
856 if (ret
>= IB_UVERBS_MAX_DEVICES
)
862 static void ib_uverbs_add_one(struct ib_device
*device
)
866 struct ib_uverbs_device
*uverbs_dev
;
868 if (!device
->alloc_ucontext
)
871 uverbs_dev
= kzalloc(sizeof *uverbs_dev
, GFP_KERNEL
);
875 kref_init(&uverbs_dev
->ref
);
876 init_completion(&uverbs_dev
->comp
);
877 uverbs_dev
->xrcd_tree
= RB_ROOT
;
878 mutex_init(&uverbs_dev
->xrcd_tree_mutex
);
880 spin_lock(&map_lock
);
881 devnum
= find_first_zero_bit(dev_map
, IB_UVERBS_MAX_DEVICES
);
882 if (devnum
>= IB_UVERBS_MAX_DEVICES
) {
883 spin_unlock(&map_lock
);
884 devnum
= find_overflow_devnum();
888 spin_lock(&map_lock
);
889 uverbs_dev
->devnum
= devnum
+ IB_UVERBS_MAX_DEVICES
;
890 base
= devnum
+ overflow_maj
;
891 set_bit(devnum
, overflow_map
);
893 uverbs_dev
->devnum
= devnum
;
894 base
= devnum
+ IB_UVERBS_BASE_DEV
;
895 set_bit(devnum
, dev_map
);
897 spin_unlock(&map_lock
);
899 uverbs_dev
->ib_dev
= device
;
900 uverbs_dev
->num_comp_vectors
= device
->num_comp_vectors
;
902 cdev_init(&uverbs_dev
->cdev
, NULL
);
903 uverbs_dev
->cdev
.owner
= THIS_MODULE
;
904 uverbs_dev
->cdev
.ops
= device
->mmap
? &uverbs_mmap_fops
: &uverbs_fops
;
905 kobject_set_name(&uverbs_dev
->cdev
.kobj
, "uverbs%d", uverbs_dev
->devnum
);
906 if (cdev_add(&uverbs_dev
->cdev
, base
, 1))
909 uverbs_dev
->dev
= device_create(uverbs_class
, device
->dma_device
,
910 uverbs_dev
->cdev
.dev
, uverbs_dev
,
911 "uverbs%d", uverbs_dev
->devnum
);
912 if (IS_ERR(uverbs_dev
->dev
))
915 if (device_create_file(uverbs_dev
->dev
, &dev_attr_ibdev
))
917 if (device_create_file(uverbs_dev
->dev
, &dev_attr_abi_version
))
920 ib_set_client_data(device
, &uverbs_client
, uverbs_dev
);
925 device_destroy(uverbs_class
, uverbs_dev
->cdev
.dev
);
928 cdev_del(&uverbs_dev
->cdev
);
929 if (uverbs_dev
->devnum
< IB_UVERBS_MAX_DEVICES
)
930 clear_bit(devnum
, dev_map
);
932 clear_bit(devnum
, overflow_map
);
935 kref_put(&uverbs_dev
->ref
, ib_uverbs_release_dev
);
936 wait_for_completion(&uverbs_dev
->comp
);
941 static void ib_uverbs_remove_one(struct ib_device
*device
)
943 struct ib_uverbs_device
*uverbs_dev
= ib_get_client_data(device
, &uverbs_client
);
948 dev_set_drvdata(uverbs_dev
->dev
, NULL
);
949 device_destroy(uverbs_class
, uverbs_dev
->cdev
.dev
);
950 cdev_del(&uverbs_dev
->cdev
);
952 if (uverbs_dev
->devnum
< IB_UVERBS_MAX_DEVICES
)
953 clear_bit(uverbs_dev
->devnum
, dev_map
);
955 clear_bit(uverbs_dev
->devnum
- IB_UVERBS_MAX_DEVICES
, overflow_map
);
957 kref_put(&uverbs_dev
->ref
, ib_uverbs_release_dev
);
958 wait_for_completion(&uverbs_dev
->comp
);
962 static char *uverbs_devnode(struct device
*dev
, umode_t
*mode
)
966 return kasprintf(GFP_KERNEL
, "infiniband/%s", dev_name(dev
));
969 static int __init
ib_uverbs_init(void)
973 ret
= register_chrdev_region(IB_UVERBS_BASE_DEV
, IB_UVERBS_MAX_DEVICES
,
976 printk(KERN_ERR
"user_verbs: couldn't register device number\n");
980 uverbs_class
= class_create(THIS_MODULE
, "infiniband_verbs");
981 if (IS_ERR(uverbs_class
)) {
982 ret
= PTR_ERR(uverbs_class
);
983 printk(KERN_ERR
"user_verbs: couldn't create class infiniband_verbs\n");
987 uverbs_class
->devnode
= uverbs_devnode
;
989 ret
= class_create_file(uverbs_class
, &class_attr_abi_version
.attr
);
991 printk(KERN_ERR
"user_verbs: couldn't create abi_version attribute\n");
995 ret
= ib_register_client(&uverbs_client
);
997 printk(KERN_ERR
"user_verbs: couldn't register client\n");
1004 class_destroy(uverbs_class
);
1007 unregister_chrdev_region(IB_UVERBS_BASE_DEV
, IB_UVERBS_MAX_DEVICES
);
1013 static void __exit
ib_uverbs_cleanup(void)
1015 ib_unregister_client(&uverbs_client
);
1016 class_destroy(uverbs_class
);
1017 unregister_chrdev_region(IB_UVERBS_BASE_DEV
, IB_UVERBS_MAX_DEVICES
);
1019 unregister_chrdev_region(overflow_maj
, IB_UVERBS_MAX_DEVICES
);
1020 idr_destroy(&ib_uverbs_pd_idr
);
1021 idr_destroy(&ib_uverbs_mr_idr
);
1022 idr_destroy(&ib_uverbs_mw_idr
);
1023 idr_destroy(&ib_uverbs_ah_idr
);
1024 idr_destroy(&ib_uverbs_cq_idr
);
1025 idr_destroy(&ib_uverbs_qp_idr
);
1026 idr_destroy(&ib_uverbs_srq_idr
);
1029 module_init(ib_uverbs_init
);
1030 module_exit(ib_uverbs_cleanup
);