/*
 * Copyright 2013 Advanced Micro Devices, Inc.
 * All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sub license, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
 * USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 * The above copyright notice and this permission notice (including the
 * next paragraph) shall be included in all copies or substantial portions
 * of the Software.
 *
 * Authors: Christian König <christian.koenig@amd.com>
 */

#include <linux/firmware.h>
#include <linux/module.h>

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

#include "amdgpu.h"
#include "amdgpu_pm.h"
#include "amdgpu_vce.h"
#include "amdgpu_cs.h"
#include "cikd.h"

/* 1 second timeout */
#define VCE_IDLE_TIMEOUT        msecs_to_jiffies(1000)

/* Firmware Names */
#ifdef CONFIG_DRM_AMDGPU_SI
#define FIRMWARE_VCE_V1_0       "amdgpu/vce_1_0_0.bin"
#endif
#ifdef CONFIG_DRM_AMDGPU_CIK
#define FIRMWARE_BONAIRE        "amdgpu/bonaire_vce.bin"
#define FIRMWARE_KABINI "amdgpu/kabini_vce.bin"
#define FIRMWARE_KAVERI "amdgpu/kaveri_vce.bin"
#define FIRMWARE_HAWAII "amdgpu/hawaii_vce.bin"
#define FIRMWARE_MULLINS        "amdgpu/mullins_vce.bin"
#endif
#define FIRMWARE_TONGA          "amdgpu/tonga_vce.bin"
#define FIRMWARE_CARRIZO        "amdgpu/carrizo_vce.bin"
#define FIRMWARE_FIJI           "amdgpu/fiji_vce.bin"
#define FIRMWARE_STONEY         "amdgpu/stoney_vce.bin"
#define FIRMWARE_POLARIS10      "amdgpu/polaris10_vce.bin"
#define FIRMWARE_POLARIS11      "amdgpu/polaris11_vce.bin"
#define FIRMWARE_POLARIS12      "amdgpu/polaris12_vce.bin"
#define FIRMWARE_VEGAM          "amdgpu/vegam_vce.bin"

#define FIRMWARE_VEGA10         "amdgpu/vega10_vce.bin"
#define FIRMWARE_VEGA12         "amdgpu/vega12_vce.bin"
#define FIRMWARE_VEGA20         "amdgpu/vega20_vce.bin"

#ifdef CONFIG_DRM_AMDGPU_SI
MODULE_FIRMWARE(FIRMWARE_VCE_V1_0);
#endif
#ifdef CONFIG_DRM_AMDGPU_CIK
MODULE_FIRMWARE(FIRMWARE_BONAIRE);
MODULE_FIRMWARE(FIRMWARE_KABINI);
MODULE_FIRMWARE(FIRMWARE_KAVERI);
MODULE_FIRMWARE(FIRMWARE_HAWAII);
MODULE_FIRMWARE(FIRMWARE_MULLINS);
#endif
MODULE_FIRMWARE(FIRMWARE_TONGA);
MODULE_FIRMWARE(FIRMWARE_CARRIZO);
MODULE_FIRMWARE(FIRMWARE_FIJI);
MODULE_FIRMWARE(FIRMWARE_STONEY);
MODULE_FIRMWARE(FIRMWARE_POLARIS10);
MODULE_FIRMWARE(FIRMWARE_POLARIS11);
MODULE_FIRMWARE(FIRMWARE_POLARIS12);
MODULE_FIRMWARE(FIRMWARE_VEGAM);

MODULE_FIRMWARE(FIRMWARE_VEGA10);
MODULE_FIRMWARE(FIRMWARE_VEGA12);
MODULE_FIRMWARE(FIRMWARE_VEGA20);

static void amdgpu_vce_idle_work_handler(struct work_struct *work);
static int amdgpu_vce_get_create_msg(struct amdgpu_ring *ring, uint32_t handle,
                                     struct dma_fence **fence);
static int amdgpu_vce_get_destroy_msg(struct amdgpu_ring *ring, uint32_t handle,
                                      bool direct, struct dma_fence **fence);

/**
 * amdgpu_vce_firmware_name() - determine the firmware file name for VCE
 *
 * @adev: amdgpu_device pointer
 *
 * Each chip that has VCE IP may need a different firmware.
 * This function returns the name of the VCE firmware file
 * appropriate for the current chip.
 */
static const char *amdgpu_vce_firmware_name(struct amdgpu_device *adev)
{
        switch (adev->asic_type) {
#ifdef CONFIG_DRM_AMDGPU_SI
        case CHIP_PITCAIRN:
        case CHIP_TAHITI:
        case CHIP_VERDE:
                return FIRMWARE_VCE_V1_0;
#endif
#ifdef CONFIG_DRM_AMDGPU_CIK
        case CHIP_BONAIRE:
                return FIRMWARE_BONAIRE;
        case CHIP_KAVERI:
                return FIRMWARE_KAVERI;
        case CHIP_KABINI:
                return FIRMWARE_KABINI;
        case CHIP_HAWAII:
                return FIRMWARE_HAWAII;
        case CHIP_MULLINS:
                return FIRMWARE_MULLINS;
#endif
        case CHIP_TONGA:
                return  FIRMWARE_TONGA;
        case CHIP_CARRIZO:
                return  FIRMWARE_CARRIZO;
        case CHIP_FIJI:
                return  FIRMWARE_FIJI;
        case CHIP_STONEY:
                return  FIRMWARE_STONEY;
        case CHIP_POLARIS10:
                return  FIRMWARE_POLARIS10;
        case CHIP_POLARIS11:
                return  FIRMWARE_POLARIS11;
        case CHIP_POLARIS12:
                return  FIRMWARE_POLARIS12;
        case CHIP_VEGAM:
                return  FIRMWARE_VEGAM;
        case CHIP_VEGA10:
                return  FIRMWARE_VEGA10;
        case CHIP_VEGA12:
                return  FIRMWARE_VEGA12;
        case CHIP_VEGA20:
                return  FIRMWARE_VEGA20;

        default:
                return NULL;
        }
}

