/*
 * Copyright (c) 2015 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 <vnet/lawful-intercept/lawful_intercept.h>

static clib_error_t *
set_li_command_fn (vlib_main_t * vm,
		 unformat_input_t * input,
		 vlib_cli_command_t * cmd)
{
  li_main_t * lm = &li_main;
  ip4_address_t collector;
  u8 collector_set = 0;
  ip4_address_t src;
  u8 src_set = 0;
  u32 tmp;
  u16 udp_port = 0;
  u8 is_add = 1;
  int i;

  while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) {
    if (unformat (input, "collector %U", unformat_ip4_address, &collector))
      collector_set = 1;
    if (unformat (input, "src %U", unformat_ip4_address, &src))
      src_set = 1;
    else if (unformat (input, "udp-port %d", &tmp))
      udp_port = tmp;
    else if (unformat (input, "del"))
      is_add = 0;
    else
      break;
  }

  if (collector_set == 0)
    return clib_error_return (0, "collector must be set...");
  if (src_set == 0)
    return clib_error_return (0, "src must be set...");
  if (udp_port == 0)
    return clib_error_return (0, "udp-port must be set...");

  if (is_add == 1)
    {
      for (i = 0; i < vec_len (lm->collectors); i++)
        {
          if (lm->collectors[i].as_u32 == collector.as_u32)
            {
              if (lm->ports[i] == udp_port)
                return clib_error_return 
                  (0, "collector %U:%d already configured", 
                   &collector, udp_port);
              else
                return clib_error_return
                  (0, "collector %U already configured with port %d", 
                   &collector, (int)(lm->ports[i]));
            }
        }
      vec_add1 (lm->collectors, collector);
      vec_add1 (lm->ports, udp_port);
      vec_add1 (lm->src_addrs, src);
      return 0;
    }
  else
    {
      for (i = 0; i < vec_len (lm->collectors); i++)
        {
          if ((lm->collectors[i].as_u32 == collector.as_u32)
              && lm->ports[i] == udp_port)
            {
              vec_delete (lm->collectors, 1, i);
              vec_delete (lm->ports, 1, i);
              vec_delete (lm->src_addrs, 1, i);
              return 0;
            }
        }
      return clib_error_return (0, "collector %U:%d not configured",
                                &collector, udp_port);
    }
  return 0;
}

VLIB_CLI_COMMAND (set_li_command, static) = {
    .path = "set li",
    .short_help = 
    "set li src <ip4-address> collector <ip4-address> udp-port <nnnn>",
    .function = set_li_command_fn,
};

static clib_error_t *
li_init (vlib_main_t * vm)
{
  li_main_t * lm = &li_main;

  lm->vlib_main = vm;
  lm->vnet_main = vnet_get_main();
  lm->hit_node_index = li_hit_node.index;
  return 0;
}

VLIB_INIT_FUNCTION(li_init);