#define TRACE_MODULE _esm_handler

#include "core_debug.h"

#include "nas_message.h"

#include "mme_context.h"
#include "mme_event.h"
#include "esm_build.h"
#include "s1ap_build.h"
#include "s1ap_path.h"
#include "nas_path.h"
#include "mme_s11_build.h"

void esm_handle_pdn_connectivity_request(mme_sess_t *sess, 
        nas_pdn_connectivity_request_t *pdn_connectivity_request)
{
    d_assert(sess, return, "Null param");

    if (pdn_connectivity_request->presencemask &
            NAS_PDN_CONNECTIVITY_REQUEST_PROTOCOL_CONFIGURATION_OPTIONS_PRESENT)
    {
        nas_protocol_configuration_options_t *protocol_configuration_options = 
            &pdn_connectivity_request->protocol_configuration_options;

        NAS_STORE_DATA(&sess->ue_pco, protocol_configuration_options);
    }
}

void esm_handle_information_response(mme_sess_t *sess, 
        nas_esm_information_response_t *esm_information_response)
{
    mme_ue_t *mme_ue = NULL;

    d_assert(sess, return, "Null param");
    mme_ue = sess->mme_ue;
    d_assert(mme_ue, return, "Null param");

    if (esm_information_response->presencemask &
            NAS_ESM_INFORMATION_RESPONSE_ACCESS_POINT_NAME_PRESENT)
    {
        sess->pdn = mme_pdn_find_by_apn(mme_ue, 
                esm_information_response->access_point_name.apn);
        d_assert(sess->pdn, return, "No PDN Context[APN:%s])", 
            esm_information_response->access_point_name.apn);
    }

    if (esm_information_response->presencemask &
            NAS_ESM_INFORMATION_RESPONSE_PROTOCOL_CONFIGURATION_OPTIONS_PRESENT)
    {
        nas_protocol_configuration_options_t *protocol_configuration_options = 
            &esm_information_response->protocol_configuration_options;
        NAS_STORE_DATA(&sess->ue_pco, protocol_configuration_options);
    }
}

void esm_handle_activate_dedicated_bearer_request(mme_bearer_t *bearer)
{
    status_t rv;
    mme_ue_t *mme_ue = NULL;
    enb_ue_t *enb_ue = NULL;
    pkbuf_t *esmbuf = NULL, *s1apbuf = NULL;

    d_assert(bearer, return, "Null param");
    mme_ue = bearer->mme_ue;
    d_assert(mme_ue, return, "Null param");
    enb_ue = mme_ue->enb_ue;
    d_assert(enb_ue, return, "Null param");

    rv = esm_build_activate_dedicated_bearer_context(&esmbuf, bearer);
    d_assert(rv == CORE_OK && esmbuf, return, "esm build error");

    d_trace(3, "[NAS] Activate dedicated bearer context request : "
            "EMM <-- ESM\n");

    rv = s1ap_build_e_rab_setup_request(&s1apbuf, bearer, esmbuf);
    d_assert(rv == CORE_OK && s1apbuf, 
            pkbuf_free(esmbuf); return, "s1ap build error");

    d_assert(nas_send_to_enb(enb_ue, s1apbuf) == CORE_OK,,);
}

void esm_handle_activate_default_bearer_accept(mme_bearer_t *bearer)
{
    status_t rv;
    gtp_header_t h;
    gtp_xact_t *xact = NULL;
    mme_ue_t *mme_ue = NULL;
    mme_sess_t *sess = NULL;
    mme_bearer_t *dedicated_bearer = NULL;
    pkbuf_t *pkbuf = NULL;

    d_assert(bearer, return, "Null param");
    sess = bearer->sess;
    d_assert(sess, return, "Null param");
    mme_ue = sess->mme_ue;
    d_assert(mme_ue, return, "Null param");

    if (MME_HAVE_ENB_S1U_PATH(bearer))
    {
        memset(&h, 0, sizeof(gtp_header_t));
        h.type = GTP_MODIFY_BEARER_REQUEST_TYPE;
        h.teid = mme_ue->sgw_s11_teid;

        rv = mme_s11_build_modify_bearer_request(&pkbuf, h.type, bearer);
        d_assert(rv == CORE_OK, return, "S11 build error");

        xact = gtp_xact_local_create(sess->sgw, &h, pkbuf);
        d_assert(xact, return, "Null param");

        rv = gtp_xact_commit(xact);
        d_assert(rv == CORE_OK, return, "xact_commit error");
    }

    dedicated_bearer = mme_bearer_next(bearer);
    while(dedicated_bearer)
    {
        if (!MME_HAVE_ENB_S1U_PATH(dedicated_bearer))
        {
            enb_ue_t *enb_ue = NULL;
            pkbuf_t *esmbuf = NULL, *s1apbuf = NULL;

            enb_ue = mme_ue->enb_ue;
            d_assert(enb_ue, return, "Null param");

            rv = esm_build_activate_dedicated_bearer_context(
                    &esmbuf, dedicated_bearer);
            d_assert(rv == CORE_OK && esmbuf, return, "esm build error");

            d_trace(3, "[NAS] Activate dedicated bearer context request : "
                    "EMM <-- ESM\n");

            rv = s1ap_build_e_rab_setup_request(
                    &s1apbuf, dedicated_bearer, esmbuf);
            d_assert(rv == CORE_OK && s1apbuf, 
                    pkbuf_free(esmbuf); return, "s1ap build error");

            d_assert(nas_send_to_enb(enb_ue, s1apbuf) == CORE_OK,,);
        }
        dedicated_bearer = mme_bearer_next(dedicated_bearer);
    }
}

void esm_handle_activate_dedicated_bearer_accept(mme_bearer_t *bearer)
{
    status_t rv;
    gtp_header_t h;
    gtp_xact_t *xact = NULL;
    mme_ue_t *mme_ue = NULL;
    mme_sess_t *sess = NULL;
    pkbuf_t *pkbuf = NULL;

    d_assert(bearer, return, "Null param");
    sess = bearer->sess;
    d_assert(sess, return, "Null param");
    mme_ue = sess->mme_ue;
    d_assert(mme_ue, return, "Null param");
    xact = bearer->xact;
    d_assert(xact, return, "Null param");

    if (MME_HAVE_ENB_S1U_PATH(bearer))
    {
        memset(&h, 0, sizeof(gtp_header_t));
        h.type = GTP_CREATE_BEARER_RESPONSE_TYPE;
        h.teid = mme_ue->sgw_s11_teid;

        rv = mme_s11_build_create_bearer_response(&pkbuf, h.type, bearer);
        d_assert(rv == CORE_OK, return, "S11 build error");

        rv = gtp_xact_update_tx(xact, &h, pkbuf);
        d_assert(xact, return, "Null param");

        rv = gtp_xact_commit(xact);
        d_assert(rv == CORE_OK, return, "xact_commit error");
    }
}