[PATCH v3 1/3] drm: add prime helpers

Aaron Plattner aplattner at nvidia.com
Fri Apr 12 08:13:09 PDT 2013


On 04/12/13 07:58, Daniel Vetter wrote:
> On Tue, Jan 15, 2013 at 12:47:42PM -0800, Aaron Plattner wrote:
>> Instead of reimplementing all of the dma_buf functionality in every driver,
>> create helpers drm_prime_import and drm_prime_export that implement them in
>> terms of new, lower-level hook functions:
>>
>>    gem_prime_pin: callback when a buffer is created, used to pin buffers into GTT
>>    gem_prime_get_sg_table: convert a drm_gem_object to an sg_table for export
>>    gem_prime_import_sg_table: convert an sg_table into a drm_gem_object
>>    gem_prime_vmap, gem_prime_vunmap: map and unmap an object
>>
>> These hooks are optional; drivers can opt in by using drm_gem_prime_import and
>> drm_gem_prime_export as the .gem_prime_import and .gem_prime_export fields of
>> struct drm_driver.
>>
>> v2:
>> - Drop .begin_cpu_access.  None of the drivers this code replaces implemented
>>    it.  Having it here was a leftover from when I was trying to include i915 in
>>    this rework.
>> - Use mutex_lock instead of mutex_lock_interruptible, as these three drivers
>>    did.  This patch series shouldn't change that behavior.
>> - Rename helpers to gem_prime_get_sg_table and gem_prime_import_sg_table.
>>    Rename struct sg_table* variables to 'sgt' for clarity.
>> - Update drm.tmpl for these new hooks.
>>
>> v3:
>> - Pass the vaddr down to the driver.  This lets drivers that just call vunmap on
>>    the pointer avoid having to store the pointer in their GEM private structures.
>> - Move documentation into a /** DOC */ comment in drm_prime.c and include it in
>>    drm.tmpl with a !P line.  I tried to use !F lines to include documentation of
>>    the individual functions from drmP.h, but the docproc / kernel-doc scripts
>>    barf on that file, so hopefully this is good enough for now.
>> - apply refcount fix from commit be8a42ae60addd8b6092535c11b42d099d6470ec
>>    ("drm/prime: drop reference on imported dma-buf come from gem")
>>
>> Signed-off-by: Aaron Plattner <aplattner at nvidia.com>
>> Cc: Daniel Vetter <daniel.vetter at ffwll.ch>
>> Cc: David Airlie <airlied at linux.ie>
>> ---
>>   Documentation/DocBook/drm.tmpl |   4 +
>>   drivers/gpu/drm/drm_prime.c    | 186 ++++++++++++++++++++++++++++++++++++++++-
>>   include/drm/drmP.h             |  12 +++
>>   3 files changed, 201 insertions(+), 1 deletion(-)
>>
>> diff --git a/Documentation/DocBook/drm.tmpl b/Documentation/DocBook/drm.tmpl
>> index 4ee2304..ed40576 100644
>> --- a/Documentation/DocBook/drm.tmpl
>> +++ b/Documentation/DocBook/drm.tmpl
>> @@ -743,6 +743,10 @@ char *date;</synopsis>
>>             These two operations are mandatory for GEM drivers that support DRM
>>             PRIME.
>>           </para>
>> +        <sect4>
>> +          <title>DRM PRIME Helper Functions Reference</title>
>> +!Pdrivers/gpu/drm/drm_prime.c PRIME Helpers
>> +        </sect4>
>>         </sect3>
>>         <sect3 id="drm-gem-objects-mapping">
>>           <title>GEM Objects Mapping</title>
>> diff --git a/drivers/gpu/drm/drm_prime.c b/drivers/gpu/drm/drm_prime.c
>> index 7f12573..366910d 100644
>> --- a/drivers/gpu/drm/drm_prime.c
>> +++ b/drivers/gpu/drm/drm_prime.c
>> @@ -53,7 +53,8 @@
>>    * Self-importing: if userspace is using PRIME as a replacement for flink
>>    * then it will get a fd->handle request for a GEM object that it created.
>>    * Drivers should detect this situation and return back the gem object
>> - * from the dma-buf private.
>> + * from the dma-buf private.  Prime will do this automatically for drivers that
>> + * use the drm_gem_prime_{import,export} helpers.
>>    */
>>
>>   struct drm_prime_member {
>> @@ -62,6 +63,137 @@ struct drm_prime_member {
>>        uint32_t handle;
>>   };
>>
>> +static struct sg_table *drm_gem_map_dma_buf(struct dma_buf_attachment *attach,
>> +             enum dma_data_direction dir)
>> +{
>> +     struct drm_gem_object *obj = attach->dmabuf->priv;
>> +     struct sg_table *sgt;
>> +
>> +     mutex_lock(&obj->dev->struct_mutex);
>> +
>> +     sgt = obj->dev->driver->gem_prime_get_sg_table(obj);
>> +
>> +     if (!IS_ERR_OR_NULL(sgt))
>> +             dma_map_sg(attach->dev, sgt->sgl, sgt->nents, dir);
>> +
>> +     mutex_unlock(&obj->dev->struct_mutex);
>> +     return sgt;
>> +}
>> +
>> +static void drm_gem_unmap_dma_buf(struct dma_buf_attachment *attach,
>> +             struct sg_table *sgt, enum dma_data_direction dir)
>> +{
>> +     dma_unmap_sg(attach->dev, sgt->sgl, sgt->nents, dir);
>> +     sg_free_table(sgt);
>> +     kfree(sgt);
>> +}
>> +
>> +static void drm_gem_dmabuf_release(struct dma_buf *dma_buf)
>> +{
>> +     struct drm_gem_object *obj = dma_buf->priv;
>> +
>> +     if (obj->export_dma_buf == dma_buf) {
>> +             /* drop the reference on the export fd holds */
>> +             obj->export_dma_buf = NULL;
>> +             drm_gem_object_unreference_unlocked(obj);
>> +     }
>> +}
>> +
>> +static void *drm_gem_dmabuf_vmap(struct dma_buf *dma_buf)
>> +{
>> +     struct drm_gem_object *obj = dma_buf->priv;
>> +     struct drm_device *dev = obj->dev;
>> +
>> +     return dev->driver->gem_prime_vmap(obj);
>> +}
>> +
>> +static void drm_gem_dmabuf_vunmap(struct dma_buf *dma_buf, void *vaddr)
>> +{
>> +     struct drm_gem_object *obj = dma_buf->priv;
>> +     struct drm_device *dev = obj->dev;
>> +
>> +     dev->driver->gem_prime_vunmap(obj, vaddr);
>> +}
>> +
>> +static void *drm_gem_dmabuf_kmap_atomic(struct dma_buf *dma_buf,
>> +             unsigned long page_num)
>> +{
>> +     return NULL;
>> +}
>> +
>> +static void drm_gem_dmabuf_kunmap_atomic(struct dma_buf *dma_buf,
>> +             unsigned long page_num, void *addr)
>> +{
>> +
>> +}
>> +static void *drm_gem_dmabuf_kmap(struct dma_buf *dma_buf,
>> +             unsigned long page_num)
>> +{
>> +     return NULL;
>> +}
>> +
>> +static void drm_gem_dmabuf_kunmap(struct dma_buf *dma_buf,
>> +             unsigned long page_num, void *addr)
>> +{
>> +
>> +}
>> +
>> +static int drm_gem_dmabuf_mmap(struct dma_buf *dma_buf,
>> +             struct vm_area_struct *vma)
>> +{
>> +     return -EINVAL;
>> +}
>> +
>> +static const struct dma_buf_ops drm_gem_prime_dmabuf_ops =  {
>> +     .map_dma_buf = drm_gem_map_dma_buf,
>> +     .unmap_dma_buf = drm_gem_unmap_dma_buf,
>> +     .release = drm_gem_dmabuf_release,
>> +     .kmap = drm_gem_dmabuf_kmap,
>> +     .kmap_atomic = drm_gem_dmabuf_kmap_atomic,
>> +     .kunmap = drm_gem_dmabuf_kunmap,
>> +     .kunmap_atomic = drm_gem_dmabuf_kunmap_atomic,
>> +     .mmap = drm_gem_dmabuf_mmap,
>> +     .vmap = drm_gem_dmabuf_vmap,
>> +     .vunmap = drm_gem_dmabuf_vunmap,
>> +};
>> +
>> +/**
>> + * DOC: PRIME Helpers
>> + *
>> + * Drivers can implement @gem_prime_export and @gem_prime_import in terms of
>> + * simpler APIs by using the helper functions @drm_gem_prime_export and
>> + * @drm_gem_prime_import.  These functions implement dma-buf support in terms of
>> + * five lower-level driver callbacks:
>> + *
>> + * Export callbacks:
>> + *
>> + *  - @gem_prime_pin (optional): prepare a GEM object for exporting
>> + *
>> + *  - @gem_prime_get_sg_table: provide a scatter/gather table of pinned pages
>> + *
>> + *  - @gem_prime_vmap: vmap a buffer exported by your driver
>> + *
>> + *  - @gem_prime_vunmap: vunmap a buffer exported by your driver
>> + *
>> + * Import callback:
>> + *
>> + *  - @gem_prime_import_sg_table (import): produce a GEM object from another
>> + *    driver's scatter/gather table
>> + */
>> +
>> +struct dma_buf *drm_gem_prime_export(struct drm_device *dev,
>> +                                  struct drm_gem_object *obj, int flags)
>> +{
>> +     if (dev->driver->gem_prime_pin) {
>> +             int ret = dev->driver->gem_prime_pin(obj);
>> +             if (ret)
>> +                     return ERR_PTR(ret);
>> +     }
>> +     return dma_buf_export(obj, &drm_gem_prime_dmabuf_ops, obj->size,
>> +                           0600);
>> +}
>> +EXPORT_SYMBOL(drm_gem_prime_export);
>> +
>>   int drm_gem_prime_handle_to_fd(struct drm_device *dev,
>>                struct drm_file *file_priv, uint32_t handle, uint32_t flags,
>>                int *prime_fd)
>> @@ -117,6 +249,58 @@ int drm_gem_prime_handle_to_fd(struct drm_device *dev,
>>   }
>>   EXPORT_SYMBOL(drm_gem_prime_handle_to_fd);
>>
>> +struct drm_gem_object *drm_gem_prime_import(struct drm_device *dev,
>> +                                         struct dma_buf *dma_buf)
>> +{
>> +     struct dma_buf_attachment *attach;
>> +     struct sg_table *sgt;
>> +     struct drm_gem_object *obj;
>> +     int ret;
>> +
>> +     if (!dev->driver->gem_prime_import_sg_table)
>> +             return ERR_PTR(-EINVAL);
>> +
>> +     if (dma_buf->ops == &drm_gem_prime_dmabuf_ops) {
>
> This here breaks self-import checks since it smashes all buffers from all
> drivers using these helpers into one set. Which means e.g. nouveau will
> happily import a buffer from radoen as it's own. The only reason afaics
> that shit didn't completely hit the fan is that i915 still has it's own
> buffer ops, and everyone seems to only care about sharing with i915.

Doesn't the (obj->dev == dev) check below guard against that?

-- Aaron

> So after a few months of subconscious pondering (well, just re-read the
> code to review Dave's patch) I now finally know why I totally didn't like
> that we move the vtable into the helpers ;-)
>
> I think someone needs to fix this or we need to revert this patch here.
>
> Cheers, Daniel
>
>> +             obj = dma_buf->priv;
>> +             if (obj->dev == dev) {
>> +                     /*
>> +                      * Importing dmabuf exported from out own gem increases
>> +                      * refcount on gem itself instead of f_count of dmabuf.
>> +                      */
>> +                     drm_gem_object_reference(obj);
>> +                     dma_buf_put(dma_buf);
>> +                     return obj;
>> +             }
>> +     }
>> +
>> +     attach = dma_buf_attach(dma_buf, dev->dev);
>> +     if (IS_ERR(attach))
>> +             return ERR_PTR(PTR_ERR(attach));
>> +
>> +     sgt = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
>> +     if (IS_ERR_OR_NULL(sgt)) {
>> +             ret = PTR_ERR(sgt);
>> +             goto fail_detach;
>> +     }
>> +
>> +     obj = dev->driver->gem_prime_import_sg_table(dev, dma_buf->size, sgt);
>> +     if (IS_ERR(obj)) {
>> +             ret = PTR_ERR(obj);
>> +             goto fail_unmap;
>> +     }
>> +
>> +     obj->import_attach = attach;
>> +
>> +     return obj;
>> +
>> +fail_unmap:
>> +     dma_buf_unmap_attachment(attach, sgt, DMA_BIDIRECTIONAL);
>> +fail_detach:
>> +     dma_buf_detach(dma_buf, attach);
>> +     return ERR_PTR(ret);
>> +}
>> +EXPORT_SYMBOL(drm_gem_prime_import);
>> +
>>   int drm_gem_prime_fd_to_handle(struct drm_device *dev,
>>                struct drm_file *file_priv, int prime_fd, uint32_t *handle)
>>   {
>> diff --git a/include/drm/drmP.h b/include/drm/drmP.h
>> index fad21c9..2d4ca6f 100644
>> --- a/include/drm/drmP.h
>> +++ b/include/drm/drmP.h
>> @@ -919,6 +919,14 @@ struct drm_driver {
>>        /* import dmabuf -> GEM */
>>        struct drm_gem_object * (*gem_prime_import)(struct drm_device *dev,
>>                                struct dma_buf *dma_buf);
>> +     /* low-level interface used by drm_gem_prime_{import,export} */
>> +     int (*gem_prime_pin)(struct drm_gem_object *obj);
>> +     struct sg_table *(*gem_prime_get_sg_table)(struct drm_gem_object *obj);
>> +     struct drm_gem_object *(*gem_prime_import_sg_table)(
>> +                             struct drm_device *dev, size_t size,
>> +                             struct sg_table *sgt);
>> +     void *(*gem_prime_vmap)(struct drm_gem_object *obj);
>> +     void (*gem_prime_vunmap)(struct drm_gem_object *obj, void *vaddr);
>>
>>        /* vga arb irq handler */
>>        void (*vgaarb_irq)(struct drm_device *dev, bool state);
>> @@ -1540,9 +1548,13 @@ extern int drm_clients_info(struct seq_file *m, void* data);
>>   extern int drm_gem_name_info(struct seq_file *m, void *data);
>>
>>
>> +extern struct dma_buf *drm_gem_prime_export(struct drm_device *dev,
>> +             struct drm_gem_object *obj, int flags);
>>   extern int drm_gem_prime_handle_to_fd(struct drm_device *dev,
>>                struct drm_file *file_priv, uint32_t handle, uint32_t flags,
>>                int *prime_fd);
>> +extern struct drm_gem_object *drm_gem_prime_import(struct drm_device *dev,
>> +             struct dma_buf *dma_buf);
>>   extern int drm_gem_prime_fd_to_handle(struct drm_device *dev,
>>                struct drm_file *file_priv, int prime_fd, uint32_t *handle);
>>
>> --
>> 1.8.1.1
>>
>
> --
> Daniel Vetter
> Software Engineer, Intel Corporation
> +41 (0) 79 365 57 48 - http://blog.ffwll.ch
>



More information about the dri-devel mailing list