/*
 * Licensed to the OpenAirInterface (OAI) Software Alliance under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The OpenAirInterface Software Alliance licenses this file to You under
 * the OAI Public License, Version 1.1  (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.openairinterface.org/?page_id=698
 *
 * 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.
 *-------------------------------------------------------------------------------
 * For more information about the OpenAirInterface (OAI) Software Alliance:
 *      contact@openairinterface.org
 */

/*! \file m2ap_eNB_management_procedures.c
 * \brief m2ap tasks for eNB
 * \author Javier Morgade  <javier.morade@ieee.org>
 * \date 2018
 * \version 1.0
 */

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>

#include "intertask_interface.h"

#include "assertions.h"
#include "conversions.h"

#include "m2ap_common.h"
#include "m2ap_eNB_defs.h"
#include "m2ap_eNB.h"


#define M2AP_DEBUG_LIST
#ifdef M2AP_DEBUG_LIST
#  define M2AP_eNB_LIST_OUT(x, args...) M2AP_DEBUG("[eNB]%*s"x"\n", 4*indent, "", ##args)
#else
#  define M2AP_eNB_LIST_OUT(x, args...)
#endif

static int                  indent = 0;


m2ap_eNB_internal_data_t m2ap_eNB_internal_data;

RB_GENERATE(m2ap_enb_map, m2ap_eNB_data_s, entry, m2ap_eNB_compare_assoc_id);

int m2ap_eNB_compare_assoc_id(
  struct m2ap_eNB_data_s *p1, struct m2ap_eNB_data_s *p2)
{
  if (p1->assoc_id == -1) {
    if (p1->cnx_id < p2->cnx_id) {
      return -1;
    }

    if (p1->cnx_id > p2->cnx_id) {
      return 1;
    }
  } else {
    if (p1->assoc_id < p2->assoc_id) {
      return -1;
    }

    if (p1->assoc_id > p2->assoc_id) {
      return 1;
    }
  }

  /* Matching reference */
  return 0;
}

uint16_t m2ap_eNB_fetch_add_global_cnx_id(void)
{
  return ++m2ap_eNB_internal_data.global_cnx_id;
}

void m2ap_eNB_prepare_internal_data(void)
{
  memset(&m2ap_eNB_internal_data, 0, sizeof(m2ap_eNB_internal_data));
  STAILQ_INIT(&m2ap_eNB_internal_data.m2ap_eNB_instances_head);
}

void m2ap_eNB_insert_new_instance(m2ap_eNB_instance_t *new_instance_p)
{
  DevAssert(new_instance_p != NULL);

  STAILQ_INSERT_TAIL(&m2ap_eNB_internal_data.m2ap_eNB_instances_head,
                     new_instance_p, m2ap_eNB_entries);
}

void dump_tree_m2(m2ap_eNB_data_t *t)
{
  if (t == NULL) return;
  printf("-----------------------\n");
  printf("eNB id %d %s\n", t->eNB_id, t->eNB_name);
  printf("state %d\n", t->state);
  printf("nextstream %d\n", t->nextstream);
  printf("in_streams %d out_streams %d\n", t->in_streams, t->out_streams);
  printf("cnx_id %d assoc_id %d\n", t->cnx_id, t->assoc_id);
  dump_tree_m2(t->entry.rbe_left);
  dump_tree_m2(t->entry.rbe_right);
}

void dump_trees_m2(void)
{
m2ap_eNB_instance_t *zz;
STAILQ_FOREACH(zz, &m2ap_eNB_internal_data.m2ap_eNB_instances_head,
               m2ap_eNB_entries) {
printf("here comes the tree (instance %d):\n---------------------------------------------\n", zz->instance);
dump_tree_m2(zz->m2ap_enb_head.rbh_root);
printf("---------------------------------------------\n");
}
}

struct m2ap_eNB_data_s *m2ap_get_eNB(m2ap_eNB_instance_t *instance_p,
				     int32_t assoc_id,
				     uint16_t cnx_id)
{
  struct m2ap_eNB_data_s  temp;
  struct m2ap_eNB_data_s *found;

printf("m2ap_get_eNB at 1 (looking for assoc_id %d cnx_id %d)\n", assoc_id, cnx_id);
dump_trees_m2();

  memset(&temp, 0, sizeof(struct m2ap_eNB_data_s));

  temp.assoc_id = assoc_id;
  temp.cnx_id   = cnx_id;

  if (instance_p == NULL) {
    STAILQ_FOREACH(instance_p, &m2ap_eNB_internal_data.m2ap_eNB_instances_head,
                   m2ap_eNB_entries) {
      found = RB_FIND(m2ap_enb_map, &instance_p->m2ap_enb_head, &temp);

      if (found != NULL) {
        return found;
      }
    }
  } else {
    return RB_FIND(m2ap_enb_map, &instance_p->m2ap_enb_head, &temp);
  }

  return NULL;
}


