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

int decode_pdu_session_modification_request(
    pdu_session_modification_request_msg *pdu_session_modification_request,
    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, PDU_SESSION_MODIFICATION_REQUEST_MINIMUM_LENGTH, len);

  while (len - decoded > 0) {
    uint8_t ieiDecoded = *(buffer + decoded);

    if (ieiDecoded == 0)
      break;

    switch (ieiDecoded & 0xf0) {
      case PDU_SESSION_MODIFICATION_REQUEST_ALWAYSON_PDU_SESSION_REQUESTED_IEI:
        if ((decoded_result = decode_alwayson_pdu_session_requested(
            &pdu_session_modification_request->alwaysonpdusessionrequested,
            PDU_SESSION_MODIFICATION_REQUEST_ALWAYSON_PDU_SESSION_REQUESTED_IEI,
            buffer + decoded, len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST_ALWAYSON_PDU_SESSION_REQUESTED_PRESENCE;
        }
        break;
    }

    switch (ieiDecoded) {
      case PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAPABILITY_IEI:
        if ((decoded_result = decode__5gsm_capability(
            &pdu_session_modification_request->_5gsmcapability,
            PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAPABILITY_IEI,
            buffer + decoded, len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAPABILITY_PRESENCE;
        }
        break;
      case PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAUSE_IEI:
        if ((decoded_result = decode__5gsm_cause(
            &pdu_session_modification_request->_5gsmcause,
            PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAUSE_IEI, buffer + decoded,
            len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAUSE_PRESENCE;
        }
        break;
      case PDU_SESSION_MODIFICATION_REQUEST_MAXIMUM_NUMBER_OF_SUPPORTED_IEI:
        if ((decoded_result = decode_maximum_number_of_supported_packet_filters(
            &pdu_session_modification_request
                ->maximumnumberofsupportedpacketfilters,
            PDU_SESSION_MODIFICATION_REQUEST_MAXIMUM_NUMBER_OF_SUPPORTED_IEI,
            buffer + decoded, len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST_MAXIMUM_NUMBER_OF_SUPPORTED_PRESENCE;
        }
        break;
      case PDU_SESSION_MODIFICATION_REQUEST_INTGRITY_PROTECTION_MAXIMUM_DATA_RATE_IEI:
        if ((decoded_result =
            decode_intergrity_protection_maximum_data_rate(
                &pdu_session_modification_request
                    ->intergrityprotectionmaximumdatarate,
                PDU_SESSION_MODIFICATION_REQUEST_INTGRITY_PROTECTION_MAXIMUM_DATA_RATE_IEI,
                buffer + decoded, len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST_INTGRITY_PROTECTION_MAXIMUM_DATA_RATE_PRESENCE;
        }
        break;
      case PDU_SESSION_MODIFICATION_REQUEST_QOS_RULES_IEI:
        if ((decoded_result = decode_qos_rules(
            &pdu_session_modification_request->qosrules,
            PDU_SESSION_MODIFICATION_REQUEST_QOS_RULES_IEI, buffer + decoded,
            len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST_QOS_RULES_PRESENCE;
        }
        break;
      case PDU_SESSION_MODIFICATION_REQUEST_QOS_FLOW_DESCRIPTIONS_IEI:
        if ((decoded_result = decode_qos_flow_descriptions(
            &pdu_session_modification_request->qosflowdescriptions,
            PDU_SESSION_MODIFICATION_REQUEST_QOS_FLOW_DESCRIPTIONS_IEI,
            buffer + decoded, len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST_QOS_FLOW_DESCRIPTIONS_PRESENCE;
        }
        break;
        /*
         case PDU_SESSION_MODIFICATION_REQUEST_MAPPED_EPS_BEARER_CONTEXTS_IEI:
         if((decoded_result = decode_mapped_eps_bearer_contexts (&pdu_session_modification_request->mappedepsbearercontexts, PDU_SESSION_MODIFICATION_REQUEST_MAPPED_EPS_BEARER_CONTEXTS_IEI, buffer+decoded,len-decoded))<0)
         return decoded_result;
         else
         {
         decoded+=decoded_result;
         pdu_session_modification_request->presence |= PDU_SESSION_MODIFICATION_REQUEST_MAPPED_EPS_BEARER_CONTEXTS_PRESENCE;
         }
         break;
         */
      case PDU_SESSION_MODIFICATION_REQUEST_E_P_C_O_IEI:
        if ((decoded_result = decode_extended_protocol_configuration_options(
            &pdu_session_modification_request
                ->extendedprotocolconfigurationoptions,
            PDU_SESSION_MODIFICATION_REQUEST_E_P_C_O_IEI, buffer + decoded,
            len - decoded)) < 0)
          return decoded_result;
        else {
          decoded += decoded_result;
          pdu_session_modification_request->presence |=
              PDU_SESSION_MODIFICATION_REQUEST_E_P_C_O_PRESENCE;
        }
        break;
    }
  }

  return decoded;
}

int encode_pdu_session_modification_request(
    pdu_session_modification_request_msg *pdu_session_modification_request,
    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, PDU_SESSION_MODIFICATION_REQUEST_MINIMUM_LENGTH, len);

  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAPABILITY_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAPABILITY_PRESENCE) {
    if ((encoded_result = encode__5gsm_capability(
        pdu_session_modification_request->_5gsmcapability,
        PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAPABILITY_IEI, buffer + encoded,
        len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAUSE_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAUSE_PRESENCE) {
    if ((encoded_result = encode__5gsm_cause(
        pdu_session_modification_request->_5gsmcause,
        PDU_SESSION_MODIFICATION_REQUEST__5GSM_CAUSE_IEI, buffer + encoded,
        len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST_MAXIMUM_NUMBER_OF_SUPPORTED_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST_MAXIMUM_NUMBER_OF_SUPPORTED_PRESENCE) {
    if ((encoded_result = encode_maximum_number_of_supported_packet_filters(
        pdu_session_modification_request->maximumnumberofsupportedpacketfilters,
        PDU_SESSION_MODIFICATION_REQUEST_MAXIMUM_NUMBER_OF_SUPPORTED_IEI,
        buffer + encoded, len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST_ALWAYSON_PDU_SESSION_REQUESTED_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST_ALWAYSON_PDU_SESSION_REQUESTED_PRESENCE) {
    if ((encoded_result = encode_alwayson_pdu_session_requested(
        pdu_session_modification_request->alwaysonpdusessionrequested,
        PDU_SESSION_MODIFICATION_REQUEST_ALWAYSON_PDU_SESSION_REQUESTED_IEI,
        buffer + encoded, len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST_INTGRITY_PROTECTION_MAXIMUM_DATA_RATE_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST_INTGRITY_PROTECTION_MAXIMUM_DATA_RATE_PRESENCE) {
    if ((encoded_result =
        encode_intergrity_protection_maximum_data_rate(
            pdu_session_modification_request
                ->intergrityprotectionmaximumdatarate,
            PDU_SESSION_MODIFICATION_REQUEST_INTGRITY_PROTECTION_MAXIMUM_DATA_RATE_IEI,
            buffer + encoded, len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST_QOS_RULES_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST_QOS_RULES_PRESENCE) {
    if ((encoded_result = encode_qos_rules(
        pdu_session_modification_request->qosrules,
        PDU_SESSION_MODIFICATION_REQUEST_QOS_RULES_IEI, buffer + encoded,
        len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST_QOS_FLOW_DESCRIPTIONS_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST_QOS_FLOW_DESCRIPTIONS_PRESENCE) {
    if ((encoded_result = encode_qos_flow_descriptions(
        pdu_session_modification_request->qosflowdescriptions,
        PDU_SESSION_MODIFICATION_REQUEST_QOS_FLOW_DESCRIPTIONS_IEI,
        buffer + encoded, len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }
  /*
   if((pdu_session_modification_request->presence & PDU_SESSION_MODIFICATION_REQUEST_MAPPED_EPS_BEARER_CONTEXTS_PRESENCE) == PDU_SESSION_MODIFICATION_REQUEST_MAPPED_EPS_BEARER_CONTEXTS_PRESENCE)
   {
   if((encoded_result = encode_mapped_eps_bearer_contexts (pdu_session_modification_request->mappedepsbearercontexts, PDU_SESSION_MODIFICATION_REQUEST_MAPPED_EPS_BEARER_CONTEXTS_IEI, buffer + encoded, len - encoded)) < 0)
   return encoded_result;
   else
   encoded += encoded_result;
   }
   */
  if ((pdu_session_modification_request->presence
      & PDU_SESSION_MODIFICATION_REQUEST_E_P_C_O_PRESENCE)
      == PDU_SESSION_MODIFICATION_REQUEST_E_P_C_O_PRESENCE) {
    if ((encoded_result = encode_extended_protocol_configuration_options(
        pdu_session_modification_request->extendedprotocolconfigurationoptions,
        PDU_SESSION_MODIFICATION_REQUEST_E_P_C_O_IEI, buffer + encoded,
        len - encoded)) < 0)
      return encoded_result;
    else
      encoded += encoded_result;
  }

  return encoded;
}
