LXR linux/drivers/gpu/drm/etnaviv/etnaviv_gem

   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
  10static struct lock_class_key etnaviv_prime_lock_class;
  11
  12struct 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
  23void *etnaviv_gem_prime_vmap(struct drm_gem_object *obj)
  24{
  25        return etnaviv_gem_vmap(obj);
  26}
  27
  28void etnaviv_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
  29{
  30        /* TODO msm_gem_vunmap() */
  31}
  32
  33int 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
  46int 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
  58void 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
  69static 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
  84static 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
  91static int etnaviv_gem_prime_mmap_obj(struct etnaviv_gem_object *etnaviv_obj,
  92                struct vm_area_struct *vma)
  93{
  94        return dma_buf_mmap(etnaviv_obj->base.dma_buf, vma, 0);
  95}
  96
  97static const struct etnaviv_gem_ops etnaviv_gem_prime_ops = {
  98        /* .get_pages should never be called */
  99        .release = etnaviv_gem_prime_release,
 100        .vmap = etnaviv_gem_prime_vmap_impl,
 101        .mmap = etnaviv_gem_prime_mmap_obj,
 102};
 103
 104struct drm_gem_object *etnaviv_gem_prime_import_sg_table(struct drm_device *dev,
 105        struct dma_buf_attachment *attach, struct sg_table *sgt)
 106{
 107        struct etnaviv_gem_object *etnaviv_obj;
 108        size_t size = PAGE_ALIGN(attach->dmabuf->size);
 109        int ret, npages;
 110
 111        ret = etnaviv_gem_new_private(dev, size, ETNA_BO_WC,
 112                                      attach->dmabuf->resv,
 113                                      &etnaviv_gem_prime_ops, &etnaviv_obj);
 114        if (ret < 0)
 115                return ERR_PTR(ret);
 116
 117        lockdep_set_class(&etnaviv_obj->lock, &etnaviv_prime_lock_class);
 118
 119        npages = size / PAGE_SIZE;
 120
 121        etnaviv_obj->sgt = sgt;
 122        etnaviv_obj->pages = kvmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
 123        if (!etnaviv_obj->pages) {
 124                ret = -ENOMEM;
 125                goto fail;
 126        }
 127
 128        ret = drm_prime_sg_to_page_addr_arrays(sgt, etnaviv_obj->pages,
 129                                               NULL, npages);
 130        if (ret)
 131                goto fail;
 132
 133        etnaviv_gem_obj_add(dev, &etnaviv_obj->base);
 134
 135        return &etnaviv_obj->base;
 136
 137fail:
 138        drm_gem_object_put_unlocked(&etnaviv_obj->base);
 139
 140        return ERR_PTR(ret);
 141}
 142