/*
 * Copyright (C) 2019-2023 by Sukchan Lee <acetcom@gmail.com>
 *
 * This file is part of Open5GS.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include "gsm-build.h"

#undef OGS_LOG_DOMAIN
#define OGS_LOG_DOMAIN __gsm_log_domain

static uint8_t *gsm_decode_base64(const char *src, int *out_len)
{
    int len, decoded;
    uint8_t *buf = NULL;

    ogs_assert(out_len);

    *out_len = 0;

    if (!src) {
        ogs_error("No base64 data");
        return NULL;
    }

    len = ogs_base64_decoded_size(src);
    if (len <= 0) {
        ogs_error("Invalid base64 data");
        return NULL;
    }

    buf = ogs_calloc(1, len);
    if (!buf) {
        ogs_error("ogs_calloc() failed [%d]", len);
        return NULL;
    }

    decoded = ogs_base64_decode_to_buffer(buf, len, src);
    if (decoded <= 0) {
        ogs_error("Invalid base64 data");
        ogs_free(buf);
        return NULL;
    }
    if (decoded > 0xffff) {
        ogs_error("Base64 data too long [%d]", decoded);
        ogs_free(buf);
        return NULL;
    }

    *out_len = decoded;

    return buf;
}

ogs_pkbuf_t *gsm_build_pdu_session_establishment_accept(smf_sess_t *sess)
{
    ogs_pkbuf_t *pkbuf = NULL;
    smf_bearer_t *qos_flow = NULL;
    int rv, len;

    ogs_nas_5gs_message_t message;
    ogs_nas_5gs_pdu_session_establishment_accept_t *
        pdu_session_establishment_accept =
            &message.gsm.pdu_session_establishment_accept;

    ogs_nas_pdu_session_type_t *selected_pdu_session_type = NULL;
    ogs_nas_qos_rules_t *authorized_qos_rules = NULL;
    ogs_nas_session_ambr_t *session_ambr = NULL;
    ogs_nas_5gsm_cause_t *gsm_cause = NULL;
    ogs_nas_pdu_address_t *pdu_address = NULL;
    ogs_nas_s_nssai_t *nas_s_nssai = NULL;
    ogs_nas_qos_flow_descriptions_t *authorized_qos_flow_descriptions = NULL;
    ogs_nas_extended_protocol_configuration_options_t
        *extended_protocol_configuration_options = NULL;
    ogs_nas_dnn_t *dnn = NULL;

    ogs_nas_qos_rule_t qos_rule[OGS_NAS_MAX_NUM_OF_QOS_RULE];
    ogs_nas_qos_flow_description_t
        qos_flow_description[OGS_NAS_MAX_NUM_OF_QOS_FLOW_DESCRIPTION];

    uint8_t *epco_buf = NULL;
    int16_t epco_len;

    OpenAPI_list_t *qosFlowsSetupList = NULL;
    OpenAPI_qos_flow_setup_item_t *qosFlowSetupItem = NULL;
    OpenAPI_lnode_t *node = NULL;
    selected_pdu_session_type = &pdu_session_establishment_accept->
        selected_pdu_session_type;
    ogs_assert(selected_pdu_session_type);
    authorized_qos_rules = &pdu_session_establishment_accept->
        authorized_qos_rules;
    ogs_assert(authorized_qos_rules);

    session_ambr = &pdu_session_establishment_accept->session_ambr;
    ogs_assert(session_ambr);
    pdu_address = &pdu_session_establishment_accept->pdu_address;
    ogs_assert(pdu_address);
    gsm_cause = &pdu_session_establishment_accept->gsm_cause;
    ogs_assert(gsm_cause);
    nas_s_nssai = &pdu_session_establishment_accept->s_nssai;
    ogs_assert(nas_s_nssai);
    authorized_qos_flow_descriptions =
        &pdu_session_establishment_accept->authorized_qos_flow_descriptions;
    ogs_assert(authorized_qos_flow_descriptions);
    extended_protocol_configuration_options =
        &pdu_session_establishment_accept->
            extended_protocol_configuration_options;
    ogs_assert(extended_protocol_configuration_options);
    dnn = &pdu_session_establishment_accept->dnn;
    ogs_assert(dnn);

    ogs_assert(sess);
    qos_flow = smf_default_bearer_in_sess(sess);
    ogs_assert(qos_flow);
    ogs_assert(ogs_list_next(qos_flow) == NULL);

    memset(&message, 0, sizeof(message));
    message.gsm.h.extended_protocol_discriminator =
            OGS_NAS_EXTENDED_PROTOCOL_DISCRIMINATOR_5GSM;
    message.gsm.h.pdu_session_identity = sess->psi;
    message.gsm.h.procedure_transaction_identity = sess->pti;
    message.gsm.h.message_type = OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT;

    selected_pdu_session_type->type = sess->session.ssc_mode;
    selected_pdu_session_type->value = sess->session.session_type;

    if (HOME_ROUTED_ROAMING_IN_VSMF(sess)) {
        qosFlowsSetupList = sess->h_smf_qos_flows_setup_list;
        ogs_assert(qosFlowsSetupList);

        node = qosFlowsSetupList->first;
        ogs_assert(node);
        qosFlowSetupItem = node->data;
        ogs_assert(qosFlowSetupItem);

        authorized_qos_rules->buffer =
            gsm_decode_base64(qosFlowSetupItem->qos_rules, &len);
        if (!authorized_qos_rules->buffer) {
            ogs_error("Failed to decode qosRules");
            goto cleanup;
        }
        authorized_qos_rules->length = len;

    } else if (HOME_ROUTED_ROAMING_IN_HSMF(sess)) {
        ogs_fatal("This should not be invoked from H-SMF during HR-Roaming");
        ogs_assert_if_reached();
    } else {
        memset(qos_rule, 0, sizeof(qos_rule));
        gsm_encode_default_qos_rule(&qos_rule[0], qos_flow);

        rv = ogs_nas_build_qos_rules(authorized_qos_rules, qos_rule, 1);
        if (rv != OGS_OK) {
            ogs_error("ogs_nas_build_qos_rules() failed");
            goto cleanup;
        }
        if (!authorized_qos_rules->length) {
            ogs_error("No length");
            goto cleanup;
        }
    }

    /* Session-AMBR */
    session_ambr->length = 6;
    ogs_nas_bitrate_from_uint64(
            &session_ambr->downlink, sess->session.ambr.downlink);
    ogs_nas_bitrate_from_uint64(
            &session_ambr->uplink, sess->session.ambr.uplink);

    /* PDU Address */
    pdu_session_establishment_accept->presencemask |=
        OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_PDU_ADDRESS_PRESENT;
    pdu_address->pdn_type = sess->session.session_type;

    if (pdu_address->pdn_type == OGS_PDU_SESSION_TYPE_IPV4) {
        pdu_address->addr = sess->paa.addr;
        pdu_address->length = OGS_NAS_PDU_ADDRESS_IPV4_LEN;
    } else if (pdu_address->pdn_type == OGS_PDU_SESSION_TYPE_IPV6) {
        memcpy(pdu_address->addr6,
                sess->paa.addr6+(OGS_IPV6_LEN>>1), OGS_IPV6_LEN>>1);
        pdu_address->length = OGS_NAS_PDU_ADDRESS_IPV6_LEN;
    } else if (pdu_address->pdn_type == OGS_PDU_SESSION_TYPE_IPV4V6) {
        pdu_address->both.addr = sess->paa.both.addr;
        memcpy(pdu_address->both.addr6,
            sess->paa.both.addr6+(OGS_IPV6_LEN>>1), OGS_IPV6_LEN>>1);
        pdu_address->length = OGS_NAS_PDU_ADDRESS_IPV4V6_LEN;
    } else {
        ogs_error("Unexpected PDN Type %u", pdu_address->pdn_type);
        goto cleanup;
    }

    /* GSM cause */
    if (HOME_ROUTED_ROAMING_IN_VSMF(sess)) {
        if (sess->h_smf_gsm_cause) {
            pdu_session_establishment_accept->presencemask |=
                OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_5GSM_CAUSE_PRESENT;
            *gsm_cause =  sess->h_smf_gsm_cause;
        }
        sess->h_smf_gsm_cause = 0; /* Clear GSM Cause */
    } else if (HOME_ROUTED_ROAMING_IN_HSMF(sess)) {
        ogs_fatal("This should not be invoked from H-SMF during HR-Roaming");
        ogs_assert_if_reached();
    } else {
        if (sess->ue_session_type == OGS_PDU_SESSION_TYPE_IPV4V6) {
            if (pdu_address->pdn_type == OGS_PDU_SESSION_TYPE_IPV4) {
                pdu_session_establishment_accept->presencemask |=
                    OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_5GSM_CAUSE_PRESENT;
                *gsm_cause = OGS_5GSM_CAUSE_PDU_SESSION_TYPE_IPV4_ONLY_ALLOWED;
            } else if (pdu_address->pdn_type == OGS_PDU_SESSION_TYPE_IPV6) {
                pdu_session_establishment_accept->presencemask |=
                    OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_5GSM_CAUSE_PRESENT;
                *gsm_cause = OGS_5GSM_CAUSE_PDU_SESSION_TYPE_IPV6_ONLY_ALLOWED;
            }
        }
    }

    /* S-NSSAI */
    pdu_session_establishment_accept->presencemask |=
        OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_S_NSSAI_PRESENT;
    ogs_nas_build_s_nssai2(nas_s_nssai, &sess->s_nssai,
        (sess->mapped_hplmn_presence) ? &sess->mapped_hplmn : NULL);

    /* QoS flow descriptions */
    if (HOME_ROUTED_ROAMING_IN_VSMF(sess)) {
        ogs_assert(qosFlowSetupItem);

        if (qosFlowSetupItem->qos_flow_description) {
            authorized_qos_flow_descriptions->buffer = gsm_decode_base64(
                    qosFlowSetupItem->qos_flow_description, &len);
            if (!authorized_qos_flow_descriptions->buffer) {
                ogs_error("Failed to decode qosFlowDescription");
                goto cleanup;
            }
            authorized_qos_flow_descriptions->length = len;

            pdu_session_establishment_accept->presencemask |=
                OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_AUTHORIZED_QOS_FLOW_DESCRIPTIONS_PRESENT;
        }
    } else if (HOME_ROUTED_ROAMING_IN_HSMF(sess)) {
        ogs_fatal("This should not be invoked from H-SMF during HR-Roaming");
        ogs_assert_if_reached();
    } else {
        memset(&qos_flow_description, 0, sizeof(qos_flow_description));
        gsm_encode_default_qos_flow_description(
                &qos_flow_description[0], qos_flow);

        pdu_session_establishment_accept->presencemask |=
            OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_AUTHORIZED_QOS_FLOW_DESCRIPTIONS_PRESENT;
        rv = ogs_nas_build_qos_flow_descriptions(
                authorized_qos_flow_descriptions, qos_flow_description, 1);
        if (rv != OGS_OK) {
            ogs_error("ogs_nas_build_qos_flow_descriptions() failed");
            goto cleanup;
        }
        if (!authorized_qos_flow_descriptions->length) {
            ogs_error("No length");
            goto cleanup;
        }
    }

    /* Extended protocol configuration options */
    if (HOME_ROUTED_ROAMING_IN_VSMF(sess)) {
        if (sess->h_smf_extended_protocol_configuration_options.buffer &&
            sess->h_smf_extended_protocol_configuration_options.length) {
            pdu_session_establishment_accept->presencemask |=
                OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_EXTENDED_PROTOCOL_CONFIGURATION_OPTIONS_PRESENT;
            OGS_NAS_STORE_DATA(
                    extended_protocol_configuration_options,
                    &sess->h_smf_extended_protocol_configuration_options);
        }
    } else if (HOME_ROUTED_ROAMING_IN_HSMF(sess)) {
        ogs_fatal("This should not be invoked from H-SMF during HR-Roaming");
        ogs_assert_if_reached();
    } else {
        if (sess->nas.ue_epco.buffer && sess->nas.ue_epco.length) {
            epco_buf = ogs_calloc(OGS_MAX_EPCO_LEN, sizeof(uint8_t));
            ogs_assert(epco_buf);
            epco_len = smf_pco_build(epco_buf,
                    sess->nas.ue_epco.buffer, sess->nas.ue_epco.length);
            if (epco_len <= 0) {
                ogs_error("smf_pco_build() failed");
                ogs_log_hexdump(OGS_LOG_ERROR,
                        sess->nas.ue_epco.buffer, sess->nas.ue_epco.length);
                goto cleanup;
            }
            pdu_session_establishment_accept->presencemask |=
                OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_EXTENDED_PROTOCOL_CONFIGURATION_OPTIONS_PRESENT;
            extended_protocol_configuration_options->buffer = epco_buf;
            extended_protocol_configuration_options->length = epco_len;
        }
    }

    /* DNN */
    pdu_session_establishment_accept->presencemask |=
        OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_ACCEPT_DNN_PRESENT;
    ogs_assert(sess->session.name);
    dnn->length = strlen(sess->session.name);
    ogs_cpystrn(dnn->value, sess->session.name,
            ogs_min(dnn->length, OGS_MAX_DNN_LEN) + 1);

    pkbuf = ogs_nas_5gs_plain_encode(&message);
    ogs_assert(pkbuf);

    sess->establishment_accept_sent = true;

