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 <linux/dma-buf.h>
   7 #include "etnaviv_drv.h"
   8 #include "etnaviv_gem.h"
   9
  10 static struct lock_class_key etnaviv_prime_lock_class;
  11
  12 struct sg_table *etnaviv_gem_prime_get_sg_table(struct drm_gem_object *obj)
  13 {
  14         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  15         int npages = obj->size >> PAGE_SHIFT;
  16
  17         if (WARN_ON(!etnaviv_obj->pages))  /* should have already pinned! */
  18                 return ERR_PTR(-EINVAL);
  19
  20         return drm_prime_pages_to_sg(etnaviv_obj->pages, npages);
  21 }
  22
  23 void *etnaviv_gem_prime_vmap(struct drm_gem_object *obj)
  24 {
  25         return etnaviv_gem_vmap(obj);
  26 }
  27
  28 void etnaviv_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
  29 {
  30         /* TODO msm_gem_vunmap() */
  31 }
  32
  33 int etnaviv_gem_prime_mmap(struct drm_gem_object *obj,
  34                            struct vm_area_struct *vma)
  35 {
  36         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  37         int ret;
  38
  39         ret = drm_gem_mmap_obj(obj, obj->size, vma);
  40         if (ret < 0)
  41                 return ret;
  42
  43         return etnaviv_obj->ops->mmap(etnaviv_obj, vma);
  44 }
  45
  46 int etnaviv_gem_prime_pin(struct drm_gem_object *obj)
  47 {
  48         if (!obj->import_attach) {
  49                 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  50
  51                 mutex_lock(&etnaviv_obj->lock);
  52                 etnaviv_gem_get_pages(etnaviv_obj);
  53                 mutex_unlock(&etnaviv_obj->lock);
  54         }
  55         return 0;
  56 }
  57
  58 void etnaviv_gem_prime_unpin(struct drm_gem_object *obj)
  59 {
  60         if (!obj->import_attach) {
  61                 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  62
  63                 mutex_lock(&etnaviv_obj->lock);
  64                 etnaviv_gem_put_pages(to_etnaviv_bo(obj));
  65                 mutex_unlock(&etnaviv_obj->lock);
  66         }
  67 }
  68
  69 static void etnaviv_gem_prime_release(struct etnaviv_gem_object *etnaviv_obj)
  70 {
  71         if (etnaviv_obj->vaddr)
  72                 dma_buf_vunmap(etnaviv_obj->base.import_attach->dmabuf,
  73                                etnaviv_obj->vaddr);
  74
  75         /* Don't drop the pages for imported dmabuf, as they are not
  76          * ours, just free the array we allocated:
  77          */
  78         if (etnaviv_obj->pages)
  79                 kvfree(etnaviv_obj->pages);
  80
  81         drm_prime_gem_destroy(&etnaviv_obj->base, etnaviv_obj->sgt);
  82 }
  83
  84 static void *etnaviv_gem_prime_vmap_impl(struct etnaviv_gem_object *etnaviv_obj)
  85 {
  86         lockdep_assert_held(&etnaviv_obj->lock);
  87
  88         return dma_buf_vmap(etnaviv_obj->base.import_attach->dmabuf);
  89 }
  90
  91 static int etnaviv_gem_prime_mmap_obj(struct etnaviv_gem_object *etnaviv_obj,
  92                 struct vm_area_struct *vma)
  93 {
  94         int ret;
  95
  96         ret = dma_buf_mmap(etnaviv_obj->base.dma_buf, vma, 0);
  97         if (!ret) {
  98                 /* Drop the reference acquired by drm_gem_mmap_obj(). */
  99                 drm_gem_object_put(&etnaviv_obj->base);
 100         }
 101
 102         return ret;
 103 }
 104
 105 static const struct etnaviv_gem_ops etnaviv_gem_prime_ops = {
 106         /* .get_pages should never be called */
 107         .release = etnaviv_gem_prime_release,
 108         .vmap = etnaviv_gem_prime_vmap_impl,
 109         .mmap = etnaviv_gem_prime_mmap_obj,
 110 };
 111
 112 struct drm_gem_object *etnaviv_gem_prime_import_sg_table(struct drm_device *dev,
 113         struct dma_buf_attachment *attach, struct sg_table *sgt)
 114 {
 115         struct etnaviv_gem_object *etnaviv_obj;
 116         size_t size = PAGE_ALIGN(attach->dmabuf->size);
 117         int ret, npages;
 118
 119         ret = etnaviv_gem_new_private(dev, size, ETNA_BO_WC,
 120                                       attach->dmabuf->resv,
 121                                       &etnaviv_gem_prime_ops, &etnaviv_obj);
 122         if (ret < 0)
 123                 return ERR_PTR(ret);
 124
 125         lockdep_set_class(&etnaviv_obj->lock, &etnaviv_prime_lock_class);
 126
 127         npages = size / PAGE_SIZE;
 128
 129         etnaviv_obj->sgt = sgt;
 130         etnaviv_obj->pages = kvmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
 131         if (!etnaviv_obj->pages) {
 132                 ret = -ENOMEM;
 133                 goto fail;
 134         }
 135
 136         ret = drm_prime_sg_to_page_addr_arrays(sgt, etnaviv_obj->pages,
 137                                                NULL, 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_unlocked(&etnaviv_obj->base);
 147
 148         return ERR_PTR(ret);
 149 }
 150
 151 struct reservation_object *etnaviv_gem_prime_res_obj(struct drm_gem_object *obj)
 152 {
 153         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
 154
 155         return etnaviv_obj->resv;
 156 }