1
2
3
4
5
6
7
8
9#include <drm/drmP.h>
10#include "udl_drv.h"
11#include <linux/shmem_fs.h>
12#include <linux/dma-buf.h>
13
14struct udl_gem_object *udl_gem_alloc_object(struct drm_device *dev,
15 size_t size)
16{
17 struct udl_gem_object *obj;
18
19 obj = kzalloc(sizeof(*obj), GFP_KERNEL);
20 if (obj == NULL)
21 return NULL;
22
23 if (drm_gem_object_init(dev, &obj->base, size) != 0) {
24 kfree(obj);
25 return NULL;
26 }
27
28 obj->flags = UDL_BO_CACHEABLE;
29 return obj;
30}
31
32static int
33udl_gem_create(struct drm_file *file,
34 struct drm_device *dev,
35 uint64_t size,
36 uint32_t *handle_p)
37{
38 struct udl_gem_object *obj;
39 int ret;
40 u32 handle;
41
42 size = roundup(size, PAGE_SIZE);
43
44 obj = udl_gem_alloc_object(dev, size);
45 if (obj == NULL)
46 return -ENOMEM;
47
48 ret = drm_gem_handle_create(file, &obj->base, &handle);
49 if (ret) {
50 drm_gem_object_release(&obj->base);
51 kfree(obj);
52 return ret;
53 }
54
55 drm_gem_object_put_unlocked(&obj->base);
56 *handle_p = handle;
57 return 0;
58}
59
60static void update_vm_cache_attr(struct udl_gem_object *obj,
61 struct vm_area_struct *vma)
62{
63 DRM_DEBUG_KMS("flags = 0x%x\n", obj->flags);
64
65
66 if (obj->flags & UDL_BO_CACHEABLE) {
67 vma->vm_page_prot = vm_get_page_prot(vma->vm_flags);
68 } else if (obj->flags & UDL_BO_WC) {
69 vma->vm_page_prot =
70 pgprot_writecombine(vm_get_page_prot(vma->vm_flags));
71 } else {
72 vma->vm_page_prot =
73 pgprot_noncached(vm_get_page_prot(vma->vm_flags));
74 }
75}
76
77int udl_dumb_create(struct drm_file *file,
78 struct drm_device *dev,
79 struct drm_mode_create_dumb *args)
80{
81 args->pitch = args->width * DIV_ROUND_UP(args->bpp, 8);
82 args->size = args->pitch * args->height;
83 return udl_gem_create(file, dev,
84 args->size, &args->handle);
85}
86
87int udl_drm_gem_mmap(struct file *filp, struct vm_area_struct *vma)
88{
89 int ret;
90
91 ret = drm_gem_mmap(filp, vma);
92 if (ret)
93 return ret;
94
95 vma->vm_flags &= ~VM_PFNMAP;
96 vma->vm_flags |= VM_MIXEDMAP;
97
98 update_vm_cache_attr(to_udl_bo(vma->vm_private_data), vma);
99
100 return ret;
101}
102
103int udl_gem_fault(struct vm_fault *vmf)
104{
105 struct vm_area_struct *vma = vmf->vma;
106 struct udl_gem_object *obj = to_udl_bo(vma->vm_private_data);
107 struct page *page;
108 unsigned int page_offset;
109 int ret = 0;
110
111 page_offset = (vmf->address - vma->vm_start) >> PAGE_SHIFT;
112
113 if (!obj->pages)
114 return VM_FAULT_SIGBUS;
115
116 page = obj->pages[page_offset];
117 ret = vm_insert_page(vma, vmf->address, page);
118 switch (ret) {
119 case -EAGAIN:
120 case 0:
121 case -ERESTARTSYS:
122 return VM_FAULT_NOPAGE;
123 case -ENOMEM:
124 return VM_FAULT_OOM;
125 default:
126 return VM_FAULT_SIGBUS;
127 }
128}
129
130int udl_gem_get_pages(struct udl_gem_object *obj)
131{
132 struct page **pages;
133
134 if (obj->pages)
135 return 0;
136
137 pages = drm_gem_get_pages(&obj->base);
138 if (IS_ERR(pages))
139 return PTR_ERR(pages);
140
141 obj->pages = pages;
142
143 return 0;
144}
145
146void udl_gem_put_pages(struct udl_gem_object *obj)
147{
148 if (obj->base.import_attach) {
149 kvfree(obj->pages);
150 obj->pages = NULL;
151 return;
152 }
153
154 drm_gem_put_pages(&obj->base, obj->pages, false, false);
155 obj->pages = NULL;
156}
157
158int udl_gem_vmap(struct udl_gem_object *obj)
159{
160 int page_count = obj->base.size / PAGE_SIZE;
161 int ret;
162
163 if (obj->base.import_attach) {
164 obj->vmapping = dma_buf_vmap(obj->base.import_attach->dmabuf);
165 if (!obj->vmapping)
166 return -ENOMEM;
167 return 0;
168 }
169
170 ret = udl_gem_get_pages(obj);
171 if (ret)
172 return ret;
173
174 obj->vmapping = vmap(obj->pages, page_count, 0, PAGE_KERNEL);
175 if (!obj->vmapping)
176 return -ENOMEM;
177 return 0;
178}
179
180void udl_gem_vunmap(struct udl_gem_object *obj)
181{
182 if (obj->base.import_attach) {
183 dma_buf_vunmap(obj->base.import_attach->dmabuf, obj->vmapping);
184 return;
185 }
186
187 vunmap(obj->vmapping);
188
189 udl_gem_put_pages(obj);
190}
191
192void udl_gem_free_object(struct drm_gem_object *gem_obj)
193{
194 struct udl_gem_object *obj = to_udl_bo(gem_obj);
195
196 if (obj->vmapping)
197 udl_gem_vunmap(obj);
198
199 if (gem_obj->import_attach) {
200 drm_prime_gem_destroy(gem_obj, obj->sg);
201 put_device(gem_obj->dev->dev);
202 }
203
204 if (obj->pages)
205 udl_gem_put_pages(obj);
206
207 drm_gem_free_mmap_offset(gem_obj);
208}
209
210
211
212int udl_gem_mmap(struct drm_file *file, struct drm_device *dev,
213 uint32_t handle, uint64_t *offset)
214{
215 struct udl_gem_object *gobj;
216 struct drm_gem_object *obj;
217 int ret = 0;
218
219 mutex_lock(&dev->struct_mutex);
220 obj = drm_gem_object_lookup(file, handle);
221 if (obj == NULL) {
222 ret = -ENOENT;
223 goto unlock;
224 }
225 gobj = to_udl_bo(obj);
226
227 ret = udl_gem_get_pages(gobj);
228 if (ret)
229 goto out;
230 ret = drm_gem_create_mmap_offset(obj);
231 if (ret)
232 goto out;
233
234 *offset = drm_vma_node_offset_addr(&gobj->base.vma_node);
235
236out:
237 drm_gem_object_put(&gobj->base);
238unlock:
239 mutex_unlock(&dev->struct_mutex);
240 return ret;
241}
242