cleanup:
    if (authorized_qos_rules->buffer)
        ogs_free(authorized_qos_rules->buffer);
    if (authorized_qos_flow_descriptions->buffer)
        ogs_free(authorized_qos_flow_descriptions->buffer);
    if (extended_protocol_configuration_options->buffer)
        ogs_free(extended_protocol_configuration_options->buffer);

    OGS_NAS_CLEAR_DATA(&sess->h_smf_extended_protocol_configuration_options);

    return pkbuf;
}

ogs_pkbuf_t *gsm_build_pdu_session_establishment_reject(
        smf_sess_t *sess, ogs_nas_5gsm_cause_t gsm_cause)
{
    ogs_nas_5gs_message_t message;
    ogs_nas_5gs_pdu_session_establishment_reject_t *
        pdu_session_establishment_reject =
            &message.gsm.pdu_session_establishment_reject;

    ogs_assert(sess);
    ogs_assert(gsm_cause);

    memset(&message, 0, sizeof(message));
    message.gsm.h.extended_protocol_discriminator =
            OGS_NAS_EXTENDED_PROTOCOL_DISCRIMINATOR_5GSM;
    message.gsm.h.pdu_session_identity = sess->psi;
    message.gsm.h.procedure_transaction_identity = sess->pti;
    message.gsm.h.message_type = OGS_NAS_5GS_PDU_SESSION_ESTABLISHMENT_REJECT;

    pdu_session_establishment_reject->gsm_cause = gsm_cause;

    return ogs_nas_5gs_plain_encode(&message);
}

