// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2013 Red Hat
 * Author: Rob Clark <robdclark@gmail.com>
 */

#include <linux/file.h>
#include <linux/sync_file.h>
#include <linux/uaccess.h>

#include <drm/drm_drv.h>
#include <drm/drm_file.h>
#include <drm/drm_syncobj.h>

#include "msm_drv.h"
#include "msm_gpu.h"
#include "msm_gem.h"
#include "msm_gpu_trace.h"

/*
 * Cmdstream submission:
 */

static struct msm_gem_submit *submit_create(struct drm_device *dev,
                struct msm_gpu *gpu,
                struct msm_gpu_submitqueue *queue, uint32_t nr_bos,
                uint32_t nr_cmds)
{
        struct msm_gem_submit *submit;
        uint64_t sz;
        int ret;

        sz = struct_size(submit, bos, nr_bos) +
                        ((u64)nr_cmds * sizeof(submit->cmd[0]));

        if (sz > SIZE_MAX)
                return ERR_PTR(-ENOMEM);

        submit = kzalloc(sz, GFP_KERNEL | __GFP_NOWARN | __GFP_NORETRY);
        if (!submit)
                return ERR_PTR(-ENOMEM);

        ret = drm_sched_job_init(&submit->base, queue->entity, queue);
        if (ret) {
                kfree(submit);
                return ERR_PTR(ret);
        }

        kref_init(&submit->ref);
        submit->dev = dev;
        submit->aspace = queue->ctx->aspace;
        submit->gpu = gpu;
        submit->cmd = (void *)&submit->bos[nr_bos];
        submit->queue = queue;
        submit->ring = gpu->rb[queue->ring_nr];
        submit->fault_dumped = false;

        INIT_LIST_HEAD(&submit->node);

        return submit;
}

void __msm_gem_submit_destroy(struct kref *kref)
{
        struct msm_gem_submit *submit =
                        container_of(kref, struct msm_gem_submit, ref);
        unsigned i;

        if (submit->fence_id) {
                mutex_lock(&submit->queue->lock);
                idr_remove(&submit->queue->fence_idr, submit->fence_id);
                mutex_unlock(&submit->queue->lock);
        }

        dma_fence_put(submit->user_fence);
        dma_fence_put(submit->hw_fence);

        put_pid(submit->pid);
        msm_submitqueue_put(submit->queue);

        for (i = 0; i < submit->nr_cmds; i++)
                kfree(submit->cmd[i].relocs);

        kfree(submit);
}

static int submit_lookup_objects(struct msm_gem_submit *submit,
                struct drm_msm_gem_submit *args, struct drm_file *file)
{
        unsigned i;
        int ret = 0;

        for (i = 0; i < args->nr_bos; i++) {
                struct drm_msm_gem_submit_bo submit_bo;
                void __user *userptr =
                        u64_to_user_ptr(args->bos + (i * sizeof(submit_bo)));

                /* make sure we don't have garbage flags, in case we hit
                 * error path before flags is initialized:
                 */
                submit->bos[i].flags = 0;

                if (copy_from_user(&submit_bo, userptr, sizeof(submit_bo))) {
                        ret = -EFAULT;
                        i = 0;
                        goto out;
                }

/* at least one of READ and/or WRITE flags should be set: */
#define MANDATORY_FLAGS (MSM_SUBMIT_BO_READ | MSM_SUBMIT_BO_WRITE)

                if ((submit_bo.flags & ~MSM_SUBMIT_BO_FLAGS) ||
                        !(submit_bo.flags & MANDATORY_FLAGS)) {
                        DRM_ERROR("invalid flags: %x\n", submit_bo.flags);
                        ret = -EINVAL;
                        i = 0;
                        goto out;
                }

                submit->bos[i].handle = submit_bo.handle;
                submit->bos[i].flags = submit_bo.flags;
                /* in validate_objects() we figure out if this is true: */
                submit->bos[i].iova  = submit_bo.presumed;
        }

        spin_lock(&file->table_lock);

