// SPDX-License-Identifier: MIT
/*
 * Copyright © 2020 Intel Corporation
 */
#include <linux/dma-resv.h>
#include "i915_gem_ww.h"
#include "gem/i915_gem_object.h"

void i915_gem_ww_ctx_init(struct i915_gem_ww_ctx *ww, bool intr)
{
        ww_acquire_init(&ww->ctx, &reservation_ww_class);
        INIT_LIST_HEAD(&ww->obj_list);
        ww->intr = intr;
        ww->contended = NULL;
}

static void i915_gem_ww_ctx_unlock_all(struct i915_gem_ww_ctx *ww)
{
        struct drm_i915_gem_object *obj;

        while ((obj = list_first_entry_or_null(&ww->obj_list, struct drm_i915_gem_object, obj_link))) {
                list_del(&obj->obj_link);
                i915_gem_object_unlock(obj);
                i915_gem_object_put(obj);
        }
}

void i915_gem_ww_unlock_single(struct drm_i915_gem_object *obj)
{
        list_del(&obj->obj_link);
        i915_gem_object_unlock(obj);
        i915_gem_object_put(obj);
}

void i915_gem_ww_ctx_fini(struct i915_gem_ww_ctx *ww)
{
        i915_gem_ww_ctx_unlock_all(ww);
        WARN_ON(ww->contended);
        ww_acquire_fini(&ww->ctx);
}

int __must_check i915_gem_ww_ctx_backoff(struct i915_gem_ww_ctx *ww)
{
        int ret = 0;

        if (WARN_ON(!ww->contended))
                return -EINVAL;

        i915_gem_ww_ctx_unlock_all(ww);
        if (ww->intr)
                ret = dma_resv_lock_slow_interruptible(ww->contended->base.resv, &ww->ctx);
        else
                dma_resv_lock_slow(ww->contended->base.resv, &ww->ctx);

        if (!ret)
                list_add_tail(&ww->contended->obj_link, &ww->obj_list);
        else
                i915_gem_object_put(ww->contended);

        ww->contended = NULL;

        return ret;
}