/*
 * Copyright (c) 2016 Hisilicon Limited.
 *
 * This software is available to you under a choice of one of two
 * licenses.  You may choose to be licensed under the terms of the GNU
 * General Public License (GPL) Version 2, available from the file
 * COPYING in the main directory of this source tree, or the
 * OpenIB.org BSD license below:
 *
 *     Redistribution and use in source and binary forms, with or
 *     without modification, are permitted provided that the following
 *     conditions are met:
 *
 *      - Redistributions of source code must retain the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer.
 *
 *      - Redistributions in binary form must reproduce the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer in the documentation and/or other materials
 *        provided with the distribution.
 *
 * 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 AUTHORS OR COPYRIGHT HOLDERS
 * 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.
 */

#include <linux/platform_device.h>
#include <rdma/ib_umem.h>
#include "hns_roce_device.h"
#include "hns_roce_cmd.h"
#include "hns_roce_hem.h"
#include "hns_roce_user.h"
#include "hns_roce_common.h"

static void hns_roce_ib_cq_comp(struct hns_roce_cq *hr_cq)
{
        struct ib_cq *ibcq = &hr_cq->ib_cq;

        ibcq->comp_handler(ibcq, ibcq->cq_context);
}

static void hns_roce_ib_cq_event(struct hns_roce_cq *hr_cq,
                                 enum hns_roce_event event_type)
{
        struct hns_roce_dev *hr_dev;
        struct ib_event event;
        struct ib_cq *ibcq;

        ibcq = &hr_cq->ib_cq;
        hr_dev = to_hr_dev(ibcq->device);

        if (event_type != HNS_ROCE_EVENT_TYPE_CQ_ID_INVALID &&
            event_type != HNS_ROCE_EVENT_TYPE_CQ_ACCESS_ERROR &&
            event_type != HNS_ROCE_EVENT_TYPE_CQ_OVERFLOW) {
                dev_err(&hr_dev->pdev->dev,
                        "hns_roce_ib: Unexpected event type 0x%x on CQ %06lx\n",
                        event_type, hr_cq->cqn);
                return;
        }

        if (ibcq->event_handler) {
                event.device = ibcq->device;
                event.event = IB_EVENT_CQ_ERR;
                event.element.cq = ibcq;
                ibcq->event_handler(&event, ibcq->cq_context);
        }
}

static int hns_roce_sw2hw_cq(struct hns_roce_dev *dev,
                             struct hns_roce_cmd_mailbox *mailbox,
                             unsigned long cq_num)
{
        return hns_roce_cmd_mbox(dev, mailbox->dma, 0, cq_num, 0,
                            HNS_ROCE_CMD_SW2HW_CQ, HNS_ROCE_CMD_TIME_CLASS_A);
}

static int hns_roce_cq_alloc(struct hns_roce_dev *hr_dev, int nent,
                             struct hns_roce_mtt *hr_mtt,
                             struct hns_roce_uar *hr_uar,
                             struct hns_roce_cq *hr_cq, int vector)
{
        struct hns_roce_cmd_mailbox *mailbox = NULL;
        struct hns_roce_cq_table *cq_table = NULL;
        struct device *dev = &hr_dev->pdev->dev;
        dma_addr_t dma_handle;
        u64 *mtts = NULL;
        int ret = 0;

        cq_table = &hr_dev->cq_table;

        /* Get the physical address of cq buf */
        mtts = hns_roce_table_find(&hr_dev->mr_table.mtt_table,
                                   hr_mtt->first_seg, &dma_handle);
        if (!mtts) {
                dev_err(dev, "CQ alloc.Failed to find cq buf addr.\n");
                return -EINVAL;
        }

        if (vector >= hr_dev->caps.num_comp_vectors) {
                dev_err(dev, "CQ alloc.Invalid vector.\n");
                return -EINVAL;
        }
        hr_cq->vector = vector;

        ret = hns_roce_bitmap_alloc(&cq_table->bitmap, &hr_cq->cqn);
        if (ret == -1) {
                dev_err(dev, "CQ alloc.Failed to alloc index.\n");
                return -ENOMEM;
        }

        /* Get CQC memory HEM(Hardware Entry Memory) table */
        ret = hns_roce_table_get(hr_dev, &cq_table->table, hr_cq->cqn);
        if (ret) {
                dev_err(dev, "CQ alloc.Failed to get context mem.\n");
                goto err_out;
        }

