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
  23 struct sg_table *etnaviv_gem_prime_get_sg_table(struct drm_gem_object *obj)
  24 {
  25         struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  26
  27         BUG_ON(!etnaviv_obj->sgt);  /* should have already pinned! */
  28
  29         return etnaviv_obj->sgt;
  30 }
  31
  32 void *etnaviv_gem_prime_vmap(struct drm_gem_object *obj)
  33 {
  34         return etnaviv_gem_vmap(obj);
  35 }
  36
  37 void etnaviv_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
  38 {
  39         /* TODO msm_gem_vunmap() */
  40 }
  41
  42 int etnaviv_gem_prime_pin(struct drm_gem_object *obj)
  43 {
  44         if (!obj->import_attach) {
  45                 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  46
  47                 mutex_lock(&etnaviv_obj->lock);
  48                 etnaviv_gem_get_pages(etnaviv_obj);
  49                 mutex_unlock(&etnaviv_obj->lock);
  50         }
  51         return 0;
  52 }
  53
  54 void etnaviv_gem_prime_unpin(struct drm_gem_object *obj)
  55 {
  56         if (!obj->import_attach) {
  57                 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj);
  58
  59                 mutex_lock(&etnaviv_obj->lock);
  60                 etnaviv_gem_put_pages(to_etnaviv_bo(obj));
  61                 mutex_unlock(&etnaviv_obj->lock);
  62         }
  63 }
  64
  65 static void etnaviv_gem_prime_release(struct etnaviv_gem_object *etnaviv_obj)
  66 {
  67         if (etnaviv_obj->vaddr)
  68                 dma_buf_vunmap(etnaviv_obj->base.import_attach->dmabuf,
  69                                etnaviv_obj->vaddr);
  70
  71         /* Don't drop the pages for imported dmabuf, as they are not
  72          * ours, just free the array we allocated:
  73          */
  74         if (etnaviv_obj->pages)
  75                 drm_free_large(etnaviv_obj->pages);
  76
  77         drm_prime_gem_destroy(&etnaviv_obj->base, etnaviv_obj->sgt);
  78 }
  79
  80 static void *etnaviv_gem_prime_vmap_impl(struct etnaviv_gem_object *etnaviv_obj)
  81 {
  82         lockdep_assert_held(&etnaviv_obj->lock);
  83
  84         return dma_buf_vmap(etnaviv_obj->base.import_attach->dmabuf);
  85 }
  86
  87 static const struct etnaviv_gem_ops etnaviv_gem_prime_ops = {
  88         /* .get_pages should never be called */
  89         .release = etnaviv_gem_prime_release,
  90         .vmap = etnaviv_gem_prime_vmap_impl,
  91 };
  92
  93 struct drm_gem_object *etnaviv_gem_prime_import_sg_table(struct drm_device *dev,
  94         struct dma_buf_attachment *attach, struct sg_table *sgt)
  95 {
  96         struct etnaviv_gem_object *etnaviv_obj;
  97         size_t size = PAGE_ALIGN(attach->dmabuf->size);
  98         int ret, npages;
  99
 100         ret = etnaviv_gem_new_private(dev, size, ETNA_BO_WC,
 101                                       attach->dmabuf->resv,
 102                                       &etnaviv_gem_prime_ops, &etnaviv_obj);
 103         if (ret < 0)
 104                 return ERR_PTR(ret);
 105
 106         npages = size / PAGE_SIZE;
 107
 108         etnaviv_obj->sgt = sgt;
 109         etnaviv_obj->pages = drm_malloc_ab(npages, sizeof(struct page *));
 110         if (!etnaviv_obj->pages) {
 111                 ret = -ENOMEM;
 112                 goto fail;
 113         }
 114
 115         ret = drm_prime_sg_to_page_addr_arrays(sgt, etnaviv_obj->pages,
 116                                                NULL, npages);
 117         if (ret)
 118                 goto fail;
 119
 120         ret = etnaviv_gem_obj_add(dev, &etnaviv_obj->base);
 121         if (ret)
 122                 goto fail;
 123
 124         return &etnaviv_obj->base;
 125
 126 fail:
 127         drm_gem_object_unreference_unlocked(&etnaviv_obj->base);
 128
 129         return ERR_PTR(ret);
 130 }