// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
// Copyright (c) 2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

#include "act.h"
#include "en/tc_priv.h"
#include "en/tc_ct.h"

static bool
tc_act_can_offload_ct(struct mlx5e_tc_act_parse_state *parse_state,
        const struct flow_action_entry *act,
        int act_index,
        struct mlx5_flow_attr *attr)
{
 return !((act->ct.action & TCA_CT_ACT_COMMIT) &&
   flow_action_is_last_entry(parse_state->flow_action, act));
}

static int
tc_act_parse_ct(struct mlx5e_tc_act_parse_state *parse_state,
  const struct flow_action_entry *act,
  struct mlx5e_priv *priv,
  struct mlx5_flow_attr *attr)
{
 int err;

 err = mlx5_tc_ct_parse_action(parse_state->ct_priv, attr, act, parse_state->extack);
 if (err)
  return err;

 if (mlx5e_is_eswitch_flow(parse_state->flow)) {
  attr->esw_attr->split_count = attr->esw_attr->out_count;
  parse_state->if_count = 0;
 }

 attr->flags |= MLX5_ATTR_FLAG_CT;

 return 0;
}

static int
tc_act_post_parse_ct(struct mlx5e_tc_act_parse_state *parse_state,
       struct mlx5e_priv *priv,
       struct mlx5_flow_attr *attr)
{
 if (!(attr->flags & MLX5_ATTR_FLAG_CT))
  return 0;

 return mlx5_tc_ct_flow_offload(parse_state->ct_priv, attr);
}

static bool
tc_act_is_multi_table_act_ct(struct mlx5e_priv *priv,
        const struct flow_action_entry *act,
        struct mlx5_flow_attr *attr)
{
 if (act->ct.action & TCA_CT_ACT_CLEAR)
  return false;

 return true;
}

static bool
tc_act_is_missable_ct(const struct flow_action_entry *act)
{
 return !(act->ct.action & TCA_CT_ACT_CLEAR);
}

struct mlx5e_tc_act mlx5e_tc_act_ct = {
 .can_offload = tc_act_can_offload_ct,
 .parse_action = tc_act_parse_ct,
 .post_parse = tc_act_post_parse_ct,
 .is_multi_table_act = tc_act_is_multi_table_act_ct,
 .is_missable = tc_act_is_missable_ct,
};