#include "radeon.h"
 
-static void radeon_sa_bo_remove_locked(struct radeon_sa_bo *sa_bo);
-static void radeon_sa_bo_try_free(struct radeon_sa_manager *sa_manager);
-
 int radeon_sa_bo_manager_init(struct radeon_device *rdev,
                              struct radeon_sa_manager *sa_manager,
-                             unsigned size, u32 align, u32 domain, u32 flags)
+                             unsigned int size, u32 sa_align, u32 domain,
+                             u32 flags)
 {
-       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 < RADEON_NUM_RINGS; ++i) {
-               INIT_LIST_HEAD(&sa_manager->flist[i]);
-       }
+       int r;
 
-       r = radeon_bo_create(rdev, size, align, true,
+       r = radeon_bo_create(rdev, size, RADEON_GPU_PAGE_SIZE, true,
                             domain, flags, NULL, NULL, &sa_manager->bo);
        if (r) {
                dev_err(rdev->dev, "(%d) failed to allocate bo for manager\n", r);
                return r;
        }
 
+       sa_manager->domain = domain;
+
+       drm_suballoc_manager_init(&sa_manager->base, size, sa_align);
+
        return r;
 }
 
 void radeon_sa_bo_manager_fini(struct radeon_device *rdev,
                               struct radeon_sa_manager *sa_manager)
 {
-       struct radeon_sa_bo *sa_bo, *tmp;
-
-       if (!list_empty(&sa_manager->olist)) {
-               sa_manager->hole = &sa_manager->olist,
-               radeon_sa_bo_try_free(sa_manager);
-               if (!list_empty(&sa_manager->olist)) {
-                       dev_err(rdev->dev, "sa_manager is not empty, clearing anyway\n");
-               }
-       }
-       list_for_each_entry_safe(sa_bo, tmp, &sa_manager->olist, olist) {
-               radeon_sa_bo_remove_locked(sa_bo);
-       }
+       drm_suballoc_manager_fini(&sa_manager->base);
        radeon_bo_unref(&sa_manager->bo);
-       sa_manager->size = 0;
 }
 
 int radeon_sa_bo_manager_start(struct radeon_device *rdev,
        return r;
 }
 