m2ap_eNB_instance_t *m2ap_eNB_get_instance(instance_t instance)
{
  m2ap_eNB_instance_t *temp = NULL;

  STAILQ_FOREACH(temp, &m2ap_eNB_internal_data.m2ap_eNB_instances_head,
                 m2ap_eNB_entries) {
    if (temp->instance == instance) {
      /* Matching occurence */
      return temp;
    }
  }

  return NULL;
}

/// utility functions

void m2ap_dump_eNB (m2ap_eNB_data_t  * eNB_ref);

void
m2ap_dump_eNB_list (void) {
   m2ap_eNB_instance_t *inst = NULL;
   struct m2ap_eNB_data_s *found = NULL;
   struct m2ap_eNB_data_s temp;

   memset(&temp, 0, sizeof(struct m2ap_eNB_data_s));

  STAILQ_FOREACH (inst, &m2ap_eNB_internal_data.m2ap_eNB_instances_head,  m2ap_eNB_entries) {
    found = RB_FIND(m2ap_enb_map, &inst->m2ap_enb_head, &temp);
    m2ap_dump_eNB (found);
  }
}

void m2ap_dump_eNB (m2ap_eNB_data_t  * eNB_ref) {

  if (eNB_ref == NULL) {
    return;
  }

  M2AP_eNB_LIST_OUT ("");
  M2AP_eNB_LIST_OUT ("eNB name:          %s", eNB_ref->eNB_name == NULL ? "not present" : eNB_ref->eNB_name);
  M2AP_eNB_LIST_OUT ("eNB STATE:         %07x", eNB_ref->state);
  M2AP_eNB_LIST_OUT ("eNB ID:            %07x", eNB_ref->eNB_id);
  indent++;
  M2AP_eNB_LIST_OUT ("SCTP cnx id:     %d", eNB_ref->cnx_id);
  M2AP_eNB_LIST_OUT ("SCTP assoc id:     %d", eNB_ref->assoc_id);
  M2AP_eNB_LIST_OUT ("SCTP instreams:    %d", eNB_ref->in_streams);
  M2AP_eNB_LIST_OUT ("SCTP outstreams:   %d", eNB_ref->out_streams);
  indent--;
}

m2ap_eNB_data_t  * m2ap_is_eNB_pci_in_list (const uint32_t pci)
{
  m2ap_eNB_instance_t    *inst;
  struct m2ap_eNB_data_s *elm;

  STAILQ_FOREACH(inst, &m2ap_eNB_internal_data.m2ap_eNB_instances_head, m2ap_eNB_entries) {
    RB_FOREACH(elm, m2ap_enb_map, &inst->m2ap_enb_head) {
      for (int i = 0; i<elm->num_cc; i++) {
        if (elm->Nid_cell[i] == pci) {
          return elm;
        }
      }
    }
  }
  return NULL;
}

m2ap_eNB_data_t  * m2ap_is_eNB_id_in_list (const uint32_t eNB_id)
{
  m2ap_eNB_instance_t    *inst;
  struct m2ap_eNB_data_s *elm;

  STAILQ_FOREACH(inst, &m2ap_eNB_internal_data.m2ap_eNB_instances_head, m2ap_eNB_entries) {
    RB_FOREACH(elm, m2ap_enb_map, &inst->m2ap_enb_head) {
      if (elm->eNB_id == eNB_id)
        return elm;
    }
  }
  return NULL;
}

m2ap_eNB_data_t  * m2ap_is_eNB_assoc_id_in_list (const uint32_t sctp_assoc_id)
{
  m2ap_eNB_instance_t    *inst;
  struct m2ap_eNB_data_s *found;
  struct m2ap_eNB_data_s temp;

  temp.assoc_id = sctp_assoc_id;
  temp.cnx_id = -1;

  STAILQ_FOREACH(inst, &m2ap_eNB_internal_data.m2ap_eNB_instances_head, m2ap_eNB_entries) {
    found = RB_FIND(m2ap_enb_map, &inst->m2ap_enb_head, &temp);
    if (found != NULL){
      if (found->assoc_id == sctp_assoc_id) {
	return found;
      }
    }
  }
  return NULL;
}
