1 // SPDX-License-Identifier: GPL-2.0 or MIT
2 /* Copyright 2019 Linaro, Ltd, Rob Herring <robh@kernel.org> */
3 /* Copyright 2023 Collabora ltd. */
4
5 #include <linux/dma-buf.h>
6 #include <linux/dma-mapping.h>
7 #include <linux/err.h>
8 #include <linux/slab.h>
9
10 #include <drm/panthor_drm.h>
11
12 #include "panthor_device.h"
13 #include "panthor_gem.h"
14 #include "panthor_mmu.h"
15
panthor_gem_free_object(struct drm_gem_object * obj)16 static void panthor_gem_free_object(struct drm_gem_object *obj)
17 {
18 struct panthor_gem_object *bo = to_panthor_bo(obj);
19 struct drm_gem_object *vm_root_gem = bo->exclusive_vm_root_gem;
20
21 drm_gem_free_mmap_offset(&bo->base.base);
22 mutex_destroy(&bo->gpuva_list_lock);
23 drm_gem_shmem_free(&bo->base);
24 drm_gem_object_put(vm_root_gem);
25 }
26
27 /**
28 * panthor_kernel_bo_destroy() - Destroy a kernel buffer object
29 * @bo: Kernel buffer object to destroy. If NULL or an ERR_PTR(), the destruction
30 * is skipped.
31 */
panthor_kernel_bo_destroy(struct panthor_kernel_bo * bo)32 void panthor_kernel_bo_destroy(struct panthor_kernel_bo *bo)
33 {
34 struct panthor_vm *vm;
35 int ret;
36
37 if (IS_ERR_OR_NULL(bo))
38 return;
39
40 vm = bo->vm;
41 panthor_kernel_bo_vunmap(bo);
42
43 if (drm_WARN_ON(bo->obj->dev,
44 to_panthor_bo(bo->obj)->exclusive_vm_root_gem != panthor_vm_root_gem(vm)))
45 goto out_free_bo;
46
47 ret = panthor_vm_unmap_range(vm, bo->va_node.start,
48 panthor_kernel_bo_size(bo));
49 if (ret)
50 goto out_free_bo;
51
52 panthor_vm_free_va(vm, &bo->va_node);
53 drm_gem_object_put(bo->obj);
54
55 out_free_bo:
56 panthor_vm_put(vm);
57 kfree(bo);
58 }
59
60 /**
61 * panthor_kernel_bo_create() - Create and map a GEM object to a VM
62 * @ptdev: Device.
63 * @vm: VM to map the GEM to. If NULL, the kernel object is not GPU mapped.
64 * @size: Size of the buffer object.
65 * @bo_flags: Combination of drm_panthor_bo_flags flags.
66 * @vm_map_flags: Combination of drm_panthor_vm_bind_op_flags (only those
67 * that are related to map operations).
68 * @gpu_va: GPU address assigned when mapping to the VM.
69 * If gpu_va == PANTHOR_VM_KERNEL_AUTO_VA, the virtual address will be
70 * automatically allocated.
71 *
72 * Return: A valid pointer in case of success, an ERR_PTR() otherwise.
73 */
74 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)75 panthor_kernel_bo_create(struct panthor_device *ptdev, struct panthor_vm *vm,
76 size_t size, u32 bo_flags, u32 vm_map_flags,
77 u64 gpu_va)
78 {
79 struct drm_gem_shmem_object *obj;
80 struct panthor_kernel_bo *kbo;
81 struct panthor_gem_object *bo;
82 int ret;
83
84 if (drm_WARN_ON(&ptdev->base, !vm))
85 return ERR_PTR(-EINVAL);
86
87 kbo = kzalloc(sizeof(*kbo), GFP_KERNEL);
88 if (!kbo)
89 return ERR_PTR(-ENOMEM);
90
91 obj = drm_gem_shmem_create(&ptdev->base, size);
92 if (IS_ERR(obj)) {
93 ret = PTR_ERR(obj);
94 goto err_free_bo;
95 }
96
97 bo = to_panthor_bo(&obj->base);
98 size = obj->base.size;
99 kbo->obj = &obj->base;
100 bo->flags = bo_flags;
101
102 ret = panthor_vm_alloc_va(vm, gpu_va, size, &kbo->va_node);
103 if (ret)
104 goto err_put_obj;
105
106 ret = panthor_vm_map_bo_range(vm, bo, 0, size, kbo->va_node.start, vm_map_flags);
107 if (ret)
108 goto err_free_va;
109
110 kbo->vm = panthor_vm_get(vm);
111 bo->exclusive_vm_root_gem = panthor_vm_root_gem(vm);
112 drm_gem_object_get(bo->exclusive_vm_root_gem);
113 bo->base.base.resv = bo->exclusive_vm_root_gem->resv;
114 return kbo;
115
116 err_free_va:
117 panthor_vm_free_va(vm, &kbo->va_node);
118
119 err_put_obj:
120 drm_gem_object_put(&obj->base);
121
122 err_free_bo:
123 kfree(kbo);
124 return ERR_PTR(ret);
125 }
126
panthor_gem_mmap(struct drm_gem_object * obj,struct vm_area_struct * vma)127 static int panthor_gem_mmap(struct drm_gem_object *obj, struct vm_area_struct *vma)
128 {
129 struct panthor_gem_object *bo = to_panthor_bo(obj);
130
131 /* Don't allow mmap on objects that have the NO_MMAP flag set. */
132 if (bo->flags & DRM_PANTHOR_BO_NO_MMAP)
133 return -EINVAL;
134
135 return drm_gem_shmem_object_mmap(obj, vma);
136 }
137
138 static struct dma_buf *
panthor_gem_prime_export(struct drm_gem_object * obj,int flags)139 panthor_gem_prime_export(struct drm_gem_object *obj, int flags)
140 {
141 /* We can't export GEMs that have an exclusive VM. */
142 if (to_panthor_bo(obj)->exclusive_vm_root_gem)
143 return ERR_PTR(-EINVAL);
144
145 return drm_gem_prime_export(obj, flags);
146 }
147
148 static const struct drm_gem_object_funcs panthor_gem_funcs = {
149 .free = panthor_gem_free_object,
150 .print_info = drm_gem_shmem_object_print_info,
151 .pin = drm_gem_shmem_object_pin,
152 .unpin = drm_gem_shmem_object_unpin,
153 .get_sg_table = drm_gem_shmem_object_get_sg_table,
154 .vmap = drm_gem_shmem_object_vmap,
155 .vunmap = drm_gem_shmem_object_vunmap,
156 .mmap = panthor_gem_mmap,
157 .export = panthor_gem_prime_export,
158 .vm_ops = &drm_gem_shmem_vm_ops,
159 };
160
161 /**
162 * panthor_gem_create_object - Implementation of driver->gem_create_object.
163 * @ddev: DRM device
164 * @size: Size in bytes of the memory the object will reference
165 *
166 * This lets the GEM helpers allocate object structs for us, and keep
167 * our BO stats correct.
168 */
panthor_gem_create_object(struct drm_device * ddev,size_t size)169 struct drm_gem_object *panthor_gem_create_object(struct drm_device *ddev, size_t size)
170 {
171 struct panthor_device *ptdev = container_of(ddev, struct panthor_device, base);
172 struct panthor_gem_object *obj;
173
174 obj = kzalloc(sizeof(*obj), GFP_KERNEL);
175 if (!obj)
176 return ERR_PTR(-ENOMEM);
177
178 obj->base.base.funcs = &panthor_gem_funcs;
179 obj->base.map_wc = !ptdev->coherent;
180 mutex_init(&obj->gpuva_list_lock);
181 drm_gem_gpuva_set_lock(&obj->base.base, &obj->gpuva_list_lock);
182
183 return &obj->base.base;
184 }
185
186 /**
187 * panthor_gem_create_with_handle() - Create a GEM object and attach it to a handle.
188 * @file: DRM file.
189 * @ddev: DRM device.
190 * @exclusive_vm: Exclusive VM. Not NULL if the GEM object can't be shared.
191 * @size: Size of the GEM object to allocate.
192 * @flags: Combination of drm_panthor_bo_flags flags.
193 * @handle: Pointer holding the handle pointing to the new GEM object.
194 *
195 * Return: Zero on success
196 */
197 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)198 panthor_gem_create_with_handle(struct drm_file *file,
199 struct drm_device *ddev,
200 struct panthor_vm *exclusive_vm,
201 u64 *size, u32 flags, u32 *handle)
202 {
203 int ret;
204 struct drm_gem_shmem_object *shmem;
205 struct panthor_gem_object *bo;
206
207 shmem = drm_gem_shmem_create(ddev, *size);
208 if (IS_ERR(shmem))
209 return PTR_ERR(shmem);
210
211 bo = to_panthor_bo(&shmem->base);
212 bo->flags = flags;
213
214 if (exclusive_vm) {
215 bo->exclusive_vm_root_gem = panthor_vm_root_gem(exclusive_vm);
216 drm_gem_object_get(bo->exclusive_vm_root_gem);
217 bo->base.base.resv = bo->exclusive_vm_root_gem->resv;
218 }
219
220 /*
221 * Allocate an id of idr table where the obj is registered
222 * and handle has the id what user can see.
223 */
224 ret = drm_gem_handle_create(file, &shmem->base, handle);
225 if (!ret)
226 *size = bo->base.base.size;
227
228 /* drop reference from allocate - handle holds it now. */
229 drm_gem_object_put(&shmem->base);
230
231 return ret;
232 }
233