        for (i = 0; i < args->nr_bos; i++) {
                struct drm_gem_object *obj;

                /* normally use drm_gem_object_lookup(), but for bulk lookup
                 * all under single table_lock just hit object_idr directly:
                 */
                obj = idr_find(&file->object_idr, submit->bos[i].handle);
                if (!obj) {
                        DRM_ERROR("invalid handle %u at index %u\n", submit->bos[i].handle, i);
                        ret = -EINVAL;
                        goto out_unlock;
                }

                drm_gem_object_get(obj);

                submit->bos[i].obj = to_msm_bo(obj);
        }

out_unlock:
        spin_unlock(&file->table_lock);

out:
        submit->nr_bos = i;

        return ret;
}

static int submit_lookup_cmds(struct msm_gem_submit *submit,
                struct drm_msm_gem_submit *args, struct drm_file *file)
{
        unsigned i;
        size_t sz;
        int ret = 0;

        for (i = 0; i < args->nr_cmds; i++) {
                struct drm_msm_gem_submit_cmd submit_cmd;
                void __user *userptr =
                        u64_to_user_ptr(args->cmds + (i * sizeof(submit_cmd)));

                ret = copy_from_user(&submit_cmd, userptr, sizeof(submit_cmd));
                if (ret) {
                        ret = -EFAULT;
                        goto out;
                }

                /* validate input from userspace: */
                switch (submit_cmd.type) {
                case MSM_SUBMIT_CMD_BUF:
                case MSM_SUBMIT_CMD_IB_TARGET_BUF:
                case MSM_SUBMIT_CMD_CTX_RESTORE_BUF:
                        break;
                default:
                        DRM_ERROR("invalid type: %08x\n", submit_cmd.type);
                        return -EINVAL;
                }

                if (submit_cmd.size % 4) {
                        DRM_ERROR("non-aligned cmdstream buffer size: %u\n",
                                        submit_cmd.size);
                        ret = -EINVAL;
                        goto out;
                }

                submit->cmd[i].type = submit_cmd.type;
                submit->cmd[i].size = submit_cmd.size / 4;
                submit->cmd[i].offset = submit_cmd.submit_offset / 4;
                submit->cmd[i].idx  = submit_cmd.submit_idx;
                submit->cmd[i].nr_relocs = submit_cmd.nr_relocs;

                userptr = u64_to_user_ptr(submit_cmd.relocs);

                sz = array_size(submit_cmd.nr_relocs,
                                sizeof(struct drm_msm_gem_submit_reloc));
                /* check for overflow: */
                if (sz == SIZE_MAX) {
                        ret = -ENOMEM;
                        goto out;
                }
                submit->cmd[i].relocs = kmalloc(sz, GFP_KERNEL);
                ret = copy_from_user(submit->cmd[i].relocs, userptr, sz);
                if (ret) {
                        ret = -EFAULT;
                        goto out;
                }
        }

out:
        return ret;
}

/* Unwind bo state, according to cleanup_flags.  In the success case, only
 * the lock is dropped at the end of the submit (and active/pin ref is dropped
 * later when the submit is retired).
 */
static void submit_cleanup_bo(struct msm_gem_submit *submit, int i,
                unsigned cleanup_flags)
{
        struct drm_gem_object *obj = &submit->bos[i].obj->base;
        unsigned flags = submit->bos[i].flags & cleanup_flags;

        /*
         * Clear flags bit before dropping lock, so that the msm_job_run()
         * path isn't racing with submit_cleanup() (ie. the read/modify/
         * write is protected by the obj lock in all paths)
         */
        submit->bos[i].flags &= ~cleanup_flags;

        if (flags & BO_VMA_PINNED)
                msm_gem_unpin_vma(submit->bos[i].vma);

        if (flags & BO_OBJ_PINNED)
                msm_gem_unpin_locked(obj);

        if (flags & BO_ACTIVE)
                msm_gem_active_put(obj);

        if (flags & BO_LOCKED)
                dma_resv_unlock(obj->resv);
}

