d741b1d735f72e31ccc836880b08ecf984ac0cf8
[linux-2.6-microblaze.git] / drivers / gpu / drm / etnaviv / etnaviv_gem_prime.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (C) 2014-2018 Etnaviv Project
4  */
5
6 #include <drm/drm_prime.h>
7 #include <linux/dma-buf.h>
8
9 #include "etnaviv_drv.h"
10 #include "etnaviv_gem.h"
11
12 static struct lock_class_key etnaviv_prime_lock_class;
13
14 struct sg_table *etnaviv_gem_prime_get_sg_table(struct drm_gem_object *obj)
15 {
16         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
17         int npages = obj->size >> PAGE_SHIFT;
18
19         if (WARN_ON(!etnaviv_obj->pages))  /* should have already pinned! */
20                 return ERR_PTR(-EINVAL);
21
22         return drm_prime_pages_to_sg(obj->dev, etnaviv_obj->pages, npages);
23 }
24
25 int etnaviv_gem_prime_vmap(struct drm_gem_object *obj, struct dma_buf_map *map)
26 {
27         void *vaddr;
28
29         vaddr = etnaviv_gem_vmap(obj);
30         if (!vaddr)
31                 return -ENOMEM;
32         dma_buf_map_set_vaddr(map, vaddr);
33
34         return 0;
35 }
36
37 int etnaviv_gem_prime_mmap(struct drm_gem_object *obj,
38                            struct vm_area_struct *vma)
39 {
40         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
41         int ret;
42
43         ret = drm_gem_mmap_obj(obj, obj->size, vma);
44         if (ret < 0)
45                 return ret;
46
47         return etnaviv_obj->ops->mmap(etnaviv_obj, vma);
48 }
49
50 int etnaviv_gem_prime_pin(struct drm_gem_object *obj)
51 {
52         if (!obj->import_attach) {
53                 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
54
55                 mutex_lock(&etnaviv_obj->lock);
56                 etnaviv_gem_get_pages(etnaviv_obj);
57                 mutex_unlock(&etnaviv_obj->lock);
58         }
59         return 0;
60 }
61
62 void etnaviv_gem_prime_unpin(struct drm_gem_object *obj)
63 {
64         if (!obj->import_attach) {
65                 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
66
67                 mutex_lock(&etnaviv_obj->lock);
68                 etnaviv_gem_put_pages(to_etnaviv_bo(obj));
69                 mutex_unlock(&etnaviv_obj->lock);
70         }
71 }
72
73 static void etnaviv_gem_prime_release(struct etnaviv_gem_object *etnaviv_obj)
74 {
75         struct dma_buf_map map = DMA_BUF_MAP_INIT_VADDR(etnaviv_obj->vaddr);
76
77         if (etnaviv_obj->vaddr)
78                 dma_buf_vunmap(etnaviv_obj->base.import_attach->dmabuf, &map);
79
80         /* Don't drop the pages for imported dmabuf, as they are not
81          * ours, just free the array we allocated:
82          */
83         kvfree(etnaviv_obj->pages);
84
85         drm_prime_gem_destroy(&etnaviv_obj->base, etnaviv_obj->sgt);
86 }
87
88 static void *etnaviv_gem_prime_vmap_impl(struct etnaviv_gem_object *etnaviv_obj)
89 {
90         struct dma_buf_map map;
91         int ret;
92
93         lockdep_assert_held(&etnaviv_obj->lock);
94
95         ret = dma_buf_vmap(etnaviv_obj->base.import_attach->dmabuf, &map);
96         if (ret)
97                 return NULL;
98         return map.vaddr;
99 }
100
101 static int etnaviv_gem_prime_mmap_obj(struct etnaviv_gem_object *etnaviv_obj,
102                 struct vm_area_struct *vma)
103 {
104         return dma_buf_mmap(etnaviv_obj->base.dma_buf, vma, 0);
105 }
106
107 static const struct etnaviv_gem_ops etnaviv_gem_prime_ops = {
108         /* .get_pages should never be called */
109         .release = etnaviv_gem_prime_release,
110         .vmap = etnaviv_gem_prime_vmap_impl,
111         .mmap = etnaviv_gem_prime_mmap_obj,
112 };
113
114 struct drm_gem_object *etnaviv_gem_prime_import_sg_table(struct drm_device *dev,
115         struct dma_buf_attachment *attach, struct sg_table *sgt)
116 {
117         struct etnaviv_gem_object *etnaviv_obj;
118         size_t size = PAGE_ALIGN(attach->dmabuf->size);
119         int ret, npages;
120
121         ret = etnaviv_gem_new_private(dev, size, ETNA_BO_WC,
122                                       &etnaviv_gem_prime_ops, &etnaviv_obj);
123         if (ret < 0)
124                 return ERR_PTR(ret);
125
126         lockdep_set_class(&etnaviv_obj->lock, &etnaviv_prime_lock_class);
127
128         npages = size / PAGE_SIZE;
129
130         etnaviv_obj->sgt = sgt;
131         etnaviv_obj->pages = kvmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
132         if (!etnaviv_obj->pages) {
133                 ret = -ENOMEM;
134                 goto fail;
135         }
136
137         ret = drm_prime_sg_to_page_array(sgt, etnaviv_obj->pages, npages);
138         if (ret)
139                 goto fail;
140
141         etnaviv_gem_obj_add(dev, &etnaviv_obj->base);
142
143         return &etnaviv_obj->base;
144
145 fail:
146         drm_gem_object_put(&etnaviv_obj->base);
147
148         return ERR_PTR(ret);
149 }