drivers/gpu/drm/nouveau/nouveau_prime.c

   1
   2 #include "drmP.h"
   3 #include "drm.h"
   4
   5 #include "nouveau_drv.h"
   6 #include "nouveau_drm.h"
   7 #include "nouveau_dma.h"
   8
   9 #include <linux/dma-buf.h>
  10
  11 static struct sg_table *nouveau_gem_map_dma_buf(struct dma_buf_attachment *attachment,
  12                                           enum dma_data_direction dir)
  13 {
  14         struct nouveau_bo *nvbo = attachment->dmabuf->priv;
  15         struct drm_device *dev = nvbo->gem->dev;
  16         int npages = nvbo->bo.num_pages;
  17         struct sg_table *sg;
  18         int nents;
  19
  20         mutex_lock(&dev->struct_mutex);
  21         sg = drm_prime_pages_to_sg(nvbo->bo.ttm->pages, npages);
  22         nents = dma_map_sg(attachment->dev, sg->sgl, sg->nents, dir);
  23         mutex_unlock(&dev->struct_mutex);
  24         return sg;
  25 }
  26
  27 static void nouveau_gem_unmap_dma_buf(struct dma_buf_attachment *attachment,
  28                                       struct sg_table *sg, enum dma_data_direction dir)
  29 {
  30         dma_unmap_sg(attachment->dev, sg->sgl, sg->nents, dir);
  31         sg_free_table(sg);
  32         kfree(sg);
  33 }
  34
  35 static void nouveau_gem_dmabuf_release(struct dma_buf *dma_buf)
  36 {
  37         struct nouveau_bo *nvbo = dma_buf->priv;
  38
  39         if (nvbo->gem->export_dma_buf == dma_buf) {
  40                 nvbo->gem->export_dma_buf = NULL;
  41                 drm_gem_object_unreference_unlocked(nvbo->gem);
  42         }
  43 }
  44
  45 static void *nouveau_gem_kmap_atomic(struct dma_buf *dma_buf, unsigned long page_num)
  46 {
  47         return NULL;
  48 }
  49
  50 static void nouveau_gem_kunmap_atomic(struct dma_buf *dma_buf, unsigned long page_num, void *addr)
  51 {
  52
  53 }
  54 static void *nouveau_gem_kmap(struct dma_buf *dma_buf, unsigned long page_num)
  55 {
  56         return NULL;
  57 }
  58
  59 static void nouveau_gem_kunmap(struct dma_buf *dma_buf, unsigned long page_num, void *addr)
  60 {
  61
  62 }
  63
  64 static const struct dma_buf_ops nouveau_dmabuf_ops =  {
  65         .map_dma_buf = nouveau_gem_map_dma_buf,
  66         .unmap_dma_buf = nouveau_gem_unmap_dma_buf,
  67         .release = nouveau_gem_dmabuf_release,
  68         .kmap = nouveau_gem_kmap,
  69         .kmap_atomic = nouveau_gem_kmap_atomic,
  70         .kunmap = nouveau_gem_kunmap,
  71         .kunmap_atomic = nouveau_gem_kunmap_atomic,
  72 };
  73
  74 static int
  75 nouveau_prime_new(struct drm_device *dev,
  76                   size_t size,
  77                   struct sg_table *sg,
  78                   struct nouveau_bo **pnvbo)
  79 {
  80         struct nouveau_bo *nvbo;
  81         u32 flags = 0;
  82         int ret;
  83
  84         flags = TTM_PL_FLAG_TT;
  85
  86         ret = nouveau_bo_new(dev, size, 0, flags, 0, 0,
  87                              sg, pnvbo);
  88         if (ret)
  89                 return ret;
  90         nvbo = *pnvbo;
  91
  92         /* we restrict allowed domains on nv50+ to only the types
  93          * that were requested at creation time.  not possibly on
  94          * earlier chips without busting the ABI.
  95          */
  96         nvbo->valid_domains = NOUVEAU_GEM_DOMAIN_GART;
  97         nvbo->gem = drm_gem_object_alloc(dev, nvbo->bo.mem.size);
  98         if (!nvbo->gem) {
  99                 nouveau_bo_ref(NULL, pnvbo);
 100                 return -ENOMEM;
 101         }
 102
 103         nvbo->gem->driver_private = nvbo;
 104         return 0;
 105 }
 106
 107 struct dma_buf *nouveau_gem_prime_export(struct drm_device *dev,
 108                                 struct drm_gem_object *obj, int flags)
 109 {
 110         struct nouveau_bo *nvbo = nouveau_gem_object(obj);
 111         int ret = 0;
 112
 113         /* pin buffer into GTT */
 114         ret = nouveau_bo_pin(nvbo, TTM_PL_FLAG_TT);
 115         if (ret)
 116                 return ERR_PTR(-EINVAL);
 117
 118         return dma_buf_export(nvbo, &nouveau_dmabuf_ops, obj->size, flags);
 119 }
 120
 121 struct drm_gem_object *nouveau_gem_prime_import(struct drm_device *dev,
 122                                 struct dma_buf *dma_buf)
 123 {
 124         struct dma_buf_attachment *attach;
 125         struct sg_table *sg;
 126         struct nouveau_bo *nvbo;
 127         int ret;
 128
 129         if (dma_buf->ops == &nouveau_dmabuf_ops) {
 130                 nvbo = dma_buf->priv;
 131                 if (nvbo->gem) {
 132                         if (nvbo->gem->dev == dev) {
 133                                 drm_gem_object_reference(nvbo->gem);
 134                                 return nvbo->gem;
 135                         }
 136                 }
 137         }
 138         /* need to attach */
 139         attach = dma_buf_attach(dma_buf, dev->dev);
 140         if (IS_ERR(attach))
 141                 return ERR_PTR(PTR_ERR(attach));
 142
 143         sg = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
 144         if (IS_ERR(sg)) {
 145                 ret = PTR_ERR(sg);
 146                 goto fail_detach;
 147         }
 148
 149         ret = nouveau_prime_new(dev, dma_buf->size, sg, &nvbo);
 150         if (ret)
 151                 goto fail_unmap;
 152
 153         nvbo->gem->import_attach = attach;
 154
 155         return nvbo->gem;
 156
 157 fail_unmap:
 158         dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
 159 fail_detach:
 160         dma_buf_detach(dma_buf, attach);
 161         return ERR_PTR(ret);
 162 }
 163