static void submit_unlock_unpin_bo(struct msm_gem_submit *submit, int i)
{
        unsigned cleanup_flags = BO_VMA_PINNED | BO_OBJ_PINNED |
                                 BO_ACTIVE | BO_LOCKED;
        submit_cleanup_bo(submit, i, cleanup_flags);

        if (!(submit->bos[i].flags & BO_VALID))
                submit->bos[i].iova = 0;
}

/* This is where we make sure all the bo's are reserved and pin'd: */
static int submit_lock_objects(struct msm_gem_submit *submit)
{
        int contended, slow_locked = -1, i, ret = 0;

retry:
        for (i = 0; i < submit->nr_bos; i++) {
                struct msm_gem_object *msm_obj = submit->bos[i].obj;

                if (slow_locked == i)
                        slow_locked = -1;

                contended = i;

                if (!(submit->bos[i].flags & BO_LOCKED)) {
                        ret = dma_resv_lock_interruptible(msm_obj->base.resv,
                                                          &submit->ticket);
                        if (ret)
                                goto fail;
                        submit->bos[i].flags |= BO_LOCKED;
                }
        }

        ww_acquire_done(&submit->ticket);

        return 0;

fail:
        if (ret == -EALREADY) {
                DRM_ERROR("handle %u at index %u already on submit list\n",
                                submit->bos[i].handle, i);
                ret = -EINVAL;
        }

        for (; i >= 0; i--)
                submit_unlock_unpin_bo(submit, i);

        if (slow_locked > 0)
                submit_unlock_unpin_bo(submit, slow_locked);

        if (ret == -EDEADLK) {
                struct msm_gem_object *msm_obj = submit->bos[contended].obj;
                /* we lost out in a seqno race, lock and retry.. */
                ret = dma_resv_lock_slow_interruptible(msm_obj->base.resv,
                                                       &submit->ticket);
                if (!ret) {
                        submit->bos[contended].flags |= BO_LOCKED;
                        slow_locked = contended;
                        goto retry;
                }

                /* Not expecting -EALREADY here, if the bo was already
                 * locked, we should have gotten -EALREADY already from
                 * the dma_resv_lock_interruptable() call.
                 */
                WARN_ON_ONCE(ret == -EALREADY);
        }

        return ret;
}

static int submit_fence_sync(struct msm_gem_submit *submit, bool no_implicit)
{
        int i, ret = 0;

        for (i = 0; i < submit->nr_bos; i++) {
                struct drm_gem_object *obj = &submit->bos[i].obj->base;
                bool write = submit->bos[i].flags & MSM_SUBMIT_BO_WRITE;

                /* NOTE: _reserve_shared() must happen before
                 * _add_shared_fence(), which makes this a slightly
                 * strange place to call it.  OTOH this is a
                 * convenient can-fail point to hook it in.
                 */
                ret = dma_resv_reserve_fences(obj->resv, 1);
                if (ret)
                        return ret;

                /* exclusive fences must be ordered */
                if (no_implicit && !write)
                        continue;

                ret = drm_sched_job_add_implicit_dependencies(&submit->base,
                                                              obj,
                                                              write);
                if (ret)
                        break;
        }

        return ret;
}

static int submit_pin_objects(struct msm_gem_submit *submit)
{
        int i, ret = 0;

        submit->valid = true;

        /*
         * Increment active_count first, so if under memory pressure, we
         * don't inadvertently evict a bo needed by the submit in order
         * to pin an earlier bo in the same submit.
         */
        for (i = 0; i < submit->nr_bos; i++) {
                struct drm_gem_object *obj = &submit->bos[i].obj->base;

                msm_gem_active_get(obj, submit->gpu);
                submit->bos[i].flags |= BO_ACTIVE;
        }

        for (i = 0; i < submit->nr_bos; i++) {
                struct drm_gem_object *obj = &submit->bos[i].obj->base;
                struct msm_gem_vma *vma;

                /* if locking succeeded, pin bo: */
                vma = msm_gem_get_vma_locked(obj, submit->aspace);
                if (IS_ERR(vma)) {
                        ret = PTR_ERR(vma);
                        break;
                }

                ret = msm_gem_pin_vma_locked(obj, vma);
                if (ret)
                        break;

                submit->bos[i].flags |= BO_OBJ_PINNED | BO_VMA_PINNED;
                submit->bos[i].vma = vma;

                if (vma->iova == submit->bos[i].iova) {
                        submit->bos[i].flags |= BO_VALID;
                } else {
                        submit->bos[i].iova = vma->iova;
                        /* iova changed, so address in cmdstream is not valid: */
                        submit->bos[i].flags &= ~BO_VALID;
                        submit->valid = false;
                }
        }

        return ret;
}

