/*****************************************************************************
#                                                                            *
# Copyright 2019 AT&T Intellectual Property                                  *
# Copyright 2019 Nokia                                                       *
#                                                                            *
# 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 "e2ap_asn1c_codec.h"


void e2ap_asn1c_print_pdu(const E2AP_PDU_t* pdu)
{
  printf("before\n");
  //  xer_fprint(stdout, &asn_DEF_E2AP_PDU, (void *)pdu);
  xer_fprint(stdout, &asn_DEF_E2AP_PDU, pdu);
  printf("after\n");
  printf("\n");
}

void asn1c_xer_print(asn_TYPE_descriptor_t *typeDescriptor, void *data)
{
  xer_fprint(stdout, typeDescriptor, (void *)data);
  printf("\n");
}


E2AP_PDU_t* e2ap_xml_to_pdu(char const* xml_message)
{
  // E2AP_PDU_t *pdu = new E2AP_PDU_t();
  E2AP_PDU_t *pdu = calloc(1, sizeof(E2AP_PDU_t));

  assert(pdu != 0);

  printf("xmlpdu1\n");

  uint8_t         buf[MAX_XML_BUFFER];
  asn_dec_rval_t  rval;
  size_t          size;
  FILE            *f;

  char XML_path[300];
  char *work_dir = getenv(WORKDIR_ENV);

  printf("xmlpdu2\n");  

  strcpy(XML_path, work_dir);
  strcat(XML_path, E2AP_XML_DIR);
  strcat(XML_path, xml_message);

  printf("xmlpdu4\n");  

  LOG_D("Generate E2AP PDU from XML file: %s\n", XML_path);
  memset(buf, 0, sizeof(buf));

  printf("xmlpdu3\n");  

  f = fopen(XML_path, "r");
  if(!f){
     LOG_E("Unable to open %s. Make sure you have set the Environment Variable E2SIM_DIR, see README", XML_path)
  }

  printf("xmlpdu5\n");  

  assert(f);

  printf("xmlpdu6\n");  

  size = fread(buf, 1, sizeof(buf), f);
  if(size == 0 || size == sizeof(buf))
  {
    LOG_E("Input too long: %s", XML_path);
    exit(1);
  }

  fclose(f);

  printf("xmlpdu7\n");  

  rval = xer_decode(0, &asn_DEF_E2AP_PDU, (void **)&pdu, buf, size);

  printf("xmlpdu8\n");  

  assert(rval.code == RC_OK);

  return pdu;
}


E2setupRequest_t* smaller_e2ap_xml_to_pdu(char const* xml_message)
{
  // E2AP_PDU_t *pdu = new E2AP_PDU_t();
  E2AP_PDU_t *pdu = calloc(1, sizeof(E2AP_PDU_t));

  //  GlobalE2node_ID_t *globale2nodeid = (GlobalE2node_ID_t*)calloc(1, sizeof(GlobalE2node_ID_t));
  GlobalE2node_ID_t *globale2nodeid = (GlobalE2node_ID_t*)calloc(1, sizeof(GlobalE2node_ID_t));   
  E2setupRequest_t *e2setuprequest = (E2setupRequest_t*)calloc(1,sizeof(E2setupRequest_t));

  printf("xmlpdu1\n");

  uint8_t         buf[MAX_XML_BUFFER];
  asn_dec_rval_t  rval;
  size_t          size;
  FILE            *f;

  char XML_path[300];
  char *work_dir = getenv(WORKDIR_ENV);

  printf("xmlpdu2\n");  

  strcpy(XML_path, work_dir);
  strcat(XML_path, E2AP_XML_DIR);
  strcat(XML_path, xml_message);

  printf("xmlpdu4\n");  

  LOG_D("Generate E2AP PDU from XML file: %s\n", XML_path);
  memset(buf, 0, sizeof(buf));

  printf("xmlpdu3\n");  

  f = fopen(XML_path, "r");
  if(!f){
     LOG_E("Unable to open %s. Make sure you have set the Environment Variable E2SIM_DIR, see README", XML_path)
  }

  printf("xmlpdu5\n");  

  assert(f);

  printf("xmlpdu6\n");  

  size = fread(buf, 1, sizeof(buf), f);
  if(size == 0 || size == sizeof(buf))
  {
    LOG_E("Input too long: %s", XML_path);
    exit(1);
  }

  fclose(f);

  printf("xmlpdu7\n");

  rval = xer_decode(0, &asn_DEF_E2setupRequest, (void **)&e2setuprequest, buf, size);

  printf("xmlpdu8\n");  

  assert(rval.code == RC_OK);

  return e2setuprequest;
}


int e2ap_asn1c_encode_pdu(E2AP_PDU_t* pdu, unsigned char **buffer)
{
  int len;

  *buffer = NULL;
  assert(pdu != NULL);
  assert(buffer != NULL);

  len = aper_encode_to_new_buffer(&asn_DEF_E2AP_PDU, 0, pdu, (void **)buffer);

  if (len < 0)  {
    LOG_E("[E2AP ASN] Unable to aper encode");
    exit(1);
  }
  else {
    LOG_D("[E2AP ASN] Encoded succesfully, encoded size = %d", len);
  }

  ASN_STRUCT_FREE_CONTENTS_ONLY(asn_DEF_E2AP_PDU, pdu);

  return len;
}

void e2ap_asn1c_decode_pdu(E2AP_PDU_t* pdu, unsigned char *buffer, int len)
{
  asn_dec_rval_t dec_ret;

  assert(buffer != NULL);

  dec_ret = aper_decode(NULL, &asn_DEF_E2AP_PDU, (void **)&pdu, buffer, len, 0, 0);

  if (dec_ret.code != RC_OK) {
    LOG_E("[E2AP ASN] Failed to decode pdu");
    exit(1);
  }
  else {
    LOG_D("[E2AP ASN] Decoded succesfully");
  }
}

int e2ap_asn1c_get_procedureCode(E2AP_PDU_t* pdu)
{
  int procedureCode = -1;

  switch(pdu->present)
  {
    case E2AP_PDU_PR_initiatingMessage:
      fprintf(stderr,"initiating message\n");
      procedureCode = pdu->choice.initiatingMessage->procedureCode;
      break;

    case E2AP_PDU_PR_successfulOutcome:
      procedureCode = pdu->choice.successfulOutcome->procedureCode;
      break;

    case E2AP_PDU_PR_unsuccessfulOutcome:
      procedureCode = pdu->choice.unsuccessfulOutcome->procedureCode;
      break;

    default:
      LOG_E("[E2AP] Error: Unknown index %d in E2AP PDU", (int)pdu->present);
      break;
  }

  return procedureCode;
}