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

#include "lan966x_main.h"

int lan966x_tbf_add(struct lan966x_port *port,
      struct tc_tbf_qopt_offload *qopt)
{
 struct lan966x *lan966x = port->lan966x;
 bool root = qopt->parent == TC_H_ROOT;
 u32 queue = 0;
 u32 cir, cbs;
 u32 se_idx;

 if (!root) {
  queue = TC_H_MIN(qopt->parent) - 1;
  if (queue >= NUM_PRIO_QUEUES)
   return -EOPNOTSUPP;
 }

 if (root)
  se_idx = SE_IDX_PORT + port->chip_port;
 else
  se_idx = SE_IDX_QUEUE + port->chip_port * NUM_PRIO_QUEUES + queue;

 cir = div_u64(qopt->replace_params.rate.rate_bytes_ps, 1000) * 8;
 cbs = qopt->replace_params.max_size;

 /* 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(0) |
  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_tbf_del(struct lan966x_port *port,
      struct tc_tbf_qopt_offload *qopt)
{
 struct lan966x *lan966x = port->lan966x;
 bool root = qopt->parent == TC_H_ROOT;
 u32 queue = 0;
 u32 se_idx;

 if (!root) {
  queue = TC_H_MIN(qopt->parent) - 1;
  if (queue >= NUM_PRIO_QUEUES)
   return -EOPNOTSUPP;
 }

 if (root)
  se_idx = SE_IDX_PORT + port->chip_port;
 else
  se_idx = SE_IDX_QUEUE + port->chip_port * NUM_PRIO_QUEUES + queue;

 lan_rmw(QSYS_SE_CFG_SE_AVB_ENA_SET(0) |
  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;
}