static void submit_attach_object_fences(struct msm_gem_submit *submit)
{
        int i;

        for (i = 0; i < submit->nr_bos; i++) {
                struct drm_gem_object *obj = &submit->bos[i].obj->base;

                if (submit->bos[i].flags & MSM_SUBMIT_BO_WRITE)
                        dma_resv_add_fence(obj->resv, submit->user_fence,
                                           DMA_RESV_USAGE_WRITE);
                else if (submit->bos[i].flags & MSM_SUBMIT_BO_READ)
                        dma_resv_add_fence(obj->resv, submit->user_fence,
                                           DMA_RESV_USAGE_READ);
        }
}

static int submit_bo(struct msm_gem_submit *submit, uint32_t idx,
                struct msm_gem_object **obj, uint64_t *iova, bool *valid)
{
        if (idx >= submit->nr_bos) {
                DRM_ERROR("invalid buffer index: %u (out of %u)\n",
                                idx, submit->nr_bos);
                return -EINVAL;
        }

        if (obj)
                *obj = submit->bos[idx].obj;
        if (iova)
                *iova = submit->bos[idx].iova;
        if (valid)
                *valid = !!(submit->bos[idx].flags & BO_VALID);

        return 0;
}

/* process the reloc's and patch up the cmdstream as needed: */
static int submit_reloc(struct msm_gem_submit *submit, struct msm_gem_object *obj,
                uint32_t offset, uint32_t nr_relocs, struct drm_msm_gem_submit_reloc *relocs)
{
        uint32_t i, last_offset = 0;
        uint32_t *ptr;
        int ret = 0;

        if (!nr_relocs)
                return 0;

        if (offset % 4) {
                DRM_ERROR("non-aligned cmdstream buffer: %u\n", offset);
                return -EINVAL;
        }

        /* For now, just map the entire thing.  Eventually we probably
         * to do it page-by-page, w/ kmap() if not vmap()d..
         */
        ptr = msm_gem_get_vaddr_locked(&obj->base);

        if (IS_ERR(ptr)) {
                ret = PTR_ERR(ptr);
                DBG("failed to map: %d", ret);
                return ret;
        }

        for (i = 0; i < nr_relocs; i++) {
                struct drm_msm_gem_submit_reloc submit_reloc = relocs[i];
                uint32_t off;
                uint64_t iova;
                bool valid;

                if (submit_reloc.submit_offset % 4) {
                        DRM_ERROR("non-aligned reloc offset: %u\n",
                                        submit_reloc.submit_offset);
                        ret = -EINVAL;
                        goto out;
                }

                /* offset in dwords: */
                off = submit_reloc.submit_offset / 4;

                if ((off >= (obj->base.size / 4)) ||
                                (off < last_offset)) {
                        DRM_ERROR("invalid offset %u at reloc %u\n", off, i);
                        ret = -EINVAL;
                        goto out;
                }

                ret = submit_bo(submit, submit_reloc.reloc_idx, NULL, &iova, &valid);
                if (ret)
                        goto out;

                if (valid)
                        continue;

                iova += submit_reloc.reloc_offset;

                if (submit_reloc.shift < 0)
                        iova >>= -submit_reloc.shift;
                else
                        iova <<= submit_reloc.shift;

                ptr[off] = iova | submit_reloc.or;

                last_offset = off;
        }

out:
        msm_gem_put_vaddr_locked(&obj->base);

        return ret;
}

