// SPDX-License-Identifier: GPL-2.0
// Copyright (c) 2025 Broadcom.

#include <linux/module.h>
#include <linux/pci.h>
#include <linux/auxiliary_bus.h>

#include <rdma/ib_verbs.h>

#include "bng_res.h"
#include "bng_sp.h"
#include "bng_fw.h"
#include "bnge.h"
#include "bnge_auxr.h"
#include "bng_re.h"
#include "bnge_hwrm.h"
#include "bng_debugfs.h"

MODULE_AUTHOR("Siva Reddy Kallam <siva.kallam@broadcom.com>");
MODULE_DESCRIPTION(BNG_RE_DESC);
MODULE_LICENSE("Dual BSD/GPL");

static struct bng_re_dev *bng_re_dev_add(struct auxiliary_device *adev,
                                         struct bnge_auxr_dev *aux_dev)
{
        struct bng_re_dev *rdev;

        /* Allocate bng_re_dev instance */
        rdev = ib_alloc_device(bng_re_dev, ibdev);
        if (!rdev) {
                pr_err("%s: bng_re_dev allocation failure!", KBUILD_MODNAME);
                return NULL;
        }

        /* Assign auxiliary device specific data */
        rdev->netdev = aux_dev->net;
        rdev->aux_dev = aux_dev;
        rdev->adev = adev;
        rdev->fn_id = rdev->aux_dev->pdev->devfn;

        return rdev;
}


static int bng_re_register_netdev(struct bng_re_dev *rdev)
{
        struct bnge_auxr_dev *aux_dev;

        aux_dev = rdev->aux_dev;
        return bnge_register_dev(aux_dev, rdev->adev);
}

static void bng_re_destroy_chip_ctx(struct bng_re_dev *rdev)
{
        struct bng_re_chip_ctx *chip_ctx;

        kfree(rdev->dev_attr);
        rdev->dev_attr = NULL;

        chip_ctx = rdev->chip_ctx;
        rdev->chip_ctx = NULL;
        rdev->rcfw.res = NULL;
        rdev->bng_res.cctx = NULL;
        rdev->bng_res.pdev = NULL;
        kfree(chip_ctx);
}

static int bng_re_setup_chip_ctx(struct bng_re_dev *rdev)
{
        struct bng_re_chip_ctx *chip_ctx;
        struct bnge_auxr_dev *aux_dev;
        int rc = -ENOMEM;

        aux_dev = rdev->aux_dev;
        rdev->bng_res.pdev = aux_dev->pdev;
        rdev->rcfw.res = &rdev->bng_res;
        chip_ctx = kzalloc_obj(*chip_ctx);
        if (!chip_ctx)
                return -ENOMEM;
        chip_ctx->chip_num = aux_dev->chip_num;
        chip_ctx->hw_stats_size = aux_dev->hw_ring_stats_size;

        rdev->chip_ctx = chip_ctx;
        rdev->bng_res.cctx = rdev->chip_ctx;
        rdev->dev_attr = kzalloc_obj(*rdev->dev_attr);
        if (!rdev->dev_attr)
                goto free_chip_ctx;
        rdev->bng_res.dattr = rdev->dev_attr;

        return 0;
free_chip_ctx:
        kfree(rdev->chip_ctx);
        rdev->chip_ctx = NULL;
        return rc;
}

static void bng_re_init_hwrm_hdr(struct input *hdr, u16 opcd)
{
        hdr->req_type = cpu_to_le16(opcd);
        hdr->cmpl_ring = cpu_to_le16(-1);
        hdr->target_id = cpu_to_le16(-1);
}

static void bng_re_fill_fw_msg(struct bnge_fw_msg *fw_msg, void *msg,
                               int msg_len, void *resp, int resp_max_len,
                               int timeout)
{
        fw_msg->msg = msg;
        fw_msg->msg_len = msg_len;
        fw_msg->resp = resp;
        fw_msg->resp_max_len = resp_max_len;
        fw_msg->timeout = timeout;
}

