drivers/gpu/drm/udl/udl_gem.c

   1 /*
   2  * Copyright (C) 2012 Red Hat
   3  *
   4  * This file is subject to the terms and conditions of the GNU General Public
   5  * License v2. See the file COPYING in the main directory of this archive for
   6  * more details.
   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
  14 struct 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         return obj;
  29 }
  30
  31 static int
  32 udl_gem_create(struct drm_file *file,
  33                struct drm_device *dev,
  34                uint64_t size,
  35                uint32_t *handle_p)
  36 {
  37         struct udl_gem_object *obj;
  38         int ret;
  39         u32 handle;
  40
  41         size = roundup(size, PAGE_SIZE);
  42
  43         obj = udl_gem_alloc_object(dev, size);
  44         if (obj == NULL)
  45                 return -ENOMEM;
  46
  47         ret = drm_gem_handle_create(file, &obj->base, &handle);
  48         if (ret) {
  49                 drm_gem_object_release(&obj->base);
  50                 kfree(obj);
  51                 return ret;
  52         }
  53
  54         drm_gem_object_unreference(&obj->base);
  55         *handle_p = handle;
  56         return 0;
  57 }
  58
  59 int udl_dumb_create(struct drm_file *file,
  60                     struct drm_device *dev,
  61                     struct drm_mode_create_dumb *args)
  62 {
  63         args->pitch = args->width * ((args->bpp + 1) / 8);
  64         args->size = args->pitch * args->height;
  65         return udl_gem_create(file, dev,
  66                               args->size, &args->handle);
  67 }
  68
  69 int udl_drm_gem_mmap(struct file *filp, struct vm_area_struct *vma)
  70 {
  71         int ret;
  72
  73         ret = drm_gem_mmap(filp, vma);
  74         if (ret)
  75                 return ret;
  76
  77         vma->vm_flags &= ~VM_PFNMAP;
  78         vma->vm_flags |= VM_MIXEDMAP;
  79
  80         return ret;
  81 }
  82
  83 int udl_gem_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
  84 {
  85         struct udl_gem_object *obj = to_udl_bo(vma->vm_private_data);
  86         struct page *page;
  87         unsigned int page_offset;
  88         int ret = 0;
  89
  90         page_offset = ((unsigned long)vmf->virtual_address - vma->vm_start) >>
  91                 PAGE_SHIFT;
  92
  93         if (!obj->pages)
  94                 return VM_FAULT_SIGBUS;
  95
  96         page = obj->pages[page_offset];
  97         ret = vm_insert_page(vma, (unsigned long)vmf->virtual_address, page);
  98         switch (ret) {
  99         case -EAGAIN:
 100         case 0:
 101         case -ERESTARTSYS:
 102                 return VM_FAULT_NOPAGE;
 103         case -ENOMEM:
 104                 return VM_FAULT_OOM;
 105         default:
 106                 return VM_FAULT_SIGBUS;
 107         }
 108 }
 109
 110 static int udl_gem_get_pages(struct udl_gem_object *obj, gfp_t gfpmask)
 111 {
 112         struct page **pages;
 113
 114         if (obj->pages)
 115                 return 0;
 116
 117         pages = drm_gem_get_pages(&obj->base, gfpmask);
 118         if (IS_ERR(pages))
 119                 return PTR_ERR(pages);
 120
 121         obj->pages = pages;
 122
 123         return 0;
 124 }
 125
 126 static void udl_gem_put_pages(struct udl_gem_object *obj)
 127 {
 128         if (obj->base.import_attach) {
 129                 drm_free_large(obj->pages);
 130                 obj->pages = NULL;
 131                 return;
 132         }
 133
 134         drm_gem_put_pages(&obj->base, obj->pages, false, false);
 135         obj->pages = NULL;
 136 }
 137
 138 int udl_gem_vmap(struct udl_gem_object *obj)
 139 {
 140         int page_count = obj->base.size / PAGE_SIZE;
 141         int ret;
 142
 143         if (obj->base.import_attach) {
 144                 obj->vmapping = dma_buf_vmap(obj->base.import_attach->dmabuf);
 145                 if (!obj->vmapping)
 146                         return -ENOMEM;
 147                 return 0;
 148         }
 149
 150         ret = udl_gem_get_pages(obj, GFP_KERNEL);
 151         if (ret)
 152                 return ret;
 153
 154         obj->vmapping = vmap(obj->pages, page_count, 0, PAGE_KERNEL);
 155         if (!obj->vmapping)
 156                 return -ENOMEM;
 157         return 0;
 158 }
 159
 160 void udl_gem_vunmap(struct udl_gem_object *obj)
 161 {
 162         if (obj->base.import_attach) {
 163                 dma_buf_vunmap(obj->base.import_attach->dmabuf, obj->vmapping);
 164                 return;
 165         }
 166
 167         if (obj->vmapping)
 168                 vunmap(obj->vmapping);
 169
 170         udl_gem_put_pages(obj);
 171 }
 172
 173 void udl_gem_free_object(struct drm_gem_object *gem_obj)
 174 {
 175         struct udl_gem_object *obj = to_udl_bo(gem_obj);
 176
 177         if (obj->vmapping)
 178                 udl_gem_vunmap(obj);
 179
 180         if (gem_obj->import_attach) {
 181                 drm_prime_gem_destroy(gem_obj, obj->sg);
 182                 put_device(gem_obj->dev->dev);
 183         }
 184
 185         if (obj->pages)
 186                 udl_gem_put_pages(obj);
 187
 188         drm_gem_free_mmap_offset(gem_obj);
 189 }
 190
 191 /* the dumb interface doesn't work with the GEM straight MMAP
 192    interface, it expects to do MMAP on the drm fd, like normal */
 193 int udl_gem_mmap(struct drm_file *file, struct drm_device *dev,
 194                  uint32_t handle, uint64_t *offset)
 195 {
 196         struct udl_gem_object *gobj;
 197         struct drm_gem_object *obj;
 198         int ret = 0;
 199
 200         mutex_lock(&dev->struct_mutex);
 201         obj = drm_gem_object_lookup(dev, file, handle);
 202         if (obj == NULL) {
 203                 ret = -ENOENT;
 204                 goto unlock;
 205         }
 206         gobj = to_udl_bo(obj);
 207
 208         ret = udl_gem_get_pages(gobj, GFP_KERNEL);
 209         if (ret)
 210                 goto out;
 211         ret = drm_gem_create_mmap_offset(obj);
 212         if (ret)
 213                 goto out;
 214
 215         *offset = drm_vma_node_offset_addr(&gobj->base.vma_node);
 216
 217 out:
 218         drm_gem_object_unreference(&gobj->base);
 219 unlock:
 220         mutex_unlock(&dev->struct_mutex);
 221         return ret;
 222 }
 223
 224 static int udl_prime_create(struct drm_device *dev,
 225                             size_t size,
 226                             struct sg_table *sg,
 227                             struct udl_gem_object **obj_p)
 228 {
 229         struct udl_gem_object *obj;
 230         int npages;
 231
 232         npages = size / PAGE_SIZE;
 233
 234         *obj_p = NULL;
 235         obj = udl_gem_alloc_object(dev, npages * PAGE_SIZE);
 236         if (!obj)
 237                 return -ENOMEM;
 238
 239         obj->sg = sg;
 240         obj->pages = drm_malloc_ab(npages, sizeof(struct page *));
 241         if (obj->pages == NULL) {
 242                 DRM_ERROR("obj pages is NULL %d\n", npages);
 243                 return -ENOMEM;
 244         }
 245
 246         drm_prime_sg_to_page_addr_arrays(sg, obj->pages, NULL, npages);
 247
 248         *obj_p = obj;
 249         return 0;
 250 }
 251
 252 struct drm_gem_object *udl_gem_prime_import(struct drm_device *dev,
 253                                 struct dma_buf *dma_buf)
 254 {
 255         struct dma_buf_attachment *attach;
 256         struct sg_table *sg;
 257         struct udl_gem_object *uobj;
 258         int ret;
 259
 260         /* need to attach */
 261         get_device(dev->dev);
 262         attach = dma_buf_attach(dma_buf, dev->dev);
 263         if (IS_ERR(attach)) {
 264                 put_device(dev->dev);
 265                 return ERR_CAST(attach);
 266         }
 267
 268         get_dma_buf(dma_buf);
 269
 270         sg = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
 271         if (IS_ERR(sg)) {
 272                 ret = PTR_ERR(sg);
 273                 goto fail_detach;
 274         }
 275
 276         ret = udl_prime_create(dev, dma_buf->size, sg, &uobj);
 277         if (ret) {
 278                 goto fail_unmap;
 279         }
 280
 281         uobj->base.import_attach = attach;
 282
 283         return &uobj->base;
 284
 285 fail_unmap:
 286         dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
 287 fail_detach:
 288         dma_buf_detach(dma_buf, attach);
 289         dma_buf_put(dma_buf);
 290         put_device(dev->dev);
 291         return ERR_PTR(ret);
 292 }