/* Cleanup submit at end of ioctl.  In the error case, this also drops
 * references, unpins, and drops active refcnt.  In the non-error case,
 * this is done when the submit is retired.
 */
static void submit_cleanup(struct msm_gem_submit *submit, bool error)
{
        unsigned cleanup_flags = BO_LOCKED;
        unsigned i;

        if (error)
                cleanup_flags |= BO_VMA_PINNED | BO_OBJ_PINNED | BO_ACTIVE;

        for (i = 0; i < submit->nr_bos; i++) {
                struct msm_gem_object *msm_obj = submit->bos[i].obj;
                submit_cleanup_bo(submit, i, cleanup_flags);
                if (error)
                        drm_gem_object_put(&msm_obj->base);
        }
}

void msm_submit_retire(struct msm_gem_submit *submit)
{
        int i;

        for (i = 0; i < submit->nr_bos; i++) {
                struct drm_gem_object *obj = &submit->bos[i].obj->base;

                msm_gem_lock(obj);
                /* Note, VMA already fence-unpinned before submit: */
                submit_cleanup_bo(submit, i, BO_OBJ_PINNED | BO_ACTIVE);
                msm_gem_unlock(obj);
                drm_gem_object_put(obj);
        }
}

struct msm_submit_post_dep {
        struct drm_syncobj *syncobj;
        uint64_t point;
        struct dma_fence_chain *chain;
};

static struct drm_syncobj **msm_parse_deps(struct msm_gem_submit *submit,
                                           struct drm_file *file,
                                           uint64_t in_syncobjs_addr,
                                           uint32_t nr_in_syncobjs,
                                           size_t syncobj_stride,
                                           struct msm_ringbuffer *ring)
{
        struct drm_syncobj **syncobjs = NULL;
        struct drm_msm_gem_submit_syncobj syncobj_desc = {0};
        int ret = 0;
        uint32_t i, j;

        syncobjs = kcalloc(nr_in_syncobjs, sizeof(*syncobjs),
                           GFP_KERNEL | __GFP_NOWARN | __GFP_NORETRY);
        if (!syncobjs)
                return ERR_PTR(-ENOMEM);

        for (i = 0; i < nr_in_syncobjs; ++i) {
                uint64_t address = in_syncobjs_addr + i * syncobj_stride;
                struct dma_fence *fence;

                if (copy_from_user(&syncobj_desc,
                                   u64_to_user_ptr(address),
                                   min(syncobj_stride, sizeof(syncobj_desc)))) {
                        ret = -EFAULT;
                        break;
                }

                if (syncobj_desc.point &&
                    !drm_core_check_feature(submit->dev, DRIVER_SYNCOBJ_TIMELINE)) {
                        ret = -EOPNOTSUPP;
                        break;
                }

                if (syncobj_desc.flags & ~MSM_SUBMIT_SYNCOBJ_FLAGS) {
                        ret = -EINVAL;
                        break;
                }

                ret = drm_syncobj_find_fence(file, syncobj_desc.handle,
                                             syncobj_desc.point, 0, &fence);
                if (ret)
                        break;

                ret = drm_sched_job_add_dependency(&submit->base, fence);
                if (ret)
                        break;

                if (syncobj_desc.flags & MSM_SUBMIT_SYNCOBJ_RESET) {
                        syncobjs[i] =
                                drm_syncobj_find(file, syncobj_desc.handle);
                        if (!syncobjs[i]) {
                                ret = -EINVAL;
                                break;
                        }
                }
        }

        if (ret) {
                for (j = 0; j <= i; ++j) {
                        if (syncobjs[j])
                                drm_syncobj_put(syncobjs[j]);
                }
                kfree(syncobjs);
                return ERR_PTR(ret);
        }
        return syncobjs;
}

static void msm_reset_syncobjs(struct drm_syncobj **syncobjs,
                               uint32_t nr_syncobjs)
{
        uint32_t i;

        for (i = 0; syncobjs && i < nr_syncobjs; ++i) {
                if (syncobjs[i])
                        drm_syncobj_replace_fence(syncobjs[i], NULL);
        }
}