static int bng_re_net_ring_free(struct bng_re_dev *rdev,
                                u16 fw_ring_id, int type)
{
        struct bnge_auxr_dev *aux_dev = rdev->aux_dev;
        struct hwrm_ring_free_input req = {};
        struct hwrm_ring_free_output resp;
        struct bnge_fw_msg fw_msg = {};
        int rc = -EINVAL;

        bng_re_init_hwrm_hdr((void *)&req, HWRM_RING_FREE);
        req.ring_type = type;
        req.ring_id = cpu_to_le16(fw_ring_id);
        bng_re_fill_fw_msg(&fw_msg, (void *)&req, sizeof(req), (void *)&resp,
                            sizeof(resp), BNGE_DFLT_HWRM_CMD_TIMEOUT);
        rc = bnge_send_msg(aux_dev, &fw_msg);
        if (rc)
                ibdev_err(&rdev->ibdev, "Failed to free HW ring:%d :%#x",
                          req.ring_id, rc);
        return rc;
}

static int bng_re_net_ring_alloc(struct bng_re_dev *rdev,
                                 struct bng_re_ring_attr *ring_attr,
                                 u16 *fw_ring_id)
{
        struct bnge_auxr_dev *aux_dev = rdev->aux_dev;
        struct hwrm_ring_alloc_input req = {};
        struct hwrm_ring_alloc_output resp;
        struct bnge_fw_msg fw_msg = {};
        int rc;

        bng_re_init_hwrm_hdr((void *)&req, HWRM_RING_ALLOC);
        req.enables = 0;
        req.page_tbl_addr =  cpu_to_le64(ring_attr->dma_arr[0]);
        if (ring_attr->pages > 1) {
                /* Page size is in log2 units */
                req.page_size = BNGE_PAGE_SHIFT;
                req.page_tbl_depth = 1;
        }
        req.fbo = 0;
        /* Association of ring index with doorbell index and MSIX number */
        req.logical_id = cpu_to_le16(ring_attr->lrid);
        req.length = cpu_to_le32(ring_attr->depth + 1);
        req.ring_type = ring_attr->type;
        req.int_mode = ring_attr->mode;
        bng_re_fill_fw_msg(&fw_msg, (void *)&req, sizeof(req), (void *)&resp,
                           sizeof(resp), BNGE_DFLT_HWRM_CMD_TIMEOUT);
        rc = bnge_send_msg(aux_dev, &fw_msg);
        if (!rc)
                *fw_ring_id = le16_to_cpu(resp.ring_id);

        return rc;
}

static int bng_re_stats_ctx_free(struct bng_re_dev *rdev)
{
        struct bnge_auxr_dev *aux_dev = rdev->aux_dev;
        struct hwrm_stat_ctx_free_input req = {};
        struct hwrm_stat_ctx_free_output resp = {};
        struct bnge_fw_msg fw_msg = {};
        int rc;

        bng_re_init_hwrm_hdr((void *)&req, HWRM_STAT_CTX_FREE);
        req.stat_ctx_id = cpu_to_le32(rdev->stats_ctx.fw_id);
        bng_re_fill_fw_msg(&fw_msg, (void *)&req, sizeof(req), (void *)&resp,
                           sizeof(resp), BNGE_DFLT_HWRM_CMD_TIMEOUT);
        rc = bnge_send_msg(aux_dev, &fw_msg);
        if (rc)
                ibdev_err(&rdev->ibdev, "Failed to free HW stats context %#x",
                          rc);

        return rc;
}

static int bng_re_stats_ctx_alloc(struct bng_re_dev *rdev)
{
        struct bnge_auxr_dev *aux_dev = rdev->aux_dev;
        struct bng_re_stats *stats = &rdev->stats_ctx;
        struct hwrm_stat_ctx_alloc_output resp = {};
        struct hwrm_stat_ctx_alloc_input req = {};
        struct bnge_fw_msg fw_msg = {};
        int rc;

        stats->fw_id = BNGE_INVALID_STATS_CTX_ID;

        bng_re_init_hwrm_hdr((void *)&req, HWRM_STAT_CTX_ALLOC);
        req.update_period_ms = cpu_to_le32(1000);
        req.stats_dma_addr = cpu_to_le64(stats->dma_map);
        req.stats_dma_length = cpu_to_le16(rdev->chip_ctx->hw_stats_size);
        req.stat_ctx_flags = STAT_CTX_ALLOC_REQ_STAT_CTX_FLAGS_ROCE;
        bng_re_fill_fw_msg(&fw_msg, (void *)&req, sizeof(req), (void *)&resp,
                           sizeof(resp), BNGE_DFLT_HWRM_CMD_TIMEOUT);
        rc = bnge_send_msg(aux_dev, &fw_msg);
        if (!rc)
                stats->fw_id = le32_to_cpu(resp.stat_ctx_id);
        return rc;
}

