open5gs/src/sgw/sgw_sm.c

#include "gtp/gtp_node.h"

#include "sgw_context.h"
#include "sgw_event.h"
#include "sgw_sm.h"
#include "sgw_gtp_path.h"
#include "sgw_s11_handler.h"
#include "sgw_s5c_handler.h"

void sgw_state_initial(ogs_fsm_t *s, sgw_event_t *e)
{
    sgw_sm_debug(e);

    ogs_assert(s);

    OGS_FSM_TRAN(s, &sgw_state_operational);
}

void sgw_state_final(ogs_fsm_t *s, sgw_event_t *e)
{
    sgw_sm_debug(e);

    ogs_assert(s);
}

void sgw_state_operational(ogs_fsm_t *s, sgw_event_t *e)
{
    int rv;

    sgw_sm_debug(e);

    ogs_assert(s);

    switch (e->id)
    {
        case OGS_FSM_ENTRY_SIG:
        {
            rv = sgw_gtp_open();
            if (rv != OGS_OK)
            {
                ogs_error("Can't establish SGW path");
                break;
            }
            break;
        }
        case OGS_FSM_EXIT_SIG:
        {
            sgw_gtp_close();
            break;
        }
        case SGW_EVT_S11_MESSAGE:
        {
            int rv;
            ogs_pkbuf_t *pkbuf = NULL;
            gtp_xact_t *xact = NULL;
            gtp_message_t message;
            sgw_ue_t *sgw_ue = NULL;

            ogs_assert(e);
            pkbuf = e->pkbuf;
            ogs_assert(pkbuf);
            rv = gtp_parse_msg(&message, pkbuf);
            ogs_assert(rv == OGS_OK);

            if (message.h.teid == 0)
            {
                gtp_node_t *mme = sgw_mme_add_by_message(&message);
                ogs_assert(mme);
                sgw_ue = sgw_ue_add_by_message(&message);
                SETUP_GTP_NODE(sgw_ue, mme);
            }
            else
            {
                sgw_ue = sgw_ue_find_by_teid(message.h.teid);
            }
            ogs_assert(sgw_ue);

            rv = gtp_xact_receive(sgw_ue->gnode, &message.h, &xact);
            if (rv != OGS_OK)
            {
                ogs_pkbuf_free(pkbuf);
                break;
            }

            switch(message.h.type)
            {
                case GTP_CREATE_SESSION_REQUEST_TYPE:
                    sgw_s11_handle_create_session_request(xact, sgw_ue,
                            &message);
                    break;
                case GTP_MODIFY_BEARER_REQUEST_TYPE:
                    sgw_s11_handle_modify_bearer_request(xact, sgw_ue,
                            &message.modify_bearer_request);
                    break;
                case GTP_DELETE_SESSION_REQUEST_TYPE:
                    sgw_s11_handle_delete_session_request(xact, sgw_ue,
                            &message);
                    break;
                case GTP_CREATE_BEARER_RESPONSE_TYPE:
                    sgw_s11_handle_create_bearer_response(xact, sgw_ue,
                            &message);
                    break;
                case GTP_UPDATE_BEARER_RESPONSE_TYPE:
                    sgw_s11_handle_update_bearer_response(xact, sgw_ue,
                            &message);
                    break;
                case GTP_DELETE_BEARER_RESPONSE_TYPE:
                    sgw_s11_handle_delete_bearer_response(xact, sgw_ue,
                            &message);
                    break;
                case GTP_RELEASE_ACCESS_BEARERS_REQUEST_TYPE:
                    sgw_s11_handle_release_access_bearers_request(xact, sgw_ue,
                        &message.release_access_bearers_request);
                    break;
                case GTP_DOWNLINK_DATA_NOTIFICATION_ACKNOWLEDGE_TYPE:
                    sgw_s11_handle_downlink_data_notification_ack(xact, sgw_ue,
                        &message.downlink_data_notification_acknowledge);
                    break;
                case GTP_CREATE_INDIRECT_DATA_FORWARDING_TUNNEL_REQUEST_TYPE:
                    sgw_s11_handle_create_indirect_data_forwarding_tunnel_request(
                        xact, sgw_ue,
                        &message.
                            create_indirect_data_forwarding_tunnel_request);
                    break;
                case GTP_DELETE_INDIRECT_DATA_FORWARDING_TUNNEL_REQUEST_TYPE:
                    sgw_s11_handle_delete_indirect_data_forwarding_tunnel_request(
                        xact, sgw_ue);
                    break;
                default:
                    ogs_warn("Not implmeneted(type:%d)", message.h.type);
                    break;
            }
            ogs_pkbuf_free(pkbuf);
            break;
        }
        case SGW_EVT_S5C_MESSAGE:
        {
            int rv;
            ogs_pkbuf_t *pkbuf = NULL;
            gtp_xact_t *xact = NULL;
            gtp_message_t message;
            sgw_sess_t *sess = NULL;

            ogs_assert(e);
            pkbuf = e->pkbuf;

            ogs_assert(pkbuf);
            rv = gtp_parse_msg(&message, pkbuf);
            ogs_assert(rv == OGS_OK);

            sess = sgw_sess_find_by_teid(message.h.teid);
            ogs_assert(sess);

            rv = gtp_xact_receive(sess->gnode, &message.h, &xact);
            if (rv != OGS_OK)
            {
                ogs_pkbuf_free(pkbuf);
                break;
            }

            switch(message.h.type)
            {
                case GTP_CREATE_SESSION_RESPONSE_TYPE:
                    sgw_s5c_handle_create_session_response(xact, sess,
                            &message);
                    break;
                case GTP_DELETE_SESSION_RESPONSE_TYPE:
                    sgw_s5c_handle_delete_session_response(xact, sess,
                            &message);
                    break;
                case GTP_CREATE_BEARER_REQUEST_TYPE:
                    sgw_s5c_handle_create_bearer_request(xact, sess,
                            &message);
                    break;
                case GTP_UPDATE_BEARER_REQUEST_TYPE:
                    sgw_s5c_handle_update_bearer_request(xact, sess,
                            &message);
                    break;
                case GTP_DELETE_BEARER_REQUEST_TYPE:
                    sgw_s5c_handle_delete_bearer_request(xact, sess,
                            &message);
                    break;
                default:
                    ogs_warn("Not implmeneted(type:%d)", message.h.type);
                    break;
            }
            ogs_pkbuf_free(pkbuf);
            break;
        }
        case SGW_EVT_LO_DLDATA_NOTI:
        {
            sgw_bearer_t* bearer = NULL;
            ogs_assert(e);

            bearer = e->bearer;
            ogs_assert(bearer);

            sgw_s11_handle_lo_dldata_notification(bearer);

            break;
        }
        default:
        {
            ogs_error("No handler for event %s", sgw_event_get_name(e));
            break;
        }
    }
}