static struct msm_submit_post_dep *msm_parse_post_deps(struct drm_device *dev,
                                                       struct drm_file *file,
                                                       uint64_t syncobjs_addr,
                                                       uint32_t nr_syncobjs,
                                                       size_t syncobj_stride)
{
        struct msm_submit_post_dep *post_deps;
        struct drm_msm_gem_submit_syncobj syncobj_desc = {0};
        int ret = 0;
        uint32_t i, j;

        post_deps = kmalloc_array(nr_syncobjs, sizeof(*post_deps),
                                  GFP_KERNEL | __GFP_NOWARN | __GFP_NORETRY);
        if (!post_deps)
                return ERR_PTR(-ENOMEM);

        for (i = 0; i < nr_syncobjs; ++i) {
                uint64_t address = syncobjs_addr + i * syncobj_stride;

                if (copy_from_user(&syncobj_desc,
                                   u64_to_user_ptr(address),
                                   min(syncobj_stride, sizeof(syncobj_desc)))) {
                        ret = -EFAULT;
                        break;
                }

                post_deps[i].point = syncobj_desc.point;
                post_deps[i].chain = NULL;

                if (syncobj_desc.flags) {
                        ret = -EINVAL;
                        break;
                }

                if (syncobj_desc.point) {
                        if (!drm_core_check_feature(dev,
                                                    DRIVER_SYNCOBJ_TIMELINE)) {
                                ret = -EOPNOTSUPP;
                                break;
                        }

                        post_deps[i].chain = dma_fence_chain_alloc();
                        if (!post_deps[i].chain) {
                                ret = -ENOMEM;
                                break;
                        }
                }

                post_deps[i].syncobj =
                        drm_syncobj_find(file, syncobj_desc.handle);
                if (!post_deps[i].syncobj) {
                        ret = -EINVAL;
                        break;
                }
        }

        if (ret) {
                for (j = 0; j <= i; ++j) {
                        dma_fence_chain_free(post_deps[j].chain);
                        if (post_deps[j].syncobj)
                                drm_syncobj_put(post_deps[j].syncobj);
                }

                kfree(post_deps);
                return ERR_PTR(ret);
        }

        return post_deps;
}

static void msm_process_post_deps(struct msm_submit_post_dep *post_deps,
                                  uint32_t count, struct dma_fence *fence)
{
        uint32_t i;

        for (i = 0; post_deps && i < count; ++i) {
                if (post_deps[i].chain) {
                        drm_syncobj_add_point(post_deps[i].syncobj,
                                              post_deps[i].chain,
                                              fence, post_deps[i].point);
                        post_deps[i].chain = NULL;
                } else {
                        drm_syncobj_replace_fence(post_deps[i].syncobj,
                                                  fence);
                }
        }
}

int msm_ioctl_gem_submit(struct drm_device *dev, void *data,
                struct drm_file *file)
{
        static atomic_t ident = ATOMIC_INIT(0);
        struct msm_drm_private *priv = dev->dev_private;
        struct drm_msm_gem_submit *args = data;
        struct msm_file_private *ctx = file->driver_priv;
        struct msm_gem_submit *submit = NULL;
        struct msm_gpu *gpu = priv->gpu;
        struct msm_gpu_submitqueue *queue;
        struct msm_ringbuffer *ring;
        struct msm_submit_post_dep *post_deps = NULL;
        struct drm_syncobj **syncobjs_to_reset = NULL;
        int out_fence_fd = -1;
        struct pid *pid = get_pid(task_pid(current));
        bool has_ww_ticket = false;
        unsigned i;
        int ret, submitid;

        if (!gpu)
                return -ENXIO;

        if (args->pad)
                return -EINVAL;

        if (unlikely(!ctx->aspace) && !capable(CAP_SYS_RAWIO)) {
                DRM_ERROR_RATELIMITED("IOMMU support or CAP_SYS_RAWIO required!\n");
                return -EPERM;
        }

