drivers/gpu/drm/etnaviv/etnaviv_gem_prime.c

   1 /*
   2  * Copyright (C) 2013 Red Hat
   3  * Author: Rob Clark <robdclark@gmail.com>
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 as published by
   7  * the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  12  * more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #include <linux/dma-buf.h>
  19 #include "etnaviv_drv.h"
  20 #include "etnaviv_gem.h"
  21
  22 static struct lock_class_key etnaviv_prime_lock_class;
  23
  24 struct sg_table *etnaviv_gem_prime_get_sg_table(struct drm_gem_object *obj)
  25 {
  26         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  27         int npages = obj->size >> PAGE_SHIFT;
  28
  29         if (WARN_ON(!etnaviv_obj->pages))  /* should have already pinned! */
  30                 return NULL;
  31
  32         return drm_prime_pages_to_sg(etnaviv_obj->pages, npages);
  33 }
  34
  35 void *etnaviv_gem_prime_vmap(struct drm_gem_object *obj)
  36 {
  37         return etnaviv_gem_vmap(obj);
  38 }
  39
  40 void etnaviv_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
  41 {
  42         /* TODO msm_gem_vunmap() */
  43 }
  44
  45 int etnaviv_gem_prime_mmap(struct drm_gem_object *obj,
  46                            struct vm_area_struct *vma)
  47 {
  48         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  49         int ret;
  50
  51         ret = drm_gem_mmap_obj(obj, obj->size, vma);
  52         if (ret < 0)
  53                 return ret;
  54
  55         return etnaviv_obj->ops->mmap(etnaviv_obj, vma);
  56 }
  57
  58 int etnaviv_gem_prime_pin(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_get_pages(etnaviv_obj);
  65                 mutex_unlock(&etnaviv_obj->lock);
  66         }
  67         return 0;
  68 }
  69
  70 void etnaviv_gem_prime_unpin(struct drm_gem_object *obj)
  71 {
  72         if (!obj->import_attach) {
  73                 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  74
  75                 mutex_lock(&etnaviv_obj->lock);
  76                 etnaviv_gem_put_pages(to_etnaviv_bo(obj));
  77                 mutex_unlock(&etnaviv_obj->lock);
  78         }
  79 }
  80
  81 static void etnaviv_gem_prime_release(struct etnaviv_gem_object *etnaviv_obj)
  82 {
  83         if (etnaviv_obj->vaddr)
  84                 dma_buf_vunmap(etnaviv_obj->base.import_attach->dmabuf,
  85                                etnaviv_obj->vaddr);
  86
  87         /* Don't drop the pages for imported dmabuf, as they are not
  88          * ours, just free the array we allocated:
  89          */
  90         if (etnaviv_obj->pages)
  91                 kvfree(etnaviv_obj->pages);
  92
  93         drm_prime_gem_destroy(&etnaviv_obj->base, etnaviv_obj->sgt);
  94 }
  95
  96 static void *etnaviv_gem_prime_vmap_impl(struct etnaviv_gem_object *etnaviv_obj)
  97 {
  98         lockdep_assert_held(&etnaviv_obj->lock);
  99
 100         return dma_buf_vmap(etnaviv_obj->base.import_attach->dmabuf);
 101 }
 102
 103 static int etnaviv_gem_prime_mmap_obj(struct etnaviv_gem_object *etnaviv_obj,
 104                 struct vm_area_struct *vma)
 105 {
 106         return dma_buf_mmap(etnaviv_obj->base.dma_buf, vma, 0);
 107 }
 108
 109 static const struct etnaviv_gem_ops etnaviv_gem_prime_ops = {
 110         /* .get_pages should never be called */
 111         .release = etnaviv_gem_prime_release,
 112         .vmap = etnaviv_gem_prime_vmap_impl,
 113         .mmap = etnaviv_gem_prime_mmap_obj,
 114 };
 115
 116 struct drm_gem_object *etnaviv_gem_prime_import_sg_table(struct drm_device *dev,
 117         struct dma_buf_attachment *attach, struct sg_table *sgt)
 118 {
 119         struct etnaviv_gem_object *etnaviv_obj;
 120         size_t size = PAGE_ALIGN(attach->dmabuf->size);
 121         int ret, npages;
 122
 123         ret = etnaviv_gem_new_private(dev, size, ETNA_BO_WC,
 124                                       attach->dmabuf->resv,
 125                                       &etnaviv_gem_prime_ops, &etnaviv_obj);
 126         if (ret < 0)
 127                 return ERR_PTR(ret);
 128
 129         lockdep_set_class(&etnaviv_obj->lock, &etnaviv_prime_lock_class);
 130
 131         npages = size / PAGE_SIZE;
 132
 133         etnaviv_obj->sgt = sgt;
 134         etnaviv_obj->pages = kvmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
 135         if (!etnaviv_obj->pages) {
 136                 ret = -ENOMEM;
 137                 goto fail;
 138         }
 139
 140         ret = drm_prime_sg_to_page_addr_arrays(sgt, etnaviv_obj->pages,
 141                                                NULL, npages);
 142         if (ret)
 143                 goto fail;
 144
 145         etnaviv_gem_obj_add(dev, &etnaviv_obj->base);
 146
 147         return &etnaviv_obj->base;
 148
 149 fail:
 150         drm_gem_object_put_unlocked(&etnaviv_obj->base);
 151
 152         return ERR_PTR(ret);
 153 }
 154
 155 struct reservation_object *etnaviv_gem_prime_res_obj(struct drm_gem_object *obj)
 156 {
 157         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
 158
 159         return etnaviv_obj->resv;
 160 }