/*
 * Licensed to Open Source Radio Access Network(OS-RAN) Alliance and OAI
 * Software Alliance under one or more contributor license agreements. The
 * initial OpenXG series projects are derided from OAI projects, the files from
 * OAI projects are all in compliance with OAI Public License, Version 1.1.
 * codes and files developed from OpenXG projects and from OS-RAN Alliance
 * are all under OS-RAN licenses; you may not use this file except in compliance
 * with the license.  You may get a copy of the license at:
 *	http://www.openxg.org.cn/?falu_69.html
 * For more information about OpenXG, please contact:
 * contact@openxg.org.cn
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include "TLVEncoder.h"
#include "TLVDecoder.h"
#include "RegistrationReject.h"

int decode_registration_reject(
    registration_reject_msg* registration_reject, uint8_t* buffer,
    uint32_t len) {
  uint32_t decoded   = 0;
  int decoded_result = 0;

  // Check if we got a NULL pointer and if buffer length is >= minimum length
  // expected for the message.
  CHECK_PDU_POINTER_AND_LENGTH_DECODER(
      buffer, REGISTRATION_REJECT_MINIMUM_LENGTH, len);

  if ((decoded_result = decode__5gmm_cause(
           &registration_reject->_5gmmcause, 0, buffer + decoded,
           len - decoded)) < 0)
    return decoded_result;
  else
    decoded += decoded_result;

  while (len - decoded > 0) {
    uint8_t ieiDecoded = *(buffer + decoded);
    if ((ieiDecoded != REGISTRATION_REJECT_GPRSTIMER2_T3346_VALUE_IEI) &&
        (ieiDecoded != REGISTRATION_REJECT_GPRSTIMER2_T3502_VALUE_IEI) &&
        (ieiDecoded != REGISTRATION_REJECT_EAP_MESSAGE_IEI))
      break;
    switch (ieiDecoded) {
      case REGISTRATION_REJECT_GPRSTIMER2_T3346_VALUE_IEI:
        if ((decoded_result = decode_gprs_timer2(
                 &registration_reject->t3346,
                 REGISTRATION_REJECT_GPRSTIMER2_T3346_VALUE_IEI,
                 buffer + decoded, len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          registration_reject->presence |=
              REGISTRATION_REJECT_GPRSTIMER2_T3346_VALUE_PRESENT;
        }
        break;
      case REGISTRATION_REJECT_GPRSTIMER2_T3502_VALUE_IEI:
        if ((decoded_result = decode_gprs_timer2(
                 &registration_reject->t3502,
                 REGISTRATION_REJECT_GPRSTIMER2_T3502_VALUE_IEI,
                 buffer + decoded, len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          registration_reject->presence |=
              REGISTRATION_REJECT_GPRSTIMER2_T3502_VALUE_PRESENT;
        }
        break;
      case REGISTRATION_REJECT_EAP_MESSAGE_IEI:
        if ((decoded_result = decode_eap_message(
                 &registration_reject->eapmessage,
                 REGISTRATION_REJECT_EAP_MESSAGE_IEI, buffer + decoded,
                 len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          registration_reject->presence |=
              REGISTRATION_REJECT_EAP_MESSAGE_PRESENT;
        }
        break;
    }
  }

  return decoded;
}

int encode_registration_reject(
    registration_reject_msg* registration_reject, uint8_t* buffer,
    uint32_t len) {
  uint32_t encoded   = 0;
  int encoded_result = 0;

  // Check if we got a NULL pointer and if buffer length is >= minimum length
  // expected for the message.
  CHECK_PDU_POINTER_AND_LENGTH_ENCODER(
      buffer, REGISTRATION_REJECT_MINIMUM_LENGTH, len);

  if ((encoded_result = encode__5gmm_cause(
           registration_reject->_5gmmcause, 0, buffer + encoded,
           len - encoded)) < 0)
    return encoded_result;
  else
    encoded += encoded_result;

  if ((registration_reject->presence &
       REGISTRATION_REJECT_GPRSTIMER2_T3346_VALUE_PRESENT) ==
      REGISTRATION_REJECT_GPRSTIMER2_T3346_VALUE_PRESENT) {
    if ((encoded_result = encode_gprs_timer2(
             registration_reject->t3346,
             REGISTRATION_REJECT_GPRSTIMER2_T3346_VALUE_IEI, buffer + encoded,
             len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((registration_reject->presence &
       REGISTRATION_REJECT_GPRSTIMER2_T3502_VALUE_PRESENT) ==
      REGISTRATION_REJECT_GPRSTIMER2_T3502_VALUE_PRESENT) {
    if ((encoded_result = encode_gprs_timer2(
             registration_reject->t3502,
             REGISTRATION_REJECT_GPRSTIMER2_T3502_VALUE_IEI, buffer + encoded,
             len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((registration_reject->presence &
       REGISTRATION_REJECT_EAP_MESSAGE_PRESENT) ==
      REGISTRATION_REJECT_EAP_MESSAGE_PRESENT) {
    if ((encoded_result = encode_eap_message(
             registration_reject->eapmessage,
             REGISTRATION_REJECT_EAP_MESSAGE_IEI, buffer + encoded,
             len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  return encoded;
}
