This source file includes following definitions.
- vgem_gem_free_object
- vgem_gem_fault
- vgem_open
- vgem_postclose
- __vgem_gem_create
- __vgem_gem_destroy
- vgem_gem_create
- vgem_gem_dumb_create
- vgem_gem_dumb_map
- vgem_mmap
- vgem_pin_pages
- vgem_unpin_pages
- vgem_prime_pin
- vgem_prime_unpin
- vgem_prime_get_sg_table
- vgem_prime_import
- vgem_prime_import_sg_table
- vgem_prime_vmap
- vgem_prime_vunmap
- vgem_prime_mmap
- vgem_release
- vgem_init
- vgem_exit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33 #include <linux/dma-buf.h>
34 #include <linux/module.h>
35 #include <linux/platform_device.h>
36 #include <linux/shmem_fs.h>
37 #include <linux/vmalloc.h>
38
39 #include <drm/drm_drv.h>
40 #include <drm/drm_file.h>
41 #include <drm/drm_ioctl.h>
42 #include <drm/drm_prime.h>
43
44 #include "vgem_drv.h"
45
46 #define DRIVER_NAME "vgem"
47 #define DRIVER_DESC "Virtual GEM provider"
48 #define DRIVER_DATE "20120112"
49 #define DRIVER_MAJOR 1
50 #define DRIVER_MINOR 0
51
52 static struct vgem_device {
53 struct drm_device drm;
54 struct platform_device *platform;
55 } *vgem_device;
56
57 static void vgem_gem_free_object(struct drm_gem_object *obj)
58 {
59 struct drm_vgem_gem_object *vgem_obj = to_vgem_bo(obj);
60
61 kvfree(vgem_obj->pages);
62 mutex_destroy(&vgem_obj->pages_lock);
63
64 if (obj->import_attach)
65 drm_prime_gem_destroy(obj, vgem_obj->table);
66
67 drm_gem_object_release(obj);
68 kfree(vgem_obj);
69 }
70
71 static vm_fault_t vgem_gem_fault(struct vm_fault *vmf)
72 {
73 struct vm_area_struct *vma = vmf->vma;
74 struct drm_vgem_gem_object *obj = vma->vm_private_data;
75
76 unsigned long vaddr = vmf->address;
77 vm_fault_t ret = VM_FAULT_SIGBUS;
78 loff_t num_pages;
79 pgoff_t page_offset;
80 page_offset = (vaddr - vma->vm_start) >> PAGE_SHIFT;
81
82 num_pages = DIV_ROUND_UP(obj->base.size, PAGE_SIZE);
83
84 if (page_offset >= num_pages)
85 return VM_FAULT_SIGBUS;
86
87 mutex_lock(&obj->pages_lock);
88 if (obj->pages) {
89 get_page(obj->pages[page_offset]);
90 vmf->page = obj->pages[page_offset];
91 ret = 0;
92 }
93 mutex_unlock(&obj->pages_lock);
94 if (ret) {
95 struct page *page;
96
97 page = shmem_read_mapping_page(
98 file_inode(obj->base.filp)->i_mapping,
99 page_offset);
100 if (!IS_ERR(page)) {
101 vmf->page = page;
102 ret = 0;
103 } else switch (PTR_ERR(page)) {
104 case -ENOSPC:
105 case -ENOMEM:
106 ret = VM_FAULT_OOM;
107 break;
108 case -EBUSY:
109 ret = VM_FAULT_RETRY;
110 break;
111 case -EFAULT:
112 case -EINVAL:
113 ret = VM_FAULT_SIGBUS;
114 break;
115 default:
116 WARN_ON(PTR_ERR(page));
117 ret = VM_FAULT_SIGBUS;
118 break;
119 }
120
121 }
122 return ret;
123 }
124
125 static const struct vm_operations_struct vgem_gem_vm_ops = {
126 .fault = vgem_gem_fault,
127 .open = drm_gem_vm_open,
128 .close = drm_gem_vm_close,
129 };
130
131 static int vgem_open(struct drm_device *dev, struct drm_file *file)
132 {
133 struct vgem_file *vfile;
134 int ret;
135
136 vfile = kzalloc(sizeof(*vfile), GFP_KERNEL);
137 if (!vfile)
138 return -ENOMEM;
139
140 file->driver_priv = vfile;
141
142 ret = vgem_fence_open(vfile);
143 if (ret) {
144 kfree(vfile);
145 return ret;
146 }
147
148 return 0;
149 }
150
151 static void vgem_postclose(struct drm_device *dev, struct drm_file *file)
152 {
153 struct vgem_file *vfile = file->driver_priv;
154
155 vgem_fence_close(vfile);
156 kfree(vfile);
157 }
158
159 static struct drm_vgem_gem_object *__vgem_gem_create(struct drm_device *dev,
160 unsigned long size)
161 {
162 struct drm_vgem_gem_object *obj;
163 int ret;
164
165 obj = kzalloc(sizeof(*obj), GFP_KERNEL);
166 if (!obj)
167 return ERR_PTR(-ENOMEM);
168
169 ret = drm_gem_object_init(dev, &obj->base, roundup(size, PAGE_SIZE));
170 if (ret) {
171 kfree(obj);
172 return ERR_PTR(ret);
173 }
174
175 mutex_init(&obj->pages_lock);
176
177 return obj;
178 }
179
180 static void __vgem_gem_destroy(struct drm_vgem_gem_object *obj)
181 {
182 drm_gem_object_release(&obj->base);
183 kfree(obj);
184 }
185
186 static struct drm_gem_object *vgem_gem_create(struct drm_device *dev,
187 struct drm_file *file,
188 unsigned int *handle,
189 unsigned long size)
190 {
191 struct drm_vgem_gem_object *obj;
192 int ret;
193
194 obj = __vgem_gem_create(dev, size);
195 if (IS_ERR(obj))
196 return ERR_CAST(obj);
197
198 ret = drm_gem_handle_create(file, &obj->base, handle);
199 if (ret) {
200 drm_gem_object_put_unlocked(&obj->base);
201 return ERR_PTR(ret);
202 }
203
204 return &obj->base;
205 }
206
207 static int vgem_gem_dumb_create(struct drm_file *file, struct drm_device *dev,
208 struct drm_mode_create_dumb *args)
209 {
210 struct drm_gem_object *gem_object;
211 u64 pitch, size;
212
213 pitch = args->width * DIV_ROUND_UP(args->bpp, 8);
214 size = args->height * pitch;
215 if (size == 0)
216 return -EINVAL;
217
218 gem_object = vgem_gem_create(dev, file, &args->handle, size);
219 if (IS_ERR(gem_object))
220 return PTR_ERR(gem_object);
221
222 args->size = gem_object->size;
223 args->pitch = pitch;
224
225 drm_gem_object_put_unlocked(gem_object);
226
227 DRM_DEBUG("Created object of size %llu\n", args->size);
228
229 return 0;
230 }
231
232 static int vgem_gem_dumb_map(struct drm_file *file, struct drm_device *dev,
233 uint32_t handle, uint64_t *offset)
234 {
235 struct drm_gem_object *obj;
236 int ret;
237
238 obj = drm_gem_object_lookup(file, handle);
239 if (!obj)
240 return -ENOENT;
241
242 if (!obj->filp) {
243 ret = -EINVAL;
244 goto unref;
245 }
246
247 ret = drm_gem_create_mmap_offset(obj);
248 if (ret)
249 goto unref;
250
251 *offset = drm_vma_node_offset_addr(&obj->vma_node);
252 unref:
253 drm_gem_object_put_unlocked(obj);
254
255 return ret;
256 }
257
258 static struct drm_ioctl_desc vgem_ioctls[] = {
259 DRM_IOCTL_DEF_DRV(VGEM_FENCE_ATTACH, vgem_fence_attach_ioctl, DRM_RENDER_ALLOW),
260 DRM_IOCTL_DEF_DRV(VGEM_FENCE_SIGNAL, vgem_fence_signal_ioctl, DRM_RENDER_ALLOW),
261 };
262
263 static int vgem_mmap(struct file *filp, struct vm_area_struct *vma)
264 {
265 unsigned long flags = vma->vm_flags;
266 int ret;
267
268 ret = drm_gem_mmap(filp, vma);
269 if (ret)
270 return ret;
271
272
273
274
275 vma->vm_flags = flags | VM_DONTEXPAND | VM_DONTDUMP;
276 return 0;
277 }
278
279 static const struct file_operations vgem_driver_fops = {
280 .owner = THIS_MODULE,
281 .open = drm_open,
282 .mmap = vgem_mmap,
283 .poll = drm_poll,
284 .read = drm_read,
285 .unlocked_ioctl = drm_ioctl,
286 .compat_ioctl = drm_compat_ioctl,
287 .release = drm_release,
288 };
289
290 static struct page **vgem_pin_pages(struct drm_vgem_gem_object *bo)
291 {
292 mutex_lock(&bo->pages_lock);
293 if (bo->pages_pin_count++ == 0) {
294 struct page **pages;
295
296 pages = drm_gem_get_pages(&bo->base);
297 if (IS_ERR(pages)) {
298 bo->pages_pin_count--;
299 mutex_unlock(&bo->pages_lock);
300 return pages;
301 }
302
303 bo->pages = pages;
304 }
305 mutex_unlock(&bo->pages_lock);
306
307 return bo->pages;
308 }
309
310 static void vgem_unpin_pages(struct drm_vgem_gem_object *bo)
311 {
312 mutex_lock(&bo->pages_lock);
313 if (--bo->pages_pin_count == 0) {
314 drm_gem_put_pages(&bo->base, bo->pages, true, true);
315 bo->pages = NULL;
316 }
317 mutex_unlock(&bo->pages_lock);
318 }
319
320 static int vgem_prime_pin(struct drm_gem_object *obj)
321 {
322 struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
323 long n_pages = obj->size >> PAGE_SHIFT;
324 struct page **pages;
325
326 pages = vgem_pin_pages(bo);
327 if (IS_ERR(pages))
328 return PTR_ERR(pages);
329
330
331
332
333 drm_clflush_pages(pages, n_pages);
334
335 return 0;
336 }
337
338 static void vgem_prime_unpin(struct drm_gem_object *obj)
339 {
340 struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
341
342 vgem_unpin_pages(bo);
343 }
344
345 static struct sg_table *vgem_prime_get_sg_table(struct drm_gem_object *obj)
346 {
347 struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
348
349 return drm_prime_pages_to_sg(bo->pages, bo->base.size >> PAGE_SHIFT);
350 }
351
352 static struct drm_gem_object* vgem_prime_import(struct drm_device *dev,
353 struct dma_buf *dma_buf)
354 {
355 struct vgem_device *vgem = container_of(dev, typeof(*vgem), drm);
356
357 return drm_gem_prime_import_dev(dev, dma_buf, &vgem->platform->dev);
358 }
359
360 static struct drm_gem_object *vgem_prime_import_sg_table(struct drm_device *dev,
361 struct dma_buf_attachment *attach, struct sg_table *sg)
362 {
363 struct drm_vgem_gem_object *obj;
364 int npages;
365
366 obj = __vgem_gem_create(dev, attach->dmabuf->size);
367 if (IS_ERR(obj))
368 return ERR_CAST(obj);
369
370 npages = PAGE_ALIGN(attach->dmabuf->size) / PAGE_SIZE;
371
372 obj->table = sg;
373 obj->pages = kvmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
374 if (!obj->pages) {
375 __vgem_gem_destroy(obj);
376 return ERR_PTR(-ENOMEM);
377 }
378
379 obj->pages_pin_count++;
380 drm_prime_sg_to_page_addr_arrays(obj->table, obj->pages, NULL,
381 npages);
382 return &obj->base;
383 }
384
385 static void *vgem_prime_vmap(struct drm_gem_object *obj)
386 {
387 struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
388 long n_pages = obj->size >> PAGE_SHIFT;
389 struct page **pages;
390
391 pages = vgem_pin_pages(bo);
392 if (IS_ERR(pages))
393 return NULL;
394
395 return vmap(pages, n_pages, 0, pgprot_writecombine(PAGE_KERNEL));
396 }
397
398 static void vgem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
399 {
400 struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
401
402 vunmap(vaddr);
403 vgem_unpin_pages(bo);
404 }
405
406 static int vgem_prime_mmap(struct drm_gem_object *obj,
407 struct vm_area_struct *vma)
408 {
409 int ret;
410
411 if (obj->size < vma->vm_end - vma->vm_start)
412 return -EINVAL;
413
414 if (!obj->filp)
415 return -ENODEV;
416
417 ret = call_mmap(obj->filp, vma);
418 if (ret)
419 return ret;
420
421 fput(vma->vm_file);
422 vma->vm_file = get_file(obj->filp);
423 vma->vm_flags |= VM_DONTEXPAND | VM_DONTDUMP;
424 vma->vm_page_prot = pgprot_writecombine(vm_get_page_prot(vma->vm_flags));
425
426 return 0;
427 }
428
429 static void vgem_release(struct drm_device *dev)
430 {
431 struct vgem_device *vgem = container_of(dev, typeof(*vgem), drm);
432
433 platform_device_unregister(vgem->platform);
434 drm_dev_fini(&vgem->drm);
435
436 kfree(vgem);
437 }
438
439 static struct drm_driver vgem_driver = {
440 .driver_features = DRIVER_GEM | DRIVER_RENDER,
441 .release = vgem_release,
442 .open = vgem_open,
443 .postclose = vgem_postclose,
444 .gem_free_object_unlocked = vgem_gem_free_object,
445 .gem_vm_ops = &vgem_gem_vm_ops,
446 .ioctls = vgem_ioctls,
447 .num_ioctls = ARRAY_SIZE(vgem_ioctls),
448 .fops = &vgem_driver_fops,
449
450 .dumb_create = vgem_gem_dumb_create,
451 .dumb_map_offset = vgem_gem_dumb_map,
452
453 .prime_handle_to_fd = drm_gem_prime_handle_to_fd,
454 .prime_fd_to_handle = drm_gem_prime_fd_to_handle,
455 .gem_prime_pin = vgem_prime_pin,
456 .gem_prime_unpin = vgem_prime_unpin,
457 .gem_prime_import = vgem_prime_import,
458 .gem_prime_import_sg_table = vgem_prime_import_sg_table,
459 .gem_prime_get_sg_table = vgem_prime_get_sg_table,
460 .gem_prime_vmap = vgem_prime_vmap,
461 .gem_prime_vunmap = vgem_prime_vunmap,
462 .gem_prime_mmap = vgem_prime_mmap,
463
464 .name = DRIVER_NAME,
465 .desc = DRIVER_DESC,
466 .date = DRIVER_DATE,
467 .major = DRIVER_MAJOR,
468 .minor = DRIVER_MINOR,
469 };
470
471 static int __init vgem_init(void)
472 {
473 int ret;
474
475 vgem_device = kzalloc(sizeof(*vgem_device), GFP_KERNEL);
476 if (!vgem_device)
477 return -ENOMEM;
478
479 vgem_device->platform =
480 platform_device_register_simple("vgem", -1, NULL, 0);
481 if (IS_ERR(vgem_device->platform)) {
482 ret = PTR_ERR(vgem_device->platform);
483 goto out_free;
484 }
485
486 dma_coerce_mask_and_coherent(&vgem_device->platform->dev,
487 DMA_BIT_MASK(64));
488 ret = drm_dev_init(&vgem_device->drm, &vgem_driver,
489 &vgem_device->platform->dev);
490 if (ret)
491 goto out_unregister;
492
493
494 ret = drm_dev_register(&vgem_device->drm, 0);
495 if (ret)
496 goto out_fini;
497
498 return 0;
499
500 out_fini:
501 drm_dev_fini(&vgem_device->drm);
502 out_unregister:
503 platform_device_unregister(vgem_device->platform);
504 out_free:
505 kfree(vgem_device);
506 return ret;
507 }
508
509 static void __exit vgem_exit(void)
510 {
511 drm_dev_unregister(&vgem_device->drm);
512 drm_dev_put(&vgem_device->drm);
513 }
514
515 module_init(vgem_init);
516 module_exit(vgem_exit);
517
518 MODULE_AUTHOR("Red Hat, Inc.");
519 MODULE_AUTHOR("Intel Corporation");
520 MODULE_DESCRIPTION(DRIVER_DESC);
521 MODULE_LICENSE("GPL and additional rights");