ogs_pkbuf_t *gsm_build_pdu_session_modification_command(
        smf_sess_t *sess,
        uint8_t qos_rule_code, uint8_t qos_flow_description_code)
{
    ogs_pkbuf_t *pkbuf = NULL;
    smf_bearer_t *qos_flow = NULL;
    int rv, i, len;

    ogs_nas_5gs_message_t message;
    ogs_nas_5gs_pdu_session_modification_command_t
        *pdu_session_modification_command =
            &message.gsm.pdu_session_modification_command;

    ogs_nas_qos_rules_t *authorized_qos_rules = NULL;
    ogs_nas_qos_flow_descriptions_t *authorized_qos_flow_descriptions = NULL;

    ogs_nas_qos_rule_t qos_rule[OGS_NAS_MAX_NUM_OF_QOS_RULE];
    ogs_nas_qos_flow_description_t
        qos_flow_description[OGS_NAS_MAX_NUM_OF_QOS_FLOW_DESCRIPTION];

    OpenAPI_qos_flow_add_modify_request_item_t *qosFlowAddModRequestItem = NULL;
    OpenAPI_qos_flow_release_request_item_t *qosFlowRelRequestItem = NULL;
    OpenAPI_lnode_t *node = NULL;
    int num;

    authorized_qos_rules = &pdu_session_modification_command->
        authorized_qos_rules;
    ogs_assert(authorized_qos_rules);
    authorized_qos_flow_descriptions =
        &pdu_session_modification_command->authorized_qos_flow_descriptions;
    ogs_assert(authorized_qos_flow_descriptions);

    ogs_assert(sess);

    memset(&message, 0, sizeof(message));
    message.gsm.h.extended_protocol_discriminator =
            OGS_NAS_EXTENDED_PROTOCOL_DISCRIMINATOR_5GSM;
    message.gsm.h.pdu_session_identity = sess->psi;
    message.gsm.h.procedure_transaction_identity = sess->pti;
    message.gsm.h.message_type = OGS_NAS_5GS_PDU_SESSION_MODIFICATION_COMMAND;

    /* QoS rule */
    if (HOME_ROUTED_ROAMING_IN_VSMF(sess)) {
        num = 0;
        OpenAPI_list_for_each(
                sess->h_smf_qos_flows_add_mod_request_list, node) {
            qosFlowAddModRequestItem = node->data;
            if (qosFlowAddModRequestItem &&
                qosFlowAddModRequestItem->qos_rules) {
                ogs_nas_qos_rules_t qos_rules;

                qos_rules.buffer = gsm_decode_base64(
                        qosFlowAddModRequestItem->qos_rules, &len);
                if (!qos_rules.buffer) {
                    ogs_error("Failed to decode qosRules in qosFlowsAddModRequestList");
                    goto cleanup;
                }
                qos_rules.length = len;

                if (1 != ogs_nas_parse_qos_rules(&qos_rule[num],
                            &qos_rules)) {
                    ogs_error("ogs_nas_parse_qos_rules() failed");
                    ogs_free(qos_rules.buffer);
                    goto cleanup;
                }

                ogs_free(qos_rules.buffer);

                num++;
            }
        }

        OpenAPI_list_for_each(sess->h_smf_qos_flows_rel_request_list, node) {
            qosFlowRelRequestItem = node->data;
            if (qosFlowRelRequestItem && qosFlowRelRequestItem->qos_rules) {
                ogs_nas_qos_rules_t qos_rules;

                qos_rules.buffer = gsm_decode_base64(
                        qosFlowRelRequestItem->qos_rules, &len);
                if (!qos_rules.buffer) {
                    ogs_error("Failed to decode qosRules in qosFlowsRelRequestList");
                    goto cleanup;
                }
                qos_rules.length = len;

                if (1 != ogs_nas_parse_qos_rules(&qos_rule[num], &qos_rules)) {
                    ogs_error("ogs_nas_parse_qos_rules() failed");
                    ogs_free(qos_rules.buffer);
                    goto cleanup;
                }

                ogs_free(qos_rules.buffer);

                num++;
            }
        }

        if (num) {
            ogs_assert(OGS_OK ==
                    ogs_nas_build_qos_rules(
                        authorized_qos_rules, qos_rule, num));

            pdu_session_modification_command->presencemask |=
                OGS_NAS_5GS_PDU_SESSION_MODIFICATION_COMMAND_AUTHORIZED_QOS_RULES_PRESENT;
        }
    } else if (HOME_ROUTED_ROAMING_IN_HSMF(sess)) {
        ogs_fatal("This should not be invoked from H-SMF during HR-Roaming");
        ogs_assert_if_reached();
    } else if (qos_rule_code) {

        i = 0;
        memset(qos_rule, 0, sizeof(qos_rule));

        ogs_list_for_each_entry(
                &sess->qos_flow_to_modify_list, qos_flow, to_modify_node) {
            ogs_assert(i < OGS_MAX_NUM_OF_BEARER);
            gsm_encode_qos_rule(&qos_rule[i], qos_flow, qos_rule_code);
            i++;
        }

        rv = ogs_nas_build_qos_rules(authorized_qos_rules, qos_rule, i);
        if (rv != OGS_OK) {
            ogs_error("ogs_nas_build_qos_rules() failed");
            return NULL;
        }
        if (!authorized_qos_rules->length) {
            ogs_error("No length");
            return NULL;
        }

        pdu_session_modification_command->presencemask |=
            OGS_NAS_5GS_PDU_SESSION_MODIFICATION_COMMAND_AUTHORIZED_QOS_RULES_PRESENT;
    }

    /* QoS flow descriptions */
    if (HOME_ROUTED_ROAMING_IN_VSMF(sess)) {
        num = 0;
        OpenAPI_list_for_each(
                sess->h_smf_qos_flows_add_mod_request_list, node) {
            qosFlowAddModRequestItem = node->data;
            if (qosFlowAddModRequestItem &&
                qosFlowAddModRequestItem->qos_flow_description) {
                ogs_nas_qos_flow_descriptions_t qos_flow_descriptions;

                qos_flow_descriptions.buffer = gsm_decode_base64(
                        qosFlowAddModRequestItem->qos_flow_description, &len);
                if (!qos_flow_descriptions.buffer) {
                    ogs_error("Failed to decode qosFlowDescription in qosFlowsAddModRequestList");
                    goto cleanup;
                }
                qos_flow_descriptions.length = len;

                if (1 != ogs_nas_parse_qos_flow_descriptions(
                            &qos_flow_description[num],
                            &qos_flow_descriptions)) {
                    ogs_error("ogs_nas_parse_qos_flow_descriptions() failed");
                    ogs_free(qos_flow_descriptions.buffer);
                    goto cleanup;
                }

                ogs_free(qos_flow_descriptions.buffer);

                num++;
            }
        }

        OpenAPI_list_for_each(
                sess->h_smf_qos_flows_rel_request_list, node) {
            qosFlowRelRequestItem = node->data;
            if (qosFlowRelRequestItem &&
                qosFlowRelRequestItem->qos_flow_description) {
                ogs_nas_qos_flow_descriptions_t qos_flow_descriptions;

                qos_flow_descriptions.buffer = gsm_decode_base64(
                        qosFlowRelRequestItem->qos_flow_description, &len);
                if (!qos_flow_descriptions.buffer) {
                    ogs_error("Failed to decode qosFlowDescription in qosFlowsRelRequestList");
                    goto cleanup;
                }
                qos_flow_descriptions.length = len;

                if (1 != ogs_nas_parse_qos_flow_descriptions(
                            &qos_flow_description[num],
                            &qos_flow_descriptions)) {
                    ogs_error("ogs_nas_parse_qos_flow_descriptions() failed");
                    ogs_free(qos_flow_descriptions.buffer);
                    goto cleanup;
                }

                ogs_free(qos_flow_descriptions.buffer);

                num++;
            }
        }

        if (num) {
            ogs_assert(OGS_OK ==
                    ogs_nas_build_qos_flow_descriptions(
                        authorized_qos_flow_descriptions,
                        qos_flow_description, num));

            pdu_session_modification_command->presencemask |=
                OGS_NAS_5GS_PDU_SESSION_MODIFICATION_COMMAND_AUTHORIZED_QOS_FLOW_DESCRIPTIONS_PRESENT;
        }
    } else if (HOME_ROUTED_ROAMING_IN_HSMF(sess)) {
        ogs_fatal("This should not be invoked from H-SMF during HR-Roaming");
        ogs_assert_if_reached();
    } else if (qos_flow_description_code) {

        i = 0;
        memset(&qos_flow_description, 0, sizeof(qos_flow_description));

        ogs_list_for_each_entry(
                &sess->qos_flow_to_modify_list, qos_flow, to_modify_node) {
            ogs_assert(i < OGS_MAX_NUM_OF_BEARER);

            gsm_encode_qos_flow_description(
                &qos_flow_description[i], qos_flow, qos_flow_description_code);

            i++;
        }

        rv = ogs_nas_build_qos_flow_descriptions(
                authorized_qos_flow_descriptions, qos_flow_description, i);
        if (rv != OGS_OK) {
            ogs_error("ogs_nas_build_qos_flow_descriptions() failed");
            return NULL;
        }
        if (!authorized_qos_flow_descriptions->length) {
            ogs_error("No length");
            return NULL;
        }

        pdu_session_modification_command->presencemask |=
            OGS_NAS_5GS_PDU_SESSION_MODIFICATION_COMMAND_AUTHORIZED_QOS_FLOW_DESCRIPTIONS_PRESENT;
    }

    pkbuf = ogs_nas_5gs_plain_encode(&message);
    ogs_assert(pkbuf);

cleanup:
    if (authorized_qos_rules->buffer)
        ogs_free(authorized_qos_rules->buffer);
    if (authorized_qos_flow_descriptions->buffer)
        ogs_free(authorized_qos_flow_descriptions->buffer);

    return pkbuf;
}

