#include "amdgpu.h"
 
-static void amdgpu_sa_bo_remove_locked(struct amdgpu_sa_bo *sa_bo);
-static void amdgpu_sa_bo_try_free(struct amdgpu_sa_manager *sa_manager);
-
 int amdgpu_sa_bo_manager_init(struct amdgpu_device *adev,
                              struct amdgpu_sa_manager *sa_manager,
-                             unsigned size, u32 align, u32 domain)
+                             unsigned int size, u32 suballoc_align, u32 domain)
 {
-       int i, r;
-
-       init_waitqueue_head(&sa_manager->wq);
-       sa_manager->bo = NULL;
-       sa_manager->size = size;
-       sa_manager->domain = domain;
-       sa_manager->align = align;
-       sa_manager->hole = &sa_manager->olist;
-       INIT_LIST_HEAD(&sa_manager->olist);
-       for (i = 0; i < AMDGPU_SA_NUM_FENCE_LISTS; ++i)
-               INIT_LIST_HEAD(&sa_manager->flist[i]);
+       int r;
 
-       r = amdgpu_bo_create_kernel(adev, size, align, domain, &sa_manager->bo,
-                               &sa_manager->gpu_addr, &sa_manager->cpu_ptr);
+       r = amdgpu_bo_create_kernel(adev, size, AMDGPU_GPU_PAGE_SIZE, domain,
+                                   &sa_manager->bo, &sa_manager->gpu_addr,
+                                   &sa_manager->cpu_ptr);
        if (r) {
                dev_err(adev->dev, "(%d) failed to allocate bo for manager\n", r);
                return r;
        }
 
-       memset(sa_manager->cpu_ptr, 0, sa_manager->size);
+       memset(sa_manager->cpu_ptr, 0, size);
+       drm_suballoc_manager_init(&sa_manager->base, size, suballoc_align);
        return r;
 }
 
 void amdgpu_sa_bo_manager_fini(struct amdgpu_device *adev,
                               struct amdgpu_sa_manager *sa_manager)
 {
-       struct amdgpu_sa_bo *sa_bo, *tmp;
-
        if (sa_manager->bo == NULL) {
                dev_err(adev->dev, "no bo for sa manager\n");
                return;
        }
 
-       if (!list_empty(&sa_manager->olist)) {
-               sa_manager->hole = &sa_manager->olist,
-               amdgpu_sa_bo_try_free(sa_manager);
-               if (!list_empty(&sa_manager->olist)) {
-                       dev_err(adev->dev, "sa_manager is not empty, clearing anyway\n");
-               }
-       }
-       list_for_each_entry_safe(sa_bo, tmp, &sa_manager->olist, olist) {
-               amdgpu_sa_bo_remove_locked(sa_bo);
-       }
+       drm_suballoc_manager_fini(&sa_manager->base);
 
        amdgpu_bo_free_kernel(&sa_manager->bo, &sa_manager->gpu_addr, &sa_manager->cpu_ptr);
-       sa_manager->size = 0;
 }
 
