1 /*
2  * Copyright (C) 2015 Etnaviv Project
3  *
4  * This program is free software; you can redistribute it and/or modify it
5  * under the terms of the GNU General Public License version 2 as published by
6  * the Free Software Foundation.
7  *
8  * This program is distributed in the hope that it will be useful, but WITHOUT
9  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
11  * more details.
12  *
13  * You should have received a copy of the GNU General Public License along with
14  * this program.  If not, see <http://www.gnu.org/licenses/>.
15  */
16 
17 #include <linux/spinlock.h>
18 #include <linux/shmem_fs.h>
19 
20 #include "etnaviv_drv.h"
21 #include "etnaviv_gem.h"
22 #include "etnaviv_gpu.h"
23 #include "etnaviv_mmu.h"
24 
25 static void etnaviv_gem_scatter_map(struct etnaviv_gem_object *etnaviv_obj)
26 {
27 	struct drm_device *dev = etnaviv_obj->base.dev;
28 	struct sg_table *sgt = etnaviv_obj->sgt;
29 
30 	/*
31 	 * For non-cached buffers, ensure the new pages are clean
32 	 * because display controller, GPU, etc. are not coherent.
33 	 */
34 	if (etnaviv_obj->flags & ETNA_BO_CACHE_MASK)
35 		dma_map_sg(dev->dev, sgt->sgl, sgt->nents, DMA_BIDIRECTIONAL);
36 }
37 
38 static void etnaviv_gem_scatterlist_unmap(struct etnaviv_gem_object *etnaviv_obj)
39 {
40 	struct drm_device *dev = etnaviv_obj->base.dev;
41 	struct sg_table *sgt = etnaviv_obj->sgt;
42 
43 	/*
44 	 * For non-cached buffers, ensure the new pages are clean
45 	 * because display controller, GPU, etc. are not coherent:
46 	 *
47 	 * WARNING: The DMA API does not support concurrent CPU
48 	 * and device access to the memory area.  With BIDIRECTIONAL,
49 	 * we will clean the cache lines which overlap the region,
50 	 * and invalidate all cache lines (partially) contained in
51 	 * the region.
52 	 *
53 	 * If you have dirty data in the overlapping cache lines,
54 	 * that will corrupt the GPU-written data.  If you have
55 	 * written into the remainder of the region, this can
56 	 * discard those writes.
57 	 */
58 	if (etnaviv_obj->flags & ETNA_BO_CACHE_MASK)
59 		dma_unmap_sg(dev->dev, sgt->sgl, sgt->nents, DMA_BIDIRECTIONAL);
60 }
61 
62 /* called with etnaviv_obj->lock held */
63 static int etnaviv_gem_shmem_get_pages(struct etnaviv_gem_object *etnaviv_obj)
64 {
65 	struct drm_device *dev = etnaviv_obj->base.dev;
66 	struct page **p = drm_gem_get_pages(&etnaviv_obj->base);
67 
68 	if (IS_ERR(p)) {
69 		dev_err(dev->dev, "could not get pages: %ld\n", PTR_ERR(p));
70 		return PTR_ERR(p);
71 	}
72 
73 	etnaviv_obj->pages = p;
74 
75 	return 0;
76 }
77 
78 static void put_pages(struct etnaviv_gem_object *etnaviv_obj)
79 {
80 	if (etnaviv_obj->sgt) {
81 		etnaviv_gem_scatterlist_unmap(etnaviv_obj);
82 		sg_free_table(etnaviv_obj->sgt);
83 		kfree(etnaviv_obj->sgt);
84 		etnaviv_obj->sgt = NULL;
85 	}
86 	if (etnaviv_obj->pages) {
87 		drm_gem_put_pages(&etnaviv_obj->base, etnaviv_obj->pages,
88 				  true, false);
89 
90 		etnaviv_obj->pages = NULL;
91 	}
92 }
93 
94 struct page **etnaviv_gem_get_pages(struct etnaviv_gem_object *etnaviv_obj)
95 {
96 	int ret;
97 
98 	lockdep_assert_held(&etnaviv_obj->lock);
99 
100 	if (!etnaviv_obj->pages) {
101 		ret = etnaviv_obj->ops->get_pages(etnaviv_obj);
102 		if (ret < 0)
103 			return ERR_PTR(ret);
104 	}
105 
106 	if (!etnaviv_obj->sgt) {
107 		struct drm_device *dev = etnaviv_obj->base.dev;
108 		int npages = etnaviv_obj->base.size >> PAGE_SHIFT;
109 		struct sg_table *sgt;
110 
111 		sgt = drm_prime_pages_to_sg(etnaviv_obj->pages, npages);
112 		if (IS_ERR(sgt)) {
113 			dev_err(dev->dev, "failed to allocate sgt: %ld\n",
114 				PTR_ERR(sgt));
115 			return ERR_CAST(sgt);
116 		}
117 
118 		etnaviv_obj->sgt = sgt;
119 
120 		etnaviv_gem_scatter_map(etnaviv_obj);
121 	}
122 
123 	return etnaviv_obj->pages;
124 }
125 
126 void etnaviv_gem_put_pages(struct etnaviv_gem_object *etnaviv_obj)
127 {
128 	lockdep_assert_held(&etnaviv_obj->lock);
129 	/* when we start tracking the pin count, then do something here */
130 }
131 
132 static int etnaviv_gem_mmap_obj(struct etnaviv_gem_object *etnaviv_obj,
133 		struct vm_area_struct *vma)
134 {
135 	pgprot_t vm_page_prot;
136 
137 	vma->vm_flags &= ~VM_PFNMAP;
138 	vma->vm_flags |= VM_MIXEDMAP;
139 
140 	vm_page_prot = vm_get_page_prot(vma->vm_flags);
141 
142 	if (etnaviv_obj->flags & ETNA_BO_WC) {
143 		vma->vm_page_prot = pgprot_writecombine(vm_page_prot);
144 	} else if (etnaviv_obj->flags & ETNA_BO_UNCACHED) {
145 		vma->vm_page_prot = pgprot_noncached(vm_page_prot);
146 	} else {
147 		/*
148 		 * Shunt off cached objs to shmem file so they have their own
149 		 * address_space (so unmap_mapping_range does what we want,
150 		 * in particular in the case of mmap'd dmabufs)
151 		 */
152 		fput(vma->vm_file);
153 		get_file(etnaviv_obj->base.filp);
154 		vma->vm_pgoff = 0;
155 		vma->vm_file  = etnaviv_obj->base.filp;
156 
157 		vma->vm_page_prot = vm_page_prot;
158 	}
159 
160 	return 0;
161 }
162 
163 int etnaviv_gem_mmap(struct file *filp, struct vm_area_struct *vma)
164 {
165 	struct etnaviv_gem_object *obj;
166 	int ret;
167 
168 	ret = drm_gem_mmap(filp, vma);
169 	if (ret) {
170 		DBG("mmap failed: %d", ret);
171 		return ret;
172 	}
173 
174 	obj = to_etnaviv_bo(vma->vm_private_data);
175 	return obj->ops->mmap(obj, vma);
176 }
177 
178 int etnaviv_gem_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
179 {
180 	struct drm_gem_object *obj = vma->vm_private_data;
181 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
182 	struct page **pages, *page;
183 	pgoff_t pgoff;
184 	int ret;
185 
186 	/*
187 	 * Make sure we don't parallel update on a fault, nor move or remove
188 	 * something from beneath our feet.  Note that vm_insert_page() is
189 	 * specifically coded to take care of this, so we don't have to.
190 	 */
191 	ret = mutex_lock_interruptible(&etnaviv_obj->lock);
192 	if (ret)
193 		goto out;
194 
195 	/* make sure we have pages attached now */
196 	pages = etnaviv_gem_get_pages(etnaviv_obj);
197 	mutex_unlock(&etnaviv_obj->lock);
198 
199 	if (IS_ERR(pages)) {
200 		ret = PTR_ERR(pages);
201 		goto out;
202 	}
203 
204 	/* We don't use vmf->pgoff since that has the fake offset: */
205 	pgoff = ((unsigned long)vmf->virtual_address -
206 			vma->vm_start) >> PAGE_SHIFT;
207 
208 	page = pages[pgoff];
209 
210 	VERB("Inserting %p pfn %lx, pa %lx", vmf->virtual_address,
211 	     page_to_pfn(page), page_to_pfn(page) << PAGE_SHIFT);
212 
213 	ret = vm_insert_page(vma, (unsigned long)vmf->virtual_address, page);
214 
215 out:
216 	switch (ret) {
217 	case -EAGAIN:
218 	case 0:
219 	case -ERESTARTSYS:
220 	case -EINTR:
221 	case -EBUSY:
222 		/*
223 		 * EBUSY is ok: this just means that another thread
224 		 * already did the job.
225 		 */
226 		return VM_FAULT_NOPAGE;
227 	case -ENOMEM:
228 		return VM_FAULT_OOM;
229 	default:
230 		return VM_FAULT_SIGBUS;
231 	}
232 }
233 
234 int etnaviv_gem_mmap_offset(struct drm_gem_object *obj, u64 *offset)
235 {
236 	int ret;
237 
238 	/* Make it mmapable */
239 	ret = drm_gem_create_mmap_offset(obj);
240 	if (ret)
241 		dev_err(obj->dev->dev, "could not allocate mmap offset\n");
242 	else
243 		*offset = drm_vma_node_offset_addr(&obj->vma_node);
244 
245 	return ret;
246 }
247 
248 static struct etnaviv_vram_mapping *
249 etnaviv_gem_get_vram_mapping(struct etnaviv_gem_object *obj,
250 			     struct etnaviv_iommu *mmu)
251 {
252 	struct etnaviv_vram_mapping *mapping;
253 
254 	list_for_each_entry(mapping, &obj->vram_list, obj_node) {
255 		if (mapping->mmu == mmu)
256 			return mapping;
257 	}
258 
259 	return NULL;
260 }
261 
262 void etnaviv_gem_mapping_reference(struct etnaviv_vram_mapping *mapping)
263 {
264 	struct etnaviv_gem_object *etnaviv_obj = mapping->object;
265 
266 	drm_gem_object_reference(&etnaviv_obj->base);
267 
268 	mutex_lock(&etnaviv_obj->lock);
269 	WARN_ON(mapping->use == 0);
270 	mapping->use += 1;
271 	mutex_unlock(&etnaviv_obj->lock);
272 }
273 
274 void etnaviv_gem_mapping_unreference(struct etnaviv_vram_mapping *mapping)
275 {
276 	struct etnaviv_gem_object *etnaviv_obj = mapping->object;
277 
278 	mutex_lock(&etnaviv_obj->lock);
279 	WARN_ON(mapping->use == 0);
280 	mapping->use -= 1;
281 	mutex_unlock(&etnaviv_obj->lock);
282 
283 	drm_gem_object_unreference_unlocked(&etnaviv_obj->base);
284 }
285 
286 struct etnaviv_vram_mapping *etnaviv_gem_mapping_get(
287 	struct drm_gem_object *obj, struct etnaviv_gpu *gpu)
288 {
289 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
290 	struct etnaviv_vram_mapping *mapping;
291 	struct page **pages;
292 	int ret = 0;
293 
294 	mutex_lock(&etnaviv_obj->lock);
295 	mapping = etnaviv_gem_get_vram_mapping(etnaviv_obj, gpu->mmu);
296 	if (mapping) {
297 		/*
298 		 * Holding the object lock prevents the use count changing
299 		 * beneath us.  If the use count is zero, the MMU might be
300 		 * reaping this object, so take the lock and re-check that
301 		 * the MMU owns this mapping to close this race.
302 		 */
303 		if (mapping->use == 0) {
304 			mutex_lock(&gpu->mmu->lock);
305 			if (mapping->mmu == gpu->mmu)
306 				mapping->use += 1;
307 			else
308 				mapping = NULL;
309 			mutex_unlock(&gpu->mmu->lock);
310 			if (mapping)
311 				goto out;
312 		} else {
313 			mapping->use += 1;
314 			goto out;
315 		}
316 	}
317 
318 	pages = etnaviv_gem_get_pages(etnaviv_obj);
319 	if (IS_ERR(pages)) {
320 		ret = PTR_ERR(pages);
321 		goto out;
322 	}
323 
324 	/*
325 	 * See if we have a reaped vram mapping we can re-use before
326 	 * allocating a fresh mapping.
327 	 */
328 	mapping = etnaviv_gem_get_vram_mapping(etnaviv_obj, NULL);
329 	if (!mapping) {
330 		mapping = kzalloc(sizeof(*mapping), GFP_KERNEL);
331 		if (!mapping) {
332 			ret = -ENOMEM;
333 			goto out;
334 		}
335 
336 		INIT_LIST_HEAD(&mapping->scan_node);
337 		mapping->object = etnaviv_obj;
338 	} else {
339 		list_del(&mapping->obj_node);
340 	}
341 
342 	mapping->mmu = gpu->mmu;
343 	mapping->use = 1;
344 
345 	ret = etnaviv_iommu_map_gem(gpu->mmu, etnaviv_obj, gpu->memory_base,
346 				    mapping);
347 	if (ret < 0)
348 		kfree(mapping);
349 	else
350 		list_add_tail(&mapping->obj_node, &etnaviv_obj->vram_list);
351 
352 out:
353 	mutex_unlock(&etnaviv_obj->lock);
354 
355 	if (ret)
356 		return ERR_PTR(ret);
357 
358 	/* Take a reference on the object */
359 	drm_gem_object_reference(obj);
360 	return mapping;
361 }
362 
363 void *etnaviv_gem_vmap(struct drm_gem_object *obj)
364 {
365 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
366 
367 	if (etnaviv_obj->vaddr)
368 		return etnaviv_obj->vaddr;
369 
370 	mutex_lock(&etnaviv_obj->lock);
371 	/*
372 	 * Need to check again, as we might have raced with another thread
373 	 * while waiting for the mutex.
374 	 */
375 	if (!etnaviv_obj->vaddr)
376 		etnaviv_obj->vaddr = etnaviv_obj->ops->vmap(etnaviv_obj);
377 	mutex_unlock(&etnaviv_obj->lock);
378 
379 	return etnaviv_obj->vaddr;
380 }
381 
382 static void *etnaviv_gem_vmap_impl(struct etnaviv_gem_object *obj)
383 {
384 	struct page **pages;
385 
386 	lockdep_assert_held(&obj->lock);
387 
388 	pages = etnaviv_gem_get_pages(obj);
389 	if (IS_ERR(pages))
390 		return NULL;
391 
392 	return vmap(pages, obj->base.size >> PAGE_SHIFT,
393 			VM_MAP, pgprot_writecombine(PAGE_KERNEL));
394 }
395 
396 static inline enum dma_data_direction etnaviv_op_to_dma_dir(u32 op)
397 {
398 	if (op & ETNA_PREP_READ)
399 		return DMA_FROM_DEVICE;
400 	else if (op & ETNA_PREP_WRITE)
401 		return DMA_TO_DEVICE;
402 	else
403 		return DMA_BIDIRECTIONAL;
404 }
405 
406 int etnaviv_gem_cpu_prep(struct drm_gem_object *obj, u32 op,
407 		struct timespec *timeout)
408 {
409 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
410 	struct drm_device *dev = obj->dev;
411 	bool write = !!(op & ETNA_PREP_WRITE);
412 	int ret;
413 
414 	if (op & ETNA_PREP_NOSYNC) {
415 		if (!reservation_object_test_signaled_rcu(etnaviv_obj->resv,
416 							  write))
417 			return -EBUSY;
418 	} else {
419 		unsigned long remain = etnaviv_timeout_to_jiffies(timeout);
420 
421 		ret = reservation_object_wait_timeout_rcu(etnaviv_obj->resv,
422 							  write, true, remain);
423 		if (ret <= 0)
424 			return ret == 0 ? -ETIMEDOUT : ret;
425 	}
426 
427 	if (etnaviv_obj->flags & ETNA_BO_CACHED) {
428 		if (!etnaviv_obj->sgt) {
429 			void *ret;
430 
431 			mutex_lock(&etnaviv_obj->lock);
432 			ret = etnaviv_gem_get_pages(etnaviv_obj);
433 			mutex_unlock(&etnaviv_obj->lock);
434 			if (IS_ERR(ret))
435 				return PTR_ERR(ret);
436 		}
437 
438 		dma_sync_sg_for_cpu(dev->dev, etnaviv_obj->sgt->sgl,
439 				    etnaviv_obj->sgt->nents,
440 				    etnaviv_op_to_dma_dir(op));
441 		etnaviv_obj->last_cpu_prep_op = op;
442 	}
443 
444 	return 0;
445 }
446 
447 int etnaviv_gem_cpu_fini(struct drm_gem_object *obj)
448 {
449 	struct drm_device *dev = obj->dev;
450 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
451 
452 	if (etnaviv_obj->flags & ETNA_BO_CACHED) {
453 		/* fini without a prep is almost certainly a userspace error */
454 		WARN_ON(etnaviv_obj->last_cpu_prep_op == 0);
455 		dma_sync_sg_for_device(dev->dev, etnaviv_obj->sgt->sgl,
456 			etnaviv_obj->sgt->nents,
457 			etnaviv_op_to_dma_dir(etnaviv_obj->last_cpu_prep_op));
458 		etnaviv_obj->last_cpu_prep_op = 0;
459 	}
460 
461 	return 0;
462 }
463 
464 int etnaviv_gem_wait_bo(struct etnaviv_gpu *gpu, struct drm_gem_object *obj,
465 	struct timespec *timeout)
466 {
467 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
468 
469 	return etnaviv_gpu_wait_obj_inactive(gpu, etnaviv_obj, timeout);
470 }
471 
472 #ifdef CONFIG_DEBUG_FS
473 static void etnaviv_gem_describe_fence(struct fence *fence,
474 	const char *type, struct seq_file *m)
475 {
476 	if (!test_bit(FENCE_FLAG_SIGNALED_BIT, &fence->flags))
477 		seq_printf(m, "\t%9s: %s %s seq %u\n",
478 			   type,
479 			   fence->ops->get_driver_name(fence),
480 			   fence->ops->get_timeline_name(fence),
481 			   fence->seqno);
482 }
483 
484 static void etnaviv_gem_describe(struct drm_gem_object *obj, struct seq_file *m)
485 {
486 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
487 	struct reservation_object *robj = etnaviv_obj->resv;
488 	struct reservation_object_list *fobj;
489 	struct fence *fence;
490 	unsigned long off = drm_vma_node_start(&obj->vma_node);
491 
492 	seq_printf(m, "%08x: %c %2d (%2d) %08lx %p %zd\n",
493 			etnaviv_obj->flags, is_active(etnaviv_obj) ? 'A' : 'I',
494 			obj->name, obj->refcount.refcount.counter,
495 			off, etnaviv_obj->vaddr, obj->size);
496 
497 	rcu_read_lock();
498 	fobj = rcu_dereference(robj->fence);
499 	if (fobj) {
500 		unsigned int i, shared_count = fobj->shared_count;
501 
502 		for (i = 0; i < shared_count; i++) {
503 			fence = rcu_dereference(fobj->shared[i]);
504 			etnaviv_gem_describe_fence(fence, "Shared", m);
505 		}
506 	}
507 
508 	fence = rcu_dereference(robj->fence_excl);
509 	if (fence)
510 		etnaviv_gem_describe_fence(fence, "Exclusive", m);
511 	rcu_read_unlock();
512 }
513 
514 void etnaviv_gem_describe_objects(struct etnaviv_drm_private *priv,
515 	struct seq_file *m)
516 {
517 	struct etnaviv_gem_object *etnaviv_obj;
518 	int count = 0;
519 	size_t size = 0;
520 
521 	mutex_lock(&priv->gem_lock);
522 	list_for_each_entry(etnaviv_obj, &priv->gem_list, gem_node) {
523 		struct drm_gem_object *obj = &etnaviv_obj->base;
524 
525 		seq_puts(m, "   ");
526 		etnaviv_gem_describe(obj, m);
527 		count++;
528 		size += obj->size;
529 	}
530 	mutex_unlock(&priv->gem_lock);
531 
532 	seq_printf(m, "Total %d objects, %zu bytes\n", count, size);
533 }
534 #endif
535 
536 static void etnaviv_gem_shmem_release(struct etnaviv_gem_object *etnaviv_obj)
537 {
538 	vunmap(etnaviv_obj->vaddr);
539 	put_pages(etnaviv_obj);
540 }
541 
542 static const struct etnaviv_gem_ops etnaviv_gem_shmem_ops = {
543 	.get_pages = etnaviv_gem_shmem_get_pages,
544 	.release = etnaviv_gem_shmem_release,
545 	.vmap = etnaviv_gem_vmap_impl,
546 	.mmap = etnaviv_gem_mmap_obj,
547 };
548 
549 void etnaviv_gem_free_object(struct drm_gem_object *obj)
550 {
551 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
552 	struct etnaviv_vram_mapping *mapping, *tmp;
553 
554 	/* object should not be active */
555 	WARN_ON(is_active(etnaviv_obj));
556 
557 	list_del(&etnaviv_obj->gem_node);
558 
559 	list_for_each_entry_safe(mapping, tmp, &etnaviv_obj->vram_list,
560 				 obj_node) {
561 		struct etnaviv_iommu *mmu = mapping->mmu;
562 
563 		WARN_ON(mapping->use);
564 
565 		if (mmu)
566 			etnaviv_iommu_unmap_gem(mmu, mapping);
567 
568 		list_del(&mapping->obj_node);
569 		kfree(mapping);
570 	}
571 
572 	drm_gem_free_mmap_offset(obj);
573 	etnaviv_obj->ops->release(etnaviv_obj);
574 	if (etnaviv_obj->resv == &etnaviv_obj->_resv)
575 		reservation_object_fini(&etnaviv_obj->_resv);
576 	drm_gem_object_release(obj);
577 
578 	kfree(etnaviv_obj);
579 }
580 
581 int etnaviv_gem_obj_add(struct drm_device *dev, struct drm_gem_object *obj)
582 {
583 	struct etnaviv_drm_private *priv = dev->dev_private;
584 	struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
585 
586 	mutex_lock(&priv->gem_lock);
587 	list_add_tail(&etnaviv_obj->gem_node, &priv->gem_list);
588 	mutex_unlock(&priv->gem_lock);
589 
590 	return 0;
591 }
592 
593 static int etnaviv_gem_new_impl(struct drm_device *dev, u32 size, u32 flags,
594 	struct reservation_object *robj, const struct etnaviv_gem_ops *ops,
595 	struct drm_gem_object **obj)
596 {
597 	struct etnaviv_gem_object *etnaviv_obj;
598 	unsigned sz = sizeof(*etnaviv_obj);
599 	bool valid = true;
600 
601 	/* validate flags */
602 	switch (flags & ETNA_BO_CACHE_MASK) {
603 	case ETNA_BO_UNCACHED:
604 	case ETNA_BO_CACHED:
605 	case ETNA_BO_WC:
606 		break;
607 	default:
608 		valid = false;
609 	}
610 
611 	if (!valid) {
612 		dev_err(dev->dev, "invalid cache flag: %x\n",
613 			(flags & ETNA_BO_CACHE_MASK));
614 		return -EINVAL;
615 	}
616 
617 	etnaviv_obj = kzalloc(sz, GFP_KERNEL);
618 	if (!etnaviv_obj)
619 		return -ENOMEM;
620 
621 	etnaviv_obj->flags = flags;
622 	etnaviv_obj->ops = ops;
623 	if (robj) {
624 		etnaviv_obj->resv = robj;
625 	} else {
626 		etnaviv_obj->resv = &etnaviv_obj->_resv;
627 		reservation_object_init(&etnaviv_obj->_resv);
628 	}
629 
630 	mutex_init(&etnaviv_obj->lock);
631 	INIT_LIST_HEAD(&etnaviv_obj->vram_list);
632 
633 	*obj = &etnaviv_obj->base;
634 
635 	return 0;
636 }
637 
638 static struct drm_gem_object *__etnaviv_gem_new(struct drm_device *dev,
639 		u32 size, u32 flags)
640 {
641 	struct drm_gem_object *obj = NULL;
642 	int ret;
643 
644 	size = PAGE_ALIGN(size);
645 
646 	ret = etnaviv_gem_new_impl(dev, size, flags, NULL,
647 				   &etnaviv_gem_shmem_ops, &obj);
648 	if (ret)
649 		goto fail;
650 
651 	ret = drm_gem_object_init(dev, obj, size);
652 	if (ret == 0) {
653 		struct address_space *mapping;
654 
655 		/*
656 		 * Our buffers are kept pinned, so allocating them
657 		 * from the MOVABLE zone is a really bad idea, and
658 		 * conflicts with CMA.  See coments above new_inode()
659 		 * why this is required _and_ expected if you're
660 		 * going to pin these pages.
661 		 */
662 		mapping = obj->filp->f_mapping;
663 		mapping_set_gfp_mask(mapping, GFP_HIGHUSER);
664 	}
665 
666 	if (ret)
667 		goto fail;
668 
669 	return obj;
670 
671 fail:
672 	drm_gem_object_unreference_unlocked(obj);
673 	return ERR_PTR(ret);
674 }
675 
676 /* convenience method to construct a GEM buffer object, and userspace handle */
677 int etnaviv_gem_new_handle(struct drm_device *dev, struct drm_file *file,
678 		u32 size, u32 flags, u32 *handle)
679 {
680 	struct drm_gem_object *obj;
681 	int ret;
682 
683 	obj = __etnaviv_gem_new(dev, size, flags);
684 	if (IS_ERR(obj))
685 		return PTR_ERR(obj);
686 
687 	ret = etnaviv_gem_obj_add(dev, obj);
688 	if (ret < 0) {
689 		drm_gem_object_unreference_unlocked(obj);
690 		return ret;
691 	}
692 
693 	ret = drm_gem_handle_create(file, obj, handle);
694 
695 	/* drop reference from allocate - handle holds it now */
696 	drm_gem_object_unreference_unlocked(obj);
697 
698 	return ret;
699 }
700 
701 struct drm_gem_object *etnaviv_gem_new(struct drm_device *dev,
702 		u32 size, u32 flags)
703 {
704 	struct drm_gem_object *obj;
705 	int ret;
706 
707 	obj = __etnaviv_gem_new(dev, size, flags);
708 	if (IS_ERR(obj))
709 		return obj;
710 
711 	ret = etnaviv_gem_obj_add(dev, obj);
712 	if (ret < 0) {
713 		drm_gem_object_unreference_unlocked(obj);
714 		return ERR_PTR(ret);
715 	}
716 
717 	return obj;
718 }
719 
720 int etnaviv_gem_new_private(struct drm_device *dev, size_t size, u32 flags,
721 	struct reservation_object *robj, const struct etnaviv_gem_ops *ops,
722 	struct etnaviv_gem_object **res)
723 {
724 	struct drm_gem_object *obj;
725 	int ret;
726 
727 	ret = etnaviv_gem_new_impl(dev, size, flags, robj, ops, &obj);
728 	if (ret)
729 		return ret;
730 
731 	drm_gem_private_object_init(dev, obj, size);
732 
733 	*res = to_etnaviv_bo(obj);
734 
735 	return 0;
736 }
737 
738 struct get_pages_work {
739 	struct work_struct work;
740 	struct mm_struct *mm;
741 	struct task_struct *task;
742 	struct etnaviv_gem_object *etnaviv_obj;
743 };
744 
745 static struct page **etnaviv_gem_userptr_do_get_pages(
746 	struct etnaviv_gem_object *etnaviv_obj, struct mm_struct *mm, struct task_struct *task)
747 {
748 	int ret = 0, pinned, npages = etnaviv_obj->base.size >> PAGE_SHIFT;
749 	struct page **pvec;
750 	uintptr_t ptr;
751 
752 	pvec = drm_malloc_ab(npages, sizeof(struct page *));
753 	if (!pvec)
754 		return ERR_PTR(-ENOMEM);
755 
756 	pinned = 0;
757 	ptr = etnaviv_obj->userptr.ptr;
758 
759 	down_read(&mm->mmap_sem);
760 	while (pinned < npages) {
761 		ret = get_user_pages_remote(task, mm, ptr, npages - pinned,
762 					    !etnaviv_obj->userptr.ro, 0,
763 					    pvec + pinned, NULL);
764 		if (ret < 0)
765 			break;
766 
767 		ptr += ret * PAGE_SIZE;
768 		pinned += ret;
769 	}
770 	up_read(&mm->mmap_sem);
771 
772 	if (ret < 0) {
773 		release_pages(pvec, pinned, 0);
774 		drm_free_large(pvec);
775 		return ERR_PTR(ret);
776 	}
777 
778 	return pvec;
779 }
780 
781 static void __etnaviv_gem_userptr_get_pages(struct work_struct *_work)
782 {
783 	struct get_pages_work *work = container_of(_work, typeof(*work), work);
784 	struct etnaviv_gem_object *etnaviv_obj = work->etnaviv_obj;
785 	struct page **pvec;
786 
787 	pvec = etnaviv_gem_userptr_do_get_pages(etnaviv_obj, work->mm, work->task);
788 
789 	mutex_lock(&etnaviv_obj->lock);
790 	if (IS_ERR(pvec)) {
791 		etnaviv_obj->userptr.work = ERR_CAST(pvec);
792 	} else {
793 		etnaviv_obj->userptr.work = NULL;
794 		etnaviv_obj->pages = pvec;
795 	}
796 
797 	mutex_unlock(&etnaviv_obj->lock);
798 	drm_gem_object_unreference_unlocked(&etnaviv_obj->base);
799 
800 	mmput(work->mm);
801 	put_task_struct(work->task);
802 	kfree(work);
803 }
804 
805 static int etnaviv_gem_userptr_get_pages(struct etnaviv_gem_object *etnaviv_obj)
806 {
807 	struct page **pvec = NULL;
808 	struct get_pages_work *work;
809 	struct mm_struct *mm;
810 	int ret, pinned, npages = etnaviv_obj->base.size >> PAGE_SHIFT;
811 
812 	if (etnaviv_obj->userptr.work) {
813 		if (IS_ERR(etnaviv_obj->userptr.work)) {
814 			ret = PTR_ERR(etnaviv_obj->userptr.work);
815 			etnaviv_obj->userptr.work = NULL;
816 		} else {
817 			ret = -EAGAIN;
818 		}
819 		return ret;
820 	}
821 
822 	mm = get_task_mm(etnaviv_obj->userptr.task);
823 	pinned = 0;
824 	if (mm == current->mm) {
825 		pvec = drm_malloc_ab(npages, sizeof(struct page *));
826 		if (!pvec) {
827 			mmput(mm);
828 			return -ENOMEM;
829 		}
830 
831 		pinned = __get_user_pages_fast(etnaviv_obj->userptr.ptr, npages,
832 					       !etnaviv_obj->userptr.ro, pvec);
833 		if (pinned < 0) {
834 			drm_free_large(pvec);
835 			mmput(mm);
836 			return pinned;
837 		}
838 
839 		if (pinned == npages) {
840 			etnaviv_obj->pages = pvec;
841 			mmput(mm);
842 			return 0;
843 		}
844 	}
845 
846 	release_pages(pvec, pinned, 0);
847 	drm_free_large(pvec);
848 
849 	work = kmalloc(sizeof(*work), GFP_KERNEL);
850 	if (!work) {
851 		mmput(mm);
852 		return -ENOMEM;
853 	}
854 
855 	get_task_struct(current);
856 	drm_gem_object_reference(&etnaviv_obj->base);
857 
858 	work->mm = mm;
859 	work->task = current;
860 	work->etnaviv_obj = etnaviv_obj;
861 
862 	etnaviv_obj->userptr.work = &work->work;
863 	INIT_WORK(&work->work, __etnaviv_gem_userptr_get_pages);
864 
865 	etnaviv_queue_work(etnaviv_obj->base.dev, &work->work);
866 
867 	return -EAGAIN;
868 }
869 
870 static void etnaviv_gem_userptr_release(struct etnaviv_gem_object *etnaviv_obj)
871 {
872 	if (etnaviv_obj->sgt) {
873 		etnaviv_gem_scatterlist_unmap(etnaviv_obj);
874 		sg_free_table(etnaviv_obj->sgt);
875 		kfree(etnaviv_obj->sgt);
876 	}
877 	if (etnaviv_obj->pages) {
878 		int npages = etnaviv_obj->base.size >> PAGE_SHIFT;
879 
880 		release_pages(etnaviv_obj->pages, npages, 0);
881 		drm_free_large(etnaviv_obj->pages);
882 	}
883 	put_task_struct(etnaviv_obj->userptr.task);
884 }
885 
886 static int etnaviv_gem_userptr_mmap_obj(struct etnaviv_gem_object *etnaviv_obj,
887 		struct vm_area_struct *vma)
888 {
889 	return -EINVAL;
890 }
891 
892 static const struct etnaviv_gem_ops etnaviv_gem_userptr_ops = {
893 	.get_pages = etnaviv_gem_userptr_get_pages,
894 	.release = etnaviv_gem_userptr_release,
895 	.vmap = etnaviv_gem_vmap_impl,
896 	.mmap = etnaviv_gem_userptr_mmap_obj,
897 };
898 
899 int etnaviv_gem_new_userptr(struct drm_device *dev, struct drm_file *file,
900 	uintptr_t ptr, u32 size, u32 flags, u32 *handle)
901 {
902 	struct etnaviv_gem_object *etnaviv_obj;
903 	int ret;
904 
905 	ret = etnaviv_gem_new_private(dev, size, ETNA_BO_CACHED, NULL,
906 				      &etnaviv_gem_userptr_ops, &etnaviv_obj);
907 	if (ret)
908 		return ret;
909 
910 	etnaviv_obj->userptr.ptr = ptr;
911 	etnaviv_obj->userptr.task = current;
912 	etnaviv_obj->userptr.ro = !(flags & ETNA_USERPTR_WRITE);
913 	get_task_struct(current);
914 
915 	ret = etnaviv_gem_obj_add(dev, &etnaviv_obj->base);
916 	if (ret)
917 		goto unreference;
918 
919 	ret = drm_gem_handle_create(file, &etnaviv_obj->base, handle);
920 unreference:
921 	/* drop reference from allocate - handle holds it now */
922 	drm_gem_object_unreference_unlocked(&etnaviv_obj->base);
923 	return ret;
924 }
925