/*
 * 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 <hicn/transport/config.h>

#ifdef __vpp__

#include <hicn/transport/core/hicn_vapi.h>
#include <hicn/transport/core/memif_vapi.h>
#include <hicn/transport/core/prefix.h>
#include <hicn/transport/core/vpp_forwarder_interface.h>

extern "C" {
#include <memif/libmemif.h>
};

typedef enum { MASTER = 0, SLAVE = 1 } memif_role_t;

#define MEMIF_DEFAULT_RING_SIZE 2048
#define MEMIF_DEFAULT_RX_QUEUES 1
#define MEMIF_DEFAULT_TX_QUEUES 1
#define MEMIF_DEFAULT_BUFFER_SIZE 2048

namespace transport {

namespace core {

std::mutex VPPForwarderInterface::global_lock_;

VPPForwarderInterface::VPPForwarderInterface(MemifConnector &connector)
    : ForwarderInterface<VPPForwarderInterface, MemifConnector>(connector),
      sw_if_index_(~0),
      face_id1_(~0),
      face_id2_(~0),
      is_consumer_(false){
        sock_ = hc_sock_create();
      }

VPPForwarderInterface::~VPPForwarderInterface() {}

/**
 * @brief Create a memif interface in the local VPP forwarder.
 */
uint32_t VPPForwarderInterface::getMemifConfiguration() {
  memif_create_params_t input_params = {0};

  int ret = memif_vapi_get_next_memif_id(
      *(vapi_ctx_t *)VPPForwarderInterface::sock_, &memif_id_);

  if (ret < 0) {
    throw errors::RuntimeException(
        "Error getting next memif id. Could not create memif interface.");
  }

  input_params.id = memif_id_;
  input_params.role = memif_role_t::MASTER;
  input_params.mode = memif_interface_mode_t::MEMIF_INTERFACE_MODE_IP;
  input_params.rx_queues = MEMIF_DEFAULT_RX_QUEUES;
  input_params.tx_queues = MEMIF_DEFAULT_TX_QUEUES;
  input_params.ring_size = MEMIF_DEFAULT_RING_SIZE;
  input_params.buffer_size = MEMIF_DEFAULT_BUFFER_SIZE;

  memif_output_params_t output_params = {0};

  ret = memif_vapi_create_memif(*(vapi_ctx_t *)VPPForwarderInterface::sock_,
                                      &input_params, &output_params);

  if (ret < 0) {
    throw errors::RuntimeException(
        "Error creating memif interface in the local VPP forwarder.");
  }

  return output_params.sw_if_index;
}

void VPPForwarderInterface::consumerConnection() {
  hicn_consumer_input_params input = {0};
  hicn_consumer_output_params output = {0};
  ip_address_t ip4_address;
  ip_address_t ip6_address;

  output.src4 = &ip4_address;
  output.src6 = &ip6_address;
  input.swif = sw_if_index_;



  int ret = hicn_vapi_register_cons_app(*(vapi_ctx_t *)VPPForwarderInterface::sock_,
                                              &input, &output);

  if (ret < 0) {
    throw errors::RuntimeException(hicn_vapi_get_error_string(ret));
  }

  face_id1_ = output.face_id1;
  face_id2_ = output.face_id2;


  std::memcpy(inet_address_.v4.as_u8, output.src4->v4.as_u8, IPV4_ADDR_LEN);

  std::memcpy(inet6_address_.v6.as_u8, output.src6->v6.as_u8, IPV6_ADDR_LEN);
}

void VPPForwarderInterface::producerConnection() {
  // Producer connection will be set when we set the first route.
}

void VPPForwarderInterface::connect(bool is_consumer) {
  std::lock_guard<std::mutex> connection_lock(global_lock_);

  hc_sock_connect(sock_);

  sw_if_index_ = getMemifConfiguration();

  is_consumer_ = is_consumer;
  if (is_consumer_) {
    consumerConnection();
  }

  connector_.connect(memif_id_, 0);
}

void VPPForwarderInterface::registerRoute(Prefix &prefix) {
  ip_prefix_t &addr = prefix.toIpPrefixStruct();

  ip_prefix_t producer_prefix;
  ip_address_t producer_locator;

  if (face_id1_ == uint32_t(~0)) {
    hicn_producer_input_params input;
    std::memset(&input, 0, sizeof(input));

    hicn_producer_output_params output;
    std::memset(&output, 0, sizeof(output));

    input.prefix = &producer_prefix;
    output.prod_addr = &producer_locator;

    // Here we have to ask to the actual connector what is the
    // memif_id, since this function should be called after the
    // memif creation.
    input.swif = sw_if_index_;
    input.prefix->address = addr.address;
    input.prefix->family = addr.family;
    input.prefix->len = addr.len;
    input.cs_reserved = content_store_reserved_;

    int ret = hicn_vapi_register_prod_app(
        *(vapi_ctx_t *)VPPForwarderInterface::sock_, &input, &output);

    if (ret < 0) {
      throw errors::RuntimeException(hicn_vapi_get_error_string(ret));
    }

    inet6_address_ = *output.prod_addr;

    face_id1_ = output.face_id;
  } else {
    hicn_producer_set_route_params params;
    params.prefix = &producer_prefix;
    params.prefix->address = addr.address;
    params.prefix->family = addr.family;
    params.prefix->len = addr.len;
    params.face_id = face_id1_;

    int ret = hicn_vapi_register_route(*(vapi_ctx_t *)VPPForwarderInterface::sock_,
                                             &params);

    if (ret < 0) {
      throw errors::RuntimeException(hicn_vapi_get_error_string(ret));
    }
  }
}

void VPPForwarderInterface::closeConnection() {
  if (VPPForwarderInterface::sock_) {
    connector_.close();

    if (is_consumer_) {
      hicn_del_face_app_input_params params;
      params.face_id = face_id1_;
      hicn_vapi_face_cons_del(*(vapi_ctx_t *)VPPForwarderInterface::sock_, &params);
      params.face_id = face_id2_;
      hicn_vapi_face_cons_del(*(vapi_ctx_t *)VPPForwarderInterface::sock_, &params);
    }
    else {
      hicn_del_face_app_input_params params;
      params.face_id = face_id1_;
      hicn_vapi_face_prod_del(*(vapi_ctx_t *)VPPForwarderInterface::sock_, &params);
    }

    if (sw_if_index_ != uint32_t(~0)) {
      int ret = memif_vapi_delete_memif(*(vapi_ctx_t *)VPPForwarderInterface::sock_,
                                              sw_if_index_);
      if (ret < 0) {
        TRANSPORT_LOGE("Error deleting memif with sw idx %u.", sw_if_index_);
      }
    }

    hc_sock_free(sock_);
    VPPForwarderInterface::sock_ = nullptr;
  }
}

}  // namespace core

}  // namespace transport

#endif