        /* The cq insert radix tree */
        spin_lock_irq(&cq_table->lock);
        /* Radix_tree: The associated pointer and long integer key value like */
        ret = radix_tree_insert(&cq_table->tree, hr_cq->cqn, hr_cq);
        spin_unlock_irq(&cq_table->lock);
        if (ret) {
                dev_err(dev, "CQ alloc.Failed to radix_tree_insert.\n");
                goto err_put;
        }

        /* Allocate mailbox memory */
        mailbox = hns_roce_alloc_cmd_mailbox(hr_dev);
        if (IS_ERR(mailbox)) {
                ret = PTR_ERR(mailbox);
                goto err_radix;
        }

        hr_dev->hw->write_cqc(hr_dev, hr_cq, mailbox->buf, mtts, dma_handle,
                              nent, vector);

        /* Send mailbox to hw */
        ret = hns_roce_sw2hw_cq(hr_dev, mailbox, hr_cq->cqn);
        hns_roce_free_cmd_mailbox(hr_dev, mailbox);
        if (ret) {
                dev_err(dev, "CQ alloc.Failed to cmd mailbox.\n");
                goto err_radix;
        }

        hr_cq->cons_index = 0;
        hr_cq->uar = hr_uar;

        atomic_set(&hr_cq->refcount, 1);
        init_completion(&hr_cq->free);

        return 0;

err_radix:
        spin_lock_irq(&cq_table->lock);
        radix_tree_delete(&cq_table->tree, hr_cq->cqn);
        spin_unlock_irq(&cq_table->lock);

err_put:
        hns_roce_table_put(hr_dev, &cq_table->table, hr_cq->cqn);

err_out:
        hns_roce_bitmap_free(&cq_table->bitmap, hr_cq->cqn);
        return ret;
}

static int hns_roce_hw2sw_cq(struct hns_roce_dev *dev,
                             struct hns_roce_cmd_mailbox *mailbox,
                             unsigned long cq_num)
{
        return hns_roce_cmd_mbox(dev, 0, mailbox ? mailbox->dma : 0, cq_num,
                                 mailbox ? 0 : 1, HNS_ROCE_CMD_HW2SW_CQ,
                                 HNS_ROCE_CMD_TIME_CLASS_A);
}

static void hns_roce_free_cq(struct hns_roce_dev *hr_dev,
                             struct hns_roce_cq *hr_cq)
{
        struct hns_roce_cq_table *cq_table = &hr_dev->cq_table;
        struct device *dev = &hr_dev->pdev->dev;
        int ret;

        ret = hns_roce_hw2sw_cq(hr_dev, NULL, hr_cq->cqn);
        if (ret)
                dev_err(dev, "HW2SW_CQ failed (%d) for CQN %06lx\n", ret,
                        hr_cq->cqn);

        /* Waiting interrupt process procedure carried out */
        synchronize_irq(hr_dev->eq_table.eq[hr_cq->vector].irq);

        /* wait for all interrupt processed */
        if (atomic_dec_and_test(&hr_cq->refcount))
                complete(&hr_cq->free);
        wait_for_completion(&hr_cq->free);

        spin_lock_irq(&cq_table->lock);
        radix_tree_delete(&cq_table->tree, hr_cq->cqn);
        spin_unlock_irq(&cq_table->lock);

        hns_roce_table_put(hr_dev, &cq_table->table, hr_cq->cqn);
        hns_roce_bitmap_free(&cq_table->bitmap, hr_cq->cqn);
}

static int hns_roce_ib_get_cq_umem(struct hns_roce_dev *hr_dev,
                                   struct ib_ucontext *context,
                                   struct hns_roce_cq_buf *buf,
                                   struct ib_umem **umem, u64 buf_addr, int cqe)
{
        int ret;

        *umem = ib_umem_get(context, buf_addr, cqe * hr_dev->caps.cq_entry_sz,
                            IB_ACCESS_LOCAL_WRITE, 1);
        if (IS_ERR(*umem))
                return PTR_ERR(*umem);

        ret = hns_roce_mtt_init(hr_dev, ib_umem_page_count(*umem),
                                ilog2((unsigned int)(*umem)->page_size),
                                &buf->hr_mtt);
        if (ret)
                goto err_buf;

