drivers/gpu/drm/panfrost/panfrost_gem.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright 2019 Linaro, Ltd, Rob Herring <robh@kernel.org> */
   3
   4 #include <linux/err.h>
   5 #include <linux/slab.h>
   6 #include <linux/dma-buf.h>
   7 #include <linux/dma-mapping.h>
   8
   9 #include <drm/panfrost_drm.h>
  10 #include "panfrost_device.h"
  11 #include "panfrost_gem.h"
  12 #include "panfrost_mmu.h"
  13
  14 /* Called DRM core on the last userspace/kernel unreference of the
  15  * BO.
  16  */
  17 static void panfrost_gem_free_object(struct drm_gem_object *obj)
  18 {
  19         struct panfrost_gem_object *bo = to_panfrost_bo(obj);
  20         struct panfrost_device *pfdev = obj->dev->dev_private;
  21
  22         /*
  23          * Make sure the BO is no longer inserted in the shrinker list before
  24          * taking care of the destruction itself. If we don't do that we have a
  25          * race condition between this function and what's done in
  26          * panfrost_gem_shrinker_scan().
  27          */
  28         mutex_lock(&pfdev->shrinker_lock);
  29         list_del_init(&bo->base.madv_list);
  30         mutex_unlock(&pfdev->shrinker_lock);
  31
  32         /*
  33          * If we still have mappings attached to the BO, there's a problem in
  34          * our refcounting.
  35          */
  36         WARN_ON_ONCE(!list_empty(&bo->mappings.list));
  37
  38         if (bo->sgts) {
  39                 int i;
  40                 int n_sgt = bo->base.base.size / SZ_2M;
  41
  42                 for (i = 0; i < n_sgt; i++) {
  43                         if (bo->sgts[i].sgl) {
  44                                 dma_unmap_sgtable(pfdev->dev, &bo->sgts[i],
  45                                                   DMA_BIDIRECTIONAL, 0);
  46                                 sg_free_table(&bo->sgts[i]);
  47                         }
  48                 }
  49                 kvfree(bo->sgts);
  50         }
  51
  52         drm_gem_shmem_free_object(obj);
  53 }
  54
  55 struct panfrost_gem_mapping *
  56 panfrost_gem_mapping_get(struct panfrost_gem_object *bo,
  57                          struct panfrost_file_priv *priv)
  58 {
  59         struct panfrost_gem_mapping *iter, *mapping = NULL;
  60
  61         mutex_lock(&bo->mappings.lock);
  62         list_for_each_entry(iter, &bo->mappings.list, node) {
  63                 if (iter->mmu == &priv->mmu) {
  64                         kref_get(&iter->refcount);
  65                         mapping = iter;
  66                         break;
  67                 }
  68         }
  69         mutex_unlock(&bo->mappings.lock);
  70
  71         return mapping;
  72 }
  73
  74 static void
  75 panfrost_gem_teardown_mapping(struct panfrost_gem_mapping *mapping)
  76 {
  77         struct panfrost_file_priv *priv;
  78
  79         if (mapping->active)
  80                 panfrost_mmu_unmap(mapping);
  81
  82         priv = container_of(mapping->mmu, struct panfrost_file_priv, mmu);
  83         spin_lock(&priv->mm_lock);
  84         if (drm_mm_node_allocated(&mapping->mmnode))
  85                 drm_mm_remove_node(&mapping->mmnode);
  86         spin_unlock(&priv->mm_lock);
  87 }
  88
  89 static void panfrost_gem_mapping_release(struct kref *kref)
  90 {
  91         struct panfrost_gem_mapping *mapping;
  92
  93         mapping = container_of(kref, struct panfrost_gem_mapping, refcount);
  94
  95         panfrost_gem_teardown_mapping(mapping);
  96         drm_gem_object_put(&mapping->obj->base.base);
  97         kfree(mapping);
  98 }
  99
 100 void panfrost_gem_mapping_put(struct panfrost_gem_mapping *mapping)
 101 {
 102         if (!mapping)
 103                 return;
 104
 105         kref_put(&mapping->refcount, panfrost_gem_mapping_release);
 106 }
 107
 108 void panfrost_gem_teardown_mappings_locked(struct panfrost_gem_object *bo)
 109 {
 110         struct panfrost_gem_mapping *mapping;
 111
 112         list_for_each_entry(mapping, &bo->mappings.list, node)
 113                 panfrost_gem_teardown_mapping(mapping);
 114 }
 115
 116 int panfrost_gem_open(struct drm_gem_object *obj, struct drm_file *file_priv)
 117 {
 118         int ret;
 119         size_t size = obj->size;
 120         u64 align;
 121         struct panfrost_gem_object *bo = to_panfrost_bo(obj);
 122         unsigned long color = bo->noexec ? PANFROST_BO_NOEXEC : 0;
 123         struct panfrost_file_priv *priv = file_priv->driver_priv;
 124         struct panfrost_gem_mapping *mapping;
 125
 126         mapping = kzalloc(sizeof(*mapping), GFP_KERNEL);
 127         if (!mapping)
 128                 return -ENOMEM;
 129
 130         INIT_LIST_HEAD(&mapping->node);
 131         kref_init(&mapping->refcount);
 132         drm_gem_object_get(obj);
 133         mapping->obj = bo;
 134
 135         /*
 136          * Executable buffers cannot cross a 16MB boundary as the program
 137          * counter is 24-bits. We assume executable buffers will be less than
 138          * 16MB and aligning executable buffers to their size will avoid
 139          * crossing a 16MB boundary.
 140          */
 141         if (!bo->noexec)
 142                 align = size >> PAGE_SHIFT;
 143         else
 144                 align = size >= SZ_2M ? SZ_2M >> PAGE_SHIFT : 0;
 145
 146         mapping->mmu = &priv->mmu;
 147         spin_lock(&priv->mm_lock);
 148         ret = drm_mm_insert_node_generic(&priv->mm, &mapping->mmnode,
 149                                          size >> PAGE_SHIFT, align, color, 0);
 150         spin_unlock(&priv->mm_lock);
 151         if (ret)
 152                 goto err;
 153
 154         if (!bo->is_heap) {
 155                 ret = panfrost_mmu_map(mapping);
 156                 if (ret)
 157                         goto err;
 158         }
 159
 160         mutex_lock(&bo->mappings.lock);
 161         WARN_ON(bo->base.madv != PANFROST_MADV_WILLNEED);
 162         list_add_tail(&mapping->node, &bo->mappings.list);
 163         mutex_unlock(&bo->mappings.lock);
 164
 165 err:
 166         if (ret)
 167                 panfrost_gem_mapping_put(mapping);
 168         return ret;
 169 }
 170
 171 void panfrost_gem_close(struct drm_gem_object *obj, struct drm_file *file_priv)
 172 {
 173         struct panfrost_file_priv *priv = file_priv->driver_priv;
 174         struct panfrost_gem_object *bo = to_panfrost_bo(obj);
 175         struct panfrost_gem_mapping *mapping = NULL, *iter;
 176
 177         mutex_lock(&bo->mappings.lock);
 178         list_for_each_entry(iter, &bo->mappings.list, node) {
 179                 if (iter->mmu == &priv->mmu) {
 180                         mapping = iter;
 181                         list_del(&iter->node);
 182                         break;
 183                 }
 184         }
 185         mutex_unlock(&bo->mappings.lock);
 186
 187         panfrost_gem_mapping_put(mapping);
 188 }
 189
 190 static int panfrost_gem_pin(struct drm_gem_object *obj)
 191 {
 192         if (to_panfrost_bo(obj)->is_heap)
 193                 return -EINVAL;
 194
 195         return drm_gem_shmem_pin(obj);
 196 }
 197
 198 static const struct drm_gem_object_funcs panfrost_gem_funcs = {
 199         .free = panfrost_gem_free_object,
 200         .open = panfrost_gem_open,
 201         .close = panfrost_gem_close,
 202         .print_info = drm_gem_shmem_print_info,
 203         .pin = panfrost_gem_pin,
 204         .unpin = drm_gem_shmem_unpin,
 205         .get_sg_table = drm_gem_shmem_get_sg_table,
 206         .vmap = drm_gem_shmem_vmap,
 207         .vunmap = drm_gem_shmem_vunmap,
 208         .mmap = drm_gem_shmem_mmap,
 209 };
 210
 211 /**
 212  * panfrost_gem_create_object - Implementation of driver->gem_create_object.
 213  * @dev: DRM device
 214  * @size: Size in bytes of the memory the object will reference
 215  *
 216  * This lets the GEM helpers allocate object structs for us, and keep
 217  * our BO stats correct.
 218  */
 219 struct drm_gem_object *panfrost_gem_create_object(struct drm_device *dev, size_t size)
 220 {
 221         struct panfrost_device *pfdev = dev->dev_private;
 222         struct panfrost_gem_object *obj;
 223
 224         obj = kzalloc(sizeof(*obj), GFP_KERNEL);
 225         if (!obj)
 226                 return NULL;
 227
 228         INIT_LIST_HEAD(&obj->mappings.list);
 229         mutex_init(&obj->mappings.lock);
 230         obj->base.base.funcs = &panfrost_gem_funcs;
 231         obj->base.map_cached = pfdev->coherent;
 232
 233         return &obj->base.base;
 234 }
 235
 236 struct panfrost_gem_object *
 237 panfrost_gem_create_with_handle(struct drm_file *file_priv,
 238                                 struct drm_device *dev, size_t size,
 239                                 u32 flags,
 240                                 uint32_t *handle)
 241 {
 242         int ret;
 243         struct drm_gem_shmem_object *shmem;
 244         struct panfrost_gem_object *bo;
 245
 246         /* Round up heap allocations to 2MB to keep fault handling simple */
 247         if (flags & PANFROST_BO_HEAP)
 248                 size = roundup(size, SZ_2M);
 249
 250         shmem = drm_gem_shmem_create(dev, size);
 251         if (IS_ERR(shmem))
 252                 return ERR_CAST(shmem);
 253
 254         bo = to_panfrost_bo(&shmem->base);
 255         bo->noexec = !!(flags & PANFROST_BO_NOEXEC);
 256         bo->is_heap = !!(flags & PANFROST_BO_HEAP);
 257
 258         /*
 259          * Allocate an id of idr table where the obj is registered
 260          * and handle has the id what user can see.
 261          */
 262         ret = drm_gem_handle_create(file_priv, &shmem->base, handle);
 263         /* drop reference from allocate - handle holds it now. */
 264         drm_gem_object_put(&shmem->base);
 265         if (ret)
 266                 return ERR_PTR(ret);
 267
 268         return bo;
 269 }
 270
 271 struct drm_gem_object *
 272 panfrost_gem_prime_import_sg_table(struct drm_device *dev,
 273                                    struct dma_buf_attachment *attach,
 274                                    struct sg_table *sgt)
 275 {
 276         struct drm_gem_object *obj;
 277         struct panfrost_gem_object *bo;
 278
 279         obj = drm_gem_shmem_prime_import_sg_table(dev, attach, sgt);
 280         if (IS_ERR(obj))
 281                 return ERR_CAST(obj);
 282
 283         bo = to_panfrost_bo(obj);
 284         bo->noexec = true;
 285
 286         return obj;
 287 }