ogs_pkbuf_t *gsm_build_pdu_session_modification_reject(
        smf_sess_t *sess, ogs_nas_5gsm_cause_t gsm_cause)
{
    ogs_nas_5gs_message_t message;
    ogs_nas_5gs_pdu_session_modification_reject_t *
        pdu_session_modification_reject =
            &message.gsm.pdu_session_modification_reject;

    ogs_assert(sess);
    ogs_assert(gsm_cause);

    memset(&message, 0, sizeof(message));
    message.gsm.h.extended_protocol_discriminator =
            OGS_NAS_EXTENDED_PROTOCOL_DISCRIMINATOR_5GSM;
    message.gsm.h.pdu_session_identity = sess->psi;
    message.gsm.h.procedure_transaction_identity = sess->pti;
    message.gsm.h.message_type = OGS_NAS_5GS_PDU_SESSION_MODIFICATION_REJECT;

    pdu_session_modification_reject->gsm_cause = gsm_cause;

    return ogs_nas_5gs_plain_encode(&message);
}

ogs_pkbuf_t *gsm_build_pdu_session_release_command(
        smf_sess_t *sess, ogs_nas_5gsm_cause_t gsm_cause)
{
    ogs_nas_5gs_message_t message;
    ogs_nas_5gs_pdu_session_release_command_t *pdu_session_release_command =
        &message.gsm.pdu_session_release_command;

    ogs_assert(sess);
    ogs_assert(gsm_cause);

    memset(&message, 0, sizeof(message));
    message.gsm.h.extended_protocol_discriminator =
            OGS_NAS_EXTENDED_PROTOCOL_DISCRIMINATOR_5GSM;
    message.gsm.h.pdu_session_identity = sess->psi;
    message.gsm.h.procedure_transaction_identity = sess->pti;
    message.gsm.h.message_type = OGS_NAS_5GS_PDU_SESSION_RELEASE_COMMAND;

    pdu_session_release_command->gsm_cause = gsm_cause;

    return ogs_nas_5gs_plain_encode(&message);
}