-static void radeon_sa_bo_remove_locked(struct radeon_sa_bo *sa_bo)
+int radeon_sa_bo_new(struct radeon_sa_manager *sa_manager,
+                    struct drm_suballoc **sa_bo,
+                    unsigned int size, unsigned int align)
 {
-       struct radeon_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);
-       radeon_fence_unref(&sa_bo->fence);
-       kfree(sa_bo);
-}
-
-static void radeon_sa_bo_try_free(struct radeon_sa_manager *sa_manager)
-{
-       struct radeon_sa_bo *sa_bo, *tmp;
-
-       if (sa_manager->hole->next == &sa_manager->olist)
-               return;
+       struct drm_suballoc *sa = drm_suballoc_new(&sa_manager->base, size,
+                                                  GFP_KERNEL, true, align);
 
-       sa_bo = list_entry(sa_manager->hole->next, struct radeon_sa_bo, olist);
-       list_for_each_entry_safe_from(sa_bo, tmp, &sa_manager->olist, olist) {
-               if (sa_bo->fence == NULL || !radeon_fence_signaled(sa_bo->fence)) {
-                       return;
-               }
-               radeon_sa_bo_remove_locked(sa_bo);
+       if (IS_ERR(sa)) {
+               *sa_bo = NULL;
+               return PTR_ERR(sa);
        }
-}
 
-static inline unsigned radeon_sa_bo_hole_soffset(struct radeon_sa_manager *sa_manager)
-{
-       struct list_head *hole = sa_manager->hole;
-
-       if (hole != &sa_manager->olist) {
-               return list_entry(hole, struct radeon_sa_bo, olist)->eoffset;
-       }
+       *sa_bo = sa;
        return 0;
 }
 
-static inline unsigned radeon_sa_bo_hole_eoffset(struct radeon_sa_manager *sa_manager)
-{
-       struct list_head *hole = sa_manager->hole;
-
-       if (hole->next != &sa_manager->olist) {
-               return list_entry(hole->next, struct radeon_sa_bo, olist)->soffset;
-       }
-       return sa_manager->size;
-}
-
-static bool radeon_sa_bo_try_alloc(struct radeon_sa_manager *sa_manager,
-                                  struct radeon_sa_bo *sa_bo,
-                                  unsigned size, unsigned align)
-{
-       unsigned soffset, eoffset, wasted;
-
-       soffset = radeon_sa_bo_hole_soffset(sa_manager);
-       eoffset = radeon_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;
-}
-
-/**
- * radeon_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 radeon_sa_event(struct radeon_sa_manager *sa_manager,
-                           unsigned size, unsigned align)
-{
-       unsigned soffset, eoffset, wasted;
-       int i;
-
-       for (i = 0; i < RADEON_NUM_RINGS; ++i) {
-               if (!list_empty(&sa_manager->flist[i])) {
-                       return true;
-               }
-       }
-
-       soffset = radeon_sa_bo_hole_soffset(sa_manager);
-       eoffset = radeon_sa_bo_hole_eoffset(sa_manager);
-       wasted = (align - (soffset % align)) % align;
-
-       if ((eoffset - soffset) >= (size + wasted)) {
-               return true;
-       }
-
-       return false;
-}
-
-static bool radeon_sa_bo_next_hole(struct radeon_sa_manager *sa_manager,
-                                  struct radeon_fence **fences,
-                                  unsigned *tries)
-{
-       struct radeon_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 = radeon_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 < RADEON_NUM_RINGS; ++i) {
-               struct radeon_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 radeon_sa_bo, flist);
-
-               if (!radeon_fence_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) {
-               ++tries[best_bo->fence->ring];
-               sa_manager->hole = best_bo->olist.prev;
-
-               /* we knew that this one is signaled,
-                  so it's save to remote it */
-               radeon_sa_bo_remove_locked(best_bo);
-               return true;
-       }
-       return false;
-}
-
-int radeon_sa_bo_new(struct radeon_device *rdev,
-                    struct radeon_sa_manager *sa_manager,
-                    struct radeon_sa_bo **sa_bo,
-                    unsigned size, unsigned align)
-{
-       struct radeon_fence *fences[RADEON_NUM_RINGS];
-       unsigned tries[RADEON_NUM_RINGS];
-       int i, r;
-
-       BUG_ON(align > sa_manager->align);
-       BUG_ON(size > sa_manager->size);
-
-       *sa_bo = kmalloc(sizeof(struct radeon_sa_bo), GFP_KERNEL);
-       if ((*sa_bo) == NULL) {
-               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 < RADEON_NUM_RINGS; ++i)
-                       tries[i] = 0;
-
-               do {
-                       radeon_sa_bo_try_free(sa_manager);
-
-                       if (radeon_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 (radeon_sa_bo_next_hole(sa_manager, fences, tries));
-
-               for (i = 0; i < RADEON_NUM_RINGS; ++i)
-                       radeon_fence_ref(fences[i]);
-
-               spin_unlock(&sa_manager->wq.lock);
-               r = radeon_fence_wait_any(rdev, fences, false);
-               for (i = 0; i < RADEON_NUM_RINGS; ++i)
-                       radeon_fence_unref(&fences[i]);
-               spin_lock(&sa_manager->wq.lock);
-               /* if we have nothing to wait for block */
-               if (r == -ENOENT) {
-                       r = wait_event_interruptible_locked(
-                               sa_manager->wq, 
-                               radeon_sa_event(sa_manager, size, align)
-                       );
-               }
-
-       } while (!r);
-
-       spin_unlock(&sa_manager->wq.lock);
-       kfree(*sa_bo);
-       *sa_bo = NULL;
-       return r;
-}
-
-void radeon_sa_bo_free(struct radeon_device *rdev, struct radeon_sa_bo **sa_bo,
+void radeon_sa_bo_free(struct drm_suballoc **sa_bo,
                       struct radeon_fence *fence)
 {
-       struct radeon_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 && !radeon_fence_signaled(fence)) {
-               (*sa_bo)->fence = radeon_fence_ref(fence);
-               list_add_tail(&(*sa_bo)->flist,
-                             &sa_manager->flist[fence->ring]);
-       } else {
-               radeon_sa_bo_remove_locked(*sa_bo);
-       }
-       wake_up_all_locked(&sa_manager->wq);
-       spin_unlock(&sa_manager->wq.lock);
+       if (fence)
+               drm_suballoc_free(*sa_bo, &fence->base);
+       else
+               drm_suballoc_free(*sa_bo, NULL);
+
        *sa_bo = NULL;
 }
 
 void radeon_sa_bo_dump_debug_info(struct radeon_sa_manager *sa_manager,
                                  struct seq_file *m)
 {
-       struct radeon_sa_bo *i;
+       struct drm_printer p = drm_seq_file_printer(m);
 
-       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);
-               if (i->fence) {
-                       seq_printf(m, " protected by 0x%016llx on ring %d",
-                                  i->fence->seq, i->fence->ring);
-               }
-               seq_printf(m, "\n");
-       }
-       spin_unlock(&sa_manager->wq.lock);
+       drm_suballoc_dump_debug_info(&sa_manager->base, &p, sa_manager->gpu_addr);
 }
 #endif