drm/msm: change gem->vmap() to get/put
[cascardo/linux.git] / drivers / gpu / drm / msm / msm_gem.c
1 /*
2  * Copyright (C) 2013 Red Hat
3  * Author: Rob Clark <robdclark@gmail.com>
4  *
5  * This program is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 as published by
7  * the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
12  * more details.
13  *
14  * You should have received a copy of the GNU General Public License along with
15  * this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
17
18 #include <linux/spinlock.h>
19 #include <linux/shmem_fs.h>
20 #include <linux/dma-buf.h>
21 #include <linux/pfn_t.h>
22
23 #include "msm_drv.h"
24 #include "msm_fence.h"
25 #include "msm_gem.h"
26 #include "msm_gpu.h"
27 #include "msm_mmu.h"
28
29 static dma_addr_t physaddr(struct drm_gem_object *obj)
30 {
31         struct msm_gem_object *msm_obj = to_msm_bo(obj);
32         struct msm_drm_private *priv = obj->dev->dev_private;
33         return (((dma_addr_t)msm_obj->vram_node->start) << PAGE_SHIFT) +
34                         priv->vram.paddr;
35 }
36
37 static bool use_pages(struct drm_gem_object *obj)
38 {
39         struct msm_gem_object *msm_obj = to_msm_bo(obj);
40         return !msm_obj->vram_node;
41 }
42
43 /* allocate pages from VRAM carveout, used when no IOMMU: */
44 static struct page **get_pages_vram(struct drm_gem_object *obj,
45                 int npages)
46 {
47         struct msm_gem_object *msm_obj = to_msm_bo(obj);
48         struct msm_drm_private *priv = obj->dev->dev_private;
49         dma_addr_t paddr;
50         struct page **p;
51         int ret, i;
52
53         p = drm_malloc_ab(npages, sizeof(struct page *));
54         if (!p)
55                 return ERR_PTR(-ENOMEM);
56
57         ret = drm_mm_insert_node(&priv->vram.mm, msm_obj->vram_node,
58                         npages, 0, DRM_MM_SEARCH_DEFAULT);
59         if (ret) {
60                 drm_free_large(p);
61                 return ERR_PTR(ret);
62         }
63
64         paddr = physaddr(obj);
65         for (i = 0; i < npages; i++) {
66                 p[i] = phys_to_page(paddr);
67                 paddr += PAGE_SIZE;
68         }
69
70         return p;
71 }
72
73 /* called with dev->struct_mutex held */
74 static struct page **get_pages(struct drm_gem_object *obj)
75 {
76         struct msm_gem_object *msm_obj = to_msm_bo(obj);
77
78         if (!msm_obj->pages) {
79                 struct drm_device *dev = obj->dev;
80                 struct page **p;
81                 int npages = obj->size >> PAGE_SHIFT;
82
83                 if (use_pages(obj))
84                         p = drm_gem_get_pages(obj);
85                 else
86                         p = get_pages_vram(obj, npages);
87
88                 if (IS_ERR(p)) {
89                         dev_err(dev->dev, "could not get pages: %ld\n",
90                                         PTR_ERR(p));
91                         return p;
92                 }
93
94                 msm_obj->sgt = drm_prime_pages_to_sg(p, npages);
95                 if (IS_ERR(msm_obj->sgt)) {
96                         dev_err(dev->dev, "failed to allocate sgt\n");
97                         return ERR_CAST(msm_obj->sgt);
98                 }
99
100                 msm_obj->pages = p;
101
102                 /* For non-cached buffers, ensure the new pages are clean
103                  * because display controller, GPU, etc. are not coherent:
104                  */
105                 if (msm_obj->flags & (MSM_BO_WC|MSM_BO_UNCACHED))
106                         dma_map_sg(dev->dev, msm_obj->sgt->sgl,
107                                         msm_obj->sgt->nents, DMA_BIDIRECTIONAL);
108         }
109
110         return msm_obj->pages;
111 }
112
113 static void put_pages(struct drm_gem_object *obj)
114 {
115         struct msm_gem_object *msm_obj = to_msm_bo(obj);
116
117         if (msm_obj->pages) {
118                 /* For non-cached buffers, ensure the new pages are clean
119                  * because display controller, GPU, etc. are not coherent:
120                  */
121                 if (msm_obj->flags & (MSM_BO_WC|MSM_BO_UNCACHED))
122                         dma_unmap_sg(obj->dev->dev, msm_obj->sgt->sgl,
123                                         msm_obj->sgt->nents, DMA_BIDIRECTIONAL);
124                 sg_free_table(msm_obj->sgt);
125                 kfree(msm_obj->sgt);
126
127                 if (use_pages(obj))
128                         drm_gem_put_pages(obj, msm_obj->pages, true, false);
129                 else {
130                         drm_mm_remove_node(msm_obj->vram_node);
131                         drm_free_large(msm_obj->pages);
132                 }
133
134                 msm_obj->pages = NULL;
135         }
136 }
137
138 struct page **msm_gem_get_pages(struct drm_gem_object *obj)
139 {
140         struct drm_device *dev = obj->dev;
141         struct page **p;
142         mutex_lock(&dev->struct_mutex);
143         p = get_pages(obj);
144         mutex_unlock(&dev->struct_mutex);
145         return p;
146 }
147
148 void msm_gem_put_pages(struct drm_gem_object *obj)
149 {
150         /* when we start tracking the pin count, then do something here */
151 }
152
153 int msm_gem_mmap_obj(struct drm_gem_object *obj,
154                 struct vm_area_struct *vma)
155 {
156         struct msm_gem_object *msm_obj = to_msm_bo(obj);
157
158         vma->vm_flags &= ~VM_PFNMAP;
159         vma->vm_flags |= VM_MIXEDMAP;
160
161         if (msm_obj->flags & MSM_BO_WC) {
162                 vma->vm_page_prot = pgprot_writecombine(vm_get_page_prot(vma->vm_flags));
163         } else if (msm_obj->flags & MSM_BO_UNCACHED) {
164                 vma->vm_page_prot = pgprot_noncached(vm_get_page_prot(vma->vm_flags));
165         } else {
166                 /*
167                  * Shunt off cached objs to shmem file so they have their own
168                  * address_space (so unmap_mapping_range does what we want,
169                  * in particular in the case of mmap'd dmabufs)
170                  */
171                 fput(vma->vm_file);
172                 get_file(obj->filp);
173                 vma->vm_pgoff = 0;
174                 vma->vm_file  = obj->filp;
175
176                 vma->vm_page_prot = vm_get_page_prot(vma->vm_flags);
177         }
178
179         return 0;
180 }
181
182 int msm_gem_mmap(struct file *filp, struct vm_area_struct *vma)
183 {
184         int ret;
185
186         ret = drm_gem_mmap(filp, vma);
187         if (ret) {
188                 DBG("mmap failed: %d", ret);
189                 return ret;
190         }
191
192         return msm_gem_mmap_obj(vma->vm_private_data, vma);
193 }
194
195 int msm_gem_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
196 {
197         struct drm_gem_object *obj = vma->vm_private_data;
198         struct drm_device *dev = obj->dev;
199         struct page **pages;
200         unsigned long pfn;
201         pgoff_t pgoff;
202         int ret;
203
204         /* Make sure we don't parallel update on a fault, nor move or remove
205          * something from beneath our feet
206          */
207         ret = mutex_lock_interruptible(&dev->struct_mutex);
208         if (ret)
209                 goto out;
210
211         /* make sure we have pages attached now */
212         pages = get_pages(obj);
213         if (IS_ERR(pages)) {
214                 ret = PTR_ERR(pages);
215                 goto out_unlock;
216         }
217
218         /* We don't use vmf->pgoff since that has the fake offset: */
219         pgoff = ((unsigned long)vmf->virtual_address -
220                         vma->vm_start) >> PAGE_SHIFT;
221
222         pfn = page_to_pfn(pages[pgoff]);
223
224         VERB("Inserting %p pfn %lx, pa %lx", vmf->virtual_address,
225                         pfn, pfn << PAGE_SHIFT);
226
227         ret = vm_insert_mixed(vma, (unsigned long)vmf->virtual_address,
228                         __pfn_to_pfn_t(pfn, PFN_DEV));
229
230 out_unlock:
231         mutex_unlock(&dev->struct_mutex);
232 out:
233         switch (ret) {
234         case -EAGAIN:
235         case 0:
236         case -ERESTARTSYS:
237         case -EINTR:
238         case -EBUSY:
239                 /*
240                  * EBUSY is ok: this just means that another thread
241                  * already did the job.
242                  */
243                 return VM_FAULT_NOPAGE;
244         case -ENOMEM:
245                 return VM_FAULT_OOM;
246         default:
247                 return VM_FAULT_SIGBUS;
248         }
249 }
250
251 /** get mmap offset */
252 static uint64_t mmap_offset(struct drm_gem_object *obj)
253 {
254         struct drm_device *dev = obj->dev;
255         int ret;
256
257         WARN_ON(!mutex_is_locked(&dev->struct_mutex));
258
259         /* Make it mmapable */
260         ret = drm_gem_create_mmap_offset(obj);
261
262         if (ret) {
263                 dev_err(dev->dev, "could not allocate mmap offset\n");
264                 return 0;
265         }
266
267         return drm_vma_node_offset_addr(&obj->vma_node);
268 }
269
270 uint64_t msm_gem_mmap_offset(struct drm_gem_object *obj)
271 {
272         uint64_t offset;
273         mutex_lock(&obj->dev->struct_mutex);
274         offset = mmap_offset(obj);
275         mutex_unlock(&obj->dev->struct_mutex);
276         return offset;
277 }
278
279 static void
280 put_iova(struct drm_gem_object *obj)
281 {
282         struct drm_device *dev = obj->dev;
283         struct msm_drm_private *priv = obj->dev->dev_private;
284         struct msm_gem_object *msm_obj = to_msm_bo(obj);
285         int id;
286
287         WARN_ON(!mutex_is_locked(&dev->struct_mutex));
288
289         for (id = 0; id < ARRAY_SIZE(msm_obj->domain); id++) {
290                 struct msm_mmu *mmu = priv->mmus[id];
291                 if (mmu && msm_obj->domain[id].iova) {
292                         uint32_t offset = msm_obj->domain[id].iova;
293                         mmu->funcs->unmap(mmu, offset, msm_obj->sgt, obj->size);
294                         msm_obj->domain[id].iova = 0;
295                 }
296         }
297 }
298
299 /* should be called under struct_mutex.. although it can be called
300  * from atomic context without struct_mutex to acquire an extra
301  * iova ref if you know one is already held.
302  *
303  * That means when I do eventually need to add support for unpinning
304  * the refcnt counter needs to be atomic_t.
305  */
306 int msm_gem_get_iova_locked(struct drm_gem_object *obj, int id,
307                 uint32_t *iova)
308 {
309         struct msm_gem_object *msm_obj = to_msm_bo(obj);
310         int ret = 0;
311
312         if (!msm_obj->domain[id].iova) {
313                 struct msm_drm_private *priv = obj->dev->dev_private;
314                 struct page **pages = get_pages(obj);
315
316                 if (IS_ERR(pages))
317                         return PTR_ERR(pages);
318
319                 if (iommu_present(&platform_bus_type)) {
320                         struct msm_mmu *mmu = priv->mmus[id];
321                         uint32_t offset;
322
323                         if (WARN_ON(!mmu))
324                                 return -EINVAL;
325
326                         offset = (uint32_t)mmap_offset(obj);
327                         ret = mmu->funcs->map(mmu, offset, msm_obj->sgt,
328                                         obj->size, IOMMU_READ | IOMMU_WRITE);
329                         msm_obj->domain[id].iova = offset;
330                 } else {
331                         msm_obj->domain[id].iova = physaddr(obj);
332                 }
333         }
334
335         if (!ret)
336                 *iova = msm_obj->domain[id].iova;
337
338         return ret;
339 }
340
341 /* get iova, taking a reference.  Should have a matching put */
342 int msm_gem_get_iova(struct drm_gem_object *obj, int id, uint32_t *iova)
343 {
344         struct msm_gem_object *msm_obj = to_msm_bo(obj);
345         int ret;
346
347         /* this is safe right now because we don't unmap until the
348          * bo is deleted:
349          */
350         if (msm_obj->domain[id].iova) {
351                 *iova = msm_obj->domain[id].iova;
352                 return 0;
353         }
354
355         mutex_lock(&obj->dev->struct_mutex);
356         ret = msm_gem_get_iova_locked(obj, id, iova);
357         mutex_unlock(&obj->dev->struct_mutex);
358         return ret;
359 }
360
361 /* get iova without taking a reference, used in places where you have
362  * already done a 'msm_gem_get_iova()'.
363  */
364 uint32_t msm_gem_iova(struct drm_gem_object *obj, int id)
365 {
366         struct msm_gem_object *msm_obj = to_msm_bo(obj);
367         WARN_ON(!msm_obj->domain[id].iova);
368         return msm_obj->domain[id].iova;
369 }
370
371 void msm_gem_put_iova(struct drm_gem_object *obj, int id)
372 {
373         // XXX TODO ..
374         // NOTE: probably don't need a _locked() version.. we wouldn't
375         // normally unmap here, but instead just mark that it could be
376         // unmapped (if the iova refcnt drops to zero), but then later
377         // if another _get_iova_locked() fails we can start unmapping
378         // things that are no longer needed..
379 }
380
381 int msm_gem_dumb_create(struct drm_file *file, struct drm_device *dev,
382                 struct drm_mode_create_dumb *args)
383 {
384         args->pitch = align_pitch(args->width, args->bpp);
385         args->size  = PAGE_ALIGN(args->pitch * args->height);
386         return msm_gem_new_handle(dev, file, args->size,
387                         MSM_BO_SCANOUT | MSM_BO_WC, &args->handle);
388 }
389
390 int msm_gem_dumb_map_offset(struct drm_file *file, struct drm_device *dev,
391                 uint32_t handle, uint64_t *offset)
392 {
393         struct drm_gem_object *obj;
394         int ret = 0;
395
396         /* GEM does all our handle to object mapping */
397         obj = drm_gem_object_lookup(file, handle);
398         if (obj == NULL) {
399                 ret = -ENOENT;
400                 goto fail;
401         }
402
403         *offset = msm_gem_mmap_offset(obj);
404
405         drm_gem_object_unreference_unlocked(obj);
406
407 fail:
408         return ret;
409 }
410
411 void *msm_gem_get_vaddr_locked(struct drm_gem_object *obj)
412 {
413         struct msm_gem_object *msm_obj = to_msm_bo(obj);
414         WARN_ON(!mutex_is_locked(&obj->dev->struct_mutex));
415         if (!msm_obj->vaddr) {
416                 struct page **pages = get_pages(obj);
417                 if (IS_ERR(pages))
418                         return ERR_CAST(pages);
419                 msm_obj->vaddr = vmap(pages, obj->size >> PAGE_SHIFT,
420                                 VM_MAP, pgprot_writecombine(PAGE_KERNEL));
421                 if (msm_obj->vaddr == NULL)
422                         return ERR_PTR(-ENOMEM);
423         }
424         return msm_obj->vaddr;
425 }
426
427 void *msm_gem_get_vaddr(struct drm_gem_object *obj)
428 {
429         void *ret;
430         mutex_lock(&obj->dev->struct_mutex);
431         ret = msm_gem_get_vaddr_locked(obj);
432         mutex_unlock(&obj->dev->struct_mutex);
433         return ret;
434 }
435
436 void msm_gem_put_vaddr_locked(struct drm_gem_object *obj)
437 {
438         WARN_ON(!mutex_is_locked(&obj->dev->struct_mutex));
439         /* no-op for now */
440 }
441
442 void msm_gem_put_vaddr(struct drm_gem_object *obj)
443 {
444         /* no-op for now */
445 }
446
447 /* Update madvise status, returns true if not purged, else
448  * false or -errno.
449  */
450 int msm_gem_madvise(struct drm_gem_object *obj, unsigned madv)
451 {
452         struct msm_gem_object *msm_obj = to_msm_bo(obj);
453
454         WARN_ON(!mutex_is_locked(&obj->dev->struct_mutex));
455
456         if (msm_obj->madv != __MSM_MADV_PURGED)
457                 msm_obj->madv = madv;
458
459         return (msm_obj->madv != __MSM_MADV_PURGED);
460 }
461
462 void msm_gem_purge(struct drm_gem_object *obj)
463 {
464         struct drm_device *dev = obj->dev;
465         struct msm_gem_object *msm_obj = to_msm_bo(obj);
466
467         WARN_ON(!mutex_is_locked(&dev->struct_mutex));
468         WARN_ON(!is_purgeable(msm_obj));
469         WARN_ON(obj->import_attach);
470
471         put_iova(obj);
472
473         vunmap(msm_obj->vaddr);
474         msm_obj->vaddr = NULL;
475
476         put_pages(obj);
477
478         msm_obj->madv = __MSM_MADV_PURGED;
479
480         drm_vma_node_unmap(&obj->vma_node, dev->anon_inode->i_mapping);
481         drm_gem_free_mmap_offset(obj);
482
483         /* Our goal here is to return as much of the memory as
484          * is possible back to the system as we are called from OOM.
485          * To do this we must instruct the shmfs to drop all of its
486          * backing pages, *now*.
487          */
488         shmem_truncate_range(file_inode(obj->filp), 0, (loff_t)-1);
489
490         invalidate_mapping_pages(file_inode(obj->filp)->i_mapping,
491                         0, (loff_t)-1);
492 }
493
494 /* must be called before _move_to_active().. */
495 int msm_gem_sync_object(struct drm_gem_object *obj,
496                 struct msm_fence_context *fctx, bool exclusive)
497 {
498         struct msm_gem_object *msm_obj = to_msm_bo(obj);
499         struct reservation_object_list *fobj;
500         struct fence *fence;
501         int i, ret;
502
503         if (!exclusive) {
504                 /* NOTE: _reserve_shared() must happen before _add_shared_fence(),
505                  * which makes this a slightly strange place to call it.  OTOH this
506                  * is a convenient can-fail point to hook it in.  (And similar to
507                  * how etnaviv and nouveau handle this.)
508                  */
509                 ret = reservation_object_reserve_shared(msm_obj->resv);
510                 if (ret)
511                         return ret;
512         }
513
514         fobj = reservation_object_get_list(msm_obj->resv);
515         if (!fobj || (fobj->shared_count == 0)) {
516                 fence = reservation_object_get_excl(msm_obj->resv);
517                 /* don't need to wait on our own fences, since ring is fifo */
518                 if (fence && (fence->context != fctx->context)) {
519                         ret = fence_wait(fence, true);
520                         if (ret)
521                                 return ret;
522                 }
523         }
524
525         if (!exclusive || !fobj)
526                 return 0;
527
528         for (i = 0; i < fobj->shared_count; i++) {
529                 fence = rcu_dereference_protected(fobj->shared[i],
530                                                 reservation_object_held(msm_obj->resv));
531                 if (fence->context != fctx->context) {
532                         ret = fence_wait(fence, true);
533                         if (ret)
534                                 return ret;
535                 }
536         }
537
538         return 0;
539 }
540
541 void msm_gem_move_to_active(struct drm_gem_object *obj,
542                 struct msm_gpu *gpu, bool exclusive, struct fence *fence)
543 {
544         struct msm_gem_object *msm_obj = to_msm_bo(obj);
545         WARN_ON(msm_obj->madv != MSM_MADV_WILLNEED);
546         msm_obj->gpu = gpu;
547         if (exclusive)
548                 reservation_object_add_excl_fence(msm_obj->resv, fence);
549         else
550                 reservation_object_add_shared_fence(msm_obj->resv, fence);
551         list_del_init(&msm_obj->mm_list);
552         list_add_tail(&msm_obj->mm_list, &gpu->active_list);
553 }
554
555 void msm_gem_move_to_inactive(struct drm_gem_object *obj)
556 {
557         struct drm_device *dev = obj->dev;
558         struct msm_drm_private *priv = dev->dev_private;
559         struct msm_gem_object *msm_obj = to_msm_bo(obj);
560
561         WARN_ON(!mutex_is_locked(&dev->struct_mutex));
562
563         msm_obj->gpu = NULL;
564         list_del_init(&msm_obj->mm_list);
565         list_add_tail(&msm_obj->mm_list, &priv->inactive_list);
566 }
567
568 int msm_gem_cpu_prep(struct drm_gem_object *obj, uint32_t op, ktime_t *timeout)
569 {
570         struct msm_gem_object *msm_obj = to_msm_bo(obj);
571         bool write = !!(op & MSM_PREP_WRITE);
572
573         if (op & MSM_PREP_NOSYNC) {
574                 if (!reservation_object_test_signaled_rcu(msm_obj->resv, write))
575                         return -EBUSY;
576         } else {
577                 int ret;
578
579                 ret = reservation_object_wait_timeout_rcu(msm_obj->resv, write,
580                                 true, timeout_to_jiffies(timeout));
581                 if (ret <= 0)
582                         return ret == 0 ? -ETIMEDOUT : ret;
583         }
584
585         /* TODO cache maintenance */
586
587         return 0;
588 }
589
590 int msm_gem_cpu_fini(struct drm_gem_object *obj)
591 {
592         /* TODO cache maintenance */
593         return 0;
594 }
595
596 #ifdef CONFIG_DEBUG_FS
597 static void describe_fence(struct fence *fence, const char *type,
598                 struct seq_file *m)
599 {
600         if (!fence_is_signaled(fence))
601                 seq_printf(m, "\t%9s: %s %s seq %u\n", type,
602                                 fence->ops->get_driver_name(fence),
603                                 fence->ops->get_timeline_name(fence),
604                                 fence->seqno);
605 }
606
607 void msm_gem_describe(struct drm_gem_object *obj, struct seq_file *m)
608 {
609         struct msm_gem_object *msm_obj = to_msm_bo(obj);
610         struct reservation_object *robj = msm_obj->resv;
611         struct reservation_object_list *fobj;
612         struct fence *fence;
613         uint64_t off = drm_vma_node_start(&obj->vma_node);
614         const char *madv;
615
616         WARN_ON(!mutex_is_locked(&obj->dev->struct_mutex));
617
618         switch (msm_obj->madv) {
619         case __MSM_MADV_PURGED:
620                 madv = " purged";
621                 break;
622         case MSM_MADV_DONTNEED:
623                 madv = " purgeable";
624                 break;
625         case MSM_MADV_WILLNEED:
626         default:
627                 madv = "";
628                 break;
629         }
630
631         seq_printf(m, "%08x: %c %2d (%2d) %08llx %p %zu%s\n",
632                         msm_obj->flags, is_active(msm_obj) ? 'A' : 'I',
633                         obj->name, obj->refcount.refcount.counter,
634                         off, msm_obj->vaddr, obj->size, madv);
635
636         rcu_read_lock();
637         fobj = rcu_dereference(robj->fence);
638         if (fobj) {
639                 unsigned int i, shared_count = fobj->shared_count;
640
641                 for (i = 0; i < shared_count; i++) {
642                         fence = rcu_dereference(fobj->shared[i]);
643                         describe_fence(fence, "Shared", m);
644                 }
645         }
646
647         fence = rcu_dereference(robj->fence_excl);
648         if (fence)
649                 describe_fence(fence, "Exclusive", m);
650         rcu_read_unlock();
651 }
652
653 void msm_gem_describe_objects(struct list_head *list, struct seq_file *m)
654 {
655         struct msm_gem_object *msm_obj;
656         int count = 0;
657         size_t size = 0;
658
659         list_for_each_entry(msm_obj, list, mm_list) {
660                 struct drm_gem_object *obj = &msm_obj->base;
661                 seq_printf(m, "   ");
662                 msm_gem_describe(obj, m);
663                 count++;
664                 size += obj->size;
665         }
666
667         seq_printf(m, "Total %d objects, %zu bytes\n", count, size);
668 }
669 #endif
670
671 void msm_gem_free_object(struct drm_gem_object *obj)
672 {
673         struct drm_device *dev = obj->dev;
674         struct msm_gem_object *msm_obj = to_msm_bo(obj);
675
676         WARN_ON(!mutex_is_locked(&dev->struct_mutex));
677
678         /* object should not be on active list: */
679         WARN_ON(is_active(msm_obj));
680
681         list_del(&msm_obj->mm_list);
682
683         put_iova(obj);
684
685         if (obj->import_attach) {
686                 if (msm_obj->vaddr)
687                         dma_buf_vunmap(obj->import_attach->dmabuf, msm_obj->vaddr);
688
689                 /* Don't drop the pages for imported dmabuf, as they are not
690                  * ours, just free the array we allocated:
691                  */
692                 if (msm_obj->pages)
693                         drm_free_large(msm_obj->pages);
694
695                 drm_prime_gem_destroy(obj, msm_obj->sgt);
696         } else {
697                 vunmap(msm_obj->vaddr);
698                 put_pages(obj);
699         }
700
701         if (msm_obj->resv == &msm_obj->_resv)
702                 reservation_object_fini(msm_obj->resv);
703
704         drm_gem_object_release(obj);
705
706         kfree(msm_obj);
707 }
708
709 /* convenience method to construct a GEM buffer object, and userspace handle */
710 int msm_gem_new_handle(struct drm_device *dev, struct drm_file *file,
711                 uint32_t size, uint32_t flags, uint32_t *handle)
712 {
713         struct drm_gem_object *obj;
714         int ret;
715
716         ret = mutex_lock_interruptible(&dev->struct_mutex);
717         if (ret)
718                 return ret;
719
720         obj = msm_gem_new(dev, size, flags);
721
722         mutex_unlock(&dev->struct_mutex);
723
724         if (IS_ERR(obj))
725                 return PTR_ERR(obj);
726
727         ret = drm_gem_handle_create(file, obj, handle);
728
729         /* drop reference from allocate - handle holds it now */
730         drm_gem_object_unreference_unlocked(obj);
731
732         return ret;
733 }
734
735 static int msm_gem_new_impl(struct drm_device *dev,
736                 uint32_t size, uint32_t flags,
737                 struct reservation_object *resv,
738                 struct drm_gem_object **obj)
739 {
740         struct msm_drm_private *priv = dev->dev_private;
741         struct msm_gem_object *msm_obj;
742         unsigned sz;
743         bool use_vram = false;
744
745         switch (flags & MSM_BO_CACHE_MASK) {
746         case MSM_BO_UNCACHED:
747         case MSM_BO_CACHED:
748         case MSM_BO_WC:
749                 break;
750         default:
751                 dev_err(dev->dev, "invalid cache flag: %x\n",
752                                 (flags & MSM_BO_CACHE_MASK));
753                 return -EINVAL;
754         }
755
756         if (!iommu_present(&platform_bus_type))
757                 use_vram = true;
758         else if ((flags & MSM_BO_STOLEN) && priv->vram.size)
759                 use_vram = true;
760
761         if (WARN_ON(use_vram && !priv->vram.size))
762                 return -EINVAL;
763
764         sz = sizeof(*msm_obj);
765         if (use_vram)
766                 sz += sizeof(struct drm_mm_node);
767
768         msm_obj = kzalloc(sz, GFP_KERNEL);
769         if (!msm_obj)
770                 return -ENOMEM;
771
772         if (use_vram)
773                 msm_obj->vram_node = (void *)&msm_obj[1];
774
775         msm_obj->flags = flags;
776         msm_obj->madv = MSM_MADV_WILLNEED;
777
778         if (resv) {
779                 msm_obj->resv = resv;
780         } else {
781                 msm_obj->resv = &msm_obj->_resv;
782                 reservation_object_init(msm_obj->resv);
783         }
784
785         INIT_LIST_HEAD(&msm_obj->submit_entry);
786         list_add_tail(&msm_obj->mm_list, &priv->inactive_list);
787
788         *obj = &msm_obj->base;
789
790         return 0;
791 }
792
793 struct drm_gem_object *msm_gem_new(struct drm_device *dev,
794                 uint32_t size, uint32_t flags)
795 {
796         struct drm_gem_object *obj = NULL;
797         int ret;
798
799         WARN_ON(!mutex_is_locked(&dev->struct_mutex));
800
801         size = PAGE_ALIGN(size);
802
803         ret = msm_gem_new_impl(dev, size, flags, NULL, &obj);
804         if (ret)
805                 goto fail;
806
807         if (use_pages(obj)) {
808                 ret = drm_gem_object_init(dev, obj, size);
809                 if (ret)
810                         goto fail;
811         } else {
812                 drm_gem_private_object_init(dev, obj, size);
813         }
814
815         return obj;
816
817 fail:
818         if (obj)
819                 drm_gem_object_unreference(obj);
820
821         return ERR_PTR(ret);
822 }
823
824 struct drm_gem_object *msm_gem_import(struct drm_device *dev,
825                 struct dma_buf *dmabuf, struct sg_table *sgt)
826 {
827         struct msm_gem_object *msm_obj;
828         struct drm_gem_object *obj;
829         uint32_t size;
830         int ret, npages;
831
832         /* if we don't have IOMMU, don't bother pretending we can import: */
833         if (!iommu_present(&platform_bus_type)) {
834                 dev_err(dev->dev, "cannot import without IOMMU\n");
835                 return ERR_PTR(-EINVAL);
836         }
837
838         size = PAGE_ALIGN(dmabuf->size);
839
840         ret = msm_gem_new_impl(dev, size, MSM_BO_WC, dmabuf->resv, &obj);
841         if (ret)
842                 goto fail;
843
844         drm_gem_private_object_init(dev, obj, size);
845
846         npages = size / PAGE_SIZE;
847
848         msm_obj = to_msm_bo(obj);
849         msm_obj->sgt = sgt;
850         msm_obj->pages = drm_malloc_ab(npages, sizeof(struct page *));
851         if (!msm_obj->pages) {
852                 ret = -ENOMEM;
853                 goto fail;
854         }
855
856         ret = drm_prime_sg_to_page_addr_arrays(sgt, msm_obj->pages, NULL, npages);
857         if (ret)
858                 goto fail;
859
860         return obj;
861
862 fail:
863         if (obj)
864                 drm_gem_object_unreference_unlocked(obj);
865
866         return ERR_PTR(ret);
867 }