static int bng_re_query_hwrm_version(struct bng_re_dev *rdev)
{
        struct bnge_auxr_dev *aux_dev = rdev->aux_dev;
        struct hwrm_ver_get_output ver_get_resp = {};
        struct hwrm_ver_get_input ver_get_req = {};
        struct bng_re_chip_ctx *cctx;
        struct bnge_fw_msg fw_msg = {};
        int rc;

        bng_re_init_hwrm_hdr((void *)&ver_get_req, HWRM_VER_GET);
        ver_get_req.hwrm_intf_maj = HWRM_VERSION_MAJOR;
        ver_get_req.hwrm_intf_min = HWRM_VERSION_MINOR;
        ver_get_req.hwrm_intf_upd = HWRM_VERSION_UPDATE;
        bng_re_fill_fw_msg(&fw_msg, (void *)&ver_get_req, sizeof(ver_get_req),
                            (void *)&ver_get_resp, sizeof(ver_get_resp),
                            BNGE_DFLT_HWRM_CMD_TIMEOUT);
        rc = bnge_send_msg(aux_dev, &fw_msg);
        if (rc) {
                ibdev_err(&rdev->ibdev, "Failed to query HW version, rc = 0x%x",
                          rc);
                return rc;
        }

        cctx = rdev->chip_ctx;
        cctx->hwrm_intf_ver =
                (u64)le16_to_cpu(ver_get_resp.hwrm_intf_major) << 48 |
                (u64)le16_to_cpu(ver_get_resp.hwrm_intf_minor) << 32 |
                (u64)le16_to_cpu(ver_get_resp.hwrm_intf_build) << 16 |
                le16_to_cpu(ver_get_resp.hwrm_intf_patch);

        cctx->hwrm_cmd_max_timeout = le16_to_cpu(ver_get_resp.max_req_timeout);

        if (!cctx->hwrm_cmd_max_timeout)
                cctx->hwrm_cmd_max_timeout = BNG_ROCE_FW_MAX_TIMEOUT;

        return 0;
}

static void bng_re_dev_uninit(struct bng_re_dev *rdev)
{
        int rc;
        bng_re_debugfs_rem_pdev(rdev);

        if (test_and_clear_bit(BNG_RE_FLAG_RCFW_CHANNEL_EN, &rdev->flags)) {
                rc = bng_re_deinit_rcfw(&rdev->rcfw);
                if (rc)
                        ibdev_warn(&rdev->ibdev,
                                   "Failed to deinitialize RCFW: %#x", rc);
                bng_re_stats_ctx_free(rdev);
                bng_re_free_stats_ctx_mem(rdev->bng_res.pdev, &rdev->stats_ctx);
                bng_re_disable_rcfw_channel(&rdev->rcfw);
                bng_re_net_ring_free(rdev, rdev->rcfw.creq.ring_id,
                             RING_ALLOC_REQ_RING_TYPE_NQ);
                bng_re_free_rcfw_channel(&rdev->rcfw);
        }

        kfree(rdev->nqr);
        rdev->nqr = NULL;
        bng_re_destroy_chip_ctx(rdev);
        if (test_and_clear_bit(BNG_RE_FLAG_NETDEV_REGISTERED, &rdev->flags))
                bnge_unregister_dev(rdev->aux_dev);
}