ogs_pkbuf_t *gsm_build_pdu_session_release_reject(
        smf_sess_t *sess, ogs_nas_5gsm_cause_t gsm_cause)
{
    ogs_nas_5gs_message_t message;
    ogs_nas_5gs_pdu_session_release_reject_t *
        pdu_session_release_reject = &message.gsm.pdu_session_release_reject;

    ogs_assert(sess);
    ogs_assert(gsm_cause);

    memset(&message, 0, sizeof(message));
    message.gsm.h.extended_protocol_discriminator =
            OGS_NAS_EXTENDED_PROTOCOL_DISCRIMINATOR_5GSM;
    message.gsm.h.pdu_session_identity = sess->psi;
    message.gsm.h.procedure_transaction_identity = sess->pti;
    message.gsm.h.message_type = OGS_NAS_5GS_PDU_SESSION_RELEASE_REJECT;

    pdu_session_release_reject->gsm_cause = gsm_cause;

    return ogs_nas_5gs_plain_encode(&message);
}

void gsm_encode_default_qos_rule(
        ogs_nas_qos_rule_t *qos_rule, smf_bearer_t *qos_flow)
{
    ogs_assert(qos_rule);
    ogs_assert(qos_flow);

    /*
     * TS23.501
     * 5.7.1.3 QoS Rules
     *
     * A default QoS rule is required to be sent to the UE for every PDU
     * Session establishment and it is associated with a QoS Flow. For IP type
     * PDU Session or Ethernet type PDU Session, the default QoS rule is
     * the only QoS rule of a PDU Session which may contain a Packet Filter
     * Set that allows all UL packets, and in this case, the highest
     * precedence value shall be used for the QoS rule.
     *
     * As long as the default QoS rule does not contain a Packet Filter Set or
     * contains a Packet Filter Set that allows all UL packets, Reflective QoS
     * should not be applied for the QoS Flow which the default QoS rule is
     * associated with and the RQA should not be sent for this QoS Flow.
     */
    qos_rule->identifier = qos_flow->qfi; /* Use QFI in Open5GS */
    qos_rule->code = OGS_NAS_QOS_CODE_CREATE_NEW_QOS_RULE;
    qos_rule->DQR_bit = 1;
    qos_rule->num_of_packet_filter = 1;

    qos_rule->pf[0].direction = OGS_NAS_QOS_DIRECTION_BIDIRECTIONAL;
    qos_rule->pf[0].identifier = 1;
    qos_rule->pf[0].content.length = 1;
    qos_rule->pf[0].content.num_of_component = 1;
    qos_rule->pf[0].content.component[0].type = OGS_PACKET_FILTER_MATCH_ALL;

    /*
     * TS23.501
     * 5.7.1.9 Precedence Value
     *
     * The QoS rule precedence value and the PDR precedence value determine
     * the order in which a QoS rule or a PDR, respectively, shall be evaluated.
     * The evaluation of the QoS rules or PDRs is performed in increasing order
     * of their precedence value.
     */
    qos_rule->precedence = 255; /* lowest precedence */
    qos_rule->flow.segregation = 0;
    qos_rule->flow.identifier = qos_flow->qfi;
}

