2 * Copyright (C) Fuzhou Rockchip Electronics Co.Ltd
3 * Author:Mark Yao <mark.yao@rock-chips.com>
5 * This software is licensed under the terms of the GNU General Public
6 * License version 2, as published by the Free Software Foundation, and
7 * may be copied, distributed, and modified under those terms.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
17 #include <drm/drm_gem.h>
18 #include <drm/drm_sync_helper.h>
19 #include <drm/drm_vma_manager.h>
20 #include <drm/rockchip_drm.h>
22 #include <linux/completion.h>
23 #include <linux/dma-attrs.h>
24 #include <linux/dma-buf.h>
25 #include <linux/reservation.h>
27 #include "rockchip_drm_drv.h"
28 #include "rockchip_drm_gem.h"
30 static int rockchip_gem_alloc_buf(struct rockchip_gem_object *rk_obj,
33 struct drm_gem_object *obj = &rk_obj->base;
34 struct drm_device *drm = obj->dev;
36 init_dma_attrs(&rk_obj->dma_attrs);
37 dma_set_attr(DMA_ATTR_WRITE_COMBINE, &rk_obj->dma_attrs);
40 dma_set_attr(DMA_ATTR_NO_KERNEL_MAPPING, &rk_obj->dma_attrs);
42 rk_obj->kvaddr = dma_alloc_attrs(drm->dev, obj->size,
43 &rk_obj->dma_addr, GFP_KERNEL,
45 if (!rk_obj->kvaddr) {
46 DRM_ERROR("failed to allocate %zu byte dma buffer", obj->size);
53 static void rockchip_gem_free_buf(struct rockchip_gem_object *rk_obj)
55 struct drm_gem_object *obj = &rk_obj->base;
56 struct drm_device *drm = obj->dev;
58 dma_free_attrs(drm->dev, obj->size, rk_obj->kvaddr, rk_obj->dma_addr,
62 static int rockchip_drm_gem_object_mmap(struct drm_gem_object *obj,
63 struct vm_area_struct *vma)
67 struct rockchip_gem_object *rk_obj = to_rockchip_obj(obj);
68 struct drm_device *drm = obj->dev;
71 * dma_alloc_attrs() allocated a struct page table for rk_obj, so clear
72 * VM_PFNMAP flag that was set by drm_gem_mmap_obj()/drm_gem_mmap().
74 vma->vm_flags &= ~VM_PFNMAP;
77 ret = dma_mmap_attrs(drm->dev, vma, rk_obj->kvaddr, rk_obj->dma_addr,
78 obj->size, &rk_obj->dma_attrs);
80 drm_gem_vm_close(vma);
85 int rockchip_gem_mmap_buf(struct drm_gem_object *obj,
86 struct vm_area_struct *vma)
90 ret = drm_gem_mmap_obj(obj, obj->size, vma);
94 return rockchip_drm_gem_object_mmap(obj, vma);
97 /* drm driver mmap file operations */
98 int rockchip_gem_mmap(struct file *filp, struct vm_area_struct *vma)
100 struct drm_gem_object *obj;
103 ret = drm_gem_mmap(filp, vma);
107 obj = vma->vm_private_data;
109 return rockchip_drm_gem_object_mmap(obj, vma);
112 struct drm_gem_object *
113 rockchip_gem_prime_import_sg_table(struct drm_device *drm,
114 struct dma_buf_attachment *attach,
115 struct sg_table *sgt)
117 struct rockchip_gem_object *rk_obj;
118 struct drm_gem_object *obj;
120 rk_obj = kzalloc(sizeof(*rk_obj), GFP_KERNEL);
122 return ERR_PTR(-ENOMEM);
126 drm_gem_private_object_init(drm, obj, attach->dmabuf->size);
128 rk_obj->dma_addr = sg_dma_address(sgt->sgl);
130 sg_dma_len(sgt->sgl) = obj->size;
135 struct rockchip_gem_object *
136 rockchip_gem_create_object(struct drm_device *drm, unsigned int size,
139 struct rockchip_gem_object *rk_obj;
140 struct drm_gem_object *obj;
143 size = round_up(size, PAGE_SIZE);
145 rk_obj = kzalloc(sizeof(*rk_obj), GFP_KERNEL);
147 return ERR_PTR(-ENOMEM);
151 drm_gem_private_object_init(drm, obj, size);
153 ret = rockchip_gem_alloc_buf(rk_obj, alloc_kmap);
155 goto err_free_rk_obj;
165 * rockchip_gem_free_object - (struct drm_driver)->gem_free_object callback
168 void rockchip_gem_free_object(struct drm_gem_object *obj)
170 struct rockchip_gem_object *rk_obj;
172 rk_obj = to_rockchip_obj(obj);
174 if (obj->import_attach) {
175 drm_prime_gem_destroy(obj, rk_obj->sgt);
177 drm_gem_free_mmap_offset(obj);
178 rockchip_gem_free_buf(rk_obj);
181 #ifdef CONFIG_DRM_DMA_SYNC
182 drm_fence_signal_and_put(&rk_obj->acquire_fence);
189 * rockchip_gem_create_with_handle - allocate an object with the given
190 * size and create a gem handle on it
192 * returns a struct rockchip_gem_object* on success or ERR_PTR values
195 static struct rockchip_gem_object *
196 rockchip_gem_create_with_handle(struct drm_file *file_priv,
197 struct drm_device *drm, unsigned int size,
198 unsigned int *handle)
200 struct rockchip_gem_object *rk_obj;
201 struct drm_gem_object *obj;
204 rk_obj = rockchip_gem_create_object(drm, size, false);
206 return ERR_CAST(rk_obj);
211 * allocate a id of idr table where the obj is registered
212 * and handle has the id what user can see.
214 ret = drm_gem_handle_create(file_priv, obj, handle);
216 goto err_handle_create;
218 /* drop reference from allocate - handle holds it now. */
219 drm_gem_object_unreference_unlocked(obj);
224 rockchip_gem_free_object(obj);
229 int rockchip_gem_dumb_map_offset(struct drm_file *file_priv,
230 struct drm_device *dev, uint32_t handle,
233 struct drm_gem_object *obj;
236 obj = drm_gem_object_lookup(dev, file_priv, handle);
238 DRM_ERROR("failed to lookup gem object.\n");
242 ret = drm_gem_create_mmap_offset(obj);
246 *offset = drm_vma_node_offset_addr(&obj->vma_node);
247 DRM_DEBUG_KMS("offset = 0x%llx\n", *offset);
250 drm_gem_object_unreference_unlocked(obj);
256 * rockchip_gem_dumb_create - (struct drm_driver)->dumb_create callback
259 * This aligns the pitch and size arguments to the minimum required. wrap
260 * this into your own function if you need bigger alignment.
262 int rockchip_gem_dumb_create(struct drm_file *file_priv,
263 struct drm_device *dev,
264 struct drm_mode_create_dumb *args)
266 struct rockchip_gem_object *rk_obj;
267 int min_pitch = DIV_ROUND_UP(args->width * args->bpp, 8);
270 * align to 64 bytes since Mali requires it.
272 args->pitch = ALIGN(min_pitch, 64);
273 args->size = args->pitch * args->height;
275 rk_obj = rockchip_gem_create_with_handle(file_priv, dev, args->size,
278 return PTR_ERR_OR_ZERO(rk_obj);
281 int rockchip_gem_map_offset_ioctl(struct drm_device *drm, void *data,
282 struct drm_file *file_priv)
284 struct drm_rockchip_gem_map_off *args = data;
286 return rockchip_gem_dumb_map_offset(file_priv, drm, args->handle,
290 int rockchip_gem_create_ioctl(struct drm_device *dev, void *data,
291 struct drm_file *file_priv)
293 struct drm_rockchip_gem_create *args = data;
294 struct rockchip_gem_object *rk_obj;
296 rk_obj = rockchip_gem_create_with_handle(file_priv, dev, args->size,
298 return PTR_ERR_OR_ZERO(rk_obj);
301 static struct reservation_object *drm_gem_get_resv(struct drm_gem_object *gem)
303 struct dma_buf *dma_buf = gem->dma_buf;
304 return dma_buf ? dma_buf->resv : NULL;
307 #ifdef CONFIG_DRM_DMA_SYNC
308 static void rockchip_gem_acquire_complete(struct drm_reservation_cb *rcb,
311 struct completion *compl = context;
315 static int rockchip_gem_acquire(struct drm_device *dev,
316 struct rockchip_gem_object *rockchip_gem_obj,
320 struct rockchip_drm_private *dev_priv = dev->dev_private;
321 struct reservation_object *resv =
322 drm_gem_get_resv(&rockchip_gem_obj->base);
324 struct drm_reservation_cb rcb;
325 DECLARE_COMPLETION_ONSTACK(compl);
331 !rockchip_gem_obj->acquire_exclusive &&
332 rockchip_gem_obj->acquire_fence) {
333 atomic_inc(&rockchip_gem_obj->acquire_shared_count);
337 fence = drm_sw_fence_new(dev_priv->cpu_fence_context,
338 atomic_add_return(1, &dev_priv->cpu_fence_seqno));
340 ret = PTR_ERR(fence);
341 DRM_ERROR("Failed to create acquire fence %d.\n", ret);
344 ww_mutex_lock(&resv->lock, NULL);
346 ret = reservation_object_reserve_shared(resv);
348 DRM_ERROR("Failed to reserve space for shared fence %d.\n",
353 drm_reservation_cb_init(&rcb, rockchip_gem_acquire_complete, &compl);
354 ret = drm_reservation_cb_add(&rcb, resv, exclusive);
356 DRM_ERROR("Failed to add reservation to callback %d.\n", ret);
359 drm_reservation_cb_done(&rcb);
361 reservation_object_add_excl_fence(resv, fence);
363 reservation_object_add_shared_fence(resv, fence);
365 ww_mutex_unlock(&resv->lock);
366 mutex_unlock(&dev->struct_mutex);
367 ret = wait_for_completion_interruptible(&compl);
368 mutex_lock(&dev->struct_mutex);
370 DRM_ERROR("Failed wait for reservation callback %d.\n", ret);
371 drm_reservation_cb_fini(&rcb);
372 /* somebody else may be already waiting on it */
373 drm_fence_signal_and_put(&fence);
376 rockchip_gem_obj->acquire_fence = fence;
377 rockchip_gem_obj->acquire_exclusive = exclusive;
378 atomic_set(&rockchip_gem_obj->acquire_shared_count, 1);
382 ww_mutex_unlock(&resv->lock);
387 static void rockchip_gem_release(struct rockchip_gem_object *rockchip_gem_obj)
389 BUG_ON(!rockchip_gem_obj->acquire_fence);
390 if (atomic_sub_and_test(1,
391 &rockchip_gem_obj->acquire_shared_count))
392 drm_fence_signal_and_put(&rockchip_gem_obj->acquire_fence);
396 int rockchip_gem_cpu_acquire_ioctl(struct drm_device *dev, void *data,
397 struct drm_file *file)
399 struct drm_rockchip_gem_cpu_acquire *args = data;
400 struct rockchip_drm_file_private *file_priv = file->driver_priv;
401 struct drm_gem_object *obj;
402 struct rockchip_gem_object *rockchip_gem_obj;
403 struct rockchip_gem_object_node *gem_node;
406 DRM_DEBUG_KMS("[BO:%u] flags: 0x%x\n", args->handle, args->flags);
408 mutex_lock(&dev->struct_mutex);
410 obj = drm_gem_object_lookup(dev, file, args->handle);
412 DRM_ERROR("failed to lookup gem object.\n");
417 rockchip_gem_obj = to_rockchip_obj(obj);
419 if (!drm_gem_get_resv(&rockchip_gem_obj->base)) {
420 /* If there is no reservation object present, there is no
421 * cross-process/cross-device sharing and sync is unnecessary.
427 #ifdef CONFIG_DRM_DMA_SYNC
428 ret = rockchip_gem_acquire(dev, rockchip_gem_obj,
429 args->flags & DRM_ROCKCHIP_GEM_CPU_ACQUIRE_EXCLUSIVE);
434 gem_node = kzalloc(sizeof(*gem_node), GFP_KERNEL);
436 DRM_ERROR("Failed to allocate rockchip_drm_gem_obj_node.\n");
441 gem_node->rockchip_gem_obj = rockchip_gem_obj;
442 list_add(&gem_node->list, &file_priv->gem_cpu_acquire_list);
443 mutex_unlock(&dev->struct_mutex);
447 #ifdef CONFIG_DRM_DMA_SYNC
448 rockchip_gem_release(rockchip_gem_obj);
451 drm_gem_object_unreference(obj);
454 mutex_unlock(&dev->struct_mutex);
458 int rockchip_gem_cpu_release_ioctl(struct drm_device *dev, void *data,
459 struct drm_file *file)
461 struct drm_rockchip_gem_cpu_release *args = data;
462 struct rockchip_drm_file_private *file_priv = file->driver_priv;
463 struct drm_gem_object *obj;
464 struct rockchip_gem_object *rockchip_gem_obj;
465 struct list_head *cur;
468 DRM_DEBUG_KMS("[BO:%u]\n", args->handle);
470 mutex_lock(&dev->struct_mutex);
472 obj = drm_gem_object_lookup(dev, file, args->handle);
474 DRM_ERROR("failed to lookup gem object.\n");
479 rockchip_gem_obj = to_rockchip_obj(obj);
481 if (!drm_gem_get_resv(&rockchip_gem_obj->base)) {
482 /* If there is no reservation object present, there is no
483 * cross-process/cross-device sharing and sync is unnecessary.
489 list_for_each(cur, &file_priv->gem_cpu_acquire_list) {
490 struct rockchip_gem_object_node *node = list_entry(
491 cur, struct rockchip_gem_object_node, list);
492 if (node->rockchip_gem_obj == rockchip_gem_obj)
495 if (cur == &file_priv->gem_cpu_acquire_list) {
496 DRM_ERROR("gem object not acquired for current process.\n");
501 #ifdef CONFIG_DRM_DMA_SYNC
502 rockchip_gem_release(rockchip_gem_obj);
506 kfree(list_entry(cur, struct rockchip_gem_object_node, list));
507 /* unreference for the reference held since cpu_acquire_ioctl */
508 drm_gem_object_unreference(obj);
512 /* unreference for the reference from drm_gem_object_lookup() */
513 drm_gem_object_unreference(obj);
516 mutex_unlock(&dev->struct_mutex);
521 * Allocate a sg_table for this GEM object.
522 * Note: Both the table's contents, and the sg_table itself must be freed by
524 * Returns a pointer to the newly allocated sg_table, or an ERR_PTR() error.
526 struct sg_table *rockchip_gem_prime_get_sg_table(struct drm_gem_object *obj)
528 struct rockchip_gem_object *rk_obj = to_rockchip_obj(obj);
529 struct drm_device *drm = obj->dev;
530 struct sg_table *sgt;
533 sgt = kzalloc(sizeof(*sgt), GFP_KERNEL);
535 return ERR_PTR(-ENOMEM);
537 ret = dma_get_sgtable_attrs(drm->dev, sgt, rk_obj->kvaddr,
538 rk_obj->dma_addr, obj->size,
541 DRM_ERROR("failed to allocate sgt, %d\n", ret);
549 void *rockchip_gem_prime_vmap(struct drm_gem_object *obj)
551 struct rockchip_gem_object *rk_obj = to_rockchip_obj(obj);
553 if (dma_get_attr(DMA_ATTR_NO_KERNEL_MAPPING, &rk_obj->dma_attrs))
556 return rk_obj->kvaddr;
559 void rockchip_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)