        /* for now, we just have 3d pipe.. eventually this would need to
         * be more clever to dispatch to appropriate gpu module:
         */
        if (MSM_PIPE_ID(args->flags) != MSM_PIPE_3D0)
                return -EINVAL;

        if (MSM_PIPE_FLAGS(args->flags) & ~MSM_SUBMIT_FLAGS)
                return -EINVAL;

        if (args->flags & MSM_SUBMIT_SUDO) {
                if (!IS_ENABLED(CONFIG_DRM_MSM_GPU_SUDO) ||
                    !capable(CAP_SYS_RAWIO))
                        return -EINVAL;
        }

        queue = msm_submitqueue_get(ctx, args->queueid);
        if (!queue)
                return -ENOENT;

        /* Get a unique identifier for the submission for logging purposes */
        submitid = atomic_inc_return(&ident) - 1;

        ring = gpu->rb[queue->ring_nr];
        trace_msm_gpu_submit(pid_nr(pid), ring->id, submitid,
                args->nr_bos, args->nr_cmds);

        ret = mutex_lock_interruptible(&queue->lock);
        if (ret)
                goto out_post_unlock;

        if (args->flags & MSM_SUBMIT_FENCE_FD_OUT) {
                out_fence_fd = get_unused_fd_flags(O_CLOEXEC);
                if (out_fence_fd < 0) {
                        ret = out_fence_fd;
                        goto out_unlock;
                }
        }

        submit = submit_create(dev, gpu, queue, args->nr_bos,
                args->nr_cmds);
        if (IS_ERR(submit)) {
                ret = PTR_ERR(submit);
                submit = NULL;
                goto out_unlock;
        }

        submit->pid = pid;
        submit->ident = submitid;

        if (args->flags & MSM_SUBMIT_SUDO)
                submit->in_rb = true;

        if (args->flags & MSM_SUBMIT_FENCE_FD_IN) {
                struct dma_fence *in_fence;

                in_fence = sync_file_get_fence(args->fence_fd);

                if (!in_fence) {
                        ret = -EINVAL;
                        goto out_unlock;
                }

                ret = drm_sched_job_add_dependency(&submit->base, in_fence);
                if (ret)
                        goto out_unlock;
        }

        if (args->flags & MSM_SUBMIT_SYNCOBJ_IN) {
                syncobjs_to_reset = msm_parse_deps(submit, file,
                                                   args->in_syncobjs,
                                                   args->nr_in_syncobjs,
                                                   args->syncobj_stride, ring);
                if (IS_ERR(syncobjs_to_reset)) {
                        ret = PTR_ERR(syncobjs_to_reset);
                        goto out_unlock;
                }
        }

        if (args->flags & MSM_SUBMIT_SYNCOBJ_OUT) {
                post_deps = msm_parse_post_deps(dev, file,
                                                args->out_syncobjs,
                                                args->nr_out_syncobjs,
                                                args->syncobj_stride);
                if (IS_ERR(post_deps)) {
                        ret = PTR_ERR(post_deps);
                        goto out_unlock;
                }
        }

        ret = submit_lookup_objects(submit, args, file);
        if (ret)
                goto out;

        ret = submit_lookup_cmds(submit, args, file);
        if (ret)
                goto out;

        /* copy_*_user while holding a ww ticket upsets lockdep */
        ww_acquire_init(&submit->ticket, &reservation_ww_class);
        has_ww_ticket = true;
        ret = submit_lock_objects(submit);
        if (ret)
                goto out;

        ret = submit_fence_sync(submit, !!(args->flags & MSM_SUBMIT_NO_IMPLICIT));
        if (ret)
                goto out;

        ret = submit_pin_objects(submit);
        if (ret)
                goto out;

