1 /*
2  * Copyright (c) 2005 Topspin Communications.  All rights reserved.
3  * Copyright (c) 2005, 2006, 2007 Cisco Systems.  All rights reserved.
4  * Copyright (c) 2005 PathScale, Inc.  All rights reserved.
5  * Copyright (c) 2006 Mellanox Technologies.  All rights reserved.
6  *
7  * This software is available to you under a choice of one of two
8  * licenses.  You may choose to be licensed under the terms of the GNU
9  * General Public License (GPL) Version 2, available from the file
10  * COPYING in the main directory of this source tree, or the
11  * OpenIB.org BSD license below:
12  *
13  *     Redistribution and use in source and binary forms, with or
14  *     without modification, are permitted provided that the following
15  *     conditions are met:
16  *
17  *      - Redistributions of source code must retain the above
18  *        copyright notice, this list of conditions and the following
19  *        disclaimer.
20  *
21  *      - Redistributions in binary form must reproduce the above
22  *        copyright notice, this list of conditions and the following
23  *        disclaimer in the documentation and/or other materials
24  *        provided with the distribution.
25  *
26  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
27  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
28  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
29  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
30  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
31  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
32  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
33  * SOFTWARE.
34  */
35 
36 #include <linux/file.h>
37 #include <linux/fs.h>
38 #include <linux/slab.h>
39 #include <linux/sched.h>
40 
41 #include <linux/uaccess.h>
42 
43 #include <rdma/uverbs_types.h>
44 #include <rdma/uverbs_std_types.h>
45 #include "rdma_core.h"
46 
47 #include "uverbs.h"
48 #include "core_priv.h"
49 
50 /*
51  * Copy a response to userspace. If the provided 'resp' is larger than the
52  * user buffer it is silently truncated. If the user provided a larger buffer
53  * then the trailing portion is zero filled.
54  *
55  * These semantics are intended to support future extension of the output
56  * structures.
57  */
uverbs_response(struct uverbs_attr_bundle * attrs,const void * resp,size_t resp_len)58 static int uverbs_response(struct uverbs_attr_bundle *attrs, const void *resp,
59 			   size_t resp_len)
60 {
61 	int ret;
62 
63 	if (uverbs_attr_is_valid(attrs, UVERBS_ATTR_CORE_OUT))
64 		return uverbs_copy_to_struct_or_zero(
65 			attrs, UVERBS_ATTR_CORE_OUT, resp, resp_len);
66 
67 	if (copy_to_user(attrs->ucore.outbuf, resp,
68 			 min(attrs->ucore.outlen, resp_len)))
69 		return -EFAULT;
70 
71 	if (resp_len < attrs->ucore.outlen) {
72 		/*
73 		 * Zero fill any extra memory that user
74 		 * space might have provided.
75 		 */
76 		ret = clear_user(attrs->ucore.outbuf + resp_len,
77 				 attrs->ucore.outlen - resp_len);
78 		if (ret)
79 			return -EFAULT;
80 	}
81 
82 	return 0;
83 }
84 
85 /*
86  * Copy a request from userspace. If the provided 'req' is larger than the
87  * user buffer then the user buffer is zero extended into the 'req'. If 'req'
88  * is smaller than the user buffer then the uncopied bytes in the user buffer
89  * must be zero.
90  */
uverbs_request(struct uverbs_attr_bundle * attrs,void * req,size_t req_len)91 static int uverbs_request(struct uverbs_attr_bundle *attrs, void *req,
92 			  size_t req_len)
93 {
94 	if (copy_from_user(req, attrs->ucore.inbuf,
95 			   min(attrs->ucore.inlen, req_len)))
96 		return -EFAULT;
97 
98 	if (attrs->ucore.inlen < req_len) {
99 		memset(req + attrs->ucore.inlen, 0,
100 		       req_len - attrs->ucore.inlen);
101 	} else if (attrs->ucore.inlen > req_len) {
102 		if (!ib_is_buffer_cleared(attrs->ucore.inbuf + req_len,
103 					  attrs->ucore.inlen - req_len))
104 			return -EOPNOTSUPP;
105 	}
106 	return 0;
107 }
108 
109 /*
110  * Generate the value for the 'response_length' protocol used by write_ex.
111  * This is the number of bytes the kernel actually wrote. Userspace can use
112  * this to detect what structure members in the response the kernel
113  * understood.
114  */
uverbs_response_length(struct uverbs_attr_bundle * attrs,size_t resp_len)115 static u32 uverbs_response_length(struct uverbs_attr_bundle *attrs,
116 				  size_t resp_len)
117 {
118 	return min_t(size_t, attrs->ucore.outlen, resp_len);
119 }
120 
121 /*
122  * The iterator version of the request interface is for handlers that need to
123  * step over a flex array at the end of a command header.
124  */
125 struct uverbs_req_iter {
126 	const void __user *cur;
127 	const void __user *end;
128 };
129 
uverbs_request_start(struct uverbs_attr_bundle * attrs,struct uverbs_req_iter * iter,void * req,size_t req_len)130 static int uverbs_request_start(struct uverbs_attr_bundle *attrs,
131 				struct uverbs_req_iter *iter,
132 				void *req,
133 				size_t req_len)
134 {
135 	if (attrs->ucore.inlen < req_len)
136 		return -ENOSPC;
137 
138 	if (copy_from_user(req, attrs->ucore.inbuf, req_len))
139 		return -EFAULT;
140 
141 	iter->cur = attrs->ucore.inbuf + req_len;
142 	iter->end = attrs->ucore.inbuf + attrs->ucore.inlen;
143 	return 0;
144 }
145 
uverbs_request_next(struct uverbs_req_iter * iter,void * val,size_t len)146 static int uverbs_request_next(struct uverbs_req_iter *iter, void *val,
147 			       size_t len)
148 {
149 	if (iter->cur + len > iter->end)
150 		return -ENOSPC;
151 
152 	if (copy_from_user(val, iter->cur, len))
153 		return -EFAULT;
154 
155 	iter->cur += len;
156 	return 0;
157 }
158 
uverbs_request_next_ptr(struct uverbs_req_iter * iter,size_t len)159 static const void __user *uverbs_request_next_ptr(struct uverbs_req_iter *iter,
160 						  size_t len)
161 {
162 	const void __user *res = iter->cur;
163 
164 	if (iter->cur + len > iter->end)
165 		return (void __force __user *)ERR_PTR(-ENOSPC);
166 	iter->cur += len;
167 	return res;
168 }
169 
uverbs_request_finish(struct uverbs_req_iter * iter)170 static int uverbs_request_finish(struct uverbs_req_iter *iter)
171 {
172 	if (!ib_is_buffer_cleared(iter->cur, iter->end - iter->cur))
173 		return -EOPNOTSUPP;
174 	return 0;
175 }
176 
177 /*
178  * When calling a destroy function during an error unwind we need to pass in
179  * the udata that is sanitized of all user arguments. Ie from the driver
180  * perspective it looks like no udata was passed.
181  */
uverbs_get_cleared_udata(struct uverbs_attr_bundle * attrs)182 struct ib_udata *uverbs_get_cleared_udata(struct uverbs_attr_bundle *attrs)
183 {
184 	attrs->driver_udata = (struct ib_udata){};
185 	return &attrs->driver_udata;
186 }
187 
188 static struct ib_uverbs_completion_event_file *
_ib_uverbs_lookup_comp_file(s32 fd,struct uverbs_attr_bundle * attrs)189 _ib_uverbs_lookup_comp_file(s32 fd, struct uverbs_attr_bundle *attrs)
190 {
191 	struct ib_uobject *uobj = ufd_get_read(UVERBS_OBJECT_COMP_CHANNEL,
192 					       fd, attrs);
193 
194 	if (IS_ERR(uobj))
195 		return (void *)uobj;
196 
197 	uverbs_uobject_get(uobj);
198 	uobj_put_read(uobj);
199 
200 	return container_of(uobj, struct ib_uverbs_completion_event_file,
201 			    uobj);
202 }
203 #define ib_uverbs_lookup_comp_file(_fd, _ufile)                                \
204 	_ib_uverbs_lookup_comp_file((_fd)*typecheck(s32, _fd), _ufile)
205 
ib_alloc_ucontext(struct uverbs_attr_bundle * attrs)206 int ib_alloc_ucontext(struct uverbs_attr_bundle *attrs)
207 {
208 	struct ib_uverbs_file *ufile = attrs->ufile;
209 	struct ib_ucontext *ucontext;
210 	struct ib_device *ib_dev;
211 
212 	ib_dev = srcu_dereference(ufile->device->ib_dev,
213 				  &ufile->device->disassociate_srcu);
214 	if (!ib_dev)
215 		return -EIO;
216 
217 	ucontext = rdma_zalloc_drv_obj(ib_dev, ib_ucontext);
218 	if (!ucontext)
219 		return -ENOMEM;
220 
221 	ucontext->device = ib_dev;
222 	ucontext->ufile = ufile;
223 	xa_init_flags(&ucontext->mmap_xa, XA_FLAGS_ALLOC);
224 
225 	rdma_restrack_new(&ucontext->res, RDMA_RESTRACK_CTX);
226 	rdma_restrack_set_name(&ucontext->res, NULL);
227 	attrs->context = ucontext;
228 	return 0;
229 }
230 
ib_init_ucontext(struct uverbs_attr_bundle * attrs)231 int ib_init_ucontext(struct uverbs_attr_bundle *attrs)
232 {
233 	struct ib_ucontext *ucontext = attrs->context;
234 	struct ib_uverbs_file *file = attrs->ufile;
235 	int ret;
236 
237 	if (!down_read_trylock(&file->hw_destroy_rwsem))
238 		return -EIO;
239 	mutex_lock(&file->ucontext_lock);
240 	if (file->ucontext) {
241 		ret = -EINVAL;
242 		goto err;
243 	}
244 
245 	ret = ib_rdmacg_try_charge(&ucontext->cg_obj, ucontext->device,
246 				   RDMACG_RESOURCE_HCA_HANDLE);
247 	if (ret)
248 		goto err;
249 
250 	ret = ucontext->device->ops.alloc_ucontext(ucontext,
251 						   &attrs->driver_udata);
252 	if (ret)
253 		goto err_uncharge;
254 
255 	rdma_restrack_add(&ucontext->res);
256 
257 	/*
258 	 * Make sure that ib_uverbs_get_ucontext() sees the pointer update
259 	 * only after all writes to setup the ucontext have completed
260 	 */
261 	smp_store_release(&file->ucontext, ucontext);
262 
263 	mutex_unlock(&file->ucontext_lock);
264 	up_read(&file->hw_destroy_rwsem);
265 	return 0;
266 
267 err_uncharge:
268 	ib_rdmacg_uncharge(&ucontext->cg_obj, ucontext->device,
269 			   RDMACG_RESOURCE_HCA_HANDLE);
270 err:
271 	mutex_unlock(&file->ucontext_lock);
272 	up_read(&file->hw_destroy_rwsem);
273 	return ret;
274 }
275 
ib_uverbs_get_context(struct uverbs_attr_bundle * attrs)276 static int ib_uverbs_get_context(struct uverbs_attr_bundle *attrs)
277 {
278 	struct ib_uverbs_get_context_resp resp;
279 	struct ib_uverbs_get_context cmd;
280 	struct ib_device *ib_dev;
281 	struct ib_uobject *uobj;
282 	int ret;
283 
284 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
285 	if (ret)
286 		return ret;
287 
288 	ret = ib_alloc_ucontext(attrs);
289 	if (ret)
290 		return ret;
291 
292 	uobj = uobj_alloc(UVERBS_OBJECT_ASYNC_EVENT, attrs, &ib_dev);
293 	if (IS_ERR(uobj)) {
294 		ret = PTR_ERR(uobj);
295 		goto err_ucontext;
296 	}
297 
298 	resp = (struct ib_uverbs_get_context_resp){
299 		.num_comp_vectors = attrs->ufile->device->num_comp_vectors,
300 		.async_fd = uobj->id,
301 	};
302 	ret = uverbs_response(attrs, &resp, sizeof(resp));
303 	if (ret)
304 		goto err_uobj;
305 
306 	ret = ib_init_ucontext(attrs);
307 	if (ret)
308 		goto err_uobj;
309 
310 	ib_uverbs_init_async_event_file(
311 		container_of(uobj, struct ib_uverbs_async_event_file, uobj));
312 	rdma_alloc_commit_uobject(uobj, attrs);
313 	return 0;
314 
315 err_uobj:
316 	rdma_alloc_abort_uobject(uobj, attrs, false);
317 err_ucontext:
318 	rdma_restrack_put(&attrs->context->res);
319 	kfree(attrs->context);
320 	attrs->context = NULL;
321 	return ret;
322 }
323 
copy_query_dev_fields(struct ib_ucontext * ucontext,struct ib_uverbs_query_device_resp * resp,struct ib_device_attr * attr)324 static void copy_query_dev_fields(struct ib_ucontext *ucontext,
325 				  struct ib_uverbs_query_device_resp *resp,
326 				  struct ib_device_attr *attr)
327 {
328 	struct ib_device *ib_dev = ucontext->device;
329 
330 	resp->fw_ver		= attr->fw_ver;
331 	resp->node_guid		= ib_dev->node_guid;
332 	resp->sys_image_guid	= attr->sys_image_guid;
333 	resp->max_mr_size	= attr->max_mr_size;
334 	resp->page_size_cap	= attr->page_size_cap;
335 	resp->vendor_id		= attr->vendor_id;
336 	resp->vendor_part_id	= attr->vendor_part_id;
337 	resp->hw_ver		= attr->hw_ver;
338 	resp->max_qp		= attr->max_qp;
339 	resp->max_qp_wr		= attr->max_qp_wr;
340 	resp->device_cap_flags  = lower_32_bits(attr->device_cap_flags);
341 	resp->max_sge		= min(attr->max_send_sge, attr->max_recv_sge);
342 	resp->max_sge_rd	= attr->max_sge_rd;
343 	resp->max_cq		= attr->max_cq;
344 	resp->max_cqe		= attr->max_cqe;
345 	resp->max_mr		= attr->max_mr;
346 	resp->max_pd		= attr->max_pd;
347 	resp->max_qp_rd_atom	= attr->max_qp_rd_atom;
348 	resp->max_ee_rd_atom	= attr->max_ee_rd_atom;
349 	resp->max_res_rd_atom	= attr->max_res_rd_atom;
350 	resp->max_qp_init_rd_atom	= attr->max_qp_init_rd_atom;
351 	resp->max_ee_init_rd_atom	= attr->max_ee_init_rd_atom;
352 	resp->atomic_cap		= attr->atomic_cap;
353 	resp->max_ee			= attr->max_ee;
354 	resp->max_rdd			= attr->max_rdd;
355 	resp->max_mw			= attr->max_mw;
356 	resp->max_raw_ipv6_qp		= attr->max_raw_ipv6_qp;
357 	resp->max_raw_ethy_qp		= attr->max_raw_ethy_qp;
358 	resp->max_mcast_grp		= attr->max_mcast_grp;
359 	resp->max_mcast_qp_attach	= attr->max_mcast_qp_attach;
360 	resp->max_total_mcast_qp_attach	= attr->max_total_mcast_qp_attach;
361 	resp->max_ah			= attr->max_ah;
362 	resp->max_srq			= attr->max_srq;
363 	resp->max_srq_wr		= attr->max_srq_wr;
364 	resp->max_srq_sge		= attr->max_srq_sge;
365 	resp->max_pkeys			= attr->max_pkeys;
366 	resp->local_ca_ack_delay	= attr->local_ca_ack_delay;
367 	resp->phys_port_cnt = min_t(u32, ib_dev->phys_port_cnt, U8_MAX);
368 }
369 
ib_uverbs_query_device(struct uverbs_attr_bundle * attrs)370 static int ib_uverbs_query_device(struct uverbs_attr_bundle *attrs)
371 {
372 	struct ib_uverbs_query_device      cmd;
373 	struct ib_uverbs_query_device_resp resp;
374 	struct ib_ucontext *ucontext;
375 	int ret;
376 
377 	ucontext = ib_uverbs_get_ucontext(attrs);
378 	if (IS_ERR(ucontext))
379 		return PTR_ERR(ucontext);
380 
381 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
382 	if (ret)
383 		return ret;
384 
385 	memset(&resp, 0, sizeof resp);
386 	copy_query_dev_fields(ucontext, &resp, &ucontext->device->attrs);
387 
388 	return uverbs_response(attrs, &resp, sizeof(resp));
389 }
390 
ib_uverbs_query_port(struct uverbs_attr_bundle * attrs)391 static int ib_uverbs_query_port(struct uverbs_attr_bundle *attrs)
392 {
393 	struct ib_uverbs_query_port      cmd;
394 	struct ib_uverbs_query_port_resp resp;
395 	struct ib_port_attr              attr;
396 	int                              ret;
397 	struct ib_ucontext *ucontext;
398 	struct ib_device *ib_dev;
399 
400 	ucontext = ib_uverbs_get_ucontext(attrs);
401 	if (IS_ERR(ucontext))
402 		return PTR_ERR(ucontext);
403 	ib_dev = ucontext->device;
404 
405 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
406 	if (ret)
407 		return ret;
408 
409 	ret = ib_query_port(ib_dev, cmd.port_num, &attr);
410 	if (ret)
411 		return ret;
412 
413 	memset(&resp, 0, sizeof resp);
414 	copy_port_attr_to_resp(&attr, &resp, ib_dev, cmd.port_num);
415 
416 	return uverbs_response(attrs, &resp, sizeof(resp));
417 }
418 
ib_uverbs_alloc_pd(struct uverbs_attr_bundle * attrs)419 static int ib_uverbs_alloc_pd(struct uverbs_attr_bundle *attrs)
420 {
421 	struct ib_uverbs_alloc_pd_resp resp = {};
422 	struct ib_uverbs_alloc_pd      cmd;
423 	struct ib_uobject             *uobj;
424 	struct ib_pd                  *pd;
425 	int                            ret;
426 	struct ib_device *ib_dev;
427 
428 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
429 	if (ret)
430 		return ret;
431 
432 	uobj = uobj_alloc(UVERBS_OBJECT_PD, attrs, &ib_dev);
433 	if (IS_ERR(uobj))
434 		return PTR_ERR(uobj);
435 
436 	pd = rdma_zalloc_drv_obj(ib_dev, ib_pd);
437 	if (!pd) {
438 		ret = -ENOMEM;
439 		goto err;
440 	}
441 
442 	pd->device  = ib_dev;
443 	pd->uobject = uobj;
444 	atomic_set(&pd->usecnt, 0);
445 
446 	rdma_restrack_new(&pd->res, RDMA_RESTRACK_PD);
447 	rdma_restrack_set_name(&pd->res, NULL);
448 
449 	ret = ib_dev->ops.alloc_pd(pd, &attrs->driver_udata);
450 	if (ret)
451 		goto err_alloc;
452 	rdma_restrack_add(&pd->res);
453 
454 	uobj->object = pd;
455 	uobj_finalize_uobj_create(uobj, attrs);
456 
457 	resp.pd_handle = uobj->id;
458 	return uverbs_response(attrs, &resp, sizeof(resp));
459 
460 err_alloc:
461 	rdma_restrack_put(&pd->res);
462 	kfree(pd);
463 err:
464 	uobj_alloc_abort(uobj, attrs);
465 	return ret;
466 }
467 
ib_uverbs_dealloc_pd(struct uverbs_attr_bundle * attrs)468 static int ib_uverbs_dealloc_pd(struct uverbs_attr_bundle *attrs)
469 {
470 	struct ib_uverbs_dealloc_pd cmd;
471 	int ret;
472 
473 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
474 	if (ret)
475 		return ret;
476 
477 	return uobj_perform_destroy(UVERBS_OBJECT_PD, cmd.pd_handle, attrs);
478 }
479 
480 struct xrcd_table_entry {
481 	struct rb_node  node;
482 	struct ib_xrcd *xrcd;
483 	struct inode   *inode;
484 };
485 
xrcd_table_insert(struct ib_uverbs_device * dev,struct inode * inode,struct ib_xrcd * xrcd)486 static int xrcd_table_insert(struct ib_uverbs_device *dev,
487 			    struct inode *inode,
488 			    struct ib_xrcd *xrcd)
489 {
490 	struct xrcd_table_entry *entry, *scan;
491 	struct rb_node **p = &dev->xrcd_tree.rb_node;
492 	struct rb_node *parent = NULL;
493 
494 	entry = kmalloc(sizeof *entry, GFP_KERNEL);
495 	if (!entry)
496 		return -ENOMEM;
497 
498 	entry->xrcd  = xrcd;
499 	entry->inode = inode;
500 
501 	while (*p) {
502 		parent = *p;
503 		scan = rb_entry(parent, struct xrcd_table_entry, node);
504 
505 		if (inode < scan->inode) {
506 			p = &(*p)->rb_left;
507 		} else if (inode > scan->inode) {
508 			p = &(*p)->rb_right;
509 		} else {
510 			kfree(entry);
511 			return -EEXIST;
512 		}
513 	}
514 
515 	rb_link_node(&entry->node, parent, p);
516 	rb_insert_color(&entry->node, &dev->xrcd_tree);
517 	igrab(inode);
518 	return 0;
519 }
520 
xrcd_table_search(struct ib_uverbs_device * dev,struct inode * inode)521 static struct xrcd_table_entry *xrcd_table_search(struct ib_uverbs_device *dev,
522 						  struct inode *inode)
523 {
524 	struct xrcd_table_entry *entry;
525 	struct rb_node *p = dev->xrcd_tree.rb_node;
526 
527 	while (p) {
528 		entry = rb_entry(p, struct xrcd_table_entry, node);
529 
530 		if (inode < entry->inode)
531 			p = p->rb_left;
532 		else if (inode > entry->inode)
533 			p = p->rb_right;
534 		else
535 			return entry;
536 	}
537 
538 	return NULL;
539 }
540 
find_xrcd(struct ib_uverbs_device * dev,struct inode * inode)541 static struct ib_xrcd *find_xrcd(struct ib_uverbs_device *dev, struct inode *inode)
542 {
543 	struct xrcd_table_entry *entry;
544 
545 	entry = xrcd_table_search(dev, inode);
546 	if (!entry)
547 		return NULL;
548 
549 	return entry->xrcd;
550 }
551 
xrcd_table_delete(struct ib_uverbs_device * dev,struct inode * inode)552 static void xrcd_table_delete(struct ib_uverbs_device *dev,
553 			      struct inode *inode)
554 {
555 	struct xrcd_table_entry *entry;
556 
557 	entry = xrcd_table_search(dev, inode);
558 	if (entry) {
559 		iput(inode);
560 		rb_erase(&entry->node, &dev->xrcd_tree);
561 		kfree(entry);
562 	}
563 }
564 
ib_uverbs_open_xrcd(struct uverbs_attr_bundle * attrs)565 static int ib_uverbs_open_xrcd(struct uverbs_attr_bundle *attrs)
566 {
567 	struct ib_uverbs_device *ibudev = attrs->ufile->device;
568 	struct ib_uverbs_open_xrcd_resp	resp = {};
569 	struct ib_uverbs_open_xrcd	cmd;
570 	struct ib_uxrcd_object         *obj;
571 	struct ib_xrcd                 *xrcd = NULL;
572 	struct inode                   *inode = NULL;
573 	int				new_xrcd = 0;
574 	struct ib_device *ib_dev;
575 	struct fd f = {};
576 	int ret;
577 
578 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
579 	if (ret)
580 		return ret;
581 
582 	mutex_lock(&ibudev->xrcd_tree_mutex);
583 
584 	if (cmd.fd != -1) {
585 		/* search for file descriptor */
586 		f = fdget(cmd.fd);
587 		if (!f.file) {
588 			ret = -EBADF;
589 			goto err_tree_mutex_unlock;
590 		}
591 
592 		inode = file_inode(f.file);
593 		xrcd = find_xrcd(ibudev, inode);
594 		if (!xrcd && !(cmd.oflags & O_CREAT)) {
595 			/* no file descriptor. Need CREATE flag */
596 			ret = -EAGAIN;
597 			goto err_tree_mutex_unlock;
598 		}
599 
600 		if (xrcd && cmd.oflags & O_EXCL) {
601 			ret = -EINVAL;
602 			goto err_tree_mutex_unlock;
603 		}
604 	}
605 
606 	obj = (struct ib_uxrcd_object *)uobj_alloc(UVERBS_OBJECT_XRCD, attrs,
607 						   &ib_dev);
608 	if (IS_ERR(obj)) {
609 		ret = PTR_ERR(obj);
610 		goto err_tree_mutex_unlock;
611 	}
612 
613 	if (!xrcd) {
614 		xrcd = ib_alloc_xrcd_user(ib_dev, inode, &attrs->driver_udata);
615 		if (IS_ERR(xrcd)) {
616 			ret = PTR_ERR(xrcd);
617 			goto err;
618 		}
619 		new_xrcd = 1;
620 	}
621 
622 	atomic_set(&obj->refcnt, 0);
623 	obj->uobject.object = xrcd;
624 
625 	if (inode) {
626 		if (new_xrcd) {
627 			/* create new inode/xrcd table entry */
628 			ret = xrcd_table_insert(ibudev, inode, xrcd);
629 			if (ret)
630 				goto err_dealloc_xrcd;
631 		}
632 		atomic_inc(&xrcd->usecnt);
633 	}
634 
635 	if (f.file)
636 		fdput(f);
637 
638 	mutex_unlock(&ibudev->xrcd_tree_mutex);
639 	uobj_finalize_uobj_create(&obj->uobject, attrs);
640 
641 	resp.xrcd_handle = obj->uobject.id;
642 	return uverbs_response(attrs, &resp, sizeof(resp));
643 
644 err_dealloc_xrcd:
645 	ib_dealloc_xrcd_user(xrcd, uverbs_get_cleared_udata(attrs));
646 
647 err:
648 	uobj_alloc_abort(&obj->uobject, attrs);
649 
650 err_tree_mutex_unlock:
651 	if (f.file)
652 		fdput(f);
653 
654 	mutex_unlock(&ibudev->xrcd_tree_mutex);
655 
656 	return ret;
657 }
658 
ib_uverbs_close_xrcd(struct uverbs_attr_bundle * attrs)659 static int ib_uverbs_close_xrcd(struct uverbs_attr_bundle *attrs)
660 {
661 	struct ib_uverbs_close_xrcd cmd;
662 	int ret;
663 
664 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
665 	if (ret)
666 		return ret;
667 
668 	return uobj_perform_destroy(UVERBS_OBJECT_XRCD, cmd.xrcd_handle, attrs);
669 }
670 
ib_uverbs_dealloc_xrcd(struct ib_uobject * uobject,struct ib_xrcd * xrcd,enum rdma_remove_reason why,struct uverbs_attr_bundle * attrs)671 int ib_uverbs_dealloc_xrcd(struct ib_uobject *uobject, struct ib_xrcd *xrcd,
672 			   enum rdma_remove_reason why,
673 			   struct uverbs_attr_bundle *attrs)
674 {
675 	struct inode *inode;
676 	int ret;
677 	struct ib_uverbs_device *dev = attrs->ufile->device;
678 
679 	inode = xrcd->inode;
680 	if (inode && !atomic_dec_and_test(&xrcd->usecnt))
681 		return 0;
682 
683 	ret = ib_dealloc_xrcd_user(xrcd, &attrs->driver_udata);
684 	if (ret) {
685 		atomic_inc(&xrcd->usecnt);
686 		return ret;
687 	}
688 
689 	if (inode)
690 		xrcd_table_delete(dev, inode);
691 
692 	return 0;
693 }
694 
ib_uverbs_reg_mr(struct uverbs_attr_bundle * attrs)695 static int ib_uverbs_reg_mr(struct uverbs_attr_bundle *attrs)
696 {
697 	struct ib_uverbs_reg_mr_resp resp = {};
698 	struct ib_uverbs_reg_mr      cmd;
699 	struct ib_uobject           *uobj;
700 	struct ib_pd                *pd;
701 	struct ib_mr                *mr;
702 	int                          ret;
703 	struct ib_device *ib_dev;
704 
705 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
706 	if (ret)
707 		return ret;
708 
709 	if ((cmd.start & ~PAGE_MASK) != (cmd.hca_va & ~PAGE_MASK))
710 		return -EINVAL;
711 
712 	uobj = uobj_alloc(UVERBS_OBJECT_MR, attrs, &ib_dev);
713 	if (IS_ERR(uobj))
714 		return PTR_ERR(uobj);
715 
716 	ret = ib_check_mr_access(ib_dev, cmd.access_flags);
717 	if (ret)
718 		goto err_free;
719 
720 	pd = uobj_get_obj_read(pd, UVERBS_OBJECT_PD, cmd.pd_handle, attrs);
721 	if (!pd) {
722 		ret = -EINVAL;
723 		goto err_free;
724 	}
725 
726 	mr = pd->device->ops.reg_user_mr(pd, cmd.start, cmd.length, cmd.hca_va,
727 					 cmd.access_flags,
728 					 &attrs->driver_udata);
729 	if (IS_ERR(mr)) {
730 		ret = PTR_ERR(mr);
731 		goto err_put;
732 	}
733 
734 	mr->device  = pd->device;
735 	mr->pd      = pd;
736 	mr->type    = IB_MR_TYPE_USER;
737 	mr->dm	    = NULL;
738 	mr->sig_attrs = NULL;
739 	mr->uobject = uobj;
740 	atomic_inc(&pd->usecnt);
741 	mr->iova = cmd.hca_va;
742 	mr->length = cmd.length;
743 
744 	rdma_restrack_new(&mr->res, RDMA_RESTRACK_MR);
745 	rdma_restrack_set_name(&mr->res, NULL);
746 	rdma_restrack_add(&mr->res);
747 
748 	uobj->object = mr;
749 	uobj_put_obj_read(pd);
750 	uobj_finalize_uobj_create(uobj, attrs);
751 
752 	resp.lkey = mr->lkey;
753 	resp.rkey = mr->rkey;
754 	resp.mr_handle = uobj->id;
755 	return uverbs_response(attrs, &resp, sizeof(resp));
756 
757 err_put:
758 	uobj_put_obj_read(pd);
759 err_free:
760 	uobj_alloc_abort(uobj, attrs);
761 	return ret;
762 }
763 
ib_uverbs_rereg_mr(struct uverbs_attr_bundle * attrs)764 static int ib_uverbs_rereg_mr(struct uverbs_attr_bundle *attrs)
765 {
766 	struct ib_uverbs_rereg_mr      cmd;
767 	struct ib_uverbs_rereg_mr_resp resp;
768 	struct ib_mr                *mr;
769 	int                          ret;
770 	struct ib_uobject	    *uobj;
771 	struct ib_uobject *new_uobj;
772 	struct ib_device *ib_dev;
773 	struct ib_pd *orig_pd;
774 	struct ib_pd *new_pd;
775 	struct ib_mr *new_mr;
776 
777 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
778 	if (ret)
779 		return ret;
780 
781 	if (!cmd.flags)
782 		return -EINVAL;
783 
784 	if (cmd.flags & ~IB_MR_REREG_SUPPORTED)
785 		return -EOPNOTSUPP;
786 
787 	if ((cmd.flags & IB_MR_REREG_TRANS) &&
788 	    (cmd.start & ~PAGE_MASK) != (cmd.hca_va & ~PAGE_MASK))
789 		return -EINVAL;
790 
791 	uobj = uobj_get_write(UVERBS_OBJECT_MR, cmd.mr_handle, attrs);
792 	if (IS_ERR(uobj))
793 		return PTR_ERR(uobj);
794 
795 	mr = uobj->object;
796 
797 	if (mr->dm) {
798 		ret = -EINVAL;
799 		goto put_uobjs;
800 	}
801 
802 	if (cmd.flags & IB_MR_REREG_ACCESS) {
803 		ret = ib_check_mr_access(mr->device, cmd.access_flags);
804 		if (ret)
805 			goto put_uobjs;
806 	}
807 
808 	orig_pd = mr->pd;
809 	if (cmd.flags & IB_MR_REREG_PD) {
810 		new_pd = uobj_get_obj_read(pd, UVERBS_OBJECT_PD, cmd.pd_handle,
811 					   attrs);
812 		if (!new_pd) {
813 			ret = -EINVAL;
814 			goto put_uobjs;
815 		}
816 	} else {
817 		new_pd = mr->pd;
818 	}
819 
820 	/*
821 	 * The driver might create a new HW object as part of the rereg, we need
822 	 * to have a uobject ready to hold it.
823 	 */
824 	new_uobj = uobj_alloc(UVERBS_OBJECT_MR, attrs, &ib_dev);
825 	if (IS_ERR(new_uobj)) {
826 		ret = PTR_ERR(new_uobj);
827 		goto put_uobj_pd;
828 	}
829 
830 	new_mr = ib_dev->ops.rereg_user_mr(mr, cmd.flags, cmd.start, cmd.length,
831 					   cmd.hca_va, cmd.access_flags, new_pd,
832 					   &attrs->driver_udata);
833 	if (IS_ERR(new_mr)) {
834 		ret = PTR_ERR(new_mr);
835 		goto put_new_uobj;
836 	}
837 	if (new_mr) {
838 		new_mr->device = new_pd->device;
839 		new_mr->pd = new_pd;
840 		new_mr->type = IB_MR_TYPE_USER;
841 		new_mr->uobject = uobj;
842 		atomic_inc(&new_pd->usecnt);
843 		new_uobj->object = new_mr;
844 
845 		rdma_restrack_new(&new_mr->res, RDMA_RESTRACK_MR);
846 		rdma_restrack_set_name(&new_mr->res, NULL);
847 		rdma_restrack_add(&new_mr->res);
848 
849 		/*
850 		 * The new uobj for the new HW object is put into the same spot
851 		 * in the IDR and the old uobj & HW object is deleted.
852 		 */
853 		rdma_assign_uobject(uobj, new_uobj, attrs);
854 		rdma_alloc_commit_uobject(new_uobj, attrs);
855 		uobj_put_destroy(uobj);
856 		new_uobj = NULL;
857 		uobj = NULL;
858 		mr = new_mr;
859 	} else {
860 		if (cmd.flags & IB_MR_REREG_PD) {
861 			atomic_dec(&orig_pd->usecnt);
862 			mr->pd = new_pd;
863 			atomic_inc(&new_pd->usecnt);
864 		}
865 		if (cmd.flags & IB_MR_REREG_TRANS) {
866 			mr->iova = cmd.hca_va;
867 			mr->length = cmd.length;
868 		}
869 	}
870 
871 	memset(&resp, 0, sizeof(resp));
872 	resp.lkey      = mr->lkey;
873 	resp.rkey      = mr->rkey;
874 
875 	ret = uverbs_response(attrs, &resp, sizeof(resp));
876 
877 put_new_uobj:
878 	if (new_uobj)
879 		uobj_alloc_abort(new_uobj, attrs);
880 put_uobj_pd:
881 	if (cmd.flags & IB_MR_REREG_PD)
882 		uobj_put_obj_read(new_pd);
883 
884 put_uobjs:
885 	if (uobj)
886 		uobj_put_write(uobj);
887 
888 	return ret;
889 }
890 
ib_uverbs_dereg_mr(struct uverbs_attr_bundle * attrs)891 static int ib_uverbs_dereg_mr(struct uverbs_attr_bundle *attrs)
892 {
893 	struct ib_uverbs_dereg_mr cmd;
894 	int ret;
895 
896 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
897 	if (ret)
898 		return ret;
899 
900 	return uobj_perform_destroy(UVERBS_OBJECT_MR, cmd.mr_handle, attrs);
901 }
902 
ib_uverbs_alloc_mw(struct uverbs_attr_bundle * attrs)903 static int ib_uverbs_alloc_mw(struct uverbs_attr_bundle *attrs)
904 {
905 	struct ib_uverbs_alloc_mw      cmd;
906 	struct ib_uverbs_alloc_mw_resp resp = {};
907 	struct ib_uobject             *uobj;
908 	struct ib_pd                  *pd;
909 	struct ib_mw                  *mw;
910 	int                            ret;
911 	struct ib_device *ib_dev;
912 
913 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
914 	if (ret)
915 		return ret;
916 
917 	uobj = uobj_alloc(UVERBS_OBJECT_MW, attrs, &ib_dev);
918 	if (IS_ERR(uobj))
919 		return PTR_ERR(uobj);
920 
921 	pd = uobj_get_obj_read(pd, UVERBS_OBJECT_PD, cmd.pd_handle, attrs);
922 	if (!pd) {
923 		ret = -EINVAL;
924 		goto err_free;
925 	}
926 
927 	if (cmd.mw_type != IB_MW_TYPE_1 && cmd.mw_type != IB_MW_TYPE_2) {
928 		ret = -EINVAL;
929 		goto err_put;
930 	}
931 
932 	mw = rdma_zalloc_drv_obj(ib_dev, ib_mw);
933 	if (!mw) {
934 		ret = -ENOMEM;
935 		goto err_put;
936 	}
937 
938 	mw->device = ib_dev;
939 	mw->pd = pd;
940 	mw->uobject = uobj;
941 	mw->type = cmd.mw_type;
942 
943 	ret = pd->device->ops.alloc_mw(mw, &attrs->driver_udata);
944 	if (ret)
945 		goto err_alloc;
946 
947 	atomic_inc(&pd->usecnt);
948 
949 	uobj->object = mw;
950 	uobj_put_obj_read(pd);
951 	uobj_finalize_uobj_create(uobj, attrs);
952 
953 	resp.rkey = mw->rkey;
954 	resp.mw_handle = uobj->id;
955 	return uverbs_response(attrs, &resp, sizeof(resp));
956 
957 err_alloc:
958 	kfree(mw);
959 err_put:
960 	uobj_put_obj_read(pd);
961 err_free:
962 	uobj_alloc_abort(uobj, attrs);
963 	return ret;
964 }
965 
ib_uverbs_dealloc_mw(struct uverbs_attr_bundle * attrs)966 static int ib_uverbs_dealloc_mw(struct uverbs_attr_bundle *attrs)
967 {
968 	struct ib_uverbs_dealloc_mw cmd;
969 	int ret;
970 
971 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
972 	if (ret)
973 		return ret;
974 
975 	return uobj_perform_destroy(UVERBS_OBJECT_MW, cmd.mw_handle, attrs);
976 }
977 
ib_uverbs_create_comp_channel(struct uverbs_attr_bundle * attrs)978 static int ib_uverbs_create_comp_channel(struct uverbs_attr_bundle *attrs)
979 {
980 	struct ib_uverbs_create_comp_channel	   cmd;
981 	struct ib_uverbs_create_comp_channel_resp  resp;
982 	struct ib_uobject			  *uobj;
983 	struct ib_uverbs_completion_event_file	  *ev_file;
984 	struct ib_device *ib_dev;
985 	int ret;
986 
987 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
988 	if (ret)
989 		return ret;
990 
991 	uobj = uobj_alloc(UVERBS_OBJECT_COMP_CHANNEL, attrs, &ib_dev);
992 	if (IS_ERR(uobj))
993 		return PTR_ERR(uobj);
994 
995 	ev_file = container_of(uobj, struct ib_uverbs_completion_event_file,
996 			       uobj);
997 	ib_uverbs_init_event_queue(&ev_file->ev_queue);
998 	uobj_finalize_uobj_create(uobj, attrs);
999 
1000 	resp.fd = uobj->id;
1001 	return uverbs_response(attrs, &resp, sizeof(resp));
1002 }
1003 
create_cq(struct uverbs_attr_bundle * attrs,struct ib_uverbs_ex_create_cq * cmd)1004 static int create_cq(struct uverbs_attr_bundle *attrs,
1005 		     struct ib_uverbs_ex_create_cq *cmd)
1006 {
1007 	struct ib_ucq_object           *obj;
1008 	struct ib_uverbs_completion_event_file    *ev_file = NULL;
1009 	struct ib_cq                   *cq;
1010 	int                             ret;
1011 	struct ib_uverbs_ex_create_cq_resp resp = {};
1012 	struct ib_cq_init_attr attr = {};
1013 	struct ib_device *ib_dev;
1014 
1015 	if (cmd->comp_vector >= attrs->ufile->device->num_comp_vectors)
1016 		return -EINVAL;
1017 
1018 	obj = (struct ib_ucq_object *)uobj_alloc(UVERBS_OBJECT_CQ, attrs,
1019 						 &ib_dev);
1020 	if (IS_ERR(obj))
1021 		return PTR_ERR(obj);
1022 
1023 	if (cmd->comp_channel >= 0) {
1024 		ev_file = ib_uverbs_lookup_comp_file(cmd->comp_channel, attrs);
1025 		if (IS_ERR(ev_file)) {
1026 			ret = PTR_ERR(ev_file);
1027 			goto err;
1028 		}
1029 	}
1030 
1031 	obj->uevent.uobject.user_handle = cmd->user_handle;
1032 	INIT_LIST_HEAD(&obj->comp_list);
1033 	INIT_LIST_HEAD(&obj->uevent.event_list);
1034 
1035 	attr.cqe = cmd->cqe;
1036 	attr.comp_vector = cmd->comp_vector;
1037 	attr.flags = cmd->flags;
1038 
1039 	cq = rdma_zalloc_drv_obj(ib_dev, ib_cq);
1040 	if (!cq) {
1041 		ret = -ENOMEM;
1042 		goto err_file;
1043 	}
1044 	cq->device        = ib_dev;
1045 	cq->uobject       = obj;
1046 	cq->comp_handler  = ib_uverbs_comp_handler;
1047 	cq->event_handler = ib_uverbs_cq_event_handler;
1048 	cq->cq_context    = ev_file ? &ev_file->ev_queue : NULL;
1049 	atomic_set(&cq->usecnt, 0);
1050 
1051 	rdma_restrack_new(&cq->res, RDMA_RESTRACK_CQ);
1052 	rdma_restrack_set_name(&cq->res, NULL);
1053 
1054 	ret = ib_dev->ops.create_cq(cq, &attr, &attrs->driver_udata);
1055 	if (ret)
1056 		goto err_free;
1057 	rdma_restrack_add(&cq->res);
1058 
1059 	obj->uevent.uobject.object = cq;
1060 	obj->uevent.event_file = READ_ONCE(attrs->ufile->default_async_file);
1061 	if (obj->uevent.event_file)
1062 		uverbs_uobject_get(&obj->uevent.event_file->uobj);
1063 	uobj_finalize_uobj_create(&obj->uevent.uobject, attrs);
1064 
1065 	resp.base.cq_handle = obj->uevent.uobject.id;
1066 	resp.base.cqe = cq->cqe;
1067 	resp.response_length = uverbs_response_length(attrs, sizeof(resp));
1068 	return uverbs_response(attrs, &resp, sizeof(resp));
1069 
1070 err_free:
1071 	rdma_restrack_put(&cq->res);
1072 	kfree(cq);
1073 err_file:
1074 	if (ev_file)
1075 		ib_uverbs_release_ucq(ev_file, obj);
1076 err:
1077 	uobj_alloc_abort(&obj->uevent.uobject, attrs);
1078 	return ret;
1079 }
1080 
ib_uverbs_create_cq(struct uverbs_attr_bundle * attrs)1081 static int ib_uverbs_create_cq(struct uverbs_attr_bundle *attrs)
1082 {
1083 	struct ib_uverbs_create_cq      cmd;
1084 	struct ib_uverbs_ex_create_cq	cmd_ex;
1085 	int ret;
1086 
1087 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1088 	if (ret)
1089 		return ret;
1090 
1091 	memset(&cmd_ex, 0, sizeof(cmd_ex));
1092 	cmd_ex.user_handle = cmd.user_handle;
1093 	cmd_ex.cqe = cmd.cqe;
1094 	cmd_ex.comp_vector = cmd.comp_vector;
1095 	cmd_ex.comp_channel = cmd.comp_channel;
1096 
1097 	return create_cq(attrs, &cmd_ex);
1098 }
1099 
ib_uverbs_ex_create_cq(struct uverbs_attr_bundle * attrs)1100 static int ib_uverbs_ex_create_cq(struct uverbs_attr_bundle *attrs)
1101 {
1102 	struct ib_uverbs_ex_create_cq  cmd;
1103 	int ret;
1104 
1105 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1106 	if (ret)
1107 		return ret;
1108 
1109 	if (cmd.comp_mask)
1110 		return -EINVAL;
1111 
1112 	if (cmd.reserved)
1113 		return -EINVAL;
1114 
1115 	return create_cq(attrs, &cmd);
1116 }
1117 
ib_uverbs_resize_cq(struct uverbs_attr_bundle * attrs)1118 static int ib_uverbs_resize_cq(struct uverbs_attr_bundle *attrs)
1119 {
1120 	struct ib_uverbs_resize_cq	cmd;
1121 	struct ib_uverbs_resize_cq_resp	resp = {};
1122 	struct ib_cq			*cq;
1123 	int ret;
1124 
1125 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1126 	if (ret)
1127 		return ret;
1128 
1129 	cq = uobj_get_obj_read(cq, UVERBS_OBJECT_CQ, cmd.cq_handle, attrs);
1130 	if (!cq)
1131 		return -EINVAL;
1132 
1133 	ret = cq->device->ops.resize_cq(cq, cmd.cqe, &attrs->driver_udata);
1134 	if (ret)
1135 		goto out;
1136 
1137 	resp.cqe = cq->cqe;
1138 
1139 	ret = uverbs_response(attrs, &resp, sizeof(resp));
1140 out:
1141 	rdma_lookup_put_uobject(&cq->uobject->uevent.uobject,
1142 				UVERBS_LOOKUP_READ);
1143 
1144 	return ret;
1145 }
1146 
copy_wc_to_user(struct ib_device * ib_dev,void __user * dest,struct ib_wc * wc)1147 static int copy_wc_to_user(struct ib_device *ib_dev, void __user *dest,
1148 			   struct ib_wc *wc)
1149 {
1150 	struct ib_uverbs_wc tmp;
1151 
1152 	tmp.wr_id		= wc->wr_id;
1153 	tmp.status		= wc->status;
1154 	tmp.opcode		= wc->opcode;
1155 	tmp.vendor_err		= wc->vendor_err;
1156 	tmp.byte_len		= wc->byte_len;
1157 	tmp.ex.imm_data		= wc->ex.imm_data;
1158 	tmp.qp_num		= wc->qp->qp_num;
1159 	tmp.src_qp		= wc->src_qp;
1160 	tmp.wc_flags		= wc->wc_flags;
1161 	tmp.pkey_index		= wc->pkey_index;
1162 	if (rdma_cap_opa_ah(ib_dev, wc->port_num))
1163 		tmp.slid	= OPA_TO_IB_UCAST_LID(wc->slid);
1164 	else
1165 		tmp.slid	= ib_lid_cpu16(wc->slid);
1166 	tmp.sl			= wc->sl;
1167 	tmp.dlid_path_bits	= wc->dlid_path_bits;
1168 	tmp.port_num		= wc->port_num;
1169 	tmp.reserved		= 0;
1170 
1171 	if (copy_to_user(dest, &tmp, sizeof tmp))
1172 		return -EFAULT;
1173 
1174 	return 0;
1175 }
1176 
ib_uverbs_poll_cq(struct uverbs_attr_bundle * attrs)1177 static int ib_uverbs_poll_cq(struct uverbs_attr_bundle *attrs)
1178 {
1179 	struct ib_uverbs_poll_cq       cmd;
1180 	struct ib_uverbs_poll_cq_resp  resp;
1181 	u8 __user                     *header_ptr;
1182 	u8 __user                     *data_ptr;
1183 	struct ib_cq                  *cq;
1184 	struct ib_wc                   wc;
1185 	int                            ret;
1186 
1187 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1188 	if (ret)
1189 		return ret;
1190 
1191 	cq = uobj_get_obj_read(cq, UVERBS_OBJECT_CQ, cmd.cq_handle, attrs);
1192 	if (!cq)
1193 		return -EINVAL;
1194 
1195 	/* we copy a struct ib_uverbs_poll_cq_resp to user space */
1196 	header_ptr = attrs->ucore.outbuf;
1197 	data_ptr = header_ptr + sizeof resp;
1198 
1199 	memset(&resp, 0, sizeof resp);
1200 	while (resp.count < cmd.ne) {
1201 		ret = ib_poll_cq(cq, 1, &wc);
1202 		if (ret < 0)
1203 			goto out_put;
1204 		if (!ret)
1205 			break;
1206 
1207 		ret = copy_wc_to_user(cq->device, data_ptr, &wc);
1208 		if (ret)
1209 			goto out_put;
1210 
1211 		data_ptr += sizeof(struct ib_uverbs_wc);
1212 		++resp.count;
1213 	}
1214 
1215 	if (copy_to_user(header_ptr, &resp, sizeof resp)) {
1216 		ret = -EFAULT;
1217 		goto out_put;
1218 	}
1219 	ret = 0;
1220 
1221 	if (uverbs_attr_is_valid(attrs, UVERBS_ATTR_CORE_OUT))
1222 		ret = uverbs_output_written(attrs, UVERBS_ATTR_CORE_OUT);
1223 
1224 out_put:
1225 	rdma_lookup_put_uobject(&cq->uobject->uevent.uobject,
1226 				UVERBS_LOOKUP_READ);
1227 	return ret;
1228 }
1229 
ib_uverbs_req_notify_cq(struct uverbs_attr_bundle * attrs)1230 static int ib_uverbs_req_notify_cq(struct uverbs_attr_bundle *attrs)
1231 {
1232 	struct ib_uverbs_req_notify_cq cmd;
1233 	struct ib_cq                  *cq;
1234 	int ret;
1235 
1236 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1237 	if (ret)
1238 		return ret;
1239 
1240 	cq = uobj_get_obj_read(cq, UVERBS_OBJECT_CQ, cmd.cq_handle, attrs);
1241 	if (!cq)
1242 		return -EINVAL;
1243 
1244 	ib_req_notify_cq(cq, cmd.solicited_only ?
1245 			 IB_CQ_SOLICITED : IB_CQ_NEXT_COMP);
1246 
1247 	rdma_lookup_put_uobject(&cq->uobject->uevent.uobject,
1248 				UVERBS_LOOKUP_READ);
1249 	return 0;
1250 }
1251 
ib_uverbs_destroy_cq(struct uverbs_attr_bundle * attrs)1252 static int ib_uverbs_destroy_cq(struct uverbs_attr_bundle *attrs)
1253 {
1254 	struct ib_uverbs_destroy_cq      cmd;
1255 	struct ib_uverbs_destroy_cq_resp resp;
1256 	struct ib_uobject		*uobj;
1257 	struct ib_ucq_object        	*obj;
1258 	int ret;
1259 
1260 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1261 	if (ret)
1262 		return ret;
1263 
1264 	uobj = uobj_get_destroy(UVERBS_OBJECT_CQ, cmd.cq_handle, attrs);
1265 	if (IS_ERR(uobj))
1266 		return PTR_ERR(uobj);
1267 
1268 	obj = container_of(uobj, struct ib_ucq_object, uevent.uobject);
1269 	memset(&resp, 0, sizeof(resp));
1270 	resp.comp_events_reported  = obj->comp_events_reported;
1271 	resp.async_events_reported = obj->uevent.events_reported;
1272 
1273 	uobj_put_destroy(uobj);
1274 
1275 	return uverbs_response(attrs, &resp, sizeof(resp));
1276 }
1277 
create_qp(struct uverbs_attr_bundle * attrs,struct ib_uverbs_ex_create_qp * cmd)1278 static int create_qp(struct uverbs_attr_bundle *attrs,
1279 		     struct ib_uverbs_ex_create_qp *cmd)
1280 {
1281 	struct ib_uqp_object		*obj;
1282 	struct ib_device		*device;
1283 	struct ib_pd			*pd = NULL;
1284 	struct ib_xrcd			*xrcd = NULL;
1285 	struct ib_uobject		*xrcd_uobj = ERR_PTR(-ENOENT);
1286 	struct ib_cq			*scq = NULL, *rcq = NULL;
1287 	struct ib_srq			*srq = NULL;
1288 	struct ib_qp			*qp;
1289 	struct ib_qp_init_attr		attr = {};
1290 	struct ib_uverbs_ex_create_qp_resp resp = {};
1291 	int				ret;
1292 	struct ib_rwq_ind_table *ind_tbl = NULL;
1293 	bool has_sq = true;
1294 	struct ib_device *ib_dev;
1295 
1296 	switch (cmd->qp_type) {
1297 	case IB_QPT_RAW_PACKET:
1298 		if (!capable(CAP_NET_RAW))
1299 			return -EPERM;
1300 		break;
1301 	case IB_QPT_RC:
1302 	case IB_QPT_UC:
1303 	case IB_QPT_UD:
1304 	case IB_QPT_XRC_INI:
1305 	case IB_QPT_XRC_TGT:
1306 	case IB_QPT_DRIVER:
1307 		break;
1308 	default:
1309 		return -EINVAL;
1310 	}
1311 
1312 	obj = (struct ib_uqp_object *)uobj_alloc(UVERBS_OBJECT_QP, attrs,
1313 						 &ib_dev);
1314 	if (IS_ERR(obj))
1315 		return PTR_ERR(obj);
1316 	obj->uxrcd = NULL;
1317 	obj->uevent.uobject.user_handle = cmd->user_handle;
1318 	mutex_init(&obj->mcast_lock);
1319 
1320 	if (cmd->comp_mask & IB_UVERBS_CREATE_QP_MASK_IND_TABLE) {
1321 		ind_tbl = uobj_get_obj_read(rwq_ind_table,
1322 					    UVERBS_OBJECT_RWQ_IND_TBL,
1323 					    cmd->rwq_ind_tbl_handle, attrs);
1324 		if (!ind_tbl) {
1325 			ret = -EINVAL;
1326 			goto err_put;
1327 		}
1328 
1329 		attr.rwq_ind_tbl = ind_tbl;
1330 	}
1331 
1332 	if (ind_tbl && (cmd->max_recv_wr || cmd->max_recv_sge || cmd->is_srq)) {
1333 		ret = -EINVAL;
1334 		goto err_put;
1335 	}
1336 
1337 	if (ind_tbl && !cmd->max_send_wr)
1338 		has_sq = false;
1339 
1340 	if (cmd->qp_type == IB_QPT_XRC_TGT) {
1341 		xrcd_uobj = uobj_get_read(UVERBS_OBJECT_XRCD, cmd->pd_handle,
1342 					  attrs);
1343 
1344 		if (IS_ERR(xrcd_uobj)) {
1345 			ret = -EINVAL;
1346 			goto err_put;
1347 		}
1348 
1349 		xrcd = (struct ib_xrcd *)xrcd_uobj->object;
1350 		if (!xrcd) {
1351 			ret = -EINVAL;
1352 			goto err_put;
1353 		}
1354 		device = xrcd->device;
1355 	} else {
1356 		if (cmd->qp_type == IB_QPT_XRC_INI) {
1357 			cmd->max_recv_wr = 0;
1358 			cmd->max_recv_sge = 0;
1359 		} else {
1360 			if (cmd->is_srq) {
1361 				srq = uobj_get_obj_read(srq, UVERBS_OBJECT_SRQ,
1362 							cmd->srq_handle, attrs);
1363 				if (!srq || srq->srq_type == IB_SRQT_XRC) {
1364 					ret = -EINVAL;
1365 					goto err_put;
1366 				}
1367 			}
1368 
1369 			if (!ind_tbl) {
1370 				if (cmd->recv_cq_handle != cmd->send_cq_handle) {
1371 					rcq = uobj_get_obj_read(
1372 						cq, UVERBS_OBJECT_CQ,
1373 						cmd->recv_cq_handle, attrs);
1374 					if (!rcq) {
1375 						ret = -EINVAL;
1376 						goto err_put;
1377 					}
1378 				}
1379 			}
1380 		}
1381 
1382 		if (has_sq)
1383 			scq = uobj_get_obj_read(cq, UVERBS_OBJECT_CQ,
1384 						cmd->send_cq_handle, attrs);
1385 		if (!ind_tbl && cmd->qp_type != IB_QPT_XRC_INI)
1386 			rcq = rcq ?: scq;
1387 		pd = uobj_get_obj_read(pd, UVERBS_OBJECT_PD, cmd->pd_handle,
1388 				       attrs);
1389 		if (!pd || (!scq && has_sq)) {
1390 			ret = -EINVAL;
1391 			goto err_put;
1392 		}
1393 
1394 		device = pd->device;
1395 	}
1396 
1397 	attr.event_handler = ib_uverbs_qp_event_handler;
1398 	attr.send_cq       = scq;
1399 	attr.recv_cq       = rcq;
1400 	attr.srq           = srq;
1401 	attr.xrcd	   = xrcd;
1402 	attr.sq_sig_type   = cmd->sq_sig_all ? IB_SIGNAL_ALL_WR :
1403 					      IB_SIGNAL_REQ_WR;
1404 	attr.qp_type       = cmd->qp_type;
1405 
1406 	attr.cap.max_send_wr     = cmd->max_send_wr;
1407 	attr.cap.max_recv_wr     = cmd->max_recv_wr;
1408 	attr.cap.max_send_sge    = cmd->max_send_sge;
1409 	attr.cap.max_recv_sge    = cmd->max_recv_sge;
1410 	attr.cap.max_inline_data = cmd->max_inline_data;
1411 
1412 	INIT_LIST_HEAD(&obj->uevent.event_list);
1413 	INIT_LIST_HEAD(&obj->mcast_list);
1414 
1415 	attr.create_flags = cmd->create_flags;
1416 	if (attr.create_flags & ~(IB_QP_CREATE_BLOCK_MULTICAST_LOOPBACK |
1417 				IB_QP_CREATE_CROSS_CHANNEL |
1418 				IB_QP_CREATE_MANAGED_SEND |
1419 				IB_QP_CREATE_MANAGED_RECV |
1420 				IB_QP_CREATE_SCATTER_FCS |
1421 				IB_QP_CREATE_CVLAN_STRIPPING |
1422 				IB_QP_CREATE_SOURCE_QPN |
1423 				IB_QP_CREATE_PCI_WRITE_END_PADDING)) {
1424 		ret = -EINVAL;
1425 		goto err_put;
1426 	}
1427 
1428 	if (attr.create_flags & IB_QP_CREATE_SOURCE_QPN) {
1429 		if (!capable(CAP_NET_RAW)) {
1430 			ret = -EPERM;
1431 			goto err_put;
1432 		}
1433 
1434 		attr.source_qpn = cmd->source_qpn;
1435 	}
1436 
1437 	qp = ib_create_qp_user(device, pd, &attr, &attrs->driver_udata, obj,
1438 			       KBUILD_MODNAME);
1439 	if (IS_ERR(qp)) {
1440 		ret = PTR_ERR(qp);
1441 		goto err_put;
1442 	}
1443 	ib_qp_usecnt_inc(qp);
1444 
1445 	obj->uevent.uobject.object = qp;
1446 	obj->uevent.event_file = READ_ONCE(attrs->ufile->default_async_file);
1447 	if (obj->uevent.event_file)
1448 		uverbs_uobject_get(&obj->uevent.event_file->uobj);
1449 
1450 	if (xrcd) {
1451 		obj->uxrcd = container_of(xrcd_uobj, struct ib_uxrcd_object,
1452 					  uobject);
1453 		atomic_inc(&obj->uxrcd->refcnt);
1454 		uobj_put_read(xrcd_uobj);
1455 	}
1456 
1457 	if (pd)
1458 		uobj_put_obj_read(pd);
1459 	if (scq)
1460 		rdma_lookup_put_uobject(&scq->uobject->uevent.uobject,
1461 					UVERBS_LOOKUP_READ);
1462 	if (rcq && rcq != scq)
1463 		rdma_lookup_put_uobject(&rcq->uobject->uevent.uobject,
1464 					UVERBS_LOOKUP_READ);
1465 	if (srq)
1466 		rdma_lookup_put_uobject(&srq->uobject->uevent.uobject,
1467 					UVERBS_LOOKUP_READ);
1468 	if (ind_tbl)
1469 		uobj_put_obj_read(ind_tbl);
1470 	uobj_finalize_uobj_create(&obj->uevent.uobject, attrs);
1471 
1472 	resp.base.qpn             = qp->qp_num;
1473 	resp.base.qp_handle       = obj->uevent.uobject.id;
1474 	resp.base.max_recv_sge    = attr.cap.max_recv_sge;
1475 	resp.base.max_send_sge    = attr.cap.max_send_sge;
1476 	resp.base.max_recv_wr     = attr.cap.max_recv_wr;
1477 	resp.base.max_send_wr     = attr.cap.max_send_wr;
1478 	resp.base.max_inline_data = attr.cap.max_inline_data;
1479 	resp.response_length = uverbs_response_length(attrs, sizeof(resp));
1480 	return uverbs_response(attrs, &resp, sizeof(resp));
1481 
1482 err_put:
1483 	if (!IS_ERR(xrcd_uobj))
1484 		uobj_put_read(xrcd_uobj);
1485 	if (pd)
1486 		uobj_put_obj_read(pd);
1487 	if (scq)
1488 		rdma_lookup_put_uobject(&scq->uobject->uevent.uobject,
1489 					UVERBS_LOOKUP_READ);
1490 	if (rcq && rcq != scq)
1491 		rdma_lookup_put_uobject(&rcq->uobject->uevent.uobject,
1492 					UVERBS_LOOKUP_READ);
1493 	if (srq)
1494 		rdma_lookup_put_uobject(&srq->uobject->uevent.uobject,
1495 					UVERBS_LOOKUP_READ);
1496 	if (ind_tbl)
1497 		uobj_put_obj_read(ind_tbl);
1498 
1499 	uobj_alloc_abort(&obj->uevent.uobject, attrs);
1500 	return ret;
1501 }
1502 
ib_uverbs_create_qp(struct uverbs_attr_bundle * attrs)1503 static int ib_uverbs_create_qp(struct uverbs_attr_bundle *attrs)
1504 {
1505 	struct ib_uverbs_create_qp      cmd;
1506 	struct ib_uverbs_ex_create_qp	cmd_ex;
1507 	int ret;
1508 
1509 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1510 	if (ret)
1511 		return ret;
1512 
1513 	memset(&cmd_ex, 0, sizeof(cmd_ex));
1514 	cmd_ex.user_handle = cmd.user_handle;
1515 	cmd_ex.pd_handle = cmd.pd_handle;
1516 	cmd_ex.send_cq_handle = cmd.send_cq_handle;
1517 	cmd_ex.recv_cq_handle = cmd.recv_cq_handle;
1518 	cmd_ex.srq_handle = cmd.srq_handle;
1519 	cmd_ex.max_send_wr = cmd.max_send_wr;
1520 	cmd_ex.max_recv_wr = cmd.max_recv_wr;
1521 	cmd_ex.max_send_sge = cmd.max_send_sge;
1522 	cmd_ex.max_recv_sge = cmd.max_recv_sge;
1523 	cmd_ex.max_inline_data = cmd.max_inline_data;
1524 	cmd_ex.sq_sig_all = cmd.sq_sig_all;
1525 	cmd_ex.qp_type = cmd.qp_type;
1526 	cmd_ex.is_srq = cmd.is_srq;
1527 
1528 	return create_qp(attrs, &cmd_ex);
1529 }
1530 
ib_uverbs_ex_create_qp(struct uverbs_attr_bundle * attrs)1531 static int ib_uverbs_ex_create_qp(struct uverbs_attr_bundle *attrs)
1532 {
1533 	struct ib_uverbs_ex_create_qp cmd;
1534 	int ret;
1535 
1536 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1537 	if (ret)
1538 		return ret;
1539 
1540 	if (cmd.comp_mask & ~IB_UVERBS_CREATE_QP_SUP_COMP_MASK)
1541 		return -EINVAL;
1542 
1543 	if (cmd.reserved)
1544 		return -EINVAL;
1545 
1546 	return create_qp(attrs, &cmd);
1547 }
1548 
ib_uverbs_open_qp(struct uverbs_attr_bundle * attrs)1549 static int ib_uverbs_open_qp(struct uverbs_attr_bundle *attrs)
1550 {
1551 	struct ib_uverbs_create_qp_resp resp = {};
1552 	struct ib_uverbs_open_qp        cmd;
1553 	struct ib_uqp_object           *obj;
1554 	struct ib_xrcd		       *xrcd;
1555 	struct ib_qp                   *qp;
1556 	struct ib_qp_open_attr          attr = {};
1557 	int ret;
1558 	struct ib_uobject *xrcd_uobj;
1559 	struct ib_device *ib_dev;
1560 
1561 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1562 	if (ret)
1563 		return ret;
1564 
1565 	obj = (struct ib_uqp_object *)uobj_alloc(UVERBS_OBJECT_QP, attrs,
1566 						 &ib_dev);
1567 	if (IS_ERR(obj))
1568 		return PTR_ERR(obj);
1569 
1570 	xrcd_uobj = uobj_get_read(UVERBS_OBJECT_XRCD, cmd.pd_handle, attrs);
1571 	if (IS_ERR(xrcd_uobj)) {
1572 		ret = -EINVAL;
1573 		goto err_put;
1574 	}
1575 
1576 	xrcd = (struct ib_xrcd *)xrcd_uobj->object;
1577 	if (!xrcd) {
1578 		ret = -EINVAL;
1579 		goto err_xrcd;
1580 	}
1581 
1582 	attr.event_handler = ib_uverbs_qp_event_handler;
1583 	attr.qp_num        = cmd.qpn;
1584 	attr.qp_type       = cmd.qp_type;
1585 
1586 	INIT_LIST_HEAD(&obj->uevent.event_list);
1587 	INIT_LIST_HEAD(&obj->mcast_list);
1588 
1589 	qp = ib_open_qp(xrcd, &attr);
1590 	if (IS_ERR(qp)) {
1591 		ret = PTR_ERR(qp);
1592 		goto err_xrcd;
1593 	}
1594 
1595 	obj->uevent.uobject.object = qp;
1596 	obj->uevent.uobject.user_handle = cmd.user_handle;
1597 
1598 	obj->uxrcd = container_of(xrcd_uobj, struct ib_uxrcd_object, uobject);
1599 	atomic_inc(&obj->uxrcd->refcnt);
1600 	qp->uobject = obj;
1601 	uobj_put_read(xrcd_uobj);
1602 	uobj_finalize_uobj_create(&obj->uevent.uobject, attrs);
1603 
1604 	resp.qpn = qp->qp_num;
1605 	resp.qp_handle = obj->uevent.uobject.id;
1606 	return uverbs_response(attrs, &resp, sizeof(resp));
1607 
1608 err_xrcd:
1609 	uobj_put_read(xrcd_uobj);
1610 err_put:
1611 	uobj_alloc_abort(&obj->uevent.uobject, attrs);
1612 	return ret;
1613 }
1614 
copy_ah_attr_to_uverbs(struct ib_uverbs_qp_dest * uverb_attr,struct rdma_ah_attr * rdma_attr)1615 static void copy_ah_attr_to_uverbs(struct ib_uverbs_qp_dest *uverb_attr,
1616 				   struct rdma_ah_attr *rdma_attr)
1617 {
1618 	const struct ib_global_route   *grh;
1619 
1620 	uverb_attr->dlid              = rdma_ah_get_dlid(rdma_attr);
1621 	uverb_attr->sl                = rdma_ah_get_sl(rdma_attr);
1622 	uverb_attr->src_path_bits     = rdma_ah_get_path_bits(rdma_attr);
1623 	uverb_attr->static_rate       = rdma_ah_get_static_rate(rdma_attr);
1624 	uverb_attr->is_global         = !!(rdma_ah_get_ah_flags(rdma_attr) &
1625 					 IB_AH_GRH);
1626 	if (uverb_attr->is_global) {
1627 		grh = rdma_ah_read_grh(rdma_attr);
1628 		memcpy(uverb_attr->dgid, grh->dgid.raw, 16);
1629 		uverb_attr->flow_label        = grh->flow_label;
1630 		uverb_attr->sgid_index        = grh->sgid_index;
1631 		uverb_attr->hop_limit         = grh->hop_limit;
1632 		uverb_attr->traffic_class     = grh->traffic_class;
1633 	}
1634 	uverb_attr->port_num          = rdma_ah_get_port_num(rdma_attr);
1635 }
1636 
ib_uverbs_query_qp(struct uverbs_attr_bundle * attrs)1637 static int ib_uverbs_query_qp(struct uverbs_attr_bundle *attrs)
1638 {
1639 	struct ib_uverbs_query_qp      cmd;
1640 	struct ib_uverbs_query_qp_resp resp;
1641 	struct ib_qp                   *qp;
1642 	struct ib_qp_attr              *attr;
1643 	struct ib_qp_init_attr         *init_attr;
1644 	int                            ret;
1645 
1646 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1647 	if (ret)
1648 		return ret;
1649 
1650 	attr      = kmalloc(sizeof *attr, GFP_KERNEL);
1651 	init_attr = kmalloc(sizeof *init_attr, GFP_KERNEL);
1652 	if (!attr || !init_attr) {
1653 		ret = -ENOMEM;
1654 		goto out;
1655 	}
1656 
1657 	qp = uobj_get_obj_read(qp, UVERBS_OBJECT_QP, cmd.qp_handle, attrs);
1658 	if (!qp) {
1659 		ret = -EINVAL;
1660 		goto out;
1661 	}
1662 
1663 	ret = ib_query_qp(qp, attr, cmd.attr_mask, init_attr);
1664 
1665 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
1666 				UVERBS_LOOKUP_READ);
1667 
1668 	if (ret)
1669 		goto out;
1670 
1671 	memset(&resp, 0, sizeof resp);
1672 
1673 	resp.qp_state               = attr->qp_state;
1674 	resp.cur_qp_state           = attr->cur_qp_state;
1675 	resp.path_mtu               = attr->path_mtu;
1676 	resp.path_mig_state         = attr->path_mig_state;
1677 	resp.qkey                   = attr->qkey;
1678 	resp.rq_psn                 = attr->rq_psn;
1679 	resp.sq_psn                 = attr->sq_psn;
1680 	resp.dest_qp_num            = attr->dest_qp_num;
1681 	resp.qp_access_flags        = attr->qp_access_flags;
1682 	resp.pkey_index             = attr->pkey_index;
1683 	resp.alt_pkey_index         = attr->alt_pkey_index;
1684 	resp.sq_draining            = attr->sq_draining;
1685 	resp.max_rd_atomic          = attr->max_rd_atomic;
1686 	resp.max_dest_rd_atomic     = attr->max_dest_rd_atomic;
1687 	resp.min_rnr_timer          = attr->min_rnr_timer;
1688 	resp.port_num               = attr->port_num;
1689 	resp.timeout                = attr->timeout;
1690 	resp.retry_cnt              = attr->retry_cnt;
1691 	resp.rnr_retry              = attr->rnr_retry;
1692 	resp.alt_port_num           = attr->alt_port_num;
1693 	resp.alt_timeout            = attr->alt_timeout;
1694 
1695 	copy_ah_attr_to_uverbs(&resp.dest, &attr->ah_attr);
1696 	copy_ah_attr_to_uverbs(&resp.alt_dest, &attr->alt_ah_attr);
1697 
1698 	resp.max_send_wr            = init_attr->cap.max_send_wr;
1699 	resp.max_recv_wr            = init_attr->cap.max_recv_wr;
1700 	resp.max_send_sge           = init_attr->cap.max_send_sge;
1701 	resp.max_recv_sge           = init_attr->cap.max_recv_sge;
1702 	resp.max_inline_data        = init_attr->cap.max_inline_data;
1703 	resp.sq_sig_all             = init_attr->sq_sig_type == IB_SIGNAL_ALL_WR;
1704 
1705 	ret = uverbs_response(attrs, &resp, sizeof(resp));
1706 
1707 out:
1708 	kfree(attr);
1709 	kfree(init_attr);
1710 
1711 	return ret;
1712 }
1713 
1714 /* Remove ignored fields set in the attribute mask */
modify_qp_mask(enum ib_qp_type qp_type,int mask)1715 static int modify_qp_mask(enum ib_qp_type qp_type, int mask)
1716 {
1717 	switch (qp_type) {
1718 	case IB_QPT_XRC_INI:
1719 		return mask & ~(IB_QP_MAX_DEST_RD_ATOMIC | IB_QP_MIN_RNR_TIMER);
1720 	case IB_QPT_XRC_TGT:
1721 		return mask & ~(IB_QP_MAX_QP_RD_ATOMIC | IB_QP_RETRY_CNT |
1722 				IB_QP_RNR_RETRY);
1723 	default:
1724 		return mask;
1725 	}
1726 }
1727 
copy_ah_attr_from_uverbs(struct ib_device * dev,struct rdma_ah_attr * rdma_attr,struct ib_uverbs_qp_dest * uverb_attr)1728 static void copy_ah_attr_from_uverbs(struct ib_device *dev,
1729 				     struct rdma_ah_attr *rdma_attr,
1730 				     struct ib_uverbs_qp_dest *uverb_attr)
1731 {
1732 	rdma_attr->type = rdma_ah_find_type(dev, uverb_attr->port_num);
1733 	if (uverb_attr->is_global) {
1734 		rdma_ah_set_grh(rdma_attr, NULL,
1735 				uverb_attr->flow_label,
1736 				uverb_attr->sgid_index,
1737 				uverb_attr->hop_limit,
1738 				uverb_attr->traffic_class);
1739 		rdma_ah_set_dgid_raw(rdma_attr, uverb_attr->dgid);
1740 	} else {
1741 		rdma_ah_set_ah_flags(rdma_attr, 0);
1742 	}
1743 	rdma_ah_set_dlid(rdma_attr, uverb_attr->dlid);
1744 	rdma_ah_set_sl(rdma_attr, uverb_attr->sl);
1745 	rdma_ah_set_path_bits(rdma_attr, uverb_attr->src_path_bits);
1746 	rdma_ah_set_static_rate(rdma_attr, uverb_attr->static_rate);
1747 	rdma_ah_set_port_num(rdma_attr, uverb_attr->port_num);
1748 	rdma_ah_set_make_grd(rdma_attr, false);
1749 }
1750 
modify_qp(struct uverbs_attr_bundle * attrs,struct ib_uverbs_ex_modify_qp * cmd)1751 static int modify_qp(struct uverbs_attr_bundle *attrs,
1752 		     struct ib_uverbs_ex_modify_qp *cmd)
1753 {
1754 	struct ib_qp_attr *attr;
1755 	struct ib_qp *qp;
1756 	int ret;
1757 
1758 	attr = kzalloc(sizeof(*attr), GFP_KERNEL);
1759 	if (!attr)
1760 		return -ENOMEM;
1761 
1762 	qp = uobj_get_obj_read(qp, UVERBS_OBJECT_QP, cmd->base.qp_handle,
1763 			       attrs);
1764 	if (!qp) {
1765 		ret = -EINVAL;
1766 		goto out;
1767 	}
1768 
1769 	if ((cmd->base.attr_mask & IB_QP_PORT) &&
1770 	    !rdma_is_port_valid(qp->device, cmd->base.port_num)) {
1771 		ret = -EINVAL;
1772 		goto release_qp;
1773 	}
1774 
1775 	if ((cmd->base.attr_mask & IB_QP_AV)) {
1776 		if (!rdma_is_port_valid(qp->device, cmd->base.dest.port_num)) {
1777 			ret = -EINVAL;
1778 			goto release_qp;
1779 		}
1780 
1781 		if (cmd->base.attr_mask & IB_QP_STATE &&
1782 		    cmd->base.qp_state == IB_QPS_RTR) {
1783 		/* We are in INIT->RTR TRANSITION (if we are not,
1784 		 * this transition will be rejected in subsequent checks).
1785 		 * In the INIT->RTR transition, we cannot have IB_QP_PORT set,
1786 		 * but the IB_QP_STATE flag is required.
1787 		 *
1788 		 * Since kernel 3.14 (commit dbf727de7440), the uverbs driver,
1789 		 * when IB_QP_AV is set, has required inclusion of a valid
1790 		 * port number in the primary AV. (AVs are created and handled
1791 		 * differently for infiniband and ethernet (RoCE) ports).
1792 		 *
1793 		 * Check the port number included in the primary AV against
1794 		 * the port number in the qp struct, which was set (and saved)
1795 		 * in the RST->INIT transition.
1796 		 */
1797 			if (cmd->base.dest.port_num != qp->real_qp->port) {
1798 				ret = -EINVAL;
1799 				goto release_qp;
1800 			}
1801 		} else {
1802 		/* We are in SQD->SQD. (If we are not, this transition will
1803 		 * be rejected later in the verbs layer checks).
1804 		 * Check for both IB_QP_PORT and IB_QP_AV, these can be set
1805 		 * together in the SQD->SQD transition.
1806 		 *
1807 		 * If only IP_QP_AV was set, add in IB_QP_PORT as well (the
1808 		 * verbs layer driver does not track primary port changes
1809 		 * resulting from path migration. Thus, in SQD, if the primary
1810 		 * AV is modified, the primary port should also be modified).
1811 		 *
1812 		 * Note that in this transition, the IB_QP_STATE flag
1813 		 * is not allowed.
1814 		 */
1815 			if (((cmd->base.attr_mask & (IB_QP_AV | IB_QP_PORT))
1816 			     == (IB_QP_AV | IB_QP_PORT)) &&
1817 			    cmd->base.port_num != cmd->base.dest.port_num) {
1818 				ret = -EINVAL;
1819 				goto release_qp;
1820 			}
1821 			if ((cmd->base.attr_mask & (IB_QP_AV | IB_QP_PORT))
1822 			    == IB_QP_AV) {
1823 				cmd->base.attr_mask |= IB_QP_PORT;
1824 				cmd->base.port_num = cmd->base.dest.port_num;
1825 			}
1826 		}
1827 	}
1828 
1829 	if ((cmd->base.attr_mask & IB_QP_ALT_PATH) &&
1830 	    (!rdma_is_port_valid(qp->device, cmd->base.alt_port_num) ||
1831 	    !rdma_is_port_valid(qp->device, cmd->base.alt_dest.port_num) ||
1832 	    cmd->base.alt_port_num != cmd->base.alt_dest.port_num)) {
1833 		ret = -EINVAL;
1834 		goto release_qp;
1835 	}
1836 
1837 	if ((cmd->base.attr_mask & IB_QP_CUR_STATE &&
1838 	    cmd->base.cur_qp_state > IB_QPS_ERR) ||
1839 	    (cmd->base.attr_mask & IB_QP_STATE &&
1840 	    cmd->base.qp_state > IB_QPS_ERR)) {
1841 		ret = -EINVAL;
1842 		goto release_qp;
1843 	}
1844 
1845 	if (cmd->base.attr_mask & IB_QP_STATE)
1846 		attr->qp_state = cmd->base.qp_state;
1847 	if (cmd->base.attr_mask & IB_QP_CUR_STATE)
1848 		attr->cur_qp_state = cmd->base.cur_qp_state;
1849 	if (cmd->base.attr_mask & IB_QP_PATH_MTU)
1850 		attr->path_mtu = cmd->base.path_mtu;
1851 	if (cmd->base.attr_mask & IB_QP_PATH_MIG_STATE)
1852 		attr->path_mig_state = cmd->base.path_mig_state;
1853 	if (cmd->base.attr_mask & IB_QP_QKEY) {
1854 		if (cmd->base.qkey & IB_QP_SET_QKEY && !capable(CAP_NET_RAW)) {
1855 			ret = -EPERM;
1856 			goto release_qp;
1857 		}
1858 		attr->qkey = cmd->base.qkey;
1859 	}
1860 	if (cmd->base.attr_mask & IB_QP_RQ_PSN)
1861 		attr->rq_psn = cmd->base.rq_psn;
1862 	if (cmd->base.attr_mask & IB_QP_SQ_PSN)
1863 		attr->sq_psn = cmd->base.sq_psn;
1864 	if (cmd->base.attr_mask & IB_QP_DEST_QPN)
1865 		attr->dest_qp_num = cmd->base.dest_qp_num;
1866 	if (cmd->base.attr_mask & IB_QP_ACCESS_FLAGS)
1867 		attr->qp_access_flags = cmd->base.qp_access_flags;
1868 	if (cmd->base.attr_mask & IB_QP_PKEY_INDEX)
1869 		attr->pkey_index = cmd->base.pkey_index;
1870 	if (cmd->base.attr_mask & IB_QP_EN_SQD_ASYNC_NOTIFY)
1871 		attr->en_sqd_async_notify = cmd->base.en_sqd_async_notify;
1872 	if (cmd->base.attr_mask & IB_QP_MAX_QP_RD_ATOMIC)
1873 		attr->max_rd_atomic = cmd->base.max_rd_atomic;
1874 	if (cmd->base.attr_mask & IB_QP_MAX_DEST_RD_ATOMIC)
1875 		attr->max_dest_rd_atomic = cmd->base.max_dest_rd_atomic;
1876 	if (cmd->base.attr_mask & IB_QP_MIN_RNR_TIMER)
1877 		attr->min_rnr_timer = cmd->base.min_rnr_timer;
1878 	if (cmd->base.attr_mask & IB_QP_PORT)
1879 		attr->port_num = cmd->base.port_num;
1880 	if (cmd->base.attr_mask & IB_QP_TIMEOUT)
1881 		attr->timeout = cmd->base.timeout;
1882 	if (cmd->base.attr_mask & IB_QP_RETRY_CNT)
1883 		attr->retry_cnt = cmd->base.retry_cnt;
1884 	if (cmd->base.attr_mask & IB_QP_RNR_RETRY)
1885 		attr->rnr_retry = cmd->base.rnr_retry;
1886 	if (cmd->base.attr_mask & IB_QP_ALT_PATH) {
1887 		attr->alt_port_num = cmd->base.alt_port_num;
1888 		attr->alt_timeout = cmd->base.alt_timeout;
1889 		attr->alt_pkey_index = cmd->base.alt_pkey_index;
1890 	}
1891 	if (cmd->base.attr_mask & IB_QP_RATE_LIMIT)
1892 		attr->rate_limit = cmd->rate_limit;
1893 
1894 	if (cmd->base.attr_mask & IB_QP_AV)
1895 		copy_ah_attr_from_uverbs(qp->device, &attr->ah_attr,
1896 					 &cmd->base.dest);
1897 
1898 	if (cmd->base.attr_mask & IB_QP_ALT_PATH)
1899 		copy_ah_attr_from_uverbs(qp->device, &attr->alt_ah_attr,
1900 					 &cmd->base.alt_dest);
1901 
1902 	ret = ib_modify_qp_with_udata(qp, attr,
1903 				      modify_qp_mask(qp->qp_type,
1904 						     cmd->base.attr_mask),
1905 				      &attrs->driver_udata);
1906 
1907 release_qp:
1908 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
1909 				UVERBS_LOOKUP_READ);
1910 out:
1911 	kfree(attr);
1912 
1913 	return ret;
1914 }
1915 
ib_uverbs_modify_qp(struct uverbs_attr_bundle * attrs)1916 static int ib_uverbs_modify_qp(struct uverbs_attr_bundle *attrs)
1917 {
1918 	struct ib_uverbs_ex_modify_qp cmd;
1919 	int ret;
1920 
1921 	ret = uverbs_request(attrs, &cmd.base, sizeof(cmd.base));
1922 	if (ret)
1923 		return ret;
1924 
1925 	if (cmd.base.attr_mask & ~IB_QP_ATTR_STANDARD_BITS)
1926 		return -EOPNOTSUPP;
1927 
1928 	return modify_qp(attrs, &cmd);
1929 }
1930 
ib_uverbs_ex_modify_qp(struct uverbs_attr_bundle * attrs)1931 static int ib_uverbs_ex_modify_qp(struct uverbs_attr_bundle *attrs)
1932 {
1933 	struct ib_uverbs_ex_modify_qp cmd;
1934 	struct ib_uverbs_ex_modify_qp_resp resp = {
1935 		.response_length = uverbs_response_length(attrs, sizeof(resp))
1936 	};
1937 	int ret;
1938 
1939 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1940 	if (ret)
1941 		return ret;
1942 
1943 	/*
1944 	 * Last bit is reserved for extending the attr_mask by
1945 	 * using another field.
1946 	 */
1947 	if (cmd.base.attr_mask & ~(IB_QP_ATTR_STANDARD_BITS | IB_QP_RATE_LIMIT))
1948 		return -EOPNOTSUPP;
1949 
1950 	ret = modify_qp(attrs, &cmd);
1951 	if (ret)
1952 		return ret;
1953 
1954 	return uverbs_response(attrs, &resp, sizeof(resp));
1955 }
1956 
ib_uverbs_destroy_qp(struct uverbs_attr_bundle * attrs)1957 static int ib_uverbs_destroy_qp(struct uverbs_attr_bundle *attrs)
1958 {
1959 	struct ib_uverbs_destroy_qp      cmd;
1960 	struct ib_uverbs_destroy_qp_resp resp;
1961 	struct ib_uobject		*uobj;
1962 	struct ib_uqp_object        	*obj;
1963 	int ret;
1964 
1965 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
1966 	if (ret)
1967 		return ret;
1968 
1969 	uobj = uobj_get_destroy(UVERBS_OBJECT_QP, cmd.qp_handle, attrs);
1970 	if (IS_ERR(uobj))
1971 		return PTR_ERR(uobj);
1972 
1973 	obj = container_of(uobj, struct ib_uqp_object, uevent.uobject);
1974 	memset(&resp, 0, sizeof(resp));
1975 	resp.events_reported = obj->uevent.events_reported;
1976 
1977 	uobj_put_destroy(uobj);
1978 
1979 	return uverbs_response(attrs, &resp, sizeof(resp));
1980 }
1981 
alloc_wr(size_t wr_size,__u32 num_sge)1982 static void *alloc_wr(size_t wr_size, __u32 num_sge)
1983 {
1984 	if (num_sge >= (U32_MAX - ALIGN(wr_size, sizeof(struct ib_sge))) /
1985 			       sizeof(struct ib_sge))
1986 		return NULL;
1987 
1988 	return kmalloc(ALIGN(wr_size, sizeof(struct ib_sge)) +
1989 			       num_sge * sizeof(struct ib_sge),
1990 		       GFP_KERNEL);
1991 }
1992 
ib_uverbs_post_send(struct uverbs_attr_bundle * attrs)1993 static int ib_uverbs_post_send(struct uverbs_attr_bundle *attrs)
1994 {
1995 	struct ib_uverbs_post_send      cmd;
1996 	struct ib_uverbs_post_send_resp resp;
1997 	struct ib_uverbs_send_wr       *user_wr;
1998 	struct ib_send_wr              *wr = NULL, *last, *next;
1999 	const struct ib_send_wr	       *bad_wr;
2000 	struct ib_qp                   *qp;
2001 	int                             i, sg_ind;
2002 	int				is_ud;
2003 	int ret, ret2;
2004 	size_t                          next_size;
2005 	const struct ib_sge __user *sgls;
2006 	const void __user *wqes;
2007 	struct uverbs_req_iter iter;
2008 
2009 	ret = uverbs_request_start(attrs, &iter, &cmd, sizeof(cmd));
2010 	if (ret)
2011 		return ret;
2012 	wqes = uverbs_request_next_ptr(&iter, cmd.wqe_size * cmd.wr_count);
2013 	if (IS_ERR(wqes))
2014 		return PTR_ERR(wqes);
2015 	sgls = uverbs_request_next_ptr(
2016 		&iter, cmd.sge_count * sizeof(struct ib_uverbs_sge));
2017 	if (IS_ERR(sgls))
2018 		return PTR_ERR(sgls);
2019 	ret = uverbs_request_finish(&iter);
2020 	if (ret)
2021 		return ret;
2022 
2023 	user_wr = kmalloc(cmd.wqe_size, GFP_KERNEL);
2024 	if (!user_wr)
2025 		return -ENOMEM;
2026 
2027 	qp = uobj_get_obj_read(qp, UVERBS_OBJECT_QP, cmd.qp_handle, attrs);
2028 	if (!qp) {
2029 		ret = -EINVAL;
2030 		goto out;
2031 	}
2032 
2033 	is_ud = qp->qp_type == IB_QPT_UD;
2034 	sg_ind = 0;
2035 	last = NULL;
2036 	for (i = 0; i < cmd.wr_count; ++i) {
2037 		if (copy_from_user(user_wr, wqes + i * cmd.wqe_size,
2038 				   cmd.wqe_size)) {
2039 			ret = -EFAULT;
2040 			goto out_put;
2041 		}
2042 
2043 		if (user_wr->num_sge + sg_ind > cmd.sge_count) {
2044 			ret = -EINVAL;
2045 			goto out_put;
2046 		}
2047 
2048 		if (is_ud) {
2049 			struct ib_ud_wr *ud;
2050 
2051 			if (user_wr->opcode != IB_WR_SEND &&
2052 			    user_wr->opcode != IB_WR_SEND_WITH_IMM) {
2053 				ret = -EINVAL;
2054 				goto out_put;
2055 			}
2056 
2057 			next_size = sizeof(*ud);
2058 			ud = alloc_wr(next_size, user_wr->num_sge);
2059 			if (!ud) {
2060 				ret = -ENOMEM;
2061 				goto out_put;
2062 			}
2063 
2064 			ud->ah = uobj_get_obj_read(ah, UVERBS_OBJECT_AH,
2065 						   user_wr->wr.ud.ah, attrs);
2066 			if (!ud->ah) {
2067 				kfree(ud);
2068 				ret = -EINVAL;
2069 				goto out_put;
2070 			}
2071 			ud->remote_qpn = user_wr->wr.ud.remote_qpn;
2072 			ud->remote_qkey = user_wr->wr.ud.remote_qkey;
2073 
2074 			next = &ud->wr;
2075 		} else if (user_wr->opcode == IB_WR_RDMA_WRITE_WITH_IMM ||
2076 			   user_wr->opcode == IB_WR_RDMA_WRITE ||
2077 			   user_wr->opcode == IB_WR_RDMA_READ) {
2078 			struct ib_rdma_wr *rdma;
2079 
2080 			next_size = sizeof(*rdma);
2081 			rdma = alloc_wr(next_size, user_wr->num_sge);
2082 			if (!rdma) {
2083 				ret = -ENOMEM;
2084 				goto out_put;
2085 			}
2086 
2087 			rdma->remote_addr = user_wr->wr.rdma.remote_addr;
2088 			rdma->rkey = user_wr->wr.rdma.rkey;
2089 
2090 			next = &rdma->wr;
2091 		} else if (user_wr->opcode == IB_WR_ATOMIC_CMP_AND_SWP ||
2092 			   user_wr->opcode == IB_WR_ATOMIC_FETCH_AND_ADD) {
2093 			struct ib_atomic_wr *atomic;
2094 
2095 			next_size = sizeof(*atomic);
2096 			atomic = alloc_wr(next_size, user_wr->num_sge);
2097 			if (!atomic) {
2098 				ret = -ENOMEM;
2099 				goto out_put;
2100 			}
2101 
2102 			atomic->remote_addr = user_wr->wr.atomic.remote_addr;
2103 			atomic->compare_add = user_wr->wr.atomic.compare_add;
2104 			atomic->swap = user_wr->wr.atomic.swap;
2105 			atomic->rkey = user_wr->wr.atomic.rkey;
2106 
2107 			next = &atomic->wr;
2108 		} else if (user_wr->opcode == IB_WR_SEND ||
2109 			   user_wr->opcode == IB_WR_SEND_WITH_IMM ||
2110 			   user_wr->opcode == IB_WR_SEND_WITH_INV) {
2111 			next_size = sizeof(*next);
2112 			next = alloc_wr(next_size, user_wr->num_sge);
2113 			if (!next) {
2114 				ret = -ENOMEM;
2115 				goto out_put;
2116 			}
2117 		} else {
2118 			ret = -EINVAL;
2119 			goto out_put;
2120 		}
2121 
2122 		if (user_wr->opcode == IB_WR_SEND_WITH_IMM ||
2123 		    user_wr->opcode == IB_WR_RDMA_WRITE_WITH_IMM) {
2124 			next->ex.imm_data =
2125 					(__be32 __force) user_wr->ex.imm_data;
2126 		} else if (user_wr->opcode == IB_WR_SEND_WITH_INV) {
2127 			next->ex.invalidate_rkey = user_wr->ex.invalidate_rkey;
2128 		}
2129 
2130 		if (!last)
2131 			wr = next;
2132 		else
2133 			last->next = next;
2134 		last = next;
2135 
2136 		next->next       = NULL;
2137 		next->wr_id      = user_wr->wr_id;
2138 		next->num_sge    = user_wr->num_sge;
2139 		next->opcode     = user_wr->opcode;
2140 		next->send_flags = user_wr->send_flags;
2141 
2142 		if (next->num_sge) {
2143 			next->sg_list = (void *) next +
2144 				ALIGN(next_size, sizeof(struct ib_sge));
2145 			if (copy_from_user(next->sg_list, sgls + sg_ind,
2146 					   next->num_sge *
2147 						   sizeof(struct ib_sge))) {
2148 				ret = -EFAULT;
2149 				goto out_put;
2150 			}
2151 			sg_ind += next->num_sge;
2152 		} else
2153 			next->sg_list = NULL;
2154 	}
2155 
2156 	resp.bad_wr = 0;
2157 	ret = qp->device->ops.post_send(qp->real_qp, wr, &bad_wr);
2158 	if (ret)
2159 		for (next = wr; next; next = next->next) {
2160 			++resp.bad_wr;
2161 			if (next == bad_wr)
2162 				break;
2163 		}
2164 
2165 	ret2 = uverbs_response(attrs, &resp, sizeof(resp));
2166 	if (ret2)
2167 		ret = ret2;
2168 
2169 out_put:
2170 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
2171 				UVERBS_LOOKUP_READ);
2172 
2173 	while (wr) {
2174 		if (is_ud && ud_wr(wr)->ah)
2175 			uobj_put_obj_read(ud_wr(wr)->ah);
2176 		next = wr->next;
2177 		kfree(wr);
2178 		wr = next;
2179 	}
2180 
2181 out:
2182 	kfree(user_wr);
2183 
2184 	return ret;
2185 }
2186 
2187 static struct ib_recv_wr *
ib_uverbs_unmarshall_recv(struct uverbs_req_iter * iter,u32 wr_count,u32 wqe_size,u32 sge_count)2188 ib_uverbs_unmarshall_recv(struct uverbs_req_iter *iter, u32 wr_count,
2189 			  u32 wqe_size, u32 sge_count)
2190 {
2191 	struct ib_uverbs_recv_wr *user_wr;
2192 	struct ib_recv_wr        *wr = NULL, *last, *next;
2193 	int                       sg_ind;
2194 	int                       i;
2195 	int                       ret;
2196 	const struct ib_sge __user *sgls;
2197 	const void __user *wqes;
2198 
2199 	if (wqe_size < sizeof(struct ib_uverbs_recv_wr))
2200 		return ERR_PTR(-EINVAL);
2201 
2202 	wqes = uverbs_request_next_ptr(iter, wqe_size * wr_count);
2203 	if (IS_ERR(wqes))
2204 		return ERR_CAST(wqes);
2205 	sgls = uverbs_request_next_ptr(
2206 		iter, sge_count * sizeof(struct ib_uverbs_sge));
2207 	if (IS_ERR(sgls))
2208 		return ERR_CAST(sgls);
2209 	ret = uverbs_request_finish(iter);
2210 	if (ret)
2211 		return ERR_PTR(ret);
2212 
2213 	user_wr = kmalloc(wqe_size, GFP_KERNEL);
2214 	if (!user_wr)
2215 		return ERR_PTR(-ENOMEM);
2216 
2217 	sg_ind = 0;
2218 	last = NULL;
2219 	for (i = 0; i < wr_count; ++i) {
2220 		if (copy_from_user(user_wr, wqes + i * wqe_size,
2221 				   wqe_size)) {
2222 			ret = -EFAULT;
2223 			goto err;
2224 		}
2225 
2226 		if (user_wr->num_sge + sg_ind > sge_count) {
2227 			ret = -EINVAL;
2228 			goto err;
2229 		}
2230 
2231 		if (user_wr->num_sge >=
2232 		    (U32_MAX - ALIGN(sizeof(*next), sizeof(struct ib_sge))) /
2233 			    sizeof(struct ib_sge)) {
2234 			ret = -EINVAL;
2235 			goto err;
2236 		}
2237 
2238 		next = kmalloc(ALIGN(sizeof(*next), sizeof(struct ib_sge)) +
2239 				       user_wr->num_sge * sizeof(struct ib_sge),
2240 			       GFP_KERNEL);
2241 		if (!next) {
2242 			ret = -ENOMEM;
2243 			goto err;
2244 		}
2245 
2246 		if (!last)
2247 			wr = next;
2248 		else
2249 			last->next = next;
2250 		last = next;
2251 
2252 		next->next       = NULL;
2253 		next->wr_id      = user_wr->wr_id;
2254 		next->num_sge    = user_wr->num_sge;
2255 
2256 		if (next->num_sge) {
2257 			next->sg_list = (void *)next +
2258 				ALIGN(sizeof(*next), sizeof(struct ib_sge));
2259 			if (copy_from_user(next->sg_list, sgls + sg_ind,
2260 					   next->num_sge *
2261 						   sizeof(struct ib_sge))) {
2262 				ret = -EFAULT;
2263 				goto err;
2264 			}
2265 			sg_ind += next->num_sge;
2266 		} else
2267 			next->sg_list = NULL;
2268 	}
2269 
2270 	kfree(user_wr);
2271 	return wr;
2272 
2273 err:
2274 	kfree(user_wr);
2275 
2276 	while (wr) {
2277 		next = wr->next;
2278 		kfree(wr);
2279 		wr = next;
2280 	}
2281 
2282 	return ERR_PTR(ret);
2283 }
2284 
ib_uverbs_post_recv(struct uverbs_attr_bundle * attrs)2285 static int ib_uverbs_post_recv(struct uverbs_attr_bundle *attrs)
2286 {
2287 	struct ib_uverbs_post_recv      cmd;
2288 	struct ib_uverbs_post_recv_resp resp;
2289 	struct ib_recv_wr              *wr, *next;
2290 	const struct ib_recv_wr	       *bad_wr;
2291 	struct ib_qp                   *qp;
2292 	int ret, ret2;
2293 	struct uverbs_req_iter iter;
2294 
2295 	ret = uverbs_request_start(attrs, &iter, &cmd, sizeof(cmd));
2296 	if (ret)
2297 		return ret;
2298 
2299 	wr = ib_uverbs_unmarshall_recv(&iter, cmd.wr_count, cmd.wqe_size,
2300 				       cmd.sge_count);
2301 	if (IS_ERR(wr))
2302 		return PTR_ERR(wr);
2303 
2304 	qp = uobj_get_obj_read(qp, UVERBS_OBJECT_QP, cmd.qp_handle, attrs);
2305 	if (!qp) {
2306 		ret = -EINVAL;
2307 		goto out;
2308 	}
2309 
2310 	resp.bad_wr = 0;
2311 	ret = qp->device->ops.post_recv(qp->real_qp, wr, &bad_wr);
2312 
2313 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
2314 				UVERBS_LOOKUP_READ);
2315 	if (ret) {
2316 		for (next = wr; next; next = next->next) {
2317 			++resp.bad_wr;
2318 			if (next == bad_wr)
2319 				break;
2320 		}
2321 	}
2322 
2323 	ret2 = uverbs_response(attrs, &resp, sizeof(resp));
2324 	if (ret2)
2325 		ret = ret2;
2326 out:
2327 	while (wr) {
2328 		next = wr->next;
2329 		kfree(wr);
2330 		wr = next;
2331 	}
2332 
2333 	return ret;
2334 }
2335 
ib_uverbs_post_srq_recv(struct uverbs_attr_bundle * attrs)2336 static int ib_uverbs_post_srq_recv(struct uverbs_attr_bundle *attrs)
2337 {
2338 	struct ib_uverbs_post_srq_recv      cmd;
2339 	struct ib_uverbs_post_srq_recv_resp resp;
2340 	struct ib_recv_wr                  *wr, *next;
2341 	const struct ib_recv_wr		   *bad_wr;
2342 	struct ib_srq                      *srq;
2343 	int ret, ret2;
2344 	struct uverbs_req_iter iter;
2345 
2346 	ret = uverbs_request_start(attrs, &iter, &cmd, sizeof(cmd));
2347 	if (ret)
2348 		return ret;
2349 
2350 	wr = ib_uverbs_unmarshall_recv(&iter, cmd.wr_count, cmd.wqe_size,
2351 				       cmd.sge_count);
2352 	if (IS_ERR(wr))
2353 		return PTR_ERR(wr);
2354 
2355 	srq = uobj_get_obj_read(srq, UVERBS_OBJECT_SRQ, cmd.srq_handle, attrs);
2356 	if (!srq) {
2357 		ret = -EINVAL;
2358 		goto out;
2359 	}
2360 
2361 	resp.bad_wr = 0;
2362 	ret = srq->device->ops.post_srq_recv(srq, wr, &bad_wr);
2363 
2364 	rdma_lookup_put_uobject(&srq->uobject->uevent.uobject,
2365 				UVERBS_LOOKUP_READ);
2366 
2367 	if (ret)
2368 		for (next = wr; next; next = next->next) {
2369 			++resp.bad_wr;
2370 			if (next == bad_wr)
2371 				break;
2372 		}
2373 
2374 	ret2 = uverbs_response(attrs, &resp, sizeof(resp));
2375 	if (ret2)
2376 		ret = ret2;
2377 
2378 out:
2379 	while (wr) {
2380 		next = wr->next;
2381 		kfree(wr);
2382 		wr = next;
2383 	}
2384 
2385 	return ret;
2386 }
2387 
ib_uverbs_create_ah(struct uverbs_attr_bundle * attrs)2388 static int ib_uverbs_create_ah(struct uverbs_attr_bundle *attrs)
2389 {
2390 	struct ib_uverbs_create_ah	 cmd;
2391 	struct ib_uverbs_create_ah_resp	 resp;
2392 	struct ib_uobject		*uobj;
2393 	struct ib_pd			*pd;
2394 	struct ib_ah			*ah;
2395 	struct rdma_ah_attr		attr = {};
2396 	int ret;
2397 	struct ib_device *ib_dev;
2398 
2399 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
2400 	if (ret)
2401 		return ret;
2402 
2403 	uobj = uobj_alloc(UVERBS_OBJECT_AH, attrs, &ib_dev);
2404 	if (IS_ERR(uobj))
2405 		return PTR_ERR(uobj);
2406 
2407 	if (!rdma_is_port_valid(ib_dev, cmd.attr.port_num)) {
2408 		ret = -EINVAL;
2409 		goto err;
2410 	}
2411 
2412 	pd = uobj_get_obj_read(pd, UVERBS_OBJECT_PD, cmd.pd_handle, attrs);
2413 	if (!pd) {
2414 		ret = -EINVAL;
2415 		goto err;
2416 	}
2417 
2418 	attr.type = rdma_ah_find_type(ib_dev, cmd.attr.port_num);
2419 	rdma_ah_set_make_grd(&attr, false);
2420 	rdma_ah_set_dlid(&attr, cmd.attr.dlid);
2421 	rdma_ah_set_sl(&attr, cmd.attr.sl);
2422 	rdma_ah_set_path_bits(&attr, cmd.attr.src_path_bits);
2423 	rdma_ah_set_static_rate(&attr, cmd.attr.static_rate);
2424 	rdma_ah_set_port_num(&attr, cmd.attr.port_num);
2425 
2426 	if (cmd.attr.is_global) {
2427 		rdma_ah_set_grh(&attr, NULL, cmd.attr.grh.flow_label,
2428 				cmd.attr.grh.sgid_index,
2429 				cmd.attr.grh.hop_limit,
2430 				cmd.attr.grh.traffic_class);
2431 		rdma_ah_set_dgid_raw(&attr, cmd.attr.grh.dgid);
2432 	} else {
2433 		rdma_ah_set_ah_flags(&attr, 0);
2434 	}
2435 
2436 	ah = rdma_create_user_ah(pd, &attr, &attrs->driver_udata);
2437 	if (IS_ERR(ah)) {
2438 		ret = PTR_ERR(ah);
2439 		goto err_put;
2440 	}
2441 
2442 	ah->uobject  = uobj;
2443 	uobj->user_handle = cmd.user_handle;
2444 	uobj->object = ah;
2445 	uobj_put_obj_read(pd);
2446 	uobj_finalize_uobj_create(uobj, attrs);
2447 
2448 	resp.ah_handle = uobj->id;
2449 	return uverbs_response(attrs, &resp, sizeof(resp));
2450 
2451 err_put:
2452 	uobj_put_obj_read(pd);
2453 err:
2454 	uobj_alloc_abort(uobj, attrs);
2455 	return ret;
2456 }
2457 
ib_uverbs_destroy_ah(struct uverbs_attr_bundle * attrs)2458 static int ib_uverbs_destroy_ah(struct uverbs_attr_bundle *attrs)
2459 {
2460 	struct ib_uverbs_destroy_ah cmd;
2461 	int ret;
2462 
2463 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
2464 	if (ret)
2465 		return ret;
2466 
2467 	return uobj_perform_destroy(UVERBS_OBJECT_AH, cmd.ah_handle, attrs);
2468 }
2469 
ib_uverbs_attach_mcast(struct uverbs_attr_bundle * attrs)2470 static int ib_uverbs_attach_mcast(struct uverbs_attr_bundle *attrs)
2471 {
2472 	struct ib_uverbs_attach_mcast cmd;
2473 	struct ib_qp                 *qp;
2474 	struct ib_uqp_object         *obj;
2475 	struct ib_uverbs_mcast_entry *mcast;
2476 	int                           ret;
2477 
2478 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
2479 	if (ret)
2480 		return ret;
2481 
2482 	qp = uobj_get_obj_read(qp, UVERBS_OBJECT_QP, cmd.qp_handle, attrs);
2483 	if (!qp)
2484 		return -EINVAL;
2485 
2486 	obj = qp->uobject;
2487 
2488 	mutex_lock(&obj->mcast_lock);
2489 	list_for_each_entry(mcast, &obj->mcast_list, list)
2490 		if (cmd.mlid == mcast->lid &&
2491 		    !memcmp(cmd.gid, mcast->gid.raw, sizeof mcast->gid.raw)) {
2492 			ret = 0;
2493 			goto out_put;
2494 		}
2495 
2496 	mcast = kmalloc(sizeof *mcast, GFP_KERNEL);
2497 	if (!mcast) {
2498 		ret = -ENOMEM;
2499 		goto out_put;
2500 	}
2501 
2502 	mcast->lid = cmd.mlid;
2503 	memcpy(mcast->gid.raw, cmd.gid, sizeof mcast->gid.raw);
2504 
2505 	ret = ib_attach_mcast(qp, &mcast->gid, cmd.mlid);
2506 	if (!ret)
2507 		list_add_tail(&mcast->list, &obj->mcast_list);
2508 	else
2509 		kfree(mcast);
2510 
2511 out_put:
2512 	mutex_unlock(&obj->mcast_lock);
2513 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
2514 				UVERBS_LOOKUP_READ);
2515 
2516 	return ret;
2517 }
2518 
ib_uverbs_detach_mcast(struct uverbs_attr_bundle * attrs)2519 static int ib_uverbs_detach_mcast(struct uverbs_attr_bundle *attrs)
2520 {
2521 	struct ib_uverbs_detach_mcast cmd;
2522 	struct ib_uqp_object         *obj;
2523 	struct ib_qp                 *qp;
2524 	struct ib_uverbs_mcast_entry *mcast;
2525 	int                           ret;
2526 	bool                          found = false;
2527 
2528 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
2529 	if (ret)
2530 		return ret;
2531 
2532 	qp = uobj_get_obj_read(qp, UVERBS_OBJECT_QP, cmd.qp_handle, attrs);
2533 	if (!qp)
2534 		return -EINVAL;
2535 
2536 	obj = qp->uobject;
2537 	mutex_lock(&obj->mcast_lock);
2538 
2539 	list_for_each_entry(mcast, &obj->mcast_list, list)
2540 		if (cmd.mlid == mcast->lid &&
2541 		    !memcmp(cmd.gid, mcast->gid.raw, sizeof mcast->gid.raw)) {
2542 			list_del(&mcast->list);
2543 			kfree(mcast);
2544 			found = true;
2545 			break;
2546 		}
2547 
2548 	if (!found) {
2549 		ret = -EINVAL;
2550 		goto out_put;
2551 	}
2552 
2553 	ret = ib_detach_mcast(qp, (union ib_gid *)cmd.gid, cmd.mlid);
2554 
2555 out_put:
2556 	mutex_unlock(&obj->mcast_lock);
2557 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
2558 				UVERBS_LOOKUP_READ);
2559 	return ret;
2560 }
2561 
flow_resources_alloc(size_t num_specs)2562 struct ib_uflow_resources *flow_resources_alloc(size_t num_specs)
2563 {
2564 	struct ib_uflow_resources *resources;
2565 
2566 	resources = kzalloc(sizeof(*resources), GFP_KERNEL);
2567 
2568 	if (!resources)
2569 		return NULL;
2570 
2571 	if (!num_specs)
2572 		goto out;
2573 
2574 	resources->counters =
2575 		kcalloc(num_specs, sizeof(*resources->counters), GFP_KERNEL);
2576 	resources->collection =
2577 		kcalloc(num_specs, sizeof(*resources->collection), GFP_KERNEL);
2578 
2579 	if (!resources->counters || !resources->collection)
2580 		goto err;
2581 
2582 out:
2583 	resources->max = num_specs;
2584 	return resources;
2585 
2586 err:
2587 	kfree(resources->counters);
2588 	kfree(resources);
2589 
2590 	return NULL;
2591 }
2592 EXPORT_SYMBOL(flow_resources_alloc);
2593 
ib_uverbs_flow_resources_free(struct ib_uflow_resources * uflow_res)2594 void ib_uverbs_flow_resources_free(struct ib_uflow_resources *uflow_res)
2595 {
2596 	unsigned int i;
2597 
2598 	if (!uflow_res)
2599 		return;
2600 
2601 	for (i = 0; i < uflow_res->collection_num; i++)
2602 		atomic_dec(&uflow_res->collection[i]->usecnt);
2603 
2604 	for (i = 0; i < uflow_res->counters_num; i++)
2605 		atomic_dec(&uflow_res->counters[i]->usecnt);
2606 
2607 	kfree(uflow_res->collection);
2608 	kfree(uflow_res->counters);
2609 	kfree(uflow_res);
2610 }
2611 EXPORT_SYMBOL(ib_uverbs_flow_resources_free);
2612 
flow_resources_add(struct ib_uflow_resources * uflow_res,enum ib_flow_spec_type type,void * ibobj)2613 void flow_resources_add(struct ib_uflow_resources *uflow_res,
2614 			enum ib_flow_spec_type type,
2615 			void *ibobj)
2616 {
2617 	WARN_ON(uflow_res->num >= uflow_res->max);
2618 
2619 	switch (type) {
2620 	case IB_FLOW_SPEC_ACTION_HANDLE:
2621 		atomic_inc(&((struct ib_flow_action *)ibobj)->usecnt);
2622 		uflow_res->collection[uflow_res->collection_num++] =
2623 			(struct ib_flow_action *)ibobj;
2624 		break;
2625 	case IB_FLOW_SPEC_ACTION_COUNT:
2626 		atomic_inc(&((struct ib_counters *)ibobj)->usecnt);
2627 		uflow_res->counters[uflow_res->counters_num++] =
2628 			(struct ib_counters *)ibobj;
2629 		break;
2630 	default:
2631 		WARN_ON(1);
2632 	}
2633 
2634 	uflow_res->num++;
2635 }
2636 EXPORT_SYMBOL(flow_resources_add);
2637 
kern_spec_to_ib_spec_action(struct uverbs_attr_bundle * attrs,struct ib_uverbs_flow_spec * kern_spec,union ib_flow_spec * ib_spec,struct ib_uflow_resources * uflow_res)2638 static int kern_spec_to_ib_spec_action(struct uverbs_attr_bundle *attrs,
2639 				       struct ib_uverbs_flow_spec *kern_spec,
2640 				       union ib_flow_spec *ib_spec,
2641 				       struct ib_uflow_resources *uflow_res)
2642 {
2643 	ib_spec->type = kern_spec->type;
2644 	switch (ib_spec->type) {
2645 	case IB_FLOW_SPEC_ACTION_TAG:
2646 		if (kern_spec->flow_tag.size !=
2647 		    sizeof(struct ib_uverbs_flow_spec_action_tag))
2648 			return -EINVAL;
2649 
2650 		ib_spec->flow_tag.size = sizeof(struct ib_flow_spec_action_tag);
2651 		ib_spec->flow_tag.tag_id = kern_spec->flow_tag.tag_id;
2652 		break;
2653 	case IB_FLOW_SPEC_ACTION_DROP:
2654 		if (kern_spec->drop.size !=
2655 		    sizeof(struct ib_uverbs_flow_spec_action_drop))
2656 			return -EINVAL;
2657 
2658 		ib_spec->drop.size = sizeof(struct ib_flow_spec_action_drop);
2659 		break;
2660 	case IB_FLOW_SPEC_ACTION_HANDLE:
2661 		if (kern_spec->action.size !=
2662 		    sizeof(struct ib_uverbs_flow_spec_action_handle))
2663 			return -EOPNOTSUPP;
2664 		ib_spec->action.act = uobj_get_obj_read(flow_action,
2665 							UVERBS_OBJECT_FLOW_ACTION,
2666 							kern_spec->action.handle,
2667 							attrs);
2668 		if (!ib_spec->action.act)
2669 			return -EINVAL;
2670 		ib_spec->action.size =
2671 			sizeof(struct ib_flow_spec_action_handle);
2672 		flow_resources_add(uflow_res,
2673 				   IB_FLOW_SPEC_ACTION_HANDLE,
2674 				   ib_spec->action.act);
2675 		uobj_put_obj_read(ib_spec->action.act);
2676 		break;
2677 	case IB_FLOW_SPEC_ACTION_COUNT:
2678 		if (kern_spec->flow_count.size !=
2679 			sizeof(struct ib_uverbs_flow_spec_action_count))
2680 			return -EINVAL;
2681 		ib_spec->flow_count.counters =
2682 			uobj_get_obj_read(counters,
2683 					  UVERBS_OBJECT_COUNTERS,
2684 					  kern_spec->flow_count.handle,
2685 					  attrs);
2686 		if (!ib_spec->flow_count.counters)
2687 			return -EINVAL;
2688 		ib_spec->flow_count.size =
2689 				sizeof(struct ib_flow_spec_action_count);
2690 		flow_resources_add(uflow_res,
2691 				   IB_FLOW_SPEC_ACTION_COUNT,
2692 				   ib_spec->flow_count.counters);
2693 		uobj_put_obj_read(ib_spec->flow_count.counters);
2694 		break;
2695 	default:
2696 		return -EINVAL;
2697 	}
2698 	return 0;
2699 }
2700 
spec_filter_size(const void * kern_spec_filter,u16 kern_filter_size,u16 ib_real_filter_sz)2701 static ssize_t spec_filter_size(const void *kern_spec_filter, u16 kern_filter_size,
2702 				u16 ib_real_filter_sz)
2703 {
2704 	/*
2705 	 * User space filter structures must be 64 bit aligned, otherwise this
2706 	 * may pass, but we won't handle additional new attributes.
2707 	 */
2708 
2709 	if (kern_filter_size > ib_real_filter_sz) {
2710 		if (memchr_inv(kern_spec_filter +
2711 			       ib_real_filter_sz, 0,
2712 			       kern_filter_size - ib_real_filter_sz))
2713 			return -EINVAL;
2714 		return ib_real_filter_sz;
2715 	}
2716 	return kern_filter_size;
2717 }
2718 
ib_uverbs_kern_spec_to_ib_spec_filter(enum ib_flow_spec_type type,const void * kern_spec_mask,const void * kern_spec_val,size_t kern_filter_sz,union ib_flow_spec * ib_spec)2719 int ib_uverbs_kern_spec_to_ib_spec_filter(enum ib_flow_spec_type type,
2720 					  const void *kern_spec_mask,
2721 					  const void *kern_spec_val,
2722 					  size_t kern_filter_sz,
2723 					  union ib_flow_spec *ib_spec)
2724 {
2725 	ssize_t actual_filter_sz;
2726 	ssize_t ib_filter_sz;
2727 
2728 	/* User flow spec size must be aligned to 4 bytes */
2729 	if (kern_filter_sz != ALIGN(kern_filter_sz, 4))
2730 		return -EINVAL;
2731 
2732 	ib_spec->type = type;
2733 
2734 	if (ib_spec->type == (IB_FLOW_SPEC_INNER | IB_FLOW_SPEC_VXLAN_TUNNEL))
2735 		return -EINVAL;
2736 
2737 	switch (ib_spec->type & ~IB_FLOW_SPEC_INNER) {
2738 	case IB_FLOW_SPEC_ETH:
2739 		ib_filter_sz = offsetof(struct ib_flow_eth_filter, real_sz);
2740 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2741 						    kern_filter_sz,
2742 						    ib_filter_sz);
2743 		if (actual_filter_sz <= 0)
2744 			return -EINVAL;
2745 		ib_spec->size = sizeof(struct ib_flow_spec_eth);
2746 		memcpy(&ib_spec->eth.val, kern_spec_val, actual_filter_sz);
2747 		memcpy(&ib_spec->eth.mask, kern_spec_mask, actual_filter_sz);
2748 		break;
2749 	case IB_FLOW_SPEC_IPV4:
2750 		ib_filter_sz = offsetof(struct ib_flow_ipv4_filter, real_sz);
2751 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2752 						    kern_filter_sz,
2753 						    ib_filter_sz);
2754 		if (actual_filter_sz <= 0)
2755 			return -EINVAL;
2756 		ib_spec->size = sizeof(struct ib_flow_spec_ipv4);
2757 		memcpy(&ib_spec->ipv4.val, kern_spec_val, actual_filter_sz);
2758 		memcpy(&ib_spec->ipv4.mask, kern_spec_mask, actual_filter_sz);
2759 		break;
2760 	case IB_FLOW_SPEC_IPV6:
2761 		ib_filter_sz = offsetof(struct ib_flow_ipv6_filter, real_sz);
2762 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2763 						    kern_filter_sz,
2764 						    ib_filter_sz);
2765 		if (actual_filter_sz <= 0)
2766 			return -EINVAL;
2767 		ib_spec->size = sizeof(struct ib_flow_spec_ipv6);
2768 		memcpy(&ib_spec->ipv6.val, kern_spec_val, actual_filter_sz);
2769 		memcpy(&ib_spec->ipv6.mask, kern_spec_mask, actual_filter_sz);
2770 
2771 		if ((ntohl(ib_spec->ipv6.mask.flow_label)) >= BIT(20) ||
2772 		    (ntohl(ib_spec->ipv6.val.flow_label)) >= BIT(20))
2773 			return -EINVAL;
2774 		break;
2775 	case IB_FLOW_SPEC_TCP:
2776 	case IB_FLOW_SPEC_UDP:
2777 		ib_filter_sz = offsetof(struct ib_flow_tcp_udp_filter, real_sz);
2778 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2779 						    kern_filter_sz,
2780 						    ib_filter_sz);
2781 		if (actual_filter_sz <= 0)
2782 			return -EINVAL;
2783 		ib_spec->size = sizeof(struct ib_flow_spec_tcp_udp);
2784 		memcpy(&ib_spec->tcp_udp.val, kern_spec_val, actual_filter_sz);
2785 		memcpy(&ib_spec->tcp_udp.mask, kern_spec_mask, actual_filter_sz);
2786 		break;
2787 	case IB_FLOW_SPEC_VXLAN_TUNNEL:
2788 		ib_filter_sz = offsetof(struct ib_flow_tunnel_filter, real_sz);
2789 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2790 						    kern_filter_sz,
2791 						    ib_filter_sz);
2792 		if (actual_filter_sz <= 0)
2793 			return -EINVAL;
2794 		ib_spec->tunnel.size = sizeof(struct ib_flow_spec_tunnel);
2795 		memcpy(&ib_spec->tunnel.val, kern_spec_val, actual_filter_sz);
2796 		memcpy(&ib_spec->tunnel.mask, kern_spec_mask, actual_filter_sz);
2797 
2798 		if ((ntohl(ib_spec->tunnel.mask.tunnel_id)) >= BIT(24) ||
2799 		    (ntohl(ib_spec->tunnel.val.tunnel_id)) >= BIT(24))
2800 			return -EINVAL;
2801 		break;
2802 	case IB_FLOW_SPEC_ESP:
2803 		ib_filter_sz = offsetof(struct ib_flow_esp_filter, real_sz);
2804 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2805 						    kern_filter_sz,
2806 						    ib_filter_sz);
2807 		if (actual_filter_sz <= 0)
2808 			return -EINVAL;
2809 		ib_spec->esp.size = sizeof(struct ib_flow_spec_esp);
2810 		memcpy(&ib_spec->esp.val, kern_spec_val, actual_filter_sz);
2811 		memcpy(&ib_spec->esp.mask, kern_spec_mask, actual_filter_sz);
2812 		break;
2813 	case IB_FLOW_SPEC_GRE:
2814 		ib_filter_sz = offsetof(struct ib_flow_gre_filter, real_sz);
2815 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2816 						    kern_filter_sz,
2817 						    ib_filter_sz);
2818 		if (actual_filter_sz <= 0)
2819 			return -EINVAL;
2820 		ib_spec->gre.size = sizeof(struct ib_flow_spec_gre);
2821 		memcpy(&ib_spec->gre.val, kern_spec_val, actual_filter_sz);
2822 		memcpy(&ib_spec->gre.mask, kern_spec_mask, actual_filter_sz);
2823 		break;
2824 	case IB_FLOW_SPEC_MPLS:
2825 		ib_filter_sz = offsetof(struct ib_flow_mpls_filter, real_sz);
2826 		actual_filter_sz = spec_filter_size(kern_spec_mask,
2827 						    kern_filter_sz,
2828 						    ib_filter_sz);
2829 		if (actual_filter_sz <= 0)
2830 			return -EINVAL;
2831 		ib_spec->mpls.size = sizeof(struct ib_flow_spec_mpls);
2832 		memcpy(&ib_spec->mpls.val, kern_spec_val, actual_filter_sz);
2833 		memcpy(&ib_spec->mpls.mask, kern_spec_mask, actual_filter_sz);
2834 		break;
2835 	default:
2836 		return -EINVAL;
2837 	}
2838 	return 0;
2839 }
2840 
kern_spec_to_ib_spec_filter(struct ib_uverbs_flow_spec * kern_spec,union ib_flow_spec * ib_spec)2841 static int kern_spec_to_ib_spec_filter(struct ib_uverbs_flow_spec *kern_spec,
2842 				       union ib_flow_spec *ib_spec)
2843 {
2844 	size_t kern_filter_sz;
2845 	void *kern_spec_mask;
2846 	void *kern_spec_val;
2847 
2848 	if (check_sub_overflow((size_t)kern_spec->hdr.size,
2849 			       sizeof(struct ib_uverbs_flow_spec_hdr),
2850 			       &kern_filter_sz))
2851 		return -EINVAL;
2852 
2853 	kern_filter_sz /= 2;
2854 
2855 	kern_spec_val = (void *)kern_spec +
2856 		sizeof(struct ib_uverbs_flow_spec_hdr);
2857 	kern_spec_mask = kern_spec_val + kern_filter_sz;
2858 
2859 	return ib_uverbs_kern_spec_to_ib_spec_filter(kern_spec->type,
2860 						     kern_spec_mask,
2861 						     kern_spec_val,
2862 						     kern_filter_sz, ib_spec);
2863 }
2864 
kern_spec_to_ib_spec(struct uverbs_attr_bundle * attrs,struct ib_uverbs_flow_spec * kern_spec,union ib_flow_spec * ib_spec,struct ib_uflow_resources * uflow_res)2865 static int kern_spec_to_ib_spec(struct uverbs_attr_bundle *attrs,
2866 				struct ib_uverbs_flow_spec *kern_spec,
2867 				union ib_flow_spec *ib_spec,
2868 				struct ib_uflow_resources *uflow_res)
2869 {
2870 	if (kern_spec->reserved)
2871 		return -EINVAL;
2872 
2873 	if (kern_spec->type >= IB_FLOW_SPEC_ACTION_TAG)
2874 		return kern_spec_to_ib_spec_action(attrs, kern_spec, ib_spec,
2875 						   uflow_res);
2876 	else
2877 		return kern_spec_to_ib_spec_filter(kern_spec, ib_spec);
2878 }
2879 
ib_uverbs_ex_create_wq(struct uverbs_attr_bundle * attrs)2880 static int ib_uverbs_ex_create_wq(struct uverbs_attr_bundle *attrs)
2881 {
2882 	struct ib_uverbs_ex_create_wq cmd;
2883 	struct ib_uverbs_ex_create_wq_resp resp = {};
2884 	struct ib_uwq_object           *obj;
2885 	int err = 0;
2886 	struct ib_cq *cq;
2887 	struct ib_pd *pd;
2888 	struct ib_wq *wq;
2889 	struct ib_wq_init_attr wq_init_attr = {};
2890 	struct ib_device *ib_dev;
2891 
2892 	err = uverbs_request(attrs, &cmd, sizeof(cmd));
2893 	if (err)
2894 		return err;
2895 
2896 	if (cmd.comp_mask)
2897 		return -EOPNOTSUPP;
2898 
2899 	obj = (struct ib_uwq_object *)uobj_alloc(UVERBS_OBJECT_WQ, attrs,
2900 						 &ib_dev);
2901 	if (IS_ERR(obj))
2902 		return PTR_ERR(obj);
2903 
2904 	pd = uobj_get_obj_read(pd, UVERBS_OBJECT_PD, cmd.pd_handle, attrs);
2905 	if (!pd) {
2906 		err = -EINVAL;
2907 		goto err_uobj;
2908 	}
2909 
2910 	cq = uobj_get_obj_read(cq, UVERBS_OBJECT_CQ, cmd.cq_handle, attrs);
2911 	if (!cq) {
2912 		err = -EINVAL;
2913 		goto err_put_pd;
2914 	}
2915 
2916 	wq_init_attr.cq = cq;
2917 	wq_init_attr.max_sge = cmd.max_sge;
2918 	wq_init_attr.max_wr = cmd.max_wr;
2919 	wq_init_attr.wq_type = cmd.wq_type;
2920 	wq_init_attr.event_handler = ib_uverbs_wq_event_handler;
2921 	wq_init_attr.create_flags = cmd.create_flags;
2922 	INIT_LIST_HEAD(&obj->uevent.event_list);
2923 	obj->uevent.uobject.user_handle = cmd.user_handle;
2924 
2925 	wq = pd->device->ops.create_wq(pd, &wq_init_attr, &attrs->driver_udata);
2926 	if (IS_ERR(wq)) {
2927 		err = PTR_ERR(wq);
2928 		goto err_put_cq;
2929 	}
2930 
2931 	wq->uobject = obj;
2932 	obj->uevent.uobject.object = wq;
2933 	wq->wq_type = wq_init_attr.wq_type;
2934 	wq->cq = cq;
2935 	wq->pd = pd;
2936 	wq->device = pd->device;
2937 	atomic_set(&wq->usecnt, 0);
2938 	atomic_inc(&pd->usecnt);
2939 	atomic_inc(&cq->usecnt);
2940 	obj->uevent.event_file = READ_ONCE(attrs->ufile->default_async_file);
2941 	if (obj->uevent.event_file)
2942 		uverbs_uobject_get(&obj->uevent.event_file->uobj);
2943 
2944 	uobj_put_obj_read(pd);
2945 	rdma_lookup_put_uobject(&cq->uobject->uevent.uobject,
2946 				UVERBS_LOOKUP_READ);
2947 	uobj_finalize_uobj_create(&obj->uevent.uobject, attrs);
2948 
2949 	resp.wq_handle = obj->uevent.uobject.id;
2950 	resp.max_sge = wq_init_attr.max_sge;
2951 	resp.max_wr = wq_init_attr.max_wr;
2952 	resp.wqn = wq->wq_num;
2953 	resp.response_length = uverbs_response_length(attrs, sizeof(resp));
2954 	return uverbs_response(attrs, &resp, sizeof(resp));
2955 
2956 err_put_cq:
2957 	rdma_lookup_put_uobject(&cq->uobject->uevent.uobject,
2958 				UVERBS_LOOKUP_READ);
2959 err_put_pd:
2960 	uobj_put_obj_read(pd);
2961 err_uobj:
2962 	uobj_alloc_abort(&obj->uevent.uobject, attrs);
2963 
2964 	return err;
2965 }
2966 
ib_uverbs_ex_destroy_wq(struct uverbs_attr_bundle * attrs)2967 static int ib_uverbs_ex_destroy_wq(struct uverbs_attr_bundle *attrs)
2968 {
2969 	struct ib_uverbs_ex_destroy_wq	cmd;
2970 	struct ib_uverbs_ex_destroy_wq_resp	resp = {};
2971 	struct ib_uobject		*uobj;
2972 	struct ib_uwq_object		*obj;
2973 	int				ret;
2974 
2975 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
2976 	if (ret)
2977 		return ret;
2978 
2979 	if (cmd.comp_mask)
2980 		return -EOPNOTSUPP;
2981 
2982 	resp.response_length = uverbs_response_length(attrs, sizeof(resp));
2983 	uobj = uobj_get_destroy(UVERBS_OBJECT_WQ, cmd.wq_handle, attrs);
2984 	if (IS_ERR(uobj))
2985 		return PTR_ERR(uobj);
2986 
2987 	obj = container_of(uobj, struct ib_uwq_object, uevent.uobject);
2988 	resp.events_reported = obj->uevent.events_reported;
2989 
2990 	uobj_put_destroy(uobj);
2991 
2992 	return uverbs_response(attrs, &resp, sizeof(resp));
2993 }
2994 
ib_uverbs_ex_modify_wq(struct uverbs_attr_bundle * attrs)2995 static int ib_uverbs_ex_modify_wq(struct uverbs_attr_bundle *attrs)
2996 {
2997 	struct ib_uverbs_ex_modify_wq cmd;
2998 	struct ib_wq *wq;
2999 	struct ib_wq_attr wq_attr = {};
3000 	int ret;
3001 
3002 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3003 	if (ret)
3004 		return ret;
3005 
3006 	if (!cmd.attr_mask)
3007 		return -EINVAL;
3008 
3009 	if (cmd.attr_mask > (IB_WQ_STATE | IB_WQ_CUR_STATE | IB_WQ_FLAGS))
3010 		return -EINVAL;
3011 
3012 	wq = uobj_get_obj_read(wq, UVERBS_OBJECT_WQ, cmd.wq_handle, attrs);
3013 	if (!wq)
3014 		return -EINVAL;
3015 
3016 	if (cmd.attr_mask & IB_WQ_FLAGS) {
3017 		wq_attr.flags = cmd.flags;
3018 		wq_attr.flags_mask = cmd.flags_mask;
3019 	}
3020 
3021 	if (cmd.attr_mask & IB_WQ_CUR_STATE) {
3022 		if (cmd.curr_wq_state > IB_WQS_ERR)
3023 			return -EINVAL;
3024 
3025 		wq_attr.curr_wq_state = cmd.curr_wq_state;
3026 	} else {
3027 		wq_attr.curr_wq_state = wq->state;
3028 	}
3029 
3030 	if (cmd.attr_mask & IB_WQ_STATE) {
3031 		if (cmd.wq_state > IB_WQS_ERR)
3032 			return -EINVAL;
3033 
3034 		wq_attr.wq_state = cmd.wq_state;
3035 	} else {
3036 		wq_attr.wq_state = wq_attr.curr_wq_state;
3037 	}
3038 
3039 	ret = wq->device->ops.modify_wq(wq, &wq_attr, cmd.attr_mask,
3040 					&attrs->driver_udata);
3041 	rdma_lookup_put_uobject(&wq->uobject->uevent.uobject,
3042 				UVERBS_LOOKUP_READ);
3043 	return ret;
3044 }
3045 
ib_uverbs_ex_create_rwq_ind_table(struct uverbs_attr_bundle * attrs)3046 static int ib_uverbs_ex_create_rwq_ind_table(struct uverbs_attr_bundle *attrs)
3047 {
3048 	struct ib_uverbs_ex_create_rwq_ind_table cmd;
3049 	struct ib_uverbs_ex_create_rwq_ind_table_resp  resp = {};
3050 	struct ib_uobject *uobj;
3051 	int err;
3052 	struct ib_rwq_ind_table_init_attr init_attr = {};
3053 	struct ib_rwq_ind_table *rwq_ind_tbl;
3054 	struct ib_wq **wqs = NULL;
3055 	u32 *wqs_handles = NULL;
3056 	struct ib_wq	*wq = NULL;
3057 	int i, num_read_wqs;
3058 	u32 num_wq_handles;
3059 	struct uverbs_req_iter iter;
3060 	struct ib_device *ib_dev;
3061 
3062 	err = uverbs_request_start(attrs, &iter, &cmd, sizeof(cmd));
3063 	if (err)
3064 		return err;
3065 
3066 	if (cmd.comp_mask)
3067 		return -EOPNOTSUPP;
3068 
3069 	if (cmd.log_ind_tbl_size > IB_USER_VERBS_MAX_LOG_IND_TBL_SIZE)
3070 		return -EINVAL;
3071 
3072 	num_wq_handles = 1 << cmd.log_ind_tbl_size;
3073 	wqs_handles = kcalloc(num_wq_handles, sizeof(*wqs_handles),
3074 			      GFP_KERNEL);
3075 	if (!wqs_handles)
3076 		return -ENOMEM;
3077 
3078 	err = uverbs_request_next(&iter, wqs_handles,
3079 				  num_wq_handles * sizeof(__u32));
3080 	if (err)
3081 		goto err_free;
3082 
3083 	err = uverbs_request_finish(&iter);
3084 	if (err)
3085 		goto err_free;
3086 
3087 	wqs = kcalloc(num_wq_handles, sizeof(*wqs), GFP_KERNEL);
3088 	if (!wqs) {
3089 		err = -ENOMEM;
3090 		goto  err_free;
3091 	}
3092 
3093 	for (num_read_wqs = 0; num_read_wqs < num_wq_handles;
3094 			num_read_wqs++) {
3095 		wq = uobj_get_obj_read(wq, UVERBS_OBJECT_WQ,
3096 				       wqs_handles[num_read_wqs], attrs);
3097 		if (!wq) {
3098 			err = -EINVAL;
3099 			goto put_wqs;
3100 		}
3101 
3102 		wqs[num_read_wqs] = wq;
3103 		atomic_inc(&wqs[num_read_wqs]->usecnt);
3104 	}
3105 
3106 	uobj = uobj_alloc(UVERBS_OBJECT_RWQ_IND_TBL, attrs, &ib_dev);
3107 	if (IS_ERR(uobj)) {
3108 		err = PTR_ERR(uobj);
3109 		goto put_wqs;
3110 	}
3111 
3112 	rwq_ind_tbl = rdma_zalloc_drv_obj(ib_dev, ib_rwq_ind_table);
3113 	if (!rwq_ind_tbl) {
3114 		err = -ENOMEM;
3115 		goto err_uobj;
3116 	}
3117 
3118 	init_attr.log_ind_tbl_size = cmd.log_ind_tbl_size;
3119 	init_attr.ind_tbl = wqs;
3120 
3121 	rwq_ind_tbl->ind_tbl = wqs;
3122 	rwq_ind_tbl->log_ind_tbl_size = init_attr.log_ind_tbl_size;
3123 	rwq_ind_tbl->uobject = uobj;
3124 	uobj->object = rwq_ind_tbl;
3125 	rwq_ind_tbl->device = ib_dev;
3126 	atomic_set(&rwq_ind_tbl->usecnt, 0);
3127 
3128 	err = ib_dev->ops.create_rwq_ind_table(rwq_ind_tbl, &init_attr,
3129 					       &attrs->driver_udata);
3130 	if (err)
3131 		goto err_create;
3132 
3133 	for (i = 0; i < num_wq_handles; i++)
3134 		rdma_lookup_put_uobject(&wqs[i]->uobject->uevent.uobject,
3135 					UVERBS_LOOKUP_READ);
3136 	kfree(wqs_handles);
3137 	uobj_finalize_uobj_create(uobj, attrs);
3138 
3139 	resp.ind_tbl_handle = uobj->id;
3140 	resp.ind_tbl_num = rwq_ind_tbl->ind_tbl_num;
3141 	resp.response_length = uverbs_response_length(attrs, sizeof(resp));
3142 	return uverbs_response(attrs, &resp, sizeof(resp));
3143 
3144 err_create:
3145 	kfree(rwq_ind_tbl);
3146 err_uobj:
3147 	uobj_alloc_abort(uobj, attrs);
3148 put_wqs:
3149 	for (i = 0; i < num_read_wqs; i++) {
3150 		rdma_lookup_put_uobject(&wqs[i]->uobject->uevent.uobject,
3151 					UVERBS_LOOKUP_READ);
3152 		atomic_dec(&wqs[i]->usecnt);
3153 	}
3154 err_free:
3155 	kfree(wqs_handles);
3156 	kfree(wqs);
3157 	return err;
3158 }
3159 
ib_uverbs_ex_destroy_rwq_ind_table(struct uverbs_attr_bundle * attrs)3160 static int ib_uverbs_ex_destroy_rwq_ind_table(struct uverbs_attr_bundle *attrs)
3161 {
3162 	struct ib_uverbs_ex_destroy_rwq_ind_table cmd;
3163 	int ret;
3164 
3165 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3166 	if (ret)
3167 		return ret;
3168 
3169 	if (cmd.comp_mask)
3170 		return -EOPNOTSUPP;
3171 
3172 	return uobj_perform_destroy(UVERBS_OBJECT_RWQ_IND_TBL,
3173 				    cmd.ind_tbl_handle, attrs);
3174 }
3175 
ib_uverbs_ex_create_flow(struct uverbs_attr_bundle * attrs)3176 static int ib_uverbs_ex_create_flow(struct uverbs_attr_bundle *attrs)
3177 {
3178 	struct ib_uverbs_create_flow	  cmd;
3179 	struct ib_uverbs_create_flow_resp resp = {};
3180 	struct ib_uobject		  *uobj;
3181 	struct ib_flow			  *flow_id;
3182 	struct ib_uverbs_flow_attr	  *kern_flow_attr;
3183 	struct ib_flow_attr		  *flow_attr;
3184 	struct ib_qp			  *qp;
3185 	struct ib_uflow_resources	  *uflow_res;
3186 	struct ib_uverbs_flow_spec_hdr	  *kern_spec;
3187 	struct uverbs_req_iter iter;
3188 	int err;
3189 	void *ib_spec;
3190 	int i;
3191 	struct ib_device *ib_dev;
3192 
3193 	err = uverbs_request_start(attrs, &iter, &cmd, sizeof(cmd));
3194 	if (err)
3195 		return err;
3196 
3197 	if (cmd.comp_mask)
3198 		return -EINVAL;
3199 
3200 	if (!capable(CAP_NET_RAW))
3201 		return -EPERM;
3202 
3203 	if (cmd.flow_attr.flags >= IB_FLOW_ATTR_FLAGS_RESERVED)
3204 		return -EINVAL;
3205 
3206 	if ((cmd.flow_attr.flags & IB_FLOW_ATTR_FLAGS_DONT_TRAP) &&
3207 	    ((cmd.flow_attr.type == IB_FLOW_ATTR_ALL_DEFAULT) ||
3208 	     (cmd.flow_attr.type == IB_FLOW_ATTR_MC_DEFAULT)))
3209 		return -EINVAL;
3210 
3211 	if (cmd.flow_attr.num_of_specs > IB_FLOW_SPEC_SUPPORT_LAYERS)
3212 		return -EINVAL;
3213 
3214 	if (cmd.flow_attr.size >
3215 	    (cmd.flow_attr.num_of_specs * sizeof(struct ib_uverbs_flow_spec)))
3216 		return -EINVAL;
3217 
3218 	if (cmd.flow_attr.reserved[0] ||
3219 	    cmd.flow_attr.reserved[1])
3220 		return -EINVAL;
3221 
3222 	if (cmd.flow_attr.num_of_specs) {
3223 		kern_flow_attr = kmalloc(sizeof(*kern_flow_attr) + cmd.flow_attr.size,
3224 					 GFP_KERNEL);
3225 		if (!kern_flow_attr)
3226 			return -ENOMEM;
3227 
3228 		*kern_flow_attr = cmd.flow_attr;
3229 		err = uverbs_request_next(&iter, &kern_flow_attr->flow_specs,
3230 					  cmd.flow_attr.size);
3231 		if (err)
3232 			goto err_free_attr;
3233 	} else {
3234 		kern_flow_attr = &cmd.flow_attr;
3235 	}
3236 
3237 	err = uverbs_request_finish(&iter);
3238 	if (err)
3239 		goto err_free_attr;
3240 
3241 	uobj = uobj_alloc(UVERBS_OBJECT_FLOW, attrs, &ib_dev);
3242 	if (IS_ERR(uobj)) {
3243 		err = PTR_ERR(uobj);
3244 		goto err_free_attr;
3245 	}
3246 
3247 	if (!rdma_is_port_valid(uobj->context->device, cmd.flow_attr.port)) {
3248 		err = -EINVAL;
3249 		goto err_uobj;
3250 	}
3251 
3252 	qp = uobj_get_obj_read(qp, UVERBS_OBJECT_QP, cmd.qp_handle, attrs);
3253 	if (!qp) {
3254 		err = -EINVAL;
3255 		goto err_uobj;
3256 	}
3257 
3258 	if (qp->qp_type != IB_QPT_UD && qp->qp_type != IB_QPT_RAW_PACKET) {
3259 		err = -EINVAL;
3260 		goto err_put;
3261 	}
3262 
3263 	flow_attr = kzalloc(struct_size(flow_attr, flows,
3264 				cmd.flow_attr.num_of_specs), GFP_KERNEL);
3265 	if (!flow_attr) {
3266 		err = -ENOMEM;
3267 		goto err_put;
3268 	}
3269 	uflow_res = flow_resources_alloc(cmd.flow_attr.num_of_specs);
3270 	if (!uflow_res) {
3271 		err = -ENOMEM;
3272 		goto err_free_flow_attr;
3273 	}
3274 
3275 	flow_attr->type = kern_flow_attr->type;
3276 	flow_attr->priority = kern_flow_attr->priority;
3277 	flow_attr->num_of_specs = kern_flow_attr->num_of_specs;
3278 	flow_attr->port = kern_flow_attr->port;
3279 	flow_attr->flags = kern_flow_attr->flags;
3280 	flow_attr->size = sizeof(*flow_attr);
3281 
3282 	kern_spec = kern_flow_attr->flow_specs;
3283 	ib_spec = flow_attr + 1;
3284 	for (i = 0; i < flow_attr->num_of_specs &&
3285 			cmd.flow_attr.size >= sizeof(*kern_spec) &&
3286 			cmd.flow_attr.size >= kern_spec->size;
3287 	     i++) {
3288 		err = kern_spec_to_ib_spec(
3289 				attrs, (struct ib_uverbs_flow_spec *)kern_spec,
3290 				ib_spec, uflow_res);
3291 		if (err)
3292 			goto err_free;
3293 
3294 		flow_attr->size +=
3295 			((union ib_flow_spec *) ib_spec)->size;
3296 		cmd.flow_attr.size -= kern_spec->size;
3297 		kern_spec = ((void *)kern_spec) + kern_spec->size;
3298 		ib_spec += ((union ib_flow_spec *) ib_spec)->size;
3299 	}
3300 	if (cmd.flow_attr.size || (i != flow_attr->num_of_specs)) {
3301 		pr_warn("create flow failed, flow %d: %u bytes left from uverb cmd\n",
3302 			i, cmd.flow_attr.size);
3303 		err = -EINVAL;
3304 		goto err_free;
3305 	}
3306 
3307 	flow_id = qp->device->ops.create_flow(qp, flow_attr,
3308 					      &attrs->driver_udata);
3309 
3310 	if (IS_ERR(flow_id)) {
3311 		err = PTR_ERR(flow_id);
3312 		goto err_free;
3313 	}
3314 
3315 	ib_set_flow(uobj, flow_id, qp, qp->device, uflow_res);
3316 
3317 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
3318 				UVERBS_LOOKUP_READ);
3319 	kfree(flow_attr);
3320 
3321 	if (cmd.flow_attr.num_of_specs)
3322 		kfree(kern_flow_attr);
3323 	uobj_finalize_uobj_create(uobj, attrs);
3324 
3325 	resp.flow_handle = uobj->id;
3326 	return uverbs_response(attrs, &resp, sizeof(resp));
3327 
3328 err_free:
3329 	ib_uverbs_flow_resources_free(uflow_res);
3330 err_free_flow_attr:
3331 	kfree(flow_attr);
3332 err_put:
3333 	rdma_lookup_put_uobject(&qp->uobject->uevent.uobject,
3334 				UVERBS_LOOKUP_READ);
3335 err_uobj:
3336 	uobj_alloc_abort(uobj, attrs);
3337 err_free_attr:
3338 	if (cmd.flow_attr.num_of_specs)
3339 		kfree(kern_flow_attr);
3340 	return err;
3341 }
3342 
ib_uverbs_ex_destroy_flow(struct uverbs_attr_bundle * attrs)3343 static int ib_uverbs_ex_destroy_flow(struct uverbs_attr_bundle *attrs)
3344 {
3345 	struct ib_uverbs_destroy_flow	cmd;
3346 	int				ret;
3347 
3348 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3349 	if (ret)
3350 		return ret;
3351 
3352 	if (cmd.comp_mask)
3353 		return -EINVAL;
3354 
3355 	return uobj_perform_destroy(UVERBS_OBJECT_FLOW, cmd.flow_handle, attrs);
3356 }
3357 
__uverbs_create_xsrq(struct uverbs_attr_bundle * attrs,struct ib_uverbs_create_xsrq * cmd,struct ib_udata * udata)3358 static int __uverbs_create_xsrq(struct uverbs_attr_bundle *attrs,
3359 				struct ib_uverbs_create_xsrq *cmd,
3360 				struct ib_udata *udata)
3361 {
3362 	struct ib_uverbs_create_srq_resp resp = {};
3363 	struct ib_usrq_object           *obj;
3364 	struct ib_pd                    *pd;
3365 	struct ib_srq                   *srq;
3366 	struct ib_srq_init_attr          attr;
3367 	int ret;
3368 	struct ib_uobject *xrcd_uobj;
3369 	struct ib_device *ib_dev;
3370 
3371 	obj = (struct ib_usrq_object *)uobj_alloc(UVERBS_OBJECT_SRQ, attrs,
3372 						  &ib_dev);
3373 	if (IS_ERR(obj))
3374 		return PTR_ERR(obj);
3375 
3376 	if (cmd->srq_type == IB_SRQT_TM)
3377 		attr.ext.tag_matching.max_num_tags = cmd->max_num_tags;
3378 
3379 	if (cmd->srq_type == IB_SRQT_XRC) {
3380 		xrcd_uobj = uobj_get_read(UVERBS_OBJECT_XRCD, cmd->xrcd_handle,
3381 					  attrs);
3382 		if (IS_ERR(xrcd_uobj)) {
3383 			ret = -EINVAL;
3384 			goto err;
3385 		}
3386 
3387 		attr.ext.xrc.xrcd = (struct ib_xrcd *)xrcd_uobj->object;
3388 		if (!attr.ext.xrc.xrcd) {
3389 			ret = -EINVAL;
3390 			goto err_put_xrcd;
3391 		}
3392 
3393 		obj->uxrcd = container_of(xrcd_uobj, struct ib_uxrcd_object, uobject);
3394 		atomic_inc(&obj->uxrcd->refcnt);
3395 	}
3396 
3397 	if (ib_srq_has_cq(cmd->srq_type)) {
3398 		attr.ext.cq = uobj_get_obj_read(cq, UVERBS_OBJECT_CQ,
3399 						cmd->cq_handle, attrs);
3400 		if (!attr.ext.cq) {
3401 			ret = -EINVAL;
3402 			goto err_put_xrcd;
3403 		}
3404 	}
3405 
3406 	pd = uobj_get_obj_read(pd, UVERBS_OBJECT_PD, cmd->pd_handle, attrs);
3407 	if (!pd) {
3408 		ret = -EINVAL;
3409 		goto err_put_cq;
3410 	}
3411 
3412 	attr.event_handler  = ib_uverbs_srq_event_handler;
3413 	attr.srq_type       = cmd->srq_type;
3414 	attr.attr.max_wr    = cmd->max_wr;
3415 	attr.attr.max_sge   = cmd->max_sge;
3416 	attr.attr.srq_limit = cmd->srq_limit;
3417 
3418 	INIT_LIST_HEAD(&obj->uevent.event_list);
3419 	obj->uevent.uobject.user_handle = cmd->user_handle;
3420 
3421 	srq = ib_create_srq_user(pd, &attr, obj, udata);
3422 	if (IS_ERR(srq)) {
3423 		ret = PTR_ERR(srq);
3424 		goto err_put_pd;
3425 	}
3426 
3427 	obj->uevent.uobject.object = srq;
3428 	obj->uevent.uobject.user_handle = cmd->user_handle;
3429 	obj->uevent.event_file = READ_ONCE(attrs->ufile->default_async_file);
3430 	if (obj->uevent.event_file)
3431 		uverbs_uobject_get(&obj->uevent.event_file->uobj);
3432 
3433 	if (cmd->srq_type == IB_SRQT_XRC)
3434 		resp.srqn = srq->ext.xrc.srq_num;
3435 
3436 	if (cmd->srq_type == IB_SRQT_XRC)
3437 		uobj_put_read(xrcd_uobj);
3438 
3439 	if (ib_srq_has_cq(cmd->srq_type))
3440 		rdma_lookup_put_uobject(&attr.ext.cq->uobject->uevent.uobject,
3441 					UVERBS_LOOKUP_READ);
3442 
3443 	uobj_put_obj_read(pd);
3444 	uobj_finalize_uobj_create(&obj->uevent.uobject, attrs);
3445 
3446 	resp.srq_handle = obj->uevent.uobject.id;
3447 	resp.max_wr = attr.attr.max_wr;
3448 	resp.max_sge = attr.attr.max_sge;
3449 	return uverbs_response(attrs, &resp, sizeof(resp));
3450 
3451 err_put_pd:
3452 	uobj_put_obj_read(pd);
3453 err_put_cq:
3454 	if (ib_srq_has_cq(cmd->srq_type))
3455 		rdma_lookup_put_uobject(&attr.ext.cq->uobject->uevent.uobject,
3456 					UVERBS_LOOKUP_READ);
3457 
3458 err_put_xrcd:
3459 	if (cmd->srq_type == IB_SRQT_XRC) {
3460 		atomic_dec(&obj->uxrcd->refcnt);
3461 		uobj_put_read(xrcd_uobj);
3462 	}
3463 
3464 err:
3465 	uobj_alloc_abort(&obj->uevent.uobject, attrs);
3466 	return ret;
3467 }
3468 
ib_uverbs_create_srq(struct uverbs_attr_bundle * attrs)3469 static int ib_uverbs_create_srq(struct uverbs_attr_bundle *attrs)
3470 {
3471 	struct ib_uverbs_create_srq      cmd;
3472 	struct ib_uverbs_create_xsrq     xcmd;
3473 	int ret;
3474 
3475 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3476 	if (ret)
3477 		return ret;
3478 
3479 	memset(&xcmd, 0, sizeof(xcmd));
3480 	xcmd.response	 = cmd.response;
3481 	xcmd.user_handle = cmd.user_handle;
3482 	xcmd.srq_type	 = IB_SRQT_BASIC;
3483 	xcmd.pd_handle	 = cmd.pd_handle;
3484 	xcmd.max_wr	 = cmd.max_wr;
3485 	xcmd.max_sge	 = cmd.max_sge;
3486 	xcmd.srq_limit	 = cmd.srq_limit;
3487 
3488 	return __uverbs_create_xsrq(attrs, &xcmd, &attrs->driver_udata);
3489 }
3490 
ib_uverbs_create_xsrq(struct uverbs_attr_bundle * attrs)3491 static int ib_uverbs_create_xsrq(struct uverbs_attr_bundle *attrs)
3492 {
3493 	struct ib_uverbs_create_xsrq     cmd;
3494 	int ret;
3495 
3496 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3497 	if (ret)
3498 		return ret;
3499 
3500 	return __uverbs_create_xsrq(attrs, &cmd, &attrs->driver_udata);
3501 }
3502 
ib_uverbs_modify_srq(struct uverbs_attr_bundle * attrs)3503 static int ib_uverbs_modify_srq(struct uverbs_attr_bundle *attrs)
3504 {
3505 	struct ib_uverbs_modify_srq cmd;
3506 	struct ib_srq              *srq;
3507 	struct ib_srq_attr          attr;
3508 	int                         ret;
3509 
3510 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3511 	if (ret)
3512 		return ret;
3513 
3514 	srq = uobj_get_obj_read(srq, UVERBS_OBJECT_SRQ, cmd.srq_handle, attrs);
3515 	if (!srq)
3516 		return -EINVAL;
3517 
3518 	attr.max_wr    = cmd.max_wr;
3519 	attr.srq_limit = cmd.srq_limit;
3520 
3521 	ret = srq->device->ops.modify_srq(srq, &attr, cmd.attr_mask,
3522 					  &attrs->driver_udata);
3523 
3524 	rdma_lookup_put_uobject(&srq->uobject->uevent.uobject,
3525 				UVERBS_LOOKUP_READ);
3526 
3527 	return ret;
3528 }
3529 
ib_uverbs_query_srq(struct uverbs_attr_bundle * attrs)3530 static int ib_uverbs_query_srq(struct uverbs_attr_bundle *attrs)
3531 {
3532 	struct ib_uverbs_query_srq      cmd;
3533 	struct ib_uverbs_query_srq_resp resp;
3534 	struct ib_srq_attr              attr;
3535 	struct ib_srq                   *srq;
3536 	int                             ret;
3537 
3538 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3539 	if (ret)
3540 		return ret;
3541 
3542 	srq = uobj_get_obj_read(srq, UVERBS_OBJECT_SRQ, cmd.srq_handle, attrs);
3543 	if (!srq)
3544 		return -EINVAL;
3545 
3546 	ret = ib_query_srq(srq, &attr);
3547 
3548 	rdma_lookup_put_uobject(&srq->uobject->uevent.uobject,
3549 				UVERBS_LOOKUP_READ);
3550 
3551 	if (ret)
3552 		return ret;
3553 
3554 	memset(&resp, 0, sizeof resp);
3555 
3556 	resp.max_wr    = attr.max_wr;
3557 	resp.max_sge   = attr.max_sge;
3558 	resp.srq_limit = attr.srq_limit;
3559 
3560 	return uverbs_response(attrs, &resp, sizeof(resp));
3561 }
3562 
ib_uverbs_destroy_srq(struct uverbs_attr_bundle * attrs)3563 static int ib_uverbs_destroy_srq(struct uverbs_attr_bundle *attrs)
3564 {
3565 	struct ib_uverbs_destroy_srq      cmd;
3566 	struct ib_uverbs_destroy_srq_resp resp;
3567 	struct ib_uobject		 *uobj;
3568 	struct ib_uevent_object        	 *obj;
3569 	int ret;
3570 
3571 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3572 	if (ret)
3573 		return ret;
3574 
3575 	uobj = uobj_get_destroy(UVERBS_OBJECT_SRQ, cmd.srq_handle, attrs);
3576 	if (IS_ERR(uobj))
3577 		return PTR_ERR(uobj);
3578 
3579 	obj = container_of(uobj, struct ib_uevent_object, uobject);
3580 	memset(&resp, 0, sizeof(resp));
3581 	resp.events_reported = obj->events_reported;
3582 
3583 	uobj_put_destroy(uobj);
3584 
3585 	return uverbs_response(attrs, &resp, sizeof(resp));
3586 }
3587 
ib_uverbs_ex_query_device(struct uverbs_attr_bundle * attrs)3588 static int ib_uverbs_ex_query_device(struct uverbs_attr_bundle *attrs)
3589 {
3590 	struct ib_uverbs_ex_query_device_resp resp = {};
3591 	struct ib_uverbs_ex_query_device  cmd;
3592 	struct ib_device_attr attr = {0};
3593 	struct ib_ucontext *ucontext;
3594 	struct ib_device *ib_dev;
3595 	int err;
3596 
3597 	ucontext = ib_uverbs_get_ucontext(attrs);
3598 	if (IS_ERR(ucontext))
3599 		return PTR_ERR(ucontext);
3600 	ib_dev = ucontext->device;
3601 
3602 	err = uverbs_request(attrs, &cmd, sizeof(cmd));
3603 	if (err)
3604 		return err;
3605 
3606 	if (cmd.comp_mask)
3607 		return -EINVAL;
3608 
3609 	if (cmd.reserved)
3610 		return -EINVAL;
3611 
3612 	err = ib_dev->ops.query_device(ib_dev, &attr, &attrs->driver_udata);
3613 	if (err)
3614 		return err;
3615 
3616 	copy_query_dev_fields(ucontext, &resp.base, &attr);
3617 
3618 	resp.odp_caps.general_caps = attr.odp_caps.general_caps;
3619 	resp.odp_caps.per_transport_caps.rc_odp_caps =
3620 		attr.odp_caps.per_transport_caps.rc_odp_caps;
3621 	resp.odp_caps.per_transport_caps.uc_odp_caps =
3622 		attr.odp_caps.per_transport_caps.uc_odp_caps;
3623 	resp.odp_caps.per_transport_caps.ud_odp_caps =
3624 		attr.odp_caps.per_transport_caps.ud_odp_caps;
3625 	resp.xrc_odp_caps = attr.odp_caps.per_transport_caps.xrc_odp_caps;
3626 
3627 	resp.timestamp_mask = attr.timestamp_mask;
3628 	resp.hca_core_clock = attr.hca_core_clock;
3629 	resp.device_cap_flags_ex = attr.device_cap_flags;
3630 	resp.rss_caps.supported_qpts = attr.rss_caps.supported_qpts;
3631 	resp.rss_caps.max_rwq_indirection_tables =
3632 		attr.rss_caps.max_rwq_indirection_tables;
3633 	resp.rss_caps.max_rwq_indirection_table_size =
3634 		attr.rss_caps.max_rwq_indirection_table_size;
3635 	resp.max_wq_type_rq = attr.max_wq_type_rq;
3636 	resp.raw_packet_caps = attr.raw_packet_caps;
3637 	resp.tm_caps.max_rndv_hdr_size	= attr.tm_caps.max_rndv_hdr_size;
3638 	resp.tm_caps.max_num_tags	= attr.tm_caps.max_num_tags;
3639 	resp.tm_caps.max_ops		= attr.tm_caps.max_ops;
3640 	resp.tm_caps.max_sge		= attr.tm_caps.max_sge;
3641 	resp.tm_caps.flags		= attr.tm_caps.flags;
3642 	resp.cq_moderation_caps.max_cq_moderation_count  =
3643 		attr.cq_caps.max_cq_moderation_count;
3644 	resp.cq_moderation_caps.max_cq_moderation_period =
3645 		attr.cq_caps.max_cq_moderation_period;
3646 	resp.max_dm_size = attr.max_dm_size;
3647 	resp.response_length = uverbs_response_length(attrs, sizeof(resp));
3648 
3649 	return uverbs_response(attrs, &resp, sizeof(resp));
3650 }
3651 
ib_uverbs_ex_modify_cq(struct uverbs_attr_bundle * attrs)3652 static int ib_uverbs_ex_modify_cq(struct uverbs_attr_bundle *attrs)
3653 {
3654 	struct ib_uverbs_ex_modify_cq cmd;
3655 	struct ib_cq *cq;
3656 	int ret;
3657 
3658 	ret = uverbs_request(attrs, &cmd, sizeof(cmd));
3659 	if (ret)
3660 		return ret;
3661 
3662 	if (!cmd.attr_mask || cmd.reserved)
3663 		return -EINVAL;
3664 
3665 	if (cmd.attr_mask > IB_CQ_MODERATE)
3666 		return -EOPNOTSUPP;
3667 
3668 	cq = uobj_get_obj_read(cq, UVERBS_OBJECT_CQ, cmd.cq_handle, attrs);
3669 	if (!cq)
3670 		return -EINVAL;
3671 
3672 	ret = rdma_set_cq_moderation(cq, cmd.attr.cq_count, cmd.attr.cq_period);
3673 
3674 	rdma_lookup_put_uobject(&cq->uobject->uevent.uobject,
3675 				UVERBS_LOOKUP_READ);
3676 	return ret;
3677 }
3678 
3679 /*
3680  * Describe the input structs for write(). Some write methods have an input
3681  * only struct, most have an input and output. If the struct has an output then
3682  * the 'response' u64 must be the first field in the request structure.
3683  *
3684  * If udata is present then both the request and response structs have a
3685  * trailing driver_data flex array. In this case the size of the base struct
3686  * cannot be changed.
3687  */
3688 #define UAPI_DEF_WRITE_IO(req, resp)                                           \
3689 	.write.has_resp = 1 +                                                  \
3690 			  BUILD_BUG_ON_ZERO(offsetof(req, response) != 0) +    \
3691 			  BUILD_BUG_ON_ZERO(sizeof_field(req, response) !=    \
3692 					    sizeof(u64)),                      \
3693 	.write.req_size = sizeof(req), .write.resp_size = sizeof(resp)
3694 
3695 #define UAPI_DEF_WRITE_I(req) .write.req_size = sizeof(req)
3696 
3697 #define UAPI_DEF_WRITE_UDATA_IO(req, resp)                                     \
3698 	UAPI_DEF_WRITE_IO(req, resp),                                          \
3699 		.write.has_udata =                                             \
3700 			1 +                                                    \
3701 			BUILD_BUG_ON_ZERO(offsetof(req, driver_data) !=        \
3702 					  sizeof(req)) +                       \
3703 			BUILD_BUG_ON_ZERO(offsetof(resp, driver_data) !=       \
3704 					  sizeof(resp))
3705 
3706 #define UAPI_DEF_WRITE_UDATA_I(req)                                            \
3707 	UAPI_DEF_WRITE_I(req),                                                 \
3708 		.write.has_udata =                                             \
3709 			1 + BUILD_BUG_ON_ZERO(offsetof(req, driver_data) !=    \
3710 					      sizeof(req))
3711 
3712 /*
3713  * The _EX versions are for use with WRITE_EX and allow the last struct member
3714  * to be specified. Buffers that do not include that member will be rejected.
3715  */
3716 #define UAPI_DEF_WRITE_IO_EX(req, req_last_member, resp, resp_last_member)     \
3717 	.write.has_resp = 1,                                                   \
3718 	.write.req_size = offsetofend(req, req_last_member),                   \
3719 	.write.resp_size = offsetofend(resp, resp_last_member)
3720 
3721 #define UAPI_DEF_WRITE_I_EX(req, req_last_member)                              \
3722 	.write.req_size = offsetofend(req, req_last_member)
3723 
3724 const struct uapi_definition uverbs_def_write_intf[] = {
3725 	DECLARE_UVERBS_OBJECT(
3726 		UVERBS_OBJECT_AH,
3727 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_CREATE_AH,
3728 				     ib_uverbs_create_ah,
3729 				     UAPI_DEF_WRITE_UDATA_IO(
3730 					     struct ib_uverbs_create_ah,
3731 					     struct ib_uverbs_create_ah_resp)),
3732 		DECLARE_UVERBS_WRITE(
3733 			IB_USER_VERBS_CMD_DESTROY_AH,
3734 			ib_uverbs_destroy_ah,
3735 			UAPI_DEF_WRITE_I(struct ib_uverbs_destroy_ah)),
3736 		UAPI_DEF_OBJ_NEEDS_FN(create_user_ah),
3737 		UAPI_DEF_OBJ_NEEDS_FN(destroy_ah)),
3738 
3739 	DECLARE_UVERBS_OBJECT(
3740 		UVERBS_OBJECT_COMP_CHANNEL,
3741 		DECLARE_UVERBS_WRITE(
3742 			IB_USER_VERBS_CMD_CREATE_COMP_CHANNEL,
3743 			ib_uverbs_create_comp_channel,
3744 			UAPI_DEF_WRITE_IO(
3745 				struct ib_uverbs_create_comp_channel,
3746 				struct ib_uverbs_create_comp_channel_resp))),
3747 
3748 	DECLARE_UVERBS_OBJECT(
3749 		UVERBS_OBJECT_CQ,
3750 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_CREATE_CQ,
3751 				     ib_uverbs_create_cq,
3752 				     UAPI_DEF_WRITE_UDATA_IO(
3753 					     struct ib_uverbs_create_cq,
3754 					     struct ib_uverbs_create_cq_resp),
3755 				     UAPI_DEF_METHOD_NEEDS_FN(create_cq)),
3756 		DECLARE_UVERBS_WRITE(
3757 			IB_USER_VERBS_CMD_DESTROY_CQ,
3758 			ib_uverbs_destroy_cq,
3759 			UAPI_DEF_WRITE_IO(struct ib_uverbs_destroy_cq,
3760 					  struct ib_uverbs_destroy_cq_resp),
3761 			UAPI_DEF_METHOD_NEEDS_FN(destroy_cq)),
3762 		DECLARE_UVERBS_WRITE(
3763 			IB_USER_VERBS_CMD_POLL_CQ,
3764 			ib_uverbs_poll_cq,
3765 			UAPI_DEF_WRITE_IO(struct ib_uverbs_poll_cq,
3766 					  struct ib_uverbs_poll_cq_resp),
3767 			UAPI_DEF_METHOD_NEEDS_FN(poll_cq)),
3768 		DECLARE_UVERBS_WRITE(
3769 			IB_USER_VERBS_CMD_REQ_NOTIFY_CQ,
3770 			ib_uverbs_req_notify_cq,
3771 			UAPI_DEF_WRITE_I(struct ib_uverbs_req_notify_cq),
3772 			UAPI_DEF_METHOD_NEEDS_FN(req_notify_cq)),
3773 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_RESIZE_CQ,
3774 				     ib_uverbs_resize_cq,
3775 				     UAPI_DEF_WRITE_UDATA_IO(
3776 					     struct ib_uverbs_resize_cq,
3777 					     struct ib_uverbs_resize_cq_resp),
3778 				     UAPI_DEF_METHOD_NEEDS_FN(resize_cq)),
3779 		DECLARE_UVERBS_WRITE_EX(
3780 			IB_USER_VERBS_EX_CMD_CREATE_CQ,
3781 			ib_uverbs_ex_create_cq,
3782 			UAPI_DEF_WRITE_IO_EX(struct ib_uverbs_ex_create_cq,
3783 					     reserved,
3784 					     struct ib_uverbs_ex_create_cq_resp,
3785 					     response_length),
3786 			UAPI_DEF_METHOD_NEEDS_FN(create_cq)),
3787 		DECLARE_UVERBS_WRITE_EX(
3788 			IB_USER_VERBS_EX_CMD_MODIFY_CQ,
3789 			ib_uverbs_ex_modify_cq,
3790 			UAPI_DEF_WRITE_I(struct ib_uverbs_ex_modify_cq),
3791 			UAPI_DEF_METHOD_NEEDS_FN(modify_cq))),
3792 
3793 	DECLARE_UVERBS_OBJECT(
3794 		UVERBS_OBJECT_DEVICE,
3795 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_GET_CONTEXT,
3796 				     ib_uverbs_get_context,
3797 				     UAPI_DEF_WRITE_UDATA_IO(
3798 					     struct ib_uverbs_get_context,
3799 					     struct ib_uverbs_get_context_resp)),
3800 		DECLARE_UVERBS_WRITE(
3801 			IB_USER_VERBS_CMD_QUERY_DEVICE,
3802 			ib_uverbs_query_device,
3803 			UAPI_DEF_WRITE_IO(struct ib_uverbs_query_device,
3804 					  struct ib_uverbs_query_device_resp)),
3805 		DECLARE_UVERBS_WRITE(
3806 			IB_USER_VERBS_CMD_QUERY_PORT,
3807 			ib_uverbs_query_port,
3808 			UAPI_DEF_WRITE_IO(struct ib_uverbs_query_port,
3809 					  struct ib_uverbs_query_port_resp),
3810 			UAPI_DEF_METHOD_NEEDS_FN(query_port)),
3811 		DECLARE_UVERBS_WRITE_EX(
3812 			IB_USER_VERBS_EX_CMD_QUERY_DEVICE,
3813 			ib_uverbs_ex_query_device,
3814 			UAPI_DEF_WRITE_IO_EX(
3815 				struct ib_uverbs_ex_query_device,
3816 				reserved,
3817 				struct ib_uverbs_ex_query_device_resp,
3818 				response_length),
3819 			UAPI_DEF_METHOD_NEEDS_FN(query_device)),
3820 		UAPI_DEF_OBJ_NEEDS_FN(alloc_ucontext),
3821 		UAPI_DEF_OBJ_NEEDS_FN(dealloc_ucontext)),
3822 
3823 	DECLARE_UVERBS_OBJECT(
3824 		UVERBS_OBJECT_FLOW,
3825 		DECLARE_UVERBS_WRITE_EX(
3826 			IB_USER_VERBS_EX_CMD_CREATE_FLOW,
3827 			ib_uverbs_ex_create_flow,
3828 			UAPI_DEF_WRITE_IO_EX(struct ib_uverbs_create_flow,
3829 					     flow_attr,
3830 					     struct ib_uverbs_create_flow_resp,
3831 					     flow_handle),
3832 			UAPI_DEF_METHOD_NEEDS_FN(create_flow)),
3833 		DECLARE_UVERBS_WRITE_EX(
3834 			IB_USER_VERBS_EX_CMD_DESTROY_FLOW,
3835 			ib_uverbs_ex_destroy_flow,
3836 			UAPI_DEF_WRITE_I(struct ib_uverbs_destroy_flow),
3837 			UAPI_DEF_METHOD_NEEDS_FN(destroy_flow))),
3838 
3839 	DECLARE_UVERBS_OBJECT(
3840 		UVERBS_OBJECT_MR,
3841 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_DEREG_MR,
3842 				     ib_uverbs_dereg_mr,
3843 				     UAPI_DEF_WRITE_I(struct ib_uverbs_dereg_mr),
3844 				     UAPI_DEF_METHOD_NEEDS_FN(dereg_mr)),
3845 		DECLARE_UVERBS_WRITE(
3846 			IB_USER_VERBS_CMD_REG_MR,
3847 			ib_uverbs_reg_mr,
3848 			UAPI_DEF_WRITE_UDATA_IO(struct ib_uverbs_reg_mr,
3849 						struct ib_uverbs_reg_mr_resp),
3850 			UAPI_DEF_METHOD_NEEDS_FN(reg_user_mr)),
3851 		DECLARE_UVERBS_WRITE(
3852 			IB_USER_VERBS_CMD_REREG_MR,
3853 			ib_uverbs_rereg_mr,
3854 			UAPI_DEF_WRITE_UDATA_IO(struct ib_uverbs_rereg_mr,
3855 						struct ib_uverbs_rereg_mr_resp),
3856 			UAPI_DEF_METHOD_NEEDS_FN(rereg_user_mr))),
3857 
3858 	DECLARE_UVERBS_OBJECT(
3859 		UVERBS_OBJECT_MW,
3860 		DECLARE_UVERBS_WRITE(
3861 			IB_USER_VERBS_CMD_ALLOC_MW,
3862 			ib_uverbs_alloc_mw,
3863 			UAPI_DEF_WRITE_UDATA_IO(struct ib_uverbs_alloc_mw,
3864 						struct ib_uverbs_alloc_mw_resp),
3865 			UAPI_DEF_METHOD_NEEDS_FN(alloc_mw)),
3866 		DECLARE_UVERBS_WRITE(
3867 			IB_USER_VERBS_CMD_DEALLOC_MW,
3868 			ib_uverbs_dealloc_mw,
3869 			UAPI_DEF_WRITE_I(struct ib_uverbs_dealloc_mw),
3870 			UAPI_DEF_METHOD_NEEDS_FN(dealloc_mw))),
3871 
3872 	DECLARE_UVERBS_OBJECT(
3873 		UVERBS_OBJECT_PD,
3874 		DECLARE_UVERBS_WRITE(
3875 			IB_USER_VERBS_CMD_ALLOC_PD,
3876 			ib_uverbs_alloc_pd,
3877 			UAPI_DEF_WRITE_UDATA_IO(struct ib_uverbs_alloc_pd,
3878 						struct ib_uverbs_alloc_pd_resp),
3879 			UAPI_DEF_METHOD_NEEDS_FN(alloc_pd)),
3880 		DECLARE_UVERBS_WRITE(
3881 			IB_USER_VERBS_CMD_DEALLOC_PD,
3882 			ib_uverbs_dealloc_pd,
3883 			UAPI_DEF_WRITE_I(struct ib_uverbs_dealloc_pd),
3884 			UAPI_DEF_METHOD_NEEDS_FN(dealloc_pd))),
3885 
3886 	DECLARE_UVERBS_OBJECT(
3887 		UVERBS_OBJECT_QP,
3888 		DECLARE_UVERBS_WRITE(
3889 			IB_USER_VERBS_CMD_ATTACH_MCAST,
3890 			ib_uverbs_attach_mcast,
3891 			UAPI_DEF_WRITE_I(struct ib_uverbs_attach_mcast),
3892 			UAPI_DEF_METHOD_NEEDS_FN(attach_mcast),
3893 			UAPI_DEF_METHOD_NEEDS_FN(detach_mcast)),
3894 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_CREATE_QP,
3895 				     ib_uverbs_create_qp,
3896 				     UAPI_DEF_WRITE_UDATA_IO(
3897 					     struct ib_uverbs_create_qp,
3898 					     struct ib_uverbs_create_qp_resp),
3899 				     UAPI_DEF_METHOD_NEEDS_FN(create_qp)),
3900 		DECLARE_UVERBS_WRITE(
3901 			IB_USER_VERBS_CMD_DESTROY_QP,
3902 			ib_uverbs_destroy_qp,
3903 			UAPI_DEF_WRITE_IO(struct ib_uverbs_destroy_qp,
3904 					  struct ib_uverbs_destroy_qp_resp),
3905 			UAPI_DEF_METHOD_NEEDS_FN(destroy_qp)),
3906 		DECLARE_UVERBS_WRITE(
3907 			IB_USER_VERBS_CMD_DETACH_MCAST,
3908 			ib_uverbs_detach_mcast,
3909 			UAPI_DEF_WRITE_I(struct ib_uverbs_detach_mcast),
3910 			UAPI_DEF_METHOD_NEEDS_FN(detach_mcast)),
3911 		DECLARE_UVERBS_WRITE(
3912 			IB_USER_VERBS_CMD_MODIFY_QP,
3913 			ib_uverbs_modify_qp,
3914 			UAPI_DEF_WRITE_I(struct ib_uverbs_modify_qp),
3915 			UAPI_DEF_METHOD_NEEDS_FN(modify_qp)),
3916 		DECLARE_UVERBS_WRITE(
3917 			IB_USER_VERBS_CMD_POST_RECV,
3918 			ib_uverbs_post_recv,
3919 			UAPI_DEF_WRITE_IO(struct ib_uverbs_post_recv,
3920 					  struct ib_uverbs_post_recv_resp),
3921 			UAPI_DEF_METHOD_NEEDS_FN(post_recv)),
3922 		DECLARE_UVERBS_WRITE(
3923 			IB_USER_VERBS_CMD_POST_SEND,
3924 			ib_uverbs_post_send,
3925 			UAPI_DEF_WRITE_IO(struct ib_uverbs_post_send,
3926 					  struct ib_uverbs_post_send_resp),
3927 			UAPI_DEF_METHOD_NEEDS_FN(post_send)),
3928 		DECLARE_UVERBS_WRITE(
3929 			IB_USER_VERBS_CMD_QUERY_QP,
3930 			ib_uverbs_query_qp,
3931 			UAPI_DEF_WRITE_IO(struct ib_uverbs_query_qp,
3932 					  struct ib_uverbs_query_qp_resp),
3933 			UAPI_DEF_METHOD_NEEDS_FN(query_qp)),
3934 		DECLARE_UVERBS_WRITE_EX(
3935 			IB_USER_VERBS_EX_CMD_CREATE_QP,
3936 			ib_uverbs_ex_create_qp,
3937 			UAPI_DEF_WRITE_IO_EX(struct ib_uverbs_ex_create_qp,
3938 					     comp_mask,
3939 					     struct ib_uverbs_ex_create_qp_resp,
3940 					     response_length),
3941 			UAPI_DEF_METHOD_NEEDS_FN(create_qp)),
3942 		DECLARE_UVERBS_WRITE_EX(
3943 			IB_USER_VERBS_EX_CMD_MODIFY_QP,
3944 			ib_uverbs_ex_modify_qp,
3945 			UAPI_DEF_WRITE_IO_EX(struct ib_uverbs_ex_modify_qp,
3946 					     base,
3947 					     struct ib_uverbs_ex_modify_qp_resp,
3948 					     response_length),
3949 			UAPI_DEF_METHOD_NEEDS_FN(modify_qp))),
3950 
3951 	DECLARE_UVERBS_OBJECT(
3952 		UVERBS_OBJECT_RWQ_IND_TBL,
3953 		DECLARE_UVERBS_WRITE_EX(
3954 			IB_USER_VERBS_EX_CMD_CREATE_RWQ_IND_TBL,
3955 			ib_uverbs_ex_create_rwq_ind_table,
3956 			UAPI_DEF_WRITE_IO_EX(
3957 				struct ib_uverbs_ex_create_rwq_ind_table,
3958 				log_ind_tbl_size,
3959 				struct ib_uverbs_ex_create_rwq_ind_table_resp,
3960 				ind_tbl_num),
3961 			UAPI_DEF_METHOD_NEEDS_FN(create_rwq_ind_table)),
3962 		DECLARE_UVERBS_WRITE_EX(
3963 			IB_USER_VERBS_EX_CMD_DESTROY_RWQ_IND_TBL,
3964 			ib_uverbs_ex_destroy_rwq_ind_table,
3965 			UAPI_DEF_WRITE_I(
3966 				struct ib_uverbs_ex_destroy_rwq_ind_table),
3967 			UAPI_DEF_METHOD_NEEDS_FN(destroy_rwq_ind_table))),
3968 
3969 	DECLARE_UVERBS_OBJECT(
3970 		UVERBS_OBJECT_WQ,
3971 		DECLARE_UVERBS_WRITE_EX(
3972 			IB_USER_VERBS_EX_CMD_CREATE_WQ,
3973 			ib_uverbs_ex_create_wq,
3974 			UAPI_DEF_WRITE_IO_EX(struct ib_uverbs_ex_create_wq,
3975 					     max_sge,
3976 					     struct ib_uverbs_ex_create_wq_resp,
3977 					     wqn),
3978 			UAPI_DEF_METHOD_NEEDS_FN(create_wq)),
3979 		DECLARE_UVERBS_WRITE_EX(
3980 			IB_USER_VERBS_EX_CMD_DESTROY_WQ,
3981 			ib_uverbs_ex_destroy_wq,
3982 			UAPI_DEF_WRITE_IO_EX(struct ib_uverbs_ex_destroy_wq,
3983 					     wq_handle,
3984 					     struct ib_uverbs_ex_destroy_wq_resp,
3985 					     reserved),
3986 			UAPI_DEF_METHOD_NEEDS_FN(destroy_wq)),
3987 		DECLARE_UVERBS_WRITE_EX(
3988 			IB_USER_VERBS_EX_CMD_MODIFY_WQ,
3989 			ib_uverbs_ex_modify_wq,
3990 			UAPI_DEF_WRITE_I_EX(struct ib_uverbs_ex_modify_wq,
3991 					    curr_wq_state),
3992 			UAPI_DEF_METHOD_NEEDS_FN(modify_wq))),
3993 
3994 	DECLARE_UVERBS_OBJECT(
3995 		UVERBS_OBJECT_SRQ,
3996 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_CREATE_SRQ,
3997 				     ib_uverbs_create_srq,
3998 				     UAPI_DEF_WRITE_UDATA_IO(
3999 					     struct ib_uverbs_create_srq,
4000 					     struct ib_uverbs_create_srq_resp),
4001 				     UAPI_DEF_METHOD_NEEDS_FN(create_srq)),
4002 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_CREATE_XSRQ,
4003 				     ib_uverbs_create_xsrq,
4004 				     UAPI_DEF_WRITE_UDATA_IO(
4005 					     struct ib_uverbs_create_xsrq,
4006 					     struct ib_uverbs_create_srq_resp),
4007 				     UAPI_DEF_METHOD_NEEDS_FN(create_srq)),
4008 		DECLARE_UVERBS_WRITE(
4009 			IB_USER_VERBS_CMD_DESTROY_SRQ,
4010 			ib_uverbs_destroy_srq,
4011 			UAPI_DEF_WRITE_IO(struct ib_uverbs_destroy_srq,
4012 					  struct ib_uverbs_destroy_srq_resp),
4013 			UAPI_DEF_METHOD_NEEDS_FN(destroy_srq)),
4014 		DECLARE_UVERBS_WRITE(
4015 			IB_USER_VERBS_CMD_MODIFY_SRQ,
4016 			ib_uverbs_modify_srq,
4017 			UAPI_DEF_WRITE_UDATA_I(struct ib_uverbs_modify_srq),
4018 			UAPI_DEF_METHOD_NEEDS_FN(modify_srq)),
4019 		DECLARE_UVERBS_WRITE(
4020 			IB_USER_VERBS_CMD_POST_SRQ_RECV,
4021 			ib_uverbs_post_srq_recv,
4022 			UAPI_DEF_WRITE_IO(struct ib_uverbs_post_srq_recv,
4023 					  struct ib_uverbs_post_srq_recv_resp),
4024 			UAPI_DEF_METHOD_NEEDS_FN(post_srq_recv)),
4025 		DECLARE_UVERBS_WRITE(
4026 			IB_USER_VERBS_CMD_QUERY_SRQ,
4027 			ib_uverbs_query_srq,
4028 			UAPI_DEF_WRITE_IO(struct ib_uverbs_query_srq,
4029 					  struct ib_uverbs_query_srq_resp),
4030 			UAPI_DEF_METHOD_NEEDS_FN(query_srq))),
4031 
4032 	DECLARE_UVERBS_OBJECT(
4033 		UVERBS_OBJECT_XRCD,
4034 		DECLARE_UVERBS_WRITE(
4035 			IB_USER_VERBS_CMD_CLOSE_XRCD,
4036 			ib_uverbs_close_xrcd,
4037 			UAPI_DEF_WRITE_I(struct ib_uverbs_close_xrcd)),
4038 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_OPEN_QP,
4039 				     ib_uverbs_open_qp,
4040 				     UAPI_DEF_WRITE_UDATA_IO(
4041 					     struct ib_uverbs_open_qp,
4042 					     struct ib_uverbs_create_qp_resp)),
4043 		DECLARE_UVERBS_WRITE(IB_USER_VERBS_CMD_OPEN_XRCD,
4044 				     ib_uverbs_open_xrcd,
4045 				     UAPI_DEF_WRITE_UDATA_IO(
4046 					     struct ib_uverbs_open_xrcd,
4047 					     struct ib_uverbs_open_xrcd_resp)),
4048 		UAPI_DEF_OBJ_NEEDS_FN(alloc_xrcd),
4049 		UAPI_DEF_OBJ_NEEDS_FN(dealloc_xrcd)),
4050 
4051 	{},
4052 };
4053