xref: /openbsd/sys/dev/pci/drm/drm_prime.c (revision 76d0caae)
1 /*
2  * Copyright © 2012 Red Hat
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice (including the next
12  * paragraph) shall be included in all copies or substantial portions of the
13  * Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21  * IN THE SOFTWARE.
22  *
23  * Authors:
24  *      Dave Airlie <airlied@redhat.com>
25  *      Rob Clark <rob.clark@linaro.org>
26  *
27  */
28 
29 #include <linux/export.h>
30 #include <linux/dma-buf.h>
31 #include <linux/rbtree.h>
32 
33 #include <drm/drm.h>
34 #include <drm/drm_drv.h>
35 #include <drm/drm_file.h>
36 #include <drm/drm_framebuffer.h>
37 #include <drm/drm_gem.h>
38 #include <drm/drm_prime.h>
39 
40 #include "drm_internal.h"
41 
42 /**
43  * DOC: overview and lifetime rules
44  *
45  * Similar to GEM global names, PRIME file descriptors are also used to share
46  * buffer objects across processes. They offer additional security: as file
47  * descriptors must be explicitly sent over UNIX domain sockets to be shared
48  * between applications, they can't be guessed like the globally unique GEM
49  * names.
50  *
51  * Drivers that support the PRIME API implement the
52  * &drm_driver.prime_handle_to_fd and &drm_driver.prime_fd_to_handle operations.
53  * GEM based drivers must use drm_gem_prime_handle_to_fd() and
54  * drm_gem_prime_fd_to_handle() to implement these. For GEM based drivers the
55  * actual driver interfaces is provided through the &drm_gem_object_funcs.export
56  * and &drm_driver.gem_prime_import hooks.
57  *
58  * &dma_buf_ops implementations for GEM drivers are all individually exported
59  * for drivers which need to overwrite or reimplement some of them.
60  *
61  * Reference Counting for GEM Drivers
62  * ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
63  *
64  * On the export the &dma_buf holds a reference to the exported buffer object,
65  * usually a &drm_gem_object. It takes this reference in the PRIME_HANDLE_TO_FD
66  * IOCTL, when it first calls &drm_gem_object_funcs.export
67  * and stores the exporting GEM object in the &dma_buf.priv field. This
68  * reference needs to be released when the final reference to the &dma_buf
69  * itself is dropped and its &dma_buf_ops.release function is called.  For
70  * GEM-based drivers, the &dma_buf should be exported using
71  * drm_gem_dmabuf_export() and then released by drm_gem_dmabuf_release().
72  *
73  * Thus the chain of references always flows in one direction, avoiding loops:
74  * importing GEM object -> dma-buf -> exported GEM bo. A further complication
75  * are the lookup caches for import and export. These are required to guarantee
76  * that any given object will always have only one uniqe userspace handle. This
77  * is required to allow userspace to detect duplicated imports, since some GEM
78  * drivers do fail command submissions if a given buffer object is listed more
79  * than once. These import and export caches in &drm_prime_file_private only
80  * retain a weak reference, which is cleaned up when the corresponding object is
81  * released.
82  *
83  * Self-importing: If userspace is using PRIME as a replacement for flink then
84  * it will get a fd->handle request for a GEM object that it created.  Drivers
85  * should detect this situation and return back the underlying object from the
86  * dma-buf private. For GEM based drivers this is handled in
87  * drm_gem_prime_import() already.
88  */
89 
90 struct drm_prime_member {
91 	struct dma_buf *dma_buf;
92 	uint32_t handle;
93 
94 	struct rb_node dmabuf_rb;
95 	struct rb_node handle_rb;
96 };
97 
98 static int drm_prime_add_buf_handle(struct drm_prime_file_private *prime_fpriv,
99 				    struct dma_buf *dma_buf, uint32_t handle)
100 {
101 	struct drm_prime_member *member;
102 	struct rb_node **p, *rb;
103 
104 	member = kmalloc(sizeof(*member), GFP_KERNEL);
105 	if (!member)
106 		return -ENOMEM;
107 
108 	get_dma_buf(dma_buf);
109 	member->dma_buf = dma_buf;
110 	member->handle = handle;
111 
112 	rb = NULL;
113 	p = &prime_fpriv->dmabufs.rb_node;
114 	while (*p) {
115 		struct drm_prime_member *pos;
116 
117 		rb = *p;
118 		pos = rb_entry(rb, struct drm_prime_member, dmabuf_rb);
119 		if (dma_buf > pos->dma_buf)
120 			p = &rb->rb_right;
121 		else
122 			p = &rb->rb_left;
123 	}
124 	rb_link_node(&member->dmabuf_rb, rb, p);
125 	rb_insert_color(&member->dmabuf_rb, &prime_fpriv->dmabufs);
126 
127 	rb = NULL;
128 	p = &prime_fpriv->handles.rb_node;
129 	while (*p) {
130 		struct drm_prime_member *pos;
131 
132 		rb = *p;
133 		pos = rb_entry(rb, struct drm_prime_member, handle_rb);
134 		if (handle > pos->handle)
135 			p = &rb->rb_right;
136 		else
137 			p = &rb->rb_left;
138 	}
139 	rb_link_node(&member->handle_rb, rb, p);
140 	rb_insert_color(&member->handle_rb, &prime_fpriv->handles);
141 
142 	return 0;
143 }
144 
145 static struct dma_buf *drm_prime_lookup_buf_by_handle(struct drm_prime_file_private *prime_fpriv,
146 						      uint32_t handle)
147 {
148 	struct rb_node *rb;
149 
150 	rb = prime_fpriv->handles.rb_node;
151 	while (rb) {
152 		struct drm_prime_member *member;
153 
154 		member = rb_entry(rb, struct drm_prime_member, handle_rb);
155 		if (member->handle == handle)
156 			return member->dma_buf;
157 		else if (member->handle < handle)
158 			rb = rb->rb_right;
159 		else
160 			rb = rb->rb_left;
161 	}
162 
163 	return NULL;
164 }
165 
166 static int drm_prime_lookup_buf_handle(struct drm_prime_file_private *prime_fpriv,
167 				       struct dma_buf *dma_buf,
168 				       uint32_t *handle)
169 {
170 	struct rb_node *rb;
171 
172 	rb = prime_fpriv->dmabufs.rb_node;
173 	while (rb) {
174 		struct drm_prime_member *member;
175 
176 		member = rb_entry(rb, struct drm_prime_member, dmabuf_rb);
177 		if (member->dma_buf == dma_buf) {
178 			*handle = member->handle;
179 			return 0;
180 		} else if (member->dma_buf < dma_buf) {
181 			rb = rb->rb_right;
182 		} else {
183 			rb = rb->rb_left;
184 		}
185 	}
186 
187 	return -ENOENT;
188 }
189 
190 void drm_prime_remove_buf_handle_locked(struct drm_prime_file_private *prime_fpriv,
191 					struct dma_buf *dma_buf)
192 {
193 	struct rb_node *rb;
194 
195 	rb = prime_fpriv->dmabufs.rb_node;
196 	while (rb) {
197 		struct drm_prime_member *member;
198 
199 		member = rb_entry(rb, struct drm_prime_member, dmabuf_rb);
200 		if (member->dma_buf == dma_buf) {
201 			rb_erase(&member->handle_rb, &prime_fpriv->handles);
202 			rb_erase(&member->dmabuf_rb, &prime_fpriv->dmabufs);
203 
204 			dma_buf_put(dma_buf);
205 			kfree(member);
206 			return;
207 		} else if (member->dma_buf < dma_buf) {
208 			rb = rb->rb_right;
209 		} else {
210 			rb = rb->rb_left;
211 		}
212 	}
213 }
214 
215 void drm_prime_init_file_private(struct drm_prime_file_private *prime_fpriv)
216 {
217 	rw_init(&prime_fpriv->lock, "primlk");
218 	prime_fpriv->dmabufs = RB_ROOT;
219 	prime_fpriv->handles = RB_ROOT;
220 }
221 
222 void drm_prime_destroy_file_private(struct drm_prime_file_private *prime_fpriv)
223 {
224 	/* by now drm_gem_release should've made sure the list is empty */
225 	WARN_ON(!RB_EMPTY_ROOT(&prime_fpriv->dmabufs));
226 }
227 
228 /**
229  * drm_gem_dmabuf_export - &dma_buf export implementation for GEM
230  * @dev: parent device for the exported dmabuf
231  * @exp_info: the export information used by dma_buf_export()
232  *
233  * This wraps dma_buf_export() for use by generic GEM drivers that are using
234  * drm_gem_dmabuf_release(). In addition to calling dma_buf_export(), we take
235  * a reference to the &drm_device and the exported &drm_gem_object (stored in
236  * &dma_buf_export_info.priv) which is released by drm_gem_dmabuf_release().
237  *
238  * Returns the new dmabuf.
239  */
240 struct dma_buf *drm_gem_dmabuf_export(struct drm_device *dev,
241 				      struct dma_buf_export_info *exp_info)
242 {
243 	struct drm_gem_object *obj = exp_info->priv;
244 	struct dma_buf *dma_buf;
245 
246 	dma_buf = dma_buf_export(exp_info);
247 	if (IS_ERR(dma_buf))
248 		return dma_buf;
249 
250 	drm_dev_get(dev);
251 	drm_gem_object_get(obj);
252 #ifdef __linux__
253 	dma_buf->file->f_mapping = obj->dev->anon_inode->i_mapping;
254 #endif
255 
256 	return dma_buf;
257 }
258 EXPORT_SYMBOL(drm_gem_dmabuf_export);
259 
260 /**
261  * drm_gem_dmabuf_release - &dma_buf release implementation for GEM
262  * @dma_buf: buffer to be released
263  *
264  * Generic release function for dma_bufs exported as PRIME buffers. GEM drivers
265  * must use this in their &dma_buf_ops structure as the release callback.
266  * drm_gem_dmabuf_release() should be used in conjunction with
267  * drm_gem_dmabuf_export().
268  */
269 void drm_gem_dmabuf_release(struct dma_buf *dma_buf)
270 {
271 	struct drm_gem_object *obj = dma_buf->priv;
272 	struct drm_device *dev = obj->dev;
273 
274 	/* drop the reference on the export fd holds */
275 	drm_gem_object_put(obj);
276 
277 	drm_dev_put(dev);
278 }
279 EXPORT_SYMBOL(drm_gem_dmabuf_release);
280 
281 /**
282  * drm_gem_prime_fd_to_handle - PRIME import function for GEM drivers
283  * @dev: dev to export the buffer from
284  * @file_priv: drm file-private structure
285  * @prime_fd: fd id of the dma-buf which should be imported
286  * @handle: pointer to storage for the handle of the imported buffer object
287  *
288  * This is the PRIME import function which must be used mandatorily by GEM
289  * drivers to ensure correct lifetime management of the underlying GEM object.
290  * The actual importing of GEM object from the dma-buf is done through the
291  * &drm_driver.gem_prime_import driver callback.
292  *
293  * Returns 0 on success or a negative error code on failure.
294  */
295 int drm_gem_prime_fd_to_handle(struct drm_device *dev,
296 			       struct drm_file *file_priv, int prime_fd,
297 			       uint32_t *handle)
298 {
299 	struct dma_buf *dma_buf;
300 	struct drm_gem_object *obj;
301 	int ret;
302 
303 	dma_buf = dma_buf_get(prime_fd);
304 	if (IS_ERR(dma_buf))
305 		return PTR_ERR(dma_buf);
306 
307 	mutex_lock(&file_priv->prime.lock);
308 
309 	ret = drm_prime_lookup_buf_handle(&file_priv->prime,
310 			dma_buf, handle);
311 	if (ret == 0)
312 		goto out_put;
313 
314 	/* never seen this one, need to import */
315 	mutex_lock(&dev->object_name_lock);
316 	if (dev->driver->gem_prime_import)
317 		obj = dev->driver->gem_prime_import(dev, dma_buf);
318 	else
319 		obj = drm_gem_prime_import(dev, dma_buf);
320 	if (IS_ERR(obj)) {
321 		ret = PTR_ERR(obj);
322 		goto out_unlock;
323 	}
324 
325 	if (obj->dma_buf) {
326 		WARN_ON(obj->dma_buf != dma_buf);
327 	} else {
328 		obj->dma_buf = dma_buf;
329 		get_dma_buf(dma_buf);
330 	}
331 
332 	/* _handle_create_tail unconditionally unlocks dev->object_name_lock. */
333 	ret = drm_gem_handle_create_tail(file_priv, obj, handle);
334 	drm_gem_object_put(obj);
335 	if (ret)
336 		goto out_put;
337 
338 	ret = drm_prime_add_buf_handle(&file_priv->prime,
339 			dma_buf, *handle);
340 	mutex_unlock(&file_priv->prime.lock);
341 	if (ret)
342 		goto fail;
343 
344 	dma_buf_put(dma_buf);
345 
346 	return 0;
347 
348 fail:
349 	/* hmm, if driver attached, we are relying on the free-object path
350 	 * to detach.. which seems ok..
351 	 */
352 	drm_gem_handle_delete(file_priv, *handle);
353 	dma_buf_put(dma_buf);
354 	return ret;
355 
356 out_unlock:
357 	mutex_unlock(&dev->object_name_lock);
358 out_put:
359 	mutex_unlock(&file_priv->prime.lock);
360 	dma_buf_put(dma_buf);
361 	return ret;
362 }
363 EXPORT_SYMBOL(drm_gem_prime_fd_to_handle);
364 
365 int drm_prime_fd_to_handle_ioctl(struct drm_device *dev, void *data,
366 				 struct drm_file *file_priv)
367 {
368 	struct drm_prime_handle *args = data;
369 
370 	if (!dev->driver->prime_fd_to_handle)
371 		return -ENOSYS;
372 
373 	return dev->driver->prime_fd_to_handle(dev, file_priv,
374 			args->fd, &args->handle);
375 }
376 
377 static struct dma_buf *export_and_register_object(struct drm_device *dev,
378 						  struct drm_gem_object *obj,
379 						  uint32_t flags)
380 {
381 	struct dma_buf *dmabuf;
382 
383 	/* prevent races with concurrent gem_close. */
384 	if (obj->handle_count == 0) {
385 		dmabuf = ERR_PTR(-ENOENT);
386 		return dmabuf;
387 	}
388 
389 	if (obj->funcs && obj->funcs->export)
390 		dmabuf = obj->funcs->export(obj, flags);
391 	else if (dev->driver->gem_prime_export)
392 		dmabuf = dev->driver->gem_prime_export(obj, flags);
393 	else
394 		dmabuf = drm_gem_prime_export(obj, flags);
395 	if (IS_ERR(dmabuf)) {
396 		/* normally the created dma-buf takes ownership of the ref,
397 		 * but if that fails then drop the ref
398 		 */
399 		return dmabuf;
400 	}
401 
402 	/*
403 	 * Note that callers do not need to clean up the export cache
404 	 * since the check for obj->handle_count guarantees that someone
405 	 * will clean it up.
406 	 */
407 	obj->dma_buf = dmabuf;
408 	get_dma_buf(obj->dma_buf);
409 
410 	return dmabuf;
411 }
412 
413 /**
414  * drm_gem_prime_handle_to_fd - PRIME export function for GEM drivers
415  * @dev: dev to export the buffer from
416  * @file_priv: drm file-private structure
417  * @handle: buffer handle to export
418  * @flags: flags like DRM_CLOEXEC
419  * @prime_fd: pointer to storage for the fd id of the create dma-buf
420  *
421  * This is the PRIME export function which must be used mandatorily by GEM
422  * drivers to ensure correct lifetime management of the underlying GEM object.
423  * The actual exporting from GEM object to a dma-buf is done through the
424  * &drm_driver.gem_prime_export driver callback.
425  */
426 int drm_gem_prime_handle_to_fd(struct drm_device *dev,
427 			       struct drm_file *file_priv, uint32_t handle,
428 			       uint32_t flags,
429 			       int *prime_fd)
430 {
431 	struct drm_gem_object *obj;
432 	int ret = 0;
433 	struct dma_buf *dmabuf;
434 
435 	mutex_lock(&file_priv->prime.lock);
436 	obj = drm_gem_object_lookup(file_priv, handle);
437 	if (!obj)  {
438 		ret = -ENOENT;
439 		goto out_unlock;
440 	}
441 
442 	dmabuf = drm_prime_lookup_buf_by_handle(&file_priv->prime, handle);
443 	if (dmabuf) {
444 		get_dma_buf(dmabuf);
445 		goto out_have_handle;
446 	}
447 
448 	mutex_lock(&dev->object_name_lock);
449 #ifdef notyet
450 	/* re-export the original imported object */
451 	if (obj->import_attach) {
452 		dmabuf = obj->import_attach->dmabuf;
453 		get_dma_buf(dmabuf);
454 		goto out_have_obj;
455 	}
456 #endif
457 
458 	if (obj->dma_buf) {
459 		get_dma_buf(obj->dma_buf);
460 		dmabuf = obj->dma_buf;
461 		goto out_have_obj;
462 	}
463 
464 	dmabuf = export_and_register_object(dev, obj, flags);
465 	if (IS_ERR(dmabuf)) {
466 		/* normally the created dma-buf takes ownership of the ref,
467 		 * but if that fails then drop the ref
468 		 */
469 		ret = PTR_ERR(dmabuf);
470 		mutex_unlock(&dev->object_name_lock);
471 		goto out;
472 	}
473 
474 out_have_obj:
475 	/*
476 	 * If we've exported this buffer then cheat and add it to the import list
477 	 * so we get the correct handle back. We must do this under the
478 	 * protection of dev->object_name_lock to ensure that a racing gem close
479 	 * ioctl doesn't miss to remove this buffer handle from the cache.
480 	 */
481 	ret = drm_prime_add_buf_handle(&file_priv->prime,
482 				       dmabuf, handle);
483 	mutex_unlock(&dev->object_name_lock);
484 	if (ret)
485 		goto fail_put_dmabuf;
486 
487 out_have_handle:
488 	ret = dma_buf_fd(dmabuf, flags);
489 	/*
490 	 * We must _not_ remove the buffer from the handle cache since the newly
491 	 * created dma buf is already linked in the global obj->dma_buf pointer,
492 	 * and that is invariant as long as a userspace gem handle exists.
493 	 * Closing the handle will clean out the cache anyway, so we don't leak.
494 	 */
495 	if (ret < 0) {
496 		goto fail_put_dmabuf;
497 	} else {
498 		*prime_fd = ret;
499 		ret = 0;
500 	}
501 
502 	goto out;
503 
504 fail_put_dmabuf:
505 	dma_buf_put(dmabuf);
506 out:
507 	drm_gem_object_put(obj);
508 out_unlock:
509 	mutex_unlock(&file_priv->prime.lock);
510 
511 	return ret;
512 }
513 EXPORT_SYMBOL(drm_gem_prime_handle_to_fd);
514 
515 int drm_prime_handle_to_fd_ioctl(struct drm_device *dev, void *data,
516 				 struct drm_file *file_priv)
517 {
518 	struct drm_prime_handle *args = data;
519 
520 	if (!dev->driver->prime_handle_to_fd)
521 		return -ENOSYS;
522 
523 	/* check flags are valid */
524 	if (args->flags & ~(DRM_CLOEXEC | DRM_RDWR))
525 		return -EINVAL;
526 
527 	return dev->driver->prime_handle_to_fd(dev, file_priv,
528 			args->handle, args->flags, &args->fd);
529 }
530 
531 /**
532  * DOC: PRIME Helpers
533  *
534  * Drivers can implement &drm_gem_object_funcs.export and
535  * &drm_driver.gem_prime_import in terms of simpler APIs by using the helper
536  * functions drm_gem_prime_export() and drm_gem_prime_import(). These functions
537  * implement dma-buf support in terms of some lower-level helpers, which are
538  * again exported for drivers to use individually:
539  *
540  * Exporting buffers
541  * ~~~~~~~~~~~~~~~~~
542  *
543  * Optional pinning of buffers is handled at dma-buf attach and detach time in
544  * drm_gem_map_attach() and drm_gem_map_detach(). Backing storage itself is
545  * handled by drm_gem_map_dma_buf() and drm_gem_unmap_dma_buf(), which relies on
546  * &drm_gem_object_funcs.get_sg_table.
547  *
548  * For kernel-internal access there's drm_gem_dmabuf_vmap() and
549  * drm_gem_dmabuf_vunmap(). Userspace mmap support is provided by
550  * drm_gem_dmabuf_mmap().
551  *
552  * Note that these export helpers can only be used if the underlying backing
553  * storage is fully coherent and either permanently pinned, or it is safe to pin
554  * it indefinitely.
555  *
556  * FIXME: The underlying helper functions are named rather inconsistently.
557  *
558  * Exporting buffers
559  * ~~~~~~~~~~~~~~~~~
560  *
561  * Importing dma-bufs using drm_gem_prime_import() relies on
562  * &drm_driver.gem_prime_import_sg_table.
563  *
564  * Note that similarly to the export helpers this permanently pins the
565  * underlying backing storage. Which is ok for scanout, but is not the best
566  * option for sharing lots of buffers for rendering.
567  */
568 
569 /**
570  * drm_gem_map_attach - dma_buf attach implementation for GEM
571  * @dma_buf: buffer to attach device to
572  * @attach: buffer attachment data
573  *
574  * Calls &drm_gem_object_funcs.pin for device specific handling. This can be
575  * used as the &dma_buf_ops.attach callback. Must be used together with
576  * drm_gem_map_detach().
577  *
578  * Returns 0 on success, negative error code on failure.
579  */
580 int drm_gem_map_attach(struct dma_buf *dma_buf,
581 		       struct dma_buf_attachment *attach)
582 {
583 	struct drm_gem_object *obj = dma_buf->priv;
584 
585 	return drm_gem_pin(obj);
586 }
587 EXPORT_SYMBOL(drm_gem_map_attach);
588 
589 /**
590  * drm_gem_map_detach - dma_buf detach implementation for GEM
591  * @dma_buf: buffer to detach from
592  * @attach: attachment to be detached
593  *
594  * Calls &drm_gem_object_funcs.pin for device specific handling.  Cleans up
595  * &dma_buf_attachment from drm_gem_map_attach(). This can be used as the
596  * &dma_buf_ops.detach callback.
597  */
598 void drm_gem_map_detach(struct dma_buf *dma_buf,
599 			struct dma_buf_attachment *attach)
600 {
601 	struct drm_gem_object *obj = dma_buf->priv;
602 
603 	drm_gem_unpin(obj);
604 }
605 EXPORT_SYMBOL(drm_gem_map_detach);
606 
607 #ifdef notyet
608 
609 /**
610  * drm_gem_map_dma_buf - map_dma_buf implementation for GEM
611  * @attach: attachment whose scatterlist is to be returned
612  * @dir: direction of DMA transfer
613  *
614  * Calls &drm_gem_object_funcs.get_sg_table and then maps the scatterlist. This
615  * can be used as the &dma_buf_ops.map_dma_buf callback. Should be used together
616  * with drm_gem_unmap_dma_buf().
617  *
618  * Returns:sg_table containing the scatterlist to be returned; returns ERR_PTR
619  * on error. May return -EINTR if it is interrupted by a signal.
620  */
621 struct sg_table *drm_gem_map_dma_buf(struct dma_buf_attachment *attach,
622 				     enum dma_data_direction dir)
623 {
624 	struct drm_gem_object *obj = attach->dmabuf->priv;
625 	struct sg_table *sgt;
626 	int ret;
627 
628 	if (WARN_ON(dir == DMA_NONE))
629 		return ERR_PTR(-EINVAL);
630 
631 	if (obj->funcs)
632 		sgt = obj->funcs->get_sg_table(obj);
633 	else
634 		sgt = obj->dev->driver->gem_prime_get_sg_table(obj);
635 
636 	ret = dma_map_sgtable(attach->dev, sgt, dir,
637 			      DMA_ATTR_SKIP_CPU_SYNC);
638 	if (ret) {
639 		sg_free_table(sgt);
640 		kfree(sgt);
641 		sgt = ERR_PTR(ret);
642 	}
643 
644 	return sgt;
645 }
646 EXPORT_SYMBOL(drm_gem_map_dma_buf);
647 
648 /**
649  * drm_gem_unmap_dma_buf - unmap_dma_buf implementation for GEM
650  * @attach: attachment to unmap buffer from
651  * @sgt: scatterlist info of the buffer to unmap
652  * @dir: direction of DMA transfer
653  *
654  * This can be used as the &dma_buf_ops.unmap_dma_buf callback.
655  */
656 void drm_gem_unmap_dma_buf(struct dma_buf_attachment *attach,
657 			   struct sg_table *sgt,
658 			   enum dma_data_direction dir)
659 {
660 	if (!sgt)
661 		return;
662 
663 	dma_unmap_sgtable(attach->dev, sgt, dir, DMA_ATTR_SKIP_CPU_SYNC);
664 	sg_free_table(sgt);
665 	kfree(sgt);
666 }
667 EXPORT_SYMBOL(drm_gem_unmap_dma_buf);
668 
669 #endif /* notyet */
670 
671 /**
672  * drm_gem_dmabuf_vmap - dma_buf vmap implementation for GEM
673  * @dma_buf: buffer to be mapped
674  *
675  * Sets up a kernel virtual mapping. This can be used as the &dma_buf_ops.vmap
676  * callback. Calls into &drm_gem_object_funcs.vmap for device specific handling.
677  *
678  * Returns the kernel virtual address or NULL on failure.
679  */
680 void *drm_gem_dmabuf_vmap(struct dma_buf *dma_buf)
681 {
682 	struct drm_gem_object *obj = dma_buf->priv;
683 	void *vaddr;
684 
685 	vaddr = drm_gem_vmap(obj);
686 	if (IS_ERR(vaddr))
687 		vaddr = NULL;
688 
689 	return vaddr;
690 }
691 EXPORT_SYMBOL(drm_gem_dmabuf_vmap);
692 
693 /**
694  * drm_gem_dmabuf_vunmap - dma_buf vunmap implementation for GEM
695  * @dma_buf: buffer to be unmapped
696  * @vaddr: the virtual address of the buffer
697  *
698  * Releases a kernel virtual mapping. This can be used as the
699  * &dma_buf_ops.vunmap callback. Calls into &drm_gem_object_funcs.vunmap for device specific handling.
700  */
701 void drm_gem_dmabuf_vunmap(struct dma_buf *dma_buf, void *vaddr)
702 {
703 	struct drm_gem_object *obj = dma_buf->priv;
704 
705 	drm_gem_vunmap(obj, vaddr);
706 }
707 EXPORT_SYMBOL(drm_gem_dmabuf_vunmap);
708 
709 #ifdef notyet
710 
711 /**
712  * drm_gem_prime_mmap - PRIME mmap function for GEM drivers
713  * @obj: GEM object
714  * @vma: Virtual address range
715  *
716  * This function sets up a userspace mapping for PRIME exported buffers using
717  * the same codepath that is used for regular GEM buffer mapping on the DRM fd.
718  * The fake GEM offset is added to vma->vm_pgoff and &drm_driver->fops->mmap is
719  * called to set up the mapping.
720  *
721  * Drivers can use this as their &drm_driver.gem_prime_mmap callback.
722  */
723 int drm_gem_prime_mmap(struct drm_gem_object *obj, struct vm_area_struct *vma)
724 {
725 	struct drm_file *priv;
726 	struct file *fil;
727 	int ret;
728 
729 	/* Add the fake offset */
730 	vma->vm_pgoff += drm_vma_node_start(&obj->vma_node);
731 
732 	if (obj->funcs && obj->funcs->mmap) {
733 		ret = obj->funcs->mmap(obj, vma);
734 		if (ret)
735 			return ret;
736 		vma->vm_private_data = obj;
737 		drm_gem_object_get(obj);
738 		return 0;
739 	}
740 
741 	priv = kzalloc(sizeof(*priv), GFP_KERNEL);
742 	fil = kzalloc(sizeof(*fil), GFP_KERNEL);
743 	if (!priv || !fil) {
744 		ret = -ENOMEM;
745 		goto out;
746 	}
747 
748 	/* Used by drm_gem_mmap() to lookup the GEM object */
749 	priv->minor = obj->dev->primary;
750 	fil->private_data = priv;
751 
752 	ret = drm_vma_node_allow(&obj->vma_node, priv);
753 	if (ret)
754 		goto out;
755 
756 	ret = obj->dev->driver->fops->mmap(fil, vma);
757 
758 	drm_vma_node_revoke(&obj->vma_node, priv);
759 out:
760 	kfree(priv);
761 	kfree(fil);
762 
763 	return ret;
764 }
765 EXPORT_SYMBOL(drm_gem_prime_mmap);
766 
767 /**
768  * drm_gem_dmabuf_mmap - dma_buf mmap implementation for GEM
769  * @dma_buf: buffer to be mapped
770  * @vma: virtual address range
771  *
772  * Provides memory mapping for the buffer. This can be used as the
773  * &dma_buf_ops.mmap callback. It just forwards to &drm_driver.gem_prime_mmap,
774  * which should be set to drm_gem_prime_mmap().
775  *
776  * FIXME: There's really no point to this wrapper, drivers which need anything
777  * else but drm_gem_prime_mmap can roll their own &dma_buf_ops.mmap callback.
778  *
779  * Returns 0 on success or a negative error code on failure.
780  */
781 int drm_gem_dmabuf_mmap(struct dma_buf *dma_buf, struct vm_area_struct *vma)
782 {
783 	struct drm_gem_object *obj = dma_buf->priv;
784 	struct drm_device *dev = obj->dev;
785 
786 	if (!dev->driver->gem_prime_mmap)
787 		return -ENOSYS;
788 
789 	return dev->driver->gem_prime_mmap(obj, vma);
790 }
791 EXPORT_SYMBOL(drm_gem_dmabuf_mmap);
792 
793 #endif /* notyet */
794 
795 static const struct dma_buf_ops drm_gem_prime_dmabuf_ops =  {
796 #ifdef notyet
797 	.cache_sgt_mapping = true,
798 	.attach = drm_gem_map_attach,
799 	.detach = drm_gem_map_detach,
800 	.map_dma_buf = drm_gem_map_dma_buf,
801 	.unmap_dma_buf = drm_gem_unmap_dma_buf,
802 #endif
803 	.release = drm_gem_dmabuf_release,
804 #ifdef notyet
805 	.mmap = drm_gem_dmabuf_mmap,
806 	.vmap = drm_gem_dmabuf_vmap,
807 	.vunmap = drm_gem_dmabuf_vunmap,
808 #endif
809 };
810 
811 /**
812  * drm_prime_pages_to_sg - converts a page array into an sg list
813  * @dev: DRM device
814  * @pages: pointer to the array of page pointers to convert
815  * @nr_pages: length of the page vector
816  *
817  * This helper creates an sg table object from a set of pages
818  * the driver is responsible for mapping the pages into the
819  * importers address space for use with dma_buf itself.
820  *
821  * This is useful for implementing &drm_gem_object_funcs.get_sg_table.
822  */
823 struct sg_table *drm_prime_pages_to_sg(struct drm_device *dev,
824 				       struct vm_page **pages, unsigned int nr_pages)
825 {
826 	STUB();
827 	return NULL;
828 #ifdef notyet
829 	struct sg_table *sg;
830 	struct scatterlist *sge;
831 	size_t max_segment = 0;
832 
833 	sg = kmalloc(sizeof(struct sg_table), GFP_KERNEL);
834 	if (!sg)
835 		return ERR_PTR(-ENOMEM);
836 
837 	if (dev)
838 		max_segment = dma_max_mapping_size(dev->dev);
839 	if (max_segment == 0 || max_segment > SCATTERLIST_MAX_SEGMENT)
840 		max_segment = SCATTERLIST_MAX_SEGMENT;
841 	sge = __sg_alloc_table_from_pages(sg, pages, nr_pages, 0,
842 					  nr_pages << PAGE_SHIFT,
843 					  max_segment,
844 					  NULL, 0, GFP_KERNEL);
845 	if (IS_ERR(sge)) {
846 		kfree(sg);
847 		sg = ERR_CAST(sge);
848 	}
849 	return sg;
850 #endif
851 }
852 EXPORT_SYMBOL(drm_prime_pages_to_sg);
853 
854 /**
855  * drm_prime_get_contiguous_size - returns the contiguous size of the buffer
856  * @sgt: sg_table describing the buffer to check
857  *
858  * This helper calculates the contiguous size in the DMA address space
859  * of the the buffer described by the provided sg_table.
860  *
861  * This is useful for implementing
862  * &drm_gem_object_funcs.gem_prime_import_sg_table.
863  */
864 unsigned long drm_prime_get_contiguous_size(struct sg_table *sgt)
865 {
866 	STUB();
867 	return 0;
868 #ifdef notyet
869 	dma_addr_t expected = sg_dma_address(sgt->sgl);
870 	struct scatterlist *sg;
871 	unsigned long size = 0;
872 	int i;
873 
874 	for_each_sgtable_dma_sg(sgt, sg, i) {
875 		unsigned int len = sg_dma_len(sg);
876 
877 		if (!len)
878 			break;
879 		if (sg_dma_address(sg) != expected)
880 			break;
881 		expected += len;
882 		size += len;
883 	}
884 	return size;
885 #endif
886 }
887 EXPORT_SYMBOL(drm_prime_get_contiguous_size);
888 
889 /**
890  * drm_gem_prime_export - helper library implementation of the export callback
891  * @obj: GEM object to export
892  * @flags: flags like DRM_CLOEXEC and DRM_RDWR
893  *
894  * This is the implementation of the &drm_gem_object_funcs.export functions for GEM drivers
895  * using the PRIME helpers. It is used as the default in
896  * drm_gem_prime_handle_to_fd().
897  */
898 struct dma_buf *drm_gem_prime_export(struct drm_gem_object *obj,
899 				     int flags)
900 {
901 	struct drm_device *dev = obj->dev;
902 	struct dma_buf_export_info exp_info = {
903 #ifdef __linux__
904 		.exp_name = KBUILD_MODNAME, /* white lie for debug */
905 		.owner = dev->driver->fops->owner,
906 #endif
907 		.ops = &drm_gem_prime_dmabuf_ops,
908 		.size = obj->size,
909 		.flags = flags,
910 		.priv = obj,
911 		.resv = obj->resv,
912 	};
913 
914 	return drm_gem_dmabuf_export(dev, &exp_info);
915 }
916 EXPORT_SYMBOL(drm_gem_prime_export);
917 
918 /**
919  * drm_gem_prime_import_dev - core implementation of the import callback
920  * @dev: drm_device to import into
921  * @dma_buf: dma-buf object to import
922  * @attach_dev: struct device to dma_buf attach
923  *
924  * This is the core of drm_gem_prime_import(). It's designed to be called by
925  * drivers who want to use a different device structure than &drm_device.dev for
926  * attaching via dma_buf. This function calls
927  * &drm_driver.gem_prime_import_sg_table internally.
928  *
929  * Drivers must arrange to call drm_prime_gem_destroy() from their
930  * &drm_gem_object_funcs.free hook when using this function.
931  */
932 struct drm_gem_object *drm_gem_prime_import_dev(struct drm_device *dev,
933 					    struct dma_buf *dma_buf,
934 					    struct device *attach_dev)
935 {
936 	struct dma_buf_attachment *attach;
937 #ifdef notyet
938 	struct sg_table *sgt;
939 #endif
940 	struct drm_gem_object *obj;
941 	int ret;
942 
943 	if (dma_buf->ops == &drm_gem_prime_dmabuf_ops) {
944 		obj = dma_buf->priv;
945 		if (obj->dev == dev) {
946 			/*
947 			 * Importing dmabuf exported from out own gem increases
948 			 * refcount on gem itself instead of f_count of dmabuf.
949 			 */
950 			drm_gem_object_get(obj);
951 			return obj;
952 		}
953 	}
954 
955 #ifdef notyet
956 	if (!dev->driver->gem_prime_import_sg_table)
957 		return ERR_PTR(-EINVAL);
958 #endif
959 
960 	attach = dma_buf_attach(dma_buf, attach_dev);
961 	if (IS_ERR(attach))
962 		return ERR_CAST(attach);
963 
964 #ifdef notyet
965 	get_dma_buf(dma_buf);
966 
967 	sgt = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
968 	if (IS_ERR(sgt)) {
969 		ret = PTR_ERR(sgt);
970 		goto fail_detach;
971 	}
972 
973 	obj = dev->driver->gem_prime_import_sg_table(dev, attach, sgt);
974 	if (IS_ERR(obj)) {
975 		ret = PTR_ERR(obj);
976 		goto fail_unmap;
977 	}
978 
979 	obj->import_attach = attach;
980 	obj->resv = dma_buf->resv;
981 
982 	return obj;
983 
984 fail_unmap:
985 	dma_buf_unmap_attachment(attach, sgt, DMA_BIDIRECTIONAL);
986 fail_detach:
987 	dma_buf_detach(dma_buf, attach);
988 	dma_buf_put(dma_buf);
989 
990 	return ERR_PTR(ret);
991 #else
992 	ret = 0;
993 	panic(__func__);
994 #endif
995 }
996 EXPORT_SYMBOL(drm_gem_prime_import_dev);
997 
998 /**
999  * drm_gem_prime_import - helper library implementation of the import callback
1000  * @dev: drm_device to import into
1001  * @dma_buf: dma-buf object to import
1002  *
1003  * This is the implementation of the gem_prime_import functions for GEM drivers
1004  * using the PRIME helpers. Drivers can use this as their
1005  * &drm_driver.gem_prime_import implementation. It is used as the default
1006  * implementation in drm_gem_prime_fd_to_handle().
1007  *
1008  * Drivers must arrange to call drm_prime_gem_destroy() from their
1009  * &drm_gem_object_funcs.free hook when using this function.
1010  */
1011 struct drm_gem_object *drm_gem_prime_import(struct drm_device *dev,
1012 					    struct dma_buf *dma_buf)
1013 {
1014 	return drm_gem_prime_import_dev(dev, dma_buf, dev->dev);
1015 }
1016 EXPORT_SYMBOL(drm_gem_prime_import);
1017 
1018 /**
1019  * drm_prime_sg_to_page_addr_arrays - convert an sg table into a page array
1020  * @sgt: scatter-gather table to convert
1021  * @pages: optional array of page pointers to store the page array in
1022  * @addrs: optional array to store the dma bus address of each page
1023  * @max_entries: size of both the passed-in arrays
1024  *
1025  * Exports an sg table into an array of pages and addresses. This is currently
1026  * required by the TTM driver in order to do correct fault handling.
1027  *
1028  * Drivers can use this in their &drm_driver.gem_prime_import_sg_table
1029  * implementation.
1030  */
1031 int drm_prime_sg_to_page_addr_arrays(struct sg_table *sgt, struct vm_page **pages,
1032 				     dma_addr_t *addrs, int max_entries)
1033 {
1034 	STUB();
1035 	return -ENOSYS;
1036 #ifdef notyet
1037 	struct sg_dma_page_iter dma_iter;
1038 	struct sg_page_iter page_iter;
1039 	struct vm_page **p = pages;
1040 	dma_addr_t *a = addrs;
1041 
1042 	if (pages) {
1043 		for_each_sgtable_page(sgt, &page_iter, 0) {
1044 			if (WARN_ON(p - pages >= max_entries))
1045 				return -1;
1046 			*p++ = sg_page_iter_page(&page_iter);
1047 		}
1048 	}
1049 	if (addrs) {
1050 		for_each_sgtable_dma_page(sgt, &dma_iter, 0) {
1051 			if (WARN_ON(a - addrs >= max_entries))
1052 				return -1;
1053 			*a++ = sg_page_iter_dma_address(&dma_iter);
1054 		}
1055 	}
1056 
1057 	return 0;
1058 #endif
1059 }
1060 EXPORT_SYMBOL(drm_prime_sg_to_page_addr_arrays);
1061 
1062 /**
1063  * drm_prime_gem_destroy - helper to clean up a PRIME-imported GEM object
1064  * @obj: GEM object which was created from a dma-buf
1065  * @sg: the sg-table which was pinned at import time
1066  *
1067  * This is the cleanup functions which GEM drivers need to call when they use
1068  * drm_gem_prime_import() or drm_gem_prime_import_dev() to import dma-bufs.
1069  */
1070 void drm_prime_gem_destroy(struct drm_gem_object *obj, struct sg_table *sg)
1071 {
1072 	STUB();
1073 #ifdef notyet
1074 	struct dma_buf_attachment *attach;
1075 	struct dma_buf *dma_buf;
1076 
1077 	attach = obj->import_attach;
1078 	if (sg)
1079 		dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
1080 	dma_buf = attach->dmabuf;
1081 	dma_buf_detach(attach->dmabuf, attach);
1082 	/* remove the reference */
1083 	dma_buf_put(dma_buf);
1084 #endif
1085 }
1086 EXPORT_SYMBOL(drm_prime_gem_destroy);
1087