-static void amdgpu_sa_bo_remove_locked(struct amdgpu_sa_bo *sa_bo)
-{
-       struct amdgpu_sa_manager *sa_manager = sa_bo->manager;
-       if (sa_manager->hole == &sa_bo->olist) {
-               sa_manager->hole = sa_bo->olist.prev;
-       }
-       list_del_init(&sa_bo->olist);
-       list_del_init(&sa_bo->flist);
-       dma_fence_put(sa_bo->fence);
-       kfree(sa_bo);
-}
-
-static void amdgpu_sa_bo_try_free(struct amdgpu_sa_manager *sa_manager)
+int amdgpu_sa_bo_new(struct amdgpu_sa_manager *sa_manager,
+                    struct drm_suballoc **sa_bo,
+                    unsigned int size)
 {
-       struct amdgpu_sa_bo *sa_bo, *tmp;
+       struct drm_suballoc *sa = drm_suballoc_new(&sa_manager->base, size,
+                                                  GFP_KERNEL, true, 0);
 
-       if (sa_manager->hole->next == &sa_manager->olist)
-               return;
+       if (IS_ERR(sa)) {
+               *sa_bo = NULL;
 
-       sa_bo = list_entry(sa_manager->hole->next, struct amdgpu_sa_bo, olist);
-       list_for_each_entry_safe_from(sa_bo, tmp, &sa_manager->olist, olist) {
-               if (sa_bo->fence == NULL ||
-                   !dma_fence_is_signaled(sa_bo->fence)) {
-                       return;
-               }
-               amdgpu_sa_bo_remove_locked(sa_bo);
+               return PTR_ERR(sa);
        }
-}
 
-static inline unsigned amdgpu_sa_bo_hole_soffset(struct amdgpu_sa_manager *sa_manager)
-{
-       struct list_head *hole = sa_manager->hole;
-
-       if (hole != &sa_manager->olist) {
-               return list_entry(hole, struct amdgpu_sa_bo, olist)->eoffset;
-       }
+       *sa_bo = sa;
        return 0;
 }
 
-static inline unsigned amdgpu_sa_bo_hole_eoffset(struct amdgpu_sa_manager *sa_manager)
-{
-       struct list_head *hole = sa_manager->hole;
-
-       if (hole->next != &sa_manager->olist) {
-               return list_entry(hole->next, struct amdgpu_sa_bo, olist)->soffset;
-       }
-       return sa_manager->size;
-}
-
-static bool amdgpu_sa_bo_try_alloc(struct amdgpu_sa_manager *sa_manager,
-                                  struct amdgpu_sa_bo *sa_bo,
-                                  unsigned size, unsigned align)
-{
-       unsigned soffset, eoffset, wasted;
-
-       soffset = amdgpu_sa_bo_hole_soffset(sa_manager);
-       eoffset = amdgpu_sa_bo_hole_eoffset(sa_manager);
-       wasted = (align - (soffset % align)) % align;
-
-       if ((eoffset - soffset) >= (size + wasted)) {
-               soffset += wasted;
-
-               sa_bo->manager = sa_manager;
-               sa_bo->soffset = soffset;
-               sa_bo->eoffset = soffset + size;
-               list_add(&sa_bo->olist, sa_manager->hole);
-               INIT_LIST_HEAD(&sa_bo->flist);
-               sa_manager->hole = &sa_bo->olist;
-               return true;
-       }
-       return false;
-}
-
-/**
- * amdgpu_sa_event - Check if we can stop waiting
- *
- * @sa_manager: pointer to the sa_manager
- * @size: number of bytes we want to allocate
- * @align: alignment we need to match
- *
- * Check if either there is a fence we can wait for or
- * enough free memory to satisfy the allocation directly
- */
-static bool amdgpu_sa_event(struct amdgpu_sa_manager *sa_manager,
-                           unsigned size, unsigned align)
-{
-       unsigned soffset, eoffset, wasted;
-       int i;
-
-       for (i = 0; i < AMDGPU_SA_NUM_FENCE_LISTS; ++i)
-               if (!list_empty(&sa_manager->flist[i]))
-                       return true;
-
-       soffset = amdgpu_sa_bo_hole_soffset(sa_manager);
-       eoffset = amdgpu_sa_bo_hole_eoffset(sa_manager);
-       wasted = (align - (soffset % align)) % align;
-
-       if ((eoffset - soffset) >= (size + wasted)) {
-               return true;
-       }
-
-       return false;
-}
-
-static bool amdgpu_sa_bo_next_hole(struct amdgpu_sa_manager *sa_manager,
-                                  struct dma_fence **fences,
-                                  unsigned *tries)
-{
-       struct amdgpu_sa_bo *best_bo = NULL;
-       unsigned i, soffset, best, tmp;
-
-       /* if hole points to the end of the buffer */
-       if (sa_manager->hole->next == &sa_manager->olist) {
-               /* try again with its beginning */
-               sa_manager->hole = &sa_manager->olist;
-               return true;
-       }
-
-       soffset = amdgpu_sa_bo_hole_soffset(sa_manager);
-       /* to handle wrap around we add sa_manager->size */
-       best = sa_manager->size * 2;
-       /* go over all fence list and try to find the closest sa_bo
-        * of the current last
-        */
-       for (i = 0; i < AMDGPU_SA_NUM_FENCE_LISTS; ++i) {
-               struct amdgpu_sa_bo *sa_bo;
-
-               fences[i] = NULL;
-
-               if (list_empty(&sa_manager->flist[i]))
-                       continue;
-
-               sa_bo = list_first_entry(&sa_manager->flist[i],
-                                        struct amdgpu_sa_bo, flist);
-
-               if (!dma_fence_is_signaled(sa_bo->fence)) {
-                       fences[i] = sa_bo->fence;
-                       continue;
-               }
-
-               /* limit the number of tries each ring gets */
-               if (tries[i] > 2) {
-                       continue;
-               }
-
-               tmp = sa_bo->soffset;
-               if (tmp < soffset) {
-                       /* wrap around, pretend it's after */
-                       tmp += sa_manager->size;
-               }
-               tmp -= soffset;
-               if (tmp < best) {
-                       /* this sa bo is the closest one */
-                       best = tmp;
-                       best_bo = sa_bo;
-               }
-       }
-
-       if (best_bo) {
-               uint32_t idx = best_bo->fence->context;
-
-               idx %= AMDGPU_SA_NUM_FENCE_LISTS;
-               ++tries[idx];
-               sa_manager->hole = best_bo->olist.prev;
-
-               /* we knew that this one is signaled,
-                  so it's save to remote it */
-               amdgpu_sa_bo_remove_locked(best_bo);
-               return true;
-       }
-       return false;
-}
-
-int amdgpu_sa_bo_new(struct amdgpu_sa_manager *sa_manager,
-                    struct amdgpu_sa_bo **sa_bo,
-                    unsigned size, unsigned align)
-{
-       struct dma_fence *fences[AMDGPU_SA_NUM_FENCE_LISTS];
-       unsigned tries[AMDGPU_SA_NUM_FENCE_LISTS];
-       unsigned count;
-       int i, r;
-       signed long t;
-
-       if (WARN_ON_ONCE(align > sa_manager->align))
-               return -EINVAL;
-
-       if (WARN_ON_ONCE(size > sa_manager->size))
-               return -EINVAL;
-
-       *sa_bo = kmalloc(sizeof(struct amdgpu_sa_bo), GFP_KERNEL);
-       if (!(*sa_bo))
-               return -ENOMEM;
-       (*sa_bo)->manager = sa_manager;
-       (*sa_bo)->fence = NULL;
-       INIT_LIST_HEAD(&(*sa_bo)->olist);
-       INIT_LIST_HEAD(&(*sa_bo)->flist);
-
-       spin_lock(&sa_manager->wq.lock);
-       do {
-               for (i = 0; i < AMDGPU_SA_NUM_FENCE_LISTS; ++i)
-                       tries[i] = 0;
-
-               do {
-                       amdgpu_sa_bo_try_free(sa_manager);
-
-                       if (amdgpu_sa_bo_try_alloc(sa_manager, *sa_bo,
-                                                  size, align)) {
-                               spin_unlock(&sa_manager->wq.lock);
-                               return 0;
-                       }
-
-                       /* see if we can skip over some allocations */
-               } while (amdgpu_sa_bo_next_hole(sa_manager, fences, tries));
-
-               for (i = 0, count = 0; i < AMDGPU_SA_NUM_FENCE_LISTS; ++i)
-                       if (fences[i])
-                               fences[count++] = dma_fence_get(fences[i]);
-
-               if (count) {
-                       spin_unlock(&sa_manager->wq.lock);
-                       t = dma_fence_wait_any_timeout(fences, count, false,
-                                                      MAX_SCHEDULE_TIMEOUT,
-                                                      NULL);
-                       for (i = 0; i < count; ++i)
-                               dma_fence_put(fences[i]);
-
-                       r = (t > 0) ? 0 : t;
-                       spin_lock(&sa_manager->wq.lock);
-               } else {
-                       /* if we have nothing to wait for block */
-                       r = wait_event_interruptible_locked(
-                               sa_manager->wq,
-                               amdgpu_sa_event(sa_manager, size, align)
-                       );
-               }
-
-       } while (!r);
-
-       spin_unlock(&sa_manager->wq.lock);
-       kfree(*sa_bo);
-       *sa_bo = NULL;
-       return r;
-}
-
-void amdgpu_sa_bo_free(struct amdgpu_device *adev, struct amdgpu_sa_bo **sa_bo,
+void amdgpu_sa_bo_free(struct amdgpu_device *adev, struct drm_suballoc **sa_bo,
                       struct dma_fence *fence)
 {
-       struct amdgpu_sa_manager *sa_manager;
-
        if (sa_bo == NULL || *sa_bo == NULL) {
                return;
        }
 
-       sa_manager = (*sa_bo)->manager;
-       spin_lock(&sa_manager->wq.lock);
-       if (fence && !dma_fence_is_signaled(fence)) {
-               uint32_t idx;
-
-               (*sa_bo)->fence = dma_fence_get(fence);
-               idx = fence->context % AMDGPU_SA_NUM_FENCE_LISTS;
-               list_add_tail(&(*sa_bo)->flist, &sa_manager->flist[idx]);
-       } else {
-               amdgpu_sa_bo_remove_locked(*sa_bo);
-       }
-       wake_up_all_locked(&sa_manager->wq);
-       spin_unlock(&sa_manager->wq.lock);
+       drm_suballoc_free(*sa_bo, fence);
        *sa_bo = NULL;
 }
 
 void amdgpu_sa_bo_dump_debug_info(struct amdgpu_sa_manager *sa_manager,
                                  struct seq_file *m)
 {
-       struct amdgpu_sa_bo *i;
-
-       spin_lock(&sa_manager->wq.lock);
-       list_for_each_entry(i, &sa_manager->olist, olist) {
-               uint64_t soffset = i->soffset + sa_manager->gpu_addr;
-               uint64_t eoffset = i->eoffset + sa_manager->gpu_addr;
-               if (&i->olist == sa_manager->hole) {
-                       seq_printf(m, ">");
-               } else {
-                       seq_printf(m, " ");
-               }
-               seq_printf(m, "[0x%010llx 0x%010llx] size %8lld",
-                          soffset, eoffset, eoffset - soffset);
+       struct drm_printer p = drm_seq_file_printer(m);
 
-               if (i->fence)
-                       seq_printf(m, " protected by 0x%016llx on context %llu",
-                                  i->fence->seqno, i->fence->context);
-
-               seq_printf(m, "\n");
-       }
-       spin_unlock(&sa_manager->wq.lock);
+       drm_suballoc_dump_debug_info(&sa_manager->base, &p, sa_manager->gpu_addr);
 }
 #endif