static int bng_re_dev_init(struct bng_re_dev *rdev)
{
        struct bng_re_ring_attr rattr = {};
        struct bng_re_creq_ctx *creq;
        u32 db_offt;
        int vid;
        u8 type;
        int rc;

        /* Registered a new RoCE device instance to netdev */
        rc = bng_re_register_netdev(rdev);
        if (rc) {
                ibdev_err(&rdev->ibdev,
                                "Failed to register with netedev: %#x\n", rc);
                goto reg_netdev_fail;
        }

        set_bit(BNG_RE_FLAG_NETDEV_REGISTERED, &rdev->flags);

        if (rdev->aux_dev->auxr_info->msix_requested < BNG_RE_MIN_MSIX) {
                ibdev_err(&rdev->ibdev,
                          "RoCE requires minimum 2 MSI-X vectors, but only %d reserved\n",
                          rdev->aux_dev->auxr_info->msix_requested);
                rc = -EINVAL;
                goto msix_ctx_fail;
        }
        ibdev_dbg(&rdev->ibdev, "Got %d MSI-X vectors\n",
                  rdev->aux_dev->auxr_info->msix_requested);

        rc = bng_re_setup_chip_ctx(rdev);
        if (rc) {
                ibdev_err(&rdev->ibdev, "Failed to get chip context\n");
                goto msix_ctx_fail;
        }

        rc = bng_re_query_hwrm_version(rdev);
        if (rc)
                goto destroy_chip_ctx;

        rc = bng_re_alloc_fw_channel(&rdev->bng_res, &rdev->rcfw);
        if (rc) {
                ibdev_err(&rdev->ibdev,
                          "Failed to allocate RCFW Channel: %#x\n", rc);
                goto destroy_chip_ctx;
        }

        /* Allocate nq record memory */
        rdev->nqr = kzalloc_obj(*rdev->nqr);
        if (!rdev->nqr) {
                rc = -ENOMEM;
                goto nq_alloc_fail;
        }

        rdev->nqr->num_msix = rdev->aux_dev->auxr_info->msix_requested;
        memcpy(rdev->nqr->msix_entries, rdev->aux_dev->msix_info,
               sizeof(struct bnge_msix_info) * rdev->nqr->num_msix);

        type = RING_ALLOC_REQ_RING_TYPE_NQ;
        creq = &rdev->rcfw.creq;
        rattr.dma_arr = creq->hwq.pbl[BNG_PBL_LVL_0].pg_map_arr;
        rattr.pages = creq->hwq.pbl[creq->hwq.level].pg_count;
        rattr.type = type;
        rattr.mode = RING_ALLOC_REQ_INT_MODE_MSIX;
        rattr.depth = BNG_FW_CREQE_MAX_CNT - 1;
        rattr.lrid = rdev->nqr->msix_entries[BNG_RE_CREQ_NQ_IDX].ring_idx;
        rc = bng_re_net_ring_alloc(rdev, &rattr, &creq->ring_id);
        if (rc) {
                ibdev_err(&rdev->ibdev, "Failed to allocate CREQ: %#x\n", rc);
                goto free_rcfw;
        }
        db_offt = rdev->nqr->msix_entries[BNG_RE_CREQ_NQ_IDX].db_offset;
        vid = rdev->nqr->msix_entries[BNG_RE_CREQ_NQ_IDX].vector;

        rc = bng_re_enable_fw_channel(&rdev->rcfw,
                                        vid, db_offt);
        if (rc) {
                ibdev_err(&rdev->ibdev, "Failed to enable RCFW channel: %#x\n",
                          rc);
                goto free_ring;
        }

        rc = bng_re_get_dev_attr(&rdev->rcfw);
        if (rc)
                goto disable_rcfw;

        bng_re_debugfs_add_pdev(rdev);
        rc = bng_re_alloc_stats_ctx_mem(rdev->bng_res.pdev, rdev->chip_ctx,
                                        &rdev->stats_ctx);
        if (rc) {
                ibdev_err(&rdev->ibdev,
                          "Failed to allocate stats context: %#x\n", rc);
                goto disable_rcfw;
        }

        rc = bng_re_stats_ctx_alloc(rdev);
        if (rc) {
                ibdev_err(&rdev->ibdev,
                          "Failed to allocate QPLIB context: %#x\n", rc);
                goto free_stats_ctx;
        }

        rc = bng_re_init_rcfw(&rdev->rcfw, &rdev->stats_ctx);
        if (rc) {
                ibdev_err(&rdev->ibdev,
                          "Failed to initialize RCFW: %#x\n", rc);
                goto free_sctx;
        }
        set_bit(BNG_RE_FLAG_RCFW_CHANNEL_EN, &rdev->flags);

        return 0;
free_sctx:
        bng_re_stats_ctx_free(rdev);
free_stats_ctx:
        bng_re_free_stats_ctx_mem(rdev->bng_res.pdev, &rdev->stats_ctx);
disable_rcfw:
        bng_re_disable_rcfw_channel(&rdev->rcfw);
free_ring:
        bng_re_net_ring_free(rdev, rdev->rcfw.creq.ring_id, type);
free_rcfw:
        kfree(rdev->nqr);
nq_alloc_fail:
        bng_re_free_rcfw_channel(&rdev->rcfw);
destroy_chip_ctx:
        bng_re_destroy_chip_ctx(rdev);
msix_ctx_fail:
        bnge_unregister_dev(rdev->aux_dev);
        clear_bit(BNG_RE_FLAG_NETDEV_REGISTERED, &rdev->flags);
reg_netdev_fail:
        return rc;
}

