/*
 * Copyright 2013 Red Hat Inc.
 *
 * 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, sublicense,
 * 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 above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * 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 NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) 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.
 *
 * Authors: Dave Airlie
 *          Alon Levy
 */

#include <linux/pci.h>
#include <linux/uaccess.h>

#include <drm/drm_print.h>

#include "qxl_drv.h"
#include "qxl_object.h"

/*
 * TODO: allocating a new gem(in qxl_bo) for each request.
 * This is wasteful since bo's are page aligned.
 */
int qxl_alloc_ioctl(struct drm_device *dev, void *data, struct drm_file *file_priv)
{
        struct qxl_device *qdev = to_qxl(dev);
        struct drm_qxl_alloc *qxl_alloc = data;
        int ret;
        uint32_t handle;
        u32 domain = QXL_GEM_DOMAIN_VRAM;

        if (qxl_alloc->size == 0) {
                DRM_ERROR("invalid size %d\n", qxl_alloc->size);
                return -EINVAL;
        }
        ret = qxl_gem_object_create_with_handle(qdev, file_priv,
                                                domain,
                                                qxl_alloc->size,
                                                NULL,
                                                NULL, &handle);
        if (ret) {
                DRM_ERROR("%s: failed to create gem ret=%d\n",
                          __func__, ret);
                return -ENOMEM;
        }
        qxl_alloc->handle = handle;
        return 0;
}

int qxl_map_ioctl(struct drm_device *dev, void *data, struct drm_file *file_priv)
{
        struct qxl_device *qdev = to_qxl(dev);
        struct drm_qxl_map *qxl_map = data;

        return drm_gem_ttm_dumb_map_offset(file_priv, &qdev->ddev, qxl_map->handle,
                                           &qxl_map->offset);
}

struct qxl_reloc_info {
        int type;
        struct qxl_bo *dst_bo;
        uint32_t dst_offset;
        struct qxl_bo *src_bo;
        int src_offset;
};

/*
 * dst must be validated, i.e. whole bo on vram/surfacesram (right now all bo's
 * are on vram).
 * *(dst + dst_off) = qxl_bo_physical_address(src, src_off)
 */
static void
apply_reloc(struct qxl_device *qdev, struct qxl_reloc_info *info)
{
        void *reloc_page;

        reloc_page = qxl_bo_kmap_atomic_page(qdev, info->dst_bo, info->dst_offset & PAGE_MASK);
        *(uint64_t *)(reloc_page + (info->dst_offset & ~PAGE_MASK)) = qxl_bo_physical_address(qdev,
                                                                                              info->src_bo,
                                                                                              info->src_offset);
        qxl_bo_kunmap_atomic_page(qdev, info->dst_bo, reloc_page);
}

static void
apply_surf_reloc(struct qxl_device *qdev, struct qxl_reloc_info *info)
{
        uint32_t id = 0;
        void *reloc_page;

        if (info->src_bo && !info->src_bo->is_primary)
                id = info->src_bo->surface_id;

        reloc_page = qxl_bo_kmap_atomic_page(qdev, info->dst_bo, info->dst_offset & PAGE_MASK);
        *(uint32_t *)(reloc_page + (info->dst_offset & ~PAGE_MASK)) = id;
        qxl_bo_kunmap_atomic_page(qdev, info->dst_bo, reloc_page);
}

/* return holding the reference to this object */
static int qxlhw_handle_to_bo(struct drm_file *file_priv, uint64_t handle,
                              struct qxl_release *release, struct qxl_bo **qbo_p)
{
        struct drm_gem_object *gobj;
        struct qxl_bo *qobj;
        int ret;

        gobj = drm_gem_object_lookup(file_priv, handle);
        if (!gobj)
                return -EINVAL;

        qobj = gem_to_qxl_bo(gobj);

        ret = qxl_release_list_add(release, qobj);
        drm_gem_object_put(gobj);
        if (ret)
                return ret;

        *qbo_p = qobj;
        return 0;
}

/*
 * Usage of execbuffer:
 * Relocations need to take into account the full QXLDrawable size.
 * However, the command as passed from user space must *not* contain the initial
 * QXLReleaseInfo struct (first XXX bytes)
 */
static int qxl_process_single_command(struct qxl_device *qdev,
                                      struct drm_qxl_command *cmd,
                                      struct drm_file *file_priv)
{
        struct qxl_reloc_info *reloc_info;
        int release_type;
        struct qxl_release *release;
        struct qxl_bo *cmd_bo;
        void *fb_cmd;
        int i, ret;
        int unwritten;