        ret = hns_roce_ib_umem_write_mtt(hr_dev, &buf->hr_mtt, *umem);
        if (ret)
                goto err_mtt;

        return 0;

err_mtt:
        hns_roce_mtt_cleanup(hr_dev, &buf->hr_mtt);

err_buf:
        ib_umem_release(*umem);
        return ret;
}

static int hns_roce_ib_alloc_cq_buf(struct hns_roce_dev *hr_dev,
                                    struct hns_roce_cq_buf *buf, u32 nent)
{
        int ret;

        ret = hns_roce_buf_alloc(hr_dev, nent * hr_dev->caps.cq_entry_sz,
                                 PAGE_SIZE * 2, &buf->hr_buf);
        if (ret)
                goto out;

        ret = hns_roce_mtt_init(hr_dev, buf->hr_buf.npages,
                                buf->hr_buf.page_shift, &buf->hr_mtt);
        if (ret)
                goto err_buf;

        ret = hns_roce_buf_write_mtt(hr_dev, &buf->hr_mtt, &buf->hr_buf);
        if (ret)
                goto err_mtt;

        return 0;

err_mtt:
        hns_roce_mtt_cleanup(hr_dev, &buf->hr_mtt);

err_buf:
        hns_roce_buf_free(hr_dev, nent * hr_dev->caps.cq_entry_sz,
                          &buf->hr_buf);
out:
        return ret;
}

static void hns_roce_ib_free_cq_buf(struct hns_roce_dev *hr_dev,
                                    struct hns_roce_cq_buf *buf, int cqe)
{
        hns_roce_buf_free(hr_dev, (cqe + 1) * hr_dev->caps.cq_entry_sz,
                          &buf->hr_buf);
}

struct ib_cq *hns_roce_ib_create_cq(struct ib_device *ib_dev,
                                    const struct ib_cq_init_attr *attr,
                                    struct ib_ucontext *context,
                                    struct ib_udata *udata)
{
        struct hns_roce_dev *hr_dev = to_hr_dev(ib_dev);
        struct device *dev = &hr_dev->pdev->dev;
        struct hns_roce_ib_create_cq ucmd;
        struct hns_roce_cq *hr_cq = NULL;
        struct hns_roce_uar *uar = NULL;
        int vector = attr->comp_vector;
        int cq_entries = attr->cqe;
        int ret = 0;

        if (cq_entries < 1 || cq_entries > hr_dev->caps.max_cqes) {
                dev_err(dev, "Creat CQ failed. entries=%d, max=%d\n",
                        cq_entries, hr_dev->caps.max_cqes);
                return ERR_PTR(-EINVAL);
        }

        hr_cq = kmalloc(sizeof(*hr_cq), GFP_KERNEL);
        if (!hr_cq)
                return ERR_PTR(-ENOMEM);

        /* In v1 engine, parameter verification */
        if (cq_entries < HNS_ROCE_MIN_CQE_NUM)
                cq_entries = HNS_ROCE_MIN_CQE_NUM;

        cq_entries = roundup_pow_of_two((unsigned int)cq_entries);
        hr_cq->ib_cq.cqe = cq_entries - 1;
        spin_lock_init(&hr_cq->lock);

        if (context) {
                if (ib_copy_from_udata(&ucmd, udata, sizeof(ucmd))) {
                        dev_err(dev, "Failed to copy_from_udata.\n");
                        ret = -EFAULT;
                        goto err_cq;
                }

                /* Get user space address, write it into mtt table */
                ret = hns_roce_ib_get_cq_umem(hr_dev, context, &hr_cq->hr_buf,
                                              &hr_cq->umem, ucmd.buf_addr,
                                              cq_entries);
                if (ret) {
                        dev_err(dev, "Failed to get_cq_umem.\n");
                        goto err_cq;
                }

                /* Get user space parameters */
                uar = &to_hr_ucontext(context)->uar;
        } else {
                /* Init mmt table and write buff address to mtt table */
                ret = hns_roce_ib_alloc_cq_buf(hr_dev, &hr_cq->hr_buf,
                                               cq_entries);
                if (ret) {
                        dev_err(dev, "Failed to alloc_cq_buf.\n");
                        goto err_cq;
                }

                uar = &hr_dev->priv_uar;
                hr_cq->cq_db_l = hr_dev->reg_base + ROCEE_DB_OTHERS_L_0_REG +
                                 0x1000 * uar->index;
        }

