drivers/gpu/drm/vgem/vgem_drv.c

   1 /*
   2  * Copyright 2011 Red Hat, Inc.
   3  * Copyright © 2014 The Chromium OS Authors
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software")
   7  * to deal in the software without restriction, including without limitation
   8  * on the rights to use, copy, modify, merge, publish, distribute, sub
   9  * license, and/or sell copies of the Software, and to permit persons to whom
  10  * them Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice (including the next
  13  * paragraph) shall be included in all copies or substantial portions of the
  14  * Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTIBILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.  IN NO EVENT SHALL
  19  * THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES, OR OTHER LIABILITY, WHETHER
  20  * IN AN ACTION OF CONTRACT, TORT, OR OTHERWISE, ARISING FROM, OUT OF OR IN
  21  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *      Adam Jackson <ajax@redhat.com>
  25  *      Ben Widawsky <ben@bwidawsk.net>
  26  */
  27
  28 /**
  29  * This is vgem, a (non-hardware-backed) GEM service.  This is used by Mesa's
  30  * software renderer and the X server for efficient buffer sharing.
  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_managed.h>
  43 #include <drm/drm_prime.h>
  44
  45 #include "vgem_drv.h"
  46
  47 #define DRIVER_NAME     "vgem"
  48 #define DRIVER_DESC     "Virtual GEM provider"
  49 #define DRIVER_DATE     "20120112"
  50 #define DRIVER_MAJOR    1
  51 #define DRIVER_MINOR    0
  52
  53 static const struct drm_gem_object_funcs vgem_gem_object_funcs;
  54
  55 static struct vgem_device {
  56         struct drm_device drm;
  57         struct platform_device *platform;
  58 } *vgem_device;
  59
  60 static void vgem_gem_free_object(struct drm_gem_object *obj)
  61 {
  62         struct drm_vgem_gem_object *vgem_obj = to_vgem_bo(obj);
  63
  64         kvfree(vgem_obj->pages);
  65         mutex_destroy(&vgem_obj->pages_lock);
  66
  67         if (obj->import_attach)
  68                 drm_prime_gem_destroy(obj, vgem_obj->table);
  69
  70         drm_gem_object_release(obj);
  71         kfree(vgem_obj);
  72 }
  73
  74 static vm_fault_t vgem_gem_fault(struct vm_fault *vmf)
  75 {
  76         struct vm_area_struct *vma = vmf->vma;
  77         struct drm_vgem_gem_object *obj = vma->vm_private_data;
  78         /* We don't use vmf->pgoff since that has the fake offset */
  79         unsigned long vaddr = vmf->address;
  80         vm_fault_t ret = VM_FAULT_SIGBUS;
  81         loff_t num_pages;
  82         pgoff_t page_offset;
  83         page_offset = (vaddr - vma->vm_start) >> PAGE_SHIFT;
  84
  85         num_pages = DIV_ROUND_UP(obj->base.size, PAGE_SIZE);
  86
  87         if (page_offset >= num_pages)
  88                 return VM_FAULT_SIGBUS;
  89
  90         mutex_lock(&obj->pages_lock);
  91         if (obj->pages) {
  92                 get_page(obj->pages[page_offset]);
  93                 vmf->page = obj->pages[page_offset];
  94                 ret = 0;
  95         }
  96         mutex_unlock(&obj->pages_lock);
  97         if (ret) {
  98                 struct page *page;
  99
 100                 page = shmem_read_mapping_page(
 101                                         file_inode(obj->base.filp)->i_mapping,
 102                                         page_offset);
 103                 if (!IS_ERR(page)) {
 104                         vmf->page = page;
 105                         ret = 0;
 106                 } else switch (PTR_ERR(page)) {
 107                         case -ENOSPC:
 108                         case -ENOMEM:
 109                                 ret = VM_FAULT_OOM;
 110                                 break;
 111                         case -EBUSY:
 112                                 ret = VM_FAULT_RETRY;
 113                                 break;
 114                         case -EFAULT:
 115                         case -EINVAL:
 116                                 ret = VM_FAULT_SIGBUS;
 117                                 break;
 118                         default:
 119                                 WARN_ON(PTR_ERR(page));
 120                                 ret = VM_FAULT_SIGBUS;
 121                                 break;
 122                 }
 123
 124         }
 125         return ret;
 126 }
 127
 128 static const struct vm_operations_struct vgem_gem_vm_ops = {
 129         .fault = vgem_gem_fault,
 130         .open = drm_gem_vm_open,
 131         .close = drm_gem_vm_close,
 132 };
 133
 134 static int vgem_open(struct drm_device *dev, struct drm_file *file)
 135 {
 136         struct vgem_file *vfile;
 137         int ret;
 138
 139         vfile = kzalloc(sizeof(*vfile), GFP_KERNEL);
 140         if (!vfile)
 141                 return -ENOMEM;
 142
 143         file->driver_priv = vfile;
 144
 145         ret = vgem_fence_open(vfile);
 146         if (ret) {
 147                 kfree(vfile);
 148                 return ret;
 149         }
 150
 151         return 0;
 152 }
 153
 154 static void vgem_postclose(struct drm_device *dev, struct drm_file *file)
 155 {
 156         struct vgem_file *vfile = file->driver_priv;
 157
 158         vgem_fence_close(vfile);
 159         kfree(vfile);
 160 }
 161
 162 static struct drm_vgem_gem_object *__vgem_gem_create(struct drm_device *dev,
 163                                                 unsigned long size)
 164 {
 165         struct drm_vgem_gem_object *obj;
 166         int ret;
 167
 168         obj = kzalloc(sizeof(*obj), GFP_KERNEL);
 169         if (!obj)
 170                 return ERR_PTR(-ENOMEM);
 171
 172         obj->base.funcs = &vgem_gem_object_funcs;
 173
 174         ret = drm_gem_object_init(dev, &obj->base, roundup(size, PAGE_SIZE));
 175         if (ret) {
 176                 kfree(obj);
 177                 return ERR_PTR(ret);
 178         }
 179
 180         mutex_init(&obj->pages_lock);
 181
 182         return obj;
 183 }
 184
 185 static void __vgem_gem_destroy(struct drm_vgem_gem_object *obj)
 186 {
 187         drm_gem_object_release(&obj->base);
 188         kfree(obj);
 189 }
 190
 191 static struct drm_gem_object *vgem_gem_create(struct drm_device *dev,
 192                                               struct drm_file *file,
 193                                               unsigned int *handle,
 194                                               unsigned long size)
 195 {
 196         struct drm_vgem_gem_object *obj;
 197         int ret;
 198
 199         obj = __vgem_gem_create(dev, size);
 200         if (IS_ERR(obj))
 201                 return ERR_CAST(obj);
 202
 203         ret = drm_gem_handle_create(file, &obj->base, handle);
 204         if (ret) {
 205                 drm_gem_object_put(&obj->base);
 206                 return ERR_PTR(ret);
 207         }
 208
 209         return &obj->base;
 210 }
 211
 212 static int vgem_gem_dumb_create(struct drm_file *file, struct drm_device *dev,
 213                                 struct drm_mode_create_dumb *args)
 214 {
 215         struct drm_gem_object *gem_object;
 216         u64 pitch, size;
 217
 218         pitch = args->width * DIV_ROUND_UP(args->bpp, 8);
 219         size = args->height * pitch;
 220         if (size == 0)
 221                 return -EINVAL;
 222
 223         gem_object = vgem_gem_create(dev, file, &args->handle, size);
 224         if (IS_ERR(gem_object))
 225                 return PTR_ERR(gem_object);
 226
 227         args->size = gem_object->size;
 228         args->pitch = pitch;
 229
 230         drm_gem_object_put(gem_object);
 231
 232         DRM_DEBUG("Created object of size %llu\n", args->size);
 233
 234         return 0;
 235 }
 236
 237 static struct drm_ioctl_desc vgem_ioctls[] = {
 238         DRM_IOCTL_DEF_DRV(VGEM_FENCE_ATTACH, vgem_fence_attach_ioctl, DRM_RENDER_ALLOW),
 239         DRM_IOCTL_DEF_DRV(VGEM_FENCE_SIGNAL, vgem_fence_signal_ioctl, DRM_RENDER_ALLOW),
 240 };
 241
 242 static int vgem_mmap(struct file *filp, struct vm_area_struct *vma)
 243 {
 244         unsigned long flags = vma->vm_flags;
 245         int ret;
 246
 247         ret = drm_gem_mmap(filp, vma);
 248         if (ret)
 249                 return ret;
 250
 251         /* Keep the WC mmaping set by drm_gem_mmap() but our pages
 252          * are ordinary and not special.
 253          */
 254         vma->vm_flags = flags | VM_DONTEXPAND | VM_DONTDUMP;
 255         return 0;
 256 }
 257
 258 static const struct file_operations vgem_driver_fops = {
 259         .owner          = THIS_MODULE,
 260         .open           = drm_open,
 261         .mmap           = vgem_mmap,
 262         .poll           = drm_poll,
 263         .read           = drm_read,
 264         .unlocked_ioctl = drm_ioctl,
 265         .compat_ioctl   = drm_compat_ioctl,
 266         .release        = drm_release,
 267 };
 268
 269 static struct page **vgem_pin_pages(struct drm_vgem_gem_object *bo)
 270 {
 271         mutex_lock(&bo->pages_lock);
 272         if (bo->pages_pin_count++ == 0) {
 273                 struct page **pages;
 274
 275                 pages = drm_gem_get_pages(&bo->base);
 276                 if (IS_ERR(pages)) {
 277                         bo->pages_pin_count--;
 278                         mutex_unlock(&bo->pages_lock);
 279                         return pages;
 280                 }
 281
 282                 bo->pages = pages;
 283         }
 284         mutex_unlock(&bo->pages_lock);
 285
 286         return bo->pages;
 287 }
 288
 289 static void vgem_unpin_pages(struct drm_vgem_gem_object *bo)
 290 {
 291         mutex_lock(&bo->pages_lock);
 292         if (--bo->pages_pin_count == 0) {
 293                 drm_gem_put_pages(&bo->base, bo->pages, true, true);
 294                 bo->pages = NULL;
 295         }
 296         mutex_unlock(&bo->pages_lock);
 297 }
 298
 299 static int vgem_prime_pin(struct drm_gem_object *obj)
 300 {
 301         struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
 302         long n_pages = obj->size >> PAGE_SHIFT;
 303         struct page **pages;
 304
 305         pages = vgem_pin_pages(bo);
 306         if (IS_ERR(pages))
 307                 return PTR_ERR(pages);
 308
 309         /* Flush the object from the CPU cache so that importers can rely
 310          * on coherent indirect access via the exported dma-address.
 311          */
 312         drm_clflush_pages(pages, n_pages);
 313
 314         return 0;
 315 }
 316
 317 static void vgem_prime_unpin(struct drm_gem_object *obj)
 318 {
 319         struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
 320
 321         vgem_unpin_pages(bo);
 322 }
 323
 324 static struct sg_table *vgem_prime_get_sg_table(struct drm_gem_object *obj)
 325 {
 326         struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
 327
 328         return drm_prime_pages_to_sg(obj->dev, bo->pages, bo->base.size >> PAGE_SHIFT);
 329 }
 330
 331 static struct drm_gem_object* vgem_prime_import(struct drm_device *dev,
 332                                                 struct dma_buf *dma_buf)
 333 {
 334         struct vgem_device *vgem = container_of(dev, typeof(*vgem), drm);
 335
 336         return drm_gem_prime_import_dev(dev, dma_buf, &vgem->platform->dev);
 337 }
 338
 339 static struct drm_gem_object *vgem_prime_import_sg_table(struct drm_device *dev,
 340                         struct dma_buf_attachment *attach, struct sg_table *sg)
 341 {
 342         struct drm_vgem_gem_object *obj;
 343         int npages;
 344
 345         obj = __vgem_gem_create(dev, attach->dmabuf->size);
 346         if (IS_ERR(obj))
 347                 return ERR_CAST(obj);
 348
 349         npages = PAGE_ALIGN(attach->dmabuf->size) / PAGE_SIZE;
 350
 351         obj->table = sg;
 352         obj->pages = kvmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
 353         if (!obj->pages) {
 354                 __vgem_gem_destroy(obj);
 355                 return ERR_PTR(-ENOMEM);
 356         }
 357
 358         obj->pages_pin_count++; /* perma-pinned */
 359         drm_prime_sg_to_page_addr_arrays(obj->table, obj->pages, NULL,
 360                                         npages);
 361         return &obj->base;
 362 }
 363
 364 static int vgem_prime_vmap(struct drm_gem_object *obj, struct dma_buf_map *map)
 365 {
 366         struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
 367         long n_pages = obj->size >> PAGE_SHIFT;
 368         struct page **pages;
 369         void *vaddr;
 370
 371         pages = vgem_pin_pages(bo);
 372         if (IS_ERR(pages))
 373                 return PTR_ERR(pages);
 374
 375         vaddr = vmap(pages, n_pages, 0, pgprot_writecombine(PAGE_KERNEL));
 376         if (!vaddr)
 377                 return -ENOMEM;
 378         dma_buf_map_set_vaddr(map, vaddr);
 379
 380         return 0;
 381 }
 382
 383 static void vgem_prime_vunmap(struct drm_gem_object *obj, struct dma_buf_map *map)
 384 {
 385         struct drm_vgem_gem_object *bo = to_vgem_bo(obj);
 386
 387         vunmap(map->vaddr);
 388         vgem_unpin_pages(bo);
 389 }
 390
 391 static int vgem_prime_mmap(struct drm_gem_object *obj,
 392                            struct vm_area_struct *vma)
 393 {
 394         int ret;
 395
 396         if (obj->size < vma->vm_end - vma->vm_start)
 397                 return -EINVAL;
 398
 399         if (!obj->filp)
 400                 return -ENODEV;
 401
 402         ret = call_mmap(obj->filp, vma);
 403         if (ret)
 404                 return ret;
 405
 406         fput(vma->vm_file);
 407         vma->vm_file = get_file(obj->filp);
 408         vma->vm_flags |= VM_DONTEXPAND | VM_DONTDUMP;
 409         vma->vm_page_prot = pgprot_writecombine(vm_get_page_prot(vma->vm_flags));
 410
 411         return 0;
 412 }
 413
 414 static const struct drm_gem_object_funcs vgem_gem_object_funcs = {
 415         .free = vgem_gem_free_object,
 416         .pin = vgem_prime_pin,
 417         .unpin = vgem_prime_unpin,
 418         .get_sg_table = vgem_prime_get_sg_table,
 419         .vmap = vgem_prime_vmap,
 420         .vunmap = vgem_prime_vunmap,
 421         .vm_ops = &vgem_gem_vm_ops,
 422 };
 423
 424 static const struct drm_driver vgem_driver = {
 425         .driver_features                = DRIVER_GEM | DRIVER_RENDER,
 426         .open                           = vgem_open,
 427         .postclose                      = vgem_postclose,
 428         .ioctls                         = vgem_ioctls,
 429         .num_ioctls                     = ARRAY_SIZE(vgem_ioctls),
 430         .fops                           = &vgem_driver_fops,
 431
 432         .dumb_create                    = vgem_gem_dumb_create,
 433
 434         .prime_handle_to_fd = drm_gem_prime_handle_to_fd,
 435         .prime_fd_to_handle = drm_gem_prime_fd_to_handle,
 436         .gem_prime_import = vgem_prime_import,
 437         .gem_prime_import_sg_table = vgem_prime_import_sg_table,
 438         .gem_prime_mmap = vgem_prime_mmap,
 439
 440         .name   = DRIVER_NAME,
 441         .desc   = DRIVER_DESC,
 442         .date   = DRIVER_DATE,
 443         .major  = DRIVER_MAJOR,
 444         .minor  = DRIVER_MINOR,
 445 };
 446
 447 static int __init vgem_init(void)
 448 {
 449         int ret;
 450         struct platform_device *pdev;
 451
 452         pdev = platform_device_register_simple("vgem", -1, NULL, 0);
 453         if (IS_ERR(pdev))
 454                 return PTR_ERR(pdev);
 455
 456         if (!devres_open_group(&pdev->dev, NULL, GFP_KERNEL)) {
 457                 ret = -ENOMEM;
 458                 goto out_unregister;
 459         }
 460
 461         dma_coerce_mask_and_coherent(&pdev->dev,
 462                                      DMA_BIT_MASK(64));
 463
 464         vgem_device = devm_drm_dev_alloc(&pdev->dev, &vgem_driver,
 465                                          struct vgem_device, drm);
 466         if (IS_ERR(vgem_device)) {
 467                 ret = PTR_ERR(vgem_device);
 468                 goto out_devres;
 469         }
 470         vgem_device->platform = pdev;
 471
 472         /* Final step: expose the device/driver to userspace */
 473         ret = drm_dev_register(&vgem_device->drm, 0);
 474         if (ret)
 475                 goto out_devres;
 476
 477         return 0;
 478
 479 out_devres:
 480         devres_release_group(&pdev->dev, NULL);
 481 out_unregister:
 482         platform_device_unregister(pdev);
 483         return ret;
 484 }
 485
 486 static void __exit vgem_exit(void)
 487 {
 488         struct platform_device *pdev = vgem_device->platform;
 489
 490         drm_dev_unregister(&vgem_device->drm);
 491         devres_release_group(&pdev->dev, NULL);
 492         platform_device_unregister(pdev);
 493 }
 494
 495 module_init(vgem_init);
 496 module_exit(vgem_exit);
 497
 498 MODULE_AUTHOR("Red Hat, Inc.");
 499 MODULE_AUTHOR("Intel Corporation");
 500 MODULE_DESCRIPTION(DRIVER_DESC);
 501 MODULE_LICENSE("GPL and additional rights");