/*
 * flowprobe.c - skeleton vpp-api-test plug-in
 *
 * Copyright (c) <current-year> <your-organization>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include <vat/vat.h>
#include <vlibapi/api.h>
#include <vlibmemory/api.h>
#include <vppinfra/error.h>
#include <flowprobe/flowprobe.h>

#define __plugin_msg_base flowprobe_test_main.msg_id_base
#include <vlibapi/vat_helper_macros.h>

/**
 * @file vpp_api_test plugin
 */

uword unformat_sw_if_index (unformat_input_t * input, va_list * args);

/* Declare message IDs */
#include <flowprobe/flowprobe.api_enum.h>
#include <flowprobe/flowprobe.api_types.h>

typedef struct
{
    /** API message ID base */
  u16 msg_id_base;
    /** vat_main_t pointer */
  vat_main_t *vat_main;
} flowprobe_test_main_t;

flowprobe_test_main_t flowprobe_test_main;

static int
api_flowprobe_tx_interface_add_del (vat_main_t * vam)
{
  unformat_input_t *i = vam->input;
  int enable_disable = 1;
  u8 which = FLOW_VARIANT_IP4;
  u32 sw_if_index = ~0;
  vl_api_flowprobe_tx_interface_add_del_t *mp;
  int ret;

  /* Parse args required to build the message */
  while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
    {
      if (unformat (i, "%U", unformat_sw_if_index, vam, &sw_if_index))
	;
      else if (unformat (i, "sw_if_index %d", &sw_if_index))
	;
      else if (unformat (i, "disable"))
	enable_disable = 0;
      else if (unformat (i, "ip4"))
	which = FLOW_VARIANT_IP4;
      else if (unformat (i, "ip6"))
	which = FLOW_VARIANT_IP6;
      else if (unformat (i, "l2"))
	which = FLOW_VARIANT_L2;
      else
	break;
    }

  if (sw_if_index == ~0)
    {
      errmsg ("missing interface name / explicit sw_if_index number \n");
      return -99;
    }

  /* Construct the API message */
  M (FLOWPROBE_TX_INTERFACE_ADD_DEL, mp);
  mp->sw_if_index = ntohl (sw_if_index);
  mp->is_add = enable_disable;
  mp->which = which;

  /* send it... */
  S (mp);

  /* Wait for a reply... */
  W (ret);
  return ret;
}

static int
api_flowprobe_interface_add_del (vat_main_t *vam)
{
  unformat_input_t *i = vam->input;
  int enable_disable = 1;
  u8 which = FLOWPROBE_WHICH_IP4;
  u8 direction = FLOWPROBE_DIRECTION_TX;
  u32 sw_if_index = ~0;
  vl_api_flowprobe_interface_add_del_t *mp;
  int ret;

  /* Parse args required to build the message */
  while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
    {
      if (unformat (i, "%U", unformat_sw_if_index, vam, &sw_if_index))
	;
      else if (unformat (i, "sw_if_index %d", &sw_if_index))
	;
      else if (unformat (i, "disable"))
	enable_disable = 0;
      else if (unformat (i, "ip4"))
	which = FLOWPROBE_WHICH_IP4;
      else if (unformat (i, "ip6"))
	which = FLOWPROBE_WHICH_IP6;
      else if (unformat (i, "l2"))
	which = FLOWPROBE_WHICH_L2;
      else if (unformat (i, "rx"))
	direction = FLOWPROBE_DIRECTION_RX;
      else if (unformat (i, "tx"))
	direction = FLOWPROBE_DIRECTION_TX;
      else if (unformat (i, "both"))
	direction = FLOWPROBE_DIRECTION_BOTH;
      else
	break;
    }

  if (sw_if_index == ~0)
    {
      errmsg ("Missing interface name / explicit sw_if_index number\n");
      return -99;
    }

  /* Construct the API message */
  M (FLOWPROBE_INTERFACE_ADD_DEL, mp);
  mp->sw_if_index = ntohl (sw_if_index);
  mp->is_add = enable_disable;
  mp->which = which;
  mp->direction = direction;

  /* Send it... */
  S (mp);

  /* Wait for a reply... */
  W (ret);
  return ret;
}

static int
api_flowprobe_interface_dump (vat_main_t *vam)
{
  unformat_input_t *i = vam->input;
  vl_api_flowprobe_interface_dump_t *mp;
  vl_api_control_ping_t *mp_ping;
  u32 sw_if_index = ~0;
  int ret;

  /* Parse args required to build the message */
  while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
    {
      if (unformat (i, "%d", &sw_if_index))
	;
      else
	break;
    }

  /* Construct the API message */
  M (FLOWPROBE_INTERFACE_DUMP, mp);
  mp->sw_if_index = htonl (sw_if_index);

  /* Send it... */
  S (mp);

  /* Use control ping for synchronization */
  PING (&flowprobe_test_main, mp_ping);
  S (mp_ping);

  /* Wait for a reply... */
  W (ret);
  return ret;
}

