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 22static inline struct tegra_bo *host1x_to_tegra_bo(struct host1x_bo *bo) 23{ 24 return container_of(bo, struct tegra_bo, base); 25} 26 27static 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 37static 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 44static void tegra_bo_unpin(struct host1x_bo *bo, struct sg_table *sgt) 45{ 46} 47 48static 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 55static void tegra_bo_munmap(struct host1x_bo *bo, void *addr) 56{ 57} 58 59static 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 66static void tegra_bo_kunmap(struct host1x_bo *bo, unsigned int page, 67 void *addr) 68{ 69} 70 71static 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 83static 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 94static 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 99struct 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 137err_mmap: 138 drm_gem_object_release(&bo->gem); 139err_init: 140 tegra_bo_destroy(drm, bo); 141err_dma: 142 kfree(bo); 143 144 return ERR_PTR(err); 145} 146 147struct 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 168err: 169 tegra_bo_free_object(&bo->gem); 170 return ERR_PTR(ret); 171} 172 173static 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 225detach: 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); 231free_mmap: 232 drm_gem_free_mmap_offset(&bo->gem); 233release: 234 drm_gem_object_release(&bo->gem); 235free: 236 kfree(bo); 237 238 return ERR_PTR(err); 239} 240 241void 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 259int 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 281int 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 307const struct vm_operations_struct tegra_bo_vm_ops = { 308 .open = drm_gem_vm_open, 309 .close = drm_gem_vm_close, 310}; 311 312int 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 333static struct sg_table * 334tegra_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 356static 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 364static void tegra_gem_prime_release(struct dma_buf *buf) 365{ 366 drm_gem_dmabuf_release(buf); 367} 368 369static void *tegra_gem_prime_kmap_atomic(struct dma_buf *buf, 370 unsigned long page) 371{ 372 return NULL; 373} 374 375static void tegra_gem_prime_kunmap_atomic(struct dma_buf *buf, 376 unsigned long page, 377 void *addr) 378{ 379} 380 381static void *tegra_gem_prime_kmap(struct dma_buf *buf, unsigned long page) 382{ 383 return NULL; 384} 385 386static void tegra_gem_prime_kunmap(struct dma_buf *buf, unsigned long page, 387 void *addr) 388{ 389} 390 391static int tegra_gem_prime_mmap(struct dma_buf *buf, struct vm_area_struct *vma) 392{ 393 return -EINVAL; 394} 395 396static 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 404static void tegra_gem_prime_vunmap(struct dma_buf *buf, void *vaddr) 405{ 406} 407 408static 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 421struct 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 429struct 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} 449