        switch (cmd->type) {
        case QXL_CMD_DRAW:
                release_type = QXL_RELEASE_DRAWABLE;
                break;
        case QXL_CMD_SURFACE:
        case QXL_CMD_CURSOR:
        default:
                DRM_DEBUG("Only draw commands in execbuffers\n");
                return -EINVAL;
        }

        if (cmd->command_size > PAGE_SIZE - sizeof(union qxl_release_info))
                return -EINVAL;

        if (!access_ok(u64_to_user_ptr(cmd->command),
                       cmd->command_size))
                return -EFAULT;

        reloc_info = kmalloc_objs(struct qxl_reloc_info, cmd->relocs_num);
        if (!reloc_info)
                return -ENOMEM;

        ret = qxl_alloc_release_reserved(qdev,
                                         sizeof(union qxl_release_info) +
                                         cmd->command_size,
                                         release_type,
                                         &release,
                                         &cmd_bo);
        if (ret)
                goto out_free_reloc;

        /* TODO copy slow path code from i915 */
        fb_cmd = qxl_bo_kmap_atomic_page(qdev, cmd_bo, (release->release_offset & PAGE_MASK));
        unwritten = __copy_from_user_inatomic_nocache
                (fb_cmd + sizeof(union qxl_release_info) + (release->release_offset & ~PAGE_MASK),
                 u64_to_user_ptr(cmd->command), cmd->command_size);

        {
                struct qxl_drawable *draw = fb_cmd;

                draw->mm_time = qdev->rom->mm_clock;
        }

        qxl_bo_kunmap_atomic_page(qdev, cmd_bo, fb_cmd);
        if (unwritten) {
                DRM_ERROR("got unwritten %d\n", unwritten);
                ret = -EFAULT;
                goto out_free_release;
        }

        /* fill out reloc info structs */
        for (i = 0; i < cmd->relocs_num; ++i) {
                struct drm_qxl_reloc reloc;
                struct drm_qxl_reloc __user *u = u64_to_user_ptr(cmd->relocs);

                if (copy_from_user(&reloc, u + i, sizeof(reloc))) {
                        ret = -EFAULT;
                        goto out_free_bos;
                }

                /* add the bos to the list of bos to validate -
                   need to validate first then process relocs? */
                if (reloc.reloc_type != QXL_RELOC_TYPE_BO && reloc.reloc_type != QXL_RELOC_TYPE_SURF) {
                        DRM_DEBUG("unknown reloc type %d\n", reloc.reloc_type);

                        ret = -EINVAL;
                        goto out_free_bos;
                }
                reloc_info[i].type = reloc.reloc_type;

                if (reloc.dst_handle) {
                        ret = qxlhw_handle_to_bo(file_priv, reloc.dst_handle, release,
                                                 &reloc_info[i].dst_bo);
                        if (ret)
                                goto out_free_bos;
                        reloc_info[i].dst_offset = reloc.dst_offset;
                } else {
                        reloc_info[i].dst_bo = cmd_bo;
                        reloc_info[i].dst_offset = reloc.dst_offset + release->release_offset;
                }

                /* reserve and validate the reloc dst bo */
                if (reloc.reloc_type == QXL_RELOC_TYPE_BO || reloc.src_handle) {
                        ret = qxlhw_handle_to_bo(file_priv, reloc.src_handle, release,
                                                 &reloc_info[i].src_bo);
                        if (ret)
                                goto out_free_bos;
                        reloc_info[i].src_offset = reloc.src_offset;
                } else {
                        reloc_info[i].src_bo = NULL;
                        reloc_info[i].src_offset = 0;
                }
        }

        /* validate all buffers */
        ret = qxl_release_reserve_list(release, false);
        if (ret)
                goto out_free_bos;

        for (i = 0; i < cmd->relocs_num; ++i) {
                if (reloc_info[i].type == QXL_RELOC_TYPE_BO)
                        apply_reloc(qdev, &reloc_info[i]);
                else if (reloc_info[i].type == QXL_RELOC_TYPE_SURF)
                        apply_surf_reloc(qdev, &reloc_info[i]);
        }

        qxl_release_fence_buffer_objects(release);
        ret = qxl_push_command_ring_release(qdev, release, cmd->type, true);

out_free_bos:
out_free_release:
        if (ret)
                qxl_release_free(qdev, release);
out_free_reloc:
        kfree(reloc_info);
        return ret;
}