static int bng_re_add_device(struct auxiliary_device *adev)
{
        struct bnge_auxr_priv *auxr_priv =
                container_of(adev, struct bnge_auxr_priv, aux_dev);
        struct bng_re_en_dev_info *dev_info;
        struct bng_re_dev *rdev;
        int rc;

        dev_info = auxiliary_get_drvdata(adev);

        rdev = bng_re_dev_add(adev, auxr_priv->auxr_dev);
        if (!rdev) {
                rc = -ENOMEM;
                goto exit;
        }

        dev_info->rdev = rdev;

        rc = bng_re_dev_init(rdev);
        if (rc)
                goto re_dev_dealloc;

        return 0;

re_dev_dealloc:
        ib_dealloc_device(&rdev->ibdev);
exit:
        return rc;
}


static void bng_re_remove_device(struct bng_re_dev *rdev,
                                 struct auxiliary_device *aux_dev)
{
        bng_re_dev_uninit(rdev);
        ib_dealloc_device(&rdev->ibdev);
}


static int bng_re_probe(struct auxiliary_device *adev,
                        const struct auxiliary_device_id *id)
{
        struct bnge_auxr_priv *aux_priv =
                container_of(adev, struct bnge_auxr_priv, aux_dev);
        struct bng_re_en_dev_info *en_info;
        int rc;

        en_info = kzalloc_obj(*en_info);
        if (!en_info)
                return -ENOMEM;

        en_info->auxr_dev = aux_priv->auxr_dev;

        auxiliary_set_drvdata(adev, en_info);

        rc = bng_re_add_device(adev);
        if (rc)
                kfree(en_info);

        return rc;
}

static void bng_re_remove(struct auxiliary_device *adev)
{
        struct bng_re_en_dev_info *dev_info = auxiliary_get_drvdata(adev);
        struct bng_re_dev *rdev;

        rdev = dev_info->rdev;

        bng_re_remove_device(rdev, adev);
        kfree(dev_info);
}

static const struct auxiliary_device_id bng_re_id_table[] = {
        { .name = BNG_RE_ADEV_NAME ".rdma", },
        {},
};

MODULE_DEVICE_TABLE(auxiliary, bng_re_id_table);

static struct auxiliary_driver bng_re_driver = {
        .name = "rdma",
        .probe = bng_re_probe,
        .remove = bng_re_remove,
        .id_table = bng_re_id_table,
};

static int __init bng_re_mod_init(void)
{
        int rc;


        bng_re_register_debugfs();

        rc = auxiliary_driver_register(&bng_re_driver);
        if (rc) {
                pr_err("%s: Failed to register auxiliary driver\n",
                       KBUILD_MODNAME);
                goto unreg_debugfs;
        }
        return 0;
unreg_debugfs:
        bng_re_unregister_debugfs();
        return rc;
}

static void __exit bng_re_mod_exit(void)
{
        auxiliary_driver_unregister(&bng_re_driver);
        bng_re_unregister_debugfs();
}

module_init(bng_re_mod_init);
module_exit(bng_re_mod_exit);