// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) Intel Corporation
 * Author: Shao Guohua <guohua.shao@intel.com>
 */

#include <net/datapath_api.h>
#include <net/datapath_api_qos.h>
#include "datapath.h"

int dp_node_link_add(struct dp_node_link *info, int flag)
{
	struct inst_info *dp_info;

	if (unlikely((dp_late_init() <= 0) && !dp_cpu_init_ok)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!info || is_invalid_inst(info->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(info->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_LINK_ADD, info, flag);
}
EXPORT_SYMBOL(dp_node_link_add);

int dp_node_unlink(struct dp_node_link *info, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!info || is_invalid_inst(info->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(info->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_UNLINK, info, flag);
}
EXPORT_SYMBOL(dp_node_unlink);

int dp_node_link_get(struct dp_node_link *info, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!info || is_invalid_inst(info->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(info->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_LINK_GET, info, flag);
}
EXPORT_SYMBOL(dp_node_link_get);

int dp_node_link_en_set(struct dp_node_link_enable *en, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!en || is_invalid_inst(en->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(en->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_LINK_EN_SET, en, flag);
}
EXPORT_SYMBOL(dp_node_link_en_set);

int dp_node_link_en_get(struct dp_node_link_enable *en, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!en || is_invalid_inst(en->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(en->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_LINK_EN_GET, en, flag);
}
EXPORT_SYMBOL(dp_node_link_en_get);

int dp_link_add(struct dp_qos_link *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(LINK_ADD, cfg, flag);
}
EXPORT_SYMBOL(dp_link_add);

int dp_link_get(struct dp_qos_link *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(LINK_GET, cfg, flag);
}
EXPORT_SYMBOL(dp_link_get);

int dp_qos_link_prio_set(struct dp_node_prio *info, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!info || is_invalid_inst(info->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(info->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(LINK_PRIO_SET, info, flag);
}
EXPORT_SYMBOL(dp_qos_link_prio_set);

int dp_qos_link_prio_get(struct dp_node_prio *info, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!info || is_invalid_inst(info->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(info->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(LINK_PRIO_GET, info, flag);
}
EXPORT_SYMBOL(dp_qos_link_prio_get);

int dp_queue_conf_set(struct dp_queue_conf *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(QUEUE_CFG_SET, cfg, flag);
}
EXPORT_SYMBOL(dp_queue_conf_set);

int dp_queue_conf_get(struct dp_queue_conf *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg)
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(QUEUE_CFG_GET, cfg, flag);
}
EXPORT_SYMBOL(dp_queue_conf_get);

int dp_shaper_conf_set(struct dp_shaper_conf *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg)
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(SHAPER_SET, cfg, flag);
}
EXPORT_SYMBOL(dp_shaper_conf_set);

int dp_shaper_conf_get(struct dp_shaper_conf *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(SHAPER_GET, cfg, flag);
}
EXPORT_SYMBOL(dp_shaper_conf_get);

int dp_node_alloc(struct dp_node_alloc *node, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!node || is_invalid_inst(node->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(node->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_ALLOC, node, flag);
}
EXPORT_SYMBOL(dp_node_alloc);

int dp_node_free(struct dp_node_alloc *node, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!node || is_invalid_inst(node->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(node->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_FREE, node, flag);
}
EXPORT_SYMBOL(dp_node_free);

int dp_node_children_free(struct dp_node_alloc *node, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!node || is_invalid_inst(node->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(node->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_CHILDREN_FREE, node, flag);
}
EXPORT_SYMBOL(dp_node_children_free);

int dp_deq_port_res_get(struct dp_dequeue_res *res, int flag)
{
	dp_subif_t *subif;
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!res || is_invalid_inst(res->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(res->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;

	/* if res->dev is NULL, res->dp_port should be valid */
	if (res->dev) { /*fill dp_port if dev is provided */
		subif = kzalloc(sizeof(*subif), GFP_ATOMIC);
		if (!subif)
			return DP_FAILURE;
		dp_get_netif_subifid(res->dev, NULL, NULL, NULL, subif, 0);
		if (!subif->subif_num) {
			pr_err("Not found dev %s\n", res->dev->name);
			kfree(subif);
			return DP_FAILURE;
		}
		res->dp_port = subif->port_id;
		kfree(subif);
		subif = NULL;
	}
	DP_DEBUG(DP_DBG_FLAG_QOS_DETAIL, "%s: dp_port=%d tconf_idx=%d\n",
		 __func__, res->dp_port, res->cqm_deq_idx);
	return dp_info->dp_qos_platform_set(DEQ_PORT_RES_GET, res, flag);
}
EXPORT_SYMBOL(dp_deq_port_res_get);

int dp_counter_mode_set(struct dp_counter_conf *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(COUNTER_MODE_SET, cfg, flag);
}
EXPORT_SYMBOL(dp_counter_mode_set);

int dp_counter_mode_get(struct dp_counter_conf *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(COUNTER_MODE_GET, cfg, flag);
}
EXPORT_SYMBOL(dp_counter_mode_get);

int dp_queue_map_set(struct dp_queue_map_set *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(QUEUE_MAP_SET, cfg, flag);
}
EXPORT_SYMBOL(dp_queue_map_set);

int dp_queue_map_get(struct dp_queue_map_get *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(QUEUE_MAP_GET, cfg, flag);
}
EXPORT_SYMBOL(dp_queue_map_get);

int dp_children_get(struct dp_node_child *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(NODE_CHILDREN_GET, cfg, flag);
}
EXPORT_SYMBOL(dp_children_get);

int dp_qos_level_get(struct dp_qos_level *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(QOS_LEVEL_GET, cfg, flag);
}
EXPORT_SYMBOL(dp_qos_level_get);

int dp_qos_get_q_logic(struct dp_qos_q_logic *cfg, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set) {
		cfg->q_logic_id = cfg->q_id; /* For GRX500 */
		return DP_SUCCESS;
	}
	return dp_info->dp_qos_platform_set(QOS_Q_LOGIC, cfg, flag);
}
EXPORT_SYMBOL(dp_qos_get_q_logic);

int dp_qos_global_info_get(struct dp_qos_cfg_info *info, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!info || is_invalid_inst(info->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(info->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;
	return dp_info->dp_qos_platform_set(QOS_GLOBAL_CFG_GET, info, flag);
}
EXPORT_SYMBOL(dp_qos_global_info_get);

int dp_qos_port_conf_set(struct dp_port_cfg_info *info, int flag)
{
	struct inst_info *dp_info;

	if (unlikely(dp_late_init() <= 0)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!info || is_invalid_inst(info->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(info->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;

	return dp_info->dp_qos_platform_set(QOS_PORT_CFG_SET, info, flag);
}
EXPORT_SYMBOL(dp_qos_port_conf_set);

int dp_block_flush_port(struct dp_qos_blk_flush_port *cfg, int flag)
{
	struct inst_info *dp_info;
	int ret = DP_FAILURE;

	if (unlikely(!dp_init_ok)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;

	if (cfg->dp_port >= dp_info->cap.max_num_dp_ports) {
		pr_err("%s dp_port %d should be valid\n",
		       __func__, cfg->dp_port);
		return DP_FAILURE;
	}

	DP_LIB_LOCK(&dp_lock);
	ret = dp_info->dp_qos_platform_set(QOS_BLOCK_FLUSH_PORT, cfg, flag);
	DP_LIB_UNLOCK(&dp_lock);

	return ret;
}
EXPORT_SYMBOL(dp_block_flush_port);

int dp_block_flush_queue(struct dp_qos_blk_flush_queue *cfg, int flag)
{
	struct inst_info *dp_info;
	int ret = DP_FAILURE;

	if (unlikely(!dp_init_ok)) {
		pr_err("%s failed: datapath not initialized yet\n", __func__);
		return DP_FAILURE;
	}

	if (!cfg || is_invalid_inst(cfg->inst))
		return DP_FAILURE;

	dp_info = get_dp_prop_info(cfg->inst);

	if (!dp_info || !dp_info->dp_qos_platform_set)
		return DP_FAILURE;

	DP_LIB_LOCK(&dp_lock);
	ret = dp_info->dp_qos_platform_set(QOS_BLOCK_FLUSH_QUEUE, cfg, flag);
	DP_LIB_UNLOCK(&dp_lock);

	return ret;
}
EXPORT_SYMBOL(dp_block_flush_queue);