int qxl_execbuffer_ioctl(struct drm_device *dev, void *data, struct drm_file *file_priv)
{
        struct qxl_device *qdev = to_qxl(dev);
        struct drm_qxl_execbuffer *execbuffer = data;
        struct drm_qxl_command user_cmd;
        int cmd_num;
        int ret;

        for (cmd_num = 0; cmd_num < execbuffer->commands_num; ++cmd_num) {

                struct drm_qxl_command __user *commands =
                        u64_to_user_ptr(execbuffer->commands);

                if (copy_from_user(&user_cmd, commands + cmd_num,
                                       sizeof(user_cmd)))
                        return -EFAULT;

                ret = qxl_process_single_command(qdev, &user_cmd, file_priv);
                if (ret)
                        return ret;
        }
        return 0;
}

int qxl_update_area_ioctl(struct drm_device *dev, void *data, struct drm_file *file)
{
        struct qxl_device *qdev = to_qxl(dev);
        struct drm_qxl_update_area *update_area = data;
        struct qxl_rect area = {.left = update_area->left,
                                .top = update_area->top,
                                .right = update_area->right,
                                .bottom = update_area->bottom};
        int ret;
        struct drm_gem_object *gobj = NULL;
        struct qxl_bo *qobj = NULL;
        struct ttm_operation_ctx ctx = { true, false };

        if (update_area->left >= update_area->right ||
            update_area->top >= update_area->bottom)
                return -EINVAL;

        gobj = drm_gem_object_lookup(file, update_area->handle);
        if (gobj == NULL)
                return -ENOENT;

        qobj = gem_to_qxl_bo(gobj);

        ret = qxl_bo_reserve(qobj);
        if (ret)
                goto out;

        if (!qobj->tbo.pin_count) {
                qxl_ttm_placement_from_domain(qobj, qobj->type);
                ret = ttm_bo_validate(&qobj->tbo, &qobj->placement, &ctx);
                if (unlikely(ret))
                        goto out;
        }

        ret = qxl_bo_check_id(qdev, qobj);
        if (ret)
                goto out2;
        if (!qobj->surface_id)
                DRM_ERROR("got update area for surface with no id %d\n", update_area->handle);
        ret = qxl_io_update_area(qdev, qobj, &area);

out2:
        qxl_bo_unreserve(qobj);

out:
        drm_gem_object_put(gobj);
        return ret;
}

int qxl_getparam_ioctl(struct drm_device *dev, void *data, struct drm_file *file_priv)
{
        struct qxl_device *qdev = to_qxl(dev);
        struct drm_qxl_getparam *param = data;

        switch (param->param) {
        case QXL_PARAM_NUM_SURFACES:
                param->value = qdev->rom->n_surfaces;
                break;
        case QXL_PARAM_MAX_RELOCS:
                param->value = QXL_MAX_RES;
                break;
        default:
                return -EINVAL;
        }
        return 0;
}

int qxl_clientcap_ioctl(struct drm_device *dev, void *data, struct drm_file *file_priv)
{
        struct qxl_device *qdev = to_qxl(dev);
        struct pci_dev *pdev = to_pci_dev(dev->dev);
        struct drm_qxl_clientcap *param = data;
        int byte, idx;

        byte = param->index / 8;
        idx = param->index % 8;

        if (pdev->revision < 4)
                return -ENOSYS;

        if (byte >= 58)
                return -ENOSYS;

        if (qdev->rom->client_capabilities[byte] & (1 << idx))
                return 0;
        return -ENOSYS;
}

int qxl_alloc_surf_ioctl(struct drm_device *dev, void *data, struct drm_file *file)
{
        struct qxl_device *qdev = to_qxl(dev);
        struct drm_qxl_alloc_surf *param = data;
        int handle;
        int ret;
        int size, actual_stride;
        struct qxl_surface surf;

        /* work out size allocate bo with handle */
        actual_stride = param->stride < 0 ? -param->stride : param->stride;
        size = actual_stride * param->height + actual_stride;

        surf.format = param->format;
        surf.width = param->width;
        surf.height = param->height;
        surf.stride = param->stride;
        surf.data = 0;

        ret = qxl_gem_object_create_with_handle(qdev, file,
                                                QXL_GEM_DOMAIN_SURFACE,
                                                size,
                                                &surf,
                                                NULL, &handle);
        if (ret) {
                DRM_ERROR("%s: failed to create gem ret=%d\n",
                          __func__, ret);
                return -ENOMEM;
        } else
                param->handle = handle;
        return ret;
}