drivers/gpu/drm/ttm/ttm_bo.c

   1 /**************************************************************************
   2  *
   3  * Copyright (c) 2006-2009 VMware, Inc., Palo Alto, CA., USA
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  21  * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
  22  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  23  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  24  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27 /*
  28  * Authors: Thomas Hellstrom <thellstrom-at-vmware-dot-com>
  29  */
  30
  31 #define pr_fmt(fmt) "[TTM] " fmt
  32
  33 #include <drm/ttm/ttm_module.h>
  34 #include <drm/ttm/ttm_bo_driver.h>
  35 #include <drm/ttm/ttm_placement.h>
  36 #include <linux/jiffies.h>
  37 #include <linux/slab.h>
  38 #include <linux/sched.h>
  39 #include <linux/mm.h>
  40 #include <linux/file.h>
  41 #include <linux/module.h>
  42 #include <linux/atomic.h>
  43
  44 #define TTM_ASSERT_LOCKED(param)
  45 #define TTM_DEBUG(fmt, arg...)
  46 #define TTM_BO_HASH_ORDER 13
  47
  48 static int ttm_bo_setup_vm(struct ttm_buffer_object *bo);
  49 static int ttm_bo_swapout(struct ttm_mem_shrink *shrink);
  50 static void ttm_bo_global_kobj_release(struct kobject *kobj);
  51
  52 static struct attribute ttm_bo_count = {
  53         .name = "bo_count",
  54         .mode = S_IRUGO
  55 };
  56
  57 static inline int ttm_mem_type_from_flags(uint32_t flags, uint32_t *mem_type)
  58 {
  59         int i;
  60
  61         for (i = 0; i <= TTM_PL_PRIV5; i++)
  62                 if (flags & (1 << i)) {
  63                         *mem_type = i;
  64                         return 0;
  65                 }
  66         return -EINVAL;
  67 }
  68
  69 static void ttm_mem_type_debug(struct ttm_bo_device *bdev, int mem_type)
  70 {
  71         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
  72
  73         pr_err("    has_type: %d\n", man->has_type);
  74         pr_err("    use_type: %d\n", man->use_type);
  75         pr_err("    flags: 0x%08X\n", man->flags);
  76         pr_err("    gpu_offset: 0x%08lX\n", man->gpu_offset);
  77         pr_err("    size: %llu\n", man->size);
  78         pr_err("    available_caching: 0x%08X\n", man->available_caching);
  79         pr_err("    default_caching: 0x%08X\n", man->default_caching);
  80         if (mem_type != TTM_PL_SYSTEM)
  81                 (*man->func->debug)(man, TTM_PFX);
  82 }
  83
  84 static void ttm_bo_mem_space_debug(struct ttm_buffer_object *bo,
  85                                         struct ttm_placement *placement)
  86 {
  87         int i, ret, mem_type;
  88
  89         pr_err("No space for %p (%lu pages, %luK, %luM)\n",
  90                bo, bo->mem.num_pages, bo->mem.size >> 10,
  91                bo->mem.size >> 20);
  92         for (i = 0; i < placement->num_placement; i++) {
  93                 ret = ttm_mem_type_from_flags(placement->placement[i],
  94                                                 &mem_type);
  95                 if (ret)
  96                         return;
  97                 pr_err("  placement[%d]=0x%08X (%d)\n",
  98                        i, placement->placement[i], mem_type);
  99                 ttm_mem_type_debug(bo->bdev, mem_type);
 100         }
 101 }
 102
 103 static ssize_t ttm_bo_global_show(struct kobject *kobj,
 104                                   struct attribute *attr,
 105                                   char *buffer)
 106 {
 107         struct ttm_bo_global *glob =
 108                 container_of(kobj, struct ttm_bo_global, kobj);
 109
 110         return snprintf(buffer, PAGE_SIZE, "%lu\n",
 111                         (unsigned long) atomic_read(&glob->bo_count));
 112 }
 113
 114 static struct attribute *ttm_bo_global_attrs[] = {
 115         &ttm_bo_count,
 116         NULL
 117 };
 118
 119 static const struct sysfs_ops ttm_bo_global_ops = {
 120         .show = &ttm_bo_global_show
 121 };
 122
 123 static struct kobj_type ttm_bo_glob_kobj_type  = {
 124         .release = &ttm_bo_global_kobj_release,
 125         .sysfs_ops = &ttm_bo_global_ops,
 126         .default_attrs = ttm_bo_global_attrs
 127 };
 128
 129
 130 static inline uint32_t ttm_bo_type_flags(unsigned type)
 131 {
 132         return 1 << (type);
 133 }
 134
 135 static void ttm_bo_release_list(struct kref *list_kref)
 136 {
 137         struct ttm_buffer_object *bo =
 138             container_of(list_kref, struct ttm_buffer_object, list_kref);
 139         struct ttm_bo_device *bdev = bo->bdev;
 140         size_t acc_size = bo->acc_size;
 141
 142         BUG_ON(atomic_read(&bo->list_kref.refcount));
 143         BUG_ON(atomic_read(&bo->kref.refcount));
 144         BUG_ON(atomic_read(&bo->cpu_writers));
 145         BUG_ON(bo->sync_obj != NULL);
 146         BUG_ON(bo->mem.mm_node != NULL);
 147         BUG_ON(!list_empty(&bo->lru));
 148         BUG_ON(!list_empty(&bo->ddestroy));
 149
 150         if (bo->ttm)
 151                 ttm_tt_destroy(bo->ttm);
 152         atomic_dec(&bo->glob->bo_count);
 153         if (bo->resv == &bo->ttm_resv)
 154                 reservation_object_fini(&bo->ttm_resv);
 155
 156         if (bo->destroy)
 157                 bo->destroy(bo);
 158         else {
 159                 kfree(bo);
 160         }
 161         ttm_mem_global_free(bdev->glob->mem_glob, acc_size);
 162 }
 163
 164 void ttm_bo_add_to_lru(struct ttm_buffer_object *bo)
 165 {
 166         struct ttm_bo_device *bdev = bo->bdev;
 167         struct ttm_mem_type_manager *man;
 168
 169         lockdep_assert_held(&bo->resv->lock.base);
 170
 171         if (!(bo->mem.placement & TTM_PL_FLAG_NO_EVICT)) {
 172
 173                 BUG_ON(!list_empty(&bo->lru));
 174
 175                 man = &bdev->man[bo->mem.mem_type];
 176                 list_add_tail(&bo->lru, &man->lru);
 177                 kref_get(&bo->list_kref);
 178
 179                 if (bo->ttm != NULL) {
 180                         list_add_tail(&bo->swap, &bo->glob->swap_lru);
 181                         kref_get(&bo->list_kref);
 182                 }
 183         }
 184 }
 185 EXPORT_SYMBOL(ttm_bo_add_to_lru);
 186
 187 int ttm_bo_del_from_lru(struct ttm_buffer_object *bo)
 188 {
 189         int put_count = 0;
 190
 191         if (!list_empty(&bo->swap)) {
 192                 list_del_init(&bo->swap);
 193                 ++put_count;
 194         }
 195         if (!list_empty(&bo->lru)) {
 196                 list_del_init(&bo->lru);
 197                 ++put_count;
 198         }
 199
 200         /*
 201          * TODO: Add a driver hook to delete from
 202          * driver-specific LRU's here.
 203          */
 204
 205         return put_count;
 206 }
 207
 208 static void ttm_bo_ref_bug(struct kref *list_kref)
 209 {
 210         BUG();
 211 }
 212
 213 void ttm_bo_list_ref_sub(struct ttm_buffer_object *bo, int count,
 214                          bool never_free)
 215 {
 216         kref_sub(&bo->list_kref, count,
 217                  (never_free) ? ttm_bo_ref_bug : ttm_bo_release_list);
 218 }
 219
 220 void ttm_bo_del_sub_from_lru(struct ttm_buffer_object *bo)
 221 {
 222         int put_count;
 223
 224         spin_lock(&bo->glob->lru_lock);
 225         put_count = ttm_bo_del_from_lru(bo);
 226         spin_unlock(&bo->glob->lru_lock);
 227         ttm_bo_list_ref_sub(bo, put_count, true);
 228 }
 229 EXPORT_SYMBOL(ttm_bo_del_sub_from_lru);
 230
 231 /*
 232  * Call bo->mutex locked.
 233  */
 234 static int ttm_bo_add_ttm(struct ttm_buffer_object *bo, bool zero_alloc)
 235 {
 236         struct ttm_bo_device *bdev = bo->bdev;
 237         struct ttm_bo_global *glob = bo->glob;
 238         int ret = 0;
 239         uint32_t page_flags = 0;
 240
 241         TTM_ASSERT_LOCKED(&bo->mutex);
 242         bo->ttm = NULL;
 243
 244         if (bdev->need_dma32)
 245                 page_flags |= TTM_PAGE_FLAG_DMA32;
 246
 247         switch (bo->type) {
 248         case ttm_bo_type_device:
 249                 if (zero_alloc)
 250                         page_flags |= TTM_PAGE_FLAG_ZERO_ALLOC;
 251         case ttm_bo_type_kernel:
 252                 bo->ttm = bdev->driver->ttm_tt_create(bdev, bo->num_pages << PAGE_SHIFT,
 253                                                       page_flags, glob->dummy_read_page);
 254                 if (unlikely(bo->ttm == NULL))
 255                         ret = -ENOMEM;
 256                 break;
 257         case ttm_bo_type_sg:
 258                 bo->ttm = bdev->driver->ttm_tt_create(bdev, bo->num_pages << PAGE_SHIFT,
 259                                                       page_flags | TTM_PAGE_FLAG_SG,
 260                                                       glob->dummy_read_page);
 261                 if (unlikely(bo->ttm == NULL)) {
 262                         ret = -ENOMEM;
 263                         break;
 264                 }
 265                 bo->ttm->sg = bo->sg;
 266                 break;
 267         default:
 268                 pr_err("Illegal buffer object type\n");
 269                 ret = -EINVAL;
 270                 break;
 271         }
 272
 273         return ret;
 274 }
 275
 276 static int ttm_bo_handle_move_mem(struct ttm_buffer_object *bo,
 277                                   struct ttm_mem_reg *mem,
 278                                   bool evict, bool interruptible,
 279                                   bool no_wait_gpu)
 280 {
 281         struct ttm_bo_device *bdev = bo->bdev;
 282         bool old_is_pci = ttm_mem_reg_is_pci(bdev, &bo->mem);
 283         bool new_is_pci = ttm_mem_reg_is_pci(bdev, mem);
 284         struct ttm_mem_type_manager *old_man = &bdev->man[bo->mem.mem_type];
 285         struct ttm_mem_type_manager *new_man = &bdev->man[mem->mem_type];
 286         int ret = 0;
 287
 288         if (old_is_pci || new_is_pci ||
 289             ((mem->placement & bo->mem.placement & TTM_PL_MASK_CACHING) == 0)) {
 290                 ret = ttm_mem_io_lock(old_man, true);
 291                 if (unlikely(ret != 0))
 292                         goto out_err;
 293                 ttm_bo_unmap_virtual_locked(bo);
 294                 ttm_mem_io_unlock(old_man);
 295         }
 296
 297         /*
 298          * Create and bind a ttm if required.
 299          */
 300
 301         if (!(new_man->flags & TTM_MEMTYPE_FLAG_FIXED)) {
 302                 if (bo->ttm == NULL) {
 303                         bool zero = !(old_man->flags & TTM_MEMTYPE_FLAG_FIXED);
 304                         ret = ttm_bo_add_ttm(bo, zero);
 305                         if (ret)
 306                                 goto out_err;
 307                 }
 308
 309                 ret = ttm_tt_set_placement_caching(bo->ttm, mem->placement);
 310                 if (ret)
 311                         goto out_err;
 312
 313                 if (mem->mem_type != TTM_PL_SYSTEM) {
 314                         ret = ttm_tt_bind(bo->ttm, mem);
 315                         if (ret)
 316                                 goto out_err;
 317                 }
 318
 319                 if (bo->mem.mem_type == TTM_PL_SYSTEM) {
 320                         if (bdev->driver->move_notify)
 321                                 bdev->driver->move_notify(bo, mem);
 322                         bo->mem = *mem;
 323                         mem->mm_node = NULL;
 324                         goto moved;
 325                 }
 326         }
 327
 328         if (bdev->driver->move_notify)
 329                 bdev->driver->move_notify(bo, mem);
 330
 331         if (!(old_man->flags & TTM_MEMTYPE_FLAG_FIXED) &&
 332             !(new_man->flags & TTM_MEMTYPE_FLAG_FIXED))
 333                 ret = ttm_bo_move_ttm(bo, evict, no_wait_gpu, mem);
 334         else if (bdev->driver->move)
 335                 ret = bdev->driver->move(bo, evict, interruptible,
 336                                          no_wait_gpu, mem);
 337         else
 338                 ret = ttm_bo_move_memcpy(bo, evict, no_wait_gpu, mem);
 339
 340         if (ret) {
 341                 if (bdev->driver->move_notify) {
 342                         struct ttm_mem_reg tmp_mem = *mem;
 343                         *mem = bo->mem;
 344                         bo->mem = tmp_mem;
 345                         bdev->driver->move_notify(bo, mem);
 346                         bo->mem = *mem;
 347                         *mem = tmp_mem;
 348                 }
 349
 350                 goto out_err;
 351         }
 352
 353 moved:
 354         if (bo->evicted) {
 355                 ret = bdev->driver->invalidate_caches(bdev, bo->mem.placement);
 356                 if (ret)
 357                         pr_err("Can not flush read caches\n");
 358                 bo->evicted = false;
 359         }
 360
 361         if (bo->mem.mm_node) {
 362                 bo->offset = (bo->mem.start << PAGE_SHIFT) +
 363                     bdev->man[bo->mem.mem_type].gpu_offset;
 364                 bo->cur_placement = bo->mem.placement;
 365         } else
 366                 bo->offset = 0;
 367
 368         return 0;
 369
 370 out_err:
 371         new_man = &bdev->man[bo->mem.mem_type];
 372         if ((new_man->flags & TTM_MEMTYPE_FLAG_FIXED) && bo->ttm) {
 373                 ttm_tt_unbind(bo->ttm);
 374                 ttm_tt_destroy(bo->ttm);
 375                 bo->ttm = NULL;
 376         }
 377
 378         return ret;
 379 }
 380
 381 /**
 382  * Call bo::reserved.
 383  * Will release GPU memory type usage on destruction.
 384  * This is the place to put in driver specific hooks to release
 385  * driver private resources.
 386  * Will release the bo::reserved lock.
 387  */
 388
 389 static void ttm_bo_cleanup_memtype_use(struct ttm_buffer_object *bo)
 390 {
 391         if (bo->bdev->driver->move_notify)
 392                 bo->bdev->driver->move_notify(bo, NULL);
 393
 394         if (bo->ttm) {
 395                 ttm_tt_unbind(bo->ttm);
 396                 ttm_tt_destroy(bo->ttm);
 397                 bo->ttm = NULL;
 398         }
 399         ttm_bo_mem_put(bo, &bo->mem);
 400
 401         ww_mutex_unlock (&bo->resv->lock);
 402 }
 403
 404 static void ttm_bo_cleanup_refs_or_queue(struct ttm_buffer_object *bo)
 405 {
 406         struct ttm_bo_device *bdev = bo->bdev;
 407         struct ttm_bo_global *glob = bo->glob;
 408         struct ttm_bo_driver *driver = bdev->driver;
 409         void *sync_obj = NULL;
 410         int put_count;
 411         int ret;
 412
 413         spin_lock(&glob->lru_lock);
 414         ret = ttm_bo_reserve_nolru(bo, false, true, false, 0);
 415
 416         spin_lock(&bdev->fence_lock);
 417         (void) ttm_bo_wait(bo, false, false, true);
 418         if (!ret && !bo->sync_obj) {
 419                 spin_unlock(&bdev->fence_lock);
 420                 put_count = ttm_bo_del_from_lru(bo);
 421
 422                 spin_unlock(&glob->lru_lock);
 423                 ttm_bo_cleanup_memtype_use(bo);
 424
 425                 ttm_bo_list_ref_sub(bo, put_count, true);
 426
 427                 return;
 428         }
 429         if (bo->sync_obj)
 430                 sync_obj = driver->sync_obj_ref(bo->sync_obj);
 431         spin_unlock(&bdev->fence_lock);
 432
 433         if (!ret)
 434                 ww_mutex_unlock(&bo->resv->lock);
 435
 436         kref_get(&bo->list_kref);
 437         list_add_tail(&bo->ddestroy, &bdev->ddestroy);
 438         spin_unlock(&glob->lru_lock);
 439
 440         if (sync_obj) {
 441                 driver->sync_obj_flush(sync_obj);
 442                 driver->sync_obj_unref(&sync_obj);
 443         }
 444         schedule_delayed_work(&bdev->wq,
 445                               ((HZ / 100) < 1) ? 1 : HZ / 100);
 446 }
 447
 448 /**
 449  * function ttm_bo_cleanup_refs_and_unlock
 450  * If bo idle, remove from delayed- and lru lists, and unref.
 451  * If not idle, do nothing.
 452  *
 453  * Must be called with lru_lock and reservation held, this function
 454  * will drop both before returning.
 455  *
 456  * @interruptible         Any sleeps should occur interruptibly.
 457  * @no_wait_gpu           Never wait for gpu. Return -EBUSY instead.
 458  */
 459
 460 static int ttm_bo_cleanup_refs_and_unlock(struct ttm_buffer_object *bo,
 461                                           bool interruptible,
 462                                           bool no_wait_gpu)
 463 {
 464         struct ttm_bo_device *bdev = bo->bdev;
 465         struct ttm_bo_driver *driver = bdev->driver;
 466         struct ttm_bo_global *glob = bo->glob;
 467         int put_count;
 468         int ret;
 469
 470         spin_lock(&bdev->fence_lock);
 471         ret = ttm_bo_wait(bo, false, false, true);
 472
 473         if (ret && !no_wait_gpu) {
 474                 void *sync_obj;
 475
 476                 /*
 477                  * Take a reference to the fence and unreserve,
 478                  * at this point the buffer should be dead, so
 479                  * no new sync objects can be attached.
 480                  */
 481                 sync_obj = driver->sync_obj_ref(bo->sync_obj);
 482                 spin_unlock(&bdev->fence_lock);
 483
 484                 ww_mutex_unlock(&bo->resv->lock);
 485                 spin_unlock(&glob->lru_lock);
 486
 487                 ret = driver->sync_obj_wait(sync_obj, false, interruptible);
 488                 driver->sync_obj_unref(&sync_obj);
 489                 if (ret)
 490                         return ret;
 491
 492                 /*
 493                  * remove sync_obj with ttm_bo_wait, the wait should be
 494                  * finished, and no new wait object should have been added.
 495                  */
 496                 spin_lock(&bdev->fence_lock);
 497                 ret = ttm_bo_wait(bo, false, false, true);
 498                 WARN_ON(ret);
 499                 spin_unlock(&bdev->fence_lock);
 500                 if (ret)
 501                         return ret;
 502
 503                 spin_lock(&glob->lru_lock);
 504                 ret = ttm_bo_reserve_nolru(bo, false, true, false, 0);
 505
 506                 /*
 507                  * We raced, and lost, someone else holds the reservation now,
 508                  * and is probably busy in ttm_bo_cleanup_memtype_use.
 509                  *
 510                  * Even if it's not the case, because we finished waiting any
 511                  * delayed destruction would succeed, so just return success
 512                  * here.
 513                  */
 514                 if (ret) {
 515                         spin_unlock(&glob->lru_lock);
 516                         return 0;
 517                 }
 518         } else
 519                 spin_unlock(&bdev->fence_lock);
 520
 521         if (ret || unlikely(list_empty(&bo->ddestroy))) {
 522                 ww_mutex_unlock(&bo->resv->lock);
 523                 spin_unlock(&glob->lru_lock);
 524                 return ret;
 525         }
 526
 527         put_count = ttm_bo_del_from_lru(bo);
 528         list_del_init(&bo->ddestroy);
 529         ++put_count;
 530
 531         spin_unlock(&glob->lru_lock);
 532         ttm_bo_cleanup_memtype_use(bo);
 533
 534         ttm_bo_list_ref_sub(bo, put_count, true);
 535
 536         return 0;
 537 }
 538
 539 /**
 540  * Traverse the delayed list, and call ttm_bo_cleanup_refs on all
 541  * encountered buffers.
 542  */
 543
 544 static int ttm_bo_delayed_delete(struct ttm_bo_device *bdev, bool remove_all)
 545 {
 546         struct ttm_bo_global *glob = bdev->glob;
 547         struct ttm_buffer_object *entry = NULL;
 548         int ret = 0;
 549
 550         spin_lock(&glob->lru_lock);
 551         if (list_empty(&bdev->ddestroy))
 552                 goto out_unlock;
 553
 554         entry = list_first_entry(&bdev->ddestroy,
 555                 struct ttm_buffer_object, ddestroy);
 556         kref_get(&entry->list_kref);
 557
 558         for (;;) {
 559                 struct ttm_buffer_object *nentry = NULL;
 560
 561                 if (entry->ddestroy.next != &bdev->ddestroy) {
 562                         nentry = list_first_entry(&entry->ddestroy,
 563                                 struct ttm_buffer_object, ddestroy);
 564                         kref_get(&nentry->list_kref);
 565                 }
 566
 567                 ret = ttm_bo_reserve_nolru(entry, false, true, false, 0);
 568                 if (remove_all && ret) {
 569                         spin_unlock(&glob->lru_lock);
 570                         ret = ttm_bo_reserve_nolru(entry, false, false,
 571                                                    false, 0);
 572                         spin_lock(&glob->lru_lock);
 573                 }
 574
 575                 if (!ret)
 576                         ret = ttm_bo_cleanup_refs_and_unlock(entry, false,
 577                                                              !remove_all);
 578                 else
 579                         spin_unlock(&glob->lru_lock);
 580
 581                 kref_put(&entry->list_kref, ttm_bo_release_list);
 582                 entry = nentry;
 583
 584                 if (ret || !entry)
 585                         goto out;
 586
 587                 spin_lock(&glob->lru_lock);
 588                 if (list_empty(&entry->ddestroy))
 589                         break;
 590         }
 591
 592 out_unlock:
 593         spin_unlock(&glob->lru_lock);
 594 out:
 595         if (entry)
 596                 kref_put(&entry->list_kref, ttm_bo_release_list);
 597         return ret;
 598 }
 599
 600 static void ttm_bo_delayed_workqueue(struct work_struct *work)
 601 {
 602         struct ttm_bo_device *bdev =
 603             container_of(work, struct ttm_bo_device, wq.work);
 604
 605         if (ttm_bo_delayed_delete(bdev, false)) {
 606                 schedule_delayed_work(&bdev->wq,
 607                                       ((HZ / 100) < 1) ? 1 : HZ / 100);
 608         }
 609 }
 610
 611 static void ttm_bo_release(struct kref *kref)
 612 {
 613         struct ttm_buffer_object *bo =
 614             container_of(kref, struct ttm_buffer_object, kref);
 615         struct ttm_bo_device *bdev = bo->bdev;
 616         struct ttm_mem_type_manager *man = &bdev->man[bo->mem.mem_type];
 617
 618         write_lock(&bdev->vm_lock);
 619         if (likely(bo->vm_node != NULL)) {
 620                 rb_erase(&bo->vm_rb, &bdev->addr_space_rb);
 621                 drm_mm_put_block(bo->vm_node);
 622                 bo->vm_node = NULL;
 623         }
 624         write_unlock(&bdev->vm_lock);
 625         ttm_mem_io_lock(man, false);
 626         ttm_mem_io_free_vm(bo);
 627         ttm_mem_io_unlock(man);
 628         ttm_bo_cleanup_refs_or_queue(bo);
 629         kref_put(&bo->list_kref, ttm_bo_release_list);
 630 }
 631
 632 void ttm_bo_unref(struct ttm_buffer_object **p_bo)
 633 {
 634         struct ttm_buffer_object *bo = *p_bo;
 635
 636         *p_bo = NULL;
 637         kref_put(&bo->kref, ttm_bo_release);
 638 }
 639 EXPORT_SYMBOL(ttm_bo_unref);
 640
 641 int ttm_bo_lock_delayed_workqueue(struct ttm_bo_device *bdev)
 642 {
 643         return cancel_delayed_work_sync(&bdev->wq);
 644 }
 645 EXPORT_SYMBOL(ttm_bo_lock_delayed_workqueue);
 646
 647 void ttm_bo_unlock_delayed_workqueue(struct ttm_bo_device *bdev, int resched)
 648 {
 649         if (resched)
 650                 schedule_delayed_work(&bdev->wq,
 651                                       ((HZ / 100) < 1) ? 1 : HZ / 100);
 652 }
 653 EXPORT_SYMBOL(ttm_bo_unlock_delayed_workqueue);
 654
 655 static int ttm_bo_evict(struct ttm_buffer_object *bo, bool interruptible,
 656                         bool no_wait_gpu)
 657 {
 658         struct ttm_bo_device *bdev = bo->bdev;
 659         struct ttm_mem_reg evict_mem;
 660         struct ttm_placement placement;
 661         int ret = 0;
 662
 663         spin_lock(&bdev->fence_lock);
 664         ret = ttm_bo_wait(bo, false, interruptible, no_wait_gpu);
 665         spin_unlock(&bdev->fence_lock);
 666
 667         if (unlikely(ret != 0)) {
 668                 if (ret != -ERESTARTSYS) {
 669                         pr_err("Failed to expire sync object before buffer eviction\n");
 670                 }
 671                 goto out;
 672         }
 673
 674         lockdep_assert_held(&bo->resv->lock.base);
 675
 676         evict_mem = bo->mem;
 677         evict_mem.mm_node = NULL;
 678         evict_mem.bus.io_reserved_vm = false;
 679         evict_mem.bus.io_reserved_count = 0;
 680
 681         placement.fpfn = 0;
 682         placement.lpfn = 0;
 683         placement.num_placement = 0;
 684         placement.num_busy_placement = 0;
 685         bdev->driver->evict_flags(bo, &placement);
 686         ret = ttm_bo_mem_space(bo, &placement, &evict_mem, interruptible,
 687                                 no_wait_gpu);
 688         if (ret) {
 689                 if (ret != -ERESTARTSYS) {
 690                         pr_err("Failed to find memory space for buffer 0x%p eviction\n",
 691                                bo);
 692                         ttm_bo_mem_space_debug(bo, &placement);
 693                 }
 694                 goto out;
 695         }
 696
 697         ret = ttm_bo_handle_move_mem(bo, &evict_mem, true, interruptible,
 698                                      no_wait_gpu);
 699         if (ret) {
 700                 if (ret != -ERESTARTSYS)
 701                         pr_err("Buffer eviction failed\n");
 702                 ttm_bo_mem_put(bo, &evict_mem);
 703                 goto out;
 704         }
 705         bo->evicted = true;
 706 out:
 707         return ret;
 708 }
 709
 710 static int ttm_mem_evict_first(struct ttm_bo_device *bdev,
 711                                 uint32_t mem_type,
 712                                 bool interruptible,
 713                                 bool no_wait_gpu)
 714 {
 715         struct ttm_bo_global *glob = bdev->glob;
 716         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
 717         struct ttm_buffer_object *bo;
 718         int ret = -EBUSY, put_count;
 719
 720         spin_lock(&glob->lru_lock);
 721         list_for_each_entry(bo, &man->lru, lru) {
 722                 ret = ttm_bo_reserve_nolru(bo, false, true, false, 0);
 723                 if (!ret)
 724                         break;
 725         }
 726
 727         if (ret) {
 728                 spin_unlock(&glob->lru_lock);
 729                 return ret;
 730         }
 731
 732         kref_get(&bo->list_kref);
 733
 734         if (!list_empty(&bo->ddestroy)) {
 735                 ret = ttm_bo_cleanup_refs_and_unlock(bo, interruptible,
 736                                                      no_wait_gpu);
 737                 kref_put(&bo->list_kref, ttm_bo_release_list);
 738                 return ret;
 739         }
 740
 741         put_count = ttm_bo_del_from_lru(bo);
 742         spin_unlock(&glob->lru_lock);
 743
 744         BUG_ON(ret != 0);
 745
 746         ttm_bo_list_ref_sub(bo, put_count, true);
 747
 748         ret = ttm_bo_evict(bo, interruptible, no_wait_gpu);
 749         ttm_bo_unreserve(bo);
 750
 751         kref_put(&bo->list_kref, ttm_bo_release_list);
 752         return ret;
 753 }
 754
 755 void ttm_bo_mem_put(struct ttm_buffer_object *bo, struct ttm_mem_reg *mem)
 756 {
 757         struct ttm_mem_type_manager *man = &bo->bdev->man[mem->mem_type];
 758
 759         if (mem->mm_node)
 760                 (*man->func->put_node)(man, mem);
 761 }
 762 EXPORT_SYMBOL(ttm_bo_mem_put);
 763
 764 /**
 765  * Repeatedly evict memory from the LRU for @mem_type until we create enough
 766  * space, or we've evicted everything and there isn't enough space.
 767  */
 768 static int ttm_bo_mem_force_space(struct ttm_buffer_object *bo,
 769                                         uint32_t mem_type,
 770                                         struct ttm_placement *placement,
 771                                         struct ttm_mem_reg *mem,
 772                                         bool interruptible,
 773                                         bool no_wait_gpu)
 774 {
 775         struct ttm_bo_device *bdev = bo->bdev;
 776         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
 777         int ret;
 778
 779         do {
 780                 ret = (*man->func->get_node)(man, bo, placement, mem);
 781                 if (unlikely(ret != 0))
 782                         return ret;
 783                 if (mem->mm_node)
 784                         break;
 785                 ret = ttm_mem_evict_first(bdev, mem_type,
 786                                           interruptible, no_wait_gpu);
 787                 if (unlikely(ret != 0))
 788                         return ret;
 789         } while (1);
 790         if (mem->mm_node == NULL)
 791                 return -ENOMEM;
 792         mem->mem_type = mem_type;
 793         return 0;
 794 }
 795
 796 static uint32_t ttm_bo_select_caching(struct ttm_mem_type_manager *man,
 797                                       uint32_t cur_placement,
 798                                       uint32_t proposed_placement)
 799 {
 800         uint32_t caching = proposed_placement & TTM_PL_MASK_CACHING;
 801         uint32_t result = proposed_placement & ~TTM_PL_MASK_CACHING;
 802
 803         /**
 804          * Keep current caching if possible.
 805          */
 806
 807         if ((cur_placement & caching) != 0)
 808                 result |= (cur_placement & caching);
 809         else if ((man->default_caching & caching) != 0)
 810                 result |= man->default_caching;
 811         else if ((TTM_PL_FLAG_CACHED & caching) != 0)
 812                 result |= TTM_PL_FLAG_CACHED;
 813         else if ((TTM_PL_FLAG_WC & caching) != 0)
 814                 result |= TTM_PL_FLAG_WC;
 815         else if ((TTM_PL_FLAG_UNCACHED & caching) != 0)
 816                 result |= TTM_PL_FLAG_UNCACHED;
 817
 818         return result;
 819 }
 820
 821 static bool ttm_bo_mt_compatible(struct ttm_mem_type_manager *man,
 822                                  uint32_t mem_type,
 823                                  uint32_t proposed_placement,
 824                                  uint32_t *masked_placement)
 825 {
 826         uint32_t cur_flags = ttm_bo_type_flags(mem_type);
 827
 828         if ((cur_flags & proposed_placement & TTM_PL_MASK_MEM) == 0)
 829                 return false;
 830
 831         if ((proposed_placement & man->available_caching) == 0)
 832                 return false;
 833
 834         cur_flags |= (proposed_placement & man->available_caching);
 835
 836         *masked_placement = cur_flags;
 837         return true;
 838 }
 839
 840 /**
 841  * Creates space for memory region @mem according to its type.
 842  *
 843  * This function first searches for free space in compatible memory types in
 844  * the priority order defined by the driver.  If free space isn't found, then
 845  * ttm_bo_mem_force_space is attempted in priority order to evict and find
 846  * space.
 847  */
 848 int ttm_bo_mem_space(struct ttm_buffer_object *bo,
 849                         struct ttm_placement *placement,
 850                         struct ttm_mem_reg *mem,
 851                         bool interruptible,
 852                         bool no_wait_gpu)
 853 {
 854         struct ttm_bo_device *bdev = bo->bdev;
 855         struct ttm_mem_type_manager *man;
 856         uint32_t mem_type = TTM_PL_SYSTEM;
 857         uint32_t cur_flags = 0;
 858         bool type_found = false;
 859         bool type_ok = false;
 860         bool has_erestartsys = false;
 861         int i, ret;
 862
 863         mem->mm_node = NULL;
 864         for (i = 0; i < placement->num_placement; ++i) {
 865                 ret = ttm_mem_type_from_flags(placement->placement[i],
 866                                                 &mem_type);
 867                 if (ret)
 868                         return ret;
 869                 man = &bdev->man[mem_type];
 870
 871                 type_ok = ttm_bo_mt_compatible(man,
 872                                                 mem_type,
 873                                                 placement->placement[i],
 874                                                 &cur_flags);
 875
 876                 if (!type_ok)
 877                         continue;
 878
 879                 cur_flags = ttm_bo_select_caching(man, bo->mem.placement,
 880                                                   cur_flags);
 881                 /*
 882                  * Use the access and other non-mapping-related flag bits from
 883                  * the memory placement flags to the current flags
 884                  */
 885                 ttm_flag_masked(&cur_flags, placement->placement[i],
 886                                 ~TTM_PL_MASK_MEMTYPE);
 887
 888                 if (mem_type == TTM_PL_SYSTEM)
 889                         break;
 890
 891                 if (man->has_type && man->use_type) {
 892                         type_found = true;
 893                         ret = (*man->func->get_node)(man, bo, placement, mem);
 894                         if (unlikely(ret))
 895                                 return ret;
 896                 }
 897                 if (mem->mm_node)
 898                         break;
 899         }
 900
 901         if ((type_ok && (mem_type == TTM_PL_SYSTEM)) || mem->mm_node) {
 902                 mem->mem_type = mem_type;
 903                 mem->placement = cur_flags;
 904                 return 0;
 905         }
 906
 907         if (!type_found)
 908                 return -EINVAL;
 909
 910         for (i = 0; i < placement->num_busy_placement; ++i) {
 911                 ret = ttm_mem_type_from_flags(placement->busy_placement[i],
 912                                                 &mem_type);
 913                 if (ret)
 914                         return ret;
 915                 man = &bdev->man[mem_type];
 916                 if (!man->has_type)
 917                         continue;
 918                 if (!ttm_bo_mt_compatible(man,
 919                                                 mem_type,
 920                                                 placement->busy_placement[i],
 921                                                 &cur_flags))
 922                         continue;
 923
 924                 cur_flags = ttm_bo_select_caching(man, bo->mem.placement,
 925                                                   cur_flags);
 926                 /*
 927                  * Use the access and other non-mapping-related flag bits from
 928                  * the memory placement flags to the current flags
 929                  */
 930                 ttm_flag_masked(&cur_flags, placement->busy_placement[i],
 931                                 ~TTM_PL_MASK_MEMTYPE);
 932
 933
 934                 if (mem_type == TTM_PL_SYSTEM) {
 935                         mem->mem_type = mem_type;
 936                         mem->placement = cur_flags;
 937                         mem->mm_node = NULL;
 938                         return 0;
 939                 }
 940
 941                 ret = ttm_bo_mem_force_space(bo, mem_type, placement, mem,
 942                                                 interruptible, no_wait_gpu);
 943                 if (ret == 0 && mem->mm_node) {
 944                         mem->placement = cur_flags;
 945                         return 0;
 946                 }
 947                 if (ret == -ERESTARTSYS)
 948                         has_erestartsys = true;
 949         }
 950         ret = (has_erestartsys) ? -ERESTARTSYS : -ENOMEM;
 951         return ret;
 952 }
 953 EXPORT_SYMBOL(ttm_bo_mem_space);
 954
 955 int ttm_bo_move_buffer(struct ttm_buffer_object *bo,
 956                         struct ttm_placement *placement,
 957                         bool interruptible,
 958                         bool no_wait_gpu)
 959 {
 960         int ret = 0;
 961         struct ttm_mem_reg mem;
 962         struct ttm_bo_device *bdev = bo->bdev;
 963
 964         lockdep_assert_held(&bo->resv->lock.base);
 965
 966         /*
 967          * FIXME: It's possible to pipeline buffer moves.
 968          * Have the driver move function wait for idle when necessary,
 969          * instead of doing it here.
 970          */
 971         spin_lock(&bdev->fence_lock);
 972         ret = ttm_bo_wait(bo, false, interruptible, no_wait_gpu);
 973         spin_unlock(&bdev->fence_lock);
 974         if (ret)
 975                 return ret;
 976         mem.num_pages = bo->num_pages;
 977         mem.size = mem.num_pages << PAGE_SHIFT;
 978         mem.page_alignment = bo->mem.page_alignment;
 979         mem.bus.io_reserved_vm = false;
 980         mem.bus.io_reserved_count = 0;
 981         /*
 982          * Determine where to move the buffer.
 983          */
 984         ret = ttm_bo_mem_space(bo, placement, &mem,
 985                                interruptible, no_wait_gpu);
 986         if (ret)
 987                 goto out_unlock;
 988         ret = ttm_bo_handle_move_mem(bo, &mem, false,
 989                                      interruptible, no_wait_gpu);
 990 out_unlock:
 991         if (ret && mem.mm_node)
 992                 ttm_bo_mem_put(bo, &mem);
 993         return ret;
 994 }
 995
 996 static int ttm_bo_mem_compat(struct ttm_placement *placement,
 997                              struct ttm_mem_reg *mem)
 998 {
 999         int i;
1000
1001         if (mem->mm_node && placement->lpfn != 0 &&
1002             (mem->start < placement->fpfn ||
1003              mem->start + mem->num_pages > placement->lpfn))
1004                 return -1;
1005
1006         for (i = 0; i < placement->num_placement; i++) {
1007                 if ((placement->placement[i] & mem->placement &
1008                         TTM_PL_MASK_CACHING) &&
1009                         (placement->placement[i] & mem->placement &
1010                         TTM_PL_MASK_MEM))
1011                         return i;
1012         }
1013         return -1;
1014 }
1015
1016 int ttm_bo_validate(struct ttm_buffer_object *bo,
1017                         struct ttm_placement *placement,
1018                         bool interruptible,
1019                         bool no_wait_gpu)
1020 {
1021         int ret;
1022
1023         lockdep_assert_held(&bo->resv->lock.base);
1024         /* Check that range is valid */
1025         if (placement->lpfn || placement->fpfn)
1026                 if (placement->fpfn > placement->lpfn ||
1027                         (placement->lpfn - placement->fpfn) < bo->num_pages)
1028                         return -EINVAL;
1029         /*
1030          * Check whether we need to move buffer.
1031          */
1032         ret = ttm_bo_mem_compat(placement, &bo->mem);
1033         if (ret < 0) {
1034                 ret = ttm_bo_move_buffer(bo, placement, interruptible,
1035                                          no_wait_gpu);
1036                 if (ret)
1037                         return ret;
1038         } else {
1039                 /*
1040                  * Use the access and other non-mapping-related flag bits from
1041                  * the compatible memory placement flags to the active flags
1042                  */
1043                 ttm_flag_masked(&bo->mem.placement, placement->placement[ret],
1044                                 ~TTM_PL_MASK_MEMTYPE);
1045         }
1046         /*
1047          * We might need to add a TTM.
1048          */
1049         if (bo->mem.mem_type == TTM_PL_SYSTEM && bo->ttm == NULL) {
1050                 ret = ttm_bo_add_ttm(bo, true);
1051                 if (ret)
1052                         return ret;
1053         }
1054         return 0;
1055 }
1056 EXPORT_SYMBOL(ttm_bo_validate);
1057
1058 int ttm_bo_check_placement(struct ttm_buffer_object *bo,
1059                                 struct ttm_placement *placement)
1060 {
1061         BUG_ON((placement->fpfn || placement->lpfn) &&
1062                (bo->mem.num_pages > (placement->lpfn - placement->fpfn)));
1063
1064         return 0;
1065 }
1066
1067 int ttm_bo_init(struct ttm_bo_device *bdev,
1068                 struct ttm_buffer_object *bo,
1069                 unsigned long size,
1070                 enum ttm_bo_type type,
1071                 struct ttm_placement *placement,
1072                 uint32_t page_alignment,
1073                 bool interruptible,
1074                 struct file *persistent_swap_storage,
1075                 size_t acc_size,
1076                 struct sg_table *sg,
1077                 void (*destroy) (struct ttm_buffer_object *))
1078 {
1079         int ret = 0;
1080         unsigned long num_pages;
1081         struct ttm_mem_global *mem_glob = bdev->glob->mem_glob;
1082         bool locked;
1083
1084         ret = ttm_mem_global_alloc(mem_glob, acc_size, false, false);
1085         if (ret) {
1086                 pr_err("Out of kernel memory\n");
1087                 if (destroy)
1088                         (*destroy)(bo);
1089                 else
1090                         kfree(bo);
1091                 return -ENOMEM;
1092         }
1093
1094         num_pages = (size + PAGE_SIZE - 1) >> PAGE_SHIFT;
1095         if (num_pages == 0) {
1096                 pr_err("Illegal buffer object size\n");
1097                 if (destroy)
1098                         (*destroy)(bo);
1099                 else
1100                         kfree(bo);
1101                 ttm_mem_global_free(mem_glob, acc_size);
1102                 return -EINVAL;
1103         }
1104         bo->destroy = destroy;
1105
1106         kref_init(&bo->kref);
1107         kref_init(&bo->list_kref);
1108         atomic_set(&bo->cpu_writers, 0);
1109         INIT_LIST_HEAD(&bo->lru);
1110         INIT_LIST_HEAD(&bo->ddestroy);
1111         INIT_LIST_HEAD(&bo->swap);
1112         INIT_LIST_HEAD(&bo->io_reserve_lru);
1113         bo->bdev = bdev;
1114         bo->glob = bdev->glob;
1115         bo->type = type;
1116         bo->num_pages = num_pages;
1117         bo->mem.size = num_pages << PAGE_SHIFT;
1118         bo->mem.mem_type = TTM_PL_SYSTEM;
1119         bo->mem.num_pages = bo->num_pages;
1120         bo->mem.mm_node = NULL;
1121         bo->mem.page_alignment = page_alignment;
1122         bo->mem.bus.io_reserved_vm = false;
1123         bo->mem.bus.io_reserved_count = 0;
1124         bo->priv_flags = 0;
1125         bo->mem.placement = (TTM_PL_FLAG_SYSTEM | TTM_PL_FLAG_CACHED);
1126         bo->persistent_swap_storage = persistent_swap_storage;
1127         bo->acc_size = acc_size;
1128         bo->sg = sg;
1129         bo->resv = &bo->ttm_resv;
1130         reservation_object_init(bo->resv);
1131         atomic_inc(&bo->glob->bo_count);
1132
1133         ret = ttm_bo_check_placement(bo, placement);
1134
1135         /*
1136          * For ttm_bo_type_device buffers, allocate
1137          * address space from the device.
1138          */
1139         if (likely(!ret) &&
1140             (bo->type == ttm_bo_type_device ||
1141              bo->type == ttm_bo_type_sg))
1142                 ret = ttm_bo_setup_vm(bo);
1143
1144         locked = ww_mutex_trylock(&bo->resv->lock);
1145         WARN_ON(!locked);
1146
1147         if (likely(!ret))
1148                 ret = ttm_bo_validate(bo, placement, interruptible, false);
1149
1150         ttm_bo_unreserve(bo);
1151
1152         if (unlikely(ret))
1153                 ttm_bo_unref(&bo);
1154
1155         return ret;
1156 }
1157 EXPORT_SYMBOL(ttm_bo_init);
1158
1159 size_t ttm_bo_acc_size(struct ttm_bo_device *bdev,
1160                        unsigned long bo_size,
1161                        unsigned struct_size)
1162 {
1163         unsigned npages = (PAGE_ALIGN(bo_size)) >> PAGE_SHIFT;
1164         size_t size = 0;
1165
1166         size += ttm_round_pot(struct_size);
1167         size += PAGE_ALIGN(npages * sizeof(void *));
1168         size += ttm_round_pot(sizeof(struct ttm_tt));
1169         return size;
1170 }
1171 EXPORT_SYMBOL(ttm_bo_acc_size);
1172
1173 size_t ttm_bo_dma_acc_size(struct ttm_bo_device *bdev,
1174                            unsigned long bo_size,
1175                            unsigned struct_size)
1176 {
1177         unsigned npages = (PAGE_ALIGN(bo_size)) >> PAGE_SHIFT;
1178         size_t size = 0;
1179
1180         size += ttm_round_pot(struct_size);
1181         size += PAGE_ALIGN(npages * sizeof(void *));
1182         size += PAGE_ALIGN(npages * sizeof(dma_addr_t));
1183         size += ttm_round_pot(sizeof(struct ttm_dma_tt));
1184         return size;
1185 }
1186 EXPORT_SYMBOL(ttm_bo_dma_acc_size);
1187
1188 int ttm_bo_create(struct ttm_bo_device *bdev,
1189                         unsigned long size,
1190                         enum ttm_bo_type type,
1191                         struct ttm_placement *placement,
1192                         uint32_t page_alignment,
1193                         bool interruptible,
1194                         struct file *persistent_swap_storage,
1195                         struct ttm_buffer_object **p_bo)
1196 {
1197         struct ttm_buffer_object *bo;
1198         size_t acc_size;
1199         int ret;
1200
1201         bo = kzalloc(sizeof(*bo), GFP_KERNEL);
1202         if (unlikely(bo == NULL))
1203                 return -ENOMEM;
1204
1205         acc_size = ttm_bo_acc_size(bdev, size, sizeof(struct ttm_buffer_object));
1206         ret = ttm_bo_init(bdev, bo, size, type, placement, page_alignment,
1207                           interruptible, persistent_swap_storage, acc_size,
1208                           NULL, NULL);
1209         if (likely(ret == 0))
1210                 *p_bo = bo;
1211
1212         return ret;
1213 }
1214 EXPORT_SYMBOL(ttm_bo_create);
1215
1216 static int ttm_bo_force_list_clean(struct ttm_bo_device *bdev,
1217                                         unsigned mem_type, bool allow_errors)
1218 {
1219         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
1220         struct ttm_bo_global *glob = bdev->glob;
1221         int ret;
1222
1223         /*
1224          * Can't use standard list traversal since we're unlocking.
1225          */
1226
1227         spin_lock(&glob->lru_lock);
1228         while (!list_empty(&man->lru)) {
1229                 spin_unlock(&glob->lru_lock);
1230                 ret = ttm_mem_evict_first(bdev, mem_type, false, false);
1231                 if (ret) {
1232                         if (allow_errors) {
1233                                 return ret;
1234                         } else {
1235                                 pr_err("Cleanup eviction failed\n");
1236                         }
1237                 }
1238                 spin_lock(&glob->lru_lock);
1239         }
1240         spin_unlock(&glob->lru_lock);
1241         return 0;
1242 }
1243
1244 int ttm_bo_clean_mm(struct ttm_bo_device *bdev, unsigned mem_type)
1245 {
1246         struct ttm_mem_type_manager *man;
1247         int ret = -EINVAL;
1248
1249         if (mem_type >= TTM_NUM_MEM_TYPES) {
1250                 pr_err("Illegal memory type %d\n", mem_type);
1251                 return ret;
1252         }
1253         man = &bdev->man[mem_type];
1254
1255         if (!man->has_type) {
1256                 pr_err("Trying to take down uninitialized memory manager type %u\n",
1257                        mem_type);
1258                 return ret;
1259         }
1260
1261         man->use_type = false;
1262         man->has_type = false;
1263
1264         ret = 0;
1265         if (mem_type > 0) {
1266                 ttm_bo_force_list_clean(bdev, mem_type, false);
1267
1268                 ret = (*man->func->takedown)(man);
1269         }
1270
1271         return ret;
1272 }
1273 EXPORT_SYMBOL(ttm_bo_clean_mm);
1274
1275 int ttm_bo_evict_mm(struct ttm_bo_device *bdev, unsigned mem_type)
1276 {
1277         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
1278
1279         if (mem_type == 0 || mem_type >= TTM_NUM_MEM_TYPES) {
1280                 pr_err("Illegal memory manager memory type %u\n", mem_type);
1281                 return -EINVAL;
1282         }
1283
1284         if (!man->has_type) {
1285                 pr_err("Memory type %u has not been initialized\n", mem_type);
1286                 return 0;
1287         }
1288
1289         return ttm_bo_force_list_clean(bdev, mem_type, true);
1290 }
1291 EXPORT_SYMBOL(ttm_bo_evict_mm);
1292
1293 int ttm_bo_init_mm(struct ttm_bo_device *bdev, unsigned type,
1294                         unsigned long p_size)
1295 {
1296         int ret = -EINVAL;
1297         struct ttm_mem_type_manager *man;
1298
1299         BUG_ON(type >= TTM_NUM_MEM_TYPES);
1300         man = &bdev->man[type];
1301         BUG_ON(man->has_type);
1302         man->io_reserve_fastpath = true;
1303         man->use_io_reserve_lru = false;
1304         mutex_init(&man->io_reserve_mutex);
1305         INIT_LIST_HEAD(&man->io_reserve_lru);
1306
1307         ret = bdev->driver->init_mem_type(bdev, type, man);
1308         if (ret)
1309                 return ret;
1310         man->bdev = bdev;
1311
1312         ret = 0;
1313         if (type != TTM_PL_SYSTEM) {
1314                 ret = (*man->func->init)(man, p_size);
1315                 if (ret)
1316                         return ret;
1317         }
1318         man->has_type = true;
1319         man->use_type = true;
1320         man->size = p_size;
1321
1322         INIT_LIST_HEAD(&man->lru);
1323
1324         return 0;
1325 }
1326 EXPORT_SYMBOL(ttm_bo_init_mm);
1327
1328 static void ttm_bo_global_kobj_release(struct kobject *kobj)
1329 {
1330         struct ttm_bo_global *glob =
1331                 container_of(kobj, struct ttm_bo_global, kobj);
1332
1333         ttm_mem_unregister_shrink(glob->mem_glob, &glob->shrink);
1334         __free_page(glob->dummy_read_page);
1335         kfree(glob);
1336 }
1337
1338 void ttm_bo_global_release(struct drm_global_reference *ref)
1339 {
1340         struct ttm_bo_global *glob = ref->object;
1341
1342         kobject_del(&glob->kobj);
1343         kobject_put(&glob->kobj);
1344 }
1345 EXPORT_SYMBOL(ttm_bo_global_release);
1346
1347 int ttm_bo_global_init(struct drm_global_reference *ref)
1348 {
1349         struct ttm_bo_global_ref *bo_ref =
1350                 container_of(ref, struct ttm_bo_global_ref, ref);
1351         struct ttm_bo_global *glob = ref->object;
1352         int ret;
1353
1354         mutex_init(&glob->device_list_mutex);
1355         spin_lock_init(&glob->lru_lock);
1356         glob->mem_glob = bo_ref->mem_glob;
1357         glob->dummy_read_page = alloc_page(__GFP_ZERO | GFP_DMA32);
1358
1359         if (unlikely(glob->dummy_read_page == NULL)) {
1360                 ret = -ENOMEM;
1361                 goto out_no_drp;
1362         }
1363
1364         INIT_LIST_HEAD(&glob->swap_lru);
1365         INIT_LIST_HEAD(&glob->device_list);
1366
1367         ttm_mem_init_shrink(&glob->shrink, ttm_bo_swapout);
1368         ret = ttm_mem_register_shrink(glob->mem_glob, &glob->shrink);
1369         if (unlikely(ret != 0)) {
1370                 pr_err("Could not register buffer object swapout\n");
1371                 goto out_no_shrink;
1372         }
1373
1374         atomic_set(&glob->bo_count, 0);
1375
1376         ret = kobject_init_and_add(
1377                 &glob->kobj, &ttm_bo_glob_kobj_type, ttm_get_kobj(), "buffer_objects");
1378         if (unlikely(ret != 0))
1379                 kobject_put(&glob->kobj);
1380         return ret;
1381 out_no_shrink:
1382         __free_page(glob->dummy_read_page);
1383 out_no_drp:
1384         kfree(glob);
1385         return ret;
1386 }
1387 EXPORT_SYMBOL(ttm_bo_global_init);
1388
1389
1390 int ttm_bo_device_release(struct ttm_bo_device *bdev)
1391 {
1392         int ret = 0;
1393         unsigned i = TTM_NUM_MEM_TYPES;
1394         struct ttm_mem_type_manager *man;
1395         struct ttm_bo_global *glob = bdev->glob;
1396
1397         while (i--) {
1398                 man = &bdev->man[i];
1399                 if (man->has_type) {
1400                         man->use_type = false;
1401                         if ((i != TTM_PL_SYSTEM) && ttm_bo_clean_mm(bdev, i)) {
1402                                 ret = -EBUSY;
1403                                 pr_err("DRM memory manager type %d is not clean\n",
1404                                        i);
1405                         }
1406                         man->has_type = false;
1407                 }
1408         }
1409
1410         mutex_lock(&glob->device_list_mutex);
1411         list_del(&bdev->device_list);
1412         mutex_unlock(&glob->device_list_mutex);
1413
1414         cancel_delayed_work_sync(&bdev->wq);
1415
1416         while (ttm_bo_delayed_delete(bdev, true))
1417                 ;
1418
1419         spin_lock(&glob->lru_lock);
1420         if (list_empty(&bdev->ddestroy))
1421                 TTM_DEBUG("Delayed destroy list was clean\n");
1422
1423         if (list_empty(&bdev->man[0].lru))
1424                 TTM_DEBUG("Swap list was clean\n");
1425         spin_unlock(&glob->lru_lock);
1426
1427         BUG_ON(!drm_mm_clean(&bdev->addr_space_mm));
1428         write_lock(&bdev->vm_lock);
1429         drm_mm_takedown(&bdev->addr_space_mm);
1430         write_unlock(&bdev->vm_lock);
1431
1432         return ret;
1433 }
1434 EXPORT_SYMBOL(ttm_bo_device_release);
1435
1436 int ttm_bo_device_init(struct ttm_bo_device *bdev,
1437                        struct ttm_bo_global *glob,
1438                        struct ttm_bo_driver *driver,
1439                        uint64_t file_page_offset,
1440                        bool need_dma32)
1441 {
1442         int ret = -EINVAL;
1443
1444         rwlock_init(&bdev->vm_lock);
1445         bdev->driver = driver;
1446
1447         memset(bdev->man, 0, sizeof(bdev->man));
1448
1449         /*
1450          * Initialize the system memory buffer type.
1451          * Other types need to be driver / IOCTL initialized.
1452          */
1453         ret = ttm_bo_init_mm(bdev, TTM_PL_SYSTEM, 0);
1454         if (unlikely(ret != 0))
1455                 goto out_no_sys;
1456
1457         bdev->addr_space_rb = RB_ROOT;
1458         drm_mm_init(&bdev->addr_space_mm, file_page_offset, 0x10000000);
1459
1460         INIT_DELAYED_WORK(&bdev->wq, ttm_bo_delayed_workqueue);
1461         INIT_LIST_HEAD(&bdev->ddestroy);
1462         bdev->dev_mapping = NULL;
1463         bdev->glob = glob;
1464         bdev->need_dma32 = need_dma32;
1465         bdev->val_seq = 0;
1466         spin_lock_init(&bdev->fence_lock);
1467         mutex_lock(&glob->device_list_mutex);
1468         list_add_tail(&bdev->device_list, &glob->device_list);
1469         mutex_unlock(&glob->device_list_mutex);
1470
1471         return 0;
1472 out_no_sys:
1473         return ret;
1474 }
1475 EXPORT_SYMBOL(ttm_bo_device_init);
1476
1477 /*
1478  * buffer object vm functions.
1479  */
1480
1481 bool ttm_mem_reg_is_pci(struct ttm_bo_device *bdev, struct ttm_mem_reg *mem)
1482 {
1483         struct ttm_mem_type_manager *man = &bdev->man[mem->mem_type];
1484
1485         if (!(man->flags & TTM_MEMTYPE_FLAG_FIXED)) {
1486                 if (mem->mem_type == TTM_PL_SYSTEM)
1487                         return false;
1488
1489                 if (man->flags & TTM_MEMTYPE_FLAG_CMA)
1490                         return false;
1491
1492                 if (mem->placement & TTM_PL_FLAG_CACHED)
1493                         return false;
1494         }
1495         return true;
1496 }
1497
1498 void ttm_bo_unmap_virtual_locked(struct ttm_buffer_object *bo)
1499 {
1500         struct ttm_bo_device *bdev = bo->bdev;
1501         loff_t offset = (loff_t) bo->addr_space_offset;
1502         loff_t holelen = ((loff_t) bo->mem.num_pages) << PAGE_SHIFT;
1503
1504         if (!bdev->dev_mapping)
1505                 return;
1506         unmap_mapping_range(bdev->dev_mapping, offset, holelen, 1);
1507         ttm_mem_io_free_vm(bo);
1508 }
1509
1510 void ttm_bo_unmap_virtual(struct ttm_buffer_object *bo)
1511 {
1512         struct ttm_bo_device *bdev = bo->bdev;
1513         struct ttm_mem_type_manager *man = &bdev->man[bo->mem.mem_type];
1514
1515         ttm_mem_io_lock(man, false);
1516         ttm_bo_unmap_virtual_locked(bo);
1517         ttm_mem_io_unlock(man);
1518 }
1519
1520
1521 EXPORT_SYMBOL(ttm_bo_unmap_virtual);
1522
1523 static void ttm_bo_vm_insert_rb(struct ttm_buffer_object *bo)
1524 {
1525         struct ttm_bo_device *bdev = bo->bdev;
1526         struct rb_node **cur = &bdev->addr_space_rb.rb_node;
1527         struct rb_node *parent = NULL;
1528         struct ttm_buffer_object *cur_bo;
1529         unsigned long offset = bo->vm_node->start;
1530         unsigned long cur_offset;
1531
1532         while (*cur) {
1533                 parent = *cur;
1534                 cur_bo = rb_entry(parent, struct ttm_buffer_object, vm_rb);
1535                 cur_offset = cur_bo->vm_node->start;
1536                 if (offset < cur_offset)
1537                         cur = &parent->rb_left;
1538                 else if (offset > cur_offset)
1539                         cur = &parent->rb_right;
1540                 else
1541                         BUG();
1542         }
1543
1544         rb_link_node(&bo->vm_rb, parent, cur);
1545         rb_insert_color(&bo->vm_rb, &bdev->addr_space_rb);
1546 }
1547
1548 /**
1549  * ttm_bo_setup_vm:
1550  *
1551  * @bo: the buffer to allocate address space for
1552  *
1553  * Allocate address space in the drm device so that applications
1554  * can mmap the buffer and access the contents. This only
1555  * applies to ttm_bo_type_device objects as others are not
1556  * placed in the drm device address space.
1557  */
1558
1559 static int ttm_bo_setup_vm(struct ttm_buffer_object *bo)
1560 {
1561         struct ttm_bo_device *bdev = bo->bdev;
1562         int ret;
1563
1564 retry_pre_get:
1565         ret = drm_mm_pre_get(&bdev->addr_space_mm);
1566         if (unlikely(ret != 0))
1567                 return ret;
1568
1569         write_lock(&bdev->vm_lock);
1570         bo->vm_node = drm_mm_search_free(&bdev->addr_space_mm,
1571                                          bo->mem.num_pages, 0, 0);
1572
1573         if (unlikely(bo->vm_node == NULL)) {
1574                 ret = -ENOMEM;
1575                 goto out_unlock;
1576         }
1577
1578         bo->vm_node = drm_mm_get_block_atomic(bo->vm_node,
1579                                               bo->mem.num_pages, 0);
1580
1581         if (unlikely(bo->vm_node == NULL)) {
1582                 write_unlock(&bdev->vm_lock);
1583                 goto retry_pre_get;
1584         }
1585
1586         ttm_bo_vm_insert_rb(bo);
1587         write_unlock(&bdev->vm_lock);
1588         bo->addr_space_offset = ((uint64_t) bo->vm_node->start) << PAGE_SHIFT;
1589
1590         return 0;
1591 out_unlock:
1592         write_unlock(&bdev->vm_lock);
1593         return ret;
1594 }
1595
1596 int ttm_bo_wait(struct ttm_buffer_object *bo,
1597                 bool lazy, bool interruptible, bool no_wait)
1598 {
1599         struct ttm_bo_driver *driver = bo->bdev->driver;
1600         struct ttm_bo_device *bdev = bo->bdev;
1601         void *sync_obj;
1602         int ret = 0;
1603
1604         if (likely(bo->sync_obj == NULL))
1605                 return 0;
1606
1607         while (bo->sync_obj) {
1608
1609                 if (driver->sync_obj_signaled(bo->sync_obj)) {
1610                         void *tmp_obj = bo->sync_obj;
1611                         bo->sync_obj = NULL;
1612                         clear_bit(TTM_BO_PRIV_FLAG_MOVING, &bo->priv_flags);
1613                         spin_unlock(&bdev->fence_lock);
1614                         driver->sync_obj_unref(&tmp_obj);
1615                         spin_lock(&bdev->fence_lock);
1616                         continue;
1617                 }
1618
1619                 if (no_wait)
1620                         return -EBUSY;
1621
1622                 sync_obj = driver->sync_obj_ref(bo->sync_obj);
1623                 spin_unlock(&bdev->fence_lock);
1624                 ret = driver->sync_obj_wait(sync_obj,
1625                                             lazy, interruptible);
1626                 if (unlikely(ret != 0)) {
1627                         driver->sync_obj_unref(&sync_obj);
1628                         spin_lock(&bdev->fence_lock);
1629                         return ret;
1630                 }
1631                 spin_lock(&bdev->fence_lock);
1632                 if (likely(bo->sync_obj == sync_obj)) {
1633                         void *tmp_obj = bo->sync_obj;
1634                         bo->sync_obj = NULL;
1635                         clear_bit(TTM_BO_PRIV_FLAG_MOVING,
1636                                   &bo->priv_flags);
1637                         spin_unlock(&bdev->fence_lock);
1638                         driver->sync_obj_unref(&sync_obj);
1639                         driver->sync_obj_unref(&tmp_obj);
1640                         spin_lock(&bdev->fence_lock);
1641                 } else {
1642                         spin_unlock(&bdev->fence_lock);
1643                         driver->sync_obj_unref(&sync_obj);
1644                         spin_lock(&bdev->fence_lock);
1645                 }
1646         }
1647         return 0;
1648 }
1649 EXPORT_SYMBOL(ttm_bo_wait);
1650
1651 int ttm_bo_synccpu_write_grab(struct ttm_buffer_object *bo, bool no_wait)
1652 {
1653         struct ttm_bo_device *bdev = bo->bdev;
1654         int ret = 0;
1655
1656         /*
1657          * Using ttm_bo_reserve makes sure the lru lists are updated.
1658          */
1659
1660         ret = ttm_bo_reserve(bo, true, no_wait, false, 0);
1661         if (unlikely(ret != 0))
1662                 return ret;
1663         spin_lock(&bdev->fence_lock);
1664         ret = ttm_bo_wait(bo, false, true, no_wait);
1665         spin_unlock(&bdev->fence_lock);
1666         if (likely(ret == 0))
1667                 atomic_inc(&bo->cpu_writers);
1668         ttm_bo_unreserve(bo);
1669         return ret;
1670 }
1671 EXPORT_SYMBOL(ttm_bo_synccpu_write_grab);
1672
1673 void ttm_bo_synccpu_write_release(struct ttm_buffer_object *bo)
1674 {
1675         atomic_dec(&bo->cpu_writers);
1676 }
1677 EXPORT_SYMBOL(ttm_bo_synccpu_write_release);
1678
1679 /**
1680  * A buffer object shrink method that tries to swap out the first
1681  * buffer object on the bo_global::swap_lru list.
1682  */
1683
1684 static int ttm_bo_swapout(struct ttm_mem_shrink *shrink)
1685 {
1686         struct ttm_bo_global *glob =
1687             container_of(shrink, struct ttm_bo_global, shrink);
1688         struct ttm_buffer_object *bo;
1689         int ret = -EBUSY;
1690         int put_count;
1691         uint32_t swap_placement = (TTM_PL_FLAG_CACHED | TTM_PL_FLAG_SYSTEM);
1692
1693         spin_lock(&glob->lru_lock);
1694         list_for_each_entry(bo, &glob->swap_lru, swap) {
1695                 ret = ttm_bo_reserve_nolru(bo, false, true, false, 0);
1696                 if (!ret)
1697                         break;
1698         }
1699
1700         if (ret) {
1701                 spin_unlock(&glob->lru_lock);
1702                 return ret;
1703         }
1704
1705         kref_get(&bo->list_kref);
1706
1707         if (!list_empty(&bo->ddestroy)) {
1708                 ret = ttm_bo_cleanup_refs_and_unlock(bo, false, false);
1709                 kref_put(&bo->list_kref, ttm_bo_release_list);
1710                 return ret;
1711         }
1712
1713         put_count = ttm_bo_del_from_lru(bo);
1714         spin_unlock(&glob->lru_lock);
1715
1716         ttm_bo_list_ref_sub(bo, put_count, true);
1717
1718         /**
1719          * Wait for GPU, then move to system cached.
1720          */
1721
1722         spin_lock(&bo->bdev->fence_lock);
1723         ret = ttm_bo_wait(bo, false, false, false);
1724         spin_unlock(&bo->bdev->fence_lock);
1725
1726         if (unlikely(ret != 0))
1727                 goto out;
1728
1729         if ((bo->mem.placement & swap_placement) != swap_placement) {
1730                 struct ttm_mem_reg evict_mem;
1731
1732                 evict_mem = bo->mem;
1733                 evict_mem.mm_node = NULL;
1734                 evict_mem.placement = TTM_PL_FLAG_SYSTEM | TTM_PL_FLAG_CACHED;
1735                 evict_mem.mem_type = TTM_PL_SYSTEM;
1736
1737                 ret = ttm_bo_handle_move_mem(bo, &evict_mem, true,
1738                                              false, false);
1739                 if (unlikely(ret != 0))
1740                         goto out;
1741         }
1742
1743         ttm_bo_unmap_virtual(bo);
1744
1745         /**
1746          * Swap out. Buffer will be swapped in again as soon as
1747          * anyone tries to access a ttm page.
1748          */
1749
1750         if (bo->bdev->driver->swap_notify)
1751                 bo->bdev->driver->swap_notify(bo);
1752
1753         ret = ttm_tt_swapout(bo->ttm, bo->persistent_swap_storage);
1754 out:
1755
1756         /**
1757          *
1758          * Unreserve without putting on LRU to avoid swapping out an
1759          * already swapped buffer.
1760          */
1761
1762         ww_mutex_unlock(&bo->resv->lock);
1763         kref_put(&bo->list_kref, ttm_bo_release_list);
1764         return ret;
1765 }
1766
1767 void ttm_bo_swapout_all(struct ttm_bo_device *bdev)
1768 {
1769         while (ttm_bo_swapout(&bdev->glob->shrink) == 0)
1770                 ;
1771 }
1772 EXPORT_SYMBOL(ttm_bo_swapout_all);