drm/ttm: simplify ttm_bo_wait
[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  * Repeatedly evict memory from the LRU for @mem_type until we create enough
792  * space, or we've evicted everything and there isn't enough space.
793  */
794 static int ttm_bo_mem_force_space(struct ttm_buffer_object *bo,
795                                         uint32_t mem_type,
796                                         const struct ttm_place *place,
797                                         struct ttm_mem_reg *mem,
798                                         bool interruptible,
799                                         bool no_wait_gpu)
800 {
801         struct ttm_bo_device *bdev = bo->bdev;
802         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
803         int ret;
804
805         do {
806                 ret = (*man->func->get_node)(man, bo, place, mem);
807                 if (unlikely(ret != 0))
808                         return ret;
809                 if (mem->mm_node)
810                         break;
811                 ret = ttm_mem_evict_first(bdev, mem_type, place,
812                                           interruptible, no_wait_gpu);
813                 if (unlikely(ret != 0))
814                         return ret;
815         } while (1);
816         if (mem->mm_node == NULL)
817                 return -ENOMEM;
818         mem->mem_type = mem_type;
819         return 0;
820 }
821
822 static uint32_t ttm_bo_select_caching(struct ttm_mem_type_manager *man,
823                                       uint32_t cur_placement,
824                                       uint32_t proposed_placement)
825 {
826         uint32_t caching = proposed_placement & TTM_PL_MASK_CACHING;
827         uint32_t result = proposed_placement & ~TTM_PL_MASK_CACHING;
828
829         /**
830          * Keep current caching if possible.
831          */
832
833         if ((cur_placement & caching) != 0)
834                 result |= (cur_placement & caching);
835         else if ((man->default_caching & caching) != 0)
836                 result |= man->default_caching;
837         else if ((TTM_PL_FLAG_CACHED & caching) != 0)
838                 result |= TTM_PL_FLAG_CACHED;
839         else if ((TTM_PL_FLAG_WC & caching) != 0)
840                 result |= TTM_PL_FLAG_WC;
841         else if ((TTM_PL_FLAG_UNCACHED & caching) != 0)
842                 result |= TTM_PL_FLAG_UNCACHED;
843
844         return result;
845 }
846
847 static bool ttm_bo_mt_compatible(struct ttm_mem_type_manager *man,
848                                  uint32_t mem_type,
849                                  const struct ttm_place *place,
850                                  uint32_t *masked_placement)
851 {
852         uint32_t cur_flags = ttm_bo_type_flags(mem_type);
853
854         if ((cur_flags & place->flags & TTM_PL_MASK_MEM) == 0)
855                 return false;
856
857         if ((place->flags & man->available_caching) == 0)
858                 return false;
859
860         cur_flags |= (place->flags & man->available_caching);
861
862         *masked_placement = cur_flags;
863         return true;
864 }
865
866 /**
867  * Creates space for memory region @mem according to its type.
868  *
869  * This function first searches for free space in compatible memory types in
870  * the priority order defined by the driver.  If free space isn't found, then
871  * ttm_bo_mem_force_space is attempted in priority order to evict and find
872  * space.
873  */
874 int ttm_bo_mem_space(struct ttm_buffer_object *bo,
875                         struct ttm_placement *placement,
876                         struct ttm_mem_reg *mem,
877                         bool interruptible,
878                         bool no_wait_gpu)
879 {
880         struct ttm_bo_device *bdev = bo->bdev;
881         struct ttm_mem_type_manager *man;
882         uint32_t mem_type = TTM_PL_SYSTEM;
883         uint32_t cur_flags = 0;
884         bool type_found = false;
885         bool type_ok = false;
886         bool has_erestartsys = false;
887         int i, ret;
888
889         mem->mm_node = NULL;
890         for (i = 0; i < placement->num_placement; ++i) {
891                 const struct ttm_place *place = &placement->placement[i];
892
893                 ret = ttm_mem_type_from_place(place, &mem_type);
894                 if (ret)
895                         return ret;
896                 man = &bdev->man[mem_type];
897                 if (!man->has_type || !man->use_type)
898                         continue;
899
900                 type_ok = ttm_bo_mt_compatible(man, mem_type, place,
901                                                 &cur_flags);
902
903                 if (!type_ok)
904                         continue;
905
906                 type_found = true;
907                 cur_flags = ttm_bo_select_caching(man, bo->mem.placement,
908                                                   cur_flags);
909                 /*
910                  * Use the access and other non-mapping-related flag bits from
911                  * the memory placement flags to the current flags
912                  */
913                 ttm_flag_masked(&cur_flags, place->flags,
914                                 ~TTM_PL_MASK_MEMTYPE);
915
916                 if (mem_type == TTM_PL_SYSTEM)
917                         break;
918
919                 ret = (*man->func->get_node)(man, bo, place, mem);
920                 if (unlikely(ret))
921                         return ret;
922                 
923                 if (mem->mm_node)
924                         break;
925         }
926
927         if ((type_ok && (mem_type == TTM_PL_SYSTEM)) || mem->mm_node) {
928                 mem->mem_type = mem_type;
929                 mem->placement = cur_flags;
930                 return 0;
931         }
932
933         for (i = 0; i < placement->num_busy_placement; ++i) {
934                 const struct ttm_place *place = &placement->busy_placement[i];
935
936                 ret = ttm_mem_type_from_place(place, &mem_type);
937                 if (ret)
938                         return ret;
939                 man = &bdev->man[mem_type];
940                 if (!man->has_type || !man->use_type)
941                         continue;
942                 if (!ttm_bo_mt_compatible(man, mem_type, place, &cur_flags))
943                         continue;
944
945                 type_found = true;
946                 cur_flags = ttm_bo_select_caching(man, bo->mem.placement,
947                                                   cur_flags);
948                 /*
949                  * Use the access and other non-mapping-related flag bits from
950                  * the memory placement flags to the current flags
951                  */
952                 ttm_flag_masked(&cur_flags, place->flags,
953                                 ~TTM_PL_MASK_MEMTYPE);
954
955                 if (mem_type == TTM_PL_SYSTEM) {
956                         mem->mem_type = mem_type;
957                         mem->placement = cur_flags;
958                         mem->mm_node = NULL;
959                         return 0;
960                 }
961
962                 ret = ttm_bo_mem_force_space(bo, mem_type, place, mem,
963                                                 interruptible, no_wait_gpu);
964                 if (ret == 0 && mem->mm_node) {
965                         mem->placement = cur_flags;
966                         return 0;
967                 }
968                 if (ret == -ERESTARTSYS)
969                         has_erestartsys = true;
970         }
971
972         if (!type_found) {
973                 printk(KERN_ERR TTM_PFX "No compatible memory type found.\n");
974                 return -EINVAL;
975         }
976
977         return (has_erestartsys) ? -ERESTARTSYS : -ENOMEM;
978 }
979 EXPORT_SYMBOL(ttm_bo_mem_space);
980
981 static int ttm_bo_move_buffer(struct ttm_buffer_object *bo,
982                         struct ttm_placement *placement,
983                         bool interruptible,
984                         bool no_wait_gpu)
985 {
986         int ret = 0;
987         struct ttm_mem_reg mem;
988
989         lockdep_assert_held(&bo->resv->lock.base);
990
991         mem.num_pages = bo->num_pages;
992         mem.size = mem.num_pages << PAGE_SHIFT;
993         mem.page_alignment = bo->mem.page_alignment;
994         mem.bus.io_reserved_vm = false;
995         mem.bus.io_reserved_count = 0;
996         /*
997          * Determine where to move the buffer.
998          */
999         ret = ttm_bo_mem_space(bo, placement, &mem,
1000                                interruptible, no_wait_gpu);
1001         if (ret)
1002                 goto out_unlock;
1003         ret = ttm_bo_handle_move_mem(bo, &mem, false,
1004                                      interruptible, no_wait_gpu);
1005 out_unlock:
1006         if (ret && mem.mm_node)
1007                 ttm_bo_mem_put(bo, &mem);
1008         return ret;
1009 }
1010
1011 static bool ttm_bo_mem_compat(struct ttm_placement *placement,
1012                               struct ttm_mem_reg *mem,
1013                               uint32_t *new_flags)
1014 {
1015         int i;
1016
1017         for (i = 0; i < placement->num_placement; i++) {
1018                 const struct ttm_place *heap = &placement->placement[i];
1019                 if (mem->mm_node &&
1020                     (mem->start < heap->fpfn ||
1021                      (heap->lpfn != 0 && (mem->start + mem->num_pages) > heap->lpfn)))
1022                         continue;
1023
1024                 *new_flags = heap->flags;
1025                 if ((*new_flags & mem->placement & TTM_PL_MASK_CACHING) &&
1026                     (*new_flags & mem->placement & TTM_PL_MASK_MEM))
1027                         return true;
1028         }
1029
1030         for (i = 0; i < placement->num_busy_placement; i++) {
1031                 const struct ttm_place *heap = &placement->busy_placement[i];
1032                 if (mem->mm_node &&
1033                     (mem->start < heap->fpfn ||
1034                      (heap->lpfn != 0 && (mem->start + mem->num_pages) > heap->lpfn)))
1035                         continue;
1036
1037                 *new_flags = heap->flags;
1038                 if ((*new_flags & mem->placement & TTM_PL_MASK_CACHING) &&
1039                     (*new_flags & mem->placement & TTM_PL_MASK_MEM))
1040                         return true;
1041         }
1042
1043         return false;
1044 }
1045
1046 int ttm_bo_validate(struct ttm_buffer_object *bo,
1047                         struct ttm_placement *placement,
1048                         bool interruptible,
1049                         bool no_wait_gpu)
1050 {
1051         int ret;
1052         uint32_t new_flags;
1053
1054         lockdep_assert_held(&bo->resv->lock.base);
1055         /*
1056          * Check whether we need to move buffer.
1057          */
1058         if (!ttm_bo_mem_compat(placement, &bo->mem, &new_flags)) {
1059                 ret = ttm_bo_move_buffer(bo, placement, interruptible,
1060                                          no_wait_gpu);
1061                 if (ret)
1062                         return ret;
1063         } else {
1064                 /*
1065                  * Use the access and other non-mapping-related flag bits from
1066                  * the compatible memory placement flags to the active flags
1067                  */
1068                 ttm_flag_masked(&bo->mem.placement, new_flags,
1069                                 ~TTM_PL_MASK_MEMTYPE);
1070         }
1071         /*
1072          * We might need to add a TTM.
1073          */
1074         if (bo->mem.mem_type == TTM_PL_SYSTEM && bo->ttm == NULL) {
1075                 ret = ttm_bo_add_ttm(bo, true);
1076                 if (ret)
1077                         return ret;
1078         }
1079         return 0;
1080 }
1081 EXPORT_SYMBOL(ttm_bo_validate);
1082
1083 int ttm_bo_init(struct ttm_bo_device *bdev,
1084                 struct ttm_buffer_object *bo,
1085                 unsigned long size,
1086                 enum ttm_bo_type type,
1087                 struct ttm_placement *placement,
1088                 uint32_t page_alignment,
1089                 bool interruptible,
1090                 struct file *persistent_swap_storage,
1091                 size_t acc_size,
1092                 struct sg_table *sg,
1093                 struct reservation_object *resv,
1094                 void (*destroy) (struct ttm_buffer_object *))
1095 {
1096         int ret = 0;
1097         unsigned long num_pages;
1098         struct ttm_mem_global *mem_glob = bdev->glob->mem_glob;
1099         bool locked;
1100
1101         ret = ttm_mem_global_alloc(mem_glob, acc_size, false, false);
1102         if (ret) {
1103                 pr_err("Out of kernel memory\n");
1104                 if (destroy)
1105                         (*destroy)(bo);
1106                 else
1107                         kfree(bo);
1108                 return -ENOMEM;
1109         }
1110
1111         num_pages = (size + PAGE_SIZE - 1) >> PAGE_SHIFT;
1112         if (num_pages == 0) {
1113                 pr_err("Illegal buffer object size\n");
1114                 if (destroy)
1115                         (*destroy)(bo);
1116                 else
1117                         kfree(bo);
1118                 ttm_mem_global_free(mem_glob, acc_size);
1119                 return -EINVAL;
1120         }
1121         bo->destroy = destroy;
1122
1123         kref_init(&bo->kref);
1124         kref_init(&bo->list_kref);
1125         atomic_set(&bo->cpu_writers, 0);
1126         INIT_LIST_HEAD(&bo->lru);
1127         INIT_LIST_HEAD(&bo->ddestroy);
1128         INIT_LIST_HEAD(&bo->swap);
1129         INIT_LIST_HEAD(&bo->io_reserve_lru);
1130         mutex_init(&bo->wu_mutex);
1131         bo->bdev = bdev;
1132         bo->glob = bdev->glob;
1133         bo->type = type;
1134         bo->num_pages = num_pages;
1135         bo->mem.size = num_pages << PAGE_SHIFT;
1136         bo->mem.mem_type = TTM_PL_SYSTEM;
1137         bo->mem.num_pages = bo->num_pages;
1138         bo->mem.mm_node = NULL;
1139         bo->mem.page_alignment = page_alignment;
1140         bo->mem.bus.io_reserved_vm = false;
1141         bo->mem.bus.io_reserved_count = 0;
1142         bo->moving = NULL;
1143         bo->mem.placement = (TTM_PL_FLAG_SYSTEM | TTM_PL_FLAG_CACHED);
1144         bo->persistent_swap_storage = persistent_swap_storage;
1145         bo->acc_size = acc_size;
1146         bo->sg = sg;
1147         if (resv) {
1148                 bo->resv = resv;
1149                 lockdep_assert_held(&bo->resv->lock.base);
1150         } else {
1151                 bo->resv = &bo->ttm_resv;
1152                 reservation_object_init(&bo->ttm_resv);
1153         }
1154         atomic_inc(&bo->glob->bo_count);
1155         drm_vma_node_reset(&bo->vma_node);
1156
1157         /*
1158          * For ttm_bo_type_device buffers, allocate
1159          * address space from the device.
1160          */
1161         if (bo->type == ttm_bo_type_device ||
1162             bo->type == ttm_bo_type_sg)
1163                 ret = drm_vma_offset_add(&bdev->vma_manager, &bo->vma_node,
1164                                          bo->mem.num_pages);
1165
1166         /* passed reservation objects should already be locked,
1167          * since otherwise lockdep will be angered in radeon.
1168          */
1169         if (!resv) {
1170                 locked = ww_mutex_trylock(&bo->resv->lock);
1171                 WARN_ON(!locked);
1172         }
1173
1174         if (likely(!ret))
1175                 ret = ttm_bo_validate(bo, placement, interruptible, false);
1176
1177         if (!resv) {
1178                 ttm_bo_unreserve(bo);
1179
1180         } else if (!(bo->mem.placement & TTM_PL_FLAG_NO_EVICT)) {
1181                 spin_lock(&bo->glob->lru_lock);
1182                 ttm_bo_add_to_lru(bo);
1183                 spin_unlock(&bo->glob->lru_lock);
1184         }
1185
1186         if (unlikely(ret))
1187                 ttm_bo_unref(&bo);
1188
1189         return ret;
1190 }
1191 EXPORT_SYMBOL(ttm_bo_init);
1192
1193 size_t ttm_bo_acc_size(struct ttm_bo_device *bdev,
1194                        unsigned long bo_size,
1195                        unsigned struct_size)
1196 {
1197         unsigned npages = (PAGE_ALIGN(bo_size)) >> PAGE_SHIFT;
1198         size_t size = 0;
1199
1200         size += ttm_round_pot(struct_size);
1201         size += ttm_round_pot(npages * sizeof(void *));
1202         size += ttm_round_pot(sizeof(struct ttm_tt));
1203         return size;
1204 }
1205 EXPORT_SYMBOL(ttm_bo_acc_size);
1206
1207 size_t ttm_bo_dma_acc_size(struct ttm_bo_device *bdev,
1208                            unsigned long bo_size,
1209                            unsigned struct_size)
1210 {
1211         unsigned npages = (PAGE_ALIGN(bo_size)) >> PAGE_SHIFT;
1212         size_t size = 0;
1213
1214         size += ttm_round_pot(struct_size);
1215         size += ttm_round_pot(npages * (2*sizeof(void *) + sizeof(dma_addr_t)));
1216         size += ttm_round_pot(sizeof(struct ttm_dma_tt));
1217         return size;
1218 }
1219 EXPORT_SYMBOL(ttm_bo_dma_acc_size);
1220
1221 int ttm_bo_create(struct ttm_bo_device *bdev,
1222                         unsigned long size,
1223                         enum ttm_bo_type type,
1224                         struct ttm_placement *placement,
1225                         uint32_t page_alignment,
1226                         bool interruptible,
1227                         struct file *persistent_swap_storage,
1228                         struct ttm_buffer_object **p_bo)
1229 {
1230         struct ttm_buffer_object *bo;
1231         size_t acc_size;
1232         int ret;
1233
1234         bo = kzalloc(sizeof(*bo), GFP_KERNEL);
1235         if (unlikely(bo == NULL))
1236                 return -ENOMEM;
1237
1238         acc_size = ttm_bo_acc_size(bdev, size, sizeof(struct ttm_buffer_object));
1239         ret = ttm_bo_init(bdev, bo, size, type, placement, page_alignment,
1240                           interruptible, persistent_swap_storage, acc_size,
1241                           NULL, NULL, NULL);
1242         if (likely(ret == 0))
1243                 *p_bo = bo;
1244
1245         return ret;
1246 }
1247 EXPORT_SYMBOL(ttm_bo_create);
1248
1249 static int ttm_bo_force_list_clean(struct ttm_bo_device *bdev,
1250                                         unsigned mem_type, bool allow_errors)
1251 {
1252         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
1253         struct ttm_bo_global *glob = bdev->glob;
1254         int ret;
1255
1256         /*
1257          * Can't use standard list traversal since we're unlocking.
1258          */
1259
1260         spin_lock(&glob->lru_lock);
1261         while (!list_empty(&man->lru)) {
1262                 spin_unlock(&glob->lru_lock);
1263                 ret = ttm_mem_evict_first(bdev, mem_type, NULL, false, false);
1264                 if (ret) {
1265                         if (allow_errors) {
1266                                 return ret;
1267                         } else {
1268                                 pr_err("Cleanup eviction failed\n");
1269                         }
1270                 }
1271                 spin_lock(&glob->lru_lock);
1272         }
1273         spin_unlock(&glob->lru_lock);
1274         return 0;
1275 }
1276
1277 int ttm_bo_clean_mm(struct ttm_bo_device *bdev, unsigned mem_type)
1278 {
1279         struct ttm_mem_type_manager *man;
1280         int ret = -EINVAL;
1281
1282         if (mem_type >= TTM_NUM_MEM_TYPES) {
1283                 pr_err("Illegal memory type %d\n", mem_type);
1284                 return ret;
1285         }
1286         man = &bdev->man[mem_type];
1287
1288         if (!man->has_type) {
1289                 pr_err("Trying to take down uninitialized memory manager type %u\n",
1290                        mem_type);
1291                 return ret;
1292         }
1293
1294         man->use_type = false;
1295         man->has_type = false;
1296
1297         ret = 0;
1298         if (mem_type > 0) {
1299                 ttm_bo_force_list_clean(bdev, mem_type, false);
1300
1301                 ret = (*man->func->takedown)(man);
1302         }
1303
1304         return ret;
1305 }
1306 EXPORT_SYMBOL(ttm_bo_clean_mm);
1307
1308 int ttm_bo_evict_mm(struct ttm_bo_device *bdev, unsigned mem_type)
1309 {
1310         struct ttm_mem_type_manager *man = &bdev->man[mem_type];
1311
1312         if (mem_type == 0 || mem_type >= TTM_NUM_MEM_TYPES) {
1313                 pr_err("Illegal memory manager memory type %u\n", mem_type);
1314                 return -EINVAL;
1315         }
1316
1317         if (!man->has_type) {
1318                 pr_err("Memory type %u has not been initialized\n", mem_type);
1319                 return 0;
1320         }
1321
1322         return ttm_bo_force_list_clean(bdev, mem_type, true);
1323 }
1324 EXPORT_SYMBOL(ttm_bo_evict_mm);
1325
1326 int ttm_bo_init_mm(struct ttm_bo_device *bdev, unsigned type,
1327                         unsigned long p_size)
1328 {
1329         int ret = -EINVAL;
1330         struct ttm_mem_type_manager *man;
1331
1332         BUG_ON(type >= TTM_NUM_MEM_TYPES);
1333         man = &bdev->man[type];
1334         BUG_ON(man->has_type);
1335         man->io_reserve_fastpath = true;
1336         man->use_io_reserve_lru = false;
1337         mutex_init(&man->io_reserve_mutex);
1338         INIT_LIST_HEAD(&man->io_reserve_lru);
1339
1340         ret = bdev->driver->init_mem_type(bdev, type, man);
1341         if (ret)
1342                 return ret;
1343         man->bdev = bdev;
1344
1345         ret = 0;
1346         if (type != TTM_PL_SYSTEM) {
1347                 ret = (*man->func->init)(man, p_size);
1348                 if (ret)
1349                         return ret;
1350         }
1351         man->has_type = true;
1352         man->use_type = true;
1353         man->size = p_size;
1354
1355         INIT_LIST_HEAD(&man->lru);
1356
1357         return 0;
1358 }
1359 EXPORT_SYMBOL(ttm_bo_init_mm);
1360
1361 static void ttm_bo_global_kobj_release(struct kobject *kobj)
1362 {
1363         struct ttm_bo_global *glob =
1364                 container_of(kobj, struct ttm_bo_global, kobj);
1365
1366         ttm_mem_unregister_shrink(glob->mem_glob, &glob->shrink);
1367         __free_page(glob->dummy_read_page);
1368         kfree(glob);
1369 }
1370
1371 void ttm_bo_global_release(struct drm_global_reference *ref)
1372 {
1373         struct ttm_bo_global *glob = ref->object;
1374
1375         kobject_del(&glob->kobj);
1376         kobject_put(&glob->kobj);
1377 }
1378 EXPORT_SYMBOL(ttm_bo_global_release);
1379
1380 int ttm_bo_global_init(struct drm_global_reference *ref)
1381 {
1382         struct ttm_bo_global_ref *bo_ref =
1383                 container_of(ref, struct ttm_bo_global_ref, ref);
1384         struct ttm_bo_global *glob = ref->object;
1385         int ret;
1386
1387         mutex_init(&glob->device_list_mutex);
1388         spin_lock_init(&glob->lru_lock);
1389         glob->mem_glob = bo_ref->mem_glob;
1390         glob->dummy_read_page = alloc_page(__GFP_ZERO | GFP_DMA32);
1391
1392         if (unlikely(glob->dummy_read_page == NULL)) {
1393                 ret = -ENOMEM;
1394                 goto out_no_drp;
1395         }
1396
1397         INIT_LIST_HEAD(&glob->swap_lru);
1398         INIT_LIST_HEAD(&glob->device_list);
1399
1400         ttm_mem_init_shrink(&glob->shrink, ttm_bo_swapout);
1401         ret = ttm_mem_register_shrink(glob->mem_glob, &glob->shrink);
1402         if (unlikely(ret != 0)) {
1403                 pr_err("Could not register buffer object swapout\n");
1404                 goto out_no_shrink;
1405         }
1406
1407         atomic_set(&glob->bo_count, 0);
1408
1409         ret = kobject_init_and_add(
1410                 &glob->kobj, &ttm_bo_glob_kobj_type, ttm_get_kobj(), "buffer_objects");
1411         if (unlikely(ret != 0))
1412                 kobject_put(&glob->kobj);
1413         return ret;
1414 out_no_shrink:
1415         __free_page(glob->dummy_read_page);
1416 out_no_drp:
1417         kfree(glob);
1418         return ret;
1419 }
1420 EXPORT_SYMBOL(ttm_bo_global_init);
1421
1422
1423 int ttm_bo_device_release(struct ttm_bo_device *bdev)
1424 {
1425         int ret = 0;
1426         unsigned i = TTM_NUM_MEM_TYPES;
1427         struct ttm_mem_type_manager *man;
1428         struct ttm_bo_global *glob = bdev->glob;
1429
1430         while (i--) {
1431                 man = &bdev->man[i];
1432                 if (man->has_type) {
1433                         man->use_type = false;
1434                         if ((i != TTM_PL_SYSTEM) && ttm_bo_clean_mm(bdev, i)) {
1435                                 ret = -EBUSY;
1436                                 pr_err("DRM memory manager type %d is not clean\n",
1437                                        i);
1438                         }
1439                         man->has_type = false;
1440                 }
1441         }
1442
1443         mutex_lock(&glob->device_list_mutex);
1444         list_del(&bdev->device_list);
1445         mutex_unlock(&glob->device_list_mutex);
1446
1447         cancel_delayed_work_sync(&bdev->wq);
1448
1449         while (ttm_bo_delayed_delete(bdev, true))
1450                 ;
1451
1452         spin_lock(&glob->lru_lock);
1453         if (list_empty(&bdev->ddestroy))
1454                 TTM_DEBUG("Delayed destroy list was clean\n");
1455
1456         if (list_empty(&bdev->man[0].lru))
1457                 TTM_DEBUG("Swap list was clean\n");
1458         spin_unlock(&glob->lru_lock);
1459
1460         drm_vma_offset_manager_destroy(&bdev->vma_manager);
1461
1462         return ret;
1463 }
1464 EXPORT_SYMBOL(ttm_bo_device_release);
1465
1466 int ttm_bo_device_init(struct ttm_bo_device *bdev,
1467                        struct ttm_bo_global *glob,
1468                        struct ttm_bo_driver *driver,
1469                        struct address_space *mapping,
1470                        uint64_t file_page_offset,
1471                        bool need_dma32)
1472 {
1473         int ret = -EINVAL;
1474
1475         bdev->driver = driver;
1476
1477         memset(bdev->man, 0, sizeof(bdev->man));
1478
1479         /*
1480          * Initialize the system memory buffer type.
1481          * Other types need to be driver / IOCTL initialized.
1482          */
1483         ret = ttm_bo_init_mm(bdev, TTM_PL_SYSTEM, 0);
1484         if (unlikely(ret != 0))
1485                 goto out_no_sys;
1486
1487         drm_vma_offset_manager_init(&bdev->vma_manager, file_page_offset,
1488                                     0x10000000);
1489         INIT_DELAYED_WORK(&bdev->wq, ttm_bo_delayed_workqueue);
1490         INIT_LIST_HEAD(&bdev->ddestroy);
1491         bdev->dev_mapping = mapping;
1492         bdev->glob = glob;
1493         bdev->need_dma32 = need_dma32;
1494         mutex_lock(&glob->device_list_mutex);
1495         list_add_tail(&bdev->device_list, &glob->device_list);
1496         mutex_unlock(&glob->device_list_mutex);
1497
1498         return 0;
1499 out_no_sys:
1500         return ret;
1501 }
1502 EXPORT_SYMBOL(ttm_bo_device_init);
1503
1504 /*
1505  * buffer object vm functions.
1506  */
1507
1508 bool ttm_mem_reg_is_pci(struct ttm_bo_device *bdev, struct ttm_mem_reg *mem)
1509 {
1510         struct ttm_mem_type_manager *man = &bdev->man[mem->mem_type];
1511
1512         if (!(man->flags & TTM_MEMTYPE_FLAG_FIXED)) {
1513                 if (mem->mem_type == TTM_PL_SYSTEM)
1514                         return false;
1515
1516                 if (man->flags & TTM_MEMTYPE_FLAG_CMA)
1517                         return false;
1518
1519                 if (mem->placement & TTM_PL_FLAG_CACHED)
1520                         return false;
1521         }
1522         return true;
1523 }
1524
1525 void ttm_bo_unmap_virtual_locked(struct ttm_buffer_object *bo)
1526 {
1527         struct ttm_bo_device *bdev = bo->bdev;
1528
1529         drm_vma_node_unmap(&bo->vma_node, bdev->dev_mapping);
1530         ttm_mem_io_free_vm(bo);
1531 }
1532
1533 void ttm_bo_unmap_virtual(struct ttm_buffer_object *bo)
1534 {
1535         struct ttm_bo_device *bdev = bo->bdev;
1536         struct ttm_mem_type_manager *man = &bdev->man[bo->mem.mem_type];
1537
1538         ttm_mem_io_lock(man, false);
1539         ttm_bo_unmap_virtual_locked(bo);
1540         ttm_mem_io_unlock(man);
1541 }
1542
1543
1544 EXPORT_SYMBOL(ttm_bo_unmap_virtual);
1545
1546 int ttm_bo_wait(struct ttm_buffer_object *bo,
1547                 bool interruptible, bool no_wait)
1548 {
1549         long timeout = no_wait ? 0 : 15 * HZ;
1550
1551         timeout = reservation_object_wait_timeout_rcu(bo->resv, true,
1552                                                       interruptible, timeout);
1553         if (timeout < 0)
1554                 return timeout;
1555
1556         if (timeout == 0)
1557                 return -EBUSY;
1558
1559         reservation_object_add_excl_fence(bo->resv, NULL);
1560         return 0;
1561 }
1562 EXPORT_SYMBOL(ttm_bo_wait);
1563
1564 int ttm_bo_synccpu_write_grab(struct ttm_buffer_object *bo, bool no_wait)
1565 {
1566         int ret = 0;
1567
1568         /*
1569          * Using ttm_bo_reserve makes sure the lru lists are updated.
1570          */
1571
1572         ret = ttm_bo_reserve(bo, true, no_wait, NULL);
1573         if (unlikely(ret != 0))
1574                 return ret;
1575         ret = ttm_bo_wait(bo, true, no_wait);
1576         if (likely(ret == 0))
1577                 atomic_inc(&bo->cpu_writers);
1578         ttm_bo_unreserve(bo);
1579         return ret;
1580 }
1581 EXPORT_SYMBOL(ttm_bo_synccpu_write_grab);
1582
1583 void ttm_bo_synccpu_write_release(struct ttm_buffer_object *bo)
1584 {
1585         atomic_dec(&bo->cpu_writers);
1586 }
1587 EXPORT_SYMBOL(ttm_bo_synccpu_write_release);
1588
1589 /**
1590  * A buffer object shrink method that tries to swap out the first
1591  * buffer object on the bo_global::swap_lru list.
1592  */
1593
1594 static int ttm_bo_swapout(struct ttm_mem_shrink *shrink)
1595 {
1596         struct ttm_bo_global *glob =
1597             container_of(shrink, struct ttm_bo_global, shrink);
1598         struct ttm_buffer_object *bo;
1599         int ret = -EBUSY;
1600         int put_count;
1601         uint32_t swap_placement = (TTM_PL_FLAG_CACHED | TTM_PL_FLAG_SYSTEM);
1602
1603         spin_lock(&glob->lru_lock);
1604         list_for_each_entry(bo, &glob->swap_lru, swap) {
1605                 ret = __ttm_bo_reserve(bo, false, true, NULL);
1606                 if (!ret)
1607                         break;
1608         }
1609
1610         if (ret) {
1611                 spin_unlock(&glob->lru_lock);
1612                 return ret;
1613         }
1614
1615         kref_get(&bo->list_kref);
1616
1617         if (!list_empty(&bo->ddestroy)) {
1618                 ret = ttm_bo_cleanup_refs_and_unlock(bo, false, false);
1619                 kref_put(&bo->list_kref, ttm_bo_release_list);
1620                 return ret;
1621         }
1622
1623         put_count = ttm_bo_del_from_lru(bo);
1624         spin_unlock(&glob->lru_lock);
1625
1626         ttm_bo_list_ref_sub(bo, put_count, true);
1627
1628         /**
1629          * Move to system cached
1630          */
1631
1632         if ((bo->mem.placement & swap_placement) != swap_placement) {
1633                 struct ttm_mem_reg evict_mem;
1634
1635                 evict_mem = bo->mem;
1636                 evict_mem.mm_node = NULL;
1637                 evict_mem.placement = TTM_PL_FLAG_SYSTEM | TTM_PL_FLAG_CACHED;
1638                 evict_mem.mem_type = TTM_PL_SYSTEM;
1639
1640                 ret = ttm_bo_handle_move_mem(bo, &evict_mem, true,
1641                                              false, false);
1642                 if (unlikely(ret != 0))
1643                         goto out;
1644         }
1645
1646         /**
1647          * Make sure BO is idle.
1648          */
1649
1650         ret = ttm_bo_wait(bo, false, false);
1651         if (unlikely(ret != 0))
1652                 goto out;
1653
1654         ttm_bo_unmap_virtual(bo);
1655
1656         /**
1657          * Swap out. Buffer will be swapped in again as soon as
1658          * anyone tries to access a ttm page.
1659          */
1660
1661         if (bo->bdev->driver->swap_notify)
1662                 bo->bdev->driver->swap_notify(bo);
1663
1664         ret = ttm_tt_swapout(bo->ttm, bo->persistent_swap_storage);
1665 out:
1666
1667         /**
1668          *
1669          * Unreserve without putting on LRU to avoid swapping out an
1670          * already swapped buffer.
1671          */
1672
1673         __ttm_bo_unreserve(bo);
1674         kref_put(&bo->list_kref, ttm_bo_release_list);
1675         return ret;
1676 }
1677
1678 void ttm_bo_swapout_all(struct ttm_bo_device *bdev)
1679 {
1680         while (ttm_bo_swapout(&bdev->glob->shrink) == 0)
1681                 ;
1682 }
1683 EXPORT_SYMBOL(ttm_bo_swapout_all);
1684
1685 /**
1686  * ttm_bo_wait_unreserved - interruptible wait for a buffer object to become
1687  * unreserved
1688  *
1689  * @bo: Pointer to buffer
1690  */
1691 int ttm_bo_wait_unreserved(struct ttm_buffer_object *bo)
1692 {
1693         int ret;
1694
1695         /*
1696          * In the absense of a wait_unlocked API,
1697          * Use the bo::wu_mutex to avoid triggering livelocks due to
1698          * concurrent use of this function. Note that this use of
1699          * bo::wu_mutex can go away if we change locking order to
1700          * mmap_sem -> bo::reserve.
1701          */
1702         ret = mutex_lock_interruptible(&bo->wu_mutex);
1703         if (unlikely(ret != 0))
1704                 return -ERESTARTSYS;
1705         if (!ww_mutex_is_locked(&bo->resv->lock))
1706                 goto out_unlock;
1707         ret = __ttm_bo_reserve(bo, true, false, NULL);
1708         if (unlikely(ret != 0))
1709                 goto out_unlock;
1710         __ttm_bo_unreserve(bo);
1711
1712 out_unlock:
1713         mutex_unlock(&bo->wu_mutex);
1714         return ret;
1715 }