18a1cc075SBoris Brezillon // SPDX-License-Identifier: GPL-2.0 or MIT
28a1cc075SBoris Brezillon /* Copyright 2019 Linaro, Ltd, Rob Herring <robh@kernel.org> */
38a1cc075SBoris Brezillon /* Copyright 2023 Collabora ltd. */
48a1cc075SBoris Brezillon
58a1cc075SBoris Brezillon #include <linux/dma-buf.h>
68a1cc075SBoris Brezillon #include <linux/dma-mapping.h>
78a1cc075SBoris Brezillon #include <linux/err.h>
88a1cc075SBoris Brezillon #include <linux/slab.h>
98a1cc075SBoris Brezillon
108a1cc075SBoris Brezillon #include <drm/panthor_drm.h>
118a1cc075SBoris Brezillon
128a1cc075SBoris Brezillon #include "panthor_device.h"
138a1cc075SBoris Brezillon #include "panthor_gem.h"
148a1cc075SBoris Brezillon #include "panthor_mmu.h"
158a1cc075SBoris Brezillon
panthor_gem_free_object(struct drm_gem_object * obj)168a1cc075SBoris Brezillon static void panthor_gem_free_object(struct drm_gem_object *obj)
178a1cc075SBoris Brezillon {
188a1cc075SBoris Brezillon struct panthor_gem_object *bo = to_panthor_bo(obj);
198a1cc075SBoris Brezillon struct drm_gem_object *vm_root_gem = bo->exclusive_vm_root_gem;
208a1cc075SBoris Brezillon
218a1cc075SBoris Brezillon drm_gem_free_mmap_offset(&bo->base.base);
228a1cc075SBoris Brezillon mutex_destroy(&bo->gpuva_list_lock);
238a1cc075SBoris Brezillon drm_gem_shmem_free(&bo->base);
248a1cc075SBoris Brezillon drm_gem_object_put(vm_root_gem);
258a1cc075SBoris Brezillon }
268a1cc075SBoris Brezillon
278a1cc075SBoris Brezillon /**
288a1cc075SBoris Brezillon * panthor_kernel_bo_destroy() - Destroy a kernel buffer object
298a1cc075SBoris Brezillon * @bo: Kernel buffer object to destroy. If NULL or an ERR_PTR(), the destruction
308a1cc075SBoris Brezillon * is skipped.
318a1cc075SBoris Brezillon */
panthor_kernel_bo_destroy(struct panthor_kernel_bo * bo)32*ff60c8daSBoris Brezillon void panthor_kernel_bo_destroy(struct panthor_kernel_bo *bo)
338a1cc075SBoris Brezillon {
34*ff60c8daSBoris Brezillon struct panthor_vm *vm;
358a1cc075SBoris Brezillon int ret;
368a1cc075SBoris Brezillon
378a1cc075SBoris Brezillon if (IS_ERR_OR_NULL(bo))
388a1cc075SBoris Brezillon return;
398a1cc075SBoris Brezillon
40*ff60c8daSBoris Brezillon vm = bo->vm;
418a1cc075SBoris Brezillon panthor_kernel_bo_vunmap(bo);
428a1cc075SBoris Brezillon
438a1cc075SBoris Brezillon if (drm_WARN_ON(bo->obj->dev,
448a1cc075SBoris Brezillon to_panthor_bo(bo->obj)->exclusive_vm_root_gem != panthor_vm_root_gem(vm)))
458a1cc075SBoris Brezillon goto out_free_bo;
468a1cc075SBoris Brezillon
478a1cc075SBoris Brezillon ret = panthor_vm_unmap_range(vm, bo->va_node.start,
488a1cc075SBoris Brezillon panthor_kernel_bo_size(bo));
498a1cc075SBoris Brezillon if (ret)
508a1cc075SBoris Brezillon goto out_free_bo;
518a1cc075SBoris Brezillon
528a1cc075SBoris Brezillon panthor_vm_free_va(vm, &bo->va_node);
538a1cc075SBoris Brezillon drm_gem_object_put(bo->obj);
548a1cc075SBoris Brezillon
558a1cc075SBoris Brezillon out_free_bo:
56*ff60c8daSBoris Brezillon panthor_vm_put(vm);
578a1cc075SBoris Brezillon kfree(bo);
588a1cc075SBoris Brezillon }
598a1cc075SBoris Brezillon
608a1cc075SBoris Brezillon /**
618a1cc075SBoris Brezillon * panthor_kernel_bo_create() - Create and map a GEM object to a VM
628a1cc075SBoris Brezillon * @ptdev: Device.
638a1cc075SBoris Brezillon * @vm: VM to map the GEM to. If NULL, the kernel object is not GPU mapped.
648a1cc075SBoris Brezillon * @size: Size of the buffer object.
658a1cc075SBoris Brezillon * @bo_flags: Combination of drm_panthor_bo_flags flags.
668a1cc075SBoris Brezillon * @vm_map_flags: Combination of drm_panthor_vm_bind_op_flags (only those
678a1cc075SBoris Brezillon * that are related to map operations).
688a1cc075SBoris Brezillon * @gpu_va: GPU address assigned when mapping to the VM.
698a1cc075SBoris Brezillon * If gpu_va == PANTHOR_VM_KERNEL_AUTO_VA, the virtual address will be
708a1cc075SBoris Brezillon * automatically allocated.
718a1cc075SBoris Brezillon *
728a1cc075SBoris Brezillon * Return: A valid pointer in case of success, an ERR_PTR() otherwise.
738a1cc075SBoris Brezillon */
748a1cc075SBoris Brezillon struct panthor_kernel_bo *
panthor_kernel_bo_create(struct panthor_device * ptdev,struct panthor_vm * vm,size_t size,u32 bo_flags,u32 vm_map_flags,u64 gpu_va)758a1cc075SBoris Brezillon panthor_kernel_bo_create(struct panthor_device *ptdev, struct panthor_vm *vm,
768a1cc075SBoris Brezillon size_t size, u32 bo_flags, u32 vm_map_flags,
778a1cc075SBoris Brezillon u64 gpu_va)
788a1cc075SBoris Brezillon {
798a1cc075SBoris Brezillon struct drm_gem_shmem_object *obj;
808a1cc075SBoris Brezillon struct panthor_kernel_bo *kbo;
818a1cc075SBoris Brezillon struct panthor_gem_object *bo;
828a1cc075SBoris Brezillon int ret;
838a1cc075SBoris Brezillon
848a1cc075SBoris Brezillon if (drm_WARN_ON(&ptdev->base, !vm))
858a1cc075SBoris Brezillon return ERR_PTR(-EINVAL);
868a1cc075SBoris Brezillon
878a1cc075SBoris Brezillon kbo = kzalloc(sizeof(*kbo), GFP_KERNEL);
888a1cc075SBoris Brezillon if (!kbo)
898a1cc075SBoris Brezillon return ERR_PTR(-ENOMEM);
908a1cc075SBoris Brezillon
918a1cc075SBoris Brezillon obj = drm_gem_shmem_create(&ptdev->base, size);
928a1cc075SBoris Brezillon if (IS_ERR(obj)) {
938a1cc075SBoris Brezillon ret = PTR_ERR(obj);
948a1cc075SBoris Brezillon goto err_free_bo;
958a1cc075SBoris Brezillon }
968a1cc075SBoris Brezillon
978a1cc075SBoris Brezillon bo = to_panthor_bo(&obj->base);
988a1cc075SBoris Brezillon size = obj->base.size;
998a1cc075SBoris Brezillon kbo->obj = &obj->base;
1008a1cc075SBoris Brezillon bo->flags = bo_flags;
1018a1cc075SBoris Brezillon
1028a1cc075SBoris Brezillon ret = panthor_vm_alloc_va(vm, gpu_va, size, &kbo->va_node);
1038a1cc075SBoris Brezillon if (ret)
1048a1cc075SBoris Brezillon goto err_put_obj;
1058a1cc075SBoris Brezillon
1068a1cc075SBoris Brezillon ret = panthor_vm_map_bo_range(vm, bo, 0, size, kbo->va_node.start, vm_map_flags);
1078a1cc075SBoris Brezillon if (ret)
1088a1cc075SBoris Brezillon goto err_free_va;
1098a1cc075SBoris Brezillon
110*ff60c8daSBoris Brezillon kbo->vm = panthor_vm_get(vm);
1118a1cc075SBoris Brezillon bo->exclusive_vm_root_gem = panthor_vm_root_gem(vm);
1128a1cc075SBoris Brezillon drm_gem_object_get(bo->exclusive_vm_root_gem);
1138a1cc075SBoris Brezillon bo->base.base.resv = bo->exclusive_vm_root_gem->resv;
1148a1cc075SBoris Brezillon return kbo;
1158a1cc075SBoris Brezillon
1168a1cc075SBoris Brezillon err_free_va:
1178a1cc075SBoris Brezillon panthor_vm_free_va(vm, &kbo->va_node);
1188a1cc075SBoris Brezillon
1198a1cc075SBoris Brezillon err_put_obj:
1208a1cc075SBoris Brezillon drm_gem_object_put(&obj->base);
1218a1cc075SBoris Brezillon
1228a1cc075SBoris Brezillon err_free_bo:
1238a1cc075SBoris Brezillon kfree(kbo);
1248a1cc075SBoris Brezillon return ERR_PTR(ret);
1258a1cc075SBoris Brezillon }
1268a1cc075SBoris Brezillon
panthor_gem_mmap(struct drm_gem_object * obj,struct vm_area_struct * vma)1278a1cc075SBoris Brezillon static int panthor_gem_mmap(struct drm_gem_object *obj, struct vm_area_struct *vma)
1288a1cc075SBoris Brezillon {
1298a1cc075SBoris Brezillon struct panthor_gem_object *bo = to_panthor_bo(obj);
1308a1cc075SBoris Brezillon
1318a1cc075SBoris Brezillon /* Don't allow mmap on objects that have the NO_MMAP flag set. */
1328a1cc075SBoris Brezillon if (bo->flags & DRM_PANTHOR_BO_NO_MMAP)
1338a1cc075SBoris Brezillon return -EINVAL;
1348a1cc075SBoris Brezillon
1358a1cc075SBoris Brezillon return drm_gem_shmem_object_mmap(obj, vma);
1368a1cc075SBoris Brezillon }
1378a1cc075SBoris Brezillon
1388a1cc075SBoris Brezillon static struct dma_buf *
panthor_gem_prime_export(struct drm_gem_object * obj,int flags)1398a1cc075SBoris Brezillon panthor_gem_prime_export(struct drm_gem_object *obj, int flags)
1408a1cc075SBoris Brezillon {
1418a1cc075SBoris Brezillon /* We can't export GEMs that have an exclusive VM. */
1428a1cc075SBoris Brezillon if (to_panthor_bo(obj)->exclusive_vm_root_gem)
1438a1cc075SBoris Brezillon return ERR_PTR(-EINVAL);
1448a1cc075SBoris Brezillon
1458a1cc075SBoris Brezillon return drm_gem_prime_export(obj, flags);
1468a1cc075SBoris Brezillon }
1478a1cc075SBoris Brezillon
1488a1cc075SBoris Brezillon static const struct drm_gem_object_funcs panthor_gem_funcs = {
1498a1cc075SBoris Brezillon .free = panthor_gem_free_object,
1508a1cc075SBoris Brezillon .print_info = drm_gem_shmem_object_print_info,
1518a1cc075SBoris Brezillon .pin = drm_gem_shmem_object_pin,
1528a1cc075SBoris Brezillon .unpin = drm_gem_shmem_object_unpin,
1538a1cc075SBoris Brezillon .get_sg_table = drm_gem_shmem_object_get_sg_table,
1548a1cc075SBoris Brezillon .vmap = drm_gem_shmem_object_vmap,
1558a1cc075SBoris Brezillon .vunmap = drm_gem_shmem_object_vunmap,
1568a1cc075SBoris Brezillon .mmap = panthor_gem_mmap,
1578a1cc075SBoris Brezillon .export = panthor_gem_prime_export,
1588a1cc075SBoris Brezillon .vm_ops = &drm_gem_shmem_vm_ops,
1598a1cc075SBoris Brezillon };
1608a1cc075SBoris Brezillon
1618a1cc075SBoris Brezillon /**
1628a1cc075SBoris Brezillon * panthor_gem_create_object - Implementation of driver->gem_create_object.
1638a1cc075SBoris Brezillon * @ddev: DRM device
1648a1cc075SBoris Brezillon * @size: Size in bytes of the memory the object will reference
1658a1cc075SBoris Brezillon *
1668a1cc075SBoris Brezillon * This lets the GEM helpers allocate object structs for us, and keep
1678a1cc075SBoris Brezillon * our BO stats correct.
1688a1cc075SBoris Brezillon */
panthor_gem_create_object(struct drm_device * ddev,size_t size)1698a1cc075SBoris Brezillon struct drm_gem_object *panthor_gem_create_object(struct drm_device *ddev, size_t size)
1708a1cc075SBoris Brezillon {
1718a1cc075SBoris Brezillon struct panthor_device *ptdev = container_of(ddev, struct panthor_device, base);
1728a1cc075SBoris Brezillon struct panthor_gem_object *obj;
1738a1cc075SBoris Brezillon
1748a1cc075SBoris Brezillon obj = kzalloc(sizeof(*obj), GFP_KERNEL);
1758a1cc075SBoris Brezillon if (!obj)
1768a1cc075SBoris Brezillon return ERR_PTR(-ENOMEM);
1778a1cc075SBoris Brezillon
1788a1cc075SBoris Brezillon obj->base.base.funcs = &panthor_gem_funcs;
1798a1cc075SBoris Brezillon obj->base.map_wc = !ptdev->coherent;
1808a1cc075SBoris Brezillon mutex_init(&obj->gpuva_list_lock);
1818a1cc075SBoris Brezillon drm_gem_gpuva_set_lock(&obj->base.base, &obj->gpuva_list_lock);
1828a1cc075SBoris Brezillon
1838a1cc075SBoris Brezillon return &obj->base.base;
1848a1cc075SBoris Brezillon }
1858a1cc075SBoris Brezillon
1868a1cc075SBoris Brezillon /**
1878a1cc075SBoris Brezillon * panthor_gem_create_with_handle() - Create a GEM object and attach it to a handle.
1888a1cc075SBoris Brezillon * @file: DRM file.
1898a1cc075SBoris Brezillon * @ddev: DRM device.
1908a1cc075SBoris Brezillon * @exclusive_vm: Exclusive VM. Not NULL if the GEM object can't be shared.
1918a1cc075SBoris Brezillon * @size: Size of the GEM object to allocate.
1928a1cc075SBoris Brezillon * @flags: Combination of drm_panthor_bo_flags flags.
1938a1cc075SBoris Brezillon * @handle: Pointer holding the handle pointing to the new GEM object.
1948a1cc075SBoris Brezillon *
1958a1cc075SBoris Brezillon * Return: Zero on success
1968a1cc075SBoris Brezillon */
1978a1cc075SBoris Brezillon int
panthor_gem_create_with_handle(struct drm_file * file,struct drm_device * ddev,struct panthor_vm * exclusive_vm,u64 * size,u32 flags,u32 * handle)1988a1cc075SBoris Brezillon panthor_gem_create_with_handle(struct drm_file *file,
1998a1cc075SBoris Brezillon struct drm_device *ddev,
2008a1cc075SBoris Brezillon struct panthor_vm *exclusive_vm,
2018a1cc075SBoris Brezillon u64 *size, u32 flags, u32 *handle)
2028a1cc075SBoris Brezillon {
2038a1cc075SBoris Brezillon int ret;
2048a1cc075SBoris Brezillon struct drm_gem_shmem_object *shmem;
2058a1cc075SBoris Brezillon struct panthor_gem_object *bo;
2068a1cc075SBoris Brezillon
2078a1cc075SBoris Brezillon shmem = drm_gem_shmem_create(ddev, *size);
2088a1cc075SBoris Brezillon if (IS_ERR(shmem))
2098a1cc075SBoris Brezillon return PTR_ERR(shmem);
2108a1cc075SBoris Brezillon
2118a1cc075SBoris Brezillon bo = to_panthor_bo(&shmem->base);
2128a1cc075SBoris Brezillon bo->flags = flags;
2138a1cc075SBoris Brezillon
2148a1cc075SBoris Brezillon if (exclusive_vm) {
2158a1cc075SBoris Brezillon bo->exclusive_vm_root_gem = panthor_vm_root_gem(exclusive_vm);
2168a1cc075SBoris Brezillon drm_gem_object_get(bo->exclusive_vm_root_gem);
2178a1cc075SBoris Brezillon bo->base.base.resv = bo->exclusive_vm_root_gem->resv;
2188a1cc075SBoris Brezillon }
2198a1cc075SBoris Brezillon
2208a1cc075SBoris Brezillon /*
2218a1cc075SBoris Brezillon * Allocate an id of idr table where the obj is registered
2228a1cc075SBoris Brezillon * and handle has the id what user can see.
2238a1cc075SBoris Brezillon */
2248a1cc075SBoris Brezillon ret = drm_gem_handle_create(file, &shmem->base, handle);
2258a1cc075SBoris Brezillon if (!ret)
2268a1cc075SBoris Brezillon *size = bo->base.base.size;
2278a1cc075SBoris Brezillon
2288a1cc075SBoris Brezillon /* drop reference from allocate - handle holds it now. */
2298a1cc075SBoris Brezillon drm_gem_object_put(&shmem->base);
2308a1cc075SBoris Brezillon
2318a1cc075SBoris Brezillon return ret;
2328a1cc075SBoris Brezillon }
233