void gsm_encode_default_qos_flow_description(
        ogs_nas_qos_flow_description_t *qos_flow_description,
        smf_bearer_t *qos_flow)
{
    int num_of_param;

    ogs_assert(qos_flow_description);
    ogs_assert(qos_flow);

    qos_flow_description->identifier = qos_flow->qfi;
    qos_flow_description->code = OGS_NAS_CREATE_NEW_QOS_FLOW_DESCRIPTION;
    qos_flow_description->E_bit = 1;

    num_of_param = 0;

    qos_flow_description->param[num_of_param].identifier =
        OGS_NAX_QOS_FLOW_PARAMETER_ID_5QI;
    qos_flow_description->param[num_of_param].len =
        sizeof(qos_flow_description->param[num_of_param].qos_index);
    qos_flow_description->param[num_of_param].qos_index = qos_flow->qos.index;
    num_of_param++;

    qos_flow_description->num_of_parameter = num_of_param;
}

void gsm_encode_qos_rule_packet_filter(
        ogs_nas_qos_rule_t *qos_rule, smf_bearer_t *qos_flow)
{
    int i;
    smf_pf_t *pf = NULL;

    ogs_assert(qos_rule);
    ogs_assert(qos_flow);

    if (qos_rule->code == OGS_NAS_QOS_CODE_MODIFY_EXISTING_QOS_RULE_AND_DELETE_PACKET_FILTERS) {

        for (i = 0; i < qos_flow->num_of_pf_to_delete; i++) {
            qos_rule->pf[i].identifier = qos_flow->pf_to_delete[i];
        }
        qos_rule->num_of_packet_filter = qos_flow->num_of_pf_to_delete;

    } else {

        i = 0;
        ogs_list_for_each_entry(&qos_flow->pf_to_add_list, pf, to_add_node) {
            ogs_assert(i < OGS_MAX_NUM_OF_FLOW_IN_NAS);
            qos_rule->pf[i].direction = pf->direction;
            qos_rule->pf[i].identifier = pf->identifier;

            ogs_pf_content_from_ipfw_rule(
                    pf->direction, &qos_rule->pf[i].content, &pf->ipfw_rule,
                    ogs_global_conf()->parameter.
                    no_ipv4v6_local_addr_in_packet_filter);
            i++;
        }
        qos_rule->num_of_packet_filter = i;

    }
}

