Simplify the Nouveau prime implementation by using the default behavior provided
by drm_gem_prime_import and drm_gem_prime_export.

v2: Rename functions to nouveau_gem_prime_get_sg_table and
nouveau_gem_prime_import_sg_table.

Signed-off-by: Aaron Plattner <aplattner at nvidia.com>
Cc: Daniel Vetter <daniel.vetter at ffwll.ch>
Cc: David Airlie <airlied at linux.ie>
---
 drivers/gpu/drm/nouveau/nouveau_bo.h    |   1 -
 drivers/gpu/drm/nouveau/nouveau_drm.c   |   9 +-
 drivers/gpu/drm/nouveau/nouveau_gem.c   |   2 -
 drivers/gpu/drm/nouveau/nouveau_gem.h   |  10 +-
 drivers/gpu/drm/nouveau/nouveau_prime.c | 173 ++++----------------------------
 5 files changed, 34 insertions(+), 161 deletions(-)

diff --git a/drivers/gpu/drm/nouveau/nouveau_bo.h 
b/drivers/gpu/drm/nouveau/nouveau_bo.h
index 25ca379..836677a 100644
--- a/drivers/gpu/drm/nouveau/nouveau_bo.h
+++ b/drivers/gpu/drm/nouveau/nouveau_bo.h
@@ -31,7 +31,6 @@ struct nouveau_bo {
        int pin_refcnt;

        struct ttm_bo_kmap_obj dma_buf_vmap;
-       int vmapping_count;
 };

 static inline struct nouveau_bo *
