/*
 * Copyright (c) 2017-2019 Cisco and/or its affiliates.
 * 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 "face_udp.h"
#include "dpo_udp.h"

#include <vnet/vnet.h>
#include <vnet/dpo/dpo.h>
#include <vlib/vlib.h>
#include <vnet/ip/format.h>

#define HICN_FACE_NONE 0
#define HICN_FACE_DELETE 1
#define HICN_FACE_ADD 2


static clib_error_t *
hicn_face_udp_cli_set_command_fn (vlib_main_t * vm,
				  unformat_input_t * main_input,
				  vlib_cli_command_t * cmd)
{
  vnet_main_t *vnm = vnet_get_main ();
  ip46_address_t src_addr;
  u32 src_port = 0;
  ip46_address_t dst_addr;
  u32 dst_port = 0;
  hicn_face_id_t face_id = HICN_FACE_NULL;
  int ret = HICN_ERROR_NONE;
  int sw_if;
  int face_op = HICN_FACE_NONE;

  ip46_address_reset (&src_addr);
  ip46_address_reset (&dst_addr);
  /* Get a line of input. */
  unformat_input_t _line_input, *line_input = &_line_input;
  if (!unformat_user (main_input, unformat_line_input, line_input))
    {
      return (0);
    }

  while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT)
    {
      if (unformat (line_input, "del"))
	{
	  if (unformat (line_input, "id %d", &face_id))
	    face_op = HICN_FACE_DELETE;
	  else
	    {
	      return clib_error_return (0, "missing face id");
	    }
	}
      else if (unformat (line_input, "add"))
	{
	  face_op = HICN_FACE_ADD;
	  if (unformat
	      (line_input, "src_addr %U port %u dst_addr %U port %u intfc %U",
	       unformat_ip46_address, &src_addr, IP46_TYPE_ANY, &src_port,
	       unformat_ip46_address, &dst_addr, IP46_TYPE_ANY, &dst_port,
	       unformat_vnet_sw_interface, vnm, &sw_if));
	  else
	    {
	      return clib_error_return (0, "%s '%U'",
					get_error_string
					(HICN_ERROR_CLI_INVAL),
					format_unformat_error, line_input);
	    }
	}
      else
	{
	  return clib_error_return (0, "%s '%U'",
				    get_error_string (HICN_ERROR_CLI_INVAL),
				    format_unformat_error, line_input);
	}
    }

  if (face_id != HICN_FACE_NULL)
    {
      if (!hicn_dpoi_idx_is_valid (face_id))
	{
	  return clib_error_return (0, "%s, face_id %d not valid",
				    get_error_string (ret), face_id);
	}
    }

  int rv;
  switch (face_op)
    {
    case HICN_FACE_ADD:

      /* Check for presence of next hop address */
      if (((dst_addr.as_u64[0] == (u64) 0) && (dst_addr.as_u64[1] == (u64) 0))
	  || dst_port == 0)
	{
	  return clib_error_return (0, "dst address and port not specified");
	}

      if (((src_addr.as_u64[0] == (u64) 0) && (src_addr.as_u64[1] == (u64) 0))
	  || src_port == 0)
	{
	  return clib_error_return (0, "src address not specified");
	}

      rv =
	hicn_face_udp_add (&src_addr, &dst_addr,
			   clib_host_to_net_u16 (src_port),
			   clib_host_to_net_u16 (dst_port), sw_if, &face_id);
      if (rv == HICN_ERROR_NONE)
	{
	  vlib_cli_output (vm, "Face id: %d", face_id);
	}
      else
	{
	  return clib_error_return (0, get_error_string (rv));
	}
      break;
    case HICN_FACE_DELETE:
      rv = hicn_face_udp_del (face_id);
      if (rv == HICN_ERROR_NONE)
	{
	  vlib_cli_output (vm, "Face %d deleted", face_id);
	}
      else
	{
	  return clib_error_return (0, get_error_string (rv));
	}
      break;
    default:
      return clib_error_return (0, "Operation (%d) not implemented", face_op);
      break;
    }
  return (rv == HICN_ERROR_NONE) ? 0 : clib_error_return (0, "%s\n",
							  get_error_string
							  (rv));
}

/* cli declaration for 'cfg face' */
/* *INDENT-OFF* */
VLIB_CLI_COMMAND (hicn_face_udp_cli_set_command, static) =
{
  .path = "hicn face udp",
  .short_help = "hicn face udp {add src_addr <src_address> port <src_port > dst_addr <dst_address> port <dst_port>} intfc <interface> | {del id <face_id>}",
  .function = hicn_face_udp_cli_set_command_fn,
};
/* *INDENT-ON* */

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