xref: /openbmc/linux/drivers/gpu/drm/drm_syncobj.c (revision 2f828fb2)
1 /*
2  * Copyright 2017 Red Hat
3  * Parts ported from amdgpu (fence wait code).
4  * Copyright 2016 Advanced Micro Devices, Inc.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the "Software"),
8  * to deal in the Software without restriction, including without limitation
9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10  * and/or sell copies of the Software, and to permit persons to whom the
11  * Software is furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice (including the next
14  * paragraph) shall be included in all copies or substantial portions of the
15  * Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
20  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
22  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
23  * IN THE SOFTWARE.
24  *
25  * Authors:
26  *
27  */
28 
29 /**
30  * DOC: Overview
31  *
32  * DRM synchronisation objects (syncobj) are a persistent objects,
33  * that contain an optional fence. The fence can be updated with a new
34  * fence, or be NULL.
35  *
36  * syncobj's can be waited upon, where it will wait for the underlying
37  * fence.
38  *
39  * syncobj's can be export to fd's and back, these fd's are opaque and
40  * have no other use case, except passing the syncobj between processes.
41  *
42  * Their primary use-case is to implement Vulkan fences and semaphores.
43  *
44  * syncobj have a kref reference count, but also have an optional file.
45  * The file is only created once the syncobj is exported.
46  * The file takes a reference on the kref.
47  */
48 
49 #include <drm/drmP.h>
50 #include <linux/file.h>
51 #include <linux/fs.h>
52 #include <linux/anon_inodes.h>
53 #include <linux/sync_file.h>
54 #include <linux/sched/signal.h>
55 
56 #include "drm_internal.h"
57 #include <drm/drm_syncobj.h>
58 
59 /**
60  * drm_syncobj_find - lookup and reference a sync object.
61  * @file_private: drm file private pointer
62  * @handle: sync object handle to lookup.
63  *
64  * Returns a reference to the syncobj pointed to by handle or NULL.
65  */
66 struct drm_syncobj *drm_syncobj_find(struct drm_file *file_private,
67 				     u32 handle)
68 {
69 	struct drm_syncobj *syncobj;
70 
71 	spin_lock(&file_private->syncobj_table_lock);
72 
73 	/* Check if we currently have a reference on the object */
74 	syncobj = idr_find(&file_private->syncobj_idr, handle);
75 	if (syncobj)
76 		drm_syncobj_get(syncobj);
77 
78 	spin_unlock(&file_private->syncobj_table_lock);
79 
80 	return syncobj;
81 }
82 EXPORT_SYMBOL(drm_syncobj_find);
83 
84 static void drm_syncobj_add_callback_locked(struct drm_syncobj *syncobj,
85 					    struct drm_syncobj_cb *cb,
86 					    drm_syncobj_func_t func)
87 {
88 	cb->func = func;
89 	list_add_tail(&cb->node, &syncobj->cb_list);
90 }
91 
92 static int drm_syncobj_fence_get_or_add_callback(struct drm_syncobj *syncobj,
93 						 struct dma_fence **fence,
94 						 struct drm_syncobj_cb *cb,
95 						 drm_syncobj_func_t func)
96 {
97 	int ret;
98 
99 	*fence = drm_syncobj_fence_get(syncobj);
100 	if (*fence)
101 		return 1;
102 
103 	spin_lock(&syncobj->lock);
104 	/* We've already tried once to get a fence and failed.  Now that we
105 	 * have the lock, try one more time just to be sure we don't add a
106 	 * callback when a fence has already been set.
107 	 */
108 	if (syncobj->fence) {
109 		*fence = dma_fence_get(syncobj->fence);
110 		ret = 1;
111 	} else {
112 		*fence = NULL;
113 		drm_syncobj_add_callback_locked(syncobj, cb, func);
114 		ret = 0;
115 	}
116 	spin_unlock(&syncobj->lock);
117 
118 	return ret;
119 }
120 
121 /**
122  * drm_syncobj_add_callback - adds a callback to syncobj::cb_list
123  * @syncobj: Sync object to which to add the callback
124  * @cb: Callback to add
125  * @func: Func to use when initializing the drm_syncobj_cb struct
126  *
127  * This adds a callback to be called next time the fence is replaced
128  */
129 void drm_syncobj_add_callback(struct drm_syncobj *syncobj,
130 			      struct drm_syncobj_cb *cb,
131 			      drm_syncobj_func_t func)
132 {
133 	spin_lock(&syncobj->lock);
134 	drm_syncobj_add_callback_locked(syncobj, cb, func);
135 	spin_unlock(&syncobj->lock);
136 }
137 EXPORT_SYMBOL(drm_syncobj_add_callback);
138 
139 /**
140  * drm_syncobj_add_callback - removes a callback to syncobj::cb_list
141  * @syncobj: Sync object from which to remove the callback
142  * @cb: Callback to remove
143  */
144 void drm_syncobj_remove_callback(struct drm_syncobj *syncobj,
145 				 struct drm_syncobj_cb *cb)
146 {
147 	spin_lock(&syncobj->lock);
148 	list_del_init(&cb->node);
149 	spin_unlock(&syncobj->lock);
150 }
151 EXPORT_SYMBOL(drm_syncobj_remove_callback);
152 
153 /**
154  * drm_syncobj_replace_fence - replace fence in a sync object.
155  * @syncobj: Sync object to replace fence in
156  * @fence: fence to install in sync file.
157  *
158  * This replaces the fence on a sync object.
159  */
160 void drm_syncobj_replace_fence(struct drm_syncobj *syncobj,
161 			       struct dma_fence *fence)
162 {
163 	struct dma_fence *old_fence;
164 	struct drm_syncobj_cb *cur, *tmp;
165 
166 	if (fence)
167 		dma_fence_get(fence);
168 
169 	spin_lock(&syncobj->lock);
170 
171 	old_fence = syncobj->fence;
172 	syncobj->fence = fence;
173 
174 	if (fence != old_fence) {
175 		list_for_each_entry_safe(cur, tmp, &syncobj->cb_list, node) {
176 			list_del_init(&cur->node);
177 			cur->func(syncobj, cur);
178 		}
179 	}
180 
181 	spin_unlock(&syncobj->lock);
182 
183 	dma_fence_put(old_fence);
184 }
185 EXPORT_SYMBOL(drm_syncobj_replace_fence);
186 
187 struct drm_syncobj_null_fence {
188 	struct dma_fence base;
189 	spinlock_t lock;
190 };
191 
192 static const char *drm_syncobj_null_fence_get_name(struct dma_fence *fence)
193 {
194         return "syncobjnull";
195 }
196 
197 static bool drm_syncobj_null_fence_enable_signaling(struct dma_fence *fence)
198 {
199     dma_fence_enable_sw_signaling(fence);
200     return !dma_fence_is_signaled(fence);
201 }
202 
203 static const struct dma_fence_ops drm_syncobj_null_fence_ops = {
204 	.get_driver_name = drm_syncobj_null_fence_get_name,
205 	.get_timeline_name = drm_syncobj_null_fence_get_name,
206 	.enable_signaling = drm_syncobj_null_fence_enable_signaling,
207 	.wait = dma_fence_default_wait,
208 	.release = NULL,
209 };
210 
211 static int drm_syncobj_assign_null_handle(struct drm_syncobj *syncobj)
212 {
213 	struct drm_syncobj_null_fence *fence;
214 	fence = kzalloc(sizeof(*fence), GFP_KERNEL);
215 	if (fence == NULL)
216 		return -ENOMEM;
217 
218 	spin_lock_init(&fence->lock);
219 	dma_fence_init(&fence->base, &drm_syncobj_null_fence_ops,
220 		       &fence->lock, 0, 0);
221 	dma_fence_signal(&fence->base);
222 
223 	drm_syncobj_replace_fence(syncobj, &fence->base);
224 
225 	dma_fence_put(&fence->base);
226 
227 	return 0;
228 }
229 
230 int drm_syncobj_find_fence(struct drm_file *file_private,
231 			   u32 handle,
232 			   struct dma_fence **fence)
233 {
234 	struct drm_syncobj *syncobj = drm_syncobj_find(file_private, handle);
235 	int ret = 0;
236 
237 	if (!syncobj)
238 		return -ENOENT;
239 
240 	*fence = drm_syncobj_fence_get(syncobj);
241 	if (!*fence) {
242 		ret = -EINVAL;
243 	}
244 	drm_syncobj_put(syncobj);
245 	return ret;
246 }
247 EXPORT_SYMBOL(drm_syncobj_find_fence);
248 
249 /**
250  * drm_syncobj_free - free a sync object.
251  * @kref: kref to free.
252  *
253  * Only to be called from kref_put in drm_syncobj_put.
254  */
255 void drm_syncobj_free(struct kref *kref)
256 {
257 	struct drm_syncobj *syncobj = container_of(kref,
258 						   struct drm_syncobj,
259 						   refcount);
260 	drm_syncobj_replace_fence(syncobj, NULL);
261 	kfree(syncobj);
262 }
263 EXPORT_SYMBOL(drm_syncobj_free);
264 
265 /**
266  * drm_syncobj_create - create a new syncobj
267  * @out_syncobj: returned syncobj
268  * @flags: DRM_SYNCOBJ_* flags
269  * @fence: if non-NULL, the syncobj will represent this fence
270  */
271 int drm_syncobj_create(struct drm_syncobj **out_syncobj, uint32_t flags,
272 		       struct dma_fence *fence)
273 {
274 	int ret;
275 	struct drm_syncobj *syncobj;
276 
277 	syncobj = kzalloc(sizeof(struct drm_syncobj), GFP_KERNEL);
278 	if (!syncobj)
279 		return -ENOMEM;
280 
281 	kref_init(&syncobj->refcount);
282 	INIT_LIST_HEAD(&syncobj->cb_list);
283 	spin_lock_init(&syncobj->lock);
284 
285 	if (flags & DRM_SYNCOBJ_CREATE_SIGNALED) {
286 		ret = drm_syncobj_assign_null_handle(syncobj);
287 		if (ret < 0) {
288 			drm_syncobj_put(syncobj);
289 			return ret;
290 		}
291 	}
292 
293 	if (fence)
294 		drm_syncobj_replace_fence(syncobj, fence);
295 
296 	*out_syncobj = syncobj;
297 	return 0;
298 }
299 EXPORT_SYMBOL(drm_syncobj_create);
300 
301 /**
302  * drm_syncobj_get_handle - get a handle from a syncobj
303  */
304 int drm_syncobj_get_handle(struct drm_file *file_private,
305 			   struct drm_syncobj *syncobj, u32 *handle)
306 {
307 	int ret;
308 
309 	/* take a reference to put in the idr */
310 	drm_syncobj_get(syncobj);
311 
312 	idr_preload(GFP_KERNEL);
313 	spin_lock(&file_private->syncobj_table_lock);
314 	ret = idr_alloc(&file_private->syncobj_idr, syncobj, 1, 0, GFP_NOWAIT);
315 	spin_unlock(&file_private->syncobj_table_lock);
316 
317 	idr_preload_end();
318 
319 	if (ret < 0) {
320 		drm_syncobj_put(syncobj);
321 		return ret;
322 	}
323 
324 	*handle = ret;
325 	return 0;
326 }
327 EXPORT_SYMBOL(drm_syncobj_get_handle);
328 
329 static int drm_syncobj_create_as_handle(struct drm_file *file_private,
330 					u32 *handle, uint32_t flags)
331 {
332 	int ret;
333 	struct drm_syncobj *syncobj;
334 
335 	ret = drm_syncobj_create(&syncobj, flags, NULL);
336 	if (ret)
337 		return ret;
338 
339 	ret = drm_syncobj_get_handle(file_private, syncobj, handle);
340 	drm_syncobj_put(syncobj);
341 	return ret;
342 }
343 
344 static int drm_syncobj_destroy(struct drm_file *file_private,
345 			       u32 handle)
346 {
347 	struct drm_syncobj *syncobj;
348 
349 	spin_lock(&file_private->syncobj_table_lock);
350 	syncobj = idr_remove(&file_private->syncobj_idr, handle);
351 	spin_unlock(&file_private->syncobj_table_lock);
352 
353 	if (!syncobj)
354 		return -EINVAL;
355 
356 	drm_syncobj_put(syncobj);
357 	return 0;
358 }
359 
360 static int drm_syncobj_file_release(struct inode *inode, struct file *file)
361 {
362 	struct drm_syncobj *syncobj = file->private_data;
363 
364 	drm_syncobj_put(syncobj);
365 	return 0;
366 }
367 
368 static const struct file_operations drm_syncobj_file_fops = {
369 	.release = drm_syncobj_file_release,
370 };
371 
372 static int drm_syncobj_alloc_file(struct drm_syncobj *syncobj)
373 {
374 	struct file *file = anon_inode_getfile("syncobj_file",
375 					       &drm_syncobj_file_fops,
376 					       syncobj, 0);
377 	if (IS_ERR(file))
378 		return PTR_ERR(file);
379 
380 	drm_syncobj_get(syncobj);
381 	if (cmpxchg(&syncobj->file, NULL, file)) {
382 		/* lost the race */
383 		fput(file);
384 	}
385 
386 	return 0;
387 }
388 
389 int drm_syncobj_get_fd(struct drm_syncobj *syncobj, int *p_fd)
390 {
391 	int ret;
392 	int fd;
393 
394 	fd = get_unused_fd_flags(O_CLOEXEC);
395 	if (fd < 0)
396 		return fd;
397 
398 	if (!syncobj->file) {
399 		ret = drm_syncobj_alloc_file(syncobj);
400 		if (ret) {
401 			put_unused_fd(fd);
402 			return ret;
403 		}
404 	}
405 	fd_install(fd, syncobj->file);
406 	*p_fd = fd;
407 	return 0;
408 }
409 EXPORT_SYMBOL(drm_syncobj_get_fd);
410 
411 static int drm_syncobj_handle_to_fd(struct drm_file *file_private,
412 				    u32 handle, int *p_fd)
413 {
414 	struct drm_syncobj *syncobj = drm_syncobj_find(file_private, handle);
415 	int ret;
416 
417 	if (!syncobj)
418 		return -EINVAL;
419 
420 	ret = drm_syncobj_get_fd(syncobj, p_fd);
421 	drm_syncobj_put(syncobj);
422 	return ret;
423 }
424 
425 static struct drm_syncobj *drm_syncobj_fdget(int fd)
426 {
427 	struct file *file = fget(fd);
428 
429 	if (!file)
430 		return NULL;
431 	if (file->f_op != &drm_syncobj_file_fops)
432 		goto err;
433 
434 	return file->private_data;
435 err:
436 	fput(file);
437 	return NULL;
438 };
439 
440 static int drm_syncobj_fd_to_handle(struct drm_file *file_private,
441 				    int fd, u32 *handle)
442 {
443 	struct drm_syncobj *syncobj = drm_syncobj_fdget(fd);
444 	int ret;
445 
446 	if (!syncobj)
447 		return -EINVAL;
448 
449 	/* take a reference to put in the idr */
450 	drm_syncobj_get(syncobj);
451 
452 	idr_preload(GFP_KERNEL);
453 	spin_lock(&file_private->syncobj_table_lock);
454 	ret = idr_alloc(&file_private->syncobj_idr, syncobj, 1, 0, GFP_NOWAIT);
455 	spin_unlock(&file_private->syncobj_table_lock);
456 	idr_preload_end();
457 
458 	if (ret < 0) {
459 		fput(syncobj->file);
460 		return ret;
461 	}
462 	*handle = ret;
463 	return 0;
464 }
465 
466 static int drm_syncobj_import_sync_file_fence(struct drm_file *file_private,
467 					      int fd, int handle)
468 {
469 	struct dma_fence *fence = sync_file_get_fence(fd);
470 	struct drm_syncobj *syncobj;
471 
472 	if (!fence)
473 		return -EINVAL;
474 
475 	syncobj = drm_syncobj_find(file_private, handle);
476 	if (!syncobj) {
477 		dma_fence_put(fence);
478 		return -ENOENT;
479 	}
480 
481 	drm_syncobj_replace_fence(syncobj, fence);
482 	dma_fence_put(fence);
483 	drm_syncobj_put(syncobj);
484 	return 0;
485 }
486 
487 static int drm_syncobj_export_sync_file(struct drm_file *file_private,
488 					int handle, int *p_fd)
489 {
490 	int ret;
491 	struct dma_fence *fence;
492 	struct sync_file *sync_file;
493 	int fd = get_unused_fd_flags(O_CLOEXEC);
494 
495 	if (fd < 0)
496 		return fd;
497 
498 	ret = drm_syncobj_find_fence(file_private, handle, &fence);
499 	if (ret)
500 		goto err_put_fd;
501 
502 	sync_file = sync_file_create(fence);
503 
504 	dma_fence_put(fence);
505 
506 	if (!sync_file) {
507 		ret = -EINVAL;
508 		goto err_put_fd;
509 	}
510 
511 	fd_install(fd, sync_file->file);
512 
513 	*p_fd = fd;
514 	return 0;
515 err_put_fd:
516 	put_unused_fd(fd);
517 	return ret;
518 }
519 /**
520  * drm_syncobj_open - initalizes syncobj file-private structures at devnode open time
521  * @file_private: drm file-private structure to set up
522  *
523  * Called at device open time, sets up the structure for handling refcounting
524  * of sync objects.
525  */
526 void
527 drm_syncobj_open(struct drm_file *file_private)
528 {
529 	idr_init(&file_private->syncobj_idr);
530 	spin_lock_init(&file_private->syncobj_table_lock);
531 }
532 
533 static int
534 drm_syncobj_release_handle(int id, void *ptr, void *data)
535 {
536 	struct drm_syncobj *syncobj = ptr;
537 
538 	drm_syncobj_put(syncobj);
539 	return 0;
540 }
541 
542 /**
543  * drm_syncobj_release - release file-private sync object resources
544  * @file_private: drm file-private structure to clean up
545  *
546  * Called at close time when the filp is going away.
547  *
548  * Releases any remaining references on objects by this filp.
549  */
550 void
551 drm_syncobj_release(struct drm_file *file_private)
552 {
553 	idr_for_each(&file_private->syncobj_idr,
554 		     &drm_syncobj_release_handle, file_private);
555 	idr_destroy(&file_private->syncobj_idr);
556 }
557 
558 int
559 drm_syncobj_create_ioctl(struct drm_device *dev, void *data,
560 			 struct drm_file *file_private)
561 {
562 	struct drm_syncobj_create *args = data;
563 
564 	if (!drm_core_check_feature(dev, DRIVER_SYNCOBJ))
565 		return -ENODEV;
566 
567 	/* no valid flags yet */
568 	if (args->flags & ~DRM_SYNCOBJ_CREATE_SIGNALED)
569 		return -EINVAL;
570 
571 	return drm_syncobj_create_as_handle(file_private,
572 					    &args->handle, args->flags);
573 }
574 
575 int
576 drm_syncobj_destroy_ioctl(struct drm_device *dev, void *data,
577 			  struct drm_file *file_private)
578 {
579 	struct drm_syncobj_destroy *args = data;
580 
581 	if (!drm_core_check_feature(dev, DRIVER_SYNCOBJ))
582 		return -ENODEV;
583 
584 	/* make sure padding is empty */
585 	if (args->pad)
586 		return -EINVAL;
587 	return drm_syncobj_destroy(file_private, args->handle);
588 }
589 
590 int
591 drm_syncobj_handle_to_fd_ioctl(struct drm_device *dev, void *data,
592 				   struct drm_file *file_private)
593 {
594 	struct drm_syncobj_handle *args = data;
595 
596 	if (!drm_core_check_feature(dev, DRIVER_SYNCOBJ))
597 		return -ENODEV;
598 
599 	if (args->pad)
600 		return -EINVAL;
601 
602 	if (args->flags != 0 &&
603 	    args->flags != DRM_SYNCOBJ_HANDLE_TO_FD_FLAGS_EXPORT_SYNC_FILE)
604 		return -EINVAL;
605 
606 	if (args->flags & DRM_SYNCOBJ_HANDLE_TO_FD_FLAGS_EXPORT_SYNC_FILE)
607 		return drm_syncobj_export_sync_file(file_private, args->handle,
608 						    &args->fd);
609 
610 	return drm_syncobj_handle_to_fd(file_private, args->handle,
611 					&args->fd);
612 }
613 
614 int
615 drm_syncobj_fd_to_handle_ioctl(struct drm_device *dev, void *data,
616 				   struct drm_file *file_private)
617 {
618 	struct drm_syncobj_handle *args = data;
619 
620 	if (!drm_core_check_feature(dev, DRIVER_SYNCOBJ))
621 		return -ENODEV;
622 
623 	if (args->pad)
624 		return -EINVAL;
625 
626 	if (args->flags != 0 &&
627 	    args->flags != DRM_SYNCOBJ_FD_TO_HANDLE_FLAGS_IMPORT_SYNC_FILE)
628 		return -EINVAL;
629 
630 	if (args->flags & DRM_SYNCOBJ_FD_TO_HANDLE_FLAGS_IMPORT_SYNC_FILE)
631 		return drm_syncobj_import_sync_file_fence(file_private,
632 							  args->fd,
633 							  args->handle);
634 
635 	return drm_syncobj_fd_to_handle(file_private, args->fd,
636 					&args->handle);
637 }
638 
639 struct syncobj_wait_entry {
640 	struct task_struct *task;
641 	struct dma_fence *fence;
642 	struct dma_fence_cb fence_cb;
643 	struct drm_syncobj_cb syncobj_cb;
644 };
645 
646 static void syncobj_wait_fence_func(struct dma_fence *fence,
647 				    struct dma_fence_cb *cb)
648 {
649 	struct syncobj_wait_entry *wait =
650 		container_of(cb, struct syncobj_wait_entry, fence_cb);
651 
652 	wake_up_process(wait->task);
653 }
654 
655 static void syncobj_wait_syncobj_func(struct drm_syncobj *syncobj,
656 				      struct drm_syncobj_cb *cb)
657 {
658 	struct syncobj_wait_entry *wait =
659 		container_of(cb, struct syncobj_wait_entry, syncobj_cb);
660 
661 	/* This happens inside the syncobj lock */
662 	wait->fence = dma_fence_get(syncobj->fence);
663 	wake_up_process(wait->task);
664 }
665 
666 static signed long drm_syncobj_array_wait_timeout(struct drm_syncobj **syncobjs,
667 						  uint32_t count,
668 						  uint32_t flags,
669 						  signed long timeout,
670 						  uint32_t *idx)
671 {
672 	struct syncobj_wait_entry *entries;
673 	struct dma_fence *fence;
674 	signed long ret;
675 	uint32_t signaled_count, i;
676 
677 	entries = kcalloc(count, sizeof(*entries), GFP_KERNEL);
678 	if (!entries)
679 		return -ENOMEM;
680 
681 	/* Walk the list of sync objects and initialize entries.  We do
682 	 * this up-front so that we can properly return -EINVAL if there is
683 	 * a syncobj with a missing fence and then never have the chance of
684 	 * returning -EINVAL again.
685 	 */
686 	signaled_count = 0;
687 	for (i = 0; i < count; ++i) {
688 		entries[i].task = current;
689 		entries[i].fence = drm_syncobj_fence_get(syncobjs[i]);
690 		if (!entries[i].fence) {
691 			if (flags & DRM_SYNCOBJ_WAIT_FLAGS_WAIT_FOR_SUBMIT) {
692 				continue;
693 			} else {
694 				ret = -EINVAL;
695 				goto cleanup_entries;
696 			}
697 		}
698 
699 		if (dma_fence_is_signaled(entries[i].fence)) {
700 			if (signaled_count == 0 && idx)
701 				*idx = i;
702 			signaled_count++;
703 		}
704 	}
705 
706 	/* Initialize ret to the max of timeout and 1.  That way, the
707 	 * default return value indicates a successful wait and not a
708 	 * timeout.
709 	 */
710 	ret = max_t(signed long, timeout, 1);
711 
712 	if (signaled_count == count ||
713 	    (signaled_count > 0 &&
714 	     !(flags & DRM_SYNCOBJ_WAIT_FLAGS_WAIT_ALL)))
715 		goto cleanup_entries;
716 
717 	/* There's a very annoying laxness in the dma_fence API here, in
718 	 * that backends are not required to automatically report when a
719 	 * fence is signaled prior to fence->ops->enable_signaling() being
720 	 * called.  So here if we fail to match signaled_count, we need to
721 	 * fallthough and try a 0 timeout wait!
722 	 */
723 
724 	if (flags & DRM_SYNCOBJ_WAIT_FLAGS_WAIT_FOR_SUBMIT) {
725 		for (i = 0; i < count; ++i) {
726 			drm_syncobj_fence_get_or_add_callback(syncobjs[i],
727 							      &entries[i].fence,
728 							      &entries[i].syncobj_cb,
729 							      syncobj_wait_syncobj_func);
730 		}
731 	}
732 
733 	do {
734 		set_current_state(TASK_INTERRUPTIBLE);
735 
736 		signaled_count = 0;
737 		for (i = 0; i < count; ++i) {
738 			fence = entries[i].fence;
739 			if (!fence)
740 				continue;
741 
742 			if (dma_fence_is_signaled(fence) ||
743 			    (!entries[i].fence_cb.func &&
744 			     dma_fence_add_callback(fence,
745 						    &entries[i].fence_cb,
746 						    syncobj_wait_fence_func))) {
747 				/* The fence has been signaled */
748 				if (flags & DRM_SYNCOBJ_WAIT_FLAGS_WAIT_ALL) {
749 					signaled_count++;
750 				} else {
751 					if (idx)
752 						*idx = i;
753 					goto done_waiting;
754 				}
755 			}
756 		}
757 
758 		if (signaled_count == count)
759 			goto done_waiting;
760 
761 		if (timeout == 0) {
762 			/* If we are doing a 0 timeout wait and we got
763 			 * here, then we just timed out.
764 			 */
765 			ret = 0;
766 			goto done_waiting;
767 		}
768 
769 		ret = schedule_timeout(ret);
770 
771 		if (ret > 0 && signal_pending(current))
772 			ret = -ERESTARTSYS;
773 	} while (ret > 0);
774 
775 done_waiting:
776 	__set_current_state(TASK_RUNNING);
777 
778 cleanup_entries:
779 	for (i = 0; i < count; ++i) {
780 		if (entries[i].syncobj_cb.func)
781 			drm_syncobj_remove_callback(syncobjs[i],
782 						    &entries[i].syncobj_cb);
783 		if (entries[i].fence_cb.func)
784 			dma_fence_remove_callback(entries[i].fence,
785 						  &entries[i].fence_cb);
786 		dma_fence_put(entries[i].fence);
787 	}
788 	kfree(entries);
789 
790 	return ret;
791 }
792 
793 /**
794  * drm_timeout_abs_to_jiffies - calculate jiffies timeout from absolute value
795  *
796  * @timeout_nsec: timeout nsec component in ns, 0 for poll
797  *
798  * Calculate the timeout in jiffies from an absolute time in sec/nsec.
799  */
800 static signed long drm_timeout_abs_to_jiffies(int64_t timeout_nsec)
801 {
802 	ktime_t abs_timeout, now;
803 	u64 timeout_ns, timeout_jiffies64;
804 
805 	/* make 0 timeout means poll - absolute 0 doesn't seem valid */
806 	if (timeout_nsec == 0)
807 		return 0;
808 
809 	abs_timeout = ns_to_ktime(timeout_nsec);
810 	now = ktime_get();
811 
812 	if (!ktime_after(abs_timeout, now))
813 		return 0;
814 
815 	timeout_ns = ktime_to_ns(ktime_sub(abs_timeout, now));
816 
817 	timeout_jiffies64 = nsecs_to_jiffies64(timeout_ns);
818 	/*  clamp timeout to avoid infinite timeout */
819 	if (timeout_jiffies64 >= MAX_SCHEDULE_TIMEOUT - 1)
820 		return MAX_SCHEDULE_TIMEOUT - 1;
821 
822 	return timeout_jiffies64 + 1;
823 }
824 
825 static int drm_syncobj_array_wait(struct drm_device *dev,
826 				  struct drm_file *file_private,
827 				  struct drm_syncobj_wait *wait,
828 				  struct drm_syncobj **syncobjs)
829 {
830 	signed long timeout = drm_timeout_abs_to_jiffies(wait->timeout_nsec);
831 	signed long ret = 0;
832 	uint32_t first = ~0;
833 
834 	ret = drm_syncobj_array_wait_timeout(syncobjs,
835 					     wait->count_handles,
836 					     wait->flags,
837 					     timeout, &first);
838 	if (ret < 0)
839 		return ret;
840 
841 	wait->first_signaled = first;
842 	if (ret == 0)
843 		return -ETIME;
844 	return 0;
845 }
846 
847 static int drm_syncobj_array_find(struct drm_file *file_private,
848 				  void __user *user_handles,
849 				  uint32_t count_handles,
850 				  struct drm_syncobj ***syncobjs_out)
851 {
852 	uint32_t i, *handles;
853 	struct drm_syncobj **syncobjs;
854 	int ret;
855 
856 	handles = kmalloc_array(count_handles, sizeof(*handles), GFP_KERNEL);
857 	if (handles == NULL)
858 		return -ENOMEM;
859 
860 	if (copy_from_user(handles, user_handles,
861 			   sizeof(uint32_t) * count_handles)) {
862 		ret = -EFAULT;
863 		goto err_free_handles;
864 	}
865 
866 	syncobjs = kmalloc_array(count_handles, sizeof(*syncobjs), GFP_KERNEL);
867 	if (syncobjs == NULL) {
868 		ret = -ENOMEM;
869 		goto err_free_handles;
870 	}
871 
872 	for (i = 0; i < count_handles; i++) {
873 		syncobjs[i] = drm_syncobj_find(file_private, handles[i]);
874 		if (!syncobjs[i]) {
875 			ret = -ENOENT;
876 			goto err_put_syncobjs;
877 		}
878 	}
879 
880 	kfree(handles);
881 	*syncobjs_out = syncobjs;
882 	return 0;
883 
884 err_put_syncobjs:
885 	while (i-- > 0)
886 		drm_syncobj_put(syncobjs[i]);
887 	kfree(syncobjs);
888 err_free_handles:
889 	kfree(handles);
890 
891 	return ret;
892 }
893 
894 static void drm_syncobj_array_free(struct drm_syncobj **syncobjs,
895 				   uint32_t count)
896 {
897 	uint32_t i;
898 	for (i = 0; i < count; i++)
899 		drm_syncobj_put(syncobjs[i]);
900 	kfree(syncobjs);
901 }
902 
903 int
904 drm_syncobj_wait_ioctl(struct drm_device *dev, void *data,
905 		       struct drm_file *file_private)
906 {
907 	struct drm_syncobj_wait *args = data;
908 	struct drm_syncobj **syncobjs;
909 	int ret = 0;
910 
911 	if (!drm_core_check_feature(dev, DRIVER_SYNCOBJ))
912 		return -ENODEV;
913 
914 	if (args->flags & ~(DRM_SYNCOBJ_WAIT_FLAGS_WAIT_ALL |
915 			    DRM_SYNCOBJ_WAIT_FLAGS_WAIT_FOR_SUBMIT))
916 		return -EINVAL;
917 
918 	if (args->count_handles == 0)
919 		return -EINVAL;
920 
921 	ret = drm_syncobj_array_find(file_private,
922 				     u64_to_user_ptr(args->handles),
923 				     args->count_handles,
924 				     &syncobjs);
925 	if (ret < 0)
926 		return ret;
927 
928 	ret = drm_syncobj_array_wait(dev, file_private,
929 				     args, syncobjs);
930 
931 	drm_syncobj_array_free(syncobjs, args->count_handles);
932 
933 	return ret;
934 }
935 
936 int
937 drm_syncobj_reset_ioctl(struct drm_device *dev, void *data,
938 			struct drm_file *file_private)
939 {
940 	struct drm_syncobj_array *args = data;
941 	struct drm_syncobj **syncobjs;
942 	uint32_t i;
943 	int ret;
944 
945 	if (!drm_core_check_feature(dev, DRIVER_SYNCOBJ))
946 		return -ENODEV;
947 
948 	if (args->pad != 0)
949 		return -EINVAL;
950 
951 	if (args->count_handles == 0)
952 		return -EINVAL;
953 
954 	ret = drm_syncobj_array_find(file_private,
955 				     u64_to_user_ptr(args->handles),
956 				     args->count_handles,
957 				     &syncobjs);
958 	if (ret < 0)
959 		return ret;
960 
961 	for (i = 0; i < args->count_handles; i++)
962 		drm_syncobj_replace_fence(syncobjs[i], NULL);
963 
964 	drm_syncobj_array_free(syncobjs, args->count_handles);
965 
966 	return 0;
967 }
968 
969 int
970 drm_syncobj_signal_ioctl(struct drm_device *dev, void *data,
971 			 struct drm_file *file_private)
972 {
973 	struct drm_syncobj_array *args = data;
974 	struct drm_syncobj **syncobjs;
975 	uint32_t i;
976 	int ret;
977 
978 	if (!drm_core_check_feature(dev, DRIVER_SYNCOBJ))
979 		return -ENODEV;
980 
981 	if (args->pad != 0)
982 		return -EINVAL;
983 
984 	if (args->count_handles == 0)
985 		return -EINVAL;
986 
987 	ret = drm_syncobj_array_find(file_private,
988 				     u64_to_user_ptr(args->handles),
989 				     args->count_handles,
990 				     &syncobjs);
991 	if (ret < 0)
992 		return ret;
993 
994 	for (i = 0; i < args->count_handles; i++) {
995 		ret = drm_syncobj_assign_null_handle(syncobjs[i]);
996 		if (ret < 0)
997 			break;
998 	}
999 
1000 	drm_syncobj_array_free(syncobjs, args->count_handles);
1001 
1002 	return ret;
1003 }
1004