/**
 * amdgpu_vce_early_init() - try to load VCE firmware
 *
 * @adev: amdgpu_device pointer
 *
 * Tries to load the VCE firmware.
 *
 * When not found, returns ENOENT so that the driver can
 * still load and initialize the rest of the IP blocks.
 * The GPU can function just fine without VCE, they will just
 * not support video encoding.
 */
int amdgpu_vce_early_init(struct amdgpu_device *adev)
{
        const char *fw_name = amdgpu_vce_firmware_name(adev);
        const struct common_firmware_header *hdr;
        unsigned int ucode_version, version_major, version_minor, binary_id;
        int r;

        if (!fw_name)
                return -ENOENT;

        r = amdgpu_ucode_request(adev, &adev->vce.fw, AMDGPU_UCODE_REQUIRED, "%s", fw_name);
        if (r) {
                dev_err(adev->dev,
                        "amdgpu_vce: Firmware \"%s\" not found or failed to validate (%d)\n",
                        fw_name, r);

                amdgpu_ucode_release(&adev->vce.fw);
                return -ENOENT;
        }

        hdr = (const struct common_firmware_header *)adev->vce.fw->data;

        ucode_version = le32_to_cpu(hdr->ucode_version);
        version_major = (ucode_version >> 20) & 0xfff;
        version_minor = (ucode_version >> 8) & 0xfff;
        binary_id = ucode_version & 0xff;
        dev_info(adev->dev, "Found VCE firmware Version: %d.%d Binary ID: %d\n",
                version_major, version_minor, binary_id);
        adev->vce.fw_version = ((version_major << 24) | (version_minor << 16) |
                                (binary_id << 8));

        return 0;
}

/**
 * amdgpu_vce_sw_init() - allocate memory for VCE BO
 *
 * @adev: amdgpu_device pointer
 * @size: size for the new BO
 *
 * First step to get VCE online: allocate memory for VCE BO.
 * The VCE firmware binary is copied into the VCE BO later,
 * in amdgpu_vce_resume. The VCE executes its code from the
 * VCE BO and also uses the space in this BO for its stack and data.
 *
 * Ideally this BO should be placed in VRAM for optimal performance,
 * although technically it also runs from system RAM (albeit slowly).
 */
int amdgpu_vce_sw_init(struct amdgpu_device *adev, unsigned long size)
{
        int i, r;

        if (!adev->vce.fw)
                return -ENOENT;

        r = amdgpu_bo_create_kernel(adev, size, PAGE_SIZE,
                                    AMDGPU_GEM_DOMAIN_VRAM |
                                    AMDGPU_GEM_DOMAIN_GTT,
                                    &adev->vce.vcpu_bo,
                                    &adev->vce.gpu_addr, &adev->vce.cpu_addr);
        if (r) {
                dev_err(adev->dev, "(%d) failed to allocate VCE bo\n", r);
                return r;
        }

        for (i = 0; i < AMDGPU_MAX_VCE_HANDLES; ++i) {
                atomic_set(&adev->vce.handles[i], 0);
                adev->vce.filp[i] = NULL;
        }

        INIT_DELAYED_WORK(&adev->vce.idle_work, amdgpu_vce_idle_work_handler);
        mutex_init(&adev->vce.idle_mutex);

        return 0;
}

/**
 * amdgpu_vce_sw_fini - free memory
 *
 * @adev: amdgpu_device pointer
 *
 * Last step on VCE teardown, free firmware memory
 */
int amdgpu_vce_sw_fini(struct amdgpu_device *adev)
{
        unsigned int i;

        if (adev->vce.vcpu_bo == NULL)
                return 0;

        drm_sched_entity_destroy(&adev->vce.entity);

        for (i = 0; i < adev->vce.num_rings; i++)
                amdgpu_ring_fini(&adev->vce.ring[i]);

        amdgpu_ucode_release(&adev->vce.fw);
        mutex_destroy(&adev->vce.idle_mutex);

        amdgpu_bo_free_kernel(&adev->vce.vcpu_bo, &adev->vce.gpu_addr,
                (void **)&adev->vce.cpu_addr);

        return 0;
}

/**
 * amdgpu_vce_entity_init - init entity
 *
 * @adev: amdgpu_device pointer
 * @ring: amdgpu_ring pointer to check
 *
 * Initialize the entity used for handle management in the kernel driver.
 */
int amdgpu_vce_entity_init(struct amdgpu_device *adev, struct amdgpu_ring *ring)
{
        if (ring == &adev->vce.ring[0]) {
                struct drm_gpu_scheduler *sched = &ring->sched;
                int r;

                r = drm_sched_entity_init(&adev->vce.entity, DRM_SCHED_PRIORITY_NORMAL,
                                          &sched, 1, NULL);
                if (r != 0) {
                        DRM_ERROR("Failed setting up VCE run queue.\n");
                        return r;
                }
        }

        return 0;
}