        for (i = 0; i < args->nr_cmds; i++) {
                struct msm_gem_object *msm_obj;
                uint64_t iova;

                ret = submit_bo(submit, submit->cmd[i].idx,
                                &msm_obj, &iova, NULL);
                if (ret)
                        goto out;

                if (!submit->cmd[i].size ||
                        ((submit->cmd[i].size + submit->cmd[i].offset) >
                                msm_obj->base.size / 4)) {
                        DRM_ERROR("invalid cmdstream size: %u\n", submit->cmd[i].size * 4);
                        ret = -EINVAL;
                        goto out;
                }

                submit->cmd[i].iova = iova + (submit->cmd[i].offset * 4);

                if (submit->valid)
                        continue;

                ret = submit_reloc(submit, msm_obj, submit->cmd[i].offset * 4,
                                submit->cmd[i].nr_relocs, submit->cmd[i].relocs);
                if (ret)
                        goto out;
        }

        submit->nr_cmds = i;

        /*
         * If using userspace provided seqno fence, validate that the id
         * is available before arming sched job.  Since access to fence_idr
         * is serialized on the queue lock, the slot should be still avail
         * after the job is armed
         */
        if ((args->flags & MSM_SUBMIT_FENCE_SN_IN) &&
                        idr_find(&queue->fence_idr, args->fence)) {
                ret = -EINVAL;
                goto out;
        }

        drm_sched_job_arm(&submit->base);

        submit->user_fence = dma_fence_get(&submit->base.s_fence->finished);

        if (args->flags & MSM_SUBMIT_FENCE_SN_IN) {
                /*
                 * Userspace has assigned the seqno fence that it wants
                 * us to use.  It is an error to pick a fence sequence
                 * number that is not available.
                 */
                submit->fence_id = args->fence;
                ret = idr_alloc_u32(&queue->fence_idr, submit->user_fence,
                                    &submit->fence_id, submit->fence_id,
                                    GFP_KERNEL);
                /*
                 * We've already validated that the fence_id slot is valid,
                 * so if idr_alloc_u32 failed, it is a kernel bug
                 */
                WARN_ON(ret);
        } else {
                /*
                 * Allocate an id which can be used by WAIT_FENCE ioctl to map
                 * back to the underlying fence.
                 */
                submit->fence_id = idr_alloc_cyclic(&queue->fence_idr,
                                                    submit->user_fence, 1,
                                                    INT_MAX, GFP_KERNEL);
        }
        if (submit->fence_id < 0) {
                ret = submit->fence_id;
                submit->fence_id = 0;
        }

        if (ret == 0 && args->flags & MSM_SUBMIT_FENCE_FD_OUT) {
                struct sync_file *sync_file = sync_file_create(submit->user_fence);
                if (!sync_file) {
                        ret = -ENOMEM;
                } else {
                        fd_install(out_fence_fd, sync_file->file);
                        args->fence_fd = out_fence_fd;
                }
        }

        submit_attach_object_fences(submit);

        /* The scheduler owns a ref now: */
        msm_gem_submit_get(submit);

        drm_sched_entity_push_job(&submit->base);

        args->fence = submit->fence_id;
        queue->last_fence = submit->fence_id;

        msm_reset_syncobjs(syncobjs_to_reset, args->nr_in_syncobjs);
        msm_process_post_deps(post_deps, args->nr_out_syncobjs,
                              submit->user_fence);


out:
        submit_cleanup(submit, !!ret);
        if (has_ww_ticket)
                ww_acquire_fini(&submit->ticket);
out_unlock:
        if (ret && (out_fence_fd >= 0))
                put_unused_fd(out_fence_fd);
        mutex_unlock(&queue->lock);
        if (submit)
                msm_gem_submit_put(submit);
out_post_unlock:
        if (!IS_ERR_OR_NULL(post_deps)) {
                for (i = 0; i < args->nr_out_syncobjs; ++i) {
                        kfree(post_deps[i].chain);
                        drm_syncobj_put(post_deps[i].syncobj);
                }
                kfree(post_deps);
        }

        if (!IS_ERR_OR_NULL(syncobjs_to_reset)) {
                for (i = 0; i < args->nr_in_syncobjs; ++i) {
                        if (syncobjs_to_reset[i])
                                drm_syncobj_put(syncobjs_to_reset[i]);
                }
                kfree(syncobjs_to_reset);
        }

        return ret;
}