1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * Copyright (C) 2015-2018 Etnaviv Project 4 */ 5 6 #include <drm/drm_prime.h> 7 #include <drm/drm_print.h> 8 #include <linux/dma-mapping.h> 9 #include <linux/pagemap.h> 10 #include <linux/shmem_fs.h> 11 #include <linux/spinlock.h> 12 #include <linux/vmalloc.h> 13 14 #include "etnaviv_drv.h" 15 #include "etnaviv_gem.h" 16 #include "etnaviv_gpu.h" 17 #include "etnaviv_mmu.h" 18 19 static struct lock_class_key etnaviv_shm_lock_class; 20 static struct lock_class_key etnaviv_userptr_lock_class; 21 22 static void etnaviv_gem_scatter_map(struct etnaviv_gem_object *etnaviv_obj) 23 { 24 struct drm_device *dev = etnaviv_obj->base.dev; 25 struct sg_table *sgt = etnaviv_obj->sgt; 26 27 /* 28 * For non-cached buffers, ensure the new pages are clean 29 * because display controller, GPU, etc. are not coherent. 30 */ 31 if (etnaviv_obj->flags & ETNA_BO_CACHE_MASK) 32 dma_map_sgtable(dev->dev, sgt, DMA_BIDIRECTIONAL, 0); 33 } 34 35 static void etnaviv_gem_scatterlist_unmap(struct etnaviv_gem_object *etnaviv_obj) 36 { 37 struct drm_device *dev = etnaviv_obj->base.dev; 38 struct sg_table *sgt = etnaviv_obj->sgt; 39 40 /* 41 * For non-cached buffers, ensure the new pages are clean 42 * because display controller, GPU, etc. are not coherent: 43 * 44 * WARNING: The DMA API does not support concurrent CPU 45 * and device access to the memory area. With BIDIRECTIONAL, 46 * we will clean the cache lines which overlap the region, 47 * and invalidate all cache lines (partially) contained in 48 * the region. 49 * 50 * If you have dirty data in the overlapping cache lines, 51 * that will corrupt the GPU-written data. If you have 52 * written into the remainder of the region, this can 53 * discard those writes. 54 */ 55 if (etnaviv_obj->flags & ETNA_BO_CACHE_MASK) 56 dma_unmap_sgtable(dev->dev, sgt, DMA_BIDIRECTIONAL, 0); 57 } 58 59 /* called with etnaviv_obj->lock held */ 60 static int etnaviv_gem_shmem_get_pages(struct etnaviv_gem_object *etnaviv_obj) 61 { 62 struct drm_device *dev = etnaviv_obj->base.dev; 63 struct page **p = drm_gem_get_pages(&etnaviv_obj->base); 64 65 if (IS_ERR(p)) { 66 dev_dbg(dev->dev, "could not get pages: %ld\n", PTR_ERR(p)); 67 return PTR_ERR(p); 68 } 69 70 etnaviv_obj->pages = p; 71 72 return 0; 73 } 74 75 static void put_pages(struct etnaviv_gem_object *etnaviv_obj) 76 { 77 if (etnaviv_obj->sgt) { 78 etnaviv_gem_scatterlist_unmap(etnaviv_obj); 79 sg_free_table(etnaviv_obj->sgt); 80 kfree(etnaviv_obj->sgt); 81 etnaviv_obj->sgt = NULL; 82 } 83 if (etnaviv_obj->pages) { 84 drm_gem_put_pages(&etnaviv_obj->base, etnaviv_obj->pages, 85 true, false); 86 87 etnaviv_obj->pages = NULL; 88 } 89 } 90 91 struct page **etnaviv_gem_get_pages(struct etnaviv_gem_object *etnaviv_obj) 92 { 93 int ret; 94 95 lockdep_assert_held(&etnaviv_obj->lock); 96 97 if (!etnaviv_obj->pages) { 98 ret = etnaviv_obj->ops->get_pages(etnaviv_obj); 99 if (ret < 0) 100 return ERR_PTR(ret); 101 } 102 103 if (!etnaviv_obj->sgt) { 104 struct drm_device *dev = etnaviv_obj->base.dev; 105 unsigned int npages = etnaviv_obj->base.size >> PAGE_SHIFT; 106 struct sg_table *sgt; 107 108 sgt = drm_prime_pages_to_sg(dev, etnaviv_obj->pages, npages); 109 if (IS_ERR(sgt)) { 110 dev_err(dev->dev, "failed to allocate sgt: %ld\n", 111 PTR_ERR(sgt)); 112 return ERR_CAST(sgt); 113 } 114 115 etnaviv_obj->sgt = sgt; 116 117 etnaviv_gem_scatter_map(etnaviv_obj); 118 } 119 120 return etnaviv_obj->pages; 121 } 122 123 void etnaviv_gem_put_pages(struct etnaviv_gem_object *etnaviv_obj) 124 { 125 lockdep_assert_held(&etnaviv_obj->lock); 126 /* when we start tracking the pin count, then do something here */ 127 } 128 129 static int etnaviv_gem_mmap_obj(struct etnaviv_gem_object *etnaviv_obj, 130 struct vm_area_struct *vma) 131 { 132 pgprot_t vm_page_prot; 133 134 vm_flags_set(vma, VM_PFNMAP | VM_DONTEXPAND | VM_DONTDUMP); 135 136 vm_page_prot = vma_get_page_prot(vma); 137 138 if (etnaviv_obj->flags & ETNA_BO_WC) { 139 vma->vm_page_prot = pgprot_writecombine(vm_page_prot); 140 } else if (etnaviv_obj->flags & ETNA_BO_UNCACHED) { 141 vma->vm_page_prot = pgprot_noncached(vm_page_prot); 142 } else { 143 /* 144 * Shunt off cached objs to shmem file so they have their own 145 * address_space (so unmap_mapping_range does what we want, 146 * in particular in the case of mmap'd dmabufs) 147 */ 148 vma->vm_pgoff = 0; 149 vma_set_file(vma, etnaviv_obj->base.filp); 150 151 vma->vm_page_prot = vm_page_prot; 152 } 153 154 return 0; 155 } 156 157 static int etnaviv_gem_mmap(struct drm_gem_object *obj, struct vm_area_struct *vma) 158 { 159 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 160 161 return etnaviv_obj->ops->mmap(etnaviv_obj, vma); 162 } 163 164 static vm_fault_t etnaviv_gem_fault(struct vm_fault *vmf) 165 { 166 struct vm_area_struct *vma = vmf->vma; 167 struct drm_gem_object *obj = vma->vm_private_data; 168 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 169 struct page **pages; 170 unsigned long pfn; 171 pgoff_t pgoff; 172 int err; 173 174 /* 175 * Make sure we don't parallel update on a fault, nor move or remove 176 * something from beneath our feet. Note that vmf_insert_page() is 177 * specifically coded to take care of this, so we don't have to. 178 */ 179 err = mutex_lock_interruptible(&etnaviv_obj->lock); 180 if (err) 181 return VM_FAULT_NOPAGE; 182 /* make sure we have pages attached now */ 183 pages = etnaviv_gem_get_pages(etnaviv_obj); 184 mutex_unlock(&etnaviv_obj->lock); 185 186 if (IS_ERR(pages)) { 187 err = PTR_ERR(pages); 188 return vmf_error(err); 189 } 190 191 /* We don't use vmf->pgoff since that has the fake offset: */ 192 pgoff = linear_page_delta(vma, vmf->address); 193 194 pfn = page_to_pfn(pages[pgoff]); 195 196 VERB("Inserting %p pfn %lx, pa %lx", (void *)vmf->address, 197 pfn, pfn << PAGE_SHIFT); 198 199 return vmf_insert_pfn(vma, vmf->address, pfn); 200 } 201 202 int etnaviv_gem_mmap_offset(struct drm_gem_object *obj, u64 *offset) 203 { 204 int ret; 205 206 /* Make it mmapable */ 207 ret = drm_gem_create_mmap_offset(obj); 208 if (ret) 209 dev_err(obj->dev->dev, "could not allocate mmap offset\n"); 210 else 211 *offset = drm_vma_node_offset_addr(&obj->vma_node); 212 213 return ret; 214 } 215 216 static struct etnaviv_vram_mapping * 217 etnaviv_gem_get_vram_mapping(struct etnaviv_gem_object *obj, 218 struct etnaviv_iommu_context *context) 219 { 220 struct etnaviv_vram_mapping *mapping; 221 222 list_for_each_entry(mapping, &obj->vram_list, obj_node) { 223 if (mapping->context == context) 224 return mapping; 225 } 226 227 return NULL; 228 } 229 230 void etnaviv_gem_mapping_unreference(struct etnaviv_vram_mapping *mapping) 231 { 232 struct etnaviv_gem_object *etnaviv_obj = mapping->object; 233 234 mutex_lock(&etnaviv_obj->lock); 235 WARN_ON(mapping->use == 0); 236 mapping->use -= 1; 237 mutex_unlock(&etnaviv_obj->lock); 238 239 drm_gem_object_put(&etnaviv_obj->base); 240 } 241 242 struct etnaviv_vram_mapping *etnaviv_gem_mapping_get( 243 struct drm_gem_object *obj, struct etnaviv_iommu_context *mmu_context, 244 u64 va) 245 { 246 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 247 struct etnaviv_vram_mapping *mapping; 248 struct page **pages; 249 int ret = 0; 250 251 mutex_lock(&etnaviv_obj->lock); 252 mapping = etnaviv_gem_get_vram_mapping(etnaviv_obj, mmu_context); 253 if (mapping) { 254 /* 255 * Holding the object lock prevents the use count changing 256 * beneath us. If the use count is zero, the MMU might be 257 * reaping this object, so take the lock and re-check that 258 * the MMU owns this mapping to close this race. 259 */ 260 if (mapping->use == 0) { 261 mutex_lock(&mmu_context->lock); 262 if (mapping->context == mmu_context) 263 if (va && mapping->iova != va) { 264 etnaviv_iommu_reap_mapping(mapping); 265 mapping = NULL; 266 } else { 267 mapping->use += 1; 268 } 269 else 270 mapping = NULL; 271 mutex_unlock(&mmu_context->lock); 272 if (mapping) 273 goto out; 274 } else { 275 mapping->use += 1; 276 goto out; 277 } 278 } 279 280 pages = etnaviv_gem_get_pages(etnaviv_obj); 281 if (IS_ERR(pages)) { 282 ret = PTR_ERR(pages); 283 goto out; 284 } 285 286 /* 287 * See if we have a reaped vram mapping we can re-use before 288 * allocating a fresh mapping. 289 */ 290 mapping = etnaviv_gem_get_vram_mapping(etnaviv_obj, NULL); 291 if (!mapping) { 292 mapping = kzalloc_obj(*mapping); 293 if (!mapping) { 294 ret = -ENOMEM; 295 goto out; 296 } 297 298 INIT_LIST_HEAD(&mapping->scan_node); 299 mapping->object = etnaviv_obj; 300 } else { 301 list_del(&mapping->obj_node); 302 } 303 304 mapping->use = 1; 305 306 ret = etnaviv_iommu_map_gem(mmu_context, etnaviv_obj, 307 mmu_context->global->memory_base, 308 mapping, va); 309 if (ret < 0) 310 kfree(mapping); 311 else 312 list_add_tail(&mapping->obj_node, &etnaviv_obj->vram_list); 313 314 out: 315 mutex_unlock(&etnaviv_obj->lock); 316 317 if (ret) 318 return ERR_PTR(ret); 319 320 /* Take a reference on the object */ 321 drm_gem_object_get(obj); 322 return mapping; 323 } 324 325 void *etnaviv_gem_vmap(struct drm_gem_object *obj) 326 { 327 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 328 329 if (etnaviv_obj->vaddr) 330 return etnaviv_obj->vaddr; 331 332 mutex_lock(&etnaviv_obj->lock); 333 /* 334 * Need to check again, as we might have raced with another thread 335 * while waiting for the mutex. 336 */ 337 if (!etnaviv_obj->vaddr) 338 etnaviv_obj->vaddr = etnaviv_obj->ops->vmap(etnaviv_obj); 339 mutex_unlock(&etnaviv_obj->lock); 340 341 return etnaviv_obj->vaddr; 342 } 343 344 static void *etnaviv_gem_vmap_impl(struct etnaviv_gem_object *obj) 345 { 346 struct page **pages; 347 pgprot_t prot; 348 349 lockdep_assert_held(&obj->lock); 350 351 pages = etnaviv_gem_get_pages(obj); 352 if (IS_ERR(pages)) 353 return NULL; 354 355 switch (obj->flags & ETNA_BO_CACHE_MASK) { 356 case ETNA_BO_CACHED: 357 prot = PAGE_KERNEL; 358 break; 359 case ETNA_BO_UNCACHED: 360 prot = pgprot_noncached(PAGE_KERNEL); 361 break; 362 case ETNA_BO_WC: 363 default: 364 prot = pgprot_writecombine(PAGE_KERNEL); 365 } 366 367 return vmap(pages, obj->base.size >> PAGE_SHIFT, VM_MAP, prot); 368 } 369 370 static inline enum dma_data_direction etnaviv_op_to_dma_dir(u32 op) 371 { 372 op &= ETNA_PREP_READ | ETNA_PREP_WRITE; 373 374 if (op == ETNA_PREP_READ) 375 return DMA_FROM_DEVICE; 376 else if (op == ETNA_PREP_WRITE) 377 return DMA_TO_DEVICE; 378 else 379 return DMA_BIDIRECTIONAL; 380 } 381 382 int etnaviv_gem_cpu_prep(struct drm_gem_object *obj, u32 op, 383 struct drm_etnaviv_timespec *timeout) 384 { 385 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 386 struct drm_device *dev = obj->dev; 387 bool write = !!(op & ETNA_PREP_WRITE); 388 int ret; 389 390 if (!etnaviv_obj->sgt) { 391 void *ret; 392 393 mutex_lock(&etnaviv_obj->lock); 394 ret = etnaviv_gem_get_pages(etnaviv_obj); 395 mutex_unlock(&etnaviv_obj->lock); 396 if (IS_ERR(ret)) 397 return PTR_ERR(ret); 398 } 399 400 if (op & ETNA_PREP_NOSYNC) { 401 if (!dma_resv_test_signaled(obj->resv, 402 dma_resv_usage_rw(write))) 403 return -EBUSY; 404 } else { 405 unsigned long remain = etnaviv_timeout_to_jiffies(timeout); 406 407 ret = dma_resv_wait_timeout(obj->resv, dma_resv_usage_rw(write), 408 true, remain); 409 if (ret <= 0) 410 return ret == 0 ? -ETIMEDOUT : ret; 411 } 412 413 if (etnaviv_obj->flags & ETNA_BO_CACHED) { 414 dma_sync_sgtable_for_cpu(dev->dev, etnaviv_obj->sgt, 415 etnaviv_op_to_dma_dir(op)); 416 etnaviv_obj->last_cpu_prep_op = op; 417 } 418 419 return 0; 420 } 421 422 int etnaviv_gem_cpu_fini(struct drm_gem_object *obj) 423 { 424 struct drm_device *dev = obj->dev; 425 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 426 427 if (etnaviv_obj->flags & ETNA_BO_CACHED) { 428 /* fini without a prep is almost certainly a userspace error */ 429 WARN_ON(etnaviv_obj->last_cpu_prep_op == 0); 430 dma_sync_sgtable_for_device(dev->dev, etnaviv_obj->sgt, 431 etnaviv_op_to_dma_dir(etnaviv_obj->last_cpu_prep_op)); 432 etnaviv_obj->last_cpu_prep_op = 0; 433 } 434 435 return 0; 436 } 437 438 int etnaviv_gem_wait_bo(struct etnaviv_gpu *gpu, struct drm_gem_object *obj, 439 struct drm_etnaviv_timespec *timeout) 440 { 441 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 442 443 return etnaviv_gpu_wait_obj_inactive(gpu, etnaviv_obj, timeout); 444 } 445 446 #ifdef CONFIG_DEBUG_FS 447 static void etnaviv_gem_describe(struct drm_gem_object *obj, struct seq_file *m) 448 { 449 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 450 struct dma_resv *robj = obj->resv; 451 unsigned long off = drm_vma_node_start(&obj->vma_node); 452 int r; 453 454 seq_printf(m, "%08x: %c %2d (%2d) %08lx %p %zd\n", 455 etnaviv_obj->flags, is_active(etnaviv_obj) ? 'A' : 'I', 456 obj->name, kref_read(&obj->refcount), 457 off, etnaviv_obj->vaddr, obj->size); 458 459 r = dma_resv_lock(robj, NULL); 460 if (r) 461 return; 462 463 dma_resv_describe(robj, m); 464 dma_resv_unlock(robj); 465 } 466 467 void etnaviv_gem_describe_objects(struct etnaviv_drm_private *priv, 468 struct seq_file *m) 469 { 470 struct etnaviv_gem_object *etnaviv_obj; 471 int count = 0; 472 size_t size = 0; 473 474 mutex_lock(&priv->gem_lock); 475 list_for_each_entry(etnaviv_obj, &priv->gem_list, gem_node) { 476 struct drm_gem_object *obj = &etnaviv_obj->base; 477 478 seq_puts(m, " "); 479 etnaviv_gem_describe(obj, m); 480 count++; 481 size += obj->size; 482 } 483 mutex_unlock(&priv->gem_lock); 484 485 seq_printf(m, "Total %d objects, %zu bytes\n", count, size); 486 } 487 #endif 488 489 static void etnaviv_gem_shmem_release(struct etnaviv_gem_object *etnaviv_obj) 490 { 491 vunmap(etnaviv_obj->vaddr); 492 put_pages(etnaviv_obj); 493 } 494 495 static const struct etnaviv_gem_ops etnaviv_gem_shmem_ops = { 496 .get_pages = etnaviv_gem_shmem_get_pages, 497 .release = etnaviv_gem_shmem_release, 498 .vmap = etnaviv_gem_vmap_impl, 499 .mmap = etnaviv_gem_mmap_obj, 500 }; 501 502 void etnaviv_gem_free_object(struct drm_gem_object *obj) 503 { 504 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 505 struct etnaviv_drm_private *priv = obj->dev->dev_private; 506 struct etnaviv_vram_mapping *mapping, *tmp; 507 508 /* object should not be active */ 509 WARN_ON(is_active(etnaviv_obj)); 510 511 mutex_lock(&priv->gem_lock); 512 list_del(&etnaviv_obj->gem_node); 513 mutex_unlock(&priv->gem_lock); 514 515 list_for_each_entry_safe(mapping, tmp, &etnaviv_obj->vram_list, 516 obj_node) { 517 struct etnaviv_iommu_context *context = mapping->context; 518 519 WARN_ON(mapping->use); 520 521 if (context) 522 etnaviv_iommu_unmap_gem(context, mapping); 523 524 list_del(&mapping->obj_node); 525 kfree(mapping); 526 } 527 528 etnaviv_obj->ops->release(etnaviv_obj); 529 drm_gem_object_release(obj); 530 531 mutex_destroy(&etnaviv_obj->lock); 532 kfree(etnaviv_obj); 533 } 534 535 void etnaviv_gem_obj_add(struct drm_device *dev, struct drm_gem_object *obj) 536 { 537 struct etnaviv_drm_private *priv = dev->dev_private; 538 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 539 540 mutex_lock(&priv->gem_lock); 541 list_add_tail(&etnaviv_obj->gem_node, &priv->gem_list); 542 mutex_unlock(&priv->gem_lock); 543 } 544 545 static enum drm_gem_object_status etnaviv_gem_status(struct drm_gem_object *obj) 546 { 547 struct etnaviv_gem_object *etnaviv_obj = to_etnaviv_bo(obj); 548 enum drm_gem_object_status status = 0; 549 550 if (etnaviv_obj->pages) 551 status |= DRM_GEM_OBJECT_RESIDENT; 552 553 return status; 554 } 555 556 static const struct vm_operations_struct vm_ops = { 557 .fault = etnaviv_gem_fault, 558 .open = drm_gem_vm_open, 559 .close = drm_gem_vm_close, 560 }; 561 562 static const struct drm_gem_object_funcs etnaviv_gem_object_funcs = { 563 .free = etnaviv_gem_free_object, 564 .pin = etnaviv_gem_prime_pin, 565 .unpin = etnaviv_gem_prime_unpin, 566 .get_sg_table = etnaviv_gem_prime_get_sg_table, 567 .vmap = etnaviv_gem_prime_vmap, 568 .mmap = etnaviv_gem_mmap, 569 .status = etnaviv_gem_status, 570 .vm_ops = &vm_ops, 571 }; 572 573 static int etnaviv_gem_new_impl(struct drm_device *dev, u32 size, u32 flags, 574 const struct etnaviv_gem_ops *ops, struct drm_gem_object **obj) 575 { 576 struct etnaviv_gem_object *etnaviv_obj; 577 unsigned sz = sizeof(*etnaviv_obj); 578 bool valid = true; 579 580 /* validate flags */ 581 switch (flags & ETNA_BO_CACHE_MASK) { 582 case ETNA_BO_UNCACHED: 583 case ETNA_BO_CACHED: 584 case ETNA_BO_WC: 585 break; 586 default: 587 valid = false; 588 } 589 590 if (!valid) { 591 dev_err(dev->dev, "invalid cache flag: %x\n", 592 (flags & ETNA_BO_CACHE_MASK)); 593 return -EINVAL; 594 } 595 596 etnaviv_obj = kzalloc(sz, GFP_KERNEL); 597 if (!etnaviv_obj) 598 return -ENOMEM; 599 600 etnaviv_obj->size = ALIGN(size, SZ_4K); 601 etnaviv_obj->flags = flags; 602 etnaviv_obj->ops = ops; 603 604 mutex_init(&etnaviv_obj->lock); 605 INIT_LIST_HEAD(&etnaviv_obj->vram_list); 606 607 *obj = &etnaviv_obj->base; 608 (*obj)->funcs = &etnaviv_gem_object_funcs; 609 610 return 0; 611 } 612 613 /* convenience method to construct a GEM buffer object, and userspace handle */ 614 int etnaviv_gem_new_handle(struct drm_device *dev, struct drm_file *file, 615 u32 size, u32 flags, u32 *handle) 616 { 617 struct etnaviv_drm_private *priv = dev->dev_private; 618 struct drm_gem_object *obj = NULL; 619 int ret; 620 621 ret = etnaviv_gem_new_impl(dev, size, flags, &etnaviv_gem_shmem_ops, &obj); 622 if (ret) 623 goto fail; 624 625 lockdep_set_class(&to_etnaviv_bo(obj)->lock, &etnaviv_shm_lock_class); 626 627 ret = drm_gem_object_init(dev, obj, PAGE_ALIGN(size)); 628 if (ret) 629 goto fail; 630 631 /* 632 * Our buffers are kept pinned, so allocating them from the MOVABLE 633 * zone is a really bad idea, and conflicts with CMA. See comments 634 * above new_inode() why this is required _and_ expected if you're 635 * going to pin these pages. 636 */ 637 mapping_set_gfp_mask(obj->filp->f_mapping, priv->shm_gfp_mask); 638 639 etnaviv_gem_obj_add(dev, obj); 640 641 ret = drm_gem_handle_create(file, obj, handle); 642 643 /* drop reference from allocate - handle holds it now */ 644 fail: 645 drm_gem_object_put(obj); 646 647 return ret; 648 } 649 650 int etnaviv_gem_new_private(struct drm_device *dev, size_t size, u32 flags, 651 const struct etnaviv_gem_ops *ops, struct etnaviv_gem_object **res) 652 { 653 struct drm_gem_object *obj; 654 int ret; 655 656 ret = etnaviv_gem_new_impl(dev, size, flags, ops, &obj); 657 if (ret) 658 return ret; 659 660 drm_gem_private_object_init(dev, obj, size); 661 662 *res = to_etnaviv_bo(obj); 663 664 return 0; 665 } 666 667 static int etnaviv_gem_userptr_get_pages(struct etnaviv_gem_object *etnaviv_obj) 668 { 669 struct page **pvec = NULL; 670 struct etnaviv_gem_userptr *userptr = &etnaviv_obj->userptr; 671 int ret, pinned = 0, npages = etnaviv_obj->base.size >> PAGE_SHIFT; 672 unsigned int gup_flags = FOLL_LONGTERM; 673 674 might_lock_read(¤t->mm->mmap_lock); 675 676 if (userptr->mm != current->mm) 677 return -EPERM; 678 679 pvec = kvmalloc_objs(struct page *, npages); 680 if (!pvec) 681 return -ENOMEM; 682 683 if (!userptr->ro) 684 gup_flags |= FOLL_WRITE; 685 686 do { 687 unsigned num_pages = npages - pinned; 688 uint64_t ptr = userptr->ptr + pinned * PAGE_SIZE; 689 struct page **pages = pvec + pinned; 690 691 ret = pin_user_pages_fast(ptr, num_pages, gup_flags, pages); 692 if (ret < 0) { 693 unpin_user_pages(pvec, pinned); 694 kvfree(pvec); 695 return ret; 696 } 697 698 pinned += ret; 699 700 } while (pinned < npages); 701 702 etnaviv_obj->pages = pvec; 703 704 return 0; 705 } 706 707 static void etnaviv_gem_userptr_release(struct etnaviv_gem_object *etnaviv_obj) 708 { 709 if (etnaviv_obj->sgt) { 710 etnaviv_gem_scatterlist_unmap(etnaviv_obj); 711 sg_free_table(etnaviv_obj->sgt); 712 kfree(etnaviv_obj->sgt); 713 } 714 if (etnaviv_obj->pages) { 715 unsigned int npages = etnaviv_obj->base.size >> PAGE_SHIFT; 716 717 unpin_user_pages(etnaviv_obj->pages, npages); 718 kvfree(etnaviv_obj->pages); 719 } 720 } 721 722 static int etnaviv_gem_userptr_mmap_obj(struct etnaviv_gem_object *etnaviv_obj, 723 struct vm_area_struct *vma) 724 { 725 return -EINVAL; 726 } 727 728 static const struct etnaviv_gem_ops etnaviv_gem_userptr_ops = { 729 .get_pages = etnaviv_gem_userptr_get_pages, 730 .release = etnaviv_gem_userptr_release, 731 .vmap = etnaviv_gem_vmap_impl, 732 .mmap = etnaviv_gem_userptr_mmap_obj, 733 }; 734 735 int etnaviv_gem_new_userptr(struct drm_device *dev, struct drm_file *file, 736 uintptr_t ptr, u32 size, u32 flags, u32 *handle) 737 { 738 struct etnaviv_gem_object *etnaviv_obj; 739 int ret; 740 741 ret = etnaviv_gem_new_private(dev, size, ETNA_BO_CACHED, 742 &etnaviv_gem_userptr_ops, &etnaviv_obj); 743 if (ret) 744 return ret; 745 746 lockdep_set_class(&etnaviv_obj->lock, &etnaviv_userptr_lock_class); 747 748 etnaviv_obj->userptr.ptr = ptr; 749 etnaviv_obj->userptr.mm = current->mm; 750 etnaviv_obj->userptr.ro = !(flags & ETNA_USERPTR_WRITE); 751 752 etnaviv_gem_obj_add(dev, &etnaviv_obj->base); 753 754 ret = drm_gem_handle_create(file, &etnaviv_obj->base, handle); 755 756 /* drop reference from allocate - handle holds it now */ 757 drm_gem_object_put(&etnaviv_obj->base); 758 return ret; 759 } 760