// SPDX-License-Identifier: GPL-2.0+

#include "lan966x_main.h"

int lan966x_cbs_add(struct lan966x_port *port,
                    struct tc_cbs_qopt_offload *qopt)
{
        struct lan966x *lan966x = port->lan966x;
        u32 cir, cbs;
        u8 se_idx;

        /* Check for invalid values */
        if (qopt->idleslope <= 0 ||
            qopt->sendslope >= 0 ||
            qopt->locredit >= qopt->hicredit)
                return -EINVAL;

        se_idx = SE_IDX_QUEUE + port->chip_port * NUM_PRIO_QUEUES + qopt->queue;
        cir = qopt->idleslope;
        cbs = (qopt->idleslope - qopt->sendslope) *
                (qopt->hicredit - qopt->locredit) /
                -qopt->sendslope;

        /* Rate unit is 100 kbps */
        cir = DIV_ROUND_UP(cir, 100);
        /* Avoid using zero rate */
        cir = cir ?: 1;
        /* Burst unit is 4kB */
        cbs = DIV_ROUND_UP(cbs, 4096);
        /* Avoid using zero burst */
        cbs = cbs ?: 1;

        /* Check that actually the result can be written */
        if (cir > GENMASK(15, 0) ||
            cbs > GENMASK(6, 0))
                return -EINVAL;

        lan_rmw(QSYS_SE_CFG_SE_AVB_ENA_SET(1) |
                QSYS_SE_CFG_SE_FRM_MODE_SET(1),
                QSYS_SE_CFG_SE_AVB_ENA |
                QSYS_SE_CFG_SE_FRM_MODE,
                lan966x, QSYS_SE_CFG(se_idx));

        lan_wr(QSYS_CIR_CFG_CIR_RATE_SET(cir) |
               QSYS_CIR_CFG_CIR_BURST_SET(cbs),
               lan966x, QSYS_CIR_CFG(se_idx));

        return 0;
}

int lan966x_cbs_del(struct lan966x_port *port,
                    struct tc_cbs_qopt_offload *qopt)
{
        struct lan966x *lan966x = port->lan966x;
        u8 se_idx;

        se_idx = SE_IDX_QUEUE + port->chip_port * NUM_PRIO_QUEUES + qopt->queue;

        lan_rmw(QSYS_SE_CFG_SE_AVB_ENA_SET(1) |
                QSYS_SE_CFG_SE_FRM_MODE_SET(0),
                QSYS_SE_CFG_SE_AVB_ENA |
                QSYS_SE_CFG_SE_FRM_MODE,
                lan966x, QSYS_SE_CFG(se_idx));

        lan_wr(QSYS_CIR_CFG_CIR_RATE_SET(0) |
               QSYS_CIR_CFG_CIR_BURST_SET(0),
               lan966x, QSYS_CIR_CFG(se_idx));

        return 0;
}