/**
 * amdgpu_vce_suspend - unpin VCE fw memory
 *
 * @adev: amdgpu_device pointer
 *
 */
int amdgpu_vce_suspend(struct amdgpu_device *adev)
{
        int i;

        cancel_delayed_work_sync(&adev->vce.idle_work);

        if (adev->vce.vcpu_bo == NULL)
                return 0;

        for (i = 0; i < AMDGPU_MAX_VCE_HANDLES; ++i)
                if (atomic_read(&adev->vce.handles[i]))
                        break;

        if (i == AMDGPU_MAX_VCE_HANDLES)
                return 0;

        /* TODO: suspending running encoding sessions isn't supported */
        return -EINVAL;
}

/**
 * amdgpu_vce_resume - pin VCE fw memory
 *
 * @adev: amdgpu_device pointer
 *
 */
int amdgpu_vce_resume(struct amdgpu_device *adev)
{
        const struct common_firmware_header *hdr;
        unsigned int offset;
        int idx;

        if (adev->vce.vcpu_bo == NULL)
                return -EINVAL;

        hdr = (const struct common_firmware_header *)adev->vce.fw->data;
        offset = le32_to_cpu(hdr->ucode_array_offset_bytes);

        if (drm_dev_enter(adev_to_drm(adev), &idx)) {
                memset_io(adev->vce.cpu_addr, 0, amdgpu_bo_size(adev->vce.vcpu_bo));
                memcpy_toio(adev->vce.cpu_addr, adev->vce.fw->data + offset,
                            adev->vce.fw->size - offset);
                drm_dev_exit(idx);
        }

        return 0;
}

/**
 * amdgpu_vce_idle_work_handler - power off VCE
 *
 * @work: pointer to work structure
 *
 * power of VCE when it's not used any more
 */
static void amdgpu_vce_idle_work_handler(struct work_struct *work)
{
        struct amdgpu_device *adev =
                container_of(work, struct amdgpu_device, vce.idle_work.work);
        unsigned int i, count = 0;

        for (i = 0; i < adev->vce.num_rings; i++)
                count += amdgpu_fence_count_emitted(&adev->vce.ring[i]);

        if (count == 0) {
                if (adev->pm.dpm_enabled) {
                        amdgpu_dpm_enable_vce(adev, false);
                } else {
                        amdgpu_asic_set_vce_clocks(adev, 0, 0);
                        amdgpu_device_ip_set_powergating_state(adev, AMD_IP_BLOCK_TYPE_VCE,
                                                               AMD_PG_STATE_GATE);
                        amdgpu_device_ip_set_clockgating_state(adev, AMD_IP_BLOCK_TYPE_VCE,
                                                               AMD_CG_STATE_GATE);
                }
        } else {
                schedule_delayed_work(&adev->vce.idle_work, VCE_IDLE_TIMEOUT);
        }
}

/**
 * amdgpu_vce_ring_begin_use - power up VCE
 *
 * @ring: amdgpu ring
 *
 * Make sure VCE is powerd up when we want to use it
 */
void amdgpu_vce_ring_begin_use(struct amdgpu_ring *ring)
{
        struct amdgpu_device *adev = ring->adev;
        bool set_clocks;

        if (amdgpu_sriov_vf(adev))
                return;

        mutex_lock(&adev->vce.idle_mutex);
        set_clocks = !cancel_delayed_work_sync(&adev->vce.idle_work);
        if (set_clocks) {
                if (adev->pm.dpm_enabled) {
                        amdgpu_dpm_enable_vce(adev, true);
                } else {
                        amdgpu_asic_set_vce_clocks(adev, 53300, 40000);
                        amdgpu_device_ip_set_clockgating_state(adev, AMD_IP_BLOCK_TYPE_VCE,
                                                               AMD_CG_STATE_UNGATE);
                        amdgpu_device_ip_set_powergating_state(adev, AMD_IP_BLOCK_TYPE_VCE,
                                                               AMD_PG_STATE_UNGATE);

                }
        }
        mutex_unlock(&adev->vce.idle_mutex);
}

/**
 * amdgpu_vce_ring_end_use - power VCE down
 *
 * @ring: amdgpu ring
 *
 * Schedule work to power VCE down again
 */
void amdgpu_vce_ring_end_use(struct amdgpu_ring *ring)
{
        if (!amdgpu_sriov_vf(ring->adev))
                schedule_delayed_work(&ring->adev->vce.idle_work, VCE_IDLE_TIMEOUT);
}

/**
 * amdgpu_vce_free_handles - free still open VCE handles
 *
 * @adev: amdgpu_device pointer
 * @filp: drm file pointer
 *
 * Close all VCE handles still open by this file pointer
 */
void amdgpu_vce_free_handles(struct amdgpu_device *adev, struct drm_file *filp)
{
        struct amdgpu_ring *ring = &adev->vce.ring[0];
        int i, r;

        for (i = 0; i < AMDGPU_MAX_VCE_HANDLES; ++i) {
                uint32_t handle = atomic_read(&adev->vce.handles[i]);

                if (!handle || adev->vce.filp[i] != filp)
                        continue;

                r = amdgpu_vce_get_destroy_msg(ring, handle, false, NULL);
                if (r)
                        DRM_ERROR("Error destroying VCE handle (%d)!\n", r);

                adev->vce.filp[i] = NULL;
                atomic_set(&adev->vce.handles[i], 0);
        }
}

