1 /* exynos_drm_gem.c
2  *
3  * Copyright (c) 2011 Samsung Electronics Co., Ltd.
4  * Author: Inki Dae <inki.dae@samsung.com>
5  *
6  * This program is free software; you can redistribute  it and/or modify it
7  * under  the terms of  the GNU General  Public License as published by the
8  * Free Software Foundation;  either version 2 of the  License, or (at your
9  * option) any later version.
10  */
11 
12 #include <drm/drmP.h>
13 #include <drm/drm_vma_manager.h>
14 
15 #include <linux/shmem_fs.h>
16 #include <linux/dma-buf.h>
17 #include <linux/pfn_t.h>
18 #include <drm/exynos_drm.h>
19 
20 #include "exynos_drm_drv.h"
21 #include "exynos_drm_gem.h"
22 #include "exynos_drm_iommu.h"
23 
24 static int exynos_drm_alloc_buf(struct exynos_drm_gem *exynos_gem)
25 {
26 	struct drm_device *dev = exynos_gem->base.dev;
27 	unsigned long attr;
28 	unsigned int nr_pages;
29 	struct sg_table sgt;
30 	int ret = -ENOMEM;
31 
32 	if (exynos_gem->dma_addr) {
33 		DRM_DEBUG_KMS("already allocated.\n");
34 		return 0;
35 	}
36 
37 	exynos_gem->dma_attrs = 0;
38 
39 	/*
40 	 * if EXYNOS_BO_CONTIG, fully physically contiguous memory
41 	 * region will be allocated else physically contiguous
42 	 * as possible.
43 	 */
44 	if (!(exynos_gem->flags & EXYNOS_BO_NONCONTIG))
45 		exynos_gem->dma_attrs |= DMA_ATTR_FORCE_CONTIGUOUS;
46 
47 	/*
48 	 * if EXYNOS_BO_WC or EXYNOS_BO_NONCACHABLE, writecombine mapping
49 	 * else cachable mapping.
50 	 */
51 	if (exynos_gem->flags & EXYNOS_BO_WC ||
52 			!(exynos_gem->flags & EXYNOS_BO_CACHABLE))
53 		attr = DMA_ATTR_WRITE_COMBINE;
54 	else
55 		attr = DMA_ATTR_NON_CONSISTENT;
56 
57 	exynos_gem->dma_attrs |= attr;
58 	exynos_gem->dma_attrs |= DMA_ATTR_NO_KERNEL_MAPPING;
59 
60 	nr_pages = exynos_gem->size >> PAGE_SHIFT;
61 
62 	exynos_gem->pages = kvmalloc_array(nr_pages, sizeof(struct page *),
63 			GFP_KERNEL | __GFP_ZERO);
64 	if (!exynos_gem->pages) {
65 		DRM_ERROR("failed to allocate pages.\n");
66 		return -ENOMEM;
67 	}
68 
69 	exynos_gem->cookie = dma_alloc_attrs(to_dma_dev(dev), exynos_gem->size,
70 					     &exynos_gem->dma_addr, GFP_KERNEL,
71 					     exynos_gem->dma_attrs);
72 	if (!exynos_gem->cookie) {
73 		DRM_ERROR("failed to allocate buffer.\n");
74 		goto err_free;
75 	}
76 
77 	ret = dma_get_sgtable_attrs(to_dma_dev(dev), &sgt, exynos_gem->cookie,
78 				    exynos_gem->dma_addr, exynos_gem->size,
79 				    exynos_gem->dma_attrs);
80 	if (ret < 0) {
81 		DRM_ERROR("failed to get sgtable.\n");
82 		goto err_dma_free;
83 	}
84 
85 	if (drm_prime_sg_to_page_addr_arrays(&sgt, exynos_gem->pages, NULL,
86 					     nr_pages)) {
87 		DRM_ERROR("invalid sgtable.\n");
88 		ret = -EINVAL;
89 		goto err_sgt_free;
90 	}
91 
92 	sg_free_table(&sgt);
93 
94 	DRM_DEBUG_KMS("dma_addr(0x%lx), size(0x%lx)\n",
95 			(unsigned long)exynos_gem->dma_addr, exynos_gem->size);
96 
97 	return 0;
98 
99 err_sgt_free:
100 	sg_free_table(&sgt);
101 err_dma_free:
102 	dma_free_attrs(to_dma_dev(dev), exynos_gem->size, exynos_gem->cookie,
103 		       exynos_gem->dma_addr, exynos_gem->dma_attrs);
104 err_free:
105 	kvfree(exynos_gem->pages);
106 
107 	return ret;
108 }
109 
110 static void exynos_drm_free_buf(struct exynos_drm_gem *exynos_gem)
111 {
112 	struct drm_device *dev = exynos_gem->base.dev;
113 
114 	if (!exynos_gem->dma_addr) {
115 		DRM_DEBUG_KMS("dma_addr is invalid.\n");
116 		return;
117 	}
118 
119 	DRM_DEBUG_KMS("dma_addr(0x%lx), size(0x%lx)\n",
120 			(unsigned long)exynos_gem->dma_addr, exynos_gem->size);
121 
122 	dma_free_attrs(to_dma_dev(dev), exynos_gem->size, exynos_gem->cookie,
123 			(dma_addr_t)exynos_gem->dma_addr,
124 			exynos_gem->dma_attrs);
125 
126 	kvfree(exynos_gem->pages);
127 }
128 
129 static int exynos_drm_gem_handle_create(struct drm_gem_object *obj,
130 					struct drm_file *file_priv,
131 					unsigned int *handle)
132 {
133 	int ret;
134 
135 	/*
136 	 * allocate a id of idr table where the obj is registered
137 	 * and handle has the id what user can see.
138 	 */
139 	ret = drm_gem_handle_create(file_priv, obj, handle);
140 	if (ret)
141 		return ret;
142 
143 	DRM_DEBUG_KMS("gem handle = 0x%x\n", *handle);
144 
145 	/* drop reference from allocate - handle holds it now. */
146 	drm_gem_object_unreference_unlocked(obj);
147 
148 	return 0;
149 }
150 
151 void exynos_drm_gem_destroy(struct exynos_drm_gem *exynos_gem)
152 {
153 	struct drm_gem_object *obj = &exynos_gem->base;
154 
155 	DRM_DEBUG_KMS("handle count = %d\n", obj->handle_count);
156 
157 	/*
158 	 * do not release memory region from exporter.
159 	 *
160 	 * the region will be released by exporter
161 	 * once dmabuf's refcount becomes 0.
162 	 */
163 	if (obj->import_attach)
164 		drm_prime_gem_destroy(obj, exynos_gem->sgt);
165 	else
166 		exynos_drm_free_buf(exynos_gem);
167 
168 	/* release file pointer to gem object. */
169 	drm_gem_object_release(obj);
170 
171 	kfree(exynos_gem);
172 }
173 
174 unsigned long exynos_drm_gem_get_size(struct drm_device *dev,
175 						unsigned int gem_handle,
176 						struct drm_file *file_priv)
177 {
178 	struct exynos_drm_gem *exynos_gem;
179 	struct drm_gem_object *obj;
180 
181 	obj = drm_gem_object_lookup(file_priv, gem_handle);
182 	if (!obj) {
183 		DRM_ERROR("failed to lookup gem object.\n");
184 		return 0;
185 	}
186 
187 	exynos_gem = to_exynos_gem(obj);
188 
189 	drm_gem_object_unreference_unlocked(obj);
190 
191 	return exynos_gem->size;
192 }
193 
194 static struct exynos_drm_gem *exynos_drm_gem_init(struct drm_device *dev,
195 						  unsigned long size)
196 {
197 	struct exynos_drm_gem *exynos_gem;
198 	struct drm_gem_object *obj;
199 	int ret;
200 
201 	exynos_gem = kzalloc(sizeof(*exynos_gem), GFP_KERNEL);
202 	if (!exynos_gem)
203 		return ERR_PTR(-ENOMEM);
204 
205 	exynos_gem->size = size;
206 	obj = &exynos_gem->base;
207 
208 	ret = drm_gem_object_init(dev, obj, size);
209 	if (ret < 0) {
210 		DRM_ERROR("failed to initialize gem object\n");
211 		kfree(exynos_gem);
212 		return ERR_PTR(ret);
213 	}
214 
215 	ret = drm_gem_create_mmap_offset(obj);
216 	if (ret < 0) {
217 		drm_gem_object_release(obj);
218 		kfree(exynos_gem);
219 		return ERR_PTR(ret);
220 	}
221 
222 	DRM_DEBUG_KMS("created file object = %pK\n", obj->filp);
223 
224 	return exynos_gem;
225 }
226 
227 struct exynos_drm_gem *exynos_drm_gem_create(struct drm_device *dev,
228 					     unsigned int flags,
229 					     unsigned long size)
230 {
231 	struct exynos_drm_gem *exynos_gem;
232 	int ret;
233 
234 	if (flags & ~(EXYNOS_BO_MASK)) {
235 		DRM_ERROR("invalid GEM buffer flags: %u\n", flags);
236 		return ERR_PTR(-EINVAL);
237 	}
238 
239 	if (!size) {
240 		DRM_ERROR("invalid GEM buffer size: %lu\n", size);
241 		return ERR_PTR(-EINVAL);
242 	}
243 
244 	size = roundup(size, PAGE_SIZE);
245 
246 	exynos_gem = exynos_drm_gem_init(dev, size);
247 	if (IS_ERR(exynos_gem))
248 		return exynos_gem;
249 
250 	/* set memory type and cache attribute from user side. */
251 	exynos_gem->flags = flags;
252 
253 	ret = exynos_drm_alloc_buf(exynos_gem);
254 	if (ret < 0) {
255 		drm_gem_object_release(&exynos_gem->base);
256 		kfree(exynos_gem);
257 		return ERR_PTR(ret);
258 	}
259 
260 	return exynos_gem;
261 }
262 
263 int exynos_drm_gem_create_ioctl(struct drm_device *dev, void *data,
264 				struct drm_file *file_priv)
265 {
266 	struct drm_exynos_gem_create *args = data;
267 	struct exynos_drm_gem *exynos_gem;
268 	int ret;
269 
270 	exynos_gem = exynos_drm_gem_create(dev, args->flags, args->size);
271 	if (IS_ERR(exynos_gem))
272 		return PTR_ERR(exynos_gem);
273 
274 	ret = exynos_drm_gem_handle_create(&exynos_gem->base, file_priv,
275 					   &args->handle);
276 	if (ret) {
277 		exynos_drm_gem_destroy(exynos_gem);
278 		return ret;
279 	}
280 
281 	return 0;
282 }
283 
284 int exynos_drm_gem_map_ioctl(struct drm_device *dev, void *data,
285 			     struct drm_file *file_priv)
286 {
287 	struct drm_exynos_gem_map *args = data;
288 
289 	return drm_gem_dumb_map_offset(file_priv, dev, args->handle,
290 				       &args->offset);
291 }
292 
293 dma_addr_t *exynos_drm_gem_get_dma_addr(struct drm_device *dev,
294 					unsigned int gem_handle,
295 					struct drm_file *filp)
296 {
297 	struct exynos_drm_gem *exynos_gem;
298 	struct drm_gem_object *obj;
299 
300 	obj = drm_gem_object_lookup(filp, gem_handle);
301 	if (!obj) {
302 		DRM_ERROR("failed to lookup gem object.\n");
303 		return ERR_PTR(-EINVAL);
304 	}
305 
306 	exynos_gem = to_exynos_gem(obj);
307 
308 	return &exynos_gem->dma_addr;
309 }
310 
311 void exynos_drm_gem_put_dma_addr(struct drm_device *dev,
312 					unsigned int gem_handle,
313 					struct drm_file *filp)
314 {
315 	struct drm_gem_object *obj;
316 
317 	obj = drm_gem_object_lookup(filp, gem_handle);
318 	if (!obj) {
319 		DRM_ERROR("failed to lookup gem object.\n");
320 		return;
321 	}
322 
323 	drm_gem_object_unreference_unlocked(obj);
324 
325 	/*
326 	 * decrease obj->refcount one more time because we has already
327 	 * increased it at exynos_drm_gem_get_dma_addr().
328 	 */
329 	drm_gem_object_unreference_unlocked(obj);
330 }
331 
332 static int exynos_drm_gem_mmap_buffer(struct exynos_drm_gem *exynos_gem,
333 				      struct vm_area_struct *vma)
334 {
335 	struct drm_device *drm_dev = exynos_gem->base.dev;
336 	unsigned long vm_size;
337 	int ret;
338 
339 	vma->vm_flags &= ~VM_PFNMAP;
340 	vma->vm_pgoff = 0;
341 
342 	vm_size = vma->vm_end - vma->vm_start;
343 
344 	/* check if user-requested size is valid. */
345 	if (vm_size > exynos_gem->size)
346 		return -EINVAL;
347 
348 	ret = dma_mmap_attrs(to_dma_dev(drm_dev), vma, exynos_gem->cookie,
349 			     exynos_gem->dma_addr, exynos_gem->size,
350 			     exynos_gem->dma_attrs);
351 	if (ret < 0) {
352 		DRM_ERROR("failed to mmap.\n");
353 		return ret;
354 	}
355 
356 	return 0;
357 }
358 
359 int exynos_drm_gem_get_ioctl(struct drm_device *dev, void *data,
360 				      struct drm_file *file_priv)
361 {
362 	struct exynos_drm_gem *exynos_gem;
363 	struct drm_exynos_gem_info *args = data;
364 	struct drm_gem_object *obj;
365 
366 	obj = drm_gem_object_lookup(file_priv, args->handle);
367 	if (!obj) {
368 		DRM_ERROR("failed to lookup gem object.\n");
369 		return -EINVAL;
370 	}
371 
372 	exynos_gem = to_exynos_gem(obj);
373 
374 	args->flags = exynos_gem->flags;
375 	args->size = exynos_gem->size;
376 
377 	drm_gem_object_unreference_unlocked(obj);
378 
379 	return 0;
380 }
381 
382 void exynos_drm_gem_free_object(struct drm_gem_object *obj)
383 {
384 	exynos_drm_gem_destroy(to_exynos_gem(obj));
385 }
386 
387 int exynos_drm_gem_dumb_create(struct drm_file *file_priv,
388 			       struct drm_device *dev,
389 			       struct drm_mode_create_dumb *args)
390 {
391 	struct exynos_drm_gem *exynos_gem;
392 	unsigned int flags;
393 	int ret;
394 
395 	/*
396 	 * allocate memory to be used for framebuffer.
397 	 * - this callback would be called by user application
398 	 *	with DRM_IOCTL_MODE_CREATE_DUMB command.
399 	 */
400 
401 	args->pitch = args->width * ((args->bpp + 7) / 8);
402 	args->size = args->pitch * args->height;
403 
404 	if (is_drm_iommu_supported(dev))
405 		flags = EXYNOS_BO_NONCONTIG | EXYNOS_BO_WC;
406 	else
407 		flags = EXYNOS_BO_CONTIG | EXYNOS_BO_WC;
408 
409 	exynos_gem = exynos_drm_gem_create(dev, flags, args->size);
410 	if (IS_ERR(exynos_gem)) {
411 		dev_warn(dev->dev, "FB allocation failed.\n");
412 		return PTR_ERR(exynos_gem);
413 	}
414 
415 	ret = exynos_drm_gem_handle_create(&exynos_gem->base, file_priv,
416 					   &args->handle);
417 	if (ret) {
418 		exynos_drm_gem_destroy(exynos_gem);
419 		return ret;
420 	}
421 
422 	return 0;
423 }
424 
425 int exynos_drm_gem_fault(struct vm_fault *vmf)
426 {
427 	struct vm_area_struct *vma = vmf->vma;
428 	struct drm_gem_object *obj = vma->vm_private_data;
429 	struct exynos_drm_gem *exynos_gem = to_exynos_gem(obj);
430 	unsigned long pfn;
431 	pgoff_t page_offset;
432 	int ret;
433 
434 	page_offset = (vmf->address - vma->vm_start) >> PAGE_SHIFT;
435 
436 	if (page_offset >= (exynos_gem->size >> PAGE_SHIFT)) {
437 		DRM_ERROR("invalid page offset\n");
438 		ret = -EINVAL;
439 		goto out;
440 	}
441 
442 	pfn = page_to_pfn(exynos_gem->pages[page_offset]);
443 	ret = vm_insert_mixed(vma, vmf->address, __pfn_to_pfn_t(pfn, PFN_DEV));
444 
445 out:
446 	switch (ret) {
447 	case 0:
448 	case -ERESTARTSYS:
449 	case -EINTR:
450 		return VM_FAULT_NOPAGE;
451 	case -ENOMEM:
452 		return VM_FAULT_OOM;
453 	default:
454 		return VM_FAULT_SIGBUS;
455 	}
456 }
457 
458 static int exynos_drm_gem_mmap_obj(struct drm_gem_object *obj,
459 				   struct vm_area_struct *vma)
460 {
461 	struct exynos_drm_gem *exynos_gem = to_exynos_gem(obj);
462 	int ret;
463 
464 	DRM_DEBUG_KMS("flags = 0x%x\n", exynos_gem->flags);
465 
466 	/* non-cachable as default. */
467 	if (exynos_gem->flags & EXYNOS_BO_CACHABLE)
468 		vma->vm_page_prot = vm_get_page_prot(vma->vm_flags);
469 	else if (exynos_gem->flags & EXYNOS_BO_WC)
470 		vma->vm_page_prot =
471 			pgprot_writecombine(vm_get_page_prot(vma->vm_flags));
472 	else
473 		vma->vm_page_prot =
474 			pgprot_noncached(vm_get_page_prot(vma->vm_flags));
475 
476 	ret = exynos_drm_gem_mmap_buffer(exynos_gem, vma);
477 	if (ret)
478 		goto err_close_vm;
479 
480 	return ret;
481 
482 err_close_vm:
483 	drm_gem_vm_close(vma);
484 
485 	return ret;
486 }
487 
488 int exynos_drm_gem_mmap(struct file *filp, struct vm_area_struct *vma)
489 {
490 	struct drm_gem_object *obj;
491 	int ret;
492 
493 	/* set vm_area_struct. */
494 	ret = drm_gem_mmap(filp, vma);
495 	if (ret < 0) {
496 		DRM_ERROR("failed to mmap.\n");
497 		return ret;
498 	}
499 
500 	obj = vma->vm_private_data;
501 
502 	if (obj->import_attach)
503 		return dma_buf_mmap(obj->dma_buf, vma, 0);
504 
505 	return exynos_drm_gem_mmap_obj(obj, vma);
506 }
507 
508 /* low-level interface prime helpers */
509 struct sg_table *exynos_drm_gem_prime_get_sg_table(struct drm_gem_object *obj)
510 {
511 	struct exynos_drm_gem *exynos_gem = to_exynos_gem(obj);
512 	int npages;
513 
514 	npages = exynos_gem->size >> PAGE_SHIFT;
515 
516 	return drm_prime_pages_to_sg(exynos_gem->pages, npages);
517 }
518 
519 struct drm_gem_object *
520 exynos_drm_gem_prime_import_sg_table(struct drm_device *dev,
521 				     struct dma_buf_attachment *attach,
522 				     struct sg_table *sgt)
523 {
524 	struct exynos_drm_gem *exynos_gem;
525 	int npages;
526 	int ret;
527 
528 	exynos_gem = exynos_drm_gem_init(dev, attach->dmabuf->size);
529 	if (IS_ERR(exynos_gem)) {
530 		ret = PTR_ERR(exynos_gem);
531 		return ERR_PTR(ret);
532 	}
533 
534 	exynos_gem->dma_addr = sg_dma_address(sgt->sgl);
535 
536 	npages = exynos_gem->size >> PAGE_SHIFT;
537 	exynos_gem->pages = kvmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
538 	if (!exynos_gem->pages) {
539 		ret = -ENOMEM;
540 		goto err;
541 	}
542 
543 	ret = drm_prime_sg_to_page_addr_arrays(sgt, exynos_gem->pages, NULL,
544 					       npages);
545 	if (ret < 0)
546 		goto err_free_large;
547 
548 	exynos_gem->sgt = sgt;
549 
550 	if (sgt->nents == 1) {
551 		/* always physically continuous memory if sgt->nents is 1. */
552 		exynos_gem->flags |= EXYNOS_BO_CONTIG;
553 	} else {
554 		/*
555 		 * this case could be CONTIG or NONCONTIG type but for now
556 		 * sets NONCONTIG.
557 		 * TODO. we have to find a way that exporter can notify
558 		 * the type of its own buffer to importer.
559 		 */
560 		exynos_gem->flags |= EXYNOS_BO_NONCONTIG;
561 	}
562 
563 	return &exynos_gem->base;
564 
565 err_free_large:
566 	kvfree(exynos_gem->pages);
567 err:
568 	drm_gem_object_release(&exynos_gem->base);
569 	kfree(exynos_gem);
570 	return ERR_PTR(ret);
571 }
572 
573 void *exynos_drm_gem_prime_vmap(struct drm_gem_object *obj)
574 {
575 	return NULL;
576 }
577 
578 void exynos_drm_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
579 {
580 	/* Nothing to do */
581 }
582 
583 int exynos_drm_gem_prime_mmap(struct drm_gem_object *obj,
584 			      struct vm_area_struct *vma)
585 {
586 	int ret;
587 
588 	ret = drm_gem_mmap_obj(obj, obj->size, vma);
589 	if (ret < 0)
590 		return ret;
591 
592 	return exynos_drm_gem_mmap_obj(obj, vma);
593 }
594