/* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */
/* Copyright (c) 2025 NVIDIA Corporation & Affiliates */

#ifndef _MLX5_FS_HWS_
#define _MLX5_FS_HWS_

#include "mlx5hws.h"
#include "fs_hws_pools.h"

struct mlx5_fs_hws_actions_pool {
	struct mlx5hws_action *tag_action;
	struct mlx5hws_action *pop_vlan_action;
	struct mlx5hws_action *push_vlan_action;
	struct mlx5hws_action *drop_action;
	struct mlx5hws_action *decapl2_action;
	struct mlx5hws_action *remove_hdr_vlan_action;
	struct mlx5_fs_pool insert_hdr_pool;
	struct mlx5_fs_pool dl3tnltol2_pool;
	struct xarray el2tol3tnl_pools;
	struct xarray el2tol2tnl_pools;
	struct xarray mh_pools;
	struct xarray table_dests;
	struct xarray vport_vhca_dests;
	struct xarray vport_dests;
	struct xarray aso_meters;
	struct xarray sample_dests;
};

struct mlx5_fs_hws_context {
	struct mlx5hws_context	*hws_ctx;
	struct mlx5_fs_hws_actions_pool hws_pool;
};

struct mlx5_fs_hws_table {
	struct mlx5hws_table *hws_table;
	bool miss_ft_set;
};

struct mlx5_fs_hws_action {
	struct mlx5hws_action *hws_action;
	struct mlx5_fs_pool *fs_pool;
	struct mlx5_fs_hws_pr *pr_data;
	struct mlx5_fs_hws_mh *mh_data;
	u32 fw_reformat_id;
	/* Protect `fw_reformat_id` against being initialized from multiple
	 * threads.
	 */
	struct mutex lock;
};

struct mlx5_fs_hws_matcher {
	struct mlx5hws_bwc_matcher *matcher;
};

struct mlx5_fs_hws_rule_action {
	struct mlx5hws_action *action;
	union {
		struct mlx5_fc *counter;
		struct mlx5_exe_aso *exe_aso;
		u32 sampler_id;
	};
};

struct mlx5_fs_hws_rule {
	struct mlx5hws_bwc_rule *bwc_rule;
	struct mlx5_fs_hws_rule_action *hws_fs_actions;
	int num_fs_actions;
};

struct mlx5_fs_hws_data {
	struct mlx5hws_action *hws_action;
	struct mutex lock; /* protects hws_action */
	refcount_t hws_action_refcount;
};

struct mlx5_fs_hws_create_action_ctx {
	enum mlx5hws_action_type actions_type;
	struct mlx5hws_context *hws_ctx;
	u32 id;
	union {
		u8 return_reg_id;
	};
};

struct mlx5hws_action *
mlx5_fs_get_hws_action(struct mlx5_fs_hws_data *fs_hws_data,
		       struct mlx5_fs_hws_create_action_ctx *create_ctx);
void mlx5_fs_put_hws_action(struct mlx5_fs_hws_data *fs_hws_data);

#ifdef CONFIG_MLX5_HW_STEERING

int
mlx5_fs_hws_action_get_pkt_reformat_id(struct mlx5_pkt_reformat *pkt_reformat,
				       u32 *reformat_id);

bool mlx5_fs_hws_is_supported(struct mlx5_core_dev *dev);

const struct mlx5_flow_cmds *mlx5_fs_cmd_get_hws_cmds(void);

#else

static inline int
mlx5_fs_hws_action_get_pkt_reformat_id(struct mlx5_pkt_reformat *pkt_reformat,
				       u32 *reformat_id)
{
	return -EOPNOTSUPP;
}

static inline bool mlx5_fs_hws_is_supported(struct mlx5_core_dev *dev)
{
	return false;
}

static inline const struct mlx5_flow_cmds *mlx5_fs_cmd_get_hws_cmds(void)
{
	return NULL;
}

#endif /* CONFIG_MLX5_HWS_STEERING */
#endif