/**
 * amdgpu_vce_required_gart_pages() - gets number of GART pages required by VCE
 *
 * @adev: amdgpu_device pointer
 *
 * Returns how many GART pages we need before GTT for the VCE IP block.
 * For VCE1, see vce_v1_0_ensure_vcpu_bo_32bit_addr for details.
 * For VCE2+, this is not needed so return zero.
 */
u32 amdgpu_vce_required_gart_pages(struct amdgpu_device *adev)
{
        /* VCE IP block not added yet, so can't use amdgpu_ip_version */
        if (adev->family == AMDGPU_FAMILY_SI)
                return 512;

        return 0;
}

/**
 * amdgpu_vce_get_create_msg - generate a VCE create msg
 *
 * @ring: ring we should submit the msg to
 * @handle: VCE session handle to use
 * @fence: optional fence to return
 *
 * Open up a stream for HW test
 */
static int amdgpu_vce_get_create_msg(struct amdgpu_ring *ring, uint32_t handle,
                                     struct dma_fence **fence)
{
        const unsigned int ib_size_dw = 1024;
        struct amdgpu_job *job;
        struct amdgpu_ib *ib;
        struct amdgpu_ib ib_msg;
        struct dma_fence *f = NULL;
        uint64_t addr;
        int i, r;

        r = amdgpu_job_alloc_with_ib(ring->adev, &ring->adev->vce.entity,
                                     AMDGPU_FENCE_OWNER_UNDEFINED,
                                     ib_size_dw * 4, AMDGPU_IB_POOL_DIRECT,
                                     &job, AMDGPU_KERNEL_JOB_ID_VCN_RING_TEST);
        if (r)
                return r;

        memset(&ib_msg, 0, sizeof(ib_msg));
        /* only one gpu page is needed, alloc +1 page to make addr aligned. */
        r = amdgpu_ib_get(ring->adev, NULL, AMDGPU_GPU_PAGE_SIZE * 2,
                          AMDGPU_IB_POOL_DIRECT,
                          &ib_msg);
        if (r)
                goto err;

        ib = &job->ibs[0];
        /* let addr point to page boundary */
        addr = AMDGPU_GPU_PAGE_ALIGN(ib_msg.gpu_addr);

        /* stitch together an VCE create msg */
        ib->length_dw = 0;
        ib->ptr[ib->length_dw++] = 0x0000000c; /* len */
        ib->ptr[ib->length_dw++] = 0x00000001; /* session cmd */
        ib->ptr[ib->length_dw++] = handle;

        if ((ring->adev->vce.fw_version >> 24) >= 52)
                ib->ptr[ib->length_dw++] = 0x00000040; /* len */
        else
                ib->ptr[ib->length_dw++] = 0x00000030; /* len */
        ib->ptr[ib->length_dw++] = 0x01000001; /* create cmd */
        ib->ptr[ib->length_dw++] = 0x00000000;
        ib->ptr[ib->length_dw++] = 0x00000042;
        ib->ptr[ib->length_dw++] = 0x0000000a;
        ib->ptr[ib->length_dw++] = 0x00000001;
        ib->ptr[ib->length_dw++] = 0x00000080;
        ib->ptr[ib->length_dw++] = 0x00000060;
        ib->ptr[ib->length_dw++] = 0x00000100;
        ib->ptr[ib->length_dw++] = 0x00000100;
        ib->ptr[ib->length_dw++] = 0x0000000c;
        ib->ptr[ib->length_dw++] = 0x00000000;
        if ((ring->adev->vce.fw_version >> 24) >= 52) {
                ib->ptr[ib->length_dw++] = 0x00000000;
                ib->ptr[ib->length_dw++] = 0x00000000;
                ib->ptr[ib->length_dw++] = 0x00000000;
                ib->ptr[ib->length_dw++] = 0x00000000;
        }

        ib->ptr[ib->length_dw++] = 0x00000014; /* len */
        ib->ptr[ib->length_dw++] = 0x05000005; /* feedback buffer */
        ib->ptr[ib->length_dw++] = upper_32_bits(addr);
        ib->ptr[ib->length_dw++] = addr;
        ib->ptr[ib->length_dw++] = 0x00000001;

        for (i = ib->length_dw; i < ib_size_dw; ++i)
                ib->ptr[i] = 0x0;

        r = amdgpu_job_submit_direct(job, ring, &f);
        amdgpu_ib_free(&ib_msg, f);
        if (r)
                goto err;

        if (fence)
                *fence = dma_fence_get(f);
        dma_fence_put(f);
        return 0;

err:
        amdgpu_job_free(job);
        return r;
}

/**
 * amdgpu_vce_get_destroy_msg - generate a VCE destroy msg
 *
 * @ring: ring we should submit the msg to
 * @handle: VCE session handle to use
 * @direct: direct or delayed pool
 * @fence: optional fence to return
 *
 * Close up a stream for HW test or if userspace failed to do so
 */