void gsm_encode_qos_rule(
        ogs_nas_qos_rule_t *qos_rule,
        smf_bearer_t *qos_flow, uint8_t qos_rule_code)
{
    ogs_pfcp_pdr_t *dl_pdr = NULL;

    ogs_assert(qos_rule);
    ogs_assert(qos_flow);
    ogs_assert(qos_rule_code);

    dl_pdr = qos_flow->dl_pdr;
    ogs_assert(dl_pdr);

    qos_rule->identifier = qos_flow->qfi;
    qos_rule->code = qos_rule_code;

    if (qos_rule_code != OGS_NAS_QOS_CODE_DELETE_EXISTING_QOS_RULE &&
        qos_rule_code != OGS_NAS_QOS_CODE_MODIFY_EXISTING_QOS_RULE_WITHOUT_MODIFYING_PACKET_FILTERS) {
        gsm_encode_qos_rule_packet_filter(qos_rule, qos_flow);
    }
    if (qos_rule_code != OGS_NAS_QOS_CODE_DELETE_EXISTING_QOS_RULE &&
        qos_rule_code != OGS_NAS_QOS_CODE_MODIFY_EXISTING_QOS_RULE_AND_DELETE_PACKET_FILTERS &&
        qos_rule_code != OGS_NAS_QOS_CODE_MODIFY_EXISTING_QOS_RULE_WITHOUT_MODIFYING_PACKET_FILTERS)
    {
        qos_rule->precedence = dl_pdr->precedence;
        qos_rule->flow.segregation = 0;
        qos_rule->flow.identifier = qos_flow->qfi;
    }
}

