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