static int amdgpu_vce_get_destroy_msg(struct amdgpu_ring *ring, uint32_t handle,
                                      bool direct, struct dma_fence **fence)
{
        const unsigned int ib_size_dw = 1024;
        struct amdgpu_job *job;
        struct amdgpu_ib *ib;
        struct dma_fence *f = NULL;
        int i, r;

        r = amdgpu_job_alloc_with_ib(ring->adev, &ring->adev->vce.entity,
                                     AMDGPU_FENCE_OWNER_UNDEFINED,
                                     ib_size_dw * 4,
                                     direct ? AMDGPU_IB_POOL_DIRECT :
                                     AMDGPU_IB_POOL_DELAYED, &job,
                                     AMDGPU_KERNEL_JOB_ID_VCN_RING_TEST);
        if (r)
                return r;

        ib = &job->ibs[0];

        /* stitch together an VCE destroy msg */
        ib->length_dw = 0;
        ib->ptr[ib->length_dw++] = 0x0000000c; /* len */
        ib->ptr[ib->length_dw++] = 0x00000001; /* session cmd */
        ib->ptr[ib->length_dw++] = handle;

        ib->ptr[ib->length_dw++] = 0x00000020; /* len */
        ib->ptr[ib->length_dw++] = 0x00000002; /* task info */
        ib->ptr[ib->length_dw++] = 0xffffffff; /* next task info, set to 0xffffffff if no */
        ib->ptr[ib->length_dw++] = 0x00000001; /* destroy session */
        ib->ptr[ib->length_dw++] = 0x00000000;
        ib->ptr[ib->length_dw++] = 0x00000000;
        ib->ptr[ib->length_dw++] = 0xffffffff; /* feedback is not needed, set to 0xffffffff and firmware will not output feedback */
        ib->ptr[ib->length_dw++] = 0x00000000;

        ib->ptr[ib->length_dw++] = 0x00000008; /* len */
        ib->ptr[ib->length_dw++] = 0x02000001; /* destroy cmd */

        for (i = ib->length_dw; i < ib_size_dw; ++i)
                ib->ptr[i] = 0x0;

        if (direct)
                r = amdgpu_job_submit_direct(job, ring, &f);
        else
                f = amdgpu_job_submit(job);
        if (r)
                goto err;

        if (fence)
                *fence = dma_fence_get(f);
        dma_fence_put(f);
        return 0;

err:
        amdgpu_job_free(job);
        return r;
}

/**
 * amdgpu_vce_validate_bo - make sure not to cross 4GB boundary
 *
 * @p: cs parser
 * @ib: indirect buffer to use
 * @lo: address of lower dword
 * @hi: address of higher dword
 * @size: minimum size
 * @index: bs/fb index
 *
 * Make sure that no BO cross a 4GB boundary.
 */
static int amdgpu_vce_validate_bo(struct amdgpu_cs_parser *p,
                                  struct amdgpu_ib *ib, int lo, int hi,
                                  unsigned int size, int32_t index)
{
        int64_t offset = ((uint64_t)size) * ((int64_t)index);
        struct ttm_operation_ctx ctx = { false, false };
        struct amdgpu_bo_va_mapping *mapping;
        unsigned int i, fpfn, lpfn;
        struct amdgpu_bo *bo;
        uint64_t addr;
        int r;

        addr = ((uint64_t)amdgpu_ib_get_value(ib, lo)) |
               ((uint64_t)amdgpu_ib_get_value(ib, hi)) << 32;
        if (index >= 0) {
                addr += offset;
                fpfn = PAGE_ALIGN(offset) >> PAGE_SHIFT;
                lpfn = 0x100000000ULL >> PAGE_SHIFT;
        } else {
                fpfn = 0;
                lpfn = (0x100000000ULL - PAGE_ALIGN(offset)) >> PAGE_SHIFT;
        }

        r = amdgpu_cs_find_mapping(p, addr, &bo, &mapping);
        if (r) {
                DRM_ERROR("Can't find BO for addr 0x%010llx %d %d %d %d\n",
                          addr, lo, hi, size, index);
                return r;
        }

        for (i = 0; i < bo->placement.num_placement; ++i) {
                bo->placements[i].fpfn = max(bo->placements[i].fpfn, fpfn);
                bo->placements[i].lpfn = bo->placements[i].lpfn ?
                        min(bo->placements[i].lpfn, lpfn) : lpfn;
        }
        return ttm_bo_validate(&bo->tbo, &bo->placement, &ctx);
}


/**
 * amdgpu_vce_cs_reloc - command submission relocation
 *
 * @p: parser context
 * @ib: indirect buffer to use
 * @lo: address of lower dword
 * @hi: address of higher dword
 * @size: minimum size
 * @index: bs/fb index
 *
 * Patch relocation inside command stream with real buffer address
 */