static void
vl_api_flowprobe_interface_details_t_handler (
  vl_api_flowprobe_interface_details_t *mp)
{
  vat_main_t *vam = flowprobe_test_main.vat_main;
  u32 sw_if_index;
  u8 which;
  u8 direction;
  u8 *out = 0;
  const char *variants[] = {
    [FLOWPROBE_WHICH_IP4] = "ip4",
    [FLOWPROBE_WHICH_IP6] = "ip6",
    [FLOWPROBE_WHICH_L2] = "l2",
    "Erroneous variant",
  };
  const char *directions[] = {
    [FLOWPROBE_DIRECTION_RX] = "rx",
    [FLOWPROBE_DIRECTION_TX] = "tx",
    [FLOWPROBE_DIRECTION_BOTH] = "rx tx",
    "Erroneous direction",
  };

  sw_if_index = ntohl (mp->sw_if_index);

  which = mp->which;
  if (which > ARRAY_LEN (variants) - 2)
    which = ARRAY_LEN (variants) - 1;

  direction = mp->direction;
  if (direction > ARRAY_LEN (directions) - 2)
    direction = ARRAY_LEN (directions) - 1;

  out = format (0, "sw_if_index: %u, variant: %s, direction: %s\n%c",
		sw_if_index, variants[which], directions[direction], 0);

  fformat (vam->ofp, (char *) out);
  vec_free (out);
}

static int
api_flowprobe_params (vat_main_t * vam)
{
  unformat_input_t *i = vam->input;
  u32 active_timer = ~0;
  u32 passive_timer = ~0;
  vl_api_flowprobe_params_t *mp;
  int ret;
  u8 record_flags = 0;

  while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
    {
      if (unformat (i, "active %d", &active_timer))
	;
      else if (unformat (i, "passive %d", &passive_timer))
	;
      else if (unformat (i, "record"))
	while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
	  {
	    if (unformat (i, "l2"))
	      record_flags |= FLOWPROBE_RECORD_FLAG_L2;
	    else if (unformat (i, "l3"))
	      record_flags |= FLOWPROBE_RECORD_FLAG_L3;
	    else if (unformat (i, "l4"))
	      record_flags |= FLOWPROBE_RECORD_FLAG_L4;
	    else
	      break;
	  }
      else
	break;
    }

  if (passive_timer > 0 && active_timer > passive_timer)
    {
      errmsg ("Passive timer has to be greater than active one...\n");
      return -99;
    }

  /* Construct the API message */
  M (FLOWPROBE_PARAMS, mp);
  mp->record_flags = record_flags;
  mp->active_timer = ntohl (active_timer);
  mp->passive_timer = ntohl (passive_timer);

  /* send it... */
  S (mp);

  /* Wait for a reply... */
  W (ret);

  return ret;
}

static int
api_flowprobe_set_params (vat_main_t *vam)
{
  unformat_input_t *i = vam->input;
  vl_api_flowprobe_set_params_t *mp;
  u32 active_timer = ~0;
  u32 passive_timer = ~0;
  u8 record_flags = 0;
  int ret;

  while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
    {
      if (unformat (i, "active %d", &active_timer))
	;
      else if (unformat (i, "passive %d", &passive_timer))
	;
      else if (unformat (i, "record"))
	while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
	  {
	    if (unformat (i, "l2"))
	      record_flags |= FLOWPROBE_RECORD_FLAG_L2;
	    else if (unformat (i, "l3"))
	      record_flags |= FLOWPROBE_RECORD_FLAG_L3;
	    else if (unformat (i, "l4"))
	      record_flags |= FLOWPROBE_RECORD_FLAG_L4;
	    else
	      break;
	  }
      else
	break;
    }

  /* Construct the API message */
  M (FLOWPROBE_SET_PARAMS, mp);
  mp->record_flags = record_flags;
  mp->active_timer = ntohl (active_timer);
  mp->passive_timer = ntohl (passive_timer);

  /* send it... */
  S (mp);

  /* Wait for a reply... */
  W (ret);

  return ret;
}

static int
api_flowprobe_get_params (vat_main_t *vam)
{
  vl_api_flowprobe_get_params_t *mp;
  int ret;

  /* Construct the API message */
  M (FLOWPROBE_GET_PARAMS, mp);

  /* Send it... */
  S (mp);

  /* Wait for a reply... */
  W (ret);
  return ret;
}

static void
vl_api_flowprobe_get_params_reply_t_handler (
  vl_api_flowprobe_get_params_reply_t *mp)
{
  vat_main_t *vam = flowprobe_test_main.vat_main;
  u8 *out = 0;

  out =
    format (0, "active: %u, passive: %u, record:", ntohl (mp->active_timer),
	    ntohl (mp->passive_timer));

  if (mp->record_flags & FLOWPROBE_RECORD_FLAG_L2)
    out = format (out, " l2");
  if (mp->record_flags & FLOWPROBE_RECORD_FLAG_L3)
    out = format (out, " l3");
  if (mp->record_flags & FLOWPROBE_RECORD_FLAG_L4)
    out = format (out, " l4");

  out = format (out, "\n%c", 0);
  fformat (vam->ofp, (char *) out);
  vec_free (out);
  vam->result_ready = 1;
}

/*
 * List of messages that the api test plugin sends,
 * and that the data plane plugin processes
 */
#include <flowprobe/flowprobe.api_test.c>

/*
 * fd.io coding-style-patch-verification: ON
 *
 * Local Variables:
 * eval: (c-set-style "gnu")
 * End:
 */