void gsm_encode_qos_flow_description(
        ogs_nas_qos_flow_description_t *qos_flow_description,
        smf_bearer_t *qos_flow, uint8_t qos_flow_description_code)
{
    int num_of_param = 0;

    ogs_assert(qos_flow_description);
    ogs_assert(qos_flow);
    ogs_assert(qos_flow_description_code);

    qos_flow_description->identifier = qos_flow->qfi;
    qos_flow_description->code = qos_flow_description_code;

    if (qos_flow_description_code != OGS_NAS_DELETE_NEW_QOS_FLOW_DESCRIPTION) {
        qos_flow_description->E_bit = 1;
        qos_flow_description->param[num_of_param].identifier =
            OGS_NAX_QOS_FLOW_PARAMETER_ID_5QI;
        qos_flow_description->param[num_of_param].len =
            sizeof(qos_flow_description->param[num_of_param].qos_index);
        qos_flow_description->param[num_of_param++].qos_index =
            qos_flow->qos.index;

        if (qos_flow->qos.gbr.uplink) {
            qos_flow_description->param[num_of_param].identifier =
                OGS_NAX_QOS_FLOW_PARAMETER_ID_GFBR_UPLINK;
            qos_flow_description->param[num_of_param].len =
                sizeof(qos_flow_description->param[num_of_param].br);
            ogs_nas_bitrate_from_uint64(
                &qos_flow_description->param[num_of_param].br,
                qos_flow->qos.gbr.uplink);
            num_of_param++;
        }
        if (qos_flow->qos.gbr.downlink) {
            qos_flow_description->param[num_of_param].identifier =
                OGS_NAX_QOS_FLOW_PARAMETER_ID_GFBR_DOWNLINK;
            qos_flow_description->param[num_of_param].len =
                sizeof(qos_flow_description->param[num_of_param].br);
            ogs_nas_bitrate_from_uint64(
                &qos_flow_description->param[num_of_param].br,
                qos_flow->qos.gbr.downlink);
            num_of_param++;
        }
        if (qos_flow->qos.mbr.uplink) {
            qos_flow_description->param[num_of_param].identifier =
                OGS_NAX_QOS_FLOW_PARAMETER_ID_MFBR_UPLINK;
            qos_flow_description->param[num_of_param].len =
                sizeof(qos_flow_description->param[num_of_param].br);
            ogs_nas_bitrate_from_uint64(
                &qos_flow_description->param[num_of_param].br,
                qos_flow->qos.mbr.uplink);
            num_of_param++;
        }
        if (qos_flow->qos.mbr.downlink) {
            qos_flow_description->param[num_of_param].identifier =
                OGS_NAX_QOS_FLOW_PARAMETER_ID_MFBR_DOWNLINK;
            qos_flow_description->param[num_of_param].len =
                sizeof(qos_flow_description->param[num_of_param].br);
            ogs_nas_bitrate_from_uint64(
                &qos_flow_description->param[num_of_param].br,
                qos_flow->qos.mbr.downlink);
            num_of_param++;
        }
    }
    qos_flow_description->num_of_parameter = num_of_param;
}