static int amdgpu_vce_cs_reloc(struct amdgpu_cs_parser *p, struct amdgpu_ib *ib,
                               int lo, int hi, unsigned int size, uint32_t index)
{
        struct amdgpu_bo_va_mapping *mapping;
        struct amdgpu_bo *bo;
        uint64_t addr;
        int r;

        if (index == 0xffffffff)
                index = 0;

        addr = ((uint64_t)amdgpu_ib_get_value(ib, lo)) |
               ((uint64_t)amdgpu_ib_get_value(ib, hi)) << 32;
        addr += ((uint64_t)size) * ((uint64_t)index);

        r = amdgpu_cs_find_mapping(p, addr, &bo, &mapping);
        if (r) {
                DRM_ERROR("Can't find BO for addr 0x%010llx %d %d %d %d\n",
                          addr, lo, hi, size, index);
                return r;
        }

        if ((addr + (uint64_t)size) >
            (mapping->last + 1) * AMDGPU_GPU_PAGE_SIZE) {
                DRM_ERROR("BO too small for addr 0x%010llx %d %d\n",
                          addr, lo, hi);
                return -EINVAL;
        }

        addr -= mapping->start * AMDGPU_GPU_PAGE_SIZE;
        addr += amdgpu_bo_gpu_offset(bo);
        addr -= ((uint64_t)size) * ((uint64_t)index);

        amdgpu_ib_set_value(ib, lo, lower_32_bits(addr));
        amdgpu_ib_set_value(ib, hi, upper_32_bits(addr));

        return 0;
}

/**
 * amdgpu_vce_validate_handle - validate stream handle
 *
 * @p: parser context
 * @handle: handle to validate
 * @allocated: allocated a new handle?
 *
 * Validates the handle and return the found session index or -EINVAL
 * we don't have another free session index.
 */
static int amdgpu_vce_validate_handle(struct amdgpu_cs_parser *p,
                                      uint32_t handle, uint32_t *allocated)
{
        unsigned int i;

        /* validate the handle */
        for (i = 0; i < AMDGPU_MAX_VCE_HANDLES; ++i) {
                if (atomic_read(&p->adev->vce.handles[i]) == handle) {
                        if (p->adev->vce.filp[i] != p->filp) {
                                DRM_ERROR("VCE handle collision detected!\n");
                                return -EINVAL;
                        }
                        return i;
                }
        }

        /* handle not found try to alloc a new one */
        for (i = 0; i < AMDGPU_MAX_VCE_HANDLES; ++i) {
                if (!atomic_cmpxchg(&p->adev->vce.handles[i], 0, handle)) {
                        p->adev->vce.filp[i] = p->filp;
                        p->adev->vce.img_size[i] = 0;
                        *allocated |= 1 << i;
                        return i;
                }
        }

        DRM_ERROR("No more free VCE handles!\n");
        return -EINVAL;
}

/**
 * amdgpu_vce_ring_parse_cs - parse and validate the command stream
 *
 * @p: parser context
 * @job: the job to parse
 * @ib: the IB to patch
 */
int amdgpu_vce_ring_parse_cs(struct amdgpu_cs_parser *p,
                             struct amdgpu_job *job,
                             struct amdgpu_ib *ib)
{
        unsigned int fb_idx = 0, bs_idx = 0;
        int session_idx = -1;
        uint32_t destroyed = 0;
        uint32_t created = 0;
        uint32_t allocated = 0;
        uint32_t tmp, handle = 0;
        uint32_t dummy = 0xffffffff;
        uint32_t *size = &dummy;
        unsigned int idx;
        int i, r = 0;

        job->vm = NULL;

        for (idx = 0; idx < ib->length_dw;) {
                uint32_t len = amdgpu_ib_get_value(ib, idx);
                uint32_t cmd = amdgpu_ib_get_value(ib, idx + 1);

                if ((len < 8) || (len & 3)) {
                        DRM_ERROR("invalid VCE command length (%d)!\n", len);
                        r = -EINVAL;
                        goto out;
                }

                switch (cmd) {
                case 0x00000002: /* task info */
                        fb_idx = amdgpu_ib_get_value(ib, idx + 6);
                        bs_idx = amdgpu_ib_get_value(ib, idx + 7);
                        break;

                case 0x03000001: /* encode */
                        r = amdgpu_vce_validate_bo(p, ib, idx + 10, idx + 9,
                                                   0, 0);
                        if (r)
                                goto out;

                        r = amdgpu_vce_validate_bo(p, ib, idx + 12, idx + 11,
                                                   0, 0);
                        if (r)
                                goto out;
                        break;

                case 0x05000001: /* context buffer */
                        r = amdgpu_vce_validate_bo(p, ib, idx + 3, idx + 2,
                                                   0, 0);
                        if (r)
                                goto out;
                        break;

                case 0x05000004: /* video bitstream buffer */
                        tmp = amdgpu_ib_get_value(ib, idx + 4);
                        r = amdgpu_vce_validate_bo(p, ib, idx + 3, idx + 2,
                                                   tmp, bs_idx);
                        if (r)
                                goto out;
                        break;

                case 0x05000005: /* feedback buffer */
                        r = amdgpu_vce_validate_bo(p, ib, idx + 3, idx + 2,
                                                   4096, fb_idx);
                        if (r)
                                goto out;
                        break;

                case 0x0500000d: /* MV buffer */
                        r = amdgpu_vce_validate_bo(p, ib, idx + 3, idx + 2,
                                                   0, 0);
                        if (r)
                                goto out;

                        r = amdgpu_vce_validate_bo(p, ib, idx + 8, idx + 7,
                                                   0, 0);
                        if (r)
                                goto out;
                        break;
                }

                idx += len / 4;
        }