        /* Allocate cq index, fill cq_context */
        ret = hns_roce_cq_alloc(hr_dev, cq_entries, &hr_cq->hr_buf.hr_mtt, uar,
                                hr_cq, vector);
        if (ret) {
                dev_err(dev, "Creat CQ .Failed to cq_alloc.\n");
                goto err_mtt;
        }

        /* Get created cq handler and carry out event */
        hr_cq->comp = hns_roce_ib_cq_comp;
        hr_cq->event = hns_roce_ib_cq_event;
        hr_cq->cq_depth = cq_entries;

        if (context) {
                if (ib_copy_to_udata(udata, &hr_cq->cqn, sizeof(u64))) {
                        ret = -EFAULT;
                        goto err_cqc;
                }
        }

        return &hr_cq->ib_cq;

err_cqc:
        hns_roce_free_cq(hr_dev, hr_cq);

err_mtt:
        hns_roce_mtt_cleanup(hr_dev, &hr_cq->hr_buf.hr_mtt);
        if (context)
                ib_umem_release(hr_cq->umem);
        else
                hns_roce_ib_free_cq_buf(hr_dev, &hr_cq->hr_buf,
                                        hr_cq->ib_cq.cqe);

err_cq:
        kfree(hr_cq);
        return ERR_PTR(ret);
}

int hns_roce_ib_destroy_cq(struct ib_cq *ib_cq)
{
        struct hns_roce_dev *hr_dev = to_hr_dev(ib_cq->device);
        struct hns_roce_cq *hr_cq = to_hr_cq(ib_cq);

        hns_roce_free_cq(hr_dev, hr_cq);
        hns_roce_mtt_cleanup(hr_dev, &hr_cq->hr_buf.hr_mtt);

        if (ib_cq->uobject)
                ib_umem_release(hr_cq->umem);
        else
                /* Free the buff of stored cq */
                hns_roce_ib_free_cq_buf(hr_dev, &hr_cq->hr_buf, ib_cq->cqe);

        kfree(hr_cq);

        return 0;
}

void hns_roce_cq_completion(struct hns_roce_dev *hr_dev, u32 cqn)
{
        struct device *dev = &hr_dev->pdev->dev;
        struct hns_roce_cq *cq;

        cq = radix_tree_lookup(&hr_dev->cq_table.tree,
                               cqn & (hr_dev->caps.num_cqs - 1));
        if (!cq) {
                dev_warn(dev, "Completion event for bogus CQ 0x%08x\n", cqn);
                return;
        }

        cq->comp(cq);
}

void hns_roce_cq_event(struct hns_roce_dev *hr_dev, u32 cqn, int event_type)
{
        struct hns_roce_cq_table *cq_table = &hr_dev->cq_table;
        struct device *dev = &hr_dev->pdev->dev;
        struct hns_roce_cq *cq;

        cq = radix_tree_lookup(&cq_table->tree,
                               cqn & (hr_dev->caps.num_cqs - 1));
        if (cq)
                atomic_inc(&cq->refcount);

        if (!cq) {
                dev_warn(dev, "Async event for bogus CQ %08x\n", cqn);
                return;
        }

        cq->event(cq, (enum hns_roce_event)event_type);

        if (atomic_dec_and_test(&cq->refcount))
                complete(&cq->free);
}

int hns_roce_init_cq_table(struct hns_roce_dev *hr_dev)
{
        struct hns_roce_cq_table *cq_table = &hr_dev->cq_table;

        spin_lock_init(&cq_table->lock);
        INIT_RADIX_TREE(&cq_table->tree, GFP_ATOMIC);

        return hns_roce_bitmap_init(&cq_table->bitmap, hr_dev->caps.num_cqs,
                                    hr_dev->caps.num_cqs - 1,
                                    hr_dev->caps.reserved_cqs, 0);
}

void hns_roce_cleanup_cq_table(struct hns_roce_dev *hr_dev)
{
        hns_roce_bitmap_cleanup(&hr_dev->cq_table.bitmap);
}