drivers/gpu/drm/tegra/gem.c

   1 /*
   2  * NVIDIA Tegra DRM GEM helper functions
   3  *
   4  * Copyright (C) 2012 Sascha Hauer, Pengutronix
   5  * Copyright (C) 2013 NVIDIA CORPORATION, All rights reserved.
   6  *
   7  * Based on the GEM/CMA helpers
   8  *
   9  * Copyright (c) 2011 Samsung Electronics Co., Ltd.
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License version 2 as
  13  * published by the Free Software Foundation.
  14  */
  15
  16 #include <linux/dma-buf.h>
  17 #include <drm/tegra_drm.h>
  18
  19 #include "drm.h"
  20 #include "gem.h"
  21
  22 static inline struct tegra_bo *host1x_to_tegra_bo(struct host1x_bo *bo)
  23 {
  24         return container_of(bo, struct tegra_bo, base);
  25 }
  26
  27 static void tegra_bo_put(struct host1x_bo *bo)
  28 {
  29         struct tegra_bo *obj = host1x_to_tegra_bo(bo);
  30         struct drm_device *drm = obj->gem.dev;
  31
  32         mutex_lock(&drm->struct_mutex);
  33         drm_gem_object_unreference(&obj->gem);
  34         mutex_unlock(&drm->struct_mutex);
  35 }
  36
  37 static dma_addr_t tegra_bo_pin(struct host1x_bo *bo, struct sg_table **sgt)
  38 {
  39         struct tegra_bo *obj = host1x_to_tegra_bo(bo);
  40
  41         return obj->paddr;
  42 }
  43
  44 static void tegra_bo_unpin(struct host1x_bo *bo, struct sg_table *sgt)
  45 {
  46 }
  47
  48 static void *tegra_bo_mmap(struct host1x_bo *bo)
  49 {
  50         struct tegra_bo *obj = host1x_to_tegra_bo(bo);
  51
  52         return obj->vaddr;
  53 }
  54
  55 static void tegra_bo_munmap(struct host1x_bo *bo, void *addr)
  56 {
  57 }
  58
  59 static void *tegra_bo_kmap(struct host1x_bo *bo, unsigned int page)
  60 {
  61         struct tegra_bo *obj = host1x_to_tegra_bo(bo);
  62
  63         return obj->vaddr + page * PAGE_SIZE;
  64 }
  65
  66 static void tegra_bo_kunmap(struct host1x_bo *bo, unsigned int page,
  67                             void *addr)
  68 {
  69 }
  70
  71 static struct host1x_bo *tegra_bo_get(struct host1x_bo *bo)
  72 {
  73         struct tegra_bo *obj = host1x_to_tegra_bo(bo);
  74         struct drm_device *drm = obj->gem.dev;
  75
  76         mutex_lock(&drm->struct_mutex);
  77         drm_gem_object_reference(&obj->gem);
  78         mutex_unlock(&drm->struct_mutex);
  79
  80         return bo;
  81 }
  82
  83 static const struct host1x_bo_ops tegra_bo_ops = {
  84         .get = tegra_bo_get,
  85         .put = tegra_bo_put,
  86         .pin = tegra_bo_pin,
  87         .unpin = tegra_bo_unpin,
  88         .mmap = tegra_bo_mmap,
  89         .munmap = tegra_bo_munmap,
  90         .kmap = tegra_bo_kmap,
  91         .kunmap = tegra_bo_kunmap,
  92 };
  93
  94 static void tegra_bo_destroy(struct drm_device *drm, struct tegra_bo *bo)
  95 {
  96         dma_free_writecombine(drm->dev, bo->gem.size, bo->vaddr, bo->paddr);
  97 }
  98
  99 struct tegra_bo *tegra_bo_create(struct drm_device *drm, unsigned int size,
 100                                  unsigned long flags)
 101 {
 102         struct tegra_bo *bo;
 103         int err;
 104
 105         bo = kzalloc(sizeof(*bo), GFP_KERNEL);
 106         if (!bo)
 107                 return ERR_PTR(-ENOMEM);
 108
 109         host1x_bo_init(&bo->base, &tegra_bo_ops);
 110         size = round_up(size, PAGE_SIZE);
 111
 112         bo->vaddr = dma_alloc_writecombine(drm->dev, size, &bo->paddr,
 113                                            GFP_KERNEL | __GFP_NOWARN);
 114         if (!bo->vaddr) {
 115                 dev_err(drm->dev, "failed to allocate buffer with size %u\n",
 116                         size);
 117                 err = -ENOMEM;
 118                 goto err_dma;
 119         }
 120
 121         err = drm_gem_object_init(drm, &bo->gem, size);
 122         if (err)
 123                 goto err_init;
 124
 125         err = drm_gem_create_mmap_offset(&bo->gem);
 126         if (err)
 127                 goto err_mmap;
 128
 129         if (flags & DRM_TEGRA_GEM_CREATE_TILED)
 130                 bo->tiling.mode = TEGRA_BO_TILING_MODE_TILED;
 131
 132         if (flags & DRM_TEGRA_GEM_CREATE_BOTTOM_UP)
 133                 bo->flags |= TEGRA_BO_BOTTOM_UP;
 134
 135         return bo;
 136
 137 err_mmap:
 138         drm_gem_object_release(&bo->gem);
 139 err_init:
 140         tegra_bo_destroy(drm, bo);
 141 err_dma:
 142         kfree(bo);
 143
 144         return ERR_PTR(err);
 145 }
 146
 147 struct tegra_bo *tegra_bo_create_with_handle(struct drm_file *file,
 148                                              struct drm_device *drm,
 149                                              unsigned int size,
 150                                              unsigned long flags,
 151                                              unsigned int *handle)
 152 {
 153         struct tegra_bo *bo;
 154         int ret;
 155
 156         bo = tegra_bo_create(drm, size, flags);
 157         if (IS_ERR(bo))
 158                 return bo;
 159
 160         ret = drm_gem_handle_create(file, &bo->gem, handle);
 161         if (ret)
 162                 goto err;
 163
 164         drm_gem_object_unreference_unlocked(&bo->gem);
 165
 166         return bo;
 167
 168 err:
 169         tegra_bo_free_object(&bo->gem);
 170         return ERR_PTR(ret);
 171 }
 172
 173 static struct tegra_bo *tegra_bo_import(struct drm_device *drm,
 174                                         struct dma_buf *buf)
 175 {
 176         struct dma_buf_attachment *attach;
 177         struct tegra_bo *bo;
 178         ssize_t size;
 179         int err;
 180
 181         bo = kzalloc(sizeof(*bo), GFP_KERNEL);
 182         if (!bo)
 183                 return ERR_PTR(-ENOMEM);
 184
 185         host1x_bo_init(&bo->base, &tegra_bo_ops);
 186         size = round_up(buf->size, PAGE_SIZE);
 187
 188         err = drm_gem_object_init(drm, &bo->gem, size);
 189         if (err < 0)
 190                 goto free;
 191
 192         err = drm_gem_create_mmap_offset(&bo->gem);
 193         if (err < 0)
 194                 goto release;
 195
 196         attach = dma_buf_attach(buf, drm->dev);
 197         if (IS_ERR(attach)) {
 198                 err = PTR_ERR(attach);
 199                 goto free_mmap;
 200         }
 201
 202         get_dma_buf(buf);
 203
 204         bo->sgt = dma_buf_map_attachment(attach, DMA_TO_DEVICE);
 205         if (!bo->sgt) {
 206                 err = -ENOMEM;
 207                 goto detach;
 208         }
 209
 210         if (IS_ERR(bo->sgt)) {
 211                 err = PTR_ERR(bo->sgt);
 212                 goto detach;
 213         }
 214
 215         if (bo->sgt->nents > 1) {
 216                 err = -EINVAL;
 217                 goto detach;
 218         }
 219
 220         bo->paddr = sg_dma_address(bo->sgt->sgl);
 221         bo->gem.import_attach = attach;
 222
 223         return bo;
 224
 225 detach:
 226         if (!IS_ERR_OR_NULL(bo->sgt))
 227                 dma_buf_unmap_attachment(attach, bo->sgt, DMA_TO_DEVICE);
 228
 229         dma_buf_detach(buf, attach);
 230         dma_buf_put(buf);
 231 free_mmap:
 232         drm_gem_free_mmap_offset(&bo->gem);
 233 release:
 234         drm_gem_object_release(&bo->gem);
 235 free:
 236         kfree(bo);
 237
 238         return ERR_PTR(err);
 239 }
 240
 241 void tegra_bo_free_object(struct drm_gem_object *gem)
 242 {
 243         struct tegra_bo *bo = to_tegra_bo(gem);
 244
 245         if (gem->import_attach) {
 246                 dma_buf_unmap_attachment(gem->import_attach, bo->sgt,
 247                                          DMA_TO_DEVICE);
 248                 drm_prime_gem_destroy(gem, NULL);
 249         } else {
 250                 tegra_bo_destroy(gem->dev, bo);
 251         }
 252
 253         drm_gem_free_mmap_offset(gem);
 254         drm_gem_object_release(gem);
 255
 256         kfree(bo);
 257 }
 258
 259 int tegra_bo_dumb_create(struct drm_file *file, struct drm_device *drm,
 260                          struct drm_mode_create_dumb *args)
 261 {
 262         int min_pitch = DIV_ROUND_UP(args->width * args->bpp, 8);
 263         struct tegra_drm *tegra = drm->dev_private;
 264         struct tegra_bo *bo;
 265
 266         min_pitch = round_up(min_pitch, tegra->pitch_align);
 267         if (args->pitch < min_pitch)
 268                 args->pitch = min_pitch;
 269
 270         if (args->size < args->pitch * args->height)
 271                 args->size = args->pitch * args->height;
 272
 273         bo = tegra_bo_create_with_handle(file, drm, args->size, 0,
 274                                          &args->handle);
 275         if (IS_ERR(bo))
 276                 return PTR_ERR(bo);
 277
 278         return 0;
 279 }
 280
 281 int tegra_bo_dumb_map_offset(struct drm_file *file, struct drm_device *drm,
 282                              uint32_t handle, uint64_t *offset)
 283 {
 284         struct drm_gem_object *gem;
 285         struct tegra_bo *bo;
 286
 287         mutex_lock(&drm->struct_mutex);
 288
 289         gem = drm_gem_object_lookup(drm, file, handle);
 290         if (!gem) {
 291                 dev_err(drm->dev, "failed to lookup GEM object\n");
 292                 mutex_unlock(&drm->struct_mutex);
 293                 return -EINVAL;
 294         }
 295
 296         bo = to_tegra_bo(gem);
 297
 298         *offset = drm_vma_node_offset_addr(&bo->gem.vma_node);
 299
 300         drm_gem_object_unreference(gem);
 301
 302         mutex_unlock(&drm->struct_mutex);
 303
 304         return 0;
 305 }
 306
 307 const struct vm_operations_struct tegra_bo_vm_ops = {
 308         .open = drm_gem_vm_open,
 309         .close = drm_gem_vm_close,
 310 };
 311
 312 int tegra_drm_mmap(struct file *file, struct vm_area_struct *vma)
 313 {
 314         struct drm_gem_object *gem;
 315         struct tegra_bo *bo;
 316         int ret;
 317
 318         ret = drm_gem_mmap(file, vma);
 319         if (ret)
 320                 return ret;
 321
 322         gem = vma->vm_private_data;
 323         bo = to_tegra_bo(gem);
 324
 325         ret = remap_pfn_range(vma, vma->vm_start, bo->paddr >> PAGE_SHIFT,
 326                               vma->vm_end - vma->vm_start, vma->vm_page_prot);
 327         if (ret)
 328                 drm_gem_vm_close(vma);
 329
 330         return ret;
 331 }
 332
 333 static struct sg_table *
 334 tegra_gem_prime_map_dma_buf(struct dma_buf_attachment *attach,
 335                             enum dma_data_direction dir)
 336 {
 337         struct drm_gem_object *gem = attach->dmabuf->priv;
 338         struct tegra_bo *bo = to_tegra_bo(gem);
 339         struct sg_table *sgt;
 340
 341         sgt = kmalloc(sizeof(*sgt), GFP_KERNEL);
 342         if (!sgt)
 343                 return NULL;
 344
 345         if (sg_alloc_table(sgt, 1, GFP_KERNEL)) {
 346                 kfree(sgt);
 347                 return NULL;
 348         }
 349
 350         sg_dma_address(sgt->sgl) = bo->paddr;
 351         sg_dma_len(sgt->sgl) = gem->size;
 352
 353         return sgt;
 354 }
 355
 356 static void tegra_gem_prime_unmap_dma_buf(struct dma_buf_attachment *attach,
 357                                           struct sg_table *sgt,
 358                                           enum dma_data_direction dir)
 359 {
 360         sg_free_table(sgt);
 361         kfree(sgt);
 362 }
 363
 364 static void tegra_gem_prime_release(struct dma_buf *buf)
 365 {
 366         drm_gem_dmabuf_release(buf);
 367 }
 368
 369 static void *tegra_gem_prime_kmap_atomic(struct dma_buf *buf,
 370                                          unsigned long page)
 371 {
 372         return NULL;
 373 }
 374
 375 static void tegra_gem_prime_kunmap_atomic(struct dma_buf *buf,
 376                                           unsigned long page,
 377                                           void *addr)
 378 {
 379 }
 380
 381 static void *tegra_gem_prime_kmap(struct dma_buf *buf, unsigned long page)
 382 {
 383         return NULL;
 384 }
 385
 386 static void tegra_gem_prime_kunmap(struct dma_buf *buf, unsigned long page,
 387                                    void *addr)
 388 {
 389 }
 390
 391 static int tegra_gem_prime_mmap(struct dma_buf *buf, struct vm_area_struct *vma)
 392 {
 393         return -EINVAL;
 394 }
 395
 396 static void *tegra_gem_prime_vmap(struct dma_buf *buf)
 397 {
 398         struct drm_gem_object *gem = buf->priv;
 399         struct tegra_bo *bo = to_tegra_bo(gem);
 400
 401         return bo->vaddr;
 402 }
 403
 404 static void tegra_gem_prime_vunmap(struct dma_buf *buf, void *vaddr)
 405 {
 406 }
 407
 408 static const struct dma_buf_ops tegra_gem_prime_dmabuf_ops = {
 409         .map_dma_buf = tegra_gem_prime_map_dma_buf,
 410         .unmap_dma_buf = tegra_gem_prime_unmap_dma_buf,
 411         .release = tegra_gem_prime_release,
 412         .kmap_atomic = tegra_gem_prime_kmap_atomic,
 413         .kunmap_atomic = tegra_gem_prime_kunmap_atomic,
 414         .kmap = tegra_gem_prime_kmap,
 415         .kunmap = tegra_gem_prime_kunmap,
 416         .mmap = tegra_gem_prime_mmap,
 417         .vmap = tegra_gem_prime_vmap,
 418         .vunmap = tegra_gem_prime_vunmap,
 419 };
 420
 421 struct dma_buf *tegra_gem_prime_export(struct drm_device *drm,
 422                                        struct drm_gem_object *gem,
 423                                        int flags)
 424 {
 425         return dma_buf_export(gem, &tegra_gem_prime_dmabuf_ops, gem->size,
 426                               flags, NULL);
 427 }
 428
 429 struct drm_gem_object *tegra_gem_prime_import(struct drm_device *drm,
 430                                               struct dma_buf *buf)
 431 {
 432         struct tegra_bo *bo;
 433
 434         if (buf->ops == &tegra_gem_prime_dmabuf_ops) {
 435                 struct drm_gem_object *gem = buf->priv;
 436
 437                 if (gem->dev == drm) {
 438                         drm_gem_object_reference(gem);
 439                         return gem;
 440                 }
 441         }
 442
 443         bo = tegra_bo_import(drm, buf);
 444         if (IS_ERR(bo))
 445                 return ERR_CAST(bo);
 446
 447         return &bo->gem;
 448 }