        for (idx = 0; idx < ib->length_dw;) {
                uint32_t len = amdgpu_ib_get_value(ib, idx);
                uint32_t cmd = amdgpu_ib_get_value(ib, idx + 1);

                switch (cmd) {
                case 0x00000001: /* session */
                        handle = amdgpu_ib_get_value(ib, idx + 2);
                        session_idx = amdgpu_vce_validate_handle(p, handle,
                                                                 &allocated);
                        if (session_idx < 0) {
                                r = session_idx;
                                goto out;
                        }
                        size = &p->adev->vce.img_size[session_idx];
                        break;

                case 0x00000002: /* task info */
                        fb_idx = amdgpu_ib_get_value(ib, idx + 6);
                        bs_idx = amdgpu_ib_get_value(ib, idx + 7);
                        break;

                case 0x01000001: /* create */
                        created |= 1 << session_idx;
                        if (destroyed & (1 << session_idx)) {
                                destroyed &= ~(1 << session_idx);
                                allocated |= 1 << session_idx;

                        } else if (!(allocated & (1 << session_idx))) {
                                DRM_ERROR("Handle already in use!\n");
                                r = -EINVAL;
                                goto out;
                        }

                        *size = amdgpu_ib_get_value(ib, idx + 8) *
                                amdgpu_ib_get_value(ib, idx + 10) *
                                8 * 3 / 2;
                        break;

                case 0x04000001: /* config extension */
                case 0x04000002: /* pic control */
                case 0x04000005: /* rate control */
                case 0x04000007: /* motion estimation */
                case 0x04000008: /* rdo */
                case 0x04000009: /* vui */
                case 0x05000002: /* auxiliary buffer */
                case 0x05000009: /* clock table */
                        break;

                case 0x0500000c: /* hw config */
                        switch (p->adev->asic_type) {
#ifdef CONFIG_DRM_AMDGPU_CIK
                        case CHIP_KAVERI:
                        case CHIP_MULLINS:
#endif
                        case CHIP_CARRIZO:
                                break;
                        default:
                                r = -EINVAL;
                                goto out;
                        }
                        break;

                case 0x03000001: /* encode */
                        r = amdgpu_vce_cs_reloc(p, ib, idx + 10, idx + 9,
                                                *size, 0);
                        if (r)
                                goto out;

                        r = amdgpu_vce_cs_reloc(p, ib, idx + 12, idx + 11,
                                                *size / 3, 0);
                        if (r)
                                goto out;
                        break;

                case 0x02000001: /* destroy */
                        destroyed |= 1 << session_idx;
                        break;

                case 0x05000001: /* context buffer */
                        r = amdgpu_vce_cs_reloc(p, ib, idx + 3, idx + 2,
                                                *size * 2, 0);
                        if (r)
                                goto out;
                        break;

                case 0x05000004: /* video bitstream buffer */
                        tmp = amdgpu_ib_get_value(ib, idx + 4);
                        r = amdgpu_vce_cs_reloc(p, ib, idx + 3, idx + 2,
                                                tmp, bs_idx);
                        if (r)
                                goto out;
                        break;

                case 0x05000005: /* feedback buffer */
                        r = amdgpu_vce_cs_reloc(p, ib, idx + 3, idx + 2,
                                                4096, fb_idx);
                        if (r)
                                goto out;
                        break;

                case 0x0500000d: /* MV buffer */
                        r = amdgpu_vce_cs_reloc(p, ib, idx + 3,
                                                idx + 2, *size, 0);
                        if (r)
                                goto out;

                        r = amdgpu_vce_cs_reloc(p, ib, idx + 8,
                                                idx + 7, *size / 12, 0);
                        if (r)
                                goto out;
                        break;

                default:
                        DRM_ERROR("invalid VCE command (0x%x)!\n", cmd);
                        r = -EINVAL;
                        goto out;
                }

                if (session_idx == -1) {
                        DRM_ERROR("no session command at start of IB\n");
                        r = -EINVAL;
                        goto out;
                }

                idx += len / 4;
        }

        if (allocated & ~created) {
                DRM_ERROR("New session without create command!\n");
                r = -ENOENT;
        }

out:
        if (!r) {
                /* No error, free all destroyed handle slots */
                tmp = destroyed;
        } else {
                /* Error during parsing, free all allocated handle slots */
                tmp = allocated;
        }

        for (i = 0; i < AMDGPU_MAX_VCE_HANDLES; ++i)
                if (tmp & (1 << i))
                        atomic_set(&p->adev->vce.handles[i], 0);

        return r;
}

/**
 * amdgpu_vce_ring_parse_cs_vm - parse the command stream in VM mode
 *
 * @p: parser context
 * @job: the job to parse
 * @ib: the IB to patch
 */
int amdgpu_vce_ring_parse_cs_vm(struct amdgpu_cs_parser *p,
                                struct amdgpu_job *job,
                                struct amdgpu_ib *ib)
{
        int session_idx = -1;
        uint32_t destroyed = 0;
        uint32_t created = 0;
        uint32_t allocated = 0;
        uint32_t tmp, handle = 0;
        int i, r = 0, idx = 0;