diff --git a/drivers/gpu/drm/nouveau/nouveau_drm.c 
b/drivers/gpu/drm/nouveau/nouveau_drm.c
index 180a45e..5de1f9a 100644
--- a/drivers/gpu/drm/nouveau/nouveau_drm.c
+++ b/drivers/gpu/drm/nouveau/nouveau_drm.c
@@ -645,8 +645,13 @@ driver = {

        .prime_handle_to_fd = drm_gem_prime_handle_to_fd,
        .prime_fd_to_handle = drm_gem_prime_fd_to_handle,
-       .gem_prime_export = nouveau_gem_prime_export,
-       .gem_prime_import = nouveau_gem_prime_import,
+       .gem_prime_export = drm_gem_prime_export,
+       .gem_prime_import = drm_gem_prime_import,
+       .gem_prime_pin = nouveau_gem_prime_pin,
+       .gem_prime_get_sg_table = nouveau_gem_prime_get_sg_table,
+       .gem_prime_import_sg_table = nouveau_gem_prime_import_sg_table,
+       .gem_prime_vmap = nouveau_gem_prime_vmap,
+       .gem_prime_vunmap = nouveau_gem_prime_vunmap,

        .gem_init_object = nouveau_gem_object_new,
        .gem_free_object = nouveau_gem_object_del,
diff --git a/drivers/gpu/drm/nouveau/nouveau_gem.c 
b/drivers/gpu/drm/nouveau/nouveau_gem.c
index 8bf695c..24e0aab 100644
--- a/drivers/gpu/drm/nouveau/nouveau_gem.c
+++ b/drivers/gpu/drm/nouveau/nouveau_gem.c
@@ -24,8 +24,6 @@
  *
  */

-#include <linux/dma-buf.h>
-
 #include <subdev/fb.h>

 #include "nouveau_drm.h"
diff --git a/drivers/gpu/drm/nouveau/nouveau_gem.h 
b/drivers/gpu/drm/nouveau/nouveau_gem.h
index 5c10492..8d7a3f0 100644
--- a/drivers/gpu/drm/nouveau/nouveau_gem.h
+++ b/drivers/gpu/drm/nouveau/nouveau_gem.h
@@ -35,9 +35,11 @@ extern int nouveau_gem_ioctl_cpu_fini(struct drm_device *, 
void *,
 extern int nouveau_gem_ioctl_info(struct drm_device *, void *,
                                  struct drm_file *);

-extern struct dma_buf *nouveau_gem_prime_export(struct drm_device *dev,
-                               struct drm_gem_object *obj, int flags);
-extern struct drm_gem_object *nouveau_gem_prime_import(struct drm_device *dev,
-                               struct dma_buf *dma_buf);
+extern int nouveau_gem_prime_pin(struct drm_gem_object *);
+extern struct sg_table *nouveau_gem_prime_get_sg_table(struct drm_gem_object 
*);
+extern struct drm_gem_object *nouveau_gem_prime_import_sg_table(
+       struct drm_device *, size_t size, struct sg_table *);
+extern void *nouveau_gem_prime_vmap(struct drm_gem_object *);
+extern void nouveau_gem_prime_vunmap(struct drm_gem_object *, void *);

 #endif
diff --git a/drivers/gpu/drm/nouveau/nouveau_prime.c 
b/drivers/gpu/drm/nouveau/nouveau_prime.c
index b8e05ae..f53e108 100644
--- a/drivers/gpu/drm/nouveau/nouveau_prime.c
+++ b/drivers/gpu/drm/nouveau/nouveau_prime.c
@@ -22,126 +22,42 @@
  * Authors: Dave Airlie
  */

-#include <linux/dma-buf.h>
-
 #include <drm/drmP.h>

 #include "nouveau_drm.h"
 #include "nouveau_gem.h"

-static struct sg_table *nouveau_gem_map_dma_buf(struct dma_buf_attachment 
*attachment,
-                                         enum dma_data_direction dir)
+struct sg_table *nouveau_gem_prime_get_sg_table(struct drm_gem_object *obj)
 {
-       struct nouveau_bo *nvbo = attachment->dmabuf->priv;
-       struct drm_device *dev = nvbo->gem->dev;
+       struct nouveau_bo *nvbo = nouveau_gem_object(obj);
        int npages = nvbo->bo.num_pages;
-       struct sg_table *sg;
-       int nents;
-
-       mutex_lock(&dev->struct_mutex);
-       sg = drm_prime_pages_to_sg(nvbo->bo.ttm->pages, npages);
-       nents = dma_map_sg(attachment->dev, sg->sgl, sg->nents, dir);
-       mutex_unlock(&dev->struct_mutex);
-       return sg;
-}
-
-static void nouveau_gem_unmap_dma_buf(struct dma_buf_attachment *attachment,
-                                     struct sg_table *sg, enum 
dma_data_direction dir)
-{
-       dma_unmap_sg(attachment->dev, sg->sgl, sg->nents, dir);
-       sg_free_table(sg);
-       kfree(sg);
-}
-
-static void nouveau_gem_dmabuf_release(struct dma_buf *dma_buf)
-{
-       struct nouveau_bo *nvbo = dma_buf->priv;
-
-       if (nvbo->gem->export_dma_buf == dma_buf) {
-               nvbo->gem->export_dma_buf = NULL;
-               drm_gem_object_unreference_unlocked(nvbo->gem);
-       }
-}
-
-static void *nouveau_gem_kmap_atomic(struct dma_buf *dma_buf, unsigned long 
page_num)
-{
-       return NULL;
-}
-
-static void nouveau_gem_kunmap_atomic(struct dma_buf *dma_buf, unsigned long 
page_num, void *addr)
-{

-}
-static void *nouveau_gem_kmap(struct dma_buf *dma_buf, unsigned long page_num)
-{
-       return NULL;
+       return drm_prime_pages_to_sg(nvbo->bo.ttm->pages, npages);
 }

-static void nouveau_gem_kunmap(struct dma_buf *dma_buf, unsigned long 
page_num, void *addr)
+void *nouveau_gem_prime_vmap(struct drm_gem_object *obj)
 {
-
-}
-
-static int nouveau_gem_prime_mmap(struct dma_buf *dma_buf, struct 
vm_area_struct *vma)
-{
-       return -EINVAL;
-}
-
-static void *nouveau_gem_prime_vmap(struct dma_buf *dma_buf)
-{
-       struct nouveau_bo *nvbo = dma_buf->priv;
-       struct drm_device *dev = nvbo->gem->dev;
+       struct nouveau_bo *nvbo = nouveau_gem_object(obj);
        int ret;

-       mutex_lock(&dev->struct_mutex);
-       if (nvbo->vmapping_count) {
-               nvbo->vmapping_count++;
-               goto out_unlock;
-       }
-
        ret = ttm_bo_kmap(&nvbo->bo, 0, nvbo->bo.num_pages,
                          &nvbo->dma_buf_vmap);
-       if (ret) {
-               mutex_unlock(&dev->struct_mutex);
+       if (ret)
                return ERR_PTR(ret);
-       }
-       nvbo->vmapping_count = 1;
-out_unlock:
-       mutex_unlock(&dev->struct_mutex);
+
        return nvbo->dma_buf_vmap.virtual;
 }

-static void nouveau_gem_prime_vunmap(struct dma_buf *dma_buf, void *vaddr)
+void nouveau_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr)
 {
-       struct nouveau_bo *nvbo = dma_buf->priv;
-       struct drm_device *dev = nvbo->gem->dev;
+       struct nouveau_bo *nvbo = nouveau_gem_object(obj);

-       mutex_lock(&dev->struct_mutex);
-       nvbo->vmapping_count--;
-       if (nvbo->vmapping_count == 0) {
-               ttm_bo_kunmap(&nvbo->dma_buf_vmap);
-       }
-       mutex_unlock(&dev->struct_mutex);
+       ttm_bo_kunmap(&nvbo->dma_buf_vmap);
 }

-static const struct dma_buf_ops nouveau_dmabuf_ops =  {
-       .map_dma_buf = nouveau_gem_map_dma_buf,
-       .unmap_dma_buf = nouveau_gem_unmap_dma_buf,
-       .release = nouveau_gem_dmabuf_release,
-       .kmap = nouveau_gem_kmap,
-       .kmap_atomic = nouveau_gem_kmap_atomic,
-       .kunmap = nouveau_gem_kunmap,
-       .kunmap_atomic = nouveau_gem_kunmap_atomic,
-       .mmap = nouveau_gem_prime_mmap,
-       .vmap = nouveau_gem_prime_vmap,
-       .vunmap = nouveau_gem_prime_vunmap,
-};
-
-static int
-nouveau_prime_new(struct drm_device *dev,
-                 size_t size,
-                 struct sg_table *sg,
-                 struct nouveau_bo **pnvbo)
+struct drm_gem_object *nouveau_gem_prime_import_sg_table(struct drm_device 
*dev,
+                                                        size_t size,
+                                                        struct sg_table *sg)
 {
        struct nouveau_bo *nvbo;
        u32 flags = 0;
@@ -150,24 +66,22 @@ nouveau_prime_new(struct drm_device *dev,
        flags = TTM_PL_FLAG_TT;

        ret = nouveau_bo_new(dev, size, 0, flags, 0, 0,
-                            sg, pnvbo);
+                            sg, &nvbo);
        if (ret)
-               return ret;
-       nvbo = *pnvbo;
+               return ERR_PTR(ret);

        nvbo->valid_domains = NOUVEAU_GEM_DOMAIN_GART;
        nvbo->gem = drm_gem_object_alloc(dev, nvbo->bo.mem.size);
        if (!nvbo->gem) {
-               nouveau_bo_ref(NULL, pnvbo);
-               return -ENOMEM;
+               nouveau_bo_ref(NULL, &nvbo);
+               return ERR_PTR(-ENOMEM);
        }

        nvbo->gem->driver_private = nvbo;
-       return 0;
+       return nvbo->gem;
 }

-struct dma_buf *nouveau_gem_prime_export(struct drm_device *dev,
-                               struct drm_gem_object *obj, int flags)
+int nouveau_gem_prime_pin(struct drm_gem_object *obj)
 {
        struct nouveau_bo *nvbo = nouveau_gem_object(obj);
        int ret = 0;
@@ -175,52 +89,7 @@ struct dma_buf *nouveau_gem_prime_export(struct drm_device 
*dev,
        /* pin buffer into GTT */
        ret = nouveau_bo_pin(nvbo, TTM_PL_FLAG_TT);
        if (ret)
-               return ERR_PTR(-EINVAL);
-
-       return dma_buf_export(nvbo, &nouveau_dmabuf_ops, obj->size, flags);
-}
-
-struct drm_gem_object *nouveau_gem_prime_import(struct drm_device *dev,
-                               struct dma_buf *dma_buf)
-{
-       struct dma_buf_attachment *attach;
-       struct sg_table *sg;
-       struct nouveau_bo *nvbo;
-       int ret;
-
-       if (dma_buf->ops == &nouveau_dmabuf_ops) {
-               nvbo = dma_buf->priv;
-               if (nvbo->gem) {
-                       if (nvbo->gem->dev == dev) {
-                               drm_gem_object_reference(nvbo->gem);
-                               dma_buf_put(dma_buf);
-                               return nvbo->gem;
-                       }
-               }
-       }
-       /* need to attach */
-       attach = dma_buf_attach(dma_buf, dev->dev);
-       if (IS_ERR(attach))
-               return ERR_PTR(PTR_ERR(attach));
-
-       sg = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
-       if (IS_ERR(sg)) {
-               ret = PTR_ERR(sg);
-               goto fail_detach;
-       }
-
-       ret = nouveau_prime_new(dev, dma_buf->size, sg, &nvbo);
-       if (ret)
-               goto fail_unmap;
-
-       nvbo->gem->import_attach = attach;
-
-       return nvbo->gem;
+               return -EINVAL;

-fail_unmap:
-       dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
-fail_detach:
-       dma_buf_detach(dma_buf, attach);
-       return ERR_PTR(ret);
+       return 0;
 }
-
-- 
1.8.1.1

Reply via email to