1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright 2019 Linaro, Ltd, Rob Herring <robh@kernel.org> */
3 
4 #include <linux/err.h>
5 #include <linux/slab.h>
6 #include <linux/dma-buf.h>
7 #include <linux/dma-mapping.h>
8 
9 #include <drm/panfrost_drm.h>
10 #include "panfrost_device.h"
11 #include "panfrost_gem.h"
12 #include "panfrost_mmu.h"
13 
14 /* Called DRM core on the last userspace/kernel unreference of the
15  * BO.
16  */
17 static void panfrost_gem_free_object(struct drm_gem_object *obj)
18 {
19 	struct panfrost_gem_object *bo = to_panfrost_bo(obj);
20 	struct panfrost_device *pfdev = obj->dev->dev_private;
21 
22 	/*
23 	 * Make sure the BO is no longer inserted in the shrinker list before
24 	 * taking care of the destruction itself. If we don't do that we have a
25 	 * race condition between this function and what's done in
26 	 * panfrost_gem_shrinker_scan().
27 	 */
28 	mutex_lock(&pfdev->shrinker_lock);
29 	list_del_init(&bo->base.madv_list);
30 	mutex_unlock(&pfdev->shrinker_lock);
31 
32 	/*
33 	 * If we still have mappings attached to the BO, there's a problem in
34 	 * our refcounting.
35 	 */
36 	WARN_ON_ONCE(!list_empty(&bo->mappings.list));
37 
38 	if (bo->sgts) {
39 		int i;
40 		int n_sgt = bo->base.base.size / SZ_2M;
41 
42 		for (i = 0; i < n_sgt; i++) {
43 			if (bo->sgts[i].sgl) {
44 				dma_unmap_sgtable(pfdev->dev, &bo->sgts[i],
45 						  DMA_BIDIRECTIONAL, 0);
46 				sg_free_table(&bo->sgts[i]);
47 			}
48 		}
49 		kvfree(bo->sgts);
50 	}
51 
52 	drm_gem_shmem_free_object(obj);
53 }
54 
55 struct panfrost_gem_mapping *
56 panfrost_gem_mapping_get(struct panfrost_gem_object *bo,
57 			 struct panfrost_file_priv *priv)
58 {
59 	struct panfrost_gem_mapping *iter, *mapping = NULL;
60 
61 	mutex_lock(&bo->mappings.lock);
62 	list_for_each_entry(iter, &bo->mappings.list, node) {
63 		if (iter->mmu == &priv->mmu) {
64 			kref_get(&iter->refcount);
65 			mapping = iter;
66 			break;
67 		}
68 	}
69 	mutex_unlock(&bo->mappings.lock);
70 
71 	return mapping;
72 }
73 
74 static void
75 panfrost_gem_teardown_mapping(struct panfrost_gem_mapping *mapping)
76 {
77 	struct panfrost_file_priv *priv;
78 
79 	if (mapping->active)
80 		panfrost_mmu_unmap(mapping);
81 
82 	priv = container_of(mapping->mmu, struct panfrost_file_priv, mmu);
83 	spin_lock(&priv->mm_lock);
84 	if (drm_mm_node_allocated(&mapping->mmnode))
85 		drm_mm_remove_node(&mapping->mmnode);
86 	spin_unlock(&priv->mm_lock);
87 }
88 
89 static void panfrost_gem_mapping_release(struct kref *kref)
90 {
91 	struct panfrost_gem_mapping *mapping;
92 
93 	mapping = container_of(kref, struct panfrost_gem_mapping, refcount);
94 
95 	panfrost_gem_teardown_mapping(mapping);
96 	drm_gem_object_put(&mapping->obj->base.base);
97 	kfree(mapping);
98 }
99 
100 void panfrost_gem_mapping_put(struct panfrost_gem_mapping *mapping)
101 {
102 	if (!mapping)
103 		return;
104 
105 	kref_put(&mapping->refcount, panfrost_gem_mapping_release);
106 }
107 
108 void panfrost_gem_teardown_mappings_locked(struct panfrost_gem_object *bo)
109 {
110 	struct panfrost_gem_mapping *mapping;
111 
112 	list_for_each_entry(mapping, &bo->mappings.list, node)
113 		panfrost_gem_teardown_mapping(mapping);
114 }
115 
116 int panfrost_gem_open(struct drm_gem_object *obj, struct drm_file *file_priv)
117 {
118 	int ret;
119 	size_t size = obj->size;
120 	u64 align;
121 	struct panfrost_gem_object *bo = to_panfrost_bo(obj);
122 	unsigned long color = bo->noexec ? PANFROST_BO_NOEXEC : 0;
123 	struct panfrost_file_priv *priv = file_priv->driver_priv;
124 	struct panfrost_gem_mapping *mapping;
125 
126 	mapping = kzalloc(sizeof(*mapping), GFP_KERNEL);
127 	if (!mapping)
128 		return -ENOMEM;
129 
130 	INIT_LIST_HEAD(&mapping->node);
131 	kref_init(&mapping->refcount);
132 	drm_gem_object_get(obj);
133 	mapping->obj = bo;
134 
135 	/*
136 	 * Executable buffers cannot cross a 16MB boundary as the program
137 	 * counter is 24-bits. We assume executable buffers will be less than
138 	 * 16MB and aligning executable buffers to their size will avoid
139 	 * crossing a 16MB boundary.
140 	 */
141 	if (!bo->noexec)
142 		align = size >> PAGE_SHIFT;
143 	else
144 		align = size >= SZ_2M ? SZ_2M >> PAGE_SHIFT : 0;
145 
146 	mapping->mmu = &priv->mmu;
147 	spin_lock(&priv->mm_lock);
148 	ret = drm_mm_insert_node_generic(&priv->mm, &mapping->mmnode,
149 					 size >> PAGE_SHIFT, align, color, 0);
150 	spin_unlock(&priv->mm_lock);
151 	if (ret)
152 		goto err;
153 
154 	if (!bo->is_heap) {
155 		ret = panfrost_mmu_map(mapping);
156 		if (ret)
157 			goto err;
158 	}
159 
160 	mutex_lock(&bo->mappings.lock);
161 	WARN_ON(bo->base.madv != PANFROST_MADV_WILLNEED);
162 	list_add_tail(&mapping->node, &bo->mappings.list);
163 	mutex_unlock(&bo->mappings.lock);
164 
165 err:
166 	if (ret)
167 		panfrost_gem_mapping_put(mapping);
168 	return ret;
169 }
170 
171 void panfrost_gem_close(struct drm_gem_object *obj, struct drm_file *file_priv)
172 {
173 	struct panfrost_file_priv *priv = file_priv->driver_priv;
174 	struct panfrost_gem_object *bo = to_panfrost_bo(obj);
175 	struct panfrost_gem_mapping *mapping = NULL, *iter;
176 
177 	mutex_lock(&bo->mappings.lock);
178 	list_for_each_entry(iter, &bo->mappings.list, node) {
179 		if (iter->mmu == &priv->mmu) {
180 			mapping = iter;
181 			list_del(&iter->node);
182 			break;
183 		}
184 	}
185 	mutex_unlock(&bo->mappings.lock);
186 
187 	panfrost_gem_mapping_put(mapping);
188 }
189 
190 static int panfrost_gem_pin(struct drm_gem_object *obj)
191 {
192 	if (to_panfrost_bo(obj)->is_heap)
193 		return -EINVAL;
194 
195 	return drm_gem_shmem_pin(obj);
196 }
197 
198 static const struct drm_gem_object_funcs panfrost_gem_funcs = {
199 	.free = panfrost_gem_free_object,
200 	.open = panfrost_gem_open,
201 	.close = panfrost_gem_close,
202 	.print_info = drm_gem_shmem_print_info,
203 	.pin = panfrost_gem_pin,
204 	.unpin = drm_gem_shmem_unpin,
205 	.get_sg_table = drm_gem_shmem_get_sg_table,
206 	.vmap = drm_gem_shmem_vmap,
207 	.vunmap = drm_gem_shmem_vunmap,
208 	.mmap = drm_gem_shmem_mmap,
209 };
210 
211 /**
212  * panfrost_gem_create_object - Implementation of driver->gem_create_object.
213  * @dev: DRM device
214  * @size: Size in bytes of the memory the object will reference
215  *
216  * This lets the GEM helpers allocate object structs for us, and keep
217  * our BO stats correct.
218  */
219 struct drm_gem_object *panfrost_gem_create_object(struct drm_device *dev, size_t size)
220 {
221 	struct panfrost_device *pfdev = dev->dev_private;
222 	struct panfrost_gem_object *obj;
223 
224 	obj = kzalloc(sizeof(*obj), GFP_KERNEL);
225 	if (!obj)
226 		return NULL;
227 
228 	INIT_LIST_HEAD(&obj->mappings.list);
229 	mutex_init(&obj->mappings.lock);
230 	obj->base.base.funcs = &panfrost_gem_funcs;
231 	obj->base.map_wc = !pfdev->coherent;
232 
233 	return &obj->base.base;
234 }
235 
236 struct panfrost_gem_object *
237 panfrost_gem_create_with_handle(struct drm_file *file_priv,
238 				struct drm_device *dev, size_t size,
239 				u32 flags,
240 				uint32_t *handle)
241 {
242 	int ret;
243 	struct drm_gem_shmem_object *shmem;
244 	struct panfrost_gem_object *bo;
245 
246 	/* Round up heap allocations to 2MB to keep fault handling simple */
247 	if (flags & PANFROST_BO_HEAP)
248 		size = roundup(size, SZ_2M);
249 
250 	shmem = drm_gem_shmem_create(dev, size);
251 	if (IS_ERR(shmem))
252 		return ERR_CAST(shmem);
253 
254 	bo = to_panfrost_bo(&shmem->base);
255 	bo->noexec = !!(flags & PANFROST_BO_NOEXEC);
256 	bo->is_heap = !!(flags & PANFROST_BO_HEAP);
257 
258 	/*
259 	 * Allocate an id of idr table where the obj is registered
260 	 * and handle has the id what user can see.
261 	 */
262 	ret = drm_gem_handle_create(file_priv, &shmem->base, handle);
263 	/* drop reference from allocate - handle holds it now. */
264 	drm_gem_object_put(&shmem->base);
265 	if (ret)
266 		return ERR_PTR(ret);
267 
268 	return bo;
269 }
270 
271 struct drm_gem_object *
272 panfrost_gem_prime_import_sg_table(struct drm_device *dev,
273 				   struct dma_buf_attachment *attach,
274 				   struct sg_table *sgt)
275 {
276 	struct drm_gem_object *obj;
277 	struct panfrost_gem_object *bo;
278 
279 	obj = drm_gem_shmem_prime_import_sg_table(dev, attach, sgt);
280 	if (IS_ERR(obj))
281 		return ERR_CAST(obj);
282 
283 	bo = to_panfrost_bo(obj);
284 	bo->noexec = true;
285 
286 	return obj;
287 }
288