        while (idx < ib->length_dw) {
                uint32_t len = amdgpu_ib_get_value(ib, idx);
                uint32_t cmd = amdgpu_ib_get_value(ib, idx + 1);

                if ((len < 8) || (len & 3)) {
                        DRM_ERROR("invalid VCE command length (%d)!\n", len);
                        r = -EINVAL;
                        goto out;
                }

                switch (cmd) {
                case 0x00000001: /* session */
                        handle = amdgpu_ib_get_value(ib, idx + 2);
                        session_idx = amdgpu_vce_validate_handle(p, handle,
                                                                 &allocated);
                        if (session_idx < 0) {
                                r = session_idx;
                                goto out;
                        }
                        break;

                case 0x01000001: /* create */
                        created |= 1 << session_idx;
                        if (destroyed & (1 << session_idx)) {
                                destroyed &= ~(1 << session_idx);
                                allocated |= 1 << session_idx;

                        } else if (!(allocated & (1 << session_idx))) {
                                DRM_ERROR("Handle already in use!\n");
                                r = -EINVAL;
                                goto out;
                        }

                        break;

                case 0x02000001: /* destroy */
                        destroyed |= 1 << session_idx;
                        break;

                default:
                        break;
                }

                if (session_idx == -1) {
                        DRM_ERROR("no session command at start of IB\n");
                        r = -EINVAL;
                        goto out;
                }

                idx += len / 4;
        }

        if (allocated & ~created) {
                DRM_ERROR("New session without create command!\n");
                r = -ENOENT;
        }

out:
        if (!r) {
                /* No error, free all destroyed handle slots */
                tmp = destroyed;
        } else {
                /* Error during parsing, free all allocated handle slots */
                tmp = allocated;
        }

        for (i = 0; i < AMDGPU_MAX_VCE_HANDLES; ++i)
                if (tmp & (1 << i))
                        atomic_set(&p->adev->vce.handles[i], 0);

        return r;
}

/**
 * amdgpu_vce_ring_emit_ib - execute indirect buffer
 *
 * @ring: engine to use
 * @job: job to retrieve vmid from
 * @ib: the IB to execute
 * @flags: unused
 *
 */
void amdgpu_vce_ring_emit_ib(struct amdgpu_ring *ring,
                                struct amdgpu_job *job,
                                struct amdgpu_ib *ib,
                                uint32_t flags)
{
        amdgpu_ring_write(ring, VCE_CMD_IB);
        amdgpu_ring_write(ring, lower_32_bits(ib->gpu_addr));
        amdgpu_ring_write(ring, upper_32_bits(ib->gpu_addr));
        amdgpu_ring_write(ring, ib->length_dw);
}

/**
 * amdgpu_vce_ring_emit_fence - add a fence command to the ring
 *
 * @ring: engine to use
 * @addr: address
 * @seq: sequence number
 * @flags: fence related flags
 *
 */
void amdgpu_vce_ring_emit_fence(struct amdgpu_ring *ring, u64 addr, u64 seq,
                                unsigned int flags)
{
        WARN_ON(flags & AMDGPU_FENCE_FLAG_64BIT);

        amdgpu_ring_write(ring, VCE_CMD_FENCE);
        amdgpu_ring_write(ring, addr);
        amdgpu_ring_write(ring, upper_32_bits(addr));
        amdgpu_ring_write(ring, seq);
        amdgpu_ring_write(ring, VCE_CMD_TRAP);
        amdgpu_ring_write(ring, VCE_CMD_END);
}

/**
 * amdgpu_vce_ring_test_ring - test if VCE ring is working
 *
 * @ring: the engine to test on
 *
 */
int amdgpu_vce_ring_test_ring(struct amdgpu_ring *ring)
{
        struct amdgpu_device *adev = ring->adev;
        uint32_t rptr;
        unsigned int i;
        int r, timeout = adev->usec_timeout;

        /* skip ring test for sriov*/
        if (amdgpu_sriov_vf(adev))
                return 0;

        r = amdgpu_ring_alloc(ring, 16);
        if (r)
                return r;

        rptr = amdgpu_ring_get_rptr(ring);

        amdgpu_ring_write(ring, VCE_CMD_END);
        amdgpu_ring_commit(ring);

        for (i = 0; i < timeout; i++) {
                if (amdgpu_ring_get_rptr(ring) != rptr)
                        break;
                udelay(1);
        }

        if (i >= timeout)
                r = -ETIMEDOUT;

        return r;
}

/**
 * amdgpu_vce_ring_test_ib - test if VCE IBs are working
 *
 * @ring: the engine to test on
 * @timeout: timeout value in jiffies, or MAX_SCHEDULE_TIMEOUT
 *
 */
int amdgpu_vce_ring_test_ib(struct amdgpu_ring *ring, long timeout)
{
        struct dma_fence *fence = NULL;
        long r;

        /* skip vce ring1/2 ib test for now, since it's not reliable */
        if (ring != &ring->adev->vce.ring[0])
                return 0;

        r = amdgpu_vce_get_create_msg(ring, 1, NULL);
        if (r)
                goto error;

        r = amdgpu_vce_get_destroy_msg(ring, 1, true, &fence);
        if (r)
                goto error;

        r = dma_fence_wait_timeout(fence, false, timeout);
        if (r == 0)
                r = -ETIMEDOUT;
        else if (r > 0)
                r = 0;

error:
        dma_fence_put(fence);
        return r;
}

enum amdgpu_ring_priority_level amdgpu_vce_get_ring_prio(int ring)
{
        switch (ring) {
        case 0:
                return AMDGPU_RING_PRIO_0;
        case 1:
                return AMDGPU_RING_PRIO_1;
        case 2:
                return AMDGPU_RING_PRIO_2;
        default:
                return